1227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks/* 2227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * Copyright (C) 2012 The Android Open Source Project 3227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * 4227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * Licensed under the Apache License, Version 2.0 (the "License"); 5227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * you may not use this file except in compliance with the License. 6227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * You may obtain a copy of the License at 7227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * 8227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * http://www.apache.org/licenses/LICENSE-2.0 9227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * 10227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * Unless required by applicable law or agreed to in writing, software 11227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * distributed under the License is distributed on an "AS IS" BASIS, 12227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * See the License for the specific language governing permissions and 14227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * limitations under the License. 15227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks */ 16227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 17227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendrickspackage androidx.media.filterfw.decoder; 18227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 19227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport android.annotation.TargetApi; 20227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport android.media.MediaCodec; 21227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport android.media.MediaCodec.BufferInfo; 22227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport android.media.MediaFormat; 23227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 24227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport androidx.media.filterfw.FrameValue; 25227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 26227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport java.io.ByteArrayOutputStream; 27227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport java.io.IOException; 28227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport java.nio.ByteBuffer; 29227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 30227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks/** 31227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * {@link TrackDecoder} for decoding audio tracks. 32227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * 33227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * TODO: find out if we always get 16 bits per channel and document. 34227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks */ 35227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks@TargetApi(16) 36227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendrickspublic class AudioTrackDecoder extends TrackDecoder { 37227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 38227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks private final ByteArrayOutputStream mAudioByteStream; // Guarded by mAudioByteStreamLock. 39227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks private final Object mAudioByteStreamLock; 40227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 41227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks private int mAudioSampleRate; 42227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks private int mAudioChannelCount; 43227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks private long mAudioPresentationTimeUs; 44227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 45227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks public AudioTrackDecoder(int trackIndex, MediaFormat format, Listener listener) { 46227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks super(trackIndex, format, listener); 47227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 48227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks if (!DecoderUtil.isAudioFormat(format)) { 49227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks throw new IllegalArgumentException( 50227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks "AudioTrackDecoder can only be used with audio formats"); 51227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 52227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 53227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks mAudioByteStream = new ByteArrayOutputStream(); 54227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks mAudioByteStreamLock = new Object(); 55227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 56227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks mAudioSampleRate = format.getInteger(MediaFormat.KEY_SAMPLE_RATE); 57227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks mAudioChannelCount = format.getInteger(MediaFormat.KEY_CHANNEL_COUNT); 58227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 59227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 60227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks @Override 61227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks protected MediaCodec initMediaCodec(MediaFormat format) { 6283511d2f49c9e272f328730586c3d0a7852247f2Andy Hung MediaCodec mediaCodec; 6383511d2f49c9e272f328730586c3d0a7852247f2Andy Hung try { 6483511d2f49c9e272f328730586c3d0a7852247f2Andy Hung mediaCodec = MediaCodec.createDecoderByType( 6583511d2f49c9e272f328730586c3d0a7852247f2Andy Hung format.getString(MediaFormat.KEY_MIME)); 6683511d2f49c9e272f328730586c3d0a7852247f2Andy Hung } catch (IOException e) { 6783511d2f49c9e272f328730586c3d0a7852247f2Andy Hung throw new RuntimeException( 6883511d2f49c9e272f328730586c3d0a7852247f2Andy Hung "failed to create decoder for " 6983511d2f49c9e272f328730586c3d0a7852247f2Andy Hung + format.getString(MediaFormat.KEY_MIME), e); 7083511d2f49c9e272f328730586c3d0a7852247f2Andy Hung } 71227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks mediaCodec.configure(format, null, null, 0); 72227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks return mediaCodec; 73227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 74227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 75227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks @Override 76227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks protected boolean onDataAvailable( 77227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks MediaCodec codec, ByteBuffer[] buffers, int bufferIndex, BufferInfo info) { 78227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks ByteBuffer buffer = buffers[bufferIndex]; 79227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks byte[] data = new byte[info.size]; 80227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks buffer.position(info.offset); 81227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks buffer.get(data, 0, info.size); 82227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 83227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks synchronized (mAudioByteStreamLock) { 84227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks try { 85227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks if (mAudioByteStream.size() == 0 && data.length > 0) { 86227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks mAudioPresentationTimeUs = info.presentationTimeUs; 87227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 88227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 89227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks mAudioByteStream.write(data); 90227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } catch (IOException e) { 91227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks // Just drop the audio sample. 92227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 93227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 94227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 95227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks buffer.clear(); 96227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks codec.releaseOutputBuffer(bufferIndex, false); 97227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks notifyListener(); 98227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks return true; 99227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 100227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 101227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks /** 102227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * Fills the argument {@link FrameValue} with an audio sample containing the audio that was 103227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * decoded since the last call of this method. The decoder's buffer is cleared as a result. 104227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks */ 105227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks public void grabSample(FrameValue audioFrame) { 106227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks synchronized (mAudioByteStreamLock) { 107227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks if (audioFrame != null) { 108227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks AudioSample sample = new AudioSample( 109227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks mAudioSampleRate, mAudioChannelCount, mAudioByteStream.toByteArray()); 110227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks audioFrame.setValue(sample); 111227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks audioFrame.setTimestamp(mAudioPresentationTimeUs * 1000); 112227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 113227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks clearBuffer(); 114227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 115227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 116227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 117227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks /** 118227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * Clears the decoder's buffer. 119227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks */ 120227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks public void clearBuffer() { 121227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks synchronized (mAudioByteStreamLock) { 122227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks mAudioByteStream.reset(); 123227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 124227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks } 125227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks 126227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks} 127