| /* |
| * Copyright (C) 2004 Michael Niedermayer <michaelni@gmx.at> |
| * |
| * 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/snowenc.c" |
| |
| #undef malloc |
| #undef free |
| #undef printf |
| |
| #include "libavutil/lfg.h" |
| #include "libavutil/mathematics.h" |
| |
| int main(void){ |
| #define width 256 |
| #define height 256 |
| int buffer[2][width*height]; |
| SnowContext s; |
| int i; |
| AVLFG prng; |
| s.spatial_decomposition_count=6; |
| s.spatial_decomposition_type=1; |
| |
| s.temp_dwt_buffer = av_mallocz_array(width, sizeof(DWTELEM)); |
| s.temp_idwt_buffer = av_mallocz_array(width, sizeof(IDWTELEM)); |
| |
| if (!s.temp_dwt_buffer || !s.temp_idwt_buffer) { |
| fprintf(stderr, "Failed to allocate memory\n"); |
| return 1; |
| } |
| |
| av_lfg_init(&prng, 1); |
| |
| printf("testing 5/3 DWT\n"); |
| for(i=0; i<width*height; i++) |
| buffer[0][i] = buffer[1][i] = av_lfg_get(&prng) % 54321 - 12345; |
| |
| ff_spatial_dwt(buffer[0], s.temp_dwt_buffer, width, height, width, s.spatial_decomposition_type, s.spatial_decomposition_count); |
| ff_spatial_idwt((IDWTELEM*)buffer[0], s.temp_idwt_buffer, width, height, width, s.spatial_decomposition_type, s.spatial_decomposition_count); |
| |
| for(i=0; i<width*height; i++) |
| if(buffer[0][i]!= buffer[1][i]) printf("fsck: %6d %12d %7d\n",i, buffer[0][i], buffer[1][i]); |
| |
| printf("testing 9/7 DWT\n"); |
| s.spatial_decomposition_type=0; |
| for(i=0; i<width*height; i++) |
| buffer[0][i] = buffer[1][i] = av_lfg_get(&prng) % 54321 - 12345; |
| |
| ff_spatial_dwt(buffer[0], s.temp_dwt_buffer, width, height, width, s.spatial_decomposition_type, s.spatial_decomposition_count); |
| ff_spatial_idwt((IDWTELEM*)buffer[0], s.temp_idwt_buffer, width, height, width, s.spatial_decomposition_type, s.spatial_decomposition_count); |
| |
| for(i=0; i<width*height; i++) |
| if(FFABS(buffer[0][i] - buffer[1][i])>20) printf("fsck: %6d %12d %7d\n",i, buffer[0][i], buffer[1][i]); |
| |
| { |
| int level, orientation, x, y; |
| int64_t errors[8][4]; |
| int64_t g=0; |
| |
| memset(errors, 0, sizeof(errors)); |
| s.spatial_decomposition_count=3; |
| s.spatial_decomposition_type=0; |
| for(level=0; level<s.spatial_decomposition_count; level++){ |
| for(orientation=level ? 1 : 0; orientation<4; orientation++){ |
| int w= width >> (s.spatial_decomposition_count-level); |
| int h= height >> (s.spatial_decomposition_count-level); |
| int stride= width << (s.spatial_decomposition_count-level); |
| DWTELEM *buf= buffer[0]; |
| int64_t error=0; |
| |
| if(orientation&1) buf+=w; |
| if(orientation>1) buf+=stride>>1; |
| |
| memset(buffer[0], 0, sizeof(int)*width*height); |
| buf[w/2 + h/2*stride]= 256*256; |
| ff_spatial_idwt((IDWTELEM*)buffer[0], s.temp_idwt_buffer, width, height, width, s.spatial_decomposition_type, s.spatial_decomposition_count); |
| for(y=0; y<height; y++){ |
| for(x=0; x<width; x++){ |
| int64_t d= buffer[0][x + y*width]; |
| error += d*d; |
| if(FFABS(width/2-x)<9 && FFABS(height/2-y)<9 && level==2) printf("%8"PRId64" ", d); |
| } |
| if(FFABS(height/2-y)<9 && level==2) printf("\n"); |
| } |
| error= (int)(sqrt(error)+0.5); |
| errors[level][orientation]= error; |
| if(g) g=av_gcd(g, error); |
| else g= error; |
| } |
| } |
| printf("static int const visual_weight[][4]={\n"); |
| for(level=0; level<s.spatial_decomposition_count; level++){ |
| printf(" {"); |
| for(orientation=0; orientation<4; orientation++){ |
| printf("%8"PRId64",", errors[level][orientation]/g); |
| } |
| printf("},\n"); |
| } |
| printf("};\n"); |
| { |
| int level=2; |
| int w= width >> (s.spatial_decomposition_count-level); |
| //int h= height >> (s.spatial_decomposition_count-level); |
| int stride= width << (s.spatial_decomposition_count-level); |
| DWTELEM *buf= buffer[0]; |
| int64_t error=0; |
| |
| buf+=w; |
| buf+=stride>>1; |
| |
| memset(buffer[0], 0, sizeof(int)*width*height); |
| for(y=0; y<height; y++){ |
| for(x=0; x<width; x++){ |
| int tab[4]={0,2,3,1}; |
| buffer[0][x+width*y]= 256*256*tab[(x&1) + 2*(y&1)]; |
| } |
| } |
| ff_spatial_dwt(buffer[0], s.temp_dwt_buffer, width, height, width, s.spatial_decomposition_type, s.spatial_decomposition_count); |
| for(y=0; y<height; y++){ |
| for(x=0; x<width; x++){ |
| int64_t d= buffer[0][x + y*width]; |
| error += d*d; |
| if(FFABS(width/2-x)<9 && FFABS(height/2-y)<9) printf("%8"PRId64" ", d); |
| } |
| if(FFABS(height/2-y)<9) printf("\n"); |
| } |
| } |
| |
| } |
| return 0; |
| } |