#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)
Esempio n. 2
0
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"
Esempio n. 5
0
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"
Esempio n. 6
0
            "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))
Esempio n. 7
0
                            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,)
Esempio n. 8
0
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"
Esempio n. 9
0
    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"
Esempio n. 13
0
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"