PhysDESDM_DV.py

Go to the documentation of this file.

# Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
#
#
# ##########################################################################################
# Electron/photon/muon augmentation
# Use the derivation framework tools directly
# ##########################################################################################
 
# Cache the container names (used throughout)
jetContainer = primRPVLLDESDM.DV_containerFlags.jetCollectionName
METContainer= primRPVLLDESDM.DV_containerFlags.METCollectionName
photonContainer = primRPVLLDESDM.DV_containerFlags.photonCollectionName
muonContainer = primRPVLLDESDM.DV_containerFlags.muonCollectionName
electronContainer = primRPVLLDESDM.DV_containerFlags.electronCollectionName
 
 
# ##########################################################################################
# Selection algorithms setup
# ##########################################################################################
 
def DVSelectionString(flags, container):
    """Construct object selection string based on the flags provided (from DVFlags)
    """
 
    cutList = []
    try:  
        cutList.append('%s.pt > %s'%(container,flags.cutEtMin))
    except:
        pass
    try:
        cutList.append('abs(%s.eta) < %s'%(container,flags.cutEtaMax))
    except:
        pass
    try:   
        cutList.append("DV%s%s"%(container,flags.cutIsEM))
    except:
        pass
    cutString = 'count(' + ' && '.join(cutList) + ') > %i'%(flags.nPassed-1)
    return cutString
 
from DerivationFrameworkTools.DerivationFrameworkToolsConf import DerivationFramework__xAODStringSkimmingTool as skimtool
from DerivationFrameworkCore.DerivationFrameworkCoreConf import DerivationFramework__DerivationKernel as kernel
 
def DVTriggerSelectionString(flags):
    cutString=""
    if flags.triggers.__len__() >=1:
        cutString+=flags.triggers[0]
        if flags.triggers.__len__() >1:
            for trigger in flags.triggers[1:]:
                cutString+=" || "+trigger
    return cutString
 
 
from DerivationFrameworkTools.DerivationFrameworkToolsConf import DerivationFramework__FilterCombinationAND
from DerivationFrameworkTools.DerivationFrameworkToolsConf import DerivationFramework__FilterCombinationOR
 
 
 
 
 
 
from LongLivedParticleDPDMaker.LongLivedParticleDPDMakerConf import DerivationFramework__DVTracklessJetFilterTool
 
DVSingleTracklessJetFilterTool = DerivationFramework__DVTracklessJetFilterTool( name = "DVSingleTracklessJetFilterTool",
                                                                                JetContainerKey = jetContainer,
                                                                                JetPtCut= primRPVLLDESDM.DV_SingleTracklessJetFilterFlags.cutEtMin,
                                                                                JetEtaCut= primRPVLLDESDM.DV_SingleTracklessJetFilterFlags.cutEtaMax,
                                                                                sumPtTrkCut = primRPVLLDESDM.DV_SingleTracklessJetFilterFlags.cutSumPtTrkMax,
                                                                                nJetsRequired = 1)
 
ToolSvc+=DVSingleTracklessJetFilterTool
 
DVDoubleTracklessJetFilterTool = DerivationFramework__DVTracklessJetFilterTool( name = "DVDoubleTracklessJetFilterTool",
                                                                                JetContainerKey = jetContainer,
                                                                                JetPtCut= primRPVLLDESDM.DV_DoubleTracklessJetFilterFlags.cutEtMin,
                                                                                JetEtaCut= primRPVLLDESDM.DV_DoubleTracklessJetFilterFlags.cutEtaMax,
                                                                                sumPtTrkCut = primRPVLLDESDM.DV_DoubleTracklessJetFilterFlags.cutSumPtTrkMax,
                                                                                nJetsRequired = 2)
 
ToolSvc+=DVDoubleTracklessJetFilterTool
 
 
 
 
from DerivationFrameworkTools.DerivationFrameworkToolsConf import DerivationFramework__PrescaleTool
DVPrescaleToolForMultijet = DerivationFramework__PrescaleTool(name = "DVPrescaleToolForMultijet",
                                                   Prescale = primRPVLLDESDM.DV_PrescalerFlags.prescale
                                                   )
