Refactored ImageSequence to provided a cleaner and more robust implementation

This commit is contained in:
Robert Osfield 2008-10-06 17:02:20 +00:00
parent 7e43decf4c
commit 8147bd4df8
4 changed files with 216 additions and 192 deletions

View File

@ -19,6 +19,7 @@
#include <osg/StateAttribute> #include <osg/StateAttribute>
#include <list> #include <list>
#include <set>
namespace osg { namespace osg {
@ -48,8 +49,8 @@ class OSG_EXPORT ImageSequence : public ImageStream
virtual void setTimeMultiplier(double tm) { _timeMultiplier = tm; } virtual void setTimeMultiplier(double tm) { _timeMultiplier = tm; }
virtual double getTimeMultiplier() const { return _timeMultiplier; } virtual double getTimeMultiplier() const { return _timeMultiplier; }
typedef std::list< osg::ref_ptr<osg::Image> > Images; typedef std::vector< osg::ref_ptr<osg::Image> > Images;
typedef std::list< std::string > FileNames; typedef std::vector< std::string > FileNames;
virtual void seek(double time); virtual void seek(double time);
@ -75,11 +76,29 @@ class OSG_EXPORT ImageSequence : public ImageStream
void addImageFile(const std::string& fileName); void addImageFile(const std::string& fileName);
void setImageFile(unsigned int pos, const std::string& fileName);
std::string getImageFile(unsigned int pos) const;
unsigned int getNumImageFiles() const { return _fileNames.size(); }
FileNames& getFileNames() { return _fileNames; } FileNames& getFileNames() { return _fileNames; }
const FileNames& getFileNames() const { return _fileNames; } const FileNames& getFileNames() const { return _fileNames; }
void addImage(osg::Image* image); void addImage(osg::Image* image);
void setImage(int s,int t,int r,
GLint internalTextureformat,
GLenum pixelFormat,GLenum type,
unsigned char* data,
AllocationMode mode,
int packing=1) { Image::setImage(s,t,r,internalTextureformat, pixelFormat, type, data, mode, packing); }
void setImage(unsigned int pos, osg::Image* image);
Image* getImage(unsigned int pos);
const Image* getImage(unsigned int pos) const;
unsigned int getNumImages() const { return _images.size(); }
Images& getImages() { return _images; } Images& getImages() { return _images; }
const Images& getImages() const { return _images; } const Images& getImages() const { return _images; }
@ -95,10 +114,15 @@ class OSG_EXPORT ImageSequence : public ImageStream
virtual ~ImageSequence() {} virtual ~ImageSequence() {}
virtual void applyLoopingMode();
void setImageToChild(const osg::Image* image); void setImageToChild(const osg::Image* image);
void computeTimePerImage(); void computeTimePerImage();
int imageIndex(double time);
double _referenceTime; double _referenceTime;
double _timeMultiplier; double _timeMultiplier;
@ -107,24 +131,22 @@ class OSG_EXPORT ImageSequence : public ImageStream
double _timePerImage; double _timePerImage;
OpenThreads::Mutex _mutex; mutable OpenThreads::Mutex _mutex;
FileNames _fileNames; FileNames _fileNames;
FileNames::iterator _fileNamesIterator;
double _fileNamesIteratorTime;
Images _images; Images _images;
typedef std::pair< std::string, osg::ref_ptr<osg::Image> > FileNameImagePair; typedef std::set< std::string > FilesRequested;
typedef std::list< FileNameImagePair > FileNameImageList; FilesRequested _filesRequested;
FileNameImageList _filesRequested;
int _previousAppliedImageIndex;
Images::iterator _imageIterator;
double _imageIteratorTime;
bool _seekTimeSet; bool _seekTimeSet;
double _seekTime; double _seekTime;
}; };
} // namespace } // namespace

View File

@ -65,7 +65,14 @@ class OSG_EXPORT ImageStream : public Image
LOOPING LOOPING
}; };
void setLoopingMode(LoopingMode mode) { _loopingMode = mode; applyLoopingMode(); } void setLoopingMode(LoopingMode mode)
{
if (_loopingMode == mode) return;
_loopingMode = mode;
applyLoopingMode();
}
LoopingMode getLoopingMode() const { return _loopingMode; } LoopingMode getLoopingMode() const { return _loopingMode; }

View File

@ -296,7 +296,7 @@ class OSG_EXPORT NodeVisitor : public virtual Referenced
virtual osg::Image* readImageFile(const std::string& fileName) = 0; virtual osg::Image* readImageFile(const std::string& fileName) = 0;
virtual void requestImageFile(const std::string& fileName,osg::Object* attachmentPoint, double timeToMergeBy, const FrameStamp* framestamp) = 0; virtual void requestImageFile(const std::string& fileName,osg::Object* attachmentPoint, int attachmentIndex, double timeToMergeBy, const FrameStamp* framestamp) = 0;
protected: protected:
virtual ~ImageRequestHandler() {} virtual ~ImageRequestHandler() {}

View File

@ -45,14 +45,10 @@ ImageSequence::ImageSequence()
_length = 1.0; _length = 1.0;
_timePerImage = 1.0; _timePerImage = 1.0;
_fileNamesIterator = _fileNames.end();
_fileNamesIteratorTime = 0.0;
_imageIterator = _images.end();
_imageIteratorTime = 0.0;
_seekTime = 0.0; _seekTime = 0.0;
_seekTimeSet = false; _seekTimeSet = false;
_previousAppliedImageIndex = -1;
} }
ImageSequence::ImageSequence(const ImageSequence& is,const CopyOp& copyop): ImageSequence::ImageSequence(const ImageSequence& is,const CopyOp& copyop):
@ -63,14 +59,10 @@ ImageSequence::ImageSequence(const ImageSequence& is,const CopyOp& copyop):
_length(is._length), _length(is._length),
_timePerImage(is._timePerImage) _timePerImage(is._timePerImage)
{ {
_fileNamesIterator = _fileNames.end();
_fileNamesIteratorTime = 0.0;
_imageIterator = _images.end();
_imageIteratorTime = 0.0;
_seekTime = is._seekTime; _seekTime = is._seekTime;
_seekTimeSet = is._seekTimeSet; _seekTimeSet = is._seekTimeSet;
_previousAppliedImageIndex = -1;
} }
int ImageSequence::compare(const Image& rhs) const int ImageSequence::compare(const Image& rhs) const
@ -106,6 +98,12 @@ void ImageSequence::setMode(Mode mode)
void ImageSequence::setLength(double length) void ImageSequence::setLength(double length)
{ {
if (length<=0.0)
{
osg::notify(osg::NOTICE)<<"ImageSequence::setLength("<<length<<") invalid length value, must be greater than 0."<<std::endl;
return;
}
_length = length; _length = length;
computeTimePerImage(); computeTimePerImage();
} }
@ -117,74 +115,68 @@ void ImageSequence::computeTimePerImage()
else _timePerImage = _length; else _timePerImage = _length;
} }
void ImageSequence::setImageFile(unsigned int pos, const std::string& fileName)
{
OpenThreads::ScopedLock<OpenThreads::Mutex> lock(_mutex);
if (pos>=_fileNames.size()) _fileNames.resize(pos);
_fileNames[pos] = fileName;
}
std::string ImageSequence::getImageFile(unsigned int pos) const
{
OpenThreads::ScopedLock<OpenThreads::Mutex> lock(_mutex);
return pos<_fileNames.size() ? _fileNames[pos] : std::string();
}
void ImageSequence::addImageFile(const std::string& fileName) void ImageSequence::addImageFile(const std::string& fileName)
{ {
OpenThreads::ScopedLock<OpenThreads::Mutex> lock(_mutex); OpenThreads::ScopedLock<OpenThreads::Mutex> lock(_mutex);
_fileNames.push_back(fileName); _fileNames.push_back(fileName);
computeTimePerImage(); computeTimePerImage();
if (_fileNamesIterator==_fileNames.end())
{
_fileNamesIterator = _fileNames.begin();
_fileNamesIteratorTime = _referenceTime;
} }
void ImageSequence::setImage(unsigned int pos, osg::Image* image)
{
OpenThreads::ScopedLock<OpenThreads::Mutex> lock(_mutex);
osg::notify(osg::INFO)<<"ImageSequence::setImage("<<pos<<","<<image->getFileName()<<")"<<std::endl;
if (pos>=_images.size()) _images.resize(pos+1);
_images[pos] = image;
// prune from file requested list.
FilesRequested::iterator itr = _filesRequested.find(image->getFileName());
if (itr!=_filesRequested.end()) _filesRequested.erase(itr);
}
Image* ImageSequence::getImage(unsigned int pos)
{
OpenThreads::ScopedLock<OpenThreads::Mutex> lock(_mutex);
return pos<_images.size() ? _images[pos].get() : 0;
}
const Image* ImageSequence::getImage(unsigned int pos) const
{
OpenThreads::ScopedLock<OpenThreads::Mutex> lock(_mutex);
return pos<_images.size() ? _images[pos].get() : 0;
} }
void ImageSequence::addImage(osg::Image* image) void ImageSequence::addImage(osg::Image* image)
{ {
if (image==0) return;
OpenThreads::ScopedLock<OpenThreads::Mutex> lock(_mutex); OpenThreads::ScopedLock<OpenThreads::Mutex> lock(_mutex);
if (!_filesRequested.empty())
{
// follows is a mechanism that ensures that requested images
// get merged in the correct time order.
if (_filesRequested.front().first != image->getFileName())
{
for(FileNameImageList::iterator itr = _filesRequested.begin();
itr != _filesRequested.end();
++itr)
{
if (itr->first == image->getFileName())
{
osg::notify(osg::NOTICE)<<"inserting image into waiting queue : "<<image->getFileName()<<std::endl;
itr->second = image;
return;
}
}
// osg::notify(osg::NOTICE)<<"image not expected : "<<image->getFileName()<<std::endl;
_images.push_back(image);
}
else
{
// osg::notify(osg::NOTICE)<<"merging image in order expected : "<<image->getFileName()<<std::endl; // osg::notify(osg::NOTICE)<<"merging image in order expected : "<<image->getFileName()<<std::endl;
_images.push_back(image); _images.push_back(image);
_filesRequested.pop_front();
FileNameImageList::iterator itr;
for(itr = _filesRequested.begin();
itr != _filesRequested.end() && itr->second.valid();
++itr)
{
// osg::notify(osg::NOTICE)<<" merging previously loaded, but out of order file : "<<itr->first<<std::endl;
_images.push_back(itr->second);
}
_filesRequested.erase(_filesRequested.begin(), itr);
}
}
else
{
_images.push_back(image);
}
computeTimePerImage(); computeTimePerImage();
if (data()==0) if (data()==0)
{ {
_imageIterator = _images.begin(); setImageToChild(_images.front().get());
_imageIteratorTime = 0.0;
setImageToChild(_imageIterator->get());
} }
} }
@ -194,6 +186,9 @@ void ImageSequence::setImageToChild(const osg::Image* image)
if (image==0) return; if (image==0) return;
// check to see if data is changing, if not don't apply
if (image->data() == data()) return;
setImage(image->s(),image->t(),image->r(), setImage(image->s(),image->t(),image->r(),
image->getInternalTextureFormat(), image->getInternalTextureFormat(),
image->getPixelFormat(),image->getDataType(), image->getPixelFormat(),image->getDataType(),
@ -202,6 +197,24 @@ void ImageSequence::setImageToChild(const osg::Image* image)
image->getPacking()); image->getPacking());
} }
void ImageSequence::applyLoopingMode()
{
}
int ImageSequence::imageIndex(double time)
{
if (getLoopingMode()==LOOPING)
{
double positionRatio = time/_length;
time = (positionRatio - floor(positionRatio))*_length;
}
if (time<0.0) return 0;
int index = int(time/_timePerImage);
if (index>=_images.size()) return _images.size()-1;
return index;
}
void ImageSequence::update(osg::NodeVisitor* nv) void ImageSequence::update(osg::NodeVisitor* nv)
{ {
OpenThreads::ScopedLock<OpenThreads::Mutex> lock(_mutex); OpenThreads::ScopedLock<OpenThreads::Mutex> lock(_mutex);
@ -224,17 +237,28 @@ void ImageSequence::update(osg::NodeVisitor* nv)
time = _seekTime; time = _seekTime;
_referenceTime = fs->getSimulationTime() - time/_timeMultiplier; _referenceTime = fs->getSimulationTime() - time/_timeMultiplier;
} }
else
{
if (looping)
{
while (time>_length)
{
_referenceTime += _length/_timeMultiplier;
time -= _length;
}
}
else
{
if (time>_length) if (time>_length)
{ {
time -= floor(time/_length)*_length; _referenceTime = fs->getSimulationTime() - _length/_timeMultiplier;
time = _length;
}
}
} }
_seekTime = time; _seekTime = time;
// _seekTimeSet = false; _seekTimeSet = false;
FileNames::iterator previous_fileNamesIterator = _fileNamesIterator;
Images::iterator previous_imageIterator = _imageIterator;
bool pruneOldImages = false; bool pruneOldImages = false;
@ -269,133 +293,104 @@ void ImageSequence::update(osg::NodeVisitor* nv)
} }
} }
int index = int(time/_timePerImage);
// osg::notify(osg::NOTICE)<<"time= "<<time<<" _timePerImage="<<_timePerImage<<" index="<<index<<" _length="<<_length<<std::endl;
if (index>=_images.size()) index = _images.size()-1;
if (index>=0 && index<_images.size())
{
if (pruneOldImages)
{
while (index>=0 && !_images[index].valid())
{
--index;
}
}
if (index>=0)
{
// osg::notify(osg::NOTICE)<<"at time "<<time<<" setting child = "<<index<<std::endl;
if (_previousAppliedImageIndex!=index)
{
if (_previousAppliedImageIndex >= 0 &&
_previousAppliedImageIndex<_images.size() &&
pruneOldImages)
{
_images[_previousAppliedImageIndex] = 0;
}
setImageToChild(_images[index].get());
_previousAppliedImageIndex = index;
}
}
}
// osg::notify(osg::NOTICE)<<"time = "<<time<<std::endl; // osg::notify(osg::NOTICE)<<"time = "<<time<<std::endl;
if (irh && _images.size()<_fileNames.size()) if (irh)
{ {
double preLoadTime = time + osg::minimum(irh->getPreLoadTime()*_timeMultiplier, _length); double preLoadTime = time + osg::minimum(irh->getPreLoadTime()*_timeMultiplier, _length);
if (preLoadTime>=_length) int startLoadIndex = int(time/_timePerImage);
if (startLoadIndex>=_images.size()) startLoadIndex = _images.size()-1;
if (startLoadIndex<0) startLoadIndex = 0;
int endLoadIndex = int(preLoadTime/_timePerImage);
if (endLoadIndex>=_fileNames.size())
{ {
//
// Advance fileNameIterator to end and wrap around
//
for(;
_fileNamesIterator != _fileNames.end();
++_fileNamesIterator)
{
_fileNamesIteratorTime += _timePerImage;
double effectTime = fs->getSimulationTime() + (preLoadTime - _fileNamesIteratorTime);
_filesRequested.push_back(FileNameImagePair(*_fileNamesIterator,0));
irh->requestImageFile(*_fileNamesIterator, this, effectTime, fs);
}
preLoadTime -= _length;
if (looping) if (looping)
{ {
_fileNamesIterator = _fileNames.begin(); endLoadIndex -= _fileNames.size();
_fileNamesIteratorTime = 0.0;
}
}
if (_fileNamesIterator!=_fileNames.end())
{
//
// Advance fileNameIterator to encmpass preLoadTime
//
//osg::notify(osg::NOTICE)<<" _fileNamesIteratorTime = "<<_fileNamesIteratorTime<<" "<<_timePerImage<<std::endl;
while(preLoadTime > (_fileNamesIteratorTime + _timePerImage))
{
_fileNamesIteratorTime += _timePerImage;
//osg::notify(osg::NOTICE)<<" _fileNamesIteratorTime = "<<_fileNamesIteratorTime<<std::endl;
//osg::notify(osg::NOTICE)<<" need to preLoad = "<<*_fileNamesIterator<<std::endl;
++_fileNamesIterator;
if (previous_fileNamesIterator==_fileNamesIterator) break;
if (_fileNamesIterator ==_fileNames.end())
{
// return iterator to begining of set.
if (looping) _fileNamesIterator = _fileNames.begin();
else break;
}
double effectTime = fs->getSimulationTime() + (preLoadTime - _fileNamesIteratorTime);
_filesRequested.push_back(FileNameImagePair(*_fileNamesIterator,0));
irh->requestImageFile(*_fileNamesIterator, this, effectTime, fs);
}
}
if (looping && _fileNamesIterator==_fileNames.end())
{
_fileNamesIterator = _fileNames.begin();
_fileNamesIteratorTime = 0.0;
}
}
{
//
// Advance imageIterator
//
if ((looping || _seekTimeSet) && time<_imageIteratorTime)
{
_imageIterator = _images.begin();
_imageIteratorTime = 0.0;
}
if (_imageIterator!=_images.end())
{
// osg::notify(osg::NOTICE)<<" _imageIteratorTime = "<<_imageIteratorTime<<std::endl;
while(time > (_imageIteratorTime + _timePerImage))
{
_imageIteratorTime += _timePerImage;
// osg::notify(osg::NOTICE)<<" _imageIteratorTime = "<<_imageIteratorTime<<std::endl;
++_imageIterator;
if (_imageIterator ==_images.end())
{
if (pruneOldImages)
{
_images.erase(previous_imageIterator, _imageIterator);
previous_imageIterator = _images.begin();
}
if (looping)
{
// return iterator to begining of set.
_imageIterator = _images.begin();
} }
else else
{ {
break; endLoadIndex = _fileNames.size()-1;
} }
} }
if (endLoadIndex<0) endLoadIndex = 0;
double requestTime = time;
if (endLoadIndex<startLoadIndex)
{
for(int i=startLoadIndex; i<_fileNames.size(); ++i)
{
if ((i>=_images.size() || !_images[i]) && _filesRequested.count(_fileNames[i])==0)
{
_filesRequested.insert(_fileNames[i]);
irh->requestImageFile(_fileNames[i], this, i, requestTime, fs);
}
requestTime += _timePerImage;
}
for(int i=0; i<=endLoadIndex; ++i)
{
if ((i>=_images.size() || !_images[i]) && _filesRequested.count(_fileNames[i])==0)
{
_filesRequested.insert(_fileNames[i]);
irh->requestImageFile(_fileNames[i], this, i, requestTime, fs);
}
requestTime += _timePerImage;
}
}
else
{
for(int i=startLoadIndex; i<=endLoadIndex; ++i)
{
if ((i>=_images.size() || !_images[i]) && _filesRequested.count(_fileNames[i])==0)
{
_filesRequested.insert(_fileNames[i]);
irh->requestImageFile(_fileNames[i], this, i, requestTime, fs);
}
requestTime += _timePerImage;
} }
} }
if (looping && _imageIterator==_images.end())
{
_imageIterator = _images.begin();
}
} }
if (_imageIterator!=_images.end() && previous_imageIterator != _imageIterator)
{
if (pruneOldImages)
{
_images.erase(previous_imageIterator, _imageIterator);
}
setImageToChild(_imageIterator->get());
}
_seekTimeSet = false;
} }