2 * Copyright (C) 2004 Michael Niedermayer <michaelni@gmx.at>
4 * This file is part of FFmpeg.
6 * FFmpeg is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * FFmpeg is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with FFmpeg; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21 #include "libavcodec/snowenc.c"
27 #include "libavutil/lfg.h"
28 #include "libavutil/mathematics.h"
33 int buffer[2][width*height];
37 s.spatial_decomposition_count=6;
38 s.spatial_decomposition_type=1;
40 s.temp_dwt_buffer = av_mallocz_array(width, sizeof(DWTELEM));
41 s.temp_idwt_buffer = av_mallocz_array(width, sizeof(IDWTELEM));
43 if (!s.temp_dwt_buffer || !s.temp_idwt_buffer) {
44 fprintf(stderr, "Failed to allocate memory\n");
48 av_lfg_init(&prng, 1);
50 printf("testing 5/3 DWT\n");
51 for(i=0; i<width*height; i++)
52 buffer[0][i] = buffer[1][i] = av_lfg_get(&prng) % 54321 - 12345;
54 ff_spatial_dwt(buffer[0], s.temp_dwt_buffer, width, height, width, s.spatial_decomposition_type, s.spatial_decomposition_count);
55 ff_spatial_idwt((IDWTELEM*)buffer[0], s.temp_idwt_buffer, width, height, width, s.spatial_decomposition_type, s.spatial_decomposition_count);
57 for(i=0; i<width*height; i++)
58 if(buffer[0][i]!= buffer[1][i]) printf("fsck: %6d %12d %7d\n",i, buffer[0][i], buffer[1][i]);
60 printf("testing 9/7 DWT\n");
61 s.spatial_decomposition_type=0;
62 for(i=0; i<width*height; i++)
63 buffer[0][i] = buffer[1][i] = av_lfg_get(&prng) % 54321 - 12345;
65 ff_spatial_dwt(buffer[0], s.temp_dwt_buffer, width, height, width, s.spatial_decomposition_type, s.spatial_decomposition_count);
66 ff_spatial_idwt((IDWTELEM*)buffer[0], s.temp_idwt_buffer, width, height, width, s.spatial_decomposition_type, s.spatial_decomposition_count);
68 for(i=0; i<width*height; i++)
69 if(FFABS(buffer[0][i] - buffer[1][i])>20) printf("fsck: %6d %12d %7d\n",i, buffer[0][i], buffer[1][i]);
72 int level, orientation, x, y;
76 memset(errors, 0, sizeof(errors));
77 s.spatial_decomposition_count=3;
78 s.spatial_decomposition_type=0;
79 for(level=0; level<s.spatial_decomposition_count; level++){
80 for(orientation=level ? 1 : 0; orientation<4; orientation++){
81 int w= width >> (s.spatial_decomposition_count-level);
82 int h= height >> (s.spatial_decomposition_count-level);
83 int stride= width << (s.spatial_decomposition_count-level);
84 DWTELEM *buf= buffer[0];
87 if(orientation&1) buf+=w;
88 if(orientation>1) buf+=stride>>1;
90 memset(buffer[0], 0, sizeof(int)*width*height);
91 buf[w/2 + h/2*stride]= 256*256;
92 ff_spatial_idwt((IDWTELEM*)buffer[0], s.temp_idwt_buffer, width, height, width, s.spatial_decomposition_type, s.spatial_decomposition_count);
93 for(y=0; y<height; y++){
94 for(x=0; x<width; x++){
95 int64_t d= buffer[0][x + y*width];
97 if(FFABS(width/2-x)<9 && FFABS(height/2-y)<9 && level==2) printf("%8"PRId64" ", d);
99 if(FFABS(height/2-y)<9 && level==2) printf("\n");
101 error= (int)(sqrt(error)+0.5);
102 errors[level][orientation]= error;
103 if(g) g=av_gcd(g, error);
107 printf("static int const visual_weight[][4]={\n");
108 for(level=0; level<s.spatial_decomposition_count; level++){
110 for(orientation=0; orientation<4; orientation++){
111 printf("%8"PRId64",", errors[level][orientation]/g);
118 int w= width >> (s.spatial_decomposition_count-level);
119 //int h= height >> (s.spatial_decomposition_count-level);
120 int stride= width << (s.spatial_decomposition_count-level);
121 DWTELEM *buf= buffer[0];
127 memset(buffer[0], 0, sizeof(int)*width*height);
128 for(y=0; y<height; y++){
129 for(x=0; x<width; x++){
130 int tab[4]={0,2,3,1};
131 buffer[0][x+width*y]= 256*256*tab[(x&1) + 2*(y&1)];
134 ff_spatial_dwt(buffer[0], s.temp_dwt_buffer, width, height, width, s.spatial_decomposition_type, s.spatial_decomposition_count);
135 for(y=0; y<height; y++){
136 for(x=0; x<width; x++){
137 int64_t d= buffer[0][x + y*width];
139 if(FFABS(width/2-x)<9 && FFABS(height/2-y)<9) printf("%8"PRId64" ", d);
141 if(FFABS(height/2-y)<9) printf("\n");