from dace import *

# output_dir = "datadir/hot"
# temperature_range=[50,3000]
# pressure_range=[-8,5]

output_dir = "datadir/cold"
temperature_range=[50,300]
pressure_range=[-8,2]

### 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,
"CO":   default_linelist,
# "CO2":  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=output_dir, force_download=True)


#### 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")