L1THGCalUtilities/test/testHGCalL1T_multialgo_V11_cfg.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 from Configuration.Eras.Era_Phase2C9_cff import Phase2C9
0004 process = cms.Process('DIGI',Phase2C9)
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.GeometryExtended2026D49Reco_cff')
0013 process.load('Configuration.Geometry.GeometryExtended2026D49_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/Phase2HLTTDRWinter20DIGI/SingleElectron_PT2to200/GEN-SIM-DIGI-RAW/PU200_110X_mcRun4_realistic_v3_ext2-v2/40000/00582F93-5A2A-5847-8162-D81EE503500F.root'),
0033        inputCommands=cms.untracked.vstring(
0034            'keep *',
0035            'drop l1tTkPrimaryVertexs_L1TkPrimaryVertex__RECO',
0036            )
0037        )
0038
0039 process.options = cms.untracked.PSet(
0040
0041 )
0042
0043 # Production Info
0044 process.configurationMetadata = cms.untracked.PSet(
0045     version = cms.untracked.string('$Revision: 1.20 $'),
0046     annotation = cms.untracked.string('SingleElectronPt10_cfi nevts:10'),
0047     name = cms.untracked.string('Applications')
0048 )
0049
0050 # Output definition
0051 process.TFileService = cms.Service(
0052     "TFileService",
0053     fileName = cms.string("ntuple.root")
0054     )
0055
0056 # Other statements
0057 from Configuration.AlCa.GlobalTag import GlobalTag
0058 process.GlobalTag = GlobalTag(process.GlobalTag, 'auto:phase2_realistic_T15', '')
0059
0060 # load HGCAL TPG simulation
0061 process.load('L1Trigger.L1THGCal.hgcalTriggerPrimitives_cff')
0062 process.load('L1Trigger.L1THGCalUtilities.HGC3DClusterGenMatchSelector_cff')
0063 process.load('L1Trigger.L1THGCalUtilities.hgcalTriggerNtuples_cff')
0064 from L1Trigger.L1THGCalUtilities.hgcalTriggerChains import HGCalTriggerChains
0065 import L1Trigger.L1THGCalUtilities.vfe as vfe
0066 import L1Trigger.L1THGCalUtilities.concentrator as concentrator
0067 import L1Trigger.L1THGCalUtilities.clustering2d as clustering2d
0068 import L1Trigger.L1THGCalUtilities.clustering3d as clustering3d
0069 import L1Trigger.L1THGCalUtilities.selectors as selectors
0070 import L1Trigger.L1THGCalUtilities.customNtuples as ntuple
0071
0072
0073 chains = HGCalTriggerChains()
0074 # Register algorithms
0075 ## VFE
0076 chains.register_vfe("Floatingpoint", vfe.CreateVfe())
0077 ## ECON
0078 chains.register_concentrator("Supertriggercell", concentrator.CreateSuperTriggerCell())
0079 chains.register_concentrator("Threshold", concentrator.CreateThreshold())
0080 chains.register_concentrator("Bestchoice", concentrator.CreateBestChoice())
0081 chains.register_concentrator("AutoEncoder", concentrator.CreateAutoencoder())
0082 ## BE1
0083 chains.register_backend1("Dummy", clustering2d.CreateDummy())
0084 ## BE2
0085 chains.register_backend2("Histomax", clustering3d.CreateHistoMax())
0086 # Register selector
0087 chains.register_selector("Genmatch", selectors.CreateGenMatch())
0088
0089
0090 # Register ntuples
0091 ntuple_list = ['event', 'gen', 'multiclusters']
0092 chains.register_ntuple("Genclustersntuple", ntuple.CreateNtuple(ntuple_list))
0093
0094 # Register trigger chains
0095 concentrator_algos = ['Supertriggercell', 'Threshold', 'Bestchoice', 'AutoEncoder']
0096 backend_algos = ['Histomax']
0097 ## Make cross product fo ECON and BE algos
0098 import itertools
0099 for cc,be in itertools.product(concentrator_algos,backend_algos):
0100     chains.register_chain('Floatingpoint', cc, 'Dummy', be, 'Genmatch', 'Genclustersntuple')
0101
0102 process = chains.create_sequences(process)
0103
0104 # Remove towers from sequence
0105 process.L1THGCalTriggerPrimitives.remove(process.L1THGCalTowerMap)
0106 process.L1THGCalTriggerPrimitives.remove(process.L1THGCalTower)
0107
0108 process.hgcl1tpg_step = cms.Path(process.L1THGCalTriggerPrimitives)
0109 process.selector_step = cms.Path(process.L1THGCalTriggerSelector)
0110 process.ntuple_step = cms.Path(process.L1THGCalTriggerNtuples)
0111
0112 # Schedule definition
0113 process.schedule = cms.Schedule(process.hgcl1tpg_step, process.selector_step, process.ntuple_step)
0114
0115 # Add early deletion of temporary data products to reduce peak memory need
0116 from Configuration.StandardSequences.earlyDeleteSettings_cff import customiseEarlyDelete
0117 process = customiseEarlyDelete(process)
0118 # End adding early deletion
0119