# Path: tests/convenience.py
# def plot_aa_style(cols=1):
#     """Use a plotting style specifically for Astronomy & Astrophyiscs."""
#     # Use a style file
#     dir = os.path.abspath(os.path.dirname(__file__))
#     style_file = os.path.join(dir, './aa.mplstyle')
#     plt.style.use(style_file)
# 
#     # Check whether interactive backend is present
#     if os.name == 'posix' and "DISPLAY" not in os.environ:
#         # On a cluster
#         plt.switch_backend('Agg')
#         # Use just the basic latex package allowing \text to be used
#         plt.rcParams['text.latex.preamble'] = r'\usepackage{amsmath}'
# 
#     if cols == 2:
#         plt.rcParams["figure.figsize"] = (5.75373, 3.556)
# 
# def rel_path(path):
#     """Return the relative path name for this directory."""
#     return os.path.join(os.path.abspath(os.path.dirname(__file__)), path)


from matplotlib.lines import Line2D
from matplotlib.patches import Patch
from tests.convenience import plot_aa_style, rel_path
from alpha_complex import complex_rates
from alpha_real import real_rates
from alpha_simple import simple_rates
from alpha_analytical import analytical_rates
import matplotlib.pyplot as plt
import numpy as np
"""Show frbpoppy matches analytical models and predict the event rates."""


REMAKE = True
SIZE = 1e4
SURVEYS = ('askap-fly', 'fast-crafts', 'parkes-htru', 'wsrt-apertif', 'arecibo-palfa')
ELEMENTS = {'analytical': True, 'real': True, 'simple': False, 'complex': True}
ALPHAS = np.around(np.linspace(-0.5, -2.0, 7), decimals=2)


def plot(analytical=True, simple=False, complex=False, real=True):
    """Plot rates panel."""
    surveys = SURVEYS

    # If needing two panels
    if simple and complex:
        plot_aa_style(cols=2)
        fig, (ax1, ax2) = plt.subplots(1, 2, sharey=True)
    else:

# Path: tests/convenience.py
# def plot_aa_style(cols=1):
#     """Use a plotting style specifically for Astronomy & Astrophyiscs."""
#     # Use a style file
#     dir = os.path.abspath(os.path.dirname(__file__))
#     style_file = os.path.join(dir, './aa.mplstyle')
#     plt.style.use(style_file)
# 
#     # Check whether interactive backend is present
#     if os.name == 'posix' and "DISPLAY" not in os.environ:
#         # On a cluster
#         plt.switch_backend('Agg')
#         # Use just the basic latex package allowing \text to be used
#         plt.rcParams['text.latex.preamble'] = r'\usepackage{amsmath}'
# 
#     if cols == 2:
#         plt.rcParams["figure.figsize"] = (5.75373, 3.556)
# 
# def rel_path(path):
#     """Return the relative path name for this directory."""
#     return os.path.join(os.path.abspath(os.path.dirname(__file__)), path)


# Path: tests/rates/alpha_overview.py
from matplotlib.lines import Line2D
from matplotlib.patches import Patch
from tests.convenience import plot_aa_style, rel_path
from alpha_complex import complex_rates
from alpha_real import real_rates
from alpha_simple import simple_rates
from alpha_analytical import analytical_rates
import matplotlib.pyplot as plt
import numpy as np

"""Show frbpoppy matches analytical models and predict the event rates."""


REMAKE = True
SIZE = 1e4
SURVEYS = ('askap-fly', 'fast-crafts', 'parkes-htru', 'wsrt-apertif', 'arecibo-palfa')
ELEMENTS = {'analytical': True, 'real': True, 'simple': False, 'complex': True}
ALPHAS = np.around(np.linspace(-0.5, -2.0, 7), decimals=2)


def plot(analytical=True, simple=False, complex=False, real=True):
    """Plot rates panel."""
    surveys = SURVEYS

    # If needing two panels
    if simple and complex:
        plot_aa_style(cols=2)
        fig, (ax1, ax2) = plt.subplots(1, 2, sharey=True)
    else:

        plot_aa_style(cols=1)