bsnes/nall/string/eval/literal.hpp

100 lines
2.5 KiB
C++
Executable File

#pragma once
namespace nall::Eval {
inline auto isLiteral(const char*& s) -> bool {
char n = s[0];
return (n >= 'A' && n <= 'Z')
|| (n >= 'a' && n <= 'z')
|| (n >= '0' && n <= '9')
|| (n == '%' || n == '$' || n == '_' || n == '.')
|| (n == '\'' || n == '\"');
}
inline auto literalNumber(const char*& s) -> string {
const char* p = s;
//binary
if(p[0] == '%' || (p[0] == '0' && p[1] == 'b')) {
uint prefix = 1 + (p[0] == '0');
p += prefix;
while(p[0] == '\'' || p[0] == '0' || p[0] == '1') p++;
if(p - s <= prefix) throw "invalid binary literal";
string result = slice(s, 0, p - s);
s = p;
return result;
}
//octal
if(p[0] == '0' && p[1] == 'o') {
uint prefix = 1 + (p[0] == '0');
p += prefix;
while(p[0] == '\'' || (p[0] >= '0' && p[0] <= '7')) p++;
if(p - s <= prefix) throw "invalid octal literal";
string result = slice(s, 0, p - s);
s = p;
return result;
}
//hex
if(p[0] == '$' || (p[0] == '0' && p[1] == 'x')) {
uint prefix = 1 + (p[0] == '0');
p += prefix;
while(p[0] == '\'' || (p[0] >= '0' && p[0] <= '9') || (p[0] >= 'A' && p[0] <= 'F') || (p[0] >= 'a' && p[0] <= 'f')) p++;
if(p - s <= prefix) throw "invalid hex literal";
string result = slice(s, 0, p - s);
s = p;
return result;
}
//decimal
while(p[0] == '\'' || (p[0] >= '0' && p[0] <= '9')) p++;
if(p[0] != '.') {
string result = slice(s, 0, p - s);
s = p;
return result;
}
//floating-point
p++;
while(p[0] == '\'' || (p[0] >= '0' && p[0] <= '9')) p++;
string result = slice(s, 0, p - s);
s = p;
return result;
}
inline auto literalString(const char*& s) -> string {
const char* p = s;
char escape = *p++;
while(p[0] && p[0] != escape) p++;
if(*p++ != escape) throw "unclosed string literal";
string result = slice(s, 0, p - s);
s = p;
return result;
}
inline auto literalVariable(const char*& s) -> string {
const char* p = s;
while(p[0] == '_' || p[0] == '.' || (p[0] >= 'A' && p[0] <= 'Z') || (p[0] >= 'a' && p[0] <= 'z') || (p[0] >= '0' && p[0] <= '9')) p++;
string result = slice(s, 0, p - s);
s = p;
return result;
}
inline auto literal(const char*& s) -> string {
const char* p = s;
if(p[0] >= '0' && p[0] <= '9') return literalNumber(s);
if(p[0] == '%' || p[0] == '$') return literalNumber(s);
if(p[0] == '\'' || p[0] == '\"') return literalString(s);
if(p[0] == '_' || p[0] == '.' || (p[0] >= 'A' && p[0] <= 'Z') || (p[0] >= 'a' && p[0] <= 'z')) return literalVariable(s);
throw "invalid literal";
}
}