Add comparison ops

leonardt · Dec 2, 2019 · fa0b7c3 · fa0b7c3
1 parent e56e7f4
commit fa0b7c3
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 1 deletion.
diff --git a/include/verilogAST.hpp b/include/verilogAST.hpp
@@ -118,7 +118,11 @@ enum BinOp {
   POW,
   MOD,
   ALSHIFT,
-  ARSHIFT
+  ARSHIFT,
+  LT,
+  LTE,
+  GT,
+  GTE
 };
 }
 

diff --git a/src/verilogAST.cpp b/src/verilogAST.cpp
@@ -161,6 +161,18 @@ std::string BinaryOp::toString() {
     case BinOp::ARSHIFT:
       op_str = ">>>";
       break;
+    case BinOp::LT:
+      op_str = "<";
+      break;
+    case BinOp::LTE:
+      op_str = "<=";
+      break;
+    case BinOp::GT:
+      op_str = ">";
+      break;
+    case BinOp::GTE:
+      op_str = ">=";
+      break;
   }
   return left->toString() + ' ' + op_str + ' ' + right->toString();
 }

diff --git a/tests/basic.cpp b/tests/basic.cpp
@@ -86,6 +86,10 @@ TEST(BasicTests, TestBinaryOp) {
   ops.push_back(std::make_pair(vAST::BinOp::MOD, "%"));
   ops.push_back(std::make_pair(vAST::BinOp::ALSHIFT, "<<<"));
   ops.push_back(std::make_pair(vAST::BinOp::ARSHIFT, ">>>"));
+  ops.push_back(std::make_pair(vAST::BinOp::LT, "<"));
+  ops.push_back(std::make_pair(vAST::BinOp::LTE, "<="));
+  ops.push_back(std::make_pair(vAST::BinOp::GT, ">"));
+  ops.push_back(std::make_pair(vAST::BinOp::GTE, ">="));
   for (auto it : ops) {
     vAST::BinOp::BinOp op = it.first;
     std::string op_str = it.second;