bstream_iterator.h

Go to the documentation of this file.

#ifndef bstream_iterator_H
#define bstream_iterator_H

#include "binary_ifstream.h"
#include "binary_ofstream.h"
#include <iterator>
#include "FWCore/Utilities/interface/Visibility.h"

template <typename T>
class bistream_iterator : 
    public std::iterator<std::input_iterator_tag, T, ptrdiff_t, const T*, const T&>
{
public:

    bistream_iterator() : stream_(0) {}

    bistream_iterator( binary_ifstream& s) : stream_(&s) {
    read();
    }

    const T& operator*() const {return value_;}

    const T* operator->() const {return &value_;}

    bistream_iterator& operator++() {read(); return *this;}

    bistream_iterator& operator++(int) {
    bistream_iterator tmp;
    read();
    return tmp;
    }

    bool operator==(const bistream_iterator& rhs) {
    return stream_ == rhs.stream_;
    }

    bool operator!=(const bistream_iterator& rhs) {
    return !operator==(rhs);
    }

private:

    binary_ifstream* stream_;
    T               value_;

    void read() {
    if (stream_ != 0) {
        // if (!(*stream_ >> value_)) stream_ = 0;
        if (!(*stream_ >> value_)) {
        stream_ = 0;
        // std::cout << "istream_iterator: stream turned bad, set stream_ to zero" << std::endl;
        }
    }
    }

};

template <typename T> 
class dso_internal bostream_iterator :
    public std::iterator<std::output_iterator_tag,void,void,void,void> {
public:

    bostream_iterator( binary_ofstream& s) : stream_(&s) {}

    bostream_iterator& operator=( const T& t) {
    *stream_ << t;
    return *this;
    }

    bostream_iterator& operator*() {return *this;}
    bostream_iterator& operator++() {return *this;}
    bostream_iterator& operator++(int) {return *this;}

private:

    binary_ofstream* stream_;

};

#endif