1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
|
/* mpc_get_dc, mpc_get_ldc -- Transform mpc number into C complex number
mpc_get_str -- Convert a complex number into a string.
Copyright (C) 2009, 2010, 2011 INRIA
This file is part of GNU MPC.
GNU MPC is free software; you can redistribute it and/or modify it under
the terms of the GNU Lesser General Public License as published by the
Free Software Foundation; either version 3 of the License, or (at your
option) any later version.
GNU MPC is distributed in the hope that it will be useful, but WITHOUT ANY
WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for
more details.
You should have received a copy of the GNU Lesser General Public License
along with this program. If not, see http://www.gnu.org/licenses/ .
*/
#include "config.h"
#ifdef HAVE_COMPLEX_H
#include <complex.h>
#endif
#ifdef HAVE_LOCALE_H
#include <locale.h>
#endif
#include <stdio.h> /* for sprintf, fprintf */
#include <ctype.h>
#include <string.h>
#include "mpc-impl.h"
#ifdef HAVE_COMPLEX_H
double _Complex
mpc_get_dc (mpc_srcptr op, mpc_rnd_t rnd) {
return I * mpfr_get_d (mpc_imagref (op), MPC_RND_IM (rnd))
+ mpfr_get_d (mpc_realref (op), MPC_RND_RE (rnd));
}
long double _Complex
mpc_get_ldc (mpc_srcptr op, mpc_rnd_t rnd) {
return I * mpfr_get_ld (mpc_imagref (op), MPC_RND_IM (rnd))
+ mpfr_get_ld (mpc_realref (op), MPC_RND_RE (rnd));
}
#endif
/* Code for mpc_get_str. The output format is "(real imag)", the decimal point
of the locale is used. */
/* mpfr_prec_t can be either int or long int */
#if (__GMP_MP_SIZE_T_INT == 1)
#define MPC_EXP_FORMAT_SPEC "i"
#elif (__GMP_MP_SIZE_T_INT == 0)
#define MPC_EXP_FORMAT_SPEC "li"
#else
#error "mpfr_exp_t size not supported"
#endif
static char *
pretty_zero (mpfr_srcptr zero)
{
char *pretty;
pretty = mpc_alloc_str (3);
pretty[0] = mpfr_signbit (zero) ? '-' : '+';
pretty[1] = '0';
pretty[2] = '\0';
return pretty;
}
static char *
prettify (const char *str, const mp_exp_t expo, int base, int special)
{
size_t sz;
char *pretty;
char *p;
const char *s;
mp_exp_t x;
int sign;
sz = strlen (str) + 1; /* + terminal '\0' */
if (special)
{
/* special number: nan or inf */
pretty = mpc_alloc_str (sz);
strcpy (pretty, str);
return pretty;
}
/* regular number */
sign = (str[0] == '-' || str[0] == '+');
x = expo - 1; /* expo is the exponent value with decimal point BEFORE
the first digit, we wants decimal point AFTER the first
digit */
if (base == 16)
x <<= 2; /* the output exponent is a binary exponent */
++sz; /* + decimal point */
if (x != 0)
{
/* augment sz with the size needed for an exponent written in base
ten */
mp_exp_t xx;
sz += 3; /* + exponent char + sign + 1 digit */
if (x < 0)
{
/* avoid overflow when changing sign (assuming that, for the
mp_exp_t type, (max value) is greater than (- min value / 10)) */
if (x < -10)
{
xx = - (x / 10);
sz++;
}
else
xx = -x;
}
else
xx = x;
/* compute sz += floor(log(expo)/log(10)) without using libm
functions */
while (xx > 9)
{
sz++;
xx /= 10;
}
}
pretty = mpc_alloc_str (sz);
p = pretty;
/* 1. optional sign plus first digit */
s = str;
*p++ = *s++;
if (sign)
*p++ = *s++;
/* 2. decimal point */
#ifdef HAVE_LOCALECONV
*p++ = *localeconv ()->decimal_point;
#else
*p++ = '.';
#endif
*p = '\0';
/* 3. other significant digits */
strcat (pretty, s);
/* 4. exponent (in base ten) */
if (x == 0)
return pretty;
p = pretty + strlen (str) + 1;
switch (base)
{
case 10:
*p++ = 'e';
break;
case 2:
case 16:
*p++ = 'p';
break;
default:
*p++ = '@';
}
*p = '\0';
sprintf (p, "%+"MPC_EXP_FORMAT_SPEC, x);
return pretty;
}
static char *
get_pretty_str (const int base, const size_t n, mpfr_srcptr x, mpfr_rnd_t rnd)
{
mp_exp_t expo;
char *ugly;
char *pretty;
if (mpfr_zero_p (x))
return pretty_zero (x);
ugly = mpfr_get_str (NULL, &expo, base, n, x, rnd);
MPC_ASSERT (ugly != NULL);
pretty = prettify (ugly, expo, base, !mpfr_number_p (x));
mpfr_free_str (ugly);
return pretty;
}
char *
mpc_get_str (int base, size_t n, mpc_srcptr op, mpc_rnd_t rnd)
{
size_t needed_size;
char *real_str;
char *imag_str;
char *complex_str = NULL;
if (base < 2 || base > 36)
return NULL;
real_str = get_pretty_str (base, n, mpc_realref (op), MPC_RND_RE (rnd));
imag_str = get_pretty_str (base, n, mpc_imagref (op), MPC_RND_IM (rnd));
needed_size = strlen (real_str) + strlen (imag_str) + 4;
complex_str = mpc_alloc_str (needed_size);
MPC_ASSERT (complex_str != NULL);
strcpy (complex_str, "(");
strcat (complex_str, real_str);
strcat (complex_str, " ");
strcat (complex_str, imag_str);
strcat (complex_str, ")");
mpc_free_str (real_str);
mpc_free_str (imag_str);
return complex_str;
}
|
