TruthNodeMaker.h

Go to the documentation of this file.

#ifndef _TRUTH_NODE_MAKER__H_
#define _TRUTH_NODE_MAKER__H_
#include <fun4all/SubsysReco.h>
#include <map>
#include <string>

class TVector3;
class TLorentzVector;
namespace HepMC { 
  class GenParticle; 
};
class PHHepMCGenEventMap;
class PHG4TruthInfoContainer;
class PHG4HitContainer;
class SQEvent;
class SQMCEvent;
class SQTrack;
class SQTrackVector;
class SQDimuonVector;
class SQHitVector;
class SRecEvent;
class SQHit;
class GeomSvc;


class TruthNodeMaker: public SubsysReco {
  // input nodes
  PHHepMCGenEventMap* genevtmap;
  PHG4TruthInfoContainer* g4true;
  SQHitVector*      m_vec_hit;
  SRecEvent*        m_rec_evt;
  SQTrackVector*    m_vec_rec_trk;
  std::map<int, PHG4HitContainer*> m_g4hc;

  // output nodes
  SQEvent*        m_evt;
  SQMCEvent*      m_mcevt;
  SQTrackVector*  m_vec_trk;
  SQDimuonVector* m_vec_dim;

  // rec input container type
  bool m_legacy_rec_container;
  bool m_do_truthtrk_tagging;
  bool m_do_evt_header;

  // true - rec mathcing nhits threshold
  double m_matching_threshold;

 public:
  TruthNodeMaker();
  virtual ~TruthNodeMaker();
  int Init(PHCompositeNode *topNode);
  int InitRun(PHCompositeNode *topNode);
  int process_event(PHCompositeNode *topNode);
  int End(PHCompositeNode *topNode);

  void set_legacy_rec_container(bool b = true) { m_legacy_rec_container = b; }
  void set_matching_threshold(double threshold) { m_matching_threshold = threshold; }

 private:
  int  GetNodes(PHCompositeNode *topNode);
  int MakeNodes(PHCompositeNode *topNode);

  bool FindHitAtStation(int target_detIDs[], int trkid, const std::vector<SQHit*>& hitvec, TVector3& pos, TLorentzVector& mom);
  int  FindCommonHitIDs(std::vector<int>& hitidvec1, std::vector<int>& hitidvec2);
};

#endif /* _TRUTH_NODE_MAKER__H_ */