snoutStudy2.py

Go to the documentation of this file.

import ROOT,sys
from array import array

ROOT.TGaxis.SetMaxDigits(3)


#Draw
canvas = ROOT.TCanvas("canvas","canvas");

f_1 = ROOT.TFile("/pnfs/dune/persistent/users/ljf26/fluxfiles/g4lbne/v3r5p4/QGSP_BERT/OptimizedEngineeredNov2017/neutrino/flux/dk2nu/g4lbne_v3r5p4_QGSP_BERT_OptimizedEngineeredNov2017_neutrino_00250.dk2nu.root");

tree = f_1.Get("dk2nuTree");
#tree.SetBranchStatus("*",0) 
#tree.SetBranchStatus("ancestor",1)

for event in tree:
    print len(event.ancestor)

"""


tree.Print()

ROOT.gStyle.SetOptStat(0)

myhist = ROOT.TH1D("","myhist",20,0,10);
tree.Draw("360/(2*pi)*acos(ancestor.startpz[1]/sqrt(ancestor.startpx[1]*ancestor.startpx[1]+ancestor.startpy[1]*ancestor[1].startpy[1]+ancestor.startpz[1]*ancestor.startpz[1]))>>myhist","decay.nimpwt*nuray.wgt[1]");
#/(ancestor.startpx*ancestor.startpx+ancestor.startpy*ancestor.startpy+ancestory.startpz*ancestor.startpz))","decay.nimpwt*nuray.wgt[1]");

myhist.GetXaxis().SetTitle("Angle WRT Beam Axis (Degrees)")
myhist.GetYaxis().SetTitle("Number of Particles / 0.5 Degree")

myhist.Draw()

myhist2 = ROOT.TH2D("myhist2","",500,-1000,1000,500,-1000,1000);
#tree.Draw("ancestor.starty:ancestor.startx","decay.nimpwt*nuray.wgt[1]*ancestor.startz<20");
tree.Draw("ancestor.startx+(2100-ancestor.startz)*ancestor.startpx/ancestor.startpz:ancestor.starty+(2100-ancestor.startz)*ancestor.startpy/ancestor.startpz>>myhist2","decay.nimpwt*nuray.wgt[1]*ancestor.startz<2100")

myhist2.GetXaxis().SetTitle("X position (cm) at z = 21 cm")
myhist2.GetYaxis().SetTitle("Y position (cm) at z = 21 cm")

myhist2.Draw("colz")


myhist3 = ROOT.TH1D("myhist3","",50,-100,100);
tree.Draw("ancestor.startx+(2100-ancestor.startz)*ancestor.startpx/ancestor.startpz>>myhist3","decay.nimpwt*nuray.wgt[1]*ancestor.startz<21000")

myhist3.GetXaxis().SetTitle("X position (cm) at z = 21 cm")

canvas.Print("test.eps");
canvas.Print("test.png");


"""