blob: 84eee110f42a512459f46fc89170fe0a8324205b [file] [log] [blame]
Guido van Rossum96783941997-05-20 18:21:34 +00001
Guido van Rossumf9fca921996-01-12 00:47:05 +00002/* Complex object implementation */
3
4/* Borrows heavily from floatobject.c */
5
Guido van Rossum96783941997-05-20 18:21:34 +00006/* Submitted by Jim Hugunin */
7
Guido van Rossumf9fca921996-01-12 00:47:05 +00008#ifndef WITHOUT_COMPLEX
9
Guido van Rossumc0b618a1997-05-02 03:12:38 +000010#include "Python.h"
Tim Peters6d6c1a32001-08-02 04:15:00 +000011#include "structmember.h"
Guido van Rossumf9fca921996-01-12 00:47:05 +000012
Tim Peters70695122001-03-11 08:37:29 +000013/* Precisions used by repr() and str(), respectively.
14
15 The repr() precision (17 significant decimal digits) is the minimal number
16 that is guaranteed to have enough precision so that if the number is read
17 back in the exact same binary value is recreated. This is true for IEEE
18 floating point by design, and also happens to work for all other modern
19 hardware.
20
21 The str() precision is chosen so that in most cases, the rounding noise
22 created by various operations is suppressed, while giving plenty of
23 precision for practical use.
24*/
25
26#define PREC_REPR 17
27#define PREC_STR 12
Guido van Rossumf9fca921996-01-12 00:47:05 +000028
29/* elementary operations on complex numbers */
30
Guido van Rossum9e720e31996-07-21 02:31:35 +000031static Py_complex c_1 = {1., 0.};
Guido van Rossumf9fca921996-01-12 00:47:05 +000032
Tim Peters0f336042001-03-18 08:21:57 +000033Py_complex
34c_sum(Py_complex a, Py_complex b)
Guido van Rossumf9fca921996-01-12 00:47:05 +000035{
Guido van Rossum9e720e31996-07-21 02:31:35 +000036 Py_complex r;
Guido van Rossumf9fca921996-01-12 00:47:05 +000037 r.real = a.real + b.real;
38 r.imag = a.imag + b.imag;
39 return r;
40}
41
Tim Peters0f336042001-03-18 08:21:57 +000042Py_complex
43c_diff(Py_complex a, Py_complex b)
Guido van Rossumf9fca921996-01-12 00:47:05 +000044{
Guido van Rossum9e720e31996-07-21 02:31:35 +000045 Py_complex r;
Guido van Rossumf9fca921996-01-12 00:47:05 +000046 r.real = a.real - b.real;
47 r.imag = a.imag - b.imag;
48 return r;
49}
50
Tim Peters0f336042001-03-18 08:21:57 +000051Py_complex
52c_neg(Py_complex a)
Guido van Rossumf9fca921996-01-12 00:47:05 +000053{
Guido van Rossum9e720e31996-07-21 02:31:35 +000054 Py_complex r;
Guido van Rossumf9fca921996-01-12 00:47:05 +000055 r.real = -a.real;
56 r.imag = -a.imag;
57 return r;
58}
59
Tim Peters0f336042001-03-18 08:21:57 +000060Py_complex
61c_prod(Py_complex a, Py_complex b)
Guido van Rossumf9fca921996-01-12 00:47:05 +000062{
Guido van Rossum9e720e31996-07-21 02:31:35 +000063 Py_complex r;
Guido van Rossumf9fca921996-01-12 00:47:05 +000064 r.real = a.real*b.real - a.imag*b.imag;
65 r.imag = a.real*b.imag + a.imag*b.real;
66 return r;
67}
68
Tim Peters0f336042001-03-18 08:21:57 +000069Py_complex
70c_quot(Py_complex a, Py_complex b)
Guido van Rossumf9fca921996-01-12 00:47:05 +000071{
Tim Peters0f336042001-03-18 08:21:57 +000072 /******************************************************************
73 This was the original algorithm. It's grossly prone to spurious
74 overflow and underflow errors. It also merrily divides by 0 despite
75 checking for that(!). The code still serves a doc purpose here, as
76 the algorithm following is a simple by-cases transformation of this
77 one:
78
Guido van Rossum9e720e31996-07-21 02:31:35 +000079 Py_complex r;
Guido van Rossumf9fca921996-01-12 00:47:05 +000080 double d = b.real*b.real + b.imag*b.imag;
81 if (d == 0.)
Guido van Rossum96783941997-05-20 18:21:34 +000082 errno = EDOM;
Guido van Rossumf9fca921996-01-12 00:47:05 +000083 r.real = (a.real*b.real + a.imag*b.imag)/d;
84 r.imag = (a.imag*b.real - a.real*b.imag)/d;
85 return r;
Tim Peters0f336042001-03-18 08:21:57 +000086 ******************************************************************/
87
88 /* This algorithm is better, and is pretty obvious: first divide the
89 * numerators and denominator by whichever of {b.real, b.imag} has
90 * larger magnitude. The earliest reference I found was to CACM
91 * Algorithm 116 (Complex Division, Robert L. Smith, Stanford
92 * University). As usual, though, we're still ignoring all IEEE
93 * endcases.
94 */
95 Py_complex r; /* the result */
96 const double abs_breal = b.real < 0 ? -b.real : b.real;
97 const double abs_bimag = b.imag < 0 ? -b.imag : b.imag;
98
99 if (abs_breal >= abs_bimag) {
100 /* divide tops and bottom by b.real */
101 if (abs_breal == 0.0) {
102 errno = EDOM;
103 r.real = r.imag = 0.0;
104 }
105 else {
106 const double ratio = b.imag / b.real;
107 const double denom = b.real + b.imag * ratio;
108 r.real = (a.real + a.imag * ratio) / denom;
109 r.imag = (a.imag - a.real * ratio) / denom;
110 }
111 }
112 else {
113 /* divide tops and bottom by b.imag */
114 const double ratio = b.real / b.imag;
115 const double denom = b.real * ratio + b.imag;
116 assert(b.imag != 0.0);
117 r.real = (a.real * ratio + a.imag) / denom;
118 r.imag = (a.imag * ratio - a.real) / denom;
119 }
120 return r;
Guido van Rossumf9fca921996-01-12 00:47:05 +0000121}
122
Tim Peters0f336042001-03-18 08:21:57 +0000123Py_complex
124c_pow(Py_complex a, Py_complex b)
Guido van Rossumf9fca921996-01-12 00:47:05 +0000125{
Guido van Rossum9e720e31996-07-21 02:31:35 +0000126 Py_complex r;
Guido van Rossumf9fca921996-01-12 00:47:05 +0000127 double vabs,len,at,phase;
128 if (b.real == 0. && b.imag == 0.) {
129 r.real = 1.;
130 r.imag = 0.;
131 }
132 else if (a.real == 0. && a.imag == 0.) {
133 if (b.imag != 0. || b.real < 0.)
Guido van Rossum96783941997-05-20 18:21:34 +0000134 errno = ERANGE;
Guido van Rossumf9fca921996-01-12 00:47:05 +0000135 r.real = 0.;
136 r.imag = 0.;
137 }
138 else {
139 vabs = hypot(a.real,a.imag);
140 len = pow(vabs,b.real);
141 at = atan2(a.imag, a.real);
142 phase = at*b.real;
143 if (b.imag != 0.0) {
144 len /= exp(at*b.imag);
145 phase += b.imag*log(vabs);
146 }
147 r.real = len*cos(phase);
148 r.imag = len*sin(phase);
149 }
150 return r;
151}
152
Tim Peters0f336042001-03-18 08:21:57 +0000153static Py_complex
154c_powu(Py_complex x, long n)
Guido van Rossumf9fca921996-01-12 00:47:05 +0000155{
Guido van Rossum926518b1996-08-19 19:30:45 +0000156 Py_complex r, p;
Guido van Rossumf9fca921996-01-12 00:47:05 +0000157 long mask = 1;
Guido van Rossum926518b1996-08-19 19:30:45 +0000158 r = c_1;
159 p = x;
Guido van Rossumf9fca921996-01-12 00:47:05 +0000160 while (mask > 0 && n >= mask) {
161 if (n & mask)
162 r = c_prod(r,p);
163 mask <<= 1;
164 p = c_prod(p,p);
165 }
166 return r;
167}
168
Tim Peters0f336042001-03-18 08:21:57 +0000169static Py_complex
170c_powi(Py_complex x, long n)
Guido van Rossumf9fca921996-01-12 00:47:05 +0000171{
Guido van Rossum9e720e31996-07-21 02:31:35 +0000172 Py_complex cn;
Guido van Rossumf9fca921996-01-12 00:47:05 +0000173
174 if (n > 100 || n < -100) {
175 cn.real = (double) n;
176 cn.imag = 0.;
177 return c_pow(x,cn);
178 }
179 else if (n > 0)
180 return c_powu(x,n);
181 else
182 return c_quot(c_1,c_powu(x,-n));
183
184}
185
Tim Peters6d6c1a32001-08-02 04:15:00 +0000186static PyObject *
187complex_subtype_from_c_complex(PyTypeObject *type, Py_complex cval)
188{
189 PyObject *op;
190
191 op = PyType_GenericAlloc(type, 0);
192 if (op != NULL)
193 ((PyComplexObject *)op)->cval = cval;
194 return op;
195}
196
Guido van Rossumf9fca921996-01-12 00:47:05 +0000197PyObject *
Fred Drake4288c802000-07-09 04:36:04 +0000198PyComplex_FromCComplex(Py_complex cval)
Guido van Rossumf9fca921996-01-12 00:47:05 +0000199{
Guido van Rossumb18618d2000-05-03 23:44:39 +0000200 register PyComplexObject *op;
201
202 /* PyObject_New is inlined */
203 op = (PyComplexObject *) PyObject_MALLOC(sizeof(PyComplexObject));
Guido van Rossumf9fca921996-01-12 00:47:05 +0000204 if (op == NULL)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000205 return PyErr_NoMemory();
Guido van Rossumb18618d2000-05-03 23:44:39 +0000206 PyObject_INIT(op, &PyComplex_Type);
Guido van Rossumf9fca921996-01-12 00:47:05 +0000207 op->cval = cval;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000208 return (PyObject *) op;
Guido van Rossumf9fca921996-01-12 00:47:05 +0000209}
210
Tim Peters6d6c1a32001-08-02 04:15:00 +0000211static PyObject *
212complex_subtype_from_doubles(PyTypeObject *type, double real, double imag)
213{
214 Py_complex c;
215 c.real = real;
216 c.imag = imag;
217 return complex_subtype_from_c_complex(type, c);
218}
219
Guido van Rossumf9fca921996-01-12 00:47:05 +0000220PyObject *
Fred Drake4288c802000-07-09 04:36:04 +0000221PyComplex_FromDoubles(double real, double imag)
Guido van Rossum926518b1996-08-19 19:30:45 +0000222{
223 Py_complex c;
224 c.real = real;
225 c.imag = imag;
226 return PyComplex_FromCComplex(c);
Guido van Rossumf9fca921996-01-12 00:47:05 +0000227}
228
229double
Fred Drake4288c802000-07-09 04:36:04 +0000230PyComplex_RealAsDouble(PyObject *op)
Guido van Rossum926518b1996-08-19 19:30:45 +0000231{
232 if (PyComplex_Check(op)) {
233 return ((PyComplexObject *)op)->cval.real;
Fred Drake4288c802000-07-09 04:36:04 +0000234 }
235 else {
Guido van Rossum926518b1996-08-19 19:30:45 +0000236 return PyFloat_AsDouble(op);
237 }
Guido van Rossumf9fca921996-01-12 00:47:05 +0000238}
239
240double
Fred Drake4288c802000-07-09 04:36:04 +0000241PyComplex_ImagAsDouble(PyObject *op)
Guido van Rossum926518b1996-08-19 19:30:45 +0000242{
243 if (PyComplex_Check(op)) {
244 return ((PyComplexObject *)op)->cval.imag;
Fred Drake4288c802000-07-09 04:36:04 +0000245 }
246 else {
Guido van Rossum926518b1996-08-19 19:30:45 +0000247 return 0.0;
248 }
Guido van Rossumf9fca921996-01-12 00:47:05 +0000249}
250
Guido van Rossum9e720e31996-07-21 02:31:35 +0000251Py_complex
Fred Drake4288c802000-07-09 04:36:04 +0000252PyComplex_AsCComplex(PyObject *op)
Guido van Rossum926518b1996-08-19 19:30:45 +0000253{
Guido van Rossum9e720e31996-07-21 02:31:35 +0000254 Py_complex cv;
Guido van Rossumcf3d1081996-01-12 01:21:14 +0000255 if (PyComplex_Check(op)) {
256 return ((PyComplexObject *)op)->cval;
Fred Drake4288c802000-07-09 04:36:04 +0000257 }
258 else {
Guido van Rossumcf3d1081996-01-12 01:21:14 +0000259 cv.real = PyFloat_AsDouble(op);
260 cv.imag = 0.;
261 return cv;
Tim Peters70695122001-03-11 08:37:29 +0000262 }
Guido van Rossumcf3d1081996-01-12 01:21:14 +0000263}
264
Guido van Rossumf9fca921996-01-12 00:47:05 +0000265static void
Fred Drake4288c802000-07-09 04:36:04 +0000266complex_dealloc(PyObject *op)
Guido van Rossumf9fca921996-01-12 00:47:05 +0000267{
Guido van Rossumb18618d2000-05-03 23:44:39 +0000268 PyObject_DEL(op);
Guido van Rossumf9fca921996-01-12 00:47:05 +0000269}
270
271
Guido van Rossum363078a1996-05-24 20:45:01 +0000272static void
Tim Peters70695122001-03-11 08:37:29 +0000273complex_to_buf(char *buf, PyComplexObject *v, int precision)
Guido van Rossumf9fca921996-01-12 00:47:05 +0000274{
275 if (v->cval.real == 0.)
Tim Peters70695122001-03-11 08:37:29 +0000276 sprintf(buf, "%.*gj", precision, v->cval.imag);
Guido van Rossumf9fca921996-01-12 00:47:05 +0000277 else
Tim Peters70695122001-03-11 08:37:29 +0000278 sprintf(buf, "(%.*g%+.*gj)", precision, v->cval.real,
279 precision, v->cval.imag);
Guido van Rossumf9fca921996-01-12 00:47:05 +0000280}
281
282static int
Fred Drake4288c802000-07-09 04:36:04 +0000283complex_print(PyComplexObject *v, FILE *fp, int flags)
Guido van Rossumf9fca921996-01-12 00:47:05 +0000284{
285 char buf[100];
Tim Peters70695122001-03-11 08:37:29 +0000286 complex_to_buf(buf, v,
287 (flags & Py_PRINT_RAW) ? PREC_STR : PREC_REPR);
Guido van Rossumf9fca921996-01-12 00:47:05 +0000288 fputs(buf, fp);
289 return 0;
290}
291
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000292static PyObject *
Fred Drake4288c802000-07-09 04:36:04 +0000293complex_repr(PyComplexObject *v)
Guido van Rossumf9fca921996-01-12 00:47:05 +0000294{
295 char buf[100];
Tim Peters70695122001-03-11 08:37:29 +0000296 complex_to_buf(buf, v, PREC_REPR);
297 return PyString_FromString(buf);
298}
299
300static PyObject *
301complex_str(PyComplexObject *v)
302{
303 char buf[100];
304 complex_to_buf(buf, v, PREC_STR);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000305 return PyString_FromString(buf);
Guido van Rossumf9fca921996-01-12 00:47:05 +0000306}
307
Guido van Rossumf9fca921996-01-12 00:47:05 +0000308static long
Fred Drake4288c802000-07-09 04:36:04 +0000309complex_hash(PyComplexObject *v)
Guido van Rossumf9fca921996-01-12 00:47:05 +0000310{
Tim Peters39dce292000-08-15 03:34:48 +0000311 long hashreal, hashimag, combined;
312 hashreal = _Py_HashDouble(v->cval.real);
313 if (hashreal == -1)
314 return -1;
315 hashimag = _Py_HashDouble(v->cval.imag);
316 if (hashimag == -1)
317 return -1;
318 /* Note: if the imaginary part is 0, hashimag is 0 now,
319 * so the following returns hashreal unchanged. This is
320 * important because numbers of different types that
321 * compare equal must have the same hash value, so that
322 * hash(x + 0*j) must equal hash(x).
323 */
324 combined = hashreal + 1000003 * hashimag;
325 if (combined == -1)
326 combined = -2;
327 return combined;
Guido van Rossumf9fca921996-01-12 00:47:05 +0000328}
329
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000330static PyObject *
Fred Drake4288c802000-07-09 04:36:04 +0000331complex_add(PyComplexObject *v, PyComplexObject *w)
Guido van Rossumf9fca921996-01-12 00:47:05 +0000332{
Guido van Rossum09e6ad01997-02-14 22:54:21 +0000333 Py_complex result;
334 PyFPE_START_PROTECT("complex_add", return 0)
335 result = c_sum(v->cval,w->cval);
Guido van Rossum45b83911997-03-14 04:32:50 +0000336 PyFPE_END_PROTECT(result)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000337 return PyComplex_FromCComplex(result);
Guido van Rossumf9fca921996-01-12 00:47:05 +0000338}
339
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000340static PyObject *
Fred Drake4288c802000-07-09 04:36:04 +0000341complex_sub(PyComplexObject *v, PyComplexObject *w)
Guido van Rossumf9fca921996-01-12 00:47:05 +0000342{
Guido van Rossum09e6ad01997-02-14 22:54:21 +0000343 Py_complex result;
344 PyFPE_START_PROTECT("complex_sub", return 0)
345 result = c_diff(v->cval,w->cval);
Guido van Rossum45b83911997-03-14 04:32:50 +0000346 PyFPE_END_PROTECT(result)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000347 return PyComplex_FromCComplex(result);
Guido van Rossumf9fca921996-01-12 00:47:05 +0000348}
349
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000350static PyObject *
Fred Drake4288c802000-07-09 04:36:04 +0000351complex_mul(PyComplexObject *v, PyComplexObject *w)
Guido van Rossumf9fca921996-01-12 00:47:05 +0000352{
Guido van Rossum09e6ad01997-02-14 22:54:21 +0000353 Py_complex result;
354 PyFPE_START_PROTECT("complex_mul", return 0)
355 result = c_prod(v->cval,w->cval);
Guido van Rossum45b83911997-03-14 04:32:50 +0000356 PyFPE_END_PROTECT(result)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000357 return PyComplex_FromCComplex(result);
Guido van Rossumf9fca921996-01-12 00:47:05 +0000358}
359
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000360static PyObject *
Fred Drake4288c802000-07-09 04:36:04 +0000361complex_div(PyComplexObject *v, PyComplexObject *w)
Guido van Rossumf9fca921996-01-12 00:47:05 +0000362{
Guido van Rossum9e720e31996-07-21 02:31:35 +0000363 Py_complex quot;
Guido van Rossum09e6ad01997-02-14 22:54:21 +0000364 PyFPE_START_PROTECT("complex_div", return 0)
Guido van Rossum96783941997-05-20 18:21:34 +0000365 errno = 0;
Guido van Rossumf9fca921996-01-12 00:47:05 +0000366 quot = c_quot(v->cval,w->cval);
Guido van Rossum45b83911997-03-14 04:32:50 +0000367 PyFPE_END_PROTECT(quot)
Guido van Rossum96783941997-05-20 18:21:34 +0000368 if (errno == EDOM) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000369 PyErr_SetString(PyExc_ZeroDivisionError, "complex division");
Guido van Rossumf9fca921996-01-12 00:47:05 +0000370 return NULL;
371 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000372 return PyComplex_FromCComplex(quot);
Guido van Rossumf9fca921996-01-12 00:47:05 +0000373}
374
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000375static PyObject *
Fred Drake4288c802000-07-09 04:36:04 +0000376complex_remainder(PyComplexObject *v, PyComplexObject *w)
Guido van Rossumee09fc11996-09-11 13:55:55 +0000377{
Guido van Rossum3be12e91996-09-12 20:56:18 +0000378 Py_complex div, mod;
Guido van Rossum96783941997-05-20 18:21:34 +0000379 errno = 0;
Guido van Rossum3be12e91996-09-12 20:56:18 +0000380 div = c_quot(v->cval,w->cval); /* The raw divisor value. */
Guido van Rossum96783941997-05-20 18:21:34 +0000381 if (errno == EDOM) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000382 PyErr_SetString(PyExc_ZeroDivisionError, "complex remainder");
Guido van Rossum3be12e91996-09-12 20:56:18 +0000383 return NULL;
384 }
385 div.real = floor(div.real); /* Use the floor of the real part. */
386 div.imag = 0.0;
387 mod = c_diff(v->cval, c_prod(w->cval, div));
388
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000389 return PyComplex_FromCComplex(mod);
Guido van Rossumee09fc11996-09-11 13:55:55 +0000390}
391
Guido van Rossumee09fc11996-09-11 13:55:55 +0000392
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000393static PyObject *
Fred Drake4288c802000-07-09 04:36:04 +0000394complex_divmod(PyComplexObject *v, PyComplexObject *w)
Guido van Rossum3be12e91996-09-12 20:56:18 +0000395{
396 Py_complex div, mod;
397 PyObject *d, *m, *z;
Guido van Rossum96783941997-05-20 18:21:34 +0000398 errno = 0;
Guido van Rossum3be12e91996-09-12 20:56:18 +0000399 div = c_quot(v->cval,w->cval); /* The raw divisor value. */
Guido van Rossum96783941997-05-20 18:21:34 +0000400 if (errno == EDOM) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000401 PyErr_SetString(PyExc_ZeroDivisionError, "complex divmod()");
Guido van Rossum3be12e91996-09-12 20:56:18 +0000402 return NULL;
403 }
404 div.real = floor(div.real); /* Use the floor of the real part. */
405 div.imag = 0.0;
406 mod = c_diff(v->cval, c_prod(w->cval, div));
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000407 d = PyComplex_FromCComplex(div);
408 m = PyComplex_FromCComplex(mod);
409 z = Py_BuildValue("(OO)", d, m);
Guido van Rossum3be12e91996-09-12 20:56:18 +0000410 Py_XDECREF(d);
411 Py_XDECREF(m);
412 return z;
413}
Guido van Rossumf9fca921996-01-12 00:47:05 +0000414
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000415static PyObject *
Fred Drake4288c802000-07-09 04:36:04 +0000416complex_pow(PyComplexObject *v, PyObject *w, PyComplexObject *z)
Guido van Rossumf9fca921996-01-12 00:47:05 +0000417{
Guido van Rossum9e720e31996-07-21 02:31:35 +0000418 Py_complex p;
419 Py_complex exponent;
Guido van Rossumf9fca921996-01-12 00:47:05 +0000420 long int_exponent;
421
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000422 if ((PyObject *)z!=Py_None) {
423 PyErr_SetString(PyExc_ValueError, "complex modulo");
Guido van Rossumf9fca921996-01-12 00:47:05 +0000424 return NULL;
425 }
Guido van Rossum09e6ad01997-02-14 22:54:21 +0000426 PyFPE_START_PROTECT("complex_pow", return 0)
Guido van Rossum96783941997-05-20 18:21:34 +0000427 errno = 0;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000428 exponent = ((PyComplexObject*)w)->cval;
Guido van Rossumf9fca921996-01-12 00:47:05 +0000429 int_exponent = (long)exponent.real;
430 if (exponent.imag == 0. && exponent.real == int_exponent)
431 p = c_powi(v->cval,int_exponent);
432 else
433 p = c_pow(v->cval,exponent);
434
Guido van Rossum45b83911997-03-14 04:32:50 +0000435 PyFPE_END_PROTECT(p)
Guido van Rossum96783941997-05-20 18:21:34 +0000436 if (errno == ERANGE) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000437 PyErr_SetString(PyExc_ValueError,
438 "0.0 to a negative or complex power");
Guido van Rossumf9fca921996-01-12 00:47:05 +0000439 return NULL;
440 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000441 return PyComplex_FromCComplex(p);
Guido van Rossumf9fca921996-01-12 00:47:05 +0000442}
443
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000444static PyObject *
Guido van Rossum4668b002001-08-08 05:00:18 +0000445complex_int_div(PyComplexObject *v, PyComplexObject *w)
446{
447 PyObject *t, *r;
448
449 t = complex_divmod(v, w);
450 if (t != NULL) {
451 r = PyTuple_GET_ITEM(t, 0);
452 Py_INCREF(r);
453 Py_DECREF(t);
454 return r;
455 }
456 return NULL;
457}
458
459static PyObject *
Fred Drake4288c802000-07-09 04:36:04 +0000460complex_neg(PyComplexObject *v)
Guido van Rossumf9fca921996-01-12 00:47:05 +0000461{
Guido van Rossum9e720e31996-07-21 02:31:35 +0000462 Py_complex neg;
Guido van Rossumf9fca921996-01-12 00:47:05 +0000463 neg.real = -v->cval.real;
464 neg.imag = -v->cval.imag;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000465 return PyComplex_FromCComplex(neg);
Guido van Rossumf9fca921996-01-12 00:47:05 +0000466}
467
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000468static PyObject *
Fred Drake4288c802000-07-09 04:36:04 +0000469complex_pos(PyComplexObject *v)
Guido van Rossumf9fca921996-01-12 00:47:05 +0000470{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000471 Py_INCREF(v);
472 return (PyObject *)v;
Guido van Rossumf9fca921996-01-12 00:47:05 +0000473}
474
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000475static PyObject *
Fred Drake4288c802000-07-09 04:36:04 +0000476complex_abs(PyComplexObject *v)
Guido van Rossumf9fca921996-01-12 00:47:05 +0000477{
Guido van Rossum09e6ad01997-02-14 22:54:21 +0000478 double result;
479 PyFPE_START_PROTECT("complex_abs", return 0)
480 result = hypot(v->cval.real,v->cval.imag);
Guido van Rossum45b83911997-03-14 04:32:50 +0000481 PyFPE_END_PROTECT(result)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000482 return PyFloat_FromDouble(result);
Guido van Rossumf9fca921996-01-12 00:47:05 +0000483}
484
485static int
Fred Drake4288c802000-07-09 04:36:04 +0000486complex_nonzero(PyComplexObject *v)
Guido van Rossumf9fca921996-01-12 00:47:05 +0000487{
Guido van Rossum3bbef601999-01-25 19:42:19 +0000488 return v->cval.real != 0.0 || v->cval.imag != 0.0;
Guido van Rossumf9fca921996-01-12 00:47:05 +0000489}
490
491static int
Fred Drake4288c802000-07-09 04:36:04 +0000492complex_coerce(PyObject **pv, PyObject **pw)
Guido van Rossumf9fca921996-01-12 00:47:05 +0000493{
Guido van Rossum9e720e31996-07-21 02:31:35 +0000494 Py_complex cval;
Guido van Rossumf9fca921996-01-12 00:47:05 +0000495 cval.imag = 0.;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000496 if (PyInt_Check(*pw)) {
497 cval.real = (double)PyInt_AsLong(*pw);
498 *pw = PyComplex_FromCComplex(cval);
499 Py_INCREF(*pv);
Guido van Rossumf9fca921996-01-12 00:47:05 +0000500 return 0;
501 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000502 else if (PyLong_Check(*pw)) {
503 cval.real = PyLong_AsDouble(*pw);
504 *pw = PyComplex_FromCComplex(cval);
505 Py_INCREF(*pv);
Guido van Rossumf9fca921996-01-12 00:47:05 +0000506 return 0;
507 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000508 else if (PyFloat_Check(*pw)) {
509 cval.real = PyFloat_AsDouble(*pw);
510 *pw = PyComplex_FromCComplex(cval);
511 Py_INCREF(*pv);
Guido van Rossumf9fca921996-01-12 00:47:05 +0000512 return 0;
513 }
514 return 1; /* Can't do it */
515}
516
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000517static PyObject *
Guido van Rossumbe4cbb12001-01-18 01:12:39 +0000518complex_richcompare(PyObject *v, PyObject *w, int op)
519{
520 int c;
521 Py_complex i, j;
522 PyObject *res;
523
524 if (op != Py_EQ && op != Py_NE) {
525 PyErr_SetString(PyExc_TypeError,
526 "cannot compare complex numbers using <, <=, >, >=");
527 return NULL;
528 }
529
530 c = PyNumber_CoerceEx(&v, &w);
531 if (c < 0)
532 return NULL;
533 if (c > 0) {
534 Py_INCREF(Py_NotImplemented);
535 return Py_NotImplemented;
536 }
537 if (!PyComplex_Check(v) || !PyComplex_Check(w)) {
538 Py_DECREF(v);
539 Py_DECREF(w);
540 Py_INCREF(Py_NotImplemented);
541 return Py_NotImplemented;
542 }
543
544 i = ((PyComplexObject *)v)->cval;
545 j = ((PyComplexObject *)w)->cval;
546 Py_DECREF(v);
547 Py_DECREF(w);
548
549 if ((i.real == j.real && i.imag == j.imag) == (op == Py_EQ))
550 res = Py_True;
551 else
552 res = Py_False;
553
554 Py_INCREF(res);
555 return res;
556}
557
558static PyObject *
Fred Drake4288c802000-07-09 04:36:04 +0000559complex_int(PyObject *v)
Guido van Rossumf9fca921996-01-12 00:47:05 +0000560{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000561 PyErr_SetString(PyExc_TypeError,
Guido van Rossumd4ab3cd1996-09-11 22:54:37 +0000562 "can't convert complex to int; use e.g. int(abs(z))");
563 return NULL;
Guido van Rossumf9fca921996-01-12 00:47:05 +0000564}
565
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000566static PyObject *
Fred Drake4288c802000-07-09 04:36:04 +0000567complex_long(PyObject *v)
Guido van Rossumf9fca921996-01-12 00:47:05 +0000568{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000569 PyErr_SetString(PyExc_TypeError,
Guido van Rossumd4ab3cd1996-09-11 22:54:37 +0000570 "can't convert complex to long; use e.g. long(abs(z))");
571 return NULL;
Guido van Rossumf9fca921996-01-12 00:47:05 +0000572}
573
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000574static PyObject *
Fred Drake4288c802000-07-09 04:36:04 +0000575complex_float(PyObject *v)
Guido van Rossumf9fca921996-01-12 00:47:05 +0000576{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000577 PyErr_SetString(PyExc_TypeError,
Guido van Rossumd4ab3cd1996-09-11 22:54:37 +0000578 "can't convert complex to float; use e.g. abs(z)");
579 return NULL;
Guido van Rossumf9fca921996-01-12 00:47:05 +0000580}
581
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000582static PyObject *
Martin v. Löwise3eb1f22001-08-16 13:15:00 +0000583complex_conjugate(PyObject *self)
Guido van Rossumf9fca921996-01-12 00:47:05 +0000584{
Guido van Rossum926518b1996-08-19 19:30:45 +0000585 Py_complex c;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000586 c = ((PyComplexObject *)self)->cval;
Guido van Rossumf9fca921996-01-12 00:47:05 +0000587 c.imag = -c.imag;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000588 return PyComplex_FromCComplex(c);
Guido van Rossumf9fca921996-01-12 00:47:05 +0000589}
590
591static PyMethodDef complex_methods[] = {
Martin v. Löwise3eb1f22001-08-16 13:15:00 +0000592 {"conjugate", (PyCFunction)complex_conjugate, METH_NOARGS},
Guido van Rossumf9fca921996-01-12 00:47:05 +0000593 {NULL, NULL} /* sentinel */
594};
595
Tim Peters6d6c1a32001-08-02 04:15:00 +0000596static struct memberlist complex_members[] = {
597 {"real", T_DOUBLE, offsetof(PyComplexObject, cval.real), 0},
598 {"imag", T_DOUBLE, offsetof(PyComplexObject, cval.imag), 0},
599 {0},
600};
Guido van Rossumf9fca921996-01-12 00:47:05 +0000601
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000602static PyObject *
Tim Peters6d6c1a32001-08-02 04:15:00 +0000603complex_subtype_from_string(PyTypeObject *type, PyObject *v)
Guido van Rossumf9fca921996-01-12 00:47:05 +0000604{
Tim Peters6d6c1a32001-08-02 04:15:00 +0000605 extern double strtod(const char *, char **);
606 const char *s, *start;
607 char *end;
608 double x=0.0, y=0.0, z;
609 int got_re=0, got_im=0, done=0;
610 int digit_or_dot;
611 int sw_error=0;
612 int sign;
613 char buffer[256]; /* For errors */
614 char s_buffer[256];
615 int len;
616
617 if (PyString_Check(v)) {
618 s = PyString_AS_STRING(v);
619 len = PyString_GET_SIZE(v);
620 }
621 else if (PyUnicode_Check(v)) {
622 if (PyUnicode_GET_SIZE(v) >= sizeof(s_buffer)) {
623 PyErr_SetString(PyExc_ValueError,
624 "complex() literal too large to convert");
625 return NULL;
626 }
627 if (PyUnicode_EncodeDecimal(PyUnicode_AS_UNICODE(v),
628 PyUnicode_GET_SIZE(v),
629 s_buffer,
630 NULL))
631 return NULL;
632 s = s_buffer;
633 len = (int)strlen(s);
634 }
635 else if (PyObject_AsCharBuffer(v, &s, &len)) {
636 PyErr_SetString(PyExc_TypeError,
637 "complex() arg is not a string");
638 return NULL;
639 }
640
641 /* position on first nonblank */
642 start = s;
643 while (*s && isspace(Py_CHARMASK(*s)))
644 s++;
645 if (s[0] == '\0') {
646 PyErr_SetString(PyExc_ValueError,
647 "complex() arg is an empty string");
648 return NULL;
649 }
650
651 z = -1.0;
652 sign = 1;
653 do {
654
655 switch (*s) {
656
657 case '\0':
658 if (s-start != len) {
659 PyErr_SetString(
660 PyExc_ValueError,
661 "complex() arg contains a null byte");
662 return NULL;
663 }
664 if(!done) sw_error=1;
665 break;
666
667 case '-':
668 sign = -1;
669 /* Fallthrough */
670 case '+':
671 if (done) sw_error=1;
672 s++;
673 if ( *s=='\0'||*s=='+'||*s=='-' ||
674 isspace(Py_CHARMASK(*s)) ) sw_error=1;
675 break;
676
677 case 'J':
678 case 'j':
679 if (got_im || done) {
680 sw_error = 1;
681 break;
682 }
683 if (z<0.0) {
684 y=sign;
685 }
686 else{
687 y=sign*z;
688 }
689 got_im=1;
690 s++;
691 if (*s!='+' && *s!='-' )
692 done=1;
693 break;
694
695 default:
696 if (isspace(Py_CHARMASK(*s))) {
697 while (*s && isspace(Py_CHARMASK(*s)))
698 s++;
699 if (s[0] != '\0')
700 sw_error=1;
701 else
702 done = 1;
703 break;
704 }
705 digit_or_dot =
706 (*s=='.' || isdigit(Py_CHARMASK(*s)));
707 if (done||!digit_or_dot) {
708 sw_error=1;
709 break;
710 }
711 errno = 0;
712 PyFPE_START_PROTECT("strtod", return 0)
713 z = strtod(s, &end) ;
714 PyFPE_END_PROTECT(z)
715 if (errno != 0) {
716 sprintf(buffer,
717 "float() out of range: %.150s", s);
718 PyErr_SetString(
719 PyExc_ValueError,
720 buffer);
721 return NULL;
722 }
723 s=end;
724 if (*s=='J' || *s=='j') {
725
726 break;
727 }
728 if (got_re) {
729 sw_error=1;
730 break;
731 }
732
733 /* accept a real part */
734 x=sign*z;
735 got_re=1;
736 if (got_im) done=1;
737 z = -1.0;
738 sign = 1;
739 break;
740
741 } /* end of switch */
742
743 } while (*s!='\0' && !sw_error);
744
745 if (sw_error) {
746 PyErr_SetString(PyExc_ValueError,
747 "complex() arg is a malformed string");
748 return NULL;
749 }
750
751 return complex_subtype_from_doubles(type, x, y);
Guido van Rossumf9fca921996-01-12 00:47:05 +0000752}
753
Tim Peters6d6c1a32001-08-02 04:15:00 +0000754static PyObject *
755complex_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
756{
757 PyObject *r, *i, *tmp;
758 PyNumberMethods *nbr, *nbi = NULL;
759 Py_complex cr, ci;
760 int own_r = 0;
761 static char *kwlist[] = {"real", "imag", 0};
762
763 r = Py_False;
764 i = NULL;
765 if (!PyArg_ParseTupleAndKeywords(args, kwds, "|OO:complex", kwlist,
766 &r, &i))
767 return NULL;
768 if (PyString_Check(r) || PyUnicode_Check(r))
769 return complex_subtype_from_string(type, r);
770 if ((nbr = r->ob_type->tp_as_number) == NULL ||
771 nbr->nb_float == NULL ||
772 (i != NULL &&
773 ((nbi = i->ob_type->tp_as_number) == NULL ||
774 nbi->nb_float == NULL))) {
775 PyErr_SetString(PyExc_TypeError,
776 "complex() arg can't be converted to complex");
777 return NULL;
778 }
779 /* XXX Hack to support classes with __complex__ method */
780 if (PyInstance_Check(r)) {
781 static PyObject *complexstr;
782 PyObject *f;
783 if (complexstr == NULL) {
784 complexstr = PyString_InternFromString("__complex__");
785 if (complexstr == NULL)
786 return NULL;
787 }
788 f = PyObject_GetAttr(r, complexstr);
789 if (f == NULL)
790 PyErr_Clear();
791 else {
792 PyObject *args = Py_BuildValue("()");
793 if (args == NULL)
794 return NULL;
795 r = PyEval_CallObject(f, args);
796 Py_DECREF(args);
797 Py_DECREF(f);
798 if (r == NULL)
799 return NULL;
800 own_r = 1;
801 }
802 }
803 if (PyComplex_Check(r)) {
804 cr = ((PyComplexObject*)r)->cval;
805 if (own_r) {
806 Py_DECREF(r);
807 }
808 }
809 else {
810 tmp = PyNumber_Float(r);
811 if (own_r) {
812 Py_DECREF(r);
813 }
814 if (tmp == NULL)
815 return NULL;
816 if (!PyFloat_Check(tmp)) {
817 PyErr_SetString(PyExc_TypeError,
818 "float(r) didn't return a float");
819 Py_DECREF(tmp);
820 return NULL;
821 }
822 cr.real = PyFloat_AsDouble(tmp);
823 Py_DECREF(tmp);
824 cr.imag = 0.0;
825 }
826 if (i == NULL) {
827 ci.real = 0.0;
828 ci.imag = 0.0;
829 }
830 else if (PyComplex_Check(i))
831 ci = ((PyComplexObject*)i)->cval;
832 else {
833 tmp = (*nbi->nb_float)(i);
834 if (tmp == NULL)
835 return NULL;
836 ci.real = PyFloat_AsDouble(tmp);
837 Py_DECREF(tmp);
838 ci.imag = 0.;
839 }
840 cr.real -= ci.imag;
841 cr.imag += ci.real;
842 return complex_subtype_from_c_complex(type, cr);
843}
844
845static char complex_doc[] =
846"complex(real[, imag]) -> complex number\n\
847\n\
848Create a complex number from a real part and an optional imaginary part.\n\
849This is equivalent to (real + imag*1j) where imag defaults to 0.";
850
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000851static PyNumberMethods complex_as_number = {
Guido van Rossumbe4cbb12001-01-18 01:12:39 +0000852 (binaryfunc)complex_add, /* nb_add */
853 (binaryfunc)complex_sub, /* nb_subtract */
854 (binaryfunc)complex_mul, /* nb_multiply */
855 (binaryfunc)complex_div, /* nb_divide */
856 (binaryfunc)complex_remainder, /* nb_remainder */
857 (binaryfunc)complex_divmod, /* nb_divmod */
858 (ternaryfunc)complex_pow, /* nb_power */
859 (unaryfunc)complex_neg, /* nb_negative */
860 (unaryfunc)complex_pos, /* nb_positive */
861 (unaryfunc)complex_abs, /* nb_absolute */
862 (inquiry)complex_nonzero, /* nb_nonzero */
863 0, /* nb_invert */
864 0, /* nb_lshift */
865 0, /* nb_rshift */
866 0, /* nb_and */
867 0, /* nb_xor */
868 0, /* nb_or */
869 (coercion)complex_coerce, /* nb_coerce */
870 (unaryfunc)complex_int, /* nb_int */
871 (unaryfunc)complex_long, /* nb_long */
872 (unaryfunc)complex_float, /* nb_float */
873 0, /* nb_oct */
874 0, /* nb_hex */
Guido van Rossum4668b002001-08-08 05:00:18 +0000875 0, /* nb_inplace_add */
876 0, /* nb_inplace_subtract */
877 0, /* nb_inplace_multiply*/
878 0, /* nb_inplace_divide */
879 0, /* nb_inplace_remainder */
880 0, /* nb_inplace_power */
881 0, /* nb_inplace_lshift */
882 0, /* nb_inplace_rshift */
883 0, /* nb_inplace_and */
884 0, /* nb_inplace_xor */
885 0, /* nb_inplace_or */
886 (binaryfunc)complex_int_div, /* nb_floor_divide */
887 (binaryfunc)complex_div, /* nb_true_divide */
888 0, /* nb_inplace_floor_divide */
889 0, /* nb_inplace_true_divide */
Guido van Rossumf9fca921996-01-12 00:47:05 +0000890};
891
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000892PyTypeObject PyComplex_Type = {
893 PyObject_HEAD_INIT(&PyType_Type)
Guido van Rossumf9fca921996-01-12 00:47:05 +0000894 0,
895 "complex",
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000896 sizeof(PyComplexObject),
Guido van Rossumf9fca921996-01-12 00:47:05 +0000897 0,
Guido van Rossumbe4cbb12001-01-18 01:12:39 +0000898 (destructor)complex_dealloc, /* tp_dealloc */
899 (printfunc)complex_print, /* tp_print */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000900 0, /* tp_getattr */
Guido van Rossumbe4cbb12001-01-18 01:12:39 +0000901 0, /* tp_setattr */
902 0, /* tp_compare */
903 (reprfunc)complex_repr, /* tp_repr */
904 &complex_as_number, /* tp_as_number */
905 0, /* tp_as_sequence */
906 0, /* tp_as_mapping */
907 (hashfunc)complex_hash, /* tp_hash */
908 0, /* tp_call */
Tim Peters70695122001-03-11 08:37:29 +0000909 (reprfunc)complex_str, /* tp_str */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000910 PyObject_GenericGetAttr, /* tp_getattro */
Guido van Rossumbe4cbb12001-01-18 01:12:39 +0000911 0, /* tp_setattro */
912 0, /* tp_as_buffer */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000913 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
914 complex_doc, /* tp_doc */
Guido van Rossumbe4cbb12001-01-18 01:12:39 +0000915 0, /* tp_traverse */
916 0, /* tp_clear */
917 complex_richcompare, /* tp_richcompare */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000918 0, /* tp_weaklistoffset */
919 0, /* tp_iter */
920 0, /* tp_iternext */
921 complex_methods, /* tp_methods */
922 complex_members, /* tp_members */
923 0, /* tp_getset */
924 0, /* tp_base */
925 0, /* tp_dict */
926 0, /* tp_descr_get */
927 0, /* tp_descr_set */
928 0, /* tp_dictoffset */
929 0, /* tp_init */
930 0, /* tp_alloc */
931 complex_new, /* tp_new */
Guido van Rossumf9fca921996-01-12 00:47:05 +0000932};
933
934#endif