plot_validation_datamc_2018 Namespace Reference

Classes
class	Prediction
class	Ratio
class	Spectrum

Functions
def	Caption (fName, axisTitle, decomp)
def	New (cons, args)
def	Clone (obj)
def	POTStr (pot)
def	DeriveScalePow (spect, pot)
def	One ()
def	Preliminary ()
def	BeamModeLabel (beamMode='fhc')
def	SplitCanvas (ysplit)
def	LegendSplit (spec, decomp=False, ratio=False)
def	Legend (isDecomp=False, isSplit=False)
def	IsLeftAlign (fname)
def	ColorLegend (isDecomp=False, isSplit=False, leftAlign=True)
def	TopLegend (isDecomp=False, isSplit=False)
def	RatioLegend (isDecomp=False)

Variables
list	gLeaked = []
	parser = argparse.ArgumentParser()
	action
	help
	opts = parser.parse_args()
	beamMode = opts.beamMode
	fdata = TFile(opts.dataFile)
	fmc = TFile(opts.mcFile)
	fdecomp = TFile(opts.decompFile)
	isLeft = IsLeftAlign(k.GetName())
string	decompStr = ''
int	lineStyle = 1
	data = Spectrum(k.ReadObj())
	scalePow = DeriveScalePow(data, data.pot)
	decomp = Prediction(fdecomp,k.GetName())
	pred = Prediction(fmc, k.GetName())
	lStyle
	fname = re.sub(r'\{group=Cut,cat=._(.*)\}', r'_\1', k.GetName())
	leftAlign
	r = Ratio(Clone(data), Clone(pred.tot))
	rdecomp = Ratio(Clone(data), Clone(decomp.tot))
	color
	c = New(TCanvas)
	p1
	p2
	isDecomp
	isSplit
	True
string	splitName = '{}_plots/{}_{}{}_split.pdf'
	local
	banner

Function Documentation

def plot_validation_datamc_2018.BeamModeLabel

(

beamMode = 'fhc'

)

Definition at line 73 of file plot_validation_datamc_2018.py.

References New().

def BeamModeLabel(beamMode = 'fhc'):
    txt = 'Neutrino Mode'
    if 'rhc' in beamMode:
        txt = 'Antineutrino Mode'
    lb = New(TLatex, .1, .95, txt)
    lb.SetNDC()
    lb.SetTextAlign(13)
    lb.SetTextColor(kGray+3)
    lb.SetTextSize(2/45.)
    lb.Draw()

def plot_validation_datamc_2018.Caption	(		fName,
			axisTitle,
			decomp
	)

Definition at line 11 of file plot_validation_datamc_2018.py.

References novadaq::HexUtils.format(), and open().

def Caption(fName, axisTitle, decomp):
    beamMode = 'neutrino'
    if 'rhc' in fName:
        beamMode = 'antineutrino'

    lcvn = 'CVNeLow' in fName
    hcvn = 'CVNeHigh' in fName
    cut = 'preselection cuts'
    if lcvn: 
        cut = 'low CVN cut'
    if hcvn: 
        cut = 'high CVN cut'
        
    decompStr = ''
    if decomp:
        if 'rhc' in fName:
            decompType = 'proportional'
        else:
            decompType = 'combo'
        decompStr = 'Solid line indicates simulation has been corrected by {} decomposition.'.format(decompType)

    caption = '{} in {} beam mode with {}. Top panel shows data (markers) and simulation (colors) spectra. {} Bottom panel shows the ratio of data to simulation.'.format(axisTitle, beamMode, cut, decompStr)
    
    txtName = fName.replace('pdf', 'txt')
    textFile = open(txtName, 'w')
    textFile.write(caption)
    textFile.close()

def plot_validation_datamc_2018.Clone

(

obj

)

Definition at line 44 of file plot_validation_datamc_2018.py.

Referenced by ColorLegend(), Legend(), LegendSplit(), RatioLegend(), and TopLegend().

def Clone(obj):
    ret = obj.Clone()
    gLeaked.append(ret)
    return ret

def plot_validation_datamc_2018.ColorLegend	(		isDecomp = False,
			isSplit = False,
			leftAlign = True
	)

Definition at line 292 of file plot_validation_datamc_2018.py.

References Clone(), New(), and RatioLegend().

