2018-01-08 08:26:26 +00:00
|
|
|
/*
|
|
|
|
Open Asset Import Library (assimp)
|
|
|
|
----------------------------------------------------------------------
|
|
|
|
|
2021-02-28 11:17:54 +00:00
|
|
|
Copyright (c) 2006-2021, assimp team
|
2018-01-28 18:42:05 +00:00
|
|
|
|
2018-01-08 08:26:26 +00:00
|
|
|
|
|
|
|
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.
|
|
|
|
|
|
|
|
----------------------------------------------------------------------
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include "EmbedTexturesProcess.h"
|
2018-01-11 21:21:40 +00:00
|
|
|
#include <assimp/ParsingUtils.h>
|
2018-01-08 08:26:26 +00:00
|
|
|
#include "ProcessHelper.h"
|
|
|
|
|
|
|
|
#include <fstream>
|
|
|
|
|
|
|
|
using namespace Assimp;
|
|
|
|
|
|
|
|
EmbedTexturesProcess::EmbedTexturesProcess()
|
|
|
|
: BaseProcess() {
|
|
|
|
}
|
|
|
|
|
|
|
|
EmbedTexturesProcess::~EmbedTexturesProcess() {
|
|
|
|
}
|
|
|
|
|
|
|
|
bool EmbedTexturesProcess::IsActive(unsigned int pFlags) const {
|
|
|
|
return (pFlags & aiProcess_EmbedTextures) != 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
void EmbedTexturesProcess::SetupProperties(const Importer* pImp) {
|
|
|
|
mRootPath = pImp->GetPropertyString("sourceFilePath");
|
|
|
|
mRootPath = mRootPath.substr(0, mRootPath.find_last_of("\\/") + 1u);
|
|
|
|
}
|
|
|
|
|
|
|
|
void EmbedTexturesProcess::Execute(aiScene* pScene) {
|
|
|
|
if (pScene == nullptr || pScene->mRootNode == nullptr) return;
|
|
|
|
|
|
|
|
aiString path;
|
|
|
|
|
|
|
|
uint32_t embeddedTexturesCount = 0u;
|
|
|
|
|
|
|
|
for (auto matId = 0u; matId < pScene->mNumMaterials; ++matId) {
|
|
|
|
auto material = pScene->mMaterials[matId];
|
|
|
|
|
|
|
|
for (auto ttId = 1u; ttId < AI_TEXTURE_TYPE_MAX; ++ttId) {
|
|
|
|
auto tt = static_cast<aiTextureType>(ttId);
|
|
|
|
auto texturesCount = material->GetTextureCount(tt);
|
|
|
|
|
|
|
|
for (auto texId = 0u; texId < texturesCount; ++texId) {
|
|
|
|
material->GetTexture(tt, texId, &path);
|
|
|
|
if (path.data[0] == '*') continue; // Already embedded
|
|
|
|
|
|
|
|
// Indeed embed
|
|
|
|
if (addTexture(pScene, path.data)) {
|
|
|
|
auto embeddedTextureId = pScene->mNumTextures - 1u;
|
|
|
|
::ai_snprintf(path.data, 1024, "*%u", embeddedTextureId);
|
|
|
|
material->AddProperty(&path, AI_MATKEY_TEXTURE(tt, texId));
|
|
|
|
embeddedTexturesCount++;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-05-13 09:25:27 +00:00
|
|
|
ASSIMP_LOG_INFO("EmbedTexturesProcess finished. Embedded ", embeddedTexturesCount, " textures." );
|
2018-01-08 08:26:26 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
bool EmbedTexturesProcess::addTexture(aiScene* pScene, std::string path) const {
|
2018-03-19 17:22:27 +00:00
|
|
|
std::streampos imageSize = 0;
|
|
|
|
std::string imagePath = path;
|
2018-01-08 08:26:26 +00:00
|
|
|
|
|
|
|
// Test path directly
|
|
|
|
std::ifstream file(imagePath, std::ios::binary | std::ios::ate);
|
2018-03-19 17:22:27 +00:00
|
|
|
if ((imageSize = file.tellg()) == std::streampos(-1)) {
|
2021-05-13 09:25:27 +00:00
|
|
|
ASSIMP_LOG_WARN("EmbedTexturesProcess: Cannot find image: ", imagePath, ". Will try to find it in root folder.");
|
2018-01-08 08:26:26 +00:00
|
|
|
|
|
|
|
// Test path in root path
|
|
|
|
imagePath = mRootPath + path;
|
|
|
|
file.open(imagePath, std::ios::binary | std::ios::ate);
|
2018-03-19 17:22:27 +00:00
|
|
|
if ((imageSize = file.tellg()) == std::streampos(-1)) {
|
2018-01-08 08:26:26 +00:00
|
|
|
// Test path basename in root path
|
|
|
|
imagePath = mRootPath + path.substr(path.find_last_of("\\/") + 1u);
|
|
|
|
file.open(imagePath, std::ios::binary | std::ios::ate);
|
2018-03-19 17:22:27 +00:00
|
|
|
if ((imageSize = file.tellg()) == std::streampos(-1)) {
|
2021-05-13 09:25:27 +00:00
|
|
|
ASSIMP_LOG_ERROR("EmbedTexturesProcess: Unable to embed texture: ", path, ".");
|
2018-01-08 08:26:26 +00:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-09-21 14:07:09 +00:00
|
|
|
aiTexel* imageContent = new aiTexel[ 1ul + static_cast<unsigned long>( imageSize ) / sizeof(aiTexel)];
|
2018-01-08 08:26:26 +00:00
|
|
|
file.seekg(0, std::ios::beg);
|
|
|
|
file.read(reinterpret_cast<char*>(imageContent), imageSize);
|
|
|
|
|
|
|
|
// Enlarging the textures table
|
2018-12-31 13:47:54 +00:00
|
|
|
unsigned int textureId = pScene->mNumTextures++;
|
2018-01-08 08:26:26 +00:00
|
|
|
auto oldTextures = pScene->mTextures;
|
|
|
|
pScene->mTextures = new aiTexture*[pScene->mNumTextures];
|
2018-12-31 13:47:54 +00:00
|
|
|
::memmove(pScene->mTextures, oldTextures, sizeof(aiTexture*) * (pScene->mNumTextures - 1u));
|
2020-01-09 21:04:46 +00:00
|
|
|
delete [] oldTextures;
|
|
|
|
|
2018-01-08 08:26:26 +00:00
|
|
|
// Add the new texture
|
2018-12-31 13:47:54 +00:00
|
|
|
auto pTexture = new aiTexture;
|
2018-01-08 08:26:26 +00:00
|
|
|
pTexture->mHeight = 0; // Means that this is still compressed
|
2018-03-20 20:56:26 +00:00
|
|
|
pTexture->mWidth = static_cast<uint32_t>(imageSize);
|
2018-01-08 08:26:26 +00:00
|
|
|
pTexture->pcData = imageContent;
|
|
|
|
|
|
|
|
auto extension = path.substr(path.find_last_of('.') + 1u);
|
2021-04-03 08:25:03 +00:00
|
|
|
extension = ai_tolower(extension);
|
2018-12-31 13:47:54 +00:00
|
|
|
if (extension == "jpeg") {
|
|
|
|
extension = "jpg";
|
|
|
|
}
|
|
|
|
|
2018-12-31 11:16:20 +00:00
|
|
|
size_t len = extension.size();
|
2018-12-31 13:47:54 +00:00
|
|
|
if (len > HINTMAXTEXTURELEN -1 ) {
|
2018-12-31 11:16:20 +00:00
|
|
|
len = HINTMAXTEXTURELEN - 1;
|
|
|
|
}
|
2018-12-31 13:47:54 +00:00
|
|
|
::strncpy(pTexture->achFormatHint, extension.c_str(), len);
|
2018-01-08 08:26:26 +00:00
|
|
|
pScene->mTextures[textureId] = pTexture;
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|