Proto4SampleFrac.h

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

#ifndef __Proto4SampleFrac_H__
#define __Proto4SampleFrac_H__

#include <TFile.h>
#include <TNtuple.h>
#include <fun4all/SubsysReco.h>
#include <stdint.h>
#include <fstream>
#include <string>

class PHCompositeNode;
class PHG4HitContainer;
class PHG4TruthInfoContainer;
class Fun4AllHistoManager;
class TH1F;
class TH2F;
class TH3F;
class TTree;
class PHG4Particle;
class RawTowerGeom;
class RawTowerContainer;

class Proto4SampleFrac : public SubsysReco
{
 public:
  Proto4SampleFrac(const std::string &calo_name, const std::string &filename);

  virtual ~Proto4SampleFrac();

  int Init(PHCompositeNode *topNode);

  int InitRun(PHCompositeNode *topNode);

  int process_event(PHCompositeNode *topNode);

  int End(PHCompositeNode *topNode);

  void
  is_sim(bool b)
  {
    _is_sim = b;
  }

  std::string get_histo_prefix();

 private:
  // calorimeter size
  enum
  {
    n_size = 8
  };

  bool _is_sim;

  Fun4AllHistoManager *get_HistoManager();

  std::pair<int, int>
  find_max(RawTowerContainer *towers, int cluster_size);

  std::string _filename;
  std::string _calo_name;

  PHG4HitContainer* _calo_hit_container; // G4Hit for calorimeter
  PHG4HitContainer* _calo_abs_hit_container; // G4Hit for absorber
  PHG4TruthInfoContainer* _truth_container;
};

#endif  // __Proto4SampleFrac_H__