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

orto/__version__.py

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

orto/cli.py

@@ -527,6 +527,14 @@ def gen_job_func(uargs):
         else:
             nbo_module = None
 
+        # Check structure in input file or xyz file
+        # unless skip_structure flag is set
+        if not uargs.skip_structure:
+            try:
+                inp.check_structure(oj.input_file)
+            except (DataNotFoundError, DataFormattingError) as e:
+                ut.red_exit(str(e))
+
         # Set SLURM error and output file names
         config['error'] = 'slurm.%j.e'
         config['output'] = 'slurm.%j.o'
@@ -1880,6 +1888,15 @@ def read_args(arg_list=None):
         )
     )
 
+    gen_job.add_argument(
+        '--skip_structure',
+        '-ss',
+        action='store_true',
+        help=(
+            'Disables checking of structure in input file or xyz file'
+        )
+    )
+
     gen_job.add_argument(
         '-om',
         '--orca_load',

orto/extractor.py

@@ -2204,6 +2204,85 @@ class XYZInputExtractor(extto.LineExtractor):
         return _ext.data
 
 
+class StructureInputExtractor(extto.BetweenExtractor):
+    '''
+    Extracts structure from xyz file
+    '''
+
+    # Regex Start Pattern
+    START_PATTERN = rb'(?<=\*xyz )'
+
+    # Regex End Pattern
+    END_PATTERN = rb'(?=\*)'
+
+    @property
+    def data(self) -> tuple[list[str], NDArray]:
+        '''
+        list of atom labels and ndarray of coordinates (N, 3)
+        '''
+        return self._data
+
+    @staticmethod
+    def _process_block(block: str) -> tuple[list[str], NDArray]:
+        '''
+        Converts single block into data entries described in self.data
+
+        Parameters
+        ----------
+        block: str
+            String block extracted from file
+
+        Returns
+        -------
+        tuple[list[str], NDArray]
+        '''
+
+        # Get labels and coordinates
+        all_info = re.findall(
+            r'([A-Za-z]+\s*-?\d\.\d*\s+-?\d\.\d*\s+-?\d\.\d*)',
+            block
+        )
+
+        labels = re.findall(
+            r'([A-Za-z]+)',
+            block
+        )
+
+        coords = np.asarray([
+            [float(v) for v in val.split()[1:]]
+            for val in all_info
+        ])
+
+        if len(labels) != coords.shape[0]:
+            raise DataFormattingError(
+                'Number of atom labels does not match number of coordinates'
+            )
+
+        data = (labels, coords)
+
+        return data
+
+    @classmethod
+    def extract(cls, file_name: str | pathlib.Path) -> tuple[list[str], NDArray]: # noqa
+        '''
+        Convenience method which instantiates class, extracts blocks, and
+        returns processed datasets
+
+        Parameters
+        ----------
+        file_name: str | pathlib.Path
+            File to parse
+
+        Returns
+        -------
+        tuple[list[str], NDArray]
+            Each entry contains processed data, as defined in cls.data
+        '''
+        _ext = cls()
+        _ext(file_name, process=True)
+        return _ext.data
+
+
 class IntInputExtractor(XYZInputExtractor):
     '''
     Extracts *int line of an input file

orto/input.py

@@ -1,6 +1,7 @@
 from pathlib import Path
 import re
 import numpy as np
+import numpy.linalg as la
 import copy
 import xyz_py as xyzp
 
@@ -304,6 +305,77 @@ def check_xyz(file_name: str | Path, skip_check) -> None:
     return
 
 
+def check_structure(file_name: str | Path, threshold: float = 0.1) -> None:
+    '''
+    Checks that no atoms of the structure are overlapping.\n
+
+    Parameters
+    ----------
+    file_name: str | Path
+        Orca input file as either name or Path object
+    threshold: float
+        Minimum allowed distance between atoms in Angstroms
+
+    Returns
+    -------
+    None
+
+    Raises
+    ------
+    DataFormattingError
+        If any atoms are overlapping
+    '''
+
+    # Get xyz file name and check it exists and is formatted correctly
+    try:
+        xyz_file = oe.XYZFileInputExtractor.extract(file_name)
+    except DataNotFoundError:
+        xyz_file = []
+
+    try:
+        xyzline = oe.XYZInputExtractor.extract(file_name)
+    except DataNotFoundError:
+        xyzline = []
+
+    if not len(xyz_file) and not len(xyzline):
+        ut.red_exit(
+            'Error: missing or incorrect *xyzfile or *xyz line in input'
+        )
+
+    if len(xyz_file):
+        # Load structure from xyz file
+        xyz_file = Path(xyz_file[0])
+        labels, coords = xyzp.load_xyz(xyz_file)
+    else:
+        # Extract structure from input file
+        labels, coords = oe.StructureInputExtractor.extract(file_name)[0]
+
+    # Compute pairwise differences using broadcasting
+    # Broadcast to shape (N, N, 3)
+    # from (N, 1, 3) and (1, N, 3)
+    diffs = coords[:, np.newaxis, :] - coords[np.newaxis, :, :]
+
+    # Compute Euclidean distances
+    distances = la.norm(diffs, axis=-1)  # shape (N, N)
+
+    # Add large value to diagonal to avoid zero self-distances
+    distances += np.diag([100]*len(labels))
+    # Find index of zero distances
+    zero_dist = np.where(
+        distances < threshold
+    )
+    # and report
+    if len(zero_dist[0]):
+        raise DataFormattingError(
+            'Error: Overlapping atoms detected in structure'
+            ' at atom numbers:\n' + ', '.join(
+                str(i+1) for i in zero_dist[0]
+            )
+        )
+
+    return
+
+
 def check_moinp_moread(file_name: str | Path) -> None:
     '''
     Checks if MORead and/or MOInp are present in the input file.\n

orto/plotter.py

@@ -154,7 +154,10 @@ def plot_absorption_spectrum(abs_data: data.AbsorptionData,
             linefmt=stickcolour
         )
         oax.yaxis.set_minor_locator(AutoMinorLocator())
-        oax.set_ylabel(r'$f_\mathregular{osc}$')
+        if normalise:
+            oax.set_ylabel(r'Normalised $f_\mathregular{osc}$')
+        else:
+            oax.set_ylabel(r'$f_\mathregular{osc}$')
         oax.set_ylim([0., oax.get_ylim()[1]])
     elif osc_style == 'combined':
         ax.stem(

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

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

orto-1.8.3.dist-info/RECORD

@@ -0,0 +1,17 @@
+orto/__init__.py,sha256=IedlltYr3qYZxChNUdz62qogXA9Pos_MUvXdGXqAa0E,41
+orto/__version__.py,sha256=U3WnIim1rPfsaJ6EKERUuRl6DH-DG8tqNKxxzBZArOA,22
+orto/cli.py,sha256=DIEcU9jMP5BPtItaDmb6vS00GJa8amoBxUwUsu6nhrQ,82209
+orto/constants.py,sha256=anxaiTykO8Q_CXliR7zuOAdnXZrQ2-C4ndaviyl7kGc,419
+orto/data.py,sha256=960LHFeG7l626X_WA-8YxvG2toNAsgNvLo86OoYAmBY,14910
+orto/exceptions.py,sha256=D7oNeAEGeJNt5thzt6PaCn5FY6JcbJOWUE1N1LVhhuE,159
+orto/extractor.py,sha256=S27297-gasqLaLbMOqW9sj1X0U0mr3OjikxHcz9Es3o,77867
+orto/input.py,sha256=uUQV6A-8D0GZpRoY1rKK_aUPmk9kVVTnMzTHuisP5t4,11888
+orto/job.py,sha256=SM0nlc_bqhhPvfuuykhMvaUnkwC3Gp-6RvYw_a0TyGc,5855
+orto/plotter.py,sha256=_t9bow6sUMoRAvD6gVFGJTc-_ifW7RmeR0JAPWK_lm4,17799
+orto/utils.py,sha256=GcMZ9uebOSnkPQT_U5O0X49LUtTu8YpXZxEsNKgXNTY,9838
+orto-1.8.3.dist-info/licenses/LICENSE,sha256=46mU2C5kSwOnkqkw9XQAJlhBL2JAf1_uCD8lVcXyMRg,7652
+orto-1.8.3.dist-info/METADATA,sha256=fokjznTSffyKFIn8KxX5G3iItEjjDqMNvbsacMNAukY,1184
+orto-1.8.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+orto-1.8.3.dist-info/entry_points.txt,sha256=HXenCglMp_03JkN34pK2phkjXK9CFcXTGHKv5QaVY8I,39
+orto-1.8.3.dist-info/top_level.txt,sha256=hQ-z28gTN_FZ2B5Kiwxr_9cUTcCoib9W5HjbkceDXw4,5
+orto-1.8.3.dist-info/RECORD,,

orto-1.8.1.dist-info/RECORD

@@ -1,17 +0,0 @@
-orto/__init__.py,sha256=IedlltYr3qYZxChNUdz62qogXA9Pos_MUvXdGXqAa0E,41
-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=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.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,,