BankStream_AMOS.hh

Go to the documentation of this file.

#ifndef __BankStream_AMOS_HH
#define __BankStream_AMOS_HH 1

#include "Bank_AMOS.hh"
#include <vector>




namespace AMOS {

//================================================ BankStream_t ================
//==============================================================================
class BankStream_t : private Bank_t
{

protected:

  static const Size_t DEFAULT_BUFFER_SIZE;

  static const Size_t MAX_OPEN_PARTITIONS;


  //--------------------------------------------------- init -------------------
  void init ( )
  {
    eof_m = false;
    curr_bid_m = 1;
    triples_m . resize (1);
  }


  //--------------------------------------------------- inrange ----------------
  bool inrange ( )
  {
    return ( curr_bid_m > 0  &&  curr_bid_m <= last_bid_m );
  }
  

  bool eof_m;                         
  ID_t curr_bid_m;                    
  std::vector<const IDMap_t::HashTriple_t *> triples_m;   


public:

  enum bankseekdir
    {
      BEGIN,
      CURR,
      END,
    };


  //--------------------------------------------------- BankStream_t -----------
  BankStream_t (NCode_t type)
    : Bank_t (type)
  {
    init( );
    triples_m [NULL_ID] = NULL;
    buffer_size_m = DEFAULT_BUFFER_SIZE;
    max_partitions_m = MAX_OPEN_PARTITIONS;
  }


  //--------------------------------------------------- ~BankStream_t ----------
  ~BankStream_t ( )
  {
    if ( is_open_m )
      close( );
  }


  //--------------------------------------------------- clear ------------------
  void clear ( )
  {
    Bank_t::clear( );
    init( );
  }


  //--------------------------------------------------- close ------------------
  void close ( )
  {
    Bank_t::close( );
    init( );
  }


  //--------------------------------------------------- create -----------------
  void create (const std::string & dir, BankMode_t mode = B_READ | B_WRITE)
  {
    Bank_t::create (dir, mode);
    init( );
  }


  //--------------------------------------------------- destroy ----------------
  void destroy ( )
  {
    Bank_t::destroy( );
    init( );
  }


  //--------------------------------------------------- empty ------------------
  bool empty ( ) const { return Bank_t::empty( ); }


  //--------------------------------------------------- eof --------------------
  bool eof ( ) const
  {
    return ( is_open_m ? eof_m : true );
  }


  //--------------------------------------------------- exists -----------------
  bool exists (const std::string & dir) { return Bank_t::exists(dir); }


  //--------------------------------------------------- existsEID --------------
  bool existsEID (const std::string & eid) const
  { return Bank_t::existsEID(eid); }


  //--------------------------------------------------- existsIID --------------
  bool existsIID (ID_t iid) const { return Bank_t::existsIID(iid); }


  //--------------------------------------------------- getIDMap ---------------
  const IDMap_t & getIDMap ( ) const { return Bank_t::getIDMap( ); }


  //--------------------------------------------------- getIDMapSize -----------
  Size_t getIDMapSize ( ) const { return Bank_t::getIDMapSize( ); }


  //--------------------------------------------------- getMaxIID --------------
  ID_t getMaxIID ( ) const { return Bank_t::getMaxIID( ); }


  //--------------------------------------------------- getSize ----------------
  Size_t getSize ( ) const { return Bank_t::getSize( ); }


  //--------------------------------------------------- getStatus --------------
  signed char getStatus ( ) const { return Bank_t::getStatus( ); }


  //--------------------------------------------------- getType ----------------
  NCode_t getType ( ) const { return Bank_t::getType( ); }


  //--------------------------------------------------- ignore -----------------
  BankStream_t & ignore (bankstreamoff n);


  //--------------------------------------------------- isOpen -----------------
  bool isOpen ( ) const { return Bank_t::isOpen( ); }


  //--------------------------------------------------- lookupBID --------------
  ID_t lookupBID (ID_t iid) const
  { return Bank_t::lookupBID(iid); }


  //--------------------------------------------------- lookupBID --------------
  ID_t lookupBID (const std::string & eid) const
  { return Bank_t::lookupBID(eid); }


  //--------------------------------------------------- lookupEID --------------
  const std::string & lookupEID (ID_t iid) const
  { return Bank_t::lookupEID(iid); }


  //--------------------------------------------------- lookupIID --------------
  ID_t lookupIID (const std::string & eid) const
  { return Bank_t::lookupIID(eid); }


  //--------------------------------------------------- open -------------------
  void open (const std::string & dir, BankMode_t mode = B_READ | B_WRITE);


  //--------------------------------------------------- seekg ------------------
  BankStream_t & seekg (bankstreamoff off, bankseekdir dir)
  {
    if ( ! is_open_m  ||  ! (mode_m & B_READ) )
      AMOS_THROW_IO ("Cannot seekg: bank not open for reading");
    
    if ( dir == BEGIN ) curr_bid_m = 1;
    else if ( dir == END ) curr_bid_m = last_bid_m + 1;
    // else dir == CURR, do nothing
    ignore (off);
    return *this;
  }


  //--------------------------------------------------- seekg ------------------
  BankStream_t & seekg (ID_t pos)
  {
    if ( ! is_open_m  ||  ! (mode_m & B_READ) )
      AMOS_THROW_IO ("Cannot seekg: bank not open for reading");

    curr_bid_m = pos;
    eof_m = !inrange( );
    return *this;
  }


  //--------------------------------------------------- setStatus --------------
  void setStatus (signed char status) { Bank_t::setStatus(status); }


  //--------------------------------------------------- tellg ------------------
  ID_t tellg ( )
  {
    return curr_bid_m;
  }


  //--------------------------------------------------- tellp ------------------
  ID_t tellp ( )
  {
    return last_bid_m + 1;
  }


  //--------------------------------------------------- operator bool( ) -------
  operator bool ( ) const
  {
    return ! eof( );
  }


  //--------------------------------------------------- operator! --------------
  bool operator! ( ) const
  {
    return eof ( );
  }


  //--------------------------------------------------- operator>> -------------
  BankStream_t & operator>> (IBankable_t & obj);


  //--------------------------------------------------- operator<< -------------
  BankStream_t & operator<< (IBankable_t & obj);
};

} // namespace AMOS

#endif // #ifndef __BankStream_AMOS_HH

---