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.
121 lines
4.2 KiB
C
121 lines
4.2 KiB
C
8 years ago
|
/* ---------------------------------------------------------------------------
|
||
|
** 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.
|
||
|
**
|
||
|
** MJPEGVideoSource.h
|
||
|
**
|
||
|
** V4L2 RTSP streamer
|
||
|
**
|
||
|
**
|
||
|
** -------------------------------------------------------------------------*/
|
||
|
|
||
|
#include "JPEGVideoSource.hh"
|
||
|
|
||
|
class MJPEGVideoSource : public JPEGVideoSource
|
||
|
{
|
||
|
public:
|
||
|
static MJPEGVideoSource* createNew (UsageEnvironment& env, FramedSource* source)
|
||
|
{
|
||
|
return new MJPEGVideoSource(env,source);
|
||
|
}
|
||
|
virtual void doGetNextFrame()
|
||
|
{
|
||
|
if (m_inputSource)
|
||
|
m_inputSource->getNextFrame(fTo, fMaxSize, afterGettingFrameSub, this, FramedSource::handleClosure, this);
|
||
|
}
|
||
|
virtual void doStopGettingFrames()
|
||
|
{
|
||
|
FramedSource::doStopGettingFrames();
|
||
|
if (m_inputSource)
|
||
|
m_inputSource->stopGettingFrames();
|
||
|
}
|
||
|
static void afterGettingFrameSub(void* clientData, unsigned frameSize,unsigned numTruncatedBytes,struct timeval presentationTime,unsigned durationInMicroseconds)
|
||
|
{
|
||
|
MJPEGVideoSource* source = (MJPEGVideoSource*)clientData;
|
||
|
source->afterGettingFrame(frameSize, numTruncatedBytes, presentationTime, durationInMicroseconds);
|
||
|
}
|
||
|
void afterGettingFrame(unsigned frameSize,unsigned numTruncatedBytes,struct timeval presentationTime,unsigned durationInMicroseconds)
|
||
|
{
|
||
|
int headerSize = 0;
|
||
|
bool headerOk = false;
|
||
|
fFrameSize = 0;
|
||
|
|
||
|
for (unsigned int i = 0; i < frameSize ; ++i)
|
||
|
{
|
||
|
// SOF
|
||
|
if ( (i+11) < frameSize && fTo[i] == 0xFF && fTo[i+1] == 0xC0 )
|
||
|
{
|
||
|
m_height = (fTo[i+5]<<5)|(fTo[i+6]>>3);
|
||
|
m_width = (fTo[i+7]<<5)|(fTo[i+8]>>3);
|
||
|
m_type = (fTo[i+11] - 0x21);
|
||
|
}
|
||
|
// DQT
|
||
|
if ( (i+5+64) < frameSize && (fTo[i] == 0xFF) && (fTo[i+1] == 0xDB))
|
||
|
{
|
||
|
int quantSize = fTo[i+3];
|
||
|
int quantIdx = fTo[i+4];
|
||
|
if (quantIdx < 3)
|
||
|
{
|
||
|
if ( quantIdx+1 > m_qTableCount )
|
||
|
m_qTableCount = quantIdx+1;
|
||
|
memcpy(m_qTable + quantIdx*64, fTo + i + 5, 64);
|
||
|
}
|
||
|
}
|
||
|
// End of header
|
||
|
if ( (i+1) < frameSize && fTo[i] == 0x3F && fTo[i+1] == 0x00 )
|
||
|
{
|
||
|
headerOk = true;
|
||
|
headerSize = i+2;
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (headerOk)
|
||
|
{
|
||
|
fFrameSize = frameSize - headerSize;
|
||
|
memmove( fTo, fTo + headerSize, fFrameSize );
|
||
|
}
|
||
|
|
||
|
fNumTruncatedBytes = numTruncatedBytes;
|
||
|
fPresentationTime = presentationTime;
|
||
|
fDurationInMicroseconds = durationInMicroseconds;
|
||
|
afterGetting(this);
|
||
|
}
|
||
|
virtual u_int8_t type() { return m_type; };
|
||
|
virtual u_int8_t qFactor() { return 128; };
|
||
|
virtual u_int8_t width() { return m_width; };
|
||
|
virtual u_int8_t height() { return m_height; };
|
||
|
u_int8_t const* quantizationTables( u_int8_t& precision, u_int16_t& length )
|
||
|
{
|
||
|
length = 0;
|
||
|
precision = 0;
|
||
|
if (m_qTableCount > 0)
|
||
|
{
|
||
|
length = 64*m_qTableCount;
|
||
|
}
|
||
|
return m_qTable;
|
||
|
}
|
||
|
|
||
|
protected:
|
||
|
MJPEGVideoSource(UsageEnvironment& env, FramedSource* source) : JPEGVideoSource(env),
|
||
|
m_inputSource(source),
|
||
|
m_width(0), m_height(0), m_qTableCount(0),
|
||
|
m_type(0)
|
||
|
{
|
||
|
memset(&m_qTable,0,sizeof(m_qTable));
|
||
|
}
|
||
|
virtual ~MJPEGVideoSource()
|
||
|
{
|
||
|
Medium::close(m_inputSource);
|
||
|
}
|
||
|
|
||
|
protected:
|
||
|
FramedSource* m_inputSource;
|
||
|
u_int8_t m_width;
|
||
|
u_int8_t m_height;
|
||
|
u_int8_t m_qTable[64*3];
|
||
|
int m_qTableCount;
|
||
|
u_int8_t m_type;
|
||
|
};
|