AudioPlayer.cpp revision 20111aa043c5f404472bc63b90bc5aad906b1101 
1/*
2 * Copyright (C) 2009 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *      http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17#undef NDEBUG
18#include <assert.h>
19
20#define LOG_TAG "AudioPlayer"
21#include <utils/Log.h>
22
23#include <media/AudioTrack.h>
24#include <media/stagefright/AudioPlayer.h>
25#include <media/stagefright/MediaSource.h>
26#include <media/stagefright/MetaData.h>
27
28namespace android {
29
30AudioPlayer::AudioPlayer(const sp<MediaPlayerBase::AudioSink> &audioSink)
31    : mSource(NULL),
32      mAudioTrack(NULL),
33      mInputBuffer(NULL),
34      mSampleRate(0),
35      mLatencyUs(0),
36      mFrameSize(0),
37      mNumFramesPlayed(0),
38      mPositionTimeMediaUs(-1),
39      mPositionTimeRealUs(-1),
40      mSeeking(false),
41      mStarted(false),
42      mAudioSink(audioSink) {
43}
44
45AudioPlayer::~AudioPlayer() {
46    if (mStarted) {
47        stop();
48    }
49}
50
51void AudioPlayer::setSource(MediaSource *source) {
52    assert(mSource == NULL);
53    mSource = source;
54}
55
56void AudioPlayer::start() {
57    assert(!mStarted);
58    assert(mSource != NULL);
59
60    status_t err = mSource->start();
61    assert(err == OK);
62
63    sp<MetaData> format = mSource->getFormat();
64    const char *mime;
65    bool success = format->findCString(kKeyMIMEType, &mime);
66    assert(success);
67    assert(!strcasecmp(mime, "audio/raw"));
68
69    success = format->findInt32(kKeySampleRate, &mSampleRate);
70    assert(success);
71
72    int32_t numChannels;
73    success = format->findInt32(kKeyChannelCount, &numChannels);
74    assert(success);
75
76    if (mAudioSink.get() != NULL) {
77        status_t err = mAudioSink->open(
78                mSampleRate, numChannels, AudioSystem::PCM_16_BIT,
79                DEFAULT_AUDIOSINK_BUFFERCOUNT,
80                &AudioPlayer::AudioSinkCallback, this);
81        assert(err == OK);
82
83        mLatencyUs = (int64_t)mAudioSink->latency() * 1000;
84        mFrameSize = mAudioSink->frameSize();
85
86        mAudioSink->start();
87    } else {
88        mAudioTrack = new AudioTrack(
89                AudioSystem::MUSIC, mSampleRate, AudioSystem::PCM_16_BIT,
90                numChannels, 8192, 0, &AudioCallback, this, 0);
91
92        assert(mAudioTrack->initCheck() == OK);
93
94        mLatencyUs = (int64_t)mAudioTrack->latency() * 1000;
95        mFrameSize = mAudioTrack->frameSize();
96
97        mAudioTrack->start();
98    }
99
100    mStarted = true;
101}
102
103void AudioPlayer::pause() {
104    assert(mStarted);
105
106    if (mAudioSink.get() != NULL) {
107        mAudioSink->pause();
108    } else {
109        mAudioTrack->stop();
110    }
111}
112
113void AudioPlayer::resume() {
114    assert(mStarted);
115
116    if (mAudioSink.get() != NULL) {
117        mAudioSink->start();
118    } else {
119        mAudioTrack->start();
120    }
121}
122
123void AudioPlayer::stop() {
124    assert(mStarted);
125
126    if (mAudioSink.get() != NULL) {
127        mAudioSink->stop();
128    } else {
129        mAudioTrack->stop();
130
131        delete mAudioTrack;
132        mAudioTrack = NULL;
133    }
134
135    // Make sure to release any buffer we hold onto so that the
136    // source is able to stop().
137    if (mInputBuffer != NULL) {
138        LOGI("AudioPlayer releasing input buffer.");
139
140        mInputBuffer->release();
141        mInputBuffer = NULL;
142    }
143
144    mSource->stop();
145
146    mNumFramesPlayed = 0;
147    mPositionTimeMediaUs = -1;
148    mPositionTimeRealUs = -1;
149    mSeeking = false;
150    mStarted = false;
151}
152
153// static
154void AudioPlayer::AudioCallback(int event, void *user, void *info) {
155    static_cast<AudioPlayer *>(user)->AudioCallback(event, info);
156}
157
158// static
159void AudioPlayer::AudioSinkCallback(
160        MediaPlayerBase::AudioSink *audioSink,
161        void *buffer, size_t size, void *cookie) {
162    AudioPlayer *me = (AudioPlayer *)cookie;
163
164    me->fillBuffer(buffer, size);
165}
166
167void AudioPlayer::AudioCallback(int event, void *info) {
168    if (event != AudioTrack::EVENT_MORE_DATA) {
169        return;
170    }
171
172    AudioTrack::Buffer *buffer = (AudioTrack::Buffer *)info;
173    fillBuffer(buffer->raw, buffer->size);
174}
175
176void AudioPlayer::fillBuffer(void *data, size_t size) {
177    if (mNumFramesPlayed == 0) {
178        LOGI("AudioCallback");
179    }
180
181    size_t size_done = 0;
182    size_t size_remaining = size;
183    while (size_remaining > 0) {
184        MediaSource::ReadOptions options;
185
186        {
187            Mutex::Autolock autoLock(mLock);
188
189            if (mSeeking) {
190                options.setSeekTo(mSeekTimeUs);
191
192                if (mInputBuffer != NULL) {
193                    mInputBuffer->release();
194                    mInputBuffer = NULL;
195                }
196                mSeeking = false;
197            }
198        }
199
200        if (mInputBuffer == NULL) {
201            status_t err = mSource->read(&mInputBuffer, &options);
202
203            assert((err == OK && mInputBuffer != NULL)
204                   || (err != OK && mInputBuffer == NULL));
205
206            if (err != OK) {
207                memset((char *)data + size_done, 0, size_remaining);
208                break;
209            }
210
211            int32_t units, scale;
212            bool success =
213                mInputBuffer->meta_data()->findInt32(kKeyTimeUnits, &units);
214            success = success &&
215                mInputBuffer->meta_data()->findInt32(kKeyTimeScale, &scale);
216            assert(success);
217
218            Mutex::Autolock autoLock(mLock);
219            mPositionTimeMediaUs = (int64_t)units * 1000000 / scale;
220            mPositionTimeRealUs =
221                ((mNumFramesPlayed + size_done / 4) * 1000000) / mSampleRate; // XXX
222        }
223
224        if (mInputBuffer->range_length() == 0) {
225            mInputBuffer->release();
226            mInputBuffer = NULL;
227
228            continue;
229        }
230
231        size_t copy = size_remaining;
232        if (copy > mInputBuffer->range_length()) {
233            copy = mInputBuffer->range_length();
234        }
235
236        memcpy((char *)data + size_done,
237               (const char *)mInputBuffer->data() + mInputBuffer->range_offset(),
238               copy);
239
240        mInputBuffer->set_range(mInputBuffer->range_offset() + copy,
241                                mInputBuffer->range_length() - copy);
242
243        size_done += copy;
244        size_remaining -= copy;
245    }
246
247    Mutex::Autolock autoLock(mLock);
248    mNumFramesPlayed += size / mFrameSize;
249}
250
251int64_t AudioPlayer::getRealTimeUs() {
252    Mutex::Autolock autoLock(mLock);
253    return getRealTimeUsLocked();
254}
255
256int64_t AudioPlayer::getRealTimeUsLocked() const {
257    return -mLatencyUs + (mNumFramesPlayed * 1000000) / mSampleRate;
258}
259
260int64_t AudioPlayer::getMediaTimeUs() {
261    Mutex::Autolock autoLock(mLock);
262
263    return mPositionTimeMediaUs + (getRealTimeUsLocked() - mPositionTimeRealUs);
264}
265
266bool AudioPlayer::getMediaTimeMapping(
267        int64_t *realtime_us, int64_t *mediatime_us) {
268    Mutex::Autolock autoLock(mLock);
269
270    *realtime_us = mPositionTimeRealUs;
271    *mediatime_us = mPositionTimeMediaUs;
272
273    return mPositionTimeRealUs != -1 || mPositionTimeMediaUs != -1;
274}
275
276status_t AudioPlayer::seekTo(int64_t time_us) {
277    Mutex::Autolock autoLock(mLock);
278
279    mSeeking = true;
280    mSeekTimeUs = time_us;
281
282    return OK;
283}
284
285}
286