blob: ca94750f02714003792c85b1725e55c97eb2fca8 [file] [log] [blame]
Guido van Rossumf70e43a1991-02-19 12:39:46 +00001
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002/* Float object implementation */
3
Guido van Rossum2a9096b1990-10-21 22:15:08 +00004/* XXX There should be overflow checks here, but it's hard to check
5 for any kind of float exception without losing portability. */
6
Guido van Rossumc0b618a1997-05-02 03:12:38 +00007#include "Python.h"
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00008
Eric Smith8c663262007-08-25 02:26:07 +00009#include "formatter_unicode.h"
10
Guido van Rossum3f5da241990-12-20 15:06:42 +000011#include <ctype.h>
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000012
Jack Janseneddc1442003-11-20 01:44:59 +000013#if !defined(__STDC__)
Tim Petersdbd9ba62000-07-09 03:09:57 +000014extern double fmod(double, double);
15extern double pow(double, double);
Guido van Rossum6923e131990-11-02 17:50:43 +000016#endif
17
Guido van Rossum93ad0df1997-05-13 21:00:42 +000018/* Special free list -- see comments for same code in intobject.c. */
Guido van Rossum93ad0df1997-05-13 21:00:42 +000019#define BLOCK_SIZE 1000 /* 1K less typical malloc overhead */
Guido van Rossum3fce8831999-03-12 19:43:17 +000020#define BHEAD_SIZE 8 /* Enough for a 64-bit pointer */
Guido van Rossumf61bbc81999-03-12 00:12:21 +000021#define N_FLOATOBJECTS ((BLOCK_SIZE - BHEAD_SIZE) / sizeof(PyFloatObject))
Guido van Rossum3fce8831999-03-12 19:43:17 +000022
Guido van Rossum3fce8831999-03-12 19:43:17 +000023struct _floatblock {
24 struct _floatblock *next;
25 PyFloatObject objects[N_FLOATOBJECTS];
26};
27
28typedef struct _floatblock PyFloatBlock;
29
30static PyFloatBlock *block_list = NULL;
31static PyFloatObject *free_list = NULL;
32
Guido van Rossum93ad0df1997-05-13 21:00:42 +000033static PyFloatObject *
Fred Drakefd99de62000-07-09 05:02:18 +000034fill_free_list(void)
Guido van Rossum93ad0df1997-05-13 21:00:42 +000035{
36 PyFloatObject *p, *q;
Guido van Rossumb18618d2000-05-03 23:44:39 +000037 /* XXX Float blocks escape the object heap. Use PyObject_MALLOC ??? */
38 p = (PyFloatObject *) PyMem_MALLOC(sizeof(PyFloatBlock));
Guido van Rossum93ad0df1997-05-13 21:00:42 +000039 if (p == NULL)
Guido van Rossumb18618d2000-05-03 23:44:39 +000040 return (PyFloatObject *) PyErr_NoMemory();
Guido van Rossum3fce8831999-03-12 19:43:17 +000041 ((PyFloatBlock *)p)->next = block_list;
42 block_list = (PyFloatBlock *)p;
43 p = &((PyFloatBlock *)p)->objects[0];
Guido van Rossum93ad0df1997-05-13 21:00:42 +000044 q = p + N_FLOATOBJECTS;
45 while (--q > p)
Martin v. Löwis9f2e3462007-07-21 17:22:18 +000046 Py_Type(q) = (struct _typeobject *)(q-1);
47 Py_Type(q) = NULL;
Guido van Rossum93ad0df1997-05-13 21:00:42 +000048 return p + N_FLOATOBJECTS - 1;
49}
50
Guido van Rossumc0b618a1997-05-02 03:12:38 +000051PyObject *
Guido van Rossumc0b618a1997-05-02 03:12:38 +000052PyFloat_FromDouble(double fval)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000053{
Guido van Rossum93ad0df1997-05-13 21:00:42 +000054 register PyFloatObject *op;
55 if (free_list == NULL) {
56 if ((free_list = fill_free_list()) == NULL)
57 return NULL;
58 }
Guido van Rossume3a8e7e2002-08-19 19:26:42 +000059 /* Inline PyObject_New */
Guido van Rossum93ad0df1997-05-13 21:00:42 +000060 op = free_list;
Martin v. Löwis9f2e3462007-07-21 17:22:18 +000061 free_list = (PyFloatObject *)Py_Type(op);
Guido van Rossumb18618d2000-05-03 23:44:39 +000062 PyObject_INIT(op, &PyFloat_Type);
Guido van Rossum2a9096b1990-10-21 22:15:08 +000063 op->ob_fval = fval;
Guido van Rossumc0b618a1997-05-02 03:12:38 +000064 return (PyObject *) op;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000065}
66
Barry Warsaw226ae6c1999-10-12 19:54:53 +000067PyObject *
Georg Brandl428f0642007-03-18 18:35:15 +000068PyFloat_FromString(PyObject *v)
Barry Warsaw226ae6c1999-10-12 19:54:53 +000069{
Guido van Rossum4c08d552000-03-10 22:55:18 +000070 const char *s, *last, *end;
Barry Warsaw226ae6c1999-10-12 19:54:53 +000071 double x;
Tim Petersef14d732000-09-23 03:39:17 +000072 char buffer[256]; /* for errors */
Guido van Rossum2be161d2007-05-15 20:43:51 +000073 char *s_buffer = NULL;
Martin v. Löwis18e16552006-02-15 17:27:45 +000074 Py_ssize_t len;
Guido van Rossum2be161d2007-05-15 20:43:51 +000075 PyObject *result = NULL;
Barry Warsaw226ae6c1999-10-12 19:54:53 +000076
Guido van Rossum4c08d552000-03-10 22:55:18 +000077 if (PyString_Check(v)) {
78 s = PyString_AS_STRING(v);
79 len = PyString_GET_SIZE(v);
80 }
Guido van Rossum9e896b32000-04-05 20:11:21 +000081 else if (PyUnicode_Check(v)) {
Guido van Rossum2be161d2007-05-15 20:43:51 +000082 s_buffer = (char *)PyMem_MALLOC(PyUnicode_GET_SIZE(v)+1);
83 if (s_buffer == NULL)
84 return PyErr_NoMemory();
Tim Petersef14d732000-09-23 03:39:17 +000085 if (PyUnicode_EncodeDecimal(PyUnicode_AS_UNICODE(v),
Guido van Rossum9e896b32000-04-05 20:11:21 +000086 PyUnicode_GET_SIZE(v),
Tim Petersd2364e82001-11-01 20:09:42 +000087 s_buffer,
Guido van Rossum9e896b32000-04-05 20:11:21 +000088 NULL))
Neal Norwitz447e7c32007-08-12 07:11:25 +000089 goto error;
Guido van Rossum9e896b32000-04-05 20:11:21 +000090 s = s_buffer;
Martin v. Löwis18e16552006-02-15 17:27:45 +000091 len = strlen(s);
Guido van Rossum9e896b32000-04-05 20:11:21 +000092 }
Guido van Rossum4c08d552000-03-10 22:55:18 +000093 else if (PyObject_AsCharBuffer(v, &s, &len)) {
94 PyErr_SetString(PyExc_TypeError,
Skip Montanaro71390a92002-05-02 13:03:22 +000095 "float() argument must be a string or a number");
Barry Warsaw226ae6c1999-10-12 19:54:53 +000096 return NULL;
Guido van Rossum4c08d552000-03-10 22:55:18 +000097 }
Barry Warsaw226ae6c1999-10-12 19:54:53 +000098
Guido van Rossum4c08d552000-03-10 22:55:18 +000099 last = s + len;
Barry Warsaw226ae6c1999-10-12 19:54:53 +0000100 while (*s && isspace(Py_CHARMASK(*s)))
101 s++;
Tim Petersef14d732000-09-23 03:39:17 +0000102 if (*s == '\0') {
Barry Warsaw226ae6c1999-10-12 19:54:53 +0000103 PyErr_SetString(PyExc_ValueError, "empty string for float()");
Guido van Rossum2be161d2007-05-15 20:43:51 +0000104 goto error;
Barry Warsaw226ae6c1999-10-12 19:54:53 +0000105 }
Tim Petersef14d732000-09-23 03:39:17 +0000106 /* We don't care about overflow or underflow. If the platform supports
107 * them, infinities and signed zeroes (on underflow) are fine.
108 * However, strtod can return 0 for denormalized numbers, where atof
109 * does not. So (alas!) we special-case a zero result. Note that
110 * whether strtod sets errno on underflow is not defined, so we can't
111 * key off errno.
112 */
Guido van Rossum2be161d2007-05-15 20:43:51 +0000113 PyFPE_START_PROTECT("strtod", goto error)
Martin v. Löwis737ea822004-06-08 18:52:54 +0000114 x = PyOS_ascii_strtod(s, (char **)&end);
Tim Peters858346e2000-09-25 21:01:28 +0000115 PyFPE_END_PROTECT(x)
Barry Warsaw226ae6c1999-10-12 19:54:53 +0000116 errno = 0;
Barry Warsaw226ae6c1999-10-12 19:54:53 +0000117 /* Believe it or not, Solaris 2.6 can move end *beyond* the null
Tim Petersef14d732000-09-23 03:39:17 +0000118 byte at the end of the string, when the input is inf(inity). */
Barry Warsaw226ae6c1999-10-12 19:54:53 +0000119 if (end > last)
120 end = last;
Tim Petersef14d732000-09-23 03:39:17 +0000121 if (end == s) {
Barry Warsawaf8aef92001-11-28 20:52:21 +0000122 PyOS_snprintf(buffer, sizeof(buffer),
123 "invalid literal for float(): %.200s", s);
Tim Petersef14d732000-09-23 03:39:17 +0000124 PyErr_SetString(PyExc_ValueError, buffer);
Guido van Rossum2be161d2007-05-15 20:43:51 +0000125 goto error;
Tim Petersef14d732000-09-23 03:39:17 +0000126 }
127 /* Since end != s, the platform made *some* kind of sense out
128 of the input. Trust it. */
Barry Warsaw226ae6c1999-10-12 19:54:53 +0000129 while (*end && isspace(Py_CHARMASK(*end)))
130 end++;
131 if (*end != '\0') {
Barry Warsawaf8aef92001-11-28 20:52:21 +0000132 PyOS_snprintf(buffer, sizeof(buffer),
133 "invalid literal for float(): %.200s", s);
Barry Warsaw226ae6c1999-10-12 19:54:53 +0000134 PyErr_SetString(PyExc_ValueError, buffer);
Guido van Rossum2be161d2007-05-15 20:43:51 +0000135 goto error;
Barry Warsaw226ae6c1999-10-12 19:54:53 +0000136 }
Guido van Rossum4c08d552000-03-10 22:55:18 +0000137 else if (end != last) {
Barry Warsaw226ae6c1999-10-12 19:54:53 +0000138 PyErr_SetString(PyExc_ValueError,
139 "null byte in argument for float()");
Guido van Rossum2be161d2007-05-15 20:43:51 +0000140 goto error;
Barry Warsaw226ae6c1999-10-12 19:54:53 +0000141 }
Tim Petersef14d732000-09-23 03:39:17 +0000142 if (x == 0.0) {
143 /* See above -- may have been strtod being anal
144 about denorms. */
Guido van Rossum2be161d2007-05-15 20:43:51 +0000145 PyFPE_START_PROTECT("atof", goto error)
Martin v. Löwis737ea822004-06-08 18:52:54 +0000146 x = PyOS_ascii_atof(s);
Tim Peters858346e2000-09-25 21:01:28 +0000147 PyFPE_END_PROTECT(x)
Tim Petersef14d732000-09-23 03:39:17 +0000148 errno = 0; /* whether atof ever set errno is undefined */
Barry Warsaw226ae6c1999-10-12 19:54:53 +0000149 }
Guido van Rossum2be161d2007-05-15 20:43:51 +0000150 result = PyFloat_FromDouble(x);
151 error:
152 if (s_buffer)
153 PyMem_FREE(s_buffer);
154 return result;
Barry Warsaw226ae6c1999-10-12 19:54:53 +0000155}
156
Guido van Rossum234f9421993-06-17 12:35:49 +0000157static void
Fred Drakefd99de62000-07-09 05:02:18 +0000158float_dealloc(PyFloatObject *op)
Guido van Rossum3132a5a1992-03-27 17:28:44 +0000159{
Guido van Rossum9475a232001-10-05 20:51:39 +0000160 if (PyFloat_CheckExact(op)) {
Martin v. Löwis9f2e3462007-07-21 17:22:18 +0000161 Py_Type(op) = (struct _typeobject *)free_list;
Guido van Rossum9475a232001-10-05 20:51:39 +0000162 free_list = op;
163 }
164 else
Martin v. Löwis9f2e3462007-07-21 17:22:18 +0000165 Py_Type(op)->tp_free((PyObject *)op);
Guido van Rossum3132a5a1992-03-27 17:28:44 +0000166}
167
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000168double
Fred Drakefd99de62000-07-09 05:02:18 +0000169PyFloat_AsDouble(PyObject *op)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000170{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000171 PyNumberMethods *nb;
172 PyFloatObject *fo;
Guido van Rossumb6775db1994-08-01 11:34:53 +0000173 double val;
Tim Petersd2364e82001-11-01 20:09:42 +0000174
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000175 if (op && PyFloat_Check(op))
176 return PyFloat_AS_DOUBLE((PyFloatObject*) op);
Tim Petersd2364e82001-11-01 20:09:42 +0000177
Neil Schemenauer2c77e902002-11-18 16:06:21 +0000178 if (op == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000179 PyErr_BadArgument();
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000180 return -1;
181 }
Tim Petersd2364e82001-11-01 20:09:42 +0000182
Martin v. Löwis9f2e3462007-07-21 17:22:18 +0000183 if ((nb = Py_Type(op)->tp_as_number) == NULL || nb->nb_float == NULL) {
Neil Schemenauer2c77e902002-11-18 16:06:21 +0000184 PyErr_SetString(PyExc_TypeError, "a float is required");
185 return -1;
186 }
187
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000188 fo = (PyFloatObject*) (*nb->nb_float) (op);
Guido van Rossumb6775db1994-08-01 11:34:53 +0000189 if (fo == NULL)
190 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000191 if (!PyFloat_Check(fo)) {
192 PyErr_SetString(PyExc_TypeError,
193 "nb_float should return float object");
Guido van Rossumb6775db1994-08-01 11:34:53 +0000194 return -1;
195 }
Tim Petersd2364e82001-11-01 20:09:42 +0000196
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000197 val = PyFloat_AS_DOUBLE(fo);
198 Py_DECREF(fo);
Tim Petersd2364e82001-11-01 20:09:42 +0000199
Guido van Rossumb6775db1994-08-01 11:34:53 +0000200 return val;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000201}
202
203/* Methods */
204
Tim Peters97019e42001-11-28 22:43:45 +0000205static void
Neal Norwitz545686b2006-12-28 04:45:06 +0000206format_double(char *buf, size_t buflen, double ob_fval, int precision)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000207{
208 register char *cp;
Martin v. Löwis737ea822004-06-08 18:52:54 +0000209 char format[32];
Guido van Rossum04dbf3b2007-08-07 19:51:00 +0000210 /* Subroutine for float_repr, float_str, and others.
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000211 We want float numbers to be recognizable as such,
212 i.e., they should contain a decimal point or an exponent.
213 However, %g may print the number as an integer;
214 in such cases, we append ".0" to the string. */
Tim Peters97019e42001-11-28 22:43:45 +0000215
Martin v. Löwis737ea822004-06-08 18:52:54 +0000216 PyOS_snprintf(format, 32, "%%.%ig", precision);
Neal Norwitz545686b2006-12-28 04:45:06 +0000217 PyOS_ascii_formatd(buf, buflen, format, ob_fval);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000218 cp = buf;
219 if (*cp == '-')
220 cp++;
221 for (; *cp != '\0'; cp++) {
222 /* Any non-digit means it's not an integer;
223 this takes care of NAN and INF as well. */
Guido van Rossum9fa2c111995-02-10 17:00:37 +0000224 if (!isdigit(Py_CHARMASK(*cp)))
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000225 break;
226 }
227 if (*cp == '\0') {
228 *cp++ = '.';
229 *cp++ = '0';
230 *cp++ = '\0';
231 }
232}
233
Neal Norwitz545686b2006-12-28 04:45:06 +0000234static void
235format_float(char *buf, size_t buflen, PyFloatObject *v, int precision)
Tim Peters97019e42001-11-28 22:43:45 +0000236{
Neal Norwitz545686b2006-12-28 04:45:06 +0000237 assert(PyFloat_Check(v));
238 format_double(buf, buflen, PyFloat_AS_DOUBLE(v), precision);
Tim Peters97019e42001-11-28 22:43:45 +0000239}
240
Neil Schemenauer32117e52001-01-04 01:44:34 +0000241/* Macro and helper that convert PyObject obj to a C double and store
242 the value in dbl; this replaces the functionality of the coercion
Tim Peters77d8a4f2001-12-11 20:31:34 +0000243 slot function. If conversion to double raises an exception, obj is
244 set to NULL, and the function invoking this macro returns NULL. If
245 obj is not of float, int or long type, Py_NotImplemented is incref'ed,
246 stored in obj, and returned from the function invoking this macro.
247*/
Neil Schemenauer32117e52001-01-04 01:44:34 +0000248#define CONVERT_TO_DOUBLE(obj, dbl) \
249 if (PyFloat_Check(obj)) \
250 dbl = PyFloat_AS_DOUBLE(obj); \
251 else if (convert_to_double(&(obj), &(dbl)) < 0) \
252 return obj;
253
254static int
Tim Peters9fffa3e2001-09-04 05:14:19 +0000255convert_to_double(PyObject **v, double *dbl)
Neil Schemenauer32117e52001-01-04 01:44:34 +0000256{
257 register PyObject *obj = *v;
Tim Peters9fffa3e2001-09-04 05:14:19 +0000258
Guido van Rossumddefaf32007-01-14 03:31:43 +0000259 if (PyLong_Check(obj)) {
Neil Schemenauer32117e52001-01-04 01:44:34 +0000260 *dbl = PyLong_AsDouble(obj);
Tim Peters9fffa3e2001-09-04 05:14:19 +0000261 if (*dbl == -1.0 && PyErr_Occurred()) {
262 *v = NULL;
263 return -1;
264 }
Neil Schemenauer32117e52001-01-04 01:44:34 +0000265 }
266 else {
267 Py_INCREF(Py_NotImplemented);
268 *v = Py_NotImplemented;
269 return -1;
270 }
271 return 0;
272}
273
Guido van Rossum57072eb1999-12-23 19:00:28 +0000274/* Precisions used by repr() and str(), respectively.
275
276 The repr() precision (17 significant decimal digits) is the minimal number
277 that is guaranteed to have enough precision so that if the number is read
278 back in the exact same binary value is recreated. This is true for IEEE
279 floating point by design, and also happens to work for all other modern
280 hardware.
281
282 The str() precision is chosen so that in most cases, the rounding noise
283 created by various operations is suppressed, while giving plenty of
284 precision for practical use.
285
286*/
287
288#define PREC_REPR 17
289#define PREC_STR 12
290
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000291static PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +0000292float_repr(PyFloatObject *v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000293{
294 char buf[100];
Tim Peters97019e42001-11-28 22:43:45 +0000295 format_float(buf, sizeof(buf), v, PREC_REPR);
Walter Dörwald1ab83302007-05-18 17:15:44 +0000296 return PyUnicode_FromString(buf);
Guido van Rossum57072eb1999-12-23 19:00:28 +0000297}
298
299static PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +0000300float_str(PyFloatObject *v)
Guido van Rossum57072eb1999-12-23 19:00:28 +0000301{
302 char buf[100];
Tim Peters97019e42001-11-28 22:43:45 +0000303 format_float(buf, sizeof(buf), v, PREC_STR);
Walter Dörwald7696ed72007-05-31 15:51:35 +0000304 return PyUnicode_FromString(buf);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000305}
306
Tim Peters307fa782004-09-23 08:06:40 +0000307/* Comparison is pretty much a nightmare. When comparing float to float,
308 * we do it as straightforwardly (and long-windedly) as conceivable, so
309 * that, e.g., Python x == y delivers the same result as the platform
310 * C x == y when x and/or y is a NaN.
311 * When mixing float with an integer type, there's no good *uniform* approach.
312 * Converting the double to an integer obviously doesn't work, since we
313 * may lose info from fractional bits. Converting the integer to a double
314 * also has two failure modes: (1) a long int may trigger overflow (too
315 * large to fit in the dynamic range of a C double); (2) even a C long may have
316 * more bits than fit in a C double (e.g., on a a 64-bit box long may have
317 * 63 bits of precision, but a C double probably has only 53), and then
318 * we can falsely claim equality when low-order integer bits are lost by
319 * coercion to double. So this part is painful too.
320 */
321
Michael W. Hudsond3b33b52004-02-19 19:35:22 +0000322static PyObject*
323float_richcompare(PyObject *v, PyObject *w, int op)
324{
325 double i, j;
326 int r = 0;
327
Tim Peters307fa782004-09-23 08:06:40 +0000328 assert(PyFloat_Check(v));
329 i = PyFloat_AS_DOUBLE(v);
Michael W. Hudsond3b33b52004-02-19 19:35:22 +0000330
Tim Peters307fa782004-09-23 08:06:40 +0000331 /* Switch on the type of w. Set i and j to doubles to be compared,
332 * and op to the richcomp to use.
333 */
334 if (PyFloat_Check(w))
335 j = PyFloat_AS_DOUBLE(w);
336
Thomas Wouters477c8d52006-05-27 19:21:47 +0000337 else if (!Py_IS_FINITE(i)) {
Tim Peters307fa782004-09-23 08:06:40 +0000338 if (PyInt_Check(w) || PyLong_Check(w))
Tim Peterse1c69b32004-09-23 19:22:41 +0000339 /* If i is an infinity, its magnitude exceeds any
340 * finite integer, so it doesn't matter which int we
341 * compare i with. If i is a NaN, similarly.
Tim Peters307fa782004-09-23 08:06:40 +0000342 */
343 j = 0.0;
344 else
345 goto Unimplemented;
346 }
347
Tim Peters307fa782004-09-23 08:06:40 +0000348 else if (PyLong_Check(w)) {
349 int vsign = i == 0.0 ? 0 : i < 0.0 ? -1 : 1;
350 int wsign = _PyLong_Sign(w);
351 size_t nbits;
Tim Peters307fa782004-09-23 08:06:40 +0000352 int exponent;
353
354 if (vsign != wsign) {
355 /* Magnitudes are irrelevant -- the signs alone
356 * determine the outcome.
357 */
358 i = (double)vsign;
359 j = (double)wsign;
360 goto Compare;
361 }
362 /* The signs are the same. */
363 /* Convert w to a double if it fits. In particular, 0 fits. */
364 nbits = _PyLong_NumBits(w);
365 if (nbits == (size_t)-1 && PyErr_Occurred()) {
366 /* This long is so large that size_t isn't big enough
Tim Peterse1c69b32004-09-23 19:22:41 +0000367 * to hold the # of bits. Replace with little doubles
368 * that give the same outcome -- w is so large that
369 * its magnitude must exceed the magnitude of any
370 * finite float.
Tim Peters307fa782004-09-23 08:06:40 +0000371 */
372 PyErr_Clear();
373 i = (double)vsign;
374 assert(wsign != 0);
375 j = wsign * 2.0;
376 goto Compare;
377 }
378 if (nbits <= 48) {
379 j = PyLong_AsDouble(w);
380 /* It's impossible that <= 48 bits overflowed. */
381 assert(j != -1.0 || ! PyErr_Occurred());
382 goto Compare;
383 }
384 assert(wsign != 0); /* else nbits was 0 */
385 assert(vsign != 0); /* if vsign were 0, then since wsign is
386 * not 0, we would have taken the
387 * vsign != wsign branch at the start */
388 /* We want to work with non-negative numbers. */
389 if (vsign < 0) {
390 /* "Multiply both sides" by -1; this also swaps the
391 * comparator.
392 */
393 i = -i;
394 op = _Py_SwappedOp[op];
395 }
396 assert(i > 0.0);
Neal Norwitzb2da01b2006-01-08 01:11:25 +0000397 (void) frexp(i, &exponent);
Tim Peters307fa782004-09-23 08:06:40 +0000398 /* exponent is the # of bits in v before the radix point;
399 * we know that nbits (the # of bits in w) > 48 at this point
400 */
401 if (exponent < 0 || (size_t)exponent < nbits) {
402 i = 1.0;
403 j = 2.0;
404 goto Compare;
405 }
406 if ((size_t)exponent > nbits) {
407 i = 2.0;
408 j = 1.0;
409 goto Compare;
410 }
411 /* v and w have the same number of bits before the radix
412 * point. Construct two longs that have the same comparison
413 * outcome.
414 */
415 {
416 double fracpart;
417 double intpart;
418 PyObject *result = NULL;
419 PyObject *one = NULL;
420 PyObject *vv = NULL;
421 PyObject *ww = w;
422
423 if (wsign < 0) {
424 ww = PyNumber_Negative(w);
425 if (ww == NULL)
426 goto Error;
427 }
428 else
429 Py_INCREF(ww);
430
431 fracpart = modf(i, &intpart);
432 vv = PyLong_FromDouble(intpart);
433 if (vv == NULL)
434 goto Error;
435
436 if (fracpart != 0.0) {
437 /* Shift left, and or a 1 bit into vv
438 * to represent the lost fraction.
439 */
440 PyObject *temp;
441
442 one = PyInt_FromLong(1);
443 if (one == NULL)
444 goto Error;
445
446 temp = PyNumber_Lshift(ww, one);
447 if (temp == NULL)
448 goto Error;
449 Py_DECREF(ww);
450 ww = temp;
451
452 temp = PyNumber_Lshift(vv, one);
453 if (temp == NULL)
454 goto Error;
455 Py_DECREF(vv);
456 vv = temp;
457
458 temp = PyNumber_Or(vv, one);
459 if (temp == NULL)
460 goto Error;
461 Py_DECREF(vv);
462 vv = temp;
463 }
464
465 r = PyObject_RichCompareBool(vv, ww, op);
466 if (r < 0)
467 goto Error;
468 result = PyBool_FromLong(r);
469 Error:
470 Py_XDECREF(vv);
471 Py_XDECREF(ww);
472 Py_XDECREF(one);
473 return result;
474 }
475 } /* else if (PyLong_Check(w)) */
476
477 else /* w isn't float, int, or long */
478 goto Unimplemented;
479
480 Compare:
Michael W. Hudsond3b33b52004-02-19 19:35:22 +0000481 PyFPE_START_PROTECT("richcompare", return NULL)
482 switch (op) {
483 case Py_EQ:
Tim Peters307fa782004-09-23 08:06:40 +0000484 r = i == j;
Michael W. Hudsond3b33b52004-02-19 19:35:22 +0000485 break;
486 case Py_NE:
Tim Peters307fa782004-09-23 08:06:40 +0000487 r = i != j;
Michael W. Hudsond3b33b52004-02-19 19:35:22 +0000488 break;
489 case Py_LE:
Tim Peters307fa782004-09-23 08:06:40 +0000490 r = i <= j;
Michael W. Hudsond3b33b52004-02-19 19:35:22 +0000491 break;
492 case Py_GE:
Tim Peters307fa782004-09-23 08:06:40 +0000493 r = i >= j;
Michael W. Hudsond3b33b52004-02-19 19:35:22 +0000494 break;
495 case Py_LT:
Tim Peters307fa782004-09-23 08:06:40 +0000496 r = i < j;
Michael W. Hudsond3b33b52004-02-19 19:35:22 +0000497 break;
498 case Py_GT:
Tim Peters307fa782004-09-23 08:06:40 +0000499 r = i > j;
Michael W. Hudsond3b33b52004-02-19 19:35:22 +0000500 break;
501 }
Michael W. Hudson957f9772004-02-26 12:33:09 +0000502 PyFPE_END_PROTECT(r)
Michael W. Hudsond3b33b52004-02-19 19:35:22 +0000503 return PyBool_FromLong(r);
Tim Peters307fa782004-09-23 08:06:40 +0000504
505 Unimplemented:
506 Py_INCREF(Py_NotImplemented);
507 return Py_NotImplemented;
Michael W. Hudsond3b33b52004-02-19 19:35:22 +0000508}
509
Guido van Rossum9bfef441993-03-29 10:43:31 +0000510static long
Fred Drakefd99de62000-07-09 05:02:18 +0000511float_hash(PyFloatObject *v)
Guido van Rossum9bfef441993-03-29 10:43:31 +0000512{
Tim Peters39dce292000-08-15 03:34:48 +0000513 return _Py_HashDouble(v->ob_fval);
Guido van Rossum9bfef441993-03-29 10:43:31 +0000514}
515
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000516static PyObject *
Neil Schemenauer32117e52001-01-04 01:44:34 +0000517float_add(PyObject *v, PyObject *w)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000518{
Neil Schemenauer32117e52001-01-04 01:44:34 +0000519 double a,b;
520 CONVERT_TO_DOUBLE(v, a);
521 CONVERT_TO_DOUBLE(w, b);
Guido van Rossum09e6ad01997-02-14 22:54:21 +0000522 PyFPE_START_PROTECT("add", return 0)
Neil Schemenauer32117e52001-01-04 01:44:34 +0000523 a = a + b;
524 PyFPE_END_PROTECT(a)
525 return PyFloat_FromDouble(a);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000526}
527
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000528static PyObject *
Neil Schemenauer32117e52001-01-04 01:44:34 +0000529float_sub(PyObject *v, PyObject *w)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000530{
Neil Schemenauer32117e52001-01-04 01:44:34 +0000531 double a,b;
532 CONVERT_TO_DOUBLE(v, a);
533 CONVERT_TO_DOUBLE(w, b);
Guido van Rossum09e6ad01997-02-14 22:54:21 +0000534 PyFPE_START_PROTECT("subtract", return 0)
Neil Schemenauer32117e52001-01-04 01:44:34 +0000535 a = a - b;
536 PyFPE_END_PROTECT(a)
537 return PyFloat_FromDouble(a);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000538}
539
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000540static PyObject *
Neil Schemenauer32117e52001-01-04 01:44:34 +0000541float_mul(PyObject *v, PyObject *w)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000542{
Neil Schemenauer32117e52001-01-04 01:44:34 +0000543 double a,b;
544 CONVERT_TO_DOUBLE(v, a);
545 CONVERT_TO_DOUBLE(w, b);
Guido van Rossum09e6ad01997-02-14 22:54:21 +0000546 PyFPE_START_PROTECT("multiply", return 0)
Neil Schemenauer32117e52001-01-04 01:44:34 +0000547 a = a * b;
548 PyFPE_END_PROTECT(a)
549 return PyFloat_FromDouble(a);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000550}
551
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000552static PyObject *
Neil Schemenauer32117e52001-01-04 01:44:34 +0000553float_div(PyObject *v, PyObject *w)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000554{
Neil Schemenauer32117e52001-01-04 01:44:34 +0000555 double a,b;
556 CONVERT_TO_DOUBLE(v, a);
557 CONVERT_TO_DOUBLE(w, b);
558 if (b == 0.0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000559 PyErr_SetString(PyExc_ZeroDivisionError, "float division");
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000560 return NULL;
561 }
Guido van Rossum09e6ad01997-02-14 22:54:21 +0000562 PyFPE_START_PROTECT("divide", return 0)
Neil Schemenauer32117e52001-01-04 01:44:34 +0000563 a = a / b;
564 PyFPE_END_PROTECT(a)
565 return PyFloat_FromDouble(a);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000566}
567
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000568static PyObject *
Neil Schemenauer32117e52001-01-04 01:44:34 +0000569float_rem(PyObject *v, PyObject *w)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000570{
Guido van Rossum56cd67a1992-01-26 18:16:35 +0000571 double vx, wx;
Guido van Rossum9263e781999-05-06 14:26:34 +0000572 double mod;
Neil Schemenauer32117e52001-01-04 01:44:34 +0000573 CONVERT_TO_DOUBLE(v, vx);
574 CONVERT_TO_DOUBLE(w, wx);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000575 if (wx == 0.0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000576 PyErr_SetString(PyExc_ZeroDivisionError, "float modulo");
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000577 return NULL;
578 }
Guido van Rossum09e6ad01997-02-14 22:54:21 +0000579 PyFPE_START_PROTECT("modulo", return 0)
Guido van Rossum56cd67a1992-01-26 18:16:35 +0000580 mod = fmod(vx, wx);
Guido van Rossum9263e781999-05-06 14:26:34 +0000581 /* note: checking mod*wx < 0 is incorrect -- underflows to
582 0 if wx < sqrt(smallest nonzero double) */
583 if (mod && ((wx < 0) != (mod < 0))) {
Guido van Rossum56cd67a1992-01-26 18:16:35 +0000584 mod += wx;
Guido van Rossum56cd67a1992-01-26 18:16:35 +0000585 }
Guido van Rossum45b83911997-03-14 04:32:50 +0000586 PyFPE_END_PROTECT(mod)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000587 return PyFloat_FromDouble(mod);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000588}
589
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000590static PyObject *
Neil Schemenauer32117e52001-01-04 01:44:34 +0000591float_divmod(PyObject *v, PyObject *w)
Guido van Rossumeba1b5e1991-05-05 20:07:00 +0000592{
Guido van Rossum15ecff41991-10-20 20:16:45 +0000593 double vx, wx;
Guido van Rossum9263e781999-05-06 14:26:34 +0000594 double div, mod, floordiv;
Neil Schemenauer32117e52001-01-04 01:44:34 +0000595 CONVERT_TO_DOUBLE(v, vx);
596 CONVERT_TO_DOUBLE(w, wx);
Guido van Rossum15ecff41991-10-20 20:16:45 +0000597 if (wx == 0.0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000598 PyErr_SetString(PyExc_ZeroDivisionError, "float divmod()");
Guido van Rossum15ecff41991-10-20 20:16:45 +0000599 return NULL;
600 }
Guido van Rossum09e6ad01997-02-14 22:54:21 +0000601 PyFPE_START_PROTECT("divmod", return 0)
Guido van Rossum15ecff41991-10-20 20:16:45 +0000602 mod = fmod(vx, wx);
Tim Peters78fc0b52000-09-16 03:54:24 +0000603 /* fmod is typically exact, so vx-mod is *mathematically* an
Guido van Rossum9263e781999-05-06 14:26:34 +0000604 exact multiple of wx. But this is fp arithmetic, and fp
605 vx - mod is an approximation; the result is that div may
606 not be an exact integral value after the division, although
607 it will always be very close to one.
608 */
Guido van Rossum15ecff41991-10-20 20:16:45 +0000609 div = (vx - mod) / wx;
Tim Petersd2e40d62001-11-01 23:12:27 +0000610 if (mod) {
611 /* ensure the remainder has the same sign as the denominator */
612 if ((wx < 0) != (mod < 0)) {
613 mod += wx;
614 div -= 1.0;
615 }
616 }
617 else {
618 /* the remainder is zero, and in the presence of signed zeroes
619 fmod returns different results across platforms; ensure
620 it has the same sign as the denominator; we'd like to do
621 "mod = wx * 0.0", but that may get optimized away */
Tim Peters4e8ab5d2001-11-01 23:59:56 +0000622 mod *= mod; /* hide "mod = +0" from optimizer */
Tim Petersd2e40d62001-11-01 23:12:27 +0000623 if (wx < 0.0)
624 mod = -mod;
Guido van Rossum15ecff41991-10-20 20:16:45 +0000625 }
Guido van Rossum9263e781999-05-06 14:26:34 +0000626 /* snap quotient to nearest integral value */
Tim Petersd2e40d62001-11-01 23:12:27 +0000627 if (div) {
628 floordiv = floor(div);
629 if (div - floordiv > 0.5)
630 floordiv += 1.0;
631 }
632 else {
633 /* div is zero - get the same sign as the true quotient */
634 div *= div; /* hide "div = +0" from optimizers */
635 floordiv = div * vx / wx; /* zero w/ sign of vx/wx */
636 }
637 PyFPE_END_PROTECT(floordiv)
Guido van Rossum9263e781999-05-06 14:26:34 +0000638 return Py_BuildValue("(dd)", floordiv, mod);
Guido van Rossumeba1b5e1991-05-05 20:07:00 +0000639}
640
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000641static PyObject *
Tim Peters63a35712001-12-11 19:57:24 +0000642float_floor_div(PyObject *v, PyObject *w)
643{
644 PyObject *t, *r;
645
646 t = float_divmod(v, w);
Tim Peters77d8a4f2001-12-11 20:31:34 +0000647 if (t == NULL || t == Py_NotImplemented)
648 return t;
649 assert(PyTuple_CheckExact(t));
650 r = PyTuple_GET_ITEM(t, 0);
651 Py_INCREF(r);
652 Py_DECREF(t);
653 return r;
Tim Peters63a35712001-12-11 19:57:24 +0000654}
655
656static PyObject *
Neil Schemenauer32117e52001-01-04 01:44:34 +0000657float_pow(PyObject *v, PyObject *w, PyObject *z)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000658{
659 double iv, iw, ix;
Tim Peters32f453e2001-09-03 08:35:41 +0000660
661 if ((PyObject *)z != Py_None) {
Tim Peters4c483c42001-09-05 06:24:58 +0000662 PyErr_SetString(PyExc_TypeError, "pow() 3rd argument not "
Tim Peters97f4a332001-09-05 23:49:24 +0000663 "allowed unless all arguments are integers");
Tim Peters32f453e2001-09-03 08:35:41 +0000664 return NULL;
665 }
666
Neil Schemenauer32117e52001-01-04 01:44:34 +0000667 CONVERT_TO_DOUBLE(v, iv);
668 CONVERT_TO_DOUBLE(w, iw);
Tim Petersc54d1902000-10-06 00:36:09 +0000669
670 /* Sort out special cases here instead of relying on pow() */
Tim Peters96685bf2001-08-23 22:31:37 +0000671 if (iw == 0) { /* v**0 is 1, even 0**0 */
Guido van Rossum360e4b82007-05-14 22:51:27 +0000672 return PyFloat_FromDouble(1.0);
Tim Petersc54d1902000-10-06 00:36:09 +0000673 }
Tim Peters96685bf2001-08-23 22:31:37 +0000674 if (iv == 0.0) { /* 0**w is error if w<0, else 1 */
Tim Petersc54d1902000-10-06 00:36:09 +0000675 if (iw < 0.0) {
676 PyErr_SetString(PyExc_ZeroDivisionError,
Fred Drake661ea262000-10-24 19:57:45 +0000677 "0.0 cannot be raised to a negative power");
Tim Petersc54d1902000-10-06 00:36:09 +0000678 return NULL;
679 }
680 return PyFloat_FromDouble(0.0);
681 }
Tim Peterse87568d2003-05-24 20:18:24 +0000682 if (iv < 0.0) {
683 /* Whether this is an error is a mess, and bumps into libm
684 * bugs so we have to figure it out ourselves.
685 */
686 if (iw != floor(iw)) {
687 PyErr_SetString(PyExc_ValueError, "negative number "
688 "cannot be raised to a fractional power");
689 return NULL;
690 }
691 /* iw is an exact integer, albeit perhaps a very large one.
692 * -1 raised to an exact integer should never be exceptional.
693 * Alas, some libms (chiefly glibc as of early 2003) return
694 * NaN and set EDOM on pow(-1, large_int) if the int doesn't
695 * happen to be representable in a *C* integer. That's a
696 * bug; we let that slide in math.pow() (which currently
697 * reflects all platform accidents), but not for Python's **.
698 */
Thomas Wouters477c8d52006-05-27 19:21:47 +0000699 if (iv == -1.0 && Py_IS_FINITE(iw)) {
Tim Peterse87568d2003-05-24 20:18:24 +0000700 /* Return 1 if iw is even, -1 if iw is odd; there's
701 * no guarantee that any C integral type is big
702 * enough to hold iw, so we have to check this
703 * indirectly.
704 */
705 ix = floor(iw * 0.5) * 2.0;
706 return PyFloat_FromDouble(ix == iw ? 1.0 : -1.0);
707 }
708 /* Else iv != -1.0, and overflow or underflow are possible.
709 * Unless we're to write pow() ourselves, we have to trust
710 * the platform to do this correctly.
711 */
Guido van Rossum86c04c21996-08-09 20:50:14 +0000712 }
Tim Peters96685bf2001-08-23 22:31:37 +0000713 errno = 0;
714 PyFPE_START_PROTECT("pow", return NULL)
715 ix = pow(iv, iw);
716 PyFPE_END_PROTECT(ix)
Tim Petersdc5a5082002-03-09 04:58:24 +0000717 Py_ADJUST_ERANGE1(ix);
Guido van Rossum2a9096b1990-10-21 22:15:08 +0000718 if (errno != 0) {
Tim Peterse87568d2003-05-24 20:18:24 +0000719 /* We don't expect any errno value other than ERANGE, but
720 * the range of libm bugs appears unbounded.
721 */
722 PyErr_SetFromErrno(errno == ERANGE ? PyExc_OverflowError :
723 PyExc_ValueError);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000724 return NULL;
Guido van Rossum2a9096b1990-10-21 22:15:08 +0000725 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000726 return PyFloat_FromDouble(ix);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000727}
728
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000729static PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +0000730float_neg(PyFloatObject *v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000731{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000732 return PyFloat_FromDouble(-v->ob_fval);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000733}
734
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000735static PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +0000736float_abs(PyFloatObject *v)
Guido van Rossumeba1b5e1991-05-05 20:07:00 +0000737{
Tim Petersfaf0cd22001-11-01 21:51:15 +0000738 return PyFloat_FromDouble(fabs(v->ob_fval));
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000739}
740
Guido van Rossum50b4ef61991-05-14 11:57:01 +0000741static int
Jack Diederich4dafcc42006-11-28 19:15:13 +0000742float_bool(PyFloatObject *v)
Guido van Rossum50b4ef61991-05-14 11:57:01 +0000743{
744 return v->ob_fval != 0.0;
745}
746
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000747static PyObject *
Guido van Rossum2fa33db2007-08-23 22:07:24 +0000748float_trunc(PyObject *v)
Guido van Rossum1899c2e1992-09-12 11:09:23 +0000749{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000750 double x = PyFloat_AsDouble(v);
Tim Peters7321ec42001-07-26 20:02:17 +0000751 double wholepart; /* integral portion of x, rounded toward 0 */
Tim Peters7321ec42001-07-26 20:02:17 +0000752
753 (void)modf(x, &wholepart);
Tim Peters7d791242002-11-21 22:26:37 +0000754 /* Try to get out cheap if this fits in a Python int. The attempt
755 * to cast to long must be protected, as C doesn't define what
756 * happens if the double is too big to fit in a long. Some rare
757 * systems raise an exception then (RISCOS was mentioned as one,
758 * and someone using a non-default option on Sun also bumped into
759 * that). Note that checking for >= and <= LONG_{MIN,MAX} would
760 * still be vulnerable: if a long has more bits of precision than
761 * a double, casting MIN/MAX to double may yield an approximation,
762 * and if that's rounded up, then, e.g., wholepart=LONG_MAX+1 would
763 * yield true from the C expression wholepart<=LONG_MAX, despite
764 * that wholepart is actually greater than LONG_MAX.
765 */
766 if (LONG_MIN < wholepart && wholepart < LONG_MAX) {
767 const long aslong = (long)wholepart;
Tim Peters7321ec42001-07-26 20:02:17 +0000768 return PyInt_FromLong(aslong);
Tim Peters7d791242002-11-21 22:26:37 +0000769 }
770 return PyLong_FromDouble(wholepart);
Guido van Rossum1899c2e1992-09-12 11:09:23 +0000771}
772
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000773static PyObject *
Guido van Rossum2fa33db2007-08-23 22:07:24 +0000774float_round(PyObject *v, PyObject *args)
775{
776#define UNDEF_NDIGITS (-0x7fffffff) /* Unlikely ndigits value */
777 double x;
778 double f;
779 double flr, cil;
780 double rounded;
781 int i;
782 int ndigits = UNDEF_NDIGITS;
783
784 if (!PyArg_ParseTuple(args, "|i", &ndigits))
785 return NULL;
786
787 x = PyFloat_AsDouble(v);
788
789 if (ndigits != UNDEF_NDIGITS) {
790 f = 1.0;
791 i = abs(ndigits);
792 while (--i >= 0)
793 f = f*10.0;
794 if (ndigits < 0)
795 x /= f;
796 else
797 x *= f;
798 }
799
800 flr = floor(x);
801 cil = ceil(x);
802
803 if (x-flr > 0.5)
804 rounded = cil;
805 else if (x-flr == 0.5)
806 rounded = fmod(flr, 2) == 0 ? flr : cil;
807 else
808 rounded = flr;
809
810 if (ndigits != UNDEF_NDIGITS) {
811 if (ndigits < 0)
812 rounded *= f;
813 else
814 rounded /= f;
815 return PyFloat_FromDouble(rounded);
816 }
817
818 return PyLong_FromDouble(rounded);
819#undef UNDEF_NDIGITS
820}
821
822static PyObject *
Fred Drakefd99de62000-07-09 05:02:18 +0000823float_float(PyObject *v)
Guido van Rossum1899c2e1992-09-12 11:09:23 +0000824{
Brett Cannonc3647ac2005-04-26 03:45:26 +0000825 if (PyFloat_CheckExact(v))
826 Py_INCREF(v);
827 else
828 v = PyFloat_FromDouble(((PyFloatObject *)v)->ob_fval);
Guido van Rossum1899c2e1992-09-12 11:09:23 +0000829 return v;
830}
831
832
Jeremy Hylton938ace62002-07-17 16:30:39 +0000833static PyObject *
Guido van Rossumbef14172001-08-29 15:47:46 +0000834float_subtype_new(PyTypeObject *type, PyObject *args, PyObject *kwds);
835
Tim Peters6d6c1a32001-08-02 04:15:00 +0000836static PyObject *
837float_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
838{
839 PyObject *x = Py_False; /* Integer zero */
Martin v. Löwis15e62742006-02-27 16:46:16 +0000840 static char *kwlist[] = {"x", 0};
Tim Peters6d6c1a32001-08-02 04:15:00 +0000841
Guido van Rossumbef14172001-08-29 15:47:46 +0000842 if (type != &PyFloat_Type)
843 return float_subtype_new(type, args, kwds); /* Wimp out */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000844 if (!PyArg_ParseTupleAndKeywords(args, kwds, "|O:float", kwlist, &x))
845 return NULL;
846 if (PyString_Check(x))
Georg Brandl428f0642007-03-18 18:35:15 +0000847 return PyFloat_FromString(x);
Tim Peters6d6c1a32001-08-02 04:15:00 +0000848 return PyNumber_Float(x);
849}
850
Guido van Rossumbef14172001-08-29 15:47:46 +0000851/* Wimpy, slow approach to tp_new calls for subtypes of float:
852 first create a regular float from whatever arguments we got,
853 then allocate a subtype instance and initialize its ob_fval
854 from the regular float. The regular float is then thrown away.
855*/
856static PyObject *
857float_subtype_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
858{
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000859 PyObject *tmp, *newobj;
Guido van Rossumbef14172001-08-29 15:47:46 +0000860
861 assert(PyType_IsSubtype(type, &PyFloat_Type));
862 tmp = float_new(&PyFloat_Type, args, kwds);
863 if (tmp == NULL)
864 return NULL;
Tim Peters2400fa42001-09-12 19:12:49 +0000865 assert(PyFloat_CheckExact(tmp));
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000866 newobj = type->tp_alloc(type, 0);
867 if (newobj == NULL) {
Raymond Hettingerf4667932003-06-28 20:04:25 +0000868 Py_DECREF(tmp);
Guido van Rossumbef14172001-08-29 15:47:46 +0000869 return NULL;
Raymond Hettingerf4667932003-06-28 20:04:25 +0000870 }
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000871 ((PyFloatObject *)newobj)->ob_fval = ((PyFloatObject *)tmp)->ob_fval;
Guido van Rossumbef14172001-08-29 15:47:46 +0000872 Py_DECREF(tmp);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000873 return newobj;
Guido van Rossumbef14172001-08-29 15:47:46 +0000874}
875
Guido van Rossum5d9113d2003-01-29 17:58:45 +0000876static PyObject *
877float_getnewargs(PyFloatObject *v)
878{
879 return Py_BuildValue("(d)", v->ob_fval);
880}
881
Michael W. Hudsonba283e22005-05-27 15:23:20 +0000882/* this is for the benefit of the pack/unpack routines below */
883
884typedef enum {
885 unknown_format, ieee_big_endian_format, ieee_little_endian_format
886} float_format_type;
887
888static float_format_type double_format, float_format;
889static float_format_type detected_double_format, detected_float_format;
890
891static PyObject *
892float_getformat(PyTypeObject *v, PyObject* arg)
893{
894 char* s;
895 float_format_type r;
896
Guido van Rossum2be161d2007-05-15 20:43:51 +0000897 if (PyUnicode_Check(arg)) {
898 arg = _PyUnicode_AsDefaultEncodedString(arg, NULL);
899 if (arg == NULL)
900 return NULL;
901 }
Michael W. Hudsonba283e22005-05-27 15:23:20 +0000902 if (!PyString_Check(arg)) {
903 PyErr_Format(PyExc_TypeError,
904 "__getformat__() argument must be string, not %.500s",
Martin v. Löwis9f2e3462007-07-21 17:22:18 +0000905 Py_Type(arg)->tp_name);
Michael W. Hudsonba283e22005-05-27 15:23:20 +0000906 return NULL;
907 }
908 s = PyString_AS_STRING(arg);
909 if (strcmp(s, "double") == 0) {
910 r = double_format;
911 }
912 else if (strcmp(s, "float") == 0) {
913 r = float_format;
914 }
915 else {
916 PyErr_SetString(PyExc_ValueError,
917 "__getformat__() argument 1 must be "
918 "'double' or 'float'");
919 return NULL;
920 }
921
922 switch (r) {
923 case unknown_format:
Walter Dörwald71044582007-06-22 12:26:52 +0000924 return PyUnicode_FromString("unknown");
Michael W. Hudsonba283e22005-05-27 15:23:20 +0000925 case ieee_little_endian_format:
Walter Dörwald71044582007-06-22 12:26:52 +0000926 return PyUnicode_FromString("IEEE, little-endian");
Michael W. Hudsonba283e22005-05-27 15:23:20 +0000927 case ieee_big_endian_format:
Walter Dörwald71044582007-06-22 12:26:52 +0000928 return PyUnicode_FromString("IEEE, big-endian");
Michael W. Hudsonba283e22005-05-27 15:23:20 +0000929 default:
930 Py_FatalError("insane float_format or double_format");
931 return NULL;
932 }
933}
934
935PyDoc_STRVAR(float_getformat_doc,
936"float.__getformat__(typestr) -> string\n"
937"\n"
938"You probably don't want to use this function. It exists mainly to be\n"
939"used in Python's test suite.\n"
940"\n"
941"typestr must be 'double' or 'float'. This function returns whichever of\n"
942"'unknown', 'IEEE, big-endian' or 'IEEE, little-endian' best describes the\n"
943"format of floating point numbers used by the C type named by typestr.");
944
945static PyObject *
946float_setformat(PyTypeObject *v, PyObject* args)
947{
948 char* typestr;
949 char* format;
950 float_format_type f;
951 float_format_type detected;
952 float_format_type *p;
953
954 if (!PyArg_ParseTuple(args, "ss:__setformat__", &typestr, &format))
955 return NULL;
956
957 if (strcmp(typestr, "double") == 0) {
958 p = &double_format;
959 detected = detected_double_format;
960 }
961 else if (strcmp(typestr, "float") == 0) {
962 p = &float_format;
963 detected = detected_float_format;
964 }
965 else {
966 PyErr_SetString(PyExc_ValueError,
967 "__setformat__() argument 1 must "
968 "be 'double' or 'float'");
969 return NULL;
970 }
971
972 if (strcmp(format, "unknown") == 0) {
973 f = unknown_format;
974 }
975 else if (strcmp(format, "IEEE, little-endian") == 0) {
976 f = ieee_little_endian_format;
977 }
978 else if (strcmp(format, "IEEE, big-endian") == 0) {
979 f = ieee_big_endian_format;
980 }
981 else {
982 PyErr_SetString(PyExc_ValueError,
983 "__setformat__() argument 2 must be "
984 "'unknown', 'IEEE, little-endian' or "
985 "'IEEE, big-endian'");
986 return NULL;
987
988 }
989
990 if (f != unknown_format && f != detected) {
991 PyErr_Format(PyExc_ValueError,
992 "can only set %s format to 'unknown' or the "
993 "detected platform value", typestr);
994 return NULL;
995 }
996
997 *p = f;
998 Py_RETURN_NONE;
999}
1000
1001PyDoc_STRVAR(float_setformat_doc,
1002"float.__setformat__(typestr, fmt) -> None\n"
1003"\n"
1004"You probably don't want to use this function. It exists mainly to be\n"
1005"used in Python's test suite.\n"
1006"\n"
1007"typestr must be 'double' or 'float'. fmt must be one of 'unknown',\n"
1008"'IEEE, big-endian' or 'IEEE, little-endian', and in addition can only be\n"
1009"one of the latter two if it appears to match the underlying C reality.\n"
1010"\n"
1011"Overrides the automatic determination of C-level floating point type.\n"
1012"This affects how floats are converted to and from binary strings.");
1013
Guido van Rossumb43daf72007-08-01 18:08:08 +00001014static PyObject *
1015float_getzero(PyObject *v, void *closure)
1016{
1017 return PyFloat_FromDouble(0.0);
1018}
1019
Eric Smith8c663262007-08-25 02:26:07 +00001020static PyObject *
1021float__format__(PyObject *self, PyObject *args)
1022{
1023 /* when back porting this to 2.6, check type of the format_spec
1024 and call either unicode_long__format__ or
1025 string_long__format__ */
1026 return unicode_float__format__(self, args);
1027}
1028
1029PyDoc_STRVAR(float__format__doc,
1030"float.__format__(format_spec) -> string\n"
1031"\n"
1032"Formats the float according to format_spec.");
1033
1034
Guido van Rossum5d9113d2003-01-29 17:58:45 +00001035static PyMethodDef float_methods[] = {
Guido van Rossumb43daf72007-08-01 18:08:08 +00001036 {"conjugate", (PyCFunction)float_float, METH_NOARGS,
1037 "Returns self, the complex conjugate of any float."},
Guido van Rossum2fa33db2007-08-23 22:07:24 +00001038 {"__trunc__", (PyCFunction)float_trunc, METH_NOARGS,
1039 "Returns the Integral closest to x between 0 and x."},
1040 {"__round__", (PyCFunction)float_round, METH_VARARGS,
1041 "Returns the Integral closest to x, rounding half toward even.\n"
1042 "When an argument is passed, works like built-in round(x, ndigits)."},
Guido van Rossum5d9113d2003-01-29 17:58:45 +00001043 {"__getnewargs__", (PyCFunction)float_getnewargs, METH_NOARGS},
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001044 {"__getformat__", (PyCFunction)float_getformat,
1045 METH_O|METH_CLASS, float_getformat_doc},
1046 {"__setformat__", (PyCFunction)float_setformat,
1047 METH_VARARGS|METH_CLASS, float_setformat_doc},
Eric Smith8c663262007-08-25 02:26:07 +00001048 {"__format__", (PyCFunction)float__format__,
1049 METH_VARARGS, float__format__doc},
Guido van Rossum5d9113d2003-01-29 17:58:45 +00001050 {NULL, NULL} /* sentinel */
1051};
1052
Guido van Rossumb43daf72007-08-01 18:08:08 +00001053static PyGetSetDef float_getset[] = {
1054 {"real",
1055 (getter)float_float, (setter)NULL,
1056 "the real part of a complex number",
1057 NULL},
1058 {"imag",
1059 (getter)float_getzero, (setter)NULL,
1060 "the imaginary part of a complex number",
1061 NULL},
1062 {NULL} /* Sentinel */
1063};
1064
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001065PyDoc_STRVAR(float_doc,
Tim Peters6d6c1a32001-08-02 04:15:00 +00001066"float(x) -> floating point number\n\
1067\n\
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00001068Convert a string or number to a floating point number, if possible.");
Tim Peters6d6c1a32001-08-02 04:15:00 +00001069
1070
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001071static PyNumberMethods float_as_number = {
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001072 float_add, /*nb_add*/
1073 float_sub, /*nb_subtract*/
1074 float_mul, /*nb_multiply*/
1075 float_rem, /*nb_remainder*/
1076 float_divmod, /*nb_divmod*/
1077 float_pow, /*nb_power*/
Guido van Rossumb6775db1994-08-01 11:34:53 +00001078 (unaryfunc)float_neg, /*nb_negative*/
Guido van Rossumb43daf72007-08-01 18:08:08 +00001079 (unaryfunc)float_float, /*nb_positive*/
Guido van Rossumb6775db1994-08-01 11:34:53 +00001080 (unaryfunc)float_abs, /*nb_absolute*/
Jack Diederich4dafcc42006-11-28 19:15:13 +00001081 (inquiry)float_bool, /*nb_bool*/
Guido van Rossum27acb331991-10-24 14:55:28 +00001082 0, /*nb_invert*/
1083 0, /*nb_lshift*/
1084 0, /*nb_rshift*/
1085 0, /*nb_and*/
1086 0, /*nb_xor*/
1087 0, /*nb_or*/
Neal Norwitz4886cc32006-08-21 17:06:07 +00001088 (coercion)0, /*nb_coerce*/
Guido van Rossum2fa33db2007-08-23 22:07:24 +00001089 float_trunc, /*nb_int*/
1090 float_trunc, /*nb_long*/
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001091 float_float, /*nb_float*/
Guido van Rossum4668b002001-08-08 05:00:18 +00001092 0, /* nb_oct */
1093 0, /* nb_hex */
1094 0, /* nb_inplace_add */
1095 0, /* nb_inplace_subtract */
1096 0, /* nb_inplace_multiply */
Guido van Rossum4668b002001-08-08 05:00:18 +00001097 0, /* nb_inplace_remainder */
1098 0, /* nb_inplace_power */
1099 0, /* nb_inplace_lshift */
1100 0, /* nb_inplace_rshift */
1101 0, /* nb_inplace_and */
1102 0, /* nb_inplace_xor */
1103 0, /* nb_inplace_or */
Tim Peters63a35712001-12-11 19:57:24 +00001104 float_floor_div, /* nb_floor_divide */
Guido van Rossum4668b002001-08-08 05:00:18 +00001105 float_div, /* nb_true_divide */
1106 0, /* nb_inplace_floor_divide */
1107 0, /* nb_inplace_true_divide */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001108};
1109
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001110PyTypeObject PyFloat_Type = {
Martin v. Löwis9f2e3462007-07-21 17:22:18 +00001111 PyVarObject_HEAD_INIT(&PyType_Type, 0)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001112 "float",
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001113 sizeof(PyFloatObject),
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001114 0,
Tim Peters6d6c1a32001-08-02 04:15:00 +00001115 (destructor)float_dealloc, /* tp_dealloc */
Guido van Rossum04dbf3b2007-08-07 19:51:00 +00001116 0, /* tp_print */
Tim Peters6d6c1a32001-08-02 04:15:00 +00001117 0, /* tp_getattr */
1118 0, /* tp_setattr */
Michael W. Hudson08678a12004-05-26 17:36:12 +00001119 0, /* tp_compare */
Tim Peters6d6c1a32001-08-02 04:15:00 +00001120 (reprfunc)float_repr, /* tp_repr */
1121 &float_as_number, /* tp_as_number */
1122 0, /* tp_as_sequence */
1123 0, /* tp_as_mapping */
1124 (hashfunc)float_hash, /* tp_hash */
1125 0, /* tp_call */
1126 (reprfunc)float_str, /* tp_str */
1127 PyObject_GenericGetAttr, /* tp_getattro */
1128 0, /* tp_setattro */
1129 0, /* tp_as_buffer */
Guido van Rossum3cf5b1e2006-07-27 21:53:35 +00001130 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
Tim Peters6d6c1a32001-08-02 04:15:00 +00001131 float_doc, /* tp_doc */
1132 0, /* tp_traverse */
1133 0, /* tp_clear */
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001134 float_richcompare, /* tp_richcompare */
Tim Peters6d6c1a32001-08-02 04:15:00 +00001135 0, /* tp_weaklistoffset */
1136 0, /* tp_iter */
1137 0, /* tp_iternext */
Guido van Rossum5d9113d2003-01-29 17:58:45 +00001138 float_methods, /* tp_methods */
Tim Peters6d6c1a32001-08-02 04:15:00 +00001139 0, /* tp_members */
Guido van Rossumb43daf72007-08-01 18:08:08 +00001140 float_getset, /* tp_getset */
Tim Peters6d6c1a32001-08-02 04:15:00 +00001141 0, /* tp_base */
1142 0, /* tp_dict */
1143 0, /* tp_descr_get */
1144 0, /* tp_descr_set */
1145 0, /* tp_dictoffset */
1146 0, /* tp_init */
1147 0, /* tp_alloc */
1148 float_new, /* tp_new */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00001149};
Guido van Rossumfbbd57e1997-08-05 02:16:08 +00001150
1151void
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001152_PyFloat_Init(void)
1153{
1154 /* We attempt to determine if this machine is using IEEE
1155 floating point formats by peering at the bits of some
1156 carefully chosen values. If it looks like we are on an
1157 IEEE platform, the float packing/unpacking routines can
1158 just copy bits, if not they resort to arithmetic & shifts
1159 and masks. The shifts & masks approach works on all finite
1160 values, but what happens to infinities, NaNs and signed
1161 zeroes on packing is an accident, and attempting to unpack
1162 a NaN or an infinity will raise an exception.
1163
1164 Note that if we're on some whacked-out platform which uses
1165 IEEE formats but isn't strictly little-endian or big-
1166 endian, we will fall back to the portable shifts & masks
1167 method. */
1168
1169#if SIZEOF_DOUBLE == 8
1170 {
1171 double x = 9006104071832581.0;
1172 if (memcmp(&x, "\x43\x3f\xff\x01\x02\x03\x04\x05", 8) == 0)
1173 detected_double_format = ieee_big_endian_format;
1174 else if (memcmp(&x, "\x05\x04\x03\x02\x01\xff\x3f\x43", 8) == 0)
1175 detected_double_format = ieee_little_endian_format;
1176 else
1177 detected_double_format = unknown_format;
1178 }
1179#else
1180 detected_double_format = unknown_format;
1181#endif
1182
1183#if SIZEOF_FLOAT == 4
1184 {
1185 float y = 16711938.0;
1186 if (memcmp(&y, "\x4b\x7f\x01\x02", 4) == 0)
1187 detected_float_format = ieee_big_endian_format;
1188 else if (memcmp(&y, "\x02\x01\x7f\x4b", 4) == 0)
1189 detected_float_format = ieee_little_endian_format;
1190 else
1191 detected_float_format = unknown_format;
1192 }
1193#else
1194 detected_float_format = unknown_format;
1195#endif
1196
1197 double_format = detected_double_format;
1198 float_format = detected_float_format;
1199}
1200
1201void
Fred Drakefd99de62000-07-09 05:02:18 +00001202PyFloat_Fini(void)
Guido van Rossumfbbd57e1997-08-05 02:16:08 +00001203{
Guido van Rossum3fce8831999-03-12 19:43:17 +00001204 PyFloatObject *p;
1205 PyFloatBlock *list, *next;
Neal Norwitz739a8f82004-07-08 01:55:58 +00001206 unsigned i;
Guido van Rossumf61bbc81999-03-12 00:12:21 +00001207 int bc, bf; /* block count, number of freed blocks */
1208 int frem, fsum; /* remaining unfreed floats per block, total */
1209
1210 bc = 0;
1211 bf = 0;
1212 fsum = 0;
1213 list = block_list;
1214 block_list = NULL;
Guido van Rossumd7b5fb81999-03-19 20:59:40 +00001215 free_list = NULL;
Guido van Rossumf61bbc81999-03-12 00:12:21 +00001216 while (list != NULL) {
Guido van Rossumf61bbc81999-03-12 00:12:21 +00001217 bc++;
1218 frem = 0;
Guido van Rossumd7b5fb81999-03-19 20:59:40 +00001219 for (i = 0, p = &list->objects[0];
1220 i < N_FLOATOBJECTS;
1221 i++, p++) {
Martin v. Löwis9f2e3462007-07-21 17:22:18 +00001222 if (PyFloat_CheckExact(p) && Py_Refcnt(p) != 0)
Guido van Rossumf61bbc81999-03-12 00:12:21 +00001223 frem++;
1224 }
Guido van Rossum3fce8831999-03-12 19:43:17 +00001225 next = list->next;
Guido van Rossumf61bbc81999-03-12 00:12:21 +00001226 if (frem) {
Guido van Rossum3fce8831999-03-12 19:43:17 +00001227 list->next = block_list;
1228 block_list = list;
Guido van Rossumd7b5fb81999-03-19 20:59:40 +00001229 for (i = 0, p = &list->objects[0];
1230 i < N_FLOATOBJECTS;
1231 i++, p++) {
Guido van Rossumdea6ef92001-09-11 16:13:52 +00001232 if (!PyFloat_CheckExact(p) ||
Martin v. Löwis9f2e3462007-07-21 17:22:18 +00001233 Py_Refcnt(p) == 0) {
1234 Py_Type(p) = (struct _typeobject *)
Guido van Rossumd7b5fb81999-03-19 20:59:40 +00001235 free_list;
1236 free_list = p;
1237 }
1238 }
Guido van Rossumf61bbc81999-03-12 00:12:21 +00001239 }
1240 else {
Guido van Rossumb18618d2000-05-03 23:44:39 +00001241 PyMem_FREE(list); /* XXX PyObject_FREE ??? */
Guido van Rossumf61bbc81999-03-12 00:12:21 +00001242 bf++;
1243 }
1244 fsum += frem;
Guido van Rossum3fce8831999-03-12 19:43:17 +00001245 list = next;
Guido van Rossumf61bbc81999-03-12 00:12:21 +00001246 }
Guido van Rossum3fce8831999-03-12 19:43:17 +00001247 if (!Py_VerboseFlag)
1248 return;
1249 fprintf(stderr, "# cleanup floats");
1250 if (!fsum) {
1251 fprintf(stderr, "\n");
1252 }
1253 else {
1254 fprintf(stderr,
1255 ": %d unfreed float%s in %d out of %d block%s\n",
1256 fsum, fsum == 1 ? "" : "s",
1257 bc - bf, bc, bc == 1 ? "" : "s");
1258 }
1259 if (Py_VerboseFlag > 1) {
1260 list = block_list;
1261 while (list != NULL) {
Guido van Rossumd7b5fb81999-03-19 20:59:40 +00001262 for (i = 0, p = &list->objects[0];
1263 i < N_FLOATOBJECTS;
1264 i++, p++) {
Guido van Rossumdea6ef92001-09-11 16:13:52 +00001265 if (PyFloat_CheckExact(p) &&
Martin v. Löwis9f2e3462007-07-21 17:22:18 +00001266 Py_Refcnt(p) != 0) {
Guido van Rossum3fce8831999-03-12 19:43:17 +00001267 char buf[100];
Neal Norwitz545686b2006-12-28 04:45:06 +00001268 format_float(buf, sizeof(buf), p, PREC_STR);
Thomas Wouters8b87a0b2006-03-01 05:41:20 +00001269 /* XXX(twouters) cast refcount to
1270 long until %zd is universally
1271 available
1272 */
Guido van Rossum3fce8831999-03-12 19:43:17 +00001273 fprintf(stderr,
Thomas Wouters8b87a0b2006-03-01 05:41:20 +00001274 "# <float at %p, refcnt=%ld, val=%s>\n",
Martin v. Löwis9f2e3462007-07-21 17:22:18 +00001275 p, (long)Py_Refcnt(p), buf);
Guido van Rossum3fce8831999-03-12 19:43:17 +00001276 }
1277 }
1278 list = list->next;
Guido van Rossumf61bbc81999-03-12 00:12:21 +00001279 }
1280 }
Guido van Rossumfbbd57e1997-08-05 02:16:08 +00001281}
Tim Peters9905b942003-03-20 20:53:32 +00001282
1283/*----------------------------------------------------------------------------
1284 * _PyFloat_{Pack,Unpack}{4,8}. See floatobject.h.
1285 *
1286 * TODO: On platforms that use the standard IEEE-754 single and double
1287 * formats natively, these routines could simply copy the bytes.
1288 */
1289int
1290_PyFloat_Pack4(double x, unsigned char *p, int le)
1291{
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001292 if (float_format == unknown_format) {
1293 unsigned char sign;
1294 int e;
1295 double f;
1296 unsigned int fbits;
1297 int incr = 1;
Tim Peters9905b942003-03-20 20:53:32 +00001298
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001299 if (le) {
1300 p += 3;
1301 incr = -1;
1302 }
Tim Peters9905b942003-03-20 20:53:32 +00001303
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001304 if (x < 0) {
1305 sign = 1;
1306 x = -x;
1307 }
1308 else
1309 sign = 0;
Tim Peters9905b942003-03-20 20:53:32 +00001310
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001311 f = frexp(x, &e);
Tim Peters9905b942003-03-20 20:53:32 +00001312
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001313 /* Normalize f to be in the range [1.0, 2.0) */
1314 if (0.5 <= f && f < 1.0) {
1315 f *= 2.0;
1316 e--;
1317 }
1318 else if (f == 0.0)
1319 e = 0;
1320 else {
1321 PyErr_SetString(PyExc_SystemError,
1322 "frexp() result out of range");
1323 return -1;
1324 }
1325
1326 if (e >= 128)
1327 goto Overflow;
1328 else if (e < -126) {
1329 /* Gradual underflow */
1330 f = ldexp(f, 126 + e);
1331 e = 0;
1332 }
1333 else if (!(e == 0 && f == 0.0)) {
1334 e += 127;
1335 f -= 1.0; /* Get rid of leading 1 */
1336 }
1337
1338 f *= 8388608.0; /* 2**23 */
1339 fbits = (unsigned int)(f + 0.5); /* Round */
1340 assert(fbits <= 8388608);
1341 if (fbits >> 23) {
1342 /* The carry propagated out of a string of 23 1 bits. */
1343 fbits = 0;
1344 ++e;
1345 if (e >= 255)
1346 goto Overflow;
1347 }
1348
1349 /* First byte */
1350 *p = (sign << 7) | (e >> 1);
1351 p += incr;
1352
1353 /* Second byte */
1354 *p = (char) (((e & 1) << 7) | (fbits >> 16));
1355 p += incr;
1356
1357 /* Third byte */
1358 *p = (fbits >> 8) & 0xFF;
1359 p += incr;
1360
1361 /* Fourth byte */
1362 *p = fbits & 0xFF;
1363
1364 /* Done */
1365 return 0;
1366
1367 Overflow:
1368 PyErr_SetString(PyExc_OverflowError,
1369 "float too large to pack with f format");
Tim Peters9905b942003-03-20 20:53:32 +00001370 return -1;
1371 }
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001372 else {
Michael W. Hudson3095ad02005-06-30 00:02:26 +00001373 float y = (float)x;
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001374 const char *s = (char*)&y;
1375 int i, incr = 1;
Tim Peters9905b942003-03-20 20:53:32 +00001376
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001377 if ((float_format == ieee_little_endian_format && !le)
1378 || (float_format == ieee_big_endian_format && le)) {
1379 p += 3;
1380 incr = -1;
1381 }
1382
1383 for (i = 0; i < 4; i++) {
1384 *p = *s++;
1385 p += incr;
1386 }
1387 return 0;
Tim Peters9905b942003-03-20 20:53:32 +00001388 }
Tim Peters9905b942003-03-20 20:53:32 +00001389}
1390
1391int
1392_PyFloat_Pack8(double x, unsigned char *p, int le)
1393{
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001394 if (double_format == unknown_format) {
1395 unsigned char sign;
1396 int e;
1397 double f;
1398 unsigned int fhi, flo;
1399 int incr = 1;
Tim Peters9905b942003-03-20 20:53:32 +00001400
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001401 if (le) {
1402 p += 7;
1403 incr = -1;
1404 }
Tim Peters9905b942003-03-20 20:53:32 +00001405
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001406 if (x < 0) {
1407 sign = 1;
1408 x = -x;
1409 }
1410 else
1411 sign = 0;
Tim Peters9905b942003-03-20 20:53:32 +00001412
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001413 f = frexp(x, &e);
Tim Peters9905b942003-03-20 20:53:32 +00001414
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001415 /* Normalize f to be in the range [1.0, 2.0) */
1416 if (0.5 <= f && f < 1.0) {
1417 f *= 2.0;
1418 e--;
1419 }
1420 else if (f == 0.0)
1421 e = 0;
1422 else {
1423 PyErr_SetString(PyExc_SystemError,
1424 "frexp() result out of range");
1425 return -1;
1426 }
1427
1428 if (e >= 1024)
1429 goto Overflow;
1430 else if (e < -1022) {
1431 /* Gradual underflow */
1432 f = ldexp(f, 1022 + e);
1433 e = 0;
1434 }
1435 else if (!(e == 0 && f == 0.0)) {
1436 e += 1023;
1437 f -= 1.0; /* Get rid of leading 1 */
1438 }
1439
1440 /* fhi receives the high 28 bits; flo the low 24 bits (== 52 bits) */
1441 f *= 268435456.0; /* 2**28 */
1442 fhi = (unsigned int)f; /* Truncate */
1443 assert(fhi < 268435456);
1444
1445 f -= (double)fhi;
1446 f *= 16777216.0; /* 2**24 */
1447 flo = (unsigned int)(f + 0.5); /* Round */
1448 assert(flo <= 16777216);
1449 if (flo >> 24) {
1450 /* The carry propagated out of a string of 24 1 bits. */
1451 flo = 0;
1452 ++fhi;
1453 if (fhi >> 28) {
1454 /* And it also progagated out of the next 28 bits. */
1455 fhi = 0;
1456 ++e;
1457 if (e >= 2047)
1458 goto Overflow;
1459 }
1460 }
1461
1462 /* First byte */
1463 *p = (sign << 7) | (e >> 4);
1464 p += incr;
1465
1466 /* Second byte */
1467 *p = (unsigned char) (((e & 0xF) << 4) | (fhi >> 24));
1468 p += incr;
1469
1470 /* Third byte */
1471 *p = (fhi >> 16) & 0xFF;
1472 p += incr;
1473
1474 /* Fourth byte */
1475 *p = (fhi >> 8) & 0xFF;
1476 p += incr;
1477
1478 /* Fifth byte */
1479 *p = fhi & 0xFF;
1480 p += incr;
1481
1482 /* Sixth byte */
1483 *p = (flo >> 16) & 0xFF;
1484 p += incr;
1485
1486 /* Seventh byte */
1487 *p = (flo >> 8) & 0xFF;
1488 p += incr;
1489
1490 /* Eighth byte */
1491 *p = flo & 0xFF;
1492 p += incr;
1493
1494 /* Done */
1495 return 0;
1496
1497 Overflow:
1498 PyErr_SetString(PyExc_OverflowError,
1499 "float too large to pack with d format");
Tim Peters9905b942003-03-20 20:53:32 +00001500 return -1;
1501 }
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001502 else {
1503 const char *s = (char*)&x;
1504 int i, incr = 1;
Tim Peters9905b942003-03-20 20:53:32 +00001505
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001506 if ((double_format == ieee_little_endian_format && !le)
1507 || (double_format == ieee_big_endian_format && le)) {
1508 p += 7;
1509 incr = -1;
Tim Peters9905b942003-03-20 20:53:32 +00001510 }
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001511
1512 for (i = 0; i < 8; i++) {
1513 *p = *s++;
1514 p += incr;
1515 }
1516 return 0;
Tim Peters9905b942003-03-20 20:53:32 +00001517 }
Tim Peters9905b942003-03-20 20:53:32 +00001518}
1519
Neal Norwitz545686b2006-12-28 04:45:06 +00001520/* Should only be used by marshal. */
1521int
1522_PyFloat_Repr(double x, char *p, size_t len)
1523{
1524 format_double(p, len, x, PREC_REPR);
1525 return (int)strlen(p);
1526}
1527
Tim Peters9905b942003-03-20 20:53:32 +00001528double
1529_PyFloat_Unpack4(const unsigned char *p, int le)
1530{
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001531 if (float_format == unknown_format) {
1532 unsigned char sign;
1533 int e;
1534 unsigned int f;
1535 double x;
1536 int incr = 1;
Tim Peters9905b942003-03-20 20:53:32 +00001537
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001538 if (le) {
1539 p += 3;
1540 incr = -1;
1541 }
1542
1543 /* First byte */
1544 sign = (*p >> 7) & 1;
1545 e = (*p & 0x7F) << 1;
1546 p += incr;
1547
1548 /* Second byte */
1549 e |= (*p >> 7) & 1;
1550 f = (*p & 0x7F) << 16;
1551 p += incr;
1552
1553 if (e == 255) {
1554 PyErr_SetString(
1555 PyExc_ValueError,
1556 "can't unpack IEEE 754 special value "
1557 "on non-IEEE platform");
1558 return -1;
1559 }
1560
1561 /* Third byte */
1562 f |= *p << 8;
1563 p += incr;
1564
1565 /* Fourth byte */
1566 f |= *p;
1567
1568 x = (double)f / 8388608.0;
1569
1570 /* XXX This sadly ignores Inf/NaN issues */
1571 if (e == 0)
1572 e = -126;
1573 else {
1574 x += 1.0;
1575 e -= 127;
1576 }
1577 x = ldexp(x, e);
1578
1579 if (sign)
1580 x = -x;
1581
1582 return x;
Tim Peters9905b942003-03-20 20:53:32 +00001583 }
Tim Peters9905b942003-03-20 20:53:32 +00001584 else {
Michael W. Hudsonb78a5fc2005-12-05 00:27:49 +00001585 float x;
1586
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001587 if ((float_format == ieee_little_endian_format && !le)
1588 || (float_format == ieee_big_endian_format && le)) {
Michael W. Hudsonb78a5fc2005-12-05 00:27:49 +00001589 char buf[4];
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001590 char *d = &buf[3];
1591 int i;
Tim Peters9905b942003-03-20 20:53:32 +00001592
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001593 for (i = 0; i < 4; i++) {
1594 *d-- = *p++;
1595 }
Michael W. Hudsonb78a5fc2005-12-05 00:27:49 +00001596 memcpy(&x, buf, 4);
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001597 }
1598 else {
Michael W. Hudsonb78a5fc2005-12-05 00:27:49 +00001599 memcpy(&x, p, 4);
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001600 }
Michael W. Hudsonb78a5fc2005-12-05 00:27:49 +00001601
1602 return x;
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001603 }
Tim Peters9905b942003-03-20 20:53:32 +00001604}
1605
1606double
1607_PyFloat_Unpack8(const unsigned char *p, int le)
1608{
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001609 if (double_format == unknown_format) {
1610 unsigned char sign;
1611 int e;
1612 unsigned int fhi, flo;
1613 double x;
1614 int incr = 1;
Tim Peters9905b942003-03-20 20:53:32 +00001615
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001616 if (le) {
1617 p += 7;
1618 incr = -1;
1619 }
1620
1621 /* First byte */
1622 sign = (*p >> 7) & 1;
1623 e = (*p & 0x7F) << 4;
1624
1625 p += incr;
1626
1627 /* Second byte */
1628 e |= (*p >> 4) & 0xF;
1629 fhi = (*p & 0xF) << 24;
1630 p += incr;
1631
1632 if (e == 2047) {
1633 PyErr_SetString(
1634 PyExc_ValueError,
1635 "can't unpack IEEE 754 special value "
1636 "on non-IEEE platform");
1637 return -1.0;
1638 }
1639
1640 /* Third byte */
1641 fhi |= *p << 16;
1642 p += incr;
1643
1644 /* Fourth byte */
1645 fhi |= *p << 8;
1646 p += incr;
1647
1648 /* Fifth byte */
1649 fhi |= *p;
1650 p += incr;
1651
1652 /* Sixth byte */
1653 flo = *p << 16;
1654 p += incr;
1655
1656 /* Seventh byte */
1657 flo |= *p << 8;
1658 p += incr;
1659
1660 /* Eighth byte */
1661 flo |= *p;
1662
1663 x = (double)fhi + (double)flo / 16777216.0; /* 2**24 */
1664 x /= 268435456.0; /* 2**28 */
1665
1666 if (e == 0)
1667 e = -1022;
1668 else {
1669 x += 1.0;
1670 e -= 1023;
1671 }
1672 x = ldexp(x, e);
1673
1674 if (sign)
1675 x = -x;
1676
1677 return x;
Tim Peters9905b942003-03-20 20:53:32 +00001678 }
Tim Peters9905b942003-03-20 20:53:32 +00001679 else {
Michael W. Hudsonb78a5fc2005-12-05 00:27:49 +00001680 double x;
1681
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001682 if ((double_format == ieee_little_endian_format && !le)
1683 || (double_format == ieee_big_endian_format && le)) {
1684 char buf[8];
1685 char *d = &buf[7];
1686 int i;
1687
1688 for (i = 0; i < 8; i++) {
1689 *d-- = *p++;
1690 }
Michael W. Hudsonb78a5fc2005-12-05 00:27:49 +00001691 memcpy(&x, buf, 8);
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001692 }
1693 else {
Michael W. Hudsonb78a5fc2005-12-05 00:27:49 +00001694 memcpy(&x, p, 8);
Michael W. Hudsonba283e22005-05-27 15:23:20 +00001695 }
Michael W. Hudsonb78a5fc2005-12-05 00:27:49 +00001696
1697 return x;
Tim Peters9905b942003-03-20 20:53:32 +00001698 }
Tim Peters9905b942003-03-20 20:53:32 +00001699}