Merge pull request #78 from RadhikaG/master

Added support for unary minus operator
BuildIt-lang · Sep 3, 2024 · c363631 · c363631
2 parents 9c4f2ee + df3d9fd
commit c363631
Show file tree

Hide file tree

Showing 9 changed files with 39 additions and 1 deletion.
diff --git a/include/blocks/block_replacer.h b/include/blocks/block_replacer.h
@@ -24,6 +24,7 @@ class block_replacer : public block_visitor {
 	virtual void visit(std::shared_ptr<expr>) override;
 	virtual void visit(std::shared_ptr<unary_expr>) override;
 	virtual void visit(std::shared_ptr<binary_expr>) override;
+	virtual void visit(std::shared_ptr<unary_minus_expr>) override;
 	virtual void visit(std::shared_ptr<not_expr>) override;
 	virtual void visit(std::shared_ptr<bitwise_not_expr>) override;
 	virtual void visit(std::shared_ptr<and_expr>) override;

diff --git a/include/blocks/block_visitor.h b/include/blocks/block_visitor.h
@@ -8,6 +8,7 @@ class expr;
 class unary_expr;
 class binary_expr;
 class not_expr;
+class unary_minus_expr;
 class bitwise_not_expr;
 class and_expr;
 class bitwise_and_expr;
@@ -73,6 +74,7 @@ class block_visitor {
 	virtual void visit(std::shared_ptr<expr>);
 	virtual void visit(std::shared_ptr<unary_expr>);
 	virtual void visit(std::shared_ptr<binary_expr>);
+	virtual void visit(std::shared_ptr<unary_minus_expr>);
 	virtual void visit(std::shared_ptr<not_expr>);
 	virtual void visit(std::shared_ptr<bitwise_not_expr>);
 	virtual void visit(std::shared_ptr<and_expr>);

diff --git a/include/blocks/c_code_generator.h b/include/blocks/c_code_generator.h
@@ -33,6 +33,7 @@ class c_code_generator : public block_visitor {
 	void nextl(void);
 
 	virtual void visit(not_expr::Ptr);
+	virtual void visit(unary_minus_expr::Ptr);
 	virtual void visit(bitwise_not_expr::Ptr);
 	virtual void visit(and_expr::Ptr);
 	virtual void visit(bitwise_and_expr::Ptr);

diff --git a/include/blocks/expr.h b/include/blocks/expr.h
@@ -82,7 +82,18 @@ class not_expr : public unary_expr {
 		return unary_is_same(self<not_expr>(), other);
 	}
 };
-
+// For the unary minus operator
+class unary_minus_expr : public unary_expr {
+public:
+	typedef std::shared_ptr<unary_minus_expr> Ptr;
+	virtual void dump(std::ostream &, int) override;
+	virtual void accept(block_visitor *a) override {
+		a->visit(self<unary_minus_expr>());
+	}
+	virtual bool is_same(block::Ptr other) override {
+		return unary_is_same(self<unary_minus_expr>(), other);
+	}
+};
 // For the bitwise not operator
 class bitwise_not_expr : public unary_expr {
 public:

diff --git a/include/builder/operator_overload.h b/include/builder/operator_overload.h
@@ -151,6 +151,12 @@ typename return_type_helper<T>::type operator!(const T &a) {
 	return ret_type(a).template builder_unary_op<block::not_expr>();
 }
 
+template <typename T>
+typename return_type_helper<T>::type operator-(const T &a) {
+	typedef typename return_type_helper<T>::type ret_type;
+	return ret_type(a).template builder_unary_op<block::unary_minus_expr>();
+}
+
 template <typename T>
 typename return_type_helper<T>::type operator~(const T &a) {
 	typedef typename return_type_helper<T>::type ret_type;

diff --git a/src/blocks/block_replacer.cpp b/src/blocks/block_replacer.cpp
@@ -27,6 +27,9 @@ void block_replacer::visit(binary_expr::Ptr a) {
 void block_replacer::visit(not_expr::Ptr a) {
 	unary_helper(a);
 }
+void block_replacer::visit(unary_minus_expr::Ptr a) {
+	unary_helper(a);
+}
 void block_replacer::visit(bitwise_not_expr::Ptr a) {
 	unary_helper(a);
 }

diff --git a/src/blocks/block_visitor.cpp b/src/blocks/block_visitor.cpp
@@ -14,6 +14,9 @@ void block_visitor::visit(binary_expr::Ptr a) {
 void block_visitor::visit(not_expr::Ptr a) {
 	a->expr1->accept(this);
 }
+void block_visitor::visit(unary_minus_expr::Ptr a) {
+	a->expr1->accept(this);
+}
 void block_visitor::visit(bitwise_not_expr::Ptr a) {
 	a->expr1->accept(this);
 }

diff --git a/src/blocks/c_code_generator.cpp b/src/blocks/c_code_generator.cpp
@@ -48,6 +48,12 @@ void c_code_generator::visit(not_expr::Ptr a) {
 	oss << ")";
 }
 
+void c_code_generator::visit(unary_minus_expr::Ptr a) {
+	oss << "-(";
+	a->expr1->accept(this);
+	oss << ")";
+}
+
 void c_code_generator::visit(bitwise_not_expr::Ptr a) {
 	oss << "~(";
 	a->expr1->accept(this);

diff --git a/src/blocks/expr.cpp b/src/blocks/expr.cpp
@@ -30,6 +30,11 @@ void not_expr::dump(std::ostream &oss, int indent) {
 	oss << "NOT_EXPR" << std::endl;
 	expr1->dump(oss, indent + 1);
 }
+void unary_minus_expr::dump(std::ostream &oss, int indent) {
+	printer::indent(oss, indent);
+	oss << "UNARY_MINUS_EXPR" << std::endl;
+	expr1->dump(oss, indent + 1);
+}
 void bitwise_not_expr::dump(std::ostream &oss, int indent) {
 	printer::indent(oss, indent);
 	oss << "BITWISE_NOT_EXPR" << std::endl;