inp: added unbuild methods

Author: BitterB0NG0

scripts/inp_code.py

@@ -269,6 +269,18 @@ def ATTRS_ASSIGN(element, t):
     return o.strip()
 
 
+def ATTRS_UNBUILDER(element, t):
+    o = ''
+
+    for attribute in element.attributes:
+        if attribute.type.startswith('types.'):
+            o += f'{TABS(t)}{attribute.name}=copy.deepcopy(ast.{attribute.name}),\n'
+        else:
+            o += f'{TABS(t)}{attribute.name}={CAMEL(attribute.type, "Builder")}.unbuild(ast.{attribute.name}),\n'
+
+    return o.strip()
+
+
 # ELEMENT #
 def INIT(element):
     return f"""
@@ -311,6 +323,7 @@ class {CAMEL(element.name, "Option")}(Option):
 def ELEMENT(element, parent_name, depth):
     return f'''
 import re
+import copy
 import typing
 import dataclasses
 
@@ -333,6 +346,8 @@ class {CAMEL(element.name)}({CAMEL(parent_name, 'Option')}):
         {ATTRS_COMMENT(element, 2)}
     """
 
+    _KEYWORD = "{element.mnemonic}"
+
     _ATTRS = {{{ATTRS_DICT(element)}}}
 
     _REGEX = re.compile(rf"\\A{GET_REGEX(element)}\\Z")
@@ -378,6 +393,19 @@ def build(self):
         return {CAMEL(element.name)}(
             {ATTRS_ASSIGN(element, 3)}
         )
