pi0ClusterAna.h

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

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

#include <fun4all/SubsysReco.h>

#include <string>
#include <vector>

class PHCompositeNode;
class TTree;
class PHG4Particle;
class CaloEvalStack;
class TFile;
class Fun4AllHistoManager;

class pi0ClusterAna : public SubsysReco
{
 public:

  pi0ClusterAna(const std::string &name, const std::string &outName);

  ~pi0ClusterAna() 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:

  float getEta(PHG4Particle *particle);
    
  TTree *clusters_Towers;
  TTree *truth_photon;
  TTree *truth_pi0;
  
  //CaloEvalStack *caloevalstack;
  
  //stuff for towers and clusters
  std::vector<float> m_eta_center;
  std::vector<float> m_phi_center;
  std::vector<float> m_tower_energy;
  std::vector<float> m_cluster_eta;
  std::vector<float> m_cluster_phi;
  std::vector<float> m_cluster_e;
  std::vector<float> m_cluster_chi2;
  std::vector<float> m_cluster_prob;
  std::vector<float> m_cluster_nTowers;
  
  //stuff for truth photons
  std::vector<float> m_asym;
  std::vector<float> m_deltaR;
  std::vector<float> m_lead_E;
  std::vector<float> m_sublead_E;
  std::vector<float> m_lead_phi;
  std::vector<float> m_lead_eta;
  std::vector<float> m_sublead_phi;
  std::vector<float> m_sublead_eta;

  //stuff for truth pi0's
  std::vector<float> m_pi0_E;
  std::vector<float> m_pi0_eta;
  std::vector<float> m_pi0_phi;
  
  TFile *out;
  Fun4AllHistoManager *hm = nullptr;
  std::string Outfile;

  
};

#endif // PI0CLUSTERANA_H