def ColorLegend(isDecomp = False, isSplit = False, leftAlign = True):
    # top legend for line/marker descriptors
    if not leftAlign: topleg = New(TLegend, .4, .8, .9, .88)
    else: topleg = New(TLegend, .11, .8, .61, .88)
    topleg.SetNColumns(3)
    topleg.SetFillStyle(0)

    # bottom legend for interaction colors
    if not leftAlign: leg = New(TLegend, .7, .6, .875, .8)
    else: leg = New(TLegend, .1, .6, .285, .8)
    leg.SetNColumns(2)
    leg.SetMargin(.22)
    leg.SetFillStyle(0)
    leg.SetTextAlign(33)
    if leftAlign:
        leg.SetTextAlign(13)
    
    # dummy histogram
    h = New(TH1F, '', '', 1, 0, 1)

    h.SetLineStyle(7)
    # if not decomp, we don't have uncorrected vs corrected
    # spectra to distinguish between. Set text color white 
    # so the other entries stay right justified
    if not leftAlign:
        if isDecomp: 
            topleg.AddEntry(Clone(h), 'Uncorr. MC', 'l')
        else:
            h.SetLineColor(kWhite)
            blank = topleg.AddEntry(Clone(h), '            ', 'l')
            blank.SetTextColor(kWhite)
            
        h.SetLineStyle(1)
        h.SetLineColor(kRed)
        mc = topleg.AddEntry(Clone(h), 'Total MC', 'l')
        
        h.SetMarkerStyle(kFullCircle)
        h.SetLineColor(kBlack)
        d = topleg.AddEntry(Clone(h), 'ND data', 'lep')
            
    else:
        h.SetMarkerStyle(kFullCircle)
        h.SetLineColor(kBlack)
        d = topleg.AddEntry(Clone(h), 'ND data', 'lep')

        h.SetLineStyle(1)
        h.SetLineColor(kRed)
        mc = topleg.AddEntry(Clone(h), 'Total MC', 'l')
        
        if isDecomp: 
            h.SetLineStyle(7)
            h.SetLineColor(kBlack)
            topleg.AddEntry(Clone(h), 'Uncorr. MC', 'l')
        else:
            h.SetLineColor(kWhite)
            blank = topleg.AddEntry(Clone(h), '            ', 'l')
            blank.SetTextColor(kWhite)
        
        
    # build the bottom legend
    h.SetLineColor(kWhite)
    l2 = leg.AddEntry(Clone(h), '#nu_{#mu} CC', 'l')
    l2.SetTextColor(kGreen+2)

    l4 = leg.AddEntry(Clone(h), '#bar{#nu}_{#mu} CC', 'l')
    l4.SetTextColor(kGreen-3)

    l3 = leg.AddEntry(Clone(h), '#nu_{e} CC', 'l')
    l3.SetTextColor(kMagenta-2)
    
    l5 = leg.AddEntry(Clone(h), '#bar{#nu}_{e} CC', 'l')
    l5.SetTextColor(kMagenta)

    # pad to make NC right justified
    if not leftAlign:
        blank = leg.AddEntry(Clone(h), '', 'l')
        blank.SetTextColor(kWhite)
        l1 = leg.AddEntry(Clone(h), 'NC', 'l')
        l1.SetTextColor(kAzure)
    else:
        l1 = leg.AddEntry(Clone(h), 'NC', 'l')
        l1.SetTextColor(kAzure)
        blank = leg.AddEntry(Clone(h), '', 'l')
        blank.SetTextColor(kWhite)


    topleg.SetTextSize(1.3*topleg.GetTextSize())
    leg.Draw()
    topleg.Draw()
    
    if isSplit: 
        RatioLegend(isDecomp)

def plot_validation_datamc_2018.DeriveScalePow	(		spect,
			pot
	)

Next smallest power of 1000 than maximum bin

Definition at line 52 of file plot_validation_datamc_2018.py.

References makeTrainCVSamples.int.

def DeriveScalePow(spect, pot):
    """Next smallest power of 1000 than maximum bin"""
    if spect.hist.GetMaximum() == 0: return 0
    return 3*(int(math.log10(spect.hist.GetMaximum()*pot/spect.pot))/3)

def plot_validation_datamc_2018.IsLeftAlign

(

fname

)

