const std = @import("std");

pub fn build(b: *std.Build) void {
    const target = b.standardTargetOptions(.{});
    const optimize = b.standardOptimizeOption(.{});

    const zig_wgpu = b.dependency("zig_wgpu", .{
        .target = target,
        .optimize = optimize,
    });

    // 1. Define the module so other projects can import it
    const mod = b.addModule("dimal", .{
        .root_source_file = b.path("src/main.zig"),
    });
    mod.addImport("gpu", zig_wgpu.module("zig-wgpu"));

    const exe_tests = b.addTest(.{
        .root_module = b.createModule(.{
            .root_source_file = b.path("src/main.zig"),
            .target = target,
            .optimize = optimize,
        }),
        .test_runner = .{ .path = b.path("test_runner.zig"), .mode = .simple },
    });

    exe_tests.root_module.addImport("gpu", zig_wgpu.module("zig-wgpu"));
    const run_exe_tests = b.addRunArtifact(exe_tests);
    const test_step = b.step("test", "Run tests");
    test_step.dependOn(&run_exe_tests.step);

    const bench_exe = b.addExecutable(.{
        .name = "benchmark",
        .root_module = b.createModule(.{
            .root_source_file = b.path("src/benchmark.zig"),
            .target = target,
            .imports = &.{},
        }),
    });
    bench_exe.root_module.addImport("gpu", zig_wgpu.module("zig-wgpu"));

    b.installArtifact(bench_exe);

    const run_bench = b.addRunArtifact(bench_exe);
    const bench_step = b.step("benchmark", "Run the benchmark suite");
    bench_step.dependOn(&run_bench.step);
}