add PV from reco to CbmKFPF

reco/KF/CbmKFParticleFinder.cxx

@@ -37,7 +37,7 @@ CbmKFParticleFinder::CbmKFParticleFinder(const char* name, Int_t iVerbose)
   , fTrackArray(0)
   , fEvents(0)
   , fTopoReconstructor(0)
-  , fPVFindMode(2)
+  , fPVFindMode(3)
   , fPID(0)
   , fSuperEventAnalysis(0)
   , fSETracks(0)
@@ -116,6 +116,8 @@ InitStatus CbmKFParticleFinder::Init() {
       }
     }
   }
+  
+  fCbmPrimVertex = (CbmVertex*) ioman->GetObject("PrimaryVertex.");
 
   return kSUCCESS;
 }
@@ -255,6 +257,12 @@ void CbmKFParticleFinder::Exec(Option_t* /*opt*/) {
       kfVertex.GetRefY() = mcpv[1];
       kfVertex.GetRefZ() = mcpv[2];
     }
+    
+    if(fPVFindMode==3) {
+      kfVertex.GetRefX() = fCbmPrimVertex->GetX();
+      kfVertex.GetRefY() = fCbmPrimVertex->GetY();
+      kfVertex.GetRefZ() = fCbmPrimVertex->GetZ();
+    }
 
     vector<L1FieldRegion> vField, vFieldAtLastPoint;
     fitter.Fit(vRTracks, pdg);
@@ -319,7 +327,7 @@ void CbmKFParticleFinder::Exec(Option_t* /*opt*/) {
 
       eventTopoReconstructor[iEvent].Init(tracks, tracksAtLastPoint);
 
-      if (fPVFindMode == 0) {
+      if (fPVFindMode == 0 || fPVFindMode == 3) {
         KFPVertex primVtx_tmp;
         primVtx_tmp.SetXYZ(
           kfVertex.GetRefX(), kfVertex.GetRefY(), kfVertex.GetRefZ());

reco/KF/CbmKFParticleFinder.h

@@ -17,6 +17,7 @@ class KFParticleFinder;
 class KFPTrackVector;
 class CbmMCEventList;
 class CbmMCDataArray;
+class CbmVertex;
 
 struct KFFieldVector {
   float fField[10];
@@ -32,6 +33,7 @@ public:
   void UseMCPV() { fPVFindMode = 0; }
   void ReconstructSinglePV() { fPVFindMode = 1; }
   void RconstructMultiplePV() { fPVFindMode = 2; }
+  void UseReconstructedPV()   { fPVFindMode = 3; }
 
   void SetStsTrackBranchName(const TString& name) {
     fStsTrackBranchName = name;
@@ -104,6 +106,7 @@ private:
   CbmMCDataArray* fMCTrackArray;     //mc tracks in timeslices
   TClonesArray* fMCTrackArrayEvent;  //mc tracks in event-by-event mode
   CbmMCEventList* fEventList;        //mc event list in timeslice
+  CbmVertex* fCbmPrimVertex;         //mc primary vertex
 
   //topology reconstructor
   KFParticleTopoReconstructor* fTopoReconstructor;