blob: 697ce8e90d45d7f67c3cf6b5494e26050cf666ef [file] [log] [blame]
/*
* Copyright (C) 2012 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 <stdint.h>
#include <math.h>
#include <utils/StrongPointer.h>
#include "audio/Buffer.h"
#include "BuiltinProcessing.h"
#include "Log.h"
#include "task/TaskCase.h"
// Buffer, Value, Value
static const bool RMS_MVA_INPUT_TYPE[] = {true, false, false};
// Value
static const bool RMS_MVA_OUTPUT_TYPE[] = {false};
BuiltinProcessing::BuiltinInfo BuiltinProcessing::BUINTIN_FN_TABLE[N_BUILTIN_FNS] =
{
{
"rms_mva", &BuiltinProcessing::rms_mva,
sizeof(RMS_MVA_INPUT_TYPE)/sizeof(bool), RMS_MVA_INPUT_TYPE,
sizeof(RMS_MVA_OUTPUT_TYPE)/sizeof(bool), RMS_MVA_OUTPUT_TYPE,
}
};
BuiltinProcessing::BuiltinProcessing()
: mRMSPasses(0)
{
}
// pass for 5 consecutive passes
TaskGeneric::ExecutionResult BuiltinProcessing::rms_mva(void** inputs, void** outputs)
{
LOGD("BuiltinProcessing::rms_mva in %x %x %x out %x",
inputs[0], inputs[1], inputs[2], outputs[0]);
android::sp<Buffer>& data(*reinterpret_cast<android::sp<Buffer>*>(inputs[0]));
int64_t passMin = (reinterpret_cast<TaskCase::Value*>(inputs[1]))->getInt64();
int64_t passMax = (reinterpret_cast<TaskCase::Value*>(inputs[2]))->getInt64();
int64_t rms = 0;
size_t samples = data->getSize()/2;
int16_t* rawData = reinterpret_cast<int16_t*>(data->getData());
double energy = 0.0f;
for (size_t i = 0; i < samples; i++) {
energy += (rawData[i] * rawData[i]);
}
rms = (int64_t)sqrt(energy/samples);
TaskGeneric::ExecutionResult result = TaskGeneric::EResultOK;
if (rms < passMin) {
MSG("Volume %lld low compared to min %lld max %lld", rms, passMin, passMax);
mRMSPasses = 0;
} else if (rms <= passMax) {
MSG("Volume %lld OK compared to min %lld max %lld", rms, passMin, passMax);
mRMSPasses++;
if (mRMSPasses >= RMS_CONTINUOUS_PASSES) {
//mRMSPasses = 0;
result = TaskGeneric::EResultPass;
}
} else {
LOGW("Volume %lld high compared to min %lld max %lld", rms, passMin, passMax);
mRMSPasses = 0;
}
TaskCase::Value* rmsVal = reinterpret_cast<TaskCase::Value*>(outputs[0]);
rmsVal->setInt64(rms);
return result;
}