d2/d14/TrkrNtuplizer_8h_source.html

#ifndef TRACKRECO_TRKRNTUPLIZER_H

#define TRACKRECO_TRKRNTUPLIZER_H


//===============================================

//===============================================


#include <fun4all/SubsysReco.h>

#include <trackbase/TrkrDefs.h>


#include <TMatrixFfwd.h>

#include <TMatrixT.h>

#include <TMatrixTUtils.h>

#include <string>

#include <map>

#include <set>


class PHCompositeNode;

class PHTimer;

class TrkrCluster;

class TFile;

class TNtuple;

class SvtxTrack;

class SvtxTrackMap;

class SvtxVertexMap;


// class TrkrClusterContainer;


class TrkrNtuplizer : public SubsysReco

{

 public:

  TrkrNtuplizer(const std::string &name = "TRKRNTUPLEIZER",

                const std::string &filename = "trkrntuple.root",

                const std::string &trackmapname = "SvtxTrackMap",

                unsigned int nlayers_maps = 3,

                unsigned int nlayers_intt = 8,

                unsigned int nlayers_tpc = 48,

                unsigned int nlayers_mms = 2);

  ~TrkrNtuplizer() override;


  int Init(PHCompositeNode *topNode) override;

  int InitRun(PHCompositeNode *topNode) override;

  int process_event(PHCompositeNode *topNode) override;

  int End(PHCompositeNode *topNode) override;

  //  void do_primaries(bool b);


  void do_info_eval(bool b) { _do_info_eval = b; }

  void do_vertex_eval(bool b) { _do_vertex_eval = b; }

  void do_hit_eval(bool b) { _do_hit_eval = b; }

  void do_cluster_eval(bool b) { _do_cluster_eval = b; }

  void do_clus_trk_eval(bool b) { _do_clus_trk_eval = b; }

  void do_track_eval(bool b) { _do_track_eval = b; }

  void do_tpcseed_eval(bool b) { _do_tpcseed_eval = b; }

  void do_siseed_eval(bool b) { _do_siseed_eval = b; }

  void set_first_event(int value) { _ievent = value; }

  void set_trkclus_seed_container(const std::string& name)

  { _clustrackseedcontainer = name; }

  SvtxTrack* best_track_from(TrkrDefs::cluskey cluster_key);

  std::set<SvtxTrack*> all_tracks_from(TrkrDefs::cluskey cluster_key);

  void create_cache_track_from_cluster();

  std::vector<TrkrDefs::cluskey> get_track_ckeys(SvtxTrack* track);


 private:

  unsigned int _ievent;

  unsigned int _iseed;

  float m_fSeed;

  // eval stack


  TMatrixF calculateClusterError(TrkrCluster *c, float &clusphi);

  void get_dca(SvtxTrack *track, SvtxVertexMap *vertexmap,

               float &dca3dxy, float &dca3dz,

               float &dca3dxysigma, float &dca3dzsigma);

  // TrkrClusterContainer *cluster_map{nullptr};


  //----------------------------------

  // evaluator output ntuples


  bool _do_info_eval;

  bool _do_vertex_eval;

  bool _do_hit_eval;

  bool _do_cluster_eval;

  bool _do_clus_trk_eval;

  bool _do_track_eval;

  bool _do_tpcseed_eval;

  bool _do_siseed_eval;


  unsigned int _nlayers_maps = 3;

  unsigned int _nlayers_intt = 4;

  unsigned int _nlayers_tpc = 48;

  unsigned int _nlayers_mms = 2;


  TNtuple *_ntp_info;

  TNtuple *_ntp_vertex;

  TNtuple *_ntp_hit;

  TNtuple *_ntp_cluster;

  TNtuple *_ntp_clus_trk;

  TNtuple *_ntp_track;

  TNtuple *_ntp_tpcseed;

  TNtuple *_ntp_siseed;


  // evaluator output file

  std::string _filename;

  // Track map name

  std::string _trackmapname;

  SvtxTrackMap* _trackmap = nullptr;


  std::string _clustrackseedcontainer = "TpcTrackSeedContainer";


  TFile *_tfile;

  PHTimer *_timer;


  // output subroutines

  void fillOutputNtuples(PHCompositeNode *topNode);

  void printInputInfo(PHCompositeNode *topNode);

  void printOutputInfo(PHCompositeNode *topNode);

  double AdcClockPeriod = 53.0;                      // ns


  bool _cache_track_from_cluster_exists = false;

  std::map<TrkrDefs::cluskey, std::set<SvtxTrack*> > _cache_all_tracks_from_cluster;

  std::map<TrkrDefs::cluskey, SvtxTrack*> _cache_best_track_from_cluster;

};


#endif  // G4EVAL_SVTXEVALUATOR_H