FGFS-4.1/OpenSceneGraph
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added support to osgProducer::Viewer for forcing an early exit form applications

Browse Source 
and recording an image on exit.
This commit is contained in:
Robert Osfield
2004-10-05 09:53:36 +00:00
parent b0d8ca535d
commit 68c7a47388
4 changed files with 200 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

87
src/osgProducer/Viewer.cpp
View File

@@ -225,7 +225,13 @@ protected:
// osgProducer::Viewer implemention
//
Viewer::Viewer():
_setDoneAtElapsedTimeEnabled(false),
_setDoneAtElapsedTime(0.0),
_setDoneAtFrameNumberEnabled(false),
_setDoneAtFrameNumber(0),
_done(false),
_writeImageWhenDone(false),
_writeImageFileName("saved_image.jpg"),
_recordingAnimationPath(false),
_recordingStartTime(0.0)
{
@@ -237,7 +243,13 @@ Viewer::Viewer():
Viewer::Viewer(Producer::CameraConfig *cfg):
OsgCameraGroup(cfg),
_setDoneAtElapsedTimeEnabled(false),
_setDoneAtElapsedTime(0.0),
_setDoneAtFrameNumberEnabled(false),
_setDoneAtFrameNumber(0),
_done(false),
_writeImageWhenDone(false),
_writeImageFileName("saved_image.jpg"),
_recordingAnimationPath(false),
_recordingStartTime(0.0)
{
@@ -250,7 +262,13 @@ Viewer::Viewer(Producer::CameraConfig *cfg):
Viewer::Viewer(const std::string& configFile):
OsgCameraGroup(configFile),
_setDoneAtElapsedTimeEnabled(false),
_setDoneAtElapsedTime(0.0),
_setDoneAtFrameNumberEnabled(false),
_setDoneAtFrameNumber(0),
_done(false),
_writeImageWhenDone(false),
_writeImageFileName("saved_image.jpg"),
_recordingAnimationPath(false),
_recordingStartTime(0.0)
{
@@ -262,7 +280,13 @@ Viewer::Viewer(const std::string& configFile):
Viewer::Viewer(osg::ArgumentParser& arguments):
OsgCameraGroup(arguments),
_setDoneAtElapsedTimeEnabled(false),
_setDoneAtElapsedTime(0.0),
_setDoneAtFrameNumberEnabled(false),
_setDoneAtFrameNumber(0),
_done(false),
_writeImageWhenDone(false),
_writeImageFileName("saved_image.jpg"),
_recordingAnimationPath(false),
_recordingStartTime(0.0)
{
@@ -275,6 +299,8 @@ Viewer::Viewer(osg::ArgumentParser& arguments):
if (arguments.getApplicationUsage())
{
arguments.getApplicationUsage()->addCommandLineOption("-p <filename>","Specify camera path file to animate the camera through the loaded scene");
arguments.getApplicationUsage()->addCommandLineOption("--run-till-frame-number <integer>","Specify the number of frame to run");
arguments.getApplicationUsage()->addCommandLineOption("--run-till-elapsed-time","Specify the about of time to run");
}
osg::DisplaySettings::instance()->readCommandLine(arguments);
@@ -290,12 +316,52 @@ Viewer::Viewer(osg::ArgumentParser& arguments):
selectCameraManipulator(num);
}
}
unsigned int frameNumber;
while (arguments.read("--run-till-frame-number",frameNumber))
{
setDoneAtFrameNumber(frameNumber);
}
double elapsedTime;
while (arguments.read("--run-till-elapsed-time",elapsedTime))
{
setDoneAtElapsedTime(elapsedTime);
}
std::string filename;
while (arguments.read("--write-image-when-done",filename))
{
setWriteImageWhenDone(true);
setWriteImageFileName(filename);
}
}
Viewer::~Viewer()
{
}
void Viewer::setWriteImageFileName(const std::string& filename)
{
_writeImageFileName = filename;
for(EventHandlerList::iterator itr = getEventHandlerList().begin();
itr != getEventHandlerList().end();
++itr)
{
ViewerEventHandler* viewerEventHandler = dynamic_cast<ViewerEventHandler*>(itr->get());
if (viewerEventHandler)
{
viewerEventHandler->setWriteImageFileName(filename);
}
}
}
const std::string& Viewer::getWriteImageFileName() const
{
return _writeImageFileName;
}
void Viewer::setCoordindateSystemNodePath(const osg::NodePath& nodePath)
{
_coordinateSystemNodePath.clear();
@@ -454,7 +520,10 @@ unsigned int Viewer::addCameraManipulator(osgGA::MatrixManipulator* cm)
bool Viewer::done() const
{
return _done || !validForRendering();
return _done ||
!validForRendering() ||
(_setDoneAtElapsedTimeEnabled && _setDoneAtElapsedTime<=getFrameStamp()->getReferenceTime()) ||
(_setDoneAtFrameNumberEnabled && _setDoneAtFrameNumber<=_frameNumber);
}
void Viewer::setViewByMatrix( const Producer::Matrix & pm)
@@ -641,6 +710,22 @@ void Viewer::frame()
getAnimationPath()->insert(_frameStamp->getReferenceTime()-_recordingStartTime,osg::AnimationPath::ControlPoint(osg::Vec3(_position[0],_position[1],_position[2]),_orientation));
}
if (done() && getWriteImageWhenDone())
{
for(EventHandlerList::iterator itr = getEventHandlerList().begin();
itr != getEventHandlerList().end();
++itr)
{
ViewerEventHandler* viewerEventHandler = dynamic_cast<ViewerEventHandler*>(itr->get());
if (viewerEventHandler)
{
osg::notify(osg::NOTICE)<<"Need to write image"<<std::endl;
viewerEventHandler->setWriteImageOnNextFrame(true);
}
}
}
OsgCameraGroup::frame();

69
src/osgProducer/ViewerEventHandler.cpp
View File

@@ -1,6 +1,7 @@
#include <osgProducer/ViewerEventHandler>
#include <osgGA/AnimationPathManipulator>
#include <osgDB/WriteFile>
#include <osgDB/FileNameUtils>
#include <osgText/Text>
#include <osg/BlendFunc>
@@ -14,12 +15,14 @@ class ViewerEventHandler::SnapImageDrawCallback : public Producer::Camera::Callb
{
public:
SnapImageDrawCallback(const std::string& filename):
_filename(filename),
SnapImageDrawCallback():
_snapImageOnNextFrame(false)
{
}
void setFileName(const std::string& filename) { _filename = filename; }
const std::string& getFileName() const { return _filename; }
void setSnapImageOnNextFrame(bool flag) { _snapImageOnNextFrame = flag; }
bool getSnapImageOnNextFrame() const { return _snapImageOnNextFrame; }
@@ -716,28 +719,56 @@ ViewerEventHandler::ViewerEventHandler(OsgCameraGroup* cg):
_statsAndHelpDrawCallback = new StatsAndHelpDrawCallback(this,0);
cam->addPostDrawCallback(_statsAndHelpDrawCallback);
std::string basename("saved_image");
std::string ext(".jpg");
if (cfg->getNumberOfCameras()==1)
for(unsigned int i=0;i<cfg->getNumberOfCameras();++i)
{
SnapImageDrawCallback* snapImageDrawCallback = new SnapImageDrawCallback(basename+ext);
cam->addPostDrawCallback(snapImageDrawCallback);
SnapImageDrawCallback* snapImageDrawCallback = new SnapImageDrawCallback();
cfg->getCamera(i)->addPostDrawCallback(snapImageDrawCallback);
_snapImageDrawCallbackList.push_back(snapImageDrawCallback);
}
else
Viewer* viewer = dynamic_cast<Viewer*>(cg);
if (viewer) setWriteImageFileName(viewer->getWriteImageFileName());
else setWriteImageFileName("saved_image.jpg");
}
void ViewerEventHandler::setWriteImageOnNextFrame(bool writeImageOnNextFrame)
{
for(SnapImageDrawCallbackList::iterator itr=_snapImageDrawCallbackList.begin();
itr!=_snapImageDrawCallbackList.end();
++itr)
{
for(unsigned int i=0;i<cfg->getNumberOfCameras();++i)
(*itr)->setSnapImageOnNextFrame(writeImageOnNextFrame);
}
}
void ViewerEventHandler::setWriteImageFileName(const std::string& filename)
{
std::string basename = osgDB::getNameLessExtension(filename);
std::string ext = osgDB::getFileExtension(filename);
unsigned int cameraNum = 0;
for(SnapImageDrawCallbackList::iterator itr=_snapImageDrawCallbackList.begin();
itr!=_snapImageDrawCallbackList.end();
++itr, ++cameraNum)
{
if (cameraNum==0)
{
std::string filename(basename+"_");
filename += ('0'+i);
filename += ext;
SnapImageDrawCallback* snapImageDrawCallback = new SnapImageDrawCallback(filename);
cfg->getCamera(i)->addPostDrawCallback(snapImageDrawCallback);
_snapImageDrawCallbackList.push_back(snapImageDrawCallback);
(*itr)->setFileName(filename);
}
else
{
std::string name(basename+"_");
name += ('0'+cameraNum);
name += '.';
name += ext;
(*itr)->setFileName(name);
}
}
}
bool ViewerEventHandler::handle(const osgGA::GUIEventAdapter& ea,osgGA::GUIActionAdapter& aa)
{
if(!_cg) return false;
@@ -832,13 +863,7 @@ bool ViewerEventHandler::handle(const osgGA::GUIEventAdapter& ea,osgGA::GUIActio
case osgGA::GUIEventAdapter::KEY_Print :
case 'O' :
{
for(SnapImageDrawCallbackList::iterator itr=_snapImageDrawCallbackList.begin();
itr!=_snapImageDrawCallbackList.end();
++itr)
{
(*itr)->setSnapImageOnNextFrame(true);
}
setWriteImageOnNextFrame(true);
return true;
}
case '+' :