Definition at line 279 of file plot_validation_datamc_2018.py.

def IsLeftAlign(fname):
    needsLeft = ['id_lid', 'shw_cos_z',
                 'shw_cos_numi', 'id_cvne',
                 'cosangle', 'cosdang',
                 'nplanestofront', 'shwhitfrac',
                 'shwMaxy', 'cvn2d'
                 'cvn_mode', 'lid2d']
    for fig in needsLeft:
        if fig.lower() in fname.lower():
            return True

    return False

def plot_validation_datamc_2018.Legend	(		isDecomp = False,
			isSplit = False
	)

Definition at line 244 of file plot_validation_datamc_2018.py.

References Clone(), New(), and RatioLegend().

def Legend(isDecomp = False, isSplit = False):
    leg = New(TLegend, .78, .4, .95, .89)
    h = New(TH1F, '', '', 1, 0, 1)

    leg.SetMargin(.22)
    leg.SetBorderSize(1)

    h.SetMarkerStyle(kFullCircle)
    leg.AddEntry(Clone(h), 'ND data', 'lep')
    h.SetLineColor(kRed)
    leg.AddEntry(Clone(h), 'Total MC', 'l')
    h.SetLineColor(kAzure)
    leg.AddEntry(Clone(h), 'NC', 'l')
    h.SetLineColor(kGreen+2)
    leg.AddEntry(Clone(h), '#nu_{#mu} CC', 'l')
    h.SetLineColor(kMagenta-2)
    leg.AddEntry(Clone(h), '#nu_{e} CC', 'l')
    h.SetLineColor(kGreen-3)
    leg.AddEntry(Clone(h), '#bar{#nu}_{#mu} CC', 'l')
    h.SetLineColor(kMagenta)
    leg.AddEntry(Clone(h), '#bar{#nu}_{e} CC', 'l')
    if decomp: 
        h.SetLineStyle(7)
        h.SetLineColor(kBlack)
        leg.AddEntry(Clone(h), 'Uncorr. MC', 'l')
        leg.SetTextSize(1.2*leg.GetTextSize())
    else:
        leg.SetTextSize(0.8*h.GetXaxis().GetTitleSize())
    leg.Draw()
    
    if isSplit: 
        RatioLegend(isDecomp)

# Add plots where the legend would fit better on the
# left side

def plot_validation_datamc_2018.LegendSplit	(		spec,
			decomp = False,
			ratio = False
	)

Definition at line 185 of file plot_validation_datamc_2018.py.

References Clone(), and New().

def LegendSplit(spec, decomp = False, ratio = False):
    y1 = .56
    y2 = .85
    if ratio: 
        y1 = .46
        
    # Make some room for the legend
    minY = spec.hist.GetMinimum()
    maxY = spec.hist.GetMaximinum()
    spec.hist.GetYaxis().SetRangeUser(minY, 1.3*maxY)

    leg = New(TLegend, .6, y1, .95, y2)
    leg.SetNColumns(2)
    leg.SetBorderSize(1)
    h = New(TH1F, '', '', 1, 0, 1)
    leg.SetTextSize(.7*h.GetXaxis().GetTitleSize())
    h.SetMarkerStyle(kFullCircle)
    leg.AddEntry(Clone(h), 'ND data', 'lep') #1,1

    h.SetLineColor(kAzure)
    leg.AddEntry(Clone(h), 'NC', 'l') #2,1

    h.SetLineColor(kRed)
    leg.AddEntry(Clone(h), 'Total MC', 'l') #1,2

    if decomp: 
        h.SetLineStyle(7)
    h.SetLineColor(kBlack)
    leg.AddEntry(Clone(h), 'Uncorr. MC', 'l') #2,2

    h.SetLineStyle(1)
    h.SetLineColor(kGreen+2)
    leg.AddEntry(Clone(h), '#nu_{#mu} CC', 'l') #1,3

    h.SetLineColor(kGreen-3)
    leg.AddEntry(Clone(h), '#bar{#nu_{#mu}} CC', 'l') #2,3

    h.SetLineColor(kMagenta-2)
    leg.AddEntry(Clone(h), '#nu_{e} CC', 'l') #1,4

    h.SetLineColor(kMagenta)
    leg.AddEntry(Clone(h), '#bar{#nu_{e}} CC', 'l') #2,4

    h.SetMarkerColor(1)
    h.SetLineColor(1)
    if decomp:
        h.SetMarkerColor(17)
        h.SetLineColor(17)
    if ratio:
        leg.AddEntry(Clone(h), '#splitline{Ratio to}{Uncorr. MC}') #1,5

    h.SetMarkerColor(1)
    h.SetLineColor(1)
    if decomp and ratio: 
        leg.AddEntry(Clone(h), '#splitline{Ratio to}{Decomp MC}') #2,5
    leg.Draw()

