TICL/python/TrkStep_cff.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 from RecoHGCal.TICL.TICLSeedingRegions_cff import ticlSeedingTrk, ticlSeedingTrkHFNose
0004 from RecoHGCal.TICL.ticlLayerTileProducer_cfi import ticlLayerTileProducer as _ticlLayerTileProducer
0005 from RecoHGCal.TICL.trackstersProducer_cfi import trackstersProducer as _trackstersProducer
0006 from RecoHGCal.TICL.filteredLayerClustersProducer_cfi import filteredLayerClustersProducer as _filteredLayerClustersProducer
0007
0008 # CLUSTER FILTERING/MASKING
0009
0010 filteredLayerClustersTrk = _filteredLayerClustersProducer.clone(
0011     clusterFilter = "ClusterFilterByAlgoAndSize",
0012     min_cluster_size = 3, # inclusive
0013     LayerClustersInputMask = 'ticlTrackstersEM',
0014     iteration_label = "Trk"
0015 )
0016
0017 # CA - PATTERN RECOGNITION
0018
0019 ticlTrackstersTrk = _trackstersProducer.clone(
0020     filtered_mask = "filteredLayerClustersTrk:Trk",
0021     seeding_regions = "ticlSeedingTrk",
0022     original_mask = 'ticlTrackstersEM',
0023     pluginPatternRecognitionByCA = dict(
0024         filter_on_categories = [2, 4], # filter muons and charged hadrons
0025         pid_threshold = 0.0,
0026         skip_layers = 3,
0027         min_layers_per_trackster = 10,
0028         min_cos_theta = 0.866, # ~30 degrees
0029         min_cos_pointing = 0.798, # ~ 37 degrees
0030         max_delta_time = -1.,
0031         algo_verbosity = 2,
0032         oneTracksterPerTrackSeed = True,
0033         promoteEmptyRegionToTrackster = True
0034     ),
0035     itername = "Trk"
0036 )
0037
0038 from Configuration.ProcessModifiers.ticl_v5_cff import ticl_v5
0039 ticl_v5.toModify(ticlTrackstersTrk.pluginPatternRecognitionByCA, computeLocalTime = cms.bool(True))
0040
0041 ticlTrkStepTask = cms.Task(ticlSeedingTrk
0042     ,filteredLayerClustersTrk
0043     ,ticlTrackstersTrk)
0044
0045 # HFNOSE CLUSTER FILTERING/MASKING
0046
0047 filteredLayerClustersHFNoseTrk = filteredLayerClustersTrk.clone(
0048     LayerClusters = 'hgcalLayerClustersHFNose',
0049     LayerClustersInputMask = 'ticlTrackstersHFNoseEM',
0050     min_cluster_size = 2, # inclusive
0051     algo_number = [9], # reco::CaloCluster::hfnose
0052     iteration_label = "Trkn"
0053 )
0054
0055 # HFNOSE CA - PATTERN RECOGNITION
0056
0057 ticlTrackstersHFNoseTrk = ticlTrackstersTrk.clone(
0058     detector = "HFNose",
0059     layer_clusters = "hgcalLayerClustersHFNose",
0060     layer_clusters_hfnose_tiles = "ticlLayerTileHFNose",
0061     original_mask = 'ticlTrackstersHFNoseEM',
0062     filtered_mask = "filteredLayerClustersHFNoseTrk:Trkn",
0063     seeding_regions = "ticlSeedingTrkHFNose",
0064     time_layerclusters = "hgcalLayerClustersHFNose:timeLayerCluster",
0065     pluginPatternRecognitionByCA = dict(
0066         filter_on_categories = [2, 4], # filter muons and charged hadrons
0067         pid_threshold = 0.0,
0068         skip_layers = 1,
0069         min_layers_per_trackster = 5,
0070         oneTracksterPerTrackSeed = True,
0071         promoteEmptyRegionToTrackster = True
0072     ),
0073     itername = "Trkn"
0074 )
0075
0076 ticl_v5.toModify(ticlTrackstersHFNoseTrk.pluginPatternRecognitionByCA, computeLocalTime = cms.bool(True))
0077
0078 ticlHFNoseTrkStepTask = cms.Task(ticlSeedingTrkHFNose
0079     ,filteredLayerClustersHFNoseTrk
0080     ,ticlTrackstersHFNoseTrk)
0081
0082