示例#1
0
        name='yield',
        texX='yield',
        texY='Number of Events',
        attribute=lambda event, sample: 0.5 + index,
        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
示例#2
0
#sample = Sample.combine( "ttg", [ttg160l, ttg161l, ttg162l, ttg170l, ttg171l, ttg172l, ttg180l, ttg181l, ttg182l] )
#sample = Sample.combine( "ttg", [ttg160l, ttg161l, ttg162l] )
sample = Sample.combine( "ttg", [ttg161l] )

def convertUnits( coll ):
    for p in coll:
        if abs(p['pdgId'])==11 and isinstance( p['lostHits'], basestring ): p['lostHits']    = ord( p['lostHits'] )
        if abs(p['pdgId'])==13 and isinstance( p['pfIsoId'], basestring ):  p['pfIsoId']     = ord( p['pfIsoId'] )
        if isinstance( p['genPartFlav'], basestring ):             p['genPartFlav'] = ord( p['genPartFlav'] )

selectionString = "overlapRemoval==1"

NanoVars = NanoVariables( 2016 )
readGenVarString      = NanoVars.getVariableString(   "Gen",      postprocessed=True, data=False )
readGenVarList        = NanoVars.getVariableNameList( "Gen",      postprocessed=True, data=False )
readPhotonVarList     = NanoVars.getVariableNameList( "Photon",   postprocessed=True, data=False, skipSyst=True)
readPhotonVarString   = NanoVars.getVariableString(   "Photon",   postprocessed=True, data=False, skipSyst=True)

genLeptonSel_CMSUnfold = genLeptonSelector( "CMSUnfolding" )
genPhotonSel_CMSUnfold = genPhotonSelector( "CMSUnfolding" )

recoPhotonSel_medium    = photonSelector( 'medium', year=2016 )

read_variables = [ TreeVariable.fromString('weight/F'),TreeVariable.fromString('nGenPart/I'),
                   VectorTreeVariable.fromString('GenPart[%s]'%readGenVarString, nMax = 1000) ] # all needed for genMatching
read_variables += [ TreeVariable.fromString('nPhoton/I'),
                    VectorTreeVariable.fromString('Photon[%s]'%readPhotonVarString) ]

# Define a reader
r = sample.treeReader( \
示例#3
0
                          legend=[
                              (0.2, 0.87 - 0.04 * sum(map(len, plot.histos)),
                               0.8, 0.87), 1
                          ],
                          drawObjects=drawObjects(True, lumi_scale),
                          copyIndexPHP=True)


# get nano variable lists
NanoVars = NanoVariables(args.year)
jetVarString = NanoVars.getVariableString("Jet",
                                          postprocessed=True,
                                          data=False,
                                          plot=True)
jetVariableNames = NanoVars.getVariableNameList("Jet",
                                                postprocessed=True,
                                                data=False,
                                                plot=True)
jetVarList = NanoVars.getVariableNameList("Jet",
                                          postprocessed=True,
                                          data=False,
                                          plot=True)
jetVariables = NanoVars.getVariables("Jet",
                                     postprocessed=True,
                                     data=False,
                                     plot=True)
bJetVariables = NanoVars.getVariables("BJet",
                                      postprocessed=True,
                                      data=False,
                                      plot=True)
leptonVariables = NanoVars.getVariables("Lepton",
                                        postprocessed=True,
示例#4
0
    elif options.year == 2018: lumi = 59.74

# get nano variable lists
NanoVars = NanoVariables( options.year )
#VarString ... "var1/type,var2/type"
#Variables ... ["var1/type","var2/type"]
#VarList   ... ["var1", "var2"]

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  )
示例#5
0
        name='yield',
        texX='yield',
        texY='Number of Events',
        attribute=lambda event, sample: event.nElectronGood,
        binning=[3, 0, 3],
    )


# get nano variable lists
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)
leptonVariables = NanoVars.getVariables("Lepton",
                                        postprocessed=True,
                                        data=(not args.noData),
                                        plot=True)
leptonVarString = NanoVars.getVariableString("Lepton",
                                             postprocessed=True,
                                             data=(not args.noData),
                                             plot=True)
photonVariables = NanoVars.getVariables("Photon",
                                        postprocessed=True,