blob: 66764616cee008673b8a18d2921a0bc1e78248f2 [file] [log] [blame]
Andreas Huberafed0e12011-09-20 15:39:58 -07001/*
2 * Copyright (C) 2012 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
Chong Zhang7e892182014-08-05 11:58:21 -070017//#define LOG_NDEBUG 0
18#define LOG_TAG "GenericSource"
19
Andreas Huberafed0e12011-09-20 15:39:58 -070020#include "GenericSource.h"
21
22#include "AnotherPacketSource.h"
23
Chong Zhanga19f33e2014-08-07 15:35:07 -070024#include <media/IMediaHTTPService.h>
Andreas Huberafed0e12011-09-20 15:39:58 -070025#include <media/stagefright/foundation/ABuffer.h>
26#include <media/stagefright/foundation/ADebug.h>
27#include <media/stagefright/foundation/AMessage.h>
28#include <media/stagefright/DataSource.h>
29#include <media/stagefright/FileSource.h>
30#include <media/stagefright/MediaBuffer.h>
31#include <media/stagefright/MediaDefs.h>
32#include <media/stagefright/MediaExtractor.h>
33#include <media/stagefright/MediaSource.h>
34#include <media/stagefright/MetaData.h>
Lajos Molnarcc227032014-07-17 15:33:06 -070035#include "../../libstagefright/include/WVMExtractor.h"
Andreas Huberafed0e12011-09-20 15:39:58 -070036
37namespace android {
38
39NuPlayer::GenericSource::GenericSource(
Andreas Huberb5f25f02013-02-05 10:14:26 -080040 const sp<AMessage> &notify,
Lajos Molnarcc227032014-07-17 15:33:06 -070041 bool uidValid,
42 uid_t uid)
Andreas Huberb5f25f02013-02-05 10:14:26 -080043 : Source(notify),
Robert Shih3423bbd2014-07-16 15:47:09 -070044 mFetchSubtitleDataGeneration(0),
Lajos Molnare26940f2014-07-31 10:31:26 -070045 mFetchTimedTextDataGeneration(0),
Andreas Huberb5f25f02013-02-05 10:14:26 -080046 mDurationUs(0ll),
Lajos Molnarcc227032014-07-17 15:33:06 -070047 mAudioIsVorbis(false),
Chong Zhang3de157d2014-08-05 20:54:44 -070048 mIsWidevine(false),
Lajos Molnarcc227032014-07-17 15:33:06 -070049 mUIDValid(uidValid),
50 mUID(uid) {
Chong Zhanga19f33e2014-08-07 15:35:07 -070051 resetDataSource();
Andreas Huberafed0e12011-09-20 15:39:58 -070052 DataSource::RegisterDefaultSniffers();
Chong Zhang3de157d2014-08-05 20:54:44 -070053}
54
Chong Zhanga19f33e2014-08-07 15:35:07 -070055void NuPlayer::GenericSource::resetDataSource() {
56 mHTTPService.clear();
57 mUri.clear();
58 mUriHeaders.clear();
59 mFd = -1;
60 mOffset = 0;
61 mLength = 0;
62}
63
64status_t NuPlayer::GenericSource::setDataSource(
Chong Zhang3de157d2014-08-05 20:54:44 -070065 const sp<IMediaHTTPService> &httpService,
66 const char *url,
67 const KeyedVector<String8, String8> *headers) {
Chong Zhanga19f33e2014-08-07 15:35:07 -070068 resetDataSource();
Chong Zhang3de157d2014-08-05 20:54:44 -070069
Chong Zhanga19f33e2014-08-07 15:35:07 -070070 mHTTPService = httpService;
71 mUri = url;
Andreas Huberafed0e12011-09-20 15:39:58 -070072
Chong Zhanga19f33e2014-08-07 15:35:07 -070073 if (headers) {
74 mUriHeaders = *headers;
Chong Zhang3de157d2014-08-05 20:54:44 -070075 }
76
Chong Zhanga19f33e2014-08-07 15:35:07 -070077 // delay data source creation to prepareAsync() to avoid blocking
78 // the calling thread in setDataSource for any significant time.
79 return OK;
Andreas Huberafed0e12011-09-20 15:39:58 -070080}
81
Chong Zhanga19f33e2014-08-07 15:35:07 -070082status_t NuPlayer::GenericSource::setDataSource(
Chong Zhang3de157d2014-08-05 20:54:44 -070083 int fd, int64_t offset, int64_t length) {
Chong Zhanga19f33e2014-08-07 15:35:07 -070084 resetDataSource();
Andreas Huberafed0e12011-09-20 15:39:58 -070085
Chong Zhanga19f33e2014-08-07 15:35:07 -070086 mFd = dup(fd);
87 mOffset = offset;
88 mLength = length;
89
90 // delay data source creation to prepareAsync() to avoid blocking
91 // the calling thread in setDataSource for any significant time.
92 return OK;
Andreas Huberafed0e12011-09-20 15:39:58 -070093}
94
Chong Zhang3de157d2014-08-05 20:54:44 -070095status_t NuPlayer::GenericSource::initFromDataSource(
96 const sp<DataSource> &dataSource,
97 const char* mime) {
Lajos Molnarcc227032014-07-17 15:33:06 -070098 sp<MediaExtractor> extractor;
99
100 if (mIsWidevine) {
101 String8 mimeType;
102 float confidence;
103 sp<AMessage> dummy;
104 bool success;
105
106 success = SniffWVM(dataSource, &mimeType, &confidence, &dummy);
107 if (!success
108 || strcasecmp(
109 mimeType.string(), MEDIA_MIMETYPE_CONTAINER_WVM)) {
110 ALOGE("unsupported widevine mime: %s", mimeType.string());
Chong Zhang3de157d2014-08-05 20:54:44 -0700111 return UNKNOWN_ERROR;
Lajos Molnarcc227032014-07-17 15:33:06 -0700112 }
113
114 sp<WVMExtractor> wvmExtractor = new WVMExtractor(dataSource);
115 wvmExtractor->setAdaptiveStreamingMode(true);
116 if (mUIDValid) {
117 wvmExtractor->setUID(mUID);
118 }
119 extractor = wvmExtractor;
120 } else {
Chong Zhang3de157d2014-08-05 20:54:44 -0700121 extractor = MediaExtractor::Create(dataSource, mime);
Lajos Molnarcc227032014-07-17 15:33:06 -0700122 }
Andreas Huberafed0e12011-09-20 15:39:58 -0700123
Chong Zhang3de157d2014-08-05 20:54:44 -0700124 if (extractor == NULL) {
125 return UNKNOWN_ERROR;
126 }
Andreas Huberafed0e12011-09-20 15:39:58 -0700127
Marco Nelissenc1f4b2b2014-06-17 14:48:32 -0700128 sp<MetaData> fileMeta = extractor->getMetaData();
129 if (fileMeta != NULL) {
130 int64_t duration;
131 if (fileMeta->findInt64(kKeyDuration, &duration)) {
132 mDurationUs = duration;
133 }
134 }
135
Andreas Huberafed0e12011-09-20 15:39:58 -0700136 for (size_t i = 0; i < extractor->countTracks(); ++i) {
137 sp<MetaData> meta = extractor->getTrackMetaData(i);
138
139 const char *mime;
140 CHECK(meta->findCString(kKeyMIMEType, &mime));
141
Robert Shihdd235722014-06-12 14:49:23 -0700142 sp<MediaSource> track = extractor->getTrack(i);
Andreas Huberafed0e12011-09-20 15:39:58 -0700143
144 if (!strncasecmp(mime, "audio/", 6)) {
145 if (mAudioTrack.mSource == NULL) {
Robert Shihdd235722014-06-12 14:49:23 -0700146 mAudioTrack.mIndex = i;
147 mAudioTrack.mSource = track;
Andreas Huberafed0e12011-09-20 15:39:58 -0700148
149 if (!strcasecmp(mime, MEDIA_MIMETYPE_AUDIO_VORBIS)) {
150 mAudioIsVorbis = true;
151 } else {
152 mAudioIsVorbis = false;
153 }
154 }
155 } else if (!strncasecmp(mime, "video/", 6)) {
156 if (mVideoTrack.mSource == NULL) {
Robert Shihdd235722014-06-12 14:49:23 -0700157 mVideoTrack.mIndex = i;
158 mVideoTrack.mSource = track;
Chong Zhang7e892182014-08-05 11:58:21 -0700159
160 // check if the source requires secure buffers
161 int32_t secure;
Chong Zhanga19f33e2014-08-07 15:35:07 -0700162 if (meta->findInt32(kKeyRequiresSecureBuffers, &secure)
163 && secure) {
Chong Zhang7e892182014-08-05 11:58:21 -0700164 mIsWidevine = true;
Chong Zhang3de157d2014-08-05 20:54:44 -0700165 if (mUIDValid) {
166 extractor->setUID(mUID);
167 }
Chong Zhang7e892182014-08-05 11:58:21 -0700168 }
Andreas Huberafed0e12011-09-20 15:39:58 -0700169 }
170 }
171
172 if (track != NULL) {
Robert Shihdd235722014-06-12 14:49:23 -0700173 mSources.push(track);
Andreas Huberafed0e12011-09-20 15:39:58 -0700174 int64_t durationUs;
175 if (meta->findInt64(kKeyDuration, &durationUs)) {
176 if (durationUs > mDurationUs) {
177 mDurationUs = durationUs;
178 }
179 }
180 }
181 }
Chong Zhang3de157d2014-08-05 20:54:44 -0700182
183 return OK;
Andreas Huberafed0e12011-09-20 15:39:58 -0700184}
185
Chong Zhanga19f33e2014-08-07 15:35:07 -0700186status_t NuPlayer::GenericSource::setBuffers(
187 bool audio, Vector<MediaBuffer *> &buffers) {
Lajos Molnarcc227032014-07-17 15:33:06 -0700188 if (mIsWidevine && !audio) {
189 return mVideoTrack.mSource->setBuffers(buffers);
190 }
191 return INVALID_OPERATION;
192}
193
Andreas Huberafed0e12011-09-20 15:39:58 -0700194NuPlayer::GenericSource::~GenericSource() {
195}
196
Andreas Huber9575c962013-02-05 13:59:56 -0800197void NuPlayer::GenericSource::prepareAsync() {
Chong Zhanga19f33e2014-08-07 15:35:07 -0700198 // delayed data source creation
199 AString sniffedMIME;
200 sp<DataSource> dataSource;
201
202 if (!mUri.empty()) {
203 mIsWidevine = !strncasecmp(mUri.c_str(), "widevine://", 11);
204
205 dataSource = DataSource::CreateFromURI(
206 mHTTPService, mUri.c_str(), &mUriHeaders, &sniffedMIME);
207 } else {
208 // set to false first, if the extractor
209 // comes back as secure, set it to true then.
210 mIsWidevine = false;
211
212 dataSource = new FileSource(mFd, mOffset, mLength);
213 }
214
215 if (dataSource == NULL) {
216 ALOGE("Failed to create data source!");
217 notifyPrepared(UNKNOWN_ERROR);
218 return;
219 }
220
221 status_t err = initFromDataSource(
222 dataSource, sniffedMIME.empty() ? NULL : sniffedMIME.c_str());
223
224 if (err != OK) {
225 ALOGE("Failed to init from data source!");
226 notifyPrepared(err);
227 return;
228 }
229
Andreas Huber9575c962013-02-05 13:59:56 -0800230 if (mVideoTrack.mSource != NULL) {
Chong Zhangced1c2f2014-08-08 15:22:35 -0700231 notifyVideoSizeChanged(getFormat(false /* audio */));
Andreas Huber9575c962013-02-05 13:59:56 -0800232 }
233
234 notifyFlagsChanged(
Lajos Molnarcc227032014-07-17 15:33:06 -0700235 (mIsWidevine ? FLAG_SECURE : 0)
236 | FLAG_CAN_PAUSE
Andreas Huber9575c962013-02-05 13:59:56 -0800237 | FLAG_CAN_SEEK_BACKWARD
238 | FLAG_CAN_SEEK_FORWARD
239 | FLAG_CAN_SEEK);
240
241 notifyPrepared();
242}
243
Andreas Huberafed0e12011-09-20 15:39:58 -0700244void NuPlayer::GenericSource::start() {
245 ALOGI("start");
246
247 if (mAudioTrack.mSource != NULL) {
Robert Shihbace25b2014-07-25 14:14:34 -0700248 CHECK_EQ(mAudioTrack.mSource->start(), (status_t)OK);
Andreas Huberafed0e12011-09-20 15:39:58 -0700249 mAudioTrack.mPackets =
250 new AnotherPacketSource(mAudioTrack.mSource->getFormat());
251
Robert Shih3423bbd2014-07-16 15:47:09 -0700252 readBuffer(MEDIA_TRACK_TYPE_AUDIO);
Andreas Huberafed0e12011-09-20 15:39:58 -0700253 }
254
255 if (mVideoTrack.mSource != NULL) {
Robert Shihbace25b2014-07-25 14:14:34 -0700256 CHECK_EQ(mVideoTrack.mSource->start(), (status_t)OK);
Andreas Huberafed0e12011-09-20 15:39:58 -0700257 mVideoTrack.mPackets =
258 new AnotherPacketSource(mVideoTrack.mSource->getFormat());
259
Robert Shih3423bbd2014-07-16 15:47:09 -0700260 readBuffer(MEDIA_TRACK_TYPE_VIDEO);
Andreas Huberafed0e12011-09-20 15:39:58 -0700261 }
262}
263
264status_t NuPlayer::GenericSource::feedMoreTSData() {
265 return OK;
266}
267
Robert Shih3423bbd2014-07-16 15:47:09 -0700268void NuPlayer::GenericSource::onMessageReceived(const sp<AMessage> &msg) {
269 switch (msg->what()) {
270 case kWhatFetchSubtitleData:
271 {
Lajos Molnare26940f2014-07-31 10:31:26 -0700272 fetchTextData(kWhatSendSubtitleData, MEDIA_TRACK_TYPE_SUBTITLE,
273 mFetchSubtitleDataGeneration, mSubtitleTrack.mPackets, msg);
274 break;
275 }
Robert Shih3423bbd2014-07-16 15:47:09 -0700276
Lajos Molnare26940f2014-07-31 10:31:26 -0700277 case kWhatFetchTimedTextData:
278 {
279 fetchTextData(kWhatSendTimedTextData, MEDIA_TRACK_TYPE_TIMEDTEXT,
280 mFetchTimedTextDataGeneration, mTimedTextTrack.mPackets, msg);
Robert Shih3423bbd2014-07-16 15:47:09 -0700281 break;
282 }
283
284 case kWhatSendSubtitleData:
285 {
Lajos Molnare26940f2014-07-31 10:31:26 -0700286 sendTextData(kWhatSubtitleData, MEDIA_TRACK_TYPE_SUBTITLE,
287 mFetchSubtitleDataGeneration, mSubtitleTrack.mPackets, msg);
288 break;
289 }
Robert Shih3423bbd2014-07-16 15:47:09 -0700290
Lajos Molnare26940f2014-07-31 10:31:26 -0700291 case kWhatSendTimedTextData:
292 {
293 sendTextData(kWhatTimedTextData, MEDIA_TRACK_TYPE_TIMEDTEXT,
294 mFetchTimedTextDataGeneration, mTimedTextTrack.mPackets, msg);
Robert Shih3423bbd2014-07-16 15:47:09 -0700295 break;
296 }
297
298 case kWhatChangeAVSource:
299 {
300 int32_t trackIndex;
301 CHECK(msg->findInt32("trackIndex", &trackIndex));
302 const sp<MediaSource> source = mSources.itemAt(trackIndex);
303
304 Track* track;
305 const char *mime;
306 media_track_type trackType, counterpartType;
307 sp<MetaData> meta = source->getFormat();
308 meta->findCString(kKeyMIMEType, &mime);
309 if (!strncasecmp(mime, "audio/", 6)) {
310 track = &mAudioTrack;
311 trackType = MEDIA_TRACK_TYPE_AUDIO;
312 counterpartType = MEDIA_TRACK_TYPE_VIDEO;;
313 } else {
314 CHECK(!strncasecmp(mime, "video/", 6));
315 track = &mVideoTrack;
316 trackType = MEDIA_TRACK_TYPE_VIDEO;
317 counterpartType = MEDIA_TRACK_TYPE_AUDIO;;
318 }
319
320
321 if (track->mSource != NULL) {
322 track->mSource->stop();
323 }
324 track->mSource = source;
325 track->mSource->start();
326 track->mIndex = trackIndex;
327
328 status_t avail;
329 if (!track->mPackets->hasBufferAvailable(&avail)) {
330 // sync from other source
331 TRESPASS();
332 break;
333 }
334
335 int64_t timeUs, actualTimeUs;
336 const bool formatChange = true;
Robert Shih309aa8b2014-07-29 18:34:36 -0700337 sp<AMessage> latestMeta = track->mPackets->getLatestEnqueuedMeta();
Robert Shih3423bbd2014-07-16 15:47:09 -0700338 CHECK(latestMeta != NULL && latestMeta->findInt64("timeUs", &timeUs));
339 readBuffer(trackType, timeUs, &actualTimeUs, formatChange);
340 readBuffer(counterpartType, -1, NULL, formatChange);
341 ALOGV("timeUs %lld actualTimeUs %lld", timeUs, actualTimeUs);
342
343 break;
344 }
345
346 default:
347 Source::onMessageReceived(msg);
348 break;
349 }
350}
351
Lajos Molnare26940f2014-07-31 10:31:26 -0700352void NuPlayer::GenericSource::fetchTextData(
353 uint32_t sendWhat,
354 media_track_type type,
355 int32_t curGen,
356 sp<AnotherPacketSource> packets,
357 sp<AMessage> msg) {
358 int32_t msgGeneration;
359 CHECK(msg->findInt32("generation", &msgGeneration));
360 if (msgGeneration != curGen) {
361 // stale
362 return;
363 }
364
365 int32_t avail;
366 if (packets->hasBufferAvailable(&avail)) {
367 return;
368 }
369
370 int64_t timeUs;
371 CHECK(msg->findInt64("timeUs", &timeUs));
372
373 int64_t subTimeUs;
374 readBuffer(type, timeUs, &subTimeUs);
375
376 int64_t delayUs = subTimeUs - timeUs;
377 if (msg->what() == kWhatFetchSubtitleData) {
378 const int64_t oneSecUs = 1000000ll;
379 delayUs -= oneSecUs;
380 }
381 sp<AMessage> msg2 = new AMessage(sendWhat, id());
382 msg2->setInt32("generation", msgGeneration);
383 msg2->post(delayUs < 0 ? 0 : delayUs);
384}
385
386void NuPlayer::GenericSource::sendTextData(
387 uint32_t what,
388 media_track_type type,
389 int32_t curGen,
390 sp<AnotherPacketSource> packets,
391 sp<AMessage> msg) {
392 int32_t msgGeneration;
393 CHECK(msg->findInt32("generation", &msgGeneration));
394 if (msgGeneration != curGen) {
395 // stale
396 return;
397 }
398
399 int64_t subTimeUs;
400 if (packets->nextBufferTime(&subTimeUs) != OK) {
401 return;
402 }
403
404 int64_t nextSubTimeUs;
405 readBuffer(type, -1, &nextSubTimeUs);
406
407 sp<ABuffer> buffer;
408 status_t dequeueStatus = packets->dequeueAccessUnit(&buffer);
409 if (dequeueStatus == OK) {
410 sp<AMessage> notify = dupNotify();
411 notify->setInt32("what", what);
412 notify->setBuffer("buffer", buffer);
413 notify->post();
414
415 const int64_t delayUs = nextSubTimeUs - subTimeUs;
416 msg->post(delayUs < 0 ? 0 : delayUs);
417 }
418}
419
Andreas Huber84066782011-08-16 09:34:26 -0700420sp<MetaData> NuPlayer::GenericSource::getFormatMeta(bool audio) {
Andreas Huberafed0e12011-09-20 15:39:58 -0700421 sp<MediaSource> source = audio ? mAudioTrack.mSource : mVideoTrack.mSource;
422
423 if (source == NULL) {
424 return NULL;
425 }
426
427 return source->getFormat();
428}
429
430status_t NuPlayer::GenericSource::dequeueAccessUnit(
431 bool audio, sp<ABuffer> *accessUnit) {
432 Track *track = audio ? &mAudioTrack : &mVideoTrack;
433
434 if (track->mSource == NULL) {
435 return -EWOULDBLOCK;
436 }
437
Lajos Molnarcc227032014-07-17 15:33:06 -0700438 if (mIsWidevine && !audio) {
439 // try to read a buffer as we may not have been able to the last time
Robert Shih3423bbd2014-07-16 15:47:09 -0700440 readBuffer(MEDIA_TRACK_TYPE_VIDEO, -1ll);
Lajos Molnarcc227032014-07-17 15:33:06 -0700441 }
442
Andreas Huberafed0e12011-09-20 15:39:58 -0700443 status_t finalResult;
444 if (!track->mPackets->hasBufferAvailable(&finalResult)) {
Lajos Molnarcc227032014-07-17 15:33:06 -0700445 return (finalResult == OK ? -EWOULDBLOCK : finalResult);
Andreas Huberafed0e12011-09-20 15:39:58 -0700446 }
447
448 status_t result = track->mPackets->dequeueAccessUnit(accessUnit);
449
Robert Shih3423bbd2014-07-16 15:47:09 -0700450 if (!track->mPackets->hasBufferAvailable(&finalResult)) {
451 readBuffer(audio? MEDIA_TRACK_TYPE_AUDIO : MEDIA_TRACK_TYPE_VIDEO, -1ll);
452 }
453
Lajos Molnare26940f2014-07-31 10:31:26 -0700454 if (mSubtitleTrack.mSource == NULL && mTimedTextTrack.mSource == NULL) {
Robert Shih3423bbd2014-07-16 15:47:09 -0700455 return result;
456 }
457
Lajos Molnare26940f2014-07-31 10:31:26 -0700458 if (mSubtitleTrack.mSource != NULL) {
459 CHECK(mSubtitleTrack.mPackets != NULL);
460 }
461 if (mTimedTextTrack.mSource != NULL) {
462 CHECK(mTimedTextTrack.mPackets != NULL);
463 }
464
Robert Shih3423bbd2014-07-16 15:47:09 -0700465 if (result != OK) {
Lajos Molnare26940f2014-07-31 10:31:26 -0700466 if (mSubtitleTrack.mSource != NULL) {
467 mSubtitleTrack.mPackets->clear();
468 mFetchSubtitleDataGeneration++;
469 }
470 if (mTimedTextTrack.mSource != NULL) {
471 mTimedTextTrack.mPackets->clear();
472 mFetchTimedTextDataGeneration++;
473 }
Robert Shih3423bbd2014-07-16 15:47:09 -0700474 return result;
475 }
476
477 int64_t timeUs;
478 status_t eosResult; // ignored
479 CHECK((*accessUnit)->meta()->findInt64("timeUs", &timeUs));
Lajos Molnare26940f2014-07-31 10:31:26 -0700480
481 if (mSubtitleTrack.mSource != NULL
482 && !mSubtitleTrack.mPackets->hasBufferAvailable(&eosResult)) {
Robert Shih3423bbd2014-07-16 15:47:09 -0700483 sp<AMessage> msg = new AMessage(kWhatFetchSubtitleData, id());
484 msg->setInt64("timeUs", timeUs);
485 msg->setInt32("generation", mFetchSubtitleDataGeneration);
486 msg->post();
487 }
Robert Shiheb1735e2014-07-23 15:53:14 -0700488
Lajos Molnare26940f2014-07-31 10:31:26 -0700489 if (mTimedTextTrack.mSource != NULL
490 && !mTimedTextTrack.mPackets->hasBufferAvailable(&eosResult)) {
491 sp<AMessage> msg = new AMessage(kWhatFetchTimedTextData, id());
492 msg->setInt64("timeUs", timeUs);
493 msg->setInt32("generation", mFetchTimedTextDataGeneration);
494 msg->post();
495 }
496
Andreas Huberafed0e12011-09-20 15:39:58 -0700497 return result;
498}
499
500status_t NuPlayer::GenericSource::getDuration(int64_t *durationUs) {
501 *durationUs = mDurationUs;
502 return OK;
503}
504
Robert Shihdd235722014-06-12 14:49:23 -0700505size_t NuPlayer::GenericSource::getTrackCount() const {
506 return mSources.size();
507}
508
509sp<AMessage> NuPlayer::GenericSource::getTrackInfo(size_t trackIndex) const {
510 size_t trackCount = mSources.size();
511 if (trackIndex >= trackCount) {
512 return NULL;
513 }
514
515 sp<AMessage> format = new AMessage();
516 sp<MetaData> meta = mSources.itemAt(trackIndex)->getFormat();
517
518 const char *mime;
519 CHECK(meta->findCString(kKeyMIMEType, &mime));
520
521 int32_t trackType;
522 if (!strncasecmp(mime, "video/", 6)) {
523 trackType = MEDIA_TRACK_TYPE_VIDEO;
524 } else if (!strncasecmp(mime, "audio/", 6)) {
525 trackType = MEDIA_TRACK_TYPE_AUDIO;
526 } else if (!strcasecmp(mime, MEDIA_MIMETYPE_TEXT_3GPP)) {
527 trackType = MEDIA_TRACK_TYPE_TIMEDTEXT;
528 } else {
529 trackType = MEDIA_TRACK_TYPE_UNKNOWN;
530 }
531 format->setInt32("type", trackType);
532
533 const char *lang;
534 if (!meta->findCString(kKeyMediaLanguage, &lang)) {
535 lang = "und";
536 }
537 format->setString("language", lang);
538
539 if (trackType == MEDIA_TRACK_TYPE_SUBTITLE) {
540 format->setString("mime", mime);
541
542 int32_t isAutoselect = 1, isDefault = 0, isForced = 0;
543 meta->findInt32(kKeyTrackIsAutoselect, &isAutoselect);
544 meta->findInt32(kKeyTrackIsDefault, &isDefault);
545 meta->findInt32(kKeyTrackIsForced, &isForced);
546
547 format->setInt32("auto", !!isAutoselect);
548 format->setInt32("default", !!isDefault);
549 format->setInt32("forced", !!isForced);
550 }
551
552 return format;
553}
554
Lajos Molnare26940f2014-07-31 10:31:26 -0700555ssize_t NuPlayer::GenericSource::getSelectedTrack(media_track_type type) const {
556 const Track *track = NULL;
557 switch (type) {
558 case MEDIA_TRACK_TYPE_VIDEO:
559 track = &mVideoTrack;
560 break;
561 case MEDIA_TRACK_TYPE_AUDIO:
562 track = &mAudioTrack;
563 break;
564 case MEDIA_TRACK_TYPE_TIMEDTEXT:
565 track = &mTimedTextTrack;
566 break;
567 case MEDIA_TRACK_TYPE_SUBTITLE:
568 track = &mSubtitleTrack;
569 break;
570 default:
571 break;
572 }
573
574 if (track != NULL && track->mSource != NULL) {
575 return track->mIndex;
576 }
577
578 return -1;
579}
580
Robert Shih3423bbd2014-07-16 15:47:09 -0700581status_t NuPlayer::GenericSource::selectTrack(size_t trackIndex, bool select) {
Lajos Molnare26940f2014-07-31 10:31:26 -0700582 ALOGV("%s track: %zu", select ? "select" : "deselect", trackIndex);
Robert Shih3423bbd2014-07-16 15:47:09 -0700583 if (trackIndex >= mSources.size()) {
584 return BAD_INDEX;
585 }
586
587 if (!select) {
Lajos Molnare26940f2014-07-31 10:31:26 -0700588 Track* track = NULL;
589 if (mSubtitleTrack.mSource != NULL && trackIndex == mSubtitleTrack.mIndex) {
590 track = &mSubtitleTrack;
591 mFetchSubtitleDataGeneration++;
592 } else if (mTimedTextTrack.mSource != NULL && trackIndex == mTimedTextTrack.mIndex) {
593 track = &mTimedTextTrack;
594 mFetchTimedTextDataGeneration++;
595 }
596 if (track == NULL) {
Robert Shih3423bbd2014-07-16 15:47:09 -0700597 return INVALID_OPERATION;
598 }
Lajos Molnare26940f2014-07-31 10:31:26 -0700599 track->mSource->stop();
600 track->mSource = NULL;
601 track->mPackets->clear();
Robert Shih3423bbd2014-07-16 15:47:09 -0700602 return OK;
603 }
604
605 const sp<MediaSource> source = mSources.itemAt(trackIndex);
606 sp<MetaData> meta = source->getFormat();
607 const char *mime;
608 CHECK(meta->findCString(kKeyMIMEType, &mime));
609 if (!strncasecmp(mime, "text/", 5)) {
Lajos Molnare26940f2014-07-31 10:31:26 -0700610 bool isSubtitle = strcasecmp(mime, MEDIA_MIMETYPE_TEXT_3GPP);
611 Track *track = isSubtitle ? &mSubtitleTrack : &mTimedTextTrack;
612 if (track->mSource != NULL && track->mIndex == trackIndex) {
Robert Shih3423bbd2014-07-16 15:47:09 -0700613 return OK;
614 }
Lajos Molnare26940f2014-07-31 10:31:26 -0700615 track->mIndex = trackIndex;
616 if (track->mSource != NULL) {
617 track->mSource->stop();
Robert Shih3423bbd2014-07-16 15:47:09 -0700618 }
Lajos Molnare26940f2014-07-31 10:31:26 -0700619 track->mSource = mSources.itemAt(trackIndex);
620 track->mSource->start();
621 if (track->mPackets == NULL) {
622 track->mPackets = new AnotherPacketSource(track->mSource->getFormat());
Robert Shih3423bbd2014-07-16 15:47:09 -0700623 } else {
Lajos Molnare26940f2014-07-31 10:31:26 -0700624 track->mPackets->clear();
625 track->mPackets->setFormat(track->mSource->getFormat());
Robert Shih3423bbd2014-07-16 15:47:09 -0700626
627 }
Lajos Molnare26940f2014-07-31 10:31:26 -0700628
629 if (isSubtitle) {
630 mFetchSubtitleDataGeneration++;
631 } else {
632 mFetchTimedTextDataGeneration++;
633 }
634
Robert Shih3423bbd2014-07-16 15:47:09 -0700635 return OK;
636 } else if (!strncasecmp(mime, "audio/", 6) || !strncasecmp(mime, "video/", 6)) {
637 bool audio = !strncasecmp(mime, "audio/", 6);
638 Track *track = audio ? &mAudioTrack : &mVideoTrack;
639 if (track->mSource != NULL && track->mIndex == trackIndex) {
640 return OK;
641 }
642
643 sp<AMessage> msg = new AMessage(kWhatChangeAVSource, id());
644 msg->setInt32("trackIndex", trackIndex);
645 msg->post();
646 return OK;
647 }
648
649 return INVALID_OPERATION;
650}
651
Andreas Huberafed0e12011-09-20 15:39:58 -0700652status_t NuPlayer::GenericSource::seekTo(int64_t seekTimeUs) {
653 if (mVideoTrack.mSource != NULL) {
654 int64_t actualTimeUs;
Robert Shih3423bbd2014-07-16 15:47:09 -0700655 readBuffer(MEDIA_TRACK_TYPE_VIDEO, seekTimeUs, &actualTimeUs);
Andreas Huberafed0e12011-09-20 15:39:58 -0700656
657 seekTimeUs = actualTimeUs;
658 }
659
660 if (mAudioTrack.mSource != NULL) {
Robert Shih3423bbd2014-07-16 15:47:09 -0700661 readBuffer(MEDIA_TRACK_TYPE_AUDIO, seekTimeUs);
Andreas Huberafed0e12011-09-20 15:39:58 -0700662 }
663
664 return OK;
665}
666
Robert Shih3423bbd2014-07-16 15:47:09 -0700667sp<ABuffer> NuPlayer::GenericSource::mediaBufferToABuffer(
668 MediaBuffer* mb,
669 media_track_type trackType,
670 int64_t *actualTimeUs) {
671 bool audio = trackType == MEDIA_TRACK_TYPE_AUDIO;
672 size_t outLength = mb->range_length();
673
674 if (audio && mAudioIsVorbis) {
675 outLength += sizeof(int32_t);
676 }
677
678 sp<ABuffer> ab;
679 if (mIsWidevine && !audio) {
680 // data is already provided in the buffer
681 ab = new ABuffer(NULL, mb->range_length());
682 ab->meta()->setPointer("mediaBuffer", mb);
683 mb->add_ref();
684 } else {
685 ab = new ABuffer(outLength);
686 memcpy(ab->data(),
687 (const uint8_t *)mb->data() + mb->range_offset(),
688 mb->range_length());
689 }
690
691 if (audio && mAudioIsVorbis) {
692 int32_t numPageSamples;
693 if (!mb->meta_data()->findInt32(kKeyValidSamples, &numPageSamples)) {
694 numPageSamples = -1;
695 }
696
697 uint8_t* abEnd = ab->data() + mb->range_length();
698 memcpy(abEnd, &numPageSamples, sizeof(numPageSamples));
699 }
700
Lajos Molnare26940f2014-07-31 10:31:26 -0700701 sp<AMessage> meta = ab->meta();
702
Robert Shih3423bbd2014-07-16 15:47:09 -0700703 int64_t timeUs;
704 CHECK(mb->meta_data()->findInt64(kKeyTime, &timeUs));
Robert Shih3423bbd2014-07-16 15:47:09 -0700705 meta->setInt64("timeUs", timeUs);
706
Lajos Molnare26940f2014-07-31 10:31:26 -0700707 if (trackType == MEDIA_TRACK_TYPE_TIMEDTEXT) {
708 const char *mime;
709 CHECK(mTimedTextTrack.mSource != NULL
710 && mTimedTextTrack.mSource->getFormat()->findCString(kKeyMIMEType, &mime));
711 meta->setString("mime", mime);
712 }
713
Robert Shih3423bbd2014-07-16 15:47:09 -0700714 int64_t durationUs;
715 if (mb->meta_data()->findInt64(kKeyDuration, &durationUs)) {
716 meta->setInt64("durationUs", durationUs);
717 }
718
719 if (trackType == MEDIA_TRACK_TYPE_SUBTITLE) {
720 meta->setInt32("trackIndex", mSubtitleTrack.mIndex);
721 }
722
723 if (actualTimeUs) {
724 *actualTimeUs = timeUs;
725 }
726
727 mb->release();
728 mb = NULL;
729
730 return ab;
731}
732
Andreas Huberafed0e12011-09-20 15:39:58 -0700733void NuPlayer::GenericSource::readBuffer(
Robert Shih3423bbd2014-07-16 15:47:09 -0700734 media_track_type trackType, int64_t seekTimeUs, int64_t *actualTimeUs, bool formatChange) {
735 Track *track;
736 switch (trackType) {
737 case MEDIA_TRACK_TYPE_VIDEO:
738 track = &mVideoTrack;
739 break;
740 case MEDIA_TRACK_TYPE_AUDIO:
741 track = &mAudioTrack;
742 break;
743 case MEDIA_TRACK_TYPE_SUBTITLE:
744 track = &mSubtitleTrack;
745 break;
Lajos Molnare26940f2014-07-31 10:31:26 -0700746 case MEDIA_TRACK_TYPE_TIMEDTEXT:
747 track = &mTimedTextTrack;
748 break;
Robert Shih3423bbd2014-07-16 15:47:09 -0700749 default:
750 TRESPASS();
751 }
752
753 if (track->mSource == NULL) {
754 return;
755 }
Andreas Huberafed0e12011-09-20 15:39:58 -0700756
757 if (actualTimeUs) {
758 *actualTimeUs = seekTimeUs;
759 }
760
761 MediaSource::ReadOptions options;
762
763 bool seeking = false;
764
765 if (seekTimeUs >= 0) {
Robert Shih3423bbd2014-07-16 15:47:09 -0700766 options.setSeekTo(seekTimeUs, MediaSource::ReadOptions::SEEK_PREVIOUS_SYNC);
Andreas Huberafed0e12011-09-20 15:39:58 -0700767 seeking = true;
768 }
769
Robert Shih3423bbd2014-07-16 15:47:09 -0700770 if (mIsWidevine && trackType != MEDIA_TRACK_TYPE_AUDIO) {
Lajos Molnarcc227032014-07-17 15:33:06 -0700771 options.setNonBlocking();
772 }
773
Andreas Huberafed0e12011-09-20 15:39:58 -0700774 for (;;) {
775 MediaBuffer *mbuf;
776 status_t err = track->mSource->read(&mbuf, &options);
777
778 options.clearSeekTo();
779
780 if (err == OK) {
Robert Shih3423bbd2014-07-16 15:47:09 -0700781 // formatChange && seeking: track whose source is changed during selection
782 // formatChange && !seeking: track whose source is not changed during selection
783 // !formatChange: normal seek
Lajos Molnare26940f2014-07-31 10:31:26 -0700784 if ((seeking || formatChange)
785 && (trackType == MEDIA_TRACK_TYPE_AUDIO
786 || trackType == MEDIA_TRACK_TYPE_VIDEO)) {
Robert Shih3423bbd2014-07-16 15:47:09 -0700787 ATSParser::DiscontinuityType type = formatChange
788 ? (seeking
789 ? ATSParser::DISCONTINUITY_FORMATCHANGE
790 : ATSParser::DISCONTINUITY_NONE)
791 : ATSParser::DISCONTINUITY_SEEK;
792 track->mPackets->queueDiscontinuity( type, NULL, true /* discard */);
Andreas Huberafed0e12011-09-20 15:39:58 -0700793 }
794
Robert Shih3423bbd2014-07-16 15:47:09 -0700795 sp<ABuffer> buffer = mediaBufferToABuffer(mbuf, trackType, actualTimeUs);
Andreas Huberafed0e12011-09-20 15:39:58 -0700796 track->mPackets->queueAccessUnit(buffer);
797 break;
Lajos Molnarcc227032014-07-17 15:33:06 -0700798 } else if (err == WOULD_BLOCK) {
799 break;
Andreas Huberafed0e12011-09-20 15:39:58 -0700800 } else if (err == INFO_FORMAT_CHANGED) {
801#if 0
802 track->mPackets->queueDiscontinuity(
Chong Zhang632740c2014-06-26 13:03:47 -0700803 ATSParser::DISCONTINUITY_FORMATCHANGE,
804 NULL,
805 false /* discard */);
Andreas Huberafed0e12011-09-20 15:39:58 -0700806#endif
807 } else {
808 track->mPackets->signalEOS(err);
809 break;
810 }
811 }
812}
813
Andreas Huberafed0e12011-09-20 15:39:58 -0700814} // namespace android