tokenizer.cpp 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836
  1. #define TOKEN_KINDS \
  2. TOKEN_KIND(Token_Invalid, "Invalid"), \
  3. TOKEN_KIND(Token_EOF, "EOF"), \
  4. \
  5. TOKEN_KIND(Token__LiteralBegin, "_LiteralBegin"), \
  6. TOKEN_KIND(Token_Identifier, "Identifier"), \
  7. TOKEN_KIND(Token_Integer, "Integer"), \
  8. TOKEN_KIND(Token_Float, "Float"), \
  9. TOKEN_KIND(Token_Rune, "Rune"), \
  10. TOKEN_KIND(Token_String, "String"), \
  11. TOKEN_KIND(Token__LiteralEnd, "_LiteralEnd"), \
  12. \
  13. TOKEN_KIND(Token__OperatorBegin, "_OperatorBegin"), \
  14. TOKEN_KIND(Token_Eq, "="), \
  15. TOKEN_KIND(Token_Not, "!"), \
  16. TOKEN_KIND(Token_Hash, "#"), \
  17. TOKEN_KIND(Token_At, "@"), \
  18. TOKEN_KIND(Token_Pointer, "^"), \
  19. TOKEN_KIND(Token_Add, "+"), \
  20. TOKEN_KIND(Token_Sub, "-"), \
  21. TOKEN_KIND(Token_Mul, "*"), \
  22. TOKEN_KIND(Token_Quo, "/"), \
  23. TOKEN_KIND(Token_Mod, "%"), \
  24. TOKEN_KIND(Token_And, "&"), \
  25. TOKEN_KIND(Token_Or, "|"), \
  26. TOKEN_KIND(Token_Xor, "~"), \
  27. TOKEN_KIND(Token_AndNot, "&~"), \
  28. TOKEN_KIND(Token_Shl, "<<"), \
  29. TOKEN_KIND(Token_Shr, ">>"), \
  30. \
  31. TOKEN_KIND(Token_as, "as"), \
  32. TOKEN_KIND(Token_transmute, "transmute"), \
  33. TOKEN_KIND(Token_down_cast, "down_cast"), \
  34. \
  35. TOKEN_KIND(Token_Prime, "'"), \
  36. TOKEN_KIND(Token_DoublePrime, "''"), \
  37. \
  38. TOKEN_KIND(Token__AssignOpBegin, "_AssignOpBegin"), \
  39. TOKEN_KIND(Token_AddEq, "+="), \
  40. TOKEN_KIND(Token_SubEq, "-="), \
  41. TOKEN_KIND(Token_MulEq, "*="), \
  42. TOKEN_KIND(Token_QuoEq, "/="), \
  43. TOKEN_KIND(Token_ModEq, "%="), \
  44. TOKEN_KIND(Token_AndEq, "&="), \
  45. TOKEN_KIND(Token_OrEq, "|="), \
  46. TOKEN_KIND(Token_XorEq, "~="), \
  47. TOKEN_KIND(Token_AndNotEq, "&~="), \
  48. TOKEN_KIND(Token_ShlEq, "<<="), \
  49. TOKEN_KIND(Token_ShrEq, ">>="), \
  50. TOKEN_KIND(Token__AssignOpEnd, "_AssignOpEnd"), \
  51. TOKEN_KIND(Token_Increment, "++"), \
  52. TOKEN_KIND(Token_Decrement, "--"), \
  53. TOKEN_KIND(Token_ArrowRight, "->"), \
  54. TOKEN_KIND(Token_ArrowLeft, "<-"), \
  55. \
  56. TOKEN_KIND(Token_CmpAnd, "&&"), \
  57. TOKEN_KIND(Token_CmpOr, "||"), \
  58. TOKEN_KIND(Token_CmpAndEq, "&&="), \
  59. TOKEN_KIND(Token_CmpOrEq, "||="), \
  60. \
  61. TOKEN_KIND(Token__ComparisonBegin, "_ComparisonBegin"), \
  62. TOKEN_KIND(Token_CmpEq, "=="), \
  63. TOKEN_KIND(Token_NotEq, "!="), \
  64. TOKEN_KIND(Token_Lt, "<"), \
  65. TOKEN_KIND(Token_Gt, ">"), \
  66. TOKEN_KIND(Token_LtEq, "<="), \
  67. TOKEN_KIND(Token_GtEq, ">="), \
  68. TOKEN_KIND(Token__ComparisonEnd, "_ComparisonEnd"), \
  69. \
  70. TOKEN_KIND(Token_OpenParen, "("), \
  71. TOKEN_KIND(Token_CloseParen, ")"), \
  72. TOKEN_KIND(Token_OpenBracket, "["), \
  73. TOKEN_KIND(Token_CloseBracket, "]"), \
  74. TOKEN_KIND(Token_OpenBrace, "{"), \
  75. TOKEN_KIND(Token_CloseBrace, "}"), \
  76. TOKEN_KIND(Token_Colon, ":"), \
  77. TOKEN_KIND(Token_Semicolon, ";"), \
  78. TOKEN_KIND(Token_Period, "."), \
  79. TOKEN_KIND(Token_Comma, ","), \
  80. TOKEN_KIND(Token_Ellipsis, ".."), \
  81. TOKEN_KIND(Token_RangeExclusive, "..<"), \
  82. TOKEN_KIND(Token__OperatorEnd, "_OperatorEnd"), \
  83. \
  84. TOKEN_KIND(Token__KeywordBegin, "_KeywordBegin"), \
  85. TOKEN_KIND(Token_type, "type"), \
  86. TOKEN_KIND(Token_proc, "proc"), \
  87. TOKEN_KIND(Token_match, "match"), \
  88. TOKEN_KIND(Token_break, "break"), \
  89. TOKEN_KIND(Token_continue, "continue"), \
  90. TOKEN_KIND(Token_fallthrough, "fallthrough"), \
  91. TOKEN_KIND(Token_case, "case"), \
  92. TOKEN_KIND(Token_default, "default"), \
  93. TOKEN_KIND(Token_then, "then"), \
  94. TOKEN_KIND(Token_if, "if"), \
  95. TOKEN_KIND(Token_else, "else"), \
  96. TOKEN_KIND(Token_for, "for"), \
  97. TOKEN_KIND(Token_range, "range"), \
  98. TOKEN_KIND(Token_defer, "defer"), \
  99. TOKEN_KIND(Token_return, "return"), \
  100. TOKEN_KIND(Token_struct, "struct"), \
  101. TOKEN_KIND(Token_union, "union"), \
  102. TOKEN_KIND(Token_raw_union, "raw_union"), \
  103. TOKEN_KIND(Token_enum, "enum"), \
  104. TOKEN_KIND(Token_using, "using"), \
  105. TOKEN_KIND(Token_asm, "asm"), \
  106. TOKEN_KIND(Token_volatile, "volatile"), \
  107. TOKEN_KIND(Token_atomic, "atomic"), \
  108. TOKEN_KIND(Token__KeywordEnd, "_KeywordEnd"), \
  109. TOKEN_KIND(Token_Count, "")
  110. enum TokenKind {
  111. #define TOKEN_KIND(e, s) e
  112. TOKEN_KINDS
  113. #undef TOKEN_KIND
  114. };
  115. String const token_strings[] = {
  116. #define TOKEN_KIND(e, s) {cast(u8 *)s, gb_size_of(s)-1}
  117. TOKEN_KINDS
  118. #undef TOKEN_KIND
  119. };
  120. struct TokenPos {
  121. String file;
  122. isize line, column;
  123. };
  124. i32 token_pos_cmp(TokenPos a, TokenPos b) {
  125. if (a.line == b.line) {
  126. if (a.column == b.column) {
  127. isize min_len = gb_min(a.file.len, b.file.len);
  128. return gb_memcompare(a.file.text, b.file.text, min_len);
  129. }
  130. return (a.column < b.column) ? -1 : +1;
  131. }
  132. return (a.line < b.line) ? -1 : +1;
  133. }
  134. b32 token_pos_are_equal(TokenPos a, TokenPos b) {
  135. return token_pos_cmp(a, b) == 0;
  136. }
  137. // NOTE(bill): Text is UTF-8, thus why u8 and not char
  138. struct Token {
  139. TokenKind kind;
  140. String string;
  141. TokenPos pos;
  142. };
  143. Token empty_token = {Token_Invalid};
  144. Token blank_token = {Token_Identifier, {cast(u8 *)"_", 1}};
  145. Token make_token_ident(String s) {
  146. Token t = {Token_Identifier};
  147. t.string = s;
  148. return t;
  149. }
  150. struct ErrorCollector {
  151. TokenPos prev;
  152. i64 count;
  153. i64 warning_count;
  154. };
  155. gb_global ErrorCollector global_error_collector;
  156. void warning(Token token, char *fmt, ...) {
  157. global_error_collector.warning_count++;
  158. // NOTE(bill): Duplicate error, skip it
  159. if (!token_pos_are_equal(global_error_collector.prev, token.pos)) {
  160. va_list va;
  161. global_error_collector.prev = token.pos;
  162. va_start(va, fmt);
  163. gb_printf_err("%.*s(%td:%td) Warning: %s\n",
  164. LIT(token.pos.file), token.pos.line, token.pos.column,
  165. gb_bprintf_va(fmt, va));
  166. va_end(va);
  167. }
  168. }
  169. void error(Token token, char *fmt, ...) {
  170. global_error_collector.count++;
  171. // NOTE(bill): Duplicate error, skip it
  172. if (!token_pos_are_equal(global_error_collector.prev, token.pos)) {
  173. va_list va;
  174. global_error_collector.prev = token.pos;
  175. va_start(va, fmt);
  176. gb_printf_err("%.*s(%td:%td) %s\n",
  177. LIT(token.pos.file), token.pos.line, token.pos.column,
  178. gb_bprintf_va(fmt, va));
  179. va_end(va);
  180. }
  181. }
  182. void syntax_error(Token token, char *fmt, ...) {
  183. global_error_collector.count++;
  184. // NOTE(bill): Duplicate error, skip it
  185. if (!token_pos_are_equal(global_error_collector.prev, token.pos)) {
  186. va_list va;
  187. global_error_collector.prev = token.pos;
  188. va_start(va, fmt);
  189. gb_printf_err("%.*s(%td:%td) Syntax Error: %s\n",
  190. LIT(token.pos.file), token.pos.line, token.pos.column,
  191. gb_bprintf_va(fmt, va));
  192. va_end(va);
  193. }
  194. }
  195. void compiler_error(char *fmt, ...) {
  196. va_list va;
  197. va_start(va, fmt);
  198. gb_printf_err("Internal Compiler Error: %s\n",
  199. gb_bprintf_va(fmt, va));
  200. va_end(va);
  201. gb_exit(1);
  202. }
  203. // NOTE(bill): result == priority
  204. i32 token_precedence(Token t) {
  205. switch (t.kind) {
  206. case Token_CmpOr:
  207. return 1;
  208. case Token_CmpAnd:
  209. return 2;
  210. case Token_CmpEq:
  211. case Token_NotEq:
  212. case Token_Lt:
  213. case Token_Gt:
  214. case Token_LtEq:
  215. case Token_GtEq:
  216. return 3;
  217. case Token_Add:
  218. case Token_Sub:
  219. case Token_Or:
  220. case Token_Xor:
  221. return 4;
  222. case Token_Mul:
  223. case Token_Quo:
  224. case Token_Mod:
  225. case Token_And:
  226. case Token_AndNot:
  227. case Token_Shl:
  228. case Token_Shr:
  229. return 5;
  230. case Token_DoublePrime:
  231. return 6;
  232. case Token_as:
  233. case Token_transmute:
  234. case Token_down_cast:
  235. return 7;
  236. }
  237. return 0;
  238. }
  239. gb_inline b32 token_is_literal(Token t) {
  240. return gb_is_between(t.kind, Token__LiteralBegin+1, Token__LiteralEnd-1);
  241. }
  242. gb_inline b32 token_is_operator(Token t) {
  243. return gb_is_between(t.kind, Token__OperatorBegin+1, Token__OperatorEnd-1);
  244. }
  245. gb_inline b32 token_is_keyword(Token t) {
  246. return gb_is_between(t.kind, Token__KeywordBegin+1, Token__KeywordEnd-1);
  247. }
  248. gb_inline b32 token_is_comparison(Token t) {
  249. return gb_is_between(t.kind, Token__ComparisonBegin+1, Token__ComparisonEnd-1);
  250. }
  251. gb_inline b32 token_is_shift(Token t) {
  252. return t.kind == Token_Shl || t.kind == Token_Shr;
  253. }
  254. gb_inline void print_token(Token t) { gb_printf("%.*s\n", LIT(t.string)); }
  255. enum TokenizerInitError {
  256. TokenizerInit_None,
  257. TokenizerInit_Invalid,
  258. TokenizerInit_NotExists,
  259. TokenizerInit_Permission,
  260. TokenizerInit_Empty,
  261. TokenizerInit_Count,
  262. };
  263. struct Tokenizer {
  264. String fullpath;
  265. u8 *start;
  266. u8 *end;
  267. Rune curr_rune; // current character
  268. u8 * curr; // character pos
  269. u8 * read_curr; // pos from start
  270. u8 * line; // current line pos
  271. isize line_count;
  272. isize error_count;
  273. gbArray(String) allocated_strings;
  274. };
  275. #define tokenizer_err(t, msg, ...) tokenizer_err_(t, __FUNCTION__, msg, ##__VA_ARGS__)
  276. void tokenizer_err_(Tokenizer *t, char *function, char *msg, ...) {
  277. va_list va;
  278. isize column = t->read_curr - t->line+1;
  279. if (column < 1)
  280. column = 1;
  281. gb_printf_err("%.*s(%td:%td) Syntax error: ", LIT(t->fullpath), t->line_count, column);
  282. va_start(va, msg);
  283. gb_printf_err_va(msg, va);
  284. va_end(va);
  285. gb_printf_err("\n");
  286. t->error_count++;
  287. }
  288. void advance_to_next_rune(Tokenizer *t) {
  289. if (t->read_curr < t->end) {
  290. Rune rune;
  291. isize width = 1;
  292. t->curr = t->read_curr;
  293. if (t->curr_rune == '\n') {
  294. t->line = t->curr;
  295. t->line_count++;
  296. }
  297. rune = *t->read_curr;
  298. if (rune == 0) {
  299. tokenizer_err(t, "Illegal character NUL");
  300. } else if (rune >= 0x80) { // not ASCII
  301. width = gb_utf8_decode(t->read_curr, t->end-t->read_curr, &rune);
  302. if (rune == GB_RUNE_INVALID && width == 1)
  303. tokenizer_err(t, "Illegal UTF-8 encoding");
  304. else if (rune == GB_RUNE_BOM && t->curr-t->start > 0)
  305. tokenizer_err(t, "Illegal byte order mark");
  306. }
  307. t->read_curr += width;
  308. t->curr_rune = rune;
  309. } else {
  310. t->curr = t->end;
  311. if (t->curr_rune == '\n') {
  312. t->line = t->curr;
  313. t->line_count++;
  314. }
  315. t->curr_rune = GB_RUNE_EOF;
  316. }
  317. }
  318. TokenizerInitError init_tokenizer(Tokenizer *t, String fullpath) {
  319. char *c_str = gb_alloc_array(gb_heap_allocator(), char, fullpath.len+1);
  320. memcpy(c_str, fullpath.text, fullpath.len);
  321. c_str[fullpath.len] = '\0';
  322. defer (gb_free(gb_heap_allocator(), c_str));
  323. gbFileContents fc = gb_file_read_contents(gb_heap_allocator(), true, c_str);
  324. gb_zero_item(t);
  325. if (fc.data != NULL) {
  326. t->start = cast(u8 *)fc.data;
  327. t->line = t->read_curr = t->curr = t->start;
  328. t->end = t->start + fc.size;
  329. t->fullpath = fullpath;
  330. t->line_count = 1;
  331. advance_to_next_rune(t);
  332. if (t->curr_rune == GB_RUNE_BOM)
  333. advance_to_next_rune(t); // Ignore BOM at file beginning
  334. gb_array_init(t->allocated_strings, gb_heap_allocator());
  335. return TokenizerInit_None;
  336. }
  337. gbFile f = {};
  338. gbFileError err = gb_file_open(&f, c_str);
  339. defer (gb_file_close(&f));
  340. switch (err) {
  341. case gbFileError_Invalid:
  342. return TokenizerInit_Invalid;
  343. case gbFileError_NotExists:
  344. return TokenizerInit_NotExists;
  345. case gbFileError_Permission:
  346. return TokenizerInit_Permission;
  347. }
  348. if (gb_file_size(&f) == 0)
  349. return TokenizerInit_Empty;
  350. return TokenizerInit_None;
  351. }
  352. gb_inline void destroy_tokenizer(Tokenizer *t) {
  353. if (t->start != NULL) {
  354. gb_free(gb_heap_allocator(), t->start);
  355. }
  356. if (t->allocated_strings != NULL) {
  357. gb_for_array(i, t->allocated_strings) {
  358. gb_free(gb_heap_allocator(), t->allocated_strings[i].text);
  359. }
  360. gb_array_free(t->allocated_strings);
  361. }
  362. }
  363. void tokenizer_skip_whitespace(Tokenizer *t) {
  364. for (;;) {
  365. if (rune_is_whitespace(t->curr_rune)) {
  366. advance_to_next_rune(t);
  367. } else if (t->curr_rune == '/') {
  368. if (t->read_curr[0] == '/') { // Line comment //
  369. while (t->curr_rune != '\n') {
  370. advance_to_next_rune(t);
  371. }
  372. } else if (t->read_curr[0] == '*') { // (Nested) Block comment /**/
  373. advance_to_next_rune(t);
  374. advance_to_next_rune(t);
  375. isize comment_scope = 1;
  376. while (comment_scope > 0) {
  377. if (t->curr_rune == '/') {
  378. advance_to_next_rune(t);
  379. if (t->curr_rune == '*') {
  380. advance_to_next_rune(t);
  381. comment_scope++;
  382. }
  383. } else if (t->curr_rune == '*') {
  384. advance_to_next_rune(t);
  385. if (t->curr_rune == '/') {
  386. advance_to_next_rune(t);
  387. comment_scope--;
  388. }
  389. } else {
  390. advance_to_next_rune(t);
  391. }
  392. }
  393. } else {
  394. break;
  395. }
  396. } else {
  397. break;
  398. }
  399. }
  400. }
  401. gb_inline i32 digit_value(Rune r) {
  402. if (gb_char_is_digit(cast(char)r))
  403. return r - '0';
  404. if (gb_is_between(cast(char)r, 'a', 'f'))
  405. return r - 'a' + 10;
  406. if (gb_is_between(cast(char)r, 'A', 'F'))
  407. return r - 'A' + 10;
  408. return 16; // NOTE(bill): Larger than highest possible
  409. }
  410. gb_inline void scan_mantissa(Tokenizer *t, i32 base) {
  411. // TODO(bill): Allow for underscores in numbers as a number separator
  412. // TODO(bill): Is this a good idea?
  413. // while (digit_value(t->curr_rune) < base || t->curr_rune == '_')
  414. while (digit_value(t->curr_rune) < base)
  415. advance_to_next_rune(t);
  416. }
  417. Token scan_number_to_token(Tokenizer *t, b32 seen_decimal_point) {
  418. Token token = {};
  419. u8 *start_curr = t->curr;
  420. token.kind = Token_Integer;
  421. token.string = make_string(start_curr, 1);
  422. token.pos.file = t->fullpath;
  423. token.pos.line = t->line_count;
  424. token.pos.column = t->curr-t->line+1;
  425. if (seen_decimal_point) {
  426. start_curr--;
  427. token.kind = Token_Float;
  428. scan_mantissa(t, 10);
  429. goto exponent;
  430. }
  431. if (t->curr_rune == '0') {
  432. u8 *prev = t->curr;
  433. advance_to_next_rune(t);
  434. if (t->curr_rune == 'b') { // Binary
  435. advance_to_next_rune(t);
  436. scan_mantissa(t, 2);
  437. if (t->curr - prev <= 2)
  438. token.kind = Token_Invalid;
  439. } else if (t->curr_rune == 'o') { // Octal
  440. advance_to_next_rune(t);
  441. scan_mantissa(t, 8);
  442. if (t->curr - prev <= 2)
  443. token.kind = Token_Invalid;
  444. } else if (t->curr_rune == 'd') { // Decimal
  445. advance_to_next_rune(t);
  446. scan_mantissa(t, 10);
  447. if (t->curr - prev <= 2)
  448. token.kind = Token_Invalid;
  449. } else if (t->curr_rune == 'x') { // Hexadecimal
  450. advance_to_next_rune(t);
  451. scan_mantissa(t, 16);
  452. if (t->curr - prev <= 2)
  453. token.kind = Token_Invalid;
  454. } else {
  455. seen_decimal_point = false;
  456. scan_mantissa(t, 10);
  457. if (t->curr_rune == '.' || t->curr_rune == 'e' || t->curr_rune == 'E') {
  458. seen_decimal_point = true;
  459. goto fraction;
  460. }
  461. }
  462. token.string.len = t->curr - token.string.text;
  463. return token;
  464. }
  465. scan_mantissa(t, 10);
  466. fraction:
  467. if (t->curr_rune == '.') {
  468. token.kind = Token_Float;
  469. advance_to_next_rune(t);
  470. scan_mantissa(t, 10);
  471. }
  472. exponent:
  473. if (t->curr_rune == 'e' || t->curr_rune == 'E') {
  474. token.kind = Token_Float;
  475. advance_to_next_rune(t);
  476. if (t->curr_rune == '-' || t->curr_rune == '+')
  477. advance_to_next_rune(t);
  478. scan_mantissa(t, 10);
  479. }
  480. token.string.len = t->curr - token.string.text;
  481. return token;
  482. }
  483. // Quote == " for string
  484. b32 scan_escape(Tokenizer *t, Rune quote) {
  485. isize len = 0;
  486. u32 base = 0, max = 0, x = 0;
  487. Rune r = t->curr_rune;
  488. if (r == 'a' ||
  489. r == 'b' ||
  490. r == 'f' ||
  491. r == 'n' ||
  492. r == 'r' ||
  493. r == 't' ||
  494. r == 'v' ||
  495. r == '\\' ||
  496. r == quote) {
  497. advance_to_next_rune(t);
  498. return true;
  499. } else if (gb_is_between(r, '0', '7')) {
  500. len = 3; base = 8; max = 255;
  501. } else if (r == 'x') {
  502. advance_to_next_rune(t);
  503. len = 2; base = 16; max = 255;
  504. } else if (r == 'u') {
  505. advance_to_next_rune(t);
  506. len = 4; base = 16; max = GB_RUNE_MAX;
  507. } else if (r == 'U') {
  508. advance_to_next_rune(t);
  509. len = 8; base = 16; max = GB_RUNE_MAX;
  510. } else {
  511. if (t->curr_rune < 0)
  512. tokenizer_err(t, "Escape sequence was not terminated");
  513. else
  514. tokenizer_err(t, "Unknown escape sequence");
  515. return false;
  516. }
  517. while (len --> 0) {
  518. u32 d = cast(u32)digit_value(t->curr_rune);
  519. if (d >= base) {
  520. if (t->curr_rune < 0)
  521. tokenizer_err(t, "Escape sequence was not terminated");
  522. else
  523. tokenizer_err(t, "Illegal character %d in escape sequence", t->curr_rune);
  524. return false;
  525. }
  526. x = x*base + d;
  527. advance_to_next_rune(t);
  528. }
  529. return true;
  530. }
  531. gb_inline TokenKind token_kind_variant2(Tokenizer *t, TokenKind a, TokenKind b) {
  532. if (t->curr_rune == '=') {
  533. advance_to_next_rune(t);
  534. return b;
  535. }
  536. return a;
  537. }
  538. gb_inline TokenKind token_kind_variant3(Tokenizer *t, TokenKind a, TokenKind b, Rune ch_c, TokenKind c) {
  539. if (t->curr_rune == '=') {
  540. advance_to_next_rune(t);
  541. return b;
  542. }
  543. if (t->curr_rune == ch_c) {
  544. advance_to_next_rune(t);
  545. return c;
  546. }
  547. return a;
  548. }
  549. gb_inline TokenKind token_kind_variant4(Tokenizer *t, TokenKind a, TokenKind b, Rune ch_c, TokenKind c, Rune ch_d, TokenKind d) {
  550. if (t->curr_rune == '=') {
  551. advance_to_next_rune(t);
  552. return b;
  553. } else if (t->curr_rune == ch_c) {
  554. advance_to_next_rune(t);
  555. return c;
  556. } else if (t->curr_rune == ch_d) {
  557. advance_to_next_rune(t);
  558. return d;
  559. }
  560. return a;
  561. }
  562. gb_inline TokenKind token_kind_dub_eq(Tokenizer *t, Rune sing_rune, TokenKind sing, TokenKind sing_eq, TokenKind dub, TokenKind dub_eq) {
  563. if (t->curr_rune == '=') {
  564. advance_to_next_rune(t);
  565. return sing_eq;
  566. } else if (t->curr_rune == sing_rune) {
  567. advance_to_next_rune(t);
  568. if (t->curr_rune == '=') {
  569. advance_to_next_rune(t);
  570. return dub_eq;
  571. }
  572. return dub;
  573. }
  574. return sing;
  575. }
  576. Token tokenizer_get_token(Tokenizer *t) {
  577. Token token = {};
  578. Rune curr_rune;
  579. tokenizer_skip_whitespace(t);
  580. token.string = make_string(t->curr, 1);
  581. token.pos.file = t->fullpath;
  582. token.pos.line = t->line_count;
  583. token.pos.column = t->curr - t->line + 1;
  584. curr_rune = t->curr_rune;
  585. if (rune_is_letter(curr_rune)) {
  586. token.kind = Token_Identifier;
  587. while (rune_is_letter(t->curr_rune) || rune_is_digit(t->curr_rune))
  588. advance_to_next_rune(t);
  589. token.string.len = t->curr - token.string.text;
  590. // NOTE(bill): All keywords are > 1
  591. if (token.string.len > 1) {
  592. if (token.string == token_strings[Token_as]) {
  593. token.kind = Token_as;
  594. } else if (token.string == token_strings[Token_transmute]) {
  595. token.kind = Token_transmute;
  596. } else if (token.string == token_strings[Token_down_cast]) {
  597. token.kind = Token_down_cast;
  598. } else {
  599. for (i32 k = Token__KeywordBegin+1; k < Token__KeywordEnd; k++) {
  600. if (token.string == token_strings[k]) {
  601. token.kind = cast(TokenKind)k;
  602. break;
  603. }
  604. }
  605. }
  606. }
  607. } else if (gb_is_between(curr_rune, '0', '9')) {
  608. token = scan_number_to_token(t, false);
  609. } else {
  610. advance_to_next_rune(t);
  611. switch (curr_rune) {
  612. case GB_RUNE_EOF:
  613. token.kind = Token_EOF;
  614. break;
  615. case '\'':
  616. token.kind = Token_Prime;
  617. if (t->curr_rune == '\'') {
  618. advance_to_next_rune(t);
  619. token.kind = Token_DoublePrime;
  620. }
  621. break;
  622. case '`': // Raw String Literal
  623. case '"': // String Literal
  624. {
  625. Rune quote = curr_rune;
  626. token.kind = Token_String;
  627. if (curr_rune == '"') {
  628. for (;;) {
  629. Rune r = t->curr_rune;
  630. if (r == '\n' || r < 0) {
  631. tokenizer_err(t, "String literal not terminated");
  632. break;
  633. }
  634. advance_to_next_rune(t);
  635. if (r == quote)
  636. break;
  637. if (r == '\\')
  638. scan_escape(t, '"');
  639. }
  640. } else {
  641. for (;;) {
  642. Rune r = t->curr_rune;
  643. if (r < 0) {
  644. tokenizer_err(t, "String literal not terminated");
  645. break;
  646. }
  647. advance_to_next_rune(t);
  648. if (r == quote)
  649. break;
  650. }
  651. }
  652. token.string.len = t->curr - token.string.text;
  653. i32 success = unquote_string(gb_heap_allocator(), &token.string);
  654. if (success > 0) {
  655. if (success == 2) {
  656. gb_array_append(t->allocated_strings, token.string);
  657. }
  658. return token;
  659. } else {
  660. tokenizer_err(t, "Invalid string literal");
  661. }
  662. } break;
  663. case '.':
  664. token.kind = Token_Period; // Default
  665. if (gb_is_between(t->curr_rune, '0', '9')) { // Might be a number
  666. token = scan_number_to_token(t, true);
  667. } else if (t->curr_rune == '.') { // Could be an ellipsis
  668. advance_to_next_rune(t);
  669. token.kind = Token_Ellipsis;
  670. if (t->curr_rune == '<') {
  671. advance_to_next_rune(t);
  672. token.kind = Token_RangeExclusive;
  673. }
  674. }
  675. break;
  676. case '#': token.kind = Token_Hash; break;
  677. case '@': token.kind = Token_At; break;
  678. case '^': token.kind = Token_Pointer; break;
  679. case ';': token.kind = Token_Semicolon; break;
  680. case ',': token.kind = Token_Comma; break;
  681. case '(': token.kind = Token_OpenParen; break;
  682. case ')': token.kind = Token_CloseParen; break;
  683. case '[': token.kind = Token_OpenBracket; break;
  684. case ']': token.kind = Token_CloseBracket; break;
  685. case '{': token.kind = Token_OpenBrace; break;
  686. case '}': token.kind = Token_CloseBrace; break;
  687. case ':': token.kind = Token_Colon; break;
  688. case '*': token.kind = token_kind_variant2(t, Token_Mul, Token_MulEq); break;
  689. case '/': token.kind = token_kind_variant2(t, Token_Quo, Token_QuoEq); break;
  690. case '%': token.kind = token_kind_variant2(t, Token_Mod, Token_ModEq); break;
  691. case '=': token.kind = token_kind_variant2(t, Token_Eq, Token_CmpEq); break;
  692. case '~': token.kind = token_kind_variant2(t, Token_Xor, Token_XorEq); break;
  693. case '!': token.kind = token_kind_variant2(t, Token_Not, Token_NotEq); break;
  694. case '+': token.kind = token_kind_variant3(t, Token_Add, Token_AddEq, '+', Token_Increment); break;
  695. case '-': token.kind = token_kind_variant4(t, Token_Sub, Token_SubEq, '-', Token_Decrement, '>', Token_ArrowRight); break;
  696. case '<':
  697. if (t->curr_rune == '-') {
  698. token.kind = Token_ArrowLeft;
  699. } else {
  700. token.kind = token_kind_dub_eq(t, '<', Token_Lt, Token_LtEq, Token_Shl, Token_ShlEq);
  701. }
  702. break;
  703. case '>':
  704. token.kind = token_kind_dub_eq(t, '>', Token_Gt, Token_GtEq, Token_Shr, Token_ShrEq);
  705. break;
  706. case '&':
  707. token.kind = Token_And;
  708. if (t->curr_rune == '~') {
  709. token.kind = Token_AndNot;
  710. advance_to_next_rune(t);
  711. if (t->curr_rune == '=') {
  712. token.kind = Token_AndNotEq;
  713. advance_to_next_rune(t);
  714. }
  715. } else {
  716. token.kind = token_kind_dub_eq(t, '&', Token_And, Token_AndEq, Token_CmpAnd, Token_CmpAndEq);
  717. }
  718. break;
  719. case '|': token.kind = token_kind_dub_eq(t, '|', Token_Or, Token_OrEq, Token_CmpOr, Token_CmpOrEq); break;
  720. default:
  721. if (curr_rune != GB_RUNE_BOM) {
  722. u8 str[4] = {};
  723. int len = cast(int)gb_utf8_encode_rune(str, curr_rune);
  724. tokenizer_err(t, "Illegal character: %.*s (%d) ", len, str, curr_rune);
  725. }
  726. token.kind = Token_Invalid;
  727. break;
  728. }
  729. }
  730. token.string.len = t->curr - token.string.text;
  731. return token;
  732. }