import matplotlib.pyplot as plt

def main():

    grid = open('/user/msullivan/Wh_Upgrade_Studies/Optimisation/mapping.txt')
    # 
    x_low = []
    y_low = []
    x_med = []
    y_med = []
    x_high = []
    y_high = []
    #
    for line in grid:
        line = line.replace(' ', '').split(',')
        x = float(line[1])
        y = float(line[2])
        dm = x-y
        if dm <= 300:
            x_low.append(x)
            y_low.append(y)
        elif dm > 300 and dm <= 600:
            x_med.append(x)
            y_med.append(y)
        elif dm > 600:
            x_high.append(x)
            y_high.append(y)
    #
    plt.scatter(x_low, y_low, label='SR-Low', color='indianred', s=3.5)
    plt.scatter(x_med, y_med, label='SR-Med', color='mediumseagreen', s=3.5)
    plt.scatter(x_high, y_high, label='SR-High', color='slateblue', s=3.5)
    plt.legend(loc='upper left')
    plt.xlabel(r'm$\left(\tilde{\chi}_{1}^{\pm}, \tilde{\chi}_{2}^{0}\right)$ [GeV]')
    plt.ylabel(r'm$\left(\tilde{\chi}_{1}^{0}\right)$ [GeV]')
    plt.xlim(100,1550)
    plt.ylim(-10,1050)
    #
    plt.savefig('hllhc-grid.png', quality=95, dpi=500)        






if __name__=='__main__':
    main()