[svn] / branches / dev-api-4 / xvidcore / vfw / src / codec.c Repository:
ViewVC logotype

Annotation of /branches/dev-api-4/xvidcore/vfw/src/codec.c

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1202 - (view) (download)

1 : suxen_drol 889 /**************************************************************************
2 :     *
3 :     * XVID VFW FRONTEND
4 :     * codec
5 :     *
6 :     * This program is free software; you can redistribute it and/or modify
7 :     * it under the terms of the GNU General Public License as published by
8 :     * the Free Software Foundation; either version 2 of the License, or
9 :     * (at your option) any later version.
10 :     *
11 :     * This program 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
14 :     * GNU General Public License for more details.
15 :     *
16 :     * You should have received a copy of the GNU General Public License
17 :     * along with this program; if not, write to the Free Software
18 :     * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
19 :     *
20 :     *************************************************************************/
21 :    
22 :     /**************************************************************************
23 :     *
24 :     * History:
25 :     *
26 :     * 12.07.2002 num_threads
27 :     * 23.06.2002 XVID_CPU_CHKONLY; loading speed up
28 :     * 25.04.2002 ICDECOMPRESS_PREROLL
29 :     * 17.04.2002 re-enabled lumi masking for 1st pass
30 :     * 15.04.2002 updated cbr support
31 :     * 04.04.2002 separated 2-pass code to 2pass.c
32 :     * interlacing support
33 :     * hinted ME support
34 :     * 23.03.2002 daniel smith <danielsmith@astroboymail.com>
35 :     * changed inter4v to only be in modes 5 or 6
36 :     * fixed null mode crash ?
37 :     * merged foxer's alternative 2-pass code
38 :     * added DEBUGERR output on errors instead of returning
39 :     * 16.03.2002 daniel smith <danielsmith@astroboymail.com>
40 :     * changed BITMAPV4HEADER to BITMAPINFOHEADER
41 :     * - prevents memcpy crash in compress_get_format()
42 :     * credits are processed in external 2pass mode
43 :     * motion search precision = 0 now effective in 2-pass
44 :     * modulated quantization
45 :     * added DX50 fourcc
46 :     * 01.12.2001 inital version; (c)2001 peter ross <pross@xvid.org>
47 :     *
48 :     *************************************************************************/
49 :    
50 :     #include <windows.h>
51 :     #include <vfw.h>
52 : suxen_drol 1017 #include "vfwext.h"
53 : suxen_drol 889
54 : suxen_drol 983 #include <xvid.h>
55 :     #include "debug.h"
56 : suxen_drol 889 #include "codec.h"
57 : suxen_drol 1061 #include "status.h"
58 : suxen_drol 889
59 : suxen_drol 983
60 :     static const int pmvfast_presets[7] = {
61 : suxen_drol 889 0, 0, 0, 0,
62 : edgomez 949 0 | XVID_ME_HALFPELREFINE16 | 0,
63 :     0 | XVID_ME_HALFPELREFINE16 | 0 |
64 :     XVID_ME_ADVANCEDDIAMOND16, XVID_ME_HALFPELREFINE16 | XVID_ME_EXTSEARCH16 |
65 :     XVID_ME_HALFPELREFINE8 | 0 | XVID_ME_USESQUARES16
66 : suxen_drol 889 };
67 :    
68 : suxen_drol 983
69 :    
70 : suxen_drol 889 /* return xvid compatbile colorspace,
71 :     or XVID_CSP_NULL if failure
72 :     */
73 :    
74 :     int get_colorspace(BITMAPINFOHEADER * hdr)
75 :     {
76 :     /* rgb only: negative height specifies top down image */
77 :     int rgb_flip = (hdr->biHeight < 0 ? 0 : XVID_CSP_VFLIP);
78 :    
79 :     switch(hdr->biCompression)
80 :     {
81 :     case BI_RGB :
82 :     if (hdr->biBitCount == 16)
83 :     {
84 : suxen_drol 983 DPRINTF("RGB16 (RGB555)");
85 : suxen_drol 889 return rgb_flip | XVID_CSP_RGB555;
86 :     }
87 :     if (hdr->biBitCount == 24)
88 :     {
89 : suxen_drol 983 DPRINTF("RGB24");
90 : suxen_drol 889 return rgb_flip | XVID_CSP_BGR;
91 :     }
92 :     if (hdr->biBitCount == 32)
93 :     {
94 : suxen_drol 983 DPRINTF("RGB32");
95 : suxen_drol 889 return rgb_flip | XVID_CSP_BGRA;
96 :     }
97 :    
98 : suxen_drol 983 DPRINTF("unsupported BI_RGB biBitCount=%i", hdr->biBitCount);
99 : suxen_drol 889 return XVID_CSP_NULL;
100 :    
101 :     case BI_BITFIELDS :
102 :     if (hdr->biSize >= sizeof(BITMAPV4HEADER))
103 :     {
104 :     BITMAPV4HEADER * hdr4 = (BITMAPV4HEADER *)hdr;
105 :    
106 :     if (hdr4->bV4BitCount == 16 &&
107 :     hdr4->bV4RedMask == 0x7c00 &&
108 :     hdr4->bV4GreenMask == 0x3e0 &&
109 :     hdr4->bV4BlueMask == 0x1f)
110 :     {
111 : suxen_drol 983 DPRINTF("RGB555");
112 : suxen_drol 889 return rgb_flip | XVID_CSP_RGB555;
113 :     }
114 :    
115 :     if(hdr4->bV4BitCount == 16 &&
116 :     hdr4->bV4RedMask == 0xf800 &&
117 :     hdr4->bV4GreenMask == 0x7e0 &&
118 :     hdr4->bV4BlueMask == 0x1f)
119 :     {
120 : suxen_drol 983 DPRINTF("RGB565");
121 : suxen_drol 889 return rgb_flip | XVID_CSP_RGB565;
122 :     }
123 :    
124 : suxen_drol 983 DPRINTF("unsupported BI_BITFIELDS mode");
125 : suxen_drol 889 return XVID_CSP_NULL;
126 :     }
127 :    
128 : suxen_drol 983 DPRINTF("unsupported BI_BITFIELDS/BITMAPHEADER combination");
129 : suxen_drol 889 return XVID_CSP_NULL;
130 :    
131 :     case FOURCC_I420 :
132 :     case FOURCC_IYUV :
133 : suxen_drol 983 DPRINTF("IYUY");
134 : suxen_drol 889 return XVID_CSP_I420;
135 :    
136 :     case FOURCC_YV12 :
137 : suxen_drol 983 DPRINTF("YV12");
138 : suxen_drol 889 return XVID_CSP_YV12;
139 :    
140 :     case FOURCC_YUYV :
141 :     case FOURCC_YUY2 :
142 : suxen_drol 983 DPRINTF("YUY2");
143 : suxen_drol 889 return XVID_CSP_YUY2;
144 :    
145 :     case FOURCC_YVYU :
146 : suxen_drol 983 DPRINTF("YVYU");
147 : suxen_drol 889 return XVID_CSP_YVYU;
148 :    
149 :     case FOURCC_UYVY :
150 : suxen_drol 983 DPRINTF("UYVY");
151 : suxen_drol 889 return XVID_CSP_UYVY;
152 :    
153 :     default :
154 : suxen_drol 983 DPRINTF("unsupported colorspace %c%c%c%c",
155 :     hdr->biCompression&0xff,
156 :     (hdr->biCompression>>8)&0xff,
157 :     (hdr->biCompression>>16)&0xff,
158 :     (hdr->biCompression>>24)&0xff);
159 : suxen_drol 889 return XVID_CSP_NULL;
160 :     }
161 :     }
162 :    
163 :    
164 :     /* compressor */
165 :    
166 :    
167 :     /* test the output format */
168 :    
169 :     LRESULT compress_query(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)
170 :     {
171 :     BITMAPINFOHEADER * inhdr = &lpbiInput->bmiHeader;
172 :     BITMAPINFOHEADER * outhdr = &lpbiOutput->bmiHeader;
173 :    
174 : suxen_drol 1017 /* VFWEXT detection */
175 :     if (inhdr->biCompression == VFWEXT_FOURCC) {
176 :     return (ICM_USER+0x0fff);
177 :     }
178 :    
179 : suxen_drol 889 if (get_colorspace(inhdr) == XVID_CSP_NULL)
180 :     {
181 :     return ICERR_BADFORMAT;
182 :     }
183 :    
184 :     if (lpbiOutput == NULL)
185 :     {
186 :     return ICERR_OK;
187 :     }
188 :    
189 :     if (inhdr->biWidth != outhdr->biWidth || inhdr->biHeight != outhdr->biHeight ||
190 :     (outhdr->biCompression != FOURCC_XVID && outhdr->biCompression != FOURCC_DIVX && outhdr->biCompression != FOURCC_DX50))
191 :     {
192 :     return ICERR_BADFORMAT;
193 :     }
194 :    
195 :     return ICERR_OK;
196 :     }
197 :    
198 :    
199 :     LRESULT compress_get_format(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)
200 :     {
201 :     BITMAPINFOHEADER * inhdr = &lpbiInput->bmiHeader;
202 :     BITMAPINFOHEADER * outhdr = &lpbiOutput->bmiHeader;
203 :    
204 :     if (get_colorspace(inhdr) == XVID_CSP_NULL)
205 :     {
206 :     return ICERR_BADFORMAT;
207 :     }
208 :    
209 :     if (lpbiOutput == NULL)
210 :     {
211 : syskin 1091 return sizeof(BITMAPINFOHEADER);
212 : suxen_drol 889 }
213 :    
214 :     memcpy(outhdr, inhdr, sizeof(BITMAPINFOHEADER));
215 :     outhdr->biSize = sizeof(BITMAPINFOHEADER);
216 :     outhdr->biSizeImage = compress_get_size(codec, lpbiInput, lpbiOutput);
217 :     outhdr->biXPelsPerMeter = 0;
218 :     outhdr->biYPelsPerMeter = 0;
219 :     outhdr->biClrUsed = 0;
220 :     outhdr->biClrImportant = 0;
221 :    
222 :     if (codec->config.fourcc_used == 0)
223 :     {
224 :     outhdr->biCompression = FOURCC_XVID;
225 :     }
226 :     else if (codec->config.fourcc_used == 1)
227 :     {
228 :     outhdr->biCompression = FOURCC_DIVX;
229 :     }
230 :     else
231 :     {
232 :     outhdr->biCompression = FOURCC_DX50;
233 :     }
234 :    
235 :     return ICERR_OK;
236 :     }
237 :    
238 :    
239 :     LRESULT compress_get_size(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)
240 :     {
241 :     return 2 * lpbiOutput->bmiHeader.biWidth * lpbiOutput->bmiHeader.biHeight * 3;
242 :     }
243 :    
244 :    
245 :     LRESULT compress_frames_info(CODEC * codec, ICCOMPRESSFRAMES * icf)
246 :     {
247 : edgomez 1053 #if 0
248 :     DPRINTF("%i %i", icf->lStartFrame, icf->lFrameCount);
249 :     #endif
250 : suxen_drol 889 codec->fincr = icf->dwScale;
251 :     codec->fbase = icf->dwRate;
252 :     return ICERR_OK;
253 :     }
254 :    
255 :    
256 : suxen_drol 983 const char type2char(int type)
257 :     {
258 :     if (type==XVID_TYPE_IVOP)
259 :     return 'I';
260 :     if (type==XVID_TYPE_PVOP)
261 :     return 'P';
262 :     if (type==XVID_TYPE_BVOP)
263 :     return 'B';
264 :     return 'S';
265 :     }
266 :    
267 :     int vfw_debug(void *handle,
268 :     int opt,
269 :     void *param1,
270 :     void *param2)
271 :     {
272 :     switch (opt) {
273 :     case XVID_PLG_INFO:
274 :     case XVID_PLG_CREATE:
275 :     case XVID_PLG_DESTROY:
276 :     case XVID_PLG_BEFORE:
277 :     return 0;
278 :    
279 :     case XVID_PLG_AFTER:
280 :     {
281 :     xvid_plg_data_t *data = (xvid_plg_data_t *) param1;
282 :    
283 :     DPRINTF("[%5i] type=%c Q:%2i length:%6i",
284 :     data->frame_num,
285 :     type2char(data->type),
286 :     data->quant,
287 :     data->length);
288 :     return 0;
289 :     }
290 :     }
291 :    
292 :     return XVID_ERR_FAIL;
293 :     }
294 :    
295 :    
296 :    
297 : suxen_drol 889 LRESULT compress_begin(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)
298 :     {
299 :     xvid_gbl_init_t init;
300 :     xvid_enc_create_t create;
301 : suxen_drol 983 xvid_enc_plugin_t plugins[3];
302 : suxen_drol 1017 xvid_plugin_single_t single;
303 : suxen_drol 983 xvid_plugin_2pass1_t pass1;
304 :     xvid_plugin_2pass2_t pass2;
305 : suxen_drol 1017 int i;
306 : suxen_drol 889
307 : suxen_drol 983 /* destroy previously created codec */
308 :     if(codec->ehandle) {
309 :     xvid_encore(codec->ehandle, XVID_ENC_DESTROY, NULL, NULL);
310 :     codec->ehandle = NULL;
311 :     }
312 :    
313 :     memset(&init, 0, sizeof(init));
314 :     init.version = XVID_VERSION;
315 :     init.cpu_flags = codec->config.cpu;
316 : suxen_drol 1032 init.debug = codec->config.debug;
317 : suxen_drol 983 xvid_global(0, XVID_GBL_INIT, &init, NULL);
318 :    
319 :     memset(&create, 0, sizeof(create));
320 :     create.version = XVID_VERSION;
321 : suxen_drol 1017
322 : edgomez 1053 /* zones */
323 : suxen_drol 1017 create.zones = malloc(sizeof(xvid_enc_zone_t) * codec->config.num_zones);
324 :     create.num_zones = codec->config.num_zones;
325 :     for (i=0; i < create.num_zones; i++) {
326 :     create.zones[i].frame = codec->config.zones[i].frame;
327 : suxen_drol 1032 if (codec->config.zones[i].mode == RC_ZONE_QUANT) {
328 : suxen_drol 1017 create.zones[i].mode = XVID_ZONE_QUANT;
329 :     create.zones[i].increment = codec->config.zones[i].quant;
330 :     }else{
331 :     create.zones[i].mode = XVID_ZONE_WEIGHT;
332 :     create.zones[i].increment = codec->config.zones[i].weight;
333 :     }
334 :     create.zones[i].base = 100;
335 :     }
336 :    
337 : edgomez 1053 /* plugins */
338 : suxen_drol 983 create.plugins = plugins;
339 : suxen_drol 889 switch (codec->config.mode)
340 :     {
341 : suxen_drol 1017 case RC_MODE_1PASS :
342 :     memset(&single, 0, sizeof(single));
343 :     single.version = XVID_VERSION;
344 : suxen_drol 1032 single.bitrate = codec->config.bitrate * CONFIG_KBPS;
345 : suxen_drol 1017 single.reaction_delay_factor = codec->config.rc_reaction_delay_factor;
346 :     single.averaging_period = codec->config.rc_averaging_period;
347 :     single.buffer = codec->config.rc_buffer;
348 :     plugins[create.num_plugins].func = xvid_plugin_single;
349 :     plugins[create.num_plugins].param = &single;
350 : suxen_drol 983 create.num_plugins++;
351 : suxen_drol 1032 break;
352 : suxen_drol 889
353 : suxen_drol 983 case RC_MODE_2PASS1 :
354 :     memset(&pass1, 0, sizeof(pass1));
355 :     pass1.version = XVID_VERSION;
356 : suxen_drol 1028 pass1.filename = codec->config.stats;
357 : suxen_drol 983
358 :     plugins[create.num_plugins].func = xvid_plugin_2pass1;
359 :     plugins[create.num_plugins].param = &pass1;
360 :     create.num_plugins++;
361 : suxen_drol 889 break;
362 :    
363 : suxen_drol 1017 case RC_MODE_2PASS2 :
364 : suxen_drol 983 memset(&pass2, 0, sizeof(pass2));
365 :     pass2.version = XVID_VERSION;
366 : suxen_drol 1061 if (codec->config.use_2pass_bitrate) {
367 :     pass2.bitrate = codec->config.bitrate * CONFIG_KBPS;
368 :     }else{
369 :     pass2.bitrate = -codec->config.desired_size; /* kilobytes */
370 :     }
371 : suxen_drol 983 pass2.filename = codec->config.stats;
372 : suxen_drol 1017
373 : suxen_drol 1052 pass2.keyframe_boost = codec->config.keyframe_boost; /* keyframe boost percentage: [0..100...]; */
374 :     pass2.curve_compression_high = codec->config.curve_compression_high;
375 :     pass2.curve_compression_low = codec->config.curve_compression_low;
376 : edgomez 1202 pass2.overflow_control_strength = codec->config.overflow_control_strength;
377 : suxen_drol 1052 pass2.max_overflow_improvement = codec->config.twopass_max_overflow_improvement;
378 :     pass2.max_overflow_degradation = codec->config.twopass_max_overflow_degradation;
379 :     pass2.kfreduction = codec->config.kfreduction;
380 :     pass2.min_key_interval = codec->config.min_key_interval;
381 :     pass2.container_frame_overhead = 24; /* AVI */
382 :    
383 : suxen_drol 983 plugins[create.num_plugins].func = xvid_plugin_2pass2;
384 :     plugins[create.num_plugins].param = &pass2;
385 :     create.num_plugins++;
386 : suxen_drol 889 break;
387 :    
388 : suxen_drol 983 case RC_MODE_NULL :
389 : suxen_drol 889 return ICERR_OK;
390 :    
391 :     default :
392 :     break;
393 :     }
394 :    
395 : suxen_drol 1017 if ((profiles[codec->config.profile].flags & PROFILE_ADAPTQUANT) && codec->config.lum_masking) {
396 : suxen_drol 983 plugins[create.num_plugins].func = xvid_plugin_lumimasking;
397 :     plugins[create.num_plugins].param = NULL;
398 :     create.num_plugins++;
399 : suxen_drol 889 }
400 :    
401 : suxen_drol 983 plugins[create.num_plugins].func = vfw_debug;
402 :     plugins[create.num_plugins].param = NULL;
403 :     create.num_plugins++;
404 : suxen_drol 889
405 : suxen_drol 1017 create.profile = profiles[codec->config.profile].id;
406 :    
407 : suxen_drol 889 create.width = lpbiInput->bmiHeader.biWidth;
408 :     create.height = lpbiInput->bmiHeader.biHeight;
409 :     create.fincr = codec->fincr;
410 :     create.fbase = codec->fbase;
411 :    
412 :     create.max_key_interval = codec->config.max_key_interval;
413 :    
414 : suxen_drol 1017 create.min_quant[0] = codec->config.min_iquant;
415 :     create.max_quant[0] = codec->config.max_iquant;
416 :     create.min_quant[1] = codec->config.min_pquant;
417 :     create.max_quant[1] = codec->config.max_pquant;
418 :     create.min_quant[2] = codec->config.min_bquant;
419 :     create.max_quant[2] = codec->config.max_bquant;
420 :    
421 :     if ((profiles[codec->config.profile].flags & PROFILE_BVOP) && codec->config.use_bvop) {
422 : suxen_drol 983 create.max_bframes = codec->config.max_bframes;
423 : suxen_drol 1017 create.bquant_ratio = codec->config.bquant_ratio;
424 :     create.bquant_offset = codec->config.bquant_offset;
425 :    
426 :     if (codec->config.packed)
427 :     create.global |= XVID_GLOBAL_PACKED;
428 :    
429 :     if (codec->config.closed_gov)
430 :     create.global |= XVID_GLOBAL_CLOSED_GOP;
431 :    
432 :     }
433 :    
434 : suxen_drol 889 create.frame_drop_ratio = codec->config.frame_drop_ratio;
435 :    
436 : suxen_drol 1017 create.num_threads = codec->config.num_threads;
437 : suxen_drol 889
438 : suxen_drol 983 switch(xvid_encore(0, XVID_ENC_CREATE, &create, NULL))
439 : suxen_drol 889 {
440 :     case XVID_ERR_FAIL :
441 :     return ICERR_ERROR;
442 :    
443 :     case XVID_ERR_MEMORY :
444 :     return ICERR_MEMORY;
445 :    
446 :     case XVID_ERR_FORMAT :
447 :     return ICERR_BADFORMAT;
448 :    
449 :     case XVID_ERR_VERSION :
450 :     return ICERR_UNSUPPORTED;
451 :     }
452 :    
453 :     codec->ehandle = create.handle;
454 :     codec->framenum = 0;
455 :     codec->keyspacing = 0;
456 :    
457 : suxen_drol 1061 if (codec->config.display_status) {
458 :     status_destroy_always(&codec->status);
459 :     status_create(&codec->status, codec->fincr, codec->fbase);
460 :     }
461 :    
462 : suxen_drol 889 return ICERR_OK;
463 :     }
464 :    
465 :    
466 :     LRESULT compress_end(CODEC * codec)
467 :     {
468 : suxen_drol 1061 if (codec->ehandle != NULL) {
469 : suxen_drol 889 xvid_encore(codec->ehandle, XVID_ENC_DESTROY, NULL, NULL);
470 :     codec->ehandle = NULL;
471 :     }
472 :    
473 : suxen_drol 1061 if (codec->config.display_status)
474 :     status_destroy(&codec->status);
475 :    
476 : suxen_drol 889 return ICERR_OK;
477 :     }
478 :    
479 : suxen_drol 1017
480 :     static void apply_zone_modifiers(xvid_enc_frame_t * frame, CONFIG * config, int framenum)
481 :     {
482 :     int i;
483 :    
484 :     for (i=0; i<config->num_zones && config->zones[i].frame <= framenum; i++) ;
485 :     i--;
486 :    
487 : suxen_drol 1069 frame->type = config->zones[i].type;
488 :    
489 : suxen_drol 1017 if (config->zones[i].greyscale) {
490 :     frame->vop_flags |= XVID_VOP_GREYSCALE;
491 :     }
492 :    
493 :     if (config->zones[i].chroma_opt) {
494 :     frame->vop_flags |= XVID_VOP_CHROMAOPT;
495 :     }
496 :    
497 :     if ((profiles[config->profile].flags & PROFILE_BVOP) && config->use_bvop) {
498 :     frame->bframe_threshold = config->zones[i].bvop_threshold;
499 :     }
500 :     }
501 :    
502 :    
503 : suxen_drol 889 LRESULT compress(CODEC * codec, ICCOMPRESS * icc)
504 :     {
505 :     BITMAPINFOHEADER * inhdr = icc->lpbiInput;
506 :     BITMAPINFOHEADER * outhdr = icc->lpbiOutput;
507 :     xvid_enc_frame_t frame;
508 :     xvid_enc_stats_t stats;
509 :     int length;
510 :    
511 :     memset(&frame, 0, sizeof(frame));
512 :     frame.version = XVID_VERSION;
513 :    
514 :     frame.type = XVID_TYPE_AUTO;
515 :    
516 :     /* vol stuff */
517 :    
518 : suxen_drol 1017 if ((profiles[codec->config.profile].flags & PROFILE_MPEGQUANT) &&
519 :     codec->config.quant_type != QUANT_MODE_H263)
520 : suxen_drol 889 {
521 : edgomez 949 frame.vol_flags |= XVID_VOL_MPEGQUANT;
522 : suxen_drol 889
523 : suxen_drol 1017 if (codec->config.quant_type == QUANT_MODE_CUSTOM) {
524 : suxen_drol 889 frame.quant_intra_matrix = codec->config.qmatrix_intra;
525 :     frame.quant_inter_matrix = codec->config.qmatrix_inter;
526 : suxen_drol 1017 }else{
527 : suxen_drol 889 frame.quant_intra_matrix = NULL;
528 :     frame.quant_inter_matrix = NULL;
529 :     }
530 :     }
531 :    
532 : syskin 1087 if ((profiles[codec->config.profile].flags & PROFILE_REDUCED) &&
533 : suxen_drol 1017 codec->config.reduced_resolution) {
534 : edgomez 949 frame.vol_flags |= XVID_VOL_REDUCED_ENABLE;
535 :     frame.vop_flags |= XVID_VOP_REDUCED; /* XXX: need auto decion mode */
536 : suxen_drol 889 }
537 :    
538 : suxen_drol 1017 if ((profiles[codec->config.profile].flags & PROFILE_QPEL) && codec->config.qpel) {
539 : edgomez 949 frame.vol_flags |= XVID_VOL_QUARTERPEL;
540 :     frame.motion |= XVID_ME_QUARTERPELREFINE16 | XVID_ME_QUARTERPELREFINE8;
541 : suxen_drol 889 }
542 :    
543 : syskin 1087 if ((profiles[codec->config.profile].flags & PROFILE_GMC) && codec->config.gmc) {
544 : edgomez 949 frame.vol_flags |= XVID_VOL_GMC;
545 : edgomez 1107 frame.motion |= XVID_ME_GME_REFINE;
546 : syskin 1087 }
547 : suxen_drol 889
548 : suxen_drol 1017 if ((profiles[codec->config.profile].flags & PROFILE_INTERLACE) && codec->config.interlacing)
549 : edgomez 949 frame.vol_flags |= XVID_VOL_INTERLACING;
550 : suxen_drol 889
551 :     /* vop stuff */
552 :    
553 : edgomez 949 frame.vop_flags |= XVID_VOP_HALFPEL;
554 :     frame.vop_flags |= XVID_VOP_HQACPRED;
555 : suxen_drol 889
556 : suxen_drol 1032 if (codec->config.vop_debug)
557 : edgomez 949 frame.vop_flags |= XVID_VOP_DEBUG;
558 : suxen_drol 889
559 : suxen_drol 1017 if (codec->config.trellis_quant) {
560 :     frame.vop_flags |= XVID_VOP_TRELLISQUANT;
561 :     }
562 :    
563 :     if (codec->config.motion_search > 4)
564 : edgomez 949 frame.vop_flags |= XVID_VOP_INTER4V;
565 : suxen_drol 889
566 :     if (codec->config.chromame)
567 : edgomez 1107 frame.motion |= XVID_ME_CHROMA_PVOP + XVID_ME_CHROMA_BVOP;
568 : suxen_drol 889
569 : edgomez 1193 if (codec->config.cartoon_mode) {
570 :     frame.vop_flags |= XVID_VOP_CARTOON;
571 :     #if 0 /* Seems to cause crashes with P4 cpus */
572 :     frame.motion |= XVID_ME_DETECT_STATIC_MOTION;
573 :     #endif
574 :     }
575 :    
576 : suxen_drol 889 frame.motion |= pmvfast_presets[codec->config.motion_search];
577 :    
578 :     switch (codec->config.vhq_mode)
579 :     {
580 :     case VHQ_MODE_DECISION :
581 : edgomez 1107 frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
582 : suxen_drol 889 break;
583 :    
584 :     case VHQ_LIMITED_SEARCH :
585 : edgomez 1107 frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
586 :     frame.motion |= XVID_ME_HALFPELREFINE16_RD;
587 :     frame.motion |= XVID_ME_QUARTERPELREFINE16_RD;
588 : suxen_drol 889 break;
589 :    
590 :     case VHQ_MEDIUM_SEARCH :
591 : edgomez 1107 frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
592 :     frame.motion |= XVID_ME_HALFPELREFINE16_RD;
593 :     frame.motion |= XVID_ME_HALFPELREFINE8_RD;
594 :     frame.motion |= XVID_ME_QUARTERPELREFINE16_RD;
595 :     frame.motion |= XVID_ME_QUARTERPELREFINE8_RD;
596 :     frame.motion |= XVID_ME_CHECKPREDICTION_RD;
597 : suxen_drol 889 break;
598 :    
599 :     case VHQ_WIDE_SEARCH :
600 : edgomez 1107 frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
601 :     frame.motion |= XVID_ME_HALFPELREFINE16_RD;
602 :     frame.motion |= XVID_ME_HALFPELREFINE8_RD;
603 :     frame.motion |= XVID_ME_QUARTERPELREFINE16_RD;
604 :     frame.motion |= XVID_ME_QUARTERPELREFINE8_RD;
605 :     frame.motion |= XVID_ME_CHECKPREDICTION_RD;
606 :     frame.motion |= XVID_ME_EXTSEARCH_RD;
607 : suxen_drol 889 break;
608 :    
609 :     default :
610 :     break;
611 :     }
612 :    
613 :     frame.input.plane[0] = icc->lpInput;
614 :     frame.input.stride[0] = (((icc->lpbiInput->biWidth * icc->lpbiInput->biBitCount) + 31) & ~31) >> 3;
615 :    
616 :     if ((frame.input.csp = get_colorspace(inhdr)) == XVID_CSP_NULL)
617 :     return ICERR_BADFORMAT;
618 :    
619 :     if (frame.input.csp == XVID_CSP_I420 || frame.input.csp == XVID_CSP_YV12)
620 :     frame.input.stride[0] = (frame.input.stride[0]*2)/3;
621 :    
622 :     frame.bitstream = icc->lpOutput;
623 :     frame.length = icc->lpbiOutput->biSizeImage;
624 :    
625 : suxen_drol 1017 frame.quant = 0;
626 : suxen_drol 889
627 : suxen_drol 1017 if (codec->config.mode == RC_MODE_NULL) {
628 : suxen_drol 889 outhdr->biSizeImage = 0;
629 :     *icc->lpdwFlags = AVIIF_KEYFRAME;
630 :     return ICERR_OK;
631 : suxen_drol 1017 }
632 : suxen_drol 889
633 :     // force keyframe spacing in 2-pass 1st pass
634 :     if (codec->config.motion_search == 0)
635 :     {
636 :     frame.type = XVID_TYPE_IVOP;
637 :     }
638 :     else if (codec->keyspacing < codec->config.min_key_interval && codec->framenum)
639 :     {
640 : suxen_drol 983 DPRINTF("current frame forced to p-frame");
641 : suxen_drol 889 frame.type = XVID_TYPE_PVOP;
642 :     }
643 :    
644 : suxen_drol 1017 /* frame-based stuff */
645 :     apply_zone_modifiers(&frame, &codec->config, codec->framenum);
646 :    
647 :     /* call encore */
648 :    
649 : suxen_drol 889 memset(&stats, 0, sizeof(stats));
650 :     stats.version = XVID_VERSION;
651 :    
652 :     length = xvid_encore(codec->ehandle, XVID_ENC_ENCODE, &frame, &stats);
653 :     switch (length)
654 :     {
655 :     case XVID_ERR_FAIL :
656 :     return ICERR_ERROR;
657 :    
658 :     case XVID_ERR_MEMORY :
659 :     return ICERR_MEMORY;
660 :    
661 :     case XVID_ERR_FORMAT :
662 :     return ICERR_BADFORMAT;
663 :    
664 :     case XVID_ERR_VERSION :
665 :     return ICERR_UNSUPPORTED;
666 :     }
667 :    
668 : suxen_drol 1061 if (codec->config.display_status && stats.type>0) {
669 :     status_update(&codec->status, stats.type, stats.length, stats.quant);
670 :     }
671 :    
672 : suxen_drol 983 DPRINTF("{type=%i len=%i} length=%i", stats.type, stats.length, length);
673 : suxen_drol 889
674 :     if (length == 0) /* no encoder output */
675 :     {
676 :     *icc->lpdwFlags = 0;
677 :     ((char*)icc->lpOutput)[0] = 0x7f; /* virtual dub skip frame */
678 :     outhdr->biSizeImage = 1;
679 :    
680 :     }else{
681 :     if (frame.out_flags & XVID_KEYFRAME)
682 :     {
683 :     codec->keyspacing = 0;
684 :     *icc->lpdwFlags = AVIIF_KEYFRAME;
685 :     }
686 :     else
687 :     {
688 :     *icc->lpdwFlags = 0;
689 :     }
690 :    
691 :     outhdr->biSizeImage = length;
692 :    
693 : suxen_drol 983 if (codec->config.mode == RC_MODE_2PASS1 && codec->config.discard1pass)
694 : suxen_drol 889 {
695 :     outhdr->biSizeImage = 0;
696 :     }
697 :     }
698 :    
699 : suxen_drol 1017 codec->framenum++;
700 :     codec->keyspacing++;
701 : suxen_drol 889
702 :     return ICERR_OK;
703 :     }
704 :    
705 :    
706 :     /* decompressor */
707 :    
708 :    
709 :     LRESULT decompress_query(CODEC * codec, BITMAPINFO *lpbiInput, BITMAPINFO *lpbiOutput)
710 :     {
711 :     BITMAPINFOHEADER * inhdr = &lpbiInput->bmiHeader;
712 :     BITMAPINFOHEADER * outhdr = &lpbiOutput->bmiHeader;
713 :    
714 :     if (lpbiInput == NULL)
715 :     {
716 :     return ICERR_ERROR;
717 :     }
718 :    
719 :     if (inhdr->biCompression != FOURCC_XVID && inhdr->biCompression != FOURCC_DIVX && inhdr->biCompression != FOURCC_DX50 && get_colorspace(inhdr) == XVID_CSP_NULL)
720 :     {
721 :     return ICERR_BADFORMAT;
722 :     }
723 :    
724 :     if (lpbiOutput == NULL)
725 :     {
726 :     return ICERR_OK;
727 :     }
728 :    
729 :     if (inhdr->biWidth != outhdr->biWidth ||
730 :     inhdr->biHeight != outhdr->biHeight ||
731 :     get_colorspace(outhdr) == XVID_CSP_NULL)
732 :     {
733 :     return ICERR_BADFORMAT;
734 :     }
735 :    
736 :     return ICERR_OK;
737 :     }
738 :    
739 :    
740 :     LRESULT decompress_get_format(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)
741 :     {
742 :     BITMAPINFOHEADER * inhdr = &lpbiInput->bmiHeader;
743 :     BITMAPINFOHEADER * outhdr = &lpbiOutput->bmiHeader;
744 :     LRESULT result;
745 :    
746 :     if (lpbiOutput == NULL)
747 :     {
748 :     return sizeof(BITMAPINFOHEADER);
749 :     }
750 :    
751 :     /* --- yv12 --- */
752 :    
753 :     if (get_colorspace(inhdr) != XVID_CSP_NULL) {
754 :     memcpy(outhdr, inhdr, sizeof(BITMAPINFOHEADER));
755 : edgomez 1053 /* XXX: should we set outhdr->biSize ?? */
756 : suxen_drol 889 return ICERR_OK;
757 :     }
758 :     /* --- yv12 --- */
759 :    
760 :     result = decompress_query(codec, lpbiInput, lpbiOutput);
761 :     if (result != ICERR_OK)
762 :     {
763 :     return result;
764 :     }
765 :    
766 :     outhdr->biSize = sizeof(BITMAPINFOHEADER);
767 :     outhdr->biWidth = inhdr->biWidth;
768 :     outhdr->biHeight = inhdr->biHeight;
769 :     outhdr->biPlanes = 1;
770 :     outhdr->biBitCount = 24;
771 :     outhdr->biCompression = BI_RGB; /* sonic foundry vegas video v3 only supports BI_RGB */
772 :     outhdr->biSizeImage = outhdr->biWidth * outhdr->biHeight * outhdr->biBitCount;
773 :     outhdr->biXPelsPerMeter = 0;
774 :     outhdr->biYPelsPerMeter = 0;
775 :     outhdr->biClrUsed = 0;
776 :     outhdr->biClrImportant = 0;
777 :    
778 :     return ICERR_OK;
779 :     }
780 :    
781 :    
782 :     LRESULT decompress_begin(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)
783 :     {
784 :     xvid_gbl_init_t init;
785 :     xvid_dec_create_t create;
786 :    
787 :     memset(&init, 0, sizeof(init));
788 :     init.version = XVID_VERSION;
789 :     init.cpu_flags = codec->config.cpu;
790 :     xvid_global(0, XVID_GBL_INIT, &init, NULL);
791 :    
792 :     memset(&create, 0, sizeof(create));
793 :     create.version = XVID_VERSION;
794 :     create.width = lpbiInput->bmiHeader.biWidth;
795 :     create.height = lpbiInput->bmiHeader.biHeight;
796 :    
797 :     switch(xvid_decore(0, XVID_DEC_CREATE, &create, NULL))
798 :     {
799 :     case XVID_ERR_FAIL :
800 :     return ICERR_ERROR;
801 :    
802 :     case XVID_ERR_MEMORY :
803 :     return ICERR_MEMORY;
804 :    
805 :     case XVID_ERR_FORMAT :
806 :     return ICERR_BADFORMAT;
807 :    
808 :     case XVID_ERR_VERSION :
809 :     return ICERR_UNSUPPORTED;
810 :     }
811 :    
812 :     codec->dhandle = create.handle;
813 :    
814 :     return ICERR_OK;
815 :     }
816 :    
817 :    
818 :     LRESULT decompress_end(CODEC * codec)
819 :     {
820 :     if (codec->dhandle != NULL)
821 :     {
822 :     xvid_decore(codec->dhandle, XVID_DEC_DESTROY, NULL, NULL);
823 :     codec->dhandle = NULL;
824 :     }
825 :     return ICERR_OK;
826 :     }
827 :    
828 :    
829 :     LRESULT decompress(CODEC * codec, ICDECOMPRESS * icd)
830 :     {
831 :     xvid_dec_frame_t frame;
832 :    
833 :     /* --- yv12 --- */
834 :     if (icd->lpbiInput->biCompression != FOURCC_XVID &&
835 :     icd->lpbiInput->biCompression != FOURCC_DIVX &&
836 :     icd->lpbiInput->biCompression != FOURCC_DX50)
837 :     {
838 :     xvid_gbl_convert_t convert;
839 :    
840 : suxen_drol 983 DPRINTF("input=%c%c%c%c output=%c%c%c%c",
841 :     icd->lpbiInput->biCompression&0xff,
842 :     (icd->lpbiInput->biCompression>>8)&0xff,
843 :     (icd->lpbiInput->biCompression>>16)&0xff,
844 :     (icd->lpbiInput->biCompression>>24)&0xff,
845 :     icd->lpbiOutput->biCompression&0xff,
846 :     (icd->lpbiOutput->biCompression>>8)&0xff,
847 :     (icd->lpbiOutput->biCompression>>16)&0xff,
848 :     (icd->lpbiOutput->biCompression>>24)&0xff);
849 : suxen_drol 889
850 :     memset(&convert, 0, sizeof(convert));
851 :     convert.version = XVID_VERSION;
852 :    
853 :     convert.input.csp = get_colorspace(icd->lpbiInput);
854 :     convert.input.plane[0] = icd->lpInput;
855 :     convert.input.stride[0] = (((icd->lpbiInput->biWidth *icd->lpbiInput->biBitCount) + 31) & ~31) >> 3;
856 :     if (convert.input.csp == XVID_CSP_I420 || convert.input.csp == XVID_CSP_YV12)
857 :     convert.input.stride[0] = (convert.input.stride[0]*2)/3;
858 :    
859 :     convert.output.csp = get_colorspace(icd->lpbiOutput);
860 :     convert.output.plane[0] = icd->lpOutput;
861 :     convert.output.stride[0] = (((icd->lpbiOutput->biWidth *icd->lpbiOutput->biBitCount) + 31) & ~31) >> 3;
862 :     if (convert.output.csp == XVID_CSP_I420 || convert.output.csp == XVID_CSP_YV12)
863 :     convert.output.stride[0] = (convert.output.stride[0]*2)/3;
864 :    
865 :     convert.width = icd->lpbiInput->biWidth;
866 :     convert.height = icd->lpbiInput->biHeight;
867 :     convert.interlacing = 0;
868 :     if (convert.input.csp == XVID_CSP_NULL ||
869 :     convert.output.csp == XVID_CSP_NULL ||
870 :     xvid_global(0, XVID_GBL_CONVERT, &convert, NULL) < 0)
871 :     {
872 :     return ICERR_BADFORMAT;
873 :     }
874 :     return ICERR_OK;
875 :     }
876 :     /* --- yv12 --- */
877 :    
878 :     memset(&frame, 0, sizeof(frame));
879 :     frame.version = XVID_VERSION;
880 :     frame.general = XVID_LOWDELAY; /* force low_delay_default mode */
881 :     frame.bitstream = icd->lpInput;
882 :     frame.length = icd->lpbiInput->biSizeImage;
883 :    
884 :     if (~((icd->dwFlags & ICDECOMPRESS_HURRYUP) | (icd->dwFlags & ICDECOMPRESS_UPDATE) | (icd->dwFlags & ICDECOMPRESS_PREROLL)))
885 :     {
886 :     if ((frame.output.csp = get_colorspace(icd->lpbiOutput)) == XVID_CSP_NULL)
887 :     {
888 :     return ICERR_BADFORMAT;
889 :     }
890 :     frame.output.plane[0] = icd->lpOutput;
891 :     frame.output.stride[0] = (((icd->lpbiOutput->biWidth * icd->lpbiOutput->biBitCount) + 31) & ~31) >> 3;
892 :     if (frame.output.csp == XVID_CSP_I420 || frame.output.csp == XVID_CSP_YV12)
893 :     frame.output.stride[0] = (frame.output.stride[0]*2)/3;
894 :     }
895 :     else
896 :     {
897 :     frame.output.csp = XVID_CSP_NULL;
898 :     }
899 :    
900 :     switch (xvid_decore(codec->dhandle, XVID_DEC_DECODE, &frame, NULL))
901 :     {
902 :     case XVID_ERR_FAIL :
903 :     return ICERR_ERROR;
904 :    
905 :     case XVID_ERR_MEMORY :
906 :     return ICERR_MEMORY;
907 :    
908 :     case XVID_ERR_FORMAT :
909 :     return ICERR_BADFORMAT;
910 :    
911 :     case XVID_ERR_VERSION :
912 :     return ICERR_UNSUPPORTED;
913 :     }
914 :    
915 :     return ICERR_OK;
916 :     }
917 :    

No admin address has been configured
ViewVC Help
Powered by ViewVC 1.0.4