blob: be8771dee02de0f725aa072149bc70e198bf725b [file] [log] [blame]
* Copyright (C) 2018 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
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* See the License for the specific language governing permissions and
* limitations under the License.
#include <Eigen/Dense>
#include <unsupported/Eigen/FFT>
#include "RDsp.h"
#include "SHCircularBuffer.h"
#include "DPBase.h"
namespace dp_fx {
using FXBuffer = SHCircularBuffer<float>;
class ChannelBuffer {
FXBuffer cBInput; // Circular Buffer input
FXBuffer cBOutput; // Circular Buffer output
FloatVec input; // time domain temp vector for input
FloatVec output; // time domain temp vector for output
FloatVec outTail; // time domain temp vector for output tail (for overlap-add method)
Eigen::VectorXcf complexTemp; // complex temp vector for frequency domain operations
//Current parameters
float inputGainDb;
float outputGainDb;
struct BandParams {
bool enabled;
float freqCutoffHz;
size_t binStart;
size_t binStop;
struct EqBandParams : public BandParams {
float gainDb;
struct MbcBandParams : public BandParams {
float gainPreDb;
float gainPostDb;
float attackTimeMs;
float releaseTimeMs;
float ratio;
float thresholdDb;
float kneeWidthDb;
float noiseGateThresholdDb;
float expanderRatio;
//Historic values
float previousEnvelope;
struct LimiterParams {
int32_t linkGroup;
float attackTimeMs;
float releaseTimeMs;
float ratio;
float thresholdDb;
float postGainDb;
//Historic values
float previousEnvelope;
float newFactor;
float linkFactor;
bool mPreEqInUse;
bool mPreEqEnabled;
std::vector<EqBandParams> mPreEqBands;
bool mMbcInUse;
bool mMbcEnabled;
std::vector<MbcBandParams> mMbcBands;
bool mPostEqInUse;
bool mPostEqEnabled;
std::vector<EqBandParams> mPostEqBands;
bool mLimiterInUse;
bool mLimiterEnabled;
LimiterParams mLimiterParams;
FloatVec mPreEqFactorVector; // temp pre-computed vector to shape spectrum at preEQ stage
FloatVec mPostEqFactorVector; // temp pre-computed vector to shape spectrum at postEQ stage
void initBuffers(unsigned int blockSize, unsigned int overlapSize, unsigned int halfFftSize,
unsigned int samplingRate, DPBase &dpBase);
void computeBinStartStop(BandParams &bp, size_t binStart);
unsigned int mSamplingRate;
unsigned int mBlockSize;
using CBufferVector = std::vector<ChannelBuffer>;
using GroupsMap = std::map<int32_t, IntVec>;
class LinkedLimiters {
void reset();
void update(int32_t group, int index);
void remove(int index);
GroupsMap mGroupsMap;
class DPFrequency : public DPBase {
virtual size_t processSamples(const float *in, float *out, size_t samples);
virtual void reset();
void configure(size_t blockSize, size_t overlapSize, size_t samplingRate);
static size_t getMinBockSize();
static size_t getMaxBockSize();
void updateParameters(ChannelBuffer &cb, int channelIndex);
size_t processMono(ChannelBuffer &cb);
size_t processOneVector(FloatVec &output, FloatVec &input, ChannelBuffer &cb);
size_t processChannelBuffers(CBufferVector &channelBuffers);
size_t processFirstStages(ChannelBuffer &cb);
size_t processLastStages(ChannelBuffer &cb);
void processLinkedLimiters(CBufferVector &channelBuffers);
size_t mBlockSize;
size_t mHalfFFTSize;
size_t mOverlapSize;
size_t mSamplingRate;
float mBlocksPerSecond;
CBufferVector mChannelBuffers;
LinkedLimiters mLinkedLimiters;
FloatVec mVWindow; //window class.
float mWindowRms;
Eigen::FFT<float> mFftServer;
} //namespace dp_fx
#endif // DPFREQUENCY_H_