void DiscretePeriodicWavelet::makeInverseTransform()
{
if (m_thresholdedSignal.empty()) {
if (m_transformedSignal.empty()) {
return;
}
idwt(m_transformedSignal, m_flag, toStdString(m_waveletFunction), m_outputSignal, m_length);
return;
}
idwt(m_thresholdedSignal, m_flag, toStdString(m_waveletFunction), m_outputSignal, m_length);
}
double spec_template::sample_theta_conditional_revprop_logpdf()
{
vector<double> Lbeta(L[0].size(),0);
vector<double> meanvec(L[0].size(),0);
MVprod(&L, &beta_draw, &Lbeta);
double logprob=0;
vector<double> var(psi_draw.size(),0);
vector<double> meanbuf(pars.p);
if(WAVELETS)
{
for(int i = 0; i < pars.p; i++)
{
var[i]=(1/(1+psi_draw[i]));
meanbuf[i]=var[i]*(dataWy[i]-Lbeta[i]);
}
idwt(meanbuf, pars.levsize, pars.h_vec, meanvec);
vector<double> y_draw(pars.n,0.0);
idwt(theta_draw, pars.levsize, pars.h_vec, y_draw);
for (int j = 0; j <pars.n; j++)
{
logprob=logprob+truncated_normal_logpdf(y_draw[j], meanvec[j], sqrt(Temp/lambda_draw), tau_draw[j]);
}
vector<double> addedpoints(pars.p-pars.n,0.0);
MVprod(&vex,&meanbuf,&addedpoints);
vector<double> tmp2(pars.p-pars.n,0.0);
MVprod(&vex,&theta_draw,&tmp2);
VVdif(&tmp2,&addedpoints, &tmp2);
Vconstprod(&tmp2,sqrt(Temp/lambda_draw), &tmp2);
for(int j = 0; j <pars.p-pars.n; j++)
logprob+=normlogpdf(tmp2[j], 0, 1);
}
else
{
for(unsigned int i = 0; i<psi_draw.size(); i++)
{
var[i] = (1/(1 + psi_draw[i]));
meanvec[i] = var[i]*(dataWy[i] - Lbeta[i]);
}
for (int j = 0; j < pars.p; j++)
logprob=logprob+truncated_normal_logpdf(theta_draw[j], meanvec[j],sqrt(Temp*var[j]/lambda_draw), tau_draw[j]);
}
return logprob;
}
/*!
* Matlab MEX definition for the inverse discrete wavelet transform.
*
* @param nlhs number of items on left hand side of matlab call
* @param plhs pointer to left hand side data structure
* @param nrhs number of items on right hand side of matlab call
* @param prhs pointer to right hand side data structure
*
*/
void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[]) {
double *x, *y;
rwt_init_params params = rwt_matlab_init(nlhs, plhs, nrhs, prhs, INVERSE_DWT);
y = mxGetPr(prhs[0]);
x = mxGetPr(plhs[0]);
plhs[1] = mxCreateDoubleMatrix(1, 1, mxREAL);
*mxGetPr(plhs[1]) = params.levels;
idwt(x, params.nrows, params.ncols, params.scalings, params.ncoeff, params.levels, y);
}
int test_idwt()
{
#include "test/wavelet/data_idwt/idwt_include.snip"
NUMTYPE *output;
unsigned int coef_length, output_length, i, j;
int retval = 1;
for (j = 0; j < sizeof(results) / sizeof(MultiArray); j++) {
coef_length = a_coefs[j].length;
output_length = idwtResultLength( coef_length, COIF3.len_filter );
output = (NUMTYPE*) malloc( sizeof(NUMTYPE) * (output_length + 1) );
output[ output_length ] = 3.14;
// Perform the inverset DWT.
idwt( output, a_coefs[j].array, d_coefs[j].array, coef_length, COIF3 );
// Verify that all values are within an epsilon of expected values.
for (i = 0; i < output_length; i++) {
if (fabs(output[i] - results[j].array[i]) > EPSILON) {
retval = 0;
}
}
// Verify that the final value was not overwritten.
if (output[output_length] != 3.14) {
retval = 0;
}
// Free our allocated memory.
free( output );
}
return retval;
}
int main(int argc, char *argv[]) {
FILE *in = stdin;
FILE *out = stdout;
FILE *sig = NULL;
char output_name[MAXPATHLEN] = "(stdout)";
char input_name[MAXPATHLEN] = "(stdin)";
char signature_name[MAXPATHLEN];
int i, c, w;
int row;
int n;
double alpha = 0.0;
double beta = 0.0;
int filter = 0;
int method = -1;
int level = 0;
char filter_name[MAXPATHLEN] = "";
int verbose = 0;
gray **image;
Image_tree dwts;
gray maxval;
int rows, cols, format;
double *watermark;
progname = argv[0];
pgm_init(&argc, argv); wm_init();
while ((c = getopt(argc, argv, "a:b:e:f:F:h?o:s:v:")) != EOF) {
switch (c) {
case 'a':
alpha = atof(optarg);
if (alpha <= 0.0) {
fprintf(stderr, "%s: alpha factor %f out of range\n", progname, alpha);
exit(1);
}
break;
case 'b':
beta = atof(optarg);
if (beta <= 0.0) {
fprintf(stderr, "%s: beta factor %f out of range\n", progname, beta);
exit(1);
}
break;
case 'e':
method = atoi(optarg);
if (method < 0) {
fprintf(stderr, "%s: wavelet filtering method %d out of range\n", progname, method);
exit(1);
}
break;
case 'f':
filter = atoi(optarg);
if (filter <= 0) {
fprintf(stderr, "%s: filter number %d out of range\n", progname, filter);
exit(1);
}
break;
case 'F':
strcpy(filter_name, optarg);
break;
case 'h':
case '?':
usage();
break;
case 'o':
if ((out = fopen(optarg, "wb")) == NULL) {
fprintf(stderr, "%s: unable to open output file %s\n", progname, optarg);
exit(1);
}
strcpy(output_name, optarg);
break;
case 's':
if ((sig = fopen(optarg, "r")) == NULL) {
fprintf(stderr, "%s: unable to open signature file %s\n", progname, optarg);
exit(1);
}
strcpy(signature_name, optarg);
break;
case 'v':
verbose = atoi(optarg);
if (verbose < 0) {
fprintf(stderr, "%s: verbosity level %d out of range\n", progname, verbose);
exit(1);
}
break;
}
}
argc -= optind;
argv += optind;
if (argc > 1) {
usage();
exit(1);
}
if (argc == 1 && *argv[0] != '-') {
if ((in = fopen(argv[0], "rb")) == NULL) {
fprintf(stderr, "%s: unable to open input file %s\n", progname, argv[0]);
exit(1);
}
else
strcpy(input_name, argv[0]);
}
if (sig) {
char line[32];
fgets(line, sizeof(line), sig);
if (strspn(line, "WGSG") >= 4) {
fscanf(sig, "%d\n", &n);
if (alpha == 0.0)
fscanf(sig, "%lf\n", &alpha);
else
fscanf(sig, "%*f\n");
if (beta == 0.0)
fscanf(sig, "%lf\n", &beta);
else
fscanf(sig, "%*f\n");
if (method < 0)
fscanf(sig, "%d\n", &method);
else
fscanf(sig, "%*d\n");
if (filter == 0)
fscanf(sig, "%d\n", &filter);
else
fscanf(sig, "%*d\n");
if (!strcmp(filter_name, ""))
fscanf(sig, "%[^\n\r]\n", filter_name);
else
fscanf(sig, "%*[^\n\r]\n");
}
else {
fprintf(stderr, "%s: invalid signature file %s\n", progname, signature_name);
exit(1);
}
}
else {
fprintf(stderr, "%s: signature file not specified, use -s file option\n", progname);
exit(1);
}
watermark = malloc(n * sizeof(double));
for (i = 0; i < n; i++)
fscanf(sig, "%lf\n", &watermark[i]);
fclose(sig);
pgm_readpgminit(in, &cols, &rows, &maxval, &format);
image = pgm_allocarray(cols, rows);
for (row = 0; row < rows; row++)
pgm_readpgmrow(in, image[row], cols, maxval, format);
fclose(in);
// complete decomposition
level = find_deepest_level(cols, rows) - 1;
// wavelet transform
init_dwt(cols, rows, filter_name, filter, level, method);
#ifdef POLLEN_STUFF
#include "pollen_stuff.c"
#endif
#ifdef PARAM_STUFF
#include "param_stuff.c"
#endif
dwts = fdwt(image);
// build tree for subband selection, calculate subband thresholds
init_subbands(dwts);
set_subbands_type_beta(HORIZONTAL, beta);
set_subbands_type_beta(VERTICAL, beta);
calc_subbands_threshold();
w = 0;
while (w < n) {
Subband_data s;
// select subband with max. threshold
s = select_subband();
if (verbose > 1)
fprintf(stderr, "%s: selected subband %s%d, T=%lf, beta=%lf\n", progname, subband_name(s->type), s->level, s->T, s->beta);
// watermark significant coefficients and set them selected
// check is entire signature has been embedded
c = select_subband_coeff(s);
do {
double p;
if (c < 0)
// no more significant coefficients in subband
break;
p = get_subband_coeff(s, c);
if (p < s->Cmax) {
if (verbose > 2)
fprintf(stderr, "%s: embedding sig. coeff. #%d (= %lf)\n into %s%d coeff. #%d\n",
progname, w, watermark[w], subband_name(s->type), s->level, c);
p = p + alpha * s->beta * s->T * watermark[w];
set_subband_coeff(s, c, p);
w++;
}
mark_subband_coeff(s, c);
// select next significant coefficient
c = select_subband_coeff_from(s, c);
} while (w < n);
// update subband threshold
s->T /= 2.0;
}
free_subbands();
free(watermark);
idwt(dwts, image);
pgm_writepgminit(out, cols, rows, maxval, 0);
for (row = 0; row < rows; row++)
pgm_writepgmrow(out, image[row], cols, maxval, 0);
fclose(out);
pgm_freearray(image, rows);
exit(0);
}
int main(int argc, char *argv[]) {
FILE *in = stdin;
FILE *out = stdout;
FILE *sig = NULL;
char output_name[MAXPATHLEN] = "(stdout)";
char input_name[MAXPATHLEN] = "(stdin)";
char signature_name[MAXPATHLEN];
int c;
int row, col;
int n;
double quality = 0.0;
int filter = 0;
int method = -1;
int level = 0;
char filter_name[MAXPATHLEN] = "";
int seed;
int verbose = 0;
gray **image;
Image_tree dwts;
gray maxval;
int rows, cols, colors, format;
progname = argv[0];
pgm_init(&argc, argv);
#ifdef __EMX__
_fsetmode(in, "b");
_fsetmode(out, "b");
#endif
while ((c = getopt(argc, argv, "e:f:F:h?l:o:q:s:v:")) != EOF) {
switch (c) {
case 'e':
method = atoi(optarg);
if (method < 0) {
fprintf(stderr, "%s: wavelet filtering method %d out of range\n", progname, method);
exit(1);
}
break;
case 'f':
filter = atoi(optarg);
if (filter <= 0) {
fprintf(stderr, "%s: filter number %d out of range\n", progname, filter);
exit(1);
}
break;
case 'F':
strcpy(filter_name, optarg);
break;
case 'h':
case '?':
usage();
break;
case 'l':
level = atoi(optarg);
if (level < 1) {
fprintf(stderr, "%s: embedding level out of range\n", progname);
exit(1);
}
break;
case 'o':
if ((out = fopen(optarg, "wb")) == NULL) {
fprintf(stderr, "%s: unable to open output file %s\n", progname, optarg);
exit(1);
}
strcpy(output_name, optarg);
break;
case 'q':
quality = atoi(optarg);
if (quality <= 0) {
fprintf(stderr, "%s: quality factor %d out of range\n", progname, quality);
exit(1);
}
break;
case 's':
if ((sig = fopen(optarg, "r")) == NULL) {
fprintf(stderr, "%s: unable to open signature file %s\n", progname, optarg);
exit(1);
}
strcpy(signature_name, optarg);
break;
case 'v':
verbose = atoi(optarg);
if (verbose < 0) {
fprintf(stderr, "%s: verbosity level %d out of range\n", progname, verbose);
exit(1);
}
break;
}
}
argc -= optind;
argv += optind;
if (argc > 1) {
usage();
exit(1);
}
if (argc == 1 && *argv[0] != '-')
if ((in = fopen(argv[0], "rb")) == NULL) {
fprintf(stderr, "%s: unable to open input file %s\n", progname, argv[0]);
exit(1);
}
else
strcpy(input_name, argv[0]);
if (sig) {
char line[32];
fgets(line, sizeof(line), sig);
if (strspn(line, "KDSG") >= 4) {
fscanf(sig, "%d\n", &n);
if (quality == 0.0)
fscanf(sig, "%lf\n", &quality);
else
fscanf(sig, "%*f\n");
if (method < 0)
fscanf(sig, "%d\n", &method);
else
fscanf(sig, "%*d\n");
if (filter == 0)
fscanf(sig, "%d\n", &filter);
else
fscanf(sig, "%*d\n");
if (!strcmp(filter_name, ""))
fscanf(sig, "%[^\n\r]\n", filter_name);
else
fscanf(sig, "%*[^\n\r]\n");
if (level == 0)
fscanf(sig, "%d\n", &level);
else
fscanf(sig, "%*d\n");
fscanf(sig, "%d\n", &seed);
srandom(seed);
n_signature = NBITSTOBYTES(nbit_signature);
fread(signature, sizeof(char), n_signature, sig);
fscanf(sig, "\n");
}
else {
fprintf(stderr, "%s: invalid signature file %s\n", progname, signature_name);
exit(1);
}
fclose(sig);
}
else {
fprintf(stderr, "%s: signature file not specified, use -s file option\n", progname);
exit(1);
}
pgm_readpgminit(in, &cols, &rows, &maxval, &format);
image = pgm_allocarray(cols, rows);
for (row = 0; row < rows; row++)
pgm_readpgmrow(in, image[row], cols, maxval, format);
fclose(in);
// check watermark dimensions and decomposition level
// decomposition of image
init_dwt(cols, rows, filter_name, filter, level, method);
#ifdef POLLEN_STUFF
{
double alpha, beta;
char *alpha_str = getenv("POLLEN_ALPHA"), *beta_str = getenv("POLLEN_BETA");
if (alpha_str && beta_str) {
alpha = atof(alpha_str);
beta = atof(beta_str);
if (alpha < -M_PI || alpha >= M_PI) {
fprintf(stderr, "%s: pollen - alpha %f out of range\n", progname, alpha);
exit(1);
}
if (beta < -M_PI || beta >= M_PI) {
fprintf(stderr, "%s: pollen - beta %f out of range\n", progname, beta);
exit(1);
}
if (verbose > 7)
fprintf(stderr, "%s: pollen - alpha %f, beta %f\n", progname, alpha, beta);
dwt_pollen_filter(alpha, beta);
}
}
#endif
dwts = fdwt(image);
// create 'image' from binary watermark
// decomposition of watermark
init_dwt(cols, rows, filter_name, filter, 1, method);
// dwts = fdwt(watermark);
// calculate mean value of image and set alpha
// setup of contrast sensitivity matrix
// segment detail images at each level
// calculate DFT of each segment
// compute salience for each segment
// calculate gamma or each detail image
// embed watermark
// reconstruction of watermarked image
idwt(dwts, image);
pgm_writepgminit(out, cols, rows, maxval, 0);
for (row = 0; row < rows; row++)
pgm_writepgmrow(out, image[row], cols, maxval, 0);
fclose(out);
pgm_freearray(image, rows);
exit(0);
}
int main() {
wave_object obj;
wt_object wt;
double *inp,*out,*diff;
int N, i,J;
FILE *ifp;
double temp[1200];
char *name = "db4";
obj = wave_init(name);// Initialize the wavelet
ifp = fopen(FILE_SIGNAL, "r");
i = 0;
if (!ifp) {
printf("Cannot Open File");
exit(100);
}
while (!feof(ifp)) {
fscanf(ifp, "%lf \n", &temp[i]);
i++;
}
N = 256;
inp = (double*)malloc(sizeof(double)* N);
out = (double*)malloc(sizeof(double)* N);
diff = (double*)malloc(sizeof(double)* N);
//wmean = mean(temp, N);
for (i = 0; i < N; ++i) {
inp[i] = temp[i];
//printf("%g \n",inp[i]);
}
J = 3;
wt = wt_init(obj, "dwt", N, J);// Initialize the wavelet transform object
setDWTExtension(wt, "sym");// Options are "per" and "sym". Symmetric is the default option
setWTConv(wt, "direct");
dwt(wt, inp);// Perform DWT
//DWT output can be accessed using wt->output vector. Use wt_summary to find out how to extract appx and detail coefficients
for (i = 0; i < wt->outlength; ++i) {
// printf("%g ",wt->output[i]);
}
idwt(wt, out);// Perform IDWT (if needed)
// Test Reconstruction
for (i = 0; i < wt->siglength; ++i) {
diff[i] = out[i] - inp[i];
}
printf("\n MAX %g \n", absmax(diff, wt->siglength)); // If Reconstruction succeeded then the output should be a small value.
wt_summary(wt);// Prints the full summary.
wave_free(obj);
wt_free(wt);
free(inp);
free(out);
free(diff);
return 0;
}
void ReconstructionTest()
{
wave_object obj;
wt_object wt;
double *inp,*out;
int N, i,J;
double epsilon = 1e-15;
char *type = (char*) "dwt";
N = 79926;
//N = 256;
inp = (double*)malloc(sizeof(double)* N);
out = (double*)malloc(sizeof(double)* N);
//wmean = mean(temp, N);
for (i = 0; i < N; ++i) {
inp[i] = (rand() / (double)(RAND_MAX));
}
std::vector<std::string > waveletNames;
for (unsigned int j = 0; j < 36; j++)
{
waveletNames.push_back(std::string("db") + patch::to_string(j + 1));
}
for (unsigned int j = 0; j < 17; j++)
{
waveletNames.push_back(std::string("coif") + patch::to_string(j + 1));
}
for (unsigned int j = 1; j < 20; j++)
{
waveletNames.push_back(std::string("sym") + patch::to_string(j + 1));
}
waveletNames.push_back("bior1.1");
waveletNames.push_back("bior1.3");
waveletNames.push_back("bior1.5");
waveletNames.push_back("bior2.2");
waveletNames.push_back("bior2.4");
waveletNames.push_back("bior2.6");
waveletNames.push_back("bior2.8");
waveletNames.push_back("bior3.1");
waveletNames.push_back("bior3.3");
waveletNames.push_back("bior3.5");
waveletNames.push_back("bior3.7");
waveletNames.push_back("bior3.9");
waveletNames.push_back("bior4.4");
waveletNames.push_back("bior5.5");
waveletNames.push_back("bior6.8");
waveletNames.push_back("rbior1.1");
waveletNames.push_back("rbior1.3");
waveletNames.push_back("rbior1.5");
waveletNames.push_back("rbior2.2");
waveletNames.push_back("rbior2.4");
waveletNames.push_back("rbior2.6");
waveletNames.push_back("rbior2.8");
waveletNames.push_back("rbior3.1");
waveletNames.push_back("rbior3.3");
waveletNames.push_back("rbior3.5");
waveletNames.push_back("rbior3.7");
waveletNames.push_back("rbior3.9");
waveletNames.push_back("rbior4.4");
waveletNames.push_back("rbior5.5");
waveletNames.push_back("rbior6.8");
for (unsigned int direct_fft = 0; direct_fft < 2; direct_fft++)
{
for (unsigned int sym_per = 0; sym_per < 2; sym_per++)
{
for (unsigned int j = 0; j < waveletNames.size(); j++)
{
char * name = new char[waveletNames[j].size() + 1];
memcpy(name, waveletNames[j].c_str(), waveletNames[j].size() + 1);
obj = wave_init(name);// Initialize the wavelet
for (J = 1; J < 3; J++)
{
//J = 3;
wt = wt_init(obj,(char*) "dwt", N, J);// Initialize the wavelet transform object
if (sym_per == 0)
setDWTExtension(wt, (char*) "sym");// Options are "per" and "sym". Symmetric is the default option
else
setDWTExtension(wt, (char*) "per");
if (direct_fft == 0)
setWTConv(wt, (char*) "direct");
else
setWTConv(wt, (char*) "fft");
dwt(wt, inp);// Perform DWT
idwt(wt, out);// Perform IDWT (if needed)
// Test Reconstruction
if (direct_fft == 0)
epsilon = 1e-8;
else
epsilon = 1e-10;
//BOOST_CHECK_SMALL(RMS_Error(out, inp, wt->siglength), epsilon); // If Reconstruction succeeded then the output should be a small value.
//printf("%g ",RMS_Error(out, inp, wt->siglength));
if (RMS_Error(out, inp, wt->siglength) > epsilon) {
printf("\n ERROR : DWT Reconstruction Unit Test Failed. Exiting. \n");
exit(-1);
}
wt_free(wt);
}
wave_free(obj);
delete[] name;
}
}
}
free(out);
free(inp);
}
