material.cpp

   1 #include "material.hpp"
   2 #include <iostream>
   3 #include <fstream>
   4 #include <CoreGraphics/CoreGraphics.h>
   5
   6 Material::Material(const aiMaterial &ai, GLuint progId): progId(progId) {
   7         if (ai.GetTextureCount(aiTextureType_DIFFUSE) > 0) {
   8                 aiString path;
   9                 ai.GetTexture(aiTextureType_DIFFUSE, 0, &path);
  10                 diffuseMap = new Texture(std::string(path.C_Str()));
  11         }
  12         if (ai.GetTextureCount(aiTextureType_SPECULAR) > 0) {
  13                 aiString path;
  14                 ai.GetTexture(aiTextureType_SPECULAR, 0, &path);
  15                 specularMap = new Texture(std::string(path.C_Str()));
  16         }
  17         if (ai.GetTextureCount(aiTextureType_NORMALS) > 0) {
  18                 aiString path;
  19                 ai.GetTexture(aiTextureType_NORMALS, 0, &path);
  20                 normalMap = new Texture(std::string(path.C_Str()));
  21         }
  22         ai.Get(AI_MATKEY_SHININESS, shininess);
  23         ai.Get(AI_MATKEY_COLOR_AMBIENT, ambient);
  24         ai.Get(AI_MATKEY_COLOR_DIFFUSE, diffuse);
  25         ai.Get(AI_MATKEY_COLOR_SPECULAR, specular);
  26 }
  27
  28 Material::Texture::Texture(const std::string &fileName) {
  29         auto path = "models/" + fileName;
  30         auto provider = CGDataProviderCreateWithFilename(path.c_str());
  31         std::ifstream file(path);
  32         long magic;
  33         file.read((char*)&magic, 8);
  34         file.close();
  35
  36         CGImageRef ref;
  37
  38         if (magic == 0x0a1a0a0d474e5089) // png magic number
  39                 ref = CGImageCreateWithPNGDataProvider(provider, nullptr, false, kCGRenderingIntentDefault);
  40         else
  41                 ref = CGImageCreateWithJPEGDataProvider(provider, nullptr, false, kCGRenderingIntentDefault);
  42
  43         auto dataRef = CGDataProviderCopyData(CGImageGetDataProvider(ref));
  44         auto img = (unsigned char*) CFDataGetBytePtr(dataRef);
  45
  46         glGenTextures(1, &texId);
  47         glBindTexture(GL_TEXTURE_2D, texId);
  48         auto width = CGImageGetWidth(ref), height = CGImageGetHeight(ref);
  49         glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA, width, height, 0, GL_RGBA, GL_UNSIGNED_BYTE, img);
  50         glGenerateMipmap(GL_TEXTURE_2D);
  51
  52         CGImageRelease(ref);
  53         CFRelease(dataRef);
  54 }
  55
  56 void Material::bind() const {
  57         glUniform4f(glGetUniformLocation(progId, "material.ambient"), ambient.r, ambient.g, ambient.b, ambient.a);
  58         glUniform4f(glGetUniformLocation(progId, "material.diffuse"), diffuse.r, diffuse.g, diffuse.b, diffuse.a);
  59         glUniform4f(glGetUniformLocation(progId, "material.specular"), specular.r, specular.g, specular.b, specular.a);
  60         glUniform1f(glGetUniformLocation(progId, "material.shininess"), shininess);
  61         glUniform1i(glGetUniformLocation(progId, "material.hasTexture"), diffuseMap != nullptr);
  62         glUniform1i(glGetUniformLocation(progId, "material.hasSpecularMap"), specularMap != nullptr);
  63         glUniform1i(glGetUniformLocation(progId, "material.hasNormalMap"), normalMap != nullptr);
  64
  65         if (diffuseMap) {
  66                 glUniform1i(glGetUniformLocation(progId, "material.diffuseMap"), 0);
  67                 glActiveTexture(GL_TEXTURE0);
  68                 glBindTexture(GL_TEXTURE_2D, diffuseMap->texId);
  69         }
  70         if (specularMap) {
  71                 glUniform1i(glGetUniformLocation(progId, "material.specularMap"), 1);
  72                 glActiveTexture(GL_TEXTURE1);
  73                 glBindTexture(GL_TEXTURE_2D, specularMap->texId);
  74         }
  75         if (normalMap) {
  76                 glUniform1i(glGetUniformLocation(progId, "material.normalMap"), 2);
  77                 glActiveTexture(GL_TEXTURE2);
  78                 glBindTexture(GL_TEXTURE_2D, normalMap->texId);
  79         }
  80 }