PHSiliconTruthTrackSeeding.h

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

#ifndef TRACKRECO_PHSILICONTRUTHTRACKSEEDING_H
#define TRACKRECO_PHSILICONTRUTHTRACKSEEDING_H

#include "PHTrackSeeding.h"

#include <string>  // for string

// forward declarations
class PHCompositeNode;
class PHG4TruthInfoContainer;
class PHG4HitContainer;
class TrkrClusterHitAssoc;
class TrkrHitTruthAssoc;

//class SvtxHitMap;
//class PHG4CellContainer;


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

  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;
  }

 protected:
  int Setup(PHCompositeNode* topNode) override;

  int Process(PHCompositeNode* topNode) override;

  int End() override;

 private:
  int GetNodes(PHCompositeNode* topNode);

  PHG4TruthInfoContainer* _g4truth_container = nullptr;

  PHG4HitContainer* phg4hits_tpc = nullptr;
  PHG4HitContainer* phg4hits_intt = nullptr;
  PHG4HitContainer* phg4hits_mvtx = nullptr;
  PHG4HitContainer* phg4hits_micromegas = nullptr;

  TrkrHitTruthAssoc* hittruthassoc = nullptr;
  TrkrClusterHitAssoc* clusterhitassoc = nullptr;

  unsigned int _min_clusters_per_track = 2;
  unsigned int _min_layer = 0;
  unsigned int _max_layer = 6;

  double _min_momentum = 50e-3;
};

#endif