wip: kitty image graphics

This commit is contained in:
Kalle Carlbark 2024-08-20 07:59:09 +02:00
parent cf0e38bff1
commit e85a085e41
No known key found for this signature in database
5 changed files with 212 additions and 50 deletions

View file

@ -17,6 +17,8 @@ pub fn build(b: *std.Build) void {
const vaxis_dep = b.dependency("vaxis", .{ .target = target, .optimize = optimize }); const vaxis_dep = b.dependency("vaxis", .{ .target = target, .optimize = optimize });
const vaxis = vaxis_dep.module("vaxis"); const vaxis = vaxis_dep.module("vaxis");
const objc_dep = b.dependency("objc", .{ .target = target, .optimize = optimize });
const objc = objc_dep.module("objc");
const lib = b.addStaticLibrary(.{ const lib = b.addStaticLibrary(.{
.name = "infoz", .name = "infoz",
@ -26,11 +28,20 @@ pub fn build(b: *std.Build) void {
.target = target, .target = target,
.optimize = optimize, .optimize = optimize,
}); });
const macos = b.addStaticLibrary(.{
.name = "infoz",
// In this case the main source file is merely a path, however, in more
// complicated build scripts, this could be a generated file.
.root_source_file = b.path("src/macos.zig"),
.target = target,
.optimize = optimize,
});
// This declares intent for the library to be installed into the standard // This declares intent for the library to be installed into the standard
// location when the user invokes the "install" step (the default step when // location when the user invokes the "install" step (the default step when
// running `zig build`). // running `zig build`).
b.installArtifact(lib); b.installArtifact(lib);
b.installArtifact(macos);
const exe = b.addExecutable(.{ const exe = b.addExecutable(.{
.name = "infoz", .name = "infoz",
@ -40,6 +51,7 @@ pub fn build(b: *std.Build) void {
}); });
exe.root_module.addImport("vaxis", vaxis); exe.root_module.addImport("vaxis", vaxis);
exe.root_module.addImport("objc", objc);
// This declares intent for the executable to be installed into the // This declares intent for the executable to be installed into the
// standard location when the user invokes the "install" step (the default // standard location when the user invokes the "install" step (the default

View file

@ -25,7 +25,11 @@
.dependencies = .{ .dependencies = .{
.vaxis = .{ .vaxis = .{
.url = "https://github.com/rockorager/libvaxis/archive/refs/heads/main.tar.gz", .url = "https://github.com/rockorager/libvaxis/archive/refs/heads/main.tar.gz",
.hash = "1220dab128b93a06ebca095bd38747f50cf9762b524e39e76de8771685264fd16841", .hash = "12209fb0db1467d24882f4a0fca367d25d7ef7353882270ba3b76382c185332c52ae",
},
.objc = .{
.url = "https://github.com/mitchellh/zig-objc/archive/refs/heads/main.tar.gz",
.hash = "122034b3e15d582d8d101a7713e5f13c872b8b8eb6d9cb47515b8e34ee75e122630d",
}, },
}, },
.paths = .{ .paths = .{

79
src/macos.zig Normal file
View file

@ -0,0 +1,79 @@
const objc = @import("objc");
const std = @import("std");
const utsname = @cImport({
@cInclude("sys/utsname.h");
});
const OsVersion = struct {
name: []const u8,
major: i64,
};
pub fn macosVersionAtLeast(major: i64, minor: i64, patch: i64) bool {
// Get the objc class from the runtime
const NSProcessInfo = objc.getClass("NSProcessInfo").?;
// Call a class method with no arguments that returns another objc object.
const info = NSProcessInfo.msgSend(objc.Object, "processInfo", .{});
// var NSOperatingSystemVersion = objc.getClass("NSOperatingSystemVersion").?;
// const os = info.msgSend(objc.Object, "")
// Call an instance method that returns a boolean and takes a single
// argument.
return info.msgSend(bool, "isOperatingSystemAtLeastVersion:", .{
NSOperatingSystemVersion{ .major = major, .minor = minor, .patch = patch },
});
}
pub fn macos_version(allocator: std.mem.Allocator) ![]const u8 {
const NSProcessInfo = objc.getClass("NSProcessInfo").?;
// Call a class method with no arguments that returns another objc object.
const info = NSProcessInfo.msgSend(objc.Object, "processInfo", .{});
const koko = info.msgSend(NSOperatingSystemVersion, "operatingSystemVersion", .{});
const buf = try std.fmt.allocPrint(allocator, "\n\t\t\tOS: {s} {d}.{d}.{d}", .{ version_to_name(koko), koko.major, koko.minor, koko.patch });
return buf;
}
pub fn macos_kernel_version(allocator: std.mem.Allocator) ![]const u8 {
var name: utsname.struct_utsname = undefined;
if (utsname.uname(&name) != 0) {
return error.UnameFailed;
}
const kaka = std.mem.span(@ptrCast(name.sysname));
const buf = try std.fmt.allocPrint(allocator, "\n\t\t\tKernel: {s} {s}", .{ kaka, name.version });
return buf;
}
fn version_to_name(version: NSOperatingSystemVersion) []const u8 {
switch (version.major) {
15 => return "Sequoia",
14 => return "Sonoma",
13 => return "Ventura",
12 => return "Monterey",
11 => return "Big Sur",
10 => {
switch (version.minor) {
15 => return "Catalina",
14 => return "Mojave",
else => return "Unknown",
}
},
else => return "Unknown",
}
}
/// This extern struct matches the Cocoa headers for layout.
const NSOperatingSystemVersion = extern struct {
major: i64,
minor: i64,
patch: i64,
};

View file

@ -1,5 +1,7 @@
const std = @import("std"); const std = @import("std");
const vaxis = @import("vaxis"); const vaxis = @import("vaxis");
const root = @import("root.zig");
const macos = @import("macos.zig");
const log = std.log.scoped(.main); const log = std.log.scoped(.main);
@ -17,6 +19,7 @@ pub fn main() !void {
} }
} }
// const stdout = std.io.getStdOut().writer();
const allocator = gpa.allocator(); const allocator = gpa.allocator();
var tty = try vaxis.Tty.init(); var tty = try vaxis.Tty.init();
@ -25,68 +28,100 @@ pub fn main() !void {
var vx = try vaxis.init(allocator, .{}); var vx = try vaxis.init(allocator, .{});
defer vx.deinit(allocator, tty.anyWriter()); defer vx.deinit(allocator, tty.anyWriter());
try vx.resetState(tty.anyWriter()); vx.caps.unicode = .unicode;
vx.caps.kitty_graphics = true;
var loop: vaxis.Loop(Event) = .{ .tty = &tty, .vaxis = &vx }; const os = try macos.macos_version(allocator);
try loop.init(); defer allocator.free(os);
const kernel = try macos.macos_kernel_version(allocator);
defer allocator.free(kernel);
try loop.start(); // try tty.anyWriter().writeAll(vaxis.ctlseqs.kitty_graphics_clear);
defer loop.stop(); try tty.anyWriter().writeAll(vaxis.ctlseqs.unicode_set);
defer vx.queryTerminalSend(tty.anyWriter()) catch {};
defer tty.anyWriter().writeAll(vaxis.ctlseqs.unicode_reset) catch {};
const stdout_file = std.io.getStdOut().writer(); var buf_writer = tty.bufferedWriter();
// try stdout_file.writeAll("\x1b[2J"); const writer = buf_writer.writer().any();
const winsize: vaxis.Winsize = try vaxis.Tty.getWinsize(tty.fd);
try vx.resize(allocator, tty.anyWriter(), winsize);
const win = vx.window();
win.clear();
var result: vaxis.Window.PrintResult = .{ .col = 0, .row = 0, .overflow = false };
// try tty.anyWriter().writeAll(vaxis.ctlseqs.kitty_graphics_query);
// var img1 = try vaxis.zigimg.Image.fromFilePath(allocator, "/Users/kc/Sync/Images/apple-white.png");
// defer img1.deinit();
// const koko = try vx.transmitImage(allocator, tty.anyWriter(), &img1, .png);
// win.hideCursor();
// win.showCursor(10, 1);
const system_info = [_]vaxis.Cell.Segment{
.{ .text = os },
.{ .text = kernel },
.{ .text = "\t\t\tKernel: lalal\n" },
.{ .text = "\t\t\tKernel: lalal\n" },
.{ .text = "\t\t\tKernel: lalal\n" },
.{ .text = "\t\t\tKernel: lalal\n" },
.{ .text = "\t\t\tKernel: lalal\n" },
.{ .text = "\t\t\tKernel: lalal\n" },
.{ .text = "\t\t\tKernel: lalal\n" },
.{ .text = "\t\t\tKernel: lalal\n" },
};
try root.kitty_image_inline_copy(allocator, tty.anyWriter(), "/Users/kc/Sync/Images/apple-dark.png");
try tty.anyWriter().print("\x1b[10;1H", .{});
try tty.anyWriter().print("\r\t\tTEST: testest\n", .{});
try tty.anyWriter().print("\r\t\tTEST: testest\n", .{});
try tty.anyWriter().print("\r\t\tTEST: testest\n", .{});
try tty.anyWriter().print("\r\t\tTEST: testest\n", .{});
try tty.anyWriter().print("\r\t\tTEST: testest\n", .{});
result = try win.print(&system_info, .{ .wrap = .word });
try vx.prettyPrint(writer);
// const dimensions = try koko.cellSize(win);
// const image = try vx.transmitLocalImagePath(allocator, tty.anyWriter(), "/Users/kc/Sync/Images/apple-white.png", dimensions.rows, dimensions.cols, .file, .png);
// defer vx.freeImage(tty.anyWriter(), image.id);
// try root.kitty_image_inline_copy(allocator, tty.anyWriter(), "/Users/kc/Sync/Images/apple-white.png");
// try image.draw(win, .{});
// try vx.render(tty.anyWriter());
try buf_writer.flush();
// try vx.queryTerminalSend(tty.anyWriter());
// Clear from cursor to beginning of the screen
// try stdout.writeAll("\x1b[1J");
// Move cursor to the upper left corner of the screen
// try stdout.writeAll("\x1b[1;1H");
// try stdout_file.writeAll("\x1b[9B"); // try stdout_file.writeAll("\x1b[9B");
// try vx.enterAltScreen(tty.anyWriter()); // try vx.enterAltScreen(tty.anyWriter());
try vx.queryTerminal(tty.anyWriter(), 1 * std.time.ns_per_s);
const image = [_]vaxis.Image{ // var img2 = try vaxis.
try vx.loadImage(allocator, tty.anyWriter(), .{ .path = "/Users/kc/Sync/Images/apple-dark.png" }), // const koko = try vx.loadImage(allocator, writer, .{ .path = "/Users/kc/Sync/Images/apple-white.png" });
// vaxis.Image.DrawOptions{ .clip_region = .{ .x = 0, .y = 0 } }, // const koko = try vx.transmitImage(allocator, writer, &img1, .png);
}; // const image = [_]vaxis.Image{
// // try vx.transmitImage(allocator, writer, &img1, .png),
// try vx.loadImage(allocator, writer, .{ .path = "/Users/kc/Sync/Images/apple-white.png" }),
// // vaxis.Image.DrawOptions{ .clip_region = .{ .x = 0, .y = 0 } },
// };
// defer vx.freeImage(tty.anyWriter(), image[0].id);
const event = loop.nextEvent();
switch (event) {
.winsize => |ws| try vx.resize(allocator, tty.anyWriter(), ws),
else => {},
}
_ = loop.nextEvent();
const win = vx.window();
// win.clear();
// vx.refresh = true;
// win.hideCursor(); // win.hideCursor();
// _ = loop.nextEvent(); // const dims = try image[0].cellSize(win);
const dims = try image[0].cellSize(win); // const logo_win = win.child(.{ .y_off = 0, .x_off = 0, .width = .{ .limit = dims.cols }, .height = .{ .limit = dims.rows } });
// const logo_win = win.child(.{ .y_off = 0, .x_off = 0 });
// const os_info = win.child(.{ .y_off = 0, .x_off = dims.cols, .width = .{ .limit = 30 }, .height = .{ .limit = 10 } });
// try stdout_file.print("{d} - {d}\n", .{ dims.cols, dims.rows }); // try vx.render(writer);
// try stdout.print("\n\n\n\n\n\n\n\n\n", .{});
const logo_win = win.child(.{ .y_off = 0, .x_off = 0, .width = .{ .limit = dims.cols }, .height = .{ .limit = dims.rows } }); // try writer.writeFile()
logo_win.clear(); // try stdout.writeAll("\x1b[m");
const os_info = win.child(.{ .y_off = 0, .x_off = dims.cols, .width = .{ .limit = 30 }, .height = .{ .limit = 8 } });
const system_info = [_]vaxis.Cell.Segment{
.{ .text = "\tOS: lalal\n" },
.{ .text = "\tKernel: lalal\n" },
.{ .text = "\tKernel: lalal\n" },
.{ .text = "\tKernel: lalal\n" },
.{ .text = "\tKernel: lalal\n" },
.{ .text = "\tKernel: lalal\n" },
.{ .text = "\tKernel: lalal\n" },
.{ .text = "\tKernel: lalal\n" },
.{ .text = "\tKernel: lalal\n" },
.{ .text = "\tKernel: lalal\n" },
};
try image[0].draw(logo_win, .{});
_ = try os_info.print(&system_info, .{ .wrap = .word });
try vx.render(tty.anyWriter());
// try stdout_file.writeAll("\x1b[9B");
try stdout_file.print("\n\n\n\n\n\n\n\n\n", .{});
// win.showCursor(); // win.showCursor();
std.time.sleep(10000000); // std.time.sleep(10000000);
} }
test "simple test" { test "simple test" {

View file

@ -1,10 +1,42 @@
const std = @import("std"); const std = @import("std");
const testing = std.testing; const testing = std.testing;
const vaxis = @import("vaxis");
const base64Encoder = std.base64.standard.Encoder;
export fn add(a: i32, b: i32) i32 { export fn add(a: i32, b: i32) i32 {
return a + b; return a + b;
} }
pub fn kitty_image_inline_copy(allocator: std.mem.Allocator, tty: std.io.AnyWriter, path: []const u8) !void {
var arena = std.heap.ArenaAllocator.init(allocator);
defer arena.deinit();
// const png_buf = try arena.allocator().alloc(u8, img.imageByteSize());
// const png = try img.writeToMemory(png_buf, .{ .png = .{} });
const b64_buf = try arena.allocator().alloc(u8, base64Encoder.calcSize(path.len));
const encoded = base64Encoder.encode(b64_buf, path);
try tty.print("\x1b_Ga=T,q=2,f=100,t=f,p=0,s=267,v=314;{s}\x1b\\", .{encoded});
try tty.print("\x1b\\", .{});
// if (encoded.len <= 4096) {
// try tty.print("\x1b_Ga=T,q=2,f=100,t=f,s=267,v=314,m=1;{s}\x1b\\", .{encoded});
// } else {
// var n: usize = 4096;
// try tty.print("\x1b_Ga=T,q=2,f=100,t=f,s=267,v=314,m=1;{s}\x1b\\", .{encoded[0..n]});
// while (n < encoded.len) : (n += 4096) {
// const end: usize = @min(n + 4096, encoded.len);
// const m: u2 = if (end == encoded.len) 0 else 1;
// try tty.print("\x1b_Gm={d};{s}\x1b\\", .{ m, encoded[n..end] });
// }
// }
// try tty.print("_Ga=T,q=2,f=100,t=f,s=267,v=314;{s}\\", .{encoded});
}
test "basic add functionality" { test "basic add functionality" {
try testing.expect(add(3, 7) == 10); try testing.expect(add(3, 7) == 10);
} }