angular_resolution_vs_energy.py

# Authors: S. Einecke <sabrina.einecke@adelaide.edu.au>
#          K. Brueege <kai.bruegge@tu-dortmund.de>

import click
import numpy as np
import pandas as pd

import matplotlib.pyplot as plt
import astropy.units as u

from utilities import read_data, make_energy_bins, add_theta
from plotting import plot_percentile

@click.command()
@click.argument('input_file', type=click.Path(exists=True))
@click.option('-o', '--output', type=click.Path(exists=False))
@click.option('-t', '--thresholds', multiple=True, default=0.0)

def main(input_file, output, thresholds):
    
    df = read_data(input_file)
    df = add_theta(df)

    if not thresholds:
        thresholds = [0.0]

    bins, bin_centers, bin_widths = make_energy_bins(
        e_min=0.08 * u.TeV, 
        e_max=300 * u.TeV, 
        bins=15,
        centering='log'
        )

    ax = None

    for t in thresholds:
        x = df[df.gamma_score_mean > t].energy_mean.values
        y = df[df.gamma_score_mean > t].theta
        
        ax = plot_percentile(x, y, t, bins, bin_centers, bin_widths, ax=ax)

    ref = np.loadtxt('references/South-SST-AngRes.txt')
    plt.plot(10**ref[:,0], ref[:,1],
        '--', label='SST sub-system', color='silver')

    ax.set_xscale('log')
    ax.set_xlim([0.5,300])
    ax.set_ylim([0,0.5])
    ax.set_ylabel('Angular Resolution / deg')
    ax.set_xlabel(r'$\mathrm{Reconstructed}\ \mathrm{Energy}\ /\  \mathrm{TeV}$')
    ax.legend()
    plt.tight_layout()

    if output:
        plt.savefig(output)
    else:
        plt.show()


if __name__ == "__main__":
    main()