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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
//! Print utilities
const colors = @import("colors");
const io = @import("io");
pub fn print(text: []const u8) void {
_ = io.mark(io.stream, text, colors.white) catch {};
}
pub fn println(text: []const u8) void {
_ = io.mark(io.stream, text, colors.white) catch {};
_ = io.mark(io.stream, "\n", colors.white) catch {};
}
pub fn color(text: []const u8, col: u32) void {
_ = io.mark(io.stream, text, col) catch {};
}
pub fn colorln(text: []const u8, col: u32) void {
_ = io.mark(io.stream, text, col) catch {};
_ = io.mark(io.stream, "\n", colors.white) catch {};
}
pub fn printf(comptime fmt: []const u8, args: anytype, buf: []u8) void {
const len = formatBuf(fmt, args, buf);
_ = io.mark(io.stream, buf[0..len], colors.white) catch {};
}
pub fn colorf(comptime fmt: []const u8, args: anytype, buf: []u8, col: u32) void {
const len = formatBuf(fmt, args, buf);
_ = io.mark(io.stream, buf[0..len], col) catch {};
}
fn formatBuf(comptime fmt: []const u8, args: anytype, buf: []u8) usize {
var pos: usize = 0;
comptime var arg_idx: usize = 0;
comptime var i: usize = 0;
inline while (i < fmt.len) {
if (fmt[i] == '{' and i + 1 < fmt.len and fmt[i + 1] == '}') {
const arg = args[arg_idx];
const T = @TypeOf(arg);
if (T == []const u8 or T == []u8) {
for (arg) |c| {
if (pos >= buf.len) break;
buf[pos] = c;
pos += 1;
}
} else if (@typeInfo(T) == .int or @typeInfo(T) == .comptime_int) {
pos += writeInt(@intCast(arg), buf[pos..]);
}
arg_idx += 1;
i += 2;
} else {
if (pos < buf.len) {
buf[pos] = fmt[i];
pos += 1;
}
i += 1;
}
}
return pos;
}
fn writeInt(num: u64, buf: []u8) usize {
if (num == 0) {
if (buf.len > 0) {
buf[0] = '0';
return 1;
}
return 0;
}
var temp: [20]u8 = undefined;
var n = num;
var i: usize = 0;
while (n > 0) : (i += 1) {
temp[i] = @as(u8, @intCast(n % 10)) + '0';
n /= 10;
}
var written: usize = 0;
while (i > 0 and written < buf.len) {
i -= 1;
buf[written] = temp[i];
written += 1;
}
return written;
}
|
