feat: Added pointer and reference unary operators

alinalihassan · May 27, 2022 · ce4ea17 · ce4ea17
1 parent 04cf47f
commit ce4ea17
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 2 deletions.
diff --git a/src/liblesma/Backend/Codegen.cpp b/src/liblesma/Backend/Codegen.cpp
@@ -1040,7 +1040,14 @@ llvm::Value *Codegen::visit(UnaryOp *node) {
             return Builder->CreateNot(operand, ".tmp");
         else
             throw CodegenError(node->getSpan(), "Cannot apply {} to {}", NAMEOF_ENUM(node->getOperator()), node->getExpression()->toString(SourceManager.get(), 0));
-    } else
+    } else if (node->getOperator() == TokenType::STAR) {
+        return Builder->CreateLoad(operand->getType()->getPointerElementType(), operand);
+    } else if (node->getOperator() == TokenType::AMPERSAND) {
+        auto ptr = Builder->CreateAlloca(operand->getType(), nullptr, ".tmp");
+        Builder->CreateStore(operand, ptr);
+        return ptr;
+    }
+    else
         throw CodegenError(node->getSpan(), "Unknown unary operator, cannot apply {} to {}", NAMEOF_ENUM(node->getOperator()), node->getExpression()->toString(SourceManager.get(), 0));
 }
 

diff --git a/src/liblesma/Frontend/Lexer.cpp b/src/liblesma/Frontend/Lexer.cpp
@@ -67,6 +67,8 @@ Token *Lexer::ScanOne(bool continuation) {
                 return AddToken(TokenType::STAR_EQUAL);
             return AddToken(TokenType::STAR);
         }
+        case '&':
+            return AddToken(TokenType::AMPERSAND);
         case '!':
             return AddToken(MatchAndAdvance('=') ? TokenType::BANG_EQUAL : TokenType::BANG);
         case '=': {

diff --git a/src/liblesma/Frontend/Parser.cpp b/src/liblesma/Frontend/Parser.cpp
@@ -178,7 +178,7 @@ Expression *Parser::ParseDot() {
 
 Expression *Parser::ParseUnary() {
     Expression *left = nullptr;
-    while (AdvanceIfMatchAny<TokenType::MINUS>()) {
+    while (AdvanceIfMatchAny<TokenType::MINUS, TokenType::STAR, TokenType::AMPERSAND>()) {
         auto op = Previous();
         auto expr = ParseDot();
         left = new UnaryOp({op->getStart(), expr->getEnd()}, op->type, expr);

diff --git a/src/liblesma/Token/TokenType.h b/src/liblesma/Token/TokenType.h
@@ -27,6 +27,7 @@ namespace lesma {
         STAR,
         MOD,
         POWER,
+        AMPERSAND,
         RANGE,
         ELLIPSIS,