diff options
Diffstat (limited to 'src/3rdparty/assimp/code/LWOLoader.cpp')
-rw-r--r-- | src/3rdparty/assimp/code/LWOLoader.cpp | 2481 |
1 files changed, 1259 insertions, 1222 deletions
diff --git a/src/3rdparty/assimp/code/LWOLoader.cpp b/src/3rdparty/assimp/code/LWOLoader.cpp index b23a9fe6e..222ec8e2c 100644 --- a/src/3rdparty/assimp/code/LWOLoader.cpp +++ b/src/3rdparty/assimp/code/LWOLoader.cpp @@ -3,12 +3,12 @@ Open Asset Import Library (assimp) --------------------------------------------------------------------------- -Copyright (c) 2006-2012, assimp team +Copyright (c) 2006-2016, assimp team All rights reserved. -Redistribution and use of this software in source and binary forms, -with or without modification, are permitted provided that the following +Redistribution and use of this software in source and binary forms, +with or without modification, are permitted provided that the following conditions are met: * Redistributions of source code must retain the above @@ -25,16 +25,16 @@ conditions are met: derived from this software without specific prior written permission of the assimp team. -THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS -"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT +THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS +"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR -A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT +A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, -SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT +SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, -DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY -THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT -(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE +DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY +THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT +(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. --------------------------------------------------------------------------- */ @@ -43,1399 +43,1436 @@ OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. * @brief Implementation of the LWO importer class */ -#include "AssimpPCH.h" + #ifndef ASSIMP_BUILD_NO_LWO_IMPORTER // internal headers #include "LWOLoader.h" #include "StringComparison.h" #include "SGSpatialSort.h" -#include "ByteSwap.h" +#include "ByteSwapper.h" #include "ProcessHelper.h" #include "ConvertToLHProcess.h" +#include <assimp/IOSystem.hpp> +#include <memory> +#include <sstream> +#include <iomanip> + using namespace Assimp; static const aiImporterDesc desc = { - "LightWave/Modo Object Importer", - "", - "", - "http://www.newtek.com/lightwave.html\nhttp://www.luxology.com/modo/", - aiImporterFlags_SupportTextFlavour, - 0, - 0, - 0, - 0, - "lwo lxo" + "LightWave/Modo Object Importer", + "", + "", + "http://www.newtek.com/lightwave.html\nhttp://www.luxology.com/modo/", + aiImporterFlags_SupportTextFlavour, + 0, + 0, + 0, + 0, + "lwo lxo" }; // ------------------------------------------------------------------------------------------------ // Constructor to be privately used by Importer LWOImporter::LWOImporter() + : mIsLWO2(), + mIsLXOB(), + mLayers(), + mCurLayer(), + mTags(), + mMapping(), + mSurfaces(), + mFileBuffer(), + fileSize(), + pScene(), + configSpeedFlag(), + configLayerIndex(), + hasNamedLayer() {} // ------------------------------------------------------------------------------------------------ -// Destructor, private as well +// Destructor, private as well LWOImporter::~LWOImporter() {} // ------------------------------------------------------------------------------------------------ -// Returns whether the class can handle the format of the given file. +// Returns whether the class can handle the format of the given file. bool LWOImporter::CanRead( const std::string& pFile, IOSystem* pIOHandler, bool checkSig) const { - const std::string extension = GetExtension(pFile); - if (extension == "lwo" || extension == "lxo") { - return true; - } - - // if check for extension is not enough, check for the magic tokens - if (!extension.length() || checkSig) { - uint32_t tokens[3]; - tokens[0] = AI_LWO_FOURCC_LWOB; - tokens[1] = AI_LWO_FOURCC_LWO2; - tokens[2] = AI_LWO_FOURCC_LXOB; - return CheckMagicToken(pIOHandler,pFile,tokens,3,8); - } - return false; + const std::string extension = GetExtension(pFile); + if (extension == "lwo" || extension == "lxo") { + return true; + } + + // if check for extension is not enough, check for the magic tokens + if (!extension.length() || checkSig) { + uint32_t tokens[3]; + tokens[0] = AI_LWO_FOURCC_LWOB; + tokens[1] = AI_LWO_FOURCC_LWO2; + tokens[2] = AI_LWO_FOURCC_LXOB; + return CheckMagicToken(pIOHandler,pFile,tokens,3,8); + } + return false; } // ------------------------------------------------------------------------------------------------ // Setup configuration properties void LWOImporter::SetupProperties(const Importer* pImp) { - configSpeedFlag = ( 0 != pImp->GetPropertyInteger(AI_CONFIG_FAVOUR_SPEED,0) ? true : false); - configLayerIndex = pImp->GetPropertyInteger (AI_CONFIG_IMPORT_LWO_ONE_LAYER_ONLY,UINT_MAX); - configLayerName = pImp->GetPropertyString (AI_CONFIG_IMPORT_LWO_ONE_LAYER_ONLY,""); + configSpeedFlag = ( 0 != pImp->GetPropertyInteger(AI_CONFIG_FAVOUR_SPEED,0) ? true : false); + configLayerIndex = pImp->GetPropertyInteger (AI_CONFIG_IMPORT_LWO_ONE_LAYER_ONLY,UINT_MAX); + configLayerName = pImp->GetPropertyString (AI_CONFIG_IMPORT_LWO_ONE_LAYER_ONLY,""); } // ------------------------------------------------------------------------------------------------ // Get list of file extensions const aiImporterDesc* LWOImporter::GetInfo () const { - return &desc; + return &desc; } // ------------------------------------------------------------------------------------------------ -// Imports the given file into the given scene structure. -void LWOImporter::InternReadFile( const std::string& pFile, - aiScene* pScene, - IOSystem* pIOHandler) +// Imports the given file into the given scene structure. +void LWOImporter::InternReadFile( const std::string& pFile, + aiScene* pScene, + IOSystem* pIOHandler) { - boost::scoped_ptr<IOStream> file( pIOHandler->Open( pFile, "rb")); - - // Check whether we can read from the file - if( file.get() == NULL) - throw DeadlyImportError( "Failed to open LWO file " + pFile + "."); - - if((this->fileSize = (unsigned int)file->FileSize()) < 12) - throw DeadlyImportError("LWO: The file is too small to contain the IFF header"); - - // Allocate storage and copy the contents of the file to a memory buffer - std::vector< uint8_t > mBuffer(fileSize); - file->Read( &mBuffer[0], 1, fileSize); - this->pScene = pScene; - - // Determine the type of the file - uint32_t fileType; - const char* sz = IFF::ReadHeader(&mBuffer[0],fileType); - if (sz)throw DeadlyImportError(sz); - - mFileBuffer = &mBuffer[0] + 12; - fileSize -= 12; - - // Initialize some members with their default values - hasNamedLayer = false; - - // Create temporary storage on the stack but store pointers to it in the class - // instance. Therefore everything will be destructed properly if an exception - // is thrown and we needn't take care of that. - LayerList _mLayers; - SurfaceList _mSurfaces; - TagList _mTags; - TagMappingTable _mMapping; - - mLayers = &_mLayers; - mTags = &_mTags; - mMapping = &_mMapping; - mSurfaces = &_mSurfaces; - - // Allocate a default layer (layer indices are 1-based from now) - mLayers->push_back(Layer()); - mCurLayer = &mLayers->back(); - mCurLayer->mName = "<LWODefault>"; - mCurLayer->mIndex = -1; - - // old lightwave file format (prior to v6) - if (AI_LWO_FOURCC_LWOB == fileType) { - DefaultLogger::get()->info("LWO file format: LWOB (<= LightWave 5.5)"); - - mIsLWO2 = false; + std::unique_ptr<IOStream> file( pIOHandler->Open( pFile, "rb")); + + // Check whether we can read from the file + if( file.get() == NULL) + throw DeadlyImportError( "Failed to open LWO file " + pFile + "."); + + if((this->fileSize = (unsigned int)file->FileSize()) < 12) + throw DeadlyImportError("LWO: The file is too small to contain the IFF header"); + + // Allocate storage and copy the contents of the file to a memory buffer + std::vector< uint8_t > mBuffer(fileSize); + file->Read( &mBuffer[0], 1, fileSize); + this->pScene = pScene; + + // Determine the type of the file + uint32_t fileType; + const char* sz = IFF::ReadHeader(&mBuffer[0],fileType); + if (sz)throw DeadlyImportError(sz); + + mFileBuffer = &mBuffer[0] + 12; + fileSize -= 12; + + // Initialize some members with their default values + hasNamedLayer = false; + + // Create temporary storage on the stack but store pointers to it in the class + // instance. Therefore everything will be destructed properly if an exception + // is thrown and we needn't take care of that. + LayerList _mLayers; + SurfaceList _mSurfaces; + TagList _mTags; + TagMappingTable _mMapping; + + mLayers = &_mLayers; + mTags = &_mTags; + mMapping = &_mMapping; + mSurfaces = &_mSurfaces; + + // Allocate a default layer (layer indices are 1-based from now) + mLayers->push_back(Layer()); + mCurLayer = &mLayers->back(); + mCurLayer->mName = "<LWODefault>"; + mCurLayer->mIndex = -1; + + // old lightwave file format (prior to v6) + if (AI_LWO_FOURCC_LWOB == fileType) { + DefaultLogger::get()->info("LWO file format: LWOB (<= LightWave 5.5)"); + + mIsLWO2 = false; mIsLXOB = false; - LoadLWOBFile(); - } - // New lightwave format - else if (AI_LWO_FOURCC_LWO2 == fileType) { + LoadLWOBFile(); + } + // New lightwave format + else if (AI_LWO_FOURCC_LWO2 == fileType) { mIsLXOB = false; - DefaultLogger::get()->info("LWO file format: LWO2 (>= LightWave 6)"); - } - // MODO file format - else if (AI_LWO_FOURCC_LXOB == fileType) { + DefaultLogger::get()->info("LWO file format: LWO2 (>= LightWave 6)"); + } + // MODO file format + else if (AI_LWO_FOURCC_LXOB == fileType) { mIsLXOB = true; - DefaultLogger::get()->info("LWO file format: LXOB (Modo)"); - } - // we don't know this format - else - { - char szBuff[5]; - szBuff[0] = (char)(fileType >> 24u); - szBuff[1] = (char)(fileType >> 16u); - szBuff[2] = (char)(fileType >> 8u); - szBuff[3] = (char)(fileType); - szBuff[4] = '\0'; - throw DeadlyImportError(std::string("Unknown LWO sub format: ") + szBuff); - } - - if (AI_LWO_FOURCC_LWOB != fileType) { - mIsLWO2 = true; - LoadLWO2File(); - - // The newer lightwave format allows the user to configure the - // loader that just one layer is used. If this is the case - // we need to check now whether the requested layer has been found. - if (UINT_MAX != configLayerIndex) { - unsigned int layerCount = 0; - for(std::list<LWO::Layer>::iterator itLayers=mLayers->begin(); itLayers!=mLayers->end(); itLayers++) - if (!itLayers->skip) - layerCount++; - if (layerCount!=2) - throw DeadlyImportError("LWO2: The requested layer was not found"); - } - - if (configLayerName.length() && !hasNamedLayer) { - throw DeadlyImportError("LWO2: Unable to find the requested layer: " - + configLayerName); - } - } - - // now, as we have loaded all data, we can resolve cross-referenced tags and clips - ResolveTags(); - ResolveClips(); - - // now process all layers and build meshes and nodes - std::vector<aiMesh*> apcMeshes; - std::map<uint16_t, aiNode*> apcNodes; - - apcMeshes.reserve(mLayers->size()*std::min(((unsigned int)mSurfaces->size()/2u), 1u)); - - unsigned int iDefaultSurface = UINT_MAX; // index of the default surface - for (LayerList::iterator lit = mLayers->begin(), lend = mLayers->end();lit != lend;++lit) { - LWO::Layer& layer = *lit; - if (layer.skip) - continue; - - // I don't know whether there could be dummy layers, but it would be possible - const unsigned int meshStart = (unsigned int)apcMeshes.size(); - if (!layer.mFaces.empty() && !layer.mTempPoints.empty()) { - - // now sort all faces by the surfaces assigned to them - std::vector<SortedRep> pSorted(mSurfaces->size()+1); - - unsigned int i = 0; - for (FaceList::iterator it = layer.mFaces.begin(), end = layer.mFaces.end();it != end;++it,++i) { - // Check whether we support this face's type - if ((*it).type != AI_LWO_FACE && (*it).type != AI_LWO_PTCH && - (*it).type != AI_LWO_BONE && (*it).type != AI_LWO_SUBD) { - continue; - } - - unsigned int idx = (*it).surfaceIndex; - if (idx >= mTags->size()) - { - DefaultLogger::get()->warn("LWO: Invalid face surface index"); - idx = UINT_MAX; - } - if(UINT_MAX == idx || UINT_MAX == (idx = _mMapping[idx])) { - if (UINT_MAX == iDefaultSurface) { - iDefaultSurface = (unsigned int)mSurfaces->size(); - mSurfaces->push_back(LWO::Surface()); - LWO::Surface& surf = mSurfaces->back(); - surf.mColor.r = surf.mColor.g = surf.mColor.b = 0.6f; - surf.mName = "LWODefaultSurface"; - } - idx = iDefaultSurface; - } - pSorted[idx].push_back(i); - } - if (UINT_MAX == iDefaultSurface) { - pSorted.erase(pSorted.end()-1); - } - for (unsigned int p = 0,i = 0;i < mSurfaces->size();++i) { - SortedRep& sorted = pSorted[i]; - if (sorted.empty()) - continue; - - // generate the mesh - aiMesh* mesh = new aiMesh(); - apcMeshes.push_back(mesh); - mesh->mNumFaces = (unsigned int)sorted.size(); - - // count the number of vertices - SortedRep::const_iterator it = sorted.begin(), end = sorted.end(); - for (;it != end;++it) { - mesh->mNumVertices += layer.mFaces[*it].mNumIndices; - } - - aiVector3D *nrm = NULL, * pv = mesh->mVertices = new aiVector3D[mesh->mNumVertices]; - aiFace* pf = mesh->mFaces = new aiFace[mesh->mNumFaces]; - mesh->mMaterialIndex = i; - - // find out which vertex color channels and which texture coordinate - // channels are really required by the material attached to this mesh - unsigned int vUVChannelIndices[AI_MAX_NUMBER_OF_TEXTURECOORDS]; - unsigned int vVColorIndices[AI_MAX_NUMBER_OF_COLOR_SETS]; + DefaultLogger::get()->info("LWO file format: LXOB (Modo)"); + } + // we don't know this format + else + { + char szBuff[5]; + szBuff[0] = (char)(fileType >> 24u); + szBuff[1] = (char)(fileType >> 16u); + szBuff[2] = (char)(fileType >> 8u); + szBuff[3] = (char)(fileType); + szBuff[4] = '\0'; + throw DeadlyImportError(std::string("Unknown LWO sub format: ") + szBuff); + } + + if (AI_LWO_FOURCC_LWOB != fileType) { + mIsLWO2 = true; + LoadLWO2File(); + + // The newer lightwave format allows the user to configure the + // loader that just one layer is used. If this is the case + // we need to check now whether the requested layer has been found. + if (UINT_MAX != configLayerIndex) { + unsigned int layerCount = 0; + for(std::list<LWO::Layer>::iterator itLayers=mLayers->begin(); itLayers!=mLayers->end(); ++itLayers) + if (!itLayers->skip) + layerCount++; + if (layerCount!=2) + throw DeadlyImportError("LWO2: The requested layer was not found"); + } + + if (configLayerName.length() && !hasNamedLayer) { + throw DeadlyImportError("LWO2: Unable to find the requested layer: " + + configLayerName); + } + } + + // now, as we have loaded all data, we can resolve cross-referenced tags and clips + ResolveTags(); + ResolveClips(); + + // now process all layers and build meshes and nodes + std::vector<aiMesh*> apcMeshes; + std::map<uint16_t, aiNode*> apcNodes; + + apcMeshes.reserve(mLayers->size()*std::min(((unsigned int)mSurfaces->size()/2u), 1u)); + + unsigned int iDefaultSurface = UINT_MAX; // index of the default surface + for (LWO::Layer &layer : *mLayers) { + if (layer.skip) + continue; + + // I don't know whether there could be dummy layers, but it would be possible + const unsigned int meshStart = (unsigned int)apcMeshes.size(); + if (!layer.mFaces.empty() && !layer.mTempPoints.empty()) { + + // now sort all faces by the surfaces assigned to them + std::vector<SortedRep> pSorted(mSurfaces->size()+1); + + unsigned int i = 0; + for (FaceList::iterator it = layer.mFaces.begin(), end = layer.mFaces.end();it != end;++it,++i) { + // Check whether we support this face's type + if ((*it).type != AI_LWO_FACE && (*it).type != AI_LWO_PTCH && + (*it).type != AI_LWO_BONE && (*it).type != AI_LWO_SUBD) { + continue; + } + + unsigned int idx = (*it).surfaceIndex; + if (idx >= mTags->size()) + { + DefaultLogger::get()->warn("LWO: Invalid face surface index"); + idx = UINT_MAX; + } + if(UINT_MAX == idx || UINT_MAX == (idx = _mMapping[idx])) { + if (UINT_MAX == iDefaultSurface) { + iDefaultSurface = (unsigned int)mSurfaces->size(); + mSurfaces->push_back(LWO::Surface()); + LWO::Surface& surf = mSurfaces->back(); + surf.mColor.r = surf.mColor.g = surf.mColor.b = 0.6f; + surf.mName = "LWODefaultSurface"; + } + idx = iDefaultSurface; + } + pSorted[idx].push_back(i); + } + if (UINT_MAX == iDefaultSurface) { + pSorted.erase(pSorted.end()-1); + } + for (unsigned int p = 0,i = 0;i < mSurfaces->size();++i) { + SortedRep& sorted = pSorted[i]; + if (sorted.empty()) + continue; + + // generate the mesh + aiMesh* mesh = new aiMesh(); + apcMeshes.push_back(mesh); + mesh->mNumFaces = (unsigned int)sorted.size(); + + // count the number of vertices + SortedRep::const_iterator it = sorted.begin(), end = sorted.end(); + for (;it != end;++it) { + mesh->mNumVertices += layer.mFaces[*it].mNumIndices; + } + + aiVector3D *nrm = NULL, * pv = mesh->mVertices = new aiVector3D[mesh->mNumVertices]; + aiFace* pf = mesh->mFaces = new aiFace[mesh->mNumFaces]; + mesh->mMaterialIndex = i; + + // find out which vertex color channels and which texture coordinate + // channels are really required by the material attached to this mesh + unsigned int vUVChannelIndices[AI_MAX_NUMBER_OF_TEXTURECOORDS]; + unsigned int vVColorIndices[AI_MAX_NUMBER_OF_COLOR_SETS]; #ifdef ASSIMP_BUILD_DEBUG - for (unsigned int mui = 0; mui < AI_MAX_NUMBER_OF_TEXTURECOORDS;++mui ) { - vUVChannelIndices[mui] = UINT_MAX; - } - for (unsigned int mui = 0; mui < AI_MAX_NUMBER_OF_COLOR_SETS;++mui ) { - vVColorIndices[mui] = UINT_MAX; - } + for (unsigned int mui = 0; mui < AI_MAX_NUMBER_OF_TEXTURECOORDS;++mui ) { + vUVChannelIndices[mui] = UINT_MAX; + } + for (unsigned int mui = 0; mui < AI_MAX_NUMBER_OF_COLOR_SETS;++mui ) { + vVColorIndices[mui] = UINT_MAX; + } #endif - FindUVChannels(_mSurfaces[i],sorted,layer,vUVChannelIndices); - FindVCChannels(_mSurfaces[i],sorted,layer,vVColorIndices); - - // allocate storage for UV and CV channels - aiVector3D* pvUV[AI_MAX_NUMBER_OF_TEXTURECOORDS]; - for (unsigned int mui = 0; mui < AI_MAX_NUMBER_OF_TEXTURECOORDS;++mui ) { - if (UINT_MAX == vUVChannelIndices[mui]) { - break; - } - - pvUV[mui] = mesh->mTextureCoords[mui] = new aiVector3D[mesh->mNumVertices]; - - // LightWave doesn't support more than 2 UV components (?) - mesh->mNumUVComponents[0] = 2; - } - - if (layer.mNormals.name.length()) - nrm = mesh->mNormals = new aiVector3D[mesh->mNumVertices]; - - aiColor4D* pvVC[AI_MAX_NUMBER_OF_COLOR_SETS]; - for (unsigned int mui = 0; mui < AI_MAX_NUMBER_OF_COLOR_SETS;++mui) { - if (UINT_MAX == vVColorIndices[mui]) { - break; - } - pvVC[mui] = mesh->mColors[mui] = new aiColor4D[mesh->mNumVertices]; - } - - // we would not need this extra array, but the code is much cleaner if we use it - std::vector<unsigned int>& smoothingGroups = layer.mPointReferrers; - smoothingGroups.erase (smoothingGroups.begin(),smoothingGroups.end()); - smoothingGroups.resize(mesh->mNumFaces,0); - - // now convert all faces - unsigned int vert = 0; - std::vector<unsigned int>::iterator outIt = smoothingGroups.begin(); - for (it = sorted.begin(); it != end;++it,++outIt) { - const LWO::Face& face = layer.mFaces[*it]; - *outIt = face.smoothGroup; - - // copy all vertices - for (unsigned int q = 0; q < face.mNumIndices;++q,++vert) { - unsigned int idx = face.mIndices[q]; - *pv++ = layer.mTempPoints[idx] /*- layer.mPivot*/; - - // process UV coordinates - for (unsigned int w = 0; w < AI_MAX_NUMBER_OF_TEXTURECOORDS;++w) { - if (UINT_MAX == vUVChannelIndices[w]) { - break; - } - aiVector3D*& pp = pvUV[w]; - const aiVector2D& src = ((aiVector2D*)&layer.mUVChannels[vUVChannelIndices[w]].rawData[0])[idx]; - pp->x = src.x; - pp->y = src.y; - pp++; - } - - // process normals (MODO extension) - if (nrm) { - *nrm = ((aiVector3D*)&layer.mNormals.rawData[0])[idx]; - nrm->z *= -1.f; - ++nrm; - } - - // process vertex colors - for (unsigned int w = 0; w < AI_MAX_NUMBER_OF_COLOR_SETS;++w) { - if (UINT_MAX == vVColorIndices[w]) { - break; - } - *pvVC[w] = ((aiColor4D*)&layer.mVColorChannels[vVColorIndices[w]].rawData[0])[idx]; - - // If a RGB color map is explicitly requested delete the - // alpha channel - it could theoretically be != 1. - if(_mSurfaces[i].mVCMapType == AI_LWO_RGB) - pvVC[w]->a = 1.f; - - pvVC[w]++; - } + FindUVChannels(_mSurfaces[i],sorted,layer,vUVChannelIndices); + FindVCChannels(_mSurfaces[i],sorted,layer,vVColorIndices); + + // allocate storage for UV and CV channels + aiVector3D* pvUV[AI_MAX_NUMBER_OF_TEXTURECOORDS]; + for (unsigned int mui = 0; mui < AI_MAX_NUMBER_OF_TEXTURECOORDS;++mui ) { + if (UINT_MAX == vUVChannelIndices[mui]) { + break; + } + + pvUV[mui] = mesh->mTextureCoords[mui] = new aiVector3D[mesh->mNumVertices]; + + // LightWave doesn't support more than 2 UV components (?) + mesh->mNumUVComponents[0] = 2; + } + + if (layer.mNormals.name.length()) + nrm = mesh->mNormals = new aiVector3D[mesh->mNumVertices]; + + aiColor4D* pvVC[AI_MAX_NUMBER_OF_COLOR_SETS]; + for (unsigned int mui = 0; mui < AI_MAX_NUMBER_OF_COLOR_SETS;++mui) { + if (UINT_MAX == vVColorIndices[mui]) { + break; + } + pvVC[mui] = mesh->mColors[mui] = new aiColor4D[mesh->mNumVertices]; + } + + // we would not need this extra array, but the code is much cleaner if we use it + std::vector<unsigned int>& smoothingGroups = layer.mPointReferrers; + smoothingGroups.erase (smoothingGroups.begin(),smoothingGroups.end()); + smoothingGroups.resize(mesh->mNumFaces,0); + + // now convert all faces + unsigned int vert = 0; + std::vector<unsigned int>::iterator outIt = smoothingGroups.begin(); + for (it = sorted.begin(); it != end;++it,++outIt) { + const LWO::Face& face = layer.mFaces[*it]; + *outIt = face.smoothGroup; + + // copy all vertices + for (unsigned int q = 0; q < face.mNumIndices;++q,++vert) { + unsigned int idx = face.mIndices[q]; + *pv++ = layer.mTempPoints[idx] /*- layer.mPivot*/; + + // process UV coordinates + for (unsigned int w = 0; w < AI_MAX_NUMBER_OF_TEXTURECOORDS;++w) { + if (UINT_MAX == vUVChannelIndices[w]) { + break; + } + aiVector3D*& pp = pvUV[w]; + const aiVector2D& src = ((aiVector2D*)&layer.mUVChannels[vUVChannelIndices[w]].rawData[0])[idx]; + pp->x = src.x; + pp->y = src.y; + pp++; + } + + // process normals (MODO extension) + if (nrm) { + *nrm = ((aiVector3D*)&layer.mNormals.rawData[0])[idx]; + nrm->z *= -1.f; + ++nrm; + } + + // process vertex colors + for (unsigned int w = 0; w < AI_MAX_NUMBER_OF_COLOR_SETS;++w) { + if (UINT_MAX == vVColorIndices[w]) { + break; + } + *pvVC[w] = ((aiColor4D*)&layer.mVColorChannels[vVColorIndices[w]].rawData[0])[idx]; + + // If a RGB color map is explicitly requested delete the + // alpha channel - it could theoretically be != 1. + if(_mSurfaces[i].mVCMapType == AI_LWO_RGB) + pvVC[w]->a = 1.f; + + pvVC[w]++; + } #if 0 - // process vertex weights. We can't properly reconstruct the whole skeleton for now, - // but we can create dummy bones for all weight channels which we have. - for (unsigned int w = 0; w < layer.mWeightChannels.size();++w) - { - } + // process vertex weights. We can't properly reconstruct the whole skeleton for now, + // but we can create dummy bones for all weight channels which we have. + for (unsigned int w = 0; w < layer.mWeightChannels.size();++w) + { + } #endif - face.mIndices[q] = vert; - } - pf->mIndices = face.mIndices; - pf->mNumIndices = face.mNumIndices; - unsigned int** p = (unsigned int**)&face.mIndices;*p = NULL; // HACK: make sure it won't be deleted - pf++; - } - - if (!mesh->mNormals) { - // Compute normal vectors for the mesh - we can't use our GenSmoothNormal- - // Step here since it wouldn't handle smoothing groups correctly for LWO. - // So we use a separate implementation. - ComputeNormals(mesh,smoothingGroups,_mSurfaces[i]); - } - else DefaultLogger::get()->debug("LWO2: No need to compute normals, they're already there"); - ++p; - } - } - - // Generate nodes to render the mesh. Store the source layer in the mParent member of the nodes - unsigned int num = apcMeshes.size() - meshStart; - if (layer.mName != "<LWODefault>" || num > 0) { - aiNode* pcNode = new aiNode(); - apcNodes[layer.mIndex] = pcNode; - pcNode->mName.Set(layer.mName); - pcNode->mParent = (aiNode*)&layer; - pcNode->mNumMeshes = num; - - if (pcNode->mNumMeshes) { - pcNode->mMeshes = new unsigned int[pcNode->mNumMeshes]; - for (unsigned int p = 0; p < pcNode->mNumMeshes;++p) - pcNode->mMeshes[p] = p + meshStart; - } - } - } - - if (apcNodes.empty() || apcMeshes.empty()) - throw DeadlyImportError("LWO: No meshes loaded"); - - // The RemoveRedundantMaterials step will clean this up later - pScene->mMaterials = new aiMaterial*[pScene->mNumMaterials = (unsigned int)mSurfaces->size()]; - for (unsigned int mat = 0; mat < pScene->mNumMaterials;++mat) { - aiMaterial* pcMat = new aiMaterial(); - pScene->mMaterials[mat] = pcMat; - ConvertMaterial((*mSurfaces)[mat],pcMat); - } - - // copy the meshes to the output structure - pScene->mMeshes = new aiMesh*[ pScene->mNumMeshes = (unsigned int)apcMeshes.size() ]; - ::memcpy(pScene->mMeshes,&apcMeshes[0],pScene->mNumMeshes*sizeof(void*)); - - // generate the final node graph - GenerateNodeGraph(apcNodes); + face.mIndices[q] = vert; + } + pf->mIndices = face.mIndices; + pf->mNumIndices = face.mNumIndices; + unsigned int** p = (unsigned int**)&face.mIndices;*p = NULL; // HACK: make sure it won't be deleted + pf++; + } + + if (!mesh->mNormals) { + // Compute normal vectors for the mesh - we can't use our GenSmoothNormal- + // Step here since it wouldn't handle smoothing groups correctly for LWO. + // So we use a separate implementation. + ComputeNormals(mesh,smoothingGroups,_mSurfaces[i]); + } + else DefaultLogger::get()->debug("LWO2: No need to compute normals, they're already there"); + ++p; + } + } + + // Generate nodes to render the mesh. Store the source layer in the mParent member of the nodes + unsigned int num = apcMeshes.size() - meshStart; + if (layer.mName != "<LWODefault>" || num > 0) { + aiNode* pcNode = new aiNode(); + apcNodes[layer.mIndex] = pcNode; + pcNode->mName.Set(layer.mName); + pcNode->mParent = (aiNode*)&layer; + pcNode->mNumMeshes = num; + + if (pcNode->mNumMeshes) { + pcNode->mMeshes = new unsigned int[pcNode->mNumMeshes]; + for (unsigned int p = 0; p < pcNode->mNumMeshes;++p) + pcNode->mMeshes[p] = p + meshStart; + } + } + } + + if (apcNodes.empty() || apcMeshes.empty()) + throw DeadlyImportError("LWO: No meshes loaded"); + + // The RemoveRedundantMaterials step will clean this up later + pScene->mMaterials = new aiMaterial*[pScene->mNumMaterials = (unsigned int)mSurfaces->size()]; + for (unsigned int mat = 0; mat < pScene->mNumMaterials;++mat) { + aiMaterial* pcMat = new aiMaterial(); + pScene->mMaterials[mat] = pcMat; + ConvertMaterial((*mSurfaces)[mat],pcMat); + } + + // copy the meshes to the output structure + pScene->mMeshes = new aiMesh*[ pScene->mNumMeshes = (unsigned int)apcMeshes.size() ]; + ::memcpy(pScene->mMeshes,&apcMeshes[0],pScene->mNumMeshes*sizeof(void*)); + + // generate the final node graph + GenerateNodeGraph(apcNodes); } // ------------------------------------------------------------------------------------------------ void LWOImporter::ComputeNormals(aiMesh* mesh, const std::vector<unsigned int>& smoothingGroups, - const LWO::Surface& surface) + const LWO::Surface& surface) { - // Allocate output storage - mesh->mNormals = new aiVector3D[mesh->mNumVertices]; - - // First generate per-face normals - aiVector3D* out; - std::vector<aiVector3D> faceNormals; - - // ... in some cases that's already enough - if (!surface.mMaximumSmoothAngle) - out = mesh->mNormals; - else { - faceNormals.resize(mesh->mNumVertices); - out = &faceNormals[0]; - } - - aiFace* begin = mesh->mFaces, *const end = mesh->mFaces+mesh->mNumFaces; - for (; begin != end; ++begin) { - aiFace& face = *begin; - - if(face.mNumIndices < 3) { - continue; - } - - // LWO doc: "the normal is defined as the cross product of the first and last edges" - aiVector3D* pV1 = mesh->mVertices + face.mIndices[0]; - aiVector3D* pV2 = mesh->mVertices + face.mIndices[1]; - aiVector3D* pV3 = mesh->mVertices + face.mIndices[face.mNumIndices-1]; - - aiVector3D vNor = ((*pV2 - *pV1) ^(*pV3 - *pV1)).Normalize(); - for (unsigned int i = 0; i < face.mNumIndices;++i) - out[face.mIndices[i]] = vNor; - } - if (!surface.mMaximumSmoothAngle)return; - const float posEpsilon = ComputePositionEpsilon(mesh); - - // Now generate the spatial sort tree - SGSpatialSort sSort; - std::vector<unsigned int>::const_iterator it = smoothingGroups.begin(); - for( begin = mesh->mFaces; begin != end; ++begin, ++it) - { - aiFace& face = *begin; - for (unsigned int i = 0; i < face.mNumIndices;++i) - { - unsigned int tt = face.mIndices[i]; - sSort.Add(mesh->mVertices[tt],tt,*it); - } - } - // Sort everything - this takes O(nlogn) time - sSort.Prepare(); - std::vector<unsigned int> poResult; - poResult.reserve(20); - - // Generate vertex normals. We have O(logn) for the binary lookup, which we need - // for n elements, thus the EXPECTED complexity is O(nlogn) - if (surface.mMaximumSmoothAngle < 3.f && !configSpeedFlag) { - const float fLimit = std::cos(surface.mMaximumSmoothAngle); - - for( begin = mesh->mFaces, it = smoothingGroups.begin(); begin != end; ++begin, ++it) { - const aiFace& face = *begin; - unsigned int* beginIdx = face.mIndices, *const endIdx = face.mIndices+face.mNumIndices; - for (; beginIdx != endIdx; ++beginIdx) - { - unsigned int idx = *beginIdx; - sSort.FindPositions(mesh->mVertices[idx],*it,posEpsilon,poResult,true); - std::vector<unsigned int>::const_iterator a, end = poResult.end(); - - aiVector3D vNormals; - for (a = poResult.begin();a != end;++a) { - const aiVector3D& v = faceNormals[*a]; - if (v * faceNormals[idx] < fLimit) - continue; - vNormals += v; - } - mesh->mNormals[idx] = vNormals.Normalize(); - } - } - } - // faster code path in case there is no smooth angle - else { - std::vector<bool> vertexDone(mesh->mNumVertices,false); - for( begin = mesh->mFaces, it = smoothingGroups.begin(); begin != end; ++begin, ++it) { - const aiFace& face = *begin; - unsigned int* beginIdx = face.mIndices, *const endIdx = face.mIndices+face.mNumIndices; - for (; beginIdx != endIdx; ++beginIdx) - { - unsigned int idx = *beginIdx; - if (vertexDone[idx]) - continue; - sSort.FindPositions(mesh->mVertices[idx],*it,posEpsilon,poResult,true); - std::vector<unsigned int>::const_iterator a, end = poResult.end(); - - aiVector3D vNormals; - for (a = poResult.begin();a != end;++a) { - const aiVector3D& v = faceNormals[*a]; - vNormals += v; - } - vNormals.Normalize(); - for (a = poResult.begin();a != end;++a) { - mesh->mNormals[*a] = vNormals; - vertexDone[*a] = true; - } - } - } - } + // Allocate output storage + mesh->mNormals = new aiVector3D[mesh->mNumVertices]; + + // First generate per-face normals + aiVector3D* out; + std::vector<aiVector3D> faceNormals; + + // ... in some cases that's already enough + if (!surface.mMaximumSmoothAngle) + out = mesh->mNormals; + else { + faceNormals.resize(mesh->mNumVertices); + out = &faceNormals[0]; + } + + aiFace* begin = mesh->mFaces, *const end = mesh->mFaces+mesh->mNumFaces; + for (; begin != end; ++begin) { + aiFace& face = *begin; + + if(face.mNumIndices < 3) { + continue; + } + + // LWO doc: "the normal is defined as the cross product of the first and last edges" + aiVector3D* pV1 = mesh->mVertices + face.mIndices[0]; + aiVector3D* pV2 = mesh->mVertices + face.mIndices[1]; + aiVector3D* pV3 = mesh->mVertices + face.mIndices[face.mNumIndices-1]; + + aiVector3D vNor = ((*pV2 - *pV1) ^(*pV3 - *pV1)).Normalize(); + for (unsigned int i = 0; i < face.mNumIndices;++i) + out[face.mIndices[i]] = vNor; + } + if (!surface.mMaximumSmoothAngle)return; + const float posEpsilon = ComputePositionEpsilon(mesh); + + // Now generate the spatial sort tree + SGSpatialSort sSort; + std::vector<unsigned int>::const_iterator it = smoothingGroups.begin(); + for( begin = mesh->mFaces; begin != end; ++begin, ++it) + { + aiFace& face = *begin; + for (unsigned int i = 0; i < face.mNumIndices;++i) + { + unsigned int tt = face.mIndices[i]; + sSort.Add(mesh->mVertices[tt],tt,*it); + } + } + // Sort everything - this takes O(nlogn) time + sSort.Prepare(); + std::vector<unsigned int> poResult; + poResult.reserve(20); + + // Generate vertex normals. We have O(logn) for the binary lookup, which we need + // for n elements, thus the EXPECTED complexity is O(nlogn) + if (surface.mMaximumSmoothAngle < 3.f && !configSpeedFlag) { + const float fLimit = std::cos(surface.mMaximumSmoothAngle); + + for( begin = mesh->mFaces, it = smoothingGroups.begin(); begin != end; ++begin, ++it) { + const aiFace& face = *begin; + unsigned int* beginIdx = face.mIndices, *const endIdx = face.mIndices+face.mNumIndices; + for (; beginIdx != endIdx; ++beginIdx) + { + unsigned int idx = *beginIdx; + sSort.FindPositions(mesh->mVertices[idx],*it,posEpsilon,poResult,true); + std::vector<unsigned int>::const_iterator a, end = poResult.end(); + + aiVector3D vNormals; + for (a = poResult.begin();a != end;++a) { + const aiVector3D& v = faceNormals[*a]; + if (v * faceNormals[idx] < fLimit) + continue; + vNormals += v; + } + mesh->mNormals[idx] = vNormals.Normalize(); + } + } + } + // faster code path in case there is no smooth angle + else { + std::vector<bool> vertexDone(mesh->mNumVertices,false); + for( begin = mesh->mFaces, it = smoothingGroups.begin(); begin != end; ++begin, ++it) { + const aiFace& face = *begin; + unsigned int* beginIdx = face.mIndices, *const endIdx = face.mIndices+face.mNumIndices; + for (; beginIdx != endIdx; ++beginIdx) + { + unsigned int idx = *beginIdx; + if (vertexDone[idx]) + continue; + sSort.FindPositions(mesh->mVertices[idx],*it,posEpsilon,poResult,true); + std::vector<unsigned int>::const_iterator a, end = poResult.end(); + + aiVector3D vNormals; + for (a = poResult.begin();a != end;++a) { + const aiVector3D& v = faceNormals[*a]; + vNormals += v; + } + vNormals.Normalize(); + for (a = poResult.begin();a != end;++a) { + mesh->mNormals[*a] = vNormals; + vertexDone[*a] = true; + } + } + } + } } // ------------------------------------------------------------------------------------------------ void LWOImporter::GenerateNodeGraph(std::map<uint16_t,aiNode*>& apcNodes) { - // now generate the final nodegraph - generate a root node and attach children - aiNode* root = pScene->mRootNode = new aiNode(); - root->mName.Set("<LWORoot>"); - - //Set parent of all children, inserting pivots - //std::cout << "Set parent of all children" << std::endl; - std::map<uint16_t, aiNode*> mapPivot; - for (std::map<uint16_t,aiNode*>::iterator itapcNodes = apcNodes.begin(); itapcNodes != apcNodes.end(); ++itapcNodes) { - - //Get the parent index - LWO::Layer* nodeLayer = (LWO::Layer*)(itapcNodes->second->mParent); - uint16_t parentIndex = nodeLayer->mParent; - - //Create pivot node, store it into the pivot map, and set the parent as the pivot - aiNode* pivotNode = new aiNode(); - pivotNode->mName.Set("Pivot-"+std::string(itapcNodes->second->mName.data)); - mapPivot[-(itapcNodes->first+2)] = pivotNode; - itapcNodes->second->mParent = pivotNode; - - //Look for the parent node to attach the pivot to - if (apcNodes.find(parentIndex) != apcNodes.end()) { - pivotNode->mParent = apcNodes[parentIndex]; - } else { - //If not, attach to the root node - pivotNode->mParent = root; - } - - //Set the node and the pivot node transformation - itapcNodes->second->mTransformation.a4 = -nodeLayer->mPivot.x; - itapcNodes->second->mTransformation.b4 = -nodeLayer->mPivot.y; - itapcNodes->second->mTransformation.c4 = -nodeLayer->mPivot.z; - pivotNode->mTransformation.a4 = nodeLayer->mPivot.x; - pivotNode->mTransformation.b4 = nodeLayer->mPivot.y; - pivotNode->mTransformation.c4 = nodeLayer->mPivot.z; - } - - //Merge pivot map into node map - //std::cout << "Merge pivot map into node map" << std::endl; - for (std::map<uint16_t, aiNode*>::iterator itMapPivot = mapPivot.begin(); itMapPivot != mapPivot.end(); ++itMapPivot) { - apcNodes[itMapPivot->first] = itMapPivot->second; - } - - //Set children of all parents - apcNodes[-1] = root; - for (std::map<uint16_t,aiNode*>::iterator itMapParentNodes = apcNodes.begin(); itMapParentNodes != apcNodes.end(); ++itMapParentNodes) { - for (std::map<uint16_t,aiNode*>::iterator itMapChildNodes = apcNodes.begin(); itMapChildNodes != apcNodes.end(); ++itMapChildNodes) { - if ((itMapParentNodes->first != itMapChildNodes->first) && (itMapParentNodes->second == itMapChildNodes->second->mParent)) { - ++(itMapParentNodes->second->mNumChildren); - } - } - if (itMapParentNodes->second->mNumChildren) { - itMapParentNodes->second->mChildren = new aiNode* [ itMapParentNodes->second->mNumChildren ]; - uint16_t p = 0; - for (std::map<uint16_t,aiNode*>::iterator itMapChildNodes = apcNodes.begin(); itMapChildNodes != apcNodes.end(); ++itMapChildNodes) { - if ((itMapParentNodes->first != itMapChildNodes->first) && (itMapParentNodes->second == itMapChildNodes->second->mParent)) { - itMapParentNodes->second->mChildren[p++] = itMapChildNodes->second; - } - } - } - } - - if (!pScene->mRootNode->mNumChildren) - throw DeadlyImportError("LWO: Unable to build a valid node graph"); - - // Remove a single root node with no meshes assigned to it ... - if (1 == pScene->mRootNode->mNumChildren) { - aiNode* pc = pScene->mRootNode->mChildren[0]; - pc->mParent = pScene->mRootNode->mChildren[0] = NULL; - delete pScene->mRootNode; - pScene->mRootNode = pc; - } - - // convert the whole stuff to RH with CCW winding - MakeLeftHandedProcess maker; - maker.Execute(pScene); - - FlipWindingOrderProcess flipper; - flipper.Execute(pScene); + // now generate the final nodegraph - generate a root node and attach children + aiNode* root = pScene->mRootNode = new aiNode(); + root->mName.Set("<LWORoot>"); + + //Set parent of all children, inserting pivots + //std::cout << "Set parent of all children" << std::endl; + std::map<uint16_t, aiNode*> mapPivot; + for (std::map<uint16_t,aiNode*>::iterator itapcNodes = apcNodes.begin(); itapcNodes != apcNodes.end(); ++itapcNodes) { + + //Get the parent index + LWO::Layer* nodeLayer = (LWO::Layer*)(itapcNodes->second->mParent); + uint16_t parentIndex = nodeLayer->mParent; + + //Create pivot node, store it into the pivot map, and set the parent as the pivot + aiNode* pivotNode = new aiNode(); + pivotNode->mName.Set("Pivot-"+std::string(itapcNodes->second->mName.data)); + mapPivot[-(itapcNodes->first+2)] = pivotNode; + itapcNodes->second->mParent = pivotNode; + + //Look for the parent node to attach the pivot to + if (apcNodes.find(parentIndex) != apcNodes.end()) { + pivotNode->mParent = apcNodes[parentIndex]; + } else { + //If not, attach to the root node + pivotNode->mParent = root; + } + + //Set the node and the pivot node transformation + itapcNodes->second->mTransformation.a4 = -nodeLayer->mPivot.x; + itapcNodes->second->mTransformation.b4 = -nodeLayer->mPivot.y; + itapcNodes->second->mTransformation.c4 = -nodeLayer->mPivot.z; + pivotNode->mTransformation.a4 = nodeLayer->mPivot.x; + pivotNode->mTransformation.b4 = nodeLayer->mPivot.y; + pivotNode->mTransformation.c4 = nodeLayer->mPivot.z; + } + + //Merge pivot map into node map + //std::cout << "Merge pivot map into node map" << std::endl; + for (std::map<uint16_t, aiNode*>::iterator itMapPivot = mapPivot.begin(); itMapPivot != mapPivot.end(); ++itMapPivot) { + apcNodes[itMapPivot->first] = itMapPivot->second; + } + + //Set children of all parents + apcNodes[-1] = root; + for (std::map<uint16_t,aiNode*>::iterator itMapParentNodes = apcNodes.begin(); itMapParentNodes != apcNodes.end(); ++itMapParentNodes) { + for (std::map<uint16_t,aiNode*>::iterator itMapChildNodes = apcNodes.begin(); itMapChildNodes != apcNodes.end(); ++itMapChildNodes) { + if ((itMapParentNodes->first != itMapChildNodes->first) && (itMapParentNodes->second == itMapChildNodes->second->mParent)) { + ++(itMapParentNodes->second->mNumChildren); + } + } + if (itMapParentNodes->second->mNumChildren) { + itMapParentNodes->second->mChildren = new aiNode* [ itMapParentNodes->second->mNumChildren ]; + uint16_t p = 0; + for (std::map<uint16_t,aiNode*>::iterator itMapChildNodes = apcNodes.begin(); itMapChildNodes != apcNodes.end(); ++itMapChildNodes) { + if ((itMapParentNodes->first != itMapChildNodes->first) && (itMapParentNodes->second == itMapChildNodes->second->mParent)) { + itMapParentNodes->second->mChildren[p++] = itMapChildNodes->second; + } + } + } + } + + if (!pScene->mRootNode->mNumChildren) + throw DeadlyImportError("LWO: Unable to build a valid node graph"); + + // Remove a single root node with no meshes assigned to it ... + if (1 == pScene->mRootNode->mNumChildren) { + aiNode* pc = pScene->mRootNode->mChildren[0]; + pc->mParent = pScene->mRootNode->mChildren[0] = NULL; + delete pScene->mRootNode; + pScene->mRootNode = pc; + } + + // convert the whole stuff to RH with CCW winding + MakeLeftHandedProcess maker; + maker.Execute(pScene); + + FlipWindingOrderProcess flipper; + flipper.Execute(pScene); } // ------------------------------------------------------------------------------------------------ void LWOImporter::ResolveTags() { - // --- this function is used for both LWO2 and LWOB - mMapping->resize(mTags->size(), UINT_MAX); - for (unsigned int a = 0; a < mTags->size();++a) { + // --- this function is used for both LWO2 and LWOB + mMapping->resize(mTags->size(), UINT_MAX); + for (unsigned int a = 0; a < mTags->size();++a) { - const std::string& c = (*mTags)[a]; - for (unsigned int i = 0; i < mSurfaces->size();++i) { + const std::string& c = (*mTags)[a]; + for (unsigned int i = 0; i < mSurfaces->size();++i) { - const std::string& d = (*mSurfaces)[i].mName; - if (!ASSIMP_stricmp(c,d)) { + const std::string& d = (*mSurfaces)[i].mName; + if (!ASSIMP_stricmp(c,d)) { - (*mMapping)[a] = i; - break; - } - } - } + (*mMapping)[a] = i; + break; + } + } + } } // ------------------------------------------------------------------------------------------------ void LWOImporter::ResolveClips() { - for( unsigned int i = 0; i < mClips.size();++i) { - - Clip& clip = mClips[i]; - if (Clip::REF == clip.type) { - - if (clip.clipRef >= mClips.size()) { - DefaultLogger::get()->error("LWO2: Clip referrer index is out of range"); - clip.clipRef = 0; - } - - Clip& dest = mClips[clip.clipRef]; - if (Clip::REF == dest.type) { - DefaultLogger::get()->error("LWO2: Clip references another clip reference"); - clip.type = Clip::UNSUPPORTED; - } - - else { - clip.path = dest.path; - clip.type = dest.type; - } - } - } + for( unsigned int i = 0; i < mClips.size();++i) { + + Clip& clip = mClips[i]; + if (Clip::REF == clip.type) { + + if (clip.clipRef >= mClips.size()) { + DefaultLogger::get()->error("LWO2: Clip referrer index is out of range"); + clip.clipRef = 0; + } + + Clip& dest = mClips[clip.clipRef]; + if (Clip::REF == dest.type) { + DefaultLogger::get()->error("LWO2: Clip references another clip reference"); + clip.type = Clip::UNSUPPORTED; + } + + else { + clip.path = dest.path; + clip.type = dest.type; + } + } + } } // ------------------------------------------------------------------------------------------------ void LWOImporter::AdjustTexturePath(std::string& out) { - // --- this function is used for both LWO2 and LWOB - if (!mIsLWO2 && ::strstr(out.c_str(), "(sequence)")) { - - // remove the (sequence) and append 000 - DefaultLogger::get()->info("LWOB: Sequence of animated texture found. It will be ignored"); - out = out.substr(0,out.length()-10) + "000"; - } - - // format: drive:path/file - we just need to insert a slash after the drive - std::string::size_type n = out.find_first_of(':'); - if (std::string::npos != n) { - out.insert(n+1,"/"); - } + // --- this function is used for both LWO2 and LWOB + if (!mIsLWO2 && ::strstr(out.c_str(), "(sequence)")) { + + // remove the (sequence) and append 000 + DefaultLogger::get()->info("LWOB: Sequence of animated texture found. It will be ignored"); + out = out.substr(0,out.length()-10) + "000"; + } + + // format: drive:path/file - we just need to insert a slash after the drive + std::string::size_type n = out.find_first_of(':'); + if (std::string::npos != n) { + out.insert(n+1,"/"); + } } // ------------------------------------------------------------------------------------------------ void LWOImporter::LoadLWOTags(unsigned int size) { - // --- this function is used for both LWO2 and LWOB - - const char* szCur = (const char*)mFileBuffer, *szLast = szCur; - const char* const szEnd = szLast+size; - while (szCur < szEnd) - { - if (!(*szCur)) - { - const size_t len = (size_t)(szCur-szLast); - // FIX: skip empty-sized tags - if (len) - mTags->push_back(std::string(szLast,len)); - szCur += (len&0x1 ? 1 : 2); - szLast = szCur; - } - szCur++; - } + // --- this function is used for both LWO2 and LWOB + + const char* szCur = (const char*)mFileBuffer, *szLast = szCur; + const char* const szEnd = szLast+size; + while (szCur < szEnd) + { + if (!(*szCur)) + { + const size_t len = (size_t)(szCur-szLast); + // FIX: skip empty-sized tags + if (len) + mTags->push_back(std::string(szLast,len)); + szCur += (len&0x1 ? 1 : 2); + szLast = szCur; + } + szCur++; + } } // ------------------------------------------------------------------------------------------------ void LWOImporter::LoadLWOPoints(unsigned int length) { - // --- this function is used for both LWO2 and LWOB but for - // LWO2 we need to allocate 25% more storage - it could be we'll - // need to duplicate some points later. - unsigned int regularSize = (unsigned int)mCurLayer->mTempPoints.size() + length / 12; - if (mIsLWO2) - { - mCurLayer->mTempPoints.reserve ( regularSize + (regularSize>>2u) ); - mCurLayer->mTempPoints.resize ( regularSize ); - - // initialize all point referrers with the default values - mCurLayer->mPointReferrers.reserve ( regularSize + (regularSize>>2u) ); - mCurLayer->mPointReferrers.resize ( regularSize, UINT_MAX ); - } - else mCurLayer->mTempPoints.resize( regularSize ); - - // perform endianess conversions + // --- this function is used for both LWO2 and LWOB but for + // LWO2 we need to allocate 25% more storage - it could be we'll + // need to duplicate some points later. + const size_t vertexLen = 12; + if ((length % vertexLen) != 0) + { + throw DeadlyImportError( "LWO2: Points chunk length is not multiple of vertexLen (12)"); + } + unsigned int regularSize = (unsigned int)mCurLayer->mTempPoints.size() + length / 12; + if (mIsLWO2) + { + mCurLayer->mTempPoints.reserve ( regularSize + (regularSize>>2u) ); + mCurLayer->mTempPoints.resize ( regularSize ); + + // initialize all point referrers with the default values + mCurLayer->mPointReferrers.reserve ( regularSize + (regularSize>>2u) ); + mCurLayer->mPointReferrers.resize ( regularSize, UINT_MAX ); + } + else mCurLayer->mTempPoints.resize( regularSize ); + + // perform endianness conversions #ifndef AI_BUILD_BIG_ENDIAN - for (unsigned int i = 0; i < length>>2;++i) - ByteSwap::Swap4( mFileBuffer + (i << 2)); + for (unsigned int i = 0; i < length>>2;++i) + ByteSwap::Swap4( mFileBuffer + (i << 2)); #endif - ::memcpy(&mCurLayer->mTempPoints[0],mFileBuffer,length); + ::memcpy(&mCurLayer->mTempPoints[0],mFileBuffer,length); } // ------------------------------------------------------------------------------------------------ void LWOImporter::LoadLWO2Polygons(unsigned int length) { - LE_NCONST uint16_t* const end = (LE_NCONST uint16_t*)(mFileBuffer+length); - const uint32_t type = GetU4(); - - // Determine the type of the polygons - switch (type) - { - // read unsupported stuff too (although we wont process it) - case AI_LWO_MBAL: - DefaultLogger::get()->warn("LWO2: Encountered unsupported primitive chunk (METABALL)"); - break; - case AI_LWO_CURV: - DefaultLogger::get()->warn("LWO2: Encountered unsupported primitive chunk (SPLINE)");; - break; - - // These are ok with no restrictions - case AI_LWO_PTCH: - case AI_LWO_FACE: - case AI_LWO_BONE: - case AI_LWO_SUBD: - break; - default: - - // hm!? wtf is this? ok ... - DefaultLogger::get()->error("LWO2: Ignoring unknown polygon type."); - break; - } - - // first find out how many faces and vertices we'll finally need - uint16_t* cursor= (uint16_t*)mFileBuffer; - - unsigned int iNumFaces = 0,iNumVertices = 0; - CountVertsAndFacesLWO2(iNumVertices,iNumFaces,cursor,end); - - // allocate the output array and copy face indices - if (iNumFaces) { - cursor = (uint16_t*)mFileBuffer; - - mCurLayer->mFaces.resize(iNumFaces,LWO::Face(type)); - FaceList::iterator it = mCurLayer->mFaces.begin(); - CopyFaceIndicesLWO2(it,cursor,end); - } + LE_NCONST uint16_t* const end = (LE_NCONST uint16_t*)(mFileBuffer+length); + const uint32_t type = GetU4(); + + // Determine the type of the polygons + switch (type) + { + // read unsupported stuff too (although we wont process it) + case AI_LWO_MBAL: + DefaultLogger::get()->warn("LWO2: Encountered unsupported primitive chunk (METABALL)"); + break; + case AI_LWO_CURV: + DefaultLogger::get()->warn("LWO2: Encountered unsupported primitive chunk (SPLINE)");; + break; + + // These are ok with no restrictions + case AI_LWO_PTCH: + case AI_LWO_FACE: + case AI_LWO_BONE: + case AI_LWO_SUBD: + break; + default: + + // hm!? wtf is this? ok ... + DefaultLogger::get()->error("LWO2: Ignoring unknown polygon type."); + break; + } + + // first find out how many faces and vertices we'll finally need + uint16_t* cursor= (uint16_t*)mFileBuffer; + + unsigned int iNumFaces = 0,iNumVertices = 0; + CountVertsAndFacesLWO2(iNumVertices,iNumFaces,cursor,end); + + // allocate the output array and copy face indices + if (iNumFaces) + { + cursor = (uint16_t*)mFileBuffer; + + mCurLayer->mFaces.resize(iNumFaces,LWO::Face(type)); + FaceList::iterator it = mCurLayer->mFaces.begin(); + CopyFaceIndicesLWO2(it,cursor,end); + } } // ------------------------------------------------------------------------------------------------ void LWOImporter::CountVertsAndFacesLWO2(unsigned int& verts, unsigned int& faces, - uint16_t*& cursor, const uint16_t* const end, unsigned int max) + uint16_t*& cursor, const uint16_t* const end, unsigned int max) { - while (cursor < end && max--) - { - AI_LSWAP2P(cursor); - uint16_t numIndices = *cursor++; - numIndices &= 0x03FF; - verts += numIndices;++faces; - - for(uint16_t i = 0; i < numIndices; i++) - ReadVSizedIntLWO2((uint8_t*&)cursor); - } + while (cursor < end && max--) + { + uint16_t numIndices; + ::memcpy(&numIndices, cursor++, 2); + AI_LSWAP2(numIndices); + numIndices &= 0x03FF; + + verts += numIndices; + ++faces; + + for(uint16_t i = 0; i < numIndices; i++) + { + ReadVSizedIntLWO2((uint8_t*&)cursor); + } + } } // ------------------------------------------------------------------------------------------------ void LWOImporter::CopyFaceIndicesLWO2(FaceList::iterator& it, - uint16_t*& cursor, - const uint16_t* const end) + uint16_t*& cursor, + const uint16_t* const end) { - while (cursor < end) { - - LWO::Face& face = *it++;; - if((face.mNumIndices = (*cursor++) & 0x03FF)) /* byte swapping has already been done */ { - face.mIndices = new unsigned int[face.mNumIndices]; - for(unsigned int i = 0; i < face.mNumIndices; i++) - { - face.mIndices[i] = ReadVSizedIntLWO2((uint8_t*&)cursor) + mCurLayer->mPointIDXOfs; - if(face.mIndices[i] > mCurLayer->mTempPoints.size()) - { - DefaultLogger::get()->warn("LWO2: Failure evaluating face record, index is out of range"); - face.mIndices[i] = (unsigned int)mCurLayer->mTempPoints.size()-1; - } - } - } - else throw DeadlyImportError("LWO2: Encountered invalid face record with zero indices"); - } + while (cursor < end) + { + LWO::Face& face = *it++; + uint16_t numIndices; + ::memcpy(&numIndices, cursor++, 2); + AI_LSWAP2(numIndices); + face.mNumIndices = numIndices & 0x03FF; + + if(face.mNumIndices) /* byte swapping has already been done */ + { + face.mIndices = new unsigned int[face.mNumIndices]; + for(unsigned int i = 0; i < face.mNumIndices; i++) + { + face.mIndices[i] = ReadVSizedIntLWO2((uint8_t*&)cursor) + mCurLayer->mPointIDXOfs; + if(face.mIndices[i] > mCurLayer->mTempPoints.size()) + { + DefaultLogger::get()->warn("LWO2: Failure evaluating face record, index is out of range"); + face.mIndices[i] = (unsigned int)mCurLayer->mTempPoints.size()-1; + } + } + } + else throw DeadlyImportError("LWO2: Encountered invalid face record with zero indices"); + } } // ------------------------------------------------------------------------------------------------ void LWOImporter::LoadLWO2PolygonTags(unsigned int length) { - LE_NCONST uint8_t* const end = mFileBuffer+length; + LE_NCONST uint8_t* const end = mFileBuffer+length; - AI_LWO_VALIDATE_CHUNK_LENGTH(length,PTAG,4); - uint32_t type = GetU4(); + AI_LWO_VALIDATE_CHUNK_LENGTH(length,PTAG,4); + uint32_t type = GetU4(); - if (type != AI_LWO_SURF && type != AI_LWO_SMGP) - return; + if (type != AI_LWO_SURF && type != AI_LWO_SMGP) + return; - while (mFileBuffer < end) { + while (mFileBuffer < end) + { + unsigned int i = ReadVSizedIntLWO2(mFileBuffer) + mCurLayer->mFaceIDXOfs; + unsigned int j = GetU2(); - unsigned int i = ReadVSizedIntLWO2(mFileBuffer) + mCurLayer->mFaceIDXOfs; - unsigned int j = GetU2(); - - if (i >= mCurLayer->mFaces.size()) { - DefaultLogger::get()->warn("LWO2: face index in PTAG is out of range"); - continue; - } + if (i >= mCurLayer->mFaces.size()) { + DefaultLogger::get()->warn("LWO2: face index in PTAG is out of range"); + continue; + } - switch (type) { + switch (type) { - case AI_LWO_SURF: - mCurLayer->mFaces[i].surfaceIndex = j; - break; - case AI_LWO_SMGP: /* is that really used? */ - mCurLayer->mFaces[i].smoothGroup = j; - break; - }; - } + case AI_LWO_SURF: + mCurLayer->mFaces[i].surfaceIndex = j; + break; + case AI_LWO_SMGP: /* is that really used? */ + mCurLayer->mFaces[i].smoothGroup = j; + break; + }; + } } // ------------------------------------------------------------------------------------------------ template <class T> VMapEntry* FindEntry(std::vector< T >& list,const std::string& name, bool perPoly) { - for (typename std::vector< T >::iterator it = list.begin(), end = list.end();it != end; ++it) { - if ((*it).name == name) { - if (!perPoly) { - DefaultLogger::get()->warn("LWO2: Found two VMAP sections with equal names"); - } - return &(*it); - } - } - list.push_back( T() ); - VMapEntry* p = &list.back(); - p->name = name; - return p; + for (auto & elem : list) { + if (elem.name == name) { + if (!perPoly) { + DefaultLogger::get()->warn("LWO2: Found two VMAP sections with equal names"); + } + return &elem; + } + } + list.push_back( T() ); + VMapEntry* p = &list.back(); + p->name = name; + return p; } // ------------------------------------------------------------------------------------------------ template <class T> inline void CreateNewEntry(T& chan, unsigned int srcIdx) { - if (!chan.name.length()) - return; + if (!chan.name.length()) + return; - chan.abAssigned[srcIdx] = true; - chan.abAssigned.resize(chan.abAssigned.size()+1,false); + chan.abAssigned[srcIdx] = true; + chan.abAssigned.resize(chan.abAssigned.size()+1,false); - for (unsigned int a = 0; a < chan.dims;++a) - chan.rawData.push_back(chan.rawData[srcIdx*chan.dims+a]); + for (unsigned int a = 0; a < chan.dims;++a) + chan.rawData.push_back(chan.rawData[srcIdx*chan.dims+a]); } // ------------------------------------------------------------------------------------------------ template <class T> inline void CreateNewEntry(std::vector< T >& list, unsigned int srcIdx) { - for (typename std::vector< T >::iterator it = list.begin(), end = list.end();it != end;++it) { - CreateNewEntry( *it, srcIdx ); - } + for (auto &elem : list) { + CreateNewEntry( elem, srcIdx ); + } } // ------------------------------------------------------------------------------------------------ -inline void LWOImporter::DoRecursiveVMAPAssignment(VMapEntry* base, unsigned int numRead, - unsigned int idx, float* data) +inline void LWOImporter::DoRecursiveVMAPAssignment(VMapEntry* base, unsigned int numRead, + unsigned int idx, float* data) { - ai_assert(NULL != data); - LWO::ReferrerList& refList = mCurLayer->mPointReferrers; - unsigned int i; - - base->abAssigned[idx] = true; - for (i = 0; i < numRead;++i) { - base->rawData[idx*base->dims+i]= data[i]; - } - - if (UINT_MAX != (i = refList[idx])) { - DoRecursiveVMAPAssignment(base,numRead,i,data); - } + ai_assert(NULL != data); + LWO::ReferrerList& refList = mCurLayer->mPointReferrers; + unsigned int i; + + if (idx >= base->abAssigned.size()) { + throw DeadlyImportError("Bad index"); + } + base->abAssigned[idx] = true; + for (i = 0; i < numRead;++i) { + base->rawData[idx*base->dims+i]= data[i]; + } + + if (UINT_MAX != (i = refList[idx])) { + DoRecursiveVMAPAssignment(base,numRead,i,data); + } } // ------------------------------------------------------------------------------------------------ inline void AddToSingleLinkedList(ReferrerList& refList, unsigned int srcIdx, unsigned int destIdx) { - if(UINT_MAX == refList[srcIdx]) { - refList[srcIdx] = destIdx; - return; - } - AddToSingleLinkedList(refList,refList[srcIdx],destIdx); + if(UINT_MAX == refList[srcIdx]) { + refList[srcIdx] = destIdx; + return; + } + AddToSingleLinkedList(refList,refList[srcIdx],destIdx); } // ------------------------------------------------------------------------------------------------ // Load LWO2 vertex map void LWOImporter::LoadLWO2VertexMap(unsigned int length, bool perPoly) { - LE_NCONST uint8_t* const end = mFileBuffer+length; - - AI_LWO_VALIDATE_CHUNK_LENGTH(length,VMAP,6); - unsigned int type = GetU4(); - unsigned int dims = GetU2(); - - VMapEntry* base; - - // read the name of the vertex map - std::string name; - GetS0(name,length); - - switch (type) - { - case AI_LWO_TXUV: - if (dims != 2) { - DefaultLogger::get()->warn("LWO2: Skipping UV channel \'" - + name + "\' with !2 components"); - return; - } - base = FindEntry(mCurLayer->mUVChannels,name,perPoly); - break; - case AI_LWO_WGHT: - case AI_LWO_MNVW: - if (dims != 1) { - DefaultLogger::get()->warn("LWO2: Skipping Weight Channel \'" - + name + "\' with !1 components"); - return; - } - base = FindEntry((type == AI_LWO_WGHT ? mCurLayer->mWeightChannels - : mCurLayer->mSWeightChannels),name,perPoly); - break; - case AI_LWO_RGB: - case AI_LWO_RGBA: - if (dims != 3 && dims != 4) { - DefaultLogger::get()->warn("LWO2: Skipping Color Map \'" - + name + "\' with a dimension > 4 or < 3"); - return; - } - base = FindEntry(mCurLayer->mVColorChannels,name,perPoly); - break; - - case AI_LWO_MODO_NORM: - /* This is a non-standard extension chunk used by Luxology's MODO. - * It stores per-vertex normals. This VMAP exists just once, has - * 3 dimensions and is btw extremely beautiful. - */ - if (name != "vert_normals" || dims != 3 || mCurLayer->mNormals.name.length()) - return; - - DefaultLogger::get()->info("Processing non-standard extension: MODO VMAP.NORM.vert_normals"); - - mCurLayer->mNormals.name = name; - base = & mCurLayer->mNormals; - break; - - case AI_LWO_PICK: /* these VMAPs are just silently dropped */ - case AI_LWO_MORF: - case AI_LWO_SPOT: - return; - - default: - if (name == "APS.Level") { - // XXX handle this (seems to be subdivision-related). - } - DefaultLogger::get()->warn("LWO2: Skipping unknown VMAP/VMAD channel \'" + name + "\'"); - return; - }; - base->Allocate((unsigned int)mCurLayer->mTempPoints.size()); - - // now read all entries in the map - type = std::min(dims,base->dims); - const unsigned int diff = (dims - type)<<2u; - - LWO::FaceList& list = mCurLayer->mFaces; - LWO::PointList& pointList = mCurLayer->mTempPoints; - LWO::ReferrerList& refList = mCurLayer->mPointReferrers; - - float temp[4]; - - const unsigned int numPoints = (unsigned int)pointList.size(); - const unsigned int numFaces = (unsigned int)list.size(); - - while (mFileBuffer < end) { - - unsigned int idx = ReadVSizedIntLWO2(mFileBuffer) + mCurLayer->mPointIDXOfs; - if (idx >= numPoints) { - DefaultLogger::get()->warn("LWO2: Failure evaluating VMAP/VMAD entry \'" + name + "\', vertex index is out of range"); - mFileBuffer += base->dims<<2u; - continue; - } - if (perPoly) { - unsigned int polyIdx = ReadVSizedIntLWO2(mFileBuffer) + mCurLayer->mFaceIDXOfs; - if (base->abAssigned[idx]) { - // we have already a VMAP entry for this vertex - thus - // we need to duplicate the corresponding polygon. - if (polyIdx >= numFaces) { - DefaultLogger::get()->warn("LWO2: Failure evaluating VMAD entry \'" + name + "\', polygon index is out of range"); - mFileBuffer += base->dims<<2u; - continue; - } - - LWO::Face& src = list[polyIdx]; - - // generate a new unique vertex for the corresponding index - but only - // if we can find the index in the face - bool had = false; - for (unsigned int i = 0; i < src.mNumIndices;++i) { - - unsigned int srcIdx = src.mIndices[i], tmp = idx; - do { - if (tmp == srcIdx) - break; - } - while ((tmp = refList[tmp]) != UINT_MAX); - if (tmp == UINT_MAX) { - continue; - } - - had = true; - refList.resize(refList.size()+1, UINT_MAX); - - idx = (unsigned int)pointList.size(); - src.mIndices[i] = (unsigned int)pointList.size(); - - // store the index of the new vertex in the old vertex - // so we get a single linked list we can traverse in - // only one direction - AddToSingleLinkedList(refList,srcIdx,src.mIndices[i]); - pointList.push_back(pointList[srcIdx]); - - CreateNewEntry(mCurLayer->mVColorChannels, srcIdx ); - CreateNewEntry(mCurLayer->mUVChannels, srcIdx ); - CreateNewEntry(mCurLayer->mWeightChannels, srcIdx ); - CreateNewEntry(mCurLayer->mSWeightChannels, srcIdx ); - CreateNewEntry(mCurLayer->mNormals, srcIdx ); - } - if (!had) { - DefaultLogger::get()->warn("LWO2: Failure evaluating VMAD entry \'" + name + "\', vertex index wasn't found in that polygon"); - ai_assert(had); - } - } - } - for (unsigned int l = 0; l < type;++l) - temp[l] = GetF4(); - - DoRecursiveVMAPAssignment(base,type,idx, temp); - mFileBuffer += diff; - } + LE_NCONST uint8_t* const end = mFileBuffer+length; + + AI_LWO_VALIDATE_CHUNK_LENGTH(length,VMAP,6); + unsigned int type = GetU4(); + unsigned int dims = GetU2(); + + VMapEntry* base; + + // read the name of the vertex map + std::string name; + GetS0(name,length); + + switch (type) + { + case AI_LWO_TXUV: + if (dims != 2) { + DefaultLogger::get()->warn("LWO2: Skipping UV channel \'" + + name + "\' with !2 components"); + return; + } + base = FindEntry(mCurLayer->mUVChannels,name,perPoly); + break; + case AI_LWO_WGHT: + case AI_LWO_MNVW: + if (dims != 1) { + DefaultLogger::get()->warn("LWO2: Skipping Weight Channel \'" + + name + "\' with !1 components"); + return; + } + base = FindEntry((type == AI_LWO_WGHT ? mCurLayer->mWeightChannels + : mCurLayer->mSWeightChannels),name,perPoly); + break; + case AI_LWO_RGB: + case AI_LWO_RGBA: + if (dims != 3 && dims != 4) { + DefaultLogger::get()->warn("LWO2: Skipping Color Map \'" + + name + "\' with a dimension > 4 or < 3"); + return; + } + base = FindEntry(mCurLayer->mVColorChannels,name,perPoly); + break; + + case AI_LWO_MODO_NORM: + /* This is a non-standard extension chunk used by Luxology's MODO. + * It stores per-vertex normals. This VMAP exists just once, has + * 3 dimensions and is btw extremely beautiful. + */ + if (name != "vert_normals" || dims != 3 || mCurLayer->mNormals.name.length()) + return; + + DefaultLogger::get()->info("Processing non-standard extension: MODO VMAP.NORM.vert_normals"); + + mCurLayer->mNormals.name = name; + base = & mCurLayer->mNormals; + break; + + case AI_LWO_PICK: /* these VMAPs are just silently dropped */ + case AI_LWO_MORF: + case AI_LWO_SPOT: + return; + + default: + if (name == "APS.Level") { + // XXX handle this (seems to be subdivision-related). + } + DefaultLogger::get()->warn("LWO2: Skipping unknown VMAP/VMAD channel \'" + name + "\'"); + return; + }; + base->Allocate((unsigned int)mCurLayer->mTempPoints.size()); + + // now read all entries in the map + type = std::min(dims,base->dims); + const unsigned int diff = (dims - type)<<2u; + + LWO::FaceList& list = mCurLayer->mFaces; + LWO::PointList& pointList = mCurLayer->mTempPoints; + LWO::ReferrerList& refList = mCurLayer->mPointReferrers; + + float temp[4]; + + const unsigned int numPoints = (unsigned int)pointList.size(); + const unsigned int numFaces = (unsigned int)list.size(); + + while (mFileBuffer < end) { + + unsigned int idx = ReadVSizedIntLWO2(mFileBuffer) + mCurLayer->mPointIDXOfs; + if (idx >= numPoints) { + DefaultLogger::get()->warn("LWO2: Failure evaluating VMAP/VMAD entry \'" + name + "\', vertex index is out of range"); + mFileBuffer += base->dims<<2u; + continue; + } + if (perPoly) { + unsigned int polyIdx = ReadVSizedIntLWO2(mFileBuffer) + mCurLayer->mFaceIDXOfs; + if (base->abAssigned[idx]) { + // we have already a VMAP entry for this vertex - thus + // we need to duplicate the corresponding polygon. + if (polyIdx >= numFaces) { + DefaultLogger::get()->warn("LWO2: Failure evaluating VMAD entry \'" + name + "\', polygon index is out of range"); + mFileBuffer += base->dims<<2u; + continue; + } + + LWO::Face& src = list[polyIdx]; + + // generate a new unique vertex for the corresponding index - but only + // if we can find the index in the face + bool had = false; + for (unsigned int i = 0; i < src.mNumIndices;++i) { + + unsigned int srcIdx = src.mIndices[i], tmp = idx; + do { + if (tmp == srcIdx) + break; + } + while ((tmp = refList[tmp]) != UINT_MAX); + if (tmp == UINT_MAX) { + continue; + } + + had = true; + refList.resize(refList.size()+1, UINT_MAX); + + idx = (unsigned int)pointList.size(); + src.mIndices[i] = (unsigned int)pointList.size(); + + // store the index of the new vertex in the old vertex + // so we get a single linked list we can traverse in + // only one direction + AddToSingleLinkedList(refList,srcIdx,src.mIndices[i]); + pointList.push_back(pointList[srcIdx]); + + CreateNewEntry(mCurLayer->mVColorChannels, srcIdx ); + CreateNewEntry(mCurLayer->mUVChannels, srcIdx ); + CreateNewEntry(mCurLayer->mWeightChannels, srcIdx ); + CreateNewEntry(mCurLayer->mSWeightChannels, srcIdx ); + CreateNewEntry(mCurLayer->mNormals, srcIdx ); + } + if (!had) { + DefaultLogger::get()->warn("LWO2: Failure evaluating VMAD entry \'" + name + "\', vertex index wasn't found in that polygon"); + ai_assert(had); + } + } + } + for (unsigned int l = 0; l < type;++l) + temp[l] = GetF4(); + + DoRecursiveVMAPAssignment(base,type,idx, temp); + mFileBuffer += diff; + } } // ------------------------------------------------------------------------------------------------ // Load LWO2 clip void LWOImporter::LoadLWO2Clip(unsigned int length) { - AI_LWO_VALIDATE_CHUNK_LENGTH(length,CLIP,10); - - mClips.push_back(LWO::Clip()); - LWO::Clip& clip = mClips.back(); - - // first - get the index of the clip - clip.idx = GetU4(); - - IFF::SubChunkHeader* const head = IFF::LoadSubChunk(mFileBuffer); - switch (head->type) - { - case AI_LWO_STIL: - AI_LWO_VALIDATE_CHUNK_LENGTH(head->length,STIL,1); - - // "Normal" texture - GetS0(clip.path,head->length); - clip.type = Clip::STILL; - break; - - case AI_LWO_ISEQ: - AI_LWO_VALIDATE_CHUNK_LENGTH(head->length,ISEQ,16); - // Image sequence. We'll later take the first. - { - uint8_t digits = GetU1(); mFileBuffer++; - int16_t offset = GetU2(); mFileBuffer+=4; - int16_t start = GetU2(); mFileBuffer+=4; - - std::string s; - std::ostringstream ss; - GetS0(s,head->length); - - head->length -= (unsigned int)s.length()+1; - ss << s; - ss << std::setw(digits) << offset + start; - GetS0(s,head->length); - ss << s; - clip.path = ss.str(); - clip.type = Clip::SEQ; - } - break; - - case AI_LWO_STCC: - DefaultLogger::get()->warn("LWO2: Color shifted images are not supported"); - break; - - case AI_LWO_ANIM: - DefaultLogger::get()->warn("LWO2: Animated textures are not supported"); - break; - - case AI_LWO_XREF: - AI_LWO_VALIDATE_CHUNK_LENGTH(head->length,XREF,4); - - // Just a cross-reference to another CLIp - clip.type = Clip::REF; - clip.clipRef = GetU4(); - break; - - case AI_LWO_NEGA: - AI_LWO_VALIDATE_CHUNK_LENGTH(head->length,NEGA,2); - clip.negate = (0 != GetU2()); - break; - - default: - DefaultLogger::get()->warn("LWO2: Encountered unknown CLIP subchunk"); - } + AI_LWO_VALIDATE_CHUNK_LENGTH(length,CLIP,10); + + mClips.push_back(LWO::Clip()); + LWO::Clip& clip = mClips.back(); + + // first - get the index of the clip + clip.idx = GetU4(); + + IFF::SubChunkHeader head = IFF::LoadSubChunk(mFileBuffer); + switch (head.type) + { + case AI_LWO_STIL: + AI_LWO_VALIDATE_CHUNK_LENGTH(head.length,STIL,1); + + // "Normal" texture + GetS0(clip.path,head.length); + clip.type = Clip::STILL; + break; + + case AI_LWO_ISEQ: + AI_LWO_VALIDATE_CHUNK_LENGTH(head.length,ISEQ,16); + // Image sequence. We'll later take the first. + { + uint8_t digits = GetU1(); mFileBuffer++; + int16_t offset = GetU2(); mFileBuffer+=4; + int16_t start = GetU2(); mFileBuffer+=4; + + std::string s; + std::ostringstream ss; + GetS0(s,head.length); + + head.length -= (uint16_t)s.length()+1; + ss << s; + ss << std::setw(digits) << offset + start; + GetS0(s,head.length); + ss << s; + clip.path = ss.str(); + clip.type = Clip::SEQ; + } + break; + + case AI_LWO_STCC: + DefaultLogger::get()->warn("LWO2: Color shifted images are not supported"); + break; + + case AI_LWO_ANIM: + DefaultLogger::get()->warn("LWO2: Animated textures are not supported"); + break; + + case AI_LWO_XREF: + AI_LWO_VALIDATE_CHUNK_LENGTH(head.length,XREF,4); + + // Just a cross-reference to another CLIp + clip.type = Clip::REF; + clip.clipRef = GetU4(); + break; + + case AI_LWO_NEGA: + AI_LWO_VALIDATE_CHUNK_LENGTH(head.length,NEGA,2); + clip.negate = (0 != GetU2()); + break; + + default: + DefaultLogger::get()->warn("LWO2: Encountered unknown CLIP subchunk"); + } } // ------------------------------------------------------------------------------------------------ // Load envelope description void LWOImporter::LoadLWO2Envelope(unsigned int length) { - LE_NCONST uint8_t* const end = mFileBuffer + length; - AI_LWO_VALIDATE_CHUNK_LENGTH(length,ENVL,4); + LE_NCONST uint8_t* const end = mFileBuffer + length; + AI_LWO_VALIDATE_CHUNK_LENGTH(length,ENVL,4); - mEnvelopes.push_back(LWO::Envelope()); - LWO::Envelope& envelope = mEnvelopes.back(); + mEnvelopes.push_back(LWO::Envelope()); + LWO::Envelope& envelope = mEnvelopes.back(); - // Get the index of the envelope - envelope.index = ReadVSizedIntLWO2(mFileBuffer); + // Get the index of the envelope + envelope.index = ReadVSizedIntLWO2(mFileBuffer); - // It looks like there might be an extra U4 right after the index, - // at least in modo (LXOB) files: we'll ignore it if it's zero, - // otherwise it represents the start of a subchunk, so we backtrack. - if (mIsLXOB) - { + // It looks like there might be an extra U4 right after the index, + // at least in modo (LXOB) files: we'll ignore it if it's zero, + // otherwise it represents the start of a subchunk, so we backtrack. + if (mIsLXOB) + { uint32_t extra = GetU4(); if (extra) { mFileBuffer -= 4; } - } - - // ... and read all subchunks - while (true) - { - if (mFileBuffer + 6 >= end)break; - LE_NCONST IFF::SubChunkHeader* const head = IFF::LoadSubChunk(mFileBuffer); - - if (mFileBuffer + head->length > end) - throw DeadlyImportError("LWO2: Invalid envelope chunk length"); - - uint8_t* const next = mFileBuffer+head->length; - switch (head->type) - { - // Type & representation of the envelope - case AI_LWO_TYPE: - AI_LWO_VALIDATE_CHUNK_LENGTH(head->length,TYPE,2); - mFileBuffer++; // skip user format - - // Determine type of envelope - envelope.type = (LWO::EnvelopeType)*mFileBuffer; - ++mFileBuffer; - break; - - // precondition - case AI_LWO_PRE: - AI_LWO_VALIDATE_CHUNK_LENGTH(head->length,PRE,2); - envelope.pre = (LWO::PrePostBehaviour)GetU2(); - break; - - // postcondition - case AI_LWO_POST: - AI_LWO_VALIDATE_CHUNK_LENGTH(head->length,POST,2); - envelope.post = (LWO::PrePostBehaviour)GetU2(); - break; - - // keyframe - case AI_LWO_KEY: - { - AI_LWO_VALIDATE_CHUNK_LENGTH(head->length,KEY,8); - - envelope.keys.push_back(LWO::Key()); - LWO::Key& key = envelope.keys.back(); - - key.time = GetF4(); - key.value = GetF4(); - break; - } - - // interval interpolation - case AI_LWO_SPAN: - { - AI_LWO_VALIDATE_CHUNK_LENGTH(head->length,SPAN,4); - if (envelope.keys.size()<2) - DefaultLogger::get()->warn("LWO2: Unexpected SPAN chunk"); - else { - LWO::Key& key = envelope.keys.back(); - switch (GetU4()) - { - case AI_LWO_STEP: - key.inter = LWO::IT_STEP;break; - case AI_LWO_LINE: - key.inter = LWO::IT_LINE;break; - case AI_LWO_TCB: - key.inter = LWO::IT_TCB;break; - case AI_LWO_HERM: - key.inter = LWO::IT_HERM;break; - case AI_LWO_BEZI: - key.inter = LWO::IT_BEZI;break; - case AI_LWO_BEZ2: - key.inter = LWO::IT_BEZ2;break; - default: - DefaultLogger::get()->warn("LWO2: Unknown interval interpolation mode"); - }; - - // todo ... read params - } - break; - } - - default: - DefaultLogger::get()->warn("LWO2: Encountered unknown ENVL subchunk"); - } - // regardless how much we did actually read, go to the next chunk - mFileBuffer = next; - } + } + + // ... and read all subchunks + while (true) + { + if (mFileBuffer + 6 >= end)break; + LE_NCONST IFF::SubChunkHeader head = IFF::LoadSubChunk(mFileBuffer); + + if (mFileBuffer + head.length > end) + throw DeadlyImportError("LWO2: Invalid envelope chunk length"); + + uint8_t* const next = mFileBuffer+head.length; + switch (head.type) + { + // Type & representation of the envelope + case AI_LWO_TYPE: + AI_LWO_VALIDATE_CHUNK_LENGTH(head.length,TYPE,2); + mFileBuffer++; // skip user format + + // Determine type of envelope + envelope.type = (LWO::EnvelopeType)*mFileBuffer; + ++mFileBuffer; + break; + + // precondition + case AI_LWO_PRE: + AI_LWO_VALIDATE_CHUNK_LENGTH(head.length,PRE,2); + envelope.pre = (LWO::PrePostBehaviour)GetU2(); + break; + + // postcondition + case AI_LWO_POST: + AI_LWO_VALIDATE_CHUNK_LENGTH(head.length,POST,2); + envelope.post = (LWO::PrePostBehaviour)GetU2(); + break; + + // keyframe + case AI_LWO_KEY: + { + AI_LWO_VALIDATE_CHUNK_LENGTH(head.length,KEY,8); + + envelope.keys.push_back(LWO::Key()); + LWO::Key& key = envelope.keys.back(); + + key.time = GetF4(); + key.value = GetF4(); + break; + } + + // interval interpolation + case AI_LWO_SPAN: + { + AI_LWO_VALIDATE_CHUNK_LENGTH(head.length,SPAN,4); + if (envelope.keys.size()<2) + DefaultLogger::get()->warn("LWO2: Unexpected SPAN chunk"); + else { + LWO::Key& key = envelope.keys.back(); + switch (GetU4()) + { + case AI_LWO_STEP: + key.inter = LWO::IT_STEP;break; + case AI_LWO_LINE: + key.inter = LWO::IT_LINE;break; + case AI_LWO_TCB: + key.inter = LWO::IT_TCB;break; + case AI_LWO_HERM: + key.inter = LWO::IT_HERM;break; + case AI_LWO_BEZI: + key.inter = LWO::IT_BEZI;break; + case AI_LWO_BEZ2: + key.inter = LWO::IT_BEZ2;break; + default: + DefaultLogger::get()->warn("LWO2: Unknown interval interpolation mode"); + }; + + // todo ... read params + } + break; + } + + default: + DefaultLogger::get()->warn("LWO2: Encountered unknown ENVL subchunk"); + } + // regardless how much we did actually read, go to the next chunk + mFileBuffer = next; + } } // ------------------------------------------------------------------------------------------------ // Load file - master function void LWOImporter::LoadLWO2File() { - bool skip = false; - - LE_NCONST uint8_t* const end = mFileBuffer + fileSize; - while (true) - { - if (mFileBuffer + sizeof(IFF::ChunkHeader) > end)break; - IFF::ChunkHeader* const head = IFF::LoadChunk(mFileBuffer); - - if (mFileBuffer + head->length > end) - { - throw DeadlyImportError("LWO2: Chunk length points behind the file"); - break; - } - uint8_t* const next = mFileBuffer+head->length; - unsigned int iUnnamed = 0; - - if(!head->length) { - mFileBuffer = next; - continue; - } - - switch (head->type) - { - // new layer - case AI_LWO_LAYR: - { - // add a new layer to the list .... - mLayers->push_back ( LWO::Layer() ); - LWO::Layer& layer = mLayers->back(); - mCurLayer = &layer; - - AI_LWO_VALIDATE_CHUNK_LENGTH(head->length,LAYR,16); - - // layer index. - layer.mIndex = GetU2(); - - // Continue loading this layer or ignore it? Check the layer index property - if (UINT_MAX != configLayerIndex && (configLayerIndex-1) != layer.mIndex) { - skip = true; - } - else skip = false; - - // pivot point - mFileBuffer += 2; /* unknown */ - mCurLayer->mPivot.x = GetF4(); - mCurLayer->mPivot.y = GetF4(); - mCurLayer->mPivot.z = GetF4(); - GetS0(layer.mName,head->length-16); - - // if the name is empty, generate a default name - if (layer.mName.empty()) { - char buffer[128]; // should be sufficiently large - ::sprintf(buffer,"Layer_%i", iUnnamed++); - layer.mName = buffer; - } - - // load this layer or ignore it? Check the layer name property - if (configLayerName.length() && configLayerName != layer.mName) { - skip = true; - } - else hasNamedLayer = true; - - // optional: parent of this layer - if (mFileBuffer + 2 <= next) - layer.mParent = GetU2(); - else layer.mParent = -1; - - // Set layer skip parameter - layer.skip = skip; - - break; - } - - // vertex list - case AI_LWO_PNTS: - { - if (skip) - break; - - unsigned int old = (unsigned int)mCurLayer->mTempPoints.size(); - LoadLWOPoints(head->length); - mCurLayer->mPointIDXOfs = old; - break; - } - // vertex tags - case AI_LWO_VMAD: - if (mCurLayer->mFaces.empty()) - { - DefaultLogger::get()->warn("LWO2: Unexpected VMAD chunk"); - break; - } - // --- intentionally no break here - case AI_LWO_VMAP: - { - if (skip) - break; - - if (mCurLayer->mTempPoints.empty()) - DefaultLogger::get()->warn("LWO2: Unexpected VMAP chunk"); - else LoadLWO2VertexMap(head->length,head->type == AI_LWO_VMAD); - break; - } - // face list - case AI_LWO_POLS: - { - if (skip) - break; - - unsigned int old = (unsigned int)mCurLayer->mFaces.size(); - LoadLWO2Polygons(head->length); - mCurLayer->mFaceIDXOfs = old; - break; - } - // polygon tags - case AI_LWO_PTAG: - { - if (skip) - break; - - if (mCurLayer->mFaces.empty()) - DefaultLogger::get()->warn("LWO2: Unexpected PTAG"); - else LoadLWO2PolygonTags(head->length); - break; - } - // list of tags - case AI_LWO_TAGS: - { - if (!mTags->empty()) - DefaultLogger::get()->warn("LWO2: SRFS chunk encountered twice"); - else LoadLWOTags(head->length); - break; - } - - // surface chunk - case AI_LWO_SURF: - { - LoadLWO2Surface(head->length); - break; - } - - // clip chunk - case AI_LWO_CLIP: - { - LoadLWO2Clip(head->length); - break; - } - - // envelope chunk - case AI_LWO_ENVL: - { - LoadLWO2Envelope(head->length); - break; - } - } - mFileBuffer = next; - } + bool skip = false; + + LE_NCONST uint8_t* const end = mFileBuffer + fileSize; + while (true) + { + if (mFileBuffer + sizeof(IFF::ChunkHeader) > end)break; + const IFF::ChunkHeader head = IFF::LoadChunk(mFileBuffer); + + if (mFileBuffer + head.length > end) + { + throw DeadlyImportError("LWO2: Chunk length points behind the file"); + break; + } + uint8_t* const next = mFileBuffer+head.length; + unsigned int iUnnamed = 0; + + if(!head.length) { + mFileBuffer = next; + continue; + } + + switch (head.type) + { + // new layer + case AI_LWO_LAYR: + { + // add a new layer to the list .... + mLayers->push_back ( LWO::Layer() ); + LWO::Layer& layer = mLayers->back(); + mCurLayer = &layer; + + AI_LWO_VALIDATE_CHUNK_LENGTH(head.length,LAYR,16); + + // layer index. + layer.mIndex = GetU2(); + + // Continue loading this layer or ignore it? Check the layer index property + if (UINT_MAX != configLayerIndex && (configLayerIndex-1) != layer.mIndex) { + skip = true; + } + else skip = false; + + // pivot point + mFileBuffer += 2; /* unknown */ + mCurLayer->mPivot.x = GetF4(); + mCurLayer->mPivot.y = GetF4(); + mCurLayer->mPivot.z = GetF4(); + GetS0(layer.mName,head.length-16); + + // if the name is empty, generate a default name + if (layer.mName.empty()) { + char buffer[128]; // should be sufficiently large + ::ai_snprintf(buffer, 128, "Layer_%i", iUnnamed++); + layer.mName = buffer; + } + + // load this layer or ignore it? Check the layer name property + if (configLayerName.length() && configLayerName != layer.mName) { + skip = true; + } + else hasNamedLayer = true; + + // optional: parent of this layer + if (mFileBuffer + 2 <= next) + layer.mParent = GetU2(); + else layer.mParent = -1; + + // Set layer skip parameter + layer.skip = skip; + + break; + } + + // vertex list + case AI_LWO_PNTS: + { + if (skip) + break; + + unsigned int old = (unsigned int)mCurLayer->mTempPoints.size(); + LoadLWOPoints(head.length); + mCurLayer->mPointIDXOfs = old; + break; + } + // vertex tags + case AI_LWO_VMAD: + if (mCurLayer->mFaces.empty()) + { + DefaultLogger::get()->warn("LWO2: Unexpected VMAD chunk"); + break; + } + // --- intentionally no break here + case AI_LWO_VMAP: + { + if (skip) + break; + + if (mCurLayer->mTempPoints.empty()) + DefaultLogger::get()->warn("LWO2: Unexpected VMAP chunk"); + else LoadLWO2VertexMap(head.length,head.type == AI_LWO_VMAD); + break; + } + // face list + case AI_LWO_POLS: + { + if (skip) + break; + + unsigned int old = (unsigned int)mCurLayer->mFaces.size(); + LoadLWO2Polygons(head.length); + mCurLayer->mFaceIDXOfs = old; + break; + } + // polygon tags + case AI_LWO_PTAG: + { + if (skip) + break; + + if (mCurLayer->mFaces.empty()) + DefaultLogger::get()->warn("LWO2: Unexpected PTAG"); + else LoadLWO2PolygonTags(head.length); + break; + } + // list of tags + case AI_LWO_TAGS: + { + if (!mTags->empty()) + DefaultLogger::get()->warn("LWO2: SRFS chunk encountered twice"); + else LoadLWOTags(head.length); + break; + } + + // surface chunk + case AI_LWO_SURF: + { + LoadLWO2Surface(head.length); + break; + } + + // clip chunk + case AI_LWO_CLIP: + { + LoadLWO2Clip(head.length); + break; + } + + // envelope chunk + case AI_LWO_ENVL: + { + LoadLWO2Envelope(head.length); + break; + } + } + mFileBuffer = next; + } } #endif // !! ASSIMP_BUILD_NO_LWO_IMPORTER |