RbTuple.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <vector>
#include <string>
 
class TFile;
 
namespace Belle2 {
  class Module;
 
  /*
     Modules with histogram definitions are supposed to be derived from
     HistoModule class and histograms are defined in HistoModule::defineHisto() function.
     It is supposed to be hooked by register_module(*this) function.
  */
  class RbTupleManager {
  public:
 
    // Constructor and destructor are private for singleton access
    RbTupleManager();
    ~RbTupleManager();
 
    RbTupleManager(int nprocess, const char* filename, const char* workdir = ".");
 
    static RbTupleManager& Instance();
 
    void init(int nprocess, const char* filename, const char* workdir = ".");
 
    //  void define ( void (*func)( void ) );
    void register_module(Module*);
 
 
    int begin(int pid);
 
    int terminate();
 
    // Functions to collect histograms from event processing on the fly
    //  int collect ();
 
    int dump();
 
    int hadd(bool deleteflag = true);
 
    const std::vector<Module*>& getHistDefiningModules() const
    {
      return m_histdefs;
    }
 
  private:
    static RbTupleManager* s_instance; 
    std::vector<Module*> m_histdefs; 
    int m_nproc{0}; 
    std::string m_filename; 
    std::string m_workdir;  
    TFile* m_root{nullptr}; 
  };
 
} // Belle2 namespace