ToolSvc+=DVPrescaleToolForMultijet
 
DVCombinedTracklessJetFilterToolForMultijet = DerivationFramework__FilterCombinationOR( name = "DVCombinedTracklessJetFilterToolForMultijet",
                                                                             FilterList = [DVSingleTracklessJetFilterTool,
                                                                                           DVDoubleTracklessJetFilterTool,
                                                                                           DVPrescaleToolForMultijet],
 
                                                                             )
ToolSvc+=DVCombinedTracklessJetFilterToolForMultijet
 
DVPrescaleToolForMET = DerivationFramework__PrescaleTool(name = "DVPrescaleToolForMET",
                                                   Prescale = primRPVLLDESDM.DV_PrescalerFlags.prescale
                                                   )
ToolSvc+=DVPrescaleToolForMET
 
DVCombinedTracklessJetFilterToolForMET = DerivationFramework__FilterCombinationOR( name = "DVCombinedTracklessJetFilterToolForMET",
                                                                             FilterList = [DVSingleTracklessJetFilterTool,
                                                                                           DVDoubleTracklessJetFilterTool,
                                                                                           DVPrescaleToolForMET],
 
                                                                             )
ToolSvc+=DVCombinedTracklessJetFilterToolForMET
 
DVPrescaleToolForPhoton = DerivationFramework__PrescaleTool(name = "DVPrescaleToolForPhoton",
                                                   Prescale = primRPVLLDESDM.DV_PrescalerFlags.prescale
                                                   )
ToolSvc+=DVPrescaleToolForPhoton
 
DVCombinedTracklessJetFilterToolForPhoton = DerivationFramework__FilterCombinationOR( name = "DVCombinedTracklessJetFilterToolForPhoton",
                                                                             FilterList = [DVSingleTracklessJetFilterTool,
                                                                                           DVDoubleTracklessJetFilterTool,
                                                                                           DVPrescaleToolForPhoton],
 
                                                                             )
ToolSvc+=DVCombinedTracklessJetFilterToolForPhoton
 
#======================================================================================================================================
 
 
 
 
 
 
 
from DerivationFrameworkCore.DerivationFrameworkCoreConf import DerivationFramework__DerivationKernel
 
# Barrel only MS only trigger 
DVMuonBarrelTriggerFilter = skimtool( name = "DVMuonBarrelTriggerFilter",
                                expression = DVTriggerSelectionString(primRPVLLDESDM.DV_MuonBarrelFilterFlags)
                                )
ToolSvc+=DVMuonBarrelTriggerFilter
 
DVMuonBarrelFilterTool = skimtool( name = "DVMuonBarrelFilterTool",
                             expression = DVSelectionString(primRPVLLDESDM.DV_MuonBarrelFilterFlags, muonContainer)
                )
ToolSvc+=DVMuonBarrelFilterTool
 
DV_MuonBarrelFilter = DerivationFramework__FilterCombinationAND( name = "DV_MuonBarrelFilter",
                                                                FilterList=[DVMuonBarrelFilterTool,DVMuonBarrelTriggerFilter],
                                                                #OutputLevel=DEBUG
                                                                )
ToolSvc+=DV_MuonBarrelFilter
# Full MS only trigger
DVMuonFullMSTriggerFilter = skimtool( name = "DVMuonFullMSTriggerFilter",
                                expression = DVTriggerSelectionString(primRPVLLDESDM.DV_MuonFullMSFilterFlags)
                                )
ToolSvc+=DVMuonFullMSTriggerFilter
 
DVMuonFullMSFilterTool = skimtool( name = "DVMuonFullMSFilterTool",
                             expression = DVSelectionString(primRPVLLDESDM.DV_MuonFullMSFilterFlags, muonContainer)
                )
