proto/Source/buffer.h

Go to the documentation of this file.

#ifndef _BUFFER_H
#define _BUFFER_H

#include <list>
// We use pair here.
#include <utility>

// Some base types
#include "config.h"
// Parser used
#include "parser.h"
// Count stats
#include "stats.h"

namespace ace {

typedef std::pair<words_store_t::const_iterator, words_store_t::const_iterator> words_range_t;

class Buffer {

    typedef std::pair<size_t, size_t> sentence_index_t;
    typedef std::list<sentence_index_t> sentences_indices_t;

    Parser& _parser;
    words_store_t _words;
    sentences_indices_t _sentences;
    sentences_indices_t::iterator _current;
    const ContextWindow _context_window;
    DataFileStats _stats;
    inline size_t _index_of(words_store_t::const_iterator iter) const { return iter - _words.begin(); }
    void _init(void);
    size_t _pop(void);
    void _push(size_t needed = 1, ContextWindow::Mode context_window_mode = ContextWindow::SENTENCE_MODE);

public:
    Buffer(Parser& parser, ContextWindow context_window): _parser(parser), _words(), _sentences(), _current(), _context_window(context_window), _stats() {
        _init();        
    }
    inline words_range_t context_range(void) const { return words_range_t(_words.begin(), _words.end()); }
    inline words_range_t context_range(words_store_t::const_iterator head) const { return context_range(head, head); }
    words_range_t context_range(words_store_t::const_iterator left, words_store_t::const_iterator right) const;
    inline bool empty(void) const { return _words.empty(); }
    bool next(void);
    words_range_t current(void) const;
    inline DataFileStats stats(void) const { return _stats; }
};

} // namespace ace

#endif

---