SYNC PR - Added new print method and basic analyzers for ParticleID

include/ral/ParticleID.h

@@ -0,0 +1,80 @@
+#ifndef PARTICLEID_ANALYZERS_H
+#define PARTICLEID_ANALYZERS_H
+
+#include "ROOT/RVec.hxx"
+#include "edm4hep/ParticleID.h"
+
+/**
+ * Main namespace of the library
+ */
+namespace k4::ral {
+
+/**
+ * Englobe analyzers that acts on the ParticleID class
+ */
+namespace ParticleID {
+
+/**
+ * Get type member from ParticleIDs
+ *
+ * Analyzer that can be use to obtain the type member in the class
+ * ParticleID from edm4hep
+ *
+ * @param particles List of particle id in an event
+ *
+ */
+ROOT::VecOps::RVec<int>
+get_type(ROOT::VecOps::RVec<edm4hep::ParticleID> particles);
+
+/**
+ * Get PDG member from ParticleIDs
+ *
+ * Analyzer that can be use to obtain the PDG member in the class
+ * ParticleID from edm4hep
+ *
+ * @param particles List of particle id in an event
+ *
+ */
+ROOT::VecOps::RVec<int>
+get_PDG(ROOT::VecOps::RVec<edm4hep::ParticleID> particles);
+
+/**
+ * Get algorithmType member from ParticleIDs
+ *
+ * Analyzer that can be use to obtain the algorithmType member in the class
+ * ParticleID from edm4hep
+ *
+ * @param particles List of particle id in an event
+ *
+ */
+ROOT::VecOps::RVec<int>
+get_algorithmType(ROOT::VecOps::RVec<edm4hep::ParticleID> particles);
+
+/**
+ * Get likelihood member from ParticleIDs
+ *
+ * Analyzer that can be use to obtain the likelihood member in the class
+ * ParticleID from edm4hep
+ *
+ * @param particles List of particle id in an event
+ *
+ */
+ROOT::VecOps::RVec<float>
+get_likelihood(ROOT::VecOps::RVec<edm4hep::ParticleID> particles);
+
+/**
+ * Get parameters member from ParticleIDs
+ *
+ * Analyzer that can be use to obtain the parameters member in the class
+ * ParticleID from edm4hep
+ *
+ * @param particles List of particle id in an event
+ *
+ */
+ROOT::VecOps::RVec<ROOT::VecOps::RVec<float>>
+get_parameters(ROOT::VecOps::RVec<edm4hep::ParticleID> particles);
+
+} // namespace ParticleID
+} // namespace k4::ral
+
+#endif

include/ral/ReconstructedParticle.h

@@ -99,6 +99,14 @@ get_mass(ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> particles);
 ROOT::VecOps::RVec<float> get_goodnessOfPID(
     ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> particles);
 
+struct print_PDG {
+  int m_n_events;
+  int m_n_printed;
+  print_PDG(int n_events);
+  int operator()(
+      ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> particles);
+};
+
 } // namespace ReconstructedParticle
 } // namespace k4::ral
 

src/ParticleID.cc

@@ -0,0 +1,58 @@
+#include "ral/ParticleID.h"
+
+namespace k4::ral {
+
+namespace ParticleID {
+
+ROOT::VecOps::RVec<int>
+get_type(ROOT::VecOps::RVec<edm4hep::ParticleID> particles) {
+  ROOT::VecOps::RVec<int> result;
+  result.reserve(particles.size());
+  for (edm4hep::ParticleID p : particles) {
+    result.emplace_back(p.type);
+  }
+  return result;
+}
+
+ROOT::VecOps::RVec<int>
+get_PDG(ROOT::VecOps::RVec<edm4hep::ParticleID> particles) {
+  ROOT::VecOps::RVec<int> result;
+  result.reserve(particles.size());
+  for (edm4hep::ParticleID p : particles) {
+    result.emplace_back(p.PDG);
+  }
+  return result;
+}
+
+ROOT::VecOps::RVec<int>
+get_algorithmType(ROOT::VecOps::RVec<edm4hep::ParticleID> particles) {
+  ROOT::VecOps::RVec<int> result;
+  result.reserve(particles.size());
+  for (edm4hep::ParticleID p : particles) {
+    result.emplace_back(p.algorithmType);
+  }
+  return result;
+}
+
+ROOT::VecOps::RVec<float>
+get_likelihood(ROOT::VecOps::RVec<edm4hep::ParticleID> particles) {
+  ROOT::VecOps::RVec<float> result;
+  result.reserve(particles.size());
+  for (edm4hep::ParticleID p : particles) {
+    result.emplace_back(p.likelihood);
+  }
+  return result;
+}
+
+ROOT::VecOps::RVec<ROOT::VecOps::RVec<float>>
+get_parameters(ROOT::VecOps::RVec<edm4hep::ParticleID> particles) {
+  ROOT::VecOps::RVec<ROOT::VecOps::RVec<float>> result;
+  result.reserve(particles.size());
+  for (edm4hep::ParticleID p : particles) {
+    result.emplace_back(p.parameters);
+  }
+  return result;
+}
+
+} // namespace ParticleID
+} // namespace k4::ral

src/ReconstructedParticle.cc

@@ -1,4 +1,6 @@
 #include "ral/ReconstructedParticle.h"
+#include <cstdlib>
+#include <iostream>
 
 namespace k4::ral {
 
@@ -75,5 +77,23 @@ ROOT::VecOps::RVec<float> get_goodnessOfPID(
   }
   return result;
 }
+
+print_PDG::print_PDG(int n_events) : m_n_events{n_events}, m_n_printed{0} {}
+int print_PDG::operator()(
+    ROOT::VecOps::RVec<edm4hep::ReconstructedParticleData> particles) {
+  if (m_n_events == m_n_printed) {
+    std::cout << "Finish printing PDG" << std::endl;
+    std::exit(0);
+  }
+  auto pdgs = get_PDG(particles);
+  std::cout << "Printing PDG from event " << m_n_printed << std::endl;
+  for (const int &pdg : pdgs) {
+    std::cout << pdg << " ";
+  }
+  std::cout << std::endl;
+  m_n_printed += 1;
+  return 0;
+}
+
 } // namespace ReconstructedParticle
 } // namespace k4::ral

test/integration/ReconstructedParticle.py

@@ -90,6 +90,8 @@ def get_file(url: str, path: str) -> None:
             "ReconstructedParticle::get_mass(ReconstructedParticles)")
     .Define("goodnessOfPID",
             "ReconstructedParticle::get_goodnessOfPID(ReconstructedParticles)")
+    .Define("void",
+            "ReconstructedParticle::print_PDG(5)(ReconstructedParticles)")
 )
 
 print("Output test result in a new dataframe")