fix StreamBase record only audio/video

pedroSG94 · Jan 24, 2025 · 7b1a37c · 7b1a37c
1 parent de6d6f2
commit 7b1a37c
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 3 deletions.
diff --git a/library/src/main/java/com/pedro/library/base/StreamBase.kt b/library/src/main/java/com/pedro/library/base/StreamBase.kt
@@ -37,6 +37,7 @@ import com.pedro.encoder.audio.AudioEncoder
 import com.pedro.encoder.audio.GetAudioData
 import com.pedro.encoder.input.audio.GetMicrophoneData
 import com.pedro.encoder.input.sources.audio.AudioSource
+import com.pedro.encoder.input.sources.audio.NoAudioSource
 import com.pedro.encoder.input.sources.video.NoVideoSource
 import com.pedro.encoder.input.sources.video.VideoSource
 import com.pedro.encoder.utils.CodecUtil
@@ -569,7 +570,8 @@ abstract class StreamBase(
     }
 
     override fun onAudioFormat(mediaFormat: MediaFormat) {
-      recordController.setAudioFormat(mediaFormat)
+      val isOnlyAudio = videoSource is NoVideoSource
+      recordController.setAudioFormat(mediaFormat, isOnlyAudio)
     }
   }
 
@@ -585,7 +587,10 @@ abstract class StreamBase(
     }
 
     override fun onVideoFormat(mediaFormat: MediaFormat) {
-      if (!differentRecordResolution) recordController.setVideoFormat(mediaFormat)
+      if (!differentRecordResolution) {
+        val isOnlyVideo = audioSource is NoAudioSource
+        recordController.setVideoFormat(mediaFormat, isOnlyVideo)
+      }
     }
   }
 
@@ -598,7 +603,8 @@ abstract class StreamBase(
     }
 
     override fun onVideoFormat(mediaFormat: MediaFormat) {
-      recordController.setVideoFormat(mediaFormat)
+      val isOnlyVideo = audioSource is NoAudioSource
+      recordController.setVideoFormat(mediaFormat, isOnlyVideo)
     }
   }
 

diff --git a/library/src/main/java/com/pedro/library/util/AndroidMuxerRecordController.java b/library/src/main/java/com/pedro/library/util/AndroidMuxerRecordController.java
@@ -155,6 +155,7 @@ private void init() {
   }
 
   private void write(int track, ByteBuffer byteBuffer, MediaCodec.BufferInfo info) {
+    if (track == -1) return;
     try {
       mediaMuxer.writeSampleData(track, byteBuffer, info);
       if (bitrateManager != null) bitrateManager.calculateBitrate(info.size * 8L, ExtensionsKt.getSuspendContext());

diff --git a/library/src/main/java/com/pedro/library/util/AndroidMuxerWebmRecordController.java b/library/src/main/java/com/pedro/library/util/AndroidMuxerWebmRecordController.java
@@ -136,6 +136,7 @@ private void init() {
   }
 
   private void write(int track, ByteBuffer byteBuffer, MediaCodec.BufferInfo info) {
+    if (track == -1) return;
     try {
       mediaMuxer.writeSampleData(track, byteBuffer, info);
       if (bitrateManager != null) bitrateManager.calculateBitrate(info.size * 8L, ExtensionsKt.getSuspendContext());