Fix Failed to stop the muxer (#159)

deepmedia · Aug 11, 2024 · d158cd6 · d158cd6
1 parent 07ecaed
commit d158cd6
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/lib/src/main/java/com/otaliastudios/transcoder/internal/data/Writer.kt b/lib/src/main/java/com/otaliastudios/transcoder/internal/data/Writer.kt
@@ -39,14 +39,16 @@ internal class Writer(
     override fun step(state: State.Ok<WriterData>, fresh: Boolean): State<Unit> {
         val (buffer, timestamp, flags) = state.value
         val eos = state is State.Eos
-        info.set(
-                buffer.position(),
-                buffer.remaining(),
-                timestamp,
-                if (eos) {
-                    flags and MediaCodec.BUFFER_FLAG_END_OF_STREAM
-                } else flags
-        )
+        if (eos) {
+            info.set(0, 0, 0, flags and MediaCodec.BUFFER_FLAG_END_OF_STREAM)
+        } else {
+            info.set(
+                    buffer.position(),
+                    buffer.remaining(),
+                    timestamp,
+                    flags
+            )
+        }
         sink.writeTrack(track, buffer, info)
         state.value.release()
         return if (eos) State.Eos(Unit) else State.Ok(Unit)