BookStoreManager.h

Go to the documentation of this file.

#pragma once

// -- marlinmt headers
#include <marlinmt/MarlinMTBookConfig.h>
#include <marlinmt/Component.h>
#include <marlinmt/Logging.h>

// -- std includes
#include <unistd.h>
#include <set>

namespace marlinmt {
  
  // forward declaration
  class Application ;
  class Processor ;

  class BookStoreManager : public Component {
  public:
    BookStoreManager() ;
    
    BookStoreManager( const BookStoreManager & ) = delete ;
    BookStoreManager &operator=( const BookStoreManager & ) = delete ;
    BookStoreManager( BookStoreManager && ) = delete ;
    BookStoreManager &operator=( BookStoreManager && ) = delete ;

    void writeToDisk() const ;
    
    void initialize() override ;
    
    template<typename HistT>
    [[nodiscard]] book::Handle<book::Entry<HistT>> bookHist (
      const std::filesystem::path &path, 
      const std::string_view &name,
      const std::string_view &title,
      const std::array<
        const AxisConfig<typename HistT::Precision_t>*,
        HistT::Dimension> &axesconfig,
      const BookFlag_t &flags ) ;

    void addToWrite(const book::EntryKey& key) ;

    void removeFromWrite(const book::EntryKey& key) ;

    struct ObjectNotFound : public std::exception {  
      ObjectNotFound(const std::string_view& msg) 
        : m_msg(msg) {}
      const char* what() const noexcept override { return m_msg.c_str(); }
    private:
      const std::string m_msg;
    };

    const book::EntryKey& getKey(
        const std::filesystem::path &path,
        const std::string_view &name) const ;


    template<typename T>
    [[nodiscard]] book::Handle<book::Entry<T>> getObject( 
        const book::EntryKey &key) const ;

    
  private:
    book::BookStore                      _bookStore {true} ;
    std::set<book::EntryKey>             _entriesToWrite {} ;
    StringParameter                      _outputFile {*this, "OutputFile", "The output file name for storage", "MarlinMT_"+details::convert<int>::to_string(::getpid())+".root"} ;
    StringParameter                      _defaultMemLayout {*this, "DefaultMemoryLayout", "The memory layout for objects", "Default"} ;
    BookFlag_t                           _defaultFlag { 0 } ;
  };

  
}