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/ActsGeometry.h>
#include <trackbase/TrkrDefs.h>
#include <string>  // for string
#include <vector>
#include <memory>
#include <gsl/gsl_rng.h>

// forward declarations
class PHCompositeNode;
class PHG4TruthInfoContainer;
class PHG4HitContainer;
class TrkrHitTruthAssoc;
class TrkrClusterContainer;
class TrkrClusterCrossingAssoc;
class SvtxClusterEval;
class TrackSeed;
class TrackSeedContainer;
class PHG4Particle;


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

 protected:
  int Setup(PHCompositeNode* topNode) override;

  int Process(PHCompositeNode* topNode) override;

  int End() override;

 private:
  int GetNodes(PHCompositeNode* topNode);
  int CreateNodes(PHCompositeNode* topNode);

  void buildTrackSeed(std::vector<TrkrDefs::cluskey> clusters, 
                      PHG4Particle *g4particle, TrackSeedContainer* container);
  PHG4TruthInfoContainer* m_g4truth_container = nullptr;

  /* this is a copy of the code in PHTruthSiliconAssociation */
  std::set<short int> getInttCrossings(TrackSeed*) const;

  TrkrClusterContainer *m_clusterMap = nullptr;
  TrkrClusterCrossingAssoc *m_cluster_crossing_map = 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;

  double _min_momentum = 50e-3;

  TrackSeedContainer *_track_map_silicon = nullptr;
  TrackSeedContainer *_track_map_combined = nullptr;

  ActsGeometry *tgeometry = nullptr;

  bool _circle_fit_seed = false;

  class Deleter
  {
    public:
    void operator() (gsl_rng* rng) const { gsl_rng_free(rng); }
  };


  std::unique_ptr<gsl_rng, Deleter> m_rng;

};

#endif