343 lines
8.3 KiB
C++
343 lines
8.3 KiB
C++
|
#include "main.h"
|
||
|
|
#include "SABuffer.h"
|
||
|
|
#include <math.h>
|
||
|
|
#include "WinampAttributes.h"
|
||
|
|
#include "fft.h"
|
||
|
|
extern int _srate;
|
||
|
|
#ifdef _M_IX86
|
||
|
|
__inline static int lrint(float flt)
|
||
|
|
{
|
||
|
|
int intgr;
|
||
|
|
|
||
|
|
_asm
|
||
|
|
{
|
||
|
|
fld flt
|
||
|
|
fistp intgr
|
||
|
|
}
|
||
|
|
|
||
|
|
return intgr;
|
||
|
|
}
|
||
|
|
#else
|
||
|
|
__inline static int lrint(float flt)
|
||
|
|
{
|
||
|
|
return (int)flt;
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
|
||
|
|
|
||
|
|
// quantizes to 23 bits - use appropriately
|
||
|
|
inline static float fastmin(float x, const float b)
|
||
|
|
{
|
||
|
|
x = b - x;
|
||
|
|
x += (float)fabs(x);
|
||
|
|
x *= 0.5f;
|
||
|
|
x = b - x;
|
||
|
|
return x;
|
||
|
|
}
|
||
|
|
#define FASTMIN(x,b) { x = b - x; x += (float)fabs(x); x *= 0.5f; x = b - x; }
|
||
|
|
inline static float fastclip(float x, const float a, const float b)
|
||
|
|
{
|
||
|
|
float x1 = (float)fabs(x-a);
|
||
|
|
float x2 = (float)fabs(x-b);
|
||
|
|
x = x1 + (a+b);
|
||
|
|
x -= x2;
|
||
|
|
x *= 0.5f;
|
||
|
|
return (x);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
void makeOscData(char *tempdata, char *data_buf, int little_block, int channels, int bits)
|
||
|
|
{
|
||
|
|
float dd = little_block/75.0f;
|
||
|
|
int x,c;
|
||
|
|
int stride=bits/8; // number of bytes between samples
|
||
|
|
|
||
|
|
// we're calculating using only the most significant byte,
|
||
|
|
// because we only end up with 6 bit data anyway
|
||
|
|
// if you want full resolution, check out CVS tag BETA_2005_1122_182830, file: vis.c
|
||
|
|
char *ptr, *sbuf = data_buf;
|
||
|
|
for (x = 0; x < 75; x ++)
|
||
|
|
{
|
||
|
|
float val=0;
|
||
|
|
int index =(int)((float)x * dd); // calculate the nearest sample for this point, interpolation is too expensive for this use
|
||
|
|
ptr=&sbuf[index*stride*channels+stride-1]; // find first sample, and offset for little endian
|
||
|
|
for (c=0;c<channels;c++)
|
||
|
|
{
|
||
|
|
val += (float)*ptr / 8.0f; // we want our final value to be -32 to 32
|
||
|
|
ptr+=stride; // jump to the next sample (channels are interleaved)
|
||
|
|
}
|
||
|
|
tempdata[x] = (char)lrint(val / (float)channels); // average the channels
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
inline double fast_exp2(const double val)
|
||
|
|
{
|
||
|
|
int e;
|
||
|
|
double ret;
|
||
|
|
|
||
|
|
if (val >= 0)
|
||
|
|
{
|
||
|
|
e = int (val);
|
||
|
|
ret = val - (e - 1);
|
||
|
|
((*(1 + (int *) &ret)) &= ~(2047 << 20)) += (e + 1023) << 20;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
e = int (val + 1023);
|
||
|
|
ret = val - (e - 1024);
|
||
|
|
((*(1 + (int *) &ret)) &= ~(2047 << 20)) += e << 20;
|
||
|
|
}
|
||
|
|
return (ret);
|
||
|
|
}
|
||
|
|
|
||
|
|
// ~6 clocks on Pentium M vs. ~24 for single precision sqrtf
|
||
|
|
#if !defined(_WIN64)
|
||
|
|
static inline float squareroot_sse_11bits(float x)
|
||
|
|
{
|
||
|
|
float z;
|
||
|
|
_asm
|
||
|
|
{
|
||
|
|
rsqrtss xmm0, x
|
||
|
|
rcpss xmm0, xmm0
|
||
|
|
movss z, xmm0 // z ~= sqrt(x) to 0.038%
|
||
|
|
}
|
||
|
|
return z;
|
||
|
|
}
|
||
|
|
|
||
|
|
static inline int floor_int(double x)
|
||
|
|
{
|
||
|
|
int i;
|
||
|
|
static const float round_toward_m_i = -0.5f;
|
||
|
|
__asm
|
||
|
|
{
|
||
|
|
fld x
|
||
|
|
fadd st, st(0)
|
||
|
|
fadd round_toward_m_i
|
||
|
|
fistp i
|
||
|
|
sar i, 1
|
||
|
|
}
|
||
|
|
|
||
|
|
return (i);
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
/*
|
||
|
|
static inline float hermite(float x, float y0, float y1, float y2, float y3)
|
||
|
|
{
|
||
|
|
// 4-point, 3rd-order Hermite (x-form)
|
||
|
|
float c0 = y1;
|
||
|
|
float c1 = 0.5f * (y2 - y0);
|
||
|
|
float c2 = y0 - 2.5f * y1 + 2.f * y2 - 0.5f * y3;
|
||
|
|
float c3 = 1.5f * (y1 - y2) + 0.5f * (y3 - y0);
|
||
|
|
|
||
|
|
return ((c3 * x + c2) * x + c1) * x + c0;
|
||
|
|
}
|
||
|
|
*/
|
||
|
|
|
||
|
|
/*
|
||
|
|
static const float c_half = 0.5f;
|
||
|
|
__declspec(naked) static float hermite(float frac_pos, const float* pntr)
|
||
|
|
{
|
||
|
|
__asm
|
||
|
|
{
|
||
|
|
push ecx;
|
||
|
|
mov ecx, dword ptr[esp + 12]; //////////////////////////////////////////////////////////////////////////////////////////////////
|
||
|
|
add ecx, 0x04; // ST(0) ST(1) ST(2) ST(3) ST(4) ST(5) ST(6) ST(7)
|
||
|
|
fld dword ptr [ecx+4]; // x1
|
||
|
|
fsub dword ptr [ecx-4]; // x1-xm1
|
||
|
|
fld dword ptr [ecx]; // x0 x1-xm1
|
||
|
|
fsub dword ptr [ecx+4]; // v x1-xm1
|
||
|
|
fld dword ptr [ecx+8]; // x2 v x1-xm1
|
||
|
|
fsub dword ptr [ecx]; // x2-x0 v x1-xm1
|
||
|
|
fxch st(2); // x1-m1 v x2-x0
|
||
|
|
fmul c_half; // c v x2-x0
|
||
|
|
fxch st(2); // x2-x0 v c
|
||
|
|
fmul c_half; // 0.5*(x2-x0) v c
|
||
|
|
fxch st(2); // c v 0.5*(x2-x0)
|
||
|
|
fst st(3); // c v 0.5*(x2-x0) c
|
||
|
|
fadd st(0), st(1); // w v 0.5*(x2-x0) c
|
||
|
|
fxch st(2); // 0.5*(x2-x0) v w c
|
||
|
|
faddp st(1), st(0); // v+.5(x2-x0) w c
|
||
|
|
fadd st(0), st(1); // a w c
|
||
|
|
fadd st(1), st(0); // a b_neg c
|
||
|
|
fmul dword ptr [esp+8]; // a*frac b_neg c
|
||
|
|
fsubrp st(1), st(0); // a*f-b c
|
||
|
|
fmul dword ptr [esp+8]; // (a*f-b)*f c
|
||
|
|
faddp st(1), st(0); // res-x0/f
|
||
|
|
fmul dword ptr [esp+8]; // res-x0
|
||
|
|
fadd dword ptr [ecx]; // res
|
||
|
|
pop ecx;
|
||
|
|
ret;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
*/
|
||
|
|
inline float hermite(float x, float y0, float y1, float y2, float y3)
|
||
|
|
{
|
||
|
|
// 4-point, 3rd-order Hermite (x-form)
|
||
|
|
float c0 = y1;
|
||
|
|
float c1 = 0.5f * (y2 - y0);
|
||
|
|
float c3 = 1.5f * (y1 - y2) + 0.5f * (y3 - y0);
|
||
|
|
float c2 = y0 - y1 + c1 - c3;
|
||
|
|
|
||
|
|
return ((c3 * x + c2) * x + c1) * x + c0;
|
||
|
|
}
|
||
|
|
|
||
|
|
static inline float fpow2(const float y)
|
||
|
|
{
|
||
|
|
union
|
||
|
|
{
|
||
|
|
float f;
|
||
|
|
int i;
|
||
|
|
} c;
|
||
|
|
|
||
|
|
int integer = lrint(floor(y));
|
||
|
|
/* cut: because we guarantee y>=0
|
||
|
|
if(y < 0)
|
||
|
|
integer = integer-1;
|
||
|
|
*/
|
||
|
|
|
||
|
|
float frac = y - (float)integer;
|
||
|
|
|
||
|
|
c.i = (integer+127) << 23;
|
||
|
|
c.f *= 0.33977f*frac*frac + (1.0f-0.33977f)*frac + 1.0f;
|
||
|
|
|
||
|
|
return c.f;
|
||
|
|
}
|
||
|
|
|
||
|
|
//#define SAPOW(x) (powf(2.f, (float)(x)/12.f))
|
||
|
|
#define SAPOW(x) (fpow2((float)(x)/12.f))
|
||
|
|
//#define WARP(x) ((powf(1.1f, (float)(x)/12.f) - 1.) * bla)
|
||
|
|
#define WARP(x) ((SAPOW(x) - 1.f) * bla)
|
||
|
|
void makeSpecData(unsigned char *tempdata, float *wavetrum)
|
||
|
|
{
|
||
|
|
//WARP(75);
|
||
|
|
float bla = (255.f/SAPOW(75.f));
|
||
|
|
fft_9(wavetrum);
|
||
|
|
|
||
|
|
float spec_scale=0.5;
|
||
|
|
if (config_replaygain)
|
||
|
|
{ // benski> i'm sure there's some math identity we can use to optimize this.
|
||
|
|
spec_scale/=pow(10.0f, config_replaygain_non_rg_gain.GetFloat() / 20.0f);
|
||
|
|
}
|
||
|
|
|
||
|
|
for (int i=0;i<256;i++)
|
||
|
|
{
|
||
|
|
//int lookup=2*i;
|
||
|
|
float sinT = wavetrum[2*i];
|
||
|
|
float cosT = wavetrum[2*i+1];
|
||
|
|
wavetrum[i] = sqrt(sinT*sinT+cosT*cosT)*spec_scale;
|
||
|
|
}
|
||
|
|
|
||
|
|
float next = WARP(0)+1 ;
|
||
|
|
for (int x = 0; x < 75; x ++)
|
||
|
|
{
|
||
|
|
//float prev = 1.+(pow(2.,(float)x/12.) -1.) * bla;
|
||
|
|
float binF = next;
|
||
|
|
next = WARP(x+1) +1;
|
||
|
|
|
||
|
|
float thisValue = 0;
|
||
|
|
int bin = lrint(floor(binF));
|
||
|
|
int end = lrint(floor(next));
|
||
|
|
end = min(end, 255);
|
||
|
|
float mult = ((float)(bin+1))-binF;
|
||
|
|
bool herm=true;
|
||
|
|
do
|
||
|
|
{
|
||
|
|
if (bin == end)
|
||
|
|
{
|
||
|
|
mult = (next-binF);
|
||
|
|
herm=true;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (herm)
|
||
|
|
{
|
||
|
|
float C=0, D=0;
|
||
|
|
if (bin<255)
|
||
|
|
{
|
||
|
|
C=wavetrum[bin+1];
|
||
|
|
if (bin<254)
|
||
|
|
D=wavetrum[bin+2];
|
||
|
|
}
|
||
|
|
|
||
|
|
//float samples[4] = { wavetrum[lookupA], wavetrum[lookupB], wavetrum[lookupC], wavetrum[lookupD] };
|
||
|
|
//thisValue += hermite(binF-bin, samples) * mult;
|
||
|
|
thisValue += hermite(binF-bin, wavetrum[bin-1], wavetrum[bin], C, D) * mult;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
thisValue += wavetrum[bin];
|
||
|
|
}
|
||
|
|
|
||
|
|
herm=false;
|
||
|
|
bin++;
|
||
|
|
binF=(float)bin;
|
||
|
|
}
|
||
|
|
while (bin <= end);
|
||
|
|
|
||
|
|
tempdata[x]=lrint(fastmin(thisValue, 255.f));
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
////////////////////////////////
|
||
|
|
|
||
|
|
SABuffer saBuffer;
|
||
|
|
|
||
|
|
void sa_addpcmdata(void *_data_buf, int numChannels, int numBits, int ts)
|
||
|
|
{
|
||
|
|
char *data_buf = reinterpret_cast<char *>(_data_buf);
|
||
|
|
char tempdata[75*2] = {0};
|
||
|
|
__declspec(align(16)) float wavetrum[512];
|
||
|
|
//extern int sa_curmode;
|
||
|
|
int vis_Csa=sa_override ? 3 : sa_curmode;
|
||
|
|
|
||
|
|
switch (vis_Csa)
|
||
|
|
{
|
||
|
|
case 4:
|
||
|
|
tempdata[0] = 0;
|
||
|
|
tempdata[1] = 0;
|
||
|
|
sa_add(tempdata,ts,4);
|
||
|
|
return;
|
||
|
|
case 2:
|
||
|
|
makeOscData(tempdata,data_buf,576,numChannels, numBits);
|
||
|
|
sa_add(tempdata,ts,2);
|
||
|
|
return ;
|
||
|
|
case 3:
|
||
|
|
makeOscData(tempdata+75,data_buf,576,numChannels, numBits);
|
||
|
|
// fall through!
|
||
|
|
case 1:
|
||
|
|
calcVuData((unsigned char*)tempdata, data_buf, numChannels, numBits);
|
||
|
|
vu_add(tempdata, ts);
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
bool done=false;
|
||
|
|
size_t samples=576;
|
||
|
|
while (samples)
|
||
|
|
{
|
||
|
|
unsigned int copied = saBuffer.AddToBuffer(data_buf, numChannels, numBits, ts, (unsigned int) samples);
|
||
|
|
samples-=copied;
|
||
|
|
data_buf+=(copied*(numBits/8)*numChannels);
|
||
|
|
if (saBuffer.Full())
|
||
|
|
{
|
||
|
|
saBuffer.WindowToFFTBuffer(wavetrum);
|
||
|
|
if (!done)
|
||
|
|
{
|
||
|
|
if (vis_Csa == 3)
|
||
|
|
{
|
||
|
|
makeSpecData((unsigned char*)tempdata, wavetrum);
|
||
|
|
sa_add(tempdata, ts, 0x80000003);
|
||
|
|
}
|
||
|
|
else if (vis_Csa == 1)
|
||
|
|
{
|
||
|
|
makeSpecData((unsigned char*)tempdata, wavetrum);
|
||
|
|
sa_add(tempdata, ts, 1);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
//done=true;
|
||
|
|
saBuffer.CopyHalf();
|
||
|
|
ts+=MulDiv(SABUFFER_WINDOW_INCREMENT,1000,_srate);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|