106 lines
3.0 KiB
C++
106 lines
3.0 KiB
C++
#include <cmath>
|
|
#include <VulcanoLE/Audio/AudioGrabber.h>
|
|
#include <VUtils/Logging.h>
|
|
#include <VulcanoLE/Audio/JackClient.h>
|
|
#include "VUtils/Math.h"
|
|
|
|
|
|
AudioGrabber::AudioGrabber() = default;
|
|
AudioGrabber::~AudioGrabber() = default;
|
|
|
|
bool AudioGrabber::read(stereoSample *buffer, uint32_t bufferSize) {
|
|
auto &client = JackClient::get();
|
|
if (!client.isData())
|
|
return false;
|
|
availableData = client.getFrames();
|
|
client.fillSamples(buffer, availableData);
|
|
return true;
|
|
}
|
|
|
|
AudioGrabber *AudioGrabber::createAudioGrabber() {
|
|
JackClient::get().start();
|
|
auto *grabber = new AudioGrabber();
|
|
JackClient::get().grabber = grabber;
|
|
return grabber;
|
|
}
|
|
|
|
void AudioGrabber::init() {
|
|
m_buffer = static_cast<stereoSample *>(calloc(BUFFER_SIZE, sizeof(stereoSample)));
|
|
if (env != nullptr)
|
|
m_scale = env->getAsDouble("audio_scale", 1.0);
|
|
m_filter.scale = 1.0;
|
|
DBG("SET Audio Scale: %.3f", m_scale)
|
|
loudness = { 0.0, 0.0 };
|
|
}
|
|
|
|
void AudioGrabber::calculateRMS(stereoSample *pFrame) {
|
|
double squareL = 0, meanL;
|
|
double squareR = 0, meanR;
|
|
for (int i = 0; i < availableData; i++) {
|
|
squareL += std::pow(pFrame[i].l * m_scale, 2);
|
|
squareR += std::pow(pFrame[i].r * m_scale, 2);
|
|
}
|
|
meanL = (squareL / (float) (availableData));
|
|
meanR = (squareR / (float) (availableData));
|
|
loudness = { (float) std::sqrt(meanL), (float) std::sqrt(meanR) };
|
|
}
|
|
|
|
void AudioGrabber::calculatePEAK(stereoSample *pFrame) {
|
|
stereoSampleFrame max = { 0, 0 };
|
|
for (int i = 0; i < availableData; i++) {
|
|
auto left = (float) std::abs(pFrame[i].l * m_scale);
|
|
auto right = (float) std::abs(pFrame[i].r * m_scale);
|
|
if (left > max.l)
|
|
max.l = left;
|
|
if (right > max.r)
|
|
max.r = right;
|
|
}
|
|
loudness = max;
|
|
}
|
|
|
|
stereoSampleFrame AudioGrabber::getLoudness() {
|
|
std::unique_lock<std::mutex> lck(m_mtx);
|
|
return loudness;
|
|
}
|
|
|
|
bool AudioGrabber::work() {
|
|
std::unique_lock<std::mutex> lck(m_mtx);
|
|
if (this->read(m_buffer, BUFFER_SIZE)) {
|
|
switch (requestMode) {
|
|
case ReqMode::FFT:
|
|
fft.process(m_buffer, m_scale);
|
|
break;
|
|
case ReqMode::RMS:
|
|
calculateRMS(m_buffer);
|
|
break;
|
|
case ReqMode::PEAK:
|
|
calculatePEAK(m_buffer);
|
|
break;
|
|
case ReqMode::FILTER:
|
|
loudness = m_filter.work(m_buffer, availableData);
|
|
break;
|
|
default: {
|
|
calculateRMS(m_buffer);
|
|
calculatePEAK(m_buffer);
|
|
fft.process(m_buffer, m_scale);
|
|
}
|
|
}
|
|
return true;
|
|
} else {
|
|
return false;
|
|
}
|
|
}
|
|
Audio::FilterHelper &AudioGrabber::getFilter() {
|
|
return m_filter;
|
|
}
|
|
double AudioGrabber::getBass() {
|
|
auto fftData = fft.getData();
|
|
auto val = 0.0;
|
|
for (int i = 1; i < 4; ++i) {
|
|
auto avg = (fftData->leftChannel[i] + fftData->rightChannel[i]) / 2.0;
|
|
if (avg > val)
|
|
val = avg;
|
|
}
|
|
return VUtils::Math::clamp(val, 1, 255);
|
|
}
|