Handle Piecewise[] in code generation

Replace Piecewise[] by IfThen[] calls. Unfortunately, this does not work with vectorisation.
ianhinder · Sep 8, 2012 · 84338f2 · 84338f2
1 parent a5c9127
commit 84338f2
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/Tools/CodeGen/CodeGenCactus.m b/Tools/CodeGen/CodeGenCactus.m
@@ -741,6 +741,16 @@
            rhs = rhs /.  8/3 ->  keightthird;
            rhs = rhs /. -8/3 -> -keightthird;
            *)
+
+        (* Handle Piecewise function *)
+        (* TODO: This does not work with vectorisation, since IfThen
+           there expects a constant condition *)
+        rhs = rhs /. Piecewise -> piecewise1
+                  //. piecewise1[pairs_List, val_:0] :>
+                         If[pairs==={}, val,
+                            IfThen[First[pairs][[2]],
+                                   First[pairs][[1]],
+                                   piecewise1[Rest[pairs], val]]];
 
         (* Remove parentheses *)
         rhs = rhs //. Parenthesis[xx_] -> xx;