MvtxPrototype2Clusterizer.h

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

// Tell emacs that this is a C++ source
//  -*- C++ -*-.
#ifndef MVTX_P2_CLUSTERIZER_H
#define MVTX_P2_CLUSTERIZER_H

#include <fun4all/SubsysReco.h>
#include <utility>
#include <trackbase/TrkrDefs.h>

class TrkrHit;
class TrkrHitSetContainer;
class TrkrClusterContainer;
class TrkrClusterHitAssoc;

class MvtxPrototype2Clusterizer : public SubsysReco
{
 public:
  typedef std::pair<unsigned int, unsigned int> pixel;

  MvtxPrototype2Clusterizer(const std::string &name = "MvtxPrototype2Clusterizer");
  virtual ~MvtxPrototype2Clusterizer() {}

  int Init(PHCompositeNode *topNode) { return 0; }

  int InitRun(PHCompositeNode *topNode);

  int process_event(PHCompositeNode *topNode);

  int End(PHCompositeNode *topNode) { return 0; }

  void SetZClustering(const bool make_z_clustering)
  {
    m_makeZClustering = make_z_clustering;
  }
  bool GetZClustering() const
  {
    return m_makeZClustering;
  }

 private:
  //bool are_adjacent(const pixel lhs, const pixel rhs);
  bool are_adjacent(const std::pair<TrkrDefs::hitkey, TrkrHit*> &lhs, const std::pair<TrkrDefs::hitkey, TrkrHit*> &rhs);

  void ClusterMvtx(PHCompositeNode *topNode);

  void PrintClusters(PHCompositeNode *topNode);

  // node tree storage pointers
  TrkrHitSetContainer *m_hits;
  TrkrClusterContainer *m_clusterlist;
  TrkrClusterHitAssoc *m_clusterhitassoc;

  // settings
  bool m_makeZClustering;  // z_clustering_option
};

#endif  // MVTX_MVTXCLUSTERIZER_H