working on example and adding opcodes
This commit is contained in:
157
example/main.zig
157
example/main.zig
@@ -4,91 +4,120 @@ const spv = @import("spv");
|
||||
|
||||
const shader_source = @embedFile("shader.spv");
|
||||
|
||||
const screen_width = 640;
|
||||
const screen_height = 480;
|
||||
const screen_width = 1250;
|
||||
const screen_height = 720;
|
||||
|
||||
pub fn main() !void {
|
||||
{
|
||||
//var gpa: std.heap.DebugAllocator(.{}) = .init;
|
||||
//defer _ = gpa.deinit();
|
||||
|
||||
var gpa = std.heap.ArenaAllocator.init(std.heap.page_allocator);
|
||||
defer gpa.deinit();
|
||||
|
||||
defer sdl3.shutdown();
|
||||
const init_flags = sdl3.InitFlags{ .video = true };
|
||||
const init_flags = sdl3.InitFlags{ .video = true, .events = true };
|
||||
try sdl3.init(init_flags);
|
||||
defer sdl3.quit(init_flags);
|
||||
|
||||
const window = try sdl3.video.Window.init("Hello triangle", screen_width, screen_height, .{});
|
||||
defer window.deinit();
|
||||
|
||||
const allocator = gpa.allocator();
|
||||
const surface = try window.getSurface();
|
||||
|
||||
const allocator = std.heap.smp_allocator;
|
||||
|
||||
var module = try spv.Module.init(allocator, @ptrCast(@alignCast(shader_source)), .{});
|
||||
defer module.deinit(allocator);
|
||||
|
||||
const surface = try window.getSurface();
|
||||
try surface.clear(.{ .r = 0.0, .g = 0.0, .b = 0.0, .a = 0.0 });
|
||||
|
||||
{
|
||||
try surface.lock();
|
||||
defer surface.unlock();
|
||||
|
||||
var pixel_map: [*]u32 = @as([*]u32, @ptrCast(@alignCast((surface.getPixels() orelse return).ptr)));
|
||||
|
||||
const margin_x = @divTrunc(screen_width, 3);
|
||||
const margin_y = @divTrunc(screen_height, 3);
|
||||
const top_y = margin_y;
|
||||
const bottom_y = (screen_height - 1) - margin_y;
|
||||
const center_x = @divTrunc(screen_width, 2);
|
||||
const tri_h = bottom_y - top_y;
|
||||
const max_half_w = @divTrunc(screen_width, 2) - margin_x;
|
||||
|
||||
var timer = try std.time.Timer.start();
|
||||
defer {
|
||||
const ns = timer.lap();
|
||||
std.log.info("Took {d:.3}s to render", .{@as(f32, @floatFromInt(ns)) / std.time.ns_per_s});
|
||||
}
|
||||
|
||||
for (top_y..bottom_y) |y| {
|
||||
const t: f32 = @as(f32, @floatFromInt(y - top_y)) / @as(f32, @floatFromInt(tri_h));
|
||||
const half_w: usize = @intFromFloat((t * @as(f32, @floatFromInt(max_half_w))) + 0.5);
|
||||
const x0 = std.math.clamp(center_x - half_w, 0, screen_width - 1);
|
||||
const x1 = std.math.clamp(center_x + half_w, 0, screen_width - 1);
|
||||
|
||||
for (x0..x1) |x| {
|
||||
var rt = try spv.Runtime.init(allocator, &module);
|
||||
defer rt.deinit(allocator);
|
||||
|
||||
var output: [4]f32 = undefined;
|
||||
|
||||
const entry = try rt.getEntryPointByName("main");
|
||||
const color = try rt.getResultByName("color");
|
||||
const dim = try rt.getResultByName("dim");
|
||||
const pos = try rt.getResultByName("pos");
|
||||
|
||||
try rt.writeInput(f32, &.{ @floatFromInt(x1 - x0), @floatFromInt(bottom_y - top_y) }, dim);
|
||||
try rt.writeInput(f32, &.{ @floatFromInt(x), @floatFromInt(y) }, pos);
|
||||
|
||||
try rt.callEntryPoint(allocator, entry);
|
||||
try rt.readOutput(f32, output[0..], color);
|
||||
|
||||
const rgba = surface.mapRgba(
|
||||
@truncate(@as(u32, @intFromFloat(output[0] * 255.0))),
|
||||
@truncate(@as(u32, @intFromFloat(output[1] * 255.0))),
|
||||
@truncate(@as(u32, @intFromFloat(output[2] * 255.0))),
|
||||
@truncate(@as(u32, @intFromFloat(output[3] * 255.0))),
|
||||
);
|
||||
|
||||
pixel_map[(y * surface.getWidth()) + x] = rgba.value;
|
||||
}
|
||||
var runner_cache: std.ArrayList(Runner) = try .initCapacity(allocator, screen_height);
|
||||
defer {
|
||||
for (runner_cache.items) |*runner| {
|
||||
runner.rt.deinit(allocator);
|
||||
}
|
||||
runner_cache.deinit(allocator);
|
||||
}
|
||||
|
||||
try window.updateSurface();
|
||||
for (0..screen_height) |_| {
|
||||
(try runner_cache.addOne(allocator)).* = .{
|
||||
.allocator = allocator,
|
||||
.surface = surface,
|
||||
.rt = try spv.Runtime.init(allocator, &module),
|
||||
};
|
||||
}
|
||||
|
||||
std.Thread.sleep(2_000_000_000);
|
||||
var thread_pool: std.Thread.Pool = undefined;
|
||||
try thread_pool.init(.{
|
||||
.allocator = allocator,
|
||||
});
|
||||
|
||||
var quit = false;
|
||||
while (!quit) {
|
||||
try surface.clear(.{ .r = 0.0, .g = 0.0, .b = 0.0, .a = 0.0 });
|
||||
|
||||
while (sdl3.events.poll()) |event|
|
||||
switch (event) {
|
||||
.quit => quit = true,
|
||||
.terminating => quit = true,
|
||||
else => {},
|
||||
};
|
||||
|
||||
{
|
||||
try surface.lock();
|
||||
defer surface.unlock();
|
||||
|
||||
const pixel_map: [*]u32 = @as([*]u32, @ptrCast(@alignCast((surface.getPixels() orelse return).ptr)));
|
||||
|
||||
var timer = try std.time.Timer.start();
|
||||
defer {
|
||||
const ns = timer.lap();
|
||||
const ms = @as(f32, @floatFromInt(ns)) / std.time.ns_per_s;
|
||||
std.log.info("Took {d:.3}s - {d:.3}fps to render", .{ ms, 1.0 / ms });
|
||||
}
|
||||
|
||||
var wait_group: std.Thread.WaitGroup = .{};
|
||||
for (0..screen_height) |y| {
|
||||
const runner = &runner_cache.items[y];
|
||||
thread_pool.spawnWg(&wait_group, Runner.run, .{ runner, y, pixel_map });
|
||||
}
|
||||
thread_pool.waitAndWork(&wait_group);
|
||||
}
|
||||
|
||||
try window.updateSurface();
|
||||
}
|
||||
}
|
||||
std.log.info("Successfully executed", .{});
|
||||
}
|
||||
|
||||
const Runner = struct {
|
||||
const Self = @This();
|
||||
|
||||
allocator: std.mem.Allocator,
|
||||
surface: sdl3.surface.Surface,
|
||||
rt: spv.Runtime,
|
||||
|
||||
fn run(self: *Self, y: usize, pixel_map: [*]u32) void {
|
||||
const entry = self.rt.getEntryPointByName("main") catch |err| std.debug.panic("Catch error {s}", .{@errorName(err)});
|
||||
const color = self.rt.getResultByName("color") catch |err| std.debug.panic("Catch error {s}", .{@errorName(err)});
|
||||
const time = self.rt.getResultByName("time") catch |err| std.debug.panic("Catch error {s}", .{@errorName(err)});
|
||||
const pos = self.rt.getResultByName("pos") catch |err| std.debug.panic("Catch error {s}", .{@errorName(err)});
|
||||
const res = self.rt.getResultByName("res") catch |err| std.debug.panic("Catch error {s}", .{@errorName(err)});
|
||||
var output: [4]f32 = undefined;
|
||||
|
||||
var rt = self.rt; // Copy to avoid pointer access of `self` at runtime. Okay as Runtime contains only pointers and trivially copyable fields
|
||||
|
||||
for (0..screen_width) |x| {
|
||||
rt.writeInput(f32, &.{@as(f32, @floatFromInt(std.time.milliTimestamp()))}, time) catch |err| std.debug.panic("Catch error {s}", .{@errorName(err)});
|
||||
rt.writeInput(f32, &.{ @floatFromInt(screen_width), @floatFromInt(screen_height) }, res) catch |err| std.debug.panic("Catch error {s}", .{@errorName(err)});
|
||||
rt.writeInput(f32, &.{ @floatFromInt(x), @floatFromInt(y) }, pos) catch |err| std.debug.panic("Catch error {s}", .{@errorName(err)});
|
||||
rt.callEntryPoint(self.allocator, entry) catch |err| std.debug.panic("Catch error {s}", .{@errorName(err)});
|
||||
rt.readOutput(f32, output[0..], color) catch |err| std.debug.panic("Catch error {s}", .{@errorName(err)});
|
||||
|
||||
const rgba = self.surface.mapRgba(
|
||||
@truncate(@as(u32, @intFromFloat(output[0] * 255.0))),
|
||||
@truncate(@as(u32, @intFromFloat(output[1] * 255.0))),
|
||||
@truncate(@as(u32, @intFromFloat(output[2] * 255.0))),
|
||||
@truncate(@as(u32, @intFromFloat(output[3] * 255.0))),
|
||||
);
|
||||
|
||||
pixel_map[(y * self.surface.getWidth()) + x] = rgba.value;
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
Reference in New Issue
Block a user