PHTrackSeeding.h

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

#ifndef TRACKRECO_PHTRACKSEEDING_H
#define TRACKRECO_PHTRACKSEEDING_H

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

// STL includes
#include <string>

// forward declarations
class PHCompositeNode;

//class SvtxClusterMap;
class TrkrClusterContainer;
class TrkrClusterHitAssoc;
class TrkrHitSetContainer;
class TrkrClusterIterationMapv1;
class SvtxVertexMap;
class TrackSeedContainer;


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

  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 set_do_hit_association(bool do_assoc){do_hit_assoc = do_assoc;}
  void SetUseTruthClusters(bool setit){_use_truth_clusters = setit;}
  void SetIteration(int iter){_n_iteration = iter;}
 protected:
  virtual int Setup(PHCompositeNode *topNode);

  virtual int Process(PHCompositeNode *topNode) = 0;

  virtual int End() = 0;

  TrkrClusterContainer *_cluster_map = nullptr;
  TrkrClusterHitAssoc *_cluster_hit_map = nullptr;
  TrkrClusterIterationMapv1* _iteration_map;
  int _n_iteration;
  bool do_hit_assoc = true;
  SvtxVertexMap *_vertex_map = nullptr;
  TrackSeedContainer *_track_map = nullptr;
  TrkrHitSetContainer  *_hitsets = nullptr;

  std::string _track_map_name = "TpcTrackSeedContainer";

  bool _use_truth_clusters = false;

 private:

  int CreateNodes(PHCompositeNode *topNode);

  int GetNodes(PHCompositeNode *topNode);
};

#endif