| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820 |
- /*
- ---------------------------------------------------------------------------
- Open Asset Import Library (ASSIMP)
- ---------------------------------------------------------------------------
- Copyright (c) 2006-2008, ASSIMP Development 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
- conditions are met:
- * Redistributions of source code must retain the above
- copyright notice, this list of conditions and the
- following disclaimer.
- * Redistributions in binary form must reproduce the above
- copyright notice, this list of conditions and the
- following disclaimer in the documentation and/or other
- materials provided with the distribution.
- * Neither the name of the ASSIMP team, nor the names of its
- contributors may be used to endorse or promote products
- derived from this software without specific prior
- written permission of the ASSIMP Development Team.
- 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
- OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
- 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
- OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- ---------------------------------------------------------------------------
- */
- /** @file Implementation of the LWO importer class */
- // internal headers
- #include "LWOLoader.h"
- #include "MaterialSystem.h"
- #include "StringComparison.h"
- #include "ByteSwap.h"
- // public assimp headers
- #include "../include/IOStream.h"
- #include "../include/IOSystem.h"
- #include "../include/aiScene.h"
- #include "../include/aiAssert.h"
- #include "../include/DefaultLogger.h"
- #include "../include/assimp.hpp"
- // boost headers
- #include <boost/scoped_ptr.hpp>
- using namespace Assimp;
- // ------------------------------------------------------------------------------------------------
- // Constructor to be privately used by Importer
- LWOImporter::LWOImporter()
- {
- }
- // ------------------------------------------------------------------------------------------------
- // Destructor, private as well
- LWOImporter::~LWOImporter()
- {
- }
- // ------------------------------------------------------------------------------------------------
- // Returns whether the class can handle the format of the given file.
- bool LWOImporter::CanRead( const std::string& pFile, IOSystem* pIOHandler) const
- {
- // simple check of file extension is enough for the moment
- std::string::size_type pos = pFile.find_last_of('.');
- // no file extension - can't read
- if( pos == std::string::npos)return false;
- std::string extension = pFile.substr( pos);
- if (extension.length() < 4)return false;
- if (extension[0] != '.')return false;
- if (extension[1] != 'l' && extension[1] != 'L')return false;
- if (extension[2] != 'w' && extension[2] != 'W')return false;
- if (extension[3] != 'o' && extension[3] != 'O')return false;
- return true;
- }
- // ------------------------------------------------------------------------------------------------
- // Setup configuration properties
- void LWOImporter::SetupProperties(const Importer* pImp)
- {
- this->configGradientResX = pImp->GetPropertyInteger(AI_CONFIG_IMPORT_LWO_GRADIENT_RESX,512);
- this->configGradientResY = pImp->GetPropertyInteger(AI_CONFIG_IMPORT_LWO_GRADIENT_RESY,512);
- }
- // ------------------------------------------------------------------------------------------------
- // 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 new ImportErrorException( "Failed to open LWO file " + pFile + ".");
- if((this->fileSize = (unsigned int)file->FileSize()) < 12)
- throw new ImportErrorException("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 new ImportErrorException(sz);
- mFileBuffer = &mBuffer[0] + 12;
- fileSize -= 12;
- // 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;
- mLayers = &_mLayers;
- TagList _mTags;
- mTags = &_mTags;
- TagMappingTable _mMapping;
- mMapping = &_mMapping;
- SurfaceList _mSurfaces;
- mSurfaces = &_mSurfaces;
- // allocate a default layer
- mLayers->push_back(Layer());
- mCurLayer = &mLayers->back();
- mCurLayer->mName = "<LWODefault>";
- // old lightwave file format (prior to v6)
- if (AI_LWO_FOURCC_LWOB == fileType)
- {
- mIsLWO2 = false;
- this->LoadLWOBFile();
- }
- // new lightwave format
- else if (AI_LWO_FOURCC_LWO2 == fileType)
- {
- mIsLWO2 = true;
- this->LoadLWO2File();
- }
- // 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);
- throw new ImportErrorException(std::string("Unknown LWO sub format: ") + szBuff);
- }
- ResolveTags();
- // now process all layers and build meshes and nodes
- std::vector<aiMesh*> apcMeshes;
- std::vector<aiNode*> apcNodes;
- apcNodes.reserve(mLayers->size());
- apcMeshes.reserve(mLayers->size()*std::min(((unsigned int)mSurfaces->size()/2u), 1u));
- // 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)
- {
- MaterialHelper* pcMat = new MaterialHelper();
- pScene->mMaterials[mat] = pcMat;
- ConvertMaterial((*mSurfaces)[mat],pcMat);
- }
- unsigned int iDefaultSurface = 0xffffffff; // index of the default surface
- for (LayerList::const_iterator lit = mLayers->begin(), lend = mLayers->end();
- lit != lend;++lit)
- {
- const LWO::Layer& layer = *lit;
- // 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
- typedef std::vector<unsigned int> SortedRep;
- std::vector<SortedRep> pSorted(mSurfaces->size()+1);
- unsigned int i = 0;
- for (FaceList::const_iterator it = layer.mFaces.begin(), end = layer.mFaces.end();
- it != end;++it,++i)
- {
- unsigned int idx = (*it).surfaceIndex;
- if (idx >= mTags->size())
- {
- DefaultLogger::get()->warn("LWO: Invalid face surface index");
- idx = (unsigned int)mTags->size()-1;
- }
- if(0xffffffff == (idx = _mMapping[idx]))
- {
- if (0xffffffff == 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;
- }
- idx = iDefaultSurface;
- }
- pSorted[idx].push_back(i);
- }
- if (0xffffffff == iDefaultSurface)pSorted.erase(pSorted.end()-1);
- // now generate output meshes
- for (unsigned int p = 0; p < mSurfaces->size();++p)
- if (!pSorted[p].empty())pScene->mNumMeshes++;
- if (!pScene->mNumMeshes)
- throw new ImportErrorException("LWO: There are no meshes");
- pScene->mMeshes = new aiMesh*[pScene->mNumMeshes];
- 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();
- for (SortedRep::const_iterator it = sorted.begin(), end = sorted.end();
- it != end;++it)
- {
- mesh->mNumVertices += layer.mFaces[*it].mNumIndices;
- }
- aiVector3D* 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];
- #if _DEBUG
- for (unsigned int mui = 0; mui < AI_MAX_NUMBER_OF_TEXTURECOORDS;++mui )
- vUVChannelIndices[mui] = 0xffffffff;
- for (unsigned int mui = 0; mui < AI_MAX_NUMBER_OF_COLOR_SETS;++mui )
- vVColorIndices[mui] = 0xffffffff;
- #endif
- FindUVChannels(_mSurfaces[i],layer,vUVChannelIndices);
- FindVCChannels(_mSurfaces[i],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 (0xffffffff == 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;
- }
-
- aiColor4D* pvVC[AI_MAX_NUMBER_OF_COLOR_SETS];
- for (unsigned int mui = 0; mui < AI_MAX_NUMBER_OF_COLOR_SETS;++mui)
- {
- if (0xffffffff == vVColorIndices[mui])break;
- pvVC[mui] = mesh->mColors[mui] = new aiColor4D[mesh->mNumVertices];
- }
- // now convert all faces
- unsigned int vert = 0;
- for (SortedRep::const_iterator it = sorted.begin(), end = sorted.end();
- it != end;++it)
- {
- const LWO::Face& face = layer.mFaces[*it];
- // copy all vertices
- for (unsigned int q = 0; q < face.mNumIndices;++q)
- {
- register unsigned int idx = face.mIndices[q];
- *pv++ = layer.mTempPoints[idx];
- // process UV coordinates
- for (unsigned int w = 0; w < AI_MAX_NUMBER_OF_TEXTURECOORDS;++w)
- {
- if (0xffffffff == vUVChannelIndices[w])break;
- *(pvUV[w])++ = layer.mUVChannels[vUVChannelIndices[w]].data[idx];
- }
- // process vertex colors
- for (unsigned int w = 0; w < AI_MAX_NUMBER_OF_COLOR_SETS;++w)
- {
- if (0xffffffff == vVColorIndices[w])break;
- *(pvVC[w])++ = layer.mVColorChannels[vVColorIndices[w]].data[idx];
- }
- #if 0
- // process vertex weights - not yet supported
- for (unsigned int w = 0; w < layer.mWeightChannels.size();++w)
- {
- }
- #endif
- face.mIndices[q] = vert++;
- }
- pf->mIndices = face.mIndices;
- pf->mNumIndices = face.mNumIndices;
- const_cast<unsigned int*>(face.mIndices) = NULL; // make sure it won't be deleted
- pf++;
- }
- ++p;
- }
- }
- // generate nodes to render the mesh. Store the parent index
- // in the mParent member of the nodes
- aiNode* pcNode = new aiNode();
- apcNodes.push_back(pcNode);
- pcNode->mName.Set(layer.mName);
- pcNode->mParent = reinterpret_cast<aiNode*>(layer.mParent);
- pcNode->mNumMeshes = (unsigned int)apcMeshes.size() - meshStart;
- pcNode->mMeshes = new unsigned int[pcNode->mNumMeshes];
- for (unsigned int p = 0; p < pcNode->mNumMeshes;++p)
- pcNode->mMeshes[p] = p + meshStart;
- }
- // generate the final node graph
- GenerateNodeGraph(apcNodes);
- // copy the meshes to the output structure
- if (apcMeshes.size()) // shouldn't occur, just to be sure we don't crash
- {
- pScene->mMeshes = new aiMesh*[ pScene->mNumMeshes = (unsigned int)apcMeshes.size() ];
- ::memcpy(pScene->mMeshes,&apcMeshes[0],pScene->mNumMeshes*sizeof(void*));
- }
- }
- // ------------------------------------------------------------------------------------------------
- void LWOImporter::GenerateNodeGraph(std::vector<aiNode*>& apcNodes)
- {
- // now generate the final nodegraph
- uint16_t curIndex = 0;
- while (curIndex < (uint16_t)apcNodes.size())
- {
- aiNode* node;
- uint16_t iCurParent = curIndex-1;
- node = curIndex ? apcNodes[iCurParent] : new aiNode("<dummy_root>");
- unsigned int numChilds = 0;
- for (unsigned int i = 0; i < apcNodes.size();++i)
- {
- if (i == iCurParent)continue;
- if ( reinterpret_cast<uint16_t>(apcNodes[i]->mParent) == iCurParent)++numChilds;
- }
- if (numChilds)
- {
- if (!pScene->mRootNode)
- {
- pScene->mRootNode = node;
- }
- node->mChildren = new aiNode* [ node->mNumChildren = numChilds ];
- for (unsigned int i = 0, p = 0; i < apcNodes.size();++i)
- {
- if (i == iCurParent)continue;
- uint16_t parent = reinterpret_cast<uint16_t>(apcNodes[i]->mParent);
- if (parent == iCurParent)
- {
- node->mChildren[p++] = apcNodes[i];
- apcNodes[i]->mParent = node;
- apcNodes[i] = NULL;
- }
- }
- }
- else if (!curIndex)delete node;
- ++curIndex;
- }
- // remove a single root node
- // TODO: implement directly in the above loop, no need to deallocate here
- if (1 == pScene->mRootNode->mNumChildren)
- {
- aiNode* pc = pScene->mRootNode->mChildren[0];
- pc->mParent = pScene->mRootNode->mChildren[0] = NULL;
- delete pScene->mRootNode;
- pScene->mRootNode = pc;
- }
- // add unreferenced nodes to a dummy root
- unsigned int m = 0;
- for (std::vector<aiNode*>::iterator it = apcNodes.begin(), end = apcNodes.end();
- it != end;++it)
- {
- aiNode* p = *it;
- if (p)++m;
- }
- if (m)
- {
- aiNode* pc = new aiNode();
- pc->mName.Set("<dummy_root>");
- aiNode** cc = pc->mChildren = new aiNode*[ pc->mNumChildren = m+1 ];
- for (std::vector<aiNode*>::iterator it = apcNodes.begin(), end = apcNodes.end();
- it != end;++it)
- {
- aiNode* p = *it;
- if (p)*cc++ = p;
- }
- if (pScene->mRootNode)
- {
- *cc = pScene->mRootNode;
- pScene->mRootNode->mParent = pc;
- }
- else --pc->mNumChildren;
- pScene->mRootNode = pc;
- }
- if (!pScene->mRootNode)throw new ImportErrorException("LWO: Unable to build a valid node graph");
- }
- // ------------------------------------------------------------------------------------------------
- void LWOImporter::ResolveTags()
- {
- // --- this function is used for both LWO2 and LWOB
- mMapping->resize(mTags->size(),0xffffffff);
- for (unsigned int a = 0; a < mTags->size();++a)
- {
- for (unsigned int i = 0; i < mSurfaces->size();++i)
- {
- const std::string& c = (*mTags)[a];
- const std::string& d = (*mSurfaces)[i].mName;
- if (!ASSIMP_stricmp(c,d))
- {
- (*mMapping)[a] = i;
- break;
- }
- }
- }
- }
- // ------------------------------------------------------------------------------------------------
- void LWOImporter::ParseString(std::string& out,unsigned int max)
- {
- // --- this function is used for both LWO2 and LWOB
- unsigned int iCursor = 0;
- const char* in = (const char*)mFileBuffer,*sz = in;
- while (*in)
- {
- if (++iCursor > max)
- {
- DefaultLogger::get()->warn("LWOB: Invalid file, string is is too long");
- break;
- }
- ++in;
- }
- unsigned int len = (unsigned int) (in-sz);
- out = std::string(sz,len);
- }
- // ------------------------------------------------------------------------------------------------
- void LWOImporter::AdjustTexturePath(std::string& out)
- {
- // --- this function is used for both LWO2 and LWOB
- if (::strstr(out.c_str(), "(sequence)"))
- {
- // remove the (sequence) and append 000
- DefaultLogger::get()->info("LWO: Sequence of animated texture found. It will be ignored");
- out = out.substr(0,out.length()-10) + "000";
- }
- }
- // ------------------------------------------------------------------------------------------------
- int LWOImporter::ReadVSizedIntLWO2(uint8_t*& inout)
- {
- int i;
- int c = *inout;inout++;
- if(c != 0xFF)
- {
- i = c << 8;
- c = *inout;inout++;
- i |= c;
- }
- else
- {
- c = *inout;inout++;
- i = c << 16;
- c = *inout;inout++;
- i |= c << 8;
- c = *inout;inout++;
- i |= c;
- }
- return i;
- }
- // ------------------------------------------------------------------------------------------------
- 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 unsigned int len = (unsigned int)(szCur-szLast);
- mTags->push_back(std::string(szLast,len));
- szCur += len & 1;
- szLast = szCur;
- }
- szCur++;
- }
- }
- // ------------------------------------------------------------------------------------------------
- void LWOImporter::LoadLWOPoints(unsigned int length)
- {
- // --- this function is used for both LWO2 and LWOB
- mCurLayer->mTempPoints.resize( length / 12 );
- // perform endianess conversions
- #ifndef AI_BUILD_BIG_ENDIAN
- for (unsigned int i = 0; i < length>>2;++i)
- ByteSwap::Swap4( mFileBuffer + (i << 2));
- #endif
- ::memcpy(&mCurLayer->mTempPoints[0],mFileBuffer,length);
- }
- // ------------------------------------------------------------------------------------------------
- void LWOImporter::LoadLWOPolygons(unsigned int length)
- {
- // --- this function is used for both LWO2 and LWOB
- if (mIsLWO2)
- {
- uint32_t type = *((LE_NCONST uint32_t*)mFileBuffer);mFileBuffer += 4;
- if (type != AI_LWO_FACE)
- {
- DefaultLogger::get()->warn("LWO2: Only POLS.FACE chunsk are supported.");
- return;
- }
- }
- // first find out how many faces and vertices we'll finally need
- LE_NCONST uint16_t* const end = (LE_NCONST uint16_t*)(mFileBuffer+length);
- LE_NCONST uint16_t* cursor = (LE_NCONST uint16_t*)mFileBuffer;
- // perform endianess conversions
- #ifndef AI_BUILD_BIG_ENDIAN
- while (cursor < end)ByteSwap::Swap2(cursor++);
- cursor = (LE_NCONST uint16_t*)mFileBuffer;
- #endif
- unsigned int iNumFaces = 0,iNumVertices = 0;
- if (mIsLWO2)CountVertsAndFacesLWO2(iNumVertices,iNumFaces,cursor,end);
- else CountVertsAndFacesLWOB(iNumVertices,iNumFaces,cursor,end);
- // allocate the output array and copy face indices
- if (iNumFaces)
- {
- cursor = (LE_NCONST uint16_t*)mFileBuffer;
- mCurLayer->mFaces.resize(iNumFaces);
- FaceList::iterator it = mCurLayer->mFaces.begin();
- if (mIsLWO2)CopyFaceIndicesLWO2(it,cursor,end);
- else CopyFaceIndicesLWOB(it,cursor,end);
- }
- }
- // ------------------------------------------------------------------------------------------------
- void LWOImporter::CountVertsAndFacesLWO2(unsigned int& verts, unsigned int& faces,
- LE_NCONST uint16_t*& cursor, const uint16_t* const end, unsigned int max)
- {
- while (cursor < end && max--)
- {
- uint16_t numIndices = *cursor++;
- numIndices &= 0x03FF;
- verts += numIndices;++faces;
- for(uint16_t i = 0; i < numIndices; i++)
- ReadVSizedIntLWO2((uint8_t*&)cursor);
- }
- }
- // ------------------------------------------------------------------------------------------------
- void LWOImporter::CopyFaceIndicesLWO2(FaceList::iterator& it,
- LE_NCONST uint16_t*& cursor,
- const uint16_t* const end,
- unsigned int max)
- {
- while (cursor < end && max--)
- {
- LWO::Face& face = *it;++it;
- if(face.mNumIndices = (*cursor++) & 0x03FF)
- {
- 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: face index is out of range");
- face.mIndices[i] = (unsigned int)mCurLayer->mTempPoints.size()-1;
- }
- }
- }
- else DefaultLogger::get()->warn("LWO2: face has 0 indices");
- }
- }
- // ------------------------------------------------------------------------------------------------
- void LWOImporter::LoadLWO2PolygonTags(unsigned int length)
- {
- uint32_t type = *((LE_NCONST uint32_t*)mFileBuffer);mFileBuffer+=4;
- AI_LSWAP4(type);
- if (type != AI_LWO_SURF && type != AI_LWO_SMGP)
- return;
- LE_NCONST uint8_t* const end = mFileBuffer+length;
- while (mFileBuffer < end)
- {
- unsigned int i = ReadVSizedIntLWO2(mFileBuffer) + mCurLayer->mFaceIDXOfs;
- unsigned int j = ReadVSizedIntLWO2(mFileBuffer);
- if (i > mCurLayer->mFaces.size())
- {
- DefaultLogger::get()->warn("LWO2: face index in ptag list is out of range");
- continue;
- }
- switch (type)
- {
- case AI_LWO_SURF:
- mCurLayer->mFaces[i].surfaceIndex = j;
- break;
- case AI_LWO_SMGP:
- mCurLayer->mFaces[i].smoothGroup = j;
- break;
- };
- }
- }
- // ------------------------------------------------------------------------------------------------
- void LWOImporter::LoadLWO2VertexMap(unsigned int length, bool perPoly)
- {
- unsigned int type = *((LE_NCONST uint32_t*)mFileBuffer);mFileBuffer+=4;
- unsigned int dims = *((LE_NCONST uint16_t*)mFileBuffer);mFileBuffer+=2;
- VMapEntry* base;
- switch (type)
- {
- case AI_LWO_TXUV:
- if (dims != 2)
- {
- DefaultLogger::get()->warn("LWO2: Found UV channel with != 2 components");
- }
- mCurLayer->mUVChannels.push_back(UVChannel(mCurLayer->mTempPoints.size()));
- base = &mCurLayer->mUVChannels.back();
- case AI_LWO_WGHT:
- if (dims != 1)
- {
- DefaultLogger::get()->warn("LWO2: found vertex weight map with != 1 components");
- }
- mCurLayer->mWeightChannels.push_back(WeightChannel(mCurLayer->mTempPoints.size()));
- base = &mCurLayer->mWeightChannels.back();
- case AI_LWO_RGB:
- case AI_LWO_RGBA:
- if (dims != 3 && dims != 4)
- {
- DefaultLogger::get()->warn("LWO2: found vertex color map with != 3&4 components");
- }
- mCurLayer->mVColorChannels.push_back(VColorChannel(mCurLayer->mTempPoints.size()));
- base = &mCurLayer->mVColorChannels.back();
- default: return;
- };
- // read the name of the vertex map
- ParseString(base->name,length);
- // now read all entries in the map
- type = std::min(dims,base->dims);
- const unsigned int diff = (dims - type)<<2;
- LE_NCONST uint8_t* const end = mFileBuffer+length;
- while (mFileBuffer < end)
- {
- unsigned int idx = ReadVSizedIntLWO2(mFileBuffer) + mCurLayer->mPointIDXOfs;
- if (idx > mCurLayer->mTempPoints.size())
- {
- DefaultLogger::get()->warn("LWO2: vertex index in vmap/vmad is out of range");
- continue;
- }
- for (unsigned int i = 0; i < type;++i)
- {
- base->rawData[idx*dims+i]= *((float*)mFileBuffer);
- mFileBuffer += 4;
- }
- mFileBuffer += diff;
- }
- }
- // ------------------------------------------------------------------------------------------------
- void LWOImporter::LoadLWO2File()
- {
- LE_NCONST uint8_t* const end = mFileBuffer + fileSize;
- while (true)
- {
- if (mFileBuffer + sizeof(IFF::ChunkHeader) > end)break;
- LE_NCONST IFF::ChunkHeader* const head = (LE_NCONST IFF::ChunkHeader*)mFileBuffer;
- AI_LSWAP4(head->length);
- AI_LSWAP4(head->type);
- mFileBuffer += sizeof(IFF::ChunkHeader);
- if (mFileBuffer + head->length > end)
- {
- throw new ImportErrorException("LWOB: Invalid file, the size attribute of "
- "a chunk points behind the end of the file");
- break;
- }
- LE_NCONST uint8_t* const next = mFileBuffer+head->length;
- unsigned int iUnnamed = 0;
- 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);
- // and parse its properties
- mFileBuffer += 16;
- ParseString(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;
- }
- if (mFileBuffer + 2 <= next)
- layer.mParent = *((uint16_t*)mFileBuffer);
- break;
- }
- // vertex list
- case AI_LWO_PNTS:
- {
- unsigned int old = (unsigned int)mCurLayer->mTempPoints.size();
- LoadLWOPoints(head->length);
- mCurLayer->mPointIDXOfs = old;
- break;
- }
- // vertex tags
- //case AI_LWO_VMAD:
- case AI_LWO_VMAP:
- {
- if (mCurLayer->mTempPoints.empty())
- DefaultLogger::get()->warn("LWO2: Unexpected VMAD/VMAP chunk");
- else LoadLWO2VertexMap(head->length,head->type == AI_LWO_VMAD);
- break;
- }
- // face list
- case AI_LWO_POLS:
- {
- unsigned int old = (unsigned int)mCurLayer->mFaces.size();
- LoadLWOPolygons(head->length);
- mCurLayer->mFaceIDXOfs = old;
- break;
- }
- // polygon tags
- case AI_LWO_PTAG:
- {
- if (mCurLayer->mFaces.empty())
- DefaultLogger::get()->warn("LWO2: Unexpected PTAG");
- else LoadLWO2PolygonTags(head->length);
- break;
- }
- // list of tags
- case AI_LWO_SRFS:
- {
- if (!mTags->empty())
- DefaultLogger::get()->warn("LWO2: SRFS chunk encountered twice");
- else LoadLWOTags(head->length);
- break;
- }
- // surface chunk
- case AI_LWO_SURF:
- {
- if (!mSurfaces->empty())
- DefaultLogger::get()->warn("LWO2: SURF chunk encountered twice");
- else LoadLWO2Surface(head->length);
- break;
- }
- }
- mFileBuffer = next;
- }
- }
|