From: Kostya Shishkov <kostya.shishkov@gmail.com>
Date: Sat, 22 Mar 2025 10:50:47 +0000 (+0100)
Subject: avimux: add palette change chunks to the index
X-Git-Url: https://git.nihav.org/?a=commitdiff_plain;h=fb37b4715eb6407484b277b43d1ac563fef25b7a;p=nihav.git

avimux: add palette change chunks to the index
---

diff --git a/nihav-commonfmt/src/muxers/avi.rs b/nihav-commonfmt/src/muxers/avi.rs
index 8071c4d..a739f49 100644
--- a/nihav-commonfmt/src/muxers/avi.rs
+++ b/nihav-commonfmt/src/muxers/avi.rs
@@ -286,9 +286,15 @@ impl<'a> MuxCore<'a> for AVIMuxer<'a> {
                     let end_clr = 256usize;
                     if start_clr < end_clr {
                         let chunk_len = ((end_clr - start_clr) as u32) * 4 + 4;
-                        self.bw.write_byte(b'0' + ((str_num / 10) as u8))?;
-                        self.bw.write_byte(b'0' + ((str_num % 10) as u8))?;
-                        self.bw.write_buf(b"pc")?;
+                        let mut tag = gen_chunk_tag(StreamType::Video, str_num as u32);
+                        tag[2] = b'p';
+                        tag[3] = b'c';
+                        self.index.push(IdxEntry {
+                                tag,
+                                key:    false,
+                                pos:    self.bw.tell(),
+                                len:    chunk_len });
+                        self.bw.write_buf(&tag)?;
                         self.bw.write_u32le(chunk_len)?;
                         self.bw.write_byte(start_clr as u8)?;
                         self.bw.write_byte((end_clr - start_clr) as u8)?;
@@ -326,7 +332,10 @@ impl<'a> MuxCore<'a> for AVIMuxer<'a> {
             self.bw.write_u32le((self.index.len() * 16) as u32)?;
             for item in self.index.iter() {
                 self.bw.write_buf(&item.tag)?;
-                self.bw.write_u32le(if item.key { 0x10 } else { 0 })?;
+                let flags = if item.key { 0x10 }
+                            else if &item.tag[2..] == b"pc" { 0x100 }
+                            else { 0 };
+                self.bw.write_u32le(flags)?;
                 self.bw.write_u32le(item.pos as u32)?;
                 self.bw.write_u32le(item.len)?;
             }