StatisticsSvc.h

Go to the documentation of this file.

/*
 *  StatisticsSvc
 *
 *  Provides 'Event Store' like access to ROOT objects (hist, graph, tree, etc.)
 *  - Data can be shared between algorithms.
 *  - Data is not cleared each event loop, so it can be used to collect
 *    statistics, fill trees, etc.
 *
 *  Created by Dan Dwyer on 2009/01/21
 */

#ifndef DATASVC_STATISTICSSVC_H
#define DATASVC_STATISTICSSVC_H

#include "StatisticsSvc/IStatisticsSvc.h"
#include "GaudiKernel/Service.h"
#include <string>
#include <vector>
#include <map>

class TFile;
class HistMan;
class IMessageSvc;

class StatisticsSvc : public Service,
                      virtual public IStatisticsSvc
{
 public:
  // Service interface implementation
  StatisticsSvc(const std::string& name, ISvcLocator *svc);
  ~StatisticsSvc();
  virtual StatusCode initialize();
  virtual StatusCode finalize();
  virtual StatusCode queryInterface(const InterfaceID& riid, 
                                    void** ppvInterface);

  // IStatisticsSvc interface implementation

  virtual StatusCode put(const std::string& path, TObject* object);
  virtual TObject* get(const std::string& path);
  virtual bool exists(const std::string& path);
  virtual std::vector<std::string> getContents(const std::string& path);
  virtual std::vector<std::string> getSubFolders(const std::string& path);

  
  virtual TH1F* getTH1F(const std::string& path);
  virtual TH2F* getTH2F(const std::string& path);
  virtual TH3F* getTH3F(const std::string& path);
  virtual TTree* getTree(const std::string& path);
  virtual TGraph* getGraph(const std::string& path);

 private:
  StatusCode checkOptions(std::string& handle, std::string& filename);
  StatusCode parsePath(const std::string& fullpath, std::string& top, 
                       std::string& rest);
  StatusCode reverseParsePath(const std::string& fullpath, 
                              std::string& rest, 
                              std::string& bottom);
  StatusCode getManager(const std::string& path, HistMan*& hm,
                        std::string& localPath, bool report = true);
  void dump();
  void dumpPath(const std::string& path, std::ostringstream& dumpStr);

 private:
  std::map<std::string,std::string> m_input;

  std::map<std::string,std::string> m_output;

  // Handle to message service
  IMessageSvc* m_msgSvc;

  // Map of all files by filename (myfile0.root, myfile1.root, etc.)
  std::map<std::string, TFile*> m_files;

  // Map of HistMan instances by logical name (path0, path1, etc.)
  std::map<std::string, HistMan*> m_histMan;

};

#endif  // DATASVC_STATISTICSSVC_H