Revert "Merge pull request pedroSG94#1882 from pedroSG94/fix/record-audio-ts"

This reverts commit c2b77fd, reversing
changes made to 87ac138.

Author: TroyTheGod

encoder/src/main/java/com/pedro/encoder/BaseEncoder.java

@@ -27,7 +27,6 @@
 import androidx.annotation.NonNull;
 import androidx.annotation.RequiresApi;
 
-import com.pedro.common.ExtensionsKt;
 import com.pedro.common.TimeUtils;
 import com.pedro.encoder.audio.G711Codec;
 import com.pedro.encoder.utils.CodecUtil;
@@ -37,7 +36,6 @@
 import java.util.concurrent.BlockingQueue;
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.Executors;
-import java.util.concurrent.PriorityBlockingQueue;
 
 /**
  * Created by pedro on 18/09/19.
@@ -51,12 +49,12 @@ public abstract class BaseEncoder implements EncoderCallback {
   private ExecutorService executorService;
   protected BlockingQueue<Frame> queue = new ArrayBlockingQueue<>(80);
   protected MediaCodec codec;
-  protected volatile long presentTimeUs;
+  protected long presentTimeUs;
   protected volatile boolean running = false;
   protected boolean isBufferMode = true;
   protected CodecUtil.CodecType codecType = CodecUtil.CodecType.FIRST_COMPATIBLE_FOUND;
   private MediaCodec.Callback callback;
-  private volatile long oldTimeStamp = 0L;
+  private long oldTimeStamp = 0L;
   protected boolean shouldReset = true;
   protected boolean prepared = false;
   private Handler handler;
@@ -134,10 +132,10 @@ private void initCodec() {
 
   protected void fixTimeStamp(MediaCodec.BufferInfo info) {
     if (oldTimeStamp > info.presentationTimeUs) {
-      final long currentTs = TimeUtils.getCurrentTimeMicro() - presentTimeUs;
-      info.presentationTimeUs = Math.max(currentTs, oldTimeStamp + 1);
+      info.presentationTimeUs = oldTimeStamp;
+    } else {
+      oldTimeStamp = info.presentationTimeUs;
     }
-    oldTimeStamp = info.presentationTimeUs;
   }
 
   private void reloadCodec(IllegalStateException e) {

library/src/main/java/com/pedro/library/base/recording/BaseRecordController.java

@@ -38,8 +38,10 @@ public abstract class BaseRecordController implements RecordController {
     protected Listener listener;
     protected int videoTrack = -1;
     protected int audioTrack = -1;
+    protected final MediaCodec.BufferInfo videoInfo = new MediaCodec.BufferInfo();
+    protected final MediaCodec.BufferInfo audioInfo = new MediaCodec.BufferInfo();
     protected BitrateManager bitrateManager;
-    protected volatile long startTs = 0;
+    protected long startTs = 0;
     protected RecordTracks tracks = RecordTracks.ALL;
 
     public void setVideoCodec(VideoCodec videoCodec) {
@@ -98,11 +100,11 @@ protected boolean isKeyFrame(ByteBuffer videoBuffer) {
     }
 
     //We can't reuse info because could produce stream issues
-    protected MediaCodec.BufferInfo updateFormat(MediaCodec.BufferInfo oldInfo) {
-        MediaCodec.BufferInfo info = new MediaCodec.BufferInfo();
+    protected void updateFormat(MediaCodec.BufferInfo newInfo, MediaCodec.BufferInfo oldInfo) {
         if (startTs <= 0) startTs = oldInfo.presentationTimeUs;
-        long ts = Math.max(0, oldInfo.presentationTimeUs - startTs - pauseTime);
-        info.set(oldInfo.offset, oldInfo.size, ts, oldInfo.flags);
-        return info;
+        newInfo.flags = oldInfo.flags;
+        newInfo.offset = oldInfo.offset;
+        newInfo.size = oldInfo.size;
+        newInfo.presentationTimeUs = Math.max(0, oldInfo.presentationTimeUs - startTs - pauseTime);
     }
 }

library/src/main/java/com/pedro/library/util/AacMuxerRecordController.java

@@ -108,8 +108,9 @@ public void recordVideo(ByteBuffer videoBuffer, MediaCodec.BufferInfo videoInfo)
     @RequiresApi(api = Build.VERSION_CODES.JELLY_BEAN_MR2)
     public void recordAudio(ByteBuffer audioBuffer, MediaCodec.BufferInfo audioInfo) {
         if (status == Status.RECORDING) {
+            updateFormat(this.audioInfo, audioInfo);
             //we need duplicate buffer to avoid problems with the buffer
-            write(audioBuffer.duplicate(), updateFormat(audioInfo));
+            write(audioBuffer.duplicate(), this.audioInfo);
         }
     }
 

library/src/main/java/com/pedro/library/util/AndroidMuxerRecordController.java

@@ -20,7 +20,6 @@
 import android.media.MediaFormat;
 import android.media.MediaMuxer;
 import android.os.Build;
-import android.util.Log;
 
 import androidx.annotation.NonNull;
 import androidx.annotation.Nullable;
@@ -115,14 +114,16 @@ public void recordVideo(ByteBuffer videoBuffer, MediaCodec.BufferInfo videoInfo)
       if (listener != null) listener.onStatusChange(status);
     }
     if (status == Status.RECORDING && tracks != RecordTracks.AUDIO) {
-      write(videoTrack, videoBuffer, videoInfo);
+      updateFormat(this.videoInfo, videoInfo);
+      write(videoTrack, videoBuffer, this.videoInfo);
     }
   }
 
   @Override
   public void recordAudio(ByteBuffer audioBuffer, MediaCodec.BufferInfo audioInfo) {
     if (status == Status.RECORDING && tracks != RecordTracks.VIDEO) {
-      write(audioTrack, audioBuffer, audioInfo);
+      updateFormat(this.audioInfo, audioInfo);
+      write(audioTrack, audioBuffer, this.audioInfo);
     }
   }
 
@@ -154,12 +155,9 @@ private void init() {
 
   private void write(int track, ByteBuffer byteBuffer, MediaCodec.BufferInfo info) {
     if (track == -1) return;
-    String trackString = track == audioTrack ? "Audio" : "Video";
     try {
-      MediaCodec.BufferInfo i =  updateFormat(info);
-      Log.i(TAG, trackString + ", ts: " + i.presentationTimeUs + ", flag: " + i.flags);
-      mediaMuxer.writeSampleData(track, byteBuffer, i);
-      if (bitrateManager != null) bitrateManager.calculateBitrate(i.size * 8L, ExtensionsKt.getSuspendContext());
+      mediaMuxer.writeSampleData(track, byteBuffer, info);
+      if (bitrateManager != null) bitrateManager.calculateBitrate(info.size * 8L, ExtensionsKt.getSuspendContext());
     } catch (Exception e) {
       if (listener != null) listener.onError(e);
     }

library/src/main/java/com/pedro/library/util/AndroidMuxerWebmRecordController.java

@@ -114,7 +114,8 @@ public void recordVideo(ByteBuffer videoBuffer, MediaCodec.BufferInfo videoInfo)
   @Override
   public void recordAudio(ByteBuffer audioBuffer, MediaCodec.BufferInfo audioInfo) {
     if (status == Status.RECORDING) {
-      write(audioTrack, audioBuffer, audioInfo);
+      updateFormat(this.audioInfo, audioInfo);
+      write(audioTrack, audioBuffer, this.audioInfo);
     }
   }
 
@@ -145,7 +146,7 @@ private void init() {
   private void write(int track, ByteBuffer byteBuffer, MediaCodec.BufferInfo info) {
     if (track == -1) return;
     try {
-      mediaMuxer.writeSampleData(track, byteBuffer, updateFormat(info));
+      mediaMuxer.writeSampleData(track, byteBuffer, info);
       if (bitrateManager != null) bitrateManager.calculateBitrate(info.size * 8L, ExtensionsKt.getSuspendContext());
     } catch (Exception e) {
       if (listener != null) listener.onError(e);