Add BOM skip and fix mtl parsing (#6253)
* Add BOM skip and fix mtl parsing * Remove old code * Fix #5635 * Add all attributes to ObjFileData --------- Co-authored-by: Kim Kulling <kimkulling@users.noreply.github.com>
This commit is contained in:
@@ -235,24 +235,64 @@ void ObjFileMtlImporter::load() {
|
||||
++m_DataIt;
|
||||
if (m_pModel->mCurrentMaterial != nullptr)
|
||||
getFloatValue(m_pModel->mCurrentMaterial->clearcoat_roughness);
|
||||
} else {
|
||||
} else if (*m_DataIt == 't') {
|
||||
++m_DataIt;
|
||||
if (m_pModel->mCurrentMaterial != nullptr)
|
||||
getFloatValue(m_pModel->mCurrentMaterial->clearcoat_thickness);
|
||||
} else {
|
||||
if (m_pModel->mCurrentMaterial != nullptr)
|
||||
getFloatValue(m_pModel->mCurrentMaterial->clearcoat);
|
||||
}
|
||||
break;
|
||||
}
|
||||
m_DataIt = skipLine<DataArrayIt>(m_DataIt, m_DataItEnd, m_uiLine);
|
||||
}
|
||||
break;
|
||||
|
||||
case 'm': // Texture or metallic
|
||||
{
|
||||
// Save start of token (after 'm')
|
||||
auto tokenStart = m_DataIt; // points to 'm'
|
||||
auto tokenEnd = getNextToken(m_DataIt, m_DataItEnd); // move iterator to end of token
|
||||
|
||||
std::string keyword(tokenStart, tokenEnd);
|
||||
m_DataIt = tokenEnd; // advance iterator
|
||||
|
||||
if (keyword.compare(0, 3, "map") == 0) {
|
||||
// starts with "map", treat as texture map
|
||||
m_DataIt = tokenStart;
|
||||
getTexture();
|
||||
} else if (keyword == "metallic" || keyword == "metal" || keyword == "metalness") {
|
||||
// parse metallic float value instead of texture
|
||||
getFloatIfMaterialValid(&ObjFile::Material::metallic);
|
||||
}
|
||||
|
||||
m_DataIt = skipLine<DataArrayIt>(m_DataIt, m_DataItEnd, m_uiLine);
|
||||
} break;
|
||||
|
||||
case 'm': // Texture
|
||||
case 'b': // quick'n'dirty - for 'bump' sections
|
||||
case 'r': // quick'n'dirty - for 'refl' sections
|
||||
{
|
||||
getTexture();
|
||||
m_DataIt = skipLine<DataArrayIt>(m_DataIt, m_DataItEnd, m_uiLine);
|
||||
} break;
|
||||
|
||||
case 'r': // refl (map) or roughness (float)
|
||||
{
|
||||
auto tokenStart = m_DataIt; // points to 'r'
|
||||
auto tokenEnd = getNextToken(m_DataIt, m_DataItEnd);
|
||||
std::string keyword(tokenStart, tokenEnd);
|
||||
m_DataIt = tokenEnd;
|
||||
|
||||
if (keyword == "roughness" || keyword == "rough") {
|
||||
getFloatIfMaterialValid(&ObjFile::Material::roughness);
|
||||
} else if (keyword == "refl" || keyword == "reflection") {
|
||||
m_DataIt = tokenStart;
|
||||
getTexture();
|
||||
}
|
||||
|
||||
m_DataIt = skipLine<DataArrayIt>(m_DataIt, m_DataItEnd, m_uiLine);
|
||||
} break;
|
||||
|
||||
case 'i': // Illumination model
|
||||
{
|
||||
m_DataIt = getNextToken<DataArrayIt>(m_DataIt, m_DataItEnd);
|
||||
@@ -261,11 +301,60 @@ void ObjFileMtlImporter::load() {
|
||||
m_DataIt = skipLine<DataArrayIt>(m_DataIt, m_DataItEnd, m_uiLine);
|
||||
} break;
|
||||
|
||||
case 'a': // Anisotropy
|
||||
{
|
||||
++m_DataIt;
|
||||
if (m_pModel->mCurrentMaterial != nullptr)
|
||||
getFloatValue(m_pModel->mCurrentMaterial->anisotropy);
|
||||
case 'a': {
|
||||
auto tokenStart = m_DataIt;
|
||||
auto tokenEnd = getNextToken(m_DataIt, m_DataItEnd);
|
||||
std::string keyword(tokenStart, tokenEnd);
|
||||
m_DataIt = tokenEnd;
|
||||
|
||||
if (keyword == "aniso" || keyword == "anisotropy") {
|
||||
getFloatIfMaterialValid(&ObjFile::Material::anisotropy);
|
||||
} else if (keyword == "ao") {
|
||||
getFloatIfMaterialValid(&ObjFile::Material::ambient_occlusion);
|
||||
} else if (keyword == "anisor" || ai_stdStrToLower(keyword) == "anisotropicrotation") {
|
||||
getFloatIfMaterialValid(&ObjFile::Material::anisotropy_rotation);
|
||||
} else {
|
||||
ASSIMP_LOG_WARN("Unhandled keyword: ", keyword );
|
||||
}
|
||||
|
||||
m_DataIt = skipLine<DataArrayIt>(m_DataIt, m_DataItEnd, m_uiLine);
|
||||
} break;
|
||||
|
||||
case 's': {
|
||||
auto tokenStart = m_DataIt;
|
||||
auto tokenEnd = getNextToken(m_DataIt, m_DataItEnd);
|
||||
std::string keyword(tokenStart, tokenEnd);
|
||||
m_DataIt = tokenEnd;
|
||||
|
||||
if (keyword == "subsurface" || keyword == "scattering") {
|
||||
getFloatIfMaterialValid(&ObjFile::Material::subsurface_scattering);
|
||||
} else if (ai_stdStrToLower(keyword) == "speculartint") {
|
||||
getFloatIfMaterialValid(&ObjFile::Material::specular_tint);
|
||||
} else if (keyword == "sheen") {
|
||||
getFloatIfMaterialValid(&ObjFile::Material::sheen_grazing);
|
||||
} else if (ai_stdStrToLower(keyword) == "sheentint") {
|
||||
getFloatIfMaterialValid(&ObjFile::Material::sheen_tint);
|
||||
} else {
|
||||
ASSIMP_LOG_WARN("Unhandled keyword: ", keyword );
|
||||
}
|
||||
|
||||
m_DataIt = skipLine<DataArrayIt>(m_DataIt, m_DataItEnd, m_uiLine);
|
||||
} break;
|
||||
|
||||
case 'c': {
|
||||
auto tokenStart = m_DataIt;
|
||||
auto tokenEnd = getNextToken(m_DataIt, m_DataItEnd);
|
||||
std::string keyword(tokenStart, tokenEnd);
|
||||
m_DataIt = tokenEnd;
|
||||
|
||||
if (ai_stdStrToLower(keyword) == "clearcoat") {
|
||||
getFloatIfMaterialValid(&ObjFile::Material::clearcoat);
|
||||
} else if (ai_stdStrToLower(keyword) == "clearcoatgloss") {
|
||||
getFloatIfMaterialValid(&ObjFile::Material::clearcoat_gloss);
|
||||
} else {
|
||||
ASSIMP_LOG_WARN("Unhandled keyword: ", keyword );
|
||||
}
|
||||
|
||||
m_DataIt = skipLine<DataArrayIt>(m_DataIt, m_DataItEnd, m_uiLine);
|
||||
} break;
|
||||
|
||||
@@ -332,6 +421,23 @@ void ObjFileMtlImporter::getFloatValue(Maybe<ai_real> &value) {
|
||||
value = Maybe<ai_real>();
|
||||
}
|
||||
|
||||
// -------------------------------------------------------------------
|
||||
// Writes a loaded single float value if material not null
|
||||
void ObjFileMtlImporter::getFloatIfMaterialValid(ai_real ObjFile::Material::*member) {
|
||||
if (m_pModel != nullptr && m_pModel->mCurrentMaterial != nullptr) {
|
||||
// This will call getFloatValue(ai_real&)
|
||||
getFloatValue(m_pModel->mCurrentMaterial->*member);
|
||||
}
|
||||
}
|
||||
|
||||
// -------------------------------------------------------------------
|
||||
void ObjFileMtlImporter::getFloatIfMaterialValid(Maybe<ai_real> ObjFile::Material::*member) {
|
||||
// It can directly access `m_pModel` because it's part of the class
|
||||
if (m_pModel != nullptr && m_pModel->mCurrentMaterial != nullptr) {
|
||||
getFloatValue(m_pModel->mCurrentMaterial->*member);
|
||||
}
|
||||
}
|
||||
|
||||
// -------------------------------------------------------------------
|
||||
// Creates a material from loaded data.
|
||||
void ObjFileMtlImporter::createMaterial() {
|
||||
|
||||
Reference in New Issue
Block a user