ECCEFastPIDReco.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file ECCEFastPIDReco.h

// Tell emacs that this is a C++ source
//  -*- C++ -*-.
#ifndef ECCEFASTPIDRECO_H
#define ECCEFASTPIDRECO_H

#include <fun4all/SubsysReco.h>

#include <eicpidbase/EICPIDDefs.h>

#include <string>

class PHCompositeNode;
class ECCEFastPIDMap;
class SvtxTrackMap;
class EICPIDParticleContainer;
class PHG4TruthInfoContainer;
class PHG4HitContainer;

class ECCEFastPIDReco : public SubsysReco {
public:
  ECCEFastPIDReco(ECCEFastPIDMap *map, EICPIDDefs::PIDDetector det,
                  const std::string &name = "ECCEFastPIDReco");

  virtual ~ECCEFastPIDReco();

  int Init(PHCompositeNode *topNode) override;

  int InitRun(PHCompositeNode *topNode) override;

  int process_event(PHCompositeNode *topNode) override;

  int End(PHCompositeNode *topNode) override;

  void setTrackmapNodeName(const std::string &name) {
    m_TrackmapNodeName = name;
  }

  void setEICPIDParticleMapNodeName(const std::string &name) {
    m_EICPIDParticleMapNodeName = name;
  }

  void setMatchG4Hit(const std::string &g4hit_node_name) {
    m_matchG4Hit = true;
    m_G4HitNodeName = g4hit_node_name;
  }

private:
  bool m_matchG4Hit = false;

  std::string m_G4HitNodeName = "Uninitialized";

  ECCEFastPIDMap *m_pidmap;
  EICPIDDefs::PIDDetector m_PIDDetector;

  std::string m_TrackmapNodeName = "TrackMap";
  std::string m_EICPIDParticleMapNodeName = "EICPIDParticleMap";

  SvtxTrackMap *m_SvtxTrackMap = nullptr;
  EICPIDParticleContainer *m_EICPIDParticleContainer = nullptr;
  PHG4TruthInfoContainer *m_truthInfo = nullptr;
  PHG4HitContainer *m_g4hits = nullptr;
};

#endif // ECCEFASTPIDRECO_H