Ejemplo n.º 1
0
    sample = TTGSemiLep_16
    #sample.files = sample.files[:1]

    # add needed variables
    NanoVars = NanoVariables(args.year)
    genVarString = NanoVars.getVariableString("Gen",
                                              postprocessed=True,
                                              data=False,
                                              plot=True)
    photonVarString = NanoVars.getVariableString("Photon",
                                                 postprocessed=True,
                                                 data=True,
                                                 plot=True)
    photonVariables = NanoVars.getVariables("Photon",
                                            postprocessed=True,
                                            data=True,
                                            plot=True)
    variables = [
        TreeVariable.fromString("event/l"),
        TreeVariable.fromString("weight/F"),
        TreeVariable.fromString('run/i'),
        TreeVariable.fromString("luminosityBlock/i"),
        VectorTreeVariable.fromString('Photon[%s]' % photonVarString,
                                      nMax=100),
        VectorTreeVariable.fromString('GenPhoton[%s]' % genVarString,
                                      nMax=100),
        VectorTreeVariable.fromString('GenPart[%s]' % genVarString, nMax=1000),
        TreeVariable.fromString('nPhotonGood/I'),
        TreeVariable.fromString('nJetGood/I'),
        TreeVariable.fromString('nBTagGood/I'),
        TreeVariable.fromString('nLeptonTight/I'),
Ejemplo n.º 2
0
readGenVarString      = NanoVars.getVariableString(   "Gen",      postprocessed=False, data=sample.isData )
readGenJetVarString   = NanoVars.getVariableString(   "GenJet",   postprocessed=False, data=sample.isData )
readJetVarString      = NanoVars.getVariableString(   "Jet",      postprocessed=False, data=sample.isData, skipSyst=True )
readElectronVarString = NanoVars.getVariableString(   "Electron", postprocessed=False, data=sample.isData )
readMuonVarString     = NanoVars.getVariableString(   "Muon",     postprocessed=False, data=sample.isData )
readPhotonVarString   = NanoVars.getVariableString(   "Photon",   postprocessed=False, data=sample.isData )

readGenVarList        = NanoVars.getVariableNameList( "Gen",      postprocessed=False, data=sample.isData )
readGenJetVarList     = NanoVars.getVariableNameList( "GenJet",   postprocessed=False, data=sample.isData )
readJetVarList        = NanoVars.getVariableNameList( "Jet",      postprocessed=False, data=sample.isData, skipSyst=True  )
readElectronVarList   = NanoVars.getVariableNameList( "Electron", postprocessed=False, data=sample.isData )
readMuonVarList       = NanoVars.getVariableNameList( "Muon",     postprocessed=False, data=sample.isData )
readPhotonVarList     = NanoVars.getVariableNameList( "Photon",   postprocessed=False, data=sample.isData )

readMuonVariables   = NanoVars.getVariables(        "Muon",   postprocessed=False, data=sample.isData )

writeGenVarString     = NanoVars.getVariableString(   "Gen",      postprocessed=True,  data=sample.isData )
writeGenJetVarString  = NanoVars.getVariableString(   "GenJet",   postprocessed=True,  data=sample.isData )
writeJetVarString     = NanoVars.getVariableString(   "Jet",      postprocessed=True,  data=sample.isData, skipSyst=True  )
writeMuonVarString  = NanoVars.getVariableString(   "Muon",   postprocessed=True,  data=sample.isData, skipSyst=True  )
writePhotonVarString  = NanoVars.getVariableString(   "Photon",   postprocessed=True,  data=sample.isData, skipSyst=True  )

writeJetVarList       = NanoVars.getVariableNameList( "Jet",      postprocessed=True,  data=sample.isData, skipSyst=True  )
writeBJetVarList      = NanoVars.getVariableNameList( "BJet",     postprocessed=True,  data=sample.isData, skipSyst=True  )
writeGenVarList       = NanoVars.getVariableNameList( "Gen",      postprocessed=True,  data=sample.isData )
writeGenJetVarList    = NanoVars.getVariableNameList( "GenJet",   postprocessed=True,  data=sample.isData )
writeMuonVarList    = NanoVars.getVariableNameList( "Muon",   postprocessed=True,  data=sample.isData, skipSyst=True  )
writePhotonVarList    = NanoVars.getVariableNameList( "Photon",   postprocessed=True,  data=sample.isData, skipSyst=True  )

writeGenVariables     = NanoVars.getVariables( "Gen",      postprocessed=True,  data=sample.isData )
Ejemplo n.º 3
0
        binning=[3, 0, 3],
    )


# get nano variable lists
NanoVars = NanoVariables(2017)
jetVarString = NanoVars.getVariableString("Jet",
                                          postprocessed=True,
                                          data=True,
                                          plot=True)
jetVariableNames = NanoVars.getVariableNameList("Jet",
                                                postprocessed=True,
                                                data=True,
                                                plot=True)
bJetVariables = NanoVars.getVariables("BJet",
                                      postprocessed=True,
                                      data=True,
                                      plot=True)
leptonVariables = NanoVars.getVariables("Lepton",
                                        postprocessed=True,
                                        data=True,
                                        plot=True)
photonVariables = NanoVars.getVariables("Photon",
                                        postprocessed=True,
                                        data=True,
                                        plot=True)

# Read variables and sequences
read_variables = [
    "weight/F",  #"ref_weight/F",
    "PV_npvs/I",
    "PV_npvsGood/I",
Ejemplo n.º 4
0
                    copyIndexPHP=True,
                )


NanoVars = NanoVariables(args.year)

jetVarString = NanoVars.getVariableString("Jet",
                                          postprocessed=True,
                                          data=(not args.noData),
                                          plot=True)
jetVariableNames = NanoVars.getVariableNameList("Jet",
                                                postprocessed=True,
                                                data=(not args.noData),
                                                plot=True)
bJetVariables = NanoVars.getVariables("BJet",
                                      postprocessed=True,
                                      data=(not args.noData),
                                      plot=True)
leptonVarString = NanoVars.getVariableString("Lepton",
                                             postprocessed=True,
                                             data=(not args.noData),
                                             plot=True)
leptonVariables = NanoVars.getVariables("Lepton",
                                        postprocessed=True,
                                        data=(not args.noData),
                                        plot=True)
leptonVarList = NanoVars.getVariableNameList("Lepton",
                                             postprocessed=True,
                                             data=(not args.noData),
                                             plot=True)
photonVariables = NanoVars.getVariables("Photon",
                                        postprocessed=True,