LinkSource.h

Go to the documentation of this file.

#ifndef SOURCE_LINK_ANALYZERS_H
#define SOURCE_LINK_ANALYZERS_H
 
// EDM4hep
#include "edm4hep/RecoMCParticleLinkCollection.h"
 
namespace FCCAnalyses ::PodioSource ::Link {
struct selPDG {
  const int m_pdg;
 
  explicit selPDG(const int pdg) : m_pdg(pdg){};
 
  template <typename T> T operator()(const T &inLinkColl) {
    T result;
    result.setSubsetCollection();
 
    for (const auto &link : inLinkColl) {
      const auto &particle = link.getTo();
      if (particle.getPDG() == m_pdg) {
        result.push_back(link);
      }
    }
 
    return result;
  }
};
 
struct selAbsPDG {
  const int m_pdg;
 
  explicit selAbsPDG(const int pdg) : m_pdg(pdg) {
    if (m_pdg < 0) {
      throw std::invalid_argument("FCCAnalyses::PodioSource::Link::sel_absPDG: "
                                  "Received negative value!");
    }
  };
 
  template <typename T> auto operator()(const T &inLinkColl) {
    T result;
    result.setSubsetCollection();
 
    for (const auto &link : inLinkColl) {
      const auto &particle = link.getTo();
      if (std::abs(particle.getPDG()) == m_pdg) {
        result.push_back(link);
      }
    }
 
    return result;
  };
};
 
struct selGenStatus {
  const int m_status;
 
  explicit selGenStatus(const int status) : m_status(status){};
 
  template <typename T> T operator()(const T &inLinkColl) {
    T result;
    result.setSubsetCollection();
 
    for (const auto &link : inLinkColl) {
      const auto &particle = link.getTo();
      if (particle.getGeneratorStatus() == m_status) {
        result.push_back(link);
      }
    }
 
    return result;
  }
};
} // namespace FCCAnalyses::PodioSource::Link
 
#endif /* SOURCE_LINK_ANALYZERS_H */