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