colourspacebuffer.h

// PAS 17/6/04 (revised 16/2/05)

#ifndef CVD_INCLUDE_COLOURSPACEBUFFER_H
#define CVD_INCLUDE_COLOURSPACEBUFFER_H

#include <cvd/colourspace_convert.h>
#include <cvd/colourspace_frame.h>
#include <cvd/convert_image.h>
#include <cvd/localvideobuffer.h>

namespace CVD
{
template <class T, class From>
class ColourspaceBuffer : public CVD::LocalVideoBuffer<T>
{
    public:
    ColourspaceBuffer(CVD::VideoBuffer<From>& buf)
        : LocalVideoBuffer<T>(buf.type())
        , m_vidbuf(buf)
        , m_size(buf.size())
    {
    }

    virtual RawVideoBuffer* source_buffer()
    {
        return &m_vidbuf;
    }

    ImageRef size()
    {
        return m_size;
    }

    virtual bool frame_pending()
    {
        return m_vidbuf.frame_pending();
    }

    virtual void seek_to(double t)
    {
        m_vidbuf.seek_to(t);
    }

    virtual double frame_rate()
    {
        return m_vidbuf.frame_rate();
    }

    virtual CVD::ColourspaceFrame<T>* get_frame()
    {
        VideoFrame<From>* fr = m_vidbuf.get_frame();
        Image<T> cv = convert_image<T>(*fr);
        ColourspaceFrame<T>* ret = new ColourspaceFrame<T>(fr->timestamp(), std::move(cv));
        m_vidbuf.put_frame(fr);
        return ret;
    }

    virtual void put_frame(CVD::VideoFrame<T>* f)
    {
        //Check that the type is correct...
        ColourspaceFrame<T>* csf = dynamic_cast<ColourspaceFrame<T>*>(f);

        if(csf == NULL)
            throw CVD::Exceptions::VideoBuffer::BadPutFrame();
        else
            delete csf;
    }

    private:
    CVD::VideoBuffer<From>& m_vidbuf;
    ImageRef m_size;
};

template <class T, class From>
class ColourspaceBuffer_managed : public ColourspaceBuffer<T, From>
{
    public:
    ColourspaceBuffer_managed(CVD::VideoBuffer<From>* buf)
        : ColourspaceBuffer<T, From>(*buf)
        , vb(buf)
    {
    }

    ~ColourspaceBuffer_managed()
    {
        delete vb;
    }

    private:
    VideoBuffer<From>* vb;
};

}
#endif