From c09f4cf28e5e100534541c30cbc6ccd2b73fc81c Mon Sep 17 00:00:00 2001
From: Lioncash <mathew1800@gmail.com>
Date: Sat, 29 Sep 2018 04:07:32 -0400
Subject: [PATCH] constant_propagation_pass: Fold NOT operations

---
 src/ir_opt/constant_propagation_pass.cpp | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/src/ir_opt/constant_propagation_pass.cpp b/src/ir_opt/constant_propagation_pass.cpp
index f315a3f1..4d323bf3 100644
--- a/src/ir_opt/constant_propagation_pass.cpp
+++ b/src/ir_opt/constant_propagation_pass.cpp
@@ -77,6 +77,21 @@ void FoldEOR(IR::Inst& inst, bool is_32_bit) {
     }
 }
 
+// Folds NOT operations if the contained value is an immediate.
+void FoldNOT(IR::Inst& inst, bool is_32_bit) {
+    const auto operand = inst.GetArg(0);
+
+    if (operand.IsImmediate()) {
+        const u64 result = ~operand.GetImmediateAsU64();
+
+        if (is_32_bit) {
+            inst.ReplaceUsesWith(IR::Value{static_cast<u32>(result)});
+        } else {
+            inst.ReplaceUsesWith(IR::Value{result});
+        }
+    }
+}
+
 // Folds OR operations based on the following:
 //
 // 1. imm_x | imm_y -> result
@@ -141,6 +156,10 @@ void ConstantPropagation(IR::Block& block) {
         case IR::Opcode::Or64:
             FoldOR(inst, opcode == IR::Opcode::Or32);
             break;
+        case IR::Opcode::Not32:
+        case IR::Opcode::Not64:
+            FoldNOT(inst, opcode == IR::Opcode::Not32);
+            break;
         case IR::Opcode::ZeroExtendByteToWord: {
             if (!inst.AreAllArgsImmediates())
                 break;