working on descriptor sets; ci skip
All checks were successful
Test / build (push) Has been skipped
Build / build (push) Has been skipped

This commit is contained in:
2026-02-01 22:19:36 +01:00
parent 57de432d0b
commit 1974afd6d7
9 changed files with 430 additions and 726 deletions

View File

@@ -3,6 +3,14 @@ const spv = @import("spv");
const shader_source = @embedFile("shader.spv"); const shader_source = @embedFile("shader.spv");
const Input = struct {
value: [4]i32 = [4]i32{ 0, 0, 0, 0 },
};
const Output = struct {
value: [4]i32 = [4]i32{ 0, 0, 0, 0 },
};
pub fn main() !void { pub fn main() !void {
{ {
var gpa: std.heap.DebugAllocator(.{ var gpa: std.heap.DebugAllocator(.{
@@ -19,21 +27,18 @@ pub fn main() !void {
defer rt.deinit(allocator); defer rt.deinit(allocator);
const entry = try rt.getEntryPointByName("main"); const entry = try rt.getEntryPointByName("main");
const color = try rt.getResultByName("color");
const time = try rt.getResultByName("time");
const pos = try rt.getResultByName("pos");
const res = try rt.getResultByName("res");
var output: [4]f32 = undefined; var input: Input = .{};
var output: Output = .{};
try rt.writeInput(f32, &.{@as(f32, @floatFromInt(std.time.milliTimestamp()))}, time); try rt.writeDescriptorSet(Input, allocator, &input, 0, 0);
try rt.writeInput(f32, &.{ 1250.0, 720.0 }, res); try rt.writeDescriptorSet(Output, allocator, &output, 0, 1);
try rt.writeInput(f32, &.{ 0.0, 0.0 }, pos);
try rt.callEntryPoint(allocator, entry); try rt.callEntryPoint(allocator, entry);
try rt.readOutput(f32, output[0..output.len], color); try rt.readDescriptorSet(Output, allocator, &output, 0, 1);
std.log.info("Output: Vec4{any}", .{output});
std.log.info("Output: {any}", .{output});
std.log.info("\nTotal memory used: {d:.3} KB\n", .{@as(f32, @floatFromInt(gpa.total_requested_bytes)) / 1000.0}); std.log.info("\nTotal memory used: {d:.3} KB\n", .{@as(f32, @floatFromInt(gpa.total_requested_bytes)) / 1000.0});
} }

View File

@@ -1,4 +1,4 @@
[nzsl_version("1.1")] [sudo mkswap /swapfilenzsl_version("1.1")]
module; module;
struct FragIn struct FragIn

Binary file not shown.

View File

@@ -1,604 +1,83 @@
; SPIR-V OpCapability Shader
; Version: 1.0 %1 = OpExtInstImport "GLSL.std.450"
; Generator: Khronos Glslang Reference Front End; 11 OpMemoryModel Logical GLSL450
; Bound: 408 OpEntryPoint GLCompute %4 "main" %11 %20
; Schema: 0 OpExecutionMode %4 LocalSize 1 1 1
OpCapability Shader OpDecorate %11 BuiltIn NumWorkgroups
%1 = OpExtInstImport "GLSL.std.450" OpDecorate %20 BuiltIn WorkgroupId
OpMemoryModel Logical GLSL450 OpMemberDecorate %37 0 Offset 0
OpEntryPoint Vertex %4 "main" %312 %314 %406 OpDecorate %38 ArrayStride 4
OpDecorate %11 RelaxedPrecision OpDecorate %39 BufferBlock
OpDecorate %18 RelaxedPrecision OpMemberDecorate %39 0 Offset 0
OpDecorate %25 RelaxedPrecision OpDecorate %41 Binding 0
OpDecorate %32 RelaxedPrecision OpDecorate %41 DescriptorSet 0
OpDecorate %39 RelaxedPrecision OpMemberDecorate %50 0 Offset 0
OpDecorate %46 RelaxedPrecision OpDecorate %51 ArrayStride 4
OpDecorate %53 RelaxedPrecision OpDecorate %52 BufferBlock
OpDecorate %61 RelaxedPrecision OpMemberDecorate %52 0 Offset 0
OpDecorate %69 RelaxedPrecision OpDecorate %54 Binding 1
OpDecorate %76 RelaxedPrecision OpDecorate %54 DescriptorSet 0
OpDecorate %84 RelaxedPrecision OpDecorate %58 BuiltIn WorkgroupSize
OpDecorate %104 RelaxedPrecision %2 = OpTypeVoid
OpDecorate %112 RelaxedPrecision %3 = OpTypeFunction %2
OpDecorate %113 RelaxedPrecision %6 = OpTypeInt 32 0
OpDecorate %122 RelaxedPrecision %7 = OpTypePointer Function %6
OpDecorate %129 RelaxedPrecision %9 = OpTypeVector %6 3
OpDecorate %130 RelaxedPrecision %10 = OpTypePointer Input %9
OpDecorate %138 RelaxedPrecision %11 = OpVariable %10 Input
OpDecorate %139 RelaxedPrecision %12 = OpConstant %6 0
OpDecorate %148 RelaxedPrecision %13 = OpTypePointer Input %6
OpDecorate %155 RelaxedPrecision %16 = OpConstant %6 1
OpDecorate %156 RelaxedPrecision %20 = OpVariable %10 Input
OpDecorate %163 RelaxedPrecision %21 = OpConstant %6 2
OpDecorate %164 RelaxedPrecision %34 = OpTypeInt 32 1
OpDecorate %172 RelaxedPrecision %35 = OpTypePointer Function %34
OpDecorate %173 RelaxedPrecision %37 = OpTypeStruct %34
OpDecorate %183 RelaxedPrecision %38 = OpTypeRuntimeArray %37
OpDecorate %191 RelaxedPrecision %39 = OpTypeStruct %38
OpDecorate %192 RelaxedPrecision %40 = OpTypePointer Uniform %39
OpDecorate %200 RelaxedPrecision %41 = OpVariable %40 Uniform
OpDecorate %201 RelaxedPrecision %42 = OpConstant %34 0
OpDecorate %209 RelaxedPrecision %44 = OpTypePointer Uniform %34
OpDecorate %210 RelaxedPrecision %50 = OpTypeStruct %34
OpDecorate %219 RelaxedPrecision %51 = OpTypeRuntimeArray %50
OpDecorate %226 RelaxedPrecision %52 = OpTypeStruct %51
OpDecorate %227 RelaxedPrecision %53 = OpTypePointer Uniform %52
OpDecorate %234 RelaxedPrecision %54 = OpVariable %53 Uniform
OpDecorate %235 RelaxedPrecision %58 = OpConstantComposite %9 %16 %16 %16
OpDecorate %242 RelaxedPrecision %4 = OpFunction %2 None %3
OpDecorate %243 RelaxedPrecision %5 = OpLabel
OpDecorate %252 RelaxedPrecision %8 = OpVariable %7 Function
OpDecorate %259 RelaxedPrecision %36 = OpVariable %35 Function
OpDecorate %260 RelaxedPrecision %47 = OpVariable %35 Function
OpDecorate %267 RelaxedPrecision %14 = OpAccessChain %13 %11 %12
OpDecorate %268 RelaxedPrecision %15 = OpLoad %6 %14
OpDecorate %275 RelaxedPrecision %17 = OpAccessChain %13 %11 %16
OpDecorate %276 RelaxedPrecision %18 = OpLoad %6 %17
OpDecorate %_struct_310 Block %19 = OpIMul %6 %15 %18
OpMemberDecorate %_struct_310 0 BuiltIn Position %22 = OpAccessChain %13 %20 %21
OpMemberDecorate %_struct_310 1 BuiltIn PointSize %23 = OpLoad %6 %22
OpMemberDecorate %_struct_310 2 BuiltIn ClipDistance %24 = OpIMul %6 %19 %23
OpMemberDecorate %_struct_310 3 BuiltIn CullDistance %25 = OpAccessChain %13 %11 %12
OpDecorate %314 Location 0 %26 = OpLoad %6 %25
OpDecorate %318 RelaxedPrecision %27 = OpAccessChain %13 %20 %16
OpDecorate %_arr_mat2v3float_uint_3 ArrayStride 48 %28 = OpLoad %6 %27
OpDecorate %_struct_322 Block %29 = OpIMul %6 %26 %28
OpMemberDecorate %_struct_322 0 Offset 0 %30 = OpIAdd %6 %24 %29
OpMemberDecorate %_struct_322 1 RelaxedPrecision %31 = OpAccessChain %13 %20 %12
OpMemberDecorate %_struct_322 1 Offset 16 %32 = OpLoad %6 %31
OpMemberDecorate %_struct_322 2 RowMajor %33 = OpIAdd %6 %30 %32
OpMemberDecorate %_struct_322 2 MatrixStride 16 OpStore %8 %33
OpMemberDecorate %_struct_322 2 Offset 32 %43 = OpLoad %6 %8
OpMemberDecorate %_struct_322 3 RowMajor %45 = OpAccessChain %44 %41 %42 %43 %42
OpMemberDecorate %_struct_322 3 MatrixStride 16 %46 = OpLoad %34 %45
OpMemberDecorate %_struct_322 3 Offset 64 OpStore %36 %46
OpDecorate %324 Binding 0 %48 = OpLoad %34 %36
OpDecorate %324 DescriptorSet 0 %49 = OpExtInst %34 %1 SAbs %48
OpDecorate %334 RelaxedPrecision OpStore %47 %49
OpDecorate %335 RelaxedPrecision %55 = OpLoad %6 %8
OpDecorate %336 RelaxedPrecision %56 = OpLoad %34 %47
OpDecorate %341 RelaxedPrecision %57 = OpAccessChain %44 %54 %42 %55 %42
OpDecorate %343 RelaxedPrecision OpStore %57 %56
OpDecorate %344 RelaxedPrecision OpReturn
OpDecorate %345 RelaxedPrecision OpFunctionEnd
OpDecorate %353 RelaxedPrecision
OpDecorate %354 RelaxedPrecision
OpDecorate %355 RelaxedPrecision
OpDecorate %_struct_356 Block
OpMemberDecorate %_struct_356 0 RelaxedPrecision
OpMemberDecorate %_struct_356 0 Offset 0
OpDecorate %358 Binding 1
OpDecorate %358 DescriptorSet 0
OpDecorate %363 RelaxedPrecision
OpDecorate %365 RelaxedPrecision
OpDecorate %366 RelaxedPrecision
OpDecorate %367 RelaxedPrecision
OpDecorate %_struct_368 Block
OpMemberDecorate %_struct_368 0 RelaxedPrecision
OpMemberDecorate %_struct_368 0 Offset 0
OpDecorate %370 Binding 2
OpDecorate %370 DescriptorSet 0
OpDecorate %375 RelaxedPrecision
OpDecorate %377 RelaxedPrecision
OpDecorate %378 RelaxedPrecision
OpDecorate %379 RelaxedPrecision
OpDecorate %_struct_382 Block
OpMemberDecorate %_struct_382 0 ColMajor
OpMemberDecorate %_struct_382 0 MatrixStride 16
OpMemberDecorate %_struct_382 0 Offset 0
OpMemberDecorate %_struct_382 1 Offset 48
OpMemberDecorate %_struct_382 2 RelaxedPrecision
OpMemberDecorate %_struct_382 2 ColMajor
OpMemberDecorate %_struct_382 2 MatrixStride 16
OpMemberDecorate %_struct_382 2 Offset 64
OpMemberDecorate %_struct_382 3 RelaxedPrecision
OpMemberDecorate %_struct_382 3 ColMajor
OpMemberDecorate %_struct_382 3 MatrixStride 16
OpMemberDecorate %_struct_382 3 Offset 128
OpDecorate %384 Binding 3
OpDecorate %384 DescriptorSet 0
OpDecorate %390 RelaxedPrecision
OpDecorate %392 RelaxedPrecision
OpDecorate %393 RelaxedPrecision
OpDecorate %394 RelaxedPrecision
OpDecorate %400 RelaxedPrecision
OpDecorate %402 RelaxedPrecision
OpDecorate %403 RelaxedPrecision
OpDecorate %404 RelaxedPrecision
OpDecorate %406 RelaxedPrecision
OpDecorate %406 Location 0
OpDecorate %407 RelaxedPrecision
%void = OpTypeVoid
%3 = OpTypeFunction %void
%float = OpTypeFloat 32
%_ptr_Function_float = OpTypePointer Function %float
%8 = OpTypeFunction %float %_ptr_Function_float %_ptr_Function_float
%v2float = OpTypeVector %float 2
%_ptr_Function_v2float = OpTypePointer Function %v2float
%15 = OpTypeFunction %float %_ptr_Function_v2float %_ptr_Function_v2float
%v3float = OpTypeVector %float 3
%_ptr_Function_v3float = OpTypePointer Function %v3float
%22 = OpTypeFunction %float %_ptr_Function_v3float %_ptr_Function_v3float
%v4float = OpTypeVector %float 4
%_ptr_Function_v4float = OpTypePointer Function %v4float
%29 = OpTypeFunction %float %_ptr_Function_v4float %_ptr_Function_v4float
%mat4v2float = OpTypeMatrix %v2float 4
%_ptr_Function_mat4v2float = OpTypePointer Function %mat4v2float
%36 = OpTypeFunction %float %_ptr_Function_mat4v2float %_ptr_Function_mat4v2float
%mat4v3float = OpTypeMatrix %v3float 4
%_ptr_Function_mat4v3float = OpTypePointer Function %mat4v3float
%43 = OpTypeFunction %float %_ptr_Function_mat4v3float %_ptr_Function_mat4v3float
%mat4v4float = OpTypeMatrix %v4float 4
%_ptr_Function_mat4v4float = OpTypePointer Function %mat4v4float
%50 = OpTypeFunction %float %_ptr_Function_mat4v4float %_ptr_Function_mat4v4float
%int = OpTypeInt 32 1
%v2int = OpTypeVector %int 2
%_ptr_Function_v2int = OpTypePointer Function %v2int
%58 = OpTypeFunction %float %_ptr_Function_v2int %_ptr_Function_v2int
%uint = OpTypeInt 32 0
%v2uint = OpTypeVector %uint 2
%_ptr_Function_v2uint = OpTypePointer Function %v2uint
%66 = OpTypeFunction %float %_ptr_Function_v2uint %_ptr_Function_v2uint
%v4uint = OpTypeVector %uint 4
%_ptr_Function_v4uint = OpTypePointer Function %v4uint
%73 = OpTypeFunction %float %_ptr_Function_v4uint %_ptr_Function_v4uint
%bool = OpTypeBool
%v3bool = OpTypeVector %bool 3
%_ptr_Function_v3bool = OpTypePointer Function %v3bool
%81 = OpTypeFunction %float %_ptr_Function_v3bool %_ptr_Function_v3bool
%float_0_0500000007 = OpConstant %float 0.0500000007
%float_1 = OpConstant %float 1
%float_0 = OpConstant %float 0
%uint_0 = OpConstant %uint 0
%uint_1 = OpConstant %uint 1
%uint_2 = OpConstant %uint 2
%uint_3 = OpConstant %uint 3
%int_0 = OpConstant %int 0
%int_1 = OpConstant %int 1
%int_2 = OpConstant %int 2
%int_3 = OpConstant %int 3
%v2bool = OpTypeVector %bool 2
%v4bool = OpTypeVector %bool 4
%_arr_float_uint_1 = OpTypeArray %float %uint_1
%_struct_310 = OpTypeStruct %v4float %float %_arr_float_uint_1 %_arr_float_uint_1
%_ptr_Output__struct_310 = OpTypePointer Output %_struct_310
%312 = OpVariable %_ptr_Output__struct_310 Output
%_ptr_Input_v4float = OpTypePointer Input %v4float
%314 = OpVariable %_ptr_Input_v4float Input
%_ptr_Output_v4float = OpTypePointer Output %v4float
%v3uint = OpTypeVector %uint 3
%mat2v3float = OpTypeMatrix %v3float 2
%_arr_mat2v3float_uint_3 = OpTypeArray %mat2v3float %uint_3
%_struct_322 = OpTypeStruct %v3uint %v2uint %mat4v2float %_arr_mat2v3float_uint_3
%_ptr_Uniform__struct_322 = OpTypePointer Uniform %_struct_322
%324 = OpVariable %_ptr_Uniform__struct_322 Uniform
%false = OpConstantFalse %bool
%326 = OpConstantComposite %v3bool %false %false %false
%_ptr_Uniform_v3uint = OpTypePointer Uniform %v3uint
%331 = OpConstantComposite %v3uint %uint_0 %uint_0 %uint_0
%337 = OpConstantComposite %v2uint %uint_0 %uint_0
%_ptr_Uniform_v2uint = OpTypePointer Uniform %v2uint
%346 = OpConstantComposite %v2float %float_0 %float_0
%347 = OpConstantComposite %mat4v2float %346 %346 %346 %346
%_ptr_Uniform_mat4v2float = OpTypePointer Uniform %mat4v2float
%_struct_356 = OpTypeStruct %v2int
%_ptr_Uniform__struct_356 = OpTypePointer Uniform %_struct_356
%358 = OpVariable %_ptr_Uniform__struct_356 Uniform
%359 = OpConstantComposite %v2int %int_0 %int_0
%_ptr_Uniform_v2int = OpTypePointer Uniform %v2int
%_struct_368 = OpTypeStruct %v4uint
%_ptr_Uniform__struct_368 = OpTypePointer Uniform %_struct_368
%370 = OpVariable %_ptr_Uniform__struct_368 Uniform
%371 = OpConstantComposite %v4uint %uint_0 %uint_0 %uint_0 %uint_0
%_ptr_Uniform_v4uint = OpTypePointer Uniform %v4uint
%mat3v3float = OpTypeMatrix %v3float 3
%v3int = OpTypeVector %int 3
%_struct_382 = OpTypeStruct %mat3v3float %v3int %mat4v4float %mat4v3float
%_ptr_Uniform__struct_382 = OpTypePointer Uniform %_struct_382
%384 = OpVariable %_ptr_Uniform__struct_382 Uniform
%385 = OpConstantComposite %v4float %float_0 %float_0 %float_0 %float_0
%386 = OpConstantComposite %mat4v4float %385 %385 %385 %385
%_ptr_Uniform_mat4v4float = OpTypePointer Uniform %mat4v4float
%395 = OpConstantComposite %v3float %float_0 %float_0 %float_0
%396 = OpConstantComposite %mat4v3float %395 %395 %395 %395
%_ptr_Uniform_mat4v3float = OpTypePointer Uniform %mat4v3float
%_ptr_Output_float = OpTypePointer Output %float
%406 = OpVariable %_ptr_Output_float Output
%4 = OpFunction %void None %3
%5 = OpLabel
%318 = OpVariable %_ptr_Function_float Function
%327 = OpVariable %_ptr_Function_v3bool Function
%333 = OpVariable %_ptr_Function_v3bool Function
%338 = OpVariable %_ptr_Function_v2uint Function
%342 = OpVariable %_ptr_Function_v2uint Function
%348 = OpVariable %_ptr_Function_mat4v2float Function
%352 = OpVariable %_ptr_Function_mat4v2float Function
%360 = OpVariable %_ptr_Function_v2int Function
%364 = OpVariable %_ptr_Function_v2int Function
%372 = OpVariable %_ptr_Function_v4uint Function
%376 = OpVariable %_ptr_Function_v4uint Function
%387 = OpVariable %_ptr_Function_mat4v4float Function
%391 = OpVariable %_ptr_Function_mat4v4float Function
%397 = OpVariable %_ptr_Function_mat4v3float Function
%401 = OpVariable %_ptr_Function_mat4v3float Function
%315 = OpLoad %v4float %314
%317 = OpAccessChain %_ptr_Output_v4float %312 %int_0
OpStore %317 %315
OpStore %318 %float_1
%329 = OpAccessChain %_ptr_Uniform_v3uint %324 %int_0
%330 = OpLoad %v3uint %329
%332 = OpINotEqual %v3bool %330 %331
OpStore %327 %332
OpStore %333 %326
%334 = OpFunctionCall %float %84 %327 %333
%335 = OpLoad %float %318
%336 = OpFMul %float %335 %334
OpStore %318 %336
%340 = OpAccessChain %_ptr_Uniform_v2uint %324 %int_1
%341 = OpLoad %v2uint %340
OpStore %338 %341
OpStore %342 %337
%343 = OpFunctionCall %float %69 %338 %342
%344 = OpLoad %float %318
%345 = OpFMul %float %344 %343
OpStore %318 %345
%350 = OpAccessChain %_ptr_Uniform_mat4v2float %324 %int_2
%351 = OpLoad %mat4v2float %350
OpStore %348 %351
OpStore %352 %347
%353 = OpFunctionCall %float %39 %348 %352
%354 = OpLoad %float %318
%355 = OpFMul %float %354 %353
OpStore %318 %355
%362 = OpAccessChain %_ptr_Uniform_v2int %358 %int_0
%363 = OpLoad %v2int %362
OpStore %360 %363
OpStore %364 %359
%365 = OpFunctionCall %float %61 %360 %364
%366 = OpLoad %float %318
%367 = OpFMul %float %366 %365
OpStore %318 %367
%374 = OpAccessChain %_ptr_Uniform_v4uint %370 %int_0
%375 = OpLoad %v4uint %374
OpStore %372 %375
OpStore %376 %371
%377 = OpFunctionCall %float %76 %372 %376
%378 = OpLoad %float %318
%379 = OpFMul %float %378 %377
OpStore %318 %379
%389 = OpAccessChain %_ptr_Uniform_mat4v4float %384 %int_2
%390 = OpLoad %mat4v4float %389
OpStore %387 %390
OpStore %391 %386
%392 = OpFunctionCall %float %53 %387 %391
%393 = OpLoad %float %318
%394 = OpFMul %float %393 %392
OpStore %318 %394
%399 = OpAccessChain %_ptr_Uniform_mat4v3float %384 %int_3
%400 = OpLoad %mat4v3float %399
OpStore %397 %400
OpStore %401 %396
%402 = OpFunctionCall %float %46 %397 %401
%403 = OpLoad %float %318
%404 = OpFMul %float %403 %402
OpStore %318 %404
%407 = OpLoad %float %318
OpStore %406 %407
OpReturn
OpFunctionEnd
%11 = OpFunction %float None %8
%9 = OpFunctionParameter %_ptr_Function_float
%10 = OpFunctionParameter %_ptr_Function_float
%12 = OpLabel
%86 = OpLoad %float %9
%87 = OpLoad %float %10
%88 = OpFSub %float %86 %87
%89 = OpExtInst %float %1 FAbs %88
%91 = OpFOrdLessThan %bool %89 %float_0_0500000007
%94 = OpSelect %float %91 %float_1 %float_0
OpReturnValue %94
OpFunctionEnd
%18 = OpFunction %float None %15
%16 = OpFunctionParameter %_ptr_Function_v2float
%17 = OpFunctionParameter %_ptr_Function_v2float
%19 = OpLabel
%97 = OpVariable %_ptr_Function_float Function
%101 = OpVariable %_ptr_Function_float Function
%105 = OpVariable %_ptr_Function_float Function
%109 = OpVariable %_ptr_Function_float Function
%99 = OpAccessChain %_ptr_Function_float %16 %uint_0
%100 = OpLoad %float %99
OpStore %97 %100
%102 = OpAccessChain %_ptr_Function_float %17 %uint_0
%103 = OpLoad %float %102
OpStore %101 %103
%104 = OpFunctionCall %float %11 %97 %101
%107 = OpAccessChain %_ptr_Function_float %16 %uint_1
%108 = OpLoad %float %107
OpStore %105 %108
%110 = OpAccessChain %_ptr_Function_float %17 %uint_1
%111 = OpLoad %float %110
OpStore %109 %111
%112 = OpFunctionCall %float %11 %105 %109
%113 = OpFMul %float %104 %112
OpReturnValue %113
OpFunctionEnd
%25 = OpFunction %float None %22
%23 = OpFunctionParameter %_ptr_Function_v3float
%24 = OpFunctionParameter %_ptr_Function_v3float
%26 = OpLabel
%116 = OpVariable %_ptr_Function_float Function
%119 = OpVariable %_ptr_Function_float Function
%123 = OpVariable %_ptr_Function_float Function
%126 = OpVariable %_ptr_Function_float Function
%131 = OpVariable %_ptr_Function_float Function
%135 = OpVariable %_ptr_Function_float Function
%117 = OpAccessChain %_ptr_Function_float %23 %uint_0
%118 = OpLoad %float %117
OpStore %116 %118
%120 = OpAccessChain %_ptr_Function_float %24 %uint_0
%121 = OpLoad %float %120
OpStore %119 %121
%122 = OpFunctionCall %float %11 %116 %119
%124 = OpAccessChain %_ptr_Function_float %23 %uint_1
%125 = OpLoad %float %124
OpStore %123 %125
%127 = OpAccessChain %_ptr_Function_float %24 %uint_1
%128 = OpLoad %float %127
OpStore %126 %128
%129 = OpFunctionCall %float %11 %123 %126
%130 = OpFMul %float %122 %129
%133 = OpAccessChain %_ptr_Function_float %23 %uint_2
%134 = OpLoad %float %133
OpStore %131 %134
%136 = OpAccessChain %_ptr_Function_float %24 %uint_2
%137 = OpLoad %float %136
OpStore %135 %137
%138 = OpFunctionCall %float %11 %131 %135
%139 = OpFMul %float %130 %138
OpReturnValue %139
OpFunctionEnd
%32 = OpFunction %float None %29
%30 = OpFunctionParameter %_ptr_Function_v4float
%31 = OpFunctionParameter %_ptr_Function_v4float
%33 = OpLabel
%142 = OpVariable %_ptr_Function_float Function
%145 = OpVariable %_ptr_Function_float Function
%149 = OpVariable %_ptr_Function_float Function
%152 = OpVariable %_ptr_Function_float Function
%157 = OpVariable %_ptr_Function_float Function
%160 = OpVariable %_ptr_Function_float Function
%165 = OpVariable %_ptr_Function_float Function
%169 = OpVariable %_ptr_Function_float Function
%143 = OpAccessChain %_ptr_Function_float %30 %uint_0
%144 = OpLoad %float %143
OpStore %142 %144
%146 = OpAccessChain %_ptr_Function_float %31 %uint_0
%147 = OpLoad %float %146
OpStore %145 %147
%148 = OpFunctionCall %float %11 %142 %145
%150 = OpAccessChain %_ptr_Function_float %30 %uint_1
%151 = OpLoad %float %150
OpStore %149 %151
%153 = OpAccessChain %_ptr_Function_float %31 %uint_1
%154 = OpLoad %float %153
OpStore %152 %154
%155 = OpFunctionCall %float %11 %149 %152
%156 = OpFMul %float %148 %155
%158 = OpAccessChain %_ptr_Function_float %30 %uint_2
%159 = OpLoad %float %158
OpStore %157 %159
%161 = OpAccessChain %_ptr_Function_float %31 %uint_2
%162 = OpLoad %float %161
OpStore %160 %162
%163 = OpFunctionCall %float %11 %157 %160
%164 = OpFMul %float %156 %163
%167 = OpAccessChain %_ptr_Function_float %30 %uint_3
%168 = OpLoad %float %167
OpStore %165 %168
%170 = OpAccessChain %_ptr_Function_float %31 %uint_3
%171 = OpLoad %float %170
OpStore %169 %171
%172 = OpFunctionCall %float %11 %165 %169
%173 = OpFMul %float %164 %172
OpReturnValue %173
OpFunctionEnd
%39 = OpFunction %float None %36
%37 = OpFunctionParameter %_ptr_Function_mat4v2float
%38 = OpFunctionParameter %_ptr_Function_mat4v2float
%40 = OpLabel
%177 = OpVariable %_ptr_Function_v2float Function
%180 = OpVariable %_ptr_Function_v2float Function
%185 = OpVariable %_ptr_Function_v2float Function
%188 = OpVariable %_ptr_Function_v2float Function
%194 = OpVariable %_ptr_Function_v2float Function
%197 = OpVariable %_ptr_Function_v2float Function
%203 = OpVariable %_ptr_Function_v2float Function
%206 = OpVariable %_ptr_Function_v2float Function
%178 = OpAccessChain %_ptr_Function_v2float %37 %int_0
%179 = OpLoad %v2float %178
OpStore %177 %179
%181 = OpAccessChain %_ptr_Function_v2float %38 %int_0
%182 = OpLoad %v2float %181
OpStore %180 %182
%183 = OpFunctionCall %float %18 %177 %180
%186 = OpAccessChain %_ptr_Function_v2float %37 %int_1
%187 = OpLoad %v2float %186
OpStore %185 %187
%189 = OpAccessChain %_ptr_Function_v2float %38 %int_1
%190 = OpLoad %v2float %189
OpStore %188 %190
%191 = OpFunctionCall %float %18 %185 %188
%192 = OpFMul %float %183 %191
%195 = OpAccessChain %_ptr_Function_v2float %37 %int_2
%196 = OpLoad %v2float %195
OpStore %194 %196
%198 = OpAccessChain %_ptr_Function_v2float %38 %int_2
%199 = OpLoad %v2float %198
OpStore %197 %199
%200 = OpFunctionCall %float %18 %194 %197
%201 = OpFMul %float %192 %200
%204 = OpAccessChain %_ptr_Function_v2float %37 %int_3
%205 = OpLoad %v2float %204
OpStore %203 %205
%207 = OpAccessChain %_ptr_Function_v2float %38 %int_3
%208 = OpLoad %v2float %207
OpStore %206 %208
%209 = OpFunctionCall %float %18 %203 %206
%210 = OpFMul %float %201 %209
OpReturnValue %210
OpFunctionEnd
%46 = OpFunction %float None %43
%44 = OpFunctionParameter %_ptr_Function_mat4v3float
%45 = OpFunctionParameter %_ptr_Function_mat4v3float
%47 = OpLabel
%213 = OpVariable %_ptr_Function_v3float Function
%216 = OpVariable %_ptr_Function_v3float Function
%220 = OpVariable %_ptr_Function_v3float Function
%223 = OpVariable %_ptr_Function_v3float Function
%228 = OpVariable %_ptr_Function_v3float Function
%231 = OpVariable %_ptr_Function_v3float Function
%236 = OpVariable %_ptr_Function_v3float Function
%239 = OpVariable %_ptr_Function_v3float Function
%214 = OpAccessChain %_ptr_Function_v3float %44 %int_0
%215 = OpLoad %v3float %214
OpStore %213 %215
%217 = OpAccessChain %_ptr_Function_v3float %45 %int_0
%218 = OpLoad %v3float %217
OpStore %216 %218
%219 = OpFunctionCall %float %25 %213 %216
%221 = OpAccessChain %_ptr_Function_v3float %44 %int_1
%222 = OpLoad %v3float %221
OpStore %220 %222
%224 = OpAccessChain %_ptr_Function_v3float %45 %int_1
%225 = OpLoad %v3float %224
OpStore %223 %225
%226 = OpFunctionCall %float %25 %220 %223
%227 = OpFMul %float %219 %226
%229 = OpAccessChain %_ptr_Function_v3float %44 %int_2
%230 = OpLoad %v3float %229
OpStore %228 %230
%232 = OpAccessChain %_ptr_Function_v3float %45 %int_2
%233 = OpLoad %v3float %232
OpStore %231 %233
%234 = OpFunctionCall %float %25 %228 %231
%235 = OpFMul %float %227 %234
%237 = OpAccessChain %_ptr_Function_v3float %44 %int_3
%238 = OpLoad %v3float %237
OpStore %236 %238
%240 = OpAccessChain %_ptr_Function_v3float %45 %int_3
%241 = OpLoad %v3float %240
OpStore %239 %241
%242 = OpFunctionCall %float %25 %236 %239
%243 = OpFMul %float %235 %242
OpReturnValue %243
OpFunctionEnd
%53 = OpFunction %float None %50
%51 = OpFunctionParameter %_ptr_Function_mat4v4float
%52 = OpFunctionParameter %_ptr_Function_mat4v4float
%54 = OpLabel
%246 = OpVariable %_ptr_Function_v4float Function
%249 = OpVariable %_ptr_Function_v4float Function
%253 = OpVariable %_ptr_Function_v4float Function
%256 = OpVariable %_ptr_Function_v4float Function
%261 = OpVariable %_ptr_Function_v4float Function
%264 = OpVariable %_ptr_Function_v4float Function
%269 = OpVariable %_ptr_Function_v4float Function
%272 = OpVariable %_ptr_Function_v4float Function
%247 = OpAccessChain %_ptr_Function_v4float %51 %int_0
%248 = OpLoad %v4float %247
OpStore %246 %248
%250 = OpAccessChain %_ptr_Function_v4float %52 %int_0
%251 = OpLoad %v4float %250
OpStore %249 %251
%252 = OpFunctionCall %float %32 %246 %249
%254 = OpAccessChain %_ptr_Function_v4float %51 %int_1
%255 = OpLoad %v4float %254
OpStore %253 %255
%257 = OpAccessChain %_ptr_Function_v4float %52 %int_1
%258 = OpLoad %v4float %257
OpStore %256 %258
%259 = OpFunctionCall %float %32 %253 %256
%260 = OpFMul %float %252 %259
%262 = OpAccessChain %_ptr_Function_v4float %51 %int_2
%263 = OpLoad %v4float %262
OpStore %261 %263
%265 = OpAccessChain %_ptr_Function_v4float %52 %int_2
%266 = OpLoad %v4float %265
OpStore %264 %266
%267 = OpFunctionCall %float %32 %261 %264
%268 = OpFMul %float %260 %267
%270 = OpAccessChain %_ptr_Function_v4float %51 %int_3
%271 = OpLoad %v4float %270
OpStore %269 %271
%273 = OpAccessChain %_ptr_Function_v4float %52 %int_3
%274 = OpLoad %v4float %273
OpStore %272 %274
%275 = OpFunctionCall %float %32 %269 %272
%276 = OpFMul %float %268 %275
OpReturnValue %276
OpFunctionEnd
%61 = OpFunction %float None %58
%59 = OpFunctionParameter %_ptr_Function_v2int
%60 = OpFunctionParameter %_ptr_Function_v2int
%62 = OpLabel
%279 = OpLoad %v2int %59
%280 = OpLoad %v2int %60
%282 = OpIEqual %v2bool %279 %280
%283 = OpAll %bool %282
%284 = OpSelect %float %283 %float_1 %float_0
OpReturnValue %284
OpFunctionEnd
%69 = OpFunction %float None %66
%67 = OpFunctionParameter %_ptr_Function_v2uint
%68 = OpFunctionParameter %_ptr_Function_v2uint
%70 = OpLabel
%287 = OpLoad %v2uint %67
%288 = OpLoad %v2uint %68
%289 = OpIEqual %v2bool %287 %288
%290 = OpAll %bool %289
%291 = OpSelect %float %290 %float_1 %float_0
OpReturnValue %291
OpFunctionEnd
%76 = OpFunction %float None %73
%74 = OpFunctionParameter %_ptr_Function_v4uint
%75 = OpFunctionParameter %_ptr_Function_v4uint
%77 = OpLabel
%294 = OpLoad %v4uint %74
%295 = OpLoad %v4uint %75
%297 = OpIEqual %v4bool %294 %295
%298 = OpAll %bool %297
%299 = OpSelect %float %298 %float_1 %float_0
OpReturnValue %299
OpFunctionEnd
%84 = OpFunction %float None %81
%82 = OpFunctionParameter %_ptr_Function_v3bool
%83 = OpFunctionParameter %_ptr_Function_v3bool
%85 = OpLabel
%302 = OpLoad %v3bool %82
%303 = OpLoad %v3bool %83
%304 = OpLogicalEqual %v3bool %302 %303
%305 = OpAll %bool %304
%306 = OpSelect %float %305 %float_1 %float_0
OpReturnValue %306
OpFunctionEnd

View File

@@ -30,13 +30,6 @@ const SpvEntryPoint = struct {
globals: []SpvWord, globals: []SpvWord,
}; };
const SpvSource = struct {
file_name: []const u8,
lang: spv.SpvSourceLanguage,
lang_version: SpvWord,
source: []const u8,
};
pub const ModuleError = error{ pub const ModuleError = error{
InvalidSpirV, InvalidSpirV,
InvalidMagic, InvalidMagic,
@@ -62,7 +55,6 @@ code: []const SpvWord,
addressing: spv.SpvAddressingModel, addressing: spv.SpvAddressingModel,
memory_model: spv.SpvMemoryModel, memory_model: spv.SpvMemoryModel,
files: std.ArrayList(SpvSource),
extensions: std.ArrayList([]const u8), extensions: std.ArrayList([]const u8),
results: []Result, results: []Result,
@@ -79,25 +71,21 @@ geometry_output_count: SpvWord,
geometry_input: SpvWord, geometry_input: SpvWord,
geometry_output: SpvWord, geometry_output: SpvWord,
input_locations: std.ArrayList(SpvWord), input_locations: [lib.SPIRV_MAX_INPUT_LOCATIONS]SpvWord,
output_locations: std.ArrayList(SpvWord), output_locations: [lib.SPIRV_MAX_OUTPUT_LOCATIONS]SpvWord,
bindings: std.AutoHashMap(SpvBinding, Value), bindings: [lib.SPIRV_MAX_SET][lib.SPIRV_MAX_SET_BINDINGS]SpvWord,
push_constants: []Value, push_constants: []Value,
pub fn init(allocator: std.mem.Allocator, source: []const SpvWord, options: ModuleOptions) ModuleError!Self { pub fn init(allocator: std.mem.Allocator, source: []const SpvWord, options: ModuleOptions) ModuleError!Self {
var self: Self = std.mem.zeroInit(Self, .{ var self: Self = std.mem.zeroInit(Self, .{
.options = options, .options = options,
.code = allocator.dupe(SpvWord, source) catch return ModuleError.OutOfMemory, .code = allocator.dupe(SpvWord, source) catch return ModuleError.OutOfMemory,
.files = std.ArrayList(SpvSource).empty,
.extensions = std.ArrayList([]const u8).empty, .extensions = std.ArrayList([]const u8).empty,
.entry_points = std.ArrayList(SpvEntryPoint).empty, .entry_points = std.ArrayList(SpvEntryPoint).empty,
.capabilities = std.EnumSet(spv.SpvCapability).initEmpty(), .capabilities = std.EnumSet(spv.SpvCapability).initEmpty(),
.local_size_x = 1, .local_size_x = 1,
.local_size_y = 1, .local_size_y = 1,
.local_size_z = 1, .local_size_z = 1,
.input_locations = std.ArrayList(SpvWord).empty,
.output_locations = std.ArrayList(SpvWord).empty,
.bindings = std.AutoHashMap(SpvBinding, Value).init(allocator),
}); });
errdefer allocator.free(self.code); errdefer allocator.free(self.code);
@@ -137,7 +125,7 @@ pub fn init(allocator: std.mem.Allocator, source: []const SpvWord, options: Modu
_ = self.it.skip(); // Skip schema _ = self.it.skip(); // Skip schema
try self.pass(allocator); // Setup pass try self.pass(allocator); // Setup pass
try self.populateMaps(allocator); try self.populateMaps();
if (std.process.hasEnvVarConstant("SPIRV_INTERPRETER_DEBUG_LOGS")) { if (std.process.hasEnvVarConstant("SPIRV_INTERPRETER_DEBUG_LOGS")) {
var capability_set_names: std.ArrayList([]const u8) = .empty; var capability_set_names: std.ArrayList([]const u8) = .empty;
@@ -218,34 +206,46 @@ fn pass(self: *Self, allocator: std.mem.Allocator) ModuleError!void {
} }
} }
fn populateMaps(self: *Self, allocator: std.mem.Allocator) ModuleError!void { fn populateMaps(self: *Self) ModuleError!void {
for (self.results, 0..) |result, id| { for (self.results, 0..) |result, id| {
if (result.variant == null or std.meta.activeTag(result.variant.?) != .Variable) continue; if (result.variant == null or std.meta.activeTag(result.variant.?) != .Variable)
switch (result.variant.?.Variable.storage_class) { continue;
.Input => for (result.decorations.items) |decoration| switch (decoration.rtype) {
.Location => self.input_locations.append(allocator, @intCast(id)) catch return ModuleError.OutOfMemory, var current_set: usize = 0;
for (result.decorations.items) |decoration| {
switch (result.variant.?.Variable.storage_class) {
.Input => {
if (decoration.rtype == .Location)
self.input_locations[decoration.literal_1] = @intCast(id);
},
.Output => {
if (decoration.rtype == .Location)
self.output_locations[decoration.literal_1] = @intCast(id);
},
.StorageBuffer,
.Uniform,
.UniformConstant,
=> {
switch (decoration.rtype) {
.Binding => self.bindings[current_set][decoration.literal_1] = @intCast(id),
.DescriptorSet => current_set = decoration.literal_1,
else => {},
}
},
else => {}, else => {},
}, }
.Output => for (result.decorations.items) |decoration| switch (decoration.rtype) {
.Location => self.output_locations.append(allocator, @intCast(id)) catch return ModuleError.OutOfMemory,
else => {},
},
else => {},
} }
} }
} }
pub fn deinit(self: *Self, allocator: std.mem.Allocator) void { pub fn deinit(self: *Self, allocator: std.mem.Allocator) void {
allocator.free(self.code); allocator.free(self.code);
self.input_locations.deinit(allocator);
self.output_locations.deinit(allocator);
self.bindings.deinit();
for (self.entry_points.items) |entry| { for (self.entry_points.items) |entry| {
allocator.free(entry.name); allocator.free(entry.name);
allocator.free(entry.globals); allocator.free(entry.globals);
} }
self.entry_points.deinit(allocator); self.entry_points.deinit(allocator);
self.files.deinit(allocator);
for (self.extensions.items) |ext| { for (self.extensions.items) |ext| {
allocator.free(ext); allocator.free(ext);

View File

@@ -107,17 +107,23 @@ pub const Value = union(Type) {
Vector2u32: Vec2u32, Vector2u32: Vec2u32,
Matrix: []Value, Matrix: []Value,
Array: []Value, Array: []Value,
RuntimeArray: struct {}, RuntimeArray: ?[]Value,
Structure: []Value, Structure: []Value,
Function: noreturn, Function: noreturn,
Image: struct {}, Image: struct {},
Sampler: struct {}, Sampler: struct {},
SampledImage: struct {}, SampledImage: struct {},
Pointer: noreturn, Pointer: union(enum) {
common: *Value,
f32_ptr: *f32,
i32_ptr: *i32, //< For vectors specializations
u32_ptr: *u32,
},
pub inline fn getCompositeDataOrNull(self: *const Value) ?[]Value { pub inline fn getCompositeDataOrNull(self: *const Value) ?[]Value {
return switch (self.*) { return switch (self.*) {
.Vector, .Matrix, .Array, .Structure => |v| v, .Vector, .Matrix, .Array, .Structure => |v| v,
.RuntimeArray => |v| v,
else => null, else => null,
}; };
} }
@@ -176,7 +182,7 @@ pub const Value = union(Type) {
} }
break :blk self; break :blk self;
}, },
.RuntimeArray => .{ .RuntimeArray = .{} }, .RuntimeArray => .{ .RuntimeArray = null },
else => unreachable, else => unreachable,
}, },
else => unreachable, else => unreachable,
@@ -207,6 +213,17 @@ pub const Value = union(Type) {
break :blk values; break :blk values;
}, },
}, },
.RuntimeArray => |opt_a| .{
.RuntimeArray = blk: {
if (opt_a) |a| {
const values = allocator.dupe(Value, a) catch return RuntimeError.OutOfMemory;
for (values, a) |*new_value, value| new_value.* = try value.dupe(allocator);
break :blk values;
} else {
break :blk null;
}
},
},
.Structure => |s| .{ .Structure => |s| .{
.Structure = blk: { .Structure = blk: {
const values = allocator.dupe(Value, s) catch return RuntimeError.OutOfMemory; const values = allocator.dupe(Value, s) catch return RuntimeError.OutOfMemory;
@@ -224,6 +241,10 @@ pub const Value = union(Type) {
for (values) |*value| value.deinit(allocator); for (values) |*value| value.deinit(allocator);
allocator.free(values); allocator.free(values);
}, },
.RuntimeArray => |opt_values| if (opt_values) |values| {
for (values) |*value| value.deinit(allocator);
allocator.free(values);
},
else => {}, else => {},
} }
} }
@@ -263,7 +284,10 @@ pub const TypeData = union(Type) {
components_type: Type, components_type: Type,
member_count: SpvWord, member_count: SpvWord,
}, },
RuntimeArray: struct {}, RuntimeArray: struct {
components_type_word: SpvWord,
components_type: Type,
},
Structure: struct { Structure: struct {
members_type_word: []const SpvWord, members_type_word: []const SpvWord,
member_names: std.ArrayList([]const u8), member_names: std.ArrayList([]const u8),
@@ -308,7 +332,7 @@ pub const VariantData = union(Variant) {
}, },
AccessChain: struct { AccessChain: struct {
target: SpvWord, target: SpvWord,
value: *Value, value: Value,
}, },
FunctionParameter: struct { FunctionParameter: struct {
type_word: SpvWord, type_word: SpvWord,
@@ -353,7 +377,6 @@ pub fn deinit(self: *Self, allocator: std.mem.Allocator) void {
}, },
.Constant => |*c| c.value.deinit(allocator), .Constant => |*c| c.value.deinit(allocator),
.Variable => |*v| v.value.deinit(allocator), .Variable => |*v| v.value.deinit(allocator),
//.AccessChain => |*a| a.value.deinit(allocator),
.Function => |f| allocator.free(f.params), .Function => |f| allocator.free(f.params),
else => {}, else => {},
} }
@@ -365,7 +388,7 @@ pub inline fn getValueTypeWord(self: *Self) RuntimeError!SpvWord {
return switch ((try self.getVariant()).*) { return switch ((try self.getVariant()).*) {
.Variable => |v| v.type_word, .Variable => |v| v.type_word,
.Constant => |c| c.type_word, .Constant => |c| c.type_word,
.AccessChain => |*a| a.target, .AccessChain => |a| a.target,
.FunctionParameter => |p| p.type_word, .FunctionParameter => |p| p.type_word,
else => RuntimeError.InvalidSpirV, else => RuntimeError.InvalidSpirV,
}; };
@@ -384,12 +407,22 @@ pub inline fn getValue(self: *Self) RuntimeError!*Value {
return switch ((try self.getVariant()).*) { return switch ((try self.getVariant()).*) {
.Variable => |*v| &v.value, .Variable => |*v| &v.value,
.Constant => |*c| &c.value, .Constant => |*c| &c.value,
.AccessChain => |a| a.value, .AccessChain => |*a| &a.value,
.FunctionParameter => |*p| p.value_ptr orelse return RuntimeError.InvalidSpirV, .FunctionParameter => |*p| p.value_ptr orelse return RuntimeError.InvalidSpirV,
else => RuntimeError.InvalidSpirV, else => RuntimeError.InvalidSpirV,
}; };
} }
pub inline fn getConstValue(self: *Self) RuntimeError!*const Value {
return switch ((try self.getVariant()).*) {
.Variable => |v| &v.value,
.Constant => |c| &c.value,
.AccessChain => |a| &a.value,
.FunctionParameter => |p| p.value_ptr orelse return RuntimeError.InvalidSpirV,
else => RuntimeError.InvalidSpirV,
};
}
pub inline fn getVariant(self: *Self) RuntimeError!*VariantData { pub inline fn getVariant(self: *Self) RuntimeError!*VariantData {
return &(self.variant orelse return RuntimeError.InvalidSpirV); return &(self.variant orelse return RuntimeError.InvalidSpirV);
} }
@@ -561,6 +594,18 @@ pub fn initValue(allocator: std.mem.Allocator, member_count: usize, results: []c
} }
break :blk value; break :blk value;
}, },
.RuntimeArray => |a| blk: {
std.debug.print("test {d}\n", .{member_count});
if (member_count == 0) {
break :blk Value{ .RuntimeArray = null };
}
const value: Value = .{ .RuntimeArray = allocator.alloc(Value, member_count) catch return RuntimeError.OutOfMemory };
errdefer allocator.free(value.RuntimeArray.?);
for (value.RuntimeArray.?) |*val| {
val.* = try Value.init(allocator, results, a.components_type_word);
}
break :blk value;
},
.Structure => |s| blk: { .Structure => |s| blk: {
const value: Value = .{ .Structure = allocator.alloc(Value, member_count) catch return RuntimeError.OutOfMemory }; const value: Value = .{ .Structure = allocator.alloc(Value, member_count) catch return RuntimeError.OutOfMemory };
errdefer allocator.free(value.Structure); errdefer allocator.free(value.Structure);

View File

@@ -3,6 +3,7 @@
const std = @import("std"); const std = @import("std");
const spv = @import("spv.zig"); const spv = @import("spv.zig");
const op = @import("opcodes.zig"); const op = @import("opcodes.zig");
const lib = @import("lib.zig");
const SpvVoid = spv.SpvVoid; const SpvVoid = spv.SpvVoid;
const SpvByte = spv.SpvByte; const SpvByte = spv.SpvByte;
@@ -150,7 +151,7 @@ pub fn callEntryPoint(self: *Self, allocator: std.mem.Allocator, entry_point_ind
} }
pub fn readOutput(self: *const Self, comptime T: type, output: []T, result: SpvWord) RuntimeError!void { pub fn readOutput(self: *const Self, comptime T: type, output: []T, result: SpvWord) RuntimeError!void {
if (std.mem.indexOf(SpvWord, self.mod.output_locations.items, &.{result})) |_| { if (std.mem.indexOfScalar(SpvWord, &self.mod.output_locations, result)) |_| {
try self.readValue(T, output, &self.results[result].variant.?.Variable.value); try self.readValue(T, output, &self.results[result].variant.?.Variable.value);
} else { } else {
return RuntimeError.NotFound; return RuntimeError.NotFound;
@@ -158,13 +159,39 @@ pub fn readOutput(self: *const Self, comptime T: type, output: []T, result: SpvW
} }
pub fn writeInput(self: *const Self, comptime T: type, input: []const T, result: SpvWord) RuntimeError!void { pub fn writeInput(self: *const Self, comptime T: type, input: []const T, result: SpvWord) RuntimeError!void {
if (std.mem.indexOf(SpvWord, self.mod.input_locations.items, &.{result})) |_| { if (std.mem.indexOfScalar(SpvWord, &self.mod.input_locations, result)) |_| {
try self.writeValue(T, input, &self.results[result].variant.?.Variable.value); try self.writeValue(T, input, &self.results[result].variant.?.Variable.value);
} else { } else {
return RuntimeError.NotFound; return RuntimeError.NotFound;
} }
} }
pub fn readDescriptorSet(self: *const Self, comptime T: type, output: *T, set: SpvWord, binding: SpvWord) RuntimeError!void {
if (set < lib.SPIRV_MAX_SET and binding < lib.SPIRV_MAX_SET_BINDINGS) {
try self.readValue(T, output, &self.results[self.mod.bindings[set][binding]].variant.?.Variable.value);
} else {
return RuntimeError.NotFound;
}
}
pub fn writeDescriptorSet(self: *const Self, comptime T: type, allocator: std.mem.Allocator, input: *const T, set: SpvWord, binding: SpvWord) RuntimeError!void {
if (set < lib.SPIRV_MAX_SET and binding < lib.SPIRV_MAX_SET_BINDINGS) {
const variable = &self.results[self.mod.bindings[set][binding]].variant.?.Variable;
switch (variable.value) {
.RuntimeArray => {
const resolved = self.results[variable.type_word].resolveType(self.results);
variable.value = try Result.initValue(allocator, input.len, self.results, resolved);
},
.Vector, .Matrix, .Array, .Structure => |v| {
},
}
try self.writeValue(T, input, &variable.value);
} else {
return RuntimeError.NotFound;
}
}
fn reset(self: *Self) void { fn reset(self: *Self) void {
self.function_stack.clearRetainingCapacity(); self.function_stack.clearRetainingCapacity();
self.current_function = null; self.current_function = null;
@@ -236,7 +263,15 @@ fn readValue(self: *const Self, comptime T: type, output: []T, value: *const Res
u32 => output[i] = vec[i], u32 => output[i] = vec[i],
inline else => return RuntimeError.InvalidValueType, inline else => return RuntimeError.InvalidValueType,
}, },
.Vector, .Matrix, .Array, .Structure => |values| for (values, 0..) |v, i| try self.readValue(T, output[i..], &v), .Array,
.Matrix,
.Structure,
.Vector,
=> |values| for (values, 0..) |v, i| try self.readValue(T, output[i..], &v),
.RuntimeArray => |opt_values| if (opt_values) |values| {
for (values, 0..) |v, i|
try self.readValue(T, output[i..], &v);
},
else => return RuntimeError.InvalidValueType, else => return RuntimeError.InvalidValueType,
} }
} }
@@ -307,7 +342,15 @@ fn writeValue(self: *const Self, comptime T: type, input: []const T, value: *Res
u32 => vec[i] = input[i], u32 => vec[i] = input[i],
inline else => return RuntimeError.InvalidValueType, inline else => return RuntimeError.InvalidValueType,
}, },
.Vector, .Matrix, .Array, .Structure => |*values| for (values.*, 0..) |*v, i| try self.writeValue(T, input[i..], v), .Array,
.Matrix,
.Structure,
.Vector,
=> |*values| for (values.*, 0..) |*v, i| try self.writeValue(T, input[i..], v),
.RuntimeArray => |opt_values| if (opt_values) |*values| {
for (values.*, 0..) |*v, i|
try self.writeValue(T, input[i..], v);
},
else => return RuntimeError.InvalidValueType, else => return RuntimeError.InvalidValueType,
} }
} }

