tiny.profiling.driver.paths

1const std = @import("std");2const sys = @import("sys");3const execute = @import("../root.zig").execute;4const host = @import("../root.zig").host;5const record = @import("../root.zig").record;67pub fn absoluteArtifactEnv(8    allocator: std.mem.Allocator,9    paths: record.WorkloadPaths,10) !execute.ArtifactEnv {11    std.debug.assert(paths.root.len > 0);12    std.debug.assert(paths.allocations.len > paths.root.len);13    return try absoluteExecutionArtifactEnv(allocator, .{14        .root = paths.root,15        .stdout = paths.stdout,16        .stderr = paths.stderr,17        .bench_jsonl = paths.bench_jsonl,18        .coz_jsonl = paths.coz_jsonl,19        .coz_analysis = paths.coz_analysis,20        .tracy_jsonl = paths.tracy_jsonl,21        .tracy_summary = paths.tracy_summary,22        .allocations = paths.allocations,23        .structured = paths.structured,24    });25}2627pub fn absoluteExecutionArtifactEnv(28    allocator: std.mem.Allocator,29    paths: record.ExecutionPaths,30) !execute.ArtifactEnv {31    std.debug.assert(paths.root.len > 0);32    std.debug.assert(paths.allocations.len > paths.root.len);33    return .{34        .stdout = paths.stdout,35        .stderr = paths.stderr,36        .bench_jsonl = try absolutePath(allocator, paths.bench_jsonl),37        .coz_jsonl = try absolutePath(allocator, paths.coz_jsonl),38        .coz_analysis = try absolutePath(allocator, paths.coz_analysis),39        .tracy_jsonl = try absolutePath(allocator, paths.tracy_jsonl),40        .tracy_summary = try absolutePath(allocator, paths.tracy_summary),41        .allocations = try absolutePath(allocator, paths.allocations),42    };43}4445pub fn absoluteWarmupArtifactEnv(46    allocator: std.mem.Allocator,47    paths: record.WarmupPaths,48) !execute.ArtifactEnv {49    std.debug.assert(paths.root.len > 0);50    return .{51        .stdout = paths.stdout,52        .stderr = paths.stderr,53        .bench_jsonl = try absolutePath(allocator, paths.bench_jsonl),54        .coz_jsonl = try absolutePath(allocator, paths.coz_jsonl),55        .coz_analysis = try absolutePath(allocator, paths.coz_analysis),56        .tracy_jsonl = try absolutePath(allocator, paths.tracy_jsonl),57        .tracy_summary = try absolutePath(allocator, paths.tracy_summary),58        .allocations = try absolutePath(allocator, paths.allocations),59    };60}6162pub fn absolutePath(allocator: std.mem.Allocator, path: []const u8) ![]const u8 {63    std.debug.assert(path.len > 0);64    if (std.fs.path.isAbsolute(path)) return path;65    const cwd = try sys.fs.cwdAlloc(allocator);66    const result = try std.fs.path.join(allocator, &.{ cwd, path });67    std.debug.assert(std.fs.path.isAbsolute(result));68    return result;69}7071pub fn rebaseCaptures(72    allocator: std.mem.Allocator,73    captures: []const host.Capture,74    absolute_root: []const u8,75    artifact_root: []const u8,76) ![]const host.Capture {77    std.debug.assert(std.fs.path.isAbsolute(absolute_root));78    std.debug.assert(artifact_root.len > 0);79    const rows = try allocator.dupe(host.Capture, captures);80    for (rows) |*row| {81        row.capture_path = try rebasePath(82            allocator,83            row.capture_path,84            absolute_root,85            artifact_root,86        );87        row.summary_path = try rebasePath(88            allocator,89            row.summary_path,90            absolute_root,91            artifact_root,92        );93    }94    return rows;95}9697fn rebasePath(98    allocator: std.mem.Allocator,99    path: []const u8,100    absolute_root: []const u8,101    artifact_root: []const u8,102) ![]const u8 {103    std.debug.assert(std.fs.path.isAbsolute(absolute_root));104    std.debug.assert(artifact_root.len > 0);105    if (!std.mem.startsWith(u8, path, absolute_root)) return path;106    if (path.len == absolute_root.len) return artifact_root;107    if (path[absolute_root.len] != std.fs.path.sep) return path;108    return try std.fs.path.join(109        allocator,110        &.{ artifact_root, path[absolute_root.len + 1 ..] },111    );112}113114test "profiling driver rebases lane capture paths onto the requested run root" {115    var arena_state = std.heap.ArenaAllocator.init(std.testing.allocator);116    defer arena_state.deinit();117    const allocator = arena_state.allocator();118    const rows = try rebaseCaptures(allocator, &.{.{119        .kind = "dhat",120        .tool = "valgrind",121        .capture_path = "/repo/zig-out/profiling/run/workloads/w/dhat.capture.txt",122        .summary_path = "/repo/zig-out/profiling/run/workloads/w/dhat.summary.json",123        .state = "summary_written",124    }}, "/repo/zig-out/profiling/run/workloads/w", "zig-out/profiling/run/workloads/w");125    try std.testing.expectEqualStrings(126        "zig-out/profiling/run/workloads/w/dhat.capture.txt",127        rows[0].capture_path,128    );129    try std.testing.expectEqualStrings(130        "zig-out/profiling/run/workloads/w/dhat.summary.json",131        rows[0].summary_path,132    );133134    const absolute_rows = try rebaseCaptures(allocator, &.{.{135        .kind = "dhat",136        .tool = "valgrind",137        .capture_path = "/tmp/profiling/run/workloads/w/dhat.capture.txt",138        .summary_path = "/tmp/profiling/run/workloads/w/dhat.summary.json",139        .state = "summary_written",140    }}, "/tmp/profiling/run/workloads/w", "/tmp/profiling/run/workloads/w");141    try std.testing.expectEqualStrings(142        "/tmp/profiling/run/workloads/w/dhat.capture.txt",143        absolute_rows[0].capture_path,144    );145    try std.testing.expectEqualStrings(146        "/tmp/profiling/run/workloads/w/dhat.summary.json",147        absolute_rows[0].summary_path,148    );149150    const untouched = try rebasePath(151        allocator,152        "/elsewhere/dhat.out",153        "/repo/zig-out/profiling/run/workloads/w",154        "zig-out/x",155    );156    try std.testing.expectEqualStrings("/elsewhere/dhat.out", untouched);157    const boundary = try rebasePath(158        allocator,159        "/repo/zig-out/profiling/run/workloads/wide/f",160        "/repo/zig-out/profiling/run/workloads/w",161        "zig-out/x",162    );163    try std.testing.expectEqualStrings("/repo/zig-out/profiling/run/workloads/wide/f", boundary);164}165166test "profiling driver resolves artifact env paths to absolute paths" {167    var arena_state = std.heap.ArenaAllocator.init(std.testing.allocator);168    defer arena_state.deinit();169    const allocator = arena_state.allocator();170    const paths = record.WorkloadPaths{171        .root = "zig-out/profiling/run/workloads/w",172        .command = "zig-out/profiling/run/workloads/w/command.json",173        .result = "zig-out/profiling/run/workloads/w/result.json",174        .stdout = "zig-out/profiling/run/workloads/w/stdout.txt",175        .stderr = "zig-out/profiling/run/workloads/w/stderr.txt",176        .setup_stdout = "zig-out/profiling/run/workloads/w/setup.stdout.txt",177        .setup_stderr = "zig-out/profiling/run/workloads/w/setup.stderr.txt",178        .prepare_stdout = "zig-out/profiling/run/workloads/w/prepare.stdout.txt",179        .prepare_stderr = "zig-out/profiling/run/workloads/w/prepare.stderr.txt",180        .bench_jsonl = "zig-out/profiling/run/workloads/w/bench.jsonl",181        .coz_jsonl = "zig-out/profiling/run/workloads/w/bench.coz.jsonl",182        .coz_analysis = "zig-out/profiling/run/workloads/w/bench.coz.analysis.json",183        .tracy_jsonl = "zig-out/profiling/run/workloads/w/bench.tracy.jsonl",184        .tracy_summary = "zig-out/profiling/run/workloads/w/bench.tracy.summary.jsonl",185        .allocations = "zig-out/profiling/run/workloads/w/allocations.jsonl",186        .allocations_summary = "zig-out/profiling/run/workloads/w/allocations.summary.jsonl",187        .allocation_repetitions = "zig-out/profiling/run/workloads/w/allocations.repetitions.jsonl",188        .structured = "zig-out/profiling/run/workloads/w/structured.jsonl",189        .warmup = .{190            .root = "zig-out/profiling/run/workloads/w/warmup",191            .stdout = "zig-out/profiling/run/workloads/w/warmup/stdout.txt",192            .stderr = "zig-out/profiling/run/workloads/w/warmup/stderr.txt",193            .bench_jsonl = "zig-out/profiling/run/workloads/w/warmup/bench.jsonl",194            .coz_jsonl = "zig-out/profiling/run/workloads/w/warmup/bench.coz.jsonl",195            .coz_analysis = "zig-out/profiling/run/workloads/w/warmup/bench.coz.analysis.json",196            .tracy_jsonl = "zig-out/profiling/run/workloads/w/warmup/bench.tracy.jsonl",197            .tracy_summary = "zig-out/profiling/run/workloads/w/warmup/bench.tracy.summary.jsonl",198            .allocations = "zig-out/profiling/run/workloads/w/warmup/allocations.jsonl",199        },200    };201    const env = try absoluteArtifactEnv(allocator, paths);202    try std.testing.expect(std.fs.path.isAbsolute(env.bench_jsonl));203    try std.testing.expect(std.fs.path.isAbsolute(env.tracy_jsonl));204    try std.testing.expect(std.fs.path.isAbsolute(env.allocations));205}

1pub const paths = @import("paths.zig");