reorganize omt to separate folder (wip)

omt/utils/receiver.zig

@@ -0,0 +1,111 @@
+const std = @import("std");
+const omt = @import("omt");
+const comt = @cImport(@cInclude("libomt.h"));
+
+fn discoverStream() ?[*c]u8 {
+    var discovery_count: i32 = 0;
+
+    while (discovery_count == 0) {
+        const discovered = comt.omt_discovery_getaddresses(&discovery_count);
+        std.log.info("Found {} streams:", .{discovery_count});
+
+        const count: usize = @intCast(discovery_count);
+        for (0..count) |i| {
+            const name = discovered[i];
+            std.log.info("  {s}", .{name});
+        }
+
+        if (discovery_count > 0) {
+            return discovered[0];
+        }
+        std.Thread.sleep(1000 * 1000 * 1000);
+    }
+    return null;
+}
+
+pub fn main() !void {
+    var gpa = std.heap.GeneralPurposeAllocator(.{}){};
+    const allocator = gpa.allocator();
+
+    std.log.info("Receiving...", .{});
+    comt.omt_setloggingfilename("omtrecvtest.log");
+
+    const stream = discoverStream() orelse undefined;
+
+    const frametype_flags = comt.OMTFrameType_Video | comt.OMTFrameType_Audio | comt.OMTFrameType_Metadata;
+    const receiver = comt.omt_receive_create(
+        stream,
+        frametype_flags,
+        comt.OMTPreferredVideoFormat_UYVYorBGRA,
+        0,
+    );
+    defer comt.omt_receive_destroy(receiver);
+
+    while (true) {
+        const frame = comt.omt_receive(receiver, frametype_flags, 40);
+        if (frame != null) {
+            dumpOMTMediaFrameInfo(allocator, frame);
+        }
+    }
+}
+
+fn dumpOMTMediaFrameInfo(allocator: std.mem.Allocator, frame: *comt.OMTMediaFrame) void {
+    std.log.info("--------- Received --------", .{});
+    switch (frame.*.Type) {
+        comt.OMTFrameType_Video => dumpVideoFrameInfo(frame),
+        comt.OMTFrameType_Audio => dumpAudioFrameInfo(frame),
+        comt.OMTFrameType_Metadata => dumpMetadataFrameInfo(allocator, frame),
+        else => std.log.err("Undefined frame type", .{}),
+    }
+}
+
+fn dumpVideoFrameInfo(frame: *comt.OMTMediaFrame) void {
+    std.log.info("Frame: Video", .{});
+    std.log.info(
+        \\Timestamp: {d}
+        \\Codec: {s}
+        \\Width: {d}
+        \\Height: {d}
+        \\Stride: {d}
+        \\FrameRateN: {d}
+        \\FrameRateD: {d}
+        \\AspectRatio: {d}
+        \\ColorSpace: {s}
+    , .{
+        frame.Timestamp,
+        omt.codecToString(frame.Codec) orelse "Invalid",
+        frame.Width,
+        frame.Height,
+        frame.Stride,
+        frame.FrameRateN,
+        frame.FrameRateD,
+        frame.AspectRatio,
+        omt.colorSpaceToString(frame.ColorSpace),
+    });
+}
+
+fn dumpAudioFrameInfo(frame: *comt.OMTMediaFrame) void {
+    std.log.info("Frame: Audio", .{});
+    std.log.info(
+        \\Timestamp: {d}
+        \\SampleRate: {d}
+        \\Channels: {d}
+        \\SamplesPerChannel: {d}
+    , .{ frame.Timestamp, frame.SampleRate, frame.Channels, frame.SamplesPerChannel });
+}
+
+fn dumpMetadataFrameInfo(allocator: std.mem.Allocator, frame: *comt.OMTMediaFrame) void {
+    std.log.info("Frame: Metadata", .{});
+    if (frame.FrameMetadataLength == 0 or frame.FrameMetadata == null) {
+        std.log.info("None", .{});
+        return;
+    }
+
+    const metadata_length: usize = @intCast(frame.FrameMetadataLength);
+    const buffer = allocator.alloc(u8, metadata_length) catch @panic("OOM");
+    defer allocator.free(buffer);
+
+    std.debug.print("metadata: {?p}\n", .{frame.FrameMetadata});
+    std.mem.copyForwards(u8, buffer, @ptrCast(@alignCast(frame.FrameMetadata)));
+    std.log.info("{s}", .{buffer});
+}