quickHIJING.h

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

// Tell emacs that this is a C++ source
//  -*- C++ -*-.
#ifndef QUICKHIJING_H
#define QUICKHIJING_H

#include <fun4all/SubsysReco.h>

#include <string>
#include <vector>
#include <TTree.h>
#include <TFile.h>

class PHCompositeNode;
class PHG4Particle;
class TTree;
class TFile;

class quickHIJING : public SubsysReco
{
 public:

  quickHIJING(const std::string &name = "quickHIJING");

  ~quickHIJING() override;

  int Init(PHCompositeNode *topNode) override;

  int InitRun(PHCompositeNode *topNode) override;

  int process_event(PHCompositeNode *topNode) override;

  int ResetEvent(PHCompositeNode *topNode) override;

  int EndRun(const int runnumber) override;

  int End(PHCompositeNode *topNode) override;

  int Reset(PHCompositeNode * /*topNode*/) override;

  void Print(const std::string &what = "ALL") const override;

 private:

  TTree *T;
  std::vector<int> m_pid = {0};
  std::vector<float> m_pt =  {0};
  std::vector<float> m_eta = {0};
  std::vector<float> m_phi = {0};
  std::vector<float> m_e = {0};
  std::vector<float> m_p = {0};
  float m_psi2 = 0;
  float m_cent = 0;
  float m_b = 0;
  std::string Outfile;
  TFile *out;

  float getpT(PHG4Particle *particle);
  float getPhi(PHG4Particle *particle);
  float getEta(PHG4Particle *particle);
  float getP(PHG4Particle *particle);
  
};

#endif // QUICKHIJING_H