exact_value.cpp 25 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009
  1. #include <math.h>
  2. // TODO(bill): Big numbers
  3. // IMPORTANT TODO(bill): This needs to be completely fixed!!!!!!!!
  4. gb_global BlockingMutex hash_exact_value_mutex;
  5. struct Ast;
  6. struct HashKey;
  7. struct Type;
  8. struct Entity;
  9. bool are_types_identical(Type *x, Type *y);
  10. struct Complex128 {
  11. f64 real, imag;
  12. };
  13. struct Quaternion256 {
  14. f64 imag, jmag, kmag, real;
  15. };
  16. Quaternion256 quaternion256_inverse(Quaternion256 x) {
  17. f64 invmag2 = 1.0 / (x.real*x.real + x.imag*x.imag + x.jmag*x.jmag + x.kmag*x.kmag);
  18. x.real = +x.real * invmag2;
  19. x.imag = -x.imag * invmag2;
  20. x.jmag = -x.jmag * invmag2;
  21. x.kmag = -x.kmag * invmag2;
  22. return x;
  23. }
  24. enum ExactValueKind {
  25. ExactValue_Invalid = 0,
  26. ExactValue_Bool = 1,
  27. ExactValue_String = 2,
  28. ExactValue_Integer = 3,
  29. ExactValue_Float = 4,
  30. ExactValue_Complex = 5,
  31. ExactValue_Quaternion = 6,
  32. ExactValue_Pointer = 7,
  33. ExactValue_Compound = 8, // TODO(bill): Is this good enough?
  34. ExactValue_Procedure = 9, // TODO(bill): Is this good enough?
  35. ExactValue_Typeid = 10,
  36. ExactValue_Count,
  37. };
  38. struct ExactValue {
  39. ExactValueKind kind;
  40. union {
  41. bool value_bool;
  42. String value_string;
  43. BigInt value_integer; // NOTE(bill): This must be an integer and not a pointer
  44. f64 value_float;
  45. i64 value_pointer;
  46. Complex128 *value_complex;
  47. Quaternion256 *value_quaternion;
  48. Ast * value_compound;
  49. Ast * value_procedure;
  50. Type * value_typeid;
  51. };
  52. };
  53. gb_global ExactValue const empty_exact_value = {};
  54. HashKey hash_exact_value(ExactValue v) {
  55. mutex_lock(&hash_exact_value_mutex);
  56. defer (mutex_unlock(&hash_exact_value_mutex));
  57. HashKey empty = {};
  58. switch (v.kind) {
  59. case ExactValue_Invalid:
  60. return empty;
  61. case ExactValue_Bool:
  62. return hash_integer(u64(v.value_bool));
  63. case ExactValue_String:
  64. {
  65. char const *str = string_intern(v.value_string);
  66. return hash_pointer(str);
  67. }
  68. case ExactValue_Integer:
  69. {
  70. HashKey key = hashing_proc(v.value_integer.dp, gb_size_of(*v.value_integer.dp) * v.value_integer.used);
  71. u8 last = (u8)v.value_integer.sign;
  72. key.key = (key.key ^ last) * 0x100000001b3ll;
  73. return key;
  74. }
  75. case ExactValue_Float:
  76. return hash_f64(v.value_float);
  77. case ExactValue_Pointer:
  78. return hash_integer(v.value_pointer);
  79. case ExactValue_Complex:
  80. return hashing_proc(v.value_complex, gb_size_of(Complex128));
  81. case ExactValue_Quaternion:
  82. return hashing_proc(v.value_quaternion, gb_size_of(Quaternion256));
  83. case ExactValue_Compound:
  84. return hash_pointer(v.value_compound);
  85. case ExactValue_Procedure:
  86. return hash_pointer(v.value_procedure);
  87. case ExactValue_Typeid:
  88. return hash_pointer(v.value_typeid);
  89. }
  90. return hashing_proc(&v, gb_size_of(ExactValue));
  91. }
  92. ExactValue exact_value_compound(Ast *node) {
  93. ExactValue result = {ExactValue_Compound};
  94. result.value_compound = node;
  95. return result;
  96. }
  97. ExactValue exact_value_bool(bool b) {
  98. ExactValue result = {ExactValue_Bool};
  99. result.value_bool = (b != 0);
  100. return result;
  101. }
  102. ExactValue exact_value_string(String string) {
  103. // TODO(bill): Allow for numbers with underscores in them
  104. ExactValue result = {ExactValue_String};
  105. result.value_string = string;
  106. return result;
  107. }
  108. ExactValue exact_value_i64(i64 i) {
  109. ExactValue result = {ExactValue_Integer};
  110. big_int_from_i64(&result.value_integer, i);
  111. return result;
  112. }
  113. ExactValue exact_value_u64(u64 i) {
  114. ExactValue result = {ExactValue_Integer};
  115. big_int_from_u64(&result.value_integer, i);
  116. return result;
  117. }
  118. ExactValue exact_value_float(f64 f) {
  119. ExactValue result = {ExactValue_Float};
  120. result.value_float = f;
  121. return result;
  122. }
  123. ExactValue exact_value_complex(f64 real, f64 imag) {
  124. ExactValue result = {ExactValue_Complex};
  125. result.value_complex = gb_alloc_item(permanent_allocator(), Complex128);
  126. result.value_complex->real = real;
  127. result.value_complex->imag = imag;
  128. return result;
  129. }
  130. ExactValue exact_value_quaternion(f64 real, f64 imag, f64 jmag, f64 kmag) {
  131. ExactValue result = {ExactValue_Quaternion};
  132. result.value_quaternion = gb_alloc_item(permanent_allocator(), Quaternion256);
  133. result.value_quaternion->real = real;
  134. result.value_quaternion->imag = imag;
  135. result.value_quaternion->jmag = jmag;
  136. result.value_quaternion->kmag = kmag;
  137. return result;
  138. }
  139. ExactValue exact_value_pointer(i64 ptr) {
  140. ExactValue result = {ExactValue_Pointer};
  141. result.value_pointer = ptr;
  142. return result;
  143. }
  144. ExactValue exact_value_procedure(Ast *node) {
  145. ExactValue result = {ExactValue_Procedure};
  146. result.value_procedure = node;
  147. return result;
  148. }
  149. ExactValue exact_value_typeid(Type *type) {
  150. ExactValue result = {ExactValue_Typeid};
  151. result.value_typeid = type;
  152. return result;
  153. }
  154. ExactValue exact_value_integer_from_string(String const &string) {
  155. ExactValue result = {ExactValue_Integer};
  156. big_int_from_string(&result.value_integer, string);
  157. return result;
  158. }
  159. f64 float_from_string(String string) {
  160. isize i = 0;
  161. u8 *str = string.text;
  162. isize len = string.len;
  163. f64 sign = 1.0;
  164. if (str[i] == '-') {
  165. sign = -1.0;
  166. i++;
  167. } else if (*str == '+') {
  168. i++;
  169. }
  170. f64 value = 0.0;
  171. for (; i < len; i++) {
  172. Rune r = cast(Rune)str[i];
  173. if (r == '_') {
  174. continue;
  175. }
  176. i64 v = digit_value(r);
  177. if (v >= 10) {
  178. break;
  179. }
  180. value *= 10.0;
  181. value += v;
  182. }
  183. if (str[i] == '.') {
  184. f64 pow10 = 10.0;
  185. i++;
  186. for (; i < string.len; i++) {
  187. Rune r = cast(Rune)str[i];
  188. if (r == '_') {
  189. continue;
  190. }
  191. i64 v = digit_value(r);
  192. if (v >= 10) {
  193. break;
  194. }
  195. value += v/pow10;
  196. pow10 *= 10.0;
  197. }
  198. }
  199. bool frac = false;
  200. f64 scale = 1.0;
  201. if ((str[i] == 'e') || (str[i] == 'E')) {
  202. i++;
  203. if (str[i] == '-') {
  204. frac = true;
  205. i++;
  206. } else if (str[i] == '+') {
  207. i++;
  208. }
  209. u32 exp = 0;
  210. for (; i < len; i++) {
  211. Rune r = cast(Rune)str[i];
  212. if (r == '_') {
  213. continue;
  214. }
  215. u32 d = cast(u32)digit_value(r);
  216. if (d >= 10) {
  217. break;
  218. }
  219. exp = exp * 10 + d;
  220. }
  221. if (exp > 308) exp = 308;
  222. while (exp >= 50) { scale *= 1e50; exp -= 50; }
  223. while (exp >= 8) { scale *= 1e8; exp -= 8; }
  224. while (exp > 0) { scale *= 10.0; exp -= 1; }
  225. }
  226. return sign * (frac ? (value / scale) : (value * scale));
  227. }
  228. ExactValue exact_value_float_from_string(String string) {
  229. if (string.len > 2 && string[0] == '0' && string[1] == 'h') {
  230. isize digit_count = 0;
  231. for (isize i = 2; i < string.len; i++) {
  232. if (string[i] != '_') {
  233. digit_count += 1;
  234. }
  235. }
  236. u64 u = u64_from_string(string);
  237. if (digit_count == 4) {
  238. u16 x = cast(u16)u;
  239. f32 f = f16_to_f32(x);
  240. return exact_value_float(cast(f64)f);
  241. } else if (digit_count == 8) {
  242. u32 x = cast(u32)u;
  243. f32 f = bit_cast<f32>(x);
  244. return exact_value_float(cast(f64)f);
  245. } else if (digit_count == 16) {
  246. f64 f = bit_cast<f64>(u);
  247. return exact_value_float(f);
  248. } else {
  249. GB_PANIC("Invalid hexadecimal float, expected 8 or 16 digits, got %td", digit_count);
  250. }
  251. }
  252. if (!string_contains_char(string, '.') && !string_contains_char(string, '-')) {
  253. // NOTE(bill): treat as integer
  254. return exact_value_integer_from_string(string);
  255. }
  256. f64 f = float_from_string(string);
  257. return exact_value_float(f);
  258. }
  259. ExactValue exact_value_from_basic_literal(TokenKind kind, String const &string) {
  260. switch (kind) {
  261. case Token_String: return exact_value_string(string);
  262. case Token_Integer: return exact_value_integer_from_string(string);
  263. case Token_Float: return exact_value_float_from_string(string);
  264. case Token_Imag: {
  265. String str = string;
  266. Rune last_rune = cast(Rune)str[str.len-1];
  267. str.len--; // Ignore the 'i|j|k'
  268. f64 imag = float_from_string(str);
  269. switch (last_rune) {
  270. case 'i': return exact_value_complex(0, imag);
  271. case 'j': return exact_value_quaternion(0, 0, imag, 0);
  272. case 'k': return exact_value_quaternion(0, 0, 0, imag);
  273. default: GB_PANIC("Invalid imaginary basic literal");
  274. }
  275. }
  276. case Token_Rune: {
  277. Rune r = GB_RUNE_INVALID;
  278. utf8_decode(string.text, string.len, &r);
  279. return exact_value_i64(r);
  280. }
  281. default:
  282. GB_PANIC("Invalid token for basic literal");
  283. break;
  284. }
  285. ExactValue result = {ExactValue_Invalid};
  286. return result;
  287. }
  288. ExactValue exact_value_to_integer(ExactValue v) {
  289. switch (v.kind) {
  290. case ExactValue_Bool: {
  291. i64 i = 0;
  292. if (v.value_bool) {
  293. i = 1;
  294. }
  295. return exact_value_i64(i);
  296. }
  297. case ExactValue_Integer:
  298. return v;
  299. case ExactValue_Float: {
  300. i64 i = cast(i64)v.value_float;
  301. f64 f = cast(f64)i;
  302. if (f == v.value_float) {
  303. return exact_value_i64(i);
  304. }
  305. break;
  306. }
  307. case ExactValue_Pointer:
  308. return exact_value_i64(cast(i64)cast(intptr)v.value_pointer);
  309. }
  310. ExactValue r = {ExactValue_Invalid};
  311. return r;
  312. }
  313. ExactValue exact_value_to_float(ExactValue v) {
  314. switch (v.kind) {
  315. case ExactValue_Integer:
  316. return exact_value_float(big_int_to_f64(&v.value_integer));
  317. case ExactValue_Float:
  318. return v;
  319. }
  320. ExactValue r = {ExactValue_Invalid};
  321. return r;
  322. }
  323. ExactValue exact_value_to_complex(ExactValue v) {
  324. switch (v.kind) {
  325. case ExactValue_Integer:
  326. return exact_value_complex(big_int_to_f64(&v.value_integer), 0);
  327. case ExactValue_Float:
  328. return exact_value_complex(v.value_float, 0);
  329. case ExactValue_Complex:
  330. return v;
  331. // case ExactValue_Quaternion:
  332. // return exact_value_complex(v.value_quaternion.real, v.value_quaternion.imag);
  333. }
  334. ExactValue r = {ExactValue_Invalid};
  335. v.value_complex = gb_alloc_item(permanent_allocator(), Complex128);
  336. return r;
  337. }
  338. ExactValue exact_value_to_quaternion(ExactValue v) {
  339. switch (v.kind) {
  340. case ExactValue_Integer:
  341. return exact_value_quaternion(big_int_to_f64(&v.value_integer), 0, 0, 0);
  342. case ExactValue_Float:
  343. return exact_value_quaternion(v.value_float, 0, 0, 0);
  344. case ExactValue_Complex:
  345. return exact_value_quaternion(v.value_complex->real, v.value_complex->imag, 0, 0);
  346. case ExactValue_Quaternion:
  347. return v;
  348. }
  349. ExactValue r = {ExactValue_Invalid};
  350. v.value_quaternion = gb_alloc_item(permanent_allocator(), Quaternion256);
  351. return r;
  352. }
  353. ExactValue exact_value_real(ExactValue v) {
  354. switch (v.kind) {
  355. case ExactValue_Integer:
  356. case ExactValue_Float:
  357. return v;
  358. case ExactValue_Complex:
  359. return exact_value_float(v.value_complex->real);
  360. case ExactValue_Quaternion:
  361. return exact_value_float(v.value_quaternion->real);
  362. }
  363. ExactValue r = {ExactValue_Invalid};
  364. return r;
  365. }
  366. ExactValue exact_value_imag(ExactValue v) {
  367. switch (v.kind) {
  368. case ExactValue_Integer:
  369. case ExactValue_Float:
  370. return exact_value_i64(0);
  371. case ExactValue_Complex:
  372. return exact_value_float(v.value_complex->imag);
  373. case ExactValue_Quaternion:
  374. return exact_value_float(v.value_quaternion->imag);
  375. }
  376. ExactValue r = {ExactValue_Invalid};
  377. return r;
  378. }
  379. ExactValue exact_value_jmag(ExactValue v) {
  380. switch (v.kind) {
  381. case ExactValue_Integer:
  382. case ExactValue_Float:
  383. case ExactValue_Complex:
  384. return exact_value_i64(0);
  385. case ExactValue_Quaternion:
  386. return exact_value_float(v.value_quaternion->jmag);
  387. }
  388. ExactValue r = {ExactValue_Invalid};
  389. return r;
  390. }
  391. ExactValue exact_value_kmag(ExactValue v) {
  392. switch (v.kind) {
  393. case ExactValue_Integer:
  394. case ExactValue_Float:
  395. case ExactValue_Complex:
  396. return exact_value_i64(0);
  397. case ExactValue_Quaternion:
  398. return exact_value_float(v.value_quaternion->kmag);
  399. }
  400. ExactValue r = {ExactValue_Invalid};
  401. return r;
  402. }
  403. ExactValue exact_value_make_imag(ExactValue v) {
  404. switch (v.kind) {
  405. case ExactValue_Integer:
  406. return exact_value_complex(0, exact_value_to_float(v).value_float);
  407. case ExactValue_Float:
  408. return exact_value_complex(0, v.value_float);
  409. default:
  410. GB_PANIC("Expected an integer or float type for 'exact_value_make_imag'");
  411. }
  412. ExactValue r = {ExactValue_Invalid};
  413. return r;
  414. }
  415. ExactValue exact_value_make_jmag(ExactValue v) {
  416. switch (v.kind) {
  417. case ExactValue_Integer:
  418. return exact_value_quaternion(0, 0, exact_value_to_float(v).value_float, 0);
  419. case ExactValue_Float:
  420. return exact_value_quaternion(0, 0, v.value_float, 0);
  421. default:
  422. GB_PANIC("Expected an integer or float type for 'exact_value_make_imag'");
  423. }
  424. ExactValue r = {ExactValue_Invalid};
  425. return r;
  426. }
  427. ExactValue exact_value_make_kmag(ExactValue v) {
  428. switch (v.kind) {
  429. case ExactValue_Integer:
  430. return exact_value_quaternion(0, 0, 0, exact_value_to_float(v).value_float);
  431. case ExactValue_Float:
  432. return exact_value_quaternion(0, 0, 0, v.value_float);
  433. default:
  434. GB_PANIC("Expected an integer or float type for 'exact_value_make_imag'");
  435. }
  436. ExactValue r = {ExactValue_Invalid};
  437. return r;
  438. }
  439. i64 exact_value_to_i64(ExactValue v) {
  440. v = exact_value_to_integer(v);
  441. if (v.kind == ExactValue_Integer) {
  442. return big_int_to_i64(&v.value_integer);
  443. }
  444. return 0;
  445. }
  446. f64 exact_value_to_f64(ExactValue v) {
  447. v = exact_value_to_float(v);
  448. if (v.kind == ExactValue_Float) {
  449. return v.value_float;
  450. }
  451. return 0.0;
  452. }
  453. ExactValue exact_unary_operator_value(TokenKind op, ExactValue v, i32 precision, bool is_unsigned) {
  454. switch (op) {
  455. case Token_Add: {
  456. switch (v.kind) {
  457. case ExactValue_Invalid:
  458. case ExactValue_Integer:
  459. case ExactValue_Float:
  460. case ExactValue_Complex:
  461. case ExactValue_Quaternion:
  462. return v;
  463. }
  464. break;
  465. }
  466. case Token_Sub: {
  467. switch (v.kind) {
  468. case ExactValue_Invalid:
  469. return v;
  470. case ExactValue_Integer: {
  471. ExactValue i = {ExactValue_Integer};
  472. big_int_neg(&i.value_integer, &v.value_integer);
  473. return i;
  474. }
  475. case ExactValue_Float: {
  476. ExactValue i = v;
  477. i.value_float = -i.value_float;
  478. return i;
  479. }
  480. case ExactValue_Complex: {
  481. f64 real = v.value_complex->real;
  482. f64 imag = v.value_complex->imag;
  483. return exact_value_complex(-real, -imag);
  484. }
  485. case ExactValue_Quaternion: {
  486. f64 real = v.value_quaternion->real;
  487. f64 imag = v.value_quaternion->imag;
  488. f64 jmag = v.value_quaternion->jmag;
  489. f64 kmag = v.value_quaternion->kmag;
  490. return exact_value_quaternion(-real, -imag, -jmag, -kmag);
  491. }
  492. }
  493. break;
  494. }
  495. case Token_Xor: {
  496. switch (v.kind) {
  497. case ExactValue_Invalid:
  498. return v;
  499. case ExactValue_Integer: {
  500. GB_ASSERT(precision != 0);
  501. ExactValue i = {ExactValue_Integer};
  502. big_int_not(&i.value_integer, &v.value_integer, precision, !is_unsigned);
  503. return i;
  504. }
  505. default:
  506. goto failure;
  507. }
  508. }
  509. case Token_Not: {
  510. switch (v.kind) {
  511. case ExactValue_Invalid: return v;
  512. case ExactValue_Bool:
  513. return exact_value_bool(!v.value_bool);
  514. }
  515. break;
  516. }
  517. }
  518. failure:
  519. GB_PANIC("Invalid unary operation, %.*s", LIT(token_strings[op]));
  520. ExactValue error_value = {};
  521. return error_value;
  522. }
  523. // NOTE(bill): Make sure things are evaluated in correct order
  524. i32 exact_value_order(ExactValue const &v) {
  525. switch (v.kind) {
  526. case ExactValue_Invalid:
  527. return 0;
  528. case ExactValue_Bool:
  529. case ExactValue_String:
  530. return 1;
  531. case ExactValue_Integer:
  532. return 2;
  533. case ExactValue_Float:
  534. return 3;
  535. case ExactValue_Complex:
  536. return 4;
  537. case ExactValue_Quaternion:
  538. return 5;
  539. case ExactValue_Pointer:
  540. return 6;
  541. case ExactValue_Procedure:
  542. return 7;
  543. // case ExactValue_Compound:
  544. // return 8;
  545. default:
  546. GB_PANIC("How'd you get here? Invalid Value.kind %d", v.kind);
  547. return -1;
  548. }
  549. }
  550. void match_exact_values(ExactValue *x, ExactValue *y) {
  551. if (exact_value_order(*y) < exact_value_order(*x)) {
  552. match_exact_values(y, x);
  553. return;
  554. }
  555. switch (x->kind) {
  556. case ExactValue_Invalid:
  557. *y = *x;
  558. return;
  559. case ExactValue_Bool:
  560. case ExactValue_String:
  561. case ExactValue_Quaternion:
  562. return;
  563. case ExactValue_Integer:
  564. switch (y->kind) {
  565. case ExactValue_Integer:
  566. return;
  567. case ExactValue_Float:
  568. // TODO(bill): Is this good enough?
  569. *x = exact_value_float(big_int_to_f64(&x->value_integer));
  570. return;
  571. case ExactValue_Complex:
  572. *x = exact_value_complex(big_int_to_f64(&x->value_integer), 0);
  573. return;
  574. case ExactValue_Quaternion:
  575. *x = exact_value_quaternion(big_int_to_f64(&x->value_integer), 0, 0, 0);
  576. return;
  577. }
  578. break;
  579. case ExactValue_Float:
  580. switch (y->kind) {
  581. case ExactValue_Float:
  582. return;
  583. case ExactValue_Complex:
  584. *x = exact_value_to_complex(*x);
  585. return;
  586. case ExactValue_Quaternion:
  587. *x = exact_value_to_quaternion(*x);
  588. return;
  589. }
  590. break;
  591. case ExactValue_Complex:
  592. switch (y->kind) {
  593. case ExactValue_Complex:
  594. return;
  595. case ExactValue_Quaternion:
  596. *x = exact_value_to_quaternion(*x);
  597. return;
  598. }
  599. break;
  600. case ExactValue_Procedure:
  601. return;
  602. }
  603. compiler_error("match_exact_values: How'd you get here? Invalid ExactValueKind %d", x->kind);
  604. }
  605. // TODO(bill): Allow for pointer arithmetic? Or are pointer slices good enough?
  606. ExactValue exact_binary_operator_value(TokenKind op, ExactValue x, ExactValue y) {
  607. match_exact_values(&x, &y);
  608. switch (x.kind) {
  609. case ExactValue_Invalid:
  610. return x;
  611. case ExactValue_Bool:
  612. switch (op) {
  613. case Token_CmpAnd: return exact_value_bool(x.value_bool && y.value_bool);
  614. case Token_CmpOr: return exact_value_bool(x.value_bool || y.value_bool);
  615. case Token_And: return exact_value_bool(x.value_bool & y.value_bool);
  616. case Token_Or: return exact_value_bool(x.value_bool | y.value_bool);
  617. case Token_AndNot: return exact_value_bool(x.value_bool & !y.value_bool);
  618. case Token_Xor: return exact_value_bool((x.value_bool && !y.value_bool) || (!x.value_bool && y.value_bool));
  619. default: goto error;
  620. }
  621. break;
  622. case ExactValue_Integer: {
  623. BigInt const *a = &x.value_integer;
  624. BigInt const *b = &y.value_integer;
  625. BigInt c = {};
  626. switch (op) {
  627. case Token_Add: big_int_add(&c, a, b); break;
  628. case Token_Sub: big_int_sub(&c, a, b); break;
  629. case Token_Mul: big_int_mul(&c, a, b); break;
  630. case Token_Quo: return exact_value_float(fmod(big_int_to_f64(a), big_int_to_f64(b)));
  631. case Token_QuoEq: big_int_quo(&c, a, b); break; // NOTE(bill): Integer division
  632. case Token_Mod: big_int_rem(&c, a, b); break;
  633. case Token_ModMod: big_int_euclidean_mod(&c, a, b); break;
  634. case Token_And: big_int_and(&c, a, b); break;
  635. case Token_Or: big_int_or(&c, a, b); break;
  636. case Token_Xor: big_int_xor(&c, a, b); break;
  637. case Token_AndNot: big_int_and_not(&c, a, b); break;
  638. case Token_Shl: big_int_shl(&c, a, b); break;
  639. case Token_Shr: big_int_shr(&c, a, b); break;
  640. default: goto error;
  641. }
  642. ExactValue res = {ExactValue_Integer};
  643. res.value_integer = c;
  644. return res;
  645. }
  646. case ExactValue_Float: {
  647. f64 a = x.value_float;
  648. f64 b = y.value_float;
  649. switch (op) {
  650. case Token_Add: return exact_value_float(a + b);
  651. case Token_Sub: return exact_value_float(a - b);
  652. case Token_Mul: return exact_value_float(a * b);
  653. case Token_Quo: return exact_value_float(a / b);
  654. default: goto error;
  655. }
  656. break;
  657. }
  658. case ExactValue_Complex: {
  659. y = exact_value_to_complex(y);
  660. f64 a = x.value_complex->real;
  661. f64 b = x.value_complex->imag;
  662. f64 c = y.value_complex->real;
  663. f64 d = y.value_complex->imag;
  664. f64 real = 0;
  665. f64 imag = 0;
  666. switch (op) {
  667. case Token_Add:
  668. real = a + c;
  669. imag = b + d;
  670. break;
  671. case Token_Sub:
  672. real = a - c;
  673. imag = b - d;
  674. break;
  675. case Token_Mul:
  676. real = (a*c - b*d);
  677. imag = (b*c + a*d);
  678. break;
  679. case Token_Quo: {
  680. f64 s = c*c + d*d;
  681. real = (a*c + b*d)/s;
  682. imag = (b*c - a*d)/s;
  683. break;
  684. }
  685. default: goto error;
  686. }
  687. return exact_value_complex(real, imag);
  688. break;
  689. }
  690. case ExactValue_Quaternion: {
  691. y = exact_value_to_quaternion(y);
  692. f64 xr = x.value_quaternion->real;
  693. f64 xi = x.value_quaternion->imag;
  694. f64 xj = x.value_quaternion->jmag;
  695. f64 xk = x.value_quaternion->kmag;
  696. f64 yr = y.value_quaternion->real;
  697. f64 yi = y.value_quaternion->imag;
  698. f64 yj = y.value_quaternion->jmag;
  699. f64 yk = y.value_quaternion->kmag;
  700. f64 real = 0;
  701. f64 imag = 0;
  702. f64 jmag = 0;
  703. f64 kmag = 0;
  704. switch (op) {
  705. case Token_Add:
  706. real = xr + yr;
  707. imag = xi + yi;
  708. jmag = xj + yj;
  709. kmag = xk + yk;
  710. break;
  711. case Token_Sub:
  712. real = xr - yr;
  713. imag = xi - yi;
  714. jmag = xj - yj;
  715. kmag = xk - yk;
  716. break;
  717. case Token_Mul:
  718. imag = xr * yi + xi * yr + xj * yk - xk * yj;
  719. jmag = xr * yj - xi * yk + xj * yr + xk * yi;
  720. kmag = xr * yk + xi * yj - xj * yi + xk * yr;
  721. real = xr * yr - xi * yi - xj * yj - xk * yk;
  722. break;
  723. case Token_Quo: {
  724. f64 invmag2 = 1.0 / (yr*yr + yi*yi + yj*yj + yk*yk);
  725. imag = (xr * -yi + xi * +yr + xj * -yk - xk * -yj) * invmag2;
  726. jmag = (xr * -yj - xi * -yk + xj * +yr + xk * -yi) * invmag2;
  727. kmag = (xr * -yk + xi * -yj - xj * -yi + xk * +yr) * invmag2;
  728. real = (xr * +yr - xi * -yi - xj * -yj - xk * -yk) * invmag2;
  729. break;
  730. }
  731. default: goto error;
  732. }
  733. return exact_value_quaternion(real, imag, jmag, kmag);
  734. break;
  735. }
  736. case ExactValue_String: {
  737. if (op != Token_Add) goto error;
  738. // NOTE(bill): How do you minimize this over allocation?
  739. String sx = x.value_string;
  740. String sy = y.value_string;
  741. isize len = sx.len+sy.len;
  742. u8 *data = gb_alloc_array(permanent_allocator(), u8, len);
  743. gb_memmove(data, sx.text, sx.len);
  744. gb_memmove(data+sx.len, sy.text, sy.len);
  745. return exact_value_string(make_string(data, len));
  746. break;
  747. }
  748. }
  749. error:; // NOTE(bill): MSVC accepts this??? apparently you cannot declare variables immediately after labels...
  750. return empty_exact_value;
  751. }
  752. gb_inline ExactValue exact_value_add(ExactValue const &x, ExactValue const &y) {
  753. return exact_binary_operator_value(Token_Add, x, y);
  754. }
  755. gb_inline ExactValue exact_value_sub(ExactValue const &x, ExactValue const &y) {
  756. return exact_binary_operator_value(Token_Sub, x, y);
  757. }
  758. gb_inline ExactValue exact_value_mul(ExactValue const &x, ExactValue const &y) {
  759. return exact_binary_operator_value(Token_Mul, x, y);
  760. }
  761. gb_inline ExactValue exact_value_quo(ExactValue const &x, ExactValue const &y) {
  762. return exact_binary_operator_value(Token_Quo, x, y);
  763. }
  764. gb_inline ExactValue exact_value_shift(TokenKind op, ExactValue const &x, ExactValue const &y) {
  765. return exact_binary_operator_value(op, x, y);
  766. }
  767. gb_inline ExactValue exact_value_increment_one(ExactValue const &x) {
  768. return exact_binary_operator_value(Token_Add, x, exact_value_i64(1));
  769. }
  770. i32 cmp_f64(f64 a, f64 b) {
  771. return (a > b) - (a < b);
  772. }
  773. bool compare_exact_values(TokenKind op, ExactValue x, ExactValue y) {
  774. match_exact_values(&x, &y);
  775. switch (x.kind) {
  776. case ExactValue_Invalid:
  777. return false;
  778. case ExactValue_Bool:
  779. switch (op) {
  780. case Token_CmpEq: return x.value_bool == y.value_bool;
  781. case Token_NotEq: return x.value_bool != y.value_bool;
  782. }
  783. break;
  784. case ExactValue_Integer: {
  785. i32 cmp = big_int_cmp(&x.value_integer, &y.value_integer);
  786. switch (op) {
  787. case Token_CmpEq: return cmp == 0;
  788. case Token_NotEq: return cmp != 0;
  789. case Token_Lt: return cmp < 0;
  790. case Token_LtEq: return cmp <= 0;
  791. case Token_Gt: return cmp > 0;
  792. case Token_GtEq: return cmp >= 0;
  793. }
  794. break;
  795. }
  796. case ExactValue_Float: {
  797. f64 a = x.value_float;
  798. f64 b = y.value_float;
  799. switch (op) {
  800. case Token_CmpEq: return cmp_f64(a, b) == 0;
  801. case Token_NotEq: return cmp_f64(a, b) != 0;
  802. case Token_Lt: return cmp_f64(a, b) < 0;
  803. case Token_LtEq: return cmp_f64(a, b) <= 0;
  804. case Token_Gt: return cmp_f64(a, b) > 0;
  805. case Token_GtEq: return cmp_f64(a, b) >= 0;
  806. }
  807. break;
  808. }
  809. case ExactValue_Complex: {
  810. f64 a = x.value_complex->real;
  811. f64 b = x.value_complex->imag;
  812. f64 c = y.value_complex->real;
  813. f64 d = y.value_complex->imag;
  814. switch (op) {
  815. case Token_CmpEq: return cmp_f64(a, c) == 0 && cmp_f64(b, d) == 0;
  816. case Token_NotEq: return cmp_f64(a, c) != 0 || cmp_f64(b, d) != 0;
  817. }
  818. break;
  819. }
  820. case ExactValue_String: {
  821. String a = x.value_string;
  822. String b = y.value_string;
  823. // TODO(bill): gb_memcompare is used because the strings are UTF-8
  824. switch (op) {
  825. case Token_CmpEq: return a == b;
  826. case Token_NotEq: return a != b;
  827. case Token_Lt: return a < b;
  828. case Token_LtEq: return a <= b;
  829. case Token_Gt: return a > b;
  830. case Token_GtEq: return a >= b;
  831. }
  832. break;
  833. }
  834. case ExactValue_Typeid:
  835. switch (op) {
  836. case Token_CmpEq: return are_types_identical(x.value_typeid, y.value_typeid);
  837. case Token_NotEq: return !are_types_identical(x.value_typeid, y.value_typeid);
  838. }
  839. break;
  840. case ExactValue_Procedure:
  841. switch (op) {
  842. case Token_CmpEq: return are_types_identical(x.value_typeid, y.value_typeid);
  843. case Token_NotEq: return !are_types_identical(x.value_typeid, y.value_typeid);
  844. }
  845. break;
  846. }
  847. GB_PANIC("Invalid comparison");
  848. return false;
  849. }
  850. Entity *strip_entity_wrapping(Ast *expr);
  851. Entity *strip_entity_wrapping(Entity *e);
  852. gbString write_expr_to_string(gbString str, Ast *node, bool shorthand);
  853. gbString write_exact_value_to_string(gbString str, ExactValue const &v, isize string_limit=36) {
  854. switch (v.kind) {
  855. case ExactValue_Invalid:
  856. return str;
  857. case ExactValue_Bool:
  858. return gb_string_appendc(str, v.value_bool ? "true" : "false");
  859. case ExactValue_String: {
  860. String s = quote_to_ascii(heap_allocator(), v.value_string);
  861. string_limit = gb_max(string_limit, 36);
  862. if (s.len <= string_limit) {
  863. str = gb_string_append_length(str, s.text, s.len);
  864. } else {
  865. isize n = string_limit/5;
  866. str = gb_string_append_length(str, s.text, n);
  867. str = gb_string_append_fmt(str, "\"..%lld chars..\"", s.len-(2*n));
  868. str = gb_string_append_length(str, s.text+s.len-n, n);
  869. }
  870. gb_free(heap_allocator(), s.text);
  871. return str;
  872. }
  873. case ExactValue_Integer: {
  874. String s = big_int_to_string(heap_allocator(), &v.value_integer);
  875. str = gb_string_append_length(str, s.text, s.len);
  876. gb_free(heap_allocator(), s.text);
  877. return str;
  878. }
  879. case ExactValue_Float:
  880. return gb_string_append_fmt(str, "%f", v.value_float);
  881. case ExactValue_Complex:
  882. return gb_string_append_fmt(str, "%f+%fi", v.value_complex->real, v.value_complex->imag);
  883. case ExactValue_Quaternion:
  884. return gb_string_append_fmt(str, "%f+%fi+%fj+%fk", v.value_quaternion->real, v.value_quaternion->imag, v.value_quaternion->jmag, v.value_quaternion->kmag);
  885. case ExactValue_Pointer:
  886. return str;
  887. case ExactValue_Compound:
  888. return write_expr_to_string(str, v.value_compound, false);
  889. case ExactValue_Procedure:
  890. return write_expr_to_string(str, v.value_procedure, false);
  891. }
  892. return str;
  893. };
  894. gbString exact_value_to_string(ExactValue const &v, isize string_limit=36) {
  895. return write_exact_value_to_string(gb_string_make(heap_allocator(), ""), v, string_limit);
  896. }