55 lines
2.1 KiB
C
55 lines
2.1 KiB
C
#include "token.h"
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <inttypes.h>
|
|
|
|
static const char* NAMES[] = {
|
|
[TOK_FN] = "fn", [TOK_LET] = "let", [TOK_VAR] = "var", [TOK_IF] = "if",
|
|
[TOK_ELSE] = "else", [TOK_WHILE] = "while", [TOK_FOR] = "for", [TOK_IN] = "in", [TOK_RETURN] = "return",
|
|
[TOK_STRUCT] = "struct", [TOK_TYPE] = "type", [TOK_ENUM] = "enum", [TOK_IMPL] = "impl",
|
|
[TOK_MATCH] = "match",
|
|
[TOK_I64] = "i64", [TOK_F64] = "f64", [TOK_BOOL] = "bool", [TOK_STR] = "str", [TOK_VOID] = "void",
|
|
[TOK_INT_LIT] = "整数", [TOK_FLOAT_LIT] = "浮点数", [TOK_STR_LIT] = "字符串",
|
|
[TOK_TRUE] = "true", [TOK_FALSE] = "false",
|
|
[TOK_IDENT] = "标识符", [TOK_UNDERSCORE] = "_",
|
|
[TOK_PLUS] = "+", [TOK_MINUS] = "-", [TOK_STAR] = "*",
|
|
[TOK_SLASH] = "/", [TOK_PERCENT] = "%",
|
|
[TOK_EQ_EQ] = "==", [TOK_BANG_EQ] = "!=",
|
|
[TOK_LT] = "<", [TOK_GT] = ">", [TOK_LT_EQ] = "<=", [TOK_GT_EQ] = ">=",
|
|
[TOK_AND_AND] = "&&", [TOK_PIPE_PIPE] = "||", [TOK_BANG] = "!",
|
|
[TOK_ARROW] = "->", [TOK_TO] = "to", [TOK_MATCH_ARROW] = "=>",
|
|
[TOK_PLUS_EQ] = "+=", [TOK_MINUS_EQ] = "-=", [TOK_STAR_EQ] = "*=", [TOK_SLASH_EQ] = "/=",
|
|
[TOK_LPAREN] = "(", [TOK_RPAREN] = ")",
|
|
[TOK_LBRACE] = "{", [TOK_RBRACE] = "}",
|
|
[TOK_LBRACKET] = "[", [TOK_RBRACKET] = "]",
|
|
[TOK_COMMA] = ",", [TOK_COLON] = ":", [TOK_SEMICOLON] = ";",
|
|
[TOK_ASSIGN] = "=",
|
|
[TOK_DOT] = ".", [TOK_COLON_COLON] = "::",
|
|
[TOK_EOF] = "EOF", [TOK_ERROR] = "错误",
|
|
};
|
|
|
|
const char* tok_name(TokenKind kind) {
|
|
return NAMES[kind];
|
|
}
|
|
|
|
bool tok_is_type(TokenKind kind) {
|
|
return kind == TOK_I64 || kind == TOK_F64 || kind == TOK_BOOL || kind == TOK_STR || kind == TOK_VOID;
|
|
}
|
|
|
|
int64_t tok_int_value(const Token* tok) {
|
|
if (tok->length <= 0 || tok->length >= 32) return 0;
|
|
char buf[32];
|
|
memcpy(buf, tok->start, tok->length);
|
|
buf[tok->length] = '\0';
|
|
return strtoll(buf, NULL, 10);
|
|
}
|
|
|
|
double tok_float_value(const Token* tok) {
|
|
if (tok->length <= 0 || tok->length >= 64) return 0.0;
|
|
char buf[64];
|
|
memcpy(buf, tok->start, tok->length);
|
|
buf[tok->length] = '\0';
|
|
return strtod(buf, NULL);
|
|
}
|