1cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber/* 2cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber * Copyright (C) 2010 The Android Open Source Project 3cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber * 4cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber * Licensed under the Apache License, Version 2.0 (the "License"); 5cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber * you may not use this file except in compliance with the License. 6cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber * You may obtain a copy of the License at 7cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber * 8cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber * http://www.apache.org/licenses/LICENSE-2.0 9cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber * 10cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber * Unless required by applicable law or agreed to in writing, software 11cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber * distributed under the License is distributed on an "AS IS" BASIS, 12cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber * See the License for the specific language governing permissions and 14cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber * limitations under the License. 15cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber */ 16cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber 17cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber#ifndef VIDEO_SOURCE_H_ 18cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber 19cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber#define VIDEO_SOURCE_H_ 20cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber 21cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber#include <media/stagefright/MediaBufferGroup.h> 22cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber#include <media/stagefright/MediaDefs.h> 23cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber#include <media/stagefright/MediaSource.h> 24cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber#include <media/stagefright/MetaData.h> 25cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber 26cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Hubernamespace android { 27cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber 28cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huberclass VideoSource : public MediaSource { 29cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber static const int32_t kFramerate = 24; // fps 30cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber 31cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huberpublic: 32cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber VideoSource(int width, int height) 33cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber : mWidth(width), 34cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber mHeight(height), 35cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber mSize((width * height * 3) / 2) { 36cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber mGroup.add_buffer(new MediaBuffer(mSize)); 37cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber } 38cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber 39cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber virtual sp<MetaData> getFormat() { 40cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber sp<MetaData> meta = new MetaData; 41cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber meta->setInt32(kKeyWidth, mWidth); 42cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber meta->setInt32(kKeyHeight, mHeight); 43cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber meta->setCString(kKeyMIMEType, MEDIA_MIMETYPE_VIDEO_RAW); 44cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber 45cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber return meta; 46cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber } 47cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber 48cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber virtual status_t start(MetaData *params) { 49cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber mNumFramesOutput = 0; 50cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber return OK; 51cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber } 52cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber 53cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber virtual status_t stop() { 54cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber return OK; 55cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber } 56cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber 57cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber virtual status_t read( 58cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber MediaBuffer **buffer, const MediaSource::ReadOptions *options) { 59cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber if (mNumFramesOutput == kFramerate * 100) { 60cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber // Stop returning data after 10 secs. 61cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber return ERROR_END_OF_STREAM; 62cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber } 63cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber 64cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber // printf("VideoSource::read\n"); 65cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber status_t err = mGroup.acquire_buffer(buffer); 66cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber if (err != OK) { 67cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber return err; 68cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber } 69cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber 70cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber char x = (char)((double)rand() / RAND_MAX * 255); 71cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber memset((*buffer)->data(), x, mSize); 72cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber (*buffer)->set_range(0, mSize); 73cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber (*buffer)->meta_data()->clear(); 74cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber (*buffer)->meta_data()->setInt64( 75cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber kKeyTime, (mNumFramesOutput * 1000000) / kFramerate); 76cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber ++mNumFramesOutput; 77cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber 78cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber // printf("VideoSource::read - returning buffer\n"); 79cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber // LOG(INFO)("VideoSource::read - returning buffer"); 80cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber return OK; 81cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber } 82cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber 83cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huberprotected: 84cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber virtual ~VideoSource() {} 85cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber 86cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huberprivate: 87cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber MediaBufferGroup mGroup; 88cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber int mWidth, mHeight; 89cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber size_t mSize; 90cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber int64_t mNumFramesOutput;; 91cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber 92cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber VideoSource(const VideoSource &); 93cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber VideoSource &operator=(const VideoSource &); 94cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber}; 95cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber 96cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber} // namespace android 97cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber 98cf7b9c7aae758ac0b99833915053c63c2ac46e09Andreas Huber#endif // VIDEO_SOURCE_H_ 99