summaryrefslogtreecommitdiff
path: root/src/tokenizer.cc
blob: 26763eb2522c16e95119060e734b6398ab052346 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
#ifndef TOKENIZER_CC
#define TOKENIZER_CC

#include "common.cc"
#include "source.cc"
#include "token.cc"
#include "utf8.cc"

struct Tokenizer {
  Buffer* buffer;
  Buffer_Stack* stack;

  Tokenizer(Buffer_Stack* stack) : buffer(nullptr), stack(stack) {}
};

static inline Span tokenizer_make_span(const Tokenizer* tokenizer, usize start,
                                       usize end) {
  assert_neq(tokenizer, nullptr);

  Buffer* buffer = tokenizer->buffer;
  return Span(buffer->file, start, end);
}

static inline String tokenizer_make_lexeme(const Tokenizer* tokenizer,
                                           usize start, usize end) {
  assert_neq(tokenizer, nullptr);

  Buffer* buffer = tokenizer->buffer;
  return String(buffer->content[start], end - start);
}

static inline void tokenizer_make_token(const Tokenizer* tokenizer,
                                        Token* token, Token_Kind kind,
                                        usize start, usize end) {
  assert_neq(tokenizer, nullptr);
  assert_neq(token, nullptr);

  String lexeme = tokenizer_make_lexeme(tokenizer, start, end);
  Span span = tokenizer_make_span(tokenizer, start, end);

  *token = Token(kind, lexeme, span);
}

static Buffer* tokenizer_get_buffer(Tokenizer* tokenizer, usize* cursor) {
  assert_neq(tokenizer, nullptr);

  Buffer* curr = tokenizer->buffer;
  if (likely(curr != nullptr)) {
    *cursor = curr->cursor;
    if (*cursor < curr->content.length) return curr;
  }

  if (!buffer_stack_pop(tokenizer->stack, &curr)) return nullptr;

  tokenizer->buffer = curr;
  *cursor = curr->cursor;
  return curr;
}

static wchar tokenizer_advance(const Tokenizer* tokenizer, usize* offset) {
  const String text = tokenizer->buffer->content;

  unsigned char c = *text[*offset];
  u8 nobytes = utf8_nobytes(c);
  if (nobytes > 1) panic("no support for multi-byte chars: %c:%d", c, nobytes);

  *offset += nobytes;
  return (wchar)c;
}

bool tokenizer_next(Tokenizer* tokenizer, Token* out) {
  assert_neq(tokenizer, nullptr);
  assert_neq(out, nullptr);

  usize cursor;
  Buffer* buffer = tokenizer_get_buffer(tokenizer, &cursor);
  if (buffer == nullptr) return false;

  usize advance = cursor;
  wchar c = tokenizer_advance(tokenizer, &advance);

  switch (c) {
    default:
      tokenizer_make_token(tokenizer, out, Token_Kind_Invalid_Char, cursor,
                           advance);
  }

  buffer->cursor = advance;
  return true;
}

#endif