mirror of
https://github.com/pineappleEA/pineapple-src.git
synced 2024-12-05 18:28:23 -05:00
142 lines
4.6 KiB
C
142 lines
4.6 KiB
C
|
/*
|
||
|
* Discrete wavelet transform
|
||
|
* Copyright (c) 2007 Kamil Nowosad
|
||
|
* Copyright (c) 2013 Nicolas Bertrand <nicoinattendu@gmail.com>
|
||
|
*
|
||
|
* This file is part of FFmpeg.
|
||
|
*
|
||
|
* FFmpeg is free software; you can redistribute it and/or
|
||
|
* modify it under the terms of the GNU Lesser General Public
|
||
|
* License as published by the Free Software Foundation; either
|
||
|
* version 2.1 of the License, or (at your option) any later version.
|
||
|
*
|
||
|
* FFmpeg is distributed in the hope that it will be useful,
|
||
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||
|
* Lesser General Public License for more details.
|
||
|
*
|
||
|
* You should have received a copy of the GNU Lesser General Public
|
||
|
* License along with FFmpeg; if not, write to the Free Software
|
||
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
||
|
*/
|
||
|
|
||
|
#include "libavcodec/jpeg2000dwt.c"
|
||
|
|
||
|
#include "libavutil/lfg.h"
|
||
|
|
||
|
#define MAX_W 256
|
||
|
|
||
|
static int test_dwt(int *array, int *ref, int border[2][2], int decomp_levels, int type, int max_diff) {
|
||
|
int ret, j;
|
||
|
DWTContext s1={{{0}}}, *s= &s1;
|
||
|
int64_t err2 = 0;
|
||
|
|
||
|
ret = ff_jpeg2000_dwt_init(s, border, decomp_levels, type);
|
||
|
if (ret < 0) {
|
||
|
fprintf(stderr, "ff_jpeg2000_dwt_init failed\n");
|
||
|
return 1;
|
||
|
}
|
||
|
ret = ff_dwt_encode(s, array);
|
||
|
if (ret < 0) {
|
||
|
fprintf(stderr, "ff_dwt_encode failed\n");
|
||
|
return 1;
|
||
|
}
|
||
|
ret = ff_dwt_decode(s, array);
|
||
|
if (ret < 0) {
|
||
|
fprintf(stderr, "ff_dwt_encode failed\n");
|
||
|
return 1;
|
||
|
}
|
||
|
for (j = 0; j<MAX_W * MAX_W; j++) {
|
||
|
if (FFABS(array[j] - ref[j]) > max_diff) {
|
||
|
fprintf(stderr, "missmatch at %d (%d != %d) decomp:%d border %d %d %d %d\n",
|
||
|
j, array[j], ref[j],decomp_levels, border[0][0], border[0][1], border[1][0], border[1][1]);
|
||
|
return 2;
|
||
|
}
|
||
|
err2 += (array[j] - ref[j]) * (array[j] - ref[j]);
|
||
|
array[j] = ref[j];
|
||
|
}
|
||
|
ff_dwt_destroy(s);
|
||
|
|
||
|
printf("%s, decomp:%2d border %3d %3d %3d %3d milli-err2:%9"PRId64"\n",
|
||
|
type == FF_DWT53 ? "5/3i" : "9/7i",
|
||
|
decomp_levels, border[0][0], border[0][1], border[1][0], border[1][1],
|
||
|
1000*err2 / ((border[0][1] - border[0][0])*(border[1][1] - border[1][0])));
|
||
|
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
static int test_dwtf(float *array, float *ref, int border[2][2], int decomp_levels, float max_diff) {
|
||
|
int ret, j;
|
||
|
DWTContext s1={{{0}}}, *s= &s1;
|
||
|
double err2 = 0;
|
||
|
|
||
|
ret = ff_jpeg2000_dwt_init(s, border, decomp_levels, FF_DWT97);
|
||
|
if (ret < 0) {
|
||
|
fprintf(stderr, "ff_jpeg2000_dwt_init failed\n");
|
||
|
return 1;
|
||
|
}
|
||
|
ret = ff_dwt_encode(s, array);
|
||
|
if (ret < 0) {
|
||
|
fprintf(stderr, "ff_dwt_encode failed\n");
|
||
|
return 1;
|
||
|
}
|
||
|
ret = ff_dwt_decode(s, array);
|
||
|
if (ret < 0) {
|
||
|
fprintf(stderr, "ff_dwt_encode failed\n");
|
||
|
return 1;
|
||
|
}
|
||
|
for (j = 0; j<MAX_W * MAX_W; j++) {
|
||
|
if (FFABS(array[j] - ref[j]) > max_diff) {
|
||
|
fprintf(stderr, "missmatch at %d (%f != %f) decomp:%d border %d %d %d %d\n",
|
||
|
j, array[j], ref[j],decomp_levels, border[0][0], border[0][1], border[1][0], border[1][1]);
|
||
|
return 2;
|
||
|
}
|
||
|
err2 += (array[j] - ref[j]) * (array[j] - ref[j]);
|
||
|
array[j] = ref[j];
|
||
|
}
|
||
|
ff_dwt_destroy(s);
|
||
|
|
||
|
printf("9/7f, decomp:%2d border %3d %3d %3d %3d err2:%20.3f\n",
|
||
|
decomp_levels, border[0][0], border[0][1], border[1][0], border[1][1],
|
||
|
err2 / ((border[0][1] - border[0][0])*(border[1][1] - border[1][0])));
|
||
|
|
||
|
return 0;
|
||
|
}
|
||
|
|
||
|
static int array[MAX_W * MAX_W];
|
||
|
static int ref [MAX_W * MAX_W];
|
||
|
static float arrayf[MAX_W * MAX_W];
|
||
|
static float reff [MAX_W * MAX_W];
|
||
|
|
||
|
int main(void) {
|
||
|
AVLFG prng;
|
||
|
int i,j;
|
||
|
int border[2][2];
|
||
|
int ret, decomp_levels;
|
||
|
|
||
|
av_lfg_init(&prng, 1);
|
||
|
|
||
|
for (i = 0; i<MAX_W * MAX_W; i++)
|
||
|
arrayf[i] = reff[i] = array[i] = ref[i] = av_lfg_get(&prng) % 2048;
|
||
|
|
||
|
for (i = 0; i < 100; i++) {
|
||
|
for (j=0; j<4; j++)
|
||
|
border[j>>1][j&1] = av_lfg_get(&prng) % MAX_W;
|
||
|
if (border[0][0] >= border[0][1] || border[1][0] >= border[1][1])
|
||
|
continue;
|
||
|
decomp_levels = av_lfg_get(&prng) % FF_DWT_MAX_DECLVLS;
|
||
|
|
||
|
ret = test_dwt(array, ref, border, decomp_levels, FF_DWT53, 0);
|
||
|
if (ret)
|
||
|
return ret;
|
||
|
ret = test_dwt(array, ref, border, decomp_levels, FF_DWT97_INT, FFMIN(7+5*decomp_levels, 15+3*decomp_levels));
|
||
|
if (ret)
|
||
|
return ret;
|
||
|
ret = test_dwtf(arrayf, reff, border, decomp_levels, 0.05);
|
||
|
if (ret)
|
||
|
return ret;
|
||
|
}
|
||
|
|
||
|
return 0;
|
||
|
}
|