Skip to main content

QRotor

Project description

QRotor is a Python package used to study molecular rotations, such as those of methyl and amine groups. It can calculate their quantum energy levels and wavefunctions, along with excitations and tunnel splittings. These quantum systems are represented by the qrotor.System() object.

QRotor can obtain custom potentials from DFT, which are used to solve the quantum system.


Installation

As always, it is recommended to install your packages in a virtual environment:

python3 -m venv .venv
source .venv/bin/activate

With pip

Install or upgrade ATON with

pip install qrotor -U

From source

Optionally, you can install ATON from the GitHub repo. Clone the repository or download the latest stable release as a ZIP, unzip it, and run inside it:

pip install .

Documentation

QRotor contains the following modules:

qrotor.system Definition of the quantum System object
qrotor.systems Functions to manage several System objects, such as a list of systems
qrotor.rotate Rotate specific atoms from structural files
qrotor.constants Common bond lengths and inertias
qrotor.potential Potential definitions and loading functions
qrotor.solve Solve rotation eigenvalues and eigenvectors
qrotor.plot Plotting functions

Check the full documentation online.


Usage

Solving quantum rotational systems

A basic calculation of the eigenvalues for a zero potential goes as follows. Note that the default energy unit is meV unless stated otherwise.

import qrotor as qr
system = qr.System()
system.gridsize = 200000  # Size of the potential grid
system.B = 1  # Rotational inertia
system.potential_name = 'zero'
system.solve()
system.eigenvalues
# [0.0, 1.0, 1.0, 4.0, 4.0, 9.0, 9.0, ...]  # approx values

The accuracy of the calculation increases with bigger gridsizes, but note that the runtime increases exponentially.

The same calculation can be performed for a methyl group, in a cosine potential of amplitude 30 meV:

import qrotor as qr
system = qr.System()
system.gridsize = 200000  # Size of the potential grid
system.B = qr.B_CH3  # Rotational inertia of a methyl group
system.potential_name = 'cosine'
system.potential_constants = [0, 30, 3, 0]  # Offset, max, freq, phase (for cos pot.)
system.solve()
# Plot potential and eigenvalues
qr.plot.energies(system)
# Plot the first wavefunctions
qr.plot.wavefunction(system, levels=[0,1,2], square=True)

Custom potentials from DFT

QRotor can be used to obtain custom rotational potentials from DFT calculations. Using Quantum ESPRESSO, running an SCF calculation for a methyl rotation every 10 degrees:

import qrotor as qr
from aton import api
# Approx crystal positions of the atoms to rotate
atoms = [
    '1.101   1.204   1.307'
    '2.102   2.205   2.308'
    '3.103   3.206   3.309'
]
# Create the input SCF files, saving the filenames to a list
scf_files = qr.rotate.structure_qe('molecule.in', positions=atoms, angle=10, repeat=True)
# Run the Quantum ESPRESSO calculations
api.slurm.sbatch(files=scf_files)

To load the calculated potential to a QRotor System,

# Compile a 'potential.csv' file with the calculated potential as a function of the angle
qr.potential.from_qe()
# Load to the system
system = qr.potential.load()
# Solve the system, interpolating to a bigger gridsize
system.B = qr.B_CH3
system.solve(200000)
qr.plot.energies(system)

Tunnel splittings and excitations

Tunnel splittings, excitations and energy level degeneracy below the potential maximum are also calculated upon solving the system:

system.solve()
system.splittings
system.excitations
system.deg

An integer System.deg degeneracy (e.g. 3 for methyls) indicates that the energy levels have been properly estimated. However, if the degeneracy is a float instead, please check the splittings and excitations manually from the system eigenvalues.

To export the energies and the tunnel splittings of several calculations to a CSV file:

calculations = [system1, system2, system3]
qr.systems.save_energies(calculations)
qr.systems.save_splittings(calculations)

Excitations are calculated using the mean for each energy level with respect to the ground state. Tunnel splittings for each level are calculated as the difference between A and E, considering the mean of the eigenvalues for each sublevel. See R. M. Dimeo, American Journal of Physics 71, 885–893 (2003) and A. J. Horsewill, Progress in Nuclear Magnetic Resonance Spectroscopy 35, 359–389 (1999) for further reference.


Contributing

If you are interested in opening an issue or a pull request, please feel free to do so on GitHub.
For major changes, please get in touch first to discuss the details.

Code style

Please try to follow some general guidelines:

  • Use a code style consistent with the rest of the project.
  • Include docstrings to document new additions.
  • Include automated tests for new features or modifications, see automated testing.
  • Arrange function arguments by order of relevance.

Automated testing

If you are modifying the source code, you should run the automated tests of the tests/ folder to check that everything works as intended. To do so, first install PyTest in your environment,

pip install pytest

And then run PyTest inside the main directory,

pytest -vv

Compiling the documentation

The documentation can be compiled automatically to docs/qrotor.html with Pdoc and ATON, by running:

python3 makedocs.py

This runs Pdoc, updating links and pictures, and using the custom theme CSS template from the css/ folder.


Citation

QRotor is currently under development. Please cite it if you use it in your research,

Pablo Gila-Herranz, QRotor, https://pablogila.github.io/qrotor


License

Copyright (C) 2025 Pablo Gila-Herranz
This program is free software: you can redistribute it and/or modify it under the terms of the GNU Affero General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.
This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
See the attached GNU Affero General Public License for more details.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

qrotor-4.0.1.tar.gz (39.8 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

qrotor-4.0.1-py3-none-any.whl (40.7 kB view details)

Uploaded Python 3

File details

Details for the file qrotor-4.0.1.tar.gz.

File metadata

  • Download URL: qrotor-4.0.1.tar.gz
  • Upload date:
  • Size: 39.8 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.12.3

File hashes

Hashes for qrotor-4.0.1.tar.gz
Algorithm Hash digest
SHA256 6dba0f821da4b79953ce9fda4e006aac11be0ca5883bdf77dd7d6886c3e9d198
MD5 b8428e28f2c38ced2ca3f0e80f9acd7a
BLAKE2b-256 02be7cf490072b36036e0e220d47d25ce50dc632c9fbfb219363407633fc8a91

See more details on using hashes here.

File details

Details for the file qrotor-4.0.1-py3-none-any.whl.

File metadata

  • Download URL: qrotor-4.0.1-py3-none-any.whl
  • Upload date:
  • Size: 40.7 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.12.3

File hashes

Hashes for qrotor-4.0.1-py3-none-any.whl
Algorithm Hash digest
SHA256 c55cadf4a8f52ebc903861353f9cd83b8406a6bf98c9c3525a32c8e7a02d8d52
MD5 a4bdd00c06fa11b5f826b797b5659a91
BLAKE2b-256 b120bb6efbaff1066870cf6935fcc2bf49af78c0e4f8c6a740aaf55780b9df11

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page