buf_ostream.h

/***************************************************************************
 *  include/stxxl/bits/mng/buf_ostream.h
 *
 *  Part of the STXXL. See http://stxxl.sourceforge.net
 *
 *  Copyright (C) 2002-2004 Roman Dementiev <[email protected]>
 *
 *  Distributed under the Boost Software License, Version 1.0.
 *  (See accompanying file LICENSE_1_0.txt or copy at
 *  http://www.boost.org/LICENSE_1_0.txt)
 **************************************************************************/

#ifndef STXXL_BUF_OSTREAM_HEADER
#define STXXL_BUF_OSTREAM_HEADER

#include <stxxl/bits/mng/buf_writer.h>


__STXXL_BEGIN_NAMESPACE



template <typename BlkTp_, typename BIDIteratorTp_>
class buf_ostream
{
    typedef BlkTp_ block_type;
    typedef BIDIteratorTp_ bid_iterator_type;

protected:
    buffered_writer<block_type> writer;
    bid_iterator_type current_bid;
    int_type current_elem;
    block_type * current_blk;

public:
    typedef typename block_type::const_reference const_reference;
    typedef typename block_type::reference reference;
    typedef buf_ostream<block_type, bid_iterator_type> _Self;

    buf_ostream(bid_iterator_type first_bid, int_type nbuffers) :
        writer(nbuffers, nbuffers / 2), current_bid(first_bid),
        current_elem(0)
    {
        current_blk = writer.get_free_block();
    }

    _Self & operator << (const_reference record)
    {
        current_blk->elem[current_elem++] = record;
        if (current_elem >= block_type::size)
        {
            current_elem = 0;
            current_blk = writer.write(current_blk, *(current_bid++));
        }
        return (*this);
    }

    reference current()
    {
        return current_blk->elem[current_elem];
    }

    reference operator * ()
    {
        return current_blk->elem[current_elem];
    }

    _Self & operator ++ ()
    {
        ++current_elem;
        if (current_elem >= block_type::size)
        {
            current_elem = 0;
            current_blk = writer.write(current_blk, *(current_bid++));
        }
        return (*this);
    }


    virtual ~buf_ostream()
    {
        assert(current_elem == 0);
    }
};


__STXXL_END_NAMESPACE

#endif // !STXXL_BUF_OSTREAM_HEADER