123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861 |
- #include "cmPrefix.h"
- #include "cmGlobal.h"
- #include "cmRpt.h"
- #include "cmErr.h"
- #include "cmCtx.h"
- #include "cmMem.h"
- #include "cmMallocDebug.h"
- #include "cmLinkedHeap.h"
- #include "cmFloatTypes.h"
- #include "cmComplexTypes.h"
- #include "cmFileSys.h"
- #include "cmJson.h"
- #include "cmSymTbl.h"
- #include "cmAudioFile.h"
- #include "cmText.h"
- #include "cmProcObj.h"
- #include "cmProcTemplate.h"
- #include "cmMath.h"
- #include "cmFile.h"
- #include "cmTime.h"
- #include "cmMidi.h"
- #include "cmProc.h"
- #include "cmProc2.h"
- #include "cmProc5.h"
-
- #include "cmVectOps.h"
-
-
- //=======================================================================================================================
- cmGoertzel* cmGoertzelAlloc( cmCtx* c, cmGoertzel* p, double srate, const double* fcHzV, unsigned chCnt, unsigned procSmpCnt, unsigned hopSmpCnt, unsigned wndSmpCnt )
- {
- cmGoertzel* op = cmObjAlloc(cmGoertzel,c,p);
-
- op->shb = cmShiftBufAlloc(c,NULL,0,0,0);
-
- if( srate > 0 )
- if( cmGoertzelInit(op,srate,fcHzV,chCnt,procSmpCnt,wndSmpCnt,hopSmpCnt) != cmOkRC )
- cmGoertzelFree(&op);
-
- return op;
- }
-
- cmRC_t cmGoertzelFree( cmGoertzel** pp )
- {
- cmRC_t rc = cmOkRC;
- if( pp==NULL || *pp==NULL )
- return rc;
-
- cmGoertzel* p = *pp;
- if((rc = cmGoertzelFinal(p)) != cmOkRC )
- return rc;
-
- cmShiftBufFree(&p->shb);
- cmMemFree(p->ch);
- cmMemFree(p->wnd);
- cmObjFree(pp);
- return rc;
-
- }
-
- cmRC_t cmGoertzelInit( cmGoertzel* p, double srate, const double* fcHzV, unsigned chCnt, unsigned procSmpCnt, unsigned hopSmpCnt, unsigned wndSmpCnt )
- {
- cmRC_t rc;
- unsigned i;
-
- if((rc = cmGoertzelFinal(p)) != cmOkRC )
- return rc;
-
- p->ch = cmMemResizeZ(cmGoertzelCh,p->ch,chCnt);
- p->chCnt = chCnt;
- p->srate = srate;
- p->wnd = cmMemResizeZ(cmSample_t,p->wnd,wndSmpCnt);
-
- cmVOS_Hann(p->wnd,wndSmpCnt);
-
- cmShiftBufInit(p->shb,procSmpCnt,wndSmpCnt,hopSmpCnt);
-
- for(i=0; i<p->chCnt; ++i)
- {
- cmGoertzelSetFcHz(p,i,fcHzV[i]);
- }
-
- return rc;
- }
-
- cmRC_t cmGoertzelFinal( cmGoertzel* p )
- { return cmOkRC; }
-
- cmRC_t cmGoertzelSetFcHz( cmGoertzel* p, unsigned chIdx, double hz )
- {
- assert( chIdx < p->chCnt );
- p->ch[chIdx].hz = hz;
- p->ch[chIdx].coeff = 2*cos(2*M_PI*hz/p->srate);
-
- return cmOkRC;
- }
-
- cmRC_t cmGoertzelExec( cmGoertzel* p, const cmSample_t* inpV, unsigned procSmpCnt, double* outV, unsigned chCnt )
- {
- unsigned i,j;
-
- while( cmShiftBufExec(p->shb,inpV,procSmpCnt) )
- {
- unsigned xn = p->shb->wndSmpCnt;
- cmSample_t x[ xn ];
-
- cmVOS_MultVVV(x,xn,p->wnd,p->shb->outV);
-
- for(i=0; i<chCnt; ++i)
- {
- cmGoertzelCh* ch = p->ch + i;
-
- ch->s2 = x[0];
- ch->s1 = x[1] + 2 * x[0] * ch->coeff;
- for(j=2; j<xn; ++j)
- {
- ch->s0 = x[j] + ch->coeff * ch->s1 - ch->s2;
- ch->s2 = ch->s1;
- ch->s1 = ch->s0;
- }
-
- outV[i] = ch->s2*ch->s2 + ch->s1*ch->s1 - ch->coeff * ch->s2 * ch->s1;
- }
- }
-
- return cmOkRC;
- }
-
-
- //=======================================================================================================================
- double _cmGoldSigSinc( double t, double T )
- {
- double x = t/T;
- return x == 0 ? 1.0 : sin(M_PI*x)/(M_PI*x);
- }
-
- void _cmGoldSigRaisedCos( cmSample_t* yV, int yN, double sPc, double beta )
- {
- int i;
-
- for(i=0; i<yN; ++i)
- {
- double t = i - yN/2;
- double den = 1 - (4*(beta*beta)*(t*t) / (sPc*sPc));
- double a;
-
- if(fabs(den) < 0.00001 )
- a = 1;
- else
- a = cos(M_PI * beta * t/ sPc ) / den;
-
- yV[i] = _cmGoldSigSinc(t,sPc) * a;
- }
- }
-
- void _cmGoldSigConv( cmGoldSig_t* p, unsigned chIdx )
- {
- int i;
- int sPc = p->a.samplesPerChip;
- int osf = p->a.rcosOSFact;
-
- // for each bit in the spreading-code
- for(i=0; i<p->mlsN; ++i)
- {
- int j = (i*sPc) + sPc/2; // index into bbV[] of center of impulse response
- int k = j - (sPc*osf)/2; // index into bbV[] of start of impulse response
- int h;
-
- // for each sample in the impulse response
- for(h=0; h<p->rcosN; ++h,++k)
- {
-
- while( k<0 )
- k += p->sigN;
-
- while( k>=p->sigN )
- k -= p->sigN;
-
- p->ch[chIdx].bbV[k] += p->ch[chIdx].pnV[i] * p->rcosV[h];
- }
- }
- }
-
- void _cmGoldSigModulate( cmGoldSig_t* p, unsigned chIdx )
- {
- unsigned i;
- double rps = 2.0 * M_PI * p->a.carrierHz / p->a.srate;
- cmSample_t* yV = p->ch[chIdx].mdV;
- cmSample_t* bbV = p->ch[chIdx].bbV;
-
- for(i=0; i<p->sigN; ++i)
- yV[ i ] = bbV[i]*cos(rps*i) + bbV[i]*sin(rps*i);
-
- // apply a half Hann envelope to the onset/offset of the id signal
- if( p->a.envMs > 0 )
- {
- unsigned wndMs = p->a.envMs * 2;
- unsigned wndN = wndMs * p->a.srate / 1000;
- wndN += wndN % 2 ? 0 : 1; // force the window length to be odd
- unsigned wNo2 = wndN/2 + 1;
- cmSample_t wndV[ wndN ];
- cmVOS_Hann(wndV,wndN);
- cmVOS_MultVV(yV,wNo2,wndV);
- cmVOS_MultVV(yV + p->sigN - wNo2, wNo2, wndV + wNo2 - 1);
- }
-
- }
-
- cmGoldSig_t* cmGoldSigAlloc( cmCtx* ctx, cmGoldSig_t* p, const cmGoldSigArg_t* a )
- {
- cmGoldSig_t* op = cmObjAlloc(cmGoldSig_t,ctx,p);
-
- if( a != NULL )
- if( cmGoldSigInit(op,a) != cmOkRC )
- cmGoldSigFree(&op);
-
- return op;
-
- }
-
- cmRC_t cmGoldSigFree( cmGoldSig_t** pp )
- {
- cmRC_t rc = cmOkRC;
-
- if( pp == NULL || *pp == NULL )
- return rc;
-
- cmGoldSig_t* p = *pp;
-
- if((rc = cmGoldSigFinal(p)) != cmOkRC )
- return rc;
-
- unsigned i;
- for(i=0; i<p->a.chN; ++i)
- {
- cmMemFree(p->ch[i].bbV);
- cmMemFree(p->ch[i].mdV);
- }
-
- cmMemFree(p->ch);
- cmMemFree(p->rcosV);
- cmMemFree(p->pnM);
- cmMemFree(p);
- *pp = NULL;
-
- return rc;
- }
-
- cmRC_t cmGoldSigInit( cmGoldSig_t* p, const cmGoldSigArg_t* a )
- {
- cmRC_t rc = cmOkRC;
- unsigned i;
-
- p->a = *a; // store arg recd
- p->ch = cmMemResizeZ(cmGoldSigCh_t,p->ch,a->chN); // alloc channel array
- p->mlsN = (1 << a->lfsrN) - 1; // calc spreading code length
- p->rcosN = a->samplesPerChip * a->rcosOSFact; // calc rcos imp. resp. length
- p->rcosN += (p->rcosN % 2)==0; // force rcos imp. length odd
- p->rcosV = cmMemResizeZ(cmSample_t,p->rcosV,p->rcosN); // alloc rcos imp. resp. vector
- p->pnM = cmMemResizeZ(int,p->pnM,p->mlsN*a->chN); // alloc spreading-code mtx
- p->sigN = p->mlsN * a->samplesPerChip; // calc audio signal length
-
- // generate spreading codes
- if( cmGenGoldCodes(a->lfsrN, a->mlsCoeff0, a->mlsCoeff1, a->chN, p->pnM, p->mlsN ) == false )
- {
- rc = cmCtxRtCondition(&p->obj,cmSubSysFailRC,"Unable to generate sufficient balanced Gold codes.");
- goto errLabel;
- }
-
- // generate the rcos impulse response
- _cmGoldSigRaisedCos(p->rcosV,p->rcosN,a->samplesPerChip,a->rcosBeta);
-
- // for each channel
- for(i=0; i<a->chN; ++i)
- {
- // Note: if (i*p->mlsN) is set to 0 in the following line then all channels
- // will use the same spreading code.
- p->ch[i].pnV = p->pnM + (i*p->mlsN); // get ch. spreading code
- p->ch[i].bbV = cmMemResizeZ(cmSample_t,p->ch[i].bbV,p->sigN); // alloc baseband signal vector
- p->ch[i].mdV = cmMemResizeZ(cmSample_t,p->ch[i].mdV,p->sigN); // alloc output audio vector
-
- // Convolve spreading code with rcos impulse reponse to form baseband signal.
- _cmGoldSigConv(p, i );
-
- // Modulate baseband signal to carrier frq. and apply attack/decay envelope.
- _cmGoldSigModulate(p, i );
- }
-
- errLabel:
- if((rc = cmErrLastRC(&p->obj.err)) != cmOkRC )
- cmGoldSigFree(&p);
-
- return rc;
- }
-
- cmRC_t cmGoldSigFinal( cmGoldSig_t* p )
- { return cmOkRC; }
-
- cmRC_t cmGoldSigWrite( cmCtx* ctx, cmGoldSig_t* p, const char* fn )
- {
- cmVectArray_t* vap = NULL;
- unsigned i;
-
- vap = cmVectArrayAlloc(ctx,kSampleVaFl);
-
- for(i=0; i<p->a.chN; ++i)
- {
- cmVectArrayAppendS(vap,p->ch[i].bbV,p->sigN);
- cmVectArrayAppendS(vap,p->ch[i].mdV,p->sigN);
- }
-
- cmVectArrayWrite(vap,fn);
-
- cmVectArrayFree(&vap);
-
- return cmOkRC;
- }
-
-
- cmRC_t cmGoldSigGen( cmGoldSig_t* p, unsigned chIdx, unsigned prefixN, unsigned dsN, unsigned *bsiV, unsigned bsiN, double noiseGain, cmSample_t** yVRef, unsigned* yNRef )
- {
- unsigned yN = prefixN + bsiN * (p->sigN + dsN);
- cmSample_t* yV = cmMemAllocZ(cmSample_t,yN);
- unsigned i;
-
- cmVOS_Random(yV, yN, -noiseGain, noiseGain );
-
- for(i=0; i<bsiN; ++i)
- {
- bsiV[i] = prefixN + i*(p->sigN + dsN);
-
- cmVOS_AddVV(yV + bsiV[i], p->sigN, p->ch[chIdx].mdV );
- }
-
- if( yVRef != NULL )
- *yVRef = yV;
-
- if( yNRef != NULL )
- *yNRef = yN;
-
- return cmOkRC;
- }
-
-
- //=======================================================================================================================
- cmPhat_t* cmPhatAlloc( cmCtx* ctx, cmPhat_t* p, unsigned chN, unsigned hN, float alpha, unsigned mult, unsigned flags )
- {
- cmPhat_t* op = cmObjAlloc(cmPhat_t,ctx,p);
-
- // The FFT buffer and the delay line is at least twice the size of the
- // id signal. This will guarantee that at least one complete id signal
- // is inside the buffer. In practice it means that it is possible
- // that there will be two id's in the buffer therefore if there are
- // two correlation spikes it is important that we take the second.
- p->fhN = cmNextPowerOfTwo(mult*hN);
-
- // allocate the FFT object
- cmFftAllocSR(ctx,&p->fft,NULL,p->fhN,kToPolarFftFl);
-
- if( chN != 0 )
- if( cmPhatInit(op,chN,hN,alpha,mult,flags) != cmOkRC )
- cmPhatFree(&op);
-
- return op;
-
- }
-
- cmRC_t cmPhatFree( cmPhat_t** pp )
- {
- cmRC_t rc = cmOkRC;
-
- if( pp == NULL || *pp == NULL )
- return rc;
-
- cmPhat_t* p = *pp;
- if((rc = cmPhatFinal(p)) != cmOkRC )
- return rc;
-
- cmMemFree(p->t0V);
- cmMemFree(p->t1V);
- cmMemFree(p->dV);
- cmMemFree(p->xV);
- cmMemFree(p->fhM);
- cmMemFree(p->mhM);
- cmMemFree(p->wndV);
- cmObjFreeStatic(cmFftFreeSR, cmFftSR, p->fft);
- cmVectArrayFree(&p->ftVa);
- cmObjFree(pp);
-
- return rc;
-
- }
-
-
- cmRC_t cmPhatInit( cmPhat_t* p, unsigned chN, unsigned hN, float alpha, unsigned mult, unsigned flags )
- {
- cmRC_t rc = cmOkRC;
- if((rc = cmPhatFinal(cmOkRC)) != cmOkRC )
- return rc;
-
- p->fhN = cmNextPowerOfTwo(mult*hN);
-
- if((cmFftInitSR(&p->fft, NULL, p->fhN, kToPolarFftFl)) != cmOkRC )
- return rc;
-
- p->alpha = alpha;
- p->flags = flags;
-
- // allocate the delay line
- p->dV = cmMemResizeZ(cmSample_t,p->dV,p->fhN);
- p->di = 0;
-
- // allocate the linear buffer
- p->xV = cmMemResizeZ(cmSample_t,p->xV,p->fhN);
- p->t0V = cmMemResizeZ(cmComplexR_t,p->t0V,p->fhN);
- p->t1V = cmMemResizeZ(cmComplexR_t,p->t1V,p->fhN);
-
- // allocate the window function
- p->wndV = cmMemResizeZ(cmSample_t,p->wndV,p->fhN);
- cmVOS_Hann(p->wndV,p->fhN);
-
- // allocate the signal id matrix
- p->chN = chN;
- p->hN = hN;
- p->binN = p->fft.binCnt; //atFftRealBinCount(p->fftH);
- p->fhM = cmMemResizeZ(cmComplexR_t, p->fhM, p->fhN * chN);
- p->mhM = cmMemResizeZ(float, p->mhM, p->binN * chN);
- cmPhatReset(p);
-
- //if( cmIsFlag(p->flags,kDebugAtPhatFl))
- // cmVectArrayAlloc(ctx, &p->ftVa, kSampleVaFl );
- //else
- // p->ftVa = NULL;
-
- return rc;
-
- }
-
- cmRC_t cmPhatFinal( cmPhat_t* p )
- { return cmOkRC; }
-
- cmRC_t cmPhatReset( cmPhat_t* p )
- {
- p->di = 0;
- p->absIdx = 0;
- cmVOS_Zero(p->dV,p->fhN);
- return cmOkRC;
- }
-
- cmRC_t cmPhatSetId( cmPhat_t* p, unsigned chIdx, const cmSample_t* hV, unsigned hN )
- {
- unsigned i;
- assert( chIdx < p->chN );
- assert( hN == p->hN );
-
- // Allocate a window vector
- cmSample_t* wndV = cmMemAllocZ(cmSample_t,hN);
- cmVOS_Hann(wndV,hN);
-
- // get ptr to output column in p->fhM[].
- cmComplexR_t* yV = p->fhM + (chIdx*p->fhN);
-
- // Zero pad hV[hN] to p->fhN;
- assert( hN <= p->fhN );
- cmVOS_Zero(p->xV,p->fhN);
- cmVOS_Copy(p->xV,hV,hN);
-
- // Apply the window function to the id signal
- if(atIsFlag(p->flags,kHannAtPhatFl) )
- cmVOS_MultVVV(p->xV,hV,wndV,hN);
-
- // take FFT of id signal. The result is in fft->complexV and fft->magV,phsV
- cmFftExecSR(p->fft, p->xV, p->fhN );
-
- // Store the magnitude of the id signal
- //atFftComplexAbs(p->mhM + (chIdx*p->binN), yV, p->binN);
- cmVOR_Copy(p->mhM + (chIdx*p->binN), p->fft->magV, p->binN );
-
- // Scale the magnitude
- cmVOS_MultVS( p->mhM + (chIdx*p->binN), p->binN, p->alpha);
-
- // store the complex conjugate of the FFT result in yV[]
- //atFftComplexConj(yV,p->binN);
- for(i=0; i<p->binN; ++i)
- yV[i].i = -(p->fft->complexV[i].i);
-
- cmMemFree(wndV);
-
- return kOkAtRC;
- }
-
- cmSample_t* _cmPhatReadVector( cmCtx* ctx, cmPhat_t* p, const char* fn, unsigned* vnRef )
- {
- cmVectArray_t* vap = NULL;
- cmSample_t* v = NULL;
-
- // instantiate a VectArray from a file
- if( cmVectArrayAllocFromFile(ctx, &vap, fn ) != kOkAtRC )
- {
- atErrMsg(&p->obj.err,kFileReadFailAtRC,"Id component vector file read failed '%s'.",fn);
- goto errLabel;
- }
-
- // get the count of elements in the vector
- *vnRef = cmVectArrayEleCount(vap);
-
- // allocate memory to hold the vector
- v = cmMemAlloc(&p->obj.err,cmSample_t,*vnRef);
-
- // copy the vector from the vector array object into v[]
- if( cmVectArrayGetF(vap,v,vnRef) != kOkAtRC )
- {
- cmMemFree(v);
- v = NULL;
- atErrMsg(&p->obj.err,kFileReadFailAtRC,"Id component vector copy out failed '%s'.",fn);
- goto errLabel;
- }
-
- cmRptPrintf(p->obj.err.rpt,"%i : %s",*vnRef,fn);
-
-
- errLabel:
- cmVectArrayFree(&vap);
-
- return v;
- }
-
-
- cmRC_t cmPhatExec( cmPhat_t* p, const cmSample_t* xV, unsigned xN )
- {
- unsigned n = atMin(xN,p->fhN-p->di);
-
- // update the delay line
- cmVOS_Copy(p->dV+p->di,xV,n);
-
- if( n < xN )
- cmVOS_Copy(p->dV,xV+n,xN-n);
-
- p->di = atModIncr(p->di,xN,p->fhN);
-
- // p->absIdx is the absolute sample index associated with di
- p->absIdx += xN;
-
- return kOkAtRC;
- }
-
-
- void cmPhatChExec(
- cmPhat_t* p,
- unsigned chIdx,
- unsigned sessionId,
- unsigned roleId)
- {
-
- unsigned n0 = p->fhN - p->di;
- unsigned n1 = p->fhN - n0;
-
- // Linearize the delay line into xV[]
- cmVOS_Copy(p->xV, p->dV + p->di, n0 );
- cmVOS_Copy(p->xV+n0, p->dV, n1 );
-
- if( atIsFlag(p->flags,kDebugAtPhatFl))
- cmVectArrayAppendS(p->ftVa, p->xV, p->fhN );
-
- // apply a window function to the incoming signal
- if( atIsFlag(p->flags,kHannAtPhatFl) )
- cmVOS_MultVV(p->xV,p->fhN,p->wndV);
-
- // Take the FFT of the delay line.
- // p->t0V[p->binN] = fft(p->xV)
- //atFftRealForward(p->fftH, p->xV, p->fhN, p->t0V, p->binN );
- cmFftExecSR(p->fft, p->xV, p->fhN );
-
- // Calc. the Cross Power Spectrum (aka cross spectral density) of the
- // input signal with the id signal.
- // Note that the CPS is equivalent to the Fourier Transform of the
- // cross-correlation of the two signals.
- // t0V[] *= p->fhM[:,chIdx]
- //atFftComplexMult( p->t0V, p->fhM + (chIdx * p->fhN), p->binN );
- cmVOCR_MultVVV( p->t0V, p->fft->complexV, p->fhM + (chIdx * p->fhN), p->binN)
-
- // Calculate the magnitude of the CPS.
- // xV[] = | t0V[] |
- cmVOCR_Abs( p->xV, p->t0V, p->binN );
-
- // Weight the CPS by the scaled magnitude of the id signal
- // (we want to emphasize the limited frequencies where the
- // id signal contains energy)
- // t0V[] *= p->mhM[:,chIdx]
- if( p->alpha > 0 )
- cmVOCR_MultR_VV( p->t0V, p->mhM + (chIdx*p->binN), p->binN);
-
- // Divide through by the magnitude of the CPS
- // This has the effect of whitening the spectram and thereby
- // minimizing the effect of the magnitude correlation
- // while maximimizing the effect of the phase correlation.
- //
- // t0V[] /= xV[]
- cmVOCR_DivR_VV( p->t0V, p->xV, p->binN );
-
- // Take the IFFT of the weighted CPS to recover the cross correlation.
- // xV[] = IFFT(t0V[])
-
-
- //// ***** atFftRealInverse( p->fftH, p->t0V, p->xV, p->fhN );
-
- // Shift the correlation spike to mark the end of the id
- cmVOS_Rotate( p->xV, p->fhN, -((int)p->hN) );
-
- // normalize by the length of the correlation
- cmVOS_DivVS(p->xV,p->fhN,p->fhN);
-
- if( atIsFlag(p->flags,kDebugAtPhatFl))
- {
- cmVectArrayAppendS(p->ftVa, p->xV, p->fhN );
-
- cmSample_t v[] = { sessionId, roleId };
- cmVectArrayAppendS(p->ftVa, v, sizeof(v)/sizeof(v[0]));
- }
-
- }
-
- cmRC_t cmPhatWrite( cmPhat_t* p, const char* dirStr )
- {
- cmRC_t rc = kOkAtRC;
-
- if( atIsFlag(p->flags, kDebugAtPhatFl))
- {
- char* path = NULL;
-
- if( p->ftVa != NULL )
- if((rc = cmVectArrayWrite(p->ftVa, path = atMakePath(&p->obj.err,path,"cmPhatFT","va",dirStr,NULL) )) != kOkAtRC )
- rc = atErrMsg(&p->obj.err,rc,"PHAT debug file write failed.");
-
- cmMemFree(path);
- }
-
- return rc;
- }
-
-
- cmRC_t cmPhatTest1( cmCtx* ctx, const char* dirStr )
- {
- cmRC_t rc = kOkAtRC;
- atSignalArg_t sa;
- atSignal_t* s = NULL;
- cmPhat_t* p = NULL;
- char* path = NULL;
- unsigned dspFrmCnt = 256;
- unsigned listenDelaySmp = 8196;
- double noiseGain = 0.05;
- unsigned chIdx = 0;
- cmSample_t* yV = NULL;
- unsigned yN = 0;
- double phatAlpha = 0.5;
- unsigned phatMult = 4.0;
- double nonLinExpo = 4.0;
- cmVectArray_t* outVA = NULL;
- cmVectArray_t* inVA = NULL;
- cmVectArray_t* statusVA = NULL;
- unsigned bsiN = 4;
- unsigned bsiV[bsiN]; // known signal onset in absolute samples
- unsigned esiV[bsiN]; // known signal offset
- unsigned lsiV[bsiN]; // end of listen time (when cmPhatChExec()) is run.
- unsigned dsiV[bsiN]; // detection time
- unsigned i,j;
-
- sa.chN = 1;
- sa.srate = 44100.0;
- sa.lfsrN = 8;
- sa.mlsCoeff0 = 0x8e;
- sa.mlsCoeff1 = 0x96;
- sa.samplesPerChip = 64;
- sa.rcosBeta = 0.5;
- sa.rcosOSFact = 4;
- sa.carrierHz = 17000.0;
- sa.envMs = 50.0;
-
- // allocate the the id signals
- if( atSignalAlloc( ctx, &s, &sa ) != kOkAtRC )
- return atErrMsg(&ctx->err, kTestFailAtRC, "Signal allocate failed.");
-
- // set the post signal listen delay to half the signal length
- listenDelaySmp = s->sigN/2;
-
- // allocate a PHAT detector
- if( cmPhatAlloc(ctx,&p,sa.chN,s->sigN, phatAlpha, phatMult, kDebugAtPhatFl ) != kOkAtRC )
- {
- rc = atErrMsg(&ctx->err, kTestFailAtRC, "PHAT allocate failed.");
- goto errLabel;
- }
-
- // register an id signal with the PHAT detector
- if( cmPhatSetId(p, chIdx, s->ch[chIdx].mdV, s->sigN ) != kOkAtRC )
- {
- rc = atErrMsg(&ctx->err, kTestFailAtRC, "PHAT setId failed.");
- goto errLabel;
- }
-
- // generate an input test signal containing bsiN id signals
- if( atSignalGen(s,chIdx,p->fhN,s->sigN,bsiV,bsiN,noiseGain,&yV,&yN) != kOkAtRC )
- {
- rc = atErrMsg(&ctx->err,kTestFailAtRC,"Signal generation failed.");
- goto errLabel;
- }
-
- // bsiV[] now holds signal onsets. Set esiV[] to signal offsets.
- atVOU_AddVVS(esiV,bsiV,bsiN,s->sigN );
-
- // set lsiV[] to end-of-listen location
- atVOU_AddVVS(lsiV,esiV,bsiN,listenDelaySmp);
-
- // zero the detection vector
- atVOU_Zero(dsiV,bsiN);
-
- // allocate a vector array to record the PHAT input signals
- if( cmVectArrayAlloc(ctx,&inVA,kSampleVaFl) != kOkAtRC )
- {
- rc = atErrMsg(&ctx->err, kTestFailAtRC, "vectArray inVA alloc failed.");
- goto errLabel;
- }
-
- // allocate a vector array to record the PHAT correlation output signals
- if( cmVectArrayAlloc(ctx,&outVA,kSampleVaFl) != kOkAtRC )
- {
- rc = atErrMsg(&ctx->err, kTestFailAtRC, "vectArray outVA alloc failed.");
- goto errLabel;
- }
-
- // allocate a vector array to record the PHAT status
- if( cmVectArrayAlloc(ctx,&statusVA,kSampleVaFl) != kOkAtRC )
- {
- rc = atErrMsg(&ctx->err, kTestFailAtRC, "vectArray statusVA alloc failed.");
- goto errLabel;
- }
-
-
- // for each 'dspFrmCnt' samples in the input signal
- for(i=0,j=0; j<bsiN && i<=yN-dspFrmCnt; i+=dspFrmCnt)
- {
- // store a copy of the input signal
- cmVectArrayAppendS(inVA,yV+i,dspFrmCnt);
-
- // feed the next dspFrmCnt samples to the PHAT detector
- cmPhatExec(p,yV+i,dspFrmCnt);
-
- // if the approximate end of an id signal is encountered
- if( lsiV[j] <= i && i < lsiV[j] + dspFrmCnt )
- {
- // execute the PHAT correlator
- cmPhatChExec( p, chIdx, -1, -1 );
-
- // apply non-linear exponent to the correlation vector
- cmVOS_PowV(p->xV,p->fhN,nonLinExpo);
-
- // locate the corr. peak inside the listening window
- // (the detection window is last 'detectWndSmp' samples in the corr. vector )
- unsigned detectWndSmp = 2*listenDelaySmp;
- dsiV[j] = cmVOS_ArgMax( p->xV + p->fhN - detectWndSmp, detectWndSmp);
-
- // convert the pk index to absolute time
- dsiV[j] = i + dspFrmCnt - detectWndSmp + dsiV[j];
-
- // sig beg sig end detect begin dtct end detect
- cmSample_t v[] = { bsiV[j], esiV[j], lsiV[j]-detectWndSmp, lsiV[j], dsiV[j] };
-
- // store the detection information
- cmVectArrayAppendS(statusVA,v,sizeof(v)/sizeof(v[0]));
-
- // store the correlation output vector
- cmVectArrayAppendS(outVA,p->xV,p->fhN);
-
- j += 1;
- }
- }
-
- // write inVA
- if( cmVectArrayWrite(inVA,path = atMakePath(&ctx->err,path,"phatIn","va",dirStr,NULL)) != kOkAtRC )
- {
- rc = atErrMsg(&ctx->err, kTestFailAtRC, "vectArray outVA write failed.");
- goto errLabel;
- }
-
- // write outVA
- if( cmVectArrayWrite(outVA,path = atMakePath(&ctx->err,path,"phatOut","va",dirStr,NULL)) != kOkAtRC )
- {
- rc = atErrMsg(&ctx->err, kTestFailAtRC, "vectArray outVA write failed.");
- goto errLabel;
- }
-
- // write statusVA
- if( cmVectArrayWrite(statusVA,path = atMakePath(&ctx->err,path,"phatStatus","va",dirStr,NULL)) != kOkAtRC )
- {
- rc = atErrMsg(&ctx->err, kTestFailAtRC, "vectArray statusVA write failed.");
- goto errLabel;
- }
-
- errLabel:
- cmVectArrayFree(&outVA);
- cmVectArrayFree(&inVA);
-
- if( cmPhatFree(&p) != kOkAtRC )
- atErrMsg(&ctx->err,kTestFailAtRC,"PHAT free failed.");
-
- if( atSignalFree(&s) != kOkAtRC )
- atErrMsg(&ctx->err,kTestFailAtRC,"Signal free failed.");
-
- return rc;
- }
-
- cmRC_t cmPhatTest2( cmCtx* ctx )
- {
- cmRC_t rc = kOkAtRC;
- cmPhat_t* p = NULL;
- unsigned hN = 16;
- float alpha = 1.0;
- unsigned mult = 4;
-
- cmSample_t hV[] = { 4,3,2,1, 0,0,0,0, 0,0,0,0, 0,0,0,0 };
- cmSample_t x0V[] = { 4,3,2,1, 0,0,0,0, 0,0,0,0, 0,0,0,0 };
- cmSample_t x1V[] = { 0,0,0,0, 4,3,2,1, 0,0,0,0, 0,0,0,0 };
- cmSample_t x2V[] = { 0,0,0,0, 0,0,0,0, 4,3,2,1, 0,0,0,0 };
- cmSample_t x3V[] = { 0,0,0,0, 0,0,0,0, 0,0,0,0, 4,3,2,1 };
-
- cmSample_t* xV[] = { x0V, x1V, x2V, x3V };
- unsigned chN = sizeof(xV)/sizeof(xV[0]);
- unsigned i;
-
- if(cmPhatAlloc(ctx,&p,chN,hN,alpha,mult,kNoFlagsAtPhatFl) != kOkAtRC )
- {
- rc = atErrMsg(&ctx->err,kTestFailAtRC,"cmPhatAlloc() failed.");
- goto errLabel;
- }
-
- for(i=0; i<chN; ++i)
- if( cmPhatSetId(p,i,hV,hN) != kOkAtRC )
- rc = atErrMsg(&ctx->err,kTestFailAtRC,"cmPhatSetId() failed.");
-
-
- for(i=0; i<chN; ++i)
- {
- cmPhatReset(p);
-
- if( cmPhatExec(p,xV[i],hN) != kOkAtRC )
- {
- rc = atErrMsg(&ctx->err,kTestFailAtRC,"cmPhatExec() failed.");
- goto errLabel;
- }
-
- cmPhatChExec(p, i, -1, -1);
- cmVOS_PrintL(&ctx->printRpt,"x:",p->xV,1,p->fhN);
- }
-
-
- errLabel:
-
- cmPhatFree(&p);
-
-
- return rc;
- }
|