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 19377b2ec9a2885f9b6405b07ba900a9e3f4349c38Kévin PETIT#include <inttypes.h> 20c6ac859f5a82ea8642bc6351a45508a15f224f32Marco Nelissen#include <sys/types.h> 21c6ac859f5a82ea8642bc6351a45508a15f224f32Marco Nelissen#include <sys/stat.h> 22c6ac859f5a82ea8642bc6351a45508a15f224f32Marco Nelissen#include <fcntl.h> 23c6ac859f5a82ea8642bc6351a45508a15f224f32Marco Nelissen 24c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong#include <binder/ProcessState.h> 25f1d5aa162c02a16b7195a43a9bcea4d592600ac4James Dong#include <media/stagefright/foundation/ADebug.h> 26bf03e42c7a246e2c8d53ba944b91f66ee3da10e6Lajos Molnar#include <media/stagefright/foundation/ALooper.h> 27bf03e42c7a246e2c8d53ba944b91f66ee3da10e6Lajos Molnar#include <media/stagefright/foundation/AMessage.h> 28c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong#include <media/stagefright/AudioPlayer.h> 29c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong#include <media/stagefright/MediaBufferGroup.h> 30bf03e42c7a246e2c8d53ba944b91f66ee3da10e6Lajos Molnar#include <media/stagefright/MediaCodecSource.h> 31c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong#include <media/stagefright/MediaDefs.h> 32c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong#include <media/stagefright/MetaData.h> 33c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong#include <media/stagefright/MPEG4Writer.h> 34c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong#include <media/MediaPlayerInterface.h> 35c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 36bf03e42c7a246e2c8d53ba944b91f66ee3da10e6Lajos Molnar#include <OMX_Video.h> 37bf03e42c7a246e2c8d53ba944b91f66ee3da10e6Lajos Molnar 38c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dongusing namespace android; 39c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 40c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong// Print usage showing how to use this utility to record videos 41c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dongstatic void usage(const char *me) { 42c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong fprintf(stderr, "usage: %s\n", me); 43c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong fprintf(stderr, " -h(elp)\n"); 44c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong fprintf(stderr, " -b bit rate in bits per second (default: 300000)\n"); 45dc7d07e5e402b31068a55c8b055d1a82d61f9ca5James Dong fprintf(stderr, " -c YUV420 color format: [0] semi planar or [1] planar or other omx YUV420 color format (default: 1)\n"); 46c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong fprintf(stderr, " -f frame rate in frames per second (default: 30)\n"); 47c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong fprintf(stderr, " -i I frame interval in seconds (default: 1)\n"); 48c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong fprintf(stderr, " -n number of frames to be recorded (default: 300)\n"); 49c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong fprintf(stderr, " -w width in pixels (default: 176)\n"); 50c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong fprintf(stderr, " -t height in pixels (default: 144)\n"); 51c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong fprintf(stderr, " -l encoder level. see omx il header (default: encoder specific)\n"); 52c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong fprintf(stderr, " -p encoder profile. see omx il header (default: encoder specific)\n"); 53c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong fprintf(stderr, " -v video codec: [0] AVC [1] M4V [2] H263 (default: 0)\n"); 54d4483f3fd0e024d833082fce3ddbc3a2f3d89b24Martin Storsjo fprintf(stderr, " -s(oftware) prefer software codec\n"); 55ba8da49e244be1e18b19bb16b16ff5c75e165712Dima Zavin fprintf(stderr, " -o filename: output file (default: /sdcard/output.mp4)\n"); 56c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong exit(1); 57c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong} 58c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 59c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dongclass DummySource : public MediaSource { 60c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 61c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dongpublic: 62c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong DummySource(int width, int height, int nFrames, int fps, int colorFormat) 63c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong : mWidth(width), 64c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong mHeight(height), 65c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong mMaxNumFrames(nFrames), 66c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong mFrameRate(fps), 67c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong mColorFormat(colorFormat), 68c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong mSize((width * height * 3) / 2) { 69c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong 70c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong mGroup.add_buffer(new MediaBuffer(mSize)); 71c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 72c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 73c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong virtual sp<MetaData> getFormat() { 74c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong sp<MetaData> meta = new MetaData; 75c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong meta->setInt32(kKeyWidth, mWidth); 76c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong meta->setInt32(kKeyHeight, mHeight); 77c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong meta->setInt32(kKeyColorFormat, mColorFormat); 78c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong meta->setCString(kKeyMIMEType, MEDIA_MIMETYPE_VIDEO_RAW); 79c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 80c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong return meta; 81c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 82c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 830ae7801bd228025dc4941be086f88e34b70df90fAndy Hung virtual status_t start(MetaData *params __unused) { 84c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong mNumFramesOutput = 0; 85c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong return OK; 86c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 87c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 88c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong virtual status_t stop() { 89c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong return OK; 90c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 91c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 92c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong virtual status_t read( 930ae7801bd228025dc4941be086f88e34b70df90fAndy Hung MediaBuffer **buffer, const MediaSource::ReadOptions *options __unused) { 94c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 95c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong if (mNumFramesOutput % 10 == 0) { 96c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong fprintf(stderr, "."); 97c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 98c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong if (mNumFramesOutput == mMaxNumFrames) { 99c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong return ERROR_END_OF_STREAM; 100c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 101c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 102c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong status_t err = mGroup.acquire_buffer(buffer); 103c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong if (err != OK) { 104c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong return err; 105c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 106c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 107c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong // We don't care about the contents. we just test video encoder 108c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong // Also, by skipping the content generation, we can return from 109c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong // read() much faster. 1100ae7801bd228025dc4941be086f88e34b70df90fAndy Hung#if 0 1110ae7801bd228025dc4941be086f88e34b70df90fAndy Hung // iterate through solid planes of color. 1120ae7801bd228025dc4941be086f88e34b70df90fAndy Hung static unsigned char x = 0x60; 1130ae7801bd228025dc4941be086f88e34b70df90fAndy Hung memset((*buffer)->data(), x, mSize); 1140ae7801bd228025dc4941be086f88e34b70df90fAndy Hung x = x >= 0xa0 ? 0x60 : x + 1; 1150ae7801bd228025dc4941be086f88e34b70df90fAndy Hung#endif 116c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong (*buffer)->set_range(0, mSize); 117c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong (*buffer)->meta_data()->clear(); 118c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong (*buffer)->meta_data()->setInt64( 119c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong kKeyTime, (mNumFramesOutput * 1000000) / mFrameRate); 120c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong ++mNumFramesOutput; 121c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 122c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong return OK; 123c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 124c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 125c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dongprotected: 126c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong virtual ~DummySource() {} 127c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 128c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dongprivate: 129c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong MediaBufferGroup mGroup; 130c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int mWidth, mHeight; 131c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int mMaxNumFrames; 132c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int mFrameRate; 133c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int mColorFormat; 134c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong size_t mSize; 135c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int64_t mNumFramesOutput;; 136c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 137c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong DummySource(const DummySource &); 138c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong DummySource &operator=(const DummySource &); 139c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong}; 140c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 141c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dongenum { 142c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong kYUV420SP = 0, 143c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong kYUV420P = 1, 144c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong}; 145c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 146c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong// returns -1 if mapping of the given color is unsuccessful 147c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong// returns an omx color enum value otherwise 148c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dongstatic int translateColorToOmxEnumValue(int color) { 149c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong switch (color) { 150c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong case kYUV420SP: 151c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong return OMX_COLOR_FormatYUV420SemiPlanar; 152c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong case kYUV420P: 153c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong return OMX_COLOR_FormatYUV420Planar; 154c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong default: 155dc7d07e5e402b31068a55c8b055d1a82d61f9ca5James Dong fprintf(stderr, "Custom OMX color format: %d\n", color); 156dc7d07e5e402b31068a55c8b055d1a82d61f9ca5James Dong if (color == OMX_TI_COLOR_FormatYUV420PackedSemiPlanar || 157dc7d07e5e402b31068a55c8b055d1a82d61f9ca5James Dong color == OMX_QCOM_COLOR_FormatYVU420SemiPlanar) { 158dc7d07e5e402b31068a55c8b055d1a82d61f9ca5James Dong return color; 159dc7d07e5e402b31068a55c8b055d1a82d61f9ca5James Dong } 160c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 161dc7d07e5e402b31068a55c8b055d1a82d61f9ca5James Dong return -1; 162c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong} 163c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 164c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dongint main(int argc, char **argv) { 165c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 166c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong // Default values for the program if not overwritten 167c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int frameRateFps = 30; 168c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int width = 176; 169c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int height = 144; 170c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int bitRateBps = 300000; 171c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int iFramesIntervalSeconds = 1; 172c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int colorFormat = OMX_COLOR_FormatYUV420Planar; 173c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int nFrames = 300; 174c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong int level = -1; // Encoder specific default 175c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong int profile = -1; // Encoder specific default 176c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int codec = 0; 1770ae7801bd228025dc4941be086f88e34b70df90fAndy Hung const char *fileName = "/sdcard/output.mp4"; 178d4483f3fd0e024d833082fce3ddbc3a2f3d89b24Martin Storsjo bool preferSoftwareCodec = false; 179c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 180c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong android::ProcessState::self()->startThreadPool(); 181c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int res; 182ba8da49e244be1e18b19bb16b16ff5c75e165712Dima Zavin while ((res = getopt(argc, argv, "b:c:f:i:n:w:t:l:p:v:o:hs")) >= 0) { 183c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong switch (res) { 184c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong case 'b': 185c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong { 186c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong bitRateBps = atoi(optarg); 187c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong break; 188c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 189c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 190c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong case 'c': 191c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong { 192c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong colorFormat = translateColorToOmxEnumValue(atoi(optarg)); 193c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong if (colorFormat == -1) { 194c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong usage(argv[0]); 195c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 196c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong break; 197c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 198c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 199c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong case 'f': 200c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong { 201c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong frameRateFps = atoi(optarg); 202c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong break; 203c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 204c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 205c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong case 'i': 206c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong { 207c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong iFramesIntervalSeconds = atoi(optarg); 208c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong break; 209c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 210c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 211c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong case 'n': 212c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong { 213c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong nFrames = atoi(optarg); 214c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong break; 215c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 216c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 217c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong case 'w': 218c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong { 219c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong width = atoi(optarg); 220c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong break; 221c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 222c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 223c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong case 't': 224c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong { 225c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong height = atoi(optarg); 226c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong break; 227c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 228c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 229c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong case 'l': 230c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong { 231c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong level = atoi(optarg); 232c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong break; 233c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong } 234c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong 235c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong case 'p': 236c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong { 237c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong profile = atoi(optarg); 238c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong break; 239c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong } 240c97a1325466bb0e6cee884ec24a04b65082b853aJames 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 250ba8da49e244be1e18b19bb16b16ff5c75e165712Dima Zavin case 'o': 251ba8da49e244be1e18b19bb16b16ff5c75e165712Dima Zavin { 252ba8da49e244be1e18b19bb16b16ff5c75e165712Dima Zavin fileName = optarg; 253ba8da49e244be1e18b19bb16b16ff5c75e165712Dima Zavin break; 254ba8da49e244be1e18b19bb16b16ff5c75e165712Dima Zavin } 255ba8da49e244be1e18b19bb16b16ff5c75e165712Dima Zavin 256d4483f3fd0e024d833082fce3ddbc3a2f3d89b24Martin Storsjo case 's': 257d4483f3fd0e024d833082fce3ddbc3a2f3d89b24Martin Storsjo { 258d4483f3fd0e024d833082fce3ddbc3a2f3d89b24Martin Storsjo preferSoftwareCodec = true; 259d4483f3fd0e024d833082fce3ddbc3a2f3d89b24Martin Storsjo break; 260d4483f3fd0e024d833082fce3ddbc3a2f3d89b24Martin Storsjo } 261d4483f3fd0e024d833082fce3ddbc3a2f3d89b24Martin Storsjo 262c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong case 'h': 263c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong default: 264c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong { 265c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong usage(argv[0]); 266c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong break; 267c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 268c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 269c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 270c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 271c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong status_t err = OK; 272c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong sp<MediaSource> source = 273c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong new DummySource(width, height, nFrames, frameRateFps, colorFormat); 274c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 275bf03e42c7a246e2c8d53ba944b91f66ee3da10e6Lajos Molnar sp<AMessage> enc_meta = new AMessage; 276c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong switch (codec) { 277c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong case 1: 278bf03e42c7a246e2c8d53ba944b91f66ee3da10e6Lajos Molnar enc_meta->setString("mime", MEDIA_MIMETYPE_VIDEO_MPEG4); 279c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong break; 280c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong case 2: 281bf03e42c7a246e2c8d53ba944b91f66ee3da10e6Lajos Molnar enc_meta->setString("mime", MEDIA_MIMETYPE_VIDEO_H263); 282c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong break; 283c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong default: 284bf03e42c7a246e2c8d53ba944b91f66ee3da10e6Lajos Molnar enc_meta->setString("mime", MEDIA_MIMETYPE_VIDEO_AVC); 285c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong break; 286c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 287bf03e42c7a246e2c8d53ba944b91f66ee3da10e6Lajos Molnar enc_meta->setInt32("width", width); 288bf03e42c7a246e2c8d53ba944b91f66ee3da10e6Lajos Molnar enc_meta->setInt32("height", height); 289bf03e42c7a246e2c8d53ba944b91f66ee3da10e6Lajos Molnar enc_meta->setInt32("frame-rate", frameRateFps); 290bf03e42c7a246e2c8d53ba944b91f66ee3da10e6Lajos Molnar enc_meta->setInt32("bitrate", bitRateBps); 291bf03e42c7a246e2c8d53ba944b91f66ee3da10e6Lajos Molnar enc_meta->setInt32("stride", width); 292bf03e42c7a246e2c8d53ba944b91f66ee3da10e6Lajos Molnar enc_meta->setInt32("slice-height", height); 293bf03e42c7a246e2c8d53ba944b91f66ee3da10e6Lajos Molnar enc_meta->setInt32("i-frame-interval", iFramesIntervalSeconds); 294bf03e42c7a246e2c8d53ba944b91f66ee3da10e6Lajos Molnar enc_meta->setInt32("color-format", colorFormat); 295c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong if (level != -1) { 296bf03e42c7a246e2c8d53ba944b91f66ee3da10e6Lajos Molnar enc_meta->setInt32("level", level); 297c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong } 298c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong if (profile != -1) { 299bf03e42c7a246e2c8d53ba944b91f66ee3da10e6Lajos Molnar enc_meta->setInt32("profile", profile); 300c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong } 301c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 302bf03e42c7a246e2c8d53ba944b91f66ee3da10e6Lajos Molnar sp<ALooper> looper = new ALooper; 303bf03e42c7a246e2c8d53ba944b91f66ee3da10e6Lajos Molnar looper->setName("recordvideo"); 304bf03e42c7a246e2c8d53ba944b91f66ee3da10e6Lajos Molnar looper->start(); 305bf03e42c7a246e2c8d53ba944b91f66ee3da10e6Lajos Molnar 306b2487f03f12dcafdb801fc0007c8df8412397f44Marco Nelissen sp<IMediaSource> encoder = 307bf03e42c7a246e2c8d53ba944b91f66ee3da10e6Lajos Molnar MediaCodecSource::Create( 308bf03e42c7a246e2c8d53ba944b91f66ee3da10e6Lajos Molnar looper, enc_meta, source, NULL /* consumer */, 309bf03e42c7a246e2c8d53ba944b91f66ee3da10e6Lajos Molnar preferSoftwareCodec ? MediaCodecSource::FLAG_PREFER_SOFTWARE_CODEC : 0); 310c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 311c6ac859f5a82ea8642bc6351a45508a15f224f32Marco Nelissen int fd = open(fileName, O_CREAT | O_LARGEFILE | O_TRUNC | O_RDWR, S_IRUSR | S_IWUSR); 312c6ac859f5a82ea8642bc6351a45508a15f224f32Marco Nelissen if (fd < 0) { 313c6ac859f5a82ea8642bc6351a45508a15f224f32Marco Nelissen fprintf(stderr, "couldn't open file"); 314c6ac859f5a82ea8642bc6351a45508a15f224f32Marco Nelissen return 1; 315c6ac859f5a82ea8642bc6351a45508a15f224f32Marco Nelissen } 316c6ac859f5a82ea8642bc6351a45508a15f224f32Marco Nelissen sp<MPEG4Writer> writer = new MPEG4Writer(fd); 317c6ac859f5a82ea8642bc6351a45508a15f224f32Marco Nelissen close(fd); 318c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong writer->addSource(encoder); 319c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int64_t start = systemTime(); 320f1d5aa162c02a16b7195a43a9bcea4d592600ac4James Dong CHECK_EQ((status_t)OK, writer->start()); 321c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong while (!writer->reachedEOS()) { 3221301a437ff9f5fa6c4ed78f88add3d9bfce97935Martin Storsjo usleep(100000); 323c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 324c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong err = writer->stop(); 325c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong int64_t end = systemTime(); 326c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 327c97a1325466bb0e6cee884ec24a04b65082b853aJames Dong fprintf(stderr, "$\n"); 328c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong 329c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong if (err != OK && err != ERROR_END_OF_STREAM) { 330c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong fprintf(stderr, "record failed: %d\n", err); 331c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong return 1; 332c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong } 333377b2ec9a2885f9b6405b07ba900a9e3f4349c38Kévin PETIT fprintf(stderr, "encoding %d frames in %" PRId64 " us\n", nFrames, (end-start)/1000); 334c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong fprintf(stderr, "encoding speed is: %.2f fps\n", (nFrames * 1E9) / (end-start)); 335c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong return 0; 336c69c13e18a88090c6b06ba8ac244e75cf5e76e38James Dong} 337