Adding vertex recovery step for 2018 heavy ion miniAOD production.

PhysicsTools/PatAlgos/python/slimming/MicroEventContent_cff.py

@@ -147,6 +147,7 @@
     'keep recoCentrality_hiCentrality_*_*',
     'keep int_centralityBin_*_*',
     'keep recoHFFilterInfo_hiHFfilters_*_*',
+    'keep *_offlineSlimmedPrimaryVerticesRecovery_*_*',
 ]
 from Configuration.Eras.Modifier_pp_on_AA_2018_cff import pp_on_AA_2018
 from Configuration.Eras.Modifier_pp_on_PbPb_run3_cff import pp_on_PbPb_run3

PhysicsTools/PatAlgos/python/slimming/slimming_cff.py

@@ -79,6 +79,11 @@
     slimmingTask, 
     cms.Task(slimmingTask.copy(), packedCandidateMuonID, packedPFCandidateTrackChi2, lostTrackChi2, centralityBin, hiHFfilters))
 
+from RecoHI.HiTracking.miniAODVertexRecovery_cff import offlinePrimaryVerticesRecovery, offlineSlimmedPrimaryVerticesRecovery
+pp_on_AA_2018.toReplaceWith(
+    slimmingTask,
+    cms.Task(slimmingTask.copy(), offlinePrimaryVerticesRecovery, offlineSlimmedPrimaryVerticesRecovery))
+
 from Configuration.Eras.Modifier_phase2_timing_cff import phase2_timing
 _phase2_timing_slimmingTask = cms.Task(slimmingTask.copy(),
                                        offlineSlimmedPrimaryVertices4D)

RecoHI/HiTracking/python/miniAODVertexRecovery_cff.py

@@ -0,0 +1,33 @@
+import FWCore.ParameterSet.Config as cms
+
+from RecoVertex.PrimaryVertexProducer.OfflinePrimaryVertices_cfi import offlinePrimaryVertices
+
+offlinePrimaryVerticesRecovery = offlinePrimaryVertices.clone(
+
+  isRecoveryIteration = True,
+  recoveryVtxCollection = "offlinePrimaryVertices",
+
+  TkFilterParameters = dict(
+        maxNormalizedChi2 = 999.0,
+        minPixelLayersWithHits= 0,
+        minSiliconLayersWithHits = 0,
+        maxD0Significance = 999.0, 
+        minPt = 0.0,
+        maxEta = 999.0,
+        trackQuality = "any"
+  ),
+
+  TkClusParameters = dict(
+        algorithm = "gap",
+        TkGapClusParameters = dict(
+            zSeparation = 1.0        
+        )
+  ),
+
+  vertexCollections = [offlinePrimaryVertices.vertexCollections[0].clone()]
+)    
+
+from PhysicsTools.PatAlgos.slimming.offlineSlimmedPrimaryVertices_cfi import offlineSlimmedPrimaryVertices
+offlineSlimmedPrimaryVerticesRecovery = cms.EDProducer("PATVertexSlimmer",
+  src = cms.InputTag("offlinePrimaryVerticesRecovery"),
+)

RecoVertex/PrimaryVertexProducer/interface/PrimaryVertexProducer.h

@@ -82,6 +82,9 @@ class PrimaryVertexProducer : public edm::stream::EDProducer<> {
   edm::ParameterSet theConfig;
   bool fVerbose;
 
+  bool fRecoveryIteration;
+  edm::EDGetTokenT<reco::VertexCollection> recoveryVtxToken;
+
   edm::EDGetTokenT<reco::BeamSpot> bsToken;
   edm::EDGetTokenT<reco::TrackCollection> trkToken;
   edm::EDGetTokenT<edm::ValueMap<float> > trkTimesToken;

RecoVertex/PrimaryVertexProducer/plugins/PrimaryVertexProducer.cc

@@ -115,6 +115,13 @@ PrimaryVertexProducer::PrimaryVertexProducer(const edm::ParameterSet& conf) : th
     algorithms.push_back(algorithm);
     produces<reco::VertexCollection>(algorithm.label);
   }
+
+  //check if this is a recovery iteration
+  fRecoveryIteration = conf.getParameter<bool>("isRecoveryIteration");
+  if (fRecoveryIteration) {
+    assert(algorithms.size() == 1);
+    recoveryVtxToken = consumes<reco::VertexCollection>(conf.getParameter<edm::InputTag>("recoveryVtxCollection"));
+  }
 }
 
 PrimaryVertexProducer::~PrimaryVertexProducer() {
@@ -149,6 +156,22 @@ void PrimaryVertexProducer::produce(edm::Event& iEvent, const edm::EventSetup& i
     edm::LogError("UnusableBeamSpot") << "Beamspot with invalid errors " << beamVertexState.error().matrix();
   }
 
+  //if this is a recovery iteration, check if we already have a valid PV
+  if (fRecoveryIteration) {
+    auto const& oldVertices = iEvent.get(recoveryVtxToken);
+    //look for the first valid (not-BeamSpot) vertex
+    for (auto const& old : oldVertices) {
+      if (!(old.isFake())) {
+        //found a valid vertex, write the first one to the collection and return
+        //otherwise continue with regular vertexing procedure
+        auto result = std::make_unique<reco::VertexCollection>();
+        result->push_back(old);
+        iEvent.put(std::move(result), algorithms.begin()->label);
+        return;
+      }
+    }
+  }
+
   // get RECO tracks from the event
   // `tks` can be used as a ptr to a reco::TrackCollection
   edm::Handle<reco::TrackCollection> tks;
@@ -414,6 +437,8 @@ void PrimaryVertexProducer::fillDescriptions(edm::ConfigurationDescriptions& des
     psd0.add<std::string>("algorithm", "DA_vect");
     desc.add<edm::ParameterSetDescription>("TkClusParameters", psd0);
   }
+  desc.add<bool>("isRecoveryIteration", false);
+  desc.add<edm::InputTag>("recoveryVtxCollection", {""});
 
   descriptions.add("primaryVertexProducer", desc);
 }

RecoVertex/PrimaryVertexProducer/python/OfflinePrimaryVertices_cfi.py

@@ -40,8 +40,10 @@
                maxDistanceToBeam = cms.double(1.0),
                )
       ]
-    )
-
+    ),
+
+    isRecoveryIteration = cms.bool(False),
+    recoveryVtxCollection = cms.InputTag("")
 
 
 )