Python EventSelection示例

示例#1

文件： job_efficiency_v1_data18_13TeV_EGAM2_probes_lhmedium.py 项目： jodafons/prometheus

acc = EventATLAS( "EventATLASLoop",
                  inputFiles = args.inputFiles, 
                  treePath= '*/HLT/Physval/Egamma/fakes',
                  #treePath= '*/HLT/Egamma/Egamma/probes',
                  #treePath= '*/HLT/Egamma/Egamma/fakes' if args.doEgam7 else '*/HLT/Egamma/Egamma/probes',
                  dataframe = DataframeEnum.Electron_v1, 
                  outputFile = args.outputFile,
                  level = LoggingLevel.INFO
                  )



from EventSelectionTool import EventSelection, SelectionType, EtCutType

evt = EventSelection('EventSelection', dataframe = DataframeEnum.Electron_v1)
evt.setCutValue( SelectionType.SelectionOnlineWithRings )
#pidname = 'MediumLLH_DataDriven_Rel21_Run2_2018'
#pidname = 'el_lhmedium'
pidname = '!el_lhvloose'
evt.setCutValue( SelectionType.SelectionPID, pidname ) 
evt.setCutValue( EtCutType.L2CaloAbove , 3)
evt.setCutValue( EtCutType.L2CaloBelow , 15)

ToolSvc += evt


from TrigEgammaEmulationTool import Chain, Group, TDT

