from Core.Dataset import Dataset
from Core.Utils.MakeComponent import makeComponents
# ________________________________________________________________________________________________ ||
sampleName = "SyncMC"
#filePath = "/cms/data/store/user/t2/users/klo/HeppyTree/heppy_80X_RA5_Legacy/SyncMC2016/TTW_RA5_sync/treeProducerSusyRA5/tree.root"
#filePath = "/raid/raid7/lucien/SUSY/RA5/HeppyTree/SyncMC2016/TTW_RA5_sync/treeProducerSusyRA5/tree.root"
#filePath = "/raid/raid7/lucien/SUSY/RA5/SkimTree/SyncMC2016v2/TTW_RA5_sync/treeProducerSusyRA5/tree.root"
#filePath = "/raid/raid7/lucien/SUSY/RA5/SkimTree/SyncMC2016v3/TTW_RA5_sync/treeProducerSusyRA5/tree.root"
#filePath = "/raid/raid7/lucien/SUSY/RA5/SkimTree/SyncMC2016v4/TTW_RA5_sync/treeProducerSusyRA5/tree.root"
filePath = "/raid/raid7/lucien/SUSY/RA5/SkimTree/SyncMC2016v5/TTW_RA5_sync/treeProducerSusyRA5/tree.root"
inUFTier2 = False
cmpList = ComponentList([Component(
sampleName,
filePath,
"tree",
inUFTier2,
)])
SyncMC = Dataset(
"SyncMC",
cmpList,
isMC=True,
xs=1,
)
SyncMC.setSumWeight(filePath, "SumGenWeights", inUFTier2)
# ________________________________________________________________________________________________ ||
# Skim Tree
skimCmpList = ComponentList([
Component(
from Core.ComponentList import *
from Core.Dataset import Dataset
from Utils.System import system
dataTreeDir = system.getStoragePath(
) + "/lucien/Higgs/DarkZ-NTuple/20181116/SkimTree_DarkPhoton_ZX_Run2017Data_m4l70/"
inUFTier2 = False
# ____________________________________________________________________________________________________________________________________________ ||
Data_Run2017_cmpList = ComponentList([
Component("Data_Run2017",
dataTreeDir +
"Data_Run2017-17Nov2017_noDuplicates_FRWeightv2.root",
"passedEvents",
inUFTier2=inUFTier2),
])
Data_Run2017 = Dataset(
"Data_Run2017",
Data_Run2017_cmpList,
isMC=False,
)
from Core.Utils.MakeComponent import makeComponents
from DataMC.Heppy.Run2016.HaddMC import TT_pow
sampleName = "TT_pow"
fileName = "treeProducerSusyRA5.root"
#TreeDir = "/cms/data/store/user/t2/users/klo/HeppyTree/heppy_80X_RA5_Legacy/July18_v2_LeptonJetRecleaner_TT_pow/"
TreeDir = "/cms/data/store/user/t2/users/klo/HeppyTree/heppy_80X_RA5_Legacy/July18_v2_LeptonJetRecleaner/"
#sumweight_path = "/cms/data/store/user/t2/users/klo/HeppyTree/heppy_80X_RA5_Legacy/July18_v1/"
inUFTier2 = True
#filePath = os.path.join(sumweight_path,sampleName,fileName)
#cmp = makeComponents(sampleName, TreeDir, "Events", inUFTier2)
cmpList = ComponentList(
# [ Component("TT_pow",TreeDir + sampleName + "/"+"TT_pow_%s_SkimTree.root"%i,"tree",inUFTier2) for i in range(0,414)]
[
Component("TT_pow", TreeDir + sampleName + "/" + "SkimTree.root",
"tree", inUFTier2)
])
TT_pow.componentList = cmpList
#TT_pow= Dataset(
# "TT_pow",
# cmpList,
# isMC = True,
# xs = 1,
# )
#TT_pow.setSumWeight(filePath,"SumGenWeights",inUFTier2)
Samples = [TT_pow]
from Core.ComponentList import *
from Core.Dataset import Dataset
from Utils.System import system
from Utils.SumWeight import handleSumWeight
import os
# ____________________________________________________________________________________________________________________________________________ ||
#
HToZaTo2l2g_M1 = Dataset(
"HToZaTo2l2g_M1",
ComponentList([
#Component("HToZaTo2l2g_M1","/Users/lucien/CMS/DelphesAnalysis/HToZaTo2l2g/Data/2020-06-02_MuonTreeProducer_combined.root","LiteTree",inUFTier2=False),
Component(
"HToZaTo2l2g_M1",
"/cmsuf/data/store/user/t2/users/klo/Delphes/ALP_HToZaTo2l2g_M1/2020-06-02/MuonTreeProducer/2020-06-02_MuonTreeProducer_combined.root",
"LiteTree",
inUFTier2=False),
]),
isMC=True,
xs=1,
)
HToZaTo2l2g_M1.sumw = 4995000.
10,
15,
20,
25,
30,
35,
]
# ____________________________________________________________________________________________________________________________________________ ||
sigSampleDict = {}
for m in mass_points:
tmpDataset = Dataset(
datasetName % m,
ComponentList([
Component(datasetName % m,
os.path.join(sigSkimTreeDir, fileNameTemplate % m),
skimTreePath,
inUFTier2=inUFTier2)
]),
isMC=True,
isSignal=True,
xs=Higgs_prod_xs * epsilon**2 * HZZd_xs_dict[m],
)
handleSumWeight(
tmpDataset,
system,
os.path.join(sigTreeDir, fileNameTemplate % m),
sumWeightHist,
True,
saveSumWeightTxt,
os.path.join(sigSkimTreeDir,
(fileNameTemplate % m).replace(".root", ".txt")),
from Utils.System import system
from Utils.SumWeight import handleSumWeight
#dataTreeDir = "/raid/raid7/lucien/Higgs/DarkZ-NTuple/20180806/SkimTree_Data80X_ZXCRSelection/"
#dataTreeDir = "/raid/raid7/lucien/Higgs/DarkZ-NTuple/20180806/SkimTree_Data80X_HIG-16-041-ZXCRSelection_v2/"
#dataTreeDir = "/raid/raid7/lucien/Higgs/DarkZ-NTuple/20180905/SkimTree_DarkPhoton_ZX_Run2016Data_m4l70/"
#dataTreeDir = "/raid/raid7/lucien/Higgs/DarkZ-NTuple/20180924/SkimTree_DarkPhoton_ZX_Run2016Data_m4l70/"
dataTreeDir = "/raid/raid7/kshi/Zprime/20190729/SkimTree_Run2016_mem_controlregion_Data/"
inUFTier2 = False
# ____________________________________________________________________________________________________________________________________________ ||
Data_memCR_Run2016_cmpList = ComponentList(
[
#Component("Data_Run2016_DoubleEG",dataTreeDir+"DoubleEG.root","passedEvents",inUFTier2=inUFTier2),
#Component("Data_Run2016_DoubleMuon",dataTreeDir+"DoubleMuon.root","passedEvents",inUFTier2=inUFTier2),
#Component("Data_Run2016_MuonEG",dataTreeDir+"MuonEG.root","passedEvents",inUFTier2=inUFTier2),
#Component("Data_Run2016_SingleElectron",dataTreeDir+"SingleElectron.root","passedEvents",inUFTier2=inUFTier2),
#Component("Data_Run2016_SingleMuon",dataTreeDir+"SingleMuon.root","passedEvents",inUFTier2=inUFTier2),
Component("Data_memCR_Run2016_totaldata",dataTreeDir+"total_Data.root","passedEvents",inUFTier2=inUFTier2),
]
)
Data_memCR_Run2016 = Dataset(
"Data_memCR_Run2016",
Data_memCR_Run2016_cmpList,
isMC = True,
skipWeight = True,
)
'''
bkgSamples = [
Data_memCR_Run2016,
]
'''
from Core.ComponentList import *
from Core.Dataset import Dataset
from Utils.System import system
from Utils.SumWeight import handleSumWeight
predCRTreeDir = system.getStoragePath(
) + "/lucien/Higgs/DarkZ-NTuple/20190218/SkimTree_HToZdZd_ZX_Run2016Data_m4l70_noZCandRatioCut/"
inUFTier2 = False
# ____________________________________________________________________________________________________________________________________________ ||
# Z+X
predCR_cmpList = ComponentList([
#Component("PredCR",predCRTreeDir+"Data_Run2016-2l_noDuplicates_FRWeight.root","passedEvents",False),
Component(
"PredCR", predCRTreeDir +
"Data_Run2016-2l_noDuplicates_FRWeightFromVukasin.root",
"passedEvents", False),
])
predCR = Dataset(
"PredCR",
predCR_cmpList,
isMC=True,
isSignal=True,
skipWeight=True,
)
from Core.Dataset import Dataset
from Utils.System import system
from Utils.SumWeight import handleSumWeight
bkgSkimTreeDir = system.getStoragePath()+"/"+os.environ["USER"]+"/Higgs/DarkZ-NTuple/20190218/SkimTree_HToZdZd_Run2016Data_m4l70_noZCandRatioCut/"
bkgTreeDir = "/cms/data/store/user/t2/users/klo/Higgs/HToZdZd/80X_MCProd_191127/"
inUFTier2 = False
sumWeightHist = "Ana/sumWeights"
sumWeightFromT2 = True
saveSumWeightTxt = False
# ____________________________________________________________________________________________________________________________________________ ||
# ttZ
ttZ_cmpList = ComponentList(
[
Component("ttZ",bkgSkimTreeDir+"ttZJets_13TeV_madgraphMLM.root","passedEvents",inUFTier2=inUFTier2),
]
)
ttZ = Dataset(
"ttZ",
ttZ_cmpList,
isMC = True,
xs = 0.6529,
)
handleSumWeight(
ttZ,
system,
bkgTreeDir+"ttZJets_13TeV_madgraphMLM.root",
sumWeightHist,
True,
from Core.ComponentList import *
from Core.Dataset import Dataset
cmp = Component(
"TGJets",
"/cms/data/store/user/klo/RA5/NTuples/2016/NanoAOD/MC/TGJets_TuneCUETP8M1_13TeV_amcatnlo_madspin_pythia8/InclusiveSelection_v1/180509_144401/0000/",
"Events",
keyword="tree",
inUFTier2=True,
)
cmpList = ComponentList([
cmp,
], )
TGJets = Dataset(
"TGJets",
cmpList,
xs=2.967, #pb
)
TGJets.setSumWeight(
"/raid/raid7/lucien/SUSY/RA5/SumGenWeight/NanoAOD_InclusiveSelection_v1/Run2016/TGJets/EventWeight.root"
)
from Core.ComponentList import *
from Core.Dataset import Dataset
from Utils.System import system
from Utils.SumWeight import handleSumWeight
import os
# ____________________________________________________________________________________________________________________________________________ ||
#
ZJetsToLL_LO_MLM = Dataset(
"ZJetsToLL_LO_MLM",
ComponentList([
Component(
"ZJetsToLL_LO_MLM",
"/cmsuf/data/store/user/t2/users/klo/Delphes/ZJetsToLL_LO_MLM/2020-07-09/MuonTreeProducer/2020-07-09_MuonTreeProducer.root",
"LiteTree",
inUFTier2=False),
]),
isMC=True,
xs=1,
)
ZJetsToLL_LO_MLM.sumw = 4899797.
from Core.ComponentList import *
from Core.Dataset import Dataset
from Utils.System import system
dataTreeDir = system.getStoragePath()+"/lucien/Higgs/DarkZ-NTuple/20190402/SkimTree_DarkPhoton_ZX_Run2018Data_m4l70/"
inUFTier2 = False
# ____________________________________________________________________________________________________________________________________________ ||
Data_Run2018_cmpList = ComponentList(
[
Component("Data_Run2018",dataTreeDir+"Data_Run2018_noDuplicates_FRWeightFromVukasinWZRemoved.root","passedEvents",inUFTier2=inUFTier2),
]
)
Data_Run2018 = Dataset(
"Data_Run2018",
Data_Run2018_cmpList,
isMC = False,
)
from Core.ComponentList import *
from Core.Dataset import Dataset
#dataTreeDir = "/raid/raid7/lucien/Higgs/DarkZ-NTuple/20180806/SkimTree_Data80X_ZXCRSelection/"
#dataTreeDir = "/raid/raid7/lucien/Higgs/DarkZ-NTuple/20180806/SkimTree_Data80X_HIG-16-041-ZXCRSelection_v2/"
#dataTreeDir = "/raid/raid7/lucien/Higgs/DarkZ-NTuple/20180905/SkimTree_DarkPhoton_ZX_Run2016Data_m4l70/"
dataTreeDir = "/raid/raid7/lucien/Higgs/DarkZ-NTuple/20180924/SkimTree_DarkPhoton_ZX_Run2016Data_m4l70/"
inUFTier2 = False
# ____________________________________________________________________________________________________________________________________________ ||
Data_Run2016_cmpList = ComponentList([
Component("Data_Run2016",
dataTreeDir + "Data_Run2016_noDuplicates_FRWeightCorr.root",
"passedEvents",
inUFTier2=inUFTier2),
])
Data_Run2016 = Dataset(
"Data_Run2016",
Data_Run2016_cmpList,
isMC=False,
)
#bkgTreeDir = "/raid/raid7/lucien/Higgs/DarkZ-NTuple/20180806/SkimTree_MC80X_ZXCRSelection/"
#bkgTreeDir = "/raid/raid7/lucien/Higgs/DarkZ-NTuple/20180806/SkimTree_MC80X_HIG-16-041-ZXCRSelection_v2/"
bkgTreeDir = system.getStoragePath(
) + "/lucien/Higgs/DarkZ-NTuple/20190402/SkimTree_DarkPhoton_ZX_Run2018Data_m4l70/"
bkgTreeDirT2 = "/cms/data/store/user/t2/users/klo/Higgs/DarkZ/NTuples/ZPlusX_Early2017_v1/"
predCRTreeDir = system.getStoragePath(
) + "/lucien/Higgs/DarkZ-NTuple/20190402/SkimTree_DarkPhoton_ZX_Run2018Data_m4l70/"
inUFTier2 = False
saveSumWeightTxt = False
sumWeightHist = "Ana/sumWeights"
# ____________________________________________________________________________________________________________________________________________ ||
# Z+X
predCR_cmpList = ComponentList([
Component(
"PredCR", predCRTreeDir +
"/Data_Run2018_noDuplicates_FRWeightFromVukasinWZRemoved.root",
"passedEvents", False),
])
predCR = Dataset(
"PredCR",
predCR_cmpList,
isMC=True,
isSignal=True,
skipWeight=True,
)
## ____________________________________________________________________________________________________________________________________________ ||
#DYJetsToLL_M50_cmpList = ComponentList(
# [
# Component("DYJetsToLL_M50",bkgTreeDir+"DYJetsToLL_M50.root","passedEvents",inUFTier2=inUFTier2),
# ]
dir_path = "/raid/raid7/lucien/SUSY/RA5/HeppyTree/Sept18_v1_GammaCRSkim/"
inUFTier2 = False
treeName = "tree"
rootFileName = "SkimTree.root"
postfix = "/"+rootFileName
if inUFTier2:
fileNames = [ n for n in listdir_uberftp(dir_path) if "txt" not in n and "skip" not in n]
else:
fileNames = [ n for n in os.listdir(dir_path) if "txt" not in n and "skip" not in n]
skimComponentDict = {}
for fileName in fileNames:
if "ext" not in fileName:
sampleName = fileName.replace(postfix,"")
else:
sampleName = fileName.split("/")[-1].split("_ext")[0]
if sampleName in componentDict:
if sampleName not in skimComponentDict:
skimComponentDict[sampleName] = componentDict[sampleName]
skimComponentDict[sampleName].componentList = ComponentList( [Component(sampleName,os.path.join(dir_path,fileName,rootFileName),treeName,inUFTier2,),] )
else:
skimComponentDict[sampleName].componentList.extend( [Component(sampleName,os.path.join(dir_path,fileName,rootFileName),treeName,inUFTier2,),] )
if sampleName in dataComponentDict:
if sampleName not in skimComponentDict:
skimComponentDict[sampleName] = dataComponentDict[sampleName]
skimComponentDict[sampleName].componentList = ComponentList( [Component(sampleName,os.path.join(dir_path,fileName,rootFileName),treeName,inUFTier2,),] )
else:
skimComponentDict[sampleName].componentList.extend( [Component(sampleName,os.path.join(dir_path,fileName,rootFileName),treeName,inUFTier2,),] )
sumWeightHist = "Ana/sumWeights"
kappa = 0.0001
saveSumWeightTxt = False
fileNameTemplate = "HToZdZdTo4L_M125_MZd%s_eps2e-2_kap1e-4_TuneCP5_13TeV_madgraph_pythia8.root"
datasetName = "HToZdZd_M%s"
skimTreePath = "passedEvents"
mass_points = [4,5,6,7,8,9,10,15,20,25,30,35,40,45,50,55,60,]
# ____________________________________________________________________________________________________________________________________________ ||
sigSampleDict = {}
for m in mass_points:
inputFilePath = os.path.join(sigSkimTreeDir,fileNameTemplate%m) if m not in [4,5,] else os.path.join(sigSkimTreeDir2,fileNameTemplate%m)
tmpDataset = Dataset(
datasetName%m,
ComponentList(
[ Component(datasetName%m,inputFilePath,skimTreePath,inUFTier2=inUFTier2) ]
),
isMC = True,
isSignal = True,
xs = Higgs_prod_xs*kappa**2*HZdZd_xs_dict[m],
)
handleSumWeight(
tmpDataset,
system,
os.path.join(sigTreeDir,fileNameTemplate%m),
sumWeightHist,
True,
saveSumWeightTxt,
inputFilePath.replace(".root",".txt"),
)
sigSampleDict[m] = tmpDataset
from Utils.SumWeight import handleSumWeight
import os
bkgSkimTreeDir = "/cmsuf/data/store/user/t2/users/klo/Zprime/EXO-18-008/102X_MCProd_DarkZNTuple/"
bkgSkimTreeDir2 = bkgSkimTreeDir
bkgTreeDir = "/cmsuf/data/store/user/t2/users/rosedj1/Higgs/HZZ4l/NTuple/Run2/MC2018_M19_Mar12_4l_2018Jets_JER_bestCandLegacy/"
dataTreeDir = bkgSkimTreeDir
inUFTier2 = True
sumWeightHist = "Ana/sumWeights"
saveSumWeightTxt = False
# ____________________________________________________________________________________________________________________________________________ ||
# qqZZ
qqZZ_cmpList = ComponentList([
Component("qqZZTo4L",
bkgSkimTreeDir + "ZZTo4L_TuneCP5_13TeV_powheg_pythia8.root",
"passedEvents",
inUFTier2=inUFTier2),
])
qqZZTo4L = Dataset(
"qqZZTo4L",
qqZZ_cmpList,
isMC=True,
xs=1.256,
)
handleSumWeight(
qqZZTo4L,
system,
bkgTreeDir + "ZZTo4L_TuneCP5_13TeV_powheg_pythia8.root",
sumWeightHist,
True,
from Core.ComponentList import *
from Core.Dataset import Dataset
inUFTier2 = False
# ____________________________________________________________________________________________________________________________________________ ||
PedjaData_Run2016_cmpList = ComponentList(
[
Component("PedjaData_Run2016","/raid/raid7/lucien/Higgs/DarkZ-NTuple/20180924/PedjaInput_rootfiles_MC80X_2lskim_M17_Feb21/Data_ZX_Run2017-03Feb2017_slimmedZX_FRWeight.root","selectedEvents",inUFTier2=inUFTier2),
]
)
PedjaData_Run2016 = Dataset(
"PedjaData_Run2016",
PedjaData_Run2016_cmpList,
isMC = False,
)
from Skimmer.ZMuMuSkimmer import SkimTreeZMuMuSkimmer
from Analyzer.GasGainAnalyzer import SkimTreeGasGainPlotter, SkimTreePositionPlotter
from Analyzer.RunAnalyzer import SkimTreeRunPlotter
from EndModule.GasGainEndModule import SkimTreeGasGainEndModule, SkimTreePositionEndModule
from Config.Chamber import *
import os
inputDir = "/cmsuf/data/store/user/t2/users/klo/CSC/Skim/210304_Data_2017A_hadd/"
componentList = []
for f in os.listdir(inputDir):
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)
#dir_path = "/cms/data/store/user/t2/users/klo/HeppyTree/heppy_80X_RA5_Legacy/Sept18_v1/"
dir_path = "/raid/raid7/lucien/SUSY/RA5/HeppyTree/Sept18_v1/"
inUFTier2 = False
treeName = "tree"
componentDict = {}
if inUFTier2:
fileNames = [ n for n in listdir_uberftp(dir_path) if n.endswith(".root") ]
else:
fileNames = [ n for n in os.listdir(dir_path) if n.endswith(".root") ]
for fileName in fileNames:
if "SMS" in fileName: continue # New signal model ntuples in Oct18_v1
if "ext" not in fileName:
sampleName = fileName.replace(".root","")
else:
sampleName = fileName.split("_ext")[0]
tmpDataset = Dataset(
sampleName,
ComponentList(
[Component(sampleName,os.path.join(dir_path,fileName),treeName,inUFTier2,),]
),
isMC = True,
isSignal = "SMS" in fileName,
)
tmpDataset.setSumWeight(os.path.join(dir_path,fileName),"SumGenWeights",inUFTier2)
if sampleName not in componentDict:
componentDict[sampleName] = tmpDataset
else:
componentDict[sampleName].add(tmpDataset)
from Core.ComponentList import *
from Core.Dataset import Dataset
from Core.Utils.MakeComponent import makeComponents
sampleName = "TTHnobb_pow"
fileName = "treeProducerSusyRA5.root"
TreeDir = "/cms/data/store/user/t2/users/klo/HeppyTree/heppy_80X_RA5_Legacy/July18_v2_LeptonJetRecleaner/"
sumweight_path = "/cms/data/store/user/t2/users/klo/HeppyTree/heppy_80X_RA5_Legacy/July18_v1/"
inUFTier2 = True
filePath = os.path.join(sumweight_path, sampleName, fileName)
#cmp = makeComponents(sampleName, TreeDir, "Events", inUFTier2)
cmpList = ComponentList([
Component("TTHnobb_pow",
TreeDir + sampleName + "/" + "TTHnobb_pow_%s_SkimTree.root" % i,
"tree", inUFTier2) for i in range(0, 1)
])
TTHnobb_pow = Dataset(
"TTHnobb_pow",
cmpList,
isMC=True,
xs=1,
)
TTHnobb_pow.setSumWeight(filePath, "SumGenWeights", inUFTier2)
Samples = [TTHnobb_pow]
from Utils.System import system
from Utils.SumWeight import handleSumWeight
bkgTreeDirT2_Feb21 = "/cms/data/store/user/t2/users/klo/Higgs/HZZ4l/NTuple/Run2/MC80X_M17_2l_Feb21/"
bkgTreeDirT2_Aug10 = "/cms/data/store/user/t2/users/klo/Higgs/HZZ4l/NTuple/Run2/MC80X_M17_2lskim_Aug10/"
#bkgTreeDir = "/raid/raid7/kshi/Zprime/20190718/SkimTree_Run2016_MC/"
bkgTreeDir = "/raid/raid7/kshi/Zprime/20190729/SkimTree_Run2016_mem_controlregion_MC/"
inUFTier2 = False
saveSumWeightTxt = False
sumWeightHist = "Ana/sumWeights"
# ____________________________________________________________________________________________________________________________________________ ||
DYJetsToLL_M50_memCR_cmpList = ComponentList([
Component("DYJetsToLL_M50_memCR",
bkgTreeDir +
"DYJetsToLL_M-50_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8.root",
"passedEvents",
inUFTier2=inUFTier2),
])
DYJetsToLL_M50_memCR = Dataset(
"DYJetsToLL_M50_memCR",
DYJetsToLL_M50_memCR_cmpList,
isMC=True,
xs=6104,
)
handleSumWeight(
DYJetsToLL_M50_memCR,
system,
bkgTreeDirT2_Feb21 +
"DYJetsToLL_M-50_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8.root",
#bkgTreeDir+"DYJetsToLL_M-50_TuneCUETP8M1_13TeV-amcatnloFXFX-pythia8.root",
bkgTreeDirAug10 = "/cms/data/store/user/t2/users/klo/Higgs/HZZ4l/NTuple/Run2/MC80X_M17_2lskim_Aug10/"
bkgTreeDirLucien = "/cms/data/store/user/t2/users/klo/Higgs/HToZdZd/BkgMC_Run2016/"
dataTreeDir = bkgSkimTreeDir
inUFTier2 = False
sumWeightHist = "Ana/sumWeights"
saveSumWeightTxt = False
xsBoost = 100
epsilon = 0.02 # not used below!
sumWeightFromT2 = True
# ____________________________________________________________________________________________________________________________________________ ||
# Z+X
ZPlusX_cmpList = ComponentList([
Component(
"ZPlusX",
system.getStoragePath() +
"/lucien/Higgs/DarkZ-NTuple/20190218/SkimTree_HToZdZd_ZX_Run2016Data_m4l70_noZCandRatioCut/Data_Run2016-2l_noDuplicates_FRWeight.root",
"passedEvents", False)
])
ZPlusX = Dataset(
"ZPlusX",
ZPlusX_cmpList,
isMC=True,
skipWeight=True,
)
# ____________________________________________________________________________________________________________________________________________ ||
# Data2016
# NTuple list
data2016_cmpList = ComponentList([
Component("Data2016",
from Core.ComponentList import *
from Core.Dataset import Dataset
from Utils.System import system
from Utils.SumWeight import handleSumWeight
# ____________________________________________________________________________________________________________________________________________ ||
Data_Run2018_cmpList = ComponentList([
Component(
"Data_Run2016",
system.getStoragePath() +
"/lucien/Higgs/DarkZ-NTuple/20190402/SkimTree_DarkPhoton_WrongFC_Run2018Data_m4l70/Data_Run2018_noDuplicates_FRWeightFromVukasinWZRemoved.root",
"passedEvents",
inUFTier2=False),
])
Data_Run2018 = Dataset(
"WrongFC_Run2018",
Data_Run2018_cmpList,
isMC=False,
)
from Core.Utils.UFTier2Utils import *
import os
common_path = "/cms/data/store/user/t2/users/klo/HeppyTree/heppy_80X_RA5_Legacy/Data2016_July18_v1/"
fileName = "treeProducerSusyRA5.root"
inUFTier2 = True
treeName = "tree"
pdNames = [
"MuonEG",
"DoubleMuon",
"DoubleEG",
]
selection = lambda x: "Run2016B" not in x
#componentList = []
sampleDict = {}
sampleFileNames = listdir_uberftp(common_path,selection=lambda x: True)
for pdName in pdNames:
sampleNameList = [k for k in sampleFileNames if pdName in k and selection(k)]
cmp_list = []
for sampleName in sampleNameList:
cmp_list.append(Component(sampleName,os.path.join(common_path,sampleName,fileName),treeName,inUFTier2,))
cmpList = ComponentList(cmp_list)
tmpDataset = Dataset(pdName,cmpList,isMC=False)
if pdName not in sampleDict:
sampleDict[pdName] = tmpDataset
else:
sampleDict[pdName].add(tmpDataset)
componentList = sampleDict.values()
import os
from Core.ComponentList import *
from Core.Dataset import Dataset
fileName = "SkimTree.root"
#common_path = "/raid/raid7/lucien/SUSY/RPV/SkimTree/ZMuMu/2018-06-19/Data_ZMuMuSelection_v1/"
common_path = "/raid/raid7/kshi/SUSY/RPV/SkimTree/data/TTbar/"
inUFTier2 = False
sampleNames = [
n for n in os.listdir(common_path)
if os.path.isdir(os.path.join(common_path, n)) and "Muon" in n
]
allDataSamples = []
for sampleName in sampleNames:
tmpList = ComponentList([
Component(sampleName,
"/".join([common_path, sampleName, fileName]),
"Events",
inUFTier2,
maxEvents=-1)
], )
tmpDataset = Dataset(
sampleName,
tmpList,
isMC=False,
json=os.environ['BASE_PATH'] +
"/DataMC/JSON/13TeV/Run2016/Cert_271036-284044_13TeV_23Sep2016ReReco_Collisions16_JSON.txt",
)
allDataSamples.append(tmpDataset)
bkgSkimTreeDir = "/raid/raid7/lucien/Higgs/DarkZ-NTuple/20181116/SkimTree_DarkPhoton_Run2017Data_m4l70/"
bkgSkimTreeDir2 = bkgSkimTreeDir
bkgTreeDir = "/cms/data/store/user/t2/users/klo/Higgs/DarkZ/NTuples/BkgMC_Run2017/"
dataTreeDir = bkgSkimTreeDir
zxSkimTreeDir = "/raid/raid7/lucien/Higgs/DarkZ-NTuple/20181116/SkimTree_DarkPhoton_ZX_Run2017Data_m4l70/"
inUFTier2 = False
sumWeightHist = "Ana/sumWeights"
sumWeightFromT2 = True
# ____________________________________________________________________________________________________________________________________________ ||
# Z+X
ZPlusX_cmpList = ComponentList(
[
Component("ZPlusX",
zxSkimTreeDir+"Data_Run2017-17Nov2017_noDuplicates_FRWeight.root",
"passedEvents",False)
]
)
ZPlusX = Dataset(
"ZPlusX",
ZPlusX_cmpList,
isMC = True,
skipWeight = True,
)
# ____________________________________________________________________________________________________________________________________________ ||
# Data2017
data2017_cmpList = ComponentList(
[
Component("Data2017",dataTreeDir+"/Data_Run2017-17Nov2017_4l_noDuplicates.root","passedEvents",inUFTier2=inUFTier2),
bkgSkimTreeDir = system.getStoragePath() + "/" + os.environ[
"USER"] + "/Higgs/DarkZ-NTuple/20190218/SkimTree_HToZdZd_Run2017Data_m4l70_noZCandRatioCut/"
bkgSkimTreeDir2 = bkgSkimTreeDir
bkgTreeDir = "/cms/data/store/user/t2/users/klo/Higgs/HToZdZd/94X_MCProd_191127/"
inUFTier2 = False
sumWeightHist = "Ana/sumWeights"
sumWeightFromT2 = True
saveSumWeightTxt = False
# ____________________________________________________________________________________________________________________________________________ ||
# ttZ
ttZ_cmpList = ComponentList([
Component("ttZ",
bkgSkimTreeDir +
"ttZJets_TuneCP5_13TeV_madgraphMLM_pythia8.root",
"passedEvents",
inUFTier2=inUFTier2),
])
ttZ = Dataset(
"ttZ",
ttZ_cmpList,
isMC=True,
xs=0.6529,
)
handleSumWeight(
ttZ,
system,
bkgTreeDir + "ttZJets_TuneCP5_13TeV_madgraphMLM_pythia8.root",
sumWeightHist,
from Core.ComponentList import *
from Core.Dataset import Dataset
from Utils.System import system
from Utils.SumWeight import handleSumWeight
import os
dataSkimTreeDir = "/cmsuf/data/store/user/t2/users/klo/Zprime/EXO-18-008/80X_Data_DarkZNTuple/"
inUFTier2 = True
# ____________________________________________________________________________________________________________________________________________ ||
# Data2016
data2016_cmpList = ComponentList([
Component("Data2016",
dataSkimTreeDir + "/Data_Run2016-03Feb2017_4l.root",
"passedEvents",
inUFTier2=inUFTier2),
])
data2016 = Dataset(
"Data2016",
data2016_cmpList,
isMC=False,
)
# ____________________________________________________________________________________________________________________________________________ ||
dataSamples = [
data2016,
]
"USER"] + "/Higgs/DarkZ-NTuple/20190408/SkimTree_HToZdZd_Run2018Data_m4l70/"
bkgSkimTreeDir2 = bkgSkimTreeDir
bkgTreeDir = "/cmsuf/data/store/user/t2/users/klo/Higgs/HZZ4l/NTuple/Run2/MC2018_M19_Feb19_fixGENjet_bestCandLegacy/"
dataTreeDir = bkgSkimTreeDir
zxSkimTreeDir = system.getStoragePath() + "/" + os.environ[
"USER"] + "/Higgs/DarkZ-NTuple/20190408/SkimTree_HToZdZd_Run2018Data_m4l70/"
inUFTier2 = False
sumWeightHist = "Ana/sumWeights"
sumWeightFromT2 = False
saveSumWeightTxt = False
# ____________________________________________________________________________________________________________________________________________ ||
# Z+X
ZPlusX_cmpList = ComponentList([
Component(
"ZPlusX", zxSkimTreeDir +
"Data_Run2018_noDuplicates_FRWeightFromVukasinWZRemoved.root",
"passedEvents", False)
])
ZPlusX = Dataset(
"ZPlusX",
ZPlusX_cmpList,
isMC=True,
skipWeight=True,
)
# ____________________________________________________________________________________________________________________________________________ ||
# Data2017
data2018_cmpList = ComponentList([
Component("Data2018",
dataTreeDir + "Data_Run2018_UFHZZNTuple_noDuplicates.root",
"passedEvents",
"SingleElectron",
"Tau",
]
#componentList = []
sampleDict = {}
sampleFileNames = listdir_uberftp(common_path,selection=lambda x: True)
for sampleName in sampleFileNames:
if "TT_pow" in sampleName: continue
if any([dataSample in sampleName for dataSample in dataSamples]): continue
parentName = sampleName
for postfix in postfixs:
if postfix in sampleName: parentName = parentName.replace("_"+postfix,"")
filePath = os.path.join(common_path,sampleName,fileName)
cmpList = ComponentList(
[Component(parentName,filePath,treeName,inUFTier2,)]
)
tmpDataset = Dataset(parentName,cmpList,)
tmpDataset.setSumWeight(filePath,sumWeight,inUFTier2)
#componentList.append(tmpDataset)
if parentName not in sampleDict:
sampleDict[parentName] = tmpDataset
else:
sampleDict[parentName].add(tmpDataset)
# Exceptionally handling TT Powheg sample
TT_pow_comps = makeComponents("TT_pow","/cms/data/store/user/klo/HeppyTree/heppy_80X_RA5_Legacy/July18_v1/TT_pow/180712_162135/0000/",treeName,inUFTier2)
TT_pow = Dataset("TT_pow",ComponentList([]),xs=1.)
TT_pow.sumw = 0.
for comp in TT_pow_comps:
temp_comp_list = ComponentList([comp])