BookStore.cc

Go to the documentation of this file.

#include "marlinmt/book/BookStore.h"

// -- std includes
#include <filesystem>
#include <unordered_map>

// -- MarlinBook includes
#include "marlinmt/book/Condition.h"
#include "marlinmt/book/Selection.h"
#include "marlinmt/book/StoreWriter.h"

class PathHash {
public:
  std::size_t operator()(const std::filesystem::path& path) const {
    return std::filesystem::hash_value(path);
  }
};

namespace marlinmt {
  namespace book {

    std::shared_ptr< details::Entry >
    BookStore::addEntry( const std::shared_ptr< EntryBase > &entry,
                         EntryKey                            key ) {
      key.idx = _entries.size() ;

      if ( !_idToEntry
              .insert(
                std::make_pair( Identifier( key.path ), key.idx ) )
              .second ) {
        MARLIN_BOOK_THROW( "Object already exist. Use store.book to avoid this." ) ;
      }
      _entries.push_back( std::make_shared< details::Entry >( details::Entry( entry, key ) ) ) ;
      return _entries.back() ;
    }
    
    //--------------------------------------------------------------------------
  
    void BookStore::store(StoreWriter& writer) const {
      writer.writeSelection(
        Selection::find(
          _entries.begin(),
          _entries.end(),
          ConditionBuilder())
      );  
    }

    //--------------------------------------------------------------------------
    
    void BookStore::storeSelection(
        StoreWriter& writer, const Selection& selection) const {
      writer.writeSelection(selection);
    }

    //--------------------------------------------------------------------------

    Selection BookStore::find( const Condition &cond ) const {
      return Selection::find( _entries.cbegin(), _entries.cend(), cond ) ;
    }

    //--------------------------------------------------------------------------
    
     WeakEntry BookStore::findFirst( const Condition &cond) const {
      return Selection::findFirst(_entries.cbegin(), _entries.cend(), cond); 
     }

    //--------------------------------------------------------------------------

    void BookStore::remove( const EntryKey &key ) { get( key ).clear(); }

    //--------------------------------------------------------------------------

    void BookStore::remove( const Selection &selection ) {
      for ( const WeakEntry &e : selection ) {
        remove( e.key() ) ;
      }
    }

    //--------------------------------------------------------------------------

    void BookStore::clear() { _entries.resize( 0 ); }

    //--------------------------------------------------------------------------

    std::size_t BookStore::Identifier::Hash::
                operator()( const Identifier &id ) const {
      return std::filesystem::hash_value(id._path);
    }

    //--------------------------------------------------------------------------
    
    std::filesystem::path BookStore::normalizeDirPath(const std::filesystem::path& path) {
      if(!path.is_absolute() || path.has_filename()) {
        MARLIN_BOOK_THROW( std::string("'") 
            + path.string() 
            + "' is not an absolute path to folder!");
      }
      return std::filesystem::absolute(path);
    }


  } // end namespace book
} // end namespace marlinmt