From 160fbb0ddaaa25a98cc1814ac503466746f29423 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Mon, 27 Aug 2012 16:15:56 -0400
Subject: [PATCH] Walk mapper support for more stuff.

---
 pymbolic/mapper/__init__.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/pymbolic/mapper/__init__.py b/pymbolic/mapper/__init__.py
index 9173b39..89c6bb2 100644
--- a/pymbolic/mapper/__init__.py
+++ b/pymbolic/mapper/__init__.py
@@ -407,6 +407,30 @@ class WalkMapper(RecursiveMapper):
 
         self.rec(expr.child)
 
+    def map_comparison(self, expr):
+        if not self.visit(expr):
+            return
+
+        self.rec(expr.left)
+        self.rec(expr.right)
+
+    def map_logical_not(self, expr):
+        if not self.visit(expr):
+            return
+
+        self.rec(expr.child)
+
+    map_logical_and = map_sum
+    map_logical_or = map_sum
+
+    def map_if(self, expr):
+        if not self.visit(expr):
+            return
+
+        self.rec(expr.condition)
+        self.rec(expr.then)
+        self.rec(expr.else_)
+
     def map_if_positive(self, expr):
         if not self.visit(expr):
             return
-- 
GitLab