ToolSvc+=DVMuonFullMSFilterTool
 
DV_MuonFullMSFilter = DerivationFramework__FilterCombinationAND( name = "DV_MuonFullMSFilter",
                                                                FilterList=[DVMuonFullMSFilterTool,DVMuonFullMSTriggerFilter],
                                                                #OutputLevel=DEBUG
                                                                )
ToolSvc+=DV_MuonFullMSFilter
 
# Or of barrel only and full MS
DV_MuonFinalFilter = DerivationFramework__FilterCombinationOR( name = "DV_MuonFinalFilter",
                                FilterList=[DV_MuonBarrelFilter,DV_MuonFullMSFilter],
                                #OutputLevel=DEBUG
                                )
ToolSvc+=DV_MuonFinalFilter
topSequence += kernel("RPVLL_DVMuonFilterKernel",
                      SkimmingTools = [DV_MuonFinalFilter])
RPVLLfilterNames.extend(["RPVLL_DVMuonFilterKernel"])
 
 
 
DVPhotonTriggerFilter = skimtool( name = "DVPhotonTriggerFilter",
                                expression = DVTriggerSelectionString(primRPVLLDESDM.DV_PhotonFilterFlags)
                                )
ToolSvc += DVPhotonTriggerFilter
 
# Photon IsEM
 
from LongLivedParticleDPDMaker.LongLivedParticleDPDMakerConf import DerivationFramework__RpvEgammaIDTool
DVPhotonSelection = DerivationFramework__RpvEgammaIDTool( name               = "DVPhotonSelection",
                                                             SelectionVariable = primRPVLLDESDM.DV_PhotonFilterFlags.cutIsEM,
                                                             CollectionName     = photonContainer,
                                                             EGammaSelectionKey = "DV"+photonContainer+primRPVLLDESDM.DV_PhotonFilterFlags.cutIsEM,
                                                           )
ToolSvc += DVPhotonSelection
topSequence += DerivationFramework__DerivationKernel( "DVPhotonAugmentationKernel",
                                                      AugmentationTools = [DVPhotonSelection]
                                                      )
 
 
 
DVPhotonFilterTool = skimtool( name = "DVPhotonFilterTool",
                               expression = DVSelectionString(primRPVLLDESDM.DV_PhotonFilterFlags, photonContainer),
                               )
ToolSvc += DVPhotonFilterTool
DV_PhotonFinalFilter = DerivationFramework__FilterCombinationAND( name = "DV_PhotonFinalFilter",
                                                                  FilterList=[DVPhotonFilterTool,DVPhotonTriggerFilter],
 
                                                                  )
 
ToolSvc+=DV_PhotonFinalFilter
 
DVPhotonPrescaleTool = DerivationFramework__PrescaleTool(name = "DVPhotonPrescaleTool",
                                                         Prescale = primRPVLLDESDM.DV_PhotonFilterFlags.prescale
                                                         )
ToolSvc+=DVPhotonPrescaleTool
 
DV_PrescaledPhotonFinalFilter = DerivationFramework__FilterCombinationAND(name="DV_PrescaledPhotonFinalFilter",
                                                                     FilterList=[DV_PhotonFinalFilter,DVPhotonPrescaleTool])
ToolSvc += DV_PrescaledPhotonFinalFilter
 
topSequence += kernel( "RPVLL_DV_PhotonFilterKernel",
                       SkimmingTools = [DV_PrescaledPhotonFinalFilter]
                       )
RPVLLfilterNames.extend(["RPVLL_DV_PhotonFilterKernel"])
 
 
 
DV_PhotonPlusTracklessJetFilter = DerivationFramework__FilterCombinationAND(name="DV_PhotonPlusTracklessJetFilter",
                                                                            FilterList=[DV_PhotonFinalFilter,DVCombinedTracklessJetFilterToolForPhoton]
                                                                            )
ToolSvc+= DV_PhotonPlusTracklessJetFilter
 
