annotate src/zamikongine/vertex.nim @ 32:9edca5dc4e93

add: working implementation of uniforms
author Sam <sam@basx.dev>
date Sat, 14 Jan 2023 23:34:50 +0700
parents b55d6ecde79d
children 94c38e4b5782
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
17
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
1 import std/macros
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
2 import std/strutils
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
3 import std/strformat
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
4 import std/typetraits
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
5
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
6 import ./math/vector
32
9edca5dc4e93 add: working implementation of uniforms
Sam <sam@basx.dev>
parents: 19
diff changeset
7 import ./math/matrix
17
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
8 import ./vulkan
32
9edca5dc4e93 add: working implementation of uniforms
Sam <sam@basx.dev>
parents: 19
diff changeset
9 import ./glsl_helpers
17
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
10
13
a571db114152 add: vector/matrix modules, tests, nim config script
Sam <sam@basx.dev>
parents:
diff changeset
11 type
17
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
12 VertexAttributeType = SomeNumber|Vec
32
9edca5dc4e93 add: working implementation of uniforms
Sam <sam@basx.dev>
parents: 19
diff changeset
13 AttributePurpose* = enum
9edca5dc4e93 add: working implementation of uniforms
Sam <sam@basx.dev>
parents: 19
diff changeset
14 Unknown, Position Color
9edca5dc4e93 add: working implementation of uniforms
Sam <sam@basx.dev>
parents: 19
diff changeset
15 GenericAttribute*[T:VertexAttributeType] = object
9edca5dc4e93 add: working implementation of uniforms
Sam <sam@basx.dev>
parents: 19
diff changeset
16 data*: seq[T]
9edca5dc4e93 add: working implementation of uniforms
Sam <sam@basx.dev>
parents: 19
diff changeset
17 PositionAttribute*[T:VertexAttributeType] = object
19
b55d6ecde79d did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents: 17
diff changeset
18 data*: seq[T]
32
9edca5dc4e93 add: working implementation of uniforms
Sam <sam@basx.dev>
parents: 19
diff changeset
19 ColorAttribute*[T:VertexAttributeType] = object
9edca5dc4e93 add: working implementation of uniforms
Sam <sam@basx.dev>
parents: 19
diff changeset
20 data*: seq[T]
9edca5dc4e93 add: working implementation of uniforms
Sam <sam@basx.dev>
parents: 19
diff changeset
21 VertexAttribute* = GenericAttribute|PositionAttribute|ColorAttribute
17
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
22
32
9edca5dc4e93 add: working implementation of uniforms
Sam <sam@basx.dev>
parents: 19
diff changeset
23 template getAttributeType*(v: VertexAttribute): auto = get(genericParams(typeof(v)), 0)
19
b55d6ecde79d did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents: 17
diff changeset
24
b55d6ecde79d did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents: 17
diff changeset
25 func datasize*(attribute: VertexAttribute): uint64 =
32
9edca5dc4e93 add: working implementation of uniforms
Sam <sam@basx.dev>
parents: 19
diff changeset
26 uint64(sizeof(getAttributeType(attribute))) * uint64(attribute.data.len)
17
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
27
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
28 # from https://registry.khronos.org/vulkan/specs/1.3-extensions/html/chap15.html
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
29 func nLocationSlots[T: VertexAttributeType](): int =
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
30 when (T is Vec3[float64] or T is Vec3[uint64] or T is Vec4[float64] or T is Vec4[float64]):
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
31 2
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
32 else:
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
33 1
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
34
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
35 # numbers
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
36 func getVkFormat[T: VertexAttributeType](): VkFormat =
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
37 when T is uint8: VK_FORMAT_R8_UINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
38 elif T is int8: VK_FORMAT_R8_SINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
39 elif T is uint16: VK_FORMAT_R16_UINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
40 elif T is int16: VK_FORMAT_R16_SINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
41 elif T is uint32: VK_FORMAT_R32_UINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
42 elif T is int32: VK_FORMAT_R32_SINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
43 elif T is uint64: VK_FORMAT_R64_UINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
44 elif T is int64: VK_FORMAT_R64_SINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
45 elif T is float32: VK_FORMAT_R32_SFLOAT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
46 elif T is float64: VK_FORMAT_R64_SFLOAT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
47 elif T is Vec2[uint8]: VK_FORMAT_R8G8_UINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
48 elif T is Vec2[int8]: VK_FORMAT_R8G8_SINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
49 elif T is Vec2[uint16]: VK_FORMAT_R16G16_UINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
50 elif T is Vec2[int16]: VK_FORMAT_R16G16_SINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
51 elif T is Vec2[uint32]: VK_FORMAT_R32G32_UINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
52 elif T is Vec2[int32]: VK_FORMAT_R32G32_SINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
53 elif T is Vec2[uint64]: VK_FORMAT_R64G64_UINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
54 elif T is Vec2[int64]: VK_FORMAT_R64G64_SINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
55 elif T is Vec2[float32]: VK_FORMAT_R32G32_SFLOAT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
56 elif T is Vec2[float64]: VK_FORMAT_R64G64_SFLOAT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
57 elif T is Vec3[uint8]: VK_FORMAT_R8G8B8_UINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
58 elif T is Vec3[int8]: VK_FORMAT_R8G8B8_SINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
59 elif T is Vec3[uint16]: VK_FORMAT_R16G16B16_UINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
60 elif T is Vec3[int16]: VK_FORMAT_R16G16B16_SINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
61 elif T is Vec3[uint32]: VK_FORMAT_R32G32B32_UINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
62 elif T is Vec3[int32]: VK_FORMAT_R32G32B32_SINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
63 elif T is Vec3[uint64]: VK_FORMAT_R64G64B64_UINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
64 elif T is Vec3[int64]: VK_FORMAT_R64G64B64_SINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
65 elif T is Vec3[float32]: VK_FORMAT_R32G32B32_SFLOAT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
66 elif T is Vec3[float64]: VK_FORMAT_R64G64B64_SFLOAT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
67 elif T is Vec4[uint8]: VK_FORMAT_R8G8B8A8_UINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
68 elif T is Vec4[int8]: VK_FORMAT_R8G8B8A8_SINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
69 elif T is Vec4[uint16]: VK_FORMAT_R16G16B16A16_UINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
70 elif T is Vec4[int16]: VK_FORMAT_R16G16B16A16_SINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
71 elif T is Vec4[uint32]: VK_FORMAT_R32G32B32A32_UINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
72 elif T is Vec4[int32]: VK_FORMAT_R32G32B32A32_SINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
73 elif T is Vec4[uint64]: VK_FORMAT_R64G64B64A64_UINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
74 elif T is Vec4[int64]: VK_FORMAT_R64G64B64A64_SINT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
75 elif T is Vec4[float32]: VK_FORMAT_R32G32B32A32_SFLOAT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
76 elif T is Vec4[float64]: VK_FORMAT_R64G64B64A64_SFLOAT
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
77
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
78
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
79
19
b55d6ecde79d did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents: 17
diff changeset
80 func VertexCount*[T](t: T): uint32 =
b55d6ecde79d did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents: 17
diff changeset
81 for name, value in t.fieldPairs:
b55d6ecde79d did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents: 17
diff changeset
82 when typeof(value) is VertexAttribute:
b55d6ecde79d did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents: 17
diff changeset
83 if result == 0:
b55d6ecde79d did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents: 17
diff changeset
84 result = uint32(value.data.len)
b55d6ecde79d did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents: 17
diff changeset
85 else:
b55d6ecde79d did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents: 17
diff changeset
86 assert result == uint32(value.data.len)
b55d6ecde79d did: introduce scene graph, meshs and generic vertex buffers
Sam <sam@basx.dev>
parents: 17
diff changeset
87
32
9edca5dc4e93 add: working implementation of uniforms
Sam <sam@basx.dev>
parents: 19
diff changeset
88 func generateGLSLVertexDeclarations*[T](): string =
17
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
89 var stmtList: seq[string]
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
90 var i = 0
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
91 for name, value in T().fieldPairs:
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
92 when typeof(value) is VertexAttribute:
32
9edca5dc4e93 add: working implementation of uniforms
Sam <sam@basx.dev>
parents: 19
diff changeset
93 let glsltype = getGLSLType[getAttributeType(value)]()
17
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
94 let n = name
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
95 stmtList.add(&"layout(location = {i}) in {glsltype} {n};")
32
9edca5dc4e93 add: working implementation of uniforms
Sam <sam@basx.dev>
parents: 19
diff changeset
96 i += nLocationSlots[getAttributeType(value)]()
17
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
97
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
98 return stmtList.join("\n")
13
a571db114152 add: vector/matrix modules, tests, nim config script
Sam <sam@basx.dev>
parents:
diff changeset
99
17
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
100 func generateInputVertexBinding*[T](bindingoffset: int = 0, locationoffset: int = 0): seq[VkVertexInputBindingDescription] =
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
101 # packed attribute data, not interleaved (aks "struct of arrays")
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
102 var binding = bindingoffset
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
103 for name, value in T().fieldPairs:
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
104 when typeof(value) is VertexAttribute:
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
105 result.add(
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
106 VkVertexInputBindingDescription(
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
107 binding: uint32(binding),
32
9edca5dc4e93 add: working implementation of uniforms
Sam <sam@basx.dev>
parents: 19
diff changeset
108 stride: uint32(sizeof(getAttributeType(value))),
17
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
109 inputRate: VK_VERTEX_INPUT_RATE_VERTEX, # VK_VERTEX_INPUT_RATE_INSTANCE for instances
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
110 )
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
111 )
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
112 binding += 1
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
113
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
114 func generateInputAttributeBinding*[T](bindingoffset: int = 0, locationoffset: int = 0): seq[VkVertexInputAttributeDescription] =
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
115 # packed attribute data, not interleaved (aks "struct of arrays")
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
116 var location = 0
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
117 var binding = bindingoffset
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
118 for name, value in T().fieldPairs:
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
119 when typeof(value) is VertexAttribute:
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
120 result.add(
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
121 VkVertexInputAttributeDescription(
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
122 binding: uint32(binding),
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
123 location: uint32(location),
32
9edca5dc4e93 add: working implementation of uniforms
Sam <sam@basx.dev>
parents: 19
diff changeset
124 format: getVkFormat[getAttributeType(value)](),
17
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
125 offset: 0,
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
126 )
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
127 )
32
9edca5dc4e93 add: working implementation of uniforms
Sam <sam@basx.dev>
parents: 19
diff changeset
128 location += nLocationSlots[getAttributeType(value)]()
17
b40466fa446a add: vertex basics, some refactoring
Sam <sam@basx.dev>
parents: 13
diff changeset
129 binding += 1