#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)
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"
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)
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"
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))
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,)
if where == "hpg": out_path = "/cms/data/store/user/t2/users/klo/HPG/RA5/Sync2016/2018-09-04/" outputDir = out_path endModuleOutputDir = out_path elif where == "ihepa": out_path = "SigBkgDistribution/2018-09-04/" outputDir = "/raid/raid7/lucien/SUSY/RA5/" + out_path endModuleOutputDir = "/home/lucien/public_html/SUSY/RA5/" + out_path lepCats = ["HH", "HL", "LL"] nCores = 5 nEvents = -1 disableProgressBar = False justEndSequence = False verbose = False componentList = allMCSamples for dataset in componentList: if dataset.isMC: dataset.lumi = 35.9 for component in dataset.componentList: component.maxEvents = nEvents sequence = sr_sequence sequence.add(YieldCounter("YieldCounter")) endSequence = EndSequence(skipHadd=justEndSequence, ) outputInfo = OutputInfo("OutputInfo") outputInfo.outputDir = outputDir outputInfo.TFileName = "Yield.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"
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"
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"
if dataset.isMC: dataset.lumi = 35.9 for component in dataset.componentList: component.maxEvents = nEvents plots = [ Plot("rpv_HH_TightLoose", ["TH1D", "rpv_HH_TightLoose", "", 16, 52.5, 68.5], LambdaFunc('x: x.SRCategory'), selFunc=LambdaFunc( 'x: x.leptonCategory == \"HH\" and x.met_pt[0] < 50.')), Plot("baseline_HH_TightLoose", ["TH1D", "baseline_HH_TightLoose", "", 52, 0.5, 52.5], LambdaFunc('x: x.SRCategory'), selFunc=LambdaFunc( 'x: x.leptonCategory == \"HH\" and x.met_pt[0] > 50.')), ] plotter = Plotter("Plotter", plots) srProducer = SRProducer("SRProducer") sequence = tl_rpv_sequence sequence.add(srProducer) sequence.add(plotter) endSequence = EndSequence(skipHadd=justEndSequence, ) endSequence.add(PlotEndModule(endModuleOutputDir, plots)) outputInfo = OutputInfo("OutputInfo") outputInfo.outputDir = outputDir outputInfo.TFileName = "YieldPlot.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"