You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
111 lines
3.5 KiB
111 lines
3.5 KiB
/*
|
|
* Copyright 2021 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.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <array>
|
|
#include <audio_effects/effect_aec.h>
|
|
#include <audio_utils/channels.h>
|
|
#include <audio_utils/primitives.h>
|
|
#include <climits>
|
|
#include <cstdlib>
|
|
#include <gtest/gtest.h>
|
|
#include <hardware/audio_effect.h>
|
|
#include <log/log.h>
|
|
#include <random>
|
|
#include <stdint.h>
|
|
#include <system/audio.h>
|
|
#include <vector>
|
|
|
|
template <typename T>
|
|
static float computeSnr(const T* ref, const T* tst, size_t count) {
|
|
double signal{};
|
|
double noise{};
|
|
|
|
for (size_t i = 0; i < count; ++i) {
|
|
const double value(ref[i]);
|
|
const double diff(tst[i] - value);
|
|
signal += value * value;
|
|
noise += diff * diff;
|
|
}
|
|
// Initialized to large value to handle
|
|
// cases where ref and tst match exactly
|
|
float snr = FLT_MAX;
|
|
if (signal > 0.0f && noise > 0.0f) {
|
|
snr = 10.f * log(signal / noise);
|
|
}
|
|
return snr;
|
|
}
|
|
|
|
class EffectTestHelper {
|
|
public:
|
|
EffectTestHelper(const effect_uuid_t* uuid, size_t chMask, size_t sampleRate, size_t loopCount)
|
|
: mUuid(uuid),
|
|
mChMask(chMask),
|
|
mChannelCount(audio_channel_count_from_in_mask(mChMask)),
|
|
mSampleRate(sampleRate),
|
|
mFrameCount(mSampleRate * kTenMilliSecVal),
|
|
mLoopCount(loopCount) {}
|
|
void createEffect();
|
|
void releaseEffect();
|
|
void setConfig(bool configReverse);
|
|
void setParam(uint32_t type, uint32_t val);
|
|
void process(int16_t* input, int16_t* output, bool setAecEchoDelay);
|
|
void process_reverse(int16_t* farInput, int16_t* output);
|
|
|
|
// Corresponds to SNR for 1 bit difference between two int16_t signals
|
|
static constexpr float kSNRThreshold = 90.308998;
|
|
|
|
static constexpr audio_channel_mask_t kChMasks[] = {
|
|
AUDIO_CHANNEL_IN_MONO,
|
|
AUDIO_CHANNEL_IN_STEREO,
|
|
AUDIO_CHANNEL_IN_FRONT_BACK,
|
|
AUDIO_CHANNEL_IN_6,
|
|
AUDIO_CHANNEL_IN_2POINT0POINT2,
|
|
AUDIO_CHANNEL_IN_2POINT1POINT2,
|
|
AUDIO_CHANNEL_IN_3POINT0POINT2,
|
|
AUDIO_CHANNEL_IN_3POINT1POINT2,
|
|
AUDIO_CHANNEL_IN_5POINT1,
|
|
AUDIO_CHANNEL_IN_VOICE_UPLINK_MONO,
|
|
AUDIO_CHANNEL_IN_VOICE_DNLINK_MONO,
|
|
AUDIO_CHANNEL_IN_VOICE_CALL_MONO,
|
|
};
|
|
|
|
static constexpr float kTenMilliSecVal = 0.01;
|
|
|
|
static constexpr size_t kNumChMasks = std::size(kChMasks);
|
|
|
|
static constexpr size_t kSampleRates[] = {8000, 11025, 12000, 16000, 22050,
|
|
24000, 32000, 44100, 48000};
|
|
|
|
static constexpr size_t kNumSampleRates = std::size(kSampleRates);
|
|
|
|
static constexpr size_t kLoopCounts[] = {1, 4};
|
|
|
|
static constexpr size_t kNumLoopCounts = std::size(kLoopCounts);
|
|
|
|
static constexpr size_t kAECDelay = 0;
|
|
|
|
private:
|
|
const effect_uuid_t* mUuid;
|
|
const size_t mChMask;
|
|
const size_t mChannelCount;
|
|
const size_t mSampleRate;
|
|
const size_t mFrameCount;
|
|
const size_t mLoopCount;
|
|
effect_handle_t mEffectHandle{};
|
|
};
|