"""
Module with a function for adding the Marleau & Cumming (2014)
evolutionary tracks to the database.
"""
import json
import os
from pathlib import Path
import h5py
import numpy as np
import pooch
from astropy.io import fits
from beartype import beartype
from species.core import constants
[docs]
@beartype
def add_marleau2014(database: h5py._hl.files.File, input_path: str) -> None:
"""
Function for adding the `Marleau & Cumming (2014)
<https://ui.adsabs.harvard.edu/abs/2014MNRAS.437.1378M>`_
isochrone data to the database.
Parameters
----------
database : h5py._hl.files.File
Database.
input_path : str
Folder where the data is located.
Returns
-------
NoneType
None
"""
iso_tag = "marleau2014"
url = "https://home.strw.leidenuniv.nl/~stolker/species/evolution_luminosity.fits"
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="218bd69105206373389c8d513969c500ad39ec32a5bd98e248aedf0b3e29af67",
fname=input_file,
path=input_path,
progressbar=True,
)
url = "https://home.strw.leidenuniv.nl/~stolker/species/evolution_radius.fits"
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="6aa0e75b3e54a427caa124ac9f1906ac180245927238a00a57acda8d5e7bf75d",
fname=input_file,
path=input_path,
progressbar=True,
)
url = "https://home.strw.leidenuniv.nl/~stolker/species/evolution_points.json"
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="592415d9a5425211b6e1bdc1b0e36c7e45d317165dca639fbea7d77cbfcafb9d",
fname=input_file,
path=input_path,
progressbar=True,
)
print("\nAdding isochrones: Marleau & Cumming (2014)...", end="", flush=True)
# Bolometric luminosity log10(L/Lsun)
data_lbol = fits.getdata(os.path.join(input_path, "evolution_luminosity.fits"))
data_lbol = np.log10(data_lbol)
# Radius (Rjup)
data_radius = fits.getdata(os.path.join(input_path, "evolution_radius.fits"))
file_name = os.path.join(input_path, "evolution_points.json")
with open(file_name, "r", encoding="utf-8") as json_file:
grid_points = json.load(json_file)
# Select Solar Helium and Deuterium fraction, and zero core mass
data_lbol = data_lbol[..., 1, 1, 0]
data_radius = data_radius[..., 1, 1, 0]
model_param = ["age", "mass", "s_init"]
dgroup = database.create_group(f"isochrones/{iso_tag}")
dgroup.attrs["model"] = iso_tag
dgroup.attrs["regular_grid"] = True
dgroup.attrs["n_param"] = len(model_param)
for i, item in enumerate(model_param):
dgroup.attrs[f"parameter{i}"] = item
data_teff = (
(10.0**data_lbol * constants.L_SUN)
/ (4.0 * np.pi * (data_radius * constants.R_JUP) ** 2 * constants.SIGMA_SB)
) ** (1.0 / 4.0)
data_mass = np.array(grid_points["mass"])
data_mass = np.broadcast_to(data_mass[None, :, None], data_lbol.shape)
# 1e2 to convert from SI to cgs
data_logg = np.log10(
1e2
* constants.GRAVITY
* (data_mass * constants.M_JUP)
/ (data_radius * constants.R_JUP) ** 2
)
database.create_dataset(
f"isochrones/{iso_tag}/mass", data=grid_points["mass"]
) # (Mjup)
database.create_dataset(
f"isochrones/{iso_tag}/age", data=grid_points["age"]
) # (Myr)
database.create_dataset(
f"isochrones/{iso_tag}/log_lum", data=data_lbol
) # log(L/Lsun)
database.create_dataset(f"isochrones/{iso_tag}/radius", data=data_radius) # (Rjup)
database.create_dataset(
f"isochrones/{iso_tag}/s_init", data=grid_points["s_i"]
) # (k_b/baryon)
database.create_dataset(f"isochrones/{iso_tag}/teff", data=data_teff) # (K)
database.create_dataset(f"isochrones/{iso_tag}/log_g", data=data_logg) # log(g)
print(" [DONE]")
