orto 1.7.1__py3-none-any.whl → 1.8.1__py3-none-any.whl

orto/__version__.py

@@ -1 +1 @@
-__version__ = '1.7.1'
+__version__ = '1.8.1'

orto/cli.py

@@ -820,8 +820,11 @@ def plot_abs_func(uargs, save_data_only=False):
 
     # Find unique name from multiple file names
     if len(uargs.output_file) > 1:
-        base_names = [of.stem for of in uargs.output_file]
-        unique_names = ut.find_unique_substring(base_names)
+        if uargs.unique_names:
+            base_names = [of.stem for of in uargs.output_file]
+            unique_names = ut.find_unique_substring(base_names)
+        else:
+            unique_names = [of.stem for of in uargs.output_file]
         legend = True
     else:
         legend = False
@@ -928,6 +931,10 @@ def plot_abs_func(uargs, save_data_only=False):
         else:
             x_max = uargs.xlim[1]
 
+        # Trim transitions to a specific number of states if requested
+        if uargs.trim_transitions is not None:
+            abs_data.trim_to_n(uargs.trim_transitions)
+
         # Generate spectrum
         abs_data.generate_spectrum(
             fwhm=uargs.linewidth,
@@ -2016,6 +2023,28 @@ def read_args(arg_list=None):
         )
     )
 
+    gen_abs.add_argument(
+        '--unique_names',
+        action='store_true',
+        default=False,
+        help=(
+            'Attempt to shorten file names in plot legend\n'
+            'Default: %(default)s'
+        )
+    )
+
+    gen_abs.add_argument(
+        '--trim_transitions',
+        '-tt',
+        type=int,
+        default=None,
+        help=(
+            'Trim to first n transitions\n'
+            'where n is an integer\n'
+            'Default: %(default)s'
+        )
+    )
+
     plot_subprog = all_subparsers.add_parser(
         'plot',
         description='Plot data from orca file',
@@ -2159,6 +2188,28 @@ def read_args(arg_list=None):
         )
     )
 
+    plot_abs.add_argument(
+        '--unique_names',
+        action='store_true',
+        default=False,
+        help=(
+            'Attempt to shorten file names in plot legend\n'
+            'Default: %(default)s'
+        )
+    )
+
+    plot_abs.add_argument(
+        '--trim_transitions',
+        '-tt',
+        type=int,
+        default=None,
+        help=(
+            'Trim to first n transitions\n'
+            'where n is an integer\n'
+            'Default: %(default)s'
+        )
+    )
+
     plot_xes = plot_parser.add_parser(
         'xes',
         description='Plots XES from CI calculation output',

orto/data.py

@@ -181,6 +181,27 @@ class AbsorptionData():
         self._spectrum = spectrum
         return
 
+    def trim_to_n(self, n_transitions: int) -> None:
+        '''
+        Trims data to given number of transitions
+
+        Parameters
+        ----------
+        n_transitions: int
+            Number of transitions to keep from the start of the data
+
+        Returns
+        -------
+        None
+        '''
+
+        self.energies = self.energies[:n_transitions]
+        self.osc_strengths = self.osc_strengths[:n_transitions]
+
+        self.spectrum = None
+
+        return
+
     @classmethod
     def from_extractor(cls, extractor: oe.AbsorptionExtractor,
                        remove_zero_osc: float = 1E-4) -> list['AbsorptionData']: # noqa

{orto-1.7.1.dist-info → orto-1.8.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: orto
-Version: 1.7.1
+Version: 1.8.1
 Summary: A package to make life easier when performing Orca calculations.
 Home-page: https://orto.kragskow.group
 Author: Jon Kragskow

{orto-1.7.1.dist-info → orto-1.8.1.dist-info}/RECORD

@@ -1,17 +1,17 @@
 orto/__init__.py,sha256=IedlltYr3qYZxChNUdz62qogXA9Pos_MUvXdGXqAa0E,41
-orto/__version__.py,sha256=t5fxYgsky1Zd4XKxuqdV_dDlnVKVgmosmBX1AjwcKE0,22
-orto/cli.py,sha256=U2vvUHdKR6mCOpA_Jo71mpCVuxRtmRXXxgmAO1j3Hl0,80426
+orto/__version__.py,sha256=7MCtJDgGo9X0p4xZ0jS5djhQPWK-HOaXAyASTwUa4Fc,22
+orto/cli.py,sha256=L2i5FHwXSH8AdE2hZHA0VcqJv1oxxGZ3MUgPSE9Dn_E,81704
 orto/constants.py,sha256=anxaiTykO8Q_CXliR7zuOAdnXZrQ2-C4ndaviyl7kGc,419
-orto/data.py,sha256=cNO8gQj6aZP-xriWUY2HchZ6iWZxVZnlrvAD4qhWT_o,14436
+orto/data.py,sha256=960LHFeG7l626X_WA-8YxvG2toNAsgNvLo86OoYAmBY,14910
 orto/exceptions.py,sha256=D7oNeAEGeJNt5thzt6PaCn5FY6JcbJOWUE1N1LVhhuE,159
 orto/extractor.py,sha256=d2_pGQeIjK_JbsMINTawUgcO2vcPq0b9uQj9VKuI720,75953
 orto/input.py,sha256=N8JbySSVEC_qmXZ7ppJsZ7Z1qel6PfalGYRtnX1hJ6U,9900
 orto/job.py,sha256=SM0nlc_bqhhPvfuuykhMvaUnkwC3Gp-6RvYw_a0TyGc,5855
 orto/plotter.py,sha256=IATM7vBauT3StFXqk_4DKzCSiTnUunrCeNYRBKM3eUE,17695
 orto/utils.py,sha256=GcMZ9uebOSnkPQT_U5O0X49LUtTu8YpXZxEsNKgXNTY,9838
-orto-1.7.1.dist-info/licenses/LICENSE,sha256=46mU2C5kSwOnkqkw9XQAJlhBL2JAf1_uCD8lVcXyMRg,7652
-orto-1.7.1.dist-info/METADATA,sha256=J4Ip0QocQOfRlV13ffkXh-sEgGKGcdGDszZvqa3ukKQ,1184
-orto-1.7.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-orto-1.7.1.dist-info/entry_points.txt,sha256=HXenCglMp_03JkN34pK2phkjXK9CFcXTGHKv5QaVY8I,39
-orto-1.7.1.dist-info/top_level.txt,sha256=hQ-z28gTN_FZ2B5Kiwxr_9cUTcCoib9W5HjbkceDXw4,5
-orto-1.7.1.dist-info/RECORD,,
+orto-1.8.1.dist-info/licenses/LICENSE,sha256=46mU2C5kSwOnkqkw9XQAJlhBL2JAf1_uCD8lVcXyMRg,7652
+orto-1.8.1.dist-info/METADATA,sha256=iFQdVfoNx5SpWOOcw4gC69tj7zewzVcQq4Wugv781xE,1184
+orto-1.8.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+orto-1.8.1.dist-info/entry_points.txt,sha256=HXenCglMp_03JkN34pK2phkjXK9CFcXTGHKv5QaVY8I,39
+orto-1.8.1.dist-info/top_level.txt,sha256=hQ-z28gTN_FZ2B5Kiwxr_9cUTcCoib9W5HjbkceDXw4,5
+orto-1.8.1.dist-info/RECORD,,