GenFitTrackProp.h

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

#ifndef __GenFitTrackProp_H__
#define __GenFitTrackProp_H__



#include <fun4all/SubsysReco.h>

//#include <g4eval/SvtxEvalStack.h>

#include <string>
//#include <memory>

//Forward declerations

class PHCompositeNode;

class PHG4TruthInfoContainer;

class SvtxEvalStack;

class SvtxClusterMap;
class SvtxTrackMap;

class TFile;
class TTree;
class TH2D;

namespace PHGenFit {
        class Fitter;
}


//Brief: basic ntuple and histogram creation for sim evaluation
class GenFitTrackProp: public SubsysReco
{
 public: 
  GenFitTrackProp(const std::string &name="GenFitTrackProp", const int pid_guess = 211);

  int Init(PHCompositeNode *);

  int InitRun(PHCompositeNode *);

  int process_event(PHCompositeNode *);

  int End(PHCompositeNode *);

  void set_filename(const char* file)
  { if(file) _outfile_name = file; }

        int get_pid_guess() const {
                return _pid_guess;
        }

        void set_pid_guess(int pidGuess) {
                _pid_guess = pidGuess;
        }

 private:

  int GetNodes(PHCompositeNode *);

  void fill_tree(PHCompositeNode*);
  void reset_variables();

 private:


  PHG4TruthInfoContainer* _truth_container;
  SvtxTrackMap* _trackmap;


  int _event;


  SvtxEvalStack* _svtxevalstack;

  PHGenFit::Fitter* _fitter;

  std::string _track_fitting_alg_name;

  bool _do_evt_display;

  int _pid_guess;


  std::string _outfile_name;

  TTree* _eval_tree_tracks;

  int event;

  int gtrackID;
  int gflavor;

  double gpx;
  double gpy;
  double gpz;

  double gpt;
  double geta;

  double gvx;
  double gvy;
  double gvz;

  int trackID;
  int charge;
  int nhits;

  double px;
  double py;
  double pz;
  double pt;

  double dca2d;

  double radius80;
  double pathlength80;
  double pathlength85;

};

#endif //* __GenFitTrackProp_H__ *//