112 lines
3.4 KiB
Zig
112 lines
3.4 KiB
Zig
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});
|
|
}
|