View File

@@ -43,3 +43,15 @@ pub const SpvWord = spv.SpvWord;
pub const SpvBool = spv.SpvBool; pub const SpvBool = spv.SpvBool;
pub const GLSL_std_450 = @import("GLSL_std_450/opcodes.zig"); pub const GLSL_std_450 = @import("GLSL_std_450/opcodes.zig");
/// Maximum number of input locations per module
pub const SPIRV_MAX_INPUT_LOCATIONS: usize = 32;
/// Maximum number of output locations per module
pub const SPIRV_MAX_OUTPUT_LOCATIONS: usize = 32;
/// Maximum number of descriptor set per module
pub const SPIRV_MAX_SET: usize = 32;
/// Maximum number of bindings per descriptor set
pub const SPIRV_MAX_SET_BINDINGS: usize = 32;

View File

@@ -149,7 +149,6 @@ pub const SetupDispatcher = block: {
.ShiftLeftLogical = autoSetupConstant, .ShiftLeftLogical = autoSetupConstant,
.ShiftRightArithmetic = autoSetupConstant, .ShiftRightArithmetic = autoSetupConstant,
.ShiftRightLogical = autoSetupConstant, .ShiftRightLogical = autoSetupConstant,
.Source = opSource,
.SourceExtension = opSourceExtension, .SourceExtension = opSourceExtension,
.TypeArray = opTypeArray, .TypeArray = opTypeArray,
.TypeBool = opTypeBool, .TypeBool = opTypeBool,
@@ -377,7 +376,6 @@ fn BitOperator(comptime T: ValueType, comptime Op: BitOp) type {
4 => &op2.*.Vector4u32, 4 => &op2.*.Vector4u32,
else => unreachable, else => unreachable,
}; };
// NOTE: the above dummy mapping isnt type-correct for i32; call sites below pass correct rhs pointer.
_ = b; _ = b;
return RuntimeError.InvalidSpirV; return RuntimeError.InvalidSpirV;
} }
@@ -796,32 +794,32 @@ fn addDecoration(allocator: std.mem.Allocator, rt: *Runtime, target: SpvWord, de
decoration.index = if (member) |memb| memb else 0; decoration.index = if (member) |memb| memb else 0;
switch (decoration_type) { switch (decoration_type) {
.SpecId,
.ArrayStride,
.MatrixStride,
.BuiltIn,
.UniformId,
.Stream,
.Location,
.Component,
.Index,
.Binding,
.DescriptorSet,
.Offset,
.XfbBuffer,
.XfbStride,
.FuncParamAttr,
.FPRoundingMode,
.FPFastMathMode,
.InputAttachmentIndex,
.Alignment, .Alignment,
.MaxByteOffset,
.AlignmentId, .AlignmentId,
.MaxByteOffsetId, .ArrayStride,
.SecondaryViewportRelativeNV, .Binding,
.BuiltIn,
.Component,
.CounterBuffer, .CounterBuffer,
.DescriptorSet,
.FPFastMathMode,
.FPRoundingMode,
.FuncParamAttr,
.Index,
.InputAttachmentIndex,
.Location,
.MatrixStride,
.MaxByteOffset,
.MaxByteOffsetId,
.Offset,
.SecondaryViewportRelativeNV,
.SpecId,
.Stream,
.UniformId,
.UserSemantic, .UserSemantic,
.UserTypeGOOGLE, .UserTypeGOOGLE,
.XfbBuffer,
.XfbStride,
=> { => {
decoration.literal_1 = try rt.it.next(); decoration.literal_1 = try rt.it.next();
decoration.literal_2 = null; decoration.literal_2 = null;
@@ -869,21 +867,123 @@ fn opBitcast(_: std.mem.Allocator, _: SpvWord, rt: *Runtime) RuntimeError!void {
} }
fn copyValue(dst: *Result.Value, src: *const Result.Value) void { fn copyValue(dst: *Result.Value, src: *const Result.Value) void {
switch (src.*) { const helpers = struct {
.Vector, .Matrix, .Array, .Structure => |src_slice| { fn copySlice(dst_slice: []Result.Value, src_slice: []const Result.Value) void {
const dst_slice = switch (dst.*) {
.Vector, .Matrix, .Array, .Structure => |d| d,
else => unreachable,
};
for (0..@min(dst_slice.len, src_slice.len)) |i| { for (0..@min(dst_slice.len, src_slice.len)) |i| {
copyValue(&dst_slice[i], &src_slice[i]); copyValue(&dst_slice[i], &src_slice[i]);
} }
}
fn getDstSlice(v: *Result.Value) ?[]Result.Value {
return switch (v.*) {
.Vector, .Matrix, .Array, .Structure => |s| s,
.RuntimeArray => |s| s,
else => null,
};
}
fn writeF32(dst_f32_ptr: *f32, src_v: *const Result.Value) void {
switch (src_v.*) {
.Pointer => |src_ptr| switch (src_ptr) {
.f32_ptr => |src_f32_ptr| dst_f32_ptr.* = src_f32_ptr.*,
.common => |src_val_ptr| switch (src_val_ptr.*) {
.Float => |f| dst_f32_ptr.* = f.float32,
else => unreachable,
},
else => unreachable,
},
.Float => |f| dst_f32_ptr.* = f.float32,
else => unreachable,
}
}
fn writeI32(dst_i32_ptr: *i32, src_v: *const Result.Value) void {
switch (src_v.*) {
.Pointer => |src_ptr| switch (src_ptr) {
.i32_ptr => |src_i32_ptr| dst_i32_ptr.* = src_i32_ptr.*,
.common => |src_val_ptr| switch (src_val_ptr.*) {
.Int => |i| dst_i32_ptr.* = i.sint32,
else => unreachable,
},
else => unreachable,
},
.Int => |i| dst_i32_ptr.* = i.sint32,
else => unreachable,
}
}
fn writeU32(dst_u32_ptr: *u32, src_v: *const Result.Value) void {
switch (src_v.*) {
.Pointer => |src_ptr| switch (src_ptr) {
.u32_ptr => |src_u32_ptr| dst_u32_ptr.* = src_u32_ptr.*,
.common => |src_val_ptr| switch (src_val_ptr.*) {
.Int => |i| dst_u32_ptr.* = i.uint32,
else => unreachable,
},
else => unreachable,
},
.Int => |i| dst_u32_ptr.* = i.uint32,
else => unreachable,
}
}
};
if (std.meta.activeTag(dst.*) == .Pointer) {
switch (dst.Pointer) {
.common => |dst_val_ptr| return switch (src.*) {
.Pointer => |src_ptr| switch (src_ptr) {
.common => |src_val_ptr| copyValue(dst_val_ptr, src_val_ptr),
else => dst_val_ptr.* = src.*,
},
else => copyValue(dst_val_ptr, src),
},
.f32_ptr => |dst_f32_ptr| {
helpers.writeF32(dst_f32_ptr, src);
return;
},
.i32_ptr => |dst_i32_ptr| {
helpers.writeI32(dst_i32_ptr, src);
return;
},
.u32_ptr => |dst_u32_ptr| {
helpers.writeU32(dst_u32_ptr, src);
return;
},
}
}
if (std.meta.activeTag(src.*) == .Pointer) {
switch (src.Pointer) {
.common => |src_val_ptr| {
copyValue(dst, src_val_ptr);
return;
},
else => {},
}
}
const dst_slice = helpers.getDstSlice(dst);
switch (src.*) {
.Vector, .Matrix, .Array, .Structure => |src_slice| {
helpers.copySlice(dst_slice.?, src_slice);
}, },
.RuntimeArray => |opt_src_slice| if (opt_src_slice) |src_slice| {
helpers.copySlice(dst_slice.?, src_slice);
} else unreachable,
else => dst.* = src.*, else => dst.* = src.*,
} }
} }
pub fn getValuePrimitiveField(comptime T: ValueType, comptime BitCount: SpvWord, v: *Result.Value) RuntimeError!*getValuePrimitiveFieldType(T, BitCount) { pub fn getValuePrimitiveField(comptime T: ValueType, comptime BitCount: SpvWord, v: *Result.Value) RuntimeError!*getValuePrimitiveFieldType(T, BitCount) {
if (std.meta.activeTag(v.*) == .Pointer) {
return switch (v.Pointer) {
.common => |value| getValuePrimitiveField(T, BitCount, value),
.f32_ptr => |ptr| @ptrCast(@alignCast(ptr)),
.u32_ptr => |ptr| @ptrCast(@alignCast(ptr)),
.i32_ptr => |ptr| @ptrCast(@alignCast(ptr)),
};
}
return switch (T) { return switch (T) {
.Bool => &v.Bool, .Bool => &v.Bool,
.Float => switch (BitCount) { .Float => switch (BitCount) {
@@ -937,25 +1037,53 @@ fn opAccessChain(_: std.mem.Allocator, word_count: SpvWord, rt: *Runtime) Runtim
if (i.uint32 > v.len) return RuntimeError.InvalidSpirV; if (i.uint32 > v.len) return RuntimeError.InvalidSpirV;
value_ptr = &v[i.uint32]; value_ptr = &v[i.uint32];
}, },
//.Vector4f32 => |v| { .RuntimeArray => |opt_v| if (opt_v) |v| {
// if (i.uint32 > 4) return RuntimeError.InvalidSpirV; if (i.uint32 > v.len) return RuntimeError.InvalidSpirV;
// break :blk .{ value_ptr = &v[i.uint32];
// .Float = .{ .float32 = v[i.uint32] }, } else return RuntimeError.InvalidSpirV,
// }; .Vector4f32 => |*v| {
//}, if (i.uint32 > 4) return RuntimeError.InvalidSpirV;
//.Vector2f32 => |v| { break :blk .{ .Pointer = .{ .f32_ptr = &v[i.uint32] } };
// if (i.uint32 > 2) return RuntimeError.InvalidSpirV; },
// break :blk .{ .Vector3f32 => |*v| {
// .Float = .{ .float32 = v[i.uint32] }, if (i.uint32 > 3) return RuntimeError.InvalidSpirV;
// }; break :blk .{ .Pointer = .{ .f32_ptr = &v[i.uint32] } };
//}, },
.Vector2f32 => |*v| {
if (i.uint32 > 2) return RuntimeError.InvalidSpirV;
break :blk .{ .Pointer = .{ .f32_ptr = &v[i.uint32] } };
},
.Vector4i32 => |*v| {
if (i.uint32 > 4) return RuntimeError.InvalidSpirV;
break :blk .{ .Pointer = .{ .i32_ptr = &v[i.uint32] } };
},
.Vector3i32 => |*v| {
if (i.uint32 > 3) return RuntimeError.InvalidSpirV;
break :blk .{ .Pointer = .{ .i32_ptr = &v[i.uint32] } };
},
.Vector2i32 => |*v| {
if (i.uint32 > 2) return RuntimeError.InvalidSpirV;
break :blk .{ .Pointer = .{ .i32_ptr = &v[i.uint32] } };
},
.Vector4u32 => |*v| {
if (i.uint32 > 4) return RuntimeError.InvalidSpirV;
break :blk .{ .Pointer = .{ .u32_ptr = &v[i.uint32] } };
},
.Vector3u32 => |*v| {
if (i.uint32 > 3) return RuntimeError.InvalidSpirV;
break :blk .{ .Pointer = .{ .u32_ptr = &v[i.uint32] } };
},
.Vector2u32 => |*v| {
if (i.uint32 > 2) return RuntimeError.InvalidSpirV;
break :blk .{ .Pointer = .{ .u32_ptr = &v[i.uint32] } };
},
else => return RuntimeError.InvalidSpirV, else => return RuntimeError.InvalidSpirV,
} }
}, },
else => return RuntimeError.InvalidSpirV, else => return RuntimeError.InvalidSpirV,
} }
} }
break :blk value_ptr; break :blk .{ .Pointer = .{ .common = value_ptr } };
}, },
}, },
}; };
@@ -1186,8 +1314,16 @@ fn opExecutionMode(_: std.mem.Allocator, _: SpvWord, rt: *Runtime) RuntimeError!
}, },
.Invocations => rt.mod.geometry_invocations = try rt.it.next(), .Invocations => rt.mod.geometry_invocations = try rt.it.next(),
.OutputVertices => rt.mod.geometry_output_count = try rt.it.next(), .OutputVertices => rt.mod.geometry_output_count = try rt.it.next(),
.InputPoints, .InputLines, .Triangles, .InputLinesAdjacency, .InputTrianglesAdjacency => rt.mod.geometry_input = @intFromEnum(mode), .InputPoints,
.OutputPoints, .OutputLineStrip, .OutputTriangleStrip => rt.mod.geometry_output = @intFromEnum(mode), .InputLines,
.Triangles,
.InputLinesAdjacency,
.InputTrianglesAdjacency,
=> rt.mod.geometry_input = @intFromEnum(mode),
.OutputPoints,
.OutputLineStrip,
.OutputTriangleStrip,
=> rt.mod.geometry_output = @intFromEnum(mode),
else => {}, else => {},
} }
} }
@@ -1393,26 +1529,6 @@ fn opReturnValue(_: std.mem.Allocator, _: SpvWord, rt: *Runtime) RuntimeError!vo
} }
} }
fn opSource(allocator: std.mem.Allocator, word_count: SpvWord, rt: *Runtime) RuntimeError!void {
var file = rt.mod.files.addOne(allocator) catch return RuntimeError.OutOfMemory;
file.lang = try rt.it.nextAs(spv.SpvSourceLanguage);
file.lang_version = try rt.it.next();
if (word_count > 2) {
const id = try rt.it.next();
if (id >= rt.mod.results.len) return RuntimeError.InvalidSpirV;
if (rt.mod.results[id].name) |name| {
file.file_name = name;
}
}
if (word_count > 3) {
const id = try rt.it.next();
if (id >= rt.mod.results.len) return RuntimeError.InvalidSpirV;
if (rt.mod.results[id].name) |name| {
file.source = name;
}
}
}
fn opSourceExtension(allocator: std.mem.Allocator, word_count: SpvWord, rt: *Runtime) RuntimeError!void { fn opSourceExtension(allocator: std.mem.Allocator, word_count: SpvWord, rt: *Runtime) RuntimeError!void {
rt.mod.extensions.append(allocator, try readStringN(allocator, &rt.it, word_count)) catch return RuntimeError.OutOfMemory; rt.mod.extensions.append(allocator, try readStringN(allocator, &rt.it, word_count)) catch return RuntimeError.OutOfMemory;
} }
@@ -1523,9 +1639,16 @@ fn opTypePointer(_: std.mem.Allocator, _: SpvWord, rt: *Runtime) RuntimeError!vo
fn opTypeRuntimeArray(_: std.mem.Allocator, _: SpvWord, rt: *Runtime) RuntimeError!void { fn opTypeRuntimeArray(_: std.mem.Allocator, _: SpvWord, rt: *Runtime) RuntimeError!void {
const id = try rt.it.next(); const id = try rt.it.next();
const components_type_word = try rt.it.next();
rt.mod.results[id].variant = .{ rt.mod.results[id].variant = .{
.Type = .{ .Type = .{
.RuntimeArray = .{}, .RuntimeArray = .{
.components_type_word = components_type_word,
.components_type = switch ((try rt.mod.results[components_type_word].getVariant()).*) {
.Type => |t| @as(Result.Type, t),
else => return RuntimeError.InvalidSpirV,
},
},
}, },
}; };
} }
@@ -1631,9 +1754,6 @@ fn opVariable(allocator: std.mem.Allocator, word_count: SpvWord, rt: *Runtime) R
const resolved = rt.mod.results[var_type].resolveType(rt.mod.results); const resolved = rt.mod.results[var_type].resolveType(rt.mod.results);
const member_count = resolved.getMemberCounts(); const member_count = resolved.getMemberCounts();
if (member_count == 0) {
return RuntimeError.InvalidSpirV;
}
target.variant = .{ target.variant = .{
.Variable = .{ .Variable = .{
.storage_class = storage_class, .storage_class = storage_class,