From 8c99fdc1a4b66baf66db4d93b03466ee53c48462 Mon Sep 17 00:00:00 2001
From: Philipp van Kempen <philipp.van-kempen@tum.de>
Date: Wed, 15 May 2024 08:46:26 +0200
Subject: [PATCH] coredsl2 writer: surround operands in parentheses

---
 seal5/backends/coredsl2/visitor.py | 23 ++++++++++++++++-------
 1 file changed, 16 insertions(+), 7 deletions(-)

diff --git a/seal5/backends/coredsl2/visitor.py b/seal5/backends/coredsl2/visitor.py
index 391ffe46..91505acd 100644
--- a/seal5/backends/coredsl2/visitor.py
+++ b/seal5/backends/coredsl2/visitor.py
@@ -31,26 +31,31 @@ def operation(self: behav.Operation, writer):
 
 def binary_operation(self: behav.BinaryOperation, writer):
     # print("binary_operation")
+    writer.write("(")
     self.left = self.left.generate(writer)
-    writer.write(f" {self.op.value} ")
+    writer.write(f") {self.op.value} (")
     self.right = self.right.generate(writer)
+    writer.write(")")
 
 
 def slice_operation(self: behav.SliceOperation, writer):
     # print("slice_operation")
     self.expr = self.expr.generate(writer)
-    writer.write("[")
+    writer.write("[(")
     self.left = self.left.generate(writer)
-    writer.write(":")
+    writer.write("):(")
     self.right = self.right.generate(writer)
-    writer.write("]")
+    writer.write(")]")
 
 
 def concat_operation(self: behav.ConcatOperation, writer):
     # print("concat_operation")
+    # TODO: only add () where required
+    writer.write("(")
     self.left = self.left.generate(writer)
-    writer.write(" :: ")
+    writer.write(") :: (")
     self.right = self.right.generate(writer)
+    writer.write(")")
 
 
 def number_literal(self: behav.IntLiteral, writer):
@@ -128,11 +133,13 @@ def loop(self: behav.Loop, writer):
 
 def ternary(self: behav.Ternary, writer):
     # print("ternary")
+    writer.write("(")
     self.cond.generate(writer)
-    writer.write(" ? ")
+    writer.write(") ? (")
     self.then_expr.generate(writer)
-    writer.write(" : ")
+    writer.write(") : (")
     self.else_expr.generate(writer)
+    writer.write(")")
 
 
 def return_(self: behav.Return, writer):
@@ -147,7 +154,9 @@ def return_(self: behav.Return, writer):
 def unary_operation(self: behav.UnaryOperation, writer):
     # print("unary_operation")
     writer.write(self.op.value)
+    writer.write("(")
     self.right.generate(writer)
+    writer.write(")")
 
 
 def named_reference(self: behav.NamedReference, writer):