SVertexer: use mean vertex

Signed-off-by: Felix Schlepper <f3sch.git@outlook.com>

Author: f3sch

Detectors/GlobalTrackingWorkflow/src/SecondaryVertexingSpec.cxx

@@ -12,6 +12,8 @@
 /// @file  SecondaryVertexingSpec.cxx
 
 #include <vector>
+#include "DataFormatsCalibration/MeanVertexObject.h"
+#include "Framework/CCDBParamSpec.h"
 #include "ReconstructionDataFormats/Decay3Body.h"
 #include "DataFormatsGlobalTracking/RecoContainer.h"
 #include "ReconstructionDataFormats/TrackTPCITS.h"
@@ -135,6 +137,11 @@ void SecondaryVertexingSpec::finaliseCCDB(ConcreteDataMatcher& matcher, void* ob
     mStrTracker.setClusterDictionary((const o2::itsmft::TopologyDictionary*)obj);
     return;
   }
+  if (matcher == ConcreteDataMatcher("GLO", "MEANVERTEX", 0)) {
+    LOG(info) << "Imposing new MeanVertex: " << ((const o2::dataformats::MeanVertexObject*)obj)->asString();
+    mVertexer.setMeanVertex((const o2::dataformats::MeanVertexObject*)obj);
+    return;
+  }
 }
 
 void SecondaryVertexingSpec::updateTimeDependentParams(ProcessingContext& pc)
@@ -179,6 +186,8 @@ void SecondaryVertexingSpec::updateTimeDependentParams(ProcessingContext& pc)
       mStrTracker.setupFitters();
     }
   }
+
+  pc.inputs().get<o2::dataformats::MeanVertexObject*>("meanvtx");
 }
 
 DataProcessorSpec getSecondaryVertexingSpec(GTrackID::mask_t src, bool enableCasc, bool enable3body, bool enableStrangenesTracking, bool useMC)
@@ -195,6 +204,7 @@ DataProcessorSpec getSecondaryVertexingSpec(GTrackID::mask_t src, bool enableCas
   }
   dataRequest->requestTracks(src, useMC);
   dataRequest->requestPrimaryVertertices(useMC);
+  dataRequest->inputs.emplace_back("meanvtx", "GLO", "MEANVERTEX", 0, Lifetime::Condition, ccdbParamSpec("GLO/Calib/MeanVertex", {}, 1));
   auto ggRequest = std::make_shared<o2::base::GRPGeomRequest>(false,                                                                                         // orbitResetTime
                                                               true,                                                                                          // GRPECS=true
                                                               false,                                                                                         // GRPLHCIF

Detectors/Vertexing/include/DetectorsVertexing/SVertexer.h

@@ -16,6 +16,7 @@
 #define O2_S_VERTEXER_H
 
 #include "gsl/span"
+#include "DataFormatsCalibration/MeanVertexObject.h"
 #include "DataFormatsGlobalTracking/RecoContainer.h"
 #include "ReconstructionDataFormats/PrimaryVertex.h"
 #include "ReconstructionDataFormats/V0.h"
@@ -49,8 +50,6 @@ class CorrectionMapsHelper;
 namespace vertexing
 {
 
-namespace o2d = o2::dataformats;
-
 class SVertexer
 {
  public:
@@ -112,7 +111,14 @@ class SVertexer
   int getN3Bodies() const { return mN3Bodies; }
   int getNStrangeTracks() const { return mNStrangeTracks; }
   auto& getMeanVertex() const { return mMeanVertex; }
-  void setMeanVertex(const o2d::VertexBase& v) { mMeanVertex = v; }
+  void setMeanVertex(const o2::dataformats::MeanVertexObject* v)
+  {
+    if (v == nullptr) {
+      return;
+    }
+    mMeanVertex = v->getMeanVertex();
+  }
+
   void setNThreads(int n);
   int getNThreads() const { return mNThreads; }
   void setUseMC(bool v) { mUseMC = v; }
@@ -161,7 +167,7 @@ class SVertexer
   std::array<std::vector<TrackCand>, 2> mTracksPool{}; // pools of positive and negative seeds sorted in min VtxID
   std::array<std::vector<int>, 2> mVtxFirstTrack{};    // 1st pos. and neg. track of the pools for each vertex
 
-  o2d::VertexBase mMeanVertex{{0., 0., 0.}, {0.1 * 0.1, 0., 0.1 * 0.1, 0., 0., 6. * 6.}};
+  o2::dataformats::VertexBase mMeanVertex{{0., 0., 0.}, {0.1 * 0.1, 0., 0.1 * 0.1, 0., 0., 6. * 6.}};
   const SVertexerParams* mSVParams = nullptr;
   std::array<SVertexHypothesis, NHypV0> mV0Hyps;
   std::array<SVertexHypothesis, NHypCascade> mCascHyps;