def plot_validation_datamc_2018.New	(		cons,
			args
	)

Definition at line 39 of file plot_validation_datamc_2018.py.

Referenced by BeamModeLabel(), ColorLegend(), Legend(), LegendSplit(), One(), Preliminary(), RatioLegend(), SplitCanvas(), and TopLegend().

def New(cons, *args):
    ret = cons(*args)
    gLeaked.append(ret)
    return ret

def plot_validation_datamc_2018.One

(

)

Definition at line 57 of file plot_validation_datamc_2018.py.

References New(), and one().

Referenced by datamc().

def One():
    g = New(TGraph)
    g.SetPoint(0, -1e10, 1)
    g.SetPoint(1, 1e10, 1)
    g.SetLineWidth(2)
    g.SetLineStyle(7)
    g.Draw('same')

def plot_validation_datamc_2018.POTStr

(

pot

)

Definition at line 49 of file plot_validation_datamc_2018.py.

Referenced by plot_validation_datamc_2018.Spectrum.Draw().

def POTStr(pot):
    return '%.3g#times10^{20} POT' % (pot/1e20)

def plot_validation_datamc_2018.Preliminary

(

)

Definition at line 65 of file plot_validation_datamc_2018.py.

References New().

def Preliminary():
    tex = New(TLatex, .9, .95, "NOvA Preliminary")
    tex.SetTextColor(kBlue)
    tex.SetNDC()
    tex.SetTextSize(2/30.)
    tex.SetTextAlign(32)
    tex.Draw()

def plot_validation_datamc_2018.RatioLegend

(

isDecomp = False

)

Definition at line 421 of file plot_validation_datamc_2018.py.

References Clone(), and New().

Referenced by ColorLegend(), Legend(), and TopLegend().

def RatioLegend(isDecomp = False):
    x1 = .12
    x2 = .32
    if isDecomp:
        x2 = .47
    l = New(TLegend, x1, .13, x2, .18)
    l.SetBorderSize(1)
    l.SetNColumns(2)
    h = New(TH1F, '', '', 1, 0, 1)
    h.SetMarkerStyle(kFullCircle)
    h.SetMarkerColor(1)
    h.SetLineColor(1)
    if isDecomp:
        h.SetMarkerColor(17)
        h.SetLineColor(17)
    l.AddEntry(Clone(h), 'Uncorr. MC', 'lep')

    if isDecomp: 
        h.SetMarkerColor(1)
        h.SetLineColor(1)
        l.AddEntry(Clone(h), 'Decomp MC', 'lep')
    l.Draw()

def plot_validation_datamc_2018.SplitCanvas

(

ysplit

)

Definition at line 168 of file plot_validation_datamc_2018.py.

References New().

def SplitCanvas(ysplit):
    if gPad is None: New(TCanvas)
    p1 = New(TPad, '', '', 0, 0, 1, 1)
    p2 = New(TPad, '', '', 0, 0, 1, 1)
    
    p1.SetBottomMargin(ysplit)
    p2.SetTopMargin(1-ysplit)
    
    # Draw p1 second since it's often the more important one, that the user    
    # would prefer to be able to interact with.                                
    for p in [p2, p1]:
      p.SetFillStyle(0)
      p.Draw()

    return (p1, p2)

def plot_validation_datamc_2018.TopLegend	(		isDecomp = False,
			isSplit = False
	)

Definition at line 386 of file plot_validation_datamc_2018.py.

References Clone(), New(), and RatioLegend().

def TopLegend(isDecomp = False, isSplit = False):
    leg = New(TLegend, .15, .75, .85, .85)
    h = New(TH1F, '', '', 1, 0, 1)

    leg.SetMargin(.22)
