PHInitVertexing.h

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

#ifndef TRACKRECO_PHINITVERTEXING_H
#define TRACKRECO_PHINITVERTEXING_H

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

// STL includes
#include <string>

// forward declarations
class PHCompositeNode;

class TrkrClusterContainer;
class TrkrHitSetContainer;
class SvtxVertexMap;

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

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

 protected:

  virtual int Setup(PHCompositeNode *topNode);

  virtual int Process(PHCompositeNode *topNode) = 0;


  TrkrClusterContainer *_cluster_map;
  TrkrHitSetContainer  *_hitsets;
  SvtxVertexMap *_vertex_map;

 private:
  int CreateNodes(PHCompositeNode *topNode);

  int GetNodes(PHCompositeNode *topNode);
};

#endif