Completed adding ROOT functions

scikit-hep · Mar 4, 2024 · 7e90d46 · 7e90d46
1 parent 8b2b178
commit 7e90d46
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/src/formulate/AST.py b/src/formulate/AST.py
@@ -240,6 +240,14 @@ def to_python(self):
                 return f"np.ceil({self.arguments[0]})"
             case "abs":
                 return f"np.abs({self.arguments[0]})"
+            case "even":
+                return f"! ({self.arguments[0]} % 2)"
+            case "factorial":
+                return f"np.math.factorial({self.arguments[0]})"
+            case "floor":
+                return f"! np.floor({self.arguments[0]})"
+            case "abs":
+                return f"np.abs({self.arguments[0]})"
             case "max":
                 return f"root_max({self.arguments[0]})"
             case "min":

diff --git a/src/formulate/toast.py b/src/formulate/toast.py
@@ -96,6 +96,9 @@
     "TMATH::ATANH": "atanh",
     "TMATH::CEIL": "ceil",
     "TMATH::ABS": "abs",
+    "TMath::Even": "even",
+    "TMath::Factorial": "factorial",
+    "TMath::Floor": "floor",
     "LENGTH$": "no_of_entries", #ak.num, axis = 1
     "ITERATION$": "current_iteration",
     "SUM$": "sum",