# Licensed under an MIT style license -- see LICENSE.md

__author__ = ["Charlie Hoy <charlie.hoy@ligo.org>"]

GWlatex_labels = {
    "luminosity_distance": r"$d_{L} [\mathrm{Mpc}]$",
    "geocent_time": r"$t_{c} [\mathrm{s}]$",
    "dec": r"$\delta [\mathrm{rad}]$",
    "ra": r"$\alpha [\mathrm{rad}]$",
    "a_1": r"$a_{1}$",
    "a_2": r"$a_{2}$",
    "phi_jl": r"$\phi_{JL} [\mathrm{rad}]$",
    "phase": r"$\phi [\mathrm{rad}]$",
    "psi": r"$\Psi [\mathrm{rad}]$",
    "psi_J": r"$\Psi_{J} [\mathrm{rad}]$",
    "iota": r"$\iota [\mathrm{rad}]$",
    "tilt_1": r"$\theta_{1} [\mathrm{rad}]$",
    "tilt_1_infinity": r"$\theta_{1,\infty} [\mathrm{rad}]$",
    "tilt_1_infinity_only_prec_avg": (
        r"$\theta_{1,\infty}^{\mathrm{only\, prec\, avg}} [\mathrm{rad}]$"
    ),
    "tilt_2": r"$\theta_{2} [\mathrm{rad}]$",
    "tilt_2_infinity": r"$\theta_{2,\infty} [\mathrm{rad}]$",
    "tilt_2_infinity_only_prec_avg": (
        r"$\theta_{2,\infty}^{\mathrm{only\, prec\, avg}} [\mathrm{rad}]$"
    ),
    "phi_12": r"$\phi_{12} [\mathrm{rad}]$",
    "mass_2": r"$m_{2} [M_{\odot}]$",
    "mass_1": r"$m_{1} [M_{\odot}]$",
    "total_mass": r"$M [M_{\odot}]$",
    "chirp_mass": r"$\mathcal{M} [M_{\odot}]$",
    "H1_matched_filter_snr": r"$\rho^{H}_{\mathrm{mf}}$",
    "L1_matched_filter_snr": r"$\rho^{L}_{\mathrm{mf}}$",
    "network_matched_filter_snr": r"$\rho^{N}_{\mathrm{mf}}$",
    "H1_optimal_snr": r"$\rho^{H}_{\mathrm{opt}}$",
    "L1_optimal_snr": r"$\rho^{L}_{\mathrm{opt}}$",
    "V1_optimal_snr": r"$\rho^{V}_{\mathrm{opt}}$",
    "E1_optimal_snr": r"$\rho^{E}_{\mathrm{opt}}$",
    "network_optimal_snr": r"$\rho^{N}_{\mathrm{opt}}$",
    "H1_matched_filter_snr_abs": r"$\mathrm{abs}(\rho^{H}_{\mathrm{mf}})$",
    "L1_matched_filter_snr_abs": r"$\mathrm{abs}(\rho^{L}_{\mathrm{mf}})$",
    "V1_matched_filter_snr_abs": r"$\mathrm{abs}(\rho^{V}_{\mathrm{mf}})$",
    "E1_matched_filter_snr_abs": r"$\mathrm{abs}(\rho^{E}_{\mathrm{mf}})$",
    "H1_matched_filter_snr_angle": r"$\mathrm{arg}(\rho^{H}_{\mathrm{mf}})$",
    "L1_matched_filter_snr_angle": r"$\mathrm{arg}(\rho^{L}_{\mathrm{mf}})$",
    "V1_matched_filter_snr_angle": r"$\mathrm{arg}(\rho^{V}_{\mathrm{mf}})$",
    "E1_matched_filter_snr_angle": r"$\mathrm{arg}(\rho^{E}_{\mathrm{mf}})$",
    "network_21_multipole_snr": r"$\rho_{21}$",
    "network_33_multipole_snr": r"$\rho_{33}$",
    "network_44_multipole_snr": r"$\rho_{44}$",
    "network_precessing_snr": r"$\rho_{\mathrm{p}}$",
    "_b_bar": r"$\bar{b}$",
    "_precessing_harmonics_overlap": r"$|\mathrm{O}^{\mathrm{prec}}_{0,1}|$",
    "H1_time": r"$t_{H} [\mathrm{s}]$",
    "L1_time": r"$t_{L} [\mathrm{s}]$",
    "V1_time": r"$t_{V} [\mathrm{s}]$",
    "E1_time": r"$t_{E} [\mathrm{s}]$",
    "spin_1x": r"$S_{1x}$",
    "spin_1y": r"$S_{1y}$",
    "spin_1z": r"$S_{1z}$",
    "spin_1z_evolved": r"$S_{1z}^{\mathrm{evol}}$",
    "spin_1z_infinity": r"$S_{1z,\infty}$",
    "spin_1z_infinity_only_prec_avg": (
        r"$S_{1z,\infty}^{\mathrm{only\, prec\, avg}}$"
    ),
    "spin_2x": r"$S_{2x}$",
    "spin_2y": r"$S_{2y}$",
    "spin_2z": r"$S_{2z}$",
    "spin_2z_evolved": r"$S_{2z}^{\mathrm{evol}}$",
    "spin_2z_infinity": r"$S_{2z,\infty}$",
    "spin_2z_infinity_only_prec_avg": (
        r"$S_{2z,\infty}^{\mathrm{only\, prec\, avg}}$"
    ),
    "chi_p": r"$\chi_{\mathrm{p}}$",
    "chi_p_infinity": r"$\chi_{\mathrm{p},\infty}$",
    "chi_p_infinity_only_prec_avg": (
        r"$\chi_{\mathrm{p},\infty}^{\mathrm{only\, prec\, avg}}$"
    ),
    "chi_p_2spin": r"$\chi_{\mathrm{p}}^{\mathrm{2spin}}$",
    "chi_eff": r"$\chi_{\mathrm{eff}}$",
    "chi_eff_infinity": r"$\chi_{\mathrm{eff},\infty}$",
    "chi_eff_infinity_only_prec_avg": (
        r"$\chi_{\mathrm{eff},\infty}^{\mathrm{only\, prec\, avg}}$"
    ),
    "mass_ratio": r"$q$",
    "symmetric_mass_ratio": r"$\eta$",
    "beta": r"$\beta$",
    "inverted_mass_ratio": r"$1/q$",
    "phi_1": r"$\phi_{1} [\mathrm{rad}]$",
    "phi_2": r"$\phi_{2} [\mathrm{rad}]$",
    "cos_tilt_1": r"$\cos{\theta_{1}}$",
    "cos_tilt_2": r"$\cos{\theta_{2}}$",
    "cos_tilt_1_infinity": r"$\cos{\theta_{1,\infty}}$",
    "cos_tilt_2_infinity": r"$\cos{\theta_{2,\infty}}$",
    "cos_tilt_1_infinity_only_prec_avg": (
        r"$\cos{\theta_{1,\infty}^{\mathrm{only\, prec\, avg}}}$"
    ),
    "cos_tilt_2_infinity_only_prec_avg": (
        r"$\cos{\theta_{2,\infty}^{\mathrm{only\, prec\, avg}}}$"
    ),
    "redshift": r"$z$",
    "comoving_distance": r"$d_{com} [\mathrm{Mpc}]$",
    "mass_1_source": r"$m_{1}^{\mathrm{source}} [M_{\odot}]$",
    "mass_2_source": r"$m_{2}^{\mathrm{source}} [M_{\odot}]$",
    "chirp_mass_source": r"$\mathcal{M}^{\mathrm{source}} [M_{\odot}]$",
    "total_mass_source": r"$M^{\mathrm{source}} [M_{\odot}]$",
    "cos_iota": r"$\cos{\iota}$",
    "theta_jn": r"$\theta_{JN} [\mathrm{rad}]$",
    "viewing_angle": r"$\Theta [\mathrm{rad}]$",
    "cos_theta_jn": r"$\cos{\theta_{JN}}$",
    "lambda_1": r"$\lambda_{1}$",
    "lambda_2": r"$\lambda_{2}$",
    "lambda_tilde": r"$\tilde{\lambda}$",
    "delta_lambda": r"$\delta\lambda$",
    "peak_luminosity": (
        r"$L_{\mathrm{peak}} [10^{56} \mathrm{ergs\;s^{-1}}]$"
    ),
    "peak_luminosity_non_evolved": (
        r"$L_{\mathrm{peak}}^{\mathrm{nonevol}} [10^{56} "
        "\mathrm{ergs\;s^{-1}}]$"
    ),
    "final_mass": r"$M_{\mathrm{final}} [M_{\odot}]$",
    "final_mass_non_evolved": (
        r"$M_{\mathrm{final}}^{\mathrm{nonevol}} [M_{\odot}]$"
    ),
    "final_mass_source": r"$M_{\mathrm{final}}^{\mathrm{source}} [M_{\odot}]$",
    "final_mass_source_non_evolved": (
        r"$M_{\mathrm{final}}^{\mathrm{source, nonevol}} [M_{\odot}]$"
    ),
    "final_spin": r"$a_{\mathrm{final}}$",
    "final_spin_non_evolved": r"$a_{\mathrm{final}}^{\mathrm{nonevol}}$",
    "radiated_energy": r"$E_{\mathrm{rad}} [M_{\odot}]$",
    "radiated_energy_non_evolved": (
        r"$E_{\mathrm{rad}}^{\mathrm{nonevol}} [M_{\odot}]$"
    ),
    "final_kick": r"$v_{\mathrm{final}} [\mathrm{km\;s^{-1}}]$",
    "log_pressure": r"$\log{\mathcal{P}}$",
    "gamma_1": r"$\Gamma_{1}$",
    "gamma_2": r"$\Gamma_{2}$",
    "gamma_3": r"$\Gamma_{3}$",
    "spectral_decomposition_gamma_0": r"$\gamma_{0}$",
    "spectral_decomposition_gamma_1": r"$\gamma_{1}$",
    "spectral_decomposition_gamma_2": r"$\gamma_{2}$",
    "spectral_decomposition_gamma_3": r"$\gamma_{3}$",
    "tidal_disruption_frequency": r"$f_{\mathrm{td}} [\mathrm{Hz}]$",
    "tidal_disruption_frequency_ratio": r"$f_{\mathrm{td}} / f_{220}$",
    "220_quasinormal_mode_frequency": r"$f_{220} [\mathrm{Hz}]$",
    "baryonic_torus_mass": r"$M_{\mathrm{torus}} [M_{\odot}]$",
    "baryonic_torus_mass_source": r"$M^{\mathrm{source}}_{\mathrm{torus}} [M_{\odot}]$",
    "compactness_1": r"$C_{1}$",
    "compactness_2": r"$C_{2}$",
    "baryonic_mass_1": r"$m_{1, \mathrm{baryonic}} [M_{\odot}]$",
    "baryonic_mass_1_source": r"$m^{\mathrm{source}}_{1, \mathrm{baryonic}} [M_{\odot}]$",
    "baryonic_mass_2": r"$m_{2, \mathrm{baryonic}} [M_{\odot}]$",
    "baryonic_mass_2_source": r"$m^{\mathrm{source}}_{2, \mathrm{baryonic}} [M_{\odot}]$"
}

public_GWlatex_labels = {"mass_1": r"$m_{1}^{\mathrm{det}} [M_{\odot}]$",
                         "mass_2": r"$m_{2}^{\mathrm{det}} [M_{\odot}]$",
                         "mass_1_source": r"$m_{1} [M_{\odot}]$",
                         "mass_2_source": r"$m_{2} [M_{\odot}]$"
                         }
