HCalJetPhiShift.h

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

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

#include <fun4all/SubsysReco.h>

#include <string>
#include <vector>

class PHCompositeNode;
class TTree;
class TFile;

class HCalJetPhiShift : public SubsysReco
{
public:
  
  HCalJetPhiShift(const std::string &name = "HCalJetPhiShift", const std::string &outputFile = "HCalJetPhiShift_test.root");
  
  ~HCalJetPhiShift() 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;
  
  void SetEventNumber(int event_number)
  {
    m_event = event_number;
  };
    
private:
  std::string m_outputFileName;
  
  TTree *m_T;
  
  int m_event;
  int m_nTow_in;
  int m_nTow_out;
  int m_nTow_emc;
  float m_eta;
  float m_phi;
  float m_e;
  float m_pt;
  float m_vx;
  float m_vy;
  float m_vz;

  std::vector<int> m_id;
  std::vector<float> m_eta_in;
  std::vector<float> m_phi_in;
  std::vector<float> m_e_in;
  std::vector<int> m_ieta_in;
  std::vector<int> m_iphi_in;
  
  std::vector<float> m_eta_out;
  std::vector<float> m_phi_out;
  std::vector<float> m_e_out;
  std::vector<int> m_ieta_out;
  std::vector<int> m_iphi_out;

  std::vector<float> m_eta_emc;
  std::vector<float> m_phi_emc;
  std::vector<float> m_e_emc;
  std::vector<int> m_ieta_emc;
  std::vector<int> m_iphi_emc;

  int FillTTree(PHCompositeNode *topNode);
};

#endif // HCALJETPHISHIFT_H