diff options
author | Sean Harmer <sean.harmer@kdab.com> | 2017-05-24 12:09:44 +0100 |
---|---|---|
committer | Sean Harmer <sean.harmer@kdab.com> | 2017-05-24 12:10:02 +0100 |
commit | 77d294db076dac19e8b549b445ffede9f7260c84 (patch) | |
tree | 828ee7a6862ec5c0bd24f97cb540625a2c647376 /src/3rdparty/assimp/code/TerragenLoader.cpp | |
parent | 59f8fec8a41606b3185fe3a4e276978e3e1ed5ef (diff) | |
parent | 939b9b4b7591e8a421cf048a0a84ed3e75d81d21 (diff) |
Merge branch 'dev' into wip/animationwip/animation
Change-Id: I6e770609c90a7745d08fa4e2f424e865678c5d6f
Diffstat (limited to 'src/3rdparty/assimp/code/TerragenLoader.cpp')
-rw-r--r-- | src/3rdparty/assimp/code/TerragenLoader.cpp | 395 |
1 files changed, 201 insertions, 194 deletions
diff --git a/src/3rdparty/assimp/code/TerragenLoader.cpp b/src/3rdparty/assimp/code/TerragenLoader.cpp index 27775dd9e..854959f19 100644 --- a/src/3rdparty/assimp/code/TerragenLoader.cpp +++ b/src/3rdparty/assimp/code/TerragenLoader.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,40 +25,47 @@ 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. --------------------------------------------------------------------------- */ /** @file Implementation of the Terragen importer class */ -#include "AssimpPCH.h" + #ifndef ASSIMP_BUILD_NO_TERRAGEN_IMPORTER + #include "TerragenLoader.h" +#include "StreamReader.h" +#include <assimp/Importer.hpp> +#include <assimp/IOSystem.hpp> +#include <assimp/scene.h> +#include <assimp/DefaultLogger.hpp> + using namespace Assimp; static const aiImporterDesc desc = { - "Terragen Heightmap Importer", - "", - "", - "http://www.planetside.co.uk/", - aiImporterFlags_SupportBinaryFlavour, - 0, - 0, - 0, - 0, - "ter" + "Terragen Heightmap Importer", + "", + "", + "http://www.planetside.co.uk/", + aiImporterFlags_SupportBinaryFlavour, + 0, + 0, + 0, + 0, + "ter" }; // ------------------------------------------------------------------------------------------------ @@ -68,201 +75,201 @@ TerragenImporter::TerragenImporter() {} // ------------------------------------------------------------------------------------------------ -// Destructor, private as well +// Destructor, private as well TerragenImporter::~TerragenImporter() {} // ------------------------------------------------------------------------------------------------ -// 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 TerragenImporter::CanRead( const std::string& pFile, IOSystem* pIOHandler, bool checkSig) const { - // check file extension - std::string extension = GetExtension(pFile); - - if( extension == "ter") - return true; - - if( !extension.length() || checkSig) { - /* If CanRead() is called in order to check whether we - * support a specific file extension in general pIOHandler - * might be NULL and it's our duty to return true here. - */ - if (!pIOHandler)return true; - const char* tokens[] = {"terragen"}; - return SearchFileHeaderForToken(pIOHandler,pFile,tokens,1); - } - return false; + // check file extension + std::string extension = GetExtension(pFile); + + if( extension == "ter") + return true; + + if( !extension.length() || checkSig) { + /* If CanRead() is called in order to check whether we + * support a specific file extension in general pIOHandler + * might be NULL and it's our duty to return true here. + */ + if (!pIOHandler)return true; + const char* tokens[] = {"terragen"}; + return SearchFileHeaderForToken(pIOHandler,pFile,tokens,1); + } + return false; } // ------------------------------------------------------------------------------------------------ // Build a string of all file extensions supported const aiImporterDesc* TerragenImporter::GetInfo () const { - return &desc; + return &desc; } // ------------------------------------------------------------------------------------------------ // Setup import properties void TerragenImporter::SetupProperties(const Importer* pImp) { - // AI_CONFIG_IMPORT_TER_MAKE_UVS - configComputeUVs = ( 0 != pImp->GetPropertyInteger(AI_CONFIG_IMPORT_TER_MAKE_UVS,0) ); + // AI_CONFIG_IMPORT_TER_MAKE_UVS + configComputeUVs = ( 0 != pImp->GetPropertyInteger(AI_CONFIG_IMPORT_TER_MAKE_UVS,0) ); } // ------------------------------------------------------------------------------------------------ -// Imports the given file into the given scene structure. -void TerragenImporter::InternReadFile( const std::string& pFile, - aiScene* pScene, IOSystem* pIOHandler) +// Imports the given file into the given scene structure. +void TerragenImporter::InternReadFile( const std::string& pFile, + aiScene* pScene, IOSystem* pIOHandler) { - IOStream* file = pIOHandler->Open( pFile, "rb"); - - // Check whether we can read from the file - if( file == NULL) - throw DeadlyImportError( "Failed to open TERRAGEN TERRAIN file " + pFile + "."); - - // Construct a stream reader to read all data in the correct endianess - StreamReaderLE reader(file); - if(reader.GetRemainingSize() < 16) - throw DeadlyImportError( "TER: file is too small" ); - - // Check for the existence of the two magic strings 'TERRAGEN' and 'TERRAIN ' - if (::strncmp((const char*)reader.GetPtr(),AI_TERR_BASE_STRING,8)) - throw DeadlyImportError( "TER: Magic string \'TERRAGEN\' not found" ); - - if (::strncmp((const char*)reader.GetPtr()+8,AI_TERR_TERRAIN_STRING,8)) - throw DeadlyImportError( "TER: Magic string \'TERRAIN\' not found" ); - - unsigned int x = 0,y = 0,mode = 0; - float rad = 6370.f; - (void)rad; - - - aiNode* root = pScene->mRootNode = new aiNode(); - root->mName.Set("<TERRAGEN.TERRAIN>"); - - // Default scaling is 30 - root->mTransformation.a1 = root->mTransformation.b2 = root->mTransformation.c3 = 30.f; - - // Now read all chunks until we're finished or an EOF marker is encountered - reader.IncPtr(16); - while (reader.GetRemainingSize() >= 4) - { - const char* head = (const char*)reader.GetPtr(); - reader.IncPtr(4); - - // EOF, break in every case - if (!::strncmp(head,AI_TERR_EOF_STRING,4)) - break; - - // Number of x-data points - if (!::strncmp(head,AI_TERR_CHUNK_XPTS,4)) - { - x = (uint16_t)reader.GetI2(); - } - // Number of y-data points - else if (!::strncmp(head,AI_TERR_CHUNK_YPTS,4)) - { - y = (uint16_t)reader.GetI2(); - } - // Squared terrains width-1. - else if (!::strncmp(head,AI_TERR_CHUNK_SIZE,4)) - { - x = y = (uint16_t)reader.GetI2()+1; - } - // terrain scaling - else if (!::strncmp(head,AI_TERR_CHUNK_SCAL,4)) - { - root->mTransformation.a1 = reader.GetF4(); - root->mTransformation.b2 = reader.GetF4(); - root->mTransformation.c3 = reader.GetF4(); - } - // mapping == 1: earth radius - else if (!::strncmp(head,AI_TERR_CHUNK_CRAD,4)) - { - rad = reader.GetF4(); - } - // mapping mode - else if (!::strncmp(head,AI_TERR_CHUNK_CRVM,4)) - { - mode = reader.GetI1(); - if (0 != mode) - DefaultLogger::get()->error("TER: Unsupported mapping mode, a flat terrain is returned"); - } - // actual terrain data - else if (!::strncmp(head,AI_TERR_CHUNK_ALTW,4)) - { - float hscale = (float)reader.GetI2() / 65536; - float bheight = (float)reader.GetI2(); - - if (!hscale)hscale = 1; - - // Ensure we have enough data - if (reader.GetRemainingSize() < x*y*2) - throw DeadlyImportError("TER: ALTW chunk is too small"); - - if (x <= 1 || y <= 1) - throw DeadlyImportError("TER: Invalid terrain size"); - - // Allocate the output mesh - pScene->mMeshes = new aiMesh*[pScene->mNumMeshes = 1]; - aiMesh* m = pScene->mMeshes[0] = new aiMesh(); - - // We return quads - aiFace* f = m->mFaces = new aiFace[m->mNumFaces = (x-1)*(y-1)]; - aiVector3D* pv = m->mVertices = new aiVector3D[m->mNumVertices = m->mNumFaces*4]; - - aiVector3D *uv( NULL ); - float step_y( 0.0f ), step_x( 0.0f ); - if (configComputeUVs) { - uv = m->mTextureCoords[0] = new aiVector3D[m->mNumVertices]; - step_y = 1.f/y; - step_x = 1.f/x; - } - const int16_t* data = (const int16_t*)reader.GetPtr(); - - for (unsigned int yy = 0, t = 0; yy < y-1;++yy) { - for (unsigned int xx = 0; xx < x-1;++xx,++f) { - - // make verts - const float fy = (float)yy, fx = (float)xx; - unsigned tmp,tmp2; - *pv++ = aiVector3D(fx,fy, (float)data[(tmp2=x*yy) + xx] * hscale + bheight); - *pv++ = aiVector3D(fx,fy+1, (float)data[(tmp=x*(yy+1)) + xx] * hscale + bheight); - *pv++ = aiVector3D(fx+1,fy+1,(float)data[tmp + xx+1] * hscale + bheight); - *pv++ = aiVector3D(fx+1,fy, (float)data[tmp2 + xx+1] * hscale + bheight); - - // also make texture coordinates, if necessary - if (configComputeUVs) { - *uv++ = aiVector3D( step_x*xx, step_y*yy, 0.f ); - *uv++ = aiVector3D( step_x*xx, step_y*(yy+1), 0.f ); - *uv++ = aiVector3D( step_x*(xx+1), step_y*(yy+1), 0.f ); - *uv++ = aiVector3D( step_x*(xx+1), step_y*yy, 0.f ); - } - - // make indices - f->mIndices = new unsigned int[f->mNumIndices = 4]; - for (unsigned int i = 0; i < 4;++i) - f->mIndices[i] = t++; - } - } - - // Add the mesh to the root node - root->mMeshes = new unsigned int[root->mNumMeshes = 1]; - root->mMeshes[0] = 0; - } - - // Get to the next chunk (4 byte aligned) - unsigned dtt; - if ((dtt = reader.GetCurrentPos() & 0x3)) - reader.IncPtr(4-dtt); - } - - // Check whether we have a mesh now - if (pScene->mNumMeshes != 1) - throw DeadlyImportError("TER: Unable to load terrain"); - - // Set the AI_SCENE_FLAGS_TERRAIN bit - pScene->mFlags |= AI_SCENE_FLAGS_TERRAIN; + IOStream* file = pIOHandler->Open( pFile, "rb"); + + // Check whether we can read from the file + if( file == NULL) + throw DeadlyImportError( "Failed to open TERRAGEN TERRAIN file " + pFile + "."); + + // Construct a stream reader to read all data in the correct endianness + StreamReaderLE reader(file); + if(reader.GetRemainingSize() < 16) + throw DeadlyImportError( "TER: file is too small" ); + + // Check for the existence of the two magic strings 'TERRAGEN' and 'TERRAIN ' + if (::strncmp((const char*)reader.GetPtr(),AI_TERR_BASE_STRING,8)) + throw DeadlyImportError( "TER: Magic string \'TERRAGEN\' not found" ); + + if (::strncmp((const char*)reader.GetPtr()+8,AI_TERR_TERRAIN_STRING,8)) + throw DeadlyImportError( "TER: Magic string \'TERRAIN\' not found" ); + + unsigned int x = 0,y = 0,mode = 0; + float rad = 6370.f; + (void)rad; + + + aiNode* root = pScene->mRootNode = new aiNode(); + root->mName.Set("<TERRAGEN.TERRAIN>"); + + // Default scaling is 30 + root->mTransformation.a1 = root->mTransformation.b2 = root->mTransformation.c3 = 30.f; + + // Now read all chunks until we're finished or an EOF marker is encountered + reader.IncPtr(16); + while (reader.GetRemainingSize() >= 4) + { + const char* head = (const char*)reader.GetPtr(); + reader.IncPtr(4); + + // EOF, break in every case + if (!::strncmp(head,AI_TERR_EOF_STRING,4)) + break; + + // Number of x-data points + if (!::strncmp(head,AI_TERR_CHUNK_XPTS,4)) + { + x = (uint16_t)reader.GetI2(); + } + // Number of y-data points + else if (!::strncmp(head,AI_TERR_CHUNK_YPTS,4)) + { + y = (uint16_t)reader.GetI2(); + } + // Squared terrains width-1. + else if (!::strncmp(head,AI_TERR_CHUNK_SIZE,4)) + { + x = y = (uint16_t)reader.GetI2()+1; + } + // terrain scaling + else if (!::strncmp(head,AI_TERR_CHUNK_SCAL,4)) + { + root->mTransformation.a1 = reader.GetF4(); + root->mTransformation.b2 = reader.GetF4(); + root->mTransformation.c3 = reader.GetF4(); + } + // mapping == 1: earth radius + else if (!::strncmp(head,AI_TERR_CHUNK_CRAD,4)) + { + rad = reader.GetF4(); + } + // mapping mode + else if (!::strncmp(head,AI_TERR_CHUNK_CRVM,4)) + { + mode = reader.GetI1(); + if (0 != mode) + DefaultLogger::get()->error("TER: Unsupported mapping mode, a flat terrain is returned"); + } + // actual terrain data + else if (!::strncmp(head,AI_TERR_CHUNK_ALTW,4)) + { + float hscale = (float)reader.GetI2() / 65536; + float bheight = (float)reader.GetI2(); + + if (!hscale)hscale = 1; + + // Ensure we have enough data + if (reader.GetRemainingSize() < x*y*2) + throw DeadlyImportError("TER: ALTW chunk is too small"); + + if (x <= 1 || y <= 1) + throw DeadlyImportError("TER: Invalid terrain size"); + + // Allocate the output mesh + pScene->mMeshes = new aiMesh*[pScene->mNumMeshes = 1]; + aiMesh* m = pScene->mMeshes[0] = new aiMesh(); + + // We return quads + aiFace* f = m->mFaces = new aiFace[m->mNumFaces = (x-1)*(y-1)]; + aiVector3D* pv = m->mVertices = new aiVector3D[m->mNumVertices = m->mNumFaces*4]; + + aiVector3D *uv( NULL ); + float step_y( 0.0f ), step_x( 0.0f ); + if (configComputeUVs) { + uv = m->mTextureCoords[0] = new aiVector3D[m->mNumVertices]; + step_y = 1.f/y; + step_x = 1.f/x; + } + const int16_t* data = (const int16_t*)reader.GetPtr(); + + for (unsigned int yy = 0, t = 0; yy < y-1;++yy) { + for (unsigned int xx = 0; xx < x-1;++xx,++f) { + + // make verts + const float fy = (float)yy, fx = (float)xx; + unsigned tmp,tmp2; + *pv++ = aiVector3D(fx,fy, (float)data[(tmp2=x*yy) + xx] * hscale + bheight); + *pv++ = aiVector3D(fx,fy+1, (float)data[(tmp=x*(yy+1)) + xx] * hscale + bheight); + *pv++ = aiVector3D(fx+1,fy+1,(float)data[tmp + xx+1] * hscale + bheight); + *pv++ = aiVector3D(fx+1,fy, (float)data[tmp2 + xx+1] * hscale + bheight); + + // also make texture coordinates, if necessary + if (configComputeUVs) { + *uv++ = aiVector3D( step_x*xx, step_y*yy, 0.f ); + *uv++ = aiVector3D( step_x*xx, step_y*(yy+1), 0.f ); + *uv++ = aiVector3D( step_x*(xx+1), step_y*(yy+1), 0.f ); + *uv++ = aiVector3D( step_x*(xx+1), step_y*yy, 0.f ); + } + + // make indices + f->mIndices = new unsigned int[f->mNumIndices = 4]; + for (unsigned int i = 0; i < 4;++i) + f->mIndices[i] = t++; + } + } + + // Add the mesh to the root node + root->mMeshes = new unsigned int[root->mNumMeshes = 1]; + root->mMeshes[0] = 0; + } + + // Get to the next chunk (4 byte aligned) + unsigned dtt; + if ((dtt = reader.GetCurrentPos() & 0x3)) + reader.IncPtr(4-dtt); + } + + // Check whether we have a mesh now + if (pScene->mNumMeshes != 1) + throw DeadlyImportError("TER: Unable to load terrain"); + + // Set the AI_SCENE_FLAGS_TERRAIN bit + pScene->mFlags |= AI_SCENE_FLAGS_TERRAIN; } #endif // !! ASSIMP_BUILD_NO_TERRAGEN_IMPORTER |