"""
Module with a function for adding the ATMO
evolutionary tracks to the database.
"""
from pathlib import Path
import h5py
import numpy as np
import pooch
from beartype import beartype
from species.core import constants
from species.util.data_util import extract_tarfile
[docs]
@beartype
def add_atmo(database: h5py._hl.files.File, input_path: str) -> None:
"""
Function for adding the AMES-Cond and AMES-Dusty
isochrone data to the database.
Parameters
----------
database : h5py._hl.files.File
Database.
input_path : str
Folder where the data is located.
Returns
-------
NoneType
None
"""
url = (
"https://home.strw.leidenuniv.nl/~stolker/"
"species/atmo_evolutionary_tracks.tgz"
)
iso_tag = "ATMO"
iso_size = "9.6 MB"
data_folder = Path(input_path) / "atmo_evolutionary_tracks"
if not data_folder.exists():
data_folder.mkdir()
input_file = url.rsplit("/", maxsplit=1)[-1]
data_file = Path(input_path) / input_file
if not data_file.exists():
print()
pooch.retrieve(
url=url,
known_hash="f905698a397b980acdc7a3cd7fa514a39e315c674b21dcd5cbba70833777cd3b",
fname=input_file,
path=input_path,
progressbar=True,
)
print(f"\nUnpacking {iso_tag} isochrones ({iso_size})...", end="", flush=True)
extract_tarfile(str(data_file), str(data_folder))
print(" [DONE]")
iso_files = [
"ATMO_CEQ",
"ATMO_NEQ_weak",
"ATMO_NEQ_strong",
]
labels = [
"ATMO equilibrium chemistry",
"ATMO weak non-equilibrium chemistry",
"ATMO strong non-equilibrium chemistry",
]
db_tags = [
"atmo-ceq",
"atmo-neq-weak",
"atmo-neq-strong",
]
for iso_idx, iso_item in enumerate(iso_files):
iso_path = Path(data_folder) / iso_item / "MKO_WISE_IRAC"
# Ignore hidden files
file_list = sorted(iso_path.glob("[!.]*.txt"))
for file_idx, file_item in enumerate(file_list):
# Mass (Msun) - Age (Gyr) - Teff (K) - log(L/Lsun) - Radius (Rsun) - log(g)
if file_idx == 0:
iso_data = np.loadtxt(str(file_item))
else:
iso_load = np.loadtxt(str(file_item))
iso_data = np.vstack((iso_data, iso_load))
with open(str(file_item), encoding="utf-8") as open_file:
parameters = open_file.readline()
filter_names = parameters.split()[7:]
iso_data[:, 0] *= constants.M_SUN / constants.M_JUP # (Msun) -> (Mjup)
iso_data[:, 1] *= 1e3 # (Gyr) -> (Myr)
iso_data[:, 4] *= constants.R_SUN / constants.R_JUP # (Rsun) -> (Rjup)
print(f"Adding isochrones: {labels[iso_idx]}...", end="", flush=True)
dtype = h5py.string_dtype(encoding="utf-8", length=None)
dset = database.create_dataset(
f"isochrones/{db_tags[iso_idx]}/filters",
(np.size(filter_names),),
dtype=dtype,
)
dset[...] = filter_names
database.create_dataset(
f"isochrones/{db_tags[iso_idx]}/mass", data=iso_data[:, 0]
) # (Mjup)
dset = database.create_dataset(
f"isochrones/{db_tags[iso_idx]}/age", data=iso_data[:, 1]
) # (Myr)
database.create_dataset(
f"isochrones/{db_tags[iso_idx]}/teff", data=iso_data[:, 2]
) # (K)
database.create_dataset(
f"isochrones/{db_tags[iso_idx]}/log_lum", data=iso_data[:, 3]
) # log(L/Lsun)
database.create_dataset(
f"isochrones/{db_tags[iso_idx]}/radius", data=iso_data[:, 4]
) # (Rjup)
database.create_dataset(
f"isochrones/{db_tags[iso_idx]}/log_g", data=iso_data[:, 5]
) # log(g)
database.create_dataset(
f"isochrones/{db_tags[iso_idx]}/magnitudes", data=iso_data[:, 6:]
)
dset.attrs["model"] = "atmo"
print(" [DONE]")
print(f"Database tag: {db_tags[iso_idx]}")
