PHG4InttHitReco.h

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

// Tell emacs that this is a C++ source
// This file is really -*- C++ -*-.
#ifndef G4INTT_PHG4INTTHITRECO_H
#define G4INTT_PHG4INTTHITRECO_H


#include <fun4all/SubsysReco.h>
#include <gsl/gsl_vector.h>  // for gsl_vector
#include <phparameter/PHParameterInterface.h>
#include <trackbase/TrkrDefs.h>

#include <cmath>
#include <map>
#include <string>

class PHCompositeNode;

class ClusHitsVerbosev1;
class PHG4Hit;
class PHG4TruthInfoContainer;
class TrkrClusterContainer;
class TrkrClusterContainer;
class TrkrHitSetContainer;
class TrkrTruthTrack;
class TrkrTruthTrackContainer;

class PHG4InttHitReco : public SubsysReco, public PHParameterInterface
{
 public:
  PHG4InttHitReco(const std::string &name = "PHG4InttHitReco");

  ~PHG4InttHitReco() override;
  int InitRun(PHCompositeNode *topNode) override;

  int process_event(PHCompositeNode *topNode) override;

  void SetDefaultParameters() override;

  void Detector(const std::string &d) { m_Detector = d; }

 protected:
  std::string m_Detector = "INTT";
  std::string m_HitNodeName;
  std::string m_CellNodeName;
  std::string m_GeoNodeName;

  double m_Tmin;
  double m_Tmax;
  double m_crossingPeriod;

  gsl_vector *m_LocalOutVec = nullptr;
  gsl_vector *m_PathVec = nullptr;
  gsl_vector *m_SegmentVec = nullptr;

  // needed for clustering truth tracks
  private:
  TrkrTruthTrackContainer* m_truthtracks     { nullptr }; // output truth tracks
  TrkrClusterContainer*    m_truthclusters   { nullptr }; // output clusters indexed to TrkrDefs::cluskeys in m_truthtracks
  PHG4TruthInfoContainer*  m_truthinfo       { nullptr };
  int                      m_trkid           { -1      };
  bool                     m_is_emb          { false   };
  TrkrTruthTrack*          m_current_track   { nullptr };
  const int                m_cluster_version { 4 };
  TrkrHitSetContainer*     m_truth_hits; // generate and delete a container for each truth track
  std::map<TrkrDefs::hitsetkey,unsigned int> m_hitsetkey_cnt {}; // counter for making ckeys form hitsetkeys

  PHG4Hit* prior_g4hit { nullptr }; // used to check for jumps in g4hits for loopers;
  void truthcheck_g4hit       ( PHG4Hit*, PHCompositeNode* topNode );
  void addtruthhitset         ( TrkrDefs::hitsetkey, TrkrDefs::hitkey, float neffelectrons );
  void cluster_truthhits      ( PHCompositeNode* topNode );
  void end_event_truthcluster ( PHCompositeNode* topNode );

  double m_pixel_thresholdrat { 0.01 };
  float  max_g4hitstep        { 2.0  };
  bool  record_ClusHitsVerbose { false };

  public:
  void set_pixel_thresholdrat (double val) { m_pixel_thresholdrat =   val; };
  void set_max_g4hitstep      (float  _)   { max_g4hitstep        =_; };

  void set_ClusHitsVerbose(bool set=true) { record_ClusHitsVerbose = set; };
  ClusHitsVerbosev1* mClusHitsVerbose { nullptr };

};

#endif