Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2025-03-23 15:57:53

0001 import FWCore.ParameterSet.Config as cms
0002 
0003 hltTiclTrackstersCLUE3DHigh = cms.EDProducer("TrackstersProducer",
0004     detector = cms.string('HGCAL'),
0005     filtered_mask = cms.InputTag("hltFilteredLayerClustersCLUE3DHigh","CLUE3DHigh"),
0006     itername = cms.string('CLUE3DHigh'),
0007     layer_clusters = cms.InputTag("hltHgcalMergeLayerClusters"),
0008     layer_clusters_hfnose_tiles = cms.InputTag("ticlLayerTileHFNose"),
0009     layer_clusters_tiles = cms.InputTag("hltTiclLayerTileProducer"),
0010     mightGet = cms.optional.untracked.vstring,
0011     original_mask = cms.InputTag("hltHgcalMergeLayerClusters","InitialLayerClustersMask"),
0012     patternRecognitionBy = cms.string('CLUE3D'),
0013     inferenceAlgo = cms.string('TracksterInferenceByCNNv4'),
0014     pluginPatternRecognitionByCA = cms.PSet(
0015         algo_verbosity = cms.int32(0),
0016         energy_em_over_total_threshold = cms.double(-1),
0017         etaLimitIncreaseWindow = cms.double(2.1),
0018         filter_on_categories = cms.vint32(0),
0019         max_delta_time = cms.double(3),
0020         max_longitudinal_sigmaPCA = cms.double(9999),
0021         max_missing_layers_in_trackster = cms.int32(9999),
0022         max_out_in_hops = cms.int32(10),
0023         min_cos_pointing = cms.double(-1),
0024         min_cos_theta = cms.double(0.915),
0025         min_layers_per_trackster = cms.int32(10),
0026         oneTracksterPerTrackSeed = cms.bool(False),
0027         out_in_dfs = cms.bool(True),
0028         pid_threshold = cms.double(0),
0029         promoteEmptyRegionToTrackster = cms.bool(False),
0030         root_doublet_max_distance_from_seed_squared = cms.double(9999),
0031         shower_start_max_layer = cms.int32(9999),
0032         siblings_maxRSquared = cms.vdouble(0.0006, 0.0006, 0.0006),
0033         skip_layers = cms.int32(0),
0034         type = cms.string('CA')
0035     ),
0036     pluginPatternRecognitionByCLUE3D = cms.PSet(
0037     algo_verbosity = cms.int32(0),
0038     criticalDensity = cms.vdouble(
0039       0.6,
0040       0.6,
0041       0.6
0042     ),
0043     criticalSelfDensity = cms.vdouble(
0044       0.15,
0045       0.15,
0046       0.15
0047     ),
0048     densitySiblingLayers = cms.vint32(
0049       3,
0050       3,
0051       3
0052     ),
0053     densityEtaPhiDistanceSqr = cms.vdouble(
0054       0.0008,
0055       0.0008,
0056       0.0008
0057     ),
0058     densityXYDistanceSqr = cms.vdouble(
0059       3.24,
0060       3.24,
0061       3.24
0062     ),
0063     kernelDensityFactor = cms.vdouble(
0064       0.2,
0065       0.2,
0066       0.2
0067     ),
0068     densityOnSameLayer = cms.bool(False),
0069     nearestHigherOnSameLayer = cms.bool(False),
0070     useAbsoluteProjectiveScale = cms.bool(True),
0071     useClusterDimensionXY = cms.bool(False),
0072     rescaleDensityByZ = cms.bool(False),
0073     criticalEtaPhiDistance = cms.vdouble(
0074       0.025,
0075       0.025,
0076       0.025
0077     ),
0078     criticalXYDistance = cms.vdouble(
0079       1.8,
0080       1.8,
0081       1.8
0082     ),
0083     criticalZDistanceLyr = cms.vint32(
0084       5,
0085       5,
0086       5
0087     ),
0088     outlierMultiplier = cms.vdouble(
0089       2,
0090       2,
0091       2
0092     ),
0093     minNumLayerCluster = cms.vint32(
0094       2,
0095       2,
0096       2
0097     ),
0098     computeLocalTime = cms.bool(False),
0099     doPidCut = cms.bool(True),
0100     cutHadProb = cms.double(999.),
0101     type = cms.string('CLUE3D')
0102 
0103     ),
0104     pluginPatternRecognitionByFastJet = cms.PSet(
0105         algo_verbosity = cms.int32(0),
0106         antikt_radius = cms.double(0.09),
0107         minNumLayerCluster = cms.int32(5),
0108         type = cms.string('FastJet')
0109     ),
0110     pluginInferenceAlgoTracksterInferenceByCNNv4 = cms.PSet(
0111         algo_verbosity = cms.int32(0),
0112         onnxModelPath = cms.FileInPath('RecoHGCal/TICL/data/ticlv4/onnx_models/energy_id_v0.onnx'),
0113         inputNames  = cms.vstring('input:0'),
0114         outputNames = cms.vstring("output/regressed_energy:0", "output/id_probabilities:0"),
0115         eid_min_cluster_energy = cms.double(1),
0116         eid_n_layers = cms.int32(50),
0117         eid_n_clusters = cms.int32(10),
0118         doPID = cms.int32(1),
0119         doRegression = cms.int32(0),
0120         type = cms.string('TracksterInferenceByCNNv4')
0121     ),
0122     pluginInferenceAlgoTracksterInferenceByDNN = cms.PSet(
0123         algo_verbosity = cms.int32(0),
0124     onnxPIDModelPath = cms.FileInPath('RecoHGCal/TICL/data/ticlv5/onnx_models/DNN/patternrecognition/id_v0.onnx'),
0125         onnxEnergyModelPath = cms.FileInPath('RecoHGCal/TICL/data/ticlv5/onnx_models/DNN/patternrecognition/energy_v0.onnx'),
0126         inputNames  = cms.vstring('input'),
0127         output_en   = cms.vstring('enreg_output'),
0128         output_id   = cms.vstring('pid_output'),
0129         eid_n_layers = cms.int32(50),
0130         eid_n_clusters = cms.int32(10),
0131         doPID = cms.int32(1),
0132         doRegression = cms.int32(0),
0133         type = cms.string('TracksterInferenceByDNN')
0134     ),
0135     pluginInferenceAlgoTracksterInferenceByPFN = cms.PSet(
0136         algo_verbosity = cms.int32(0),
0137         onnxPIDModelPath = cms.FileInPath('RecoHGCal/TICL/data/ticlv5/onnx_models/PFN/patternrecognition/id_v0.onnx'),
0138         onnxEnergyModelPath = cms.FileInPath('RecoHGCal/TICL/data/ticlv5/onnx_models/PFN/patternrecognition/energy_v0.onnx'),
0139         inputNames  = cms.vstring('input','input_tr_features'),
0140         output_en   = cms.vstring('enreg_output'),
0141         output_id   = cms.vstring('pid_output'),
0142         eid_n_layers = cms.int32(50),
0143         eid_n_clusters = cms.int32(10),
0144         doPID = cms.int32(1),
0145         doRegression = cms.int32(0),
0146         type = cms.string('TracksterInferenceByPFN')
0147     ),
0148     pluginInferenceAlgoTracksterInferenceByANN = cms.PSet(
0149         algo_verbosity = cms.int32(0),
0150         type = cms.string('TracksterInferenceByANN')
0151 
0152     ),
0153     seeding_regions = cms.InputTag("hltTiclSeedingGlobal"),
0154     time_layerclusters = cms.InputTag("hltHgcalMergeLayerClusters","timeLayerCluster")
0155     )
0156     
0157 from Configuration.ProcessModifiers.ticl_v5_cff import ticl_v5
0158 ticl_v5.toModify(hltTiclTrackstersCLUE3DHigh.pluginPatternRecognitionByCLUE3D, computeLocalTime = cms.bool(True))
0159 ticl_v5.toModify(hltTiclTrackstersCLUE3DHigh, inferenceAlgo = cms.string('TracksterInferenceByPFN'))