from numpy import array

from ase.thermochemistry import HinderedThermo

vibs = array([3049.060670,
              3040.796863,
              3001.661338,
              2997.961647,
              2866.153162,
              2750.855460,
              1436.792655,
              1431.413595,
              1415.952186,
              1395.726300,
              1358.412432,
              1335.922737,
              1167.009954,
              1142.126116,
              1013.918680,
              803.400098,
              783.026031,
              310.448278,
              136.112935,
              112.939853,
              103.926392,
              77.262869,
              60.278004,
              25.825447])
vib_energies = vibs / 8065.54429  # convert to eV from cm^-1
trans_barrier_energy = 0.049313   # eV
rot_barrier_energy = 0.017675     # eV
sitedensity = 1.5e15              # cm^-2
rotationalminima = 6
symmetrynumber = 1
mass = 30.07                      # amu
inertia = 73.149                  # amu Ang^-2

thermo = HinderedThermo(vib_energies=vib_energies,
                        trans_barrier_energy=trans_barrier_energy,
                        rot_barrier_energy=rot_barrier_energy,
                        sitedensity=sitedensity,
                        rotationalminima=rotationalminima,
                        symmetrynumber=symmetrynumber,
                        mass=mass,
                        inertia=inertia)

F = thermo.get_helmholtz_energy(temperature=298.15)
