[Day-24]Zig:編譯期 Comptime

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

表達式

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

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

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
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();
}
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
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 標記,以向編譯期表達此參數是編譯期就可以確定的。

1
2
3
4
5
6
7
8
9
10
11
12
13
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});
}
1
2
Result 1: 15
Result 2: 90

參考

本文以 Zig 0.13.0 為主。並同時發佈在:


留言可能不會立即顯示。若過了幾天仍未出現,請 Email 聯繫:)