Skip to content
Snippets Groups Projects
Commit 7ffe8dd4 authored by Sagar Addepalli's avatar Sagar Addepalli :speech_balloon: Committed by Tadej Novak
Browse files

Update auto PRW configs to support AF3

Update auto PRW configs to support AF3
parent 34d091a7
No related branches found
No related tags found
No related merge requests found
......@@ -20,18 +20,18 @@ from AthenaConfiguration.ComponentAccumulator import ComponentAccumulator
# best judgement whether to fix this configuration or to change it to
# wrap the block configuration instead.
def PileupReweightingToolCfg(flags, name="PileupReweightingTool", **kwargs):
def PileupReweightingToolCfg(flags, name="PileupReweightingTool", commonPRW=True, **kwargs):
acc = ComponentAccumulator()
from Campaigns.Utils import getMCCampaign,Campaign
from Campaigns.Utils import getMCCampaign
campaign = getMCCampaign(flags.Input.Files)
from PileupReweighting.AutoconfigurePRW import defaultConfigFiles,getConfigurationFiles,getLumicalcFiles
kwargs.setdefault("LumiCalcFiles", getLumicalcFiles(campaign))
if campaign in [Campaign.MC23a,Campaign.MC23c]:
if commonPRW:
kwargs.setdefault("ConfigFiles", defaultConfigFiles(campaign))
else:
kwargs.setdefault("ConfigFiles", getConfigurationFiles(files=flags.Input.Files))
acc.setPrivateTools(CompFactory.CP.PileupReweightingTool(**kwargs))
return acc
......
# Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
# Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
from Campaigns.Utils import Campaign, getMCCampaign
from AnalysisAlgorithmsConfig.ConfigAccumulator import DataType
......@@ -133,14 +133,14 @@ def getConfigurationFiles(campaign=None, dsid=None, data_type=None, files=None,
simulation_flavour = GetFileMD(files).get('SimulationFlavour', '')
data_type = DataType.FullSim if (not simulation_flavour or 'FullG4' in simulation_flavour) else DataType.FastSim
# data_type as in pileup analysis sequence: either 'data' or ('fullsim' or 'afii')
# data_type as in pileup analysis sequence: either 'data' or ('fullsim' or 'af3')
if data_type is DataType.Data:
raise ValueError('Data is not supported')
if data_type is DataType.FullSim:
simulation_type = 'FS'
elif data_type is DataType.FastSim:
simulation_type = 'AFII'
simulation_type = 'AF3'
else:
raise ValueError(f'Invalid data_type {data_type}')
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment