VUIを設定すると出力が異常になる場合があったのを修正。( #18 )

rigaya · Nov 23, 2020 · e34c016 · e34c016
1 parent da5ee0c
commit e34c016
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/VCECore/rgy_output.cpp b/VCECore/rgy_output.cpp
@@ -339,6 +339,7 @@ RGY_ERR RGYOutputRaw::WriteNextFrame(RGYBitstream *pBitstream) {
                 }
                 memmove(pBitstream->data() + next_nal_new_offset, pBitstream->data() + next_nal_orig_offset, stream_orig_length - next_nal_orig_offset);
                 memcpy(pBitstream->data() + sps_nal_offset, pkt.data, pkt.size);
+                pBitstream->setSize(new_data_size);
                 av_packet_unref(&pkt);
             }
         }

diff --git a/VCECore/rgy_output_avcodec.cpp b/VCECore/rgy_output_avcodec.cpp
@@ -2345,6 +2345,7 @@ RGY_ERR RGYOutputAvcodec::WriteNextFrameInternal(RGYBitstream *bitstream, int64_
             }
             memmove(bitstream->data() + next_nal_new_offset, bitstream->data() + next_nal_orig_offset, stream_orig_length - next_nal_orig_offset);
             memcpy(bitstream->data() + sps_nal_offset, pkt.data, pkt.size);
+            bitstream->setSize(new_data_size);
             av_packet_unref(&pkt);
         }
     }