Mercurial > mplayer.hg

comparison libswscale/swscale-example.c @ 23155:1befc8d767cd

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
cosmetics attack, part II: Remove all tabs and prettyprint/reindent the code.
author diego
date Sun, 29 Apr 2007 13:39:27 +0000
parents 9528d1ebe68f
children 5159b5f8784e
comparison
equal deleted inserted replaced
23154:e564b9cd7290 23155:1befc8d767cd
29 #include "swscale.h" 29 #include "swscale.h"
30 #include "swscale_internal.h" 30 #include "swscale_internal.h"
31 #include "rgb2rgb.h" 31 #include "rgb2rgb.h"
32 32
33 static uint64_t getSSD(uint8_t *src1, uint8_t *src2, int stride1, int stride2, int w, int h){ 33 static uint64_t getSSD(uint8_t *src1, uint8_t *src2, int stride1, int stride2, int w, int h){
34 int x,y; 34 int x,y;
35 uint64_t ssd=0; 35 uint64_t ssd=0;
36 36
37 //printf("%d %d\n", w, h); 37 //printf("%d %d\n", w, h);
38 38
39 for(y=0; y<h; y++){ 39 for (y=0; y<h; y++){
40 for(x=0; x<w; x++){ 40 for (x=0; x<w; x++){
41 int d= src1[x + y*stride1] - src2[x + y*stride2]; 41 int d= src1[x + y*stride1] - src2[x + y*stride2];
42 ssd+= d*d; 42 ssd+= d*d;
43 //printf("%d", abs(src1[x + y*stride1] - src2[x + y*stride2])/26 ); 43 //printf("%d", abs(src1[x + y*stride1] - src2[x + y*stride2])/26 );
44 } 44 }
45 //printf("\n"); 45 //printf("\n");
46 } 46 }
47 return ssd; 47 return ssd;
48 } 48 }
49 49
50 // test by ref -> src -> dst -> out & compare out against ref 50 // test by ref -> src -> dst -> out & compare out against ref
51 // ref & out are YV12 51 // ref & out are YV12
52 static int doTest(uint8_t *ref[3], int refStride[3], int w, int h, int srcFormat, int dstFormat, 52 static int doTest(uint8_t *ref[3], int refStride[3], int w, int h, int srcFormat, int dstFormat,
53 int srcW, int srcH, int dstW, int dstH, int flags){ 53 int srcW, int srcH, int dstW, int dstH, int flags){
54 uint8_t *src[3]; 54 uint8_t *src[3];
55 uint8_t *dst[3]; 55 uint8_t *dst[3];
56 uint8_t *out[3]; 56 uint8_t *out[3];
57 int srcStride[3], dstStride[3]; 57 int srcStride[3], dstStride[3];
58 int i; 58 int i;
59 uint64_t ssdY, ssdU, ssdV; 59 uint64_t ssdY, ssdU, ssdV;
60 struct SwsContext *srcContext, *dstContext, *outContext; 60 struct SwsContext *srcContext, *dstContext, *outContext;
61 int res; 61 int res;
62 62
63 res = 0; 63 res = 0;
64 for(i=0; i<3; i++){ 64 for (i=0; i<3; i++){
65 // avoid stride % bpp != 0 65 // avoid stride % bpp != 0
66 if(srcFormat==PIX_FMT_RGB24 || srcFormat==PIX_FMT_BGR24) 66 if (srcFormat==PIX_FMT_RGB24 || srcFormat==PIX_FMT_BGR24)
67 srcStride[i]= srcW*3; 67 srcStride[i]= srcW*3;
68 else 68 else
69 srcStride[i]= srcW*4; 69 srcStride[i]= srcW*4;
70 70
71 if(dstFormat==PIX_FMT_RGB24 || dstFormat==PIX_FMT_BGR24) 71 if (dstFormat==PIX_FMT_RGB24 || dstFormat==PIX_FMT_BGR24)
72 dstStride[i]= dstW*3; 72 dstStride[i]= dstW*3;
73 else 73 else
74 dstStride[i]= dstW*4; 74 dstStride[i]= dstW*4;
75 75
76 src[i]= (uint8_t*) malloc(srcStride[i]*srcH); 76 src[i]= (uint8_t*) malloc(srcStride[i]*srcH);
77 dst[i]= (uint8_t*) malloc(dstStride[i]*dstH); 77 dst[i]= (uint8_t*) malloc(dstStride[i]*dstH);
78 out[i]= (uint8_t*) malloc(refStride[i]*h); 78 out[i]= (uint8_t*) malloc(refStride[i]*h);
79 if ((src[i] == NULL) || (dst[i] == NULL) || (out[i] == NULL)) { 79 if ((src[i] == NULL) || (dst[i] == NULL) || (out[i] == NULL)) {
80 perror("Malloc"); 80 perror("Malloc");
81 res = -1; 81 res = -1;
82 82
83 goto end; 83 goto end;
84 } 84 }
85 } 85 }
86 86
87 dstContext = outContext = NULL; 87 dstContext = outContext = NULL;
88 srcContext= sws_getContext(w, h, PIX_FMT_YUV420P, srcW, srcH, srcFormat, flags, NULL, NULL, NULL); 88 srcContext= sws_getContext(w, h, PIX_FMT_YUV420P, srcW, srcH, srcFormat, flags, NULL, NULL, NULL);
89 if (srcContext == NULL) { 89 if (srcContext == NULL) {
90 fprintf(stderr, "Failed to get %s ---> %s\n", 90 fprintf(stderr, "Failed to get %s ---> %s\n",
91 sws_format_name(PIX_FMT_YUV420P), 91 sws_format_name(PIX_FMT_YUV420P),
92 sws_format_name(srcFormat)); 92 sws_format_name(srcFormat));
93 res = -1; 93 res = -1;
94 94
95 goto end; 95 goto end;
96 } 96 }
97 dstContext= sws_getContext(srcW, srcH, srcFormat, dstW, dstH, dstFormat, flags, NULL, NULL, NULL); 97 dstContext= sws_getContext(srcW, srcH, srcFormat, dstW, dstH, dstFormat, flags, NULL, NULL, NULL);
98 if (dstContext == NULL) { 98 if (dstContext == NULL) {
99 fprintf(stderr, "Failed to get %s ---> %s\n", 99 fprintf(stderr, "Failed to get %s ---> %s\n",
100 sws_format_name(srcFormat), 100 sws_format_name(srcFormat),
101 sws_format_name(dstFormat)); 101 sws_format_name(dstFormat));
102 res = -1; 102 res = -1;
103 103
104 goto end; 104 goto end;
105 } 105 }
106 outContext= sws_getContext(dstW, dstH, dstFormat, w, h, PIX_FMT_YUV420P, flags, NULL, NULL, NULL); 106 outContext= sws_getContext(dstW, dstH, dstFormat, w, h, PIX_FMT_YUV420P, flags, NULL, NULL, NULL);
107 if (outContext == NULL) { 107 if (outContext == NULL) {
108 fprintf(stderr, "Failed to get %s ---> %s\n", 108 fprintf(stderr, "Failed to get %s ---> %s\n",
109 sws_format_name(dstFormat), 109 sws_format_name(dstFormat),
110 sws_format_name(PIX_FMT_YUV420P)); 110 sws_format_name(PIX_FMT_YUV420P));
111 res = -1; 111 res = -1;
112 112
113 goto end; 113 goto end;
114 } 114 }
115 // printf("test %X %X %X -> %X %X %X\n", (int)ref[0], (int)ref[1], (int)ref[2], 115 // printf("test %X %X %X -> %X %X %X\n", (int)ref[0], (int)ref[1], (int)ref[2],
116 // (int)src[0], (int)src[1], (int)src[2]); 116 // (int)src[0], (int)src[1], (int)src[2]);
117 117
118 sws_scale(srcContext, ref, refStride, 0, h , src, srcStride); 118 sws_scale(srcContext, ref, refStride, 0, h , src, srcStride);
119 sws_scale(dstContext, src, srcStride, 0, srcH, dst, dstStride); 119 sws_scale(dstContext, src, srcStride, 0, srcH, dst, dstStride);
120 sws_scale(outContext, dst, dstStride, 0, dstH, out, refStride); 120 sws_scale(outContext, dst, dstStride, 0, dstH, out, refStride);
121 121
122 #if defined(ARCH_X86) 122 #if defined(ARCH_X86)
123 asm volatile ("emms\n\t"); 123 asm volatile ("emms\n\t");
124 #endif 124 #endif
125 125
126 ssdY= getSSD(ref[0], out[0], refStride[0], refStride[0], w, h); 126 ssdY= getSSD(ref[0], out[0], refStride[0], refStride[0], w, h);
127 ssdU= getSSD(ref[1], out[1], refStride[1], refStride[1], (w+1)>>1, (h+1)>>1); 127 ssdU= getSSD(ref[1], out[1], refStride[1], refStride[1], (w+1)>>1, (h+1)>>1);
128 ssdV= getSSD(ref[2], out[2], refStride[2], refStride[2], (w+1)>>1, (h+1)>>1); 128 ssdV= getSSD(ref[2], out[2], refStride[2], refStride[2], (w+1)>>1, (h+1)>>1);
129 129
130 if(srcFormat == PIX_FMT_GRAY8 || dstFormat==PIX_FMT_GRAY8) ssdU=ssdV=0; //FIXME check that output is really gray 130 if (srcFormat == PIX_FMT_GRAY8 || dstFormat==PIX_FMT_GRAY8) ssdU=ssdV=0; //FIXME check that output is really gray
131 131
132 ssdY/= w*h; 132 ssdY/= w*h;
133 ssdU/= w*h/4; 133 ssdU/= w*h/4;
134 ssdV/= w*h/4; 134 ssdV/= w*h/4;
135 135
136 if(ssdY>100 || ssdU>100 || ssdV>100){ 136 if (ssdY>100 || ssdU>100 || ssdV>100){
137 printf(" %s %dx%d -> %s %4dx%4d flags=%2d SSD=%5lld,%5lld,%5lld\n", 137 printf(" %s %dx%d -> %s %4dx%4d flags=%2d SSD=%5lld,%5lld,%5lld\n",
138 sws_format_name(srcFormat), srcW, srcH, 138 sws_format_name(srcFormat), srcW, srcH,
139 sws_format_name(dstFormat), dstW, dstH, 139 sws_format_name(dstFormat), dstW, dstH,
140 flags, 140 flags,
141 ssdY, ssdU, ssdV); 141 ssdY, ssdU, ssdV);
142 } 142 }
143 143
144 end: 144 end:
145 145
146 sws_freeContext(srcContext); 146 sws_freeContext(srcContext);
147 sws_freeContext(dstContext); 147 sws_freeContext(dstContext);
148 sws_freeContext(outContext); 148 sws_freeContext(outContext);
149 149
150 for(i=0; i<3; i++){ 150 for (i=0; i<3; i++){
151 free(src[i]); 151 free(src[i]);
152 free(dst[i]); 152 free(dst[i]);
153 free(out[i]); 153 free(out[i]);
154 } 154 }
155 155
156 return res; 156 return res;
157 } 157 }
158 158
159 void fast_memcpy(void *a, void *b, int s){ //FIXME 159 void fast_memcpy(void *a, void *b, int s){ //FIXME
160 memcpy(a, b, s); 160 memcpy(a, b, s);
161 } 161 }
162 162
163 static void selfTest(uint8_t *src[3], int stride[3], int w, int h){ 163 static void selfTest(uint8_t *src[3], int stride[3], int w, int h){
164 enum PixelFormat srcFormat, dstFormat; 164 enum PixelFormat srcFormat, dstFormat;
165 int srcW, srcH, dstW, dstH; 165 int srcW, srcH, dstW, dstH;
166 int flags; 166 int flags;
167 167
168 for(srcFormat = 0; srcFormat < PIX_FMT_NB; srcFormat++) { 168 for (srcFormat = 0; srcFormat < PIX_FMT_NB; srcFormat++) {
169 for(dstFormat = 0; dstFormat < PIX_FMT_NB; dstFormat++) { 169 for (dstFormat = 0; dstFormat < PIX_FMT_NB; dstFormat++) {
170 printf("%s -> %s\n", 170 printf("%s -> %s\n",
171 sws_format_name(srcFormat), 171 sws_format_name(srcFormat),
172 sws_format_name(dstFormat)); 172 sws_format_name(dstFormat));
173 173
174 srcW= w; 174 srcW= w;
175 srcH= h; 175 srcH= h;
176 for(dstW=w - w/3; dstW<= 4*w/3; dstW+= w/3){ 176 for (dstW=w - w/3; dstW<= 4*w/3; dstW+= w/3){
177 for(dstH=h - h/3; dstH<= 4*h/3; dstH+= h/3){ 177 for (dstH=h - h/3; dstH<= 4*h/3; dstH+= h/3){
178 for(flags=1; flags<33; flags*=2) { 178 for (flags=1; flags<33; flags*=2) {
179 int res; 179 int res;
180 180
181 res = doTest(src, stride, w, h, srcFormat, dstFormat, 181 res = doTest(src, stride, w, h, srcFormat, dstFormat,
182 srcW, srcH, dstW, dstH, flags); 182 srcW, srcH, dstW, dstH, flags);
183 if (res < 0) { 183 if (res < 0) {
184 dstW = 4 * w / 3; 184 dstW = 4 * w / 3;
185 dstH = 4 * h / 3; 185 dstH = 4 * h / 3;
186 flags = 33; 186 flags = 33;
187 } 187 }
188 } 188 }
189 } 189 }
190 } 190 }
191 } 191 }
192 } 192 }
193 } 193 }
194 194
195 #define W 96 195 #define W 96
196 #define H 96 196 #define H 96
197 197
198 int main(int argc, char **argv){ 198 int main(int argc, char **argv){
199 uint8_t *rgb_data = malloc (W*H*4); 199 uint8_t *rgb_data = malloc (W*H*4);
200 uint8_t *rgb_src[3]= {rgb_data, NULL, NULL}; 200 uint8_t *rgb_src[3]= {rgb_data, NULL, NULL};
201 int rgb_stride[3]={4*W, 0, 0}; 201 int rgb_stride[3]={4*W, 0, 0};
202 uint8_t *data = malloc (3*W*H); 202 uint8_t *data = malloc (3*W*H);
203 uint8_t *src[3]= {data, data+W*H, data+W*H*2}; 203 uint8_t *src[3]= {data, data+W*H, data+W*H*2};
204 int stride[3]={W, W, W}; 204 int stride[3]={W, W, W};
205 int x, y; 205 int x, y;
206 struct SwsContext *sws; 206 struct SwsContext *sws;
207 207
208 sws= sws_getContext(W/12, H/12, PIX_FMT_RGB32, W, H, PIX_FMT_YUV420P, 2, NULL, NULL, NULL); 208 sws= sws_getContext(W/12, H/12, PIX_FMT_RGB32, W, H, PIX_FMT_YUV420P, 2, NULL, NULL, NULL);
209 209
210 for(y=0; y<H; y++){ 210 for (y=0; y<H; y++){
211 for(x=0; x<W*4; x++){ 211 for (x=0; x<W*4; x++){
212 rgb_data[ x + y*4*W]= random(); 212 rgb_data[ x + y*4*W]= random();
213 } 213 }
214 } 214 }
215 #if defined(ARCH_X86) 215 #if defined(ARCH_X86)
216 sws_rgb2rgb_init(SWS_CPU_CAPS_MMX*0); 216 sws_rgb2rgb_init(SWS_CPU_CAPS_MMX*0);
217 #else 217 #else
218 sws_rgb2rgb_init(0); 218 sws_rgb2rgb_init(0);
219 #endif 219 #endif
220 sws_scale(sws, rgb_src, rgb_stride, 0, H , src, stride); 220 sws_scale(sws, rgb_src, rgb_stride, 0, H , src, stride);
221 221
222 #if defined(ARCH_X86) 222 #if defined(ARCH_X86)
223 asm volatile ("emms\n\t"); 223 asm volatile ("emms\n\t");
224 #endif 224 #endif
225 225
226 selfTest(src, stride, W, H); 226 selfTest(src, stride, W, H);
227 227
228 return 123; 228 return 123;
229 } 229 }