/* * 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 <inttypes.h>#include <stdio.h> #include "libavutil/common.h" #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;    }    if (type == FF_DWT97_INT) {        // pre-scaling to simulate dequantization which places the binary point at 1 bit above from LSB        for (j = 0; j< s->linelen[decomp_levels-1][0] * s->linelen[decomp_levels-1][1]; j++)            array[j] = (uint32_t)array[j] << I_PRESHIFT;    }    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] - (int64_t)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]) * (int64_t)(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;}