From a28ad1509db19c9edef3d05df5e575fe16d9fb70 Mon Sep 17 00:00:00 2001
From: Joose Sainio <joose.sainio@tuni.fi>
Date: Fri, 9 Feb 2024 14:25:43 +0200
Subject: [PATCH] Send finished CTU

---
 src/encmain.c |  2 +-
 src/search.c  | 60 +++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 61 insertions(+), 1 deletion(-)

diff --git a/src/encmain.c b/src/encmain.c
index 4bb166a0..857e38e6 100644
--- a/src/encmain.c
+++ b/src/encmain.c
@@ -273,7 +273,7 @@ static void* input_read_thread(void* in_args)
 #if defined(__GNUC__) && !defined(__MINGW32__)
     usleep(33000);
 #else
-    Sleep(33);
+    // Sleep(33);
 #endif
 
     // Wait until main thread is ready to receive the next frame.
diff --git a/src/search.c b/src/search.c
index 8732cc3e..072c1035 100644
--- a/src/search.c
+++ b/src/search.c
@@ -2482,4 +2482,64 @@ void uvg_search_lcu(encoder_state_t * const state, const int x, const int y, con
   if (state->encoder_control->cfg.jccr) {
     copy_coeffs(work_tree.coeff.joint_uv, coeff->joint_uv, LCU_WIDTH_C, LCU_WIDTH_C, LCU_WIDTH_C);
   }
+
+  if (start.x + 64 <= state->tile->frame->width && start.y + 64 <= state->tile->frame->height) {
+    uint8_t     type = 2;
+    uint8_t     buffer[8192 * 2];
+    UVG_CLOCK_T time;
+    UVG_GET_TIME(&time);
+#ifdef _MSC_VER
+    uint64_t time_high = time.dwHighDateTime & 0x000fffff;
+    time_high <<= 32;
+    uint64_t time_low  = time.dwLowDateTime;
+    uint64_t timestamp = time_high | time_low;
+    timestamp *= 100;
+#else
+    uint64_t timestamp = time.tv_sec * 1000000000 + time.tv_nsec;
+#endif
+    uint64_t bytes = 0;
+    memcpy(buffer, &type, 1);
+    bytes++;
+    bytes += 2;
+    memcpy(buffer + bytes, &timestamp, 8);
+    bytes += 8;
+    memcpy(buffer + bytes, &state->frame->num, 1);
+    bytes++;
+    uint8_t s = 64;
+    memcpy(buffer + bytes, &split_tree, 1);
+    bytes++;
+    memcpy(buffer + bytes, &s, 1);
+    bytes++;
+
+    uint64_t count = 0;
+    recreate(&work_tree, &start, split_tree.current_depth, buffer, &bytes, &count);
+    memcpy(buffer + 1, &count, 2);
+
+    uvg_pixels_blit(
+      work_tree.rec.y,
+      buffer + bytes,
+      LCU_WIDTH,
+      LCU_WIDTH,
+      LCU_WIDTH,
+      LCU_WIDTH);
+    bytes += LCU_WIDTH * LCU_WIDTH;
+    uvg_pixels_blit(
+      work_tree.rec.u,
+      buffer + bytes,
+      LCU_WIDTH_C,
+      LCU_WIDTH_C,
+      LCU_WIDTH_C,
+      LCU_WIDTH_C);
+    bytes += LCU_WIDTH_C * LCU_WIDTH_C;
+    uvg_pixels_blit(
+      work_tree.rec.v,
+      buffer + bytes,
+      LCU_WIDTH_C,
+      LCU_WIDTH_C,
+      LCU_WIDTH_C,
+      LCU_WIDTH_C);
+    bytes += LCU_WIDTH_C * LCU_WIDTH_C;
+
+    zmq_send(state->send_socket, buffer, bytes, 0);
+  }
 }