MakePlots Namespace Reference

Functions
def	make_count_plot
	Main function ~~~~~~~~~~~~~. More...
def	plot_cov_output
def	plot_chi2_output
def	compare_w_proposal
def	dm2_chi2_slice
def	sin_chi2_slice

Variables
list	buildpath = os.environ['SBN_LIB_DIR']
tuple	parser = argparse.ArgumentParser()
tuple	args = parser.parse_args()

Function Documentation

def MakePlots.compare_w_proposal

(

args

)

Definition at line 189 of file MakePlots.py.

def compare_w_proposal(args):
    
    chi2file = TFile(args.chifile)
    
    gStyle = TStyle()
    gStyle.SetPadLeftMargin(0.15); gStyle.SetPadRightMargin(0.15)
    
    colours = [30, 38, 46]
    contours = [chi2file.Get('90pct'), 
                chi2file.Get('3sigma'), 
                chi2file.Get('5sigma')]
    
    propcontours = []
    contournames = ['90pct', '3s', '5s']
    contourtitles = ['90% Confidence Level', '3#sigma Confidence Level', '5#sigma Confidence Level']
    
    gr_range = TGraph()
    gr_range.SetPoint(0, 0.001, 0.01)
    gr_range.SetPoint(1, 1, 100)
    gr_range.SetMarkerColor(0)
    
    bestfit = TGraph()
    bestfit.SetPoint(0, 0.062, 1.7)
    bestfit.SetMarkerStyle(29)
    bestfit.SetMarkerSize(1.6)
    bestfit.SetMarkerColor(40)
    
    print("contours has length " + str(len(contours)))
    
    for i in range(len(contours)):
        
        x = []
        y = []
        
        with open(args.compdir+'numu'+contournames[i]+'.txt') as f:
            for l, line in enumerate(f):
                if l == 0: continue
                x.append(float(line.split(', ')[0]))
                y.append(float(line.split(', ')[1].replace("\n", "")))
        
        propcontours.append(TGraph())
        for j in range(len(x)):
            propcontours[i].SetPoint(j, x[j], y[j])

        tempcanvas = TCanvas('temp_canvas', '', 1020, 990)

        templegend = TLegend()
        templegend.AddEntry(contours[i], 'Our contour', 'l')
        templegend.AddEntry(propcontours[i], 'From proposal', 'l')
        templegend.AddEntry(bestfit, 'Best Fit Point', 'p')

        tempcanvas.SetLogy()
        tempcanvas.SetLogx()

        gr_range.SetTitle(contourtitles[i]+' Comparison; sin^{2}(2#theta); #Delta m^{2} (eV^{2})')

        gr_range.Draw('AP')
        gr_range.GetXaxis().SetRangeUser(0.001, 1)
        gr_range.GetYaxis().SetRangeUser(0.01, 100)
        
        for lst in (contours, propcontours):
            lst[i].SetMarkerStyle(20)
            lst[i].SetMarkerSize(0.25)
            lst[i].SetMarkerColor(colours[i] if lst == contours else 1)
            lst[i].SetLineColor(colours[i] if lst == contours else 1)
        
        contours[i].Draw('P same')
        propcontours[i].Draw('P same')

        templegend.Draw()
        bestfit.Draw('P same')

        tempcanvas.SaveAs(args.outdir+contournames[i]+'_comparison.pdf')