1 files changed, 100 insertions, 0 deletions
diff --git a/libs/android-transcoder/src/main/java/net/ypresto/androidtranscoder/engine/PassThroughTrackTranscoder.java b/libs/android-transcoder/src/main/java/net/ypresto/androidtranscoder/engine/PassThroughTrackTranscoder.java
new file mode 100644
index 000000000..7608dac86
--- /dev/null
+++ b/libs/android-transcoder/src/main/java/net/ypresto/androidtranscoder/engine/PassThroughTrackTranscoder.java
@@ -0,0 +1,100 @@
+/*
+ * Copyright (C) 2014 Yuya Tanaka
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package net.ypresto.androidtranscoder.engine;
+
+import android.annotation.SuppressLint;
+import android.media.MediaCodec;
+import android.media.MediaExtractor;
+import android.media.MediaFormat;
+
+import java.nio.ByteBuffer;
+import java.nio.ByteOrder;
+
+public class PassThroughTrackTranscoder implements TrackTranscoder {
+    private final MediaExtractor mExtractor;
+    private final int mTrackIndex;
+    private final QueuedMuxer mMuxer;
+    private final QueuedMuxer.SampleType mSampleType;
+    private final MediaCodec.BufferInfo mBufferInfo = new MediaCodec.BufferInfo();
+    private int mBufferSize;
+    private ByteBuffer mBuffer;
+    private boolean mIsEOS;
+    private MediaFormat mActualOutputFormat;
+    private long mWrittenPresentationTimeUs;
+
+    public PassThroughTrackTranscoder(MediaExtractor extractor, int trackIndex,
+                                      QueuedMuxer muxer, QueuedMuxer.SampleType sampleType) {
+        mExtractor = extractor;
+        mTrackIndex = trackIndex;
+        mMuxer = muxer;
+        mSampleType = sampleType;
+
+        mActualOutputFormat = mExtractor.getTrackFormat(mTrackIndex);
+        mMuxer.setOutputFormat(mSampleType, mActualOutputFormat);
+        mBufferSize = mActualOutputFormat.getInteger(MediaFormat.KEY_MAX_INPUT_SIZE);
+        mBuffer = ByteBuffer.allocateDirect(mBufferSize).order(ByteOrder.nativeOrder());
+    }
+
+    @Override
+    public void setup() {
+    }
+
+    @Override
+    public MediaFormat getDeterminedFormat() {
+        return mActualOutputFormat;
+    }
+
+    @SuppressLint("Assert")
+    @Override
+    public boolean stepPipeline() {
+        if (mIsEOS) return false;
+        int trackIndex = mExtractor.getSampleTrackIndex();
+        if (trackIndex < 0) {
+            mBuffer.clear();
+            mBufferInfo.set(0, 0, 0, MediaCodec.BUFFER_FLAG_END_OF_STREAM);
+            mMuxer.writeSampleData(mSampleType, mBuffer, mBufferInfo);
+            mIsEOS = true;
+            return true;
+        }
+        if (trackIndex != mTrackIndex) return false;
+
+        mBuffer.clear();
+        int sampleSize = mExtractor.readSampleData(mBuffer, 0);
+        assert sampleSize <= mBufferSize;
+        boolean isKeyFrame = (mExtractor.getSampleFlags() & MediaExtractor.SAMPLE_FLAG_SYNC) != 0;
+        int flags = isKeyFrame ? MediaCodec.BUFFER_FLAG_SYNC_FRAME : 0;
+        mBufferInfo.set(0, sampleSize, mExtractor.getSampleTime(), flags);
+        mMuxer.writeSampleData(mSampleType, mBuffer, mBufferInfo);
+        mWrittenPresentationTimeUs = mBufferInfo.presentationTimeUs;
+
+        mExtractor.advance();
+        return true;
+    }
+
+    @Override
+    public long getWrittenPresentationTimeUs() {
+        return mWrittenPresentationTimeUs;
+    }
+
+    @Override
+    public boolean isFinished() {
+        return mIsEOS;
+    }
+
+    @Override
+    public void release() {
+    }
+}