TpcPrototypeGenFitTrkFinder.h

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

#ifndef TRACKRECO_TpcPrototypeGenFitTrkFinder_H
#define TRACKRECO_TpcPrototypeGenFitTrkFinder_H

#include <fun4all/SubsysReco.h>

#include <string>
#include <vector>


namespace PHGenFit
{
  class Fitter;
}

class AssocInfoContainer;
class PHCompositeNode;
class SvtxTrackMap;
class SvtxVertexMap;
class TClonesArray;
class TTree;
class TrkrCluster;
class TrkrClusterContainer;

class TpcPrototypeGenFitTrkFinder : public SubsysReco
{
 public:
  TpcPrototypeGenFitTrkFinder(const std::string& name = "TpcPrototypeGenFitTrkFinder", int layers = 16);

  ~TpcPrototypeGenFitTrkFinder();

  int Init(PHCompositeNode*);

  int InitRun(PHCompositeNode*);

  int process_event(PHCompositeNode*);

  int End(PHCompositeNode*);

  void set_eval_filename(const std::string & file)
  {
      _eval_outname = file;
  }
  std::string get_eval_filename() const
  {
    return _eval_outname;
  }

  void fill_eval_tree(PHCompositeNode*);
  void init_eval_tree();
  void reset_eval_variables();

  bool is_do_eval() const
  {
    return _do_eval;
  }

  void set_do_eval(bool doEval)
  {
    _do_eval = doEval;
  }

  bool is_do_evt_display() const
  {
    return _do_evt_display;
  }

  void set_do_evt_display(bool doEvtDisplay)
  {
    _do_evt_display = doEvtDisplay;
  }

  const std::string& get_track_fitting_alg_name() const
  {
    return _track_fitting_alg_name;
  }

  void set_track_fitting_alg_name(const std::string& trackFittingAlgName)
  {
    _track_fitting_alg_name = trackFittingAlgName;
  }

  int get_primary_pid_guess() const
  {
    return _primary_pid_guess;
  }

  void set_primary_pid_guess(int primaryPidGuess)
  {
    _primary_pid_guess = primaryPidGuess;
  }

 private:
  int _event;

  int GetNodes(PHCompositeNode*);

  int CreateNodes(PHCompositeNode*);

  typedef std::vector<const TrkrCluster*> tracklet_t;

  double getChi2Ndf(const tracklet_t & tracklet);

  PHGenFit::Fitter* _fitter;

  std::string _track_fitting_alg_name;

  int nLayer;
  unsigned int minLayer;
  unsigned int maxTracklet;
  int _primary_pid_guess;
  double rphiWindow;
  double ZWindow;

  //  PHG4TruthInfoContainer* _truth_container;
  TrkrClusterContainer* _clustermap;
  SvtxTrackMap* _trackmap;
  AssocInfoContainer *_assoc_container;
  SvtxVertexMap* _vertexmap;

  bool _do_eval;

  std::string _eval_outname;

  TTree* _eval_tree;
  int _tca_ntrack;
  TClonesArray* _tca_trackmap;

  TTree* _cluster_eval_tree;
  float _cluster_eval_tree_x;
  float _cluster_eval_tree_y;
  float _cluster_eval_tree_z;
  float _cluster_eval_tree_gx;
  float _cluster_eval_tree_gy;
  float _cluster_eval_tree_gz;

  bool _do_evt_display;
};

#endif