You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

281 lines
10 KiB

4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
  1. #include <windows.h>
  2. #include "glew.h"
  3. #include <stdio.h>
  4. #include <math.h>
  5. #include "utils.h"
  6. #include "GPUProgram.h"
  7. #include "ObjModel.h"
  8. #include "FBO.h"
  9. #include "FullScreenQuad.h"
  10. #include "Glm/glm.hpp"
  11. #include "Glm/ext.hpp"
  12. #pragma comment(lib,"opengl32.lib")
  13. #pragma comment(lib,"glew32.lib")
  14. LRESULT CALLBACK GLWindowProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
  15. {
  16. switch (msg)
  17. {
  18. case WM_CLOSE:
  19. PostQuitMessage(0);
  20. break;
  21. }
  22. return DefWindowProc(hwnd,msg,wParam,lParam);
  23. }
  24. INT WINAPI WinMain(_In_ HINSTANCE hInstance, _In_opt_ HINSTANCE hPrevInstance, _In_ LPSTR lpCmdLine, _In_ int nShowCmd)
  25. {
  26. WNDCLASSEX wndClass;
  27. wndClass.cbClsExtra = 0;
  28. wndClass.cbSize = sizeof(WNDCLASSEX);
  29. wndClass.cbWndExtra = 0;
  30. wndClass.hbrBackground = NULL;
  31. wndClass.hCursor = LoadCursor(NULL,IDC_ARROW);
  32. wndClass.hIcon = NULL;
  33. wndClass.hIconSm = NULL;
  34. wndClass.hInstance = hInstance;
  35. wndClass.lpfnWndProc=GLWindowProc;
  36. wndClass.lpszClassName = "OpenGL";
  37. wndClass.lpszMenuName = NULL;
  38. wndClass.style = CS_VREDRAW | CS_HREDRAW;
  39. ATOM atom = RegisterClassEx(&wndClass);
  40. RECT rect;
  41. rect.left = 0;
  42. rect.top = 0;
  43. rect.right = 800;
  44. rect.bottom = 600;
  45. AdjustWindowRect(&rect, WS_OVERLAPPEDWINDOW, FALSE);
  46. HWND hwnd = CreateWindowEx(NULL, "OpenGL", "RenderWindow", WS_OVERLAPPEDWINDOW, 100, 100, rect.right - rect.left, rect.bottom - rect.top, NULL, NULL, hInstance, NULL);
  47. HDC dc = GetDC(hwnd);
  48. PIXELFORMATDESCRIPTOR pfd;
  49. memset(&pfd, 0, sizeof(PIXELFORMATDESCRIPTOR));
  50. pfd.nVersion = 1;
  51. pfd.dwFlags = PFD_DRAW_TO_WINDOW | PFD_SUPPORT_OPENGL | PFD_TYPE_RGBA | PFD_DOUBLEBUFFER;
  52. pfd.iLayerType = PFD_MAIN_PLANE;
  53. pfd.iPixelType = PFD_TYPE_RGBA;
  54. pfd.cColorBits = 32;
  55. pfd.cDepthBits = 24;
  56. pfd.cStencilBits = 8;
  57. int pixelFormatID = ChoosePixelFormat(dc, &pfd);
  58. SetPixelFormat(dc,pixelFormatID,&pfd);
  59. HGLRC rc = wglCreateContext(dc);
  60. wglMakeCurrent(dc, rc);
  61. GetClientRect(hwnd, &rect);
  62. int viewportWidth = rect.right - rect.left, viewportHeight = rect.bottom - rect.top;
  63. glewInit();
  64. //init fsqgpu program
  65. GPUProgram originalProgram;
  66. originalProgram.AttachShader(GL_VERTEX_SHADER, "res/shader/fullscreenquad.vs");
  67. originalProgram.AttachShader(GL_FRAGMENT_SHADER, "res/shader/fullscreenquad.fs");
  68. originalProgram.Link();
  69. originalProgram.DetectAttribute("pos");
  70. originalProgram.DetectAttribute("texcoord");
  71. originalProgram.DetectUniform("U_MainTexture");
  72. //init hdr program
  73. GPUProgram hdrProgram;
  74. hdrProgram.AttachShader(GL_VERTEX_SHADER, "res/shader/fullscreenquad.vs");
  75. hdrProgram.AttachShader(GL_FRAGMENT_SHADER, "res/shader/hdrrender.fs");
  76. hdrProgram.Link();
  77. hdrProgram.DetectAttribute("pos");
  78. hdrProgram.DetectAttribute("texcoord");
  79. hdrProgram.DetectUniform("U_MainTexture");
  80. //combine program
  81. GPUProgram combineProgram;
  82. combineProgram.AttachShader(GL_VERTEX_SHADER, "res/shader/fullscreenquad.vs");
  83. combineProgram.AttachShader(GL_FRAGMENT_SHADER, "res/shader/combine.fs");
  84. combineProgram.Link();
  85. combineProgram.DetectAttribute("pos");
  86. combineProgram.DetectAttribute("texcoord");
  87. combineProgram.DetectUniform("U_MainTexture");
  88. combineProgram.DetectUniform("U_HDRTexture");
  89. //init light source program
  90. GPUProgram lightSourceProgram;
  91. lightSourceProgram.AttachShader(GL_VERTEX_SHADER, "res/shader/lightSource.vs");
  92. lightSourceProgram.AttachShader(GL_FRAGMENT_SHADER, "res/shader/lightSource.fs");
  93. lightSourceProgram.Link();
  94. lightSourceProgram.DetectAttribute("pos");
  95. lightSourceProgram.DetectUniform("M");
  96. lightSourceProgram.DetectUniform("V");
  97. lightSourceProgram.DetectUniform("P");
  98. //init gpu program
  99. GPUProgram gpuProgram;
  100. gpuProgram.AttachShader(GL_VERTEX_SHADER, "res/shader/test.vs");
  101. gpuProgram.AttachShader(GL_FRAGMENT_SHADER, "res/shader/hdr.fs");
  102. gpuProgram.Link();
  103. gpuProgram.DetectAttribute("pos");
  104. gpuProgram.DetectAttribute("texcoord");
  105. gpuProgram.DetectAttribute("normal");
  106. gpuProgram.DetectUniform("M");
  107. gpuProgram.DetectUniform("V");
  108. gpuProgram.DetectUniform("P");
  109. gpuProgram.DetectUniform("NM");
  110. gpuProgram.DetectUniform("U_AmbientLightColor");
  111. gpuProgram.DetectUniform("U_AmbientMaterial");
  112. gpuProgram.DetectUniform("U_DiffuseLightColor");
  113. gpuProgram.DetectUniform("U_DiffuseMaterial");
  114. gpuProgram.DetectUniform("U_DiffuseIntensity");
  115. gpuProgram.DetectUniform("U_LightPos");
  116. gpuProgram.DetectUniform("U_LightDirection");
  117. gpuProgram.DetectUniform("U_Cutoff");
  118. gpuProgram.DetectUniform("U_SpecularLightColor");
  119. gpuProgram.DetectUniform("U_SpecularMaterial");
  120. gpuProgram.DetectUniform("U_EyePos");
  121. //init 3d model
  122. ObjModel obj,quad,sphere;
  123. obj.Init("res/model/Cube.obj");
  124. quad.Init("res/model/Quad.obj");
  125. sphere.Init("res/model/Sphere.obj");
  126. float identity[] = {
  127. 1.0f,0,0,0,
  128. 0,1.0f,0,0,
  129. 0,0,1.0f,0,
  130. 0,0,0,1.0f
  131. };
  132. float ambientLightColor[] = { 0.4f,0.4f,0.4f,1.0f };
  133. float ambientMaterial[] = { 0.2f,0.2f,0.2f,1.0f };
  134. float diffuseLightColor[] = { 1.0f,1.0f,1.0f,1.0f };
  135. float diffuseMaterial[] = { 0.8f, 0.8f, 0.8f, 1.0f };
  136. float lightPos[] = { 0.1f,3.0f, -2.2f,1.0f };
  137. float diffuseIntensity = 3.0f;
  138. float spotLightDirection[] = { 0.0f, -1.0f, 0.0f, 128.0f };//������һλ�ij�......
  139. float spotLightCutoff = 0.0f;
  140. float specularLightColor[] = { 1.0f,1.0f,1.0f,1.0f };
  141. float specularMaterial[] = { 1.0f,1.0f,1.0f,1.0f };
  142. float eyePos[] = { 0.0f, 0.0f, 0.0f };
  143. glm::mat4 model = glm::translate<float>(0.0f, 0.0f, -2.0f) * glm::rotate(-60.0f, 1.0f, 1.0f, 1.0f);
  144. glm::mat4 normalMatrix = glm::inverseTranspose(model);
  145. glm::mat4 quadModel = glm::translate<float>(0.0f, 0.0f, -2.0f) * glm::rotate(-90.0f, 1.0f, 0.0f, 0.0f) * glm::scale(8.0f, 8.0f, 8.0f);
  146. glm::mat4 quadNormalMatrix = glm::inverseTranspose(quadModel);
  147. glm::mat4 sphereModel = glm::translate<float>(0.1f, 3.0f, -2.2f) * glm::scale(0.2f, 0.2f, 0.2f);
  148. glm::mat4 sphereNormalMatrix = glm::inverseTranspose(sphereModel);
  149. glm::mat4 viewMatrix = glm::lookAt(glm::vec3(1.0f, 0.5f, -10.0f), glm::vec3(0.0f, 0.0f, -1.0f), glm::vec3(0.0f, 1.0f, 0.0f));
  150. glm::mat4 projectionMatrix = glm::perspective(50.0f, 800.0f / 600.0f, 0.1f, 1000.0f);
  151. //��ʼ��fsq
  152. FullScreenQuad fsq;
  153. fsq.Init();
  154. //��ʼ��FBO
  155. FBO fbo;
  156. fbo.AttachColorBuffer("color", GL_COLOR_ATTACHMENT0, GL_RGBA, viewportWidth, viewportHeight);
  157. fbo.AttachDepthBuffer("depth", viewportWidth, viewportHeight);
  158. fbo.Finish();
  159. FBO HDRfbo;
  160. HDRfbo.AttachColorBuffer("color", GL_COLOR_ATTACHMENT0, GL_RGBA, viewportWidth, viewportHeight);
  161. HDRfbo.AttachColorBuffer("hdrcolor", GL_COLOR_ATTACHMENT1, GL_RGBA16F, viewportWidth, viewportHeight);
  162. HDRfbo.AttachDepthBuffer("depth", viewportWidth, viewportHeight);
  163. HDRfbo.Finish();
  164. ShowWindow(hwnd, SW_SHOW);
  165. UpdateWindow(hwnd);
  166. glClearColor(0.1f, 0.1f, 0.1f, 1.0f);
  167. MSG msg;
  168. while (true)
  169. {
  170. if (PeekMessage(&msg,NULL,NULL,NULL,PM_REMOVE))
  171. {
  172. if (msg.message==WM_QUIT)
  173. {
  174. break;
  175. }
  176. TranslateMessage(&msg);
  177. DispatchMessage(&msg);
  178. }
  179. glEnable(GL_DEPTH_TEST);
  180. HDRfbo.Bind();
  181. glUseProgram(gpuProgram.mProgram);
  182. glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
  183. glUniform4fv(gpuProgram.GetLocation("U_AmbientLightColor"), 1, ambientLightColor);
  184. glUniform4fv(gpuProgram.GetLocation("U_AmbientMaterial"), 1, ambientMaterial);
  185. glUniform4fv(gpuProgram.GetLocation("U_DiffuseLightColor"), 1, diffuseLightColor);
  186. glUniform4fv(gpuProgram.GetLocation("U_DiffuseMaterial"), 1, diffuseMaterial);
  187. glUniform1f(gpuProgram.GetLocation("U_DiffuseIntensity"), diffuseIntensity);
  188. glUniform4fv(gpuProgram.GetLocation("U_LightDirection"), 1, spotLightDirection);
  189. glUniform4fv(gpuProgram.GetLocation("U_SpecularLightColor"), 1, specularLightColor);
  190. glUniform4fv(gpuProgram.GetLocation("U_SpecularMaterial"), 1, specularMaterial);
  191. glUniform4fv(gpuProgram.GetLocation("U_LightPos"), 1, lightPos);
  192. glUniform3fv(gpuProgram.GetLocation("U_EyePos"), 1, eyePos);
  193. glUniform1f(gpuProgram.GetLocation("U_Cutoff"), spotLightCutoff);
  194. glUniformMatrix4fv(gpuProgram.GetLocation("M"), 1, GL_FALSE, glm::value_ptr(model));
  195. glUniformMatrix4fv(gpuProgram.GetLocation("V"), 1, GL_FALSE, glm::value_ptr(viewMatrix));
  196. glUniformMatrix4fv(gpuProgram.GetLocation("P"), 1, GL_FALSE, glm::value_ptr(projectionMatrix));
  197. glUniformMatrix4fv(gpuProgram.GetLocation("NM"), 1, GL_FALSE, glm::value_ptr(normalMatrix));
  198. obj.Bind(gpuProgram.GetLocation("pos"), gpuProgram.GetLocation("texcoord"), gpuProgram.GetLocation("normal"));
  199. obj.Draw();
  200. glUniformMatrix4fv(gpuProgram.GetLocation("M"), 1, GL_FALSE, glm::value_ptr(quadModel));
  201. glUniformMatrix4fv(gpuProgram.GetLocation("NM"), 1, GL_FALSE, glm::value_ptr(quadNormalMatrix));
  202. quad.Bind(gpuProgram.GetLocation("pos"), gpuProgram.GetLocation("texcoord"), gpuProgram.GetLocation("normal"));
  203. quad.Draw();
  204. glUseProgram(lightSourceProgram.mProgram);
  205. glUniformMatrix4fv(lightSourceProgram.GetLocation("M"), 1, GL_FALSE, glm::value_ptr(sphereModel));
  206. glUniformMatrix4fv(lightSourceProgram.GetLocation("V"), 1, GL_FALSE, glm::value_ptr(viewMatrix));
  207. glUniformMatrix4fv(lightSourceProgram.GetLocation("P"), 1, GL_FALSE, glm::value_ptr(projectionMatrix));
  208. sphere.Bind(lightSourceProgram.GetLocation("pos"), lightSourceProgram.GetLocation("texcoord"), lightSourceProgram.GetLocation("normal"));
  209. sphere.Draw();
  210. glUseProgram(0);
  211. HDRfbo.Unbind();
  212. glFlush();
  213. glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
  214. glActiveTexture(GL_TEXTURE0);
  215. glUseProgram(originalProgram.mProgram);
  216. glBindTexture(GL_TEXTURE_2D, HDRfbo.GetBuffer("color"));
  217. glUniform1i(originalProgram.GetLocation("U_MainTexture"), 0);
  218. fsq.DrawToLeftTop(originalProgram.GetLocation("pos"), originalProgram.GetLocation("texcoord"));
  219. glUseProgram(hdrProgram.mProgram);
  220. glBindTexture(GL_TEXTURE_2D, HDRfbo.GetBuffer("hdrcolor"));
  221. glUniform1i(hdrProgram.GetLocation("U_MainTexture"), 0);
  222. fsq.DrawToRightTop(hdrProgram.GetLocation("pos"), hdrProgram.GetLocation("texcoord"));
  223. glUseProgram(combineProgram.mProgram);
  224. glBindTexture(GL_TEXTURE_2D, HDRfbo.GetBuffer("color"));
  225. glUniform1i(combineProgram.GetLocation("U_MainTexture"), 0);
  226. glActiveTexture(GL_TEXTURE1);
  227. glBindTexture(GL_TEXTURE_2D, HDRfbo.GetBuffer("hdrcolor"));
  228. glUniform1i(combineProgram.GetLocation("U_HDRTexture"), 1);
  229. fsq.DrawToRightBottom(combineProgram.GetLocation("pos"), combineProgram.GetLocation("texcoord"));
  230. SwapBuffers(dc);
  231. }
  232. return 0;
  233. }