nuclear@0: /* nuclear@0: --------------------------------------------------------------------------- nuclear@0: Open Asset Import Library (assimp) nuclear@0: --------------------------------------------------------------------------- nuclear@0: nuclear@0: Copyright (c) 2006-2012, assimp team nuclear@0: nuclear@0: All rights reserved. nuclear@0: nuclear@0: Redistribution and use of this software in source and binary forms, nuclear@0: with or without modification, are permitted provided that the following nuclear@0: conditions are met: nuclear@0: nuclear@0: * Redistributions of source code must retain the above nuclear@0: copyright notice, this list of conditions and the nuclear@0: following disclaimer. nuclear@0: nuclear@0: * Redistributions in binary form must reproduce the above nuclear@0: copyright notice, this list of conditions and the nuclear@0: following disclaimer in the documentation and/or other nuclear@0: materials provided with the distribution. nuclear@0: nuclear@0: * Neither the name of the assimp team, nor the names of its nuclear@0: contributors may be used to endorse or promote products nuclear@0: derived from this software without specific prior nuclear@0: written permission of the assimp team. nuclear@0: nuclear@0: THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS nuclear@0: "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT nuclear@0: LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR nuclear@0: A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT nuclear@0: OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, nuclear@0: SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT nuclear@0: LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, nuclear@0: DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY nuclear@0: THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT nuclear@0: (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE nuclear@0: OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. nuclear@0: --------------------------------------------------------------------------- nuclear@0: */ nuclear@0: nuclear@0: /** @file Implementation of the XGL/ZGL importer class */ nuclear@0: nuclear@0: #include "AssimpPCH.h" nuclear@0: #ifndef ASSIMP_BUILD_NO_XGL_IMPORTER nuclear@0: nuclear@0: #include "XGLLoader.h" nuclear@0: #include "ParsingUtils.h" nuclear@0: #include "fast_atof.h" nuclear@0: nuclear@0: #include "StreamReader.h" nuclear@0: #include "MemoryIOWrapper.h" nuclear@0: nuclear@0: using namespace Assimp; nuclear@0: using namespace irr; nuclear@0: using namespace irr::io; nuclear@0: nuclear@0: nuclear@0: // zlib is needed for compressed XGL files nuclear@0: #ifndef ASSIMP_BUILD_NO_COMPRESSED_XGL nuclear@0: # ifdef ASSIMP_BUILD_NO_OWN_ZLIB nuclear@0: # include nuclear@0: # else nuclear@0: # include "../contrib/zlib/zlib.h" nuclear@0: # endif nuclear@0: #endif nuclear@0: nuclear@0: nuclear@0: // scopeguard for a malloc'ed buffer nuclear@0: struct free_it nuclear@0: { nuclear@0: free_it(void* free) : free(free) {} nuclear@0: ~free_it() { nuclear@0: ::free(this->free); nuclear@0: } nuclear@0: nuclear@0: void* free; nuclear@0: }; nuclear@0: nuclear@0: namespace Assimp { // this has to be in here because LogFunctions is in ::Assimp nuclear@0: template<> const std::string LogFunctions::log_prefix = "XGL: "; nuclear@0: nuclear@0: } nuclear@0: nuclear@0: static const aiImporterDesc desc = { nuclear@0: "XGL Importer", nuclear@0: "", nuclear@0: "", nuclear@0: "", nuclear@0: aiImporterFlags_SupportTextFlavour, nuclear@0: 0, nuclear@0: 0, nuclear@0: 0, nuclear@0: 0, nuclear@0: "xgl zgl" nuclear@0: }; nuclear@0: nuclear@0: nuclear@0: // ------------------------------------------------------------------------------------------------ nuclear@0: // Constructor to be privately used by Importer nuclear@0: XGLImporter::XGLImporter() nuclear@0: {} nuclear@0: nuclear@0: // ------------------------------------------------------------------------------------------------ nuclear@0: // Destructor, private as well nuclear@0: XGLImporter::~XGLImporter() nuclear@0: {} nuclear@0: nuclear@0: // ------------------------------------------------------------------------------------------------ nuclear@0: // Returns whether the class can handle the format of the given file. nuclear@0: bool XGLImporter::CanRead( const std::string& pFile, IOSystem* pIOHandler, bool checkSig) const nuclear@0: { nuclear@0: /* NOTE: A simple check for the file extension is not enough nuclear@0: * here. XGL and ZGL are ok, but xml is too generic nuclear@0: * and might be collada as well. So open the file and nuclear@0: * look for typical signal tokens. nuclear@0: */ nuclear@0: const std::string extension = GetExtension(pFile); nuclear@0: nuclear@0: if (extension == "xgl" || extension == "zgl") { nuclear@0: return true; nuclear@0: } nuclear@0: else if (extension == "xml" || checkSig) { nuclear@0: ai_assert(pIOHandler != NULL); nuclear@0: nuclear@0: const char* tokens[] = {"","",""}; nuclear@0: return SearchFileHeaderForToken(pIOHandler,pFile,tokens,3); nuclear@0: } nuclear@0: return false; nuclear@0: } nuclear@0: nuclear@0: // ------------------------------------------------------------------------------------------------ nuclear@0: // Get a list of all file extensions which are handled by this class nuclear@0: const aiImporterDesc* XGLImporter::GetInfo () const nuclear@0: { nuclear@0: return &desc; nuclear@0: } nuclear@0: nuclear@0: // ------------------------------------------------------------------------------------------------ nuclear@0: // Imports the given file into the given scene structure. nuclear@0: void XGLImporter::InternReadFile( const std::string& pFile, nuclear@0: aiScene* pScene, IOSystem* pIOHandler) nuclear@0: { nuclear@0: #ifndef ASSIMP_BUILD_NO_COMPRESSED_XGL nuclear@0: Bytef* dest = NULL; nuclear@0: free_it free_it_really(dest); nuclear@0: #endif nuclear@0: nuclear@0: scene = pScene; nuclear@0: boost::shared_ptr stream( pIOHandler->Open( pFile, "rb")); nuclear@0: nuclear@0: // check whether we can read from the file nuclear@0: if( stream.get() == NULL) { nuclear@0: throw DeadlyImportError( "Failed to open XGL/ZGL file " + pFile + ""); nuclear@0: } nuclear@0: nuclear@0: // see if its compressed, if so uncompress it nuclear@0: if (GetExtension(pFile) == "zgl") { nuclear@0: #ifdef ASSIMP_BUILD_NO_COMPRESSED_XGL nuclear@0: ThrowException("Cannot read ZGL file since Assimp was built without compression support"); nuclear@0: #else nuclear@0: boost::scoped_ptr raw_reader(new StreamReaderLE(stream)); nuclear@0: nuclear@0: // build a zlib stream nuclear@0: z_stream zstream; nuclear@0: zstream.opaque = Z_NULL; nuclear@0: zstream.zalloc = Z_NULL; nuclear@0: zstream.zfree = Z_NULL; nuclear@0: zstream.data_type = Z_BINARY; nuclear@0: nuclear@0: // raw decompression without a zlib or gzip header nuclear@0: inflateInit2(&zstream, -MAX_WBITS); nuclear@0: nuclear@0: // skip two extra bytes, zgl files do carry a crc16 upfront (I think) nuclear@0: raw_reader->IncPtr(2); nuclear@0: nuclear@0: zstream.next_in = reinterpret_cast( raw_reader->GetPtr() ); nuclear@0: zstream.avail_in = raw_reader->GetRemainingSize(); nuclear@0: nuclear@0: size_t total = 0l; nuclear@0: nuclear@0: // and decompress the data .... do 1k chunks in the hope that we won't kill the stack nuclear@0: #define MYBLOCK 1024 nuclear@0: Bytef block[MYBLOCK]; nuclear@0: int ret; nuclear@0: do { nuclear@0: zstream.avail_out = MYBLOCK; nuclear@0: zstream.next_out = block; nuclear@0: ret = inflate(&zstream, Z_NO_FLUSH); nuclear@0: nuclear@0: if (ret != Z_STREAM_END && ret != Z_OK) { nuclear@0: ThrowException("Failure decompressing this file using gzip, seemingly it is NOT a compressed .XGL file"); nuclear@0: } nuclear@0: const size_t have = MYBLOCK - zstream.avail_out; nuclear@0: total += have; nuclear@0: dest = reinterpret_cast( realloc(dest,total) ); nuclear@0: memcpy(dest + total - have,block,have); nuclear@0: } nuclear@0: while (ret != Z_STREAM_END); nuclear@0: nuclear@0: // terminate zlib nuclear@0: inflateEnd(&zstream); nuclear@0: nuclear@0: // replace the input stream with a memory stream nuclear@0: stream.reset(new MemoryIOStream(reinterpret_cast(dest),total)); nuclear@0: #endif nuclear@0: } nuclear@0: nuclear@0: // construct the irrXML parser nuclear@0: CIrrXML_IOStreamReader st(stream.get()); nuclear@0: boost::scoped_ptr read( createIrrXMLReader((IFileReadCallBack*) &st) ); nuclear@0: reader = read.get(); nuclear@0: nuclear@0: // parse the XML file nuclear@0: TempScope scope; nuclear@0: nuclear@0: while (ReadElement()) { nuclear@0: if (!ASSIMP_stricmp(reader->getNodeName(),"world")) { nuclear@0: ReadWorld(scope); nuclear@0: } nuclear@0: } nuclear@0: nuclear@0: nuclear@0: std::vector& meshes = scope.meshes_linear; nuclear@0: std::vector& materials = scope.materials_linear; nuclear@0: if(!meshes.size() || !materials.size()) { nuclear@0: ThrowException("failed to extract data from XGL file, no meshes loaded"); nuclear@0: } nuclear@0: nuclear@0: // copy meshes nuclear@0: scene->mNumMeshes = static_cast(meshes.size()); nuclear@0: scene->mMeshes = new aiMesh*[scene->mNumMeshes](); nuclear@0: std::copy(meshes.begin(),meshes.end(),scene->mMeshes); nuclear@0: nuclear@0: // copy materials nuclear@0: scene->mNumMaterials = static_cast(materials.size()); nuclear@0: scene->mMaterials = new aiMaterial*[scene->mNumMaterials](); nuclear@0: std::copy(materials.begin(),materials.end(),scene->mMaterials); nuclear@0: nuclear@0: if (scope.light) { nuclear@0: scene->mNumLights = 1; nuclear@0: scene->mLights = new aiLight*[1]; nuclear@0: scene->mLights[0] = scope.light; nuclear@0: nuclear@0: scope.light->mName = scene->mRootNode->mName; nuclear@0: } nuclear@0: nuclear@0: scope.dismiss(); nuclear@0: } nuclear@0: nuclear@0: // ------------------------------------------------------------------------------------------------ nuclear@0: bool XGLImporter::ReadElement() nuclear@0: { nuclear@0: while(reader->read()) { nuclear@0: if (reader->getNodeType() == EXN_ELEMENT) { nuclear@0: return true; nuclear@0: } nuclear@0: } nuclear@0: return false; nuclear@0: } nuclear@0: nuclear@0: // ------------------------------------------------------------------------------------------------ nuclear@0: bool XGLImporter::ReadElementUpToClosing(const char* closetag) nuclear@0: { nuclear@0: while(reader->read()) { nuclear@0: if (reader->getNodeType() == EXN_ELEMENT) { nuclear@0: return true; nuclear@0: } nuclear@0: else if (reader->getNodeType() == EXN_ELEMENT_END && !ASSIMP_stricmp(reader->getNodeName(),closetag)) { nuclear@0: return false; nuclear@0: } nuclear@0: } nuclear@0: LogError("unexpected EOF, expected closing <" + std::string(closetag) + "> tag"); nuclear@0: return false; nuclear@0: } nuclear@0: nuclear@0: // ------------------------------------------------------------------------------------------------ nuclear@0: bool XGLImporter::SkipToText() nuclear@0: { nuclear@0: while(reader->read()) { nuclear@0: if (reader->getNodeType() == EXN_TEXT) { nuclear@0: return true; nuclear@0: } nuclear@0: else if (reader->getNodeType() == EXN_ELEMENT || reader->getNodeType() == EXN_ELEMENT_END) { nuclear@0: ThrowException("expected text contents but found another element (or element end)"); nuclear@0: } nuclear@0: } nuclear@0: return false; nuclear@0: } nuclear@0: nuclear@0: // ------------------------------------------------------------------------------------------------ nuclear@0: std::string XGLImporter::GetElementName() nuclear@0: { nuclear@0: const char* s = reader->getNodeName(); nuclear@0: size_t len = strlen(s); nuclear@0: nuclear@0: std::string ret; nuclear@0: ret.resize(len); nuclear@0: nuclear@0: std::transform(s,s+len,ret.begin(),::tolower); nuclear@0: return ret; nuclear@0: } nuclear@0: nuclear@0: // ------------------------------------------------------------------------------------------------ nuclear@0: void XGLImporter::ReadWorld(TempScope& scope) nuclear@0: { nuclear@0: while (ReadElementUpToClosing("world")) { nuclear@0: const std::string& s = GetElementName(); nuclear@0: // XXX right now we'd skip if it comes after nuclear@0: //