def main() :
parser = optparse.OptionParser(description="Switch for data/MC running")
parser.add_option('-m', action="store_true", default=False, dest="isMC")
parser.add_option('-d', action="store_true", default=False, dest="debug")
parser.add_option('-s', action="store", default="gjets",dest="sample", type="string")
parser.add_option('-f', action="store", default="0", dest="subsec", type="int")
parser.add_option('-t', action="store", default="directPhotonsID",dest="treeName", type="string")
parser.add_option('-p', action="store", default=100.0,dest="minpt", type="float")
parser.add_option('-r', action="store", default=0.5,dest="cutDR", type="float")
options, args = parser.parse_args()
r.gROOT.SetBatch(True)
subfiles = [
"*",#0
"*_?_?_???",#1
"*_1?_?_???","*_2?_?_???","*_3?_?_???","*_4?_?_???","*_5?_?_???","*_6?_?_???","*_7?_?_???","*_8?_?_???","*_9?_?_???",#10
"*_11?_?_???","*_12?_?_???","*_13?_?_???","*_14?_?_???","*_15?_?_???","*_16?_?_???","*_17?_?_???","*_18?_?_???","*_19?_?_???",#19
"*_21?_?_???","*_22?_?_???","*_23?_?_???","*_24?_?_???","*_25?_?_???","*_26?_?_???","*_27?_?_???","*_28?_?_???","*_29?_?_???",#28
"*_31?_?_???","*_32?_?_???","*_33?_?_???","*_34?_?_???","*_35?_?_???","*_36?_?_???","*_37?_?_???","*_38?_?_???","*_39?_?_???",#37
"*_41?_?_???","*_42?_?_???","*_43?_?_???","*_44?_?_???","*_45?_?_???","*_46?_?_???","*_47?_?_???","*_48?_?_???","*_49?_?_???",#46
"*_51?_?_???","*_52?_?_???","*_53?_?_???","*_54?_?_???","*_55?_?_???","*_56?_?_???","*_57?_?_???","*_58?_?_???","*_59?_?_???",#55
"*_61?_?_???","*_62?_?_???","*_63?_?_???","*_64?_?_???","*_65?_?_???","*_66?_?_???","*_67?_?_???","*_68?_?_???","*_69?_?_???",#64
"*_71?_?_???","*_72?_?_???","*_73?_?_???","*_74?_?_???","*_75?_?_???","*_76?_?_???","*_77?_?_???","*_78?_?_???","*_79?_?_???",#73
"*_81?_?_???","*_82?_?_???","*_83?_?_???","*_84?_?_???","*_85?_?_???","*_86?_?_???","*_87?_?_???","*_88?_?_???","*_89?_?_???",#82
"*_91?_?_???","*_92?_?_???","*_93?_?_???","*_94?_?_???","*_95?_?_???","*_96?_?_???","*_97?_?_???","*_98?_?_???","*_99?_?_???",#91
"*_101?_?_???","*_102?_?_???","*_103?_?_???","*_104?_?_???","*_105?_?_???","*_106?_?_???","*_107?_?_???","*_108?_?_???","*_109?_?_???",#100
]
outFileName = "estimateTreeDR%2.1f_%s_%d.root"%(options.cutDR,options.sample,options.subsec)
if options.debug:
outFileName = "estimateTreeDR%2.1f_%s_test.root"%(options.cutDR,options.sample)
print outFileName
sys.stdout.flush()
outFile = r.TFile(outFileName,"RECREATE")
tree = r.TTree( 'tree', 'tree for estimate ' )
nJetsHT = array( 'i', [ 0 ] )
nJetsMHT = array( 'i', [ 0 ] )
htVal = array( 'd', [ 0. ] )
mhtVal = array( 'd', [ 0. ] )
dphi1 = array( 'd', [ 0. ] )
dphi2 = array( 'd', [ 0. ] )
dphi3 = array( 'd', [ 0. ] )
photonPt = array( 'd', [ 0. ] )
photonEta = array( 'd', [ 0. ] )
jet1Pt = array( 'd', [ 0. ] )
jet1Eta = array( 'd', [ 0. ] )
jet2Pt = array( 'd', [ 0. ] )
jet2Eta = array( 'd', [ 0. ] )
jet3Pt = array( 'd', [ 0. ] )
jet3Eta = array( 'd', [ 0. ] )
eventWt = array( 'd', [ 0. ] )
puWt = array( 'd', [ 0. ] )
phen = array( 'd', [ 0. ] )
phenErrStat = array( 'd', [ 0. ] )
acc = array( 'd', [ 0. ] )
accErrStat = array( 'd', [ 0. ] )
recoID = array( 'd', [ 0. ] )
recoIDErrStat = array( 'd', [ 0. ] )
recoPFIso = array( 'd', [ 0. ] )
recoPFIsoErrStat = array( 'd', [ 0. ] )
purity = array( 'd', [ 0. ] )
purityErr = array( 'd', [ 0. ] )
datamcScaleF = array( 'd', [ 0. ] )
datamcScaleFErrStat = array( 'd', [ 0. ] )
datamcScaleFErrSyst = array( 'd', [ 0. ] )
datamcVetoF = array( 'd', [ 0. ] )
datamcVetoFErrStat = array( 'd', [ 0. ] )
corrFactor = array( 'd', [ 0. ] )
tree.Branch( 'nJetsHT', nJetsHT, 'nJetsHT/I' )
tree.Branch( 'nJetsMHT', nJetsMHT, 'nJetsMHT/I' )
tree.Branch( 'htVal', htVal, 'htVal/D' )
tree.Branch( 'mhtVal', mhtVal, 'mhtVal/D' )
tree.Branch( 'dphi1', dphi1, 'dphi1/D' )
tree.Branch( 'dphi2', dphi2, 'dphi2/D' )
tree.Branch( 'dphi3', dphi3, 'dphi3/D' )
tree.Branch( 'photonPt', photonPt, 'photonPt/D' )
tree.Branch( 'photonEta',photonEta, 'photonEta/D' )
tree.Branch( 'jet1Pt', jet1Pt, 'jet1Pt/D' )
tree.Branch( 'jet1Eta', jet1Eta, 'jet1Eta/D' )
tree.Branch( 'jet2Pt', jet2Pt, 'jet2Pt/D' )
tree.Branch( 'jet2Eta', jet2Eta, 'jet2Eta/D' )
tree.Branch( 'jet3Pt', jet3Pt, 'jet3Pt/D' )
tree.Branch( 'jet3Eta', jet3Eta, 'jet3Eta/D' )
tree.Branch( 'eventWt', eventWt, 'eventWt/D' )
tree.Branch( 'puWt', puWt, 'puWt/D' )
tree.Branch( 'phen', phen, 'phen/D' )
tree.Branch( 'phenErrStat', phenErrStat, 'phenErrStat/D' )
tree.Branch( 'acc', acc, 'acc/D' )
tree.Branch( 'accErrStat', accErrStat, 'accErrStat/D' )
tree.Branch( 'recoID', recoID, 'recoID/D' )
tree.Branch( 'recoIDErrStat', recoIDErrStat, 'recoIDErrStat/D' )
tree.Branch( 'recoPFIso', recoPFIso, 'recoPFIso/D' )
tree.Branch( 'recoPFIsoErrStat', recoPFIsoErrStat, 'recoPFIsoErrStat/D' )
tree.Branch( 'purity', purity, 'purity/D' )
tree.Branch( 'purityErr', purityErr, 'purityErr/D' )
tree.Branch( 'datamcScaleF', datamcScaleF, 'datamcScaleF/D' )
tree.Branch( 'datamcScaleFErrStat', datamcScaleFErrStat, 'datamcScaleFErrStat/D' )
tree.Branch( 'datamcScaleFErrSyst', datamcScaleFErrSyst, 'datamcScaleFErrSyst/D' )
tree.Branch( 'datamcVetoF', datamcVetoF, 'datamcVetoF/D' )
tree.Branch( 'datamcVetoFErrStat', datamcVetoFErrStat, 'datamcVetoFErrStat/D' )
tree.Branch( 'corrFactor', corrFactor, 'corrFactor/D' )
myChain = r.TChain('%s/RA2Values'%(options.treeName))
##################
if options.debug:
if options.sample=="gjets":
##noht200## myChain.Add("/uscmst1b_scratch/lpc1/3DayLifetime/sturdy07/tmp/cmssw525p1/treeMaker/gjetsht200_reco_tree_ra2/res/*_?_?_???.root")
myChain.Add("/uscmst1b_scratch/lpc1/3DayLifetime/sturdy07/tmp/cmssw525p1/treeMaker/gjetsht400_reco_tree_ra2/res/*_?_?_???.root")
elif options.sample=="data":
myChain.Add("/uscmst1b_scratch/lpc1/3DayLifetime/sturdy07/tmp/cmssw525p1/treeMaker/run2012a_tree_ra2/res/*_?_?_???.root")
myChain.Add("/uscmst1b_scratch/lpc1/3DayLifetime/sturdy07/tmp/cmssw525p1/treeMaker/run2012b_tree_ra2/res/*_?_?_???.root")
else:
if options.sample=="gjets":
##noht200## myChain.Add("/uscmst1b_scratch/lpc1/3DayLifetime/sturdy07/tmp/cmssw525p1/treeMaker/gjetsht200_reco_tree_ra2/res/%s.root"%(subfiles[options.subsec]))
myChain.Add("/uscmst1b_scratch/lpc1/3DayLifetime/sturdy07/tmp/cmssw525p1/treeMaker/gjetsht400_reco_tree_ra2/res/%s.root"%(subfiles[options.subsec]))
elif options.sample=="data":
myChain.Add("/uscmst1b_scratch/lpc1/3DayLifetime/sturdy07/tmp/cmssw525p1/treeMaker/run2012a_tree_ra2/res/%s.root"%(subfiles[options.subsec]))
myChain.Add("/uscmst1b_scratch/lpc1/3DayLifetime/sturdy07/tmp/cmssw525p1/treeMaker/run2012b_tree_ra2/res/%s.root"%(subfiles[options.subsec]))
fChain = myChain
###Timing information
decade = 0
century = 0
tsw = r.TStopwatch()
tenpcount = 1
onepcount = 1
nentries = fChain.GetEntries()
print "nentries %d"%(nentries)
sys.stdout.flush()
i = 0
for event in fChain:
# ==============print number of events done == == == == == == == =
if ( i==0):
tsw.Start()
#print('.', end='')
sys.stdout.write('.')
sys.stdout.flush()
if ((i*10)/nentries == tenpcount ) :
tsw.Stop()
time = tsw.RealTime()
tsw.Start(r.kFALSE)
finTime = 0.
frac = (i*1.0)/(nentries*1.0)
if (frac>0):
finTime = time / frac - time
finMin = finTime / 60.
sys.stdout.write("%d%% done. "%(tenpcount*10))
# sys.stdout.write("t=7.2f"%(time))
sys.stdout.write("t="+str(time))
sys.stdout.write(" projected finish=%7d s("%(finTime))
sys.stdout.write("%2.2f min). "%(finMin))
sys.stdout.write("\n")
sys.stdout.flush()
tenpcount = tenpcount + 1
elif ( (i*100)/nentries == onepcount ) :
#print('.', end='')
sys.stdout.write('.')
sys.stdout.flush()
onepcount = onepcount + 1
nJetsHT[0] = event.ra2_nJetsPt50Eta25
nJetsMHT[0] = event.ra2_nJetsPt30Eta50
htVal[0] = event.ra2_HT
mhtVal[0] = event.ra2_MHT
dphi1[0] = event.ra2_dPhiMHT1
dphi2[0] = event.ra2_dPhiMHT2
dphi3[0] = event.ra2_dPhiMHT3
photonPt[0] = event.ra2_Photon1Pt
photonEta[0] = event.ra2_Photon1Eta
jet1Pt[0] = event.ra2_Jet1Pt
jet1Eta[0] = event.ra2_Jet1Eta
jet2Pt[0] = event.ra2_Jet2Pt
jet2Eta[0] = event.ra2_Jet2Eta
jet3Pt[0] = event.ra2_Jet3Pt
jet3Eta[0] = event.ra2_Jet3Eta
puWt[0] = event.ra2_PUWt
eventWt[0] = event.ra2_EventWt
ra2_Photon70PFHT400 = event.ra2_Photon70PFHT400
ra2_Photon70PFNoPUHT400 = event.ra2_Photon70PFNoPUHT400
triggers = (ra2_Photon70PFHT400 or ra2_Photon70PFNoPUHT400)
if options.isMC:
triggers = True
if (triggers) and (cutF.ra2PhotonSelection(event,options.minpt, options.cutDR)) and cutF.ra2Baseline(event):
pheno = corrs.pheno(nJetsHT[0],mhtVal[0])
phen[0] = pheno[0]
phenErrStat[0] = pheno[1]
result = corrs.acc(nJetsHT[0])
acc[0] = result[0]
accErrStat[0] = result[1]
result = corrs.acc(nJetsHT[0])
recoID[0] = result[0]
recoIDErrStat[0] = result[1]
result = corrs.acc(nJetsHT[0])
recoPFIso[0] = result[0]
recoPFIsoErrStat[0] = result[1]
purity[0] = 0.97
purityErr[0] = 0.01
result = corrs.datamcPhotonIDPFIsoScaleFactor(photonEta[0])
datamcScaleF[0] = result[0]
datamcScaleFErrStat[0] = result[1]
datamcScaleFErrSyst[0] = result[2]
result = corrs.datamcElectronVetoScaleFactor(photonEta[0])
# print result
datamcVetoF[0] = result[0]
datamcVetoFErrStat[0] = result[1]
corrFactor[0] = purity[0]*phen[0]/(acc[0]*recoID[0]*recoPFIso[0]*datamcScaleF[0]*datamcVetoF[0])
tree.Fill()
#########
i = i + 1
#tree.Write()
outFile.Write()
outFile.Close()
def main() :
parser = optparse.OptionParser(description="Switch for data/MC running")
parser.add_option('-m', action="store_true", default=False, dest="isMC")
parser.add_option('-d', action="store_true", default=False, dest="debug")
parser.add_option('-s', action="store", default="gjets",dest="sample", type="string")
parser.add_option('-f', action="store", default="0", dest="subsec", type="int")
parser.add_option('-t', action="store", default="directPhotonsID",dest="treeName", type="string")
parser.add_option('-p', action="store", default=100.0,dest="minpt", type="float")
parser.add_option('-r', action="store", default=0.5,dest="cutDR", type="float")
options, args = parser.parse_args()
r.gROOT.SetBatch(True)
subfiles = [
"*",#0
"*_?_?_???",#1
"*_1?_?_???","*_2?_?_???","*_3?_?_???","*_4?_?_???","*_5?_?_???","*_6?_?_???","*_7?_?_???","*_8?_?_???","*_9?_?_???",#10
"*_11?_?_???","*_12?_?_???","*_13?_?_???","*_14?_?_???","*_15?_?_???","*_16?_?_???","*_17?_?_???","*_18?_?_???","*_19?_?_???",#19
"*_21?_?_???","*_22?_?_???","*_23?_?_???","*_24?_?_???","*_25?_?_???","*_26?_?_???","*_27?_?_???","*_28?_?_???","*_29?_?_???",#28
"*_31?_?_???","*_32?_?_???","*_33?_?_???","*_34?_?_???","*_35?_?_???","*_36?_?_???","*_37?_?_???","*_38?_?_???","*_39?_?_???",#37
"*_41?_?_???","*_42?_?_???","*_43?_?_???","*_44?_?_???","*_45?_?_???","*_46?_?_???","*_47?_?_???","*_48?_?_???","*_49?_?_???",#46
"*_51?_?_???","*_52?_?_???","*_53?_?_???","*_54?_?_???","*_55?_?_???","*_56?_?_???","*_57?_?_???","*_58?_?_???","*_59?_?_???",#55
"*_61?_?_???","*_62?_?_???","*_63?_?_???","*_64?_?_???","*_65?_?_???","*_66?_?_???","*_67?_?_???","*_68?_?_???","*_69?_?_???",#64
"*_71?_?_???","*_72?_?_???","*_73?_?_???","*_74?_?_???","*_75?_?_???","*_76?_?_???","*_77?_?_???","*_78?_?_???","*_79?_?_???",#73
"*_81?_?_???","*_82?_?_???","*_83?_?_???","*_84?_?_???","*_85?_?_???","*_86?_?_???","*_87?_?_???","*_88?_?_???","*_89?_?_???",#82
"*_91?_?_???","*_92?_?_???","*_93?_?_???","*_94?_?_???","*_95?_?_???","*_96?_?_???","*_97?_?_???","*_98?_?_???","*_99?_?_???",#91
"*_101?_?_???","*_102?_?_???","*_103?_?_???","*_104?_?_???","*_105?_?_???","*_106?_?_???","*_107?_?_???","*_108?_?_???","*_109?_?_???",#100
]
outFileName = "sampleTreeDR%2.1f_%s_%d.root"%(options.cutDR,options.sample,options.subsec)
if options.debug:
outFileName = "sampleTreeDR%2.1f_%s_test.root"%(options.cutDR,options.sample)
print outFileName
sys.stdout.flush()
outFile = r.TFile(outFileName,"RECREATE")
tree = r.TTree( 'tree', 'tree for sample ' )
nJetsHT = array( 'i', [ 0 ] )
nJetsMHT = array( 'i', [ 0 ] )
htVal = array( 'd', [ 0. ] )
mhtVal = array( 'd', [ 0. ] )
dphi1 = array( 'd', [ 0. ] )
dphi2 = array( 'd', [ 0. ] )
dphi3 = array( 'd', [ 0. ] )
photonPt = array( 'd', [ 0. ] )
photonEta = array( 'd', [ 0. ] )
photonMinDR = array( 'd', [ 0. ] )
photonpfCH = array( 'd', [ 0. ] )
photonpfNU = array( 'd', [ 0. ] )
photonpfGA = array( 'd', [ 0. ] )
muon1Pt = array( 'd', [ 0. ] )
muon1Eta = array( 'd', [ 0. ] )
muon1MinDR = array( 'd', [ 0. ] )
muon2Pt = array( 'd', [ 0. ] )
muon2Eta = array( 'd', [ 0. ] )
muon2MinDR = array( 'd', [ 0. ] )
dimuonPt = array( 'd', [ 0. ] )
dimuonEta = array( 'd', [ 0. ] )
dimuonMinDR = array( 'd', [ 0. ] )
dimuonM = array( 'd', [ 0. ] )
jet1Pt = array( 'd', [ 0. ] )
jet1Eta = array( 'd', [ 0. ] )
jet2Pt = array( 'd', [ 0. ] )
jet2Eta = array( 'd', [ 0. ] )
jet3Pt = array( 'd', [ 0. ] )
jet3Eta = array( 'd', [ 0. ] )
eventWt = array( 'd', [ 0. ] )
puWt = array( 'd', [ 0. ] )
tree.Branch( 'nJetsHT', nJetsHT, 'nJetsHT/I' )
tree.Branch( 'nJetsMHT', nJetsMHT, 'nJetsMHT/I' )
tree.Branch( 'htVal', htVal, 'htVal/D' )
tree.Branch( 'mhtVal', mhtVal, 'mhtVal/D' )
tree.Branch( 'dphi1', dphi1, 'dphi1/D' )
tree.Branch( 'dphi2', dphi2, 'dphi2/D' )
tree.Branch( 'dphi3', dphi3, 'dphi3/D' )
tree.Branch( 'photonPt', photonPt, 'photonPt/D' )
tree.Branch( 'photonEta', photonEta, 'photonEta/D' )
tree.Branch( 'photonMinDR', photonMinDR, 'photonMinDR/D' )
tree.Branch( 'photonpfCH',photonpfCH, 'photonpfCH/D' )
tree.Branch( 'photonpfNU',photonpfNU, 'photonpfNU/D' )
tree.Branch( 'photonpfGA',photonpfGA, 'photonpfGA/D' )
tree.Branch( 'muon1Pt', muon1Pt, 'muon1Pt/D' )
tree.Branch( 'muon1Eta', muon1Eta, 'muon1Eta/D' )
tree.Branch( 'muon1MinDR', muon1MinDR, 'muon1MinDR/D' )
tree.Branch( 'muon2Pt', muon2Pt, 'muon2Pt/D' )
tree.Branch( 'muon2Eta', muon2Eta, 'muon2Eta/D' )
tree.Branch( 'muon2MinDR', muon2MinDR, 'muon2MinDR/D' )
tree.Branch( 'dimuonPt', dimuonPt, 'dimuonPt/D' )
tree.Branch( 'dimuonEta',dimuonEta, 'dimuonEta/D' )
tree.Branch( 'dimuonMinDR',dimuonMinDR, 'dimuonMinDR/D' )
tree.Branch( 'dimuonM', dimuonM, 'dimuonM/D' )
tree.Branch( 'jet1Pt', jet1Pt, 'jet1Pt/D' )
tree.Branch( 'jet1Eta', jet1Eta, 'jet1Eta/D' )
tree.Branch( 'jet2Pt', jet2Pt, 'jet2Pt/D' )
tree.Branch( 'jet2Eta', jet2Eta, 'jet2Eta/D' )
tree.Branch( 'jet3Pt', jet3Pt, 'jet3Pt/D' )
tree.Branch( 'jet3Eta', jet3Eta, 'jet3Eta/D' )
tree.Branch( 'eventWt', eventWt, 'eventWt/D' )
tree.Branch( 'puWt', puWt, 'puWt/D' )
myChain = r.TChain('%s/RA2Values'%(options.treeName))
##################
if options.debug:
if options.sample=="gjets":
##noht200## myChain.Add("/uscmst1b_scratch/lpc1/3DayLifetime/sturdy07/tmp/cmssw525p1/treeMaker/gjetsht200_reco_tree_ra2/res/*_?_?_???.root")
myChain.Add("/uscmst1b_scratch/lpc1/3DayLifetime/sturdy07/tmp/cmssw525p1/treeMaker/gjetsht400_reco_tree_ra2/res/*_?_?_???.root")
elif options.sample=="data":
myChain.Add("/uscmst1b_scratch/lpc1/3DayLifetime/sturdy07/tmp/cmssw525p1/treeMaker/run2012a_tree_ra2/res/*_?_?_???.root")
myChain.Add("/uscmst1b_scratch/lpc1/3DayLifetime/sturdy07/tmp/cmssw525p1/treeMaker/run2012b_tree_ra2/res/*_?_?_???.root")
elif options.sample=="zinv":
##noht200## myChain.Add("/uscmst1b_scratch/lpc1/3DayLifetime/sturdy07/tmp/cmssw525p1/treeMaker/zinvht200_reco_tree_ra2/res/*_?_?_???.root")
myChain.Add("/uscmst1b_scratch/lpc1/3DayLifetime/sturdy07/tmp/cmssw525p1/treeMaker/zinvht400_reco_tree_ra2/res/*_?_?_???.root")
elif options.sample=="zmumu":
##noht200## myChain.Add("/uscmst1b_scratch/lpc1/3DayLifetime/sturdy07/tmp/cmssw525p1/treeMaker/dyjetstoll_ht200_reco_tree_ra2/res/*_?_?_???.root")
myChain.Add("/uscmst1b_scratch/lpc1/3DayLifetime/sturdy07/tmp/cmssw525p1/treeMaker/dyjetstoll_ht400_reco_tree_ra2/res/*_?_?_???.root")
else:
if options.sample=="gjets":
##noht200## myChain.Add("/uscmst1b_scratch/lpc1/3DayLifetime/sturdy07/tmp/cmssw525p1/treeMaker/gjetsht200_reco_tree_ra2/res/%s.root"%(subfiles[options.subsec]))
myChain.Add("/uscmst1b_scratch/lpc1/3DayLifetime/sturdy07/tmp/cmssw525p1/treeMaker/gjetsht400_reco_tree_ra2/res/%s.root"%(subfiles[options.subsec]))
elif options.sample=="data":
myChain.Add("/uscmst1b_scratch/lpc1/3DayLifetime/sturdy07/tmp/cmssw525p1/treeMaker/run2012a_tree_ra2/res/%s.root"%(subfiles[options.subsec]))
myChain.Add("/uscmst1b_scratch/lpc1/3DayLifetime/sturdy07/tmp/cmssw525p1/treeMaker/run2012b_tree_ra2/res/%s.root"%(subfiles[options.subsec]))
elif options.sample=="zinv":
##noht200## myChain.Add("/uscmst1b_scratch/lpc1/3DayLifetime/sturdy07/tmp/cmssw525p1/treeMaker/zinvht200_reco_tree_ra2/res/%s.root"%(subfiles[options.subsec]))
myChain.Add("/uscmst1b_scratch/lpc1/3DayLifetime/sturdy07/tmp/cmssw525p1/treeMaker/zinvht400_reco_tree_ra2/res/%s.root"%(subfiles[options.subsec]))
elif options.sample=="zmumu":
##noht200## myChain.Add("/uscmst1b_scratch/lpc1/3DayLifetime/sturdy07/tmp/cmssw525p1/treeMaker/dyjetstoll_ht200_reco_tree_ra2/res/%s.root"%(subfiles[options.subsec]))
myChain.Add("/uscmst1b_scratch/lpc1/3DayLifetime/sturdy07/tmp/cmssw525p1/treeMaker/dyjetstoll_ht400_reco_tree_ra2/res/%s.root"%(subfiles[options.subsec]))
fChain = myChain
###Timing information
decade = 0
century = 0
tsw = r.TStopwatch()
tenpcount = 1
onepcount = 1
nentries = fChain.GetEntries()
print "nentries %d"%(nentries)
sys.stdout.flush()
i = 0
for event in fChain:
# ==============print number of events done == == == == == == == =
if ( i==0):
tsw.Start()
#print('.', end='')
sys.stdout.write('.')
sys.stdout.flush()
if ((i*10)/nentries == tenpcount ) :
tsw.Stop()
time = tsw.RealTime()
tsw.Start(r.kFALSE)
finTime = 0.
frac = (i*1.0)/(nentries*1.0)
if (frac>0):
finTime = time / frac - time
finMin = finTime / 60.
sys.stdout.write("%d%% done. "%(tenpcount*10))
# sys.stdout.write("t=7.2f"%(time))
sys.stdout.write("t="+str(time))
sys.stdout.write(" projected finish=%7d s("%(finTime))
sys.stdout.write("%2.2f min). "%(finMin))
sys.stdout.write("\n")
sys.stdout.flush()
tenpcount = tenpcount + 1
elif ( (i*100)/nentries == onepcount ) :
#print('.', end='')
sys.stdout.write('.')
sys.stdout.flush()
onepcount = onepcount + 1
nJetsHT[0] = event.ra2_nJetsPt50Eta25
nJetsMHT[0] = event.ra2_nJetsPt30Eta50
htVal[0] = event.ra2_HT
mhtVal[0] = event.ra2_MHT
dphi1[0] = event.ra2_dPhiMHT1
dphi2[0] = event.ra2_dPhiMHT2
dphi3[0] = event.ra2_dPhiMHT3
jet1Pt[0] = event.ra2_Jet1Pt
jet1Eta[0] = event.ra2_Jet1Eta
jet2Pt[0] = event.ra2_Jet2Pt
jet2Eta[0] = event.ra2_Jet2Eta
jet3Pt[0] = event.ra2_Jet3Pt
jet3Eta[0] = event.ra2_Jet3Eta
puWt[0] = event.ra2_PUWt
eventWt[0] = event.ra2_EventWt
triggers = True
if options.sample == "gjets" or options.sample == "data":
photonPt[0] = event.ra2_Photon1Pt
photonEta[0] = event.ra2_Photon1Eta
photonMinDR[0] = event.ra2_Photon1MinDR
photonpfCH[0] = event.ra2_Photon1pfCH
photonpfNU[0] = event.ra2_Photon1pfNU
photonpfGA[0] = event.ra2_Photon1pfGA
if not options.isMC:
triggers = (event.ra2_Photon70PFHT400 or event.ra2_Photon70PFNoPUHT400)
if options.sample == "zmumu":
muon1Pt[0] = event.ra2_Muon1Pt
muon1Eta[0] = event.ra2_Muon1Eta
muon1MinDR[0] = event.ra2_Muon1MinDR
muon2Pt[0] = event.ra2_Muon2Pt
muon2Eta[0] = event.ra2_Muon2Eta
muon2MinDR[0] = event.ra2_Muon2MinDR
dimuonPt[0] = event.ra2_DiMuonPt
dimuonEta[0] = event.ra2_DiMuonEta
dimuonMinDR[0] = event.ra2_DiMuonMinDR
dimuonM[0] = event.ra2_DiMuonInvM
if not options.isMC:
triggers = (event.ra2_Mu13_Mu8 or event.ra2_Mu17_Mu8)
if options.isMC:
triggers = True
extra = True
if options.sample == "gjets" or options.sample == "data":
extra = cutF.ra2PhotonSelection(event,options.minpt, options.cutDR)
if options.sample == "zmumu":
extra = cutF.ra2MuonSelection(event,options.minpt, options.cutDR)
if (triggers) and extra and cutF.ra2Baseline(event):
tree.Fill()
#########
i = i + 1
#tree.Write()
outFile.Write()
outFile.Close()