triggerList = [
                # e17 lhvloose

示例#2

args = parser.parse_args()

acc = EventATLAS(
    "EventATLASLoop",
    inputFiles=args.inputFiles,
    #treePath= '*/HLT/Physval/Egamma/fakes' if args.jf17 else '*/HLT/Physval/Egamma/probes',
    treePath='*/HLT/Egamma/Egamma/fakes'
    if args.jf17 else '*/HLT/Egamma/Egamma/probes',
    dataframe=DataframeEnum.PhysVal_v2,
    #outputFile = args.outputFile,
    outputFile='dummy.root',
    level=LoggingLevel.INFO)

from EventSelectionTool import EventSelection, SelectionType, EtCutType

evt = EventSelection('EventSelection')
evt.setCutValue(SelectionType.SelectionOnlineWithRings)

# Do not change this!
if args.doEgam7:
    #pidname = '!VeryLooseLLH_DataDriven_Rel21_Run2_2018'
    pidname = '!el_lhvloose'
elif args.jf17:
    pidname = '!mc_isTruthElectronAny'
else:
    #pidname = 'MediumLLH_DataDriven_Rel21_Run2_2018'
    pidname = 'el_lhmedium'
    #pidname = 'el_lhtight'

evt.setCutValue(SelectionType.SelectionPID, pidname)
evt.setCutValue(EtCutType.L2CaloAbove, 15)

示例#3

acc = EventATLAS(
    "EventATLASLoop",
    inputFiles=args.inputFiles,
    treePath='*/HLT/PhysVal/Egamma/fakes'
    if args.doFakes else '*/HLT/PhysVal/Egamma/photons',
    dataframe=DataframeEnum.Electron_v1
    if signature == 'electron' else DataframeEnum.Photon_v1,
    outputFile=args.outputFile,
    # outputFile = 'dummy.root',
    level=LoggingLevel.INFO,
)

from EventSelectionTool import EventSelection, SelectionType, EtCutType

evt = EventSelection('EventSelection')
# evt.setCutValue( SelectionType.SelectionOnlineWithRings )

# Do not change this!
if args.doEgam7:
    #pidname = '!VeryLooseLLH_DataDriven_Rel21_Run2_2018'
    pidname = '!el_lhvloose'
elif args.doZee or args.doJpsi:
    #pidname = 'MediumLLH_DataDriven_Rel21_Run2_2018'
    pidname = 'el_lhmedium'
elif args.doZrad:
    pidname = 'ph_medium'
elif args.doFakes:
    pidname = '!ph_loose'
else:
    pidname = 'el_lhmedium'

示例#4

from Gaugi.messenger import LoggingLevel
from Gaugi import ToolSvc, ToolMgr

datapath = '/afs/cern.ch/work/j/jodafons/public/data_samples/PhysVal/user.jodafons.data17_13TeV.00329835.physics_Main.deriv.DAOD_EGAM1.f843_m1824_p3336.Physval.GRL_v97.r7000_GLOBAL'

ToolMgr += EventATLASLoop("EventATLASLoop",
                          inputFiles=datapath,
                          treePath='*/HLT/Physval/Egamma/probes',
                          nov=1000,
                          dataframe=DataframeEnum.Electron_v1,
                          outputFile='test_output.root',
                          level=LoggingLevel.INFO)

from EventSelectionTool import EventSelection, SelectionType, EtCutType

evt = EventSelection('EventSelection', dataframe=DataframeEnum.Electron_v1)
evt.setCutValue(SelectionType.SelectionOnlineWithRings)
evt.setCutValue(SelectionType.SelectionPID, "el_lhtight")
evt.setCutValue(EtCutType.L2CaloAbove, 15)
ToolSvc += evt

from EmulationTools import EmulationTool
ToolSvc += EmulationTool("EgammaEmulation")

from QuadrantTools import QuadrantTool

alg = QuadrantTool("Quadrant", dataframe=DataframeEnum.Electron_v1)
alg.doTrigger = True
alg.add_quadrant(
    'HLT_e28_lhtight_nod0_noringer_ivarloose',
    "TDT__HLT__e28_lhtight_nod0_noringer_ivarloose",  # T2Calo

示例#5

文件： job_efficiency_mc16_13TeV_DP_probes_medium.py 项目： jodafons/prometheus

acc = EventATLAS( "EventATLASLoop",
                  inputFiles = args.inputFiles, 
                  treePath= '*/HLT/PhysVal/Egamma/photons',
                  dataframe = DataframeEnum.Photon_v1, 
                  outputFile = args.outputFile,
                  level = LoggingLevel.INFO
                  )



from EventSelectionTool import EventSelection, SelectionType, EtCutType

evt = EventSelection('EventSelection', dataframe = DataframeEnum.Photon_v1)
# evt.setCutValue( SelectionType.SelectionOnlineWithRings )
#pidname = 'MediumLLH_DataDriven_Rel21_Run2_2018'
pidname = 'ph_tight'
evt.setCutValue( SelectionType.SelectionPID, pidname ) 
evt.setCutValue( EtCutType.L2CaloAbove , 10)


ToolSvc += evt


from TrigEgammaEmulationTool import Chain, Group, TDT

triggerList = [
                Group( TDT( "HLT_g10_etcut","HLT__TDT__g10_etcut"), 'ph_tight', 10 ),
                # Group( Chain( "HLT__TDT__g10_etcut__medium","L1_EM3","HLT_g10_medium_noringer"), 'ph_medium', 10 ),

示例#6

文件： job_collector.py 项目： ringer-atlas/prometheus

    treePath='*/HLT/Physval/Egamma/fakes'
    if args.doEgam7 else '*/HLT/PhysVal/Egamma/photons',
    # treePath= '*/HLT/Egamma/Egamma/fakes' if args.doEgam7 else '*/HLT/Egamma/Egamma/probes',
    # dataframe = DataframeEnum.PhysVal_v2,
    dataframe=DataframeEnum.Electron_v1
    if signature == 'electron' else DataframeEnum.Photon_v1,
    outputFile=args.outputFile,
    # outputFile = 'dummy.root',
    level=LoggingLevel.INFO,
)

from EventSelectionTool import EventSelection, SelectionType, EtCutType

evt = EventSelection(
    'EventSelection',
    dataframe=DataframeEnum.Electron_v1
    if signature == 'electron' else DataframeEnum.Photon_v1,
)
evt.setCutValue(SelectionType.SelectionOnlineWithRings)

# Do not change this!
if args.doEgam7:
    #pidname = '!VeryLooseLLH_DataDriven_Rel21_Run2_2018'
    pidname = '!el_lhvloose'
elif args.doZee or args.doJpsi:
    #pidname = 'MediumLLH_DataDriven_Rel21_Run2_2018'
    pidname = 'el_lhmedium'
elif args.doZrad:
    pidname = 'ph_medium'
else:
    pidname = 'el_lhmedium'

示例#7

args = parser.parse_args()


#
# setup event
#
acc = EventATLAS( "EventATLASLoop",
                  inputFiles = args.inputFiles,
                  treePath= '*/HLT/Physval/Egamma/probes',
                  dataframe = DataframeEnum.Electron_v1,
                  outputFile = args.outputFile,
                  level = LoggingLevel.INFO
                  )

from EventSelectionTool import EventSelection, SelectionType, EtCutType
event_cuts = EventSelection('EventSelection')
event_cuts.setCutValue( SelectionType.SelectionOnlineWithRings )
event_cuts.setCutValue( SelectionType.SelectionPID, 'el_lhmedium' )
event_cuts.setCutValue( EtCutType.L2CaloAbove , 15)
ToolSvc += event_cuts



if args.test == 'q221':

  from TrigEgammaEmulationTool import Chain, Group, TDT

  triggerList = [
                # e28 lhtight
                Group( TDT( "TDT_e28_lhtight_nod0_noringer_ivarloose" , "HLT_e28_lhtight_nod0_noringer_ivarloose"   ), "el_lhtight", 28 ),
                Group( Chain( "EMU_e28_lhtight_nod0_noringer_ivarloose"       , "L1_EM24VHI", "HLT_e28_lhtight_nod0_noringer_ivarloose"   ), "el_lhtight", 28 ),