cmDspKr.c : Changed _cmDspGoertzel to initialize internal cmGoertzel object

in _cmDspGoertzelReset.
This commit is contained in:
kevin 2013-11-26 13:43:14 -05:00
parent 3d32b3ac51
commit da98170c43

View File

@ -2575,8 +2575,9 @@ struct cmDspClass_str* cmRecdPlayClassCons( cmDspCtx_t* ctx )
//========================================================================================================================================== //==========================================================================================================================================
enum enum
{ {
kHopFactGrId,
kInGrId, kInGrId,
kOutBaseGrId, kHzBaseGrId,
}; };
cmDspClass_t _cmGoertzelDC; cmDspClass_t _cmGoertzelDC;
@ -2586,13 +2587,15 @@ typedef struct
cmDspInst_t inst; cmDspInst_t inst;
cmGoertzel* g; cmGoertzel* g;
double outPhs; double outPhs;
unsigned outBaseGrId;
unsigned chCnt;
} cmDspGoertzel_t; } cmDspGoertzel_t;
cmDspInst_t* _cmDspGoertzelAlloc(cmDspCtx_t* ctx, cmDspClass_t* classPtr, unsigned storeSymId, unsigned instSymId, unsigned id, unsigned va_cnt, va_list vl ) cmDspInst_t* _cmDspGoertzelAlloc(cmDspCtx_t* ctx, cmDspClass_t* classPtr, unsigned storeSymId, unsigned instSymId, unsigned id, unsigned va_cnt, va_list vl )
{ {
if( va_cnt !=2 ) if( va_cnt !=3 )
{ {
cmDspClassErr(ctx,classPtr,kVarArgParseFailDspRC,"The 'Goertzel' constructor must have two arguments: a channel count and frequency array."); cmDspClassErr(ctx,classPtr,kVarArgParseFailDspRC,"The 'Goertzel' constructor must have two arguments: a channel count and frequency array.");
return NULL; return NULL;
@ -2601,22 +2604,61 @@ cmDspInst_t* _cmDspGoertzelAlloc(cmDspCtx_t* ctx, cmDspClass_t* classPtr, unsig
va_list vl1; va_list vl1;
va_copy(vl1,vl); va_copy(vl1,vl);
unsigned hopFact = va_arg(vl,unsigned);
int chCnt = va_arg(vl,int); int chCnt = va_arg(vl,int);
double* hzV = va_arg(vl,double*); double* hzV = va_arg(vl,double*);
unsigned outBaseGrId = kHzBaseGrId + chCnt;
unsigned i;
cmDspGoertzel_t* p = cmDspInstAllocV(cmDspGoertzel_t,ctx,classPtr,instSymId,id,storeSymId,0,vl1, cmDspGoertzel_t* p = cmDspInstAllocV(cmDspGoertzel_t,ctx,classPtr,instSymId,id,storeSymId,1,vl1,
1, "hop", kHopFactGrId, 0,0, kInDsvFl | kDoubleDsvFl, "Hop factor",
1, "in", kInGrId, 0,1, kInDsvFl | kAudioBufDsvFl, "Audio input", 1, "in", kInGrId, 0,1, kInDsvFl | kAudioBufDsvFl, "Audio input",
chCnt, "out", kOutBaseGrId, 0,1, kOutDsvFl | kDoubleDsvFl, "Detector output", chCnt, "hz", kHzBaseGrId, 0,0, kInDsvFl | kDoubleDsvFl, "Hz input.",
chCnt, "out", outBaseGrId, 0,1, kOutDsvFl | kDoubleDsvFl, "Detector output",
0 ); 0 );
va_end(vl1); va_end(vl1);
p->g = cmGoertzelAlloc(ctx->cmProcCtx, NULL, cmDspSysSampleRate(ctx->dspH), hzV, chCnt );
p->outBaseGrId = outBaseGrId;
p->chCnt = chCnt;
p->g = cmGoertzelAlloc(ctx->cmProcCtx, NULL, 0, NULL, 0,0,0,0 );
cmDspSetDefaultUInt(ctx,&p->inst, kHopFactGrId, 0, cmMax(hopFact,1));
for(i=0; i<chCnt; ++i)
cmDspSetDefaultDouble(ctx,&p->inst, kHzBaseGrId+i, 0.0, hzV[i] );
return &p->inst; return &p->inst;
} }
cmDspRC_t _cmDspGoertzelSetup( cmDspCtx_t* ctx, cmDspInst_t* inst )
{
cmDspRC_t rc = kOkDspRC;
cmDspGoertzel_t* p = (cmDspGoertzel_t*)inst;
unsigned hopFact = cmDspUInt(inst,kHopFactGrId);
unsigned procSmpCnt = cmDspAudioBufSmpCount(ctx,inst,kInGrId,0);
unsigned wndSmpCnt = procSmpCnt * hopFact;
double fcHzV[ p->chCnt ];
unsigned i;
for(i=0; i<p->chCnt; ++i)
{
double hz;
if( p->g->ch == NULL || p->g->ch[i].hz == 0 )
hz = cmDspDouble(inst,kHzBaseGrId);
else
hz = p->g->ch[i].hz;
fcHzV[i] = hz;
}
if( cmGoertzelInit(p->g,cmDspSysSampleRate(ctx->dspH),fcHzV,p->chCnt,procSmpCnt,procSmpCnt,wndSmpCnt) != cmOkRC )
rc = cmErrMsg(&inst->classPtr->err, kSubSysFailDspRC, "Unable to initialize the internal Goertzel detector.");
return rc;
}
cmDspRC_t _cmDspGoertzelFree(cmDspCtx_t* ctx, cmDspInst_t* inst, const cmDspEvt_t* evt ) cmDspRC_t _cmDspGoertzelFree(cmDspCtx_t* ctx, cmDspInst_t* inst, const cmDspEvt_t* evt )
{ {
cmDspRC_t rc = kOkDspRC; cmDspRC_t rc = kOkDspRC;
@ -2635,7 +2677,7 @@ cmDspRC_t _cmDspGoertzelReset(cmDspCtx_t* ctx, cmDspInst_t* inst, const cmDspEvt
p->outPhs = 0; p->outPhs = 0;
return kOkDspRC; return _cmDspGoertzelSetup(ctx, inst );
} }
cmDspRC_t _cmDspGoertzelExec(cmDspCtx_t* ctx, cmDspInst_t* inst, const cmDspEvt_t* evt ) cmDspRC_t _cmDspGoertzelExec(cmDspCtx_t* ctx, cmDspInst_t* inst, const cmDspEvt_t* evt )
@ -2646,12 +2688,12 @@ cmDspRC_t _cmDspGoertzelExec(cmDspCtx_t* ctx, cmDspInst_t* inst, const cmDspEvt_
unsigned n = cmDspAudioBufSmpCount(ctx,inst,kInGrId,0); unsigned n = cmDspAudioBufSmpCount(ctx,inst,kInGrId,0);
double outMs = 50.0; double outMs = 50.0;
double outPhsMax = outMs * cmDspSysSampleRate(ctx->dspH) / 1000.0; double outPhsMax = outMs * cmDspSysSampleRate(ctx->dspH) / 1000.0;
double outV[ p->g->chCnt ]; double outV[ p->chCnt ];
unsigned i; unsigned i;
if( x != NULL ) if( x != NULL )
{ {
cmGoertzelExec(p->g,x,n,outV,p->g->chCnt); cmGoertzelExec(p->g,x,n,outV,p->chCnt);
p->outPhs += n; p->outPhs += n;
if( p->outPhs > outPhsMax ) if( p->outPhs > outPhsMax )
@ -2659,9 +2701,9 @@ cmDspRC_t _cmDspGoertzelExec(cmDspCtx_t* ctx, cmDspInst_t* inst, const cmDspEvt_
while( p->outPhs > outPhsMax ) while( p->outPhs > outPhsMax )
p->outPhs -= outPhsMax; p->outPhs -= outPhsMax;
for(i=0; i<p->g->chCnt; ++i) for(i=0; i<p->chCnt; ++i)
{ {
cmDspSetDouble(ctx,inst,kOutBaseGrId+i,outV[i]); cmDspSetDouble(ctx,inst,p->outBaseGrId+i,outV[i]);
//printf("%f ",outV[i]); //printf("%f ",outV[i]);
} }
//printf("\n"); //printf("\n");
@ -2673,8 +2715,19 @@ cmDspRC_t _cmDspGoertzelExec(cmDspCtx_t* ctx, cmDspInst_t* inst, const cmDspEvt_
cmDspRC_t _cmDspGoertzelRecv(cmDspCtx_t* ctx, cmDspInst_t* inst, const cmDspEvt_t* evt ) cmDspRC_t _cmDspGoertzelRecv(cmDspCtx_t* ctx, cmDspInst_t* inst, const cmDspEvt_t* evt )
{ {
cmDspGoertzel_t* p = (cmDspGoertzel_t*)inst;
cmDspSetEvent(ctx,inst,evt); cmDspSetEvent(ctx,inst,evt);
if( kHzBaseGrId <= evt->dstVarId && evt->dstVarId < kHzBaseGrId+p->chCnt )
cmGoertzelSetFcHz(p->g, evt->dstVarId - kHzBaseGrId, cmDspDouble(inst,evt->dstVarId));
else
{
if( evt->dstVarId==kHopFactGrId )
{
_cmDspGoertzelSetup(ctx,inst);
}
}
return kOkDspRC; return kOkDspRC;
} }