1 files changed, 96 insertions, 0 deletions
diff --git a/tuner/src/com/google/android/exoplayer/MediaFormatUtil.java b/tuner/src/com/google/android/exoplayer/MediaFormatUtil.java
new file mode 100644
index 00000000..c7571b23
--- /dev/null
+++ b/tuner/src/com/google/android/exoplayer/MediaFormatUtil.java
@@ -0,0 +1,96 @@
+/*
+ * Copyright (C) 2016 The Android Open Source Project
+ *
+ * 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 com.google.android.exoplayer;
+
+import android.support.annotation.Nullable;
+import com.google.android.exoplayer.util.MimeTypes;
+import java.nio.ByteBuffer;
+import java.util.ArrayList;
+
+/** {@link MediaFormat} creation helper util */
+public class MediaFormatUtil {
+
+    /**
+     * Creates {@link MediaFormat} from {@link android.media.MediaFormat}. Since {@link
+     * com.google.android.exoplayer.TrackRenderer} uses {@link MediaFormat}, {@link
+     * android.media.MediaFormat} should be converted to be used with ExoPlayer.
+     */
+    public static MediaFormat createMediaFormat(android.media.MediaFormat format) {
+        String mimeType = format.getString(android.media.MediaFormat.KEY_MIME);
+        String language = getOptionalStringV16(format, android.media.MediaFormat.KEY_LANGUAGE);
+        int maxInputSize =
+                getOptionalIntegerV16(format, android.media.MediaFormat.KEY_MAX_INPUT_SIZE);
+        int width = getOptionalIntegerV16(format, android.media.MediaFormat.KEY_WIDTH);
+        int height = getOptionalIntegerV16(format, android.media.MediaFormat.KEY_HEIGHT);
+        int rotationDegrees = getOptionalIntegerV16(format, "rotation-degrees");
+        int channelCount =
+                getOptionalIntegerV16(format, android.media.MediaFormat.KEY_CHANNEL_COUNT);
+        int sampleRate = getOptionalIntegerV16(format, android.media.MediaFormat.KEY_SAMPLE_RATE);
+        int encoderDelay = getOptionalIntegerV16(format, "encoder-delay");
+        int encoderPadding = getOptionalIntegerV16(format, "encoder-padding");
+        ArrayList<byte[]> initializationData = new ArrayList<>();
+        for (int i = 0; format.containsKey("csd-" + i); i++) {
+            ByteBuffer buffer = format.getByteBuffer("csd-" + i);
+            byte[] data = new byte[buffer.limit()];
+            buffer.get(data);
+            initializationData.add(data);
+            buffer.flip();
+        }
+        long durationUs =
+                format.containsKey(android.media.MediaFormat.KEY_DURATION)
+                        ? format.getLong(android.media.MediaFormat.KEY_DURATION)
+                        : C.UNKNOWN_TIME_US;
+        int pcmEncoding =
+                MimeTypes.AUDIO_RAW.equals(mimeType) ? C.ENCODING_PCM_16BIT : MediaFormat.NO_VALUE;
+        MediaFormat mediaFormat =
+                new MediaFormat(
+                        null, // trackId
+                        mimeType,
+                        MediaFormat.NO_VALUE,
+                        maxInputSize,
+                        durationUs,
+                        width,
+                        height,
+                        rotationDegrees,
+                        MediaFormat.NO_VALUE,
+                        channelCount,
+                        sampleRate,
+                        language,
+                        MediaFormat.OFFSET_SAMPLE_RELATIVE,
+                        initializationData,
+                        false,
+                        MediaFormat.NO_VALUE,
+                        MediaFormat.NO_VALUE,
+                        pcmEncoding,
+                        encoderDelay,
+                        encoderPadding,
+                        null, // projectionData
+                        MediaFormat.NO_VALUE,
+                        null // colorValue
+                        );
+        mediaFormat.setFrameworkFormatV16(format);
+        return mediaFormat;
+    }
+
+    @Nullable
+    private static String getOptionalStringV16(android.media.MediaFormat format, String key) {
+        return format.containsKey(key) ? format.getString(key) : null;
+    }
+
+    private static int getOptionalIntegerV16(android.media.MediaFormat format, String key) {
+        return format.containsKey(key) ? format.getInteger(key) : MediaFormat.NO_VALUE;
+    }
+}