from exomol import *

### 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,
# "K":  default_linelist,
# "MgH":  default_linelist,
# "MgO":  default_linelist,
# "Na":  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_exomol(molecule=mol, line_list=molecules[mol], output_dir="datadir/exomol/taurex_R15000", output_format="taurex")
    # download_exomol(molecule=mol, line_list=molecules[mol], output_dir="datadir/exomol/petitRADTRANS_R1000", output_format="petitRADTRANS")