hcollector.h

#ifndef __HCOLLECTOR_H
#define __HCOLLECTOR_H
 
#include "hreader.h"
#include "hwriter.h"
#include "hwriterconsumer.h"
#include "hprobe.h"
 
#include <vector>
 
template <class T>
class HCollector: public HReader<T>, public HWriter<T>, public HWriterConsumer<T>
{
    private:
 
        HWriter<T>* _writer;
        HReader<T>* _reader;
        size_t _blocksizeIn;
        size_t _blocksizeOut;
 
        HProbe<T>* _probe;
 
        T* _buffer;
        int _collected;
 
    public:
 
        HCollector(HWriter<T>* writer, size_t blocksizeIn, size_t blocksizeOut, HProbe<T>* probe = nullptr);
 
        HCollector(HWriterConsumer<T>* consumer, size_t blocksizeIn, size_t blocksizeOut, HProbe<T>* probe = nullptr);
 
        HCollector(HReader<T>* reader, size_t blocksizeIn, size_t blocksizeOut, HProbe<T>* probe = nullptr);
 
        void SetWriter(HWriter<T>* writer);
 
    public:
 
        ~HCollector();
 
        int Write(T* src, size_t blocksize);
 
        int Read(T* dest, size_t blocksize);
 
        bool Start();
 
        bool Stop();
 
        bool Command(HCommand* command) {
            if( _writer != nullptr )
            {
                if( !_writer->Command(command) )
                {
                    return false;
                }
            }
            if( _reader != nullptr )
            {
                if( !_reader->Command(command) )
                {
                    return false;
                }
            }
            return true;
        }
};
 
#endif