forked from mirror/vulkan-zig
Shader compilation utility
This commit is contained in:
54
generator/build-integration.zig
Normal file
54
generator/build-integration.zig
Normal file
@@ -0,0 +1,54 @@
|
||||
const std = @import("std");
|
||||
const path = std.fs.path;
|
||||
const Builder = std.build.Builder;
|
||||
const Step = std.build.Step;
|
||||
|
||||
pub const ShaderCompileStep = struct {
|
||||
const Shader = struct {
|
||||
source_path: []const u8,
|
||||
full_out_path: []const u8,
|
||||
};
|
||||
|
||||
step: Step,
|
||||
builder: *Builder,
|
||||
glslc_path: []const u8,
|
||||
shaders: std.ArrayList(Shader),
|
||||
|
||||
pub fn init(builder: *Builder, glslc_path: []const u8) *ShaderCompileStep {
|
||||
const self = builder.allocator.create(ShaderCompileStep) catch unreachable;
|
||||
self.* = .{
|
||||
.step = Step.init(.Custom, "shader-compile", builder.allocator, make),
|
||||
.builder = builder,
|
||||
.glslc_path = glslc_path,
|
||||
.shaders = std.ArrayList(Shader).init(builder.allocator),
|
||||
};
|
||||
return self;
|
||||
}
|
||||
|
||||
pub fn add(self: *ShaderCompileStep, src: []const u8) []const u8 {
|
||||
const full_out_path = path.join(self.builder.allocator, &[_][]const u8{
|
||||
self.builder.build_root,
|
||||
self.builder.cache_root,
|
||||
"shaders",
|
||||
src,
|
||||
}) catch unreachable;
|
||||
self.shaders.append(.{.source_path = src, .full_out_path = full_out_path}) catch unreachable;
|
||||
return full_out_path;
|
||||
}
|
||||
|
||||
fn make(step: *Step) !void {
|
||||
const self = @fieldParentPtr(ShaderCompileStep, "step", step);
|
||||
const cwd = std.fs.cwd();
|
||||
|
||||
for (self.shaders.items) |shader| {
|
||||
const dir = path.dirname(shader.full_out_path).?;
|
||||
try cwd.makePath(dir);
|
||||
try self.builder.spawnChild(&[_][]const u8{
|
||||
self.glslc_path,
|
||||
shader.source_path,
|
||||
"-o",
|
||||
shader.full_out_path,
|
||||
});
|
||||
}
|
||||
}
|
||||
};
|
||||
@@ -10,16 +10,17 @@ pub const GenerateStep = struct {
|
||||
spec_path: []const u8,
|
||||
full_out_path: []const u8,
|
||||
|
||||
// relative_out_path is relative to builder.cache_root
|
||||
pub fn init(builder: *Builder, spec_path: []const u8, relative_out_path: []const u8) *GenerateStep {
|
||||
// out_path is relative to builder.cache_root
|
||||
pub fn init(builder: *Builder, spec_path: []const u8, out_path: []const u8) *GenerateStep {
|
||||
const self = builder.allocator.create(GenerateStep) catch unreachable;
|
||||
self.* = .{
|
||||
.step = Step.init(.Custom, "vulkan-generate", builder.allocator, make),
|
||||
.builder = builder,
|
||||
.spec_path = spec_path,
|
||||
.full_out_path = path.join(builder.allocator, &[_][]const u8{
|
||||
self.builder.build_root,
|
||||
builder.cache_root,
|
||||
relative_out_path,
|
||||
out_path,
|
||||
}) catch unreachable,
|
||||
};
|
||||
return self;
|
||||
@@ -28,16 +29,15 @@ pub const GenerateStep = struct {
|
||||
fn make(step: *Step) !void {
|
||||
const self = @fieldParentPtr(GenerateStep, "step", step);
|
||||
const cwd = std.fs.cwd();
|
||||
|
||||
var out_buffer = std.ArrayList(u8).init(self.builder.allocator);
|
||||
const spec = try std.fs.cwd().readFileAlloc(self.builder.allocator, self.spec_path, std.math.maxInt(usize));
|
||||
const spec = try cwd.readFileAlloc(self.builder.allocator, self.spec_path, std.math.maxInt(usize));
|
||||
try generate(self.builder.allocator, spec, out_buffer.writer());
|
||||
|
||||
const tree = try std.zig.parse(self.builder.allocator, out_buffer.items);
|
||||
|
||||
const dir = path.dirname(self.full_out_path).?;
|
||||
try std.fs.cwd().makePath(dir);
|
||||
const output_file = std.fs.cwd().createFile(self.full_out_path, .{}) catch unreachable;
|
||||
try cwd.makePath(dir);
|
||||
const output_file = cwd.createFile(self.full_out_path, .{}) catch unreachable;
|
||||
_ = try std.zig.render(self.builder.allocator, output_file.outStream(), tree);
|
||||
}
|
||||
};
|
||||
|
||||
Reference in New Issue
Block a user