RatesEmulationExample.py

Go to the documentation of this file.

#!/usr/bin/env python
#
#  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
#
 
# For processing JZ slices
def getHistogramSuffix(dsid):
  channel_to_string = {
    801165: "_JZ0",
    801166: "_JZ1",
    801167: "_JZ2",
    801168: "_JZ3",
    801169: "_JZ4",
    801170: "_JZ5",
    801171: "_JZ6",
    801172: "_JZ7",
    801173: "_JZ8",
    801174: "_JZ9"
  }
  try: 
    return channel_to_string[dsid]
  except KeyError:
    return ""
 
if __name__=='__main__':
  import sys
 
  # Initialise Athena config flags to get and customise the argument parser
  from AthenaConfiguration.AllConfigFlags import initConfigFlags
  flags = initConfigFlags()
  parser = flags.getArgumentParser()
  #
  parser.add_argument('--disableGlobalGroups', action='store_true', help='Turn off global groups')
  parser.add_argument('--disableTriggerGroups', action='store_true', help='Turn off per-trigger groups')
  parser.add_argument('--disableExpressGroup', action='store_true', help='Turn off express stream rates')
  parser.add_argument('--disableUniqueRates', action='store_true', help='Turn off unique rates (much faster!)')
  parser.add_argument('--disableLumiExtrapolation', action='store_true', help='Turn off luminosity extrapolation')
  #
  parser.add_argument('--MCDatasetName', default='', type=str, help='For MC input: Name of the dataset, can be used instead of MCCrossSection, MCFilterEfficiency')
  parser.add_argument('--MCCrossSection', default=0.0, type=float, help='For MC input: Cross section of process in nb')
  parser.add_argument('--MCFilterEfficiency', default=1.0, type=float, help='For MC input: Filter efficiency of any MC filter (0.0 - 1.0)')
  parser.add_argument('--MCKFactor', default=1.0, type=float, help='For MC input: Additional multiplicitive fudge-factor to the supplied cross section.')
  parser.add_argument('--MCIgnoreGeneratorWeights', action='store_true', help='For MC input: Flag to disregard any generator weights.')
  #
  parser.add_argument('--doRatesVsPositionInTrain', action='store_true', help='Study rates vs BCID position in bunch train')
  parser.add_argument('--vetoStartOfTrain', default=0, type=int, help='Number of BCIDs at the start of the train to veto, implies doRatesVsPositionInTrain')
  #
  parser.add_argument('--targetLuminosity', default=2e34, type=float)
  #
  parser.add_argument('--outputHist', default='RatesHistograms.root', type=str, help='Histogram output ROOT file')
  parser.add_argument('--ebWeightsDirectory', default='', type=str, help='Path to directory with local EB xml files')
  #
  parser.add_argument('--doMultiSliceDiJet', action='store_true', help='Enable the HS-softer-than-PU (HSTP) filter; reweight the Slices according to Jet/ETMiss procedure; recommended by PMG for di-jet slices.')
  #
  parser.add_argument('flags', nargs='*', help='Config flag overrides')
 
  args = parser.parse_args()
 
  flags.Exec.EventPrintoutInterval = 1000
  flags.Common.MsgSuppression = False
  flags.Concurrency.NumThreads = 1
 
  args = flags.fillFromArgs(parser=parser)
  flags.lock()
 
  useBunchCrossingData = (args.doRatesVsPositionInTrain or args.vetoStartOfTrain > 0)
 
  from AthenaConfiguration.AutoConfigFlags import GetFileMD
  metadata = GetFileMD(flags.Input.Files)
  # Initialize configuration object, add accumulator, merge, and run.
  from AthenaConfiguration.MainServicesConfig import MainServicesCfg 
  from AthenaConfiguration.ComponentFactory import CompFactory
 
  from AthenaPoolCnvSvc.PoolReadConfig import PoolReadCfg
  cfg = MainServicesCfg(flags)
  cfg.merge(PoolReadCfg(flags))
 
  histSvc = CompFactory.THistSvc()
  histSvc.Output += ["RATESTREAM DATAFILE='" + args.outputHist + "' OPT='RECREATE'"]
  cfg.addService(histSvc)
 
  # Minimal config needed to read metadata: MetaDataSvc & ProxyProviderSvc
  from AthenaServices.MetaDataSvcConfig import MetaDataSvcCfg
  cfg.merge(MetaDataSvcCfg(flags))
 
  from TrigConfxAOD.TrigConfxAODConfig import getxAODConfigSvc
  cfgsvc = cfg.getPrimaryAndMerge(getxAODConfigSvc(flags))
 
  from TrigDecisionTool.TrigDecisionToolConfig import TrigDecisionToolCfg
  tdt = cfg.getPrimaryAndMerge(TrigDecisionToolCfg(flags))
 
  # If the dataset name is in the input files path, then it will be fetched from there
  # Note to enable autolookup, first run "lsetup pyami; voms-proxy-init -voms atlas" and enter your grid pass phrase
  xsec = args.MCCrossSection
  fEff = args.MCFilterEfficiency
  dset = args.MCDatasetName
  if flags.Input.isMC and xsec == 0: # If the input file is MC then make sure we have the needed info
    from RatesAnalysis.GetCrossSectionAMITool import GetCrossSectionAMI
    amiTool = GetCrossSectionAMI()
    if dset == "": # Can we get the dataset name from the input file path?
      dset = amiTool.getDatasetNameFromPath(flags.Input.Files[0])
    amiTool.queryAmi(dset)
    xsec = amiTool.crossSection
    fEff = amiTool.filterEfficiency
 
  ebw = CompFactory.EnhancedBiasWeighter('EnhancedBiasRatesTool')
  ebw.RunNumber = flags.Input.RunNumbers[0]
  ebw.UseBunchCrossingData = useBunchCrossingData
  ebw.IsMC = flags.Input.isMC
  # The following three are only needed if isMC == true
  ebw.TargetLuminosity = args.targetLuminosity
  ebw.MCCrossSection = xsec
  ebw.MCFilterEfficiency = fEff
  ebw.MCKFactor = args.MCKFactor
  ebw.DoMultiSliceDiJet = args.doMultiSliceDiJet and flags.Input.isMC
  ebw.MCIgnoreGeneratorWeights = args.MCIgnoreGeneratorWeights
  ebw.EBWeightsDirectory = args.ebWeightsDirectory if args.ebWeightsDirectory else ""
  cfg.addPublicTool(ebw)
 
  rates = CompFactory.RatesEmulationExample()
  rates.DoTriggerGroups = not args.disableTriggerGroups
  rates.DoGlobalGroups = not args.disableGlobalGroups
  rates.DoExpressRates = not args.disableExpressGroup
  rates.DoUniqueRates = not args.disableUniqueRates
  rates.UseBunchCrossingData = useBunchCrossingData
  rates.TargetLuminosity = args.targetLuminosity
  rates.VetoStartOfTrain = args.vetoStartOfTrain
  rates.EnableLumiExtrapolation = not args.disableLumiExtrapolation and not args.doMultiSliceDiJet
  rates.EnhancedBiasRatesTool = ebw
  rates.TrigDecisionTool = tdt
  rates.TrigConfigSvc = cfgsvc
  rates.DoMultiSliceDiJet = args.doMultiSliceDiJet and flags.Input.isMC
  rates.histogramSuffix = getHistogramSuffix(metadata['mc_channel_number'])
  
  cfg.addEventAlgo(rates)
 
 
  from JetRecConfig.StandardSmallRJets import AntiKt4Truth, standardghosts, flavourghosts, calibmods, truthmods, standardmods, clustermods
  from JetRecConfig.JetRecConfig import JetRecCfg
  if 'AntiKt4TruthJets' not in flags.Input.Collections and args.doMultiSliceDiJet:
    cfg.merge(JetRecCfg(flags, AntiKt4Truth))
  if 'AntiKt4EMTopoJets' not in flags.Input.Collections:
    from JetRecConfig.JetDefinition import JetDefinition
    from JetRecConfig.StandardJetConstits import stdConstitDic as cst
    AntiKt4EMTopo = JetDefinition("AntiKt",0.4,cst.EMTopoOrigin,
                               ghostdefs = standardghosts+flavourghosts,
                               modifiers = calibmods+truthmods+standardmods+clustermods+("Filter_calibThreshold:15000","LArHVCorr","jetiso",),
                               lock = True)
    cfg.merge(JetRecCfg(flags, AntiKt4EMTopo))
 
 
  # Setup for accessing bunchgroup data from the DB
  # if useBunchCrossingData:
  #   from LumiBlockComps.BunchCrossingCondAlgConfig import BunchCrossingCondAlgCfg
  #   cfg.merge(BunchCrossingCondAlgCfg(flags))
 
  # If you want to turn on more detailed messages ...
  # exampleMonitorAcc.getEventAlgo('ExampleMonAlg').OutputLevel = 2 # DEBUG
  cfg.printConfig(withDetails=False) # set True for exhaustive info
 
  sc = cfg.run(flags.Exec.MaxEvents)
  sys.exit(0 if sc.isSuccess() else 1)