Vtx: make class printable

Signed-off-by: Felix Schlepper <felix.schlepper@cern.ch>

Author: f3sch

DataFormats/Reconstruction/include/ReconstructionDataFormats/Vertex.h

@@ -22,6 +22,8 @@
 #include <string>
 #include <type_traits>
 #include <array>
+#include <sstream>
+#include <format>
 #endif
 
 namespace o2
@@ -135,6 +137,11 @@ class Vertex : public VertexBase
   {
   }
 
+#if !defined(GPUCA_NO_FMT) && !defined(GPUCA_GPUCODE_DEVICE)
+  void print() const;
+  std::string asString() const;
+#endif
+
   GPUd() ushort getNContributors() const { return mNContributors; }
   GPUd() void setNContributors(ushort v) { mNContributors = v; }
   GPUd() void addContributor() { mNContributors++; }
@@ -162,6 +169,39 @@ class Vertex : public VertexBase
 
 #if !defined(GPUCA_GPUCODE_DEVICE) && !defined(GPUCA_NO_FMT)
 std::ostream& operator<<(std::ostream& os, const o2::dataformats::VertexBase& v);
+
+// small helper concept
+template <typename T>
+concept Streamable = requires(std::ostream& os, T a) {
+  { os << a } -> std::same_as<std::ostream&>;
+};
+
+template <typename Stamp>
+inline std::string Vertex<Stamp>::asString() const
+{
+  if constexpr (Streamable<Stamp>) {
+    std::ostringstream oss;
+    oss << mTimeStamp;
+    return std::format("{} NContrib:{} Chi2:{:.2f} Flags:{:b} Stamp:{}", VertexBase::asString(), mNContributors, mChi2, mBits, oss.str());
+  } else if constexpr (requires { mTimeStamp.getTimeStamp(); }) {
+    return std::format("{} NContrib:{} Chi2:{:.2f} Flags:{:b} Stamp:{}", VertexBase::asString(), mNContributors, mChi2, mBits, mTimeStamp.getTimeStamp());
+  } else {
+    return std::format("{} NContrib:{} Chi2:{:.2f} Flags:{:b} Stamp:X", VertexBase::asString(), mNContributors, mChi2, mBits);
+  }
+}
+
+template <typename Stamp>
+inline std::ostream& operator<<(std::ostream& os, const o2::dataformats::Vertex<Stamp>& v)
+{
+  os << v.asString();
+  return os;
+}
+
+template <typename Stamp>
+inline void Vertex<Stamp>::print() const
+{
+  std::cout << *this << '\n';
+}
 #endif
 
 } // namespace dataformats

DataFormats/Reconstruction/src/Vertex.cxx

@@ -10,9 +10,9 @@
 // or submit itself to any jurisdiction.
 
 #include "ReconstructionDataFormats/Vertex.h"
-#include <iostream>
 #ifndef GPUCA_NO_FMT
-#include <fmt/printf.h>
+#include <iostream>
+#include <format>
 #endif
 
 namespace o2
@@ -24,7 +24,7 @@ namespace dataformats
 #ifndef GPUCA_NO_FMT
 std::string VertexBase::asString() const
 {
-  return fmt::format("Vtx {{{:+.4e},{:+.4e},{:+.4e}}} Cov.:{{{{{:.3e}..}},{{{:.3e},{:.3e}..}},{{{:.3e},{:.3e},{:.3e}}}}}",
+  return std::format("Vtx {{{:+.4e},{:+.4e},{:+.4e}}} Cov.:{{{{{:.3e}..}},{{{:.3e},{:.3e}..}},{{{:.3e},{:.3e},{:.3e}}}}}",
                      mPos.X(), mPos.Y(), mPos.Z(), mCov[0], mCov[1], mCov[2], mCov[3], mCov[4], mCov[5]);
 }