You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
73 lines
2.1 KiB
C++
73 lines
2.1 KiB
C++
#include "loudness.hpp"
|
|
|
|
#include <stdint.h>
|
|
#include <limits>
|
|
#include <algorithm>
|
|
|
|
#include "soundmanagerimp.hpp"
|
|
|
|
namespace MWSound
|
|
{
|
|
|
|
void Sound_Loudness::analyzeLoudness(const std::vector< char >& data)
|
|
{
|
|
mQueue.insert( mQueue.end(), data.begin(), data.end() );
|
|
if (!mQueue.size())
|
|
return;
|
|
|
|
int samplesPerSegment = static_cast<int>(mSampleRate / mSamplesPerSec);
|
|
int numSamples = bytesToFrames(mQueue.size(), mChannelConfig, mSampleType);
|
|
int advance = framesToBytes(1, mChannelConfig, mSampleType);
|
|
|
|
|
|
int segment=0;
|
|
int sample=0;
|
|
while (segment < numSamples/samplesPerSegment)
|
|
{
|
|
float sum=0;
|
|
int samplesAdded = 0;
|
|
while (sample < numSamples && sample < (segment+1)*samplesPerSegment)
|
|
{
|
|
// get sample on a scale from -1 to 1
|
|
float value = 0;
|
|
if (mSampleType == SampleType_UInt8)
|
|
value = ((char)(mQueue[sample*advance]^0x80))/128.f;
|
|
else if (mSampleType == SampleType_Int16)
|
|
{
|
|
value = *reinterpret_cast<const int16_t*>(&mQueue[sample*advance]);
|
|
value /= float(std::numeric_limits<int16_t>::max());
|
|
}
|
|
else if (mSampleType == SampleType_Float32)
|
|
{
|
|
value = *reinterpret_cast<const float*>(&mQueue[sample*advance]);
|
|
value = std::max(-1.f, std::min(1.f, value)); // Float samples *should* be scaled to [-1,1] already.
|
|
}
|
|
|
|
sum += value*value;
|
|
++samplesAdded;
|
|
++sample;
|
|
}
|
|
|
|
float rms = 0; // root mean square
|
|
if (samplesAdded > 0)
|
|
rms = std::sqrt(sum / samplesAdded);
|
|
mSamples.push_back(rms);
|
|
++segment;
|
|
}
|
|
|
|
mQueue.erase(mQueue.begin(), mQueue.begin() + sample*advance);
|
|
}
|
|
|
|
|
|
float Sound_Loudness::getLoudnessAtTime(float sec) const
|
|
{
|
|
if(mSamplesPerSec <= 0.0f || mSamples.empty() || sec < 0.0f)
|
|
return 0.0f;
|
|
|
|
size_t index = static_cast<size_t>(sec * mSamplesPerSec);
|
|
index = std::max<size_t>(0, std::min(index, mSamples.size()-1));
|
|
return mSamples[index];
|
|
}
|
|
|
|
}
|