PHTrackSetMerging.h

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

#ifndef TRACKRECO_PHTRACKSETMERGING_H
#define TRACKRECO_PHTRACKSETMERGING_H

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

// STL includes
#include <string>

// forward declarations
class PHCompositeNode;

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

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

  int Init(PHCompositeNode *topNode) override;
  int InitRun(PHCompositeNode *topNode) override;
  int process_event(PHCompositeNode *topNode) override;
  int End(PHCompositeNode *topNode) override;
  int Setup(PHCompositeNode *topNode);
  void set_track_map_name_in1(const std::string &map_name) { _track_map_name_in1 = map_name; }
  void set_track_map_name_in2(const std::string &map_name) { _track_map_name_in2 = map_name; }
  void set_track_map_name_out(const std::string &map_name) { _track_map_name_out = map_name; }

 protected:
   //  virtual

  virtual int Process(PHCompositeNode *topNode) = 0;

    // virtual int End() = 0;

  //SvtxClusterMap *_cluster_map;
  TrkrClusterContainer *_cluster_map = nullptr;
  SvtxVertexMap *_vertex_map = nullptr;
  SvtxTrackMap *_track_map_in1 = nullptr;
  SvtxTrackMap *_track_map_in2 = nullptr;
  SvtxTrackMap *_track_map_out = nullptr;
  AssocInfoContainer *_assoc_container = nullptr;

  std::string _track_map_name_in1 = "SvtxTrackMap1";
  std::string _track_map_name_in2 = "SvtxTrackMap2";
  std::string _track_map_name_out = "SvtxTrackMapMerged";

 private:
  int CreateNodes(PHCompositeNode *topNode);

  int GetNodes(PHCompositeNode *topNode);
};

#endif