pi0ClusterAna/src/pi0ClusterAna.h

0001 // Tell emacs that this is a C++ source
0002 //  -*- C++ -*-.
0003 #ifndef PI0CLUSTERANA_H
0004 #define PI0CLUSTERANA_H
0005
0006 #include <fun4all/SubsysReco.h>
0007
0008 #include <string>
0009 #include <vector>
0010
0011 class PHCompositeNode;
0012 class TTree;
0013 class PHG4Particle;
0014 class CaloEvalStack;
0015 class TFile;
0016 class Fun4AllHistoManager;
0017
0018 class pi0ClusterAna : public SubsysReco
0019 {
0020  public:
0021
0022   pi0ClusterAna(const std::string &name, const std::string &outName);
0023
0024   ~pi0ClusterAna() override;
0025
0026   /** Called during initialization.
0027       Typically this is where you can book histograms, and e.g.
0028       register them to Fun4AllServer (so they can be output to file
0029       using Fun4AllServer::dumpHistos() method).
0030    */
0031   int Init(PHCompositeNode *topNode) override;
0032
0033   /** Called for first event when run number is known.
0034       Typically this is where you may want to fetch data from
0035       database, because you know the run number. A place
0036       to book histograms which have to know the run number.
0037    */
0038   int InitRun(PHCompositeNode *topNode) override;
0039
0040   /** Called for each event.
0041       This is where you do the real work.
0042    */
0043   int process_event(PHCompositeNode *topNode) override;
0044
0045   /// Clean up internals after each event.
0046   int ResetEvent(PHCompositeNode *topNode) override;
0047
0048   /// Called at the end of each run.
0049   int EndRun(const int runnumber) override;
0050
0051   /// Called at the end of all processing.
0052   int End(PHCompositeNode *topNode) override;
0053
0054   /// Reset
0055   int Reset(PHCompositeNode * /*topNode*/) override;
0056
0057   void Print(const std::string &what = "ALL") const override;
0058
0059  private:
0060
0061   float getEta(PHG4Particle *particle);
0062
0063   TTree *clusters_Towers;
0064   TTree *truth_photon;
0065   TTree *truth_pi0;
0066
0067   //CaloEvalStack *caloevalstack;
0068
0069   //stuff for towers and clusters
0070   std::vector<float> m_eta_center;
0071   std::vector<float> m_phi_center;
0072   std::vector<float> m_tower_energy;
0073   std::vector<float> m_cluster_eta;
0074   std::vector<float> m_cluster_phi;
0075   std::vector<float> m_cluster_e;
0076   std::vector<float> m_cluster_chi2;
0077   std::vector<float> m_cluster_prob;
0078   std::vector<float> m_cluster_nTowers;
0079
0080   //stuff for truth photons
0081   std::vector<float> m_asym;
0082   std::vector<float> m_deltaR;
0083   std::vector<float> m_lead_E;
0084   std::vector<float> m_sublead_E;
0085   std::vector<float> m_lead_phi;
0086   std::vector<float> m_lead_eta;
0087   std::vector<float> m_sublead_phi;
0088   std::vector<float> m_sublead_eta;
0089
0090   //stuff for truth pi0's
0091   std::vector<float> m_pi0_E;
0092   std::vector<float> m_pi0_eta;
0093   std::vector<float> m_pi0_phi;
0094
0095   TFile *out;
0096   Fun4AllHistoManager *hm = nullptr;
0097   std::string Outfile;
0098
0099
0100 };
0101
0102 #endif // PI0CLUSTERANA_H