mirror of
https://github.com/simon987/wavelib.git
synced 2025-12-18 04:49:02 +00:00
commit : DWPT and Wavelet Tree added
This commit is contained in:
47
test/wtreetest.c
Normal file
47
test/wtreetest.c
Normal 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;
|
||||
}
|
||||
Reference in New Issue
Block a user