+
+    @staticmethod
+    def unbuild(ast: {CAMEL(element.name)}):
+        """
+        Unbuilds ``{CAMEL(element.name)}`` into ``{CAMEL(element.name, "Builder")}``
+
+        Returns:
+            ``{CAMEL(element.name, "Builder")}`` for ``{CAMEL(element.name)}``.
+        """
+
+        return {CAMEL(element.name)}(
+            {ATTRS_UNBUILDER(element, 3)}
+        )
 '''[1:]
 
 

src/pymcnp/Inp.py

@@ -1,4 +1,5 @@
 import re
+import copy
 import typing
 import dataclasses
 
@@ -250,25 +251,6 @@ class InpBuilder:
     message: str = ''
     other: str = ''
 
-    def from_mcnp(source: str):
-        """
-        Initiailizes ``InpBuilder`` from MCNP.
-
-        Attributes:
-            source: MCNP for ``Inp``.
-        """
-
-        ast = Inp.from_mcnp(source)
-
-        return InpBuilder(
-            title=ast.title,
-            cells={cell.number.value: cell for cell in ast.cells},
-            surfaces={surface.number.value: surface for surface in ast.surfaces},
-            data={},
-            message=ast.message,
-            other=ast.other,
-        )
-
     def build(self):
         """
         Builds ``InpBuilder`` into ``Inp``.
@@ -299,7 +281,7 @@ def build(self):
 
         data = []
         for item in self.data.values():
-            if isinstance(item, inp.Surface):
+            if isinstance(item, inp.Data):
                 data.append(item)
             elif isinstance(item, str):
                 data.append(inp.Data.from_mcnp(item))
@@ -318,3 +300,24 @@ def build(self):
             data=data,
             data_comments=types.Tuple([]),
         )
+
+    @staticmethod
+    def unbuild(ast: Inp):
+        """
+        Unbuilds ``Inp`` into ``InpBuilder``
+
+        Returns:
+            ``InpBuilder`` for ``Inp``.
+        """
+
+        return InpBuilder(
+            title=copy.deepcopy(ast.title),
+            cells={cell.number.value: inp.CellBuilder.unbuild(cell) for cell in ast.cells},
+            surfaces={
+                surface.number.value: inp.SurfaceBuilder.unbuild(surface)
+                for surface in ast.surfaces
+            },
+            data={data.option._KEYWORD: inp.DataBuilder.unbuild(data) for data in ast.data},
+            message=copy.deepcopy(ast.message),
+            other=copy.deepcopy(ast.other),
+        )

src/pymcnp/inp/Cell.py

@@ -1,4 +1,5 @@
 import re
+import copy
 import dataclasses
 
 
@@ -182,3 +183,83 @@ def build(self):
             geometry=geometry,
             options=options,
         )
+
+    @staticmethod
+    def unbuild(ast: Cell):
+        """
+        Unbuilds ``Cell`` into ``CellBuilder``
+
+        Returns:
+            ``CellBuilder`` for ``Cell``.
+        """
+
+        options = []
+        for option in ast.options or []:
+            if isinstance(ast.option, cell.Imp):
+                options.append(cell.ImpBuilder.unbuild(option))
+            elif isinstance(ast.option, cell.Vol):
+                options.append(cell.VolBuilder.unbuild(option))
+            elif isinstance(ast.option, cell.Pwt):
+                options.append(cell.PwtBuilder.unbuild(option))
+            elif isinstance(ast.option, cell.Ext):
+                options.append(cell.ExtBuilder.unbuild(option))
+            elif isinstance(ast.option, cell.Fcl):
+                options.append(cell.FclBuilder.unbuild(option))
+            elif isinstance(ast.option, cell.Wwn):
+                options.append(cell.WwnBuilder.unbuild(option))
+            elif isinstance(ast.option, cell.Dxc):
+                options.append(cell.DxcBuilder.unbuild(option))
+            elif isinstance(ast.option, cell.Nonu):
+                options.append(cell.NonuBuilder.unbuild(option))
+            elif isinstance(ast.option, cell.Pd):
+                options.append(cell.PdBuilder.unbuild(option))
+            elif isinstance(ast.option, cell.U):
+                options.append(cell.UBuilder.unbuild(option))
+            elif isinstance(ast.option, cell.Trcl_0):
+                options.append(cell.TrclBuilder_0.unbuild(option))
+            elif isinstance(ast.option, cell.Trcl_1):
+                options.append(cell.TrclBuilder_1.unbuild(option))
+            elif isinstance(ast.option, cell.Trcl_2):
+                options.append(cell.TrclBuilder_2.unbuild(option))
+            elif isinstance(ast.option, cell.Trcl_3):
+                options.append(cell.TrclBuilder_3.unbuild(option))
+            elif isinstance(ast.option, cell.Trcl_4):
+                options.append(cell.TrclBuilder_4.unbuild(option))
+            elif isinstance(ast.option, cell.Trcl_5):
+                options.append(cell.TrclBuilder_5.unbuild(option))
+            elif isinstance(ast.option, cell.Lat):
+                options.append(cell.LatBuilder.unbuild(option))
+            elif isinstance(ast.option, cell.Fill_0):
+                options.append(cell.FillBuilder_0.unbuild(option))
+            elif isinstance(ast.option, cell.Fill_1):
+                options.append(cell.FillBuilder_1.unbuild(option))
+            elif isinstance(ast.option, cell.Fill_2):
+                options.append(cell.FillBuilder_2.unbuild(option))
+            elif isinstance(ast.option, cell.Fill_3):
+                options.append(cell.FillBuilder_3.unbuild(option))
+            elif isinstance(ast.option, cell.Fill_4):
+                options.append(cell.FillBuilder_4.unbuild(option))
+            elif isinstance(ast.option, cell.Fill_5):
+                options.append(cell.FillBuilder_5.unbuild(option))
+            elif isinstance(ast.option, cell.Fill_6):
+                options.append(cell.FillBuilder_6.unbuild(option))
+            elif isinstance(ast.option, cell.Elpt):
+                options.append(cell.ElptBuilder.unbuild(option))
+            elif isinstance(ast.option, cell.Tmp_0):
+                options.append(cell.TmpBuilder_0.unbuild(option))
+            elif isinstance(ast.option, cell.Tmp_1):
+                options.append(cell.TmpBuilder_1.unbuild(option))
+            elif isinstance(ast.option, cell.Cosy):
+                options.append(cell.CosyBuilder.unbuild(option))
+            elif isinstance(ast.option, cell.Bflcl):
+                options.append(cell.BflclBuilder.unbuild(option))
+            elif isinstance(ast.option, cell.Unc):
+                options.append(cell.UncBuilder.unbuild(option))
+
+        return CellBuilder(
+            number=copy.deepcopy(ast.number),
+            material=copy.deepcopy(ast.material),
+            density=copy.deepcopy(ast.density),
+            geometry=copy.deepcopy(ast.geometry),
+            options=options,
+        )

src/pymcnp/inp/Comment.py

@@ -1,4 +1,5 @@
 import re
+import copy
 import typing
 import dataclasses
 
@@ -70,3 +71,16 @@ def build(self):
             text = self.text
 
         return Comment(text=text)
+
+    @staticmethod
+    def unbuild(ast: Comment):
+        """
+        Unbuilds ``Comment`` into ``CommentBuilder``
+
+        Returns:
+            ``CommentBuilder`` for ``Comment``.
+        """
+
+        return CommentBuilder(
+            text=copy.deepcopy(ast.text),
+        )