topSequence += kernel( "RPVLL_DV_PhotonPlusTLJetFilterKernel",
                       SkimmingTools = [DV_PhotonPlusTracklessJetFilter]
                       )
RPVLLfilterNames.extend(["RPVLL_DV_PhotonPlusTLJetFilterKernel"])
 
 
 
 
DVMultiJetTriggerFilter = skimtool( name = "DVMultiJetTriggerFilter",
                                  expression = DVTriggerSelectionString(primRPVLLDESDM.DV_MultiJetTriggerFlags)
                                  )
ToolSvc += DVMultiJetTriggerFilter
 
DV2JetFilterTool = skimtool( name = "DV2JetFilterTool",
                                 expression = DVSelectionString(primRPVLLDESDM.DV_2JetFilterFlags, jetContainer),
                                        )
ToolSvc += DV2JetFilterTool
 
DV3JetFilterTool = skimtool( name = "DV3JetFilterTool",
                                 expression = DVSelectionString(primRPVLLDESDM.DV_3JetFilterFlags, jetContainer),
                                        )
ToolSvc += DV3JetFilterTool
 
 
DV3JetFinalFilter = DerivationFramework__FilterCombinationAND( name = "DV3JetFinalFilter",
                                                               FilterList=[DV3JetFilterTool,DV2JetFilterTool],
#                                                              OutputLevel=DEBUG
                                                               )
ToolSvc+=DV3JetFinalFilter
 
DV4JetFilterTool = skimtool( name = "DV4JetFilterTool",
                                 expression = DVSelectionString(primRPVLLDESDM.DV_4JetFilterFlags, jetContainer),
                                        )
ToolSvc += DV4JetFilterTool
 
 
DV5JetFilterTool = skimtool( name = "DV5JetFilterTool",
                             expression = DVSelectionString(primRPVLLDESDM.DV_5JetFilterFlags, jetContainer),
                             )
ToolSvc += DV5JetFilterTool
 
 
DV6JetFilterTool = skimtool( name = "DV6JetFilterTool",
                             expression = DVSelectionString(primRPVLLDESDM.DV_6JetFilterFlags, jetContainer),
                             )
ToolSvc += DV6JetFilterTool
 
DV7JetFilterTool = skimtool( name = "DV7JetFilterTool",
                             expression = DVSelectionString(primRPVLLDESDM.DV_7JetFilterFlags, jetContainer),
                             )
ToolSvc += DV7JetFilterTool
 
 
 
 
DV_multiJet_offlineJetFilter = DerivationFramework__FilterCombinationOR( name = "DV_multiJet_offlineJetFilter",
                                                                        FilterList=[DV3JetFinalFilter,DV4JetFilterTool,DV5JetFilterTool,DV6JetFilterTool,DV7JetFilterTool],
 
                                                                        )
ToolSvc += DV_multiJet_offlineJetFilter
 
 
 
DV_MultiJetFinalTracklessFilter = DerivationFramework__FilterCombinationAND( name = "DV_MultiJetFinalTracklessFilter",
                                                                    FilterList=[DV_multiJet_offlineJetFilter,DVCombinedTracklessJetFilterToolForMultijet,DVMultiJetTriggerFilter],
 
                                                                        )
ToolSvc+= DV_MultiJetFinalTracklessFilter
 
 
DV_MultiJet2JHighPtFilter = skimtool( name = "DV_MultiJet2JHighPtFilter",
                             expression = DVSelectionString(primRPVLLDESDM.DV_2JetFilterFlags_HighpTCut, jetContainer),
                             )
ToolSvc += DV_MultiJet2JHighPtFilter
 
DV_MultiJet3JHighPtFilter = skimtool( name = "DV_MultiJet3JHighPtFilter",
                             expression = DVSelectionString(primRPVLLDESDM.DV_3JetFilterFlags_HighpTCut, jetContainer),
                             )
