Andreas Huber | e46b7be | 2009-07-14 16:56:47 -0700 | [diff] [blame] | 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 | //#define LOG_NDEBUG 0 |
| 18 | #define LOG_TAG "MediaExtractor" |
| 19 | #include <utils/Log.h> |
| 20 | |
Andreas Huber | 57515f3 | 2009-10-23 09:55:10 -0700 | [diff] [blame] | 21 | #include "include/AMRExtractor.h" |
Andreas Huber | cabb7da | 2011-03-24 14:18:02 -0700 | [diff] [blame] | 22 | #include "include/AVIExtractor.h" |
Andreas Huber | 57515f3 | 2009-10-23 09:55:10 -0700 | [diff] [blame] | 23 | #include "include/MP3Extractor.h" |
| 24 | #include "include/MPEG4Extractor.h" |
Andreas Huber | 6bce6d8 | 2009-11-03 16:00:58 -0800 | [diff] [blame] | 25 | #include "include/WAVExtractor.h" |
Andreas Huber | 388379f | 2010-05-07 10:35:13 -0700 | [diff] [blame] | 26 | #include "include/OggExtractor.h" |
Andreas Huber | bc55495 | 2011-09-08 14:12:44 -0700 | [diff] [blame] | 27 | #include "include/MPEG2PSExtractor.h" |
Andreas Huber | a557b24 | 2010-06-07 13:05:37 -0700 | [diff] [blame] | 28 | #include "include/MPEG2TSExtractor.h" |
Gloria Wang | d577091 | 2010-06-22 13:55:38 -0700 | [diff] [blame] | 29 | #include "include/DRMExtractor.h" |
Gloria Wang | c2c22e7 | 2010-11-01 15:53:16 -0700 | [diff] [blame] | 30 | #include "include/WVMExtractor.h" |
Glenn Kasten | f9f223e | 2011-01-13 11:17:00 -0800 | [diff] [blame] | 31 | #include "include/FLACExtractor.h" |
Gloria Wang | c5b0abf | 2011-02-02 14:12:49 -0800 | [diff] [blame] | 32 | #include "include/AACExtractor.h" |
Andreas Huber | 57515f3 | 2009-10-23 09:55:10 -0700 | [diff] [blame] | 33 | |
Andreas Huber | 072f524 | 2010-05-20 14:56:53 -0700 | [diff] [blame] | 34 | #include "matroska/MatroskaExtractor.h" |
| 35 | |
Andreas Huber | efdd088 | 2010-08-25 11:09:41 -0700 | [diff] [blame] | 36 | #include <media/stagefright/foundation/AMessage.h> |
Andreas Huber | e46b7be | 2009-07-14 16:56:47 -0700 | [diff] [blame] | 37 | #include <media/stagefright/DataSource.h> |
Andreas Huber | e6c4096 | 2009-09-10 14:13:30 -0700 | [diff] [blame] | 38 | #include <media/stagefright/MediaDefs.h> |
Andreas Huber | e46b7be | 2009-07-14 16:56:47 -0700 | [diff] [blame] | 39 | #include <media/stagefright/MediaExtractor.h> |
Andreas Huber | aee3c63 | 2010-01-11 15:35:19 -0800 | [diff] [blame] | 40 | #include <media/stagefright/MetaData.h> |
Andreas Huber | e46b7be | 2009-07-14 16:56:47 -0700 | [diff] [blame] | 41 | #include <utils/String8.h> |
| 42 | |
| 43 | namespace android { |
| 44 | |
Andreas Huber | aee3c63 | 2010-01-11 15:35:19 -0800 | [diff] [blame] | 45 | sp<MetaData> MediaExtractor::getMetaData() { |
| 46 | return new MetaData; |
| 47 | } |
| 48 | |
Andreas Huber | 62f7ffe | 2010-05-06 10:18:05 -0700 | [diff] [blame] | 49 | uint32_t MediaExtractor::flags() const { |
Andreas Huber | 10b9b3f | 2010-10-08 10:16:24 -0700 | [diff] [blame] | 50 | return CAN_SEEK_BACKWARD | CAN_SEEK_FORWARD | CAN_PAUSE | CAN_SEEK; |
Andreas Huber | 62f7ffe | 2010-05-06 10:18:05 -0700 | [diff] [blame] | 51 | } |
| 52 | |
Andreas Huber | e46b7be | 2009-07-14 16:56:47 -0700 | [diff] [blame] | 53 | // static |
Andreas Huber | be06d26 | 2009-08-14 14:37:10 -0700 | [diff] [blame] | 54 | sp<MediaExtractor> MediaExtractor::Create( |
| 55 | const sp<DataSource> &source, const char *mime) { |
Andreas Huber | efdd088 | 2010-08-25 11:09:41 -0700 | [diff] [blame] | 56 | sp<AMessage> meta; |
| 57 | |
Andreas Huber | e46b7be | 2009-07-14 16:56:47 -0700 | [diff] [blame] | 58 | String8 tmp; |
| 59 | if (mime == NULL) { |
| 60 | float confidence; |
Andreas Huber | efdd088 | 2010-08-25 11:09:41 -0700 | [diff] [blame] | 61 | if (!source->sniff(&tmp, &confidence, &meta)) { |
Steve Block | 71f2cf1 | 2011-10-20 11:56:00 +0100 | [diff] [blame] | 62 | ALOGV("FAILED to autodetect media content."); |
Andreas Huber | e46b7be | 2009-07-14 16:56:47 -0700 | [diff] [blame] | 63 | |
| 64 | return NULL; |
| 65 | } |
| 66 | |
| 67 | mime = tmp.string(); |
Steve Block | 71f2cf1 | 2011-10-20 11:56:00 +0100 | [diff] [blame] | 68 | ALOGV("Autodetected media content as '%s' with confidence %.2f", |
Andreas Huber | e46b7be | 2009-07-14 16:56:47 -0700 | [diff] [blame] | 69 | mime, confidence); |
| 70 | } |
| 71 | |
Glenn Kasten | 5f63069 | 2011-02-21 09:37:13 -0800 | [diff] [blame] | 72 | bool isDrm = false; |
Glenn Kasten | df0b651 | 2011-02-18 15:42:25 -0800 | [diff] [blame] | 73 | // DRM MIME type syntax is "drm+type+original" where |
| 74 | // type is "es_based" or "container_based" and |
| 75 | // original is the content's cleartext MIME type |
| 76 | if (!strncmp(mime, "drm+", 4)) { |
| 77 | const char *originalMime = strchr(mime+4, '+'); |
| 78 | if (originalMime == NULL) { |
| 79 | // second + not found |
| 80 | return NULL; |
| 81 | } |
| 82 | ++originalMime; |
| 83 | if (!strncmp(mime, "drm+es_based+", 13)) { |
Glenn Kasten | 5f63069 | 2011-02-21 09:37:13 -0800 | [diff] [blame] | 84 | // DRMExtractor sets container metadata kKeyIsDRM to 1 |
Gloria Wang | d577091 | 2010-06-22 13:55:38 -0700 | [diff] [blame] | 85 | return new DRMExtractor(source, originalMime); |
Glenn Kasten | df0b651 | 2011-02-18 15:42:25 -0800 | [diff] [blame] | 86 | } else if (!strncmp(mime, "drm+container_based+", 20)) { |
Gloria Wang | d577091 | 2010-06-22 13:55:38 -0700 | [diff] [blame] | 87 | mime = originalMime; |
Glenn Kasten | 5f63069 | 2011-02-21 09:37:13 -0800 | [diff] [blame] | 88 | isDrm = true; |
Gloria Wang | d577091 | 2010-06-22 13:55:38 -0700 | [diff] [blame] | 89 | } else { |
| 90 | return NULL; |
| 91 | } |
| 92 | } |
| 93 | |
Glenn Kasten | 5f63069 | 2011-02-21 09:37:13 -0800 | [diff] [blame] | 94 | MediaExtractor *ret = NULL; |
Andreas Huber | e6c4096 | 2009-09-10 14:13:30 -0700 | [diff] [blame] | 95 | if (!strcasecmp(mime, MEDIA_MIMETYPE_CONTAINER_MPEG4) |
| 96 | || !strcasecmp(mime, "audio/mp4")) { |
Glenn Kasten | 5f63069 | 2011-02-21 09:37:13 -0800 | [diff] [blame] | 97 | ret = new MPEG4Extractor(source); |
Andreas Huber | e6c4096 | 2009-09-10 14:13:30 -0700 | [diff] [blame] | 98 | } else if (!strcasecmp(mime, MEDIA_MIMETYPE_AUDIO_MPEG)) { |
Glenn Kasten | 5f63069 | 2011-02-21 09:37:13 -0800 | [diff] [blame] | 99 | ret = new MP3Extractor(source, meta); |
Andreas Huber | e6c4096 | 2009-09-10 14:13:30 -0700 | [diff] [blame] | 100 | } else if (!strcasecmp(mime, MEDIA_MIMETYPE_AUDIO_AMR_NB) |
| 101 | || !strcasecmp(mime, MEDIA_MIMETYPE_AUDIO_AMR_WB)) { |
Glenn Kasten | 5f63069 | 2011-02-21 09:37:13 -0800 | [diff] [blame] | 102 | ret = new AMRExtractor(source); |
Glenn Kasten | f9f223e | 2011-01-13 11:17:00 -0800 | [diff] [blame] | 103 | } else if (!strcasecmp(mime, MEDIA_MIMETYPE_AUDIO_FLAC)) { |
Glenn Kasten | 5f63069 | 2011-02-21 09:37:13 -0800 | [diff] [blame] | 104 | ret = new FLACExtractor(source); |
Andreas Huber | 6bce6d8 | 2009-11-03 16:00:58 -0800 | [diff] [blame] | 105 | } else if (!strcasecmp(mime, MEDIA_MIMETYPE_CONTAINER_WAV)) { |
Glenn Kasten | 5f63069 | 2011-02-21 09:37:13 -0800 | [diff] [blame] | 106 | ret = new WAVExtractor(source); |
Andreas Huber | 388379f | 2010-05-07 10:35:13 -0700 | [diff] [blame] | 107 | } else if (!strcasecmp(mime, MEDIA_MIMETYPE_CONTAINER_OGG)) { |
Glenn Kasten | 5f63069 | 2011-02-21 09:37:13 -0800 | [diff] [blame] | 108 | ret = new OggExtractor(source); |
Andreas Huber | 072f524 | 2010-05-20 14:56:53 -0700 | [diff] [blame] | 109 | } else if (!strcasecmp(mime, MEDIA_MIMETYPE_CONTAINER_MATROSKA)) { |
Glenn Kasten | 5f63069 | 2011-02-21 09:37:13 -0800 | [diff] [blame] | 110 | ret = new MatroskaExtractor(source); |
Andreas Huber | a557b24 | 2010-06-07 13:05:37 -0700 | [diff] [blame] | 111 | } else if (!strcasecmp(mime, MEDIA_MIMETYPE_CONTAINER_MPEG2TS)) { |
Glenn Kasten | 5f63069 | 2011-02-21 09:37:13 -0800 | [diff] [blame] | 112 | ret = new MPEG2TSExtractor(source); |
Andreas Huber | cabb7da | 2011-03-24 14:18:02 -0700 | [diff] [blame] | 113 | } else if (!strcasecmp(mime, MEDIA_MIMETYPE_CONTAINER_AVI)) { |
| 114 | ret = new AVIExtractor(source); |
Gloria Wang | c2c22e7 | 2010-11-01 15:53:16 -0700 | [diff] [blame] | 115 | } else if (!strcasecmp(mime, MEDIA_MIMETYPE_CONTAINER_WVM)) { |
Glenn Kasten | 5f63069 | 2011-02-21 09:37:13 -0800 | [diff] [blame] | 116 | ret = new WVMExtractor(source); |
James Dong | aadbd80 | 2011-12-08 19:31:59 -0800 | [diff] [blame] | 117 | if (ret != NULL) { |
| 118 | isDrm = true; |
| 119 | } |
Gloria Wang | c5b0abf | 2011-02-02 14:12:49 -0800 | [diff] [blame] | 120 | } else if (!strcasecmp(mime, MEDIA_MIMETYPE_AUDIO_AAC_ADTS)) { |
Andreas Huber | fe5f0aa | 2011-11-01 11:01:01 -0700 | [diff] [blame] | 121 | ret = new AACExtractor(source, meta); |
Andreas Huber | bc55495 | 2011-09-08 14:12:44 -0700 | [diff] [blame] | 122 | } else if (!strcasecmp(mime, MEDIA_MIMETYPE_CONTAINER_MPEG2PS)) { |
| 123 | ret = new MPEG2PSExtractor(source); |
Glenn Kasten | 5f63069 | 2011-02-21 09:37:13 -0800 | [diff] [blame] | 124 | } |
Gloria Wang | 82428a8 | 2011-06-27 11:09:00 -0700 | [diff] [blame] | 125 | |
| 126 | if (ret != NULL) { |
| 127 | if (isDrm) { |
| 128 | ret->setDrmFlag(true); |
| 129 | } else { |
| 130 | ret->setDrmFlag(false); |
| 131 | } |
Andreas Huber | e46b7be | 2009-07-14 16:56:47 -0700 | [diff] [blame] | 132 | } |
| 133 | |
Glenn Kasten | 5f63069 | 2011-02-21 09:37:13 -0800 | [diff] [blame] | 134 | return ret; |
Andreas Huber | e46b7be | 2009-07-14 16:56:47 -0700 | [diff] [blame] | 135 | } |
| 136 | |
Andreas Huber | e46b7be | 2009-07-14 16:56:47 -0700 | [diff] [blame] | 137 | } // namespace android |