nqasm: support unary operators in immediate expressions

1 files changed, 5 insertions, 1 deletions

diff --git a/lexer.l b/lexer.l
index 053ac4e..324c3da 100644
--- a/lexer.l
+++ b/lexer.l
@@ -76,7 +76,11 @@ HEX        [0-9a-f]
 
 <args>\.                 { return T_DOT; }
 <args>@                  { return T_DEREF; }
-<args>[-+]?{DEC}+        { return intlit(yytext,   10, &yylval.lval); }
+<args>\+                 { return T_OPERATOR_ADD; }
+<args>-                  { return T_OPERATOR_SUB; }
+<args>!                  { return T_OPERATOR_NOT; }
+<args>~                  { return T_OPERATOR_INV; }
+<args>{DEC}+             { return intlit(yytext,   10, &yylval.lval); }
 <args>0x{HEX}+           { return intlit(yytext+2, 16, &yylval.lval); }
 <args>{SP}*,{SP}*        { return T_COMMA; }