[svn] / branches / release-1_3-branch / xvidcore / vfw / src / codec.c Repository:
ViewVC logotype

Annotation of /branches/release-1_3-branch/xvidcore/vfw/src/codec.c

Parent Directory Parent Directory | Revision Log Revision Log


Revision 2141 - (view) (download)

1 : edgomez 1382 /**************************************************************************
2 :     *
3 :     * XVID VFW FRONTEND
4 :     * codec
5 :     *
6 : Isibaar 1910 * Copyright(C) Peter Ross <pross@xvid.org>
7 :     *
8 : edgomez 1382 * This program is free software; you can redistribute it and/or modify
9 :     * it under the terms of the GNU General Public License as published by
10 :     * the Free Software Foundation; either version 2 of the License, or
11 :     * (at your option) any later version.
12 :     *
13 :     * This program is distributed in the hope that it will be useful,
14 :     * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 :     * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 :     * GNU General Public License for more details.
17 :     *
18 :     * You should have received a copy of the GNU General Public License
19 :     * along with this program; if not, write to the Free Software
20 :     * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
21 :     *
22 : Isibaar 1985 * $Id$
23 : edgomez 1382 *
24 :     *************************************************************************/
25 :    
26 :     #include <windows.h>
27 :     #include <vfw.h>
28 :     #include <stdio.h>
29 :     #include "vfwext.h"
30 :    
31 :     #include <xvid.h>
32 :     #include "debug.h"
33 :     #include "codec.h"
34 :     #include "status.h"
35 :    
36 :    
37 :    
38 :     static const int pmvfast_presets[7] = {
39 :     0, 0, 0, 0,
40 :     0 | XVID_ME_HALFPELREFINE16 | 0,
41 :     0 | XVID_ME_HALFPELREFINE16 | 0 |
42 :     XVID_ME_ADVANCEDDIAMOND16, XVID_ME_HALFPELREFINE16 | XVID_ME_EXTSEARCH16 |
43 :     XVID_ME_HALFPELREFINE8 | 0 | XVID_ME_USESQUARES16
44 :     };
45 :    
46 :    
47 :    
48 :     /* return xvid compatbile colorspace,
49 :     or XVID_CSP_NULL if failure
50 :     */
51 :    
52 :     static int get_colorspace(BITMAPINFOHEADER * hdr)
53 :     {
54 :     /* rgb only: negative height specifies top down image */
55 :     int rgb_flip = (hdr->biHeight < 0 ? 0 : XVID_CSP_VFLIP);
56 :    
57 :     switch(hdr->biCompression)
58 :     {
59 :     case BI_RGB :
60 :     if (hdr->biBitCount == 16)
61 :     {
62 :     DPRINTF("RGB16 (RGB555)");
63 :     return rgb_flip | XVID_CSP_RGB555;
64 :     }
65 :     if (hdr->biBitCount == 24)
66 :     {
67 :     DPRINTF("RGB24");
68 :     return rgb_flip | XVID_CSP_BGR;
69 :     }
70 :     if (hdr->biBitCount == 32)
71 :     {
72 :     DPRINTF("RGB32");
73 :     return rgb_flip | XVID_CSP_BGRA;
74 :     }
75 :    
76 :     DPRINTF("unsupported BI_RGB biBitCount=%i", hdr->biBitCount);
77 :     return XVID_CSP_NULL;
78 :    
79 :     case BI_BITFIELDS :
80 :     if (hdr->biSize >= sizeof(BITMAPV4HEADER))
81 :     {
82 :     BITMAPV4HEADER * hdr4 = (BITMAPV4HEADER *)hdr;
83 :    
84 :     if (hdr4->bV4BitCount == 16 &&
85 :     hdr4->bV4RedMask == 0x7c00 &&
86 :     hdr4->bV4GreenMask == 0x3e0 &&
87 :     hdr4->bV4BlueMask == 0x1f)
88 :     {
89 :     DPRINTF("RGB555");
90 :     return rgb_flip | XVID_CSP_RGB555;
91 :     }
92 :    
93 :     if(hdr4->bV4BitCount == 16 &&
94 :     hdr4->bV4RedMask == 0xf800 &&
95 :     hdr4->bV4GreenMask == 0x7e0 &&
96 :     hdr4->bV4BlueMask == 0x1f)
97 :     {
98 :     DPRINTF("RGB565");
99 :     return rgb_flip | XVID_CSP_RGB565;
100 :     }
101 :    
102 :     DPRINTF("unsupported BI_BITFIELDS mode");
103 :     return XVID_CSP_NULL;
104 :     }
105 :    
106 :     DPRINTF("unsupported BI_BITFIELDS/BITMAPHEADER combination");
107 :     return XVID_CSP_NULL;
108 :    
109 :     case FOURCC_I420 :
110 :     case FOURCC_IYUV :
111 :     DPRINTF("IYUY");
112 :     return XVID_CSP_I420;
113 :    
114 :     case FOURCC_YV12 :
115 :     DPRINTF("YV12");
116 :     return XVID_CSP_YV12;
117 :    
118 :     case FOURCC_YUYV :
119 :     case FOURCC_YUY2 :
120 :     DPRINTF("YUY2");
121 :     return XVID_CSP_YUY2;
122 :    
123 :     case FOURCC_YVYU :
124 :     DPRINTF("YVYU");
125 :     return XVID_CSP_YVYU;
126 :    
127 :     case FOURCC_UYVY :
128 :     DPRINTF("UYVY");
129 :     return XVID_CSP_UYVY;
130 :    
131 :     default :
132 :     DPRINTF("unsupported colorspace %c%c%c%c",
133 :     hdr->biCompression&0xff,
134 :     (hdr->biCompression>>8)&0xff,
135 :     (hdr->biCompression>>16)&0xff,
136 :     (hdr->biCompression>>24)&0xff);
137 :     return XVID_CSP_NULL;
138 :     }
139 :     }
140 :    
141 :    
142 :     /* compressor */
143 :    
144 :    
145 :     /* test the output format */
146 :    
147 :     LRESULT compress_query(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)
148 :     {
149 :     BITMAPINFOHEADER * inhdr = &lpbiInput->bmiHeader;
150 :     BITMAPINFOHEADER * outhdr = &lpbiOutput->bmiHeader;
151 :    
152 :     /* VFWEXT detection */
153 :     if (inhdr->biCompression == VFWEXT_FOURCC) {
154 :     return (ICM_USER+0x0fff);
155 :     }
156 :    
157 :     if (get_colorspace(inhdr) == XVID_CSP_NULL)
158 :     {
159 :     return ICERR_BADFORMAT;
160 :     }
161 :    
162 :     if (lpbiOutput == NULL)
163 :     {
164 :     return ICERR_OK;
165 :     }
166 :    
167 :     if (inhdr->biWidth != outhdr->biWidth || inhdr->biHeight != outhdr->biHeight ||
168 :     (outhdr->biCompression != FOURCC_XVID && outhdr->biCompression != FOURCC_DIVX && outhdr->biCompression != FOURCC_DX50))
169 :     {
170 :     return ICERR_BADFORMAT;
171 :     }
172 :    
173 :     return ICERR_OK;
174 :     }
175 :    
176 :    
177 :     LRESULT compress_get_format(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)
178 :     {
179 :     BITMAPINFOHEADER * inhdr = &lpbiInput->bmiHeader;
180 :     BITMAPINFOHEADER * outhdr = &lpbiOutput->bmiHeader;
181 :    
182 :     if (get_colorspace(inhdr) == XVID_CSP_NULL)
183 :     {
184 :     return ICERR_BADFORMAT;
185 :     }
186 :    
187 :     if (lpbiOutput == NULL)
188 :     {
189 :     return sizeof(BITMAPINFOHEADER);
190 :     }
191 :    
192 :     memcpy(outhdr, inhdr, sizeof(BITMAPINFOHEADER));
193 :     outhdr->biSize = sizeof(BITMAPINFOHEADER);
194 :     outhdr->biSizeImage = compress_get_size(codec, lpbiInput, lpbiOutput);
195 :     outhdr->biXPelsPerMeter = 0;
196 :     outhdr->biYPelsPerMeter = 0;
197 :     outhdr->biClrUsed = 0;
198 :     outhdr->biClrImportant = 0;
199 :    
200 : Isibaar 1910 if ((codec->config.fourcc_used == 0) || (profiles[codec->config.profile].flags & PROFILE_XVID))
201 : edgomez 1382 {
202 :     outhdr->biCompression = FOURCC_XVID;
203 :     }
204 :     else if (codec->config.fourcc_used == 1)
205 :     {
206 :     outhdr->biCompression = FOURCC_DIVX;
207 :     }
208 :     else
209 :     {
210 :     outhdr->biCompression = FOURCC_DX50;
211 :     }
212 :    
213 :     return ICERR_OK;
214 :     }
215 :    
216 :    
217 :     LRESULT compress_get_size(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)
218 :     {
219 :     return 2 * lpbiOutput->bmiHeader.biWidth * lpbiOutput->bmiHeader.biHeight * 3;
220 :     }
221 :    
222 :    
223 :     LRESULT compress_frames_info(CODEC * codec, ICCOMPRESSFRAMES * icf)
224 :     {
225 :     #if 0
226 :     DPRINTF("%i %i", icf->lStartFrame, icf->lFrameCount);
227 :     #endif
228 :     codec->fincr = icf->dwScale;
229 :     codec->fbase = icf->dwRate;
230 :     return ICERR_OK;
231 :     }
232 :    
233 :    
234 :     static char type2char(int type)
235 :     {
236 :     if (type==XVID_TYPE_IVOP)
237 :     return 'I';
238 :     if (type==XVID_TYPE_PVOP)
239 :     return 'P';
240 :     if (type==XVID_TYPE_BVOP)
241 :     return 'B';
242 :     return 'S';
243 :     }
244 :    
245 :     static int vfw_debug(void *handle,
246 :     int opt,
247 :     void *param1,
248 :     void *param2)
249 :     {
250 :     switch (opt) {
251 :     case XVID_PLG_CREATE:
252 :     *((void**)param2) = NULL;
253 :     case XVID_PLG_INFO:
254 :     case XVID_PLG_DESTROY:
255 :     case XVID_PLG_BEFORE:
256 :     return 0;
257 :    
258 :     case XVID_PLG_AFTER:
259 :     {
260 :     xvid_plg_data_t *data = (xvid_plg_data_t *) param1;
261 :    
262 :     /* We don't use DPRINTF here because it's active only for _DEBUG
263 :     * builds and that activates lot of other debug printfs. We only
264 :     * want these all the time */
265 :     char buf[1024];
266 :     sprintf(buf, "[%6i] type=%c Q:%2i length:%6i",
267 :     data->frame_num,
268 :     type2char(data->type),
269 :     data->quant,
270 :     data->length);
271 :     OutputDebugString(buf);
272 :    
273 :     return 0;
274 :     }
275 :     }
276 :    
277 :     return XVID_ERR_FAIL;
278 :     }
279 :    
280 :     #define XVID_DLL_NAME "xvidcore.dll"
281 :    
282 : edgomez 1398 static int init_dll(CODEC* codec)
283 : edgomez 1382 {
284 : edgomez 1398 if (codec->m_hdll != NULL)
285 :     return 0;
286 : edgomez 1382
287 :     DPRINTF("init_dll");
288 : edgomez 1398 codec->m_hdll = LoadLibrary(XVID_DLL_NAME);
289 :     if (codec->m_hdll == NULL) {
290 : edgomez 1382 DPRINTF("dll load failed");
291 :     MessageBox(0, XVID_DLL_NAME " not found!","Error!", MB_ICONEXCLAMATION|MB_OK);
292 :     return XVID_ERR_FAIL;
293 :     }
294 :    
295 : edgomez 1398 codec->xvid_global_func = (int (__cdecl *)(void *, int, void *, void *))GetProcAddress(codec->m_hdll, "xvid_global");
296 :     if (codec->xvid_global_func == NULL) {
297 : edgomez 1382 MessageBox(0, "xvid_global() not found", "Error", 0);
298 :     return XVID_ERR_FAIL;
299 :     }
300 :    
301 : edgomez 1398 codec->xvid_encore_func = (int (__cdecl *)(void *, int, void *, void *))GetProcAddress(codec->m_hdll, "xvid_encore");
302 :     if (codec->xvid_encore_func == NULL) {
303 : edgomez 1382 MessageBox(0, "xvid_encore() not found", "Error", 0);
304 :     return XVID_ERR_FAIL;
305 :     }
306 :    
307 : edgomez 1398 codec->xvid_decore_func = (int (__cdecl *)(void *, int, void *, void *))GetProcAddress(codec->m_hdll, "xvid_decore");
308 :     if (codec->xvid_decore_func == NULL) {
309 : edgomez 1382 MessageBox(0, "xvid_decore() not found", "Error", 0);
310 :     return XVID_ERR_FAIL;
311 :     }
312 :    
313 : edgomez 1398 codec->xvid_plugin_single_func =
314 :     (int (__cdecl *)(void *, int, void *, void *))(GetProcAddress(codec->m_hdll, "xvid_plugin_single"));
315 :     codec->xvid_plugin_2pass1_func =
316 :     (int (__cdecl *)(void *, int, void *, void *))(GetProcAddress(codec->m_hdll, "xvid_plugin_2pass1"));
317 :     codec->xvid_plugin_2pass2_func =
318 :     (int (__cdecl *)(void *, int, void *, void *))(GetProcAddress(codec->m_hdll, "xvid_plugin_2pass2"));
319 :     codec->xvid_plugin_lumimasking_func =
320 :     (int (__cdecl *)(void *, int, void *, void *))(GetProcAddress(codec->m_hdll, "xvid_plugin_lumimasking"));
321 :     codec->xvid_plugin_psnr_func =
322 :     (int (__cdecl *)(void *, int, void *, void *))(GetProcAddress(codec->m_hdll, "xvid_plugin_psnr"));
323 : edgomez 1382
324 :     return 0;
325 :     }
326 :    
327 :     /* constant-quant zones for fixed quant encoding */
328 :     static void
329 :     prepare_cquant_zones(CONFIG * config) {
330 :    
331 :     int i = 0;
332 :     if (config->num_zones == 0 || config->zones[0].frame != 0) {
333 :     /* first zone does not start at frame 0 or doesn't exist */
334 :    
335 :     if (config->num_zones >= MAX_ZONES) config->num_zones--; /* we scrifice last zone */
336 :    
337 :     config->zones[config->num_zones].frame = 0;
338 :     config->zones[config->num_zones].mode = RC_ZONE_QUANT;
339 :     config->zones[config->num_zones].weight = 100;
340 :     config->zones[config->num_zones].quant = config->desired_quant;
341 :     config->zones[config->num_zones].type = XVID_TYPE_AUTO;
342 :     config->zones[config->num_zones].greyscale = 0;
343 :     config->zones[config->num_zones].chroma_opt = 0;
344 :     config->zones[config->num_zones].bvop_threshold = 0;
345 :     config->num_zones++;
346 :    
347 :     sort_zones(config->zones, config->num_zones, &i);
348 :     }
349 :    
350 :     /* step 2: let's change all weight zones into quant zones */
351 :    
352 :     for(i = 0; i < config->num_zones; i++)
353 :     if (config->zones[i].mode == RC_ZONE_WEIGHT) {
354 :     config->zones[i].mode = RC_ZONE_QUANT;
355 :     config->zones[i].quant = (100*config->desired_quant) / config->zones[i].weight;
356 :     }
357 :     }
358 :    
359 :     /* full first pass zones */
360 :     static void
361 :     prepare_full1pass_zones(CONFIG * config) {
362 :    
363 :     int i = 0;
364 :     if (config->num_zones == 0 || config->zones[0].frame != 0) {
365 :     /* first zone does not start at frame 0 or doesn't exist */
366 :    
367 :     if (config->num_zones >= MAX_ZONES) config->num_zones--; /* we scrifice last zone */
368 :    
369 :     config->zones[config->num_zones].frame = 0;
370 :     config->zones[config->num_zones].mode = RC_ZONE_QUANT;
371 :     config->zones[config->num_zones].weight = 100;
372 :     config->zones[config->num_zones].quant = 200;
373 :     config->zones[config->num_zones].type = XVID_TYPE_AUTO;
374 :     config->zones[config->num_zones].greyscale = 0;
375 :     config->zones[config->num_zones].chroma_opt = 0;
376 :     config->zones[config->num_zones].bvop_threshold = 0;
377 :     config->num_zones++;
378 :    
379 :     sort_zones(config->zones, config->num_zones, &i);
380 :     }
381 :    
382 :     /* step 2: let's change all weight zones into quant zones */
383 :    
384 :     for(i = 0; i < config->num_zones; i++)
385 :     if (config->zones[i].mode == RC_ZONE_WEIGHT) {
386 :     config->zones[i].mode = RC_ZONE_QUANT;
387 :     config->zones[i].quant = 200;
388 :     }
389 :     }
390 :    
391 :    
392 :     LRESULT compress_begin(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)
393 :     {
394 :     xvid_gbl_init_t init;
395 :     xvid_enc_create_t create;
396 :     xvid_enc_plugin_t plugins[3];
397 :     xvid_plugin_single_t single;
398 :     xvid_plugin_2pass1_t pass1;
399 :     xvid_plugin_2pass2_t pass2;
400 : Isibaar 1864 xvid_plugin_lumimasking_t masking;
401 : Isibaar 1816 xvid_gbl_info_t info;
402 : edgomez 1382 int i;
403 :     HANDLE hFile;
404 : Isibaar 1914 const quality_t* quality_preset = (codec->config.quality==quality_table_num) ?
405 : suxen_drol 1649 &codec->config.quality_user : &quality_table[codec->config.quality];
406 : edgomez 1382
407 :     CONFIG tmpCfg; /* if we want to alter config to suit our needs, it shouldn't be visible to user later */
408 :     memcpy(&tmpCfg, &codec->config, sizeof(CONFIG));
409 :    
410 : edgomez 1398 if (init_dll(codec) != 0) return ICERR_ERROR;
411 : edgomez 1382 /* destroy previously created codec */
412 :     if(codec->ehandle) {
413 : edgomez 1398 codec->xvid_encore_func(codec->ehandle, XVID_ENC_DESTROY, NULL, NULL);
414 : edgomez 1382 codec->ehandle = NULL;
415 :     }
416 :    
417 :     memset(&init, 0, sizeof(init));
418 :     init.version = XVID_VERSION;
419 :     init.cpu_flags = codec->config.cpu;
420 :     init.debug = codec->config.debug;
421 : edgomez 1398 codec->xvid_global_func(0, XVID_GBL_INIT, &init, NULL);
422 : edgomez 1382
423 : Isibaar 1816 memset(&info, 0, sizeof(info));
424 :     info.version = XVID_VERSION;
425 :     codec->xvid_global_func(0, XVID_GBL_INFO, &info, NULL);
426 :    
427 : edgomez 1382 memset(&create, 0, sizeof(create));
428 :     create.version = XVID_VERSION;
429 :    
430 : Isibaar 1914 /* Encoder threads */
431 :     if (codec->config.cpu & XVID_CPU_FORCE)
432 :     create.num_threads = codec->config.num_threads;
433 :     else
434 :     create.num_threads = info.num_threads; /* Autodetect */
435 :    
436 : Isibaar 1917 /* Encoder slices */
437 :     if ((profiles[codec->config.profile].flags & PROFILE_RESYNCMARKER) && codec->config.num_slices != 1) {
438 :    
439 :     if (codec->config.num_slices == 0) { /* auto */
440 : Isibaar 1933 int mb_width = (lpbiInput->bmiHeader.biWidth + 15) / 16;
441 :     int mb_height = (lpbiInput->bmiHeader.biHeight + 15) / 16;
442 : Isibaar 1917
443 : Isibaar 1933 int slices = (int)((mb_width*mb_height) / 811); /* use multiple slices only above SD resolutions for now */
444 : Isibaar 1917
445 : Isibaar 1933 if (slices > 1) {
446 :     if (create.num_threads <= 1)
447 :     slices &= ~1; /* make even */
448 :     else if (create.num_threads <= slices)
449 :     slices = (slices / create.num_threads) * create.num_threads; /* multiple of threads */
450 :     else if (create.num_threads % slices)
451 :     slices = (!(create.num_threads%2)) ? (create.num_threads/2) : (create.num_threads/3);
452 :     }
453 :    
454 :     create.num_slices = slices;
455 : Isibaar 1917 }
456 :     else {
457 : Isibaar 1922 create.num_slices = codec->config.num_slices; /* force manual value - by registry edit */
458 : Isibaar 1917 }
459 :    
460 :     }
461 :    
462 : edgomez 1382 /* plugins */
463 :     create.plugins = plugins;
464 :     switch (codec->config.mode)
465 :     {
466 :     case RC_MODE_1PASS :
467 :     memset(&single, 0, sizeof(single));
468 :     single.version = XVID_VERSION;
469 :     single.bitrate = codec->config.bitrate * CONFIG_KBPS;
470 :     single.reaction_delay_factor = codec->config.rc_reaction_delay_factor;
471 :     single.averaging_period = codec->config.rc_averaging_period;
472 :     single.buffer = codec->config.rc_buffer;
473 : edgomez 1398 plugins[create.num_plugins].func = codec->xvid_plugin_single_func;
474 : edgomez 1382 plugins[create.num_plugins].param = &single;
475 :     create.num_plugins++;
476 :     if (!codec->config.use_2pass_bitrate) /* constant-quant mode */
477 :     prepare_cquant_zones(&tmpCfg);
478 :     break;
479 :    
480 :     case RC_MODE_2PASS1 :
481 :     memset(&pass1, 0, sizeof(pass1));
482 :     pass1.version = XVID_VERSION;
483 :     pass1.filename = codec->config.stats;
484 :     if (codec->config.full1pass)
485 :     prepare_full1pass_zones(&tmpCfg);
486 : edgomez 1398 plugins[create.num_plugins].func = codec->xvid_plugin_2pass1_func;
487 : edgomez 1382 plugins[create.num_plugins].param = &pass1;
488 :     create.num_plugins++;
489 :     break;
490 :    
491 :     case RC_MODE_2PASS2 :
492 :     memset(&pass2, 0, sizeof(pass2));
493 :     pass2.version = XVID_VERSION;
494 :     if (codec->config.use_2pass_bitrate) {
495 :     pass2.bitrate = codec->config.bitrate * CONFIG_KBPS;
496 :     } else {
497 :     pass2.bitrate = -codec->config.desired_size; /* kilobytes */
498 :     }
499 :     pass2.filename = codec->config.stats;
500 :    
501 :     hFile = CreateFile(pass2.filename, 0, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, 0, NULL);
502 :     if (hFile == INVALID_HANDLE_VALUE)
503 :     {
504 :     MessageBox(0, "Statsfile not found!","Error!", MB_ICONEXCLAMATION|MB_OK);
505 :     return XVID_ERR_FAIL;
506 :     } else
507 :     {
508 :     CloseHandle(hFile);
509 :     }
510 :    
511 :     pass2.keyframe_boost = codec->config.keyframe_boost; /* keyframe boost percentage: [0..100...]; */
512 :     pass2.curve_compression_high = codec->config.curve_compression_high;
513 :     pass2.curve_compression_low = codec->config.curve_compression_low;
514 :     pass2.overflow_control_strength = codec->config.overflow_control_strength;
515 :     pass2.max_overflow_improvement = codec->config.twopass_max_overflow_improvement;
516 :     pass2.max_overflow_degradation = codec->config.twopass_max_overflow_degradation;
517 :     pass2.kfreduction = codec->config.kfreduction;
518 :     pass2.kfthreshold = codec->config.kfthreshold;
519 :     pass2.container_frame_overhead = 24; /* AVI */
520 :    
521 : syskin 1562 /* VBV */
522 :     pass2.vbv_size = profiles[codec->config.profile].max_vbv_size;
523 : suxen_drol 1607 pass2.vbv_initial = (profiles[codec->config.profile].max_vbv_size*3)/4; /* 75% */
524 :     pass2.vbv_maxrate = profiles[codec->config.profile].max_bitrate;
525 : Isibaar 1906 pass2.vbv_peakrate = profiles[codec->config.profile].vbv_peakrate;
526 : syskin 1562
527 : edgomez 1398 plugins[create.num_plugins].func = codec->xvid_plugin_2pass2_func;
528 : edgomez 1382 plugins[create.num_plugins].param = &pass2;
529 :     create.num_plugins++;
530 :     break;
531 :    
532 :     case RC_MODE_NULL :
533 :     return ICERR_OK;
534 :    
535 :     default :
536 :     break;
537 :     }
538 :    
539 :     /* zones - copy from tmpCfg in case we automatically altered them above */
540 :     create.zones = malloc(sizeof(xvid_enc_zone_t) * tmpCfg.num_zones);
541 :     create.num_zones = tmpCfg.num_zones;
542 :     for (i=0; i < create.num_zones; i++) {
543 :     create.zones[i].frame = tmpCfg.zones[i].frame;
544 :     if (tmpCfg.zones[i].mode == RC_ZONE_QUANT) {
545 :     create.zones[i].mode = XVID_ZONE_QUANT;
546 :     create.zones[i].increment = tmpCfg.zones[i].quant;
547 :     }else{
548 :     create.zones[i].mode = XVID_ZONE_WEIGHT;
549 :     create.zones[i].increment = tmpCfg.zones[i].weight;
550 :     }
551 :     create.zones[i].base = 100;
552 :     }
553 :    
554 :     /* lumimasking plugin */
555 : Isibaar 1864 if ((profiles[codec->config.profile].flags & PROFILE_ADAPTQUANT) && (codec->config.lum_masking>0)) {
556 :     memset(&masking, 0, sizeof(masking));
557 :     masking.method = (codec->config.lum_masking==2);
558 : edgomez 1398 plugins[create.num_plugins].func = codec->xvid_plugin_lumimasking_func;
559 : Isibaar 1864 plugins[create.num_plugins].param = &masking;
560 : edgomez 1382 create.num_plugins++;
561 :     }
562 :    
563 : Isibaar 2141 if (codec->config.debug > 0) {
564 : Isibaar 2139 plugins[create.num_plugins].func = vfw_debug;
565 :     plugins[create.num_plugins].param = NULL;
566 :     create.num_plugins++;
567 :     }
568 : edgomez 1382 create.profile = profiles[codec->config.profile].id;
569 :    
570 :     create.width = lpbiInput->bmiHeader.biWidth;
571 :     create.height = lpbiInput->bmiHeader.biHeight;
572 :     create.fincr = codec->fincr;
573 :     create.fbase = codec->fbase;
574 :    
575 : suxen_drol 1649 create.max_key_interval = quality_preset->max_key_interval;
576 : edgomez 1382
577 : suxen_drol 1649 create.min_quant[0] = quality_preset->min_iquant;
578 :     create.max_quant[0] = quality_preset->max_iquant;
579 :     create.min_quant[1] = quality_preset->min_pquant;
580 :     create.max_quant[1] = quality_preset->max_pquant;
581 :     create.min_quant[2] = quality_preset->min_bquant;
582 :     create.max_quant[2] = quality_preset->max_bquant;
583 : edgomez 1382
584 :     if ((profiles[codec->config.profile].flags & PROFILE_BVOP) && codec->config.use_bvop) {
585 :    
586 : suxen_drol 1607 /* dxn: prevent bframes usage if interlacing is selected */
587 : Isibaar 1650 if (!((profiles[codec->config.profile].flags & PROFILE_EXTRA) && codec->config.interlacing)) {
588 : suxen_drol 1607 create.max_bframes = codec->config.max_bframes;
589 :     create.bquant_ratio = codec->config.bquant_ratio;
590 :     create.bquant_offset = codec->config.bquant_offset;
591 : edgomez 1382
592 : suxen_drol 1607 if (codec->config.packed)
593 :     create.global |= XVID_GLOBAL_PACKED;
594 : edgomez 1382
595 : suxen_drol 1607 create.global |= XVID_GLOBAL_CLOSED_GOP;
596 :    
597 : Isibaar 1718 /* restrict max bframes */
598 :     if ((create.max_bframes > profiles[codec->config.profile].xvid_max_bframes) && (profiles[codec->config.profile].xvid_max_bframes >= 0))
599 :     create.max_bframes = profiles[codec->config.profile].xvid_max_bframes;
600 : suxen_drol 1607
601 : Isibaar 1718 /* DXN: enable packed bframes */
602 :     if ((profiles[codec->config.profile].flags & PROFILE_PACKED)) {
603 : suxen_drol 1607 create.global |= XVID_GLOBAL_PACKED;
604 :     }
605 :     }
606 : edgomez 1382 }
607 :    
608 : Isibaar 1816 /* dxn: always write divx5 userdata */
609 :     if ((profiles[codec->config.profile].flags & PROFILE_EXTRA))
610 :     create.global |= XVID_GLOBAL_DIVX5_USERDATA;
611 : suxen_drol 1607
612 : Isibaar 1952 if ((profiles[codec->config.profile].flags & PROFILE_EXTRA) ||
613 :     (profiles[codec->config.profile].flags & PROFILE_XVID)) {
614 :     create.frame_drop_ratio = 0;
615 :     } else {
616 :     create.frame_drop_ratio = quality_preset->frame_drop_ratio;
617 :     }
618 : edgomez 1382
619 : edgomez 1398 switch(codec->xvid_encore_func(0, XVID_ENC_CREATE, &create, NULL))
620 : edgomez 1382 {
621 :     case XVID_ERR_FAIL :
622 :     return ICERR_ERROR;
623 :    
624 :     case XVID_ERR_MEMORY :
625 :     return ICERR_MEMORY;
626 :    
627 :     case XVID_ERR_FORMAT :
628 :     return ICERR_BADFORMAT;
629 :    
630 :     case XVID_ERR_VERSION :
631 :     return ICERR_UNSUPPORTED;
632 :     }
633 :    
634 : syskin 1477 free(create.zones);
635 : edgomez 1382 codec->ehandle = create.handle;
636 :     codec->framenum = 0;
637 :     codec->keyspacing = 0;
638 :    
639 :     if (codec->config.display_status) {
640 :     status_destroy_always(&codec->status);
641 :     status_create(&codec->status, codec->fincr, codec->fbase);
642 :     }
643 :    
644 :     return ICERR_OK;
645 :     }
646 :    
647 :    
648 :     LRESULT compress_end(CODEC * codec)
649 :     {
650 : suxen_drol 1685 if (codec==NULL)
651 :     return ICERR_OK;
652 :    
653 : edgomez 1398 if (codec->m_hdll != NULL) {
654 : edgomez 1382 if (codec->ehandle != NULL) {
655 : edgomez 1398 codec->xvid_encore_func(codec->ehandle, XVID_ENC_DESTROY, NULL, NULL);
656 : edgomez 1382 codec->ehandle = NULL;
657 :     }
658 :     }
659 :    
660 :     if (codec->config.display_status)
661 :     status_destroy(&codec->status);
662 :    
663 :     return ICERR_OK;
664 :     }
665 :    
666 :    
667 :     static void apply_zone_modifiers(xvid_enc_frame_t * frame, CONFIG * config, int framenum)
668 :     {
669 :     int i;
670 :    
671 :     for (i=0; i<config->num_zones && config->zones[i].frame <= framenum; i++) ;
672 :    
673 :     if (--i < 0) return; /* there are no zones, or we're before the first zone */
674 :    
675 :     if (framenum == config->zones[i].frame)
676 :     frame->type = config->zones[i].type;
677 :    
678 :     if (config->zones[i].greyscale) {
679 :     frame->vop_flags |= XVID_VOP_GREYSCALE;
680 :     }
681 :    
682 :     if (config->zones[i].chroma_opt) {
683 :     frame->vop_flags |= XVID_VOP_CHROMAOPT;
684 :     }
685 :    
686 : syskin 1588 if (config->zones[i].cartoon_mode) {
687 :     frame->vop_flags |= XVID_VOP_CARTOON;
688 :     frame->motion |= XVID_ME_DETECT_STATIC_MOTION;
689 :     }
690 :    
691 : edgomez 1382 if ((profiles[config->profile].flags & PROFILE_BVOP) && config->use_bvop) {
692 :     frame->bframe_threshold = config->zones[i].bvop_threshold;
693 :     }
694 :     }
695 :    
696 :    
697 : suxen_drol 1495 #define CALC_BI_STRIDE(width,bitcount) ((((width * bitcount) + 31) & ~31) >> 3)
698 :    
699 : edgomez 1382 LRESULT compress(CODEC * codec, ICCOMPRESS * icc)
700 :     {
701 :     BITMAPINFOHEADER * inhdr = icc->lpbiInput;
702 :     BITMAPINFOHEADER * outhdr = icc->lpbiOutput;
703 :     xvid_enc_frame_t frame;
704 :     xvid_enc_stats_t stats;
705 :     int length;
706 : suxen_drol 1649 const quality_t* quality_preset = (codec->config.quality==quality_table_num) ?
707 :     &codec->config.quality_user : &quality_table[codec->config.quality];
708 : edgomez 1382
709 :     memset(&frame, 0, sizeof(frame));
710 :     frame.version = XVID_VERSION;
711 :    
712 :     frame.type = XVID_TYPE_AUTO;
713 :    
714 :     /* vol stuff */
715 :    
716 :     if ((profiles[codec->config.profile].flags & PROFILE_MPEGQUANT) &&
717 :     codec->config.quant_type != QUANT_MODE_H263)
718 :     {
719 :     frame.vol_flags |= XVID_VOL_MPEGQUANT;
720 :    
721 :     if (codec->config.quant_type == QUANT_MODE_CUSTOM) {
722 :     frame.quant_intra_matrix = codec->config.qmatrix_intra;
723 :     frame.quant_inter_matrix = codec->config.qmatrix_inter;
724 :     }else{
725 :     frame.quant_intra_matrix = NULL;
726 :     frame.quant_inter_matrix = NULL;
727 :     }
728 :     }
729 :    
730 :     if ((profiles[codec->config.profile].flags & PROFILE_QPEL) && codec->config.qpel) {
731 :     frame.vol_flags |= XVID_VOL_QUARTERPEL;
732 :     frame.motion |= XVID_ME_QUARTERPELREFINE16 | XVID_ME_QUARTERPELREFINE8;
733 :     }
734 :    
735 :     if ((profiles[codec->config.profile].flags & PROFILE_GMC) && codec->config.gmc) {
736 :     frame.vol_flags |= XVID_VOL_GMC;
737 :     frame.motion |= XVID_ME_GME_REFINE;
738 :     }
739 :    
740 :     if ((profiles[codec->config.profile].flags & PROFILE_INTERLACE) && codec->config.interlacing)
741 :     frame.vol_flags |= XVID_VOL_INTERLACING;
742 :    
743 : suxen_drol 1607 /* dxn: force 1:1 picture aspect ration */
744 : Isibaar 1650 if ((profiles[codec->config.profile].flags & PROFILE_EXTRA)) {
745 : suxen_drol 1607 frame.par = XVID_PAR_11_VGA;
746 :     } else if (codec->config.ar_mode == 0) { /* PAR */
747 : edgomez 1382 if (codec->config.display_aspect_ratio != 5) {
748 :     frame.par = codec->config.display_aspect_ratio + 1;
749 :     } else {
750 :     frame.par = XVID_PAR_EXT;
751 :     frame.par_width = codec->config.par_x;
752 :     frame.par_height= codec->config.par_y;
753 :     }
754 :     } else { /* AR */
755 :     /* custom pixel aspect ratio -> calculated from DAR */
756 :     frame.par = XVID_PAR_EXT;
757 :     frame.par_width = (100 * inhdr->biHeight) / codec->config.ar_y;
758 :     frame.par_height= (100 * inhdr->biWidth) / codec->config.ar_x;
759 :     }
760 :    
761 :     /* vop stuff */
762 :    
763 :     frame.vop_flags |= XVID_VOP_HALFPEL;
764 :     frame.vop_flags |= XVID_VOP_HQACPRED;
765 :    
766 : syskin 1487 if (codec->config.interlacing && codec->config.tff)
767 :     frame.vop_flags |= XVID_VOP_TOPFIELDFIRST;
768 :    
769 :    
770 : edgomez 1382 if (codec->config.vop_debug)
771 :     frame.vop_flags |= XVID_VOP_DEBUG;
772 :    
773 : suxen_drol 1649 if (quality_preset->trellis_quant) {
774 : edgomez 1382 frame.vop_flags |= XVID_VOP_TRELLISQUANT;
775 :     }
776 :    
777 : suxen_drol 1607 if ((profiles[codec->config.profile].flags & PROFILE_4MV)) {
778 : suxen_drol 1649 if (quality_preset->motion_search > 4)
779 : suxen_drol 1607 frame.vop_flags |= XVID_VOP_INTER4V;
780 :     }
781 : edgomez 1382
782 : suxen_drol 1649 if (quality_preset->chromame)
783 : edgomez 1382 frame.motion |= XVID_ME_CHROMA_PVOP + XVID_ME_CHROMA_BVOP;
784 :    
785 : suxen_drol 1649 if (quality_preset->turbo)
786 : edgomez 1382 frame.motion |= XVID_ME_FASTREFINE16 | XVID_ME_FASTREFINE8 |
787 :     XVID_ME_SKIP_DELTASEARCH | XVID_ME_FAST_MODEINTERPOLATE |
788 :     XVID_ME_BFRAME_EARLYSTOP;
789 :    
790 : suxen_drol 1649 frame.motion |= pmvfast_presets[quality_preset->motion_search];
791 : edgomez 1382
792 : suxen_drol 1649 if (quality_preset->vhq_bframe) frame.vop_flags |= XVID_VOP_RD_BVOP;
793 : syskin 1510
794 :    
795 : suxen_drol 1649 switch (quality_preset->vhq_mode)
796 : edgomez 1382 {
797 :     case VHQ_MODE_DECISION :
798 :     frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
799 :     break;
800 :    
801 :     case VHQ_LIMITED_SEARCH :
802 :     frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
803 :     frame.motion |= XVID_ME_HALFPELREFINE16_RD;
804 :     frame.motion |= XVID_ME_QUARTERPELREFINE16_RD;
805 :     break;
806 :    
807 :     case VHQ_MEDIUM_SEARCH :
808 :     frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
809 :     frame.motion |= XVID_ME_HALFPELREFINE16_RD;
810 :     frame.motion |= XVID_ME_HALFPELREFINE8_RD;
811 :     frame.motion |= XVID_ME_QUARTERPELREFINE16_RD;
812 :     frame.motion |= XVID_ME_QUARTERPELREFINE8_RD;
813 :     frame.motion |= XVID_ME_CHECKPREDICTION_RD;
814 :     break;
815 :    
816 :     case VHQ_WIDE_SEARCH :
817 :     frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
818 :     frame.motion |= XVID_ME_HALFPELREFINE16_RD;
819 :     frame.motion |= XVID_ME_HALFPELREFINE8_RD;
820 :     frame.motion |= XVID_ME_QUARTERPELREFINE16_RD;
821 :     frame.motion |= XVID_ME_QUARTERPELREFINE8_RD;
822 :     frame.motion |= XVID_ME_CHECKPREDICTION_RD;
823 :     frame.motion |= XVID_ME_EXTSEARCH_RD;
824 :     break;
825 :    
826 :     default :
827 :     break;
828 :     }
829 :    
830 : Isibaar 1914 if (quality_preset->vhq_metric == 1)
831 :     frame.vop_flags |= XVID_VOP_RD_PSNRHVSM;
832 :    
833 : edgomez 1382 frame.input.plane[0] = icc->lpInput;
834 : suxen_drol 1495 frame.input.stride[0] = CALC_BI_STRIDE(icc->lpbiInput->biWidth, icc->lpbiInput->biBitCount);
835 : edgomez 1382
836 :     if ((frame.input.csp = get_colorspace(inhdr)) == XVID_CSP_NULL)
837 :     return ICERR_BADFORMAT;
838 :    
839 :     if (frame.input.csp == XVID_CSP_I420 || frame.input.csp == XVID_CSP_YV12) {
840 :     frame.input.stride[0] = (4 * icc->lpbiInput->biWidth + 3) / 4;
841 :     frame.input.stride[1] = frame.input.stride[2] = frame.input.stride[0] / 2 ;
842 :     }
843 :    
844 :     frame.bitstream = icc->lpOutput;
845 :     frame.length = icc->lpbiOutput->biSizeImage;
846 :    
847 :     frame.quant = 0;
848 :    
849 :     if (codec->config.mode == RC_MODE_NULL) {
850 :     outhdr->biSizeImage = 0;
851 :     *icc->lpdwFlags = AVIIF_KEYFRAME;
852 :     return ICERR_OK;
853 :     }
854 :    
855 :     // force keyframe spacing in 2-pass 1st pass
856 : suxen_drol 1649 if (quality_preset->motion_search == 0)
857 : edgomez 1382 frame.type = XVID_TYPE_IVOP;
858 :    
859 :     /* frame-based stuff */
860 :     apply_zone_modifiers(&frame, &codec->config, codec->framenum);
861 :    
862 :     /* call encore */
863 :    
864 :     memset(&stats, 0, sizeof(stats));
865 :     stats.version = XVID_VERSION;
866 :    
867 : edgomez 1398 length = codec->xvid_encore_func(codec->ehandle, XVID_ENC_ENCODE, &frame, &stats);
868 : edgomez 1382 switch (length)
869 :     {
870 :     case XVID_ERR_FAIL :
871 :     return ICERR_ERROR;
872 :    
873 :     case XVID_ERR_MEMORY :
874 :     return ICERR_MEMORY;
875 :    
876 :     case XVID_ERR_FORMAT :
877 :     return ICERR_BADFORMAT;
878 :    
879 :     case XVID_ERR_VERSION :
880 :     return ICERR_UNSUPPORTED;
881 :     }
882 :    
883 :     if (codec->config.display_status && stats.type>0) {
884 :     status_update(&codec->status, stats.type, stats.length, stats.quant);
885 :     }
886 :    
887 :     DPRINTF("{type=%i len=%i} length=%i", stats.type, stats.length, length);
888 :    
889 :     if (length == 0) /* no encoder output */
890 :     {
891 :     *icc->lpdwFlags = 0;
892 :     ((char*)icc->lpOutput)[0] = 0x7f; /* virtual dub skip frame */
893 :     outhdr->biSizeImage = 1;
894 :    
895 :     }else{
896 :     if (frame.out_flags & XVID_KEYFRAME)
897 :     {
898 :     codec->keyspacing = 0;
899 :     *icc->lpdwFlags = AVIIF_KEYFRAME;
900 :     }
901 :     else
902 :     {
903 :     *icc->lpdwFlags = 0;
904 :     }
905 :    
906 :     outhdr->biSizeImage = length;
907 :    
908 :     if (codec->config.mode == RC_MODE_2PASS1 && codec->config.discard1pass)
909 :     {
910 :     outhdr->biSizeImage = 0;
911 :     }
912 :     }
913 :    
914 :     codec->framenum++;
915 :     codec->keyspacing++;
916 :    
917 :     return ICERR_OK;
918 :     }
919 :    
920 :    
921 :     /* decompressor */
922 :    
923 :    
924 :     LRESULT decompress_query(CODEC * codec, BITMAPINFO *lpbiInput, BITMAPINFO *lpbiOutput)
925 :     {
926 :     BITMAPINFOHEADER * inhdr = &lpbiInput->bmiHeader;
927 :     BITMAPINFOHEADER * outhdr = &lpbiOutput->bmiHeader;
928 : Isibaar 1945 int in_csp = XVID_CSP_NULL, out_csp = XVID_CSP_NULL;
929 : edgomez 1382
930 :     if (lpbiInput == NULL)
931 :     {
932 :     return ICERR_ERROR;
933 :     }
934 :    
935 : Isibaar 1952 if (inhdr->biCompression != FOURCC_XVID && inhdr->biCompression != FOURCC_DIVX && inhdr->biCompression != FOURCC_DX50 && inhdr->biCompression != FOURCC_MP4V &&
936 :     inhdr->biCompression != FOURCC_xvid && inhdr->biCompression != FOURCC_divx && inhdr->biCompression != FOURCC_dx50 && inhdr->biCompression != FOURCC_mp4v &&
937 :     (in_csp = get_colorspace(inhdr)) != XVID_CSP_YV12)
938 : edgomez 1382 {
939 :     return ICERR_BADFORMAT;
940 :     }
941 :    
942 :     if (lpbiOutput == NULL)
943 :     {
944 :     return ICERR_OK;
945 :     }
946 :    
947 : Isibaar 1945 out_csp = get_colorspace(outhdr);
948 :    
949 : edgomez 1382 if (inhdr->biWidth != outhdr->biWidth ||
950 :     inhdr->biHeight != outhdr->biHeight ||
951 : Isibaar 1945 out_csp == XVID_CSP_NULL ||
952 :     (in_csp == XVID_CSP_YV12 && in_csp != out_csp))
953 : edgomez 1382 {
954 :     return ICERR_BADFORMAT;
955 :     }
956 :    
957 :     return ICERR_OK;
958 :     }
959 :    
960 :    
961 :     LRESULT decompress_get_format(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)
962 :     {
963 :     BITMAPINFOHEADER * inhdr = &lpbiInput->bmiHeader;
964 :     BITMAPINFOHEADER * outhdr = &lpbiOutput->bmiHeader;
965 :     LRESULT result;
966 :    
967 :     if (lpbiOutput == NULL)
968 :     {
969 :     return sizeof(BITMAPINFOHEADER);
970 :     }
971 :    
972 :     /* --- yv12 --- */
973 :    
974 :     if (get_colorspace(inhdr) != XVID_CSP_NULL) {
975 :     memcpy(outhdr, inhdr, sizeof(BITMAPINFOHEADER));
976 :     /* XXX: should we set outhdr->biSize ?? */
977 :     return ICERR_OK;
978 :     }
979 :     /* --- yv12 --- */
980 :    
981 :     result = decompress_query(codec, lpbiInput, lpbiOutput);
982 :     if (result != ICERR_OK)
983 :     {
984 :     return result;
985 :     }
986 :    
987 :     outhdr->biSize = sizeof(BITMAPINFOHEADER);
988 :     outhdr->biWidth = inhdr->biWidth;
989 :     outhdr->biHeight = inhdr->biHeight;
990 :     outhdr->biPlanes = 1;
991 :     outhdr->biBitCount = 24;
992 :     outhdr->biCompression = BI_RGB; /* sonic foundry vegas video v3 only supports BI_RGB */
993 : suxen_drol 1495 outhdr->biSizeImage = outhdr->biHeight * CALC_BI_STRIDE(outhdr->biWidth, outhdr->biBitCount);
994 :    
995 : edgomez 1382 outhdr->biXPelsPerMeter = 0;
996 :     outhdr->biYPelsPerMeter = 0;
997 :     outhdr->biClrUsed = 0;
998 :     outhdr->biClrImportant = 0;
999 :    
1000 :     return ICERR_OK;
1001 :     }
1002 :    
1003 :     #define REG_GET_N(X, Y, Z) \
1004 :     { \
1005 :     DWORD size = sizeof(int); \
1006 :     if (RegQueryValueEx(hKey, X, 0, 0, (LPBYTE)&Y, &size) != ERROR_SUCCESS) { \
1007 :     Y=Z; \
1008 :     } \
1009 :     }while(0)
1010 :    
1011 :     LRESULT decompress_begin(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)
1012 :     {
1013 : Isibaar 1890 BITMAPINFOHEADER * inhdr = &lpbiInput->bmiHeader;
1014 : edgomez 1382 xvid_gbl_init_t init;
1015 : Isibaar 1914 xvid_gbl_info_t info;
1016 : edgomez 1382 xvid_dec_create_t create;
1017 :     HKEY hKey;
1018 :    
1019 : edgomez 1398 if (init_dll(codec) != 0) return ICERR_ERROR;
1020 : edgomez 1382
1021 :     memset(&init, 0, sizeof(init));
1022 :     init.version = XVID_VERSION;
1023 :     init.cpu_flags = codec->config.cpu;
1024 : Isibaar 1890 init.debug = codec->config.debug;
1025 : edgomez 1398 codec->xvid_global_func(0, XVID_GBL_INIT, &init, NULL);
1026 : edgomez 1382
1027 : Isibaar 1914 memset(&info, 0, sizeof(info));
1028 :     info.version = XVID_VERSION;
1029 :     codec->xvid_global_func(0, XVID_GBL_INFO, &info, NULL);
1030 :    
1031 : edgomez 1382 memset(&create, 0, sizeof(create));
1032 :     create.version = XVID_VERSION;
1033 :     create.width = lpbiInput->bmiHeader.biWidth;
1034 :     create.height = lpbiInput->bmiHeader.biHeight;
1035 : Isibaar 1890 create.fourcc = inhdr->biCompression;
1036 : edgomez 1382
1037 : Isibaar 1914 /* Decoder threads */
1038 :     if (codec->config.cpu & XVID_CPU_FORCE)
1039 :     create.num_threads = codec->config.num_threads;
1040 :     else
1041 :     create.num_threads = info.num_threads; /* Autodetect */
1042 :    
1043 : edgomez 1398 switch(codec->xvid_decore_func(0, XVID_DEC_CREATE, &create, NULL))
1044 : edgomez 1382 {
1045 :     case XVID_ERR_FAIL :
1046 :     return ICERR_ERROR;
1047 :    
1048 :     case XVID_ERR_MEMORY :
1049 :     return ICERR_MEMORY;
1050 :    
1051 :     case XVID_ERR_FORMAT :
1052 :     return ICERR_BADFORMAT;
1053 :    
1054 :     case XVID_ERR_VERSION :
1055 :     return ICERR_UNSUPPORTED;
1056 :     }
1057 :    
1058 :     codec->dhandle = create.handle;
1059 :    
1060 :     RegOpenKeyEx(XVID_REG_KEY, XVID_REG_PARENT "\\" XVID_REG_CHILD, 0, KEY_READ, &hKey);
1061 :    
1062 : suxen_drol 1397 REG_GET_N("Brightness", pp_brightness, 0);
1063 : Isibaar 1887 REG_GET_N("Deblock_Y", pp_dy, 0);
1064 :     REG_GET_N("Deblock_UV", pp_duv, 0);
1065 :     REG_GET_N("Dering_Y", pp_dry, 0);
1066 :     REG_GET_N("Dering_UV", pp_druv, 0);
1067 :     REG_GET_N("FilmEffect", pp_fe, 0);
1068 : edgomez 1382
1069 :     RegCloseKey(hKey);
1070 :    
1071 :     return ICERR_OK;
1072 :     }
1073 :    
1074 :    
1075 :     LRESULT decompress_end(CODEC * codec)
1076 :     {
1077 : edgomez 1398 if (codec->m_hdll != NULL) {
1078 : edgomez 1382 if (codec->dhandle != NULL) {
1079 : edgomez 1398 codec->xvid_decore_func(codec->dhandle, XVID_DEC_DESTROY, NULL, NULL);
1080 : edgomez 1382 codec->dhandle = NULL;
1081 :     }
1082 :     }
1083 :    
1084 :     return ICERR_OK;
1085 :     }
1086 :    
1087 :    
1088 :     LRESULT decompress(CODEC * codec, ICDECOMPRESS * icd)
1089 :     {
1090 :     xvid_dec_frame_t frame;
1091 :    
1092 :     /* --- yv12 --- */
1093 :     if (icd->lpbiInput->biCompression != FOURCC_XVID &&
1094 :     icd->lpbiInput->biCompression != FOURCC_DIVX &&
1095 : Isibaar 1952 icd->lpbiInput->biCompression != FOURCC_DX50 &&
1096 :     icd->lpbiInput->biCompression != FOURCC_MP4V &&
1097 :     icd->lpbiInput->biCompression != FOURCC_xvid &&
1098 :     icd->lpbiInput->biCompression != FOURCC_divx &&
1099 :     icd->lpbiInput->biCompression != FOURCC_dx50 &&
1100 :     icd->lpbiInput->biCompression != FOURCC_mp4v)
1101 : edgomez 1382 {
1102 :     xvid_gbl_convert_t convert;
1103 :    
1104 :     DPRINTF("input=%c%c%c%c output=%c%c%c%c",
1105 :     icd->lpbiInput->biCompression&0xff,
1106 :     (icd->lpbiInput->biCompression>>8)&0xff,
1107 :     (icd->lpbiInput->biCompression>>16)&0xff,
1108 :     (icd->lpbiInput->biCompression>>24)&0xff,
1109 :     icd->lpbiOutput->biCompression&0xff,
1110 :     (icd->lpbiOutput->biCompression>>8)&0xff,
1111 :     (icd->lpbiOutput->biCompression>>16)&0xff,
1112 :     (icd->lpbiOutput->biCompression>>24)&0xff);
1113 :    
1114 :     memset(&convert, 0, sizeof(convert));
1115 :     convert.version = XVID_VERSION;
1116 :    
1117 :     convert.input.csp = get_colorspace(icd->lpbiInput);
1118 :     convert.input.plane[0] = icd->lpInput;
1119 : suxen_drol 1495 convert.input.stride[0] = CALC_BI_STRIDE(icd->lpbiInput->biWidth, icd->lpbiInput->biBitCount);
1120 : edgomez 1382 if (convert.input.csp == XVID_CSP_I420 || convert.input.csp == XVID_CSP_YV12)
1121 :     convert.input.stride[0] = (convert.input.stride[0]*2)/3;
1122 :    
1123 :     convert.output.csp = get_colorspace(icd->lpbiOutput);
1124 :     convert.output.plane[0] = icd->lpOutput;
1125 : suxen_drol 1495 convert.output.stride[0] = CALC_BI_STRIDE(icd->lpbiOutput->biWidth, icd->lpbiOutput->biBitCount);
1126 : edgomez 1382 if (convert.output.csp == XVID_CSP_I420 || convert.output.csp == XVID_CSP_YV12)
1127 :     convert.output.stride[0] = (convert.output.stride[0]*2)/3;
1128 :    
1129 :     convert.width = icd->lpbiInput->biWidth;
1130 :     convert.height = icd->lpbiInput->biHeight;
1131 :     convert.interlacing = 0;
1132 :     if (convert.input.csp == XVID_CSP_NULL ||
1133 :     convert.output.csp == XVID_CSP_NULL ||
1134 : edgomez 1398 codec->xvid_global_func(0, XVID_GBL_CONVERT, &convert, NULL) < 0)
1135 : edgomez 1382 {
1136 :     return ICERR_BADFORMAT;
1137 :     }
1138 :     return ICERR_OK;
1139 :     }
1140 :     /* --- yv12 --- */
1141 :    
1142 :     memset(&frame, 0, sizeof(frame));
1143 :     frame.version = XVID_VERSION;
1144 :     frame.general = XVID_LOWDELAY; /* force low_delay_default mode */
1145 :     frame.bitstream = icd->lpInput;
1146 :     frame.length = icd->lpbiInput->biSizeImage;
1147 :    
1148 :     if (~((icd->dwFlags & ICDECOMPRESS_HURRYUP) | (icd->dwFlags & ICDECOMPRESS_UPDATE) | (icd->dwFlags & ICDECOMPRESS_PREROLL)))
1149 :     {
1150 :     if ((frame.output.csp = get_colorspace(icd->lpbiOutput)) == XVID_CSP_NULL)
1151 :     {
1152 :     return ICERR_BADFORMAT;
1153 :     }
1154 :     frame.output.plane[0] = icd->lpOutput;
1155 : suxen_drol 1495 frame.output.stride[0] = CALC_BI_STRIDE(icd->lpbiOutput->biWidth, icd->lpbiOutput->biBitCount);
1156 : edgomez 1382 if (frame.output.csp == XVID_CSP_I420 || frame.output.csp == XVID_CSP_YV12)
1157 : suxen_drol 1558 frame.output.stride[0] = CALC_BI_STRIDE(icd->lpbiOutput->biWidth, 8);
1158 : edgomez 1382 }
1159 :     else
1160 :     {
1161 :     frame.output.csp = XVID_CSP_NULL;
1162 :     }
1163 :    
1164 :     if (pp_dy)frame.general |= XVID_DEBLOCKY;
1165 :     if (pp_duv) frame.general |= XVID_DEBLOCKUV;
1166 : syskin 1437 if (pp_dry) frame.general |= XVID_DERINGY;
1167 :     if (pp_druv) frame.general |= XVID_DERINGUV;
1168 : edgomez 1382 if (pp_fe) frame.general |= XVID_FILMEFFECT;
1169 :    
1170 : suxen_drol 1397 frame.brightness = pp_brightness;
1171 :    
1172 : edgomez 1398 switch (codec->xvid_decore_func(codec->dhandle, XVID_DEC_DECODE, &frame, NULL))
1173 : edgomez 1382 {
1174 :     case XVID_ERR_FAIL :
1175 :     return ICERR_ERROR;
1176 :    
1177 :     case XVID_ERR_MEMORY :
1178 :     return ICERR_MEMORY;
1179 :    
1180 :     case XVID_ERR_FORMAT :
1181 :     return ICERR_BADFORMAT;
1182 :    
1183 :     case XVID_ERR_VERSION :
1184 :     return ICERR_UNSUPPORTED;
1185 :     }
1186 :    
1187 :     return ICERR_OK;
1188 :     }
1189 :    

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