Add asString getter to MCCompLabel

shahor02 · shahor02 · commit d23bf266e442 · 2024-01-29T11:35:34.000+01:00
diff --git a/DataFormats/simulation/include/SimulationDataFormat/MCCompLabel.h b/DataFormats/simulation/include/SimulationDataFormat/MCCompLabel.h
@@ -14,6 +14,7 @@
 
 #include <cstdint>
 #include "GPUCommonRtypes.h"
+#include <string>
 
 namespace o2
 {
@@ -138,6 +139,7 @@ class MCCompLabel
   }
 
   void print() const;
+  std::string asString() const;
 
   static constexpr int maxSourceID() { return maskSrcID; }
   static constexpr int maxEventID() { return maskEvID; }
diff --git a/DataFormats/simulation/src/MCCompLabel.cxx b/DataFormats/simulation/src/MCCompLabel.cxx
@@ -14,6 +14,7 @@
 #include <ios>
 #include <iostream>
 #include <cassert>
+#include <fmt/format.h>
 
 namespace o2
 {
@@ -31,6 +32,16 @@ std::ostream& operator<<(std::ostream& os, MCCompLabel const& c)
   return os;
 }
 
+//_____________________________________________
+std::string MCCompLabel::asString() const
+{
+  // describe itself
+  if (isValid()) {
+    return fmt::format("[{}/{}{}/{:6d}]", getSourceID(), getEventID(), isFake() ? '-' : '+', getTrackID());
+  }
+  return isNoise() ? "[noise]" : "[unset]";
+}
+
 //_____________________________________________
 void MCCompLabel::print() const
 {

Original file line number	Diff line number	Diff line change
`@@ -14,6 +14,7 @@`
`14`	`14`
`15`	`15`	`#include <cstdint>`
`16`	`16`	`#include "GPUCommonRtypes.h"`
	`17`	`+#include <string>`
`17`	`18`
`18`	`19`	`namespace o2`
`19`	`20`	`{`
`@@ -138,6 +139,7 @@ class MCCompLabel`
`138`	`139`	`}`
`139`	`140`
`140`	`141`	`void print() const;`
	`142`	`+ std::string asString() const;`
`141`	`143`
`142`	`144`	`static constexpr int maxSourceID() { return maskSrcID; }`
`143`	`145`	`static constexpr int maxEventID() { return maskEvID; }`