diff options
| author | Marijn Tamis <[email protected]> | 2018-05-03 18:22:48 +0200 |
|---|---|---|
| committer | Marijn Tamis <[email protected]> | 2018-05-03 18:22:48 +0200 |
| commit | ca32c59a58d37c1822e185a2d5f3d0d3e8943593 (patch) | |
| tree | b06b9eec03f34344ef8fc31aa147b2714d3962ee /NvCloth/samples/external/assimp-4.1.0/code/TerragenLoader.cpp | |
| parent | Forced rename of platform folders in cmake dir. Git didn't pick this up before. (diff) | |
| download | nvcloth-ca32c59a58d37c1822e185a2d5f3d0d3e8943593.tar.xz nvcloth-ca32c59a58d37c1822e185a2d5f3d0d3e8943593.zip | |
NvCloth 1.1.4 Release. (24070740)
Diffstat (limited to 'NvCloth/samples/external/assimp-4.1.0/code/TerragenLoader.cpp')
| -rw-r--r-- | NvCloth/samples/external/assimp-4.1.0/code/TerragenLoader.cpp | 273 |
1 files changed, 273 insertions, 0 deletions
diff --git a/NvCloth/samples/external/assimp-4.1.0/code/TerragenLoader.cpp b/NvCloth/samples/external/assimp-4.1.0/code/TerragenLoader.cpp new file mode 100644 index 0000000..0126229 --- /dev/null +++ b/NvCloth/samples/external/assimp-4.1.0/code/TerragenLoader.cpp @@ -0,0 +1,273 @@ +/* +--------------------------------------------------------------------------- +Open Asset Import Library (assimp) +--------------------------------------------------------------------------- + +Copyright (c) 2006-2017, 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 +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 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 Terragen importer class */ + + + +#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> +#include <assimp/importerdesc.h> + +using namespace Assimp; + +static const aiImporterDesc desc = { + "Terragen Heightmap Importer", + "", + "", + "http://www.planetside.co.uk/", + aiImporterFlags_SupportBinaryFlavour, + 0, + 0, + 0, + 0, + "ter" +}; + +// ------------------------------------------------------------------------------------------------ +// Constructor to be privately used by Importer +TerragenImporter::TerragenImporter() +: configComputeUVs (false) +{} + +// ------------------------------------------------------------------------------------------------ +// Destructor, private as well +TerragenImporter::~TerragenImporter() +{} + +// ------------------------------------------------------------------------------------------------ +// 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; +} + +// ------------------------------------------------------------------------------------------------ +// Build a string of all file extensions supported +const aiImporterDesc* TerragenImporter::GetInfo () const +{ + 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) ); +} + +// ------------------------------------------------------------------------------------------------ +// 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 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; + + 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)) + { + 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 |