Added Awkward equivalent of some ROOT functions

src/formulate/AST.py

@@ -184,8 +184,9 @@ def to_python(self):
                 return "np.pi"
             case "e":
                 return "np.exp(1)"
+            case "e":
+                return "np.exp(1)"
             case _ :
                 raise ValueError("Not a valid function!")
 
-# come-up with a shared notation for functions
-# make a  to_python function
+

src/formulate/func_translations.py

@@ -0,0 +1,41 @@
+
+#maybe use num
+
+def root_length(array):
+    import awkward as ak
+    while array.layout.purelist_depth > 2:
+        array = ak.flatten(array, axis=-1)
+    return ak.count(array, axis=-1)
+
+
+def root_sum(array):
+    import awkward as ak
+    while array.layout.purelist_depth > 2:
+        array = ak.flatten(array, axis=-1)
+    return ak.sum(array, axis=-1)
+
+
+def root_min(array):
+    import awkward as ak
+    while array.layout.purelist_depth >= 2:
+        array = ak.min(array, axis=-1)
+    return ak.fill_none(array, 0)
+
+
+
+def root_max(array):
+    import awkward as ak
+    while array.layout.purelist_depth >= 2:
+        array = ak.max(array, axis=-1)
+    return ak.fill_none(array, 0)
+
+
+def root_min_if(array, condition):
+    import awkward as ak
+    array = array[condition != 0]
+    return ak.fill_none(ak.min(array,axis=1),0)
+
+def root_max_if(array, condition):
+    import awkward as ak
+    array = array[condition != 0]
+    return ak.fill_none(ak.max(array,axis=1),0)

src/formulate/toast.py

@@ -9,7 +9,7 @@
 
 
 UNARY_OP = {"pos", "neg", "binv", "linv"}
-with_sign = {}
+
 BINARY_OP = {
     "add",
     "sub",
@@ -59,7 +59,7 @@
 FUNC_MAPPING = {
     "MATH::PI": "pi", #np.pi
     "MATH::E": "e",
-    "LENGTH$": "no_of_entries", #
+    "LENGTH$": "no_of_entries", #ak.num, axis = 1
     "ITERATION$": "current_iteration",
     "SUM$": "sum",
     "MIN$": "min",