<reponame>deepchatterjeeligo/em-bright
import json

from astropy.utils.data import download_file

__version__ = '1.0.1'

PACKAGE_DATA_LINKS = {
    'equil_2H.dat': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/equil_2H.dat',  # noqa: E501
    'APR4_EPP.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/APR4_EPP.pickle',  # noqa: E501
    'BHF_BBB2.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/BHF_BBB2.pickle',  # noqa: E501
    'H4.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/H4.pickle',  # noqa: E501
    'HQC18.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/HQC18.pickle',  # noqa: E501
    'KDE0V1.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/KDE0V1.pickle',  # noqa: E501
    'KDE0V.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/KDE0V.pickle',  # noqa: E501
    'MPA1.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/MPA1.pickle',  # noqa: E501
    'MS1B_PP.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/MS1B_PP.pickle',  # noqa: E501
    'MS1_PP.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/MS1_PP.pickle',  # noqa: E501
    'RS.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/RS.pickle',  # noqa: E501
    'SK255.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/SK255.pickle',  # noqa: E501
    'SK272.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/SK272.pickle',  # noqa: E501
    'SKI2.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/SKI2.pickle',  # noqa: E501
    'SKI3.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/SKI3.pickle',  # noqa: E501
    'SKI4.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/SKI4.pickle',  # noqa: E501
    'SKI5.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/SKI5.pickle',  # noqa: E501
    'SKI6.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/SKI6.pickle',  # noqa: E501
    'SKMP.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/SKMP.pickle',  # noqa: E501
    'SKOP.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/SKOP.pickle',  # noqa: E501
    'SLY230A.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/SLY230A.pickle',  # noqa: E501
    'SLY2.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/SLY2.pickle',  # noqa: E501
    'SLY9.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/SLY9.pickle',  # noqa: E501
    'SLy.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/SLy.pickle',  # noqa: E501
    'EOS_BAYES_FACTOR_MAP.json': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/EOS_BAYES_FACTOR_MAP.json',  # noqa: E501
    'EOS_MAX_MASS_MAP.json': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/EOS_MAX_MASS_MAP.json'  # noqa: E501
}

PACKAGE_FILENAMES = dict.fromkeys(PACKAGE_DATA_LINKS)

for name, url in PACKAGE_DATA_LINKS.items():
    PACKAGE_FILENAMES[name] = download_file(
        url, cache=True, pkgname='ligo.em_bright'
    )

# load and normalize bayes factors
with open(PACKAGE_FILENAMES['EOS_BAYES_FACTOR_MAP.json']) as f:
    EOS_BAYES_FACTORS = json.load(f)
    """Taken from Table II (broad prior), Approx bayes factor column
    of 10.1103/PhysRevD.104.083003
    """

_BAYES_FACTOR_NORM = sum(EOS_BAYES_FACTORS.values())
"""Sum of bayes factors above"""

EOS_BAYES_FACTORS.update(
    {k: v / _BAYES_FACTOR_NORM for k, v in EOS_BAYES_FACTORS.items()}
)

with open(PACKAGE_FILENAMES['EOS_MAX_MASS_MAP.json']) as f:
    EOS_MAX_MASS = json.load(f)
    """Taken from Table I m_max column of 10.1103/PhysRevD.104.083003"""

import json

from astropy.utils.data import download_file

__version__ = '1.0.1'

PACKAGE_DATA_LINKS = {
    'equil_2H.dat': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/equil_2H.dat',  # noqa: E501
    'APR4_EPP.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/APR4_EPP.pickle',  # noqa: E501
    'BHF_BBB2.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/BHF_BBB2.pickle',  # noqa: E501
    'H4.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/H4.pickle',  # noqa: E501
    'HQC18.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/HQC18.pickle',  # noqa: E501
    'KDE0V1.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/KDE0V1.pickle',  # noqa: E501
    'KDE0V.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/KDE0V.pickle',  # noqa: E501
    'MPA1.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/MPA1.pickle',  # noqa: E501
    'MS1B_PP.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/MS1B_PP.pickle',  # noqa: E501
    'MS1_PP.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/MS1_PP.pickle',  # noqa: E501
    'RS.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/RS.pickle',  # noqa: E501
    'SK255.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/SK255.pickle',  # noqa: E501
    'SK272.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/SK272.pickle',  # noqa: E501
    'SKI2.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/SKI2.pickle',  # noqa: E501
    'SKI3.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/SKI3.pickle',  # noqa: E501
    'SKI4.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/SKI4.pickle',  # noqa: E501
    'SKI5.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/SKI5.pickle',  # noqa: E501
    'SKI6.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/SKI6.pickle',  # noqa: E501
    'SKMP.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/SKMP.pickle',  # noqa: E501
    'SKOP.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/SKOP.pickle',  # noqa: E501
    'SLY230A.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/SLY230A.pickle',  # noqa: E501
    'SLY2.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/SLY2.pickle',  # noqa: E501
    'SLY9.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/SLY9.pickle',  # noqa: E501
    'SLy.pickle': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/SLy.pickle',  # noqa: E501
    'EOS_BAYES_FACTOR_MAP.json': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/EOS_BAYES_FACTOR_MAP.json',  # noqa: E501
    'EOS_MAX_MASS_MAP.json': 'https://git.ligo.org/deep.chatterjee/em-bright/-/raw/v1.0.1/ligo/em_bright/data/EOS_MAX_MASS_MAP.json'  # noqa: E501
}

PACKAGE_FILENAMES = dict.fromkeys(PACKAGE_DATA_LINKS)

for name, url in PACKAGE_DATA_LINKS.items():
    PACKAGE_FILENAMES[name] = download_file(
        url, cache=True, pkgname='ligo.em_bright'
    )

# load and normalize bayes factors
with open(PACKAGE_FILENAMES['EOS_BAYES_FACTOR_MAP.json']) as f:
    EOS_BAYES_FACTORS = json.load(f)
    """Taken from Table II (broad prior), Approx bayes factor column
    of 10.1103/PhysRevD.104.083003
    """

_BAYES_FACTOR_NORM = sum(EOS_BAYES_FACTORS.values())
"""Sum of bayes factors above"""

EOS_BAYES_FACTORS.update(
    {k: v / _BAYES_FACTOR_NORM for k, v in EOS_BAYES_FACTORS.items()}
)

with open(PACKAGE_FILENAMES['EOS_MAX_MASS_MAP.json']) as f:
    EOS_MAX_MASS = json.load(f)
    """Taken from Table I m_max column of 10.1103/PhysRevD.104.083003"""
# noqa: E501 # noqa: E501 # noqa: E501 # noqa: E501 # noqa: E501 # noqa: E501 # noqa: E501 # noqa: E501 # noqa: E501 # noqa: E501 # noqa: E501 # noqa: E501 # noqa: E501 # noqa: E501 # noqa: E501 # noqa: E501 # noqa: E501 # noqa: E501 # noqa: E501 # noqa: E501 # noqa: E501 # noqa: E501 # noqa: E501 # noqa: E501 # noqa: E501 # noqa: E501 # load and normalize bayes factors Taken from Table II (broad prior), Approx bayes factor column     of 10.1103/PhysRevD.104.083003 Sum of bayes factors above Taken from Table I m_max column of 10.1103/PhysRevD.104.083003