SamplingFractionReco.h

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

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

#include <fun4all/SubsysReco.h>

#include <string>

class PHCompositeNode;
class TFile;
class TNtuple;

class SamplingFractionReco : public SubsysReco
{
 public:
  SamplingFractionReco(const std::string &name = "SamplingFractionReco", const std::string &filename = "output.root");

  virtual ~SamplingFractionReco();

  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 Detector(const std::string &name);

  void add_support_eloss(const int i = 1) { m_SupportFlag = i; }

 private:
  TNtuple *ntup = nullptr;
  TFile *outfile = nullptr;

  int m_SupportFlag = 0;

  std::string outfilename;
  std::string m_Detector;

  std::string m_HitNodeName;
  std::string m_AbsorberNodeName;
  std::string m_SupportNodeName;
};

#endif  // SAMPLINGFRACTIONRECO_H