expr/SelectExpr: add boolean attribute for negating the SelectExpr co…

…ndition
vmihalko · Mar 14, 2024 · ac88b7c · ac88b7c
1 parent b703560
commit ac88b7c
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/expr/Expr.cpp b/expr/Expr.cpp
@@ -204,11 +204,12 @@ bool GepExpr::classof(const Expr* expr) {
     return expr->getKind() == EK_GepExpr;
 }
 
-SelectExpr::SelectExpr(Expr* comp, Expr* l, Expr* r) :
+SelectExpr::SelectExpr(Expr* comp, Expr* l, Expr* r, bool negateCondition = false) :
     ExprBase(EK_SelectExpr),
     left(l),
     right(r),
-    comp(comp) {
+    comp(comp)
+    negateCondition(negateCondition) {
     setType(l->getType());
 }
 

diff --git a/expr/Expr.h b/expr/Expr.h
@@ -345,8 +345,9 @@ class SelectExpr : public ExprBase {
     Expr* left;
     Expr* right;
     Expr* comp;
+    bool negateCondition;
 
-    SelectExpr(Expr*, Expr*, Expr*);
+    SelectExpr(Expr*, Expr*, Expr*, bool negateCondition);
 
     void accept(ExprVisitor& visitor) override;
 

diff --git a/writer/ExprWriter.cpp b/writer/ExprWriter.cpp
@@ -292,7 +292,11 @@ void ExprWriter::visit(GepExpr& expr) {
 }
 
 void ExprWriter::visit(SelectExpr& expr) {
+    if (expr.negateCondition)
+        ss << "!(";
     parensIfNotSimple(expr.comp);
+    if (expr.negateCondition)
+        ss << ")";
     ss << " ? ";
     parensIfNotSimple(expr.left);
     ss << " : ";