EICG4ZDCHitTree.h

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

#ifndef EICG4ZDCHITTREE_H
#define EICG4ZDCHITTREE_H

#include <fun4all/SubsysReco.h>

#include <map>
#include <set>
#include <string>
#include <vector>

// Forward declerations
class Fun4AllHistoManager;
class PHCompositeNode;
class TFile;
class TTree;

class EICG4ZDCHitTree : public SubsysReco
{
 public:
  EICG4ZDCHitTree(const std::string &name = "EICG4ZDCHitTree", const std::string &filename = "EICG4ZDCHitTree.root");

  ~EICG4ZDCHitTree() override;

  int Init(PHCompositeNode *) override;

  int process_event(PHCompositeNode *) override;

  int End(PHCompositeNode *) override;

  void AddNode(const std::string &name, const int detid = 0);

 protected:
  int nblocks;
  Fun4AllHistoManager *hm;
  std::string _filename;
  std::set<std::string> _node_postfix;
  std::map<std::string, int> _detid;
  TTree *tree;
  TFile *outfile;

 private:

  int Nhit;

  std::vector<int> layerType;
  std::vector<int> layerID;
  std::vector<int> xID;
  std::vector<int> yID;
  std::vector<float> x0;
  std::vector<float> y0;
  std::vector<float> z0;
  std::vector<float> x1;
  std::vector<float> y1;
  std::vector<float> z1;
  std::vector<float> time0;
  std::vector<float> time1;
  std::vector<float> edep;

  int Ntrack;
  std::vector<float> trk_px;
  std::vector<float> trk_py;
  std::vector<float> trk_pz;
  std::vector<float> trk_e;
  std::vector<int> trk_pid;
  

};

#endif