orto 0.26.3__py3-none-any.whl → 0.27.0__py3-none-any.whl

orto/__version__.py

@@ -1 +1 @@
-__version__ = '0.26.3'
+__version__ = '0.27.0'

orto/cli.py

@@ -108,7 +108,7 @@ def extract_sf_energies_func(uargs):
                 if len(all_data) > 1:
                     f.write(f'# Section {it + 1:d}\n')
 
-                f.write('State, Root, Multiplicity, Relative Energy (cm^-1)\n')
+                f.write('State, Root, Multiplicity, Relative Energy (cm⁻¹)\n')
 
                 f.write(
                     f'1, {data['root'][0]:d}, {data['multiplicity'][0]:d}, 0\n' # noqa
@@ -190,7 +190,7 @@ def extract_so_energies_func(uargs):
             if len(all_energies) > 1:
                 f.write(f'# Section {it + 1:d}\n')
 
-            f.write('State, Relative Energy (cm^-1)\n')
+            f.write('State, Relative Energy (cm⁻¹)\n')
 
             for eit, energy in enumerate(energies):
                 f.write(
@@ -701,7 +701,7 @@ def distort_func(uargs):
     data = oe.FrequencyExtractor.extract(uargs.output_file)
 
     ut.cprint(
-        'Distorting along mode #{}:  {: .2f} cm^-1'.format(
+        'Distorting along mode #{}:  {: .2f} cm⁻¹'.format(
             uargs.mode_number,
             data[0]['energy (cm^-1)'][uargs.mode_number]
             ),
@@ -907,7 +907,7 @@ def extract_freq_func(uargs, save=True):
         uargs.num = len(data[0]['energy (cm^-1)'])
 
     if not save:
-        print('Frequencies (cm^-1) and intensities (km/mol)')
+        print('Frequencies (cm⁻¹) and intensities (km/mol)')
         for frq, inty in zip(
             data[0]['energy (cm^-1)'][:uargs.num],
             data[0]['IR Intensity (km mol^-1)'][:uargs.num]
@@ -919,7 +919,7 @@ def extract_freq_func(uargs, save=True):
         with open(out_name, 'w') as f:
             f.write(
                 f'# Frequencies and intensities from {uargs.output_file}\n')
-            f.write('# Frequency (cm^-1), Intensity (km/mol)\n')
+            f.write('# Frequency (cm⁻¹), Intensity (km/mol)\n')
             for frq, inty in zip(
                 data[0]['energy (cm^-1)'][:uargs.num],
                 data[0]['IR Intensity (km mol^-1)'][:uargs.num]

orto/extractor.py

@@ -424,8 +424,8 @@ class AILFTOrbEnergyExtractor(extto.BetweenExtractor):
 
 class FrequencyExtractor(extto.BetweenExtractor):
     '''
-    Extracts Vibrational mode energies, eigenvectors, and intensity \n
-    information from output file
+    Extracts Vibrational mode energies, eigenvectors, intensities, \n
+    and irreps from output file
     '''
 
     # Regex Start Pattern
@@ -448,6 +448,7 @@ class FrequencyExtractor(extto.BetweenExtractor):
          - tx\n
          - ty\n
          - tz\n
+         - irrep\n
         and all values are numpy arrays.\n
         Dimensions are n_atoms*3, 1 for all arrays other than displacements.
         '''
@@ -477,37 +478,51 @@ class FrequencyExtractor(extto.BetweenExtractor):
             [float(val) for val in freq_pattern.findall(block)]
         )
 
+        # Check if symmetry is enabled by searching for 
+        # "Point group" in block
+        if re.search(r'Point group', block):
+            symmetry = True
+        else:
+            symmetry = False
+
         n_atoms = len(wavenumbers) // 3
 
+        # Displacements section can be different depending on symmetry
+        if symmetry:
+            displacement_pattern = re.compile(
+                r'(?: +\d+){1,10}\s(?: +\d+\-[A-Za-z]+[0-9]*[A-Za-z]*){1,10}\s((?: +\d+ +(?: +-?\d\.\d{6}){1,10}\s)*)' # noqa
+            )
+        else:
+            displacement_pattern = re.compile(
+                r'(?: +\d+){1,6} +\s((?: +\d+ +(?: +-?\d\.\d{6}){1,6}\s)*)'
+            )
+
         # Extract Displacements
         disp_table = re.findall(
-            r'((?: +\d+){1,6} +\s(?: +\d+ +(?: +-?\d\.\d{6}){1,6}\s)*)',
+            displacement_pattern,
             block
         )
 
         # and combine in a single dataframe
-        master = pd.read_csv(
-                StringIO(disp_table[0]),
-                sep=r'\s{2,}',
-                engine='python',
-                index_col=0
-            )
-
-        for chunk in disp_table[1:]:
-            _df = pd.read_csv(
-                StringIO(chunk),
-                sep=r'\s{2,}',
-                engine='python',
-                index_col=0
+        all_df = [
+            pd.read_csv(
+                StringIO(dt),
+                sep=r'\s+',
+                index_col=[0],
+                header=None,
+                skipinitialspace=True,
             )
-            master = master.join(_df)
+            for dt in disp_table
+        ]
+        # Combine list of dataframe chunks into a single dataframe
+        combined_df = pd.concat(all_df, axis=1, join='outer')
 
         # convert to numpy array
-        disp_table = master.to_numpy()
+        disp_array = combined_df.to_numpy()
         # and reshape
-        disp_x = disp_table[0::3, :]
-        disp_y = disp_table[1::3, :]
-        disp_z = disp_table[2::3, :]
+        disp_x = disp_array[0::3, :]
+        disp_y = disp_array[1::3, :]
+        disp_z = disp_array[2::3, :]
         disp = np.zeros((n_atoms, n_atoms * 3, 3))
         disp[:, :, 0] = disp_x
         disp[:, :, 1] = disp_y
@@ -586,6 +601,25 @@ class FrequencyExtractor(extto.BetweenExtractor):
         # Convert to units of 1000 cm mol^-1
         eps = alin * 100
 
+        # Get irreducible representations
+        if symmetry:
+            irrep_pattern = re.compile(
+                r'(?: +\d+){1,10}\s((?: +\d+\-[A-Za-z]+[0-9]*[A-Za-z]*){1,10}\s)(?: +\d+ +(?: +-?\d\.\d{6}){1,10}\s)*' # noqa
+            )
+            irrep_blocks = re.findall(
+                irrep_pattern,
+                block
+            )
+            irreps = np.concatenate([
+                [
+                    irrep
+                    for irrep in iblock.replace('\n','').split()
+                ]
+                for iblock in irrep_blocks
+            ])
+        else:
+            irreps = np.array([''] * len(wavenumbers))
+
         data = {
             'energy (cm^-1)': wavenumbers,
             'displacements': disp,
@@ -594,7 +628,8 @@ class FrequencyExtractor(extto.BetweenExtractor):
             't2 (a.u.^2)': t2,
             'tx (a.u.)': tx,
             'ty (a.u.)': ty,
-            'tz (a.u.)': tz
+            'tz (a.u.)': tz,
+            'irrep': irreps
         }
 
         return data

{orto-0.26.3.dist-info → orto-0.27.0.dist-info}/METADATA

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

orto-0.27.0.dist-info/RECORD

@@ -0,0 +1,13 @@
+orto/__init__.py,sha256=iSR4TAfpJn1_mwTnnncgwmWBz3q_ImdmhnOORKcOvE0,63
+orto/__version__.py,sha256=m4LkmsY19lpwrK_42z1nyRmX2kinCvycxpRYq9Tomek,23
+orto/cli.py,sha256=ljRUy_dJyJwJv4HHZNkZeGocd16nCiT7ZUoWSUdNOpc,59395
+orto/extractor.py,sha256=SSjfDPwtDU3RiAY7qjpmD2Gmxg8AT3zOawhfkwnHpIE,67554
+orto/input.py,sha256=UMYNPGeqOCNKvPYJVfVS2JU7Ost_xM1ZG28t5f9-Er0,7084
+orto/job.py,sha256=SM0nlc_bqhhPvfuuykhMvaUnkwC3Gp-6RvYw_a0TyGc,5855
+orto/plotter.py,sha256=ICrO03T_HGe-H1XKZ2qzsKYdPY44E0PKiXqIQQawd7I,15633
+orto/utils.py,sha256=gVfGplkfc6xGYgLMi_7I_yAdWG-QKRaqQdy9v5F4Mck,7279
+orto-0.27.0.dist-info/METADATA,sha256=n5UZZmsrpg5MTuJ-ug-OESbjxeJ7AE8CC9eUv5ayhfY,1141
+orto-0.27.0.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+orto-0.27.0.dist-info/entry_points.txt,sha256=HXenCglMp_03JkN34pK2phkjXK9CFcXTGHKv5QaVY8I,39
+orto-0.27.0.dist-info/top_level.txt,sha256=hQ-z28gTN_FZ2B5Kiwxr_9cUTcCoib9W5HjbkceDXw4,5
+orto-0.27.0.dist-info/RECORD,,

{orto-0.26.3.dist-info → orto-0.27.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.7.1)
+Generator: setuptools (80.8.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

orto-0.26.3.dist-info/RECORD

@@ -1,13 +0,0 @@
-orto/__init__.py,sha256=iSR4TAfpJn1_mwTnnncgwmWBz3q_ImdmhnOORKcOvE0,63
-orto/__version__.py,sha256=2FwstevkfcPI8LAlvZ09KlLyS7-7r6zK6p3zBeY2Ms8,23
-orto/cli.py,sha256=dtFxOCyLl-onr9fr89r3frbrRWe3El7EurN85c_FYNY,59385
-orto/extractor.py,sha256=Gr3qnwHh77zrDrIe_RD8S2Lq5RXjLCa-B3SYM4PyskU,66298
-orto/input.py,sha256=UMYNPGeqOCNKvPYJVfVS2JU7Ost_xM1ZG28t5f9-Er0,7084
-orto/job.py,sha256=SM0nlc_bqhhPvfuuykhMvaUnkwC3Gp-6RvYw_a0TyGc,5855
-orto/plotter.py,sha256=ICrO03T_HGe-H1XKZ2qzsKYdPY44E0PKiXqIQQawd7I,15633
-orto/utils.py,sha256=gVfGplkfc6xGYgLMi_7I_yAdWG-QKRaqQdy9v5F4Mck,7279
-orto-0.26.3.dist-info/METADATA,sha256=LSdKlscqgLHhAMYW6ZWoDl8G_9D6a94VCrj2fVtZ_cc,1141
-orto-0.26.3.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
-orto-0.26.3.dist-info/entry_points.txt,sha256=HXenCglMp_03JkN34pK2phkjXK9CFcXTGHKv5QaVY8I,39
-orto-0.26.3.dist-info/top_level.txt,sha256=hQ-z28gTN_FZ2B5Kiwxr_9cUTcCoib9W5HjbkceDXw4,5
-orto-0.26.3.dist-info/RECORD,,