From 72a26548a9c9643c5cfc26b3f3721f430f9c1aa9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?William=20B=C3=A9rub=C3=A9?= <wberube@null.net>
Date: Sun, 14 Jul 2024 16:31:42 -0400
Subject: [PATCH] Returned NALU packet infos on infinity6 are broken (but not
 on i6b0/e)

---
 src/hal/star/i6_hal.c | 34 ++++++++++++++++++++++++++++++++--
 src/hal/support.c     |  2 +-
 2 files changed, 33 insertions(+), 3 deletions(-)

diff --git a/src/hal/star/i6_hal.c b/src/hal/star/i6_hal.c
index 72f7ec6..5468443 100644
--- a/src/hal/star/i6_hal.c
+++ b/src/hal/star/i6_hal.c
@@ -579,7 +579,8 @@ int i6_video_create(char index, hal_vidconfig *config)
     attrib->maxHeight = config->height;
     attrib->maxWidth = config->width;
     attrib->bufSize = config->height * config->width;
-    attrib->profile = config->profile;
+    attrib->profile = MIN((series == 0xEF || config->codec == HAL_VIDCODEC_H265) ? 1 : 2,
+        config->profile);
     attrib->byFrame = 1;
     attrib->height = config->height;
     attrib->width = config->width;
@@ -836,7 +837,36 @@ void *i6_video_thread(void)
                             outPack[j].data = pack->data;
                             outPack[j].length = pack->length;
                             outPack[j].naluCnt = pack->packNum;
-                            switch (i6_state[i].payload) {
+                            if (series == 0xEF) {
+                                signed char n = 0;
+                                switch (i6_state[i].payload) {
+                                    case HAL_VIDCODEC_H264:
+                                        for (unsigned int p = 0; p < pack->length - 4; p++) {
+                                            if (outPack[j].data[p] || outPack[j].data[p + 1] ||
+                                                outPack[j].data[p + 2] || outPack[j].data[p + 3] != 1) continue;
+                                            outPack[0].nalu[n].type = outPack[j].data[p + 4] & 0x1F;
+                                            outPack[0].nalu[n++].offset = p;
+                                            if (n == (outPack[j].naluCnt)) break;
+                                        }
+                                        break;
+                                    case HAL_VIDCODEC_H265:
+                                        for (unsigned int p = 0; p < pack->length - 4; p++) {
+                                            if (outPack[j].data[p] || outPack[j].data[p + 1] ||
+                                                outPack[j].data[p + 2] || outPack[j].data[p + 3] != 1) continue;
+                                            outPack[0].nalu[n].type = (outPack[j].data[p + 4] & 0x7E) >> 1;
+                                            outPack[0].nalu[n++].offset = p;
+                                            if (n == (outPack[j].naluCnt)) break;
+                                        }
+                                        break;
+                                }
+
+                                outPack[0].naluCnt = n;
+                                outPack[0].nalu[n].offset = pack->length;
+                                for (n = 0; n < outPack[0].naluCnt; n++)
+                                    outPack[0].nalu[n].length = 
+                                        outPack[0].nalu[n + 1].offset -
+                                        outPack[0].nalu[n].offset;
+                            } else switch (i6_state[i].payload) {
                                 case HAL_VIDCODEC_H264:
                                     for (char k = 0; k < outPack[j].naluCnt; k++) {
                                         outPack[j].nalu[k].length =
diff --git a/src/hal/support.c b/src/hal/support.c
index a6527a4..1ec99b0 100644
--- a/src/hal/support.c
+++ b/src/hal/support.c
@@ -69,7 +69,7 @@ void hal_identify(void) {
             case 0xF1: // Pudding (6E)
             case 0xF2: // Ispahan (6B0)
                 plat = HAL_PLATFORM_I6;
-                strcpy(chipId, val == 0xEF ? 
+                strcpy(chipId, series == 0xEF ? 
                     "SSC32x" : "SSC33x");
                 chnCount = I6_VENC_CHN_NUM;
                 chnState = (hal_chnstate*)i6_state;