ToolSvc += DV_MultiJet3JHighPtFilter
 
 
DV_Multi3_2JetHighPtFilter = DerivationFramework__FilterCombinationAND( name = "DV_Multi3_2JetHighPtFilter",
                                                               FilterList=[DV_MultiJet2JHighPtFilter,DV_MultiJet3JHighPtFilter],
 
                                                               )
ToolSvc+=DV_Multi3_2JetHighPtFilter
 
 
DV_MultiJet4JHighPtFilter = skimtool( name = "DV_MultiJet4JHighPtFilter",
                             expression = DVSelectionString(primRPVLLDESDM.DV_4JetFilterFlags_HighpTCut, jetContainer),
                             )
ToolSvc += DV_MultiJet4JHighPtFilter
 
 
DV_MultiJet5JHighPtFilter = skimtool( name = "DV_MultiJet5JHighPtFilter",
                             expression = DVSelectionString(primRPVLLDESDM.DV_5JetFilterFlags_HighpTCut, jetContainer),
                             )
ToolSvc += DV_MultiJet5JHighPtFilter
 
 
DV_MultiJet6JHighPtFilter = skimtool( name = "DV_MultiJet6JHighPtFilter",
                             expression = DVSelectionString(primRPVLLDESDM.DV_6JetFilterFlags_HighpTCut, jetContainer),
                             )
ToolSvc += DV_MultiJet6JHighPtFilter
 
 
DV_MultiJet7JHighPtFilter = skimtool( name = "DV_MultiJet7JHighPtFilter",
                             expression = DVSelectionString(primRPVLLDESDM.DV_7JetFilterFlags_HighpTCut, jetContainer),
                             )
ToolSvc += DV_MultiJet7JHighPtFilter
 
 
DV_MultiJetHighPtFilter = DerivationFramework__FilterCombinationOR( name = "DV_MultiJetHighPtFilter",
                                                               FilterList=[DV_Multi3_2JetHighPtFilter,DV_MultiJet4JHighPtFilter,DV_MultiJet5JHighPtFilter,DV_MultiJet6JHighPtFilter,DV_MultiJet7JHighPtFilter],
 
                                                               )
ToolSvc+=DV_MultiJetHighPtFilter
 
 
DV_MultiJetFinalFilter = DerivationFramework__FilterCombinationOR( name = "DV_MultiJetFinalFilter",
                                                               FilterList=[DV_MultiJetFinalTracklessFilter,DV_MultiJetHighPtFilter],
 
                                                               )
ToolSvc+= DV_MultiJetFinalFilter
 
topSequence += kernel( "RPVLL_DV_MultiJetFilterKernel",
                       SkimmingTools = [DV_MultiJetFinalFilter],
                       )
RPVLLfilterNames.extend(["RPVLL_DV_MultiJetFilterKernel"])
 
 
 
 
DVMETTriggerFilter = skimtool( name = "DVMETTriggerFilter",
                               expression = DVTriggerSelectionString(primRPVLLDESDM.DV_METFilterFlags)
                               )
 
ToolSvc+=DVMETTriggerFilter
 
from LongLivedParticleDPDMaker.LongLivedParticleDPDMakerConf import DerivationFramework__DVMissingETFilterTool
 
 
DVMETFilterTool = DerivationFramework__DVMissingETFilterTool(name = "DVMETFilterTool",
                                                             METContainerKey = METContainer,
                                                             METCut=primRPVLLDESDM.DV_METFilterFlags.cutMetMin)
ToolSvc+=DVMETFilterTool
 
DV_METFinalFilter = DerivationFramework__FilterCombinationAND( name = "DV_METFinalFilter",
                                                               FilterList=[DVMETFilterTool,DVMETTriggerFilter],
 
                                                               )
ToolSvc += DV_METFinalFilter
 
topSequence += kernel( "RPVLL_DV_METFilterKernel",
                       SkimmingTools = [DV_METFinalFilter],
                       )
RPVLLfilterNames.extend(["RPVLL_DV_METFilterKernel"])