Fix metal roughness texture on streaked metal material

[opengl.git] / main.cpp

diff --git a/main.cpp b/main.cpp
index fcc7a1eb90c033f6c1b3a5edec943ff759e16ac8..0b7cb229dd745e09d405234bfe89b5f107eab389 100644 (file)
--- a/main.cpp
+++ b/main.cpp
@@ -5,7 +5,6 @@
 #include <vector>
 #ifdef __APPLE__
 #include <GL/glew.h>
-#include "cocoa.h"
 #else
 #include <OpenGL/glew.h>
 #endif
@@ -43,14 +42,15 @@ struct Light {
 };
 
 std::vector<Light> lights = {
-       { glm::vec3(0, 0, 3), glm::vec3(1) },
-       { glm::vec3(0, 3, 0), glm::vec3(1) },
-       { glm::vec3(3, 0, 0), glm::vec3(1) },
-       { glm::vec3(3, 0, 0), glm::vec3(1) }
+       { glm::vec3(5, 2, -5), glm::vec3(1) },
+       { glm::vec3(0, 2, -5), glm::vec3(1) },
+       { glm::vec3(-5, 2, -5), glm::vec3(1) },
 };
 
 int activeLight = 0;
 
+bool discoLights = false;
+
 int windowWidth = 800, windowHeight = 600;
 
 float aspect() {
@@ -108,25 +108,51 @@ void drawLight(Light &light) {
 void display() {
        glClearColor(0.5, 0.5, 0.5, 1);
        glClear(GL_COLOR_BUFFER_BIT|GL_DEPTH_BUFFER_BIT);
+       glViewport(0, 0, windowWidth * 2, windowHeight * 2);
 
        float d = (float)glutGet(GLUT_ELAPSED_TIME) * 0.001f;
 
        glUseProgram(pbrProg->progId);
        setProjectionAndViewUniforms(pbrProg->progId);
 
-       glm::vec3 lightPositions[4], lightColors[4];
-       for (int i = 0; i < 4; i++) {
+       glm::vec3 lightPositions[6], lightColors[6];
+       for (int i = 0; i < 3; i++) {
                lightPositions[i] = lights[i].pos;
                lightColors[i] = lights[i].color;
        }
 
-       glUniform3fv(glGetUniformLocation(pbrProg->progId, "lightPositions"), 4, glm::value_ptr(lightPositions[0]));
-       glUniform3fv(glGetUniformLocation(pbrProg->progId, "lightColors"), 4, glm::value_ptr(lightColors[0]));
+       for (int i = 3; i < 6; i++) {
+               if (discoLights) {
+                       auto m = glm::translate(glm::mat4(1.f), glm::vec3(-2.5, 0, 0));
+                       m = glm::rotate(m, glm::radians(d * 100 + i * 30), glm::vec3(1, 0, 0));
+                       m = glm::rotate(m, glm::radians(d * 100 + i * 30), glm::vec3(0, 1, 0));
+                       m = glm::rotate(m, glm::radians(d * 100 + i * 30), glm::vec3(0, 0, 1));
+                       lightPositions[i] = glm::vec3(m * glm::vec4(5, 0, 0, 1));
+                       lightColors[i] = glm::vec3(0.2);
+                       if (i == 3) lightColors[i].x = sin(d);
+                       if (i == 4) lightColors[i].y = cos(d * 3);
+                       if (i == 5) lightColors[i].z = cos(d);
+               } else {
+                       lightPositions[i] = glm::vec3(0);
+                       lightColors[i] = glm::vec3(0);
+               }
+       }
+       
+       glUniform3fv(glGetUniformLocation(pbrProg->progId, "lightPositions"), 6, glm::value_ptr(lightPositions[0]));
+       glUniform3fv(glGetUniformLocation(pbrProg->progId, "lightColors"), 6, glm::value_ptr(lightColors[0]));
 
-       pbr->draw(skyboxes[activeSkybox]);
+       /* pbr->getRoot()->model = glm::rotate(glm::mat4(1.f), glm::radians(d * 10), glm::vec3(0, 1, 0)); */
+       pbr->draw(skyboxes[activeSkybox], d * 1000);
 
        for (Light &light: lights) drawLight(light);
 
+       if (discoLights) {
+               for (int i = 3; i < 6; i++) {
+                       Light l { lightPositions[i], lightColors[i] };
+                       drawLight(l);
+               }
+       }
+
        skyboxes[activeSkybox].draw(projMat(), viewMat());
 
        glutSwapBuffers();
@@ -160,12 +186,14 @@ void init() {
 
        pbrProg = new Program("pbrvert.glsl", "pbrfrag.glsl");
        glUseProgram(pbrProg->progId);
-       pbr = new Model("models/sphereMetal.gltf", *pbrProg);
+       pbr = new Model("models/newtonsCradle.glb", *pbrProg);
 
        glEnable(GL_DEPTH_TEST); 
        glEnable(GL_CULL_FACE); 
        // prevent edge artifacts in specular cubemaps
        glEnable(GL_TEXTURE_CUBE_MAP_SEAMLESS);
+
+       glViewport(0, 0, windowWidth, windowHeight);
 }
 
 bool* keyStates = new bool[256];
@@ -176,6 +204,8 @@ void keyboard(unsigned char key, int x, int y) {
                activeSkybox = (activeSkybox + 1) % skyboxes.size();
        if (key == 'x')
                activeLight = (activeLight + 1) % lights.size();
+       if (key == 'c')
+               discoLights = !discoLights;
 }
 
 void keyboardUp(unsigned char key, int x, int y) {
@@ -248,7 +278,6 @@ void mouse(int button, int state, int x, int y) {
 }
 
 void reshape(int newWidth, int newHeight) {
-       glViewport(0, 0, newWidth * 2, newHeight * 2);
        windowWidth = newWidth, windowHeight = newHeight;
 }
 
@@ -264,8 +293,6 @@ int main(int argc, char** argv) {
        
        init();
 
-       makeRetina();
-
        glutKeyboardFunc(keyboard);
        glutKeyboardUpFunc(keyboardUp);
        glutTimerFunc(16, timer, 0);