MvtxClusterizer.h

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

#ifndef MVTX_MVTXCLUSTERIZER_H
#define MVTX_MVTXCLUSTERIZER_H

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

#include <string>                // for string
#include <utility>

class ClusHitsVerbose;
class PHCompositeNode;
class TrkrHit;
class TrkrHitSetContainer;
class TrkrClusterContainer;
class TrkrClusterHitAssoc;
class RawHit;
class RawHitSet;
class RawHitSetContainer;

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

  MvtxClusterizer(const std::string &name = "MvtxClusterizer");
  ~MvtxClusterizer() override {}

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

  int InitRun(PHCompositeNode *topNode) override;

  int process_event(PHCompositeNode *topNode) override;

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

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

  void set_do_hit_association(bool do_assoc){do_hit_assoc = do_assoc;}
  void set_read_raw(bool read_raw){ do_read_raw = read_raw;}
  void set_ClusHitsVerbose(bool set=true) { record_ClusHitsVerbose = set; };
  ClusHitsVerbose* mClusHitsVerbose { nullptr };

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

  void ClusterMvtx(PHCompositeNode *topNode);
  void ClusterMvtxRaw(PHCompositeNode *topNode);
  void PrintClusters(PHCompositeNode *topNode);

  // node tree storage pointers
  TrkrHitSetContainer *m_hits;
  RawHitSetContainer *m_rawhits;
  TrkrClusterContainer *m_clusterlist; 

  TrkrClusterHitAssoc *m_clusterhitassoc;

  // settings
  bool m_makeZClustering;  // z_clustering_option
  bool do_hit_assoc = true;
  bool do_read_raw = false;
};

#endif  // MVTX_MVTXCLUSTERIZER_H