From c3112f90b0e7fe95cc153262107a47ff05c3380c Mon Sep 17 00:00:00 2001
From: Nickiel12 <nickiel@nickiel.net>
Date: Wed, 16 Oct 2024 00:52:17 +0000
Subject: [PATCH] added inference vstream loading

---
 src/main.zig | 34 +++++++++++++++++++++++++++++++---
 1 file changed, 31 insertions(+), 3 deletions(-)

diff --git a/src/main.zig b/src/main.zig
index 8f4b58a..04dbb93 100644
--- a/src/main.zig
+++ b/src/main.zig
@@ -13,10 +13,10 @@ const max_edge_layers = 32;
 
 pub fn main() void {
 
-    // var arena = std.heap.ArenaAllocator.init(std.heap.page_allocator);
-    // defer arena.deinit();
+    var arena = std.heap.ArenaAllocator.init(std.heap.page_allocator);
+    defer arena.deinit();
 
-    // const alloc = arena.allocator();
+    const alloc = arena.allocator();
 
     std.fs.cwd().access(hef_file, .{ }) catch |e| {
         std.debug.panic("Could not open hef file! '{any}'", .{ e });
@@ -85,6 +85,34 @@ pub fn main() void {
     std.debug.print("Output_vstream_size is: '{d}'\n", .{ output_vstream_size });
     std.debug.print("Input_vstream_size is: '{d}'\n", .{ input_vstream_size });
 
+
+
+    var input_frame_size: usize = 0;
+
+    status = hlo.hailo_get_input_vstream_frame_size(input_vstreams, &input_frame_size);
+    assert(status == hlo.HAILO_SUCCESS);
+
+    const frame_count = input_frame_size / @sizeOf(u8);
+    std.debug.print("Trying to create {d} u8s\n", .{ frame_count });
+    var input_data = try alloc.create(u8, frame_count);
+
+    var rnd = std.Random.init(0);
+
+    for (0..frame_count) |i| {
+        input_data[i] = rnd.random().int(u8);
+    }
+
+    // Here we simulate taking our data and chunking it into frame_size, then uploading each frame until we have uploaded
+    // all of our data
+    for (0..frame_count) |_| {
+        status = hlo.hailo_vstream_write_raw_buffer(input_vstreams, input_data, input_frame_size);
+        assert(status == hlo.HAILO_SUCCESS);
+    }
+
+    status = hlo.hailo_flush_input_vstream(input_vstreams);
+    assert(status == hlo.HAILO_SUCCESS);
+
+
     _ = hlo.hailo_release_output_vstreams(&output_vstreams, output_vstream_size);
     _ = hlo.hailo_release_input_vstreams(&input_vstreams, input_vstream_size);
     _ = hlo.hailo_release_hef(hef);