serverpushjpegbuffer.h

#ifndef CVD_INC_SERVERPUSHJPEGBUFFER_H
#define CVD_INC_SERVERPUSHJPEGBUFFER_H
#include <cvd/localvideobuffer.h>
#include <cvd/serverpushjpegframe.h>
#include <cvd/timer.h>
#include <iostream>

namespace CVD
{

template <class C>
class ServerPushJpegBuffer : public LocalVideoBuffer<C>
{
    public:
    ServerPushJpegBuffer(std::istream& i, bool warnings_ = 0, int eat_frames = 0)
        : LocalVideoBuffer<C>(VideoBufferType::Live)
        , is(i)
        , warnings(warnings_)
    {
        std::string tmp;
        //Eat the first 10 frames because the camera sometimes takes a while to
        //crank out ones of the specified size

        for(int junk = 0; junk < eat_frames; junk++)
            gimme_an_image(tmp);

        //Eat the first frame just to get the size
        Image<C> c = gimme_an_image(tmp);
        s = c.size();
    }

    virtual ImageRef size()
    {
        return s;
    }

    LocalVideoFrame<C>* get_frame()
    {
        Image<C> c;
        std::string data;

    loop:
        c = gimme_an_image(data);

        if(c.size() != s)
        {
            if(warnings)
                std::cerr << "ServerPushJpegBuffer: video frame is " << c.size() << " not " << s << std::endl;
            goto loop;
        }

        return new ServerPushJpegFrame<C>(get_time_of_day(), std::move(c), data);
    }

    void put_frame(VideoFrame<C>* f)
    {
        LocalVideoFrame<C>* g = dynamic_cast<LocalVideoFrame<C>*>(f);

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

    bool frame_pending()
    {
        return 1;
    }

    void seek_to(double) {};

    double frame_rate()
    {
        return 30;
    }

    private:
    std::istream& is;
    ImageRef s;
    bool warnings;

    Image<C> gimme_an_image(std::string& data)
    {

        std::string line;

        int length;
        getline(is, line); //Get --ImageSeparator
        getline(is, line); //Get Content-Type:
        is >> line; //Get Content-Length:
        is >> length; //Get the actual content length
        getline(is, line); //Eat the rest of the line
        getline(is, line); //Get the blank line

        data.resize(length);
        is.read(&data[0], length);

        std::istringstream ss(data);

        Image<C> c = img_load(ss);

        is.get(); //Eat the \r\n after the JPEG
        is.get();

        return c;
    }
};

}
#endif