[PP] Equality expression

WasmVM · Mar 20, 2024 · 3494c64 · 3494c64
1 parent 19e4299
commit 3494c64
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 2 deletions.
diff --git a/src/include/PreProcessor.hpp b/src/include/PreProcessor.hpp
@@ -85,7 +85,7 @@ struct PreProcessor {
         Result additive();
         Result shift();
         Result relational();
-        Result equality(); // TODO:
+        Result equality();
         Result bitwise_AND(); // TODO:
         Result bitwise_exclusive_OR(); // TODO:
         Result bitwise_inclusive_OR(); // TODO:

diff --git a/src/lib/PreProcessor/Expression.cpp b/src/lib/PreProcessor/Expression.cpp
@@ -104,7 +104,7 @@ PreProcessor::Expression::Result PreProcessor::Expression::relation_op(PreProces
 }
 
 PreProcessor::Expression::Result PreProcessor::Expression::eval(){
-    return relational(); // TODO:
+    return equality(); // TODO:
 }
 
 PreProcessor::Expression::Result PreProcessor::Expression::primary(){
@@ -310,3 +310,23 @@ PreProcessor::Expression::Result PreProcessor::Expression::relational(){
     }
     return res;
 }
+
+PreProcessor::Expression::Result PreProcessor::Expression::equality(){
+    PreProcessor::Expression::Result res = relational();
+    Line::iterator head = skip_whitespace(cur, end);
+    cur = head;
+    if(cur != end && cur->hold<TokenType::Punctuator>()){
+        TokenType::Punctuator punct = cur->value();
+        if(punct.type == TokenType::Punctuator::DoubleEqual || punct.type == TokenType::Punctuator::ExclamEqual){
+            cur = std::next(cur);
+            PreProcessor::Expression::Result operand = equality();
+            implicit_cast(res, operand);
+            if(punct.type == TokenType::Punctuator::DoubleEqual){
+                return relation_op<std::equal_to>(res, operand);
+            }else{
+                return relation_op<std::not_equal_to>(res, operand);
+            }
+        }
+    }
+    return res;
+}