L1THGCalUtilities/test/testHGCalL1T_multialgo_V16_cfg.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 from Configuration.Eras.Era_Phase2C17I13M9_cff import Phase2C17I13M9
0004 process = cms.Process('DIGI',Phase2C17I13M9)
0005
0006 # import of standard configurations
0007 process.load('Configuration.StandardSequences.Services_cff')
0008 process.load('SimGeneral.HepPDTESSource.pythiapdt_cfi')
0009 process.load('FWCore.MessageService.MessageLogger_cfi')
0010 process.load('Configuration.EventContent.EventContent_cff')
0011 process.load('SimGeneral.MixingModule.mixNoPU_cfi')
0012 process.load('Configuration.Geometry.GeometryExtendedRun4D88Reco_cff')
0013 process.load('Configuration.Geometry.GeometryExtendedRun4D88_cff')
0014 process.load('Configuration.StandardSequences.MagneticField_cff')
0015 process.load('Configuration.StandardSequences.Generator_cff')
0016 process.load('IOMC.EventVertexGenerators.VtxSmearedHLLHC14TeV_cfi')
0017 process.load('GeneratorInterface.Core.genFilterSummary_cff')
0018 process.load('Configuration.StandardSequences.SimIdeal_cff')
0019 process.load('Configuration.StandardSequences.Digi_cff')
0020 process.load('Configuration.StandardSequences.SimL1Emulator_cff')
0021 process.load('Configuration.StandardSequences.DigiToRaw_cff')
0022 process.load('Configuration.StandardSequences.EndOfProcess_cff')
0023 process.load('Configuration.StandardSequences.FrontierConditions_GlobalTag_cff')
0024
0025
0026 process.maxEvents = cms.untracked.PSet(
0027     input = cms.untracked.int32(5)
0028 )
0029
0030 # Input source
0031 process.source = cms.Source("PoolSource",
0032        fileNames = cms.untracked.vstring('/store/mc/Phase2Fall22DRMiniAOD/TT_TuneCP5_14TeV-powheg-pythia8/GEN-SIM-DIGI-RAW-MINIAOD/PU200_125X_mcRun4_realistic_v2_ext1-v1/30000/000c5e5f-78f7-44ee-95fe-7b2f2c2e2312.root'),
0033        inputCommands=cms.untracked.vstring(
0034            'keep *',
0035            )
0036        )
0037
0038 process.options = cms.untracked.PSet(
0039
0040 )
0041
0042 # Production Info
0043 process.configurationMetadata = cms.untracked.PSet(
0044     version = cms.untracked.string('$Revision: 1.20 $'),
0045     annotation = cms.untracked.string('SingleElectronPt10_cfi nevts:10'),
0046     name = cms.untracked.string('Applications')
0047 )
0048
0049 # Output definition
0050 process.TFileService = cms.Service(
0051     "TFileService",
0052     fileName = cms.string("ntuple.root")
0053     )
0054
0055 # Other statements
0056 from Configuration.AlCa.GlobalTag import GlobalTag
0057 process.GlobalTag = GlobalTag(process.GlobalTag, 'auto:phase2_realistic_T21', '')
0058
0059 # load HGCAL TPG simulation
0060 process.load('L1Trigger.L1THGCal.hgcalTriggerPrimitives_cff')
0061 process.load('L1Trigger.L1THGCalUtilities.HGC3DClusterGenMatchSelector_cff')
0062 process.load('L1Trigger.L1THGCalUtilities.hgcalTriggerNtuples_cff')
0063 from L1Trigger.L1THGCalUtilities.hgcalTriggerChains import HGCalTriggerChains
0064 import L1Trigger.L1THGCalUtilities.vfe as vfe
0065 import L1Trigger.L1THGCalUtilities.concentrator as concentrator
0066 import L1Trigger.L1THGCalUtilities.clustering2d as clustering2d
0067 import L1Trigger.L1THGCalUtilities.clustering3d as clustering3d
0068 import L1Trigger.L1THGCalUtilities.selectors as selectors
0069 import L1Trigger.L1THGCalUtilities.customNtuples as ntuple
0070
0071
0072 chains = HGCalTriggerChains()
0073 # Register algorithms
0074 ## VFE
0075 chains.register_vfe("Floatingpoint", vfe.CreateVfe())
0076 ## ECON
0077 chains.register_concentrator("Supertriggercell", concentrator.CreateSuperTriggerCell())
0078 chains.register_concentrator("Threshold", concentrator.CreateThreshold())
0079 chains.register_concentrator("Bestchoice", concentrator.CreateBestChoice())
0080 chains.register_concentrator("AutoEncoder", concentrator.CreateAutoencoder())
0081 ## BE1
0082 chains.register_backend1("Dummy", clustering2d.CreateDummy())
0083 ## BE2
0084 chains.register_backend2("Histomax", clustering3d.CreateHistoMax())
0085 # Register selector
0086 chains.register_selector("Genmatch", selectors.CreateGenMatch())
0087
0088
0089 # Register ntuples
0090 ntuple_list = ['event', 'gen', 'multiclusters']
0091 chains.register_ntuple("Genclustersntuple", ntuple.CreateNtuple(ntuple_list))
0092
0093 # Register trigger chains
0094 concentrator_algos = ['Supertriggercell', 'Threshold', 'Bestchoice', 'AutoEncoder']
0095 backend_algos = ['Histomax']
0096 ## Make cross product fo ECON and BE algos
0097 import itertools
0098 for cc,be in itertools.product(concentrator_algos,backend_algos):
0099     chains.register_chain('Floatingpoint', cc, 'Dummy', be, 'Genmatch', 'Genclustersntuple')
0100
0101 process = chains.create_sequences(process)
0102
0103 # Remove towers from sequence
0104 process.L1THGCalTriggerPrimitives.remove(process.L1THGCalTowerMap)
0105 process.L1THGCalTriggerPrimitives.remove(process.L1THGCalTower)
0106
0107 process.hgcl1tpg_step = cms.Path(process.L1THGCalTriggerPrimitives)
0108 process.selector_step = cms.Path(process.L1THGCalTriggerSelector)
0109 process.ntuple_step = cms.Path(process.L1THGCalTriggerNtuples)
0110
0111 # Schedule definition
0112 process.schedule = cms.Schedule(process.hgcl1tpg_step, process.selector_step, process.ntuple_step)
0113
0114 # Add early deletion of temporary data products to reduce peak memory need
0115 from Configuration.StandardSequences.earlyDeleteSettings_cff import customiseEarlyDelete
0116 process = customiseEarlyDelete(process)
0117 # End adding early deletion
0118