#    leg.SetBorderSize(1)
    leg.SetNColumns(8)

    h.SetMarkerStyle(kFullCircle)
    leg.AddEntry(Clone(h), 'ND data', 'lep')
    h.SetLineColor(kRed)
    leg.AddEntry(Clone(h), 'Total MC', 'l')
    h.SetLineColor(kAzure)
    leg.AddEntry(Clone(h), 'NC', 'l')
    h.SetLineColor(kGreen+2)
    leg.AddEntry(Clone(h), '#nu_{#mu} CC', 'l')
    h.SetLineColor(kMagenta-2)
    leg.AddEntry(Clone(h), '#nu_{e} CC', 'l')
    h.SetLineColor(kGreen-3)
    leg.AddEntry(Clone(h), '#bar{#nu}_{#mu} CC', 'l')
    h.SetLineColor(kMagenta)
    leg.AddEntry(Clone(h), '#bar{#nu}_{e} CC', 'l')
    if decomp: 
        h.SetLineStyle(7)
        h.SetLineColor(kBlack)
        leg.AddEntry(Clone(h), 'Uncorr. MC', 'l')
        leg.SetTextSize(1.2*leg.GetTextSize())
    else:
        leg.SetTextSize(0.8*h.GetXaxis().GetTitleSize())
    leg.Draw()
    
    if isSplit: 
        RatioLegend(isDecomp)

Variable Documentation

plot_validation_datamc_2018.action

Definition at line 446 of file plot_validation_datamc_2018.py.

plot_validation_datamc_2018.banner

Definition at line 564 of file plot_validation_datamc_2018.py.

plot_validation_datamc_2018.beamMode = opts.beamMode

Definition at line 462 of file plot_validation_datamc_2018.py.

Referenced by make_mass_and_oct_fc_input_2019().

plot_validation_datamc_2018.c = New(TCanvas)

Definition at line 520 of file plot_validation_datamc_2018.py.

plot_validation_datamc_2018.color

Definition at line 516 of file plot_validation_datamc_2018.py.

Referenced by Beam(), BestFitMarker(), BicountEllipse_dCP(), BiprobEllipse_dCP(), check_predinterp(), check_predinterp_numu(), ana::DataMCPair.CName(), compare_fits(), datamc(), demo_nueNumuSysts(), draw_vlines(), evd::GeometryDrawer.DrawBadBoxesPretty(), evd::GeometryDrawer.DrawCells2D(), evd::RecoBaseDrawer.DrawCluster2D(), ana::DataMCPair.DrawData(), DataMCPair.DrawData(), drawLongTerm(), evd::RecoBaseDrawer.DrawProng2D(), evd::RecoBaseDrawer.DrawProng3D(), evd::RecoBaseDrawer.DrawTrack3D(), drawVsPOT(), DummyGraph(), ana.FillWithDimColor(), FillWithDimmerColor(), FindSignPoint(), ana::ISurface.GetBestFitY(), ana::DataMCPair.GetBinnings(), HighlightSignPoints(), main(), MakeLegend(), MakePlot(), MakeTemplatePlotsSyst(), mre_blessed(), evd::RecoBaseDrawer.OfflineChans2D(), Plot1DSlices(), plot_contprof(), plot_ehade_quantbound(), plot_oscprob(), plot_quantile_boundaries_2020(), plotContProf(), PlotFrom2D(), preselection_cutflow(), prob_biprob_pedagogical(), ResolutionBars(), ana::SpectrumComponents.SaveTo(), set_bg_color_to_main(), ana::DataMCPair.SetComponentColor(), test_systmultiverse(), TheoryEllipse_dCP(), and TheoryLine_E().

plot_validation_datamc_2018.data = Spectrum(k.ReadObj())

Definition at line 480 of file plot_validation_datamc_2018.py.

plot_validation_datamc_2018.decomp = Prediction(fdecomp,k.GetName())

Definition at line 488 of file plot_validation_datamc_2018.py.

Referenced by AddExptCorrNue(), AddExptCorrNumu(), NDPredictionHandler.AddVar(), CAF_makeCAFSensitivities_for_FNEX(), check_predinterp(), check_predinterp_numu(), ana::NumuExtrapGenerator.Generate(), ana.GetDecomp(), getHists_FNEX(), ana.GetNumuPredictions2020(), joint_fit_2017_contours(), joint_fit_2017_slices(), joint_fit_2018_contours(), joint_fit_2018_slices(), joint_fit_future_bestfit_univ(), joint_fit_future_contour_univ(), ana.LoadPredictions(), make_fc_surfaces_2020(), make_files_for_decomp_plots(), MakeCAFSensitivities_for_FNEX(), MakeLegend(), modularextrap_demo_nue(), modularextrap_demo_numu(), monoprob(), NumuExtrap(), plot_nd_spectra_2018(), plot_nominal_spectra(), PlotMCComponentsErrorBand(), ana.PlotSyst(), run_joint_fit_2020_bestfit(), run_joint_fit_2020_contours(), run_joint_fit_2020_slices(), runCheatDecomp(), ana.SaveMaps(), systematics_table_from_pred_interp(), NDPredictionHandler.~NDPredictionHandler(), and PredictionHandler.~PredictionHandler().

string plot_validation_datamc_2018.decompStr = ''

Definition at line 473 of file plot_validation_datamc_2018.py.

Referenced by bin_composition_pie_chart().

plot_validation_datamc_2018.fdata = TFile(opts.dataFile)

Definition at line 463 of file plot_validation_datamc_2018.py.

plot_validation_datamc_2018.fdecomp = TFile(opts.decompFile)

Definition at line 465 of file plot_validation_datamc_2018.py.

Referenced by plot_michel_final(), and plot_nd_spectra_2018().

plot_validation_datamc_2018.fmc = TFile(opts.mcFile)

Definition at line 464 of file plot_validation_datamc_2018.py.

plot_validation_datamc_2018.fname = re.sub(r'\{group=Cut,cat=._(.*)\}', r'_\1', k.GetName())

Definition at line 495 of file plot_validation_datamc_2018.py.

list plot_validation_datamc_2018.gLeaked = []

Definition at line 10 of file plot_validation_datamc_2018.py.

plot_validation_datamc_2018.help

Definition at line 447 of file plot_validation_datamc_2018.py.

plot_validation_datamc_2018.isDecomp

Definition at line 533 of file plot_validation_datamc_2018.py.

plot_validation_datamc_2018.isLeft = IsLeftAlign(k.GetName())

Definition at line 469 of file plot_validation_datamc_2018.py.

plot_validation_datamc_2018.isSplit

Definition at line 533 of file plot_validation_datamc_2018.py.

plot_validation_datamc_2018.leftAlign

Definition at line 498 of file plot_validation_datamc_2018.py.

int plot_validation_datamc_2018.lineStyle = 1

Definition at line 474 of file plot_validation_datamc_2018.py.

Referenced by futureSig_reach_singlePOTcombo_syst(), and NuePID3bin().

plot_validation_datamc_2018.local

Definition at line 564 of file plot_validation_datamc_2018.py.

plot_validation_datamc_2018.lStyle

Definition at line 492 of file plot_validation_datamc_2018.py.

plot_validation_datamc_2018.opts = parser.parse_args()

Definition at line 452 of file plot_validation_datamc_2018.py.

plot_validation_datamc_2018.p1

Definition at line 526 of file plot_validation_datamc_2018.py.

plot_validation_datamc_2018.p2

Definition at line 526 of file plot_validation_datamc_2018.py.

plot_validation_datamc_2018.parser = argparse.ArgumentParser()

Definition at line 445 of file plot_validation_datamc_2018.py.

plot_validation_datamc_2018.pred = Prediction(fmc, k.GetName())

Definition at line 491 of file plot_validation_datamc_2018.py.

plot_validation_datamc_2018.r = Ratio(Clone(data), Clone(pred.tot))

Definition at line 507 of file plot_validation_datamc_2018.py.

plot_validation_datamc_2018.rdecomp = Ratio(Clone(data), Clone(decomp.tot))

Definition at line 514 of file plot_validation_datamc_2018.py.

Referenced by plot_nd_spectra_2018().

plot_validation_datamc_2018.scalePow = DeriveScalePow(data, data.pot)

Definition at line 484 of file plot_validation_datamc_2018.py.

string plot_validation_datamc_2018.splitName = '{}_plots/{}_{}{}_split.pdf'

Definition at line 550 of file plot_validation_datamc_2018.py.

plot_validation_datamc_2018.True

Definition at line 533 of file plot_validation_datamc_2018.py.