PH3DVertexing.h

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

#ifndef TRACKRECO_PH3DVERTEXING_H
#define TRACKRECO_PH3DVERTEXING_H

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

// STL includes
#include <set>
#include <string>

// forward declarations
class PHCompositeNode;

//class SvtxClusterMap;
//class TrkrClusterContainer;

class SvtxVertexMap;
class SvtxTrackMap;
class AssocInfoContainer;

class PH3DVertexing : public SubsysReco
{
 public:
  PH3DVertexing(const std::string &name = "PH3DVertexing");
  ~PH3DVertexing() 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;

 protected:
  virtual int Setup(PHCompositeNode *topNode);

  virtual int Process() = 0;

  //SvtxClusterMap *_cluster_map;
  // TrkrClusterContainer *_cluster_map;

  SvtxVertexMap *_vertex_map;
  SvtxTrackMap *_track_map;
  AssocInfoContainer *_assoc_container;

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

#endif