You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
127 lines
4.1 KiB
C++
127 lines
4.1 KiB
C++
/* ---------------------------------------------------------------------------
|
|
** This software is in the public domain, furnished "as is", without technical
|
|
** support, and with no warranty, express or implied, as to its usefulness for
|
|
** any purpose.
|
|
**
|
|
** DeviceSource.h
|
|
**
|
|
** live555 source
|
|
**
|
|
** -------------------------------------------------------------------------*/
|
|
|
|
|
|
#ifndef DEVICE_SOURCE
|
|
#define DEVICE_SOURCE
|
|
|
|
#include <string>
|
|
#include <list>
|
|
#include <iostream>
|
|
#include <iomanip>
|
|
|
|
#include <pthread.h>
|
|
|
|
// live555
|
|
#include <liveMedia.hh>
|
|
|
|
#include "DeviceInterface.h"
|
|
#include "V4l2Capture.h"
|
|
|
|
// -----------------------------------------
|
|
// Video Device Capture Interface
|
|
// -----------------------------------------
|
|
class VideoCaptureAccess : public DeviceInterface
|
|
{
|
|
public:
|
|
VideoCaptureAccess(V4l2Capture* device) : m_device(device) {}
|
|
virtual ~VideoCaptureAccess() { delete m_device; }
|
|
|
|
virtual size_t read(char* buffer, size_t bufferSize) { return m_device->read(buffer, bufferSize); }
|
|
virtual int getFd() { return m_device->getFd(); }
|
|
virtual unsigned long getBufferSize() { return m_device->getBufferSize(); }
|
|
virtual int getWidth() { return m_device->getWidth(); }
|
|
virtual int getHeight() { return m_device->getHeight(); }
|
|
virtual int getVideoFormat() { return m_device->getFormat(); }
|
|
|
|
protected:
|
|
V4l2Capture* m_device;
|
|
};
|
|
|
|
class V4L2DeviceSource: public FramedSource
|
|
{
|
|
public:
|
|
// ---------------------------------
|
|
// Captured frame
|
|
// ---------------------------------
|
|
struct Frame
|
|
{
|
|
Frame(char* buffer, int size, timeval timestamp, char * allocatedBuffer = NULL) : m_buffer(buffer), m_size(size), m_timestamp(timestamp), m_allocatedBuffer(allocatedBuffer) {};
|
|
Frame(const Frame&);
|
|
Frame& operator=(const Frame&);
|
|
~Frame() { delete [] m_allocatedBuffer; };
|
|
|
|
char* m_buffer;
|
|
unsigned int m_size;
|
|
timeval m_timestamp;
|
|
char* m_allocatedBuffer;
|
|
};
|
|
|
|
// ---------------------------------
|
|
// Compute simple stats
|
|
// ---------------------------------
|
|
class Stats
|
|
{
|
|
public:
|
|
Stats(const std::string & msg) : m_fps(0), m_fps_sec(0), m_size(0), m_msg(msg) {};
|
|
|
|
public:
|
|
int notify(int tv_sec, int framesize);
|
|
|
|
protected:
|
|
int m_fps;
|
|
int m_fps_sec;
|
|
int m_size;
|
|
const std::string m_msg;
|
|
};
|
|
|
|
public:
|
|
static V4L2DeviceSource* createNew(UsageEnvironment& env, DeviceInterface * device, int outputFd, unsigned int queueSize, bool useThread) ;
|
|
std::string getAuxLine() { return m_auxLine; }
|
|
void setAuxLine(const std::string auxLine) { m_auxLine = auxLine; }
|
|
DeviceInterface* getDevice() { return m_device; }
|
|
|
|
protected:
|
|
V4L2DeviceSource(UsageEnvironment& env, DeviceInterface * device, int outputFd, unsigned int queueSize, bool useThread);
|
|
virtual ~V4L2DeviceSource();
|
|
|
|
protected:
|
|
static void* threadStub(void* clientData) { return ((V4L2DeviceSource*) clientData)->thread();};
|
|
void* thread();
|
|
static void deliverFrameStub(void* clientData) {((V4L2DeviceSource*) clientData)->deliverFrame();};
|
|
void deliverFrame();
|
|
static void incomingPacketHandlerStub(void* clientData, int mask) { ((V4L2DeviceSource*) clientData)->incomingPacketHandler(); };
|
|
void incomingPacketHandler();
|
|
int getNextFrame();
|
|
void processFrame(char * frame, int frameSize, const timeval &ref);
|
|
void queueFrame(char * frame, int frameSize, const timeval &tv, char * allocatedBuffer = NULL);
|
|
|
|
// split packet in frames
|
|
virtual std::list< std::pair<unsigned char*,size_t> > splitFrames(unsigned char* frame, unsigned frameSize);
|
|
|
|
// overide FramedSource
|
|
virtual void doGetNextFrame();
|
|
|
|
protected:
|
|
std::list<Frame*> m_captureQueue;
|
|
Stats m_in;
|
|
Stats m_out;
|
|
EventTriggerId m_eventTriggerId;
|
|
int m_outfd;
|
|
DeviceInterface * m_device;
|
|
unsigned int m_queueSize;
|
|
pthread_t m_thid;
|
|
pthread_mutex_t m_mutex;
|
|
std::string m_auxLine;
|
|
};
|
|
|
|
#endif
|