AntiSigma/src/HijingCountNtuple.h

0001 #ifndef HijingCountNtuple_h__
0002 #define HijingCountNtuple_h__
0003
0004 #include <fun4all/SubsysReco.h>
0005 #include <map>
0006 #include <set>
0007 #include <string>
0008 #include <vector>
0009
0010 // Forward declerations
0011 class Fun4AllHistoManager;
0012 class PHCompositeNode;
0013 class TFile;
0014 class TH1;
0015 class TH2;
0016 class TNtuple;
0017
0018 class HijingCountNtuple: public SubsysReco
0019 {
0020  public:
0021
0022   //! constructor
0023   HijingCountNtuple( const std::string &name = "HijingCountNtuple", const std::string &filename = "HijingCountNtuple.root" );
0024
0025   //! destructor
0026   virtual ~HijingCountNtuple();
0027
0028   //! full initialization
0029   int Init(PHCompositeNode *);
0030
0031   //! event processing method
0032   int process_event(PHCompositeNode *);
0033
0034   //! end of run method
0035   int End(PHCompositeNode *);
0036
0037   void AddNode(const std::string &name, const int detid=0);
0038
0039 protected:
0040   int nblocks;
0041   Fun4AllHistoManager *hm;
0042   std::vector<TH1 *> nhits;
0043   std::vector<TH1 *> eloss;
0044   //  std::vector<TH2 *> nhit_edep;
0045   std::string _filename;
0046   std::set<std::string> _node_postfix;
0047   std::map<std::string, int> _detid;
0048   TNtuple *ntup;
0049   TNtuple *ntupe;
0050   TFile *outfile;
0051 };
0052
0053 #endif