263 lines
8.4 KiB
C++
263 lines
8.4 KiB
C++
#include "cwCommon.h"
|
|
#include "cwLog.h"
|
|
#include "cwCommonImpl.h"
|
|
#include "cwMem.h"
|
|
#include "cwFile.h"
|
|
#include "cwText.h"
|
|
#include "cwObject.h"
|
|
#include "cwAudioFile.h"
|
|
#include "cwUtility.h"
|
|
#include "cwFileSys.h"
|
|
#include "cwAudioFileOps.h"
|
|
#include "cwVectOps.h"
|
|
#include "cwMath.h"
|
|
#include "cwDsp.h"
|
|
#include "cwAudioTransforms.h"
|
|
|
|
namespace cw
|
|
{
|
|
namespace dsp
|
|
{
|
|
|
|
namespace wnd_func
|
|
{
|
|
|
|
idLabelPair_t wndLabelArray[] =
|
|
{
|
|
{ kHannWndId, "hann" },
|
|
{ kHammingWndId, "hamming" },
|
|
{ kTriangleWndId, "triangle" },
|
|
{ kKaiserWndId, "kaiser" },
|
|
{ kHannMatlabWndId, "hann_matlab" },
|
|
{ kUnityWndId, "unity" },
|
|
{ kInvalidWndId, "<invalid>" }
|
|
};
|
|
|
|
const char* wndIdToLabel( unsigned id )
|
|
{ return cw::idToLabel( wndLabelArray, id, kInvalidWndId ); }
|
|
|
|
unsigned wndLabelToId( const char* label )
|
|
{ return cw::labelToId( wndLabelArray, label, kInvalidWndId ); }
|
|
|
|
rc_t _test( const char* windowLabel, const double* wndV, unsigned wndN )
|
|
{
|
|
rc_t rc = kOkRC;
|
|
wnd_func::fobj_t* p = nullptr;
|
|
|
|
unsigned wndId = wndLabelToId( windowLabel );
|
|
|
|
if((rc = create(p,wndId,wndN,3)) == kOkRC )
|
|
{
|
|
vop::print(p->wndV, p->wndN, "%f ", windowLabel);
|
|
|
|
printf("diff: %f\n", vop::sum_sq_diff( wndV, p->wndV, wndN));
|
|
|
|
destroy(p);
|
|
}
|
|
|
|
return rc;
|
|
}
|
|
|
|
rc_t test( const cw::object_t* args )
|
|
{
|
|
double hann_15[] = { 0.0, 0.04951557, 0.1882551 , 0.38873953, 0.61126047, 0.8117449, 0.95048443, 1.0, 0.95048443, 0.8117449, 0.61126047, 0.38873953, 0.1882551, 0.04951557, 0.0 };
|
|
double hann_16[] = { 0.0, 0.04322727, 0.1654347 , 0.3454915 , 0.55226423, 0.75, 0.9045085 , 0.9890738, 0.9890738, 0.9045085, 0.75, 0.55226423, 0.3454915, 0.1654347 , 0.04322727, 0.0 };
|
|
double hamm_15[] = { 0.08,0.12555432, 0.25319469, 0.43764037, 0.64235963, 0.82680531, 0.95444568, 1.0, 0.95444568, 0.82680531,0.64235963, 0.43764037, 0.25319469, 0.12555432, 0.08 };
|
|
double hamm_16[] = { 0.08,0.11976909, 0.23219992, 0.39785218, 0.58808309, 0.77, 0.91214782, 0.9899479, 0.9899479 , 0.91214782,0.77, 0.58808309, 0.39785218, 0.23219992, 0.11976909, 0.08};
|
|
double tri_15[] = { 0.0, 0.14285714, 0.28571429, 0.42857143, 0.57142857, 0.71428571, 0.85714286, 1.0, 0.85714286, 0.71428571,0.57142857, 0.42857143, 0.28571429, 0.14285714, 0.0 };
|
|
double tri_16[] = { 0.0, 0.13333333, 0.26666667, 0.4, 0.53333333, 0.66666667, 0.8, 0.93333333, 0.93333333, 0.8, 0.66666667, 0.53333333, 0.4, 0.26666667, 0.13333333, 0.0 };
|
|
double ones_5[] = { 1.0, 1.0, 1.0, 1.0, 1.0 };
|
|
|
|
rc_t rc = kOkRC;
|
|
|
|
_test( "hann", hann_15, 15);
|
|
_test( "hann", hann_16, 16);
|
|
_test( "hamming", hamm_15, 15);
|
|
_test( "hamming", hamm_16, 16);
|
|
_test( "triangle", tri_15, 15);
|
|
_test( "triangle", tri_16, 16);
|
|
_test( "unity", ones_5, 5);
|
|
|
|
if( rc != kOkRC )
|
|
cwLogError(rc,"Window test failed.");
|
|
return rc;
|
|
}
|
|
}
|
|
|
|
namespace ola
|
|
{
|
|
rc_t test( const cw::object_t* args )
|
|
{
|
|
typedef float sample_t;
|
|
|
|
rc_t rc = kOkRC;
|
|
unsigned wndSmpCnt = 16;
|
|
unsigned hopSmpCnt = 4;
|
|
unsigned procSmpCnt = 2;
|
|
unsigned wndTypeId = wnd_func::kUnityWndId;
|
|
unsigned hopCnt = 8;
|
|
unsigned oSmpCnt = hopCnt * hopSmpCnt;
|
|
unsigned i,j,k;
|
|
|
|
sample_t cV[] = { 1,1,1,1, 2,2,2,2, 3,3,3,3, 4,4,4,4, 4,4,4,4, 4,4,4,4, 4,4,4,4, 4,4,4,4 };
|
|
sample_t* x = mem::allocZ<sample_t>(wndSmpCnt);
|
|
sample_t* y = mem::allocZ<sample_t>(oSmpCnt);
|
|
vop::fill(x,wndSmpCnt,1.0);
|
|
|
|
ola::fobj_t* p = nullptr;
|
|
|
|
if((rc = ola::create(p, wndSmpCnt, hopSmpCnt, procSmpCnt, wndTypeId )) == kOkRC )
|
|
{
|
|
|
|
// Each iteration represents a single audio cylce
|
|
// which sources/sinks procSmpCnt samples
|
|
for(i=0,j=0,k=0; k<oSmpCnt; i+=procSmpCnt)
|
|
{
|
|
j += procSmpCnt;
|
|
|
|
// if there are hopSmpCnt new samples available then there must
|
|
// be a new window of samples available -
|
|
if( j > hopSmpCnt )
|
|
{
|
|
j -= hopSmpCnt;
|
|
ola::exec(p,x,p->wndSmpCnt);
|
|
}
|
|
|
|
const sample_t* op;
|
|
|
|
// Get procSmpCnt samples from the output.
|
|
if( (op=ola::execOut(p)) != NULL )
|
|
{
|
|
assert( y + k + p->procSmpCnt <= y + oSmpCnt );
|
|
vop::copy<sample_t>(y+k, op, p->procSmpCnt);
|
|
k += p->procSmpCnt;
|
|
}
|
|
}
|
|
}
|
|
|
|
vop::print(cV,oSmpCnt,"%f ","Correct ");
|
|
vop::print(y, oSmpCnt,"%f ","Computed");
|
|
printf("diff:%f\n", vop::sum_sq_diff(cV,y,oSmpCnt));
|
|
|
|
ola::destroy(p);
|
|
mem::release(x);
|
|
mem::release(y);
|
|
return rc;
|
|
|
|
}
|
|
}
|
|
|
|
namespace shift_buf
|
|
{
|
|
rc_t test( const object_t* args )
|
|
{
|
|
rc_t rc = kOkRC;
|
|
typedef float sample_t;
|
|
|
|
sample_t m[] =
|
|
{
|
|
1, 2, 3, 4, 5, 6, 7,
|
|
7, 8, 9,10,11,12,13,
|
|
13,14,15,16,17,18,19,
|
|
19,20,21,22,23,24,25,
|
|
25,26,27,28,29,30,31,
|
|
31,32,33,34,35,36,37,
|
|
37,38,39,40,41,42,43,
|
|
43,44,45,46,47,48,49
|
|
};
|
|
|
|
unsigned procSmpCnt = 5; // count of samples to be fed to the shift buffer on each cycle
|
|
unsigned hopSmpCnt = 6; // count of samples between shift buffer outputs
|
|
unsigned wndSmpCnt = 7; // count of samples in each shift buffer output
|
|
|
|
unsigned iSmpCnt = 49; // count of samples in the input test signal
|
|
unsigned oColCnt = iSmpCnt / hopSmpCnt;
|
|
unsigned oSmpCnt = oColCnt * wndSmpCnt; // count of samples in the output test signal
|
|
unsigned i,j;
|
|
|
|
shift_buf::fobj_t* p = nullptr;
|
|
|
|
if((rc = shift_buf::create(p,procSmpCnt,wndSmpCnt,hopSmpCnt)) == kOkRC )
|
|
{
|
|
sample_t* x = mem::allocZ<sample_t>(iSmpCnt);
|
|
sample_t* y = mem::allocZ<sample_t>(oSmpCnt);
|
|
|
|
vop::seq(x,iSmpCnt,1.0f,1.0f);
|
|
|
|
for(i=0,j=0; i<iSmpCnt; i+=procSmpCnt)
|
|
{
|
|
// Give the shift buffer a block of procSmpCnt samples.
|
|
// If it returns 'true' then it has hopSmpCnt new samples and
|
|
// a buffer of at least wndSmpCnt.
|
|
for(; shift_buf::exec( p, x + i, procSmpCnt ); j+=wndSmpCnt )
|
|
{
|
|
|
|
// cmShiftBufExec() returned true then there are wndSmpCnt samples available.
|
|
assert( y + j + wndSmpCnt <= y + oSmpCnt );
|
|
vop::copy(y + j, p->outV, wndSmpCnt );
|
|
|
|
vop::print(p->outV, wndSmpCnt, "%f ");
|
|
vop::print(m + j, wndSmpCnt, "%f ");
|
|
|
|
if( !vop::is_equal(p->outV, m+j, wndSmpCnt ))
|
|
{
|
|
rc = cwLogError(kTestFailRC,"shift_buf test failed.");
|
|
goto errLabel;
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
mem::release(x);
|
|
mem::release(y);
|
|
shift_buf::destroy(p);
|
|
|
|
}
|
|
|
|
errLabel:
|
|
return rc;
|
|
}
|
|
}
|
|
|
|
namespace pv_anl
|
|
{
|
|
rc_t test( const object_t* args )
|
|
{
|
|
rc_t rc = kOkRC;
|
|
pv_anl::fobj_t* pva = nullptr;
|
|
pv_syn::fobj_t* pvs = nullptr;
|
|
unsigned procSmpCnt = 0;
|
|
float srate = 0;
|
|
float out_srate = 0;
|
|
unsigned wndSmpCnt = 0;
|
|
unsigned hopSmpCnt = 0;
|
|
unsigned flags = kCalcHzPvaFl;
|
|
unsigned wndTypeId = wnd_func::kHannWndId;
|
|
|
|
if((rc = create( pva, procSmpCnt, srate, wndSmpCnt, hopSmpCnt, flags )) != kOkRC )
|
|
{
|
|
}
|
|
|
|
if((rc = create( pvs, procSmpCnt, out_srate, wndSmpCnt, hopSmpCnt, wndTypeId )) != kOkRC )
|
|
{
|
|
}
|
|
|
|
|
|
destroy(pva);
|
|
destroy(pvs);
|
|
|
|
return rc;
|
|
}
|
|
|
|
}
|
|
|
|
rc_t test( const cw::object_t* args )
|
|
{
|
|
wnd_func::test(args);
|
|
ola::test(args);
|
|
shift_buf::test(args);
|
|
return kOkRC;
|
|
}
|
|
}
|
|
}
|