blob: 0a1df0504feb5e223455f10b910e3f8706c03de1 [file] [log] [blame]
/*
* Copyright (C) 2014 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.
*/
//#define LOG_NDEBUG 0
#define LOG_TAG "SaturationFilter"
#include <utils/Log.h>
#include <media/MediaCodecBuffer.h>
#include <media/stagefright/foundation/ADebug.h>
#include <media/stagefright/foundation/AMessage.h>
#include "SaturationFilter.h"
namespace android {
status_t SaturationFilter::configure(const sp<AMessage> &msg) {
status_t err = SimpleFilter::configure(msg);
if (err != OK) {
return err;
}
if (!msg->findString("cacheDir", &mCacheDir)) {
ALOGE("Failed to find cache directory in config message.");
return NAME_NOT_FOUND;
}
return OK;
}
status_t SaturationFilter::start() {
// TODO: use a single RS context object for entire application
mRS = new RSC::RS();
if (!mRS->init(mCacheDir.c_str())) {
ALOGE("Failed to initialize RenderScript context.");
return NO_INIT;
}
// 32-bit elements for ARGB8888
RSC::sp<const RSC::Element> e = RSC::Element::U8_4(mRS);
RSC::Type::Builder tb(mRS, e);
tb.setX(mWidth);
tb.setY(mHeight);
RSC::sp<const RSC::Type> t = tb.create();
mAllocIn = RSC::Allocation::createTyped(mRS, t);
mAllocOut = RSC::Allocation::createTyped(mRS, t);
mScript = new ScriptC_saturationARGB(mRS);
mScript->set_gSaturation(mSaturation);
return OK;
}
void SaturationFilter::reset() {
mScript.clear();
mAllocOut.clear();
mAllocIn.clear();
mRS.clear();
}
status_t SaturationFilter::setParameters(const sp<AMessage> &msg) {
sp<AMessage> params;
CHECK(msg->findMessage("params", &params));
float saturation;
if (params->findFloat("saturation", &saturation)) {
mSaturation = saturation;
}
return OK;
}
status_t SaturationFilter::processBuffers(
const sp<MediaCodecBuffer> &srcBuffer, const sp<MediaCodecBuffer> &outBuffer) {
mAllocIn->copy1DRangeFrom(0, mWidth * mHeight, srcBuffer->data());
mScript->forEach_root(mAllocIn, mAllocOut);
mAllocOut->copy1DRangeTo(0, mWidth * mHeight, outBuffer->data());
return OK;
}
} // namespace android