commit : DWPT and Wavelet Tree added

This commit is contained in:
Rafat Hussain
2016-01-29 17:09:44 +05:30
parent 2197daab4d
commit 66dc615ac8
9 changed files with 1657 additions and 30 deletions

61
test/dwpttest.c Normal file
View File

@@ -0,0 +1,61 @@
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#include "../header/wavelib.h"
double absmax(double *array, int N) {
double max;
int i;
max = 0.0;
for (i = 0; i < N; ++i) {
if (fabs(array[i]) >= max) {
max = fabs(array[i]);
}
}
return max;
}
int main() {
int i, J, N;
wave_object obj;
wpt_object wt;
double *inp, *oup, *diff;
char *name = "db4";
obj = wave_init(name);// Initialize the wavelet
N = 788 + 23;
inp = (double*)malloc(sizeof(double)* N);
oup = (double*)malloc(sizeof(double)* N);
diff = (double*)malloc(sizeof(double)* N);
for (i = 1; i < N + 1; ++i) {
//inp[i - 1] = -0.25*i*i*i + 25 * i *i + 10 * i;
inp[i - 1] = i;
}
J = 4;
wt = wpt_init(obj, N, J);// Initialize the wavelet transform Tree object
setDWPTExtension(wt, "per");// Options are "per" and "sym". Symmetric is the default option
setDWPTEntropy(wt, "logenergy", 0);
dwpt(wt, inp); // Discrete Wavelet Packet Transform
idwpt(wt, oup); // Inverse Discrete Wavelet Packet Transform
for (i = 0; i < N; ++i) {
diff[i] = (inp[i] - oup[i])/inp[i];
}
wpt_summary(wt); // Tree Summary
printf("\n MAX %g \n", absmax(diff, wt->siglength)); // If Reconstruction succeeded then the output should be a small value.
free(inp);
free(oup);
free(diff);
wave_free(obj);
wpt_free(wt);
return 0;
}

47
test/wtreetest.c Normal file
View File

@@ -0,0 +1,47 @@
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#include "../header/wavelib.h"
int main() {
int i, J, N, len;
int X, Y;
wave_object obj;
wtree_object wt;
double *inp, *oup;
char *name = "db3";
obj = wave_init(name);// Initialize the wavelet
N = 147;
inp = (double*)malloc(sizeof(double)* N);
for (i = 1; i < N + 1; ++i) {
inp[i - 1] = -0.25*i*i*i + 25 * i *i + 10 * i;
}
J = 3;
wt = wtree_init(obj, N, J);// Initialize the wavelet transform object
setWTREEExtension(wt, "sym");// Options are "per" and "sym". Symmetric is the default option
wtree(wt, inp);
wtree_summary(wt);
X = 3;
Y = 5;
len = getWTREENodelength(wt, X);
printf("\n %d", len);
printf("\n");
oup = (double*)malloc(sizeof(double)* len);
printf("Node [%d %d] Coefficients : \n",X,Y);
getWTREECoeffs(wt, X, Y, oup, len);
for (i = 0; i < len; ++i) {
printf("%g ", oup[i]);
}
printf("\n");
free(inp);
free(oup);
wave_free(obj);
wtree_free(wt);
return 0;
}