PHTrackPropagating.h

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

#ifndef TRACKRECO_PHTRACKPROPAGATING_H
#define TRACKRECO_PHTRACKPROPAGATING_H

// PHENIX includes
#include <fun4all/SubsysReco.h>

// STL includes
#include <string>

// forward declarations
class PHCompositeNode;

class TrkrClusterContainer;
class SvtxVertexMap;
class SvtxTrackMap;

class PHTrackPropagating : public SubsysReco
{
 public:
  PHTrackPropagating(const std::string &name = "PHTrackPropagating");
  ~PHTrackPropagating() override = default;

  int InitRun(PHCompositeNode *topNode) override;
  int process_event(PHCompositeNode *topNode) override;
  int End(PHCompositeNode *topNode) override;
  void set_track_map_name(const std::string &map_name) { _track_map_name = map_name; }
  void SetUseTruthClusters(bool setit){_use_truth_clusters = setit;}

 protected:
  virtual int Setup(PHCompositeNode *topNode);

  virtual int Process() = 0;

  virtual int End() = 0;


  //SvtxClusterMap *_cluster_map;
  TrkrClusterContainer *_cluster_map = nullptr;
  SvtxVertexMap *_vertex_map = nullptr;
  SvtxTrackMap *_track_map = nullptr;

  std::string _track_map_name = "SvtxTrackMap";

  bool _use_truth_clusters = false;

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

#endif