Added MPEG plugin
This commit is contained in:
Robert Osfield
@@ -87,6 +87,10 @@ ifeq ($(LIBPNG_INSTALLED),yes)
|
||||
PLUGIN_DIRS += png
|
||||
endif
|
||||
|
||||
ifeq ($(LIBMPEG3_INSTALLED),yes)
|
||||
EXAMPLE_DIRS += mpeg
|
||||
endif
|
||||
|
||||
|
||||
ifeq ($(COIN_INSTALLED),yes)
|
||||
PLUGIN_DIRS += Inventor
|
||||
|
||||
@@ -165,6 +165,9 @@ Registry::Registry()
|
||||
addFileExtensionAlias("tif", "tiff");
|
||||
#endif
|
||||
|
||||
addFileExtensionAlias("mpg", "mpeg");
|
||||
addFileExtensionAlias("mpv", "mpeg");
|
||||
|
||||
// remove geo to lwo alias as the new Carbon Graphics GEO format
|
||||
// also uses the .geo. It is still possible to load light wave .geo
|
||||
// files via loading the lwo plugin explicitly and then doing a readNodeFile.
|
||||
|
||||
16
src/osgPlugins/mpeg/GNUmakefile
Normal file
16
src/osgPlugins/mpeg/GNUmakefile
Normal file
@@ -0,0 +1,16 @@
|
||||
TOPDIR = ../../..
|
||||
include $(TOPDIR)/Make/makedefs
|
||||
|
||||
CXXFILES =\
|
||||
MpegImageStream.cpp\
|
||||
ReaderWriterMPEG.cpp\
|
||||
|
||||
MPEG_LIBS = -lmpeg3 -lpthread
|
||||
|
||||
LIBS += $(OSG_LIBS) $(MPEG_LIBS) $(OTHER_LIBS)
|
||||
|
||||
TARGET_BASENAME = mpeg
|
||||
include $(TOPDIR)/Make/cygwin_plugin_def
|
||||
PLUGIN = $(PLUGIN_PREFIX)$(TARGET_BASENAME).$(PLUGIN_EXT)
|
||||
|
||||
include $(TOPDIR)/Make/makerules
|
||||
289
src/osgPlugins/mpeg/MpegImageStream.cpp
Normal file
289
src/osgPlugins/mpeg/MpegImageStream.cpp
Normal file
@@ -0,0 +1,289 @@
|
||||
// -*-c++-*-
|
||||
|
||||
/*
|
||||
* Copyright (C) 2001 Ulrich Hertlein <u.hertlein@web.de>
|
||||
*
|
||||
* Uses libmpeg3 by Adam Williams
|
||||
* See http://www.heroinewarrior.com
|
||||
*
|
||||
* The Open Scene Graph (OSG) is a cross platform C++/OpenGL library for
|
||||
* real-time rendering of large 3D photo-realistic models.
|
||||
* The OSG homepage is http://www.openscenegraph.org/
|
||||
*
|
||||
* This software is free software; you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation; either version 2 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This software is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with this program; if not, write to the Free Software
|
||||
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
||||
*/
|
||||
|
||||
#include "MpegImageStream.h"
|
||||
#include <osg/Notify>
|
||||
#include <osg/Timer>
|
||||
|
||||
#include <stdio.h>
|
||||
#include <unistd.h>
|
||||
#include <sys/time.h>
|
||||
|
||||
#include "libmpeg3/libmpeg3.h"
|
||||
|
||||
using namespace osg;
|
||||
|
||||
#define IDLE_TIMEOUT 150000L
|
||||
|
||||
|
||||
// Constructor: setup and start thread
|
||||
MpegImageStream::MpegImageStream(const char* fileName) : ImageStream()
|
||||
{
|
||||
_useMMX = false;
|
||||
_fps = 0.0f;
|
||||
_frames = 0;
|
||||
_len = 0;
|
||||
_mpg = 0;
|
||||
|
||||
for (int i = 0; i < NUM_CMD_INDEX; i++)
|
||||
_cmd[i] = THREAD_IDLE;
|
||||
_wrIndex = _rdIndex = 0;
|
||||
|
||||
load(fileName);
|
||||
|
||||
::pthread_mutex_init(&_mutex, NULL);
|
||||
::pthread_create(&_id, NULL, MpegImageStream::s_decode, this);
|
||||
|
||||
if (fileName)
|
||||
setFileName(fileName);
|
||||
|
||||
|
||||
}
|
||||
|
||||
|
||||
// Deconstructor: stop and terminate thread
|
||||
MpegImageStream::~MpegImageStream()
|
||||
{
|
||||
stop();
|
||||
setCmd(THREAD_QUIT);
|
||||
::pthread_join(_id, NULL);
|
||||
::pthread_mutex_destroy(&_mutex);
|
||||
|
||||
mpeg3_t* mpg = (mpeg3_t*)_mpg;
|
||||
if (mpg) {
|
||||
mpeg3_close(mpg);
|
||||
mpg = NULL;
|
||||
}
|
||||
if (_rows) {
|
||||
::free(_rows);
|
||||
_rows = NULL;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
|
||||
// Set command
|
||||
void MpegImageStream::setCmd(ThreadCommand cmd)
|
||||
{
|
||||
lock();
|
||||
_cmd[_wrIndex] = cmd;
|
||||
_wrIndex = (_wrIndex + 1) % NUM_CMD_INDEX;
|
||||
unlock();
|
||||
}
|
||||
|
||||
|
||||
// Get command
|
||||
MpegImageStream::ThreadCommand MpegImageStream::getCmd()
|
||||
{
|
||||
ThreadCommand cmd = THREAD_IDLE;
|
||||
lock();
|
||||
if (_rdIndex != _wrIndex) {
|
||||
cmd = _cmd[_rdIndex];
|
||||
_rdIndex = (_rdIndex + 1) % NUM_CMD_INDEX;
|
||||
}
|
||||
unlock();
|
||||
return cmd;
|
||||
}
|
||||
|
||||
|
||||
/*
|
||||
* Decoder thread
|
||||
*/
|
||||
void* MpegImageStream::s_decode(void* vp)
|
||||
{
|
||||
return ((MpegImageStream*) vp)->decode(vp);
|
||||
}
|
||||
|
||||
void MpegImageStream::load(const char* fileName)
|
||||
{
|
||||
mpeg3_t* mpg = mpeg3_open((char*) fileName);
|
||||
if (!mpg) {
|
||||
osg::notify(WARN) << "Unable to open " << fileName << std::endl;
|
||||
return;
|
||||
}
|
||||
|
||||
if (!mpeg3_has_video(mpg)) {
|
||||
osg::notify(WARN) << "No video streams in" << fileName << std::endl;
|
||||
return;
|
||||
}
|
||||
if (mpeg3_has_audio(mpg)) {
|
||||
osg::notify(NOTICE) << "Stream has audio" << std::endl;
|
||||
}
|
||||
|
||||
_mpg = (void*)mpg;
|
||||
|
||||
mpeg3_set_cpus(mpg, 1);
|
||||
mpeg3_set_mmx(mpg, _useMMX);
|
||||
|
||||
int str = 0; //mpeg3_total_vstreams(mpg) - 1;
|
||||
|
||||
_fps = mpeg3_frame_rate(mpg, str);
|
||||
_frames = mpeg3_video_frames(mpg, str);
|
||||
_len = (float) _frames / _fps;
|
||||
|
||||
|
||||
int s = mpeg3_video_width(mpg, str);
|
||||
int t = mpeg3_video_height(mpg, str);
|
||||
|
||||
// Calculate texture size
|
||||
// these are also calculated and stored within osg::Texture but
|
||||
// too late (on the first apply) to be of any use...
|
||||
int texWidth = 1;
|
||||
for (; texWidth < s; texWidth <<= 1)
|
||||
;
|
||||
int texHeight = 1;
|
||||
for (; texHeight < t; texHeight <<= 1)
|
||||
;
|
||||
|
||||
|
||||
// Allocate image data
|
||||
// maybe use BGR888 and save some conversion somewhere?
|
||||
unsigned char* data = (unsigned char*) ::malloc(s * t * 3);
|
||||
|
||||
|
||||
setImage(s, t, 0,
|
||||
GL_RGB,
|
||||
GL_RGB, GL_UNSIGNED_BYTE, data,
|
||||
osg::Image::USE_MALLOC_FREE);
|
||||
|
||||
// Allocate decoder rows
|
||||
// documentation says we need add'l bytes at the end of each
|
||||
// row for MMX but this is more efficient and works so far.
|
||||
_rows = (unsigned char**) ::malloc(t * sizeof(unsigned char*));
|
||||
unsigned char* dp = data;
|
||||
for (int i = 0; i < t; i++) {
|
||||
_rows[t-i-1] = dp;
|
||||
dp += (s * 3);
|
||||
}
|
||||
|
||||
//
|
||||
// #if 0
|
||||
// // Setup texture matrix
|
||||
// Matrix mat;
|
||||
// mat.makeScale((float) s / (float) texWidth,
|
||||
// ((float) t / (float) texHeight) * -1.0f, 1.0f);
|
||||
// mat = mat * Matrix::translate(0.0f, (float) t / (float) texHeight, 0.0f);
|
||||
// _texMat->setMatrix(mat);
|
||||
// #else
|
||||
// _texMat->setMatrix(osg::Matrix::scale(s,-t,1.0f)*osg::Matrix::translate(0.0f,t,0.0f));
|
||||
// #endif
|
||||
// XXX
|
||||
osg::notify(NOTICE) << _frames << " @ " << _fps << " " << _len << "s" << std::endl;
|
||||
osg::notify(NOTICE) << "img " << s << "x" << t << std::endl;
|
||||
osg::notify(NOTICE) << "tex " << texWidth << "x" << texHeight << std::endl;
|
||||
}
|
||||
|
||||
|
||||
void* MpegImageStream::decode(void*)
|
||||
{
|
||||
bool playing = false;
|
||||
mpeg3_t* mpg = (mpeg3_t*)_mpg;
|
||||
int str = 0;
|
||||
float t0 = 0.0f;
|
||||
unsigned long delay = (unsigned long) ((1.0f / _fps) * 1000.0f);
|
||||
|
||||
bool done = false;
|
||||
|
||||
const osg::Timer* timer = osg::Timer::instance();
|
||||
osg::Timer_t start_tick = timer->tick();
|
||||
osg::Timer_t last_frame_tick = start_tick;
|
||||
double timePerFrame = 1.0f/_fps;
|
||||
double frameNumber = 0.0;
|
||||
|
||||
while (!done) {
|
||||
|
||||
// Handle commands
|
||||
ThreadCommand cmd = getCmd();
|
||||
if (cmd != THREAD_IDLE) {
|
||||
switch (cmd) {
|
||||
case THREAD_START: // Start or continue stream
|
||||
playing = true;
|
||||
break;
|
||||
case THREAD_STOP: // XXX
|
||||
osg::notify(NOTICE) << "stop at " << t0 << std::endl;
|
||||
playing = false;
|
||||
break;
|
||||
case THREAD_REWIND: // XXX
|
||||
t0 = 0.0;
|
||||
mpeg3_seek_percentage(mpg, 0.0);
|
||||
start_tick = timer->tick();
|
||||
frameNumber = 0;
|
||||
break;
|
||||
case THREAD_CLOSE: // Stop and close
|
||||
playing = false;
|
||||
if (mpg) {
|
||||
mpeg3_close(mpg);
|
||||
mpg = NULL;
|
||||
}
|
||||
break;
|
||||
case THREAD_QUIT: // XXX
|
||||
osg::notify(NOTICE) << "quit" << std::endl;
|
||||
done = true;
|
||||
break;
|
||||
default:
|
||||
osg::notify(osg::WARN) << "Unknown command " << cmd << std::endl;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if (playing) {
|
||||
|
||||
|
||||
last_frame_tick = timer->tick();
|
||||
|
||||
// XXX needs more work to be real-time
|
||||
mpeg3_read_frame(mpg, _rows,
|
||||
0, 0, _s, _t,
|
||||
_s, _t,
|
||||
MPEG3_RGB888, str);
|
||||
dirty(); //Image();
|
||||
|
||||
++frameNumber;
|
||||
|
||||
if (frameNumber>=_frames)
|
||||
{
|
||||
|
||||
rewind();
|
||||
//stop();
|
||||
}
|
||||
else
|
||||
{
|
||||
while (timePerFrame*(frameNumber+1)>timer->delta_s(start_tick,timer->tick()))
|
||||
{
|
||||
::usleep(delay);
|
||||
}
|
||||
}
|
||||
}
|
||||
else {
|
||||
::usleep(IDLE_TIMEOUT);
|
||||
}
|
||||
}
|
||||
|
||||
// Cleanup decoder
|
||||
|
||||
return NULL;
|
||||
}
|
||||
128
src/osgPlugins/mpeg/MpegImageStream.h
Normal file
128
src/osgPlugins/mpeg/MpegImageStream.h
Normal file
@@ -0,0 +1,128 @@
|
||||
// -*-c++-*-
|
||||
|
||||
/*
|
||||
* Copyright (C) 2001 Ulrich Hertlein <u.hertlein@web.de>
|
||||
*
|
||||
* Uses libmpeg3 by Adam Williams
|
||||
* See http://www.heroinewarrior.com
|
||||
*
|
||||
* The Open Scene Graph (OSG) is a cross platform C++/OpenGL library for
|
||||
* real-time rendering of large 3D photo-realistic models.
|
||||
* The OSG homepage is http://www.openscenegraph.org/
|
||||
*
|
||||
* This software is free software; you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation; either version 2 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This software is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with this program; if not, write to the Free Software
|
||||
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
||||
*/
|
||||
|
||||
#ifndef _MPEGIMAGESTREAM_H_
|
||||
#define _MPEGIMAGESTREAM_H_
|
||||
|
||||
#include <osg/ImageStream>
|
||||
//#include "ImageStream.h"
|
||||
|
||||
#include <pthread.h>
|
||||
|
||||
#define NUM_CMD_INDEX 4
|
||||
|
||||
namespace osg {
|
||||
|
||||
/**
|
||||
* MPEG1/2 Image Stream class.
|
||||
*/
|
||||
class SG_EXPORT MpegImageStream : public osg::ImageStream
|
||||
{
|
||||
public:
|
||||
MpegImageStream(const char* fileName = NULL);
|
||||
|
||||
virtual Object* clone() const { return new MpegImageStream; }
|
||||
virtual bool isSameKindAs(const Object* obj) const {
|
||||
return dynamic_cast<const MpegImageStream*>(obj) != NULL;
|
||||
}
|
||||
virtual const char* className() const { return "MpegImageStream"; }
|
||||
|
||||
/// Start or continue stream.
|
||||
virtual inline void start() { setCmd(THREAD_START); }
|
||||
|
||||
/// Stop stream at current position.
|
||||
virtual inline void stop() { setCmd(THREAD_STOP); }
|
||||
|
||||
/// Rewind stream to beginning.
|
||||
virtual inline void rewind() { setCmd(THREAD_REWIND); }
|
||||
|
||||
/// Enable/disable MMX.
|
||||
inline void enableMMX(bool b) { _useMMX = b; }
|
||||
|
||||
/**
|
||||
* Set frame rate in fps.
|
||||
* This is overwritten by the actual frame rate of the stream when
|
||||
* it is opened.
|
||||
*/
|
||||
inline void setFrameRate(float fps) { _fps = (fps < 0.0f ? 0.0f : fps); }
|
||||
|
||||
/// Get frame rate in fps.
|
||||
inline float getFrameRate() const { return _fps; }
|
||||
|
||||
/// Get number of frames.
|
||||
inline long getNumFrames() const { return _frames; }
|
||||
|
||||
/// Get total length in seconds.
|
||||
inline float getLength() const { return _len; }
|
||||
|
||||
void load(const char* fileName);
|
||||
|
||||
protected:
|
||||
virtual ~MpegImageStream();
|
||||
|
||||
private:
|
||||
bool _useMMX;
|
||||
float _fps;
|
||||
long _frames;
|
||||
float _len;
|
||||
|
||||
enum ThreadCommand {
|
||||
THREAD_IDLE = 0,
|
||||
THREAD_START,
|
||||
THREAD_STOP,
|
||||
THREAD_REWIND,
|
||||
THREAD_CLOSE,
|
||||
THREAD_QUIT
|
||||
};
|
||||
ThreadCommand _cmd[NUM_CMD_INDEX];
|
||||
int _wrIndex, _rdIndex;
|
||||
|
||||
pthread_mutex_t _mutex;
|
||||
pthread_t _id;
|
||||
|
||||
// Lock/unlock object.
|
||||
inline void lock() { ::pthread_mutex_lock(&_mutex); }
|
||||
inline void unlock() { ::pthread_mutex_unlock(&_mutex); }
|
||||
|
||||
/// Set command.
|
||||
void setCmd(ThreadCommand cmd);
|
||||
|
||||
/// Get command.
|
||||
ThreadCommand getCmd();
|
||||
|
||||
/// Decoder hook.
|
||||
static void* s_decode(void*);
|
||||
void* decode(void*);
|
||||
|
||||
void* _mpg;
|
||||
unsigned char** _rows;
|
||||
|
||||
};
|
||||
|
||||
} // namespace
|
||||
|
||||
#endif
|
||||
41
src/osgPlugins/mpeg/ReaderWriterMPEG.cpp
Normal file
41
src/osgPlugins/mpeg/ReaderWriterMPEG.cpp
Normal file
@@ -0,0 +1,41 @@
|
||||
#include <osg/Image>
|
||||
#include <osg/Notify>
|
||||
#include <osg/Geode>
|
||||
#include <osg/GL>
|
||||
|
||||
#include <osgDB/Registry>
|
||||
#include <osgDB/FileNameUtils>
|
||||
#include <osgDB/FileUtils>
|
||||
|
||||
#include "MpegImageStream.h"
|
||||
|
||||
class ReaderWriterMPEG : public osgDB::ReaderWriter
|
||||
{
|
||||
public:
|
||||
|
||||
virtual const char* className() { return "MPEG ImageStream Reader"; }
|
||||
|
||||
virtual bool acceptsExtension(const std::string& extension)
|
||||
{
|
||||
return osgDB::equalCaseInsensitive(extension,"mpg") ||
|
||||
osgDB::equalCaseInsensitive(extension,"mpv");
|
||||
}
|
||||
|
||||
virtual ReadResult readImage(const std::string& file, const osgDB::ReaderWriter::Options*)
|
||||
{
|
||||
std::string ext = osgDB::getLowerCaseFileExtension(file);
|
||||
if (!acceptsExtension(ext)) return ReadResult::FILE_NOT_HANDLED;
|
||||
|
||||
std::string fileName = osgDB::findDataFile( file );
|
||||
if (fileName.empty()) return ReadResult::FILE_NOT_FOUND;
|
||||
|
||||
osg::MpegImageStream* mpeg = new osg::MpegImageStream(fileName.c_str());
|
||||
mpeg->start();
|
||||
|
||||
return mpeg;
|
||||
}
|
||||
};
|
||||
|
||||
// now register with Registry to instantiate the above
|
||||
// reader/writer.
|
||||
osgDB::RegisterReaderWriterProxy<ReaderWriterMPEG> g_readerWriter_MPEG_Proxy;
|
||||
Reference in New Issue
Block a user