import fitsio
import numpy as np
import matplotlib.pyplot as plt

#match_file = '/data/des41.b/data/severett/Balrog/prod020419/stacked_catalogs/balrog_matched_catalog.fits'
input_file = '/data/des71.a/data/kuropat/balrog-base/inputs/BALROG_RUN1_DEEP_CAT_FLAG0INVHS1SNGTM3DET254.fits'

#cols = ['true_cm_g']
cols = ['cm_g']

print('Reading file...')
#match = fitsio.read(match_file, columns=cols)
inpt =  fitsio.read(input_file)
print('File read')

print inpt.dtype

#g1 = match['true_cm_g'][0]
#g2 = match['true_cm_g'][1]
g1 = inpt['cm_g'][:,0]
g2 = inpt['cm_g'][:,1]

med1 = np.median(g1)
med2 = np.median(g2)
mean1 = np.mean(g1)
mean2 = np.mean(g2)

print('med(g1) = {}'.format(med1))
print('med(g2) = {}'.format(med2))

print('Starting plots...')
db = 0.05
bins = np.arange(-1, 1.+db, db)
plt.hist(g1, ec='k', label='true_cm_g1', alpha=0.5, color='r')
plt.hist(g2, ec='k', label='true_cm_g2', alpha=0.5, color='b')
plt.axvline(x=mean1, label='<g1>={}'.format(mean1), color='r', ls='--')
plt.axvline(x=mean2, label='<g2>={}'.format(mean2), color='b', ls='--')
plt.legend()
plt.xlabel('True g')
plt.ylabel('Counts')
plt.title('True ellipticities for DF injection catalog')

plt.gcf().set_size_inches(8,6)

plt.rcParams.update({'font.size':14})

plt.show()