zig/mach
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

examples: zig fmt

Browse source 
Signed-off-by: Stephen Gutekanst <stephen@hexops.com>
This commit is contained in:
Stephen Gutekanst 2022-04-20 23:01:30 -07:00
parent 3b86aa08cc
commit 8137f1a914
1 changed files with 42 additions and 50 deletions
Download patch file Download diff file Expand all files Collapse all files

62
examples/boids/main.zig
View file

@ -1,6 +1,5 @@
/// A port of Austin Eng's "computeBoids" webgpu sample.
/// https://github.com/austinEng/webgpu-samples/blob/main/src/sample/computeBoids/main.ts
const std = @import("std");
const mach = @import("mach");
const gpu = @import("gpu");
@ -18,7 +17,7 @@ const App = mach.App(*FrameParams, .{});
const num_particle = 1500;
var sim_params = [_]f32 {
var sim_params = [_]f32{
0.04, // .delta_T
0.1, // .rule_1_distance
0.025, // .rule_2_distance
@ -76,37 +75,31 @@ pub fn main() !void {
.buffers = &[_]gpu.VertexBufferLayout{
.{
// instanced particles buffer
.array_stride = 4*4,
.array_stride = 4 * 4,
.step_mode = .instance,
.attribute_count = instanced_particles_attributes.len,
.attributes = &instanced_particles_attributes,
},
.{
// vertex buffer
.array_stride = 2*4,
.array_stride = 2 * 4,
.step_mode = .vertex,
.attribute_count = vertex_buffer_attributes.len,
.attributes = &vertex_buffer_attributes,
},
},
},
.fragment = &gpu.FragmentState{
.module = sprite_shader_module,
.entry_point = "frag_main",
.targets = &[_]gpu.ColorTargetState{
.fragment = &gpu.FragmentState{ .module = sprite_shader_module, .entry_point = "frag_main", .targets = &[_]gpu.ColorTargetState{
.{
.format = app.swap_chain_format,
},
}
},
} },
});
const compute_pipeline = app.device.createComputePipeline(&gpu.ComputePipeline.Descriptor{
.compute = gpu.ProgrammableStageDescriptor{
const compute_pipeline = app.device.createComputePipeline(&gpu.ComputePipeline.Descriptor{ .compute = gpu.ProgrammableStageDescriptor{
.module = update_sprite_shader_module,
.entry_point = "main",
}
});
} });
const vert_buffer_data = [_]f32{
-0.01, -0.02, 0.01,
@ -114,22 +107,22 @@ pub fn main() !void {
};
const sprite_vertex_buffer = app.device.createBuffer(&gpu.Buffer.Descriptor{
.usage = .{.vertex = true, .copy_dst = true},
.usage = .{ .vertex = true, .copy_dst = true },
.size = vert_buffer_data.len * @sizeOf(f32),
});
app.device.getQueue().writeBuffer(sprite_vertex_buffer, 0, f32, &vert_buffer_data);
const sim_param_buffer = app.device.createBuffer(&gpu.Buffer.Descriptor{
.usage = .{.uniform = true, .copy_dst = true},
.usage = .{ .uniform = true, .copy_dst = true },
.size = sim_params.len * @sizeOf(f32),
});
app.device.getQueue().writeBuffer(sim_param_buffer, 0, f32, &sim_params);
var initial_particle_data: [num_particle*4]f32 = undefined;
var initial_particle_data: [num_particle * 4]f32 = undefined;
var rng = std.rand.DefaultPrng.init(0);
const random = rng.random();
var i:usize = 0;
while(i < num_particle): (i += 1) {
var i: usize = 0;
while (i < num_particle) : (i += 1) {
initial_particle_data[4 * i + 0] = 2 * (random.float(f32) - 0.5);
initial_particle_data[4 * i + 1] = 2 * (random.float(f32) - 0.5);
initial_particle_data[4 * i + 2] = 2 * (random.float(f32) - 0.5) * 0.1;
@ -139,24 +132,25 @@ pub fn main() !void {
var particle_buffers: [2]gpu.Buffer = undefined;
var particle_bind_groups: [2]gpu.BindGroup = undefined;
i = 0;
while(i < 2): (i+=1) {
while (i < 2) : (i += 1) {
particle_buffers[i] = app.device.createBuffer(&gpu.Buffer.Descriptor{
.usage = .{.vertex = true, .copy_dst = true, .storage = true, },
.usage = .{
.vertex = true,
.copy_dst = true,
.storage = true,
},
.size = initial_particle_data.len * @sizeOf(f32),
});
app.device.getQueue().writeBuffer(particle_buffers[i], 0, f32, &initial_particle_data);
}
i = 0;
while(i < 2): (i+=1) {
particle_bind_groups[i] = app.device.createBindGroup(&gpu.BindGroup.Descriptor{
.layout = compute_pipeline.getBindGroupLayout(0),
.entries = &[_]gpu.BindGroup.Entry {
while (i < 2) : (i += 1) {
particle_bind_groups[i] = app.device.createBindGroup(&gpu.BindGroup.Descriptor{ .layout = compute_pipeline.getBindGroupLayout(0), .entries = &[_]gpu.BindGroup.Entry{
gpu.BindGroup.Entry.buffer(0, sim_param_buffer, 0, sim_params.len * @sizeOf(f32)),
gpu.BindGroup.Entry.buffer(1, particle_buffers[i], 0, initial_particle_data.len * @sizeOf(f32)),
gpu.BindGroup.Entry.buffer(2, particle_buffers[(i+1)%2], 0, initial_particle_data.len * @sizeOf(f32)),
}
});
gpu.BindGroup.Entry.buffer(2, particle_buffers[(i + 1) % 2], 0, initial_particle_data.len * @sizeOf(f32)),
} });
}
ctx.* = FrameParams{
@ -182,11 +176,9 @@ fn frame(app: *App, params: *FrameParams) !void {
.store_op = .store,
};
const render_pass_descriptor = gpu.RenderPassEncoder.Descriptor{
.color_attachments = &[_]gpu.RenderPassColorAttachment {
const render_pass_descriptor = gpu.RenderPassEncoder.Descriptor{ .color_attachments = &[_]gpu.RenderPassColorAttachment{
color_attachment,
}
};
} };
sim_params[0] = @floatCast(f32, app.delta_time);
app.device.getQueue().writeBuffer(params.sim_param_buffer, 0, f32, &sim_params);
@ -203,15 +195,15 @@ fn frame(app: *App, params: *FrameParams) !void {
{
const pass_encoder = command_encoder.beginRenderPass(&render_pass_descriptor);
pass_encoder.setPipeline(params.render_pipeline);
pass_encoder.setVertexBuffer(0, params.particle_buffers[(params.frame_counter + 1) % 2], 0, num_particle*4*@sizeOf(f32));
pass_encoder.setVertexBuffer(1, params.sprite_vertex_buffer, 0, 6*@sizeOf(f32));
pass_encoder.setVertexBuffer(0, params.particle_buffers[(params.frame_counter + 1) % 2], 0, num_particle * 4 * @sizeOf(f32));
pass_encoder.setVertexBuffer(1, params.sprite_vertex_buffer, 0, 6 * @sizeOf(f32));
pass_encoder.draw(3, num_particle, 0, 0);
pass_encoder.end();
pass_encoder.release();
}
params.frame_counter += 1;
if(params.frame_counter % 60 == 0) {
if (params.frame_counter % 60 == 0) {
std.debug.print("Frame {}\n", .{params.frame_counter});
}