fidskimmer = FiducialSkimmer("FiducialSkimmer")
fidCounter = GenWeightCounter("GenWeightCounter", postfix="FidicialCut")
plotter = Plotter("Plotter", validation_plots)
sequence = Sequence()
sequence.add(xsWeighter)
sequence.add(preskimmer)
sequence.add(preskimCounter)
sequence.add(fidskimmer)
sequence.add(fidCounter)
#sequence.add(plotter)
cutflows = [
"Preskim",
"FidicialCut",
]
endModuleOutputDir = "/home/lucien/public_html/" + out_path
cutflowEndModule = CutflowEndModule(endModuleOutputDir,
cutflows=cutflows,
ignoreSumw=True)
endSequence = EndSequence(skipHadd=justEndSequence)
endSequence.add(cutflowEndModule)
#endSequence.add(PlotEndModule(endModuleOutputDir,validation_plots))
outputInfo = OutputInfo("OutputInfo")
outputInfo.outputDir = outputDir
outputInfo.TFileName = "EventWeight.root"
#out_path = "TestPlot/2018-12-04_StatInput/"
#out_path = "TestPlot/2018-12-05_test/"
#out_path = "ThetaInput/2019-05-27_StatInput/"
out_path = "ThetaInput/2019-05-29_StatInput/"
lumi = 41.298
nCores = 5
outputDir = system.getStoragePath()+"/lucien/LJMet/B2G/"+out_path
nEvents = -1
disableProgressBar = False
#componentList = bkgSamples + dataSamples + sigSamples
componentList = sigSamples
justEndSequence = False
for dataset in componentList:
if dataset.isMC:
dataset.lumi = lumi
for component in dataset.componentList:
component.maxEvents = nEvents
thetaProducer = ThetaTemplateProducer("TemplateProducer")
sequence = sr_sequence
sequence.add(thetaProducer)
outputInfo = OutputInfo("OutputInfo")
outputInfo.outputDir = outputDir
outputInfo.TFileName = "StatInput.root"
#endSequence = EndSequence(skipHadd=False,haddDataSamples=True,skipComponentHadd=True)
endSequence = EndSequence(skipHadd=False,haddDataSamples=True,skipComponentHadd=False)
endSequence.allDataName = "DATA"
LambdaFunc("x: x.etaL4[0]"),
LambdaFunc("x: x.phiL1[0]"),
LambdaFunc("x: x.phiL2[0]"),
LambdaFunc("x: x.phiL3[0]"),
LambdaFunc("x: x.phiL4[0]"),
LambdaFunc("x: x.etaL21"),
LambdaFunc("x: x.etaL31"),
LambdaFunc("x: x.etaL41"),
LambdaFunc("x: x.phiL21"),
LambdaFunc("x: x.phiL31"),
LambdaFunc("x: x.phiL41"),
LambdaFunc("x: x.mass4l[0]"),
LambdaFunc("x: x.massZ1[0]"),
LambdaFunc("x: x.massZ2[0]"),
#LambdaFunc("x: x.cosTheta1"),
#LambdaFunc("x: x.cosTheta2"),
#LambdaFunc("x: x.cosThetaStar"),
#LambdaFunc("x: x.phi"),
#LambdaFunc("x: x.phi1"),
#cosTheta1, cosTheta2, cosThetaStar, Phi, Phi1, diMuon2_pt, diMuon2_eta, diMuon12_delta_eta, diMuon12_delta_phi, diMuon12_delta_R
]
csvFileSetting = CSVFileSetting("csv",["input.csv","w"])
csvFileProducer = CSVFileProducer("CSVFileProducer",varsToWrite,csvFileSetting)
sequence.add(csvFileProducer)
outputInfo = OutputInfo("OutputInfo")
outputInfo.outputDir = outputDir
outputInfo.TFileName = "TrainingInput.root"
endSequence = EndSequence(skipHadd=False,haddDataSamples=True,)
selFunc = LambdaFunc('x: abs(x.idL1[0]) == 11 and abs(x.idL2[0]) == 11 and abs(x.idL3[0]) == 13 and abs(x.idL4[0]) == 13'),
fillFunc = LambdaFunc('x: [x.massZ1[0],x.massZ2[0],x.weight,]'),
),
"MuEl": BaseObject(
"MuEl",
histKey = "MuEl",
setting = ["TH2D","MuEl","",800,0.,80.,800,0.,80.,],
selFunc = LambdaFunc('x: abs(x.idL1[0]) == 13 and abs(x.idL2[0]) == 13 and abs(x.idL3[0]) == 11 and abs(x.idL4[0]) == 11'),
fillFunc = LambdaFunc('x: [x.massZ1[0],x.massZ2[0],x.weight,]'),
),
"MuEl-ElMu": BaseObject(
"MuEl-ElMu",
histKey = "MuEl-ElMu",
setting = ["TH2D","MuEl-ElMu","",800,0.,80.,800,0.,80.,],
selFunc = LambdaFunc('x: (abs(x.idL1[0]) == 13 and abs(x.idL2[0]) == 13 and abs(x.idL3[0]) == 11 and abs(x.idL4[0]) == 11) or (abs(x.idL1[0]) == 11 and abs(x.idL2[0]) == 11 and abs(x.idL3[0]) == 13 and abs(x.idL4[0]) == 13)'),
fillFunc = LambdaFunc('x: [x.massZ1[0],x.massZ2[0],x.weight,]'),
),
},
)
sequence = darkphoton_signal_unblind_sequence
sequence.remove(resonaceZ1Skimmer)
sequence.remove(resonaceZ2Skimmer)
sequence.add(statProducer)
outputInfo = OutputInfo("OutputInfo")
outputInfo.outputDir = outputDir
outputInfo.TFileName = "StatInput.root"
endSequence = EndSequence(skipHadd=justEndSequence,haddDataSamples=True)
#Plot("nJet", ["TH1D","nJet","",5,-0.5,4.5], LambdaFunc('x: x.njets_pt30_eta2p5[0]'), ),
]
#____________________________________________________________________________________________________
#
plots = muon_plots + general_plots + jet_plots + signif_plots# big list of Plot objects
#for plot in plots:
#plot.plotSetting.divideByBinWidth = True
for dataset in componentList:
if dataset.isMC:
dataset.lumi = lumi
for component in dataset.componentList:
component.maxEvents = nEvents
plotter = Plotter("Plotter",plots)
sequence = darkphoton_signal_unblind_sequence
sequence.add(plotter)
outputInfo = OutputInfo("OutputInfo")
outputInfo.outputDir = outputDir
outputInfo.TFileName = "DataMCDistribution_radiuscut"+radcutstr+"GeV.root"
endSequence = EndSequence(skipHadd=justEndSequence)
endModuleOutputDir = "/home/"+User+"/public_html/Higgs/HToZdZd/"+out_path
endSequence.add(PlotEndModule(endModuleOutputDir,plots,skipSF=True))
## Put an index.php file into Plots dir for easy visualization
copyFile('/home/'+User+'/',phpFile,endModuleOutputDir)
from RPV.Skimmer.ZMuMuSkimmer import ZMuMuSkimmer
nCores = 8
outputDir = "/raid/raid7/lucien/SUSY/RPV/SkimTree/ZMuMu/2018-06-19/DataMC_ZMuMuSelection_v1/"
nEvents = -1
disableProgressBar = False
justEndSequence = False
componentList = allDataSamples + allMCSamples
for dataset in componentList:
for component in dataset.componentList:
component.maxEvents = nEvents
mediumMuonProducer = PhysObjProducer("MediumMuonProducer","Muon","MediumMuons","Moriond17MediumMuon")
mediumElectronProducer = PhysObjProducer("MediumElectronProducer","Electron","MediumElectrons","Moriond17MediumElectron")
jetProducer = JetProducer("JetProducer","Jet",["MediumMuons","MediumElectrons"],"LooseJets","Moriond17LooseJet",0.4)
eventSkimmer = ZMuMuSkimmer("ZMuMuSkim")
treeProducer = TreeProducer("TreeProducer")
sequence = Sequence()
sequence.add(mediumMuonProducer)
sequence.add(mediumElectronProducer)
sequence.add(jetProducer)
sequence.add(eventSkimmer)
sequence.add(treeProducer)
endSequence = EndSequence()
outputInfo = OutputInfo("OutputInfo")
outputInfo.outputDir = outputDir
outputInfo.TFileName = "SkimTree.root"
"ggZZ": 0.4,
"Data": 1.0,
}
p.plotSetting.scatter_density = "1.0"
p.selectedSamples = [
"Higgs",
"ZZ",
"Data",
]
#p.plotSetting.cms_lumi = True
#p.plotSetting.tdr_style = True
#p.plotSetting.SetNColumns = 4
else:
p.plotSetting.divideByBinWidth = True
plotter = Plotter("Plotter", plots)
sequence = darkphoton_signal_unblind_sequence
sequence.add(plotter)
outputInfo = OutputInfo("OutputInfo")
outputInfo.outputDir = outputDir
outputInfo.TFileName = "DataMCDistribution.root"
endSequence = EndSequence(
skipHadd=skipHadd,
haddDataSamples=True,
)
endModuleOutputDir = system.getPublicHtmlPath() + "/Higgs/HToZdZd/" + out_path
endSequence.add(PlotEndModule(endModuleOutputDir, plots, skipSF=False))
p.plotSetting.line_width_dict[sig.name] = 4
for dataset in componentList:
if dataset.isMC:
dataset.lumi = lumi
for component in dataset.componentList:
component.maxEvents = nEvents
for p in plots:
if p.dim == 2:
p.plotSetting.leg_pos = [0.14,0.65,0.34,0.87]
p.plotSetting.x_axis_title = "m_{Z1}"
p.plotSetting.y_axis_title = "m_{Z2}"
p.plotSetting.minimum = 0.
p.selectedSamples = ["ggH","ZZ",] + [sigSampleDict[m].name for m in [5,30,60]]
else:
p.plotSetting.divideByBinWidth = True
plotter = Plotter("Plotter",plots)
sequence = darkphoton_signal_unblind_sequence
sequence.add(plotter)
outputInfo = OutputInfo("OutputInfo")
outputInfo.outputDir = outputDir
outputInfo.TFileName = "DataMCDistribution.root"
endSequence = EndSequence(skipHadd=skipHadd)
endModuleOutputDir = system.getPublicHtmlPath()+"/Higgs/HToZdZd/"+out_path
endSequence.add(PlotEndModule(endModuleOutputDir,plots,skipSF=False))
elif where == "ihepa":
out_path = "TightLoose/BTagEfficiency/2018-10-16/"
outputDir = "/raid/raid7/lucien/SUSY/RA5/"+out_path
endModuleOutputDir = "/home/lucien/public_html/SUSY/RA5/"+out_path
lepCats = [""]
nCores = 5
nEvents = -1
disableProgressBar = False
justEndSequence = False
verbose = False
componentList = [ c for c in skimComponentDict.values() if c.isMC]
for dataset in componentList:
if dataset.isMC:
dataset.lumi = 35.9
for component in dataset.componentList:
component.maxEvents = nEvents
btagEffAna = BTagEffAnalyzer("BTagEffAnalyzer")
btagEffEnd = BTagEffEndModule(endModuleOutputDir,cutflow="TightLoose",)
sequence = tl_rpv_sequence
sequence.add(btagEffAna)
endSequence = EndSequence(skipHadd=False,haddAllSamples=True)
endSequence.add(btagEffEnd)
outputInfo = OutputInfo("OutputInfo")
outputInfo.outputDir = outputDir
outputInfo.TFileName = "BTagEfficiency.root"
BaseObject(
"MuEl-ElMu",
histKey="MuEl-ElMu",
setting=[
"TH2D",
"MuEl-ElMu",
"",
800,
0.,
80.,
800,
0.,
80.,
],
selFunc=LambdaFunc(
'x: (abs(x.idL1[0]) == 13 and abs(x.idL2[0]) == 13 and abs(x.idL3[0]) == 11 and abs(x.idL4[0]) == 11) or (abs(x.idL1[0]) == 11 and abs(x.idL2[0]) == 11 and abs(x.idL3[0]) == 13 and abs(x.idL4[0]) == 13)'
),
fillFunc=LambdaFunc('x: [x.massZ1[0],x.massZ2[0],x.weight,]'),
),
},
)
sequence = darkphoton_signal_unblind_sequence
sequence.add(statProducer)
outputInfo = OutputInfo("OutputInfo")
outputInfo.outputDir = outputDir
outputInfo.TFileName = "StatInput.root"
endSequence = EndSequence(skipHadd=skipHadd, haddDataSamples=True)
from Plotter.Plotter import Plotter
from Plotter.PlotEndModule import PlotEndModule
from RA5.Config.Plotter.PlotDefinition import allPlots
nCores = 8
outputDir = "/raid/raid7/lucien/SUSY/RA5/Log/MCDistributions/2018-05-15/AllMCSamples_v1/"
nEvents = -1
disableProgressBar = False
justEndSequence = False
componentList = allMCSamples
for dataset in componentList:
dataset.lumi = 35.9
sequence = Sequence()
xsWeighter = XSWeighter("XSWeighter")
sequence.add(xsWeighter)
sequence.add(mediumMuonProducer)
sequence.add(looseMuonProducer)
sequence.add(mediumElectronProducer)
sequence.add(looseElectronProducer)
sequence.add(jetProducer)
sequence.add(Plotter("Plotter", allPlots))
endSequence = EndSequence()
endModuleOutputDir = "/home/lucien/public_html/SUSY/RA5/Log/MCDistributions/2018-05-15/AllMCSamples_v1/"
endSequence.add(PlotEndModule(endModuleOutputDir, allPlots))
outputInfo = OutputInfo("OutputInfo")
outputInfo.outputDir = outputDir
outputInfo.TFileName = "MCDistributions.root"
anaProducer = AnalysisProducer("AnaProducer")
eventSkimmer = EventSkimmer("StopToBLepSkim",cutflow="htCR")
hltSkimmer = HLTSkimmer("HLTSkim",cutflow="htCR")
jsonSkimmer = JSONSkimmer("JSONSkim")
sequence = Sequence()
sequence.add(jsonSkimmer)
sequence.add(hltSkimmer)
sequence.add(mediumMuonProducer)
sequence.add(mediumElectronProducer)
sequence.add(jetProducer)
sequence.add(xsWeighter)
sequence.add(puWeighter)
sequence.add(anaProducer)
sequence.add(eventSkimmer)
sequence.add(plotter)
endSequence = EndSequence(skipHadd=justEndSequence)
if not ratio_switch:
#endModuleOutputDir = "/home/kshi/public_html/dataPlot/htCR"
endModuleOutputDir = "/home/lucien/public_html/SUSY/RPV/"+out_path
endSequence.add(PlotEndModule(endModuleOutputDir,mc_plots,ratio_switch))
else:
#endModuleOutputDir = "/home/kshi/public_html/ratioPlot/htCR"
endModuleOutputDir = "/home/lucien/public_html/SUSY/RPV/"+out_path
endSequence.add(PlotEndModule(endModuleOutputDir,ratio_plots,ratio_switch))
outputInfo = OutputInfo("OutputInfo")
outputInfo.outputDir = outputDir
outputInfo.TFileName = "DataMCDistribution.root"
nCores = 1
nEvents = -1
disableProgressBar = True
justEndSequence = False
verbose = False
#componentList = [SkimSyncMC]
componentList = [SyncMC]
for dataset in componentList:
if dataset.isMC:
dataset.lumi = 35.9
for component in dataset.componentList:
component.maxEvents = nEvents
if not os.path.exists(endModuleOutputDir):
os.makedirs(endModuleOutputDir)
sequence = sync_sequence
lepSyncPrinter = LeptonSyncPrinter(
"SyncPrinter", endModuleOutputDir + "uf_TTW_80X_dump_leptons.txt")
jetSyncPrinter = JetSyncPrinter(
"SyncPrinter", endModuleOutputDir + "uf_TTW_80X_dump_jets.txt")
sequence.add(lepSyncPrinter)
sequence.add(jetSyncPrinter)
endSequence = EndSequence(haddAllSamples=True)
outputInfo = OutputInfo("OutputInfo")
outputInfo.outputDir = outputDir
outputInfo.TFileName = "SyncFile.root"
LambdaFunc("x: x.Muon_GenPhi[0]"),
selFunc=LambdaFunc("x: len(x.Muon_GenPt) > 0"),
getEventWeight=LambdaFunc("x: 1."),
),
Plot(
"Muon_GenCharge",
[
"TH1D",
"Muon_GenCharge",
"",
3,
-1.5,
1.5,
],
LambdaFunc("x: x.Muon_GenCharge[0]"),
selFunc=LambdaFunc("x: len(x.Muon_GenCharge) > 0"),
getEventWeight=LambdaFunc("x: 1."),
),
]
sequence = Sequence()
plotter = Plotter("Plotter", plots)
sequence.add(plotter)
outputInfo = OutputInfo("OutputInfo")
outputInfo.outputDir = outputDir
outputInfo.TFileName = "DataMCDistribution.root"
endSequence = EndSequence()
endSequence.add(PlotEndModule(outputDir, plots, skipSF=False))
]
plots = muon_plots + general_plots + jet_plots
#plots = muon_plots + general_plots
#for plot in plots:
# plot.plotSetting.divideByBinWidth = True
for dataset in componentList:
if dataset.isMC:
dataset.lumi = lumi
for component in dataset.componentList:
component.maxEvents = nEvents
plotter = Plotter("Plotter",plots)
#sequence = darkphoton_fullm4l_sequence
sequence = darkphoton_sb_sequence
sequence.add(plotter)
outputInfo = OutputInfo("OutputInfo")
outputInfo.outputDir = outputDir
outputInfo.TFileName = "DataMCDistribution.root"
endSequence = EndSequence(skipHadd=justEndSequence) # compiles the histograms!
#endModuleOutputDir = "/home/lucien/public_html/Higgs/HToZdZd/"+out_path
endModuleOutputDir = "/home/rosedj1/public_html/Higgs/HToZdZd/"+out_path
endSequence.add(PlotEndModule(endModuleOutputDir,plots,skipSF=True))
## Put an index.php file into Plots dir for easy visualization
copyFile('/home/rosedj1/',phpFile,endModuleOutputDir)
p.plotSetting.leg_name_dict = {
"HZZd_M30": "H #rightarrow Z Z_{D} m_{Z_{D}} = 30 GeV #varepsilon = 0.05",
"HZZd_M15": "H #rightarrow Z Z_{D} m_{Z_{D}} = 15 GeV #varepsilon = 0.05",
"ZPlusX": "Z+X",
}
plot_el.plotSetting.linear_max_factor = 2.5
plot_el.plotSetting.custom_latex_list = [
BaseObject(plot_el.key,x_pos=4.,y_pos=13.,text_size=0.035,text="2e channel",),
]
plot_mu.plotSetting.linear_max_factor = 2.0
plot_mu.plotSetting.custom_latex_list = [
BaseObject(plot_mu.key,x_pos=4.,y_pos=16.5,text_size=0.035,text="2#mu channel",),
]
for sigName in mergeSigSampleDict:
for p in plots:
p.plotSetting.line_style_dict[sigName] = 1
p.plotSetting.line_width_dict[sigName] = 3
outputInfo = OutputInfo("OutputInfo")
outputInfo.outputDir = outputDir
outputInfo.TFileName = "DataMCDistribution.root"
endSequence = EndSequence(
skipHadd=skipHadd,
skipComponentHadd=skipHadd,
)
endModuleOutputDir = system.getPublicHtmlPath()+"/Higgs/DarkZ/"+end_out_path
endSequence.add(PaperPlotEndModule(endModuleOutputDir,plots,skipSF=True))
Plot("LepTightPt1" + lepCat,
["TH1D", "LepTightPt1" + lepCat, "", 20, 0., 200.],
LambdaFunc('x: x.firstLep.pt')),
Plot("LepTightEta1" + lepCat,
["TH1D", "LepTightEta1" + lepCat, "", 10, -3., 3.],
LambdaFunc('x: x.firstLep.eta')),
Plot("LepTightPhi1" + lepCat,
["TH1D", "LepTightPhi1" + lepCat, "", 10, -5., 5.],
LambdaFunc('x: x.firstLep.phi')),
#Plot("JetPt1"+lepCat, ["TH1D","JetPt1"+lepCat,"",20,0.,400.], LambdaFunc('x: x.selJets[0].pt') ),
#Plot("JetEta1"+lepCat, ["TH1D","JetEta1"+lepCat,"",10,-3.,3.], LambdaFunc('x: x.selJets[0].eta') ),
#Plot("JetPhi1"+lepCat, ["TH1D","JetPhi1"+lepCat,"",10,-5.,5.], LambdaFunc('x: x.selJets[0].phi') ),
#Plot("JetPt2"+lepCat, ["TH1D","JetPt2"+lepCat,"",20,0.,400.], LambdaFunc('x: x.selJets[1].pt') ),
#Plot("JetEta2"+lepCat, ["TH1D","JetEta2"+lepCat,"",10,-3.,3.], LambdaFunc('x: x.selJets[1].eta') ),
#Plot("JetPhi2"+lepCat, ["TH1D","JetPhi2"+lepCat,"",10,-5.,5.], LambdaFunc('x: x.selJets[1].phi') ),
])
plotter = Plotter("Plotter", plots)
sequence = gamma_cr_sequence
sequence.add(plotter)
endSequence = EndSequence(skipHadd=False, )
endSequence.add(PlotEndModule(endModuleOutputDir, plots, scaleToData=True))
outputInfo = OutputInfo("OutputInfo")
outputInfo.outputDir = outputDir
outputInfo.TFileName = "DataMCDistributions.root"
#outputDir = "/raid/raid7/lucien/CSC/GasGain/Log/2018-05-03/comp_SingleMuon2017E_v3_skim_anl_all_ZMuMu_allChannel_CombinedME11ab/"
#outputDir = "/raid/raid7/lucien/CSC/GasGain/Log/2018-05-03/comp_SingleMuon2017E_v3_skim_anl_all_ZMuMu_allChannel_CombinedME11ab_yloc/"
#outputDir = "/raid/raid7/lucien/CSC/GasGain/Log/2018-05-07/"+compStr+"_anl/"
outputDir = "/raid/raid7/lucien/CSC/GasGain/Log/2018-05-17/"
justEndSequence = True
sequence = Sequence()
sequence.add(SkimTreeZMuMuSkimmer("ZMuMuSkimmer"))
gasGainPlotter = SkimTreeGasGainPlotter("GasGainPlotter")
posPlotter = SkimTreePositionPlotter("SkimTreePositionPlotter", y_segment_dict)
skimTreeRunPlotter = SkimTreeRunPlotter("RunPlotter")
sequence.add(gasGainPlotter)
sequence.add(posPlotter)
sequence.add(skimTreeRunPlotter)
endSequence = EndSequence(skipHadd=True)
#endModuleOutputDir = "/home/lucien/public_html/CSC/GasGainAnalysis/Log/2018-03-21/InclusiveSelection/"
#endModuleOutputDir = "/home/lucien/public_html/CSC/GasGainAnalysis/Log/2018-03-21/ZMuMuSelection/"
#endModuleOutputDir = "/home/lucien/public_html/CSC/GasGainAnalysis/Log/2018-04-16/ZMuMuSelection_ExcludeBadChannels/"
#endModuleOutputDir = "/home/lucien/public_html/CSC/GasGainAnalysis/Log/2018-05-07/"+compStr+"_anl/"
endModuleOutputDir = "/home/lucien/public_html/CSC/GasGainAnalysis/Log/2018-05-17/comp_SingleMuon2017_skim_anl/"
skimTreeGasGainEndModule = SkimTreeGasGainEndModule(endModuleOutputDir)
endSequence.add(skimTreeGasGainEndModule)
skimTreePositionEndModule = SkimTreePositionEndModule(endModuleOutputDir,
ChamberTypes,
y_segment_dict,
normalize=True)
endSequence.add(skimTreePositionEndModule)
outputInfo = OutputInfo("OutputInfo")
outputInfo.outputDir = outputDir
input_channel_dict = {
"4mu": LambdaFunc('event: ' + mu_ch_sel_str),
}
sequence = signal_sequence
yieldProducer = ParaYieldProducer(
"ParaYieldProducer",
channelDict=input_channel_dict,
)
if mvaSkim:
mvaSkimmer = MVASkimmer(
"MVSkimmer",
BaseObject(
"MVASetting",
modelPath=
"/raid/raid7/lucien/AnalysisCode/ML/Zprime-ML/result/2019-06-04_Run2017_qqZZ-zpToMuMu-M40_m4l-cosTheta1-cosTheta2-cosThetaStar-phi-phi1/mlp-classifier.pkl",
threshold=0.45,
varFunc=LambdaFunc(
"x: [(x.mass4l[0]-80.)/20.,x.cosTheta1, x.cosTheta2, x.cosThetaStar, x.phi, x.phi1,]"
),
),
)
sequence.add(mvaSkimmer)
sequence.add(yieldProducer)
outputInfo = OutputInfo("OutputInfo")
outputInfo.outputDir = outputDir
outputInfo.TFileName = "StatInput.root"
endSequence = EndSequence(skipHadd=justEndSequence)
from RA5.Dataset.Run2016.Sept18_v1_Data import *
#from RA5.Dataset.Run2016.Sept18_v1_skim import *
out_path = "RPV/EventList/rpv_binning2016_v6_136p3fb-1/2019-01-23_v1.0_IncludeOnly2LepInLowMETBin_181019/"
outputDir = "/raid/raid7/lucien/SUSY/RA5/" + out_path
endModuleOutputDir = "/home/lucien/public_html/SUSY/RA5/" + out_path
nCores = 5
nEvents = -1
disableProgressBar = False
justEndSequence = False
verbose = False
#componentList = [ c for c in skimComponentDict.values() if c.isData ]
componentList = dataComponentDict.values()
for dataset in componentList:
if dataset.isMC:
dataset.lumi = 136.3
for component in dataset.componentList:
component.maxEvents = nEvents
eventPrinter = EventPrinter("EventPrinter", endModuleOutputDir)
sequence = low_met_sequence
sequence.add(eventPrinter)
endSequence = EndSequence(skipHadd=False, haddAllSamples=True)
outputInfo = OutputInfo("OutputInfo")
outputInfo.outputDir = outputDir
outputInfo.TFileName = "dummy.root"
higgsSR_sel_str = 'event.mass4l[0] > 118. and event.mass4l[0] < 130.'
higgsSB_sel_str = 'not (event.mass4l[0] > 118. and event.mass4l[0] < 130.) and event.mass4l[0] > 100. and event.mass4l[0] < 170.'
#higgsSB_sel_str = 'not (event.mass4l[0] > 118. and event.mass4l[0] < 130.)'
input_channel_dict = {
"2mu_HiggsSR":
LambdaFunc('event: ' + ' and '.join([mu_ch_sel_str, higgsSR_sel_str])),
"2e_HiggsSR":
LambdaFunc('event: ' + ' and '.join([el_ch_sel_str, higgsSR_sel_str])),
"2mu_HiggsSB":
LambdaFunc('event: ' + ' and '.join([mu_ch_sel_str, higgsSB_sel_str])),
"2e_HiggsSB":
LambdaFunc('event: ' + ' and '.join([el_ch_sel_str, higgsSB_sel_str])),
}
sequence = darkphoton_signal_sequence
yieldProducer = ParaYieldProducer(
"ParaYieldProducer",
systList=syst_list,
channelDict=input_channel_dict,
)
#sequence.add(variableProducer)
sequence.add(yieldProducer)
outputInfo = OutputInfo("OutputInfo")
outputInfo.outputDir = outputDir
outputInfo.TFileName = "StatInput.root"
#endSequence = EndSequence(skipHadd=justEndSequence)
endSequence = EndSequence(skipHadd=False)
anaProducer = AnalysisProducer("AnaProducer")
eventSkimmer = EventSkimmer("StopToBLepSkim", cutflow="stCR")
hltSkimmer = HLTSkimmer("HLTSkim", cutflow="stCR")
jsonSkimmer = JSONSkimmer("JSONSkim")
sequence = Sequence()
sequence.add(jsonSkimmer)
sequence.add(hltSkimmer)
sequence.add(mediumMuonProducer)
sequence.add(mediumElectronProducer)
sequence.add(jetProducer)
sequence.add(xsWeighter)
sequence.add(puWeighter)
sequence.add(anaProducer)
sequence.add(eventSkimmer)
sequence.add(plotter)
endSequence = EndSequence(skipHadd=False)
if not ratio_switch:
endModuleOutputDir = "/home/kshi/public_html/dataPlot/stCR/"
endSequence.add(PlotEndModule(endModuleOutputDir, data_plots,
ratio_switch))
else:
endModuleOutputDir = "/home/kshi/public_html/ratioPlot/stCR/"
endSequence.add(
PlotEndModule(endModuleOutputDir, ratio_plots, ratio_switch))
outputInfo = OutputInfo("OutputInfo")
outputInfo.outputDir = outputDir
outputInfo.TFileName = "DataMCDistribution.root"
for component in dataset.componentList:
component.maxEvents = nEvents
plots = []
plotter = Plotter("Plotter", plots)
leptonJetProducer = LeptonJetProducer("LeptonJetProducer", "Run2016")
xsWeighter = XSWeighter("XSWeighter")
NJet40Producer = NJet40Producer("NJet40Producer")
CategoryProducer = CategoryProducer("CategoryProducer")
YieldCounter = YieldCounter("YieldCounter")
#baselineSkimmer = BaselineSkimmer("BaselineSkimmer")
#signalRegionSkimmer = SignalRegionSkimmer("SignalRegionSkimmer")
sequence = Sequence()
#sequence.add(leptonJetProducer)
sequence.add(xsWeighter)
#sequence.add(baselineSkimmer)
sequence.add(NJet40Producer)
sequence.add(CategoryProducer)
sequence.add(YieldCounter)
sequence.add(plotter)
endSequence = EndSequence(skipHadd=False)
endModuleOutputDir = "/home/kshi/public_html/RA5/yieldcount/"
endSequence.add(PlotEndModule(endModuleOutputDir, plots))
outputInfo = OutputInfo("OutputInfo")
outputInfo.outputDir = outputDir
outputInfo.TFileName = "MCDistributions.root"
fname = f.replace(".root", "")
tmp = Dataset(
fname,
ComponentList(
[Component(fname, os.path.join(inputDir, f), "tree", False)]),
)
componentList.append(tmp)
rootTree = "tree"
nCores = 10
disableProgressBar = False
nEvents = -1
outputDir = "output/GasGain_2021-03-04/"
justEndSequence = False
sequence = Sequence()
sequence.add(SkimTreeZMuMuSkimmer("ZMuMuSkimmer"))
gasGainPlotter = SkimTreeGasGainPlotter("GasGainPlotter")
skimTreeRunPlotter = SkimTreeRunPlotter("RunPlotter")
sequence.add(gasGainPlotter)
sequence.add(skimTreeRunPlotter)
endSequence = EndSequence(skipHadd=justEndSequence)
endModuleOutputDir = outputDir
skimTreeGasGainEndModule = SkimTreeGasGainEndModule(endModuleOutputDir)
endSequence.add(skimTreeGasGainEndModule)
outputInfo = OutputInfo("OutputInfo")
outputInfo.outputDir = outputDir
outputInfo.TFileName = "test1.root"
nCores = 8
outputDir = "/raid/raid7/lucien/" + out_path
nEvents = -1
disableProgressBar = False
justEndSequence = True
componentList = sigSamples
for dataset in componentList:
if dataset.isMC:
dataset.lumi = 77.30
for component in dataset.componentList:
component.maxEvents = nEvents
anaSkimmer = AnalysisSkimmer("AnalysisSkimmer")
genWeightCounter = GenWeightCounter("GenWeightCounter")
sequence = Sequence()
sequence.add(anaSkimmer)
sequence.add(genWeightCounter)
efficiencyEndModule = EfficiencyEndModule("/home/lucien/public_html/" +
out_path)
endSequence = EndSequence(skipHadd=justEndSequence)
endSequence.add(efficiencyEndModule)
outputInfo = OutputInfo("OutputInfo")
outputInfo.outputDir = outputDir
outputInfo.TFileName = "EventWeight.root"
text="4#mu channel",
),
]
plot_2e2mu.plotSetting.custom_latex_list = [
BaseObject(
plot_2e2mu.key,
x_pos=10.,
y_pos=4.0,
text_size=0.035,
text="2e2#mu channel",
),
]
for sigName in mergeSigSampleDict:
for p in plots:
p.plotSetting.line_style_dict[sigName] = 1
p.plotSetting.line_width_dict[sigName] = 3
outputInfo = OutputInfo("OutputInfo")
outputInfo.outputDir = outputDir
outputInfo.TFileName = "DataMCDistribution.root"
endSequence = EndSequence(skipHadd=True)
endModuleOutputDir = system.getPublicHtmlPath(
) + "/Higgs/HToZdZd/" + end_out_path
endSequence.add(PaperPlotEndModule(
endModuleOutputDir,
plots,
skipSF=True,
))
