cvd/serverpushjpegbuffer.h

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

namespace CVD
{

template<class C> class ServerPushJpegBuffer: public LocalVideoBuffer<C>
{
    public:
        ServerPushJpegBuffer<C>(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(), 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