zhangyiss/wavelib
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

commit : sureshrink error correction

Browse Source
This commit is contained in:
Rafat Hussain
2017-08-09 04:50:41 +05:30
parent fa68a811af
commit e1baa7fc17
4 changed files with 25 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
denoise/denoise.c
View File

@@ -101,8 +101,8 @@ void getDWTDetail(wt_object wt, double *detail, int N, int level) {
} }
} }
void visushrink(double *signal,int N,char *wname,char *method,char *ext,char *thresh,double *denoised) { void visushrink(double *signal,int N,int J,char *wname,char *method,char *ext,char *thresh,double *denoised) {
int J,filt_len,iter,i,dlen,dwt_len,sgn; int filt_len,iter,i,dlen,dwt_len,sgn, MaxIter;
double sigma,td,tmp; double sigma,td,tmp;
wave_object wave; wave_object wave;
wt_object wt; wt_object wt;
@@ -112,10 +112,11 @@ void visushrink(double *signal,int N,char *wname,char *method,char *ext,char *th
filt_len = wave->filtlength; filt_len = wave->filtlength;
J = (int) (log((double)N / ((double)filt_len - 1.0)) / log(2.0)); MaxIter = (int) (log((double)N / ((double)filt_len - 1.0)) / log(2.0));
if (J > 50) { if (J > MaxIter) {
J = 50; printf("\n Error - The Signal Can only be iterated %d times using this wavelet. Exiting\n",MaxIter);
exit(-1);
} }
wt = wt_init(wave,method,N,J); wt = wt_init(wave,method,N,J);
@@ -168,8 +169,8 @@ void visushrink(double *signal,int N,char *wname,char *method,char *ext,char *th
wt_free(wt); wt_free(wt);
} }
void sureshrink(double *signal,int N,char *wname,char *method,char *ext,char *thresh,double *denoised) { void sureshrink(double *signal,int N,int J,char *wname,char *method,char *ext,char *thresh,double *denoised) {
int J, filt_len,i,it,len,dlen,dwt_len,min_index,sgn; int filt_len,i,it,len,dlen,dwt_len,min_index,sgn, MaxIter;
double sigma,norm,td,tv,te,ct,thr,temp,x_sum; double sigma,norm,td,tv,te,ct,thr,temp,x_sum;
wave_object wave; wave_object wave;
wt_object wt; wt_object wt;
@@ -179,10 +180,11 @@ void sureshrink(double *signal,int N,char *wname,char *method,char *ext,char *th
filt_len = wave->filtlength; filt_len = wave->filtlength;
J = (int) (log((double)N / ((double)filt_len - 1.0)) / log(2.0)); MaxIter = (int) (log((double)N / ((double)filt_len - 1.0)) / log(2.0));
if (J > 50) { if (J > MaxIter) {
J = 50; printf("\n Error - The Signal Can only be iterated %d times using this wavelet. Exiting\n",MaxIter);
exit(-1);
} }
wt = wt_init(wave,method,N,J); wt = wt_init(wave,method,N,J);
@@ -248,17 +250,17 @@ void sureshrink(double *signal,int N,char *wname,char *method,char *ext,char *th
if(!strcmp(thresh,"hard")) { if(!strcmp(thresh,"hard")) {
for(i = 0; i < dwt_len;++i) { for(i = 0; i < dwt_len;++i) {
if (fabs(wt->output[len+i]) < td) { if (fabs(wt->output[len+i]) < td) {
wt->output[i] = 0; wt->output[len+i] = 0;
} }
} }
} else if(!strcmp(thresh,"soft")) { } else if(!strcmp(thresh,"soft")) {
for(i = 0; i < dwt_len;++i) { for(i = 0; i < dwt_len;++i) {
if (fabs(wt->output[len + i]) < td) { if (fabs(wt->output[len + i]) < td) {
wt->output[i] = 0; wt->output[len+i] = 0;
} else { } else {
sgn = wt->output[len+i] >= 0 ? 1 : -1; sgn = wt->output[len+i] >= 0 ? 1 : -1;
temp = sgn * (fabs(wt->output[len+i]) - td); temp = sgn * (fabs(wt->output[len+i]) - td);
wt->output[i] = temp; wt->output[len+i] = temp;
} }
} }
} }

4
denoise/denoise.h
View File

@@ -16,9 +16,9 @@ extern "C" {
#endif #endif
void visushrink(double *signal,int N,char *wname,char *method,char *ext,char *thresh,double *denoised); void visushrink(double *signal,int N,int J,char *wname,char *method,char *ext,char *thresh,double *denoised);
void sureshrink(double *signal,int N,char *wname,char *method,char *ext,char *thresh,double *denoised); void sureshrink(double *signal,int N,int J,char *wname,char *method,char *ext,char *thresh,double *denoised);
double mad(double *x, int N); double mad(double *x, int N);

4
header/denoise.h
View File

@@ -11,7 +11,9 @@ extern "C" {
#endif #endif
void visushrink(double *signal,int N,char *wname,char *method,char *ext,char *thresh,double *denoised); void visushrink(double *signal,int N,int J,char *wname,char *method,char *ext,char *thresh,double *denoised);
void sureshrink(double *signal,int N,int J,char *wname,char *method,char *ext,char *thresh,double *denoised);
double mad(double *x, int N); double mad(double *x, int N);

8
test/denoisetest.c
View File

@@ -7,7 +7,7 @@
int main() { int main() {
// gcc -Wall -I../header -L../Bin denoisetest.c -o denoise -ldenoiselib -lwavelib -lm // gcc -Wall -I../header -L../Bin denoisetest.c -o denoise -ldenoiselib -lwavelib -lm
double *inp,*oup; double *inp,*oup;
int i,N; int i,N,J;
FILE *ifp,*ofp; FILE *ifp,*ofp;
double temp[2400]; double temp[2400];
@@ -31,6 +31,7 @@ int main() {
fclose(ifp); fclose(ifp);
N = i; N = i;
J = 5;
inp = (double*)malloc(sizeof(double)* N); inp = (double*)malloc(sizeof(double)* N);
oup = (double*)malloc(sizeof(double)* N); oup = (double*)malloc(sizeof(double)* N);
@@ -39,9 +40,10 @@ int main() {
inp[i] = temp[i]; inp[i] = temp[i];
} }
visushrink(inp,N,wname,method,ext,thresh,oup); //visushrink(inp,N,wname,method,ext,thresh,oup);
sureshrink(inp,N,J,wname,method,ext,thresh,oup);
ofp = fopen("denoised.txt", "w"); ofp = fopen("denoiseds.txt", "w");
for(i = 0; i < N;++i) { for(i = 0; i < N;++i) {
fprintf(ofp,"%g \n",oup[i]); fprintf(ofp,"%g \n",oup[i]);