Mercurial > games > semicongine

annotate src/zamikongine/shader.nim @ 26:8b4bc545958d

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