from dace import *

temperature_range=[50,300]
pressure_range=[1e-8,1e2]

### select linelist for each molecule
default_linelist = None # to print available linelists
default_linelist = "first-found" # select first one

molecules = {
# "C2H2": default_linelist,
# "C2H4": default_linelist,
"CH4":  default_linelist,
# "CO2":  default_linelist,
# "CO":   default_linelist,
# "H2O":  default_linelist,
# "H2S":  default_linelist,
# "HCN":  default_linelist,
# "MgH":  default_linelist,
# "MgO":  default_linelist,
# "NaH":  default_linelist,
# "NaOH": default_linelist,
# "NH3":  default_linelist,
# "O2":   default_linelist,
# "OCS":  default_linelist,
# "OH":   default_linelist,
# "SiH2": default_linelist,
# "SiH4": default_linelist,
# "SiH":  default_linelist,
# "SO2":  default_linelist,
# "TiO":  default_linelist,
# "VO":   default_linelist,
}
for mol in molecules:
    download_molecule(molecule=mol,temperature_range=temperature_range,pressure_range=pressure_range, line_list=molecules[mol], output_dir="datadir/dace")


#### Atoms don't work for now, reason unknown
# atoms = {
# "K":    {"linelist":default_linelist, "charge":0},
# "Na":   {"linelist":default_linelist, "charge":0},
# }
# for atom in atoms:
#     download_atom(atom,temperature_range=temperature_range,pressure_range=pressure_range, line_list=atoms[atom]["linelist"], charge=atoms[atom]["charge"],  output_dir="datadir/dace")