L1THGCalUtilities/test/testHGCalL1T_multialgo_V19_cfg.py

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