SimpleNtuple.h

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

#ifndef SIMPLENTUPLE_H
#define SIMPLENTUPLE_H

#include <fun4all/SubsysReco.h>

#include <map>
#include <set>
#include <string>
#include <vector>

// Forward declerations
class Fun4AllHistoManager;
class PHCompositeNode;
class TFile;
class TH1;
class TNtuple;

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

  virtual ~SimpleNtuple();

  int Init(PHCompositeNode *);

  int process_event(PHCompositeNode *);

  int End(PHCompositeNode *);

  void AddNode(const std::string &name, const int detid = 0);

 protected:
  Fun4AllHistoManager *m_HistoManager;
  TNtuple *m_Ntup;
  TFile *m_Outfile;
  std::string m_Filename;
  std::vector<TH1 *> m_ElossVec;
  std::set<std::string> m_NodePostfixSet;
  std::map<std::string, int> m_DetIdMap;
};

#endif