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, ))