/* -*-c++-*- OpenSceneGraph - Copyright (C) 1998-2004 Robert Osfield * * This library is open source and may be redistributed and/or modified under * the terms of the OpenSceneGraph Public License (OSGPL) version 0.0 or * (at your option) any later version. The full license is in LICENSE file * included with this distribution, and on the openscenegraph.org website. * * This library 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 * OpenSceneGraph Public License for more details. */ #include #include #include #include #include #define SERIALIZER() OpenThreads::ScopedLock lock(_serializerMutex) class OSGA_Archive : public osgDB::Archive { public: OSGA_Archive(); virtual ~OSGA_Archive(); virtual const char* libraryName() const { return "osga"; } virtual const char* className() const { return "Archive"; } virtual bool acceptsExtension(const std::string& extension) const { return osgDB::equalCaseInsensitive(extension,"osga"); } /** open the archive.*/ virtual bool open(const std::string& filename, ArchiveStatus status, unsigned int indexBlockSizeHint=4096); /** open the archive for reading.*/ virtual bool open(std::istream& fin); /** close the archive.*/ virtual void close(); /** return true if file exists in archive.*/ virtual bool fileExists(const std::string& filename) const; /** Get the file name which represents the master file recorded in the Archive.*/ virtual std::string getMasterFileName() const; typedef std::vector FileNameList; /** Get the full list of file names available in the archive.*/ virtual bool getFileNames(FileNameList& fileNameList) const; /** Read an osg::Object of specified file name from the Archive.*/ virtual ReadResult readObject(const std::string& fileName,const Options* options=NULL) const; /** Read an osg::Image of specified file name from the Archive.*/ virtual ReadResult readImage(const std::string& fileName,const Options* options=NULL) const; /** Read an osg::HeightField of specified file name from the Archive.*/ virtual ReadResult readHeightField(const std::string& fileName,const Options* options=NULL) const; /** Read an osg::Node of specified file name from the Archive.*/ virtual ReadResult readNode(const std::string& fileName,const Options* options=NULL) const; /** Write an osg::Object with specified file name to the Archive.*/ virtual WriteResult writeObject(const osg::Object& obj,const std::string& fileName,const Options* options=NULL) const; /** Write an osg::Image with specified file name to the Archive.*/ virtual WriteResult writeImage(const osg::Image& image,const std::string& fileName,const Options* options=NULL) const; /** Write an osg::HeightField with specified file name to the Archive.*/ virtual WriteResult writeHeightField(const osg::HeightField& heightField,const std::string& fileName,const Options* options=NULL) const; /** Write an osg::Node with specified file name to the Archive.*/ virtual WriteResult writeNode(const osg::Node& node,const std::string& fileName,const Options* options=NULL) const; #if defined(_MSC_VER) typedef __int64 pos_type; typedef __int64 size_type; #else typedef unsigned long long pos_type; typedef unsigned long long size_type; #endif typedef std::pair PositionSizePair; typedef std::map FileNamePositionMap; protected: mutable osgDB::ReentrantMutex _serializerMutex; class IndexBlock; friend class IndexBlock; class IndexBlock : public osg::Referenced { public: IndexBlock(unsigned int blockSize=0); inline pos_type getPosition() const { return _filePosition; } inline unsigned int getBlockSize() const { return _blockSize; } void setPositionNextIndexBlock(pos_type position); inline pos_type getPositionNextIndexBlock() const { return _filePositionNextIndexBlock; } static IndexBlock* read(std::istream& in, bool doEndianSwap); std::string getFirstFileName() const; bool getFileReferences(FileNamePositionMap& indexMap) const; inline bool requiresWrite() const { return _requiresWrite; } void write(std::ostream& out); inline bool spaceAvailable(pos_type, size_type, const std::string& filename) const { unsigned requiredSize = sizeof(pos_type)+sizeof(size_type)+sizeof(unsigned int)+filename.size(); return (_offsetOfNextAvailableSpace + requiredSize)<_blockSize; } bool addFileReference(pos_type position, size_type size, const std::string& filename); protected: void allocateData(unsigned int blockSize); virtual ~IndexBlock(); bool _requiresWrite; pos_type _filePosition; unsigned int _blockSize; pos_type _filePositionNextIndexBlock; unsigned int _offsetOfNextAvailableSpace; char* _data; }; public: /** Functor used in internal implementations.*/ struct ReadFunctor { ReadFunctor(const std::string& filename, const osgDB::ReaderWriter::Options* options): _filename(filename), _options(options) {} virtual ~ReadFunctor() {} virtual osgDB::ReaderWriter::ReadResult doRead(osgDB::ReaderWriter& rw, std::istream& input) const = 0; std::string _filename; const osgDB::ReaderWriter::Options* _options; }; /** Functor used in internal implementations.*/ struct WriteFunctor { WriteFunctor(const std::string& filename, const osgDB::ReaderWriter::Options* options): _filename(filename), _options(options) {} virtual ~WriteFunctor() {} virtual osgDB::ReaderWriter::WriteResult doWrite(osgDB::ReaderWriter& rw, std::ostream& output) const = 0; std::string _filename; const osgDB::ReaderWriter::Options* _options; }; protected: struct ReadObjectFunctor; struct ReadImageFunctor; struct ReadHeightFieldFunctor; struct ReadNodeFunctor; struct WriteObjectFunctor; struct WriteImageFunctor; struct WriteHeightFieldFunctor; struct WriteNodeFunctor; osgDB::ReaderWriter::ReadResult read(const ReadFunctor& readFunctor); osgDB::ReaderWriter::WriteResult write(const WriteFunctor& writeFunctor); typedef std::list< osg::ref_ptr > IndexBlockList; bool _open(std::istream& fin); void writeIndexBlocks(); bool addFileReference(pos_type position, size_type size, const std::string& fileName); static float s_currentSupportedVersion; float _version; ArchiveStatus _status; std::ifstream _input; std::fstream _output; std::string _masterFileName; IndexBlockList _indexBlockList; FileNamePositionMap _indexMap; template static inline void _write(char* ptr, const T& value) { std::copy(reinterpret_cast(&value),reinterpret_cast(&value)+sizeof(value),ptr); } template static inline void _read(char* ptr, T& value) { std::copy(ptr,ptr+sizeof(value),reinterpret_cast(&value)); } };