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

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

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 949, Wed Mar 26 14:56:49 2003 UTC revision 1393, Wed Mar 31 11:55:34 2004 UTC
# Line 49  Line 49 
49    
50  #include <windows.h>  #include <windows.h>
51  #include <vfw.h>  #include <vfw.h>
52    #include <stdio.h>
53    #include "vfwext.h"
54    
55    #include <xvid.h>
56    #include "debug.h"
57  #include "codec.h"  #include "codec.h"
58  #include "2pass.h"  #include "status.h"
59    
60  int pmvfast_presets[7] = {  
61    
62    static const int pmvfast_presets[7] = {
63          0, 0, 0, 0,          0, 0, 0, 0,
64          0 | XVID_ME_HALFPELREFINE16 | 0,          0 | XVID_ME_HALFPELREFINE16 | 0,
65          0 | XVID_ME_HALFPELREFINE16 | 0 |          0 | XVID_ME_HALFPELREFINE16 | 0 |
# Line 61  Line 67 
67          XVID_ME_HALFPELREFINE8 | 0 | XVID_ME_USESQUARES16          XVID_ME_HALFPELREFINE8 | 0 | XVID_ME_USESQUARES16
68  };  };
69    
70    
71    
72  /*      return xvid compatbile colorspace,  /*      return xvid compatbile colorspace,
73          or XVID_CSP_NULL if failure          or XVID_CSP_NULL if failure
74  */  */
75    
76  int get_colorspace(BITMAPINFOHEADER * hdr)  static int get_colorspace(BITMAPINFOHEADER * hdr)
77  {  {
78          /* rgb only: negative height specifies top down image */          /* rgb only: negative height specifies top down image */
79          int rgb_flip = (hdr->biHeight < 0 ? 0 : XVID_CSP_VFLIP);          int rgb_flip = (hdr->biHeight < 0 ? 0 : XVID_CSP_VFLIP);
# Line 75  Line 83 
83          case BI_RGB :          case BI_RGB :
84                  if (hdr->biBitCount == 16)                  if (hdr->biBitCount == 16)
85                  {                  {
86                          DEBUG("RGB16 (RGB555)");                          DPRINTF("RGB16 (RGB555)");
87                          return rgb_flip | XVID_CSP_RGB555;                          return rgb_flip | XVID_CSP_RGB555;
88                  }                  }
89                  if (hdr->biBitCount == 24)                  if (hdr->biBitCount == 24)
90                  {                  {
91                          DEBUG("RGB24");                          DPRINTF("RGB24");
92                          return rgb_flip | XVID_CSP_BGR;                          return rgb_flip | XVID_CSP_BGR;
93                  }                  }
94                  if (hdr->biBitCount == 32)                  if (hdr->biBitCount == 32)
95                  {                  {
96                          DEBUG("RGB32");                          DPRINTF("RGB32");
97                          return rgb_flip | XVID_CSP_BGRA;                          return rgb_flip | XVID_CSP_BGRA;
98                  }                  }
99    
100                  DEBUG1("unsupported BI_RGB biBitCount", hdr->biBitCount);                  DPRINTF("unsupported BI_RGB biBitCount=%i", hdr->biBitCount);
101                  return XVID_CSP_NULL;                  return XVID_CSP_NULL;
102    
103          case BI_BITFIELDS :          case BI_BITFIELDS :
# Line 102  Line 110 
110                                  hdr4->bV4GreenMask == 0x3e0 &&                                  hdr4->bV4GreenMask == 0x3e0 &&
111                                  hdr4->bV4BlueMask == 0x1f)                                  hdr4->bV4BlueMask == 0x1f)
112                          {                          {
113                                  DEBUG("RGB555");                                  DPRINTF("RGB555");
114                                  return rgb_flip | XVID_CSP_RGB555;                                  return rgb_flip | XVID_CSP_RGB555;
115                          }                          }
116    
# Line 111  Line 119 
119                                  hdr4->bV4GreenMask == 0x7e0 &&                                  hdr4->bV4GreenMask == 0x7e0 &&
120                                  hdr4->bV4BlueMask == 0x1f)                                  hdr4->bV4BlueMask == 0x1f)
121                          {                          {
122                                  DEBUG("RGB565");                                  DPRINTF("RGB565");
123                                  return rgb_flip | XVID_CSP_RGB565;                                  return rgb_flip | XVID_CSP_RGB565;
124                          }                          }
125    
126                          DEBUG("unsupported BI_BITFIELDS mode");                          DPRINTF("unsupported BI_BITFIELDS mode");
127                          return XVID_CSP_NULL;                          return XVID_CSP_NULL;
128                  }                  }
129    
130                  DEBUG("unsupported BI_BITFIELDS/BITMAPHEADER combination");                  DPRINTF("unsupported BI_BITFIELDS/BITMAPHEADER combination");
131                  return XVID_CSP_NULL;                  return XVID_CSP_NULL;
132    
133          case FOURCC_I420 :          case FOURCC_I420 :
134          case FOURCC_IYUV :          case FOURCC_IYUV :
135                  DEBUG("IYUY");                  DPRINTF("IYUY");
136                  return XVID_CSP_I420;                  return XVID_CSP_I420;
137    
138          case FOURCC_YV12 :          case FOURCC_YV12 :
139                  DEBUG("YV12");                  DPRINTF("YV12");
140                  return XVID_CSP_YV12;                  return XVID_CSP_YV12;
141    
142          case FOURCC_YUYV :          case FOURCC_YUYV :
143          case FOURCC_YUY2 :          case FOURCC_YUY2 :
144                  DEBUG("YUY2");                  DPRINTF("YUY2");
145                  return XVID_CSP_YUY2;                  return XVID_CSP_YUY2;
146    
147          case FOURCC_YVYU :          case FOURCC_YVYU :
148                  DEBUG("YVYU");                  DPRINTF("YVYU");
149                  return XVID_CSP_YVYU;                  return XVID_CSP_YVYU;
150    
151          case FOURCC_UYVY :          case FOURCC_UYVY :
152                  DEBUG("UYVY");                  DPRINTF("UYVY");
153                  return XVID_CSP_UYVY;                  return XVID_CSP_UYVY;
154    
155          default :          default :
156                  DEBUGFOURCC("unsupported colorspace", hdr->biCompression);                  DPRINTF("unsupported colorspace %c%c%c%c",
157                            hdr->biCompression&0xff,
158                            (hdr->biCompression>>8)&0xff,
159                            (hdr->biCompression>>16)&0xff,
160                            (hdr->biCompression>>24)&0xff);
161                  return XVID_CSP_NULL;                  return XVID_CSP_NULL;
162          }          }
163  }  }
# Line 161  Line 173 
173          BITMAPINFOHEADER * inhdr = &lpbiInput->bmiHeader;          BITMAPINFOHEADER * inhdr = &lpbiInput->bmiHeader;
174          BITMAPINFOHEADER * outhdr = &lpbiOutput->bmiHeader;          BITMAPINFOHEADER * outhdr = &lpbiOutput->bmiHeader;
175    
176            /* VFWEXT detection */
177            if (inhdr->biCompression == VFWEXT_FOURCC) {
178                    return (ICM_USER+0x0fff);
179            }
180    
181          if (get_colorspace(inhdr) == XVID_CSP_NULL)          if (get_colorspace(inhdr) == XVID_CSP_NULL)
182          {          {
183                  return ICERR_BADFORMAT;                  return ICERR_BADFORMAT;
# Line 193  Line 210 
210    
211          if (lpbiOutput == NULL)          if (lpbiOutput == NULL)
212          {          {
213                  return sizeof(BITMAPV4HEADER);                  return sizeof(BITMAPINFOHEADER);
214          }          }
215    
216          memcpy(outhdr, inhdr, sizeof(BITMAPINFOHEADER));          memcpy(outhdr, inhdr, sizeof(BITMAPINFOHEADER));
# Line 229  Line 246 
246    
247  LRESULT compress_frames_info(CODEC * codec, ICCOMPRESSFRAMES * icf)  LRESULT compress_frames_info(CODEC * codec, ICCOMPRESSFRAMES * icf)
248  {  {
249          // DEBUG2("frate fscale", codec->frate, codec->fscale);  #if 0
250            DPRINTF("%i %i", icf->lStartFrame, icf->lFrameCount);
251    #endif
252          codec->fincr = icf->dwScale;          codec->fincr = icf->dwScale;
253          codec->fbase = icf->dwRate;          codec->fbase = icf->dwRate;
254          return ICERR_OK;          return ICERR_OK;
255  }  }
256    
257    
258    static char type2char(int type)
259    {
260            if (type==XVID_TYPE_IVOP)
261                    return 'I';
262            if (type==XVID_TYPE_PVOP)
263                    return 'P';
264            if (type==XVID_TYPE_BVOP)
265                    return 'B';
266            return 'S';
267    }
268    
269    static int vfw_debug(void *handle,
270                             int opt,
271                             void *param1,
272                             void *param2)
273    {
274            switch (opt) {
275            case XVID_PLG_CREATE:
276                    *((void**)param2) = NULL;
277            case XVID_PLG_INFO:
278            case XVID_PLG_DESTROY:
279            case XVID_PLG_BEFORE:
280                    return 0;
281    
282            case XVID_PLG_AFTER:
283                    {
284                            xvid_plg_data_t *data = (xvid_plg_data_t *) param1;
285    
286                            /* We don't use DPRINTF here because it's active only for _DEBUG
287                             * builds and that activates lot of other debug printfs. We only
288                             * want these all the time */
289                            char buf[1024];
290                            sprintf(buf, "[%6i]   type=%c   Q:%2i   length:%6i",
291                                            data->frame_num,
292                                            type2char(data->type),
293                                            data->quant,
294                                            data->length);
295                            OutputDebugString(buf);
296    
297                            return 0;
298                    }
299            }
300    
301            return XVID_ERR_FAIL;
302    }
303    
304    #define XVID_DLL_NAME "xvidcore.dll"
305    
306    static int init_dll(CODEC* codec)
307    {
308            if (codec->m_hdll != NULL)
309                    return 0;
310    
311            DPRINTF("init_dll");
312            codec->m_hdll = LoadLibrary(XVID_DLL_NAME);
313            if (codec->m_hdll == NULL) {
314                    DPRINTF("dll load failed");
315                    MessageBox(0, XVID_DLL_NAME " not found!","Error!", MB_ICONEXCLAMATION|MB_OK);
316                    return XVID_ERR_FAIL;
317            }
318    
319            codec->xvid_global_func = (int (__cdecl *)(void *, int, void *, void *))GetProcAddress(codec->m_hdll, "xvid_global");
320            if (codec->xvid_global_func == NULL) {
321                    MessageBox(0, "xvid_global() not found", "Error", 0);
322                    return XVID_ERR_FAIL;
323            }
324    
325            codec->xvid_encore_func = (int (__cdecl *)(void *, int, void *, void *))GetProcAddress(codec->m_hdll, "xvid_encore");
326            if (codec->xvid_encore_func == NULL) {
327                    MessageBox(0, "xvid_encore() not found", "Error", 0);
328                    return XVID_ERR_FAIL;
329            }
330    
331            codec->xvid_decore_func = (int (__cdecl *)(void *, int, void *, void *))GetProcAddress(codec->m_hdll, "xvid_decore");
332            if (codec->xvid_decore_func == NULL) {
333                    MessageBox(0, "xvid_decore() not found", "Error", 0);
334                    return XVID_ERR_FAIL;
335            }
336    
337            codec->xvid_plugin_single_func =
338                    (int (__cdecl *)(void *, int, void *, void *))(GetProcAddress(codec->m_hdll, "xvid_plugin_single"));
339            codec->xvid_plugin_2pass1_func =
340                    (int (__cdecl *)(void *, int, void *, void *))(GetProcAddress(codec->m_hdll, "xvid_plugin_2pass1"));
341            codec->xvid_plugin_2pass2_func =
342                    (int (__cdecl *)(void *, int, void *, void *))(GetProcAddress(codec->m_hdll, "xvid_plugin_2pass2"));
343            codec->xvid_plugin_lumimasking_func =
344                    (int (__cdecl *)(void *, int, void *, void *))(GetProcAddress(codec->m_hdll, "xvid_plugin_lumimasking"));
345            codec->xvid_plugin_psnr_func =
346                    (int (__cdecl *)(void *, int, void *, void *))(GetProcAddress(codec->m_hdll, "xvid_plugin_psnr"));
347    
348            return 0;
349    }
350    
351    static int exit_dll(CODEC* codec)
352    {
353            if(codec->m_hdll)
354            {
355                    FreeLibrary(codec->m_hdll);
356                    codec->m_hdll = NULL;
357                    codec->xvid_global_func = NULL;
358                    codec->xvid_encore_func = NULL;
359                    codec->xvid_decore_func = NULL;
360                    codec->xvid_plugin_single_func = NULL;
361                    codec->xvid_plugin_2pass1_func = NULL;
362                    codec->xvid_plugin_2pass2_func = NULL;
363                    codec->xvid_plugin_lumimasking_func = NULL;
364                    codec->xvid_plugin_psnr_func = NULL;
365            }
366            return 0;
367    }
368    
369    /* constant-quant zones for fixed quant encoding */
370    static void
371    prepare_cquant_zones(CONFIG * config) {
372    
373            int i = 0;
374            if (config->num_zones == 0 || config->zones[0].frame != 0) {
375                    /* first zone does not start at frame 0 or doesn't exist */
376    
377                    if (config->num_zones >= MAX_ZONES) config->num_zones--; /* we scrifice last zone */
378    
379                    config->zones[config->num_zones].frame = 0;
380                    config->zones[config->num_zones].mode = RC_ZONE_QUANT;
381                    config->zones[config->num_zones].weight = 100;
382                    config->zones[config->num_zones].quant = config->desired_quant;
383                    config->zones[config->num_zones].type = XVID_TYPE_AUTO;
384                    config->zones[config->num_zones].greyscale = 0;
385                    config->zones[config->num_zones].chroma_opt = 0;
386                    config->zones[config->num_zones].bvop_threshold = 0;
387                    config->num_zones++;
388    
389                    sort_zones(config->zones, config->num_zones, &i);
390            }
391    
392            /* step 2: let's change all weight zones into quant zones */
393    
394            for(i = 0; i < config->num_zones; i++)
395                    if (config->zones[i].mode == RC_ZONE_WEIGHT) {
396                            config->zones[i].mode = RC_ZONE_QUANT;
397                            config->zones[i].quant = (100*config->desired_quant) / config->zones[i].weight;
398                    }
399    }
400    
401    /* full first pass zones */
402    static void
403    prepare_full1pass_zones(CONFIG * config) {
404    
405            int i = 0;
406            if (config->num_zones == 0 || config->zones[0].frame != 0) {
407                    /* first zone does not start at frame 0 or doesn't exist */
408    
409                    if (config->num_zones >= MAX_ZONES) config->num_zones--; /* we scrifice last zone */
410    
411                    config->zones[config->num_zones].frame = 0;
412                    config->zones[config->num_zones].mode = RC_ZONE_QUANT;
413                    config->zones[config->num_zones].weight = 100;
414                    config->zones[config->num_zones].quant = 200;
415                    config->zones[config->num_zones].type = XVID_TYPE_AUTO;
416                    config->zones[config->num_zones].greyscale = 0;
417                    config->zones[config->num_zones].chroma_opt = 0;
418                    config->zones[config->num_zones].bvop_threshold = 0;
419                    config->num_zones++;
420    
421                    sort_zones(config->zones, config->num_zones, &i);
422            }
423    
424            /* step 2: let's change all weight zones into quant zones */
425    
426            for(i = 0; i < config->num_zones; i++)
427                    if (config->zones[i].mode == RC_ZONE_WEIGHT) {
428                            config->zones[i].mode = RC_ZONE_QUANT;
429                            config->zones[i].quant = 200;
430                    }
431    }
432    
433    
434  LRESULT compress_begin(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)  LRESULT compress_begin(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)
435  {  {
436          xvid_gbl_init_t init;          xvid_gbl_init_t init;
437          xvid_enc_create_t create;          xvid_enc_create_t create;
438          xvid_enc_rc_t rc;          xvid_enc_plugin_t plugins[3];
439      xvid_enc_plugin_t plugins[1];          xvid_plugin_single_t single;
440            xvid_plugin_2pass1_t pass1;
441            xvid_plugin_2pass2_t pass2;
442            int i;
443            HANDLE hFile;
444    
445          memset(&rc, 0, sizeof(rc));          CONFIG tmpCfg; /* if we want to alter config to suit our needs, it shouldn't be visible to user later */
446          rc.version = XVID_VERSION;          memcpy(&tmpCfg, &codec->config, sizeof(CONFIG));
447    
448            if (init_dll(codec) != 0) return ICERR_ERROR;
449            /* destroy previously created codec */
450            if(codec->ehandle) {
451                    codec->xvid_encore_func(codec->ehandle, XVID_ENC_DESTROY, NULL, NULL);
452                    codec->ehandle = NULL;
453            }
454    
455            memset(&init, 0, sizeof(init));
456            init.version = XVID_VERSION;
457            init.cpu_flags = codec->config.cpu;
458            init.debug = codec->config.debug;
459            codec->xvid_global_func(0, XVID_GBL_INIT, &init, NULL);
460    
461            memset(&create, 0, sizeof(create));
462            create.version = XVID_VERSION;
463    
464            /* plugins */
465            create.plugins = plugins;
466          switch (codec->config.mode)          switch (codec->config.mode)
467          {          {
468          case DLG_MODE_CBR :          case RC_MODE_1PASS :
469                  //rc.type = XVID_RC_CBR;                  memset(&single, 0, sizeof(single));
470                  rc.min_iquant = codec->config.min_iquant;                  single.version = XVID_VERSION;
471                  rc.max_iquant = codec->config.max_iquant;                  single.bitrate = codec->config.bitrate * CONFIG_KBPS;
472                  rc.min_pquant = codec->config.min_pquant;                  single.reaction_delay_factor = codec->config.rc_reaction_delay_factor;
473                  rc.max_pquant = codec->config.max_pquant;                  single.averaging_period = codec->config.rc_averaging_period;
474                  rc.min_bquant = 0;      /* XXX: todo */                  single.buffer = codec->config.rc_buffer;
475                  rc.max_bquant = 0;      /* XXX: todo */                  plugins[create.num_plugins].func = codec->xvid_plugin_single_func;
476                  rc.data.cbr.bitrate = codec->config.rc_bitrate;                  plugins[create.num_plugins].param = &single;
477          rc.data.cbr.reaction_delay_factor = codec->config.rc_reaction_delay_factor;                  create.num_plugins++;
478                  rc.data.cbr.averaging_period = codec->config.rc_averaging_period;                  if (!codec->config.use_2pass_bitrate) /* constant-quant mode */
479                  rc.data.cbr.buffer = codec->config.rc_buffer;                          prepare_cquant_zones(&tmpCfg);
480                  break;                  break;
481    
482          case DLG_MODE_VBR_QUAL :          case RC_MODE_2PASS1 :
483                  codec->config.fquant = 0;                  memset(&pass1, 0, sizeof(pass1));
484                  //param.rc_bitrate = 0;                  pass1.version = XVID_VERSION;
485                    pass1.filename = codec->config.stats;
486                    if (codec->config.full1pass)
487                            prepare_full1pass_zones(&tmpCfg);
488                    plugins[create.num_plugins].func = codec->xvid_plugin_2pass1_func;
489                    plugins[create.num_plugins].param = &pass1;
490                    create.num_plugins++;
491                  break;                  break;
492    
493          case DLG_MODE_VBR_QUANT :          case RC_MODE_2PASS2 :
494                  codec->config.fquant = (float) codec->config.quant;                  memset(&pass2, 0, sizeof(pass2));
495                  //param.rc_bitrate = 0;                  pass2.version = XVID_VERSION;
496                  break;                  if (codec->config.use_2pass_bitrate) {
497                            pass2.bitrate = codec->config.bitrate * CONFIG_KBPS;
498                    } else {
499                            pass2.bitrate = -codec->config.desired_size;    /* kilobytes */
500                    }
501                    pass2.filename = codec->config.stats;
502    
503          case DLG_MODE_2PASS_1 :                  hFile = CreateFile(pass2.filename, 0, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, 0, NULL);
504          case DLG_MODE_2PASS_2_INT :                  if (hFile == INVALID_HANDLE_VALUE)
505          case DLG_MODE_2PASS_2_EXT :                  {
506                  //param.rc_bitrate = 0;                          MessageBox(0, "Statsfile not found!","Error!", MB_ICONEXCLAMATION|MB_OK);
507                  codec->twopass.max_framesize = (int)((double)codec->config.twopass_max_bitrate / 8.0 / ((double)codec->fbase / (double)codec->fincr));                          return XVID_ERR_FAIL;
508                    } else
509                    {
510                            CloseHandle(hFile);
511                    }
512    
513                    pass2.keyframe_boost = codec->config.keyframe_boost;   /* keyframe boost percentage: [0..100...]; */
514                    pass2.curve_compression_high = codec->config.curve_compression_high;
515                    pass2.curve_compression_low = codec->config.curve_compression_low;
516                    pass2.overflow_control_strength = codec->config.overflow_control_strength;
517                    pass2.max_overflow_improvement = codec->config.twopass_max_overflow_improvement;
518                    pass2.max_overflow_degradation = codec->config.twopass_max_overflow_degradation;
519                    pass2.kfreduction = codec->config.kfreduction;
520                    pass2.kfthreshold = codec->config.kfthreshold;
521                    pass2.container_frame_overhead = 24;    /* AVI */
522    
523                    plugins[create.num_plugins].func = codec->xvid_plugin_2pass2_func;
524                    plugins[create.num_plugins].param = &pass2;
525                    create.num_plugins++;
526                  break;                  break;
527    
528          case DLG_MODE_NULL :          case RC_MODE_NULL :
529                  return ICERR_OK;                  return ICERR_OK;
530    
531          default :          default :
532                  break;                  break;
533          }          }
534    
535      /* destroy previously created codec */          /* zones  - copy from tmpCfg in case we automatically altered them above */
536          if(codec->ehandle)          create.zones = malloc(sizeof(xvid_enc_zone_t) * tmpCfg.num_zones);
537          {          create.num_zones = tmpCfg.num_zones;
538                  xvid_encore(codec->ehandle, XVID_ENC_DESTROY, NULL, NULL);          for (i=0; i < create.num_zones; i++) {
539                  codec->ehandle = NULL;                  create.zones[i].frame = tmpCfg.zones[i].frame;
540                    if (tmpCfg.zones[i].mode == RC_ZONE_QUANT) {
541                            create.zones[i].mode = XVID_ZONE_QUANT;
542                            create.zones[i].increment = tmpCfg.zones[i].quant;
543                    }else{
544                            create.zones[i].mode = XVID_ZONE_WEIGHT;
545                            create.zones[i].increment = tmpCfg.zones[i].weight;
546                    }
547                    create.zones[i].base = 100;
548          }          }
549    
550      memset(&init, 0, sizeof(init));          /* lumimasking plugin */
551          init.version = XVID_VERSION;          if ((profiles[codec->config.profile].flags & PROFILE_ADAPTQUANT) && codec->config.lum_masking) {
552          init.cpu_flags = codec->config.cpu;                  plugins[create.num_plugins].func = codec->xvid_plugin_lumimasking_func;
553          xvid_global(0, XVID_GBL_INIT, &init, NULL);                  plugins[create.num_plugins].param = NULL;
554                    create.num_plugins++;
555            }
556    
557          memset(&create, 0, sizeof(create));          plugins[create.num_plugins].func = vfw_debug;
558          create.version = XVID_VERSION;          plugins[create.num_plugins].param = NULL;
559            create.num_plugins++;
560    
561            create.profile = profiles[codec->config.profile].id;
562    
563          create.width = lpbiInput->bmiHeader.biWidth;          create.width = lpbiInput->bmiHeader.biWidth;
564          create.height = lpbiInput->bmiHeader.biHeight;          create.height = lpbiInput->bmiHeader.biHeight;
565          create.fincr = codec->fincr;          create.fincr = codec->fincr;
566          create.fbase = codec->fbase;          create.fbase = codec->fbase;
567    
568      create.plugins = plugins;          create.max_key_interval = codec->config.max_key_interval;
569      create.num_plugins = 0;  
570          if (codec->config.lum_masking) {          create.min_quant[0] = codec->config.min_iquant;
571          plugins[create.num_plugins].func = xvid_plugin_lumimasking;          create.max_quant[0] = codec->config.max_iquant;
572          plugins[create.num_plugins].param = NULL;          create.min_quant[1] = codec->config.min_pquant;
573          create.num_plugins++;          create.max_quant[1] = codec->config.max_pquant;
574      }          create.min_quant[2] = codec->config.min_bquant;
575            create.max_quant[2] = codec->config.max_bquant;
576    
577            if ((profiles[codec->config.profile].flags & PROFILE_BVOP) && codec->config.use_bvop) {
578                    create.max_bframes = codec->config.max_bframes;
579                    create.bquant_ratio = codec->config.bquant_ratio;
580                    create.bquant_offset = codec->config.bquant_offset;
581    
582      if (codec->config.packed)      if (codec->config.packed)
583          create.global |= XVID_GLOBAL_PACKED;          create.global |= XVID_GLOBAL_PACKED;
584          if (codec->config.dx50bvop)  
585                    if (codec->config.closed_gov)
586                  create.global |= XVID_GLOBAL_CLOSED_GOP;                  create.global |= XVID_GLOBAL_CLOSED_GOP;
587    
588      create.max_key_interval = codec->config.max_key_interval;          }
         /* XXX: param.min_quantizer = codec->config.min_pquant;  
         param.max_quantizer = codec->config.max_pquant; */  
589    
         create.max_bframes = codec->config.max_bframes;  
590          create.frame_drop_ratio = codec->config.frame_drop_ratio;          create.frame_drop_ratio = codec->config.frame_drop_ratio;
591    
         create.bquant_ratio = codec->config.bquant_ratio;  
         create.bquant_offset = codec->config.bquant_offset;  
   
592          create.num_threads = codec->config.num_threads;          create.num_threads = codec->config.num_threads;
593    
594          switch(xvid_encore(0, XVID_ENC_CREATE, &create, (codec->config.mode==DLG_MODE_CBR)?&rc:NULL ))          switch(codec->xvid_encore_func(0, XVID_ENC_CREATE, &create, NULL))
595          {          {
596          case XVID_ERR_FAIL :          case XVID_ERR_FAIL :
597                  return ICERR_ERROR;                  return ICERR_ERROR;
# Line 350  Line 610 
610          codec->framenum = 0;          codec->framenum = 0;
611          codec->keyspacing = 0;          codec->keyspacing = 0;
612    
613          codec->twopass.hints = codec->twopass.stats1 = codec->twopass.stats2 = INVALID_HANDLE_VALUE;          if (codec->config.display_status) {
614          codec->twopass.hintstream = NULL;                  status_destroy_always(&codec->status);
615                    status_create(&codec->status, codec->fincr, codec->fbase);
616            }
617    
618          return ICERR_OK;          return ICERR_OK;
619  }  }
# Line 359  Line 621 
621    
622  LRESULT compress_end(CODEC * codec)  LRESULT compress_end(CODEC * codec)
623  {  {
624          if (codec->ehandle != NULL)          if (codec->m_hdll != NULL) {
625          {                  if (codec->ehandle != NULL) {
626                  xvid_encore(codec->ehandle, XVID_ENC_DESTROY, NULL, NULL);                          codec->xvid_encore_func(codec->ehandle, XVID_ENC_DESTROY, NULL, NULL);
627                            codec->ehandle = NULL;
                 if (codec->twopass.hints != INVALID_HANDLE_VALUE)  
                 {  
                         CloseHandle(codec->twopass.hints);  
628                  }                  }
629                  if (codec->twopass.stats1 != INVALID_HANDLE_VALUE)                  exit_dll(codec);
                 {  
                         CloseHandle(codec->twopass.stats1);  
630                  }                  }
631                  if (codec->twopass.stats2 != INVALID_HANDLE_VALUE)  
632                  {          if (codec->config.display_status)
633                          CloseHandle(codec->twopass.stats2);                  status_destroy(&codec->status);
634    
635            return ICERR_OK;
636                  }                  }
637                  if (codec->twopass.hintstream != NULL)  
638    
639    static void apply_zone_modifiers(xvid_enc_frame_t * frame, CONFIG * config, int framenum)
640                  {                  {
641                          free(codec->twopass.hintstream);          int i;
                 }  
642    
643                  codec->ehandle = NULL;          for (i=0; i<config->num_zones && config->zones[i].frame <= framenum; i++) ;
644    
645                  codec_2pass_finish(codec);          if (--i < 0) return; /* there are no zones, or we're before the first zone */
646    
647            if (framenum == config->zones[i].frame)
648                    frame->type = config->zones[i].type;
649    
650            if (config->zones[i].greyscale) {
651                    frame->vop_flags |= XVID_VOP_GREYSCALE;
652          }          }
653    
654          return ICERR_OK;          if (config->zones[i].chroma_opt) {
655                    frame->vop_flags |= XVID_VOP_CHROMAOPT;
656  }  }
657    
658            if ((profiles[config->profile].flags & PROFILE_BVOP) && config->use_bvop) {
659                    frame->bframe_threshold = config->zones[i].bvop_threshold;
660            }
661    }
662    
663    
664  LRESULT compress(CODEC * codec, ICCOMPRESS * icc)  LRESULT compress(CODEC * codec, ICCOMPRESS * icc)
665  {  {
666          BITMAPINFOHEADER * inhdr = icc->lpbiInput;          BITMAPINFOHEADER * inhdr = icc->lpbiInput;
# Line 396  Line 669 
669          xvid_enc_stats_t stats;          xvid_enc_stats_t stats;
670          int length;          int length;
671    
         // mpeg2avi yuv bug workaround (2 instances of CODEC)  
         if (codec->twopass.stats1 == INVALID_HANDLE_VALUE)  
         {  
                 if (codec_2pass_init(codec) == ICERR_ERROR)  
                 {  
                         return ICERR_ERROR;  
                 }  
         }  
   
672          memset(&frame, 0, sizeof(frame));          memset(&frame, 0, sizeof(frame));
673          frame.version = XVID_VERSION;          frame.version = XVID_VERSION;
674    
# Line 412  Line 676 
676    
677          /* vol stuff */          /* vol stuff */
678    
679          if (codec->config.quant_type != QUANT_MODE_H263)          if ((profiles[codec->config.profile].flags & PROFILE_MPEGQUANT) &&
680                    codec->config.quant_type != QUANT_MODE_H263)
681          {          {
682                  frame.vol_flags |= XVID_VOL_MPEGQUANT;                  frame.vol_flags |= XVID_VOL_MPEGQUANT;
683    
684                  // we actually need "default/custom" selectbox for both inter/intra                  if (codec->config.quant_type == QUANT_MODE_CUSTOM) {
                 // this will do for now  
                 if (codec->config.quant_type == QUANT_MODE_CUSTOM)  
                 {  
685                          frame.quant_intra_matrix = codec->config.qmatrix_intra;                          frame.quant_intra_matrix = codec->config.qmatrix_intra;
686                          frame.quant_inter_matrix = codec->config.qmatrix_inter;                          frame.quant_inter_matrix = codec->config.qmatrix_inter;
687                  }                  }else{
                 else  
                 {  
688                          frame.quant_intra_matrix = NULL;                          frame.quant_intra_matrix = NULL;
689                          frame.quant_inter_matrix = NULL;                          frame.quant_inter_matrix = NULL;
690                  }                  }
691          }          }
692    
693          if (codec->config.reduced_resolution) {          if ((profiles[codec->config.profile].flags & PROFILE_REDUCED) &&
694                    codec->config.reduced_resolution) {
695                  frame.vol_flags |= XVID_VOL_REDUCED_ENABLE;                  frame.vol_flags |= XVID_VOL_REDUCED_ENABLE;
696                  frame.vop_flags |= XVID_VOP_REDUCED;    /* XXX: need auto decion mode */                  frame.vop_flags |= XVID_VOP_REDUCED;    /* XXX: need auto decion mode */
697          }          }
698    
699          if (codec->config.qpel) {          if ((profiles[codec->config.profile].flags & PROFILE_QPEL) && codec->config.qpel) {
700                  frame.vol_flags |= XVID_VOL_QUARTERPEL;                  frame.vol_flags |= XVID_VOL_QUARTERPEL;
701                  frame.motion |= XVID_ME_QUARTERPELREFINE16 | XVID_ME_QUARTERPELREFINE8;                  frame.motion |= XVID_ME_QUARTERPELREFINE16 | XVID_ME_QUARTERPELREFINE8;
702          }          }
703    
704          if (codec->config.gmc)          if ((profiles[codec->config.profile].flags & PROFILE_GMC) && codec->config.gmc) {
705                  frame.vol_flags |= XVID_VOL_GMC;                  frame.vol_flags |= XVID_VOL_GMC;
706                    frame.motion |= XVID_ME_GME_REFINE;
707            }
708    
709          if (codec->config.interlacing)          if ((profiles[codec->config.profile].flags & PROFILE_INTERLACE) && codec->config.interlacing)
710                  frame.vol_flags |= XVID_VOL_INTERLACING;                  frame.vol_flags |= XVID_VOL_INTERLACING;
711    
712            if (codec->config.ar_mode == 0) { /* PAR */
713                    if (codec->config.display_aspect_ratio != 5) {
714                            frame.par = codec->config.display_aspect_ratio + 1;
715                    } else {
716                            frame.par = XVID_PAR_EXT;
717                            frame.par_width = codec->config.par_x;
718                            frame.par_height= codec->config.par_y;
719                    }
720            } else { /* AR */
721                    /* custom pixel aspect ratio -> calculated from DAR */
722                    frame.par = XVID_PAR_EXT;
723                    frame.par_width = (100 * inhdr->biHeight) / codec->config.ar_y;
724                    frame.par_height= (100 * inhdr->biWidth) / codec->config.ar_x;
725            }
726    
727      /* vop stuff */      /* vop stuff */
728    
729          frame.vop_flags |= XVID_VOP_HALFPEL;          frame.vop_flags |= XVID_VOP_HALFPEL;
730          frame.vop_flags |= XVID_VOP_HQACPRED;          frame.vop_flags |= XVID_VOP_HQACPRED;
731    
732          if (codec->config.debug)          if (codec->config.vop_debug)
733                  frame.vop_flags |= XVID_VOP_DEBUG;                  frame.vop_flags |= XVID_VOP_DEBUG;
734    
735            if (codec->config.trellis_quant) {
736                    frame.vop_flags |= XVID_VOP_TRELLISQUANT;
737            }
738    
739          if (codec->config.motion_search > 4)          if (codec->config.motion_search > 4)
740                  frame.vop_flags |= XVID_VOP_INTER4V;                  frame.vop_flags |= XVID_VOP_INTER4V;
741    
742          if (codec->config.chromame)          if (codec->config.chromame)
743                  frame.vop_flags |= XVID_ME_CHROMA16 + XVID_ME_CHROMA8;                  frame.motion |= XVID_ME_CHROMA_PVOP + XVID_ME_CHROMA_BVOP;
   
         if (codec->config.chroma_opt)  
                 frame.vop_flags |= XVID_VOP_CHROMAOPT;  
744    
745          check_greyscale_mode(&codec->config, &frame, codec->framenum);          if (codec->config.cartoon_mode) {
746                    frame.vop_flags |= XVID_VOP_CARTOON;
747  /* XXX: hinted me                  frame.motion |= XVID_ME_DETECT_STATIC_MOTION;
   
 // fix 1pass modes/hinted MV by koepi  
         if (codec->config.hinted_me && (codec->config.mode == DLG_MODE_CBR || codec->config.mode == DLG_MODE_VBR_QUAL || codec->config.mode == DLG_MODE_VBR_QUANT))  
         {  
                 codec->config.hinted_me = 0;  
748          }          }
 // end of ugly hack  
749    
750          if (codec->config.hinted_me && codec->config.mode == DLG_MODE_2PASS_1)          if (codec->config.turbo)
751          {                  frame.motion |= XVID_ME_FASTREFINE16 | XVID_ME_FASTREFINE8 |
752                  frame.hint.hintstream = codec->twopass.hintstream;                                                  XVID_ME_SKIP_DELTASEARCH | XVID_ME_FAST_MODEINTERPOLATE |
753                  frame.hint.rawhints = 0;                                                  XVID_ME_BFRAME_EARLYSTOP;
                 frame.general |= XVID_HINTEDME_GET;  
         }  
         else if (codec->config.hinted_me && (codec->config.mode == DLG_MODE_2PASS_2_EXT || codec->config.mode == DLG_MODE_2PASS_2_INT))  
         {  
                 DWORD read;  
                 DWORD blocksize;  
   
                 frame.hint.hintstream = codec->twopass.hintstream;  
                 frame.hint.rawhints = 0;  
                 frame.general |= XVID_HINTEDME_SET;  
   
                 if (codec->twopass.hints == INVALID_HANDLE_VALUE)  
                 {  
                         codec->twopass.hints = CreateFile(codec->config.hintfile, GENERIC_READ, 0, 0, OPEN_EXISTING, 0, 0);  
                         if (codec->twopass.hints == INVALID_HANDLE_VALUE)  
                         {  
                                 DEBUGERR("couldn't open hints file");  
                                 return ICERR_ERROR;  
                         }  
                 }  
                 if (!ReadFile(codec->twopass.hints, &blocksize, sizeof(DWORD), &read, 0) || read != sizeof(DWORD) ||  
                         !ReadFile(codec->twopass.hints, frame.hint.hintstream, blocksize, &read, 0) || read != blocksize)  
                 {  
                         DEBUGERR("couldn't read from hints file");  
                         return ICERR_ERROR;  
                 }  
         } */  
754    
755          frame.motion |= pmvfast_presets[codec->config.motion_search];          frame.motion |= pmvfast_presets[codec->config.motion_search];
756    
757          switch (codec->config.vhq_mode)          switch (codec->config.vhq_mode)
758          {          {
759          case VHQ_MODE_DECISION :          case VHQ_MODE_DECISION :
760                  frame.vop_flags |= XVID_VOP_MODEDECISION_BITS;                  frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
761                  break;                  break;
762    
763          case VHQ_LIMITED_SEARCH :          case VHQ_LIMITED_SEARCH :
764                  frame.vop_flags |= XVID_VOP_MODEDECISION_BITS;                  frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
765                  frame.motion |= XVID_ME_HALFPELREFINE16_BITS;                  frame.motion |= XVID_ME_HALFPELREFINE16_RD;
766                  frame.motion |= XVID_ME_QUARTERPELREFINE16_BITS;                  frame.motion |= XVID_ME_QUARTERPELREFINE16_RD;
767                  break;                  break;
768    
769          case VHQ_MEDIUM_SEARCH :          case VHQ_MEDIUM_SEARCH :
770                  frame.vop_flags |= XVID_VOP_MODEDECISION_BITS;                  frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
771                  frame.motion |= XVID_ME_HALFPELREFINE16_BITS;                  frame.motion |= XVID_ME_HALFPELREFINE16_RD;
772                  frame.motion |= XVID_ME_HALFPELREFINE8_BITS;                  frame.motion |= XVID_ME_HALFPELREFINE8_RD;
773                  frame.motion |= XVID_ME_QUARTERPELREFINE16_BITS;                  frame.motion |= XVID_ME_QUARTERPELREFINE16_RD;
774                  frame.motion |= XVID_ME_QUARTERPELREFINE8_BITS;                  frame.motion |= XVID_ME_QUARTERPELREFINE8_RD;
775                  frame.motion |= XVID_ME_CHECKPREDICTION_BITS;                  frame.motion |= XVID_ME_CHECKPREDICTION_RD;
776                  break;                  break;
777    
778          case VHQ_WIDE_SEARCH :          case VHQ_WIDE_SEARCH :
779                  frame.vop_flags |= XVID_VOP_MODEDECISION_BITS;                  frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
780                  frame.motion |= XVID_ME_HALFPELREFINE16_BITS;                  frame.motion |= XVID_ME_HALFPELREFINE16_RD;
781                  frame.motion |= XVID_ME_HALFPELREFINE8_BITS;                  frame.motion |= XVID_ME_HALFPELREFINE8_RD;
782                  frame.motion |= XVID_ME_QUARTERPELREFINE16_BITS;                  frame.motion |= XVID_ME_QUARTERPELREFINE16_RD;
783                  frame.motion |= XVID_ME_QUARTERPELREFINE8_BITS;                  frame.motion |= XVID_ME_QUARTERPELREFINE8_RD;
784                  frame.motion |= XVID_ME_CHECKPREDICTION_BITS;                  frame.motion |= XVID_ME_CHECKPREDICTION_RD;
785                  frame.motion |= XVID_ME_EXTSEARCH_BITS;                  frame.motion |= XVID_ME_EXTSEARCH_RD;
786                  break;                  break;
787    
788          default :          default :
# Line 549  Line 795 
795          if ((frame.input.csp = get_colorspace(inhdr)) == XVID_CSP_NULL)          if ((frame.input.csp = get_colorspace(inhdr)) == XVID_CSP_NULL)
796                  return ICERR_BADFORMAT;                  return ICERR_BADFORMAT;
797    
798          if (frame.input.csp == XVID_CSP_I420 || frame.input.csp == XVID_CSP_YV12)          if (frame.input.csp == XVID_CSP_I420 || frame.input.csp == XVID_CSP_YV12) {
799                  frame.input.stride[0] = (frame.input.stride[0]*2)/3;                  frame.input.stride[0] = (4 * icc->lpbiInput->biWidth + 3) / 4;
800                    frame.input.stride[1] = frame.input.stride[2] = frame.input.stride[0] / 2 ;
801            }
802    
803          frame.bitstream = icc->lpOutput;          frame.bitstream = icc->lpOutput;
804          frame.length = icc->lpbiOutput->biSizeImage;          frame.length = icc->lpbiOutput->biSizeImage;
805    
806          switch (codec->config.mode)          frame.quant = 0;
         {  
         case DLG_MODE_CBR :  
                 frame.quant = 0;  /* use xvidcore cbr rate control */  
                 break;  
   
         case DLG_MODE_VBR_QUAL :  
         case DLG_MODE_VBR_QUANT :  
         case DLG_MODE_2PASS_1 :  
                 if (codec_get_quant(codec, &frame) == ICERR_ERROR)  
                 {  
                         return ICERR_ERROR;  
                 }  
                 break;  
   
         case DLG_MODE_2PASS_2_EXT :  
         case DLG_MODE_2PASS_2_INT :  
                 if (codec_2pass_get_quant(codec, &frame) == ICERR_ERROR)  
                 {  
                         return ICERR_ERROR;  
                 }  
                 if (codec->config.dummy2pass)  
                 {  
                         outhdr->biSizeImage = codec->twopass.bytes2;  
                         *icc->lpdwFlags = (codec->twopass.nns1.quant & NNSTATS_KEYFRAME) ? AVIIF_KEYFRAME : 0;  
                         return ICERR_OK;  
                 }  
                 break;  
807    
808          case DLG_MODE_NULL :          if (codec->config.mode == RC_MODE_NULL) {
809                  outhdr->biSizeImage = 0;                  outhdr->biSizeImage = 0;
810                  *icc->lpdwFlags = AVIIF_KEYFRAME;                  *icc->lpdwFlags = AVIIF_KEYFRAME;
811                  return ICERR_OK;                  return ICERR_OK;
   
         default :  
                 DEBUGERR("Invalid encoding mode");  
                 return ICERR_ERROR;  
812          }          }
813    
   
814          // force keyframe spacing in 2-pass 1st pass          // force keyframe spacing in 2-pass 1st pass
815          if (codec->config.motion_search == 0)          if (codec->config.motion_search == 0)
         {  
816                  frame.type = XVID_TYPE_IVOP;                  frame.type = XVID_TYPE_IVOP;
817          }  
818          else if (codec->keyspacing < codec->config.min_key_interval && codec->framenum)          /* frame-based stuff */
819          {          apply_zone_modifiers(&frame, &codec->config, codec->framenum);
820                  DEBUG("current frame forced to p-frame");  
821                  frame.type = XVID_TYPE_PVOP;          /* call encore */
         }  
822    
823          memset(&stats, 0, sizeof(stats));          memset(&stats, 0, sizeof(stats));
824          stats.version = XVID_VERSION;          stats.version = XVID_VERSION;
825    
826      length = xvid_encore(codec->ehandle, XVID_ENC_ENCODE, &frame, &stats);          length = codec->xvid_encore_func(codec->ehandle, XVID_ENC_ENCODE, &frame, &stats);
827          switch (length)          switch (length)
828          {          {
829          case XVID_ERR_FAIL :          case XVID_ERR_FAIL :
# Line 625  Line 839 
839                  return ICERR_UNSUPPORTED;                  return ICERR_UNSUPPORTED;
840          }          }
841    
842          {  /* XXX: debug text */          if (codec->config.display_status && stats.type>0) {
843                  char tmp[100];                  status_update(&codec->status, stats.type, stats.length, stats.quant);
                 wsprintf(tmp, " {type=%i len=%i} length=%i", stats.type, stats.length, length);  
                 OutputDebugString(tmp);  
844          }          }
845    
846            DPRINTF("{type=%i len=%i} length=%i", stats.type, stats.length, length);
847    
848      if (length == 0)    /* no encoder output */      if (length == 0)    /* no encoder output */
849          {          {
850                  *icc->lpdwFlags = 0;                  *icc->lpdwFlags = 0;
# Line 650  Line 864 
864    
865          outhdr->biSizeImage = length;          outhdr->biSizeImage = length;
866    
867          if (codec->config.mode == DLG_MODE_2PASS_1 && codec->config.discard1pass)                  if (codec->config.mode == RC_MODE_2PASS1 && codec->config.discard1pass)
868              {              {
869                      outhdr->biSizeImage = 0;                      outhdr->biSizeImage = 0;
870              }              }
   
         /* XXX: hinted me  
             if (frame.general & XVID_HINTEDME_GET)  
         {  
                 DWORD wrote;  
                 DWORD blocksize = frame.hint.hintlength;  
   
                 if (codec->twopass.hints == INVALID_HANDLE_VALUE)  
                 {  
                         codec->twopass.hints = CreateFile(codec->config.hintfile, GENERIC_WRITE, 0, 0, CREATE_ALWAYS, 0, 0);  
                             if (codec->twopass.hints    == INVALID_HANDLE_VALUE)  
                         {  
                                 DEBUGERR("couldn't create hints file");  
                                 return ICERR_ERROR;  
                         }  
871                  }                  }
                     if (!WriteFile(codec->twopass.hints, &frame.hint.hintlength, sizeof(int), &wrote, 0) || wrote != sizeof(int) ||  
                             !WriteFile(codec->twopass.hints, frame.hint.hintstream, blocksize, &wrote, 0) || wrote != blocksize)  
                 {  
                         DEBUGERR("couldn't write to hints file");  
                         return ICERR_ERROR;  
                 }  
             }  
         */  
872    
873          if (stats.type > 0)          codec->framenum++;
874              {          codec->keyspacing++;
                   codec_2pass_update(codec, &stats);  
             }  
     }  
   
         ++codec->framenum;  
         ++codec->keyspacing;  
875    
876          return ICERR_OK;          return ICERR_OK;
877  }  }
# Line 741  Line 926 
926    
927          if (get_colorspace(inhdr) != XVID_CSP_NULL) {          if (get_colorspace(inhdr) != XVID_CSP_NULL) {
928                  memcpy(outhdr, inhdr, sizeof(BITMAPINFOHEADER));                  memcpy(outhdr, inhdr, sizeof(BITMAPINFOHEADER));
929                  // XXX: should we set outhdr->biSize ??                  /* XXX: should we set outhdr->biSize ?? */
930                  return ICERR_OK;                  return ICERR_OK;
931          }          }
932          /* --- yv12 --- */          /* --- yv12 --- */
# Line 767  Line 952 
952          return ICERR_OK;          return ICERR_OK;
953  }  }
954    
955    #define REG_GET_N(X, Y, Z) \
956    { \
957            DWORD size = sizeof(int); \
958            if (RegQueryValueEx(hKey, X, 0, 0, (LPBYTE)&Y, &size) != ERROR_SUCCESS) { \
959                    Y=Z; \
960            } \
961    }while(0)
962    
963  LRESULT decompress_begin(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)  LRESULT decompress_begin(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)
964  {  {
965          xvid_gbl_init_t init;          xvid_gbl_init_t init;
966          xvid_dec_create_t create;          xvid_dec_create_t create;
967            HKEY hKey;
968    
969            if (init_dll(codec) != 0) return ICERR_ERROR;
970    
971          memset(&init, 0, sizeof(init));          memset(&init, 0, sizeof(init));
972          init.version = XVID_VERSION;          init.version = XVID_VERSION;
973          init.cpu_flags = codec->config.cpu;          init.cpu_flags = codec->config.cpu;
974          xvid_global(0, XVID_GBL_INIT, &init, NULL);          codec->xvid_global_func(0, XVID_GBL_INIT, &init, NULL);
975    
976          memset(&create, 0, sizeof(create));          memset(&create, 0, sizeof(create));
977          create.version = XVID_VERSION;          create.version = XVID_VERSION;
978          create.width = lpbiInput->bmiHeader.biWidth;          create.width = lpbiInput->bmiHeader.biWidth;
979          create.height = lpbiInput->bmiHeader.biHeight;          create.height = lpbiInput->bmiHeader.biHeight;
980    
981          switch(xvid_decore(0, XVID_DEC_CREATE, &create, NULL))          switch(codec->xvid_decore_func(0, XVID_DEC_CREATE, &create, NULL))
982          {          {
983          case XVID_ERR_FAIL :          case XVID_ERR_FAIL :
984                  return ICERR_ERROR;                  return ICERR_ERROR;
# Line 800  Line 995 
995    
996          codec->dhandle = create.handle;          codec->dhandle = create.handle;
997    
998            RegOpenKeyEx(XVID_REG_KEY, XVID_REG_PARENT "\\" XVID_REG_CHILD, 0, KEY_READ, &hKey);
999    
1000            REG_GET_N("Deblock_Y",  pp_dy, 0)
1001            REG_GET_N("Deblock_UV", pp_duv, 0)
1002            REG_GET_N("Dering",  pp_dr, 0)
1003            REG_GET_N("FilmEffect", pp_fe, 0)
1004    
1005            RegCloseKey(hKey);
1006    
1007          return ICERR_OK;          return ICERR_OK;
1008  }  }
1009    
1010    
1011  LRESULT decompress_end(CODEC * codec)  LRESULT decompress_end(CODEC * codec)
1012  {  {
1013          if (codec->dhandle != NULL)          if (codec->m_hdll != NULL) {
1014          {                  if (codec->dhandle != NULL) {
1015                  xvid_decore(codec->dhandle, XVID_DEC_DESTROY, NULL, NULL);                          codec->xvid_decore_func(codec->dhandle, XVID_DEC_DESTROY, NULL, NULL);
1016                  codec->dhandle = NULL;                  codec->dhandle = NULL;
1017          }          }
1018                    exit_dll(codec);
1019            }
1020    
1021          return ICERR_OK;          return ICERR_OK;
1022  }  }
1023    
# Line 826  Line 1033 
1033          {          {
1034                  xvid_gbl_convert_t convert;                  xvid_gbl_convert_t convert;
1035    
1036                  DEBUGFOURCC("input", icd->lpbiInput->biCompression);                  DPRINTF("input=%c%c%c%c output=%c%c%c%c",
1037                  DEBUGFOURCC("output", icd->lpbiOutput->biCompression);                          icd->lpbiInput->biCompression&0xff,
1038                            (icd->lpbiInput->biCompression>>8)&0xff,
1039                            (icd->lpbiInput->biCompression>>16)&0xff,
1040                            (icd->lpbiInput->biCompression>>24)&0xff,
1041                            icd->lpbiOutput->biCompression&0xff,
1042                            (icd->lpbiOutput->biCompression>>8)&0xff,
1043                            (icd->lpbiOutput->biCompression>>16)&0xff,
1044                            (icd->lpbiOutput->biCompression>>24)&0xff);
1045    
1046                  memset(&convert, 0, sizeof(convert));                  memset(&convert, 0, sizeof(convert));
1047                  convert.version = XVID_VERSION;                  convert.version = XVID_VERSION;
# Line 849  Line 1063 
1063                  convert.interlacing = 0;                  convert.interlacing = 0;
1064                  if (convert.input.csp == XVID_CSP_NULL ||                  if (convert.input.csp == XVID_CSP_NULL ||
1065                          convert.output.csp == XVID_CSP_NULL ||                          convert.output.csp == XVID_CSP_NULL ||
1066                          xvid_global(0, XVID_GBL_CONVERT, &convert, NULL) < 0)                          codec->xvid_global_func(0, XVID_GBL_CONVERT, &convert, NULL) < 0)
1067                  {                  {
1068                           return ICERR_BADFORMAT;                           return ICERR_BADFORMAT;
1069                  }                  }
# Line 879  Line 1093 
1093                  frame.output.csp = XVID_CSP_NULL;                  frame.output.csp = XVID_CSP_NULL;
1094          }          }
1095    
1096          switch (xvid_decore(codec->dhandle, XVID_DEC_DECODE, &frame, NULL))          if (pp_dy)frame.general |= XVID_DEBLOCKY;
1097            if (pp_duv) frame.general |= XVID_DEBLOCKUV;
1098    /*      if (pp_dr) frame.general |= XVID_DERING; */
1099            if (pp_fe) frame.general |= XVID_FILMEFFECT;
1100    
1101            switch (codec->xvid_decore_func(codec->dhandle, XVID_DEC_DECODE, &frame, NULL))
1102          {          {
1103          case XVID_ERR_FAIL :          case XVID_ERR_FAIL :
1104                  return ICERR_ERROR;                  return ICERR_ERROR;
# Line 897  Line 1116 
1116          return ICERR_OK;          return ICERR_OK;
1117  }  }
1118    
 int codec_get_quant(CODEC* codec, xvid_enc_frame_t* frame)  
 {  
         switch (codec->config.mode)  
         {  
         case DLG_MODE_VBR_QUAL :  
                 if (codec_is_in_credits(&codec->config, codec->framenum))  
                 {  
 // added by koepi for credits greyscale  
   
                         check_greyscale_mode(&codec->config, frame, codec->framenum);  
   
 // end of koepi's addition  
   
                         switch (codec->config.credits_mode)  
                         {  
                         case CREDITS_MODE_RATE :  
                                 frame->quant = codec_get_vbr_quant(&codec->config, codec->config.quality * codec->config.credits_rate / 100);  
                                 break;  
   
                         case CREDITS_MODE_QUANT :  
                                 frame->quant = codec->config.credits_quant_p;  
                                 break;  
   
                         default :  
                                 DEBUGERR("Can't use credits size mode in quality mode");  
                                 return ICERR_ERROR;  
                         }  
                 }  
                 else  
                 {  
 // added by koepi for credits greyscale  
   
                         check_greyscale_mode(&codec->config, frame, codec->framenum);  
   
 // end of koepi's addition  
   
                         frame->quant = codec_get_vbr_quant(&codec->config, codec->config.quality);  
                 }  
                 return ICERR_OK;  
   
         case DLG_MODE_VBR_QUANT :  
                 if (codec_is_in_credits(&codec->config, codec->framenum))  
                 {  
 // added by koepi for credits greyscale  
   
                         check_greyscale_mode(&codec->config, frame, codec->framenum);  
   
 // end of koepi's addition  
   
                         switch (codec->config.credits_mode)  
                         {  
                         case CREDITS_MODE_RATE :  
                                 frame->quant =  
                                         codec->config.max_pquant -  
                                         ((codec->config.max_pquant - codec->config.quant) * codec->config.credits_rate / 100);  
                                 break;  
   
                         case CREDITS_MODE_QUANT :  
                                 frame->quant = codec->config.credits_quant_p;  
                                 break;  
   
                         default :  
                                 DEBUGERR("Can't use credits size mode in quantizer mode");  
                                 return ICERR_ERROR;  
                         }  
                 }  
                 else  
                 {  
 // added by koepi for credits greyscale  
   
                         check_greyscale_mode(&codec->config, frame, codec->framenum);  
   
 // end of koepi's addition  
   
                         frame->quant = codec->config.quant;  
                 }  
                 return ICERR_OK;  
   
         case DLG_MODE_2PASS_1 :  
 // added by koepi for credits greyscale  
   
                 check_greyscale_mode(&codec->config, frame, codec->framenum);  
   
 // end of koepi's addition  
   
                 if (codec->config.credits_mode == CREDITS_MODE_QUANT)  
                 {  
                         if (codec_is_in_credits(&codec->config, codec->framenum))  
                         {  
                                 frame->quant = codec->config.credits_quant_p;  
                         }  
                         else  
                         {  
                                 frame->quant = 2;  
                         }  
                 }  
                 else  
                 {  
                         frame->quant = 2;  
                 }  
                 return ICERR_OK;  
   
         default:  
                 DEBUGERR("get quant: invalid mode");  
                 return ICERR_ERROR;  
         }  
 }  
   
   
 int codec_is_in_credits(CONFIG* config, int framenum)  
 {  
         if (config->credits_start)  
         {  
                 if (framenum >= config->credits_start_begin &&  
                         framenum <= config->credits_start_end)  
                 {  
                         return CREDITS_START;  
                 }  
         }  
   
         if (config->credits_end)  
         {  
                 if (framenum >= config->credits_end_begin &&  
                         framenum <= config->credits_end_end)  
                 {  
                         return CREDITS_END;  
                 }  
         }  
   
         return 0;  
 }  
   
   
 int codec_get_vbr_quant(CONFIG* config, int quality)  
 {  
         static float fquant_running = 0;  
         static int my_quality = -1;  
         int     quant;  
   
         // if quality changes, recalculate fquant (credits)  
         if (quality != my_quality)  
         {  
                 config->fquant = 0;  
         }  
   
         my_quality = quality;  
   
         // desired quantiser = (maxQ-minQ)/100 * (100-qual) + minQ  
         if (!config->fquant)  
         {  
                 config->fquant =  
                         ((float) (config->max_pquant - config->min_pquant) / 100) *  
                         (100 - quality) +  
                         (float) config->min_pquant;  
   
                 fquant_running = config->fquant;  
         }  
   
         if (fquant_running < config->min_pquant)  
         {  
                 fquant_running = (float) config->min_pquant;  
         }  
         else if(fquant_running > config->max_pquant)  
         {  
                 fquant_running = (float) config->max_pquant;  
         }  
   
         quant = (int) fquant_running;  
   
         // add error between fquant and quant to fquant_running  
         fquant_running += config->fquant - quant;  
   
         return quant;  
 }  
   
 // added by koepi for credits greyscale  
   
 int check_greyscale_mode(CONFIG* config, xvid_enc_frame_t* frame, int framenum)  
   
 {  
   
         if ((codec_is_in_credits(config, framenum)) && (config->mode!=DLG_MODE_CBR))  
   
         {  
   
                 if (config->credits_greyscale)  
   
                 {  
   
                         if ((frame->vop_flags && XVID_VOP_GREYSCALE))  // use only if not already in greyscale  
   
                                 frame->vop_flags |= XVID_VOP_GREYSCALE;  
   
                 } else {  
   
                         if (!(frame->vop_flags && XVID_VOP_GREYSCALE))  // if movie is in greyscale, switch back  
   
                                 frame->vop_flags |= XVID_VOP_GREYSCALE;  
   
                 }  
   
         } else {  
   
                 if (config->greyscale)  
   
                 {  
   
                         if ((frame->vop_flags && XVID_VOP_GREYSCALE))  // use only if not already in greyscale  
   
                                 frame->vop_flags |= XVID_VOP_GREYSCALE;  
   
                 } else {  
   
                         if (!(frame->vop_flags && XVID_VOP_GREYSCALE))  // if credits is in greyscale, switch back  
   
                                 frame->vop_flags |= XVID_VOP_GREYSCALE;  
   
                 }  
   
         }  
   
         return 0;  
   
 }  
   
 // end of koepi's addition  
   
Legend:
Removed from v.949  
changed lines
  Added in v.1393
No admin address has been configured
 ViewVC Help
Powered by ViewVC 1.0.4