X-Git-Url: https://git.lukelau.me/?p=opengl.git;a=blobdiff_plain;f=material.cpp;h=32bf9e0fef2d9b609265955a1c455cc64ef99fef;hp=df8ef8b1867b36ba9e8a80b8eab6e17a9009d393;hb=d0c631f46c6db417e013b1bc0edec24cb9c2824a;hpb=a03f2bc751bbd7f91cbc2f52d2f0f6e753d6f776

diff --git a/material.cpp b/material.cpp
index df8ef8b..32bf9e0 100644
--- a/material.cpp
+++ b/material.cpp
@@ -1,73 +1,75 @@
 #include "material.hpp"
 #include "image.hpp"
+#include <assimp/pbrmaterial.h>
 
-Material::Material(const aiMaterial &ai, GLuint progId): progId(progId) {
-	if (ai.GetTextureCount(aiTextureType_DIFFUSE) > 0) {
-		aiString path;
-		ai.GetTexture(aiTextureType_DIFFUSE, 0, &path);
-		diffuseMap = new Texture(std::string(path.C_Str()));
+Material::Material(const aiMaterial &ai, const aiScene &scene, GLuint progId): progId(progId) {
+	aiString name;
+	ai.Get(AI_MATKEY_NAME, name);	
+	if (name == aiString("default material")) {
+		abort();
 	}
 
-	if (ai.GetTextureCount(aiTextureType_SPECULAR) > 0) {
 	aiString path;
-		ai.GetTexture(aiTextureType_SPECULAR, 0, &path);
-		specularMap = new Texture(std::string(path.C_Str()));
-	}
+	ai.GetTexture(aiTextureType_DIFFUSE, 1, &path);
+	albedo = new Texture(path, scene);
+	path = "";
 
-	if (ai.GetTextureCount(aiTextureType_NORMALS) > 0) {
-		aiString path;
 	ai.GetTexture(aiTextureType_NORMALS, 0, &path);
-		normalMap = new Texture(std::string(path.C_Str()));
-	}
-	ai.Get(AI_MATKEY_SHININESS, shininess);
-	ai.Get(AI_MATKEY_REFLECTIVITY, reflectivity);
-	ai.Get(AI_MATKEY_REFRACTI, refractiveIndex);
-	if (ai.Get(AI_MATKEY_OPACITY, opacity) == aiReturn_FAILURE)
-		opacity = 1.f;
+	normal = new Texture(path, scene);
+	path = "";
+
+	ai.GetTexture(AI_MATKEY_GLTF_PBRMETALLICROUGHNESS_METALLICROUGHNESS_TEXTURE, &path);
+	metallicRoughness = new Texture(path, scene);
+	path = "";
 	
-	ai.Get(AI_MATKEY_COLOR_AMBIENT, ambient);
-	ai.Get(AI_MATKEY_COLOR_DIFFUSE, diffuse);
-	ai.Get(AI_MATKEY_COLOR_SPECULAR, specular);
+	ai.GetTexture(aiTextureType_LIGHTMAP, 0, &path);
+	if (path == aiString("")) {
+		fprintf(stderr, "Material %s does not have an AO map", name.C_Str());
+		abort();
+	}
+	ambientOcclusion = new Texture(path, scene);
 }
 
-Material::Texture::Texture(const std::string &fileName) {
+Material::Texture::Texture(const aiString fileName, const aiScene &scene) {
 	glGenTextures(1, &texId);
 	glBindTexture(GL_TEXTURE_2D, texId);
-	Image img("models/" + fileName);
-	glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA, img.width(), img.height(), 0, GL_RGBA, GL_UNSIGNED_BYTE, img.data());
+	
+	std::string path;
+	unsigned char *data;
+	if (fileName.data[0] == '*') {
+		// embedded
+		int embIdx = atoi(&fileName.data[1]);
+		aiTexture *texture = scene.mTextures[embIdx];
+		if (texture->mHeight == 0) {
+			Image img((unsigned char*)texture->pcData, texture->mWidth, texture->achFormatHint);
+			glTexImage2D(GL_TEXTURE_2D, 0, img.internalFormat(), img.width(), img.height(), 0, img.format(), img.type(), img.data());
+		} else {
+			fprintf(stderr, "TODO: handle uncompressed embedded textures");
+			abort();
+		}
+	} else {
+		// not embedded
+		Image img("models/" + std::string(fileName.C_Str()));
+		glTexImage2D(GL_TEXTURE_2D, 0, img.internalFormat(), img.width(), img.height(), 0, img.format(), img.type(), img.data());
+	}
+
 	glGenerateMipmap(GL_TEXTURE_2D);
 }
 
 void Material::bind() const {
-	/* glUniform3f(glGetUniformLocation(progId, "albedo"), 0.5f, 0.f, 0.f); */
-	/* glUniform1f(glGetUniformLocation(progId, "metallic"), 0.2f); */
-	/* glUniform1f(glGetUniformLocation(progId, "roughness"), 0.3f); */
-	/* glUniform1f(glGetUniformLocation(progId, "ao"), 1.f); */
-
-	/* glUniform4f(glGetUniformLocation(progId, "material.ambient"), ambient.r, ambient.g, ambient.b, ambient.a); */
-	/* glUniform4f(glGetUniformLocation(progId, "material.diffuse"), diffuse.r, diffuse.g, diffuse.b, diffuse.a); */
-	/* glUniform4f(glGetUniformLocation(progId, "material.specular"), specular.r, specular.g, specular.b, specular.a); */
-	/* glUniform1f(glGetUniformLocation(progId, "material.shininess"), shininess); */
-	/* glUniform1f(glGetUniformLocation(progId, "material.reflectivity"), reflectivity); */
-	/* glUniform1f(glGetUniformLocation(progId, "material.refractiveIndex"), refractiveIndex); */
-	/* glUniform1f(glGetUniformLocation(progId, "material.opacity"), opacity); */
-
-	/* glUniform1i(glGetUniformLocation(progId, "material.hasTexture"), diffuseMap != nullptr); */
-	/* glUniform1i(glGetUniformLocation(progId, "material.hasSpecularMap"), specularMap != nullptr); */
-	/* glUniform1i(glGetUniformLocation(progId, "material.hasNormalMap"), normalMap != nullptr); */
-	/* glUniform1i(glGetUniformLocation(progId, "material.hasTexture"), diffuseMap != nullptr); */
-	/* glUniform1i(glGetUniformLocation(progId, "material.hasSpecularMap"), specularMap != nullptr); */
-	/* glUniform1i(glGetUniformLocation(progId, "material.hasNormalMap"), normalMap != nullptr); */
+	glUniform1i(glGetUniformLocation(progId, "albedoMap"), 0);
+	glActiveTexture(GL_TEXTURE0);
+	glBindTexture(GL_TEXTURE_2D, albedo->texId);
 
-	/* glUniform1i(glGetUniformLocation(progId, "material.diffuseMap"), 2); */
-	/* glActiveTexture(GL_TEXTURE2); */
-	/* if (diffuseMap) glBindTexture(GL_TEXTURE_2D, diffuseMap->texId); */
+	glUniform1i(glGetUniformLocation(progId, "normalMap"), 1);
+	glActiveTexture(GL_TEXTURE1);
+	glBindTexture(GL_TEXTURE_2D, normal->texId);
 
-	/* glUniform1i(glGetUniformLocation(progId, "material.specularMap"), 3); */
-	/* glActiveTexture(GL_TEXTURE3); */
-	/* if (specularMap) glBindTexture(GL_TEXTURE_2D, specularMap->texId); */
+	glUniform1i(glGetUniformLocation(progId, "metallicRoughnessMap"), 2);
+	glActiveTexture(GL_TEXTURE2);
+	glBindTexture(GL_TEXTURE_2D, metallicRoughness->texId);
 
-	/* glUniform1i(glGetUniformLocation(progId, "material.normalMap"), 4); */
-	/* glActiveTexture(GL_TEXTURE4); */
-	/* if (normalMap) glBindTexture(GL_TEXTURE_2D, normalMap->texId); */
+	glUniform1i(glGetUniformLocation(progId, "aoMap"), 3);
+	glActiveTexture(GL_TEXTURE3);
+	glBindTexture(GL_TEXTURE_2D, ambientOcclusion->texId);
 }