[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 1245 - (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 : edgomez 1242 pass2.kfthreshold = codec->config.kfthreshold;
388 : suxen_drol 1052 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 :    
493 : syskin 1245 if (--i < 0) return; /* there are no zones, or we're before the first zone */
494 : suxen_drol 1069
495 : syskin 1245 if (framenum == config->zones[i].frame)
496 :     frame->type = config->zones[i].type;
497 :    
498 : suxen_drol 1017 if (config->zones[i].greyscale) {
499 :     frame->vop_flags |= XVID_VOP_GREYSCALE;
500 :     }
501 :    
502 :     if (config->zones[i].chroma_opt) {
503 :     frame->vop_flags |= XVID_VOP_CHROMAOPT;
504 :     }
505 :    
506 :     if ((profiles[config->profile].flags & PROFILE_BVOP) && config->use_bvop) {
507 :     frame->bframe_threshold = config->zones[i].bvop_threshold;
508 :     }
509 :     }
510 :    
511 :    
512 : suxen_drol 889 LRESULT compress(CODEC * codec, ICCOMPRESS * icc)
513 :     {
514 :     BITMAPINFOHEADER * inhdr = icc->lpbiInput;
515 :     BITMAPINFOHEADER * outhdr = icc->lpbiOutput;
516 :     xvid_enc_frame_t frame;
517 :     xvid_enc_stats_t stats;
518 :     int length;
519 :    
520 :     memset(&frame, 0, sizeof(frame));
521 :     frame.version = XVID_VERSION;
522 :    
523 :     frame.type = XVID_TYPE_AUTO;
524 :    
525 :     /* vol stuff */
526 :    
527 : suxen_drol 1017 if ((profiles[codec->config.profile].flags & PROFILE_MPEGQUANT) &&
528 :     codec->config.quant_type != QUANT_MODE_H263)
529 : suxen_drol 889 {
530 : edgomez 949 frame.vol_flags |= XVID_VOL_MPEGQUANT;
531 : suxen_drol 889
532 : suxen_drol 1017 if (codec->config.quant_type == QUANT_MODE_CUSTOM) {
533 : suxen_drol 889 frame.quant_intra_matrix = codec->config.qmatrix_intra;
534 :     frame.quant_inter_matrix = codec->config.qmatrix_inter;
535 : suxen_drol 1017 }else{
536 : suxen_drol 889 frame.quant_intra_matrix = NULL;
537 :     frame.quant_inter_matrix = NULL;
538 :     }
539 :     }
540 :    
541 : syskin 1087 if ((profiles[codec->config.profile].flags & PROFILE_REDUCED) &&
542 : suxen_drol 1017 codec->config.reduced_resolution) {
543 : edgomez 949 frame.vol_flags |= XVID_VOL_REDUCED_ENABLE;
544 :     frame.vop_flags |= XVID_VOP_REDUCED; /* XXX: need auto decion mode */
545 : suxen_drol 889 }
546 :    
547 : suxen_drol 1017 if ((profiles[codec->config.profile].flags & PROFILE_QPEL) && codec->config.qpel) {
548 : edgomez 949 frame.vol_flags |= XVID_VOL_QUARTERPEL;
549 :     frame.motion |= XVID_ME_QUARTERPELREFINE16 | XVID_ME_QUARTERPELREFINE8;
550 : suxen_drol 889 }
551 :    
552 : syskin 1087 if ((profiles[codec->config.profile].flags & PROFILE_GMC) && codec->config.gmc) {
553 : edgomez 949 frame.vol_flags |= XVID_VOL_GMC;
554 : edgomez 1107 frame.motion |= XVID_ME_GME_REFINE;
555 : syskin 1087 }
556 : suxen_drol 889
557 : suxen_drol 1017 if ((profiles[codec->config.profile].flags & PROFILE_INTERLACE) && codec->config.interlacing)
558 : edgomez 949 frame.vol_flags |= XVID_VOL_INTERLACING;
559 : suxen_drol 889
560 :     /* vop stuff */
561 :    
562 : edgomez 949 frame.vop_flags |= XVID_VOP_HALFPEL;
563 :     frame.vop_flags |= XVID_VOP_HQACPRED;
564 : suxen_drol 889
565 : suxen_drol 1032 if (codec->config.vop_debug)
566 : edgomez 949 frame.vop_flags |= XVID_VOP_DEBUG;
567 : suxen_drol 889
568 : suxen_drol 1017 if (codec->config.trellis_quant) {
569 :     frame.vop_flags |= XVID_VOP_TRELLISQUANT;
570 :     }
571 :    
572 :     if (codec->config.motion_search > 4)
573 : edgomez 949 frame.vop_flags |= XVID_VOP_INTER4V;
574 : suxen_drol 889
575 :     if (codec->config.chromame)
576 : edgomez 1107 frame.motion |= XVID_ME_CHROMA_PVOP + XVID_ME_CHROMA_BVOP;
577 : suxen_drol 889
578 : edgomez 1193 if (codec->config.cartoon_mode) {
579 :     frame.vop_flags |= XVID_VOP_CARTOON;
580 :     frame.motion |= XVID_ME_DETECT_STATIC_MOTION;
581 :     }
582 :    
583 : suxen_drol 889 frame.motion |= pmvfast_presets[codec->config.motion_search];
584 :    
585 :     switch (codec->config.vhq_mode)
586 :     {
587 :     case VHQ_MODE_DECISION :
588 : edgomez 1107 frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
589 : suxen_drol 889 break;
590 :    
591 :     case VHQ_LIMITED_SEARCH :
592 : edgomez 1107 frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
593 :     frame.motion |= XVID_ME_HALFPELREFINE16_RD;
594 :     frame.motion |= XVID_ME_QUARTERPELREFINE16_RD;
595 : suxen_drol 889 break;
596 :    
597 :     case VHQ_MEDIUM_SEARCH :
598 : edgomez 1107 frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
599 :     frame.motion |= XVID_ME_HALFPELREFINE16_RD;
600 :     frame.motion |= XVID_ME_HALFPELREFINE8_RD;
601 :     frame.motion |= XVID_ME_QUARTERPELREFINE16_RD;
602 :     frame.motion |= XVID_ME_QUARTERPELREFINE8_RD;
603 :     frame.motion |= XVID_ME_CHECKPREDICTION_RD;
604 : suxen_drol 889 break;
605 :    
606 :     case VHQ_WIDE_SEARCH :
607 : edgomez 1107 frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
608 :     frame.motion |= XVID_ME_HALFPELREFINE16_RD;
609 :     frame.motion |= XVID_ME_HALFPELREFINE8_RD;
610 :     frame.motion |= XVID_ME_QUARTERPELREFINE16_RD;
611 :     frame.motion |= XVID_ME_QUARTERPELREFINE8_RD;
612 :     frame.motion |= XVID_ME_CHECKPREDICTION_RD;
613 :     frame.motion |= XVID_ME_EXTSEARCH_RD;
614 : suxen_drol 889 break;
615 :    
616 :     default :
617 :     break;
618 :     }
619 :    
620 :     frame.input.plane[0] = icc->lpInput;
621 :     frame.input.stride[0] = (((icc->lpbiInput->biWidth * icc->lpbiInput->biBitCount) + 31) & ~31) >> 3;
622 :    
623 :     if ((frame.input.csp = get_colorspace(inhdr)) == XVID_CSP_NULL)
624 :     return ICERR_BADFORMAT;
625 :    
626 :     if (frame.input.csp == XVID_CSP_I420 || frame.input.csp == XVID_CSP_YV12)
627 :     frame.input.stride[0] = (frame.input.stride[0]*2)/3;
628 :    
629 :     frame.bitstream = icc->lpOutput;
630 :     frame.length = icc->lpbiOutput->biSizeImage;
631 :    
632 : suxen_drol 1017 frame.quant = 0;
633 : suxen_drol 889
634 : suxen_drol 1017 if (codec->config.mode == RC_MODE_NULL) {
635 : suxen_drol 889 outhdr->biSizeImage = 0;
636 :     *icc->lpdwFlags = AVIIF_KEYFRAME;
637 :     return ICERR_OK;
638 : suxen_drol 1017 }
639 : suxen_drol 889
640 :     // force keyframe spacing in 2-pass 1st pass
641 :     if (codec->config.motion_search == 0)
642 :     frame.type = XVID_TYPE_IVOP;
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