ReconstructedParticleSource.h

Go to the documentation of this file.

#ifndef ANALYZERS_SOURCE_RECONSTRUCTED_PARTICLE_H
#define ANALYZERS_SOURCE_RECONSTRUCTED_PARTICLE_H
 
// ROOT
#include "ROOT/RVec.hxx"
 
// EDM4hep
#include "edm4hep/RecoMCParticleLinkCollection.h"
#include "edm4hep/ReconstructedParticleCollection.h"
 
namespace FCCAnalyses ::PodioSource ::ReconstructedParticle {
// --------------------  Selectors  -----------------------------------------

struct selPDG {
  const int m_pdg;
  explicit selPDG(const int pdgID);
  edm4hep::ReconstructedParticleCollection
  operator()(const edm4hep::RecoMCParticleLinkCollection &inLinkColl);
};

struct selAbsPDG {
  const int m_absPdg;
  explicit selAbsPDG(const int pdgID);
  edm4hep::ReconstructedParticleCollection
  operator()(const edm4hep::RecoMCParticleLinkCollection &inLinkColl);
};

struct selPt {
  const float m_minPt;
  explicit selPt(float minPt);
  edm4hep::ReconstructedParticleCollection
  operator()(const edm4hep::ReconstructedParticleCollection &inColl);
};

struct selUpTo {
  const size_t m_size;
  explicit selUpTo(const size_t size);
  edm4hep::ReconstructedParticleCollection
  operator()(const edm4hep::ReconstructedParticleCollection &inColl);
};

struct selGenStatus {
  const int m_status;
  explicit selGenStatus(const int status);
  edm4hep::ReconstructedParticleCollection
  operator()(const edm4hep::RecoMCParticleLinkCollection &inLinkColl);
};
 
// --------------------  Getters  -------------------------------------------

ROOT::VecOps::RVec<float>
getP(const edm4hep::ReconstructedParticleCollection &inColl);

ROOT::VecOps::RVec<float>
getPt(const edm4hep::ReconstructedParticleCollection &inColl);

ROOT::VecOps::RVec<float>
getY(const edm4hep::ReconstructedParticleCollection &inColl);

ROOT::VecOps::RVec<float>
getE(const edm4hep::ReconstructedParticleCollection &inColl);

ROOT::VecOps::RVec<float>
getMass(const edm4hep::ReconstructedParticleCollection &inColl);

ROOT::VecOps::RVec<float>
getCharge(const edm4hep::ReconstructedParticleCollection &inColl);
 
// --------------------  Transformers  --------------------------------------

edm4hep::ReconstructedParticleCollection
sortByPt(const edm4hep::ReconstructedParticleCollection &inColl);

edm4hep::ReconstructedParticleCollection
remove(const edm4hep::ReconstructedParticleCollection &inColl,
       const edm4hep::ReconstructedParticle &inPartToBeRemoved,
       const bool matching = false);

edm4hep::ReconstructedParticleCollection
remove(const edm4hep::ReconstructedParticleCollection &inColl,
       const edm4hep::ReconstructedParticleCollection &inPartsToBeRemoved,
       const bool matching = false);

edm4hep::ReconstructedParticleCollection
merge(const edm4hep::ReconstructedParticleCollection &inColl1,
      const edm4hep::ReconstructedParticleCollection &inColl2);

struct resonanceBuilder {
  const float m_resonanceMass;
  explicit resonanceBuilder(float resonanceMass);
  edm4hep::ReconstructedParticleCollection
  operator()(const edm4hep::ReconstructedParticleCollection &inColl);
};

struct recoilBuilder {
  const float m_sqrts;
  explicit recoilBuilder(float sqrts);
  edm4hep::ReconstructedParticleCollection
  operator()(const edm4hep::ReconstructedParticleCollection &inColl);
};
 
} // namespace FCCAnalyses::PodioSource::ReconstructedParticle
 
// namespace recoParticle = FCCAnalyses::PodioSource::ReconstructedParticle;
 
#endif /* ANALYZERS_SOURCE_RECONSTRUCTED_PARTICLE_H */