ttbar_categories = root2pandas.EventCategories() ttbar_categories.addCategory( "ttbb", selection="(GenEvt_I_TTPlusBB == 3 and GenEvt_I_TTPlusCC == 0)") ttbar_categories.addCategory( "tt2b", selection="(GenEvt_I_TTPlusBB == 2 and GenEvt_I_TTPlusCC == 0)") ttbar_categories.addCategory( "ttb", selection="(GenEvt_I_TTPlusBB == 1 and GenEvt_I_TTPlusCC == 0)") ttbar_categories.addCategory( "ttlf", selection="(GenEvt_I_TTPlusBB == 0 and GenEvt_I_TTPlusCC == 0)") ttbar_categories.addCategory( "ttcc", selection="(GenEvt_I_TTPlusBB == 0 and GenEvt_I_TTPlusCC == 1)") # initialize dataset class dataset = root2pandas.Dataset( outputdir="/nfs/dust/cms/user/vdlinden/DNNInputFiles/DNN_ttH_2017/", naming="_dnn_newJEC", addCNNmap=False, addMEM=True) # add base event selection dataset.addBaseSelection(base_selection) ntuplesPath = "/nfs/dust/cms/user/kelmorab/ttH_2018/ntuples_forDNN_v4/" memPath = "/nfs/dust/cms/user/vdlinden/MEM_2017/" # add samples to dataset dataset.addSample( sampleName="ttHbb", ntuples=ntuplesPath + "/ttHTobb_M125_TuneCP5_13TeV-powheg-pythia8_new_pmx/*nominal*.root", categories=ttH_categories,
ttbb_categories = root2pandas.EventCategories() ttbb_categories.addCategory("ttbb", selection = "(GenEvt_I_TTPlusBB == 3 and GenEvt_I_TTPlusCC == 0)") ttbb_categories.addCategory("tt2b", selection = "(GenEvt_I_TTPlusBB == 2 and GenEvt_I_TTPlusCC == 0)") ttbb_categories.addCategory("ttb", selection = "(GenEvt_I_TTPlusBB == 1 and GenEvt_I_TTPlusCC == 0)") ttbb_categories.addCategory("ttb_bb", selection = "(GenEvt_I_TTPlusBB == 3 and GenEvt_I_TTPlusCC == 0) or \ (GenEvt_I_TTPlusBB == 1 and GenEvt_I_TTPlusCC == 0)") ttbb_categories.addCategory("tthf", selection = "(GenEvt_I_TTPlusBB == 3 and GenEvt_I_TTPlusCC == 0) or \ (GenEvt_I_TTPlusBB == 1 and GenEvt_I_TTPlusCC == 0) or \ (GenEvt_I_TTPlusBB == 2 and GenEvt_I_TTPlusCC == 0)") # initialize dataset class dataset = root2pandas.Dataset( outputdir = outputdir, naming = options.Name, addMEM = False,#options.MEM, maxEntries = options.maxEntries) # add base event selection dataset.addBaseSelection(base_selection) ntuplesPath = "/nfs/dust/cms/user/swieland/ttH_legacy/ntuple/2017/" memPath = "/nfs/dust/cms/user/mwassmer/ttH_2018/MEMs_v2/" #outdated # add samples to dataset dataset.addSample( sampleName = "ttHbb", ntuples = ntuplesPath+"/ttHTobb_M125_TuneCP5_13TeV-powheg-pythia8_new_pmx/*nominal*.root",
ttbar_categories = root2pandas.EventCategories() ttbar_categories.addCategory( "ttbb", selection="(GenEvt_I_TTPlusBB == 3 and GenEvt_I_TTPlusCC == 0)") ttbar_categories.addCategory( "tt2b", selection="(GenEvt_I_TTPlusBB == 2 and GenEvt_I_TTPlusCC == 0)") ttbar_categories.addCategory( "ttb", selection="(GenEvt_I_TTPlusBB == 1 and GenEvt_I_TTPlusCC == 0)") ttbar_categories.addCategory( "ttlf", selection="(GenEvt_I_TTPlusBB == 0 and GenEvt_I_TTPlusCC == 0)") ttbar_categories.addCategory( "ttcc", selection="(GenEvt_I_TTPlusBB == 0 and GenEvt_I_TTPlusCC == 1)") # initialize dataset class dataset = root2pandas.Dataset( outputdir="/nfs/dust/cms/user/vdlinden/DNNInputFiles/ttZ_DNN_v2/", naming="dnn", addCNNmap=False, addMEM=False) # add base event selection dataset.addBaseSelection(base_selection) ntuplesPath = "/nfs/dust/cms/user/vdlinden/ttH_2018/ntuples/ntuples_v5_forDNN/" ttZntuples = "/nfs/dust/cms/user/vdlinden/ttZ_2019/ntuples_v1/" memPath = "/nfs/dust/cms/user/vdlinden/MEM_2017/" # add samples to dataset dataset.addSample( sampleName="ttHbb", ntuples=ntuplesPath + "/ttHTobb_M125_TuneCP5_13TeV-powheg-pythia8_new_pmx/*nominal*.root",
if not os.path.isabs(options.outputDir): outputdir = basedir + "/workdir/" + options.outputDir elif os.path.exists(options.outputDir) or os.path.exists( os.path.dirname(options.outputDir)): outputdir = options.outputDir else: sys.exit("ERROR: Output Directory does not exist!") # define a base event selection which is applied for all Samples # initialize dataset class dataset = root2pandas.Dataset( outputdir=outputdir, naming=options.Name, addMEM=options.MEM, maxEntries=options.maxEntries, tree=options.treeName, varName_Run="runNumber", varName_LumiBlock="lumiBlock", varName_Event="eventNumber", ) ntuplesPath = "/nfs/dust/cms/user/missirol/sandbox/ttHbb/output_190914_DeepJet/2016/exe1/selectionRoot_reco_liteTreeTTH/Nominal" ttH_categories = root2pandas.EventCategories() ttH_categories.addCategory("ttH", selection=None) ttbar_categories = root2pandas.EventCategories() ttbar_categories.addCategory("ttbar", selection=None) ttbar_bb_categories = root2pandas.EventCategories() ttbar_bb_categories.addCategory("ttbb", selection=None)
base_selection = "(" + base + " and (" + single_mu_sel + " or " + single_el_sel + "))" # define output classes ttH_categories = root2pandas.EventCategories() ttH_categories.addCategory("ttH", selection=None) ntuplespath = "/nfs/dust/cms/user/vdlinden/legacyTTH/ntuples/ntuples_ttZ/" friendTrees = { "RecoDNN_Z": "/nfs/dust/cms/user/vdlinden/legacyTTH/karim/workdir/ntuples/RecoDNN_Z" } # initialize dataset class dataset = root2pandas.Dataset(outputdir=outputdir, naming=options.Name, addMEM=options.MEM, maxEntries=options.maxEntries, ttbarReco=options.ttbarReco, friendTrees=friendTrees, ncores=options.ncores) # add base event selection dataset.addBaseSelection(base_selection) # add samples to dataset dataset.addSample( sampleName="ttH", ntuples=ntuplespath + "/ttHTobb_M125_TuneCP5_13TeV-powheg-pythia8_new_pmx/*nominal*.root", categories=ttH_categories, lumiWeight=41.5, )
# define a base event selection which is applied for all Samples base_selection = "(\ (N_LooseMuons == 0 and N_TightElectrons == 1) \ or \ (N_LooseElectrons == 0 and N_TightMuons == 1) \ )" categories = root2pandas.EventCategories() categories.addCategory("ttbar", selection = None) # initialize dataset class dataset = root2pandas.Dataset( outputdir = "/nfs/dust/cms/user/vdlinden/DNNInputFiles/ttbarMatcher/", naming = "input", addCNNmap = False, addMEM = False, maxEntries = 200000) # add base event selection dataset.addBaseSelection(base_selection) # add samples to dataset dataset.addSample( sampleName = "ttbar", ntuples = "/nfs/dust/cms/user/vdlinden/ttH_2018/ntuples/ntuples_ttbarMatching/TTToSemiLeptonic_TuneCP5_13TeV-powheg-pythia8_new_pmx/*nominal*.root", categories = categories) # initialize variable list dataset.addAllVariablesNoIndex()