nikolasibalic / ARC-Alkali-Rydberg-Calculator

Object-oriented Python library for computation of properties of highly-excited Rydbeg states of alkali and divalent atoms.
https://atomcalc.org
BSD 3-Clause "New" or "Revised" License
91 stars 81 forks source link

AC Stark maps #139

Closed dihm closed 1 year ago

dihm commented 1 year ago

Single atom calculations that determine Stark maps of the Rydberg manifold due to an AC electric field.

Code includes a helper class (for basis generation) and two solvers: one based on Shirley's Time-Independent Floquet Hamiltonian method and the other an approximate method assuming independent 2-level systems.

Resolves #138

dihm commented 1 year ago

I've tried to match structure and style as much as possible so hopefully full integration isn't too annoying. One outstanding issue is that I haven't tested against an atom with s!=0.5. I tried to make sure s was passed around correctly but I suspect there are going to be issues.

On a related note, I noticed that there are some unused imports and old Python2 compatibility boilerplate hanging around. Would you be interested in a PR that cleans them out?

nikolasibalic commented 1 year ago

Merging this to branch on the side to do some formatting fixes before merging to main.