Corentin Wallez | 4a9ef4e | 2018-07-18 11:40:26 +0200 | [diff] [blame] | 1 | // Copyright 2017 The Dawn Authors |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 2 | // |
| 3 | // Licensed under the Apache License, Version 2.0 (the "License"); |
| 4 | // you may not use this file except in compliance with the License. |
| 5 | // You may obtain a copy of the License at |
| 6 | // |
| 7 | // http://www.apache.org/licenses/LICENSE-2.0 |
| 8 | // |
| 9 | // Unless required by applicable law or agreed to in writing, software |
| 10 | // distributed under the License is distributed on an "AS IS" BASIS, |
| 11 | // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 12 | // See the License for the specific language governing permissions and |
| 13 | // limitations under the License. |
| 14 | |
Corentin Wallez | 9347e8f | 2017-06-19 13:15:13 -0400 | [diff] [blame] | 15 | #include "SampleUtils.h" |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 16 | |
Yan, Shaobo | a492427 | 2018-12-10 19:47:22 +0000 | [diff] [blame] | 17 | #include "utils/ComboRenderPipelineDescriptor.h" |
Austin Eng | 700a5fb | 2021-06-24 19:21:31 +0000 | [diff] [blame] | 18 | #include "utils/ScopedAutoreleasePool.h" |
Corentin Wallez | 134e080 | 2017-07-17 17:13:57 -0400 | [diff] [blame] | 19 | #include "utils/SystemUtils.h" |
Corentin Wallez | 04863c4 | 2019-10-25 11:36:47 +0000 | [diff] [blame] | 20 | #include "utils/WGPUHelpers.h" |
Corentin Wallez | 5ee7afd | 2017-06-19 13:09:41 -0400 | [diff] [blame] | 21 | |
Corentin Wallez | 4d7d169 | 2018-08-13 08:23:27 +0200 | [diff] [blame] | 22 | #include <glm/glm.hpp> |
| 23 | #include <glm/gtc/matrix_transform.hpp> |
| 24 | #include <glm/gtc/type_ptr.hpp> |
Kai Ninomiya | 2afea0c | 2020-07-10 20:33:08 +0000 | [diff] [blame] | 25 | #include <vector> |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 26 | |
Corentin Wallez | 04863c4 | 2019-10-25 11:36:47 +0000 | [diff] [blame] | 27 | wgpu::Device device; |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 28 | |
Corentin Wallez | 04863c4 | 2019-10-25 11:36:47 +0000 | [diff] [blame] | 29 | wgpu::Buffer indexBuffer; |
| 30 | wgpu::Buffer vertexBuffer; |
| 31 | wgpu::Buffer planeBuffer; |
| 32 | wgpu::Buffer cameraBuffer; |
| 33 | wgpu::Buffer transformBuffer[2]; |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 34 | |
Corentin Wallez | 04863c4 | 2019-10-25 11:36:47 +0000 | [diff] [blame] | 35 | wgpu::BindGroup cameraBindGroup; |
| 36 | wgpu::BindGroup bindGroup[2]; |
| 37 | wgpu::BindGroup cubeTransformBindGroup[2]; |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 38 | |
Corentin Wallez | 04863c4 | 2019-10-25 11:36:47 +0000 | [diff] [blame] | 39 | wgpu::Queue queue; |
| 40 | wgpu::SwapChain swapchain; |
| 41 | wgpu::TextureView depthStencilView; |
| 42 | wgpu::RenderPipeline pipeline; |
| 43 | wgpu::RenderPipeline planePipeline; |
| 44 | wgpu::RenderPipeline reflectionPipeline; |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 45 | |
| 46 | void initBuffers() { |
Kai Ninomiya | 2afea0c | 2020-07-10 20:33:08 +0000 | [diff] [blame] | 47 | static const uint32_t indexData[6 * 6] = {0, 1, 2, 0, 2, 3, |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 48 | |
Kai Ninomiya | 2afea0c | 2020-07-10 20:33:08 +0000 | [diff] [blame] | 49 | 4, 5, 6, 4, 6, 7, |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 50 | |
Kai Ninomiya | 2afea0c | 2020-07-10 20:33:08 +0000 | [diff] [blame] | 51 | 8, 9, 10, 8, 10, 11, |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 52 | |
Kai Ninomiya | 2afea0c | 2020-07-10 20:33:08 +0000 | [diff] [blame] | 53 | 12, 13, 14, 12, 14, 15, |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 54 | |
Kai Ninomiya | 2afea0c | 2020-07-10 20:33:08 +0000 | [diff] [blame] | 55 | 16, 17, 18, 16, 18, 19, |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 56 | |
Kai Ninomiya | 2afea0c | 2020-07-10 20:33:08 +0000 | [diff] [blame] | 57 | 20, 21, 22, 20, 22, 23}; |
Corentin Wallez | 9e9e29f | 2019-08-27 08:21:39 +0000 | [diff] [blame] | 58 | indexBuffer = |
Corentin Wallez | 04863c4 | 2019-10-25 11:36:47 +0000 | [diff] [blame] | 59 | utils::CreateBufferFromData(device, indexData, sizeof(indexData), wgpu::BufferUsage::Index); |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 60 | |
| 61 | static const float vertexData[6 * 4 * 6] = { |
Kai Ninomiya | 2afea0c | 2020-07-10 20:33:08 +0000 | [diff] [blame] | 62 | -1.0, -1.0, 1.0, 1.0, 0.0, 0.0, 1.0, -1.0, 1.0, 1.0, 0.0, 0.0, |
| 63 | 1.0, 1.0, 1.0, 1.0, 0.0, 0.0, -1.0, 1.0, 1.0, 1.0, 0.0, 0.0, |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 64 | |
Kai Ninomiya | 2afea0c | 2020-07-10 20:33:08 +0000 | [diff] [blame] | 65 | -1.0, -1.0, -1.0, 1.0, 1.0, 0.0, -1.0, 1.0, -1.0, 1.0, 1.0, 0.0, |
| 66 | 1.0, 1.0, -1.0, 1.0, 1.0, 0.0, 1.0, -1.0, -1.0, 1.0, 1.0, 0.0, |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 67 | |
Kai Ninomiya | 2afea0c | 2020-07-10 20:33:08 +0000 | [diff] [blame] | 68 | -1.0, 1.0, -1.0, 1.0, 0.0, 1.0, -1.0, 1.0, 1.0, 1.0, 0.0, 1.0, |
| 69 | 1.0, 1.0, 1.0, 1.0, 0.0, 1.0, 1.0, 1.0, -1.0, 1.0, 0.0, 1.0, |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 70 | |
Kai Ninomiya | 2afea0c | 2020-07-10 20:33:08 +0000 | [diff] [blame] | 71 | -1.0, -1.0, -1.0, 0.0, 1.0, 0.0, 1.0, -1.0, -1.0, 0.0, 1.0, 0.0, |
| 72 | 1.0, -1.0, 1.0, 0.0, 1.0, 0.0, -1.0, -1.0, 1.0, 0.0, 1.0, 0.0, |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 73 | |
Kai Ninomiya | 2afea0c | 2020-07-10 20:33:08 +0000 | [diff] [blame] | 74 | 1.0, -1.0, -1.0, 0.0, 1.0, 1.0, 1.0, 1.0, -1.0, 0.0, 1.0, 1.0, |
| 75 | 1.0, 1.0, 1.0, 0.0, 1.0, 1.0, 1.0, -1.0, 1.0, 0.0, 1.0, 1.0, |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 76 | |
Kai Ninomiya | 2afea0c | 2020-07-10 20:33:08 +0000 | [diff] [blame] | 77 | -1.0, -1.0, -1.0, 1.0, 1.0, 1.0, -1.0, -1.0, 1.0, 1.0, 1.0, 1.0, |
| 78 | -1.0, 1.0, 1.0, 1.0, 1.0, 1.0, -1.0, 1.0, -1.0, 1.0, 1.0, 1.0}; |
Corentin Wallez | 9e9e29f | 2019-08-27 08:21:39 +0000 | [diff] [blame] | 79 | vertexBuffer = utils::CreateBufferFromData(device, vertexData, sizeof(vertexData), |
Corentin Wallez | 04863c4 | 2019-10-25 11:36:47 +0000 | [diff] [blame] | 80 | wgpu::BufferUsage::Vertex); |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 81 | |
| 82 | static const float planeData[6 * 4] = { |
Kai Ninomiya | 2afea0c | 2020-07-10 20:33:08 +0000 | [diff] [blame] | 83 | -2.0, -1.0, -2.0, 0.5, 0.5, 0.5, 2.0, -1.0, -2.0, 0.5, 0.5, 0.5, |
| 84 | 2.0, -1.0, 2.0, 0.5, 0.5, 0.5, -2.0, -1.0, 2.0, 0.5, 0.5, 0.5, |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 85 | }; |
Corentin Wallez | 9e9e29f | 2019-08-27 08:21:39 +0000 | [diff] [blame] | 86 | planeBuffer = utils::CreateBufferFromData(device, planeData, sizeof(planeData), |
Corentin Wallez | 04863c4 | 2019-10-25 11:36:47 +0000 | [diff] [blame] | 87 | wgpu::BufferUsage::Vertex); |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 88 | } |
| 89 | |
Austin Eng | 58c76b3 | 2017-06-02 14:31:53 -0400 | [diff] [blame] | 90 | struct CameraData { |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 91 | glm::mat4 view; |
| 92 | glm::mat4 proj; |
| 93 | } cameraData; |
| 94 | |
| 95 | void init() { |
Corentin Wallez | 39039fa | 2018-07-18 14:06:10 +0200 | [diff] [blame] | 96 | device = CreateCppDawnDevice(); |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 97 | |
Corentin Wallez | 6d315da | 2021-02-04 15:33:42 +0000 | [diff] [blame] | 98 | queue = device.GetQueue(); |
Kai Ninomiya | c16a67a | 2017-07-27 18:30:57 -0700 | [diff] [blame] | 99 | swapchain = GetSwapChain(device); |
Corentin Wallez | 6b08781 | 2020-10-27 15:35:56 +0000 | [diff] [blame] | 100 | swapchain.Configure(GetPreferredSwapChainTextureFormat(), wgpu::TextureUsage::RenderAttachment, |
Corentin Wallez | 9e9e29f | 2019-08-27 08:21:39 +0000 | [diff] [blame] | 101 | 640, 480); |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 102 | |
| 103 | initBuffers(); |
| 104 | |
Corentin Wallez | 7aec4ae | 2021-03-24 15:55:32 +0000 | [diff] [blame] | 105 | wgpu::ShaderModule vsModule = utils::CreateShaderModule(device, R"( |
Corentin Wallez | 4814bdb | 2020-11-26 16:39:46 +0000 | [diff] [blame] | 106 | [[block]] struct Camera { |
Ben Clayton | c568684 | 2021-03-17 09:48:19 +0000 | [diff] [blame] | 107 | view : mat4x4<f32>; |
| 108 | proj : mat4x4<f32>; |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 109 | }; |
dan sinclair | 0f9c2d7 | 2021-01-19 14:18:51 +0000 | [diff] [blame] | 110 | [[group(0), binding(0)]] var<uniform> camera : Camera; |
Corentin Wallez | 4814bdb | 2020-11-26 16:39:46 +0000 | [diff] [blame] | 111 | |
| 112 | [[block]] struct Model { |
Ben Clayton | c568684 | 2021-03-17 09:48:19 +0000 | [diff] [blame] | 113 | matrix : mat4x4<f32>; |
Corentin Wallez | 4814bdb | 2020-11-26 16:39:46 +0000 | [diff] [blame] | 114 | }; |
dan sinclair | 0f9c2d7 | 2021-01-19 14:18:51 +0000 | [diff] [blame] | 115 | [[group(0), binding(1)]] var<uniform> model : Model; |
Corentin Wallez | 4814bdb | 2020-11-26 16:39:46 +0000 | [diff] [blame] | 116 | |
James Price | 9e0debd | 2021-04-13 14:52:44 +0000 | [diff] [blame] | 117 | struct VertexOut { |
| 118 | [[location(2)]] f_col : vec3<f32>; |
| 119 | [[builtin(position)]] Position : vec4<f32>; |
| 120 | }; |
Corentin Wallez | 4814bdb | 2020-11-26 16:39:46 +0000 | [diff] [blame] | 121 | |
James Price | 9e0debd | 2021-04-13 14:52:44 +0000 | [diff] [blame] | 122 | [[stage(vertex)]] fn main( |
| 123 | [[location(0)]] pos : vec3<f32>, |
| 124 | [[location(1)]] col : vec3<f32>) -> VertexOut { |
| 125 | var output : VertexOut; |
| 126 | output.f_col = col; |
| 127 | output.Position = camera.proj * camera.view * model.matrix * vec4<f32>(pos, 1.0); |
| 128 | return output; |
Corentin Wallez | 2a1d8c2 | 2019-07-12 17:52:22 +0000 | [diff] [blame] | 129 | })"); |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 130 | |
Corentin Wallez | 7aec4ae | 2021-03-24 15:55:32 +0000 | [diff] [blame] | 131 | wgpu::ShaderModule fsModule = utils::CreateShaderModule(device, R"( |
James Price | 9e0debd | 2021-04-13 14:52:44 +0000 | [diff] [blame] | 132 | [[stage(fragment)]] fn main( |
| 133 | [[location(2)]] f_col : vec3<f32>) -> [[location(0)]] vec4<f32> { |
| 134 | return vec4<f32>(f_col, 1.0); |
Corentin Wallez | b6fb5f3 | 2017-08-29 13:37:45 -0400 | [diff] [blame] | 135 | })"); |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 136 | |
Corentin Wallez | 7aec4ae | 2021-03-24 15:55:32 +0000 | [diff] [blame] | 137 | wgpu::ShaderModule fsReflectionModule = utils::CreateShaderModule(device, R"( |
James Price | 9e0debd | 2021-04-13 14:52:44 +0000 | [diff] [blame] | 138 | [[stage(fragment)]] fn main( |
| 139 | [[location(2)]] f_col : vec3<f32>) -> [[location(0)]] vec4<f32> { |
| 140 | return vec4<f32>(mix(f_col, vec3<f32>(0.5, 0.5, 0.5), vec3<f32>(0.5, 0.5, 0.5)), 1.0); |
Corentin Wallez | b6fb5f3 | 2017-08-29 13:37:45 -0400 | [diff] [blame] | 141 | })"); |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 142 | |
Corentin Wallez | 9895c27 | 2021-03-18 16:46:58 +0000 | [diff] [blame] | 143 | wgpu::VertexAttribute attributes[2]; |
| 144 | attributes[0].shaderLocation = 0; |
| 145 | attributes[0].offset = 0; |
| 146 | attributes[0].format = wgpu::VertexFormat::Float32x3; |
| 147 | attributes[1].shaderLocation = 1; |
| 148 | attributes[1].offset = 3 * sizeof(float); |
| 149 | attributes[1].format = wgpu::VertexFormat::Float32x3; |
Yunchao He | 1ba2cb8 | 2019-03-28 05:09:01 +0000 | [diff] [blame] | 150 | |
Corentin Wallez | 9895c27 | 2021-03-18 16:46:58 +0000 | [diff] [blame] | 151 | wgpu::VertexBufferLayout vertexBufferLayout; |
| 152 | vertexBufferLayout.attributeCount = 2; |
| 153 | vertexBufferLayout.attributes = attributes; |
| 154 | vertexBufferLayout.arrayStride = 6 * sizeof(float); |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 155 | |
Kai Ninomiya | 234becf | 2018-07-10 12:23:50 -0700 | [diff] [blame] | 156 | auto bgl = utils::MakeBindGroupLayout( |
| 157 | device, { |
Corentin Wallez | 9895c27 | 2021-03-18 16:46:58 +0000 | [diff] [blame] | 158 | {0, wgpu::ShaderStage::Vertex, wgpu::BufferBindingType::Uniform}, |
| 159 | {1, wgpu::ShaderStage::Vertex, wgpu::BufferBindingType::Uniform}, |
Kai Ninomiya | 234becf | 2018-07-10 12:23:50 -0700 | [diff] [blame] | 160 | }); |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 161 | |
Corentin Wallez | 04863c4 | 2019-10-25 11:36:47 +0000 | [diff] [blame] | 162 | wgpu::PipelineLayout pl = utils::MakeBasicPipelineLayout(device, &bgl); |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 163 | |
Corentin Wallez | 04863c4 | 2019-10-25 11:36:47 +0000 | [diff] [blame] | 164 | wgpu::BufferDescriptor cameraBufDesc; |
Corentin Wallez | 82b6573 | 2018-08-22 15:37:29 +0200 | [diff] [blame] | 165 | cameraBufDesc.size = sizeof(CameraData); |
Corentin Wallez | 04863c4 | 2019-10-25 11:36:47 +0000 | [diff] [blame] | 166 | cameraBufDesc.usage = wgpu::BufferUsage::CopyDst | wgpu::BufferUsage::Uniform; |
Corentin Wallez | 82b6573 | 2018-08-22 15:37:29 +0200 | [diff] [blame] | 167 | cameraBuffer = device.CreateBuffer(&cameraBufDesc); |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 168 | |
| 169 | glm::mat4 transform(1.0); |
Corentin Wallez | 9e9e29f | 2019-08-27 08:21:39 +0000 | [diff] [blame] | 170 | transformBuffer[0] = utils::CreateBufferFromData(device, &transform, sizeof(glm::mat4), |
Corentin Wallez | 04863c4 | 2019-10-25 11:36:47 +0000 | [diff] [blame] | 171 | wgpu::BufferUsage::Uniform); |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 172 | |
| 173 | transform = glm::translate(transform, glm::vec3(0.f, -2.f, 0.f)); |
Corentin Wallez | 9e9e29f | 2019-08-27 08:21:39 +0000 | [diff] [blame] | 174 | transformBuffer[1] = utils::CreateBufferFromData(device, &transform, sizeof(glm::mat4), |
Corentin Wallez | 04863c4 | 2019-10-25 11:36:47 +0000 | [diff] [blame] | 175 | wgpu::BufferUsage::Uniform); |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 176 | |
Kai Ninomiya | 2afea0c | 2020-07-10 20:33:08 +0000 | [diff] [blame] | 177 | bindGroup[0] = utils::MakeBindGroup( |
| 178 | device, bgl, |
| 179 | {{0, cameraBuffer, 0, sizeof(CameraData)}, {1, transformBuffer[0], 0, sizeof(glm::mat4)}}); |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 180 | |
Kai Ninomiya | 2afea0c | 2020-07-10 20:33:08 +0000 | [diff] [blame] | 181 | bindGroup[1] = utils::MakeBindGroup( |
| 182 | device, bgl, |
| 183 | {{0, cameraBuffer, 0, sizeof(CameraData)}, {1, transformBuffer[1], 0, sizeof(glm::mat4)}}); |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 184 | |
Kai Ninomiya | c16a67a | 2017-07-27 18:30:57 -0700 | [diff] [blame] | 185 | depthStencilView = CreateDefaultDepthStencilView(device); |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 186 | |
Corentin Wallez | 9895c27 | 2021-03-18 16:46:58 +0000 | [diff] [blame] | 187 | { |
Brandon Jones | 41c87d9 | 2021-05-21 05:01:38 +0000 | [diff] [blame] | 188 | utils::ComboRenderPipelineDescriptor descriptor; |
Corentin Wallez | 9895c27 | 2021-03-18 16:46:58 +0000 | [diff] [blame] | 189 | descriptor.vertex.module = vsModule; |
| 190 | descriptor.vertex.bufferCount = 1; |
| 191 | descriptor.vertex.buffers = &vertexBufferLayout; |
Yan, Shaobo | a492427 | 2018-12-10 19:47:22 +0000 | [diff] [blame] | 192 | |
Corentin Wallez | 9895c27 | 2021-03-18 16:46:58 +0000 | [diff] [blame] | 193 | descriptor.layout = pl; |
| 194 | descriptor.cFragment.module = fsModule; |
| 195 | descriptor.cTargets[0].format = GetPreferredSwapChainTextureFormat(); |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 196 | |
Corentin Wallez | 9895c27 | 2021-03-18 16:46:58 +0000 | [diff] [blame] | 197 | wgpu::DepthStencilState* depthStencil = |
| 198 | descriptor.EnableDepthStencil(wgpu::TextureFormat::Depth24PlusStencil8); |
| 199 | depthStencil->depthWriteEnabled = true; |
| 200 | depthStencil->depthCompare = wgpu::CompareFunction::Less; |
Yan, Shaobo | a492427 | 2018-12-10 19:47:22 +0000 | [diff] [blame] | 201 | |
Brandon Jones | 41c87d9 | 2021-05-21 05:01:38 +0000 | [diff] [blame] | 202 | pipeline = device.CreateRenderPipeline(&descriptor); |
Corentin Wallez | 9895c27 | 2021-03-18 16:46:58 +0000 | [diff] [blame] | 203 | } |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 204 | |
Corentin Wallez | 9895c27 | 2021-03-18 16:46:58 +0000 | [diff] [blame] | 205 | { |
Brandon Jones | 41c87d9 | 2021-05-21 05:01:38 +0000 | [diff] [blame] | 206 | utils::ComboRenderPipelineDescriptor descriptor; |
Corentin Wallez | 9895c27 | 2021-03-18 16:46:58 +0000 | [diff] [blame] | 207 | descriptor.vertex.module = vsModule; |
| 208 | descriptor.vertex.bufferCount = 1; |
| 209 | descriptor.vertex.buffers = &vertexBufferLayout; |
Yan, Shaobo | a492427 | 2018-12-10 19:47:22 +0000 | [diff] [blame] | 210 | |
Corentin Wallez | 9895c27 | 2021-03-18 16:46:58 +0000 | [diff] [blame] | 211 | descriptor.layout = pl; |
| 212 | descriptor.cFragment.module = fsModule; |
| 213 | descriptor.cTargets[0].format = GetPreferredSwapChainTextureFormat(); |
| 214 | |
| 215 | wgpu::DepthStencilState* depthStencil = |
| 216 | descriptor.EnableDepthStencil(wgpu::TextureFormat::Depth24PlusStencil8); |
| 217 | depthStencil->stencilFront.passOp = wgpu::StencilOperation::Replace; |
| 218 | depthStencil->stencilBack.passOp = wgpu::StencilOperation::Replace; |
| 219 | depthStencil->depthCompare = wgpu::CompareFunction::Less; |
| 220 | |
Brandon Jones | 41c87d9 | 2021-05-21 05:01:38 +0000 | [diff] [blame] | 221 | planePipeline = device.CreateRenderPipeline(&descriptor); |
Corentin Wallez | 9895c27 | 2021-03-18 16:46:58 +0000 | [diff] [blame] | 222 | } |
| 223 | |
| 224 | { |
Brandon Jones | 41c87d9 | 2021-05-21 05:01:38 +0000 | [diff] [blame] | 225 | utils::ComboRenderPipelineDescriptor descriptor; |
Corentin Wallez | 9895c27 | 2021-03-18 16:46:58 +0000 | [diff] [blame] | 226 | descriptor.vertex.module = vsModule; |
| 227 | descriptor.vertex.bufferCount = 1; |
| 228 | descriptor.vertex.buffers = &vertexBufferLayout; |
| 229 | |
| 230 | descriptor.layout = pl; |
| 231 | descriptor.cFragment.module = fsReflectionModule; |
| 232 | descriptor.cTargets[0].format = GetPreferredSwapChainTextureFormat(); |
| 233 | |
| 234 | wgpu::DepthStencilState* depthStencil = |
| 235 | descriptor.EnableDepthStencil(wgpu::TextureFormat::Depth24PlusStencil8); |
| 236 | depthStencil->stencilFront.compare = wgpu::CompareFunction::Equal; |
| 237 | depthStencil->stencilBack.compare = wgpu::CompareFunction::Equal; |
| 238 | depthStencil->stencilFront.passOp = wgpu::StencilOperation::Replace; |
| 239 | depthStencil->stencilBack.passOp = wgpu::StencilOperation::Replace; |
| 240 | depthStencil->depthWriteEnabled = true; |
| 241 | depthStencil->depthCompare = wgpu::CompareFunction::Less; |
| 242 | |
Brandon Jones | 41c87d9 | 2021-05-21 05:01:38 +0000 | [diff] [blame] | 243 | reflectionPipeline = device.CreateRenderPipeline(&descriptor); |
Corentin Wallez | 9895c27 | 2021-03-18 16:46:58 +0000 | [diff] [blame] | 244 | } |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 245 | |
| 246 | cameraData.proj = glm::perspective(glm::radians(45.0f), 1.f, 1.0f, 100.0f); |
| 247 | } |
| 248 | |
Kai Ninomiya | 2afea0c | 2020-07-10 20:33:08 +0000 | [diff] [blame] | 249 | struct { |
| 250 | uint32_t a; |
| 251 | float b; |
| 252 | } s; |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 253 | void frame() { |
| 254 | s.a = (s.a + 1) % 256; |
Corentin Wallez | 83e779d | 2017-07-10 21:44:06 -0400 | [diff] [blame] | 255 | s.b += 0.01f; |
Kai Ninomiya | 2afea0c | 2020-07-10 20:33:08 +0000 | [diff] [blame] | 256 | if (s.b >= 1.0f) { |
| 257 | s.b = 0.0f; |
| 258 | } |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 259 | |
Kai Ninomiya | 2afea0c | 2020-07-10 20:33:08 +0000 | [diff] [blame] | 260 | cameraData.view = glm::lookAt(glm::vec3(8.f * std::sin(glm::radians(s.b * 360.f)), 2.f, |
| 261 | 8.f * std::cos(glm::radians(s.b * 360.f))), |
| 262 | glm::vec3(0.0f, 0.0f, 0.0f), glm::vec3(0.0f, 1.0f, 0.0f)); |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 263 | |
Corentin Wallez | 47a3341 | 2020-06-02 09:24:39 +0000 | [diff] [blame] | 264 | queue.WriteBuffer(cameraBuffer, 0, &cameraData, sizeof(CameraData)); |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 265 | |
Corentin Wallez | 604072b | 2019-11-12 18:30:11 +0000 | [diff] [blame] | 266 | wgpu::TextureView backbufferView = swapchain.GetCurrentTextureView(); |
| 267 | utils::ComboRenderPassDescriptor renderPass({backbufferView}, depthStencilView); |
Kai Ninomiya | c16a67a | 2017-07-27 18:30:57 -0700 | [diff] [blame] | 268 | |
Corentin Wallez | 04863c4 | 2019-10-25 11:36:47 +0000 | [diff] [blame] | 269 | wgpu::CommandEncoder encoder = device.CreateCommandEncoder(); |
Corentin Wallez | 82fbccb | 2018-09-21 00:24:37 +0000 | [diff] [blame] | 270 | { |
Corentin Wallez | 04863c4 | 2019-10-25 11:36:47 +0000 | [diff] [blame] | 271 | wgpu::RenderPassEncoder pass = encoder.BeginRenderPass(&renderPass); |
Yan, Shaobo | 300eec0 | 2018-12-21 10:40:26 +0000 | [diff] [blame] | 272 | pass.SetPipeline(pipeline); |
Corentin Wallez | 70c8c10 | 2019-10-09 16:08:42 +0000 | [diff] [blame] | 273 | pass.SetBindGroup(0, bindGroup[0]); |
François Beaufort | 91b2142 | 2019-10-10 07:29:58 +0000 | [diff] [blame] | 274 | pass.SetVertexBuffer(0, vertexBuffer); |
Corentin Wallez | 5fad85b | 2020-11-25 08:54:14 +0000 | [diff] [blame] | 275 | pass.SetIndexBuffer(indexBuffer, wgpu::IndexFormat::Uint32); |
Corentin Wallez | 67b1ad7 | 2020-03-31 16:21:35 +0000 | [diff] [blame] | 276 | pass.DrawIndexed(36); |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 277 | |
Corentin Wallez | 82fbccb | 2018-09-21 00:24:37 +0000 | [diff] [blame] | 278 | pass.SetStencilReference(0x1); |
Yan, Shaobo | 300eec0 | 2018-12-21 10:40:26 +0000 | [diff] [blame] | 279 | pass.SetPipeline(planePipeline); |
Corentin Wallez | 70c8c10 | 2019-10-09 16:08:42 +0000 | [diff] [blame] | 280 | pass.SetBindGroup(0, bindGroup[0]); |
François Beaufort | 91b2142 | 2019-10-10 07:29:58 +0000 | [diff] [blame] | 281 | pass.SetVertexBuffer(0, planeBuffer); |
Corentin Wallez | 67b1ad7 | 2020-03-31 16:21:35 +0000 | [diff] [blame] | 282 | pass.DrawIndexed(6); |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 283 | |
Yan, Shaobo | 300eec0 | 2018-12-21 10:40:26 +0000 | [diff] [blame] | 284 | pass.SetPipeline(reflectionPipeline); |
François Beaufort | 91b2142 | 2019-10-10 07:29:58 +0000 | [diff] [blame] | 285 | pass.SetVertexBuffer(0, vertexBuffer); |
Corentin Wallez | 70c8c10 | 2019-10-09 16:08:42 +0000 | [diff] [blame] | 286 | pass.SetBindGroup(0, bindGroup[1]); |
Corentin Wallez | 67b1ad7 | 2020-03-31 16:21:35 +0000 | [diff] [blame] | 287 | pass.DrawIndexed(36); |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 288 | |
Corentin Wallez | 82fbccb | 2018-09-21 00:24:37 +0000 | [diff] [blame] | 289 | pass.EndPass(); |
| 290 | } |
| 291 | |
Corentin Wallez | 04863c4 | 2019-10-25 11:36:47 +0000 | [diff] [blame] | 292 | wgpu::CommandBuffer commands = encoder.Finish(); |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 293 | queue.Submit(1, &commands); |
Corentin Wallez | 604072b | 2019-11-12 18:30:11 +0000 | [diff] [blame] | 294 | swapchain.Present(); |
Kai Ninomiya | c16a67a | 2017-07-27 18:30:57 -0700 | [diff] [blame] | 295 | DoFlush(); |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 296 | } |
| 297 | |
| 298 | int main(int argc, const char* argv[]) { |
Corentin Wallez | 9347e8f | 2017-06-19 13:15:13 -0400 | [diff] [blame] | 299 | if (!InitSample(argc, argv)) { |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 300 | return 1; |
| 301 | } |
| 302 | init(); |
| 303 | |
| 304 | while (!ShouldQuit()) { |
Austin Eng | 700a5fb | 2021-06-24 19:21:31 +0000 | [diff] [blame] | 305 | utils::ScopedAutoreleasePool pool; |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 306 | frame(); |
Corentin Wallez | 134e080 | 2017-07-17 17:13:57 -0400 | [diff] [blame] | 307 | utils::USleep(16000); |
Austin Eng | 376f1c6 | 2017-05-30 20:03:44 -0400 | [diff] [blame] | 308 | } |
| 309 | |
| 310 | // TODO release stuff |
| 311 | } |