Zig：編譯期 Comptime

16th鐵人賽 zig

ZiTe

2024-09-16 07:28:58

166 瀏覽

分享至

comptime 是 Zig 的一大特色，使用它可以將許多運算帶到編譯期，從而提高運行時的效率。這個功能類似 C++11 的 constexpr 和 C++20 的 consteval。

表達式

將 comptime 加在表達式（Expression）前，Zig 就會將其結果在編譯期計算。如果無法在編譯期完成計算則會發出編譯期錯誤。

這邊的例子是以遞迴的方式計算費波那契數，並分成在 Run-time 和 Compile-time 計算的。使用 std.time.nanoTimestamp() 得知耗時，可以發現使用 comptime 的運行時耗時都是 0，代表它們確實在編譯期就已經完成求值。

const print = @import("std").debug.print;
const now = @import("std").time.nanoTimestamp;

fn fibonacci(index: u64) u64 {
    if (index < 2) return index;
    return fibonacci(index - 1) + fibonacci(index - 2);
}

fn run() void {
    const number = 25;

    // Run-time
    const start1 = now();
    const res1 = fibonacci(number);
    const end1 = now();

    // Compile-time
    const start2 = now();
    const res2 = comptime fibonacci(number);
    const end2 = now();

    print("Run-time:     {d} in {}ns\n", .{ res1, end1 - start1 });
    print("Compile-time: {d} in {}ns\n", .{ res2, end2 - start2 });
    print("\n", .{});
}

pub fn main() void {
    for (1..6) |_| {
        run();
    }
}

Run-time:     75025 in 1003800ns
Compile-time: 75025 in 0ns

Run-time:     75025 in 996000ns
Compile-time: 75025 in 0ns

Run-time:     75025 in 998600ns
Compile-time: 75025 in 0ns

Run-time:     75025 in 1000200ns
Compile-time: 75025 in 0ns

Run-time:     75025 in 1505500ns
Compile-time: 75025 in 0ns

泛型

泛型的型別也很適合使用 comptime 標記，以向編譯期表達此參數是編譯期就可以確定的。

const print = @import("std").debug.print;

fn add(comptime T: type, a: T, b: T) T {
    return a + b;
}

pub fn main() void {
    const res1 = add(u8, 5, 10);
    const res2 = add(f32, 100, -10);

    print("Result 1: {d}\n", .{res1});
    print("Result 2: {d}\n", .{res2});
}