recordvideo.cpp revision c69c13e18a88090c6b06ba8ac244e75cf5e76e38
1c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong/* 2c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong * Copyright (C) 2010 The Android Open Source Project 3c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong * 4c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong * Licensed under the Apache License, Version 2.0 (the "License"); 5c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong * you may not use this file except in compliance with the License. 6c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong * You may obtain a copy of the License at 7c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong * 8c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong * http://www.apache.org/licenses/LICENSE-2.0 9c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong * 10c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong * Unless required by applicable law or agreed to in writing, software 11c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong * distributed under the License is distributed on an "AS IS" BASIS, 12c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong * See the License for the specific language governing permissions and 14c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong * limitations under the License. 15c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong */ 16c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 17c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong#include "SineSource.h" 18c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 19c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong#include <binder/ProcessState.h> 20c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong#include <media/stagefright/AudioPlayer.h> 21c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong#include <media/stagefright/FileSource.h> 22c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong#include <media/stagefright/MediaBufferGroup.h> 23c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong#include <media/stagefright/MediaDebug.h> 24c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong#include <media/stagefright/MediaDefs.h> 25c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong#include <media/stagefright/MetaData.h> 26c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong#include <media/stagefright/MediaExtractor.h> 27c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong#include <media/stagefright/MPEG4Writer.h> 28c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong#include <media/stagefright/OMXClient.h> 29c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong#include <media/stagefright/OMXCodec.h> 30c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong#include <media/MediaPlayerInterface.h> 31c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 32c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dongusing namespace android; 33c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 34c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong// print usage showing how to use this utility to record videos 35c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dongstatic void usage(const char *me) { 36c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong fprintf(stderr, "usage: %s\n", me); 37c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong fprintf(stderr, " -h(elp)\n"); 38c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong fprintf(stderr, " -b bit rate in bits per second (default 300000)\n"); 39c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong fprintf(stderr, " -c YUV420 color format: [0] semi planar or [1] planar (default 1)\n"); 40c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong fprintf(stderr, " -f frame rate in frames per second (default 30)\n"); 41c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong fprintf(stderr, " -i I frame interval in seconds (default 1)\n"); 42c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong fprintf(stderr, " -n number of frames to be recorded (default 300)\n"); 43c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong fprintf(stderr, " -w width in pixels (default 176)\n"); 44c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong fprintf(stderr, " -t height in pixels (default 144)\n"); 45c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong fprintf(stderr, " -v video codec: [0] AVC [1] M4V [2] H263 (default 0)\n"); 46c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong exit(1); 47c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong} 48c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 49c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dongclass DummySource : public MediaSource { 50c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 51c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dongpublic: 52c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong DummySource(int width, int height, int nFrames, int fps, int colorFormat) 53c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong : mWidth(width), 54c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong mHeight(height), 55c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong mMaxNumFrames(nFrames), 56c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong mFrameRate(fps), 57c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong mColorFormat(colorFormat), 58c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong mSize((width * height * 3) / 2) { 59c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong mGroup.add_buffer(new MediaBuffer(mSize)); 60c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 61c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong // Check the color format to make sure 62c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong // that the buffer size mSize it set correctly above. 63c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong CHECK(colorFormat == OMX_COLOR_FormatYUV420SemiPlanar || 64c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong colorFormat == OMX_COLOR_FormatYUV420Planar); 65c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 66c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 67c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong virtual sp<MetaData> getFormat() { 68c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong sp<MetaData> meta = new MetaData; 69c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong meta->setInt32(kKeyWidth, mWidth); 70c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong meta->setInt32(kKeyHeight, mHeight); 71c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong meta->setInt32(kKeyColorFormat, mColorFormat); 72c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong meta->setCString(kKeyMIMEType, MEDIA_MIMETYPE_VIDEO_RAW); 73c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 74c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong return meta; 75c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 76c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 77c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong virtual status_t start(MetaData *params) { 78c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong mNumFramesOutput = 0; 79c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong return OK; 80c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 81c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 82c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong virtual status_t stop() { 83c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong return OK; 84c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 85c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 86c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong virtual status_t read( 87c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong MediaBuffer **buffer, const MediaSource::ReadOptions *options) { 88c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 89c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong if (mNumFramesOutput % 10 == 0) { 90c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong fprintf(stderr, "."); 91c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 92c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong if (mNumFramesOutput == mMaxNumFrames) { 93c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong return ERROR_END_OF_STREAM; 94c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 95c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 96c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong status_t err = mGroup.acquire_buffer(buffer); 97c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong if (err != OK) { 98c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong return err; 99c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 100c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 101c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong char x = (char)((double)rand() / RAND_MAX * 255); 102c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong memset((*buffer)->data(), x, mSize); 103c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong (*buffer)->set_range(0, mSize); 104c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong (*buffer)->meta_data()->clear(); 105c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong (*buffer)->meta_data()->setInt64( 106c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong kKeyTime, (mNumFramesOutput * 1000000) / mFrameRate); 107c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong ++mNumFramesOutput; 108c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 109c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong return OK; 110c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 111c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 112c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dongprotected: 113c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong virtual ~DummySource() {} 114c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 115c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dongprivate: 116c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong MediaBufferGroup mGroup; 117c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int mWidth, mHeight; 118c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int mMaxNumFrames; 119c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int mFrameRate; 120c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int mColorFormat; 121c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong size_t mSize; 122c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int64_t mNumFramesOutput;; 123c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 124c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong DummySource(const DummySource &); 125c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong DummySource &operator=(const DummySource &); 126c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong}; 127c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 128c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dongsp<MediaSource> createSource(const char *filename) { 129c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong sp<MediaSource> source; 130c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 131c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong sp<MediaExtractor> extractor = 132c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong MediaExtractor::Create(new FileSource(filename)); 133c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong if (extractor == NULL) { 134c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong return NULL; 135c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 136c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 137c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong size_t num_tracks = extractor->countTracks(); 138c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 139c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong sp<MetaData> meta; 140c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong for (size_t i = 0; i < num_tracks; ++i) { 141c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong meta = extractor->getTrackMetaData(i); 142c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong CHECK(meta.get() != NULL); 143c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 144c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong const char *mime; 145c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong if (!meta->findCString(kKeyMIMEType, &mime)) { 146c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong continue; 147c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 148c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 149c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong if (strncasecmp(mime, "video/", 6)) { 150c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong continue; 151c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 152c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 153c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong source = extractor->getTrack(i); 154c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong break; 155c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 156c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 157c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong return source; 158c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong} 159c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 160c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dongenum { 161c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong kYUV420SP = 0, 162c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong kYUV420P = 1, 163c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong}; 164c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 165c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong// returns -1 if mapping of the given color is unsuccessful 166c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong// returns an omx color enum value otherwise 167c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dongstatic int translateColorToOmxEnumValue(int color) { 168c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong switch (color) { 169c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong case kYUV420SP: 170c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong return OMX_COLOR_FormatYUV420SemiPlanar; 171c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong case kYUV420P: 172c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong return OMX_COLOR_FormatYUV420Planar; 173c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong default: 174c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong fprintf(stderr, "Unsupported color: %d\n", color); 175c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong return -1; 176c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 177c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong} 178c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 179c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dongint main(int argc, char **argv) { 180c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 181c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong // Default values for the program if not overwritten 182c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int frameRateFps = 30; 183c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int width = 176; 184c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int height = 144; 185c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int bitRateBps = 300000; 186c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int iFramesIntervalSeconds = 1; 187c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int colorFormat = OMX_COLOR_FormatYUV420Planar; 188c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int nFrames = 300; 189c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int codec = 0; 190c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong const char *fileName = "/sdcard/output.mp4"; 191c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 192c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong android::ProcessState::self()->startThreadPool(); 193c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int res; 194c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong while ((res = getopt(argc, argv, "b:c:f:i:n:w:t:v:o:h")) >= 0) { 195c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong switch (res) { 196c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong case 'b': 197c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong { 198c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong bitRateBps = atoi(optarg); 199c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong break; 200c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 201c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 202c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong case 'c': 203c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong { 204c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong colorFormat = translateColorToOmxEnumValue(atoi(optarg)); 205c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong if (colorFormat == -1) { 206c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong usage(argv[0]); 207c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 208c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong break; 209c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 210c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 211c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong case 'f': 212c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong { 213c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong frameRateFps = atoi(optarg); 214c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong break; 215c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 216c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 217c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong case 'i': 218c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong { 219c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong iFramesIntervalSeconds = atoi(optarg); 220c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong break; 221c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 222c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 223c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong case 'n': 224c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong { 225c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong nFrames = atoi(optarg); 226c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong break; 227c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 228c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 229c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong case 'w': 230c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong { 231c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong width = atoi(optarg); 232c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong break; 233c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 234c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 235c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong case 't': 236c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong { 237c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong height = atoi(optarg); 238c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong break; 239c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 240c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 241c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong case 'v': 242c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong { 243c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong codec = atoi(optarg); 244c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong if (codec < 0 || codec > 2) { 245c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong usage(argv[0]); 246c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 247c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong break; 248c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 249c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 250c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong case 'h': 251c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong default: 252c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong { 253c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong usage(argv[0]); 254c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong break; 255c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 256c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 257c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 258c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 259c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong OMXClient client; 260c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong CHECK_EQ(client.connect(), OK); 261c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 262c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong status_t err = OK; 263c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong sp<MediaSource> decoder = new DummySource(width, height, nFrames, frameRateFps, colorFormat); 264c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 265c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong sp<MetaData> enc_meta = new MetaData; 266c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong switch (codec) { 267c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong case 1: 268c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong enc_meta->setCString(kKeyMIMEType, MEDIA_MIMETYPE_VIDEO_MPEG4); 269c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong break; 270c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong case 2: 271c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong enc_meta->setCString(kKeyMIMEType, MEDIA_MIMETYPE_VIDEO_H263); 272c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong break; 273c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong default: 274c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong enc_meta->setCString(kKeyMIMEType, MEDIA_MIMETYPE_VIDEO_AVC); 275c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong break; 276c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 277c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong enc_meta->setInt32(kKeyWidth, width); 278c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong enc_meta->setInt32(kKeyHeight, height); 279c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong enc_meta->setInt32(kKeySampleRate, frameRateFps); 280c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong enc_meta->setInt32(kKeyBitRate, bitRateBps); 281c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong enc_meta->setInt32(kKeyStride, width); 282c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong enc_meta->setInt32(kKeySliceHeight, height); 283c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong enc_meta->setInt32(kKeyIFramesInterval, iFramesIntervalSeconds); 284c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong enc_meta->setInt32(kKeyColorFormat, colorFormat); 285c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 286c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong sp<MediaSource> encoder = 287c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong OMXCodec::Create( 288c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong client.interface(), enc_meta, true /* createEncoder */, decoder); 289c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 290c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong sp<MPEG4Writer> writer = new MPEG4Writer(fileName); 291c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong writer->addSource(encoder); 292c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int64_t start = systemTime(); 293c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong CHECK_EQ(OK, writer->start()); 294c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong while (!writer->reachedEOS()) { 295c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 296c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong err = writer->stop(); 297c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int64_t end = systemTime(); 298c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 299c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong printf("$\n"); 300c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong client.disconnect(); 301c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 302c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong if (err != OK && err != ERROR_END_OF_STREAM) { 303c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong fprintf(stderr, "record failed: %d\n", err); 304c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong return 1; 305c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 306c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong fprintf(stderr, "encoding %d frames in %lld us\n", nFrames, (end-start)/1000); 307c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong fprintf(stderr, "encoding speed is: %.2f fps\n", (nFrames * 1E9) / (end-start)); 308c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong return 0; 309c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong} 310