qrotor 4.1.1__tar.gz → 4.1.2__tar.gz

{qrotor-4.1.1 → qrotor-4.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: qrotor
-Version: 4.1.1
+Version: 4.1.2
 Summary: QRotor
 Author: Pablo Gila-Herranz
 Author-email: pgila001@ikasle.ehu.eus

{qrotor-4.1.1 → qrotor-4.1.2}/qrotor/_version.py

@@ -11,5 +11,5 @@ https://semver.org/
 ---
 """
 
-__version__ = "v4.1.1"
+__version__ = "v4.1.2"
 

{qrotor-4.1.1 → qrotor-4.1.2}/qrotor/constants.py

@@ -51,13 +51,13 @@ I_ND3 = 3 * (periodictable.D.mass * _amu * r_NH**2)
 # Rotational energy
 _hbar = const.physical_constants['reduced Planck constant'][0]
 B_CH3 = ((_hbar**2) / (2 * I_CH3)) * (1000 / const.eV)
-"""Rotational energy of CH3, in meV·s/kg·m^2."""
+"""Kinetic rotational energy of CH3, in meV·s/kg·m^2."""
 B_CD3 = ((_hbar**2) / (2 * I_CD3)) * (1000 / const.eV)
-"""Rotational energy of CD3, in meV·s/kg·m^2."""
+"""Kinetic rotational energy of CD3, in meV·s/kg·m^2."""
 B_NH3 = ((_hbar**2) / (2 * I_NH3)) * (1000 / const.eV)
-"""Rotational energy of NH3, in meV·s/kg·m^2."""
+"""Kinetic rotational energy of NH3, in meV·s/kg·m^2."""
 B_ND3 = ((_hbar**2) / (2 * I_ND3)) * (1000 / const.eV)
-"""Rotational energy of ND3, in meV·s/kg·m^2."""
+"""Kinetic rotational energy of ND3, in meV·s/kg·m^2."""
 
 # Potential constants from titov2023 [C1, C2, C3, C4, C5]
 constants_titov2023 = [

{qrotor-4.1.1 → qrotor-4.1.2}/qrotor/plot.py

@@ -34,6 +34,8 @@ def potential(
         marker='',
         linestyle='-',
         cm:bool=False,
+        normalize:bool=False,
+        ylim:tuple=None,
         ) -> None:
     """Plot the potential values of `data` (System object, or list of systems).
 
@@ -42,8 +44,13 @@ def potential(
 
     `marker` and `linestyle` can be a Matplotlib string or list of strings.
     Optionally, the Viridis colormap can be used with `cm = True`.
+
+    Set `normalize = True` to normalize by their respective `qrotor.system.System.potential_max`.
+    This can be useful if you have performed subtractions or similar operations.
+    In this case, you might also want to play with `ylim` to adjust the y-axis limits.
     """
-    system = systems.as_list(data)
+    data_copy = deepcopy(data)
+    system = systems.as_list(data_copy)
     title_str = title if title else (system[0].comment if (system[0].comment and (len(system) == 1 or not system[-1].comment)) else 'Rotational potential energy')
     # Marker as a list
     if isinstance(marker, list):
@@ -62,6 +69,15 @@ def potential(
     plt.title(title_str)
     plt.xlabel('Angle / rad')
     plt.ylabel('Potential energy / meV')
+
+    if normalize:
+        plt.ylabel('Energy / V$_{3}$')
+        for s in system:
+            s.potential_values = s.potential_values / s.potential_max
+
+    if ylim:
+        plt.ylim(ylim)
+
     plt.xticks([-2*np.pi, -3*np.pi/2, -np.pi, -np.pi/2, 0, np.pi/2, np.pi, 3*np.pi/2, 2*np.pi], [r'$-2\pi$', r'$-\frac{3\pi}{2}$', r'$-\pi$', r'$-\frac{\pi}{2}$', '0', r'$\frac{\pi}{2}$', r'$\pi$', r'$\frac{3\pi}{2}$', r'$2\pi$'])
 
     if cm:  # Plot using a colormap

{qrotor-4.1.1 → qrotor-4.1.2}/qrotor/potential.py

@@ -372,7 +372,10 @@ def interpolate(system:System) -> System:
     grid = system.grid
     gridsize = system.gridsize
     new_grid = np.linspace(0, 2*np.pi, gridsize)
-    cubic_spline = CubicSpline(grid, V)
+    # Impose periodic boundary conditions
+    grid_periodic = np.append(grid, grid[0] + 2*np.pi)
+    V_periodic = np.append(V, V[0])
+    cubic_spline = CubicSpline(grid_periodic, V_periodic, bc_type='periodic')
     new_V = cubic_spline(new_grid)
     system.grid = new_grid
     system.potential_values = new_V

{qrotor-4.1.1 → qrotor-4.1.2}/qrotor/system.py

@@ -61,7 +61,7 @@ class System:
         if not B:
             B = self.B
         self.B: float = B
-        """Rotational inertia, as in $B=\\frac{\\hbar^2}{2I}$.
+        """Kinetic rotational energy, as in $B=\\frac{\\hbar^2}{2I}$.
 
         Defaults to the value for a methyl group.
         """

{qrotor-4.1.1 → qrotor-4.1.2}/qrotor.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: qrotor
-Version: 4.1.1
+Version: 4.1.2
 Summary: QRotor
 Author: Pablo Gila-Herranz
 Author-email: pgila001@ikasle.ehu.eus

{qrotor-4.1.1 → qrotor-4.1.2}/qrotor.egg-info/SOURCES.txt

@@ -15,6 +15,7 @@ qrotor.egg-info/SOURCES.txt
 qrotor.egg-info/dependency_links.txt
 qrotor.egg-info/requires.txt
 qrotor.egg-info/top_level.txt
+tests/__init__.py
 tests/test_constants.py
 tests/test_potential.py
 tests/test_rotate.py

{qrotor-4.1.1 → qrotor-4.1.2}/qrotor.egg-info/top_level.txt

@@ -1 +1,2 @@
 qrotor
+tests