crashing version of jetCoreDirectSeedGenerator

cms-sw · Nov 5, 2020 · 157c1e9 · 157c1e9
1 parent 3c9bd1b
commit 157c1e9
Show file tree

Hide file tree

Showing 7 changed files with 856 additions and 2 deletions.
diff --git a/RecoTracker/IterativeTracking/python/JetCoreRegionalStep_cff.py b/RecoTracker/IterativeTracking/python/JetCoreRegionalStep_cff.py
@@ -121,9 +121,15 @@
     maxPtForLooperReconstruction = cms.double(0.7)
 )
 
+import RecoTracker.TkSeedGenerator.jetCoreDirectSeedGenerator_cfi
+jetCoreSeeds  = RecoTracker.TkSeedGenerator.jetCoreDirectSeedGenerator_cfi.jetCoreDirectSeedGenerator.clone(
+ vertices="firstStepPrimaryVertices"
+)
+
 # MAKING OF TRACK CANDIDATES
 import RecoTracker.CkfPattern.CkfTrackCandidates_cfi
 jetCoreRegionalStepTrackCandidates = RecoTracker.CkfPattern.CkfTrackCandidates_cfi.ckfTrackCandidates.clone(
+    # src                    = 'jetCoreSeeds',
     src                    = 'jetCoreRegionalStepSeeds',
     maxSeedsBeforeCleaning = 10000,
     TrajectoryBuilderPSet  = cms.PSet( refToPSet_ = cms.string('jetCoreRegionalStepTrajectoryBuilder')),
@@ -208,18 +214,19 @@
 fastSim.toModify(jetCoreRegionalStep,vertices = 'firstStepPrimaryVerticesBeforeMixing')
 
 # Final sequence
-JetCoreRegionalStepTask = cms.Task(jetsForCoreTracking,                 
+JetCoreRegionalStepTask = cms.Task(jetsForCoreTracking,
                                    firstStepGoodPrimaryVertices,
                                    #jetCoreRegionalStepClusters,
                                    jetCoreRegionalStepSeedLayers,
                                    jetCoreRegionalStepTrackingRegions,
                                    jetCoreRegionalStepHitDoublets,
                                    jetCoreRegionalStepSeeds,
+                                   jetCoreSeeds,
                                    jetCoreRegionalStepTrackCandidates,
                                    jetCoreRegionalStepTracks,
 #                                   jetCoreRegionalStepClassifier1,jetCoreRegionalStepClassifier2,
                                    jetCoreRegionalStep)
 JetCoreRegionalStep = cms.Sequence(JetCoreRegionalStepTask)
-fastSim.toReplaceWith(JetCoreRegionalStepTask, 
+fastSim.toReplaceWith(JetCoreRegionalStepTask,
                       cms.Task(jetCoreRegionalStepTracks,
                                    jetCoreRegionalStep))
diff --git a/RecoTracker/TkSeedGenerator/data/JetCoreDirectSeedGenerator_TrainedModel.pb b/RecoTracker/TkSeedGenerator/data/JetCoreDirectSeedGenerator_TrainedModel.pb
diff --git a/RecoTracker/TkSeedGenerator/plugins/BuildFile.xml b/RecoTracker/TkSeedGenerator/plugins/BuildFile.xml
@@ -5,6 +5,7 @@
 <use name="RecoPixelVertexing/PixelTriplets"/>
 <use name="RecoPixelVertexing/PixelTrackFitting"/>
 <use name="RecoPixelVertexing/PixelLowPtUtilities"/>
+<use name="PhysicsTools/TensorFlow"/>
 <library file="*.cc" name="RecoTrackerTkSeedGeneratorPlugins">
   <flags EDM_PLUGIN="1"/>
   <use name="DataFormats/TrackerRecHit2D"/>