inp: added nps & seed properties

Author: BitterB0NG0

examples/cli_check.py

@@ -1,5 +1,5 @@
 """
-Example converting INp files using ``Convert``.
+Example converting INP files using ``Convert``.
 """
 
 import pathlib

examples/cli_run.py

@@ -30,6 +30,13 @@ def posthook_batch(self, path):
 inp1 = pymcnp.Inp.from_file(path1)
 inp2 = pymcnp.Inp.from_file(path2)
 
+inp0.nps = 1e4
+inp1.nps = 1e5
+inp2.nps = 1e6
+inp0.seed = 123534727
+inp1.seed = 123534727
+inp2.seed = 123534727
+
 # Running.
 runner = MyRun([inp0, inp1, inp2], command='echo')
 runner.run('.')

src/pymcnp/Inp.py

@@ -408,3 +408,100 @@ def other(self, other: str | types.String) -> None:
                 other = types.String.from_mcnp(other)
 
         self._other: types.Integer = other
+
+    @property
+    def nps(self) -> types.Integer:
+        """
+        File nps.
+
+        Raises:
+            InpError: SEMANTICS_OPTION.
+            TypeError:
+        """
+
+        try:
+            card_nps = next(filter(lambda card: isinstance(card, inp.Data) and isinstance(card.option, inp.data.Nps), self.data))
+            return card_nps.option.npp
+        except StopIteration:
+            return None
+
+    @nps.setter
+    def nps(self, nps: str | int | types.Integer) -> None:
+        """
+        Sets ``nps``.
+
+        Parameters:
+            nps: File nps.
+
+        Raises:
+            InpError: SEMANTICS_OPTION.
+            TypeError:
+        """
+
+        if nps is not None:
+            if isinstance(nps, types.Integer):
+                nps = nps
+            elif isinstance(nps, int):
+                nps = types.Integer(nps)
+            elif isinstance(nps, str):
+                nps = types.Integer.from_mcnp(nps)
+
+        try:
+            card_nps = next(filter(lambda card: isinstance(card, inp.Data) and isinstance(card.option, inp.data.Nps), self.data))
+            card_nps.option.npp = nps
+        except StopIteration:
+            card_nps = inp.data.Nps(nps)
+            self.data = [*self.data, inp.Data(card_nps)]
+
+    @property
+    def seed(self) -> types.Integer:
+        """
+        File seed.
+
+        Raises:
+            InpError: SEMANTICS_OPTION.
+            TypeError:
+        """
+
+        try:
+            card_rand = next(filter(lambda card: isinstance(card, inp.Data) and isinstance(card.option, inp.data.Rand), self.data))
+            option_seed = next(filter(lambda option: isinstance(option, inp.data.rand.Seed), card_rand.option.options or []))
+            return option_seed.seed
+        except StopIteration:
+            return None
+
+    @seed.setter
+    def seed(self, seed: str | int | types.Integer) -> None:
+        """
+        Sets ``seed``.
+
+        Parameters:
+            seed: File seed.
+
+        Raises:
+            InpError: SEMANTICS_OPTION.
+            TypeError:
+        """
+
+        if seed is not None:
+            if isinstance(seed, types.Integer):
+                seed = seed
+            elif isinstance(seed, int):
+                seed = types.Integer(seed)
+            elif isinstance(seed, str):
+                seed = types.Integer.from_mcnp(seed)
+
+        try:
+            card_rand = next(filter(lambda card: isinstance(card, inp.Data) and isinstance(card.option, inp.data.Rand), self.data))
+
+            try:
+                option_seed = next(filter(lambda option: isinstance(option, inp.data.rand.Seed), card_rand.option.options or []))
+                option_seed.seed = seed
+            except StopIteration:
+                option_seed = inp.data.rand.Seed(seed)
+                card_rand.option.options = [*(card_rand.option.options or []), option_seed]
+
+        except StopIteration:
+            option_seed = inp.data.rand.Seed(seed)
+            card_rand = inp.Data(inp.data.Rand([option_seed]))
+            self.data = [*self.data, card_rand]

src/pymcnp/cli/run.py

@@ -109,8 +109,7 @@ def run(self, path: str | pathlib.Path):
             path_ptrac = subdirectory / f'run-{i}.ptrac'
 
             subdirectory.mkdir()
-            with path_input.open('w') as file:
-                file.write(inp.to_mcnp())
+            inp.to_file(path_input)
 
             self.prehook_file(subdirectory, i)
             process = subprocess.Popen([f'{self.command}', f'inp={path_input} outp={path_output} ptrac={path_ptrac}'])

tests/test_pymcnp/test_Inp.py

@@ -77,6 +77,52 @@ class Test_Init(classes.Test_Init):
             },
         ]
 
+    class Test_Properties:
+        EXAMPLES = [
+            {
+                'title': consts.string.type.STRING,
+                'cells': [consts.string.inp.CELL, consts.string.inp.COMMENT, consts.string.inp.LIKE],
+                'surfaces': [consts.string.inp.SURFACE, consts.string.inp.COMMENT],
+                'data': [consts.string.inp.DATA, consts.string.inp.COMMENT],
+                'message': consts.string.type.STRING,
+                'other': consts.string.type.STRING,
+            },
+        ]
+
+        def test_nps(self):
+            for example in self.EXAMPLES:
+                print(example)
+
+                inp = pymcnp.Inp(**example)
+                inp.nps
+
+                inp.nps = '1e3'
+                inp.nps = 10
+                inp.nps = pymcnp.utils.types.Integer(1234938)
+                inp.nps
+
+        def test_seed(self):
+            for example in self.EXAMPLES:
+                print(example)
+
+                inp = pymcnp.Inp(**example)
+                inp.seed
+
+                save = inp.data
+
+                inp.seed = '11'
+                inp.seed = 11
+                inp.seed = pymcnp.utils.types.Integer(11)
+                inp.seed
+
+                inp.data = save
+                inp.data = [*inp.data, pymcnp.inp.Data(pymcnp.inp.data.Rand())]
+
+                inp.seed = '11'
+                inp.seed = 11
+                inp.seed = pymcnp.utils.types.Integer(11)
+                inp.seed
+
     class Test_Mcnp(classes.Test_Mcnp):
         element = pymcnp.Inp
         EXAMPLES_VALID = []