From 94d021725e2e276d6cdf3b9157135d8012f67541 Mon Sep 17 00:00:00 2001
From: "Micah D. Gale" <mg@gale.engineer>
Date: Sun, 14 Jan 2024 17:59:09 -0600
Subject: [PATCH] Started group parsing mechanism

---
 montepy/data_inputs/tally.py | 38 ++++++++++++++++++++++++++++++++----
 1 file changed, 34 insertions(+), 4 deletions(-)

diff --git a/montepy/data_inputs/tally.py b/montepy/data_inputs/tally.py
index 6a5ca764..c9091c0b 100644
--- a/montepy/data_inputs/tally.py
+++ b/montepy/data_inputs/tally.py
@@ -24,9 +24,10 @@ def _number_validator(self, number):
 class Tally(DataInputAbstract, Numbered_MCNP_Object):
     """ """
 
+    # todo type enforcement
     _parser = TallyParser()
 
-    __slots__ = {"_groups", "_type", "_number", "_old_number"}
+    __slots__ = {"_groups", "_type", "_number", "_old_number", "_include_total"}
 
     def __init__(self, input=None):
         self._cells = Cells()
@@ -41,6 +42,11 @@ def __init__(self, input=None):
                 tally_type = TallyType(self.number % _TALLY_TYPE_MODULUS)
             except ValueError as e:
                 raise MalformedInputEror(input, f"Tally Type provided not allowed: {e}")
+            groups, has_total = TallyGroup.parse_tally_specification(
+                self._tree["tally"]
+            )
+            self._groups = group
+            self._include_total = has_total
 
     @staticmethod
     def _class_prefix():
@@ -76,7 +82,31 @@ def number(self):
 class TallyGroup:
     __slots__ = {"_cells"}
 
-    def __init__(
-        self,
-    ):
+    def __init__(self, cells=None, nodes=None):
         self._cells = montepy.cells.Cells()
+
+    @staticmethod
+    def parse_tally_specification(tally_spec):
+        # TODO type enforcement
+        ret = []
+        in_parens = False
+        buff = None
+        has_total = False
+        for node in tally_spec:
+            # TODO handle total
+            if in_parens:
+                if node.value == ")":
+                    in_parens = False
+                    buff.append(node)
+                    ret.append(buff)
+                    buff = None
+                else:
+                    buff.apend(node)
+            else:
+                if node.value == "(":
+                    in_parens = True
+                    buff = TallyGroup()
+                    buff.append(node)
+                else:
+                    ret.append(TallyGroup(nodes=[node]))
+        return (ret, has_total)