#include #include #include #include #include "model_opengl_program.h" #include "dds_file.h" static const QString &loadShaderSource(const QString &name) { static std::map s_shaderSources; auto findShader = s_shaderSources.find(name); if (findShader != s_shaderSources.end()) { return findShader->second; } QFile file(name); file.open(QFile::ReadOnly | QFile::Text); QTextStream stream(&file); auto insertResult = s_shaderSources.insert({name, stream.readAll()}); return insertResult.first->second; } void ModelOpenGLProgram::addShaderFromResource(QOpenGLShader::ShaderType type, const char *resourceName) { if (!addShaderFromSourceCode(type, loadShaderSource(resourceName))) dust3dDebug << "Failed to addShaderFromResource, resource:" << resourceName << ", " << log().toStdString(); } bool ModelOpenGLProgram::isCoreProfile() const { return m_isCoreProfile; } void ModelOpenGLProgram::updateTextureImage(std::unique_ptr image) { QMutexLocker lock(&m_imageMutex); m_textureImage = std::move(image); m_imageIsDirty = true; } void ModelOpenGLProgram::updateNormalMapImage(std::unique_ptr image) { QMutexLocker lock(&m_imageMutex); m_normalMapImage = std::move(image); m_imageIsDirty = true; } void ModelOpenGLProgram::updateMetalnessRoughnessAmbientOcclusionMapImage(std::unique_ptr image, bool hasMetalnessMap, bool hasRoughnessMap, bool hasAmbientOcclusionMap) { QMutexLocker lock(&m_imageMutex); m_metalnessRoughnessAmbientOcclusionMapImage = std::move(image); m_hasMetalnessMap = hasMetalnessMap; m_hasRoughnessMap = hasRoughnessMap; m_hasAmbientOcclusionMap = hasAmbientOcclusionMap; m_imageIsDirty = true; } void ModelOpenGLProgram::activeAndBindTexture(int location, QOpenGLTexture *texture) { if (0 == texture->textureId()) { dust3dDebug << "Expected texture with a bound id"; return; } texture->bind(location); } void ModelOpenGLProgram::load(bool isCoreProfile) { if (m_isLoaded) return; m_isCoreProfile = isCoreProfile; if (m_isCoreProfile) { addShaderFromResource(QOpenGLShader::Vertex, ":/shaders/model_core.vert"); addShaderFromResource(QOpenGLShader::Fragment, ":/shaders/model_core.frag"); } else { addShaderFromResource(QOpenGLShader::Vertex, ":/shaders/model.vert"); addShaderFromResource(QOpenGLShader::Fragment, ":/shaders/model.frag"); } bindAttributeLocation("vertex", 0); bindAttributeLocation("normal", 1); bindAttributeLocation("color", 2); bindAttributeLocation("texCoord", 3); bindAttributeLocation("metalness", 4); bindAttributeLocation("roughness", 5); bindAttributeLocation("tangent", 6); bindAttributeLocation("alpha", 7); link(); m_isLoaded = true; } void ModelOpenGLProgram::bindMaps() { int bindLocation = 1; // Bind environment maps if (m_isCoreProfile) { if (!m_environmentIrradianceMap) { DdsFileReader irradianceFile(":/resources/cedar_bridge_irradiance.dds"); m_environmentIrradianceMap.reset(irradianceFile.createOpenGLTexture()); } if (!m_environmentSpecularMap) { DdsFileReader irradianceFile(":/resources/cedar_bridge_specular.dds"); m_environmentSpecularMap.reset(irradianceFile.createOpenGLTexture()); } bindLocation++; activeAndBindTexture(bindLocation, m_environmentIrradianceMap.get()); setUniformValue(getUniformLocationByName("environmentIrradianceMapId"), bindLocation); bindLocation++; activeAndBindTexture(bindLocation, m_environmentSpecularMap.get()); setUniformValue(getUniformLocationByName("environmentSpecularMapId"), bindLocation); } else { if (!m_environmentIrradianceMaps) { DdsFileReader irradianceFile(":/resources/cedar_bridge_irradiance.dds"); m_environmentIrradianceMaps = std::move(irradianceFile.createOpenGLTextures()); } if (!m_environmentSpecularMaps) { DdsFileReader irradianceFile(":/resources/cedar_bridge_specular.dds"); m_environmentSpecularMaps = std::move(irradianceFile.createOpenGLTextures()); } auto oldBindLocationStart = bindLocation; for (size_t i = 0; i < m_environmentIrradianceMaps->size(); ++i) activeAndBindTexture(bindLocation++, m_environmentIrradianceMaps->at(i).get()); for (size_t i = 0; i < m_environmentSpecularMaps->size(); ++i) activeAndBindTexture(bindLocation++, m_environmentSpecularMaps->at(i).get()); bindLocation = oldBindLocationStart; for (size_t i = 0; i < m_environmentIrradianceMaps->size(); ++i) setUniformValue(getUniformLocationByName("environmentIrradianceMapId[" + std::to_string(i) + "]"), bindLocation++); for (size_t i = 0; i < m_environmentSpecularMaps->size(); ++i) setUniformValue(getUniformLocationByName("environmentSpecularMapId[" + std::to_string(i) + "]"), bindLocation++); } // Bind texture, normal map, and other maps if (m_imageIsDirty) { QMutexLocker lock(&m_imageMutex); if (m_imageIsDirty) { m_imageIsDirty = false; if (m_texture) { m_texture->destroy(); m_texture.reset(); } if (m_textureImage) { m_texture = std::make_unique(*m_textureImage); } if (m_normalMap) { m_normalMap->destroy(); m_normalMap.reset(); } if (m_normalMapImage) { m_normalMap = std::make_unique(*m_normalMapImage); } if (m_metalnessRoughnessAmbientOcclusionMap) { m_metalnessRoughnessAmbientOcclusionMap->destroy(); m_metalnessRoughnessAmbientOcclusionMap.reset(); } if (m_metalnessRoughnessAmbientOcclusionMapImage) { m_metalnessRoughnessAmbientOcclusionMap = std::make_unique(*m_metalnessRoughnessAmbientOcclusionMapImage); } m_metalnessMapEnabled = m_hasMetalnessMap; m_roughnessMapEnabled = m_hasRoughnessMap; m_ambientOcclusionMapEnabled = m_hasAmbientOcclusionMap; } } bindLocation++; if (m_texture) { activeAndBindTexture(bindLocation, m_texture.get()); setUniformValue(getUniformLocationByName("textureId"), bindLocation); setUniformValue(getUniformLocationByName("textureEnabled"), (int)1); } else { setUniformValue(getUniformLocationByName("textureId"), (int)0); setUniformValue(getUniformLocationByName("textureEnabled"), (int)0); } bindLocation++; if (m_normalMap) { activeAndBindTexture(bindLocation, m_normalMap.get()); setUniformValue(getUniformLocationByName("normalMapId"), bindLocation); setUniformValue(getUniformLocationByName("normalMapEnabled"), (int)1); } else { setUniformValue(getUniformLocationByName("normalMapId"), (int)0); setUniformValue(getUniformLocationByName("normalMapEnabled"), (int)0); } bindLocation++; if (m_metalnessRoughnessAmbientOcclusionMap) { activeAndBindTexture(bindLocation, m_metalnessRoughnessAmbientOcclusionMap.get()); setUniformValue(getUniformLocationByName("metalnessRoughnessAoMapId"), bindLocation); setUniformValue(getUniformLocationByName("metalnessMapEnabled"), (int)m_metalnessMapEnabled); setUniformValue(getUniformLocationByName("roughnessMapEnabled"), (int)m_roughnessMapEnabled); setUniformValue(getUniformLocationByName("aoMapEnabled"), (int)m_ambientOcclusionMapEnabled); } else { setUniformValue(getUniformLocationByName("metalnessRoughnessAoMapId"), (int)0); setUniformValue(getUniformLocationByName("metalnessMapEnabled"), (int)0); setUniformValue(getUniformLocationByName("roughnessMapEnabled"), (int)0); setUniformValue(getUniformLocationByName("aoMapEnabled"), (int)0); } } void ModelOpenGLProgram::releaseMaps() { if (m_texture) m_texture->release(); if (m_normalMap) m_normalMap->release(); if (m_metalnessRoughnessAmbientOcclusionMap) m_metalnessRoughnessAmbientOcclusionMap->release(); if (m_environmentIrradianceMap) m_environmentIrradianceMap->release(); if (m_environmentSpecularMap) m_environmentSpecularMap->release(); if (m_environmentIrradianceMaps) { for (size_t i = 0; i < m_environmentIrradianceMaps->size(); ++i) m_environmentIrradianceMaps->at(i)->release(); } if (m_environmentSpecularMaps) { for (size_t i = 0; i < m_environmentSpecularMaps->size(); ++i) m_environmentSpecularMaps->at(i)->release(); } } int ModelOpenGLProgram::getUniformLocationByName(const std::string &name) { auto findLocation = m_uniformLocationMap.find(name); if (findLocation != m_uniformLocationMap.end()) return findLocation->second; int location = uniformLocation(name.c_str()); m_uniformLocationMap.insert({name, location}); return location; }