TrackMap.h

#ifndef SIMCORE_TRACKMAP_H_
#define SIMCORE_TRACKMAP_H_
 
// STL
#include <unordered_map>
 
// Geant4
#include "G4Event.hh"
#include "G4Track.hh"
 
// LDMX
#include "SimCore/Event/SimParticle.h"
#include "SimCore/UserPrimaryParticleInformation.h"
#include "SimCore/UserTrackInformation.h"
 
namespace simcore {
 
class TrackMap {
 public:
  void insert(const G4Track* track);
 
  inline bool contains(const G4Track* track) const {
    return ancestry_.find(track->GetTrackID()) != ancestry_.end();
  }
 
  bool isDescendant(int trackID, int ancestorID, int maximum_depth) const;
 
  int findIncident(int trackID) const;
 
  inline bool isSaved(int trackID) const {
    return particle_map_.find(trackID) != particle_map_.end();
  }
 
  void save(const G4Track* track);
 
  void traceAncestry();
 
  void clear();
 
  std::map<int, ldmx::SimParticle>& getParticleMap() { return particle_map_; }
 
 private:
  bool isInCalorimeterRegion(const G4Track* track) const;
 
 private:
  std::unordered_map<int, std::pair<int, bool>> ancestry_;
 
  std::unordered_map<int, std::vector<int>> descendents_;
 
  std::map<int, ldmx::SimParticle> particle_map_;
};
 
}  // namespace simcore
 
#endif