PHTruthTrackSeeding.h

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

#ifndef TRACKRECO_PHTRUTHTRACKSEEDING_H
#define TRACKRECO_PHTRUTHTRACKSEEDING_H

#include "PHTrackSeeding.h"
#include <trackbase/TrkrDefs.h>
#include <string>  // for string
#include <vector>

// forward declarations
class PHCompositeNode;
class PHG4TruthInfoContainer;
class PHG4HitContainer;
class TrkrHitTruthAssoc;
class TrkrClusterContainer;
class SvtxClusterEval;

//class SvtxHitMap;
//class PHG4CellContainer;


class PHTruthTrackSeeding : public PHTrackSeeding
{
 public:
  PHTruthTrackSeeding(const std::string& name = "PHTruthTrackSeeding");

  unsigned int get_min_clusters_per_track() const
  {
    return _min_clusters_per_track;
  }

  void set_min_clusters_per_track(unsigned int minClustersPerTrack)
  {
    _min_clusters_per_track = minClustersPerTrack;
  }

  void set_min_layer(unsigned int minLayer)
  {
    _min_layer = minLayer;
  }

  void set_max_layer(unsigned int maxLayer)
  {
    _max_layer = maxLayer;
  }

  double get_min_momentum() const
  {
    return _min_momentum;
  }

  void set_min_momentum(double m)
  {
    _min_momentum = m;
  }
void helicalTrackFit(const bool helicalTrackFit)
{m_helicalTrackFit = helicalTrackFit; }
 protected:
  int Setup(PHCompositeNode* topNode) override;

  int Process(PHCompositeNode* topNode) override;

  int End() override;

 private:
  int GetNodes(PHCompositeNode* topNode);

  void circleFitSeed(std::vector<TrkrDefs::cluskey> clusters,
         double& x, double& y, double&z,
           double& px, double& py, double& pz, int charge);
void circleFitByTaubin(std::vector<TrkrDefs::cluskey>& clusters,
       double& R, double& X0, double& Y0);
  void findRoot(const double& R, const double& X0, const double& Y0,
    double& x, double& y);
  void lineFit(std::vector<TrkrDefs::cluskey>& clusters,
         double& A, double& B);
  PHG4TruthInfoContainer* _g4truth_container = nullptr;
  TrkrClusterContainer *m_clusterMap = nullptr;
  PHG4HitContainer* phg4hits_tpc = nullptr;
  PHG4HitContainer* phg4hits_intt = nullptr;
  PHG4HitContainer* phg4hits_mvtx = nullptr;
  PHG4HitContainer* phg4hits_micromegas = nullptr;

  TrkrHitTruthAssoc* hittruthassoc = nullptr;
  SvtxClusterEval* _clustereval;

  unsigned int _min_clusters_per_track = 3;
  unsigned int _min_layer = 0;
  unsigned int _max_layer = 60;

  bool m_helicalTrackFit = false;

  double _min_momentum = 50e-3;
};

#endif