jetrtrack.h

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

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

#include <fun4all/SubsysReco.h>

#include <string>
#include <TF1.h>
#include<string.h>
#include<TProfile.h>
#include<TH1.h>
#include<TH2.h>
#include <TFile.h>
#include <TTree.h>


class Fun4AllHistoManager;

class PHCompositeNode;
class TF1;
class TH1D;
class TH2D;
class TTree;

class jetrtrack : public SubsysReco
{
 public:

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

  virtual ~jetrtrack();

  void SetOutputFileName(std::string outfilename)
  {
    m_outfilename = outfilename;
  }

  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:
  //---------------------------------------------
  //define the histogram manager
  //---------------------------------------------
  Fun4AllHistoManager *
  get_HistoManager();

  //-------------------------------------------------------------------------------
  //define some vectors in which we store jet information
  //-------------------------------------------------------------------------------

  std::vector<float> m_tjet_pt;
  std::vector<float> m_tjet_phi;
  std::vector<float> m_tjet_eta;
  std::vector<float> m_tjet_jcpt;


  std::vector<float> m_rjet_pt;
  std::vector<float> m_rjet_phi;
  std::vector<float> m_rjet_eta;
  std::vector<float> m_dr;
  std::vector<float> m_trk_pt;
  std::vector<float> m_trk_eta;
  std::vector<float> m_trk_phi;
  std::vector<float> m_trk_qual;
  std::vector<float> m_nmvtxhits;

  std::vector<float> m_tp_pt;
  std::vector<float> m_tp_px;
  std::vector<float> m_tp_py;
  std::vector<float> m_tp_pz;
  std::vector<float> m_tp_phi;
  std::vector<float> m_tp_eta;


  std::vector<float> m_jc_pt;
  std::vector<int> m_jc_index;
  std::vector<float> m_jc_phi;
  std::vector<float> m_jc_eta;


  std::vector<int> m_tp_pid;

  float m_zvtx;
  /* int m_nmvtxhits; */
  int m_centrality; 
  int m_impactparam;
  std::string m_outfilename;

  float m_tjet_pt2;


  
  TFile* outfile;

};

#endif // JETRTRACK_H