blob: 8d5eec8b6406d3d2bcf126254bd54b5a69853cdc [file] [log] [blame]
/*
* Copyright (C) 2010 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include <media/stagefright/foundation/ADebug.h>
#include <media/stagefright/AMRWriter.h>
#include <media/stagefright/MediaBuffer.h>
#include <media/stagefright/MediaDefs.h>
#include <media/stagefright/MediaErrors.h>
#include <media/stagefright/MediaSource.h>
#include <media/stagefright/MetaData.h>
#include <media/mediarecorder.h>
#include <sys/prctl.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
namespace android {
AMRWriter::AMRWriter(const char *filename)
: mFd(-1),
mInitCheck(NO_INIT),
mStarted(false),
mPaused(false),
mResumed(false) {
mFd = open(filename, O_CREAT | O_LARGEFILE | O_TRUNC | O_RDWR, S_IRUSR | S_IWUSR);
if (mFd >= 0) {
mInitCheck = OK;
}
}
AMRWriter::AMRWriter(int fd)
: mFd(dup(fd)),
mInitCheck(mFd < 0? NO_INIT: OK),
mStarted(false),
mPaused(false),
mResumed(false) {
}
AMRWriter::~AMRWriter() {
if (mStarted) {
reset();
}
if (mFd != -1) {
close(mFd);
mFd = -1;
}
}
status_t AMRWriter::initCheck() const {
return mInitCheck;
}
status_t AMRWriter::addSource(const sp<MediaSource> &source) {
if (mInitCheck != OK) {
return mInitCheck;
}
if (mSource != NULL) {
// AMR files only support a single track of audio.
return UNKNOWN_ERROR;
}
sp<MetaData> meta = source->getFormat();
const char *mime;
CHECK(meta->findCString(kKeyMIMEType, &mime));
bool isWide = false;
if (!strcasecmp(mime, MEDIA_MIMETYPE_AUDIO_AMR_WB)) {
isWide = true;
} else if (strcasecmp(mime, MEDIA_MIMETYPE_AUDIO_AMR_NB)) {
return ERROR_UNSUPPORTED;
}
int32_t channelCount;
int32_t sampleRate;
CHECK(meta->findInt32(kKeyChannelCount, &channelCount));
CHECK_EQ(channelCount, 1);
CHECK(meta->findInt32(kKeySampleRate, &sampleRate));
CHECK_EQ(sampleRate, (isWide ? 16000 : 8000));
mSource = source;
const char *kHeader = isWide ? "#!AMR-WB\n" : "#!AMR\n";
ssize_t n = strlen(kHeader);
if (write(mFd, kHeader, n) != n) {
return ERROR_IO;
}
return OK;
}
status_t AMRWriter::start(MetaData *params) {
if (mInitCheck != OK) {
return mInitCheck;
}
if (mSource == NULL) {
return UNKNOWN_ERROR;
}
if (mStarted && mPaused) {
mPaused = false;
mResumed = true;
return OK;
} else if (mStarted) {
// Already started, does nothing
return OK;
}
status_t err = mSource->start();
if (err != OK) {
return err;
}
pthread_attr_t attr;
pthread_attr_init(&attr);
pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_JOINABLE);
mReachedEOS = false;
mDone = false;
pthread_create(&mThread, &attr, ThreadWrapper, this);
pthread_attr_destroy(&attr);
mStarted = true;
return OK;
}
status_t AMRWriter::pause() {
if (!mStarted) {
return OK;
}
mPaused = true;
return OK;
}
status_t AMRWriter::reset() {
if (!mStarted) {
return OK;
}
mDone = true;
void *dummy;
pthread_join(mThread, &dummy);
status_t err = (status_t) dummy;
{
status_t status = mSource->stop();
if (err == OK &&
(status != OK && status != ERROR_END_OF_STREAM)) {
err = status;
}
}
mStarted = false;
return err;
}
bool AMRWriter::exceedsFileSizeLimit() {
if (mMaxFileSizeLimitBytes == 0) {
return false;
}
return mEstimatedSizeBytes >= mMaxFileSizeLimitBytes;
}
bool AMRWriter::exceedsFileDurationLimit() {
if (mMaxFileDurationLimitUs == 0) {
return false;
}
return mEstimatedDurationUs >= mMaxFileDurationLimitUs;
}
// static
void *AMRWriter::ThreadWrapper(void *me) {
return (void *) static_cast<AMRWriter *>(me)->threadFunc();
}
status_t AMRWriter::threadFunc() {
mEstimatedDurationUs = 0;
mEstimatedSizeBytes = 0;
bool stoppedPrematurely = true;
int64_t previousPausedDurationUs = 0;
int64_t maxTimestampUs = 0;
status_t err = OK;
prctl(PR_SET_NAME, (unsigned long)"AMRWriter", 0, 0, 0);
while (!mDone) {
MediaBuffer *buffer;
err = mSource->read(&buffer);
if (err != OK) {
break;
}
if (mPaused) {
buffer->release();
buffer = NULL;
continue;
}
mEstimatedSizeBytes += buffer->range_length();
if (exceedsFileSizeLimit()) {
buffer->release();
buffer = NULL;
notify(MEDIA_RECORDER_EVENT_INFO, MEDIA_RECORDER_INFO_MAX_FILESIZE_REACHED, 0);
break;
}
int64_t timestampUs;
CHECK(buffer->meta_data()->findInt64(kKeyTime, &timestampUs));
if (timestampUs > mEstimatedDurationUs) {
mEstimatedDurationUs = timestampUs;
}
if (mResumed) {
previousPausedDurationUs += (timestampUs - maxTimestampUs - 20000);
mResumed = false;
}
timestampUs -= previousPausedDurationUs;
ALOGV("time stamp: %lld, previous paused duration: %lld",
timestampUs, previousPausedDurationUs);
if (timestampUs > maxTimestampUs) {
maxTimestampUs = timestampUs;
}
if (exceedsFileDurationLimit()) {
buffer->release();
buffer = NULL;
notify(MEDIA_RECORDER_EVENT_INFO, MEDIA_RECORDER_INFO_MAX_DURATION_REACHED, 0);
break;
}
ssize_t n = write(mFd,
(const uint8_t *)buffer->data() + buffer->range_offset(),
buffer->range_length());
if (n < (ssize_t)buffer->range_length()) {
buffer->release();
buffer = NULL;
err = ERROR_IO;
break;
}
if (err != OK) {
break;
}
if (stoppedPrematurely) {
stoppedPrematurely = false;
}
buffer->release();
buffer = NULL;
}
if ((err == OK || err == ERROR_END_OF_STREAM) && stoppedPrematurely) {
err = ERROR_MALFORMED;
}
close(mFd);
mFd = -1;
mReachedEOS = true;
if (err == ERROR_END_OF_STREAM) {
return OK;
}
return err;
}
bool AMRWriter::reachedEOS() {
return mReachedEOS;
}
} // namespace android