Mercurial > games > semicongine

annotate src/zamikongine/shader.nim @ 489:54a1f8ee208e

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
big refactoring, part1
author Sam <sam@basx.dev>
date Sat, 14 Jan 2023 14:08:00 +0700
parents 9231df12b222
children 680c4b8ca28a
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
483
73a0954beabd did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation
Sam <sam@basx.dev>
parents: 480
diff changeset
1 import std/strformat
480
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
2 import std/strutils
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
3 import std/tables
485
b4a972bd37d5 did: change shader compilation to run during program compilation, maybe add dynamic version later
Sam <sam@basx.dev>
parents: 483
diff changeset
4 import std/compilesettings
b4a972bd37d5 did: change shader compilation to run during program compilation, maybe add dynamic version later
Sam <sam@basx.dev>
parents: 483
diff changeset
5
480
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
6
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
7 import ./vulkan_helpers
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
8 import ./vulkan
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
9 import ./vertex
489
54a1f8ee208e big refactoring, part1
Sam <sam@basx.dev>
parents: 486
diff changeset
10 import ./math/vector
480
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
11
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
12 type
489
54a1f8ee208e big refactoring, part1
Sam <sam@basx.dev>
parents: 486
diff changeset
13 AllowedUniformType = SomeNumber|Vec
54a1f8ee208e big refactoring, part1
Sam <sam@basx.dev>
parents: 486
diff changeset
14 UniformSlot *[T:AllowedUniformType] = object
54a1f8ee208e big refactoring, part1
Sam <sam@basx.dev>
parents: 486
diff changeset
15 ShaderProgram*[Uniforms] = object
480
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
16 entryPoint*: string
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
17 programType*: VkShaderStageFlagBits
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
18 shader*: VkPipelineShaderStageCreateInfo
489
54a1f8ee208e big refactoring, part1
Sam <sam@basx.dev>
parents: 486
diff changeset
19 uniforms*: Uniforms
480
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
20
485
b4a972bd37d5 did: change shader compilation to run during program compilation, maybe add dynamic version later
Sam <sam@basx.dev>
parents: 483
diff changeset
21 func stage2string(stage: VkShaderStageFlagBits): string {.compileTime.} =
483
73a0954beabd did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation
Sam <sam@basx.dev>
parents: 480
diff changeset
22 case stage
73a0954beabd did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation
Sam <sam@basx.dev>
parents: 480
diff changeset
23 of VK_SHADER_STAGE_VERTEX_BIT: "vert"
73a0954beabd did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation
Sam <sam@basx.dev>
parents: 480
diff changeset
24 of VK_SHADER_STAGE_TESSELLATION_CONTROL_BIT: "tesc"
73a0954beabd did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation
Sam <sam@basx.dev>
parents: 480
diff changeset
25 of VK_SHADER_STAGE_TESSELLATION_EVALUATION_BIT: "tese"
73a0954beabd did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation
Sam <sam@basx.dev>
parents: 480
diff changeset
26 of VK_SHADER_STAGE_GEOMETRY_BIT: "geom"
73a0954beabd did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation
Sam <sam@basx.dev>
parents: 480
diff changeset
27 of VK_SHADER_STAGE_FRAGMENT_BIT: "frag"
73a0954beabd did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation
Sam <sam@basx.dev>
parents: 480
diff changeset
28 of VK_SHADER_STAGE_ALL_GRAPHICS: ""
73a0954beabd did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation
Sam <sam@basx.dev>
parents: 480
diff changeset
29 of VK_SHADER_STAGE_COMPUTE_BIT: "comp"
73a0954beabd did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation
Sam <sam@basx.dev>
parents: 480
diff changeset
30 of VK_SHADER_STAGE_ALL: ""
73a0954beabd did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation
Sam <sam@basx.dev>
parents: 480
diff changeset
31
485
b4a972bd37d5 did: change shader compilation to run during program compilation, maybe add dynamic version later
Sam <sam@basx.dev>
parents: 483
diff changeset
32 proc compileGLSLToSPIRV(stage: VkShaderStageFlagBits, shaderSource: string, entrypoint: string): seq[uint32] {.compileTime.} =
483
73a0954beabd did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation
Sam <sam@basx.dev>
parents: 480
diff changeset
33 let stagename = stage2string(stage)
486
9231df12b222 fix: build from scratch not working, remove temp shader files from compilation
Sam <sam@basx.dev>
parents: 485
diff changeset
34
489
54a1f8ee208e big refactoring, part1
Sam <sam@basx.dev>
parents: 486
diff changeset
35 # TODO: compiles only on linux for now (because we don't have compile-time functionality in std/tempfile)
54a1f8ee208e big refactoring, part1
Sam <sam@basx.dev>
parents: 486
diff changeset
36 if not defined(linux):
54a1f8ee208e big refactoring, part1
Sam <sam@basx.dev>
parents: 486
diff changeset
37 raise newException(Exception, "Compilation is currently only supported on linux (need mktemp command), sorry!")
486
9231df12b222 fix: build from scratch not working, remove temp shader files from compilation
Sam <sam@basx.dev>
parents: 485
diff changeset
38 let (tmpfile, exitCode) = gorgeEx(command=fmt"mktemp --tmpdir shader_XXXXXXX.{stagename}")
483
73a0954beabd did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation
Sam <sam@basx.dev>
parents: 480
diff changeset
39 if exitCode != 0:
486
9231df12b222 fix: build from scratch not working, remove temp shader files from compilation
Sam <sam@basx.dev>
parents: 485
diff changeset
40 raise newException(Exception, tmpfile)
9231df12b222 fix: build from scratch not working, remove temp shader files from compilation
Sam <sam@basx.dev>
parents: 485
diff changeset
41
9231df12b222 fix: build from scratch not working, remove temp shader files from compilation
Sam <sam@basx.dev>
parents: 485
diff changeset
42 let (output, exitCode_glsl) = gorgeEx(command=fmt"{querySetting(projectPath)}/glslangValidator --entry-point {entrypoint} -V --stdin -S {stagename} -o {tmpfile}", input=shaderSource)
9231df12b222 fix: build from scratch not working, remove temp shader files from compilation
Sam <sam@basx.dev>
parents: 485
diff changeset
43 if exitCode_glsl != 0:
483
73a0954beabd did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation
Sam <sam@basx.dev>
parents: 480
diff changeset
44 raise newException(Exception, output)
486
9231df12b222 fix: build from scratch not working, remove temp shader files from compilation
Sam <sam@basx.dev>
parents: 485
diff changeset
45 let shaderbinary = staticRead tmpfile
9231df12b222 fix: build from scratch not working, remove temp shader files from compilation
Sam <sam@basx.dev>
parents: 485
diff changeset
46
9231df12b222 fix: build from scratch not working, remove temp shader files from compilation
Sam <sam@basx.dev>
parents: 485
diff changeset
47 let (output_rm, exitCode_rm) = gorgeEx(command=fmt"rm {tmpfile}")
9231df12b222 fix: build from scratch not working, remove temp shader files from compilation
Sam <sam@basx.dev>
parents: 485
diff changeset
48 if exitCode_rm != 0:
9231df12b222 fix: build from scratch not working, remove temp shader files from compilation
Sam <sam@basx.dev>
parents: 485
diff changeset
49 raise newException(Exception, output_rm)
9231df12b222 fix: build from scratch not working, remove temp shader files from compilation
Sam <sam@basx.dev>
parents: 485
diff changeset
50
483
73a0954beabd did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation
Sam <sam@basx.dev>
parents: 480
diff changeset
51 var i = 0
73a0954beabd did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation
Sam <sam@basx.dev>
parents: 480
diff changeset
52 while i < shaderbinary.len:
73a0954beabd did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation
Sam <sam@basx.dev>
parents: 480
diff changeset
53 result.add(
73a0954beabd did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation
Sam <sam@basx.dev>
parents: 480
diff changeset
54 (uint32(shaderbinary[i + 0]) shl 0) or
73a0954beabd did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation
Sam <sam@basx.dev>
parents: 480
diff changeset
55 (uint32(shaderbinary[i + 1]) shl 8) or
73a0954beabd did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation
Sam <sam@basx.dev>
parents: 480
diff changeset
56 (uint32(shaderbinary[i + 2]) shl 16) or
73a0954beabd did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation
Sam <sam@basx.dev>
parents: 480
diff changeset
57 (uint32(shaderbinary[i + 3]) shl 24)
73a0954beabd did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation
Sam <sam@basx.dev>
parents: 480
diff changeset
58 )
73a0954beabd did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation
Sam <sam@basx.dev>
parents: 480
diff changeset
59 i += 4
73a0954beabd did: improve alotof_triangles example, remove glslang lib, use only binary, easier cross-compilation
Sam <sam@basx.dev>
parents: 480
diff changeset
60
489
54a1f8ee208e big refactoring, part1
Sam <sam@basx.dev>
parents: 486
diff changeset
61 proc initShaderProgram*[T](device: VkDevice, programType: static VkShaderStageFlagBits, shader: static string, entryPoint: static string="main"): ShaderProgram[T] =
480
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
62 result.entryPoint = entryPoint
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
63 result.programType = programType
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
64
485
b4a972bd37d5 did: change shader compilation to run during program compilation, maybe add dynamic version later
Sam <sam@basx.dev>
parents: 483
diff changeset
65 const constcode = compileGLSLToSPIRV(programType, shader, entryPoint)
b4a972bd37d5 did: change shader compilation to run during program compilation, maybe add dynamic version later
Sam <sam@basx.dev>
parents: 483
diff changeset
66 var code = constcode
480
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
67 var createInfo = VkShaderModuleCreateInfo(
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
68 sType: VK_STRUCTURE_TYPE_SHADER_MODULE_CREATE_INFO,
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
69 codeSize: uint(code.len * sizeof(uint32)),
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
70 pCode: if code.len > 0: addr(code[0]) else: nil,
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
71 )
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
72 var shaderModule: VkShaderModule
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
73 checkVkResult vkCreateShaderModule(device, addr(createInfo), nil, addr(shaderModule))
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
74
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
75 result.shader = VkPipelineShaderStageCreateInfo(
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
76 sType: VK_STRUCTURE_TYPE_PIPELINE_SHADER_STAGE_CREATE_INFO,
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
77 stage: programType,
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
78 module: shaderModule,
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
79 pName: cstring(result.entryPoint), # entry point for shader
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
80 )
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
81
489
54a1f8ee208e big refactoring, part1
Sam <sam@basx.dev>
parents: 486
diff changeset
82 func generateVertexShaderCode*[VertexType](entryPoint, positionAttrName, colorAttrName: static string): string {.compileTime.} =
480
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
83 var lines: seq[string]
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
84 lines.add "#version 450"
489
54a1f8ee208e big refactoring, part1
Sam <sam@basx.dev>
parents: 486
diff changeset
85 # lines.add "layout(binding = 0) uniform UniformBufferObject { float dt; } ubo;"
54a1f8ee208e big refactoring, part1
Sam <sam@basx.dev>
parents: 486
diff changeset
86 lines.add generateGLSLDeclarations[VertexType]()
480
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
87 lines.add "layout(location = 0) out vec3 fragColor;"
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
88 lines.add "void " & entryPoint & "() {"
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
89
489
54a1f8ee208e big refactoring, part1
Sam <sam@basx.dev>
parents: 486
diff changeset
90 for name, value in VertexType().fieldPairs:
480
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
91 when typeof(value) is VertexAttribute and name == positionAttrName:
489
54a1f8ee208e big refactoring, part1
Sam <sam@basx.dev>
parents: 486
diff changeset
92 # lines.add " vec2 tmp = " & name & " * ubo.dt;"
54a1f8ee208e big refactoring, part1
Sam <sam@basx.dev>
parents: 486
diff changeset
93 lines.add " vec2 tmp = " & name & ";"
54a1f8ee208e big refactoring, part1
Sam <sam@basx.dev>
parents: 486
diff changeset
94 lines.add " gl_Position = vec4(tmp, 0.0, 1.0);"
480
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
95 when typeof(value) is VertexAttribute and name == colorAttrName:
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
96 lines.add " fragColor = " & name & ";"
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
97 lines.add "}"
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
98 return lines.join("\n")
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
99
489
54a1f8ee208e big refactoring, part1
Sam <sam@basx.dev>
parents: 486
diff changeset
100 func generateFragmentShaderCode*[VertexType](entryPoint: static string): string {.compileTime.} =
480
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
101 var lines: seq[string]
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
102 lines.add "#version 450"
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
103 lines.add "layout(location = 0) in vec3 fragColor;"
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
104 lines.add "layout(location = 0) out vec4 outColor;"
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
105 lines.add "void " & entryPoint & "() {"
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
106 lines.add " outColor = vec4(fragColor, 1.0);"
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
107 lines.add "}"
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
108
14e5151f68d1 did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents:
diff changeset
109 return lines.join("\n")