PHTrackFitting.h

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

#ifndef TRACKRECO_PHTRACKFITTING_H
#define TRACKRECO_PHTRACKFITTING_H

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

// STL includes
#include <string>

// forward declarations
class PHCompositeNode;

//class SvtxClusterMap;
class TrkrClusterContainer;
class TrkrHitSetContainer;
class SvtxVertexMap;
class SvtxTrackMap;
class AssocInfoContainer;

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

  int Init(PHCompositeNode *topNode) override;
  int InitRun(PHCompositeNode *topNode) override;
  int process_event(PHCompositeNode *topNode) override;

  //virtual const std::set<unsigned int> &get_seeding_layers() const = 0;

  //virtual void set_seeding_layers(const unsigned int a[], const unsigned int n) = 0;

  //void set_track_map_name(const std::string &map_name) { _track_map_name = map_name; }

 protected:
  virtual int Setup(PHCompositeNode *topNode);

  virtual int Process() = 0;

  //SvtxClusterMap *_cluster_map;
  TrkrClusterContainer *_cluster_map;
  TrkrHitSetContainer  *_hitsets = nullptr;
  SvtxVertexMap *_vertex_map;
  SvtxTrackMap *_track_map;
  AssocInfoContainer *_assoc_container;

  std::string _track_map_name;

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

#endif