record.cpp revision 6d43768514650d65f4f3695da4f7c3e2eefd483c
120111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber/* 220111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber * Copyright (C) 2009 The Android Open Source Project 320111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber * 420111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber * Licensed under the Apache License, Version 2.0 (the "License"); 520111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber * you may not use this file except in compliance with the License. 620111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber * You may obtain a copy of the License at 720111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber * 820111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber * http://www.apache.org/licenses/LICENSE-2.0 920111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber * 1020111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber * Unless required by applicable law or agreed to in writing, software 1120111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber * distributed under the License is distributed on an "AS IS" BASIS, 1220111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 1320111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber * See the License for the specific language governing permissions and 1420111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber * limitations under the License. 1520111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber */ 1620111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 1720111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber#include <binder/ProcessState.h> 1820111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber#include <media/stagefright/CameraSource.h> 1920111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber#include <media/stagefright/MediaBufferGroup.h> 206d43768514650d65f4f3695da4f7c3e2eefd483cAndreas Huber#include <media/stagefright/MediaDebug.h> 2120111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber#include <media/stagefright/MetaData.h> 2220111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber#include <media/stagefright/MPEG4Extractor.h> 2320111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber#include <media/stagefright/MPEG4Writer.h> 2420111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber#include <media/stagefright/MmapSource.h> 2520111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber#include <media/stagefright/OMXClient.h> 26693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber#include <media/stagefright/OMXCodec.h> 2720111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 2820111aa043c5f404472bc63b90bc5aad906b1101Andreas Huberusing namespace android; 2920111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 3020111aa043c5f404472bc63b90bc5aad906b1101Andreas Huberclass DummySource : public MediaSource { 3120111aa043c5f404472bc63b90bc5aad906b1101Andreas Huberpublic: 3220111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber DummySource(int width, int height) 33693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber : mWidth(width), 34693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber mHeight(height), 35693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber mSize((width * height * 3) / 2) { 3620111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber mGroup.add_buffer(new MediaBuffer(mSize)); 3720111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber } 3820111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 39693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber virtual sp<MetaData> getFormat() { 40693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber sp<MetaData> meta = new MetaData; 41693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber meta->setInt32(kKeyWidth, mWidth); 42693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber meta->setInt32(kKeyHeight, mHeight); 43693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber meta->setCString(kKeyMIMEType, "video/raw"); 44693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber 45693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber return meta; 46693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber } 47693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber 48693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber virtual status_t getMaxSampleSize(size_t *max_size) { 4920111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber *max_size = mSize; 50693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber return OK; 51693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber } 52693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber 53693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber virtual status_t start(MetaData *params) { 54693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber return OK; 55693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber } 56693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber 57693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber virtual status_t stop() { 58693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber return OK; 5920111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber } 6020111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 61693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber virtual status_t read( 62693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber MediaBuffer **buffer, const MediaSource::ReadOptions *options) { 63693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber status_t err = mGroup.acquire_buffer(buffer); 64693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber if (err != OK) { 6520111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber return err; 6620111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber } 6720111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 6820111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber char x = (char)((double)rand() / RAND_MAX * 255); 6920111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber memset((*buffer)->data(), x, mSize); 7020111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber (*buffer)->set_range(0, mSize); 7120111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 72693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber return OK; 7320111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber } 7420111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 75693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huberprotected: 76693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber virtual ~DummySource() {} 77693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber 7820111aa043c5f404472bc63b90bc5aad906b1101Andreas Huberprivate: 7920111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber MediaBufferGroup mGroup; 80693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber int mWidth, mHeight; 8120111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber size_t mSize; 8220111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 8320111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber DummySource(const DummySource &); 8420111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber DummySource &operator=(const DummySource &); 8520111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber}; 8620111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 87693d271e62a3726689ff68f4505ba49228eb94b2Andreas Hubersp<MediaSource> createSource(const char *filename) { 88693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber sp<MediaSource> source; 8920111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 90693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber sp<MPEG4Extractor> extractor = 91693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber new MPEG4Extractor(new MmapSource(filename)); 92693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber 93693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber size_t num_tracks = extractor->countTracks(); 9420111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 9520111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber sp<MetaData> meta; 96693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber for (size_t i = 0; i < num_tracks; ++i) { 97693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber meta = extractor->getTrackMetaData(i); 986d43768514650d65f4f3695da4f7c3e2eefd483cAndreas Huber CHECK(meta.get() != NULL); 9920111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 10020111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber const char *mime; 10120111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber if (!meta->findCString(kKeyMIMEType, &mime)) { 10220111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber continue; 10320111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber } 10420111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 10520111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber if (strncasecmp(mime, "video/", 6)) { 10620111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber continue; 10720111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber } 10820111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 109693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber source = extractor->getTrack(i); 11020111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber break; 11120111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber } 11220111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 113693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber return source; 114693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber} 115693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber 116693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huberint main(int argc, char **argv) { 117693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber android::ProcessState::self()->startThreadPool(); 118693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber 119693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber#if 1 120693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber if (argc != 2) { 121693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber fprintf(stderr, "usage: %s filename\n", argv[0]); 12220111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber return 1; 12320111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber } 12420111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 12520111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber OMXClient client; 1266d43768514650d65f4f3695da4f7c3e2eefd483cAndreas Huber CHECK_EQ(client.connect(), OK); 12720111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 128693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber#if 0 129693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber sp<MediaSource> source = createSource(argv[1]); 130693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber 131693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber if (source == NULL) { 132693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber fprintf(stderr, "Unable to find a suitable video track.\n"); 133693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber return 1; 134693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber } 135693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber 136693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber sp<MetaData> meta = source->getFormat(); 137693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber 138693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber sp<OMXCodec> decoder = OMXCodec::Create( 139693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber client.interface(), meta, false /* createEncoder */, source); 14020111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 14120111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber int width, height; 14220111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber bool success = meta->findInt32(kKeyWidth, &width); 14320111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber success = success && meta->findInt32(kKeyHeight, &height); 1446d43768514650d65f4f3695da4f7c3e2eefd483cAndreas Huber CHECK(success); 145693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber#else 146693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber int width = 320; 147693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber int height = 240; 148693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber sp<MediaSource> decoder = new DummySource(width, height); 149693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber#endif 15020111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 15120111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber sp<MetaData> enc_meta = new MetaData; 152693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber // enc_meta->setCString(kKeyMIMEType, "video/3gpp"); 153693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber enc_meta->setCString(kKeyMIMEType, "video/mp4v-es"); 15420111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber enc_meta->setInt32(kKeyWidth, width); 15520111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber enc_meta->setInt32(kKeyHeight, height); 15620111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 157693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber sp<OMXCodec> encoder = 158693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber OMXCodec::Create( 159693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber client.interface(), enc_meta, true /* createEncoder */, decoder); 16020111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 16120111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber#if 1 162693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber sp<MPEG4Writer> writer = new MPEG4Writer("/sdcard/output.mp4"); 163693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber writer->addSource(enc_meta, encoder); 164693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber writer->start(); 165bf37f3364804f521cc61845b1f1ce16fe133814bAndreas Huber sleep(20); 166bf37f3364804f521cc61845b1f1ce16fe133814bAndreas Huber printf("stopping now.\n"); 167693d271e62a3726689ff68f4505ba49228eb94b2Andreas Huber writer->stop(); 16820111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber#else 16920111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber encoder->start(); 17020111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 17120111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber MediaBuffer *buffer; 1726d43768514650d65f4f3695da4f7c3e2eefd483cAndreas Huber while (encoder->read(&buffer) == OK) { 17320111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber printf("got an output frame of size %d\n", buffer->range_length()); 17420111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 17520111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber buffer->release(); 17620111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber buffer = NULL; 17720111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber } 17820111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 17920111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber encoder->stop(); 18020111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber#endif 18120111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 18220111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber client.disconnect(); 18320111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber#endif 18420111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 18520111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber#if 0 18620111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber CameraSource *source = CameraSource::Create(); 18720111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber printf("source = %p\n", source); 18820111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 18920111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber for (int i = 0; i < 100; ++i) { 19020111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber MediaBuffer *buffer; 19120111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber status_t err = source->read(&buffer); 1926d43768514650d65f4f3695da4f7c3e2eefd483cAndreas Huber CHECK_EQ(err, OK); 19320111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 19420111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber printf("got a frame, data=%p, size=%d\n", 19520111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber buffer->data(), buffer->range_length()); 19620111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 19720111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber buffer->release(); 19820111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber buffer = NULL; 19920111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber } 20020111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 20120111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber delete source; 20220111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber source = NULL; 20320111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber#endif 20420111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 20520111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber return 0; 20620111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber} 20720111aa043c5f404472bc63b90bc5aad906b1101Andreas Huber 208