ref: 74d4de6ad5bcac532631d8dfb82d9e6c0435fc0f
dir: /operators.c/
#include <limits.h> #include <assert.h> #include "dtypes.h" #include "utils.h" #include "ieee754.h" extern double trunc(double x); STATIC_INLINE double fpart(double arg) { return arg - trunc(arg); } // given a number, determine an appropriate type for storing it #if 0 numerictype_t effective_numerictype(double r) { double fp; fp = fpart(r); if (fp != 0 || r > U64_MAX || r < S64_MIN) { if (r > FLT_MAX || r < -FLT_MAX || (fabs(r) < FLT_MIN)) { return T_DOUBLE; } else { return T_FLOAT; } } else if (r >= SCHAR_MIN && r <= SCHAR_MAX) { return T_INT8; } else if (r >= SHRT_MIN && r <= SHRT_MAX) { return T_INT16; } else if (r >= INT_MIN && r <= INT_MAX) { return T_INT32; } else if (r <= S64_MAX) { return T_INT64; } return T_UINT64; } #else // simpler version implementing a smaller preferred type repertoire numerictype_t effective_numerictype(double r) { double fp; fp = fpart(r); if (fp != 0 || r > U64_MAX || r < S64_MIN) { return T_DOUBLE; } else if (r >= INT_MIN && r <= INT_MAX) { return T_INT32; } else if (r <= S64_MAX) { return T_INT64; } return T_UINT64; } #endif double conv_to_double(void *data, numerictype_t tag) { double d=0; switch (tag) { case T_INT8: d = (double)*(int8_t*)data; break; case T_UINT8: d = (double)*(uint8_t*)data; break; case T_INT16: d = (double)*(int16_t*)data; break; case T_UINT16: d = (double)*(uint16_t*)data; break; case T_INT32: d = (double)*(int32_t*)data; break; case T_UINT32: d = (double)*(uint32_t*)data; break; case T_INT64: d = (double)*(int64_t*)data; if (d > 0 && *(int64_t*)data < 0) // can happen! d = -d; break; case T_UINT64: d = (double)*(uint64_t*)data; break; case T_FLOAT: d = (double)*(float*)data; break; case T_DOUBLE: return *(double*)data; } return d; } void conv_from_double(void *dest, double d, numerictype_t tag) { switch (tag) { case T_INT8: *(int8_t*)dest = d; break; case T_UINT8: *(uint8_t*)dest = d; break; case T_INT16: *(int16_t*)dest = d; break; case T_UINT16: *(uint16_t*)dest = d; break; case T_INT32: *(int32_t*)dest = d; break; case T_UINT32: *(uint32_t*)dest = d; break; case T_INT64: *(int64_t*)dest = d; if (d > 0 && *(int64_t*)dest < 0) // 0x8000000000000000 is a bitch *(int64_t*)dest = S64_MAX; break; case T_UINT64: *(uint64_t*)dest = (int64_t)d; break; case T_FLOAT: *(float*)dest = d; break; case T_DOUBLE: *(double*)dest = d; break; } } #define CONV_TO_INTTYPE(type) \ type##_t conv_to_##type(void *data, numerictype_t tag) \ { \ type##_t i=0; \ switch (tag) { \ case T_INT8: i = (type##_t)*(int8_t*)data; break; \ case T_UINT8: i = (type##_t)*(uint8_t*)data; break; \ case T_INT16: i = (type##_t)*(int16_t*)data; break; \ case T_UINT16: i = (type##_t)*(uint16_t*)data; break; \ case T_INT32: i = (type##_t)*(int32_t*)data; break; \ case T_UINT32: i = (type##_t)*(uint32_t*)data; break; \ case T_INT64: i = (type##_t)*(int64_t*)data; break; \ case T_UINT64: i = (type##_t)*(uint64_t*)data; break; \ case T_FLOAT: i = (type##_t)*(float*)data; break; \ case T_DOUBLE: i = (type##_t)*(double*)data; break; \ } \ return i; \ } CONV_TO_INTTYPE(int64) CONV_TO_INTTYPE(int32) CONV_TO_INTTYPE(uint32) // this is needed to work around a possible compiler bug // casting negative floats and doubles to uint64. you need // to cast to int64 first. uint64_t conv_to_uint64(void *data, numerictype_t tag) { uint64_t i=0; switch (tag) { case T_INT8: i = (uint64_t)*(int8_t*)data; break; case T_UINT8: i = (uint64_t)*(uint8_t*)data; break; case T_INT16: i = (uint64_t)*(int16_t*)data; break; case T_UINT16: i = (uint64_t)*(uint16_t*)data; break; case T_INT32: i = (uint64_t)*(int32_t*)data; break; case T_UINT32: i = (uint64_t)*(uint32_t*)data; break; case T_INT64: i = (uint64_t)*(int64_t*)data; break; case T_UINT64: i = (uint64_t)*(uint64_t*)data; break; case T_FLOAT: if (*(float*)data >= 0) i = (uint64_t)*(float*)data; else i = (uint64_t)(int64_t)*(float*)data; break; case T_DOUBLE: if (*(double*)data >= 0) i = (uint64_t)*(double*)data; else i = (uint64_t)(int64_t)*(double*)data; break; } return i; } int cmp_same_lt(void *a, void *b, numerictype_t tag) { switch (tag) { case T_INT8: return *(int8_t*)a < *(int8_t*)b; case T_UINT8: return *(uint8_t*)a < *(uint8_t*)b; case T_INT16: return *(int16_t*)a < *(int16_t*)b; case T_UINT16: return *(uint16_t*)a < *(uint16_t*)b; case T_INT32: return *(int32_t*)a < *(int32_t*)b; case T_UINT32: return *(uint32_t*)a < *(uint32_t*)b; case T_INT64: return *(int64_t*)a < *(int64_t*)b; case T_UINT64: return *(uint64_t*)a < *(uint64_t*)b; case T_FLOAT: return *(float*)a < *(float*)b; case T_DOUBLE: return *(double*)a < *(double*)b; } return 0; } int cmp_same_eq(void *a, void *b, numerictype_t tag) { switch (tag) { case T_INT8: return *(int8_t*)a == *(int8_t*)b; case T_UINT8: return *(uint8_t*)a == *(uint8_t*)b; case T_INT16: return *(int16_t*)a == *(int16_t*)b; case T_UINT16: return *(uint16_t*)a == *(uint16_t*)b; case T_INT32: return *(int32_t*)a == *(int32_t*)b; case T_UINT32: return *(uint32_t*)a == *(uint32_t*)b; case T_INT64: return *(int64_t*)a == *(int64_t*)b; case T_UINT64: return *(uint64_t*)a == *(uint64_t*)b; case T_FLOAT: return *(float*)a == *(float*)b; case T_DOUBLE: return *(double*)a == *(double*)b; } return 0; } int cmp_lt(void *a, numerictype_t atag, void *b, numerictype_t btag) { if (atag==btag) return cmp_same_lt(a, b, atag); double da = conv_to_double(a, atag); double db = conv_to_double(b, btag); // casting to double will only get the wrong answer for big int64s // that differ in low bits if (da < db) return 1; if (db < da) return 0; if (atag == T_UINT64) { if (btag == T_INT64) { if (*(int64_t*)b >= 0) { return (*(uint64_t*)a < (uint64_t)*(int64_t*)b); } return ((int64_t)*(uint64_t*)a < *(int64_t*)b); } else if (btag == T_DOUBLE) { if (db != db) return 0; return (*(uint64_t*)a < (uint64_t)*(double*)b); } } else if (atag == T_INT64) { if (btag == T_UINT64) { if (*(int64_t*)a >= 0) { return ((uint64_t)*(int64_t*)a < *(uint64_t*)b); } return (*(int64_t*)a < (int64_t)*(uint64_t*)b); } else if (btag == T_DOUBLE) { if (db != db) return 0; return (*(int64_t*)a < (int64_t)*(double*)b); } } if (btag == T_UINT64) { if (atag == T_DOUBLE) { if (da != da) return 0; return (*(uint64_t*)b > (uint64_t)*(double*)a); } } else if (btag == T_INT64) { if (atag == T_DOUBLE) { if (da != da) return 0; return (*(int64_t*)b > (int64_t)*(double*)a); } } return 0; } int cmp_eq(void *a, numerictype_t atag, void *b, numerictype_t btag, int equalnans) { union { double d; int64_t i64; } u, v; if (atag==btag && (!equalnans || atag < T_FLOAT)) return cmp_same_eq(a, b, atag); double da = conv_to_double(a, atag); double db = conv_to_double(b, btag); if ((int)atag >= T_FLOAT && (int)btag >= T_FLOAT) { if (equalnans) { u.d = da; v.d = db; return u.i64 == v.i64; } return (da == db); } if (da != db) return 0; if (atag == T_UINT64) { // this is safe because if a had been bigger than S64_MAX, // we would already have concluded that it's bigger than b. if (btag == T_INT64) { return ((int64_t)*(uint64_t*)a == *(int64_t*)b); } else if (btag == T_DOUBLE) { return (*(uint64_t*)a == (uint64_t)(int64_t)*(double*)b); } } else if (atag == T_INT64) { if (btag == T_UINT64) { return (*(int64_t*)a == (int64_t)*(uint64_t*)b); } else if (btag == T_DOUBLE) { return (*(int64_t*)a == (int64_t)*(double*)b); } } else if (btag == T_UINT64) { if (atag == T_INT64) { return ((int64_t)*(uint64_t*)b == *(int64_t*)a); } else if (atag == T_DOUBLE) { return (*(uint64_t*)b == (uint64_t)(int64_t)*(double*)a); } } else if (btag == T_INT64) { if (atag == T_UINT64) { return (*(int64_t*)b == (int64_t)*(uint64_t*)a); } else if (atag == T_DOUBLE) { return (*(int64_t*)b == (int64_t)*(double*)a); } } return 1; }