[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 889, Sat Feb 22 08:24:01 2003 UTC revision 1267, Wed Dec 17 15:16:16 2003 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    static const int pmvfast_presets[7] = {
62          0, 0, 0, 0,          0, 0, 0, 0,
63          0 | PMV_HALFPELREFINE16 | 0,          0 | XVID_ME_HALFPELREFINE16 | 0,
64          0 | PMV_HALFPELREFINE16 | 0 |          0 | XVID_ME_HALFPELREFINE16 | 0 |
65          PMV_ADVANCEDDIAMOND16, PMV_HALFPELREFINE16 | PMV_EXTSEARCH16 |          XVID_ME_ADVANCEDDIAMOND16, XVID_ME_HALFPELREFINE16 | XVID_ME_EXTSEARCH16 |
66          PMV_HALFPELREFINE8 | 0 | PMV_USESQUARES16          XVID_ME_HALFPELREFINE8 | 0 | XVID_ME_USESQUARES16
67  };  };
68    
69    
70    
71  /*      return xvid compatbile colorspace,  /*      return xvid compatbile colorspace,
72          or XVID_CSP_NULL if failure          or XVID_CSP_NULL if failure
73  */  */
# Line 75  Line 82 
82          case BI_RGB :          case BI_RGB :
83                  if (hdr->biBitCount == 16)                  if (hdr->biBitCount == 16)
84                  {                  {
85                          DEBUG("RGB16 (RGB555)");                          DPRINTF("RGB16 (RGB555)");
86                          return rgb_flip | XVID_CSP_RGB555;                          return rgb_flip | XVID_CSP_RGB555;
87                  }                  }
88                  if (hdr->biBitCount == 24)                  if (hdr->biBitCount == 24)
89                  {                  {
90                          DEBUG("RGB24");                          DPRINTF("RGB24");
91                          return rgb_flip | XVID_CSP_BGR;                          return rgb_flip | XVID_CSP_BGR;
92                  }                  }
93                  if (hdr->biBitCount == 32)                  if (hdr->biBitCount == 32)
94                  {                  {
95                          DEBUG("RGB32");                          DPRINTF("RGB32");
96                          return rgb_flip | XVID_CSP_BGRA;                          return rgb_flip | XVID_CSP_BGRA;
97                  }                  }
98    
99                  DEBUG1("unsupported BI_RGB biBitCount", hdr->biBitCount);                  DPRINTF("unsupported BI_RGB biBitCount=%i", hdr->biBitCount);
100                  return XVID_CSP_NULL;                  return XVID_CSP_NULL;
101    
102          case BI_BITFIELDS :          case BI_BITFIELDS :
# Line 102  Line 109 
109                                  hdr4->bV4GreenMask == 0x3e0 &&                                  hdr4->bV4GreenMask == 0x3e0 &&
110                                  hdr4->bV4BlueMask == 0x1f)                                  hdr4->bV4BlueMask == 0x1f)
111                          {                          {
112                                  DEBUG("RGB555");                                  DPRINTF("RGB555");
113                                  return rgb_flip | XVID_CSP_RGB555;                                  return rgb_flip | XVID_CSP_RGB555;
114                          }                          }
115    
# Line 111  Line 118 
118                                  hdr4->bV4GreenMask == 0x7e0 &&                                  hdr4->bV4GreenMask == 0x7e0 &&
119                                  hdr4->bV4BlueMask == 0x1f)                                  hdr4->bV4BlueMask == 0x1f)
120                          {                          {
121                                  DEBUG("RGB565");                                  DPRINTF("RGB565");
122                                  return rgb_flip | XVID_CSP_RGB565;                                  return rgb_flip | XVID_CSP_RGB565;
123                          }                          }
124    
125                          DEBUG("unsupported BI_BITFIELDS mode");                          DPRINTF("unsupported BI_BITFIELDS mode");
126                          return XVID_CSP_NULL;                          return XVID_CSP_NULL;
127                  }                  }
128    
129                  DEBUG("unsupported BI_BITFIELDS/BITMAPHEADER combination");                  DPRINTF("unsupported BI_BITFIELDS/BITMAPHEADER combination");
130                  return XVID_CSP_NULL;                  return XVID_CSP_NULL;
131    
132          case FOURCC_I420 :          case FOURCC_I420 :
133          case FOURCC_IYUV :          case FOURCC_IYUV :
134                  DEBUG("IYUY");                  DPRINTF("IYUY");
135                  return XVID_CSP_I420;                  return XVID_CSP_I420;
136    
137          case FOURCC_YV12 :          case FOURCC_YV12 :
138                  DEBUG("YV12");                  DPRINTF("YV12");
139                  return XVID_CSP_YV12;                  return XVID_CSP_YV12;
140    
141          case FOURCC_YUYV :          case FOURCC_YUYV :
142          case FOURCC_YUY2 :          case FOURCC_YUY2 :
143                  DEBUG("YUY2");                  DPRINTF("YUY2");
144                  return XVID_CSP_YUY2;                  return XVID_CSP_YUY2;
145    
146          case FOURCC_YVYU :          case FOURCC_YVYU :
147                  DEBUG("YVYU");                  DPRINTF("YVYU");
148                  return XVID_CSP_YVYU;                  return XVID_CSP_YVYU;
149    
150          case FOURCC_UYVY :          case FOURCC_UYVY :
151                  DEBUG("UYVY");                  DPRINTF("UYVY");
152                  return XVID_CSP_UYVY;                  return XVID_CSP_UYVY;
153    
154          default :          default :
155                  DEBUGFOURCC("unsupported colorspace", hdr->biCompression);                  DPRINTF("unsupported colorspace %c%c%c%c",
156                hdr->biCompression&0xff,
157                (hdr->biCompression>>8)&0xff,
158                (hdr->biCompression>>16)&0xff,
159                (hdr->biCompression>>24)&0xff);
160                  return XVID_CSP_NULL;                  return XVID_CSP_NULL;
161          }          }
162  }  }
# Line 161  Line 172 
172          BITMAPINFOHEADER * inhdr = &lpbiInput->bmiHeader;          BITMAPINFOHEADER * inhdr = &lpbiInput->bmiHeader;
173          BITMAPINFOHEADER * outhdr = &lpbiOutput->bmiHeader;          BITMAPINFOHEADER * outhdr = &lpbiOutput->bmiHeader;
174    
175        /* VFWEXT detection */
176        if (inhdr->biCompression == VFWEXT_FOURCC) {
177            return (ICM_USER+0x0fff);
178        }
179    
180          if (get_colorspace(inhdr) == XVID_CSP_NULL)          if (get_colorspace(inhdr) == XVID_CSP_NULL)
181          {          {
182                  return ICERR_BADFORMAT;                  return ICERR_BADFORMAT;
# Line 193  Line 209 
209    
210          if (lpbiOutput == NULL)          if (lpbiOutput == NULL)
211          {          {
212                  return sizeof(BITMAPV4HEADER);                  return sizeof(BITMAPINFOHEADER);
213          }          }
214    
215          memcpy(outhdr, inhdr, sizeof(BITMAPINFOHEADER));          memcpy(outhdr, inhdr, sizeof(BITMAPINFOHEADER));
# Line 229  Line 245 
245    
246  LRESULT compress_frames_info(CODEC * codec, ICCOMPRESSFRAMES * icf)  LRESULT compress_frames_info(CODEC * codec, ICCOMPRESSFRAMES * icf)
247  {  {
248          // DEBUG2("frate fscale", codec->frate, codec->fscale);  #if 0
249        DPRINTF("%i %i", icf->lStartFrame, icf->lFrameCount);
250    #endif
251          codec->fincr = icf->dwScale;          codec->fincr = icf->dwScale;
252          codec->fbase = icf->dwRate;          codec->fbase = icf->dwRate;
253          return ICERR_OK;          return ICERR_OK;
254  }  }
255    
256    
257  LRESULT compress_begin(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)  const char type2char(int type)
258  {  {
259          xvid_gbl_init_t init;      if (type==XVID_TYPE_IVOP)
260          xvid_enc_create_t create;          return 'I';
261          xvid_enc_rc_t rc;      if (type==XVID_TYPE_PVOP)
262            return 'P';
263          memset(&rc, 0, sizeof(rc));      if (type==XVID_TYPE_BVOP)
264          rc.version = XVID_VERSION;          return 'B';
265        return 'S';
266    }
267    
268    int vfw_debug(void *handle,
269                             int opt,
270                             void *param1,
271                             void *param2)
272    {
273            switch (opt) {
274            case XVID_PLG_INFO:
275            case XVID_PLG_CREATE:
276            case XVID_PLG_DESTROY:
277            case XVID_PLG_BEFORE:
278                    return 0;
279    
280          switch (codec->config.mode)          case XVID_PLG_AFTER:
281          {          {
282          case DLG_MODE_CBR :                          xvid_plg_data_t *data = (xvid_plg_data_t *) param1;
                 //rc.type = XVID_RC_CBR;  
                 rc.min_iquant = codec->config.min_iquant;  
                 rc.max_iquant = codec->config.max_iquant;  
                 rc.min_pquant = codec->config.min_pquant;  
                 rc.max_pquant = codec->config.max_pquant;  
                 rc.min_bquant = 0;      /* XXX: todo */  
                 rc.max_bquant = 0;      /* XXX: todo */  
                 rc.data.cbr.bitrate = codec->config.rc_bitrate;  
         rc.data.cbr.reaction_delay_factor = codec->config.rc_reaction_delay_factor;  
                 rc.data.cbr.averaging_period = codec->config.rc_averaging_period;  
                 rc.data.cbr.buffer = codec->config.rc_buffer;  
                 break;  
283    
284          case DLG_MODE_VBR_QUAL :                          /* We don't use DPRINTF here because it's active only for _DEBUG
285                  codec->config.fquant = 0;                           * builds and that activates lot of other debug printfs. We only
286                  //param.rc_bitrate = 0;                           * want these all the time */
287                  break;                          char buf[1024];
288                            sprintf(buf, "[%6i]   type=%c   Q:%2i   length:%6i",
289                                            data->frame_num,
290                                            type2char(data->type),
291                                            data->quant,
292                                            data->length);
293                            OutputDebugString(buf);
294    
295          case DLG_MODE_VBR_QUANT :                          return 0;
296                  codec->config.fquant = (float) codec->config.quant;                  }
297                  //param.rc_bitrate = 0;          }
                 break;  
298    
299          case DLG_MODE_2PASS_1 :          return XVID_ERR_FAIL;
300          case DLG_MODE_2PASS_2_INT :  }
         case DLG_MODE_2PASS_2_EXT :  
                 //param.rc_bitrate = 0;  
                 codec->twopass.max_framesize = (int)((double)codec->config.twopass_max_bitrate / 8.0 / ((double)codec->fbase / (double)codec->fincr));  
                 break;  
301    
         case DLG_MODE_NULL :  
                 return ICERR_OK;  
302    
         default :  
                 break;  
         }  
303    
304      /* destroy previously created codec */  LRESULT compress_begin(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)
         if(codec->ehandle)  
305          {          {
306            xvid_gbl_init_t init;
307            xvid_enc_create_t create;
308        xvid_enc_plugin_t plugins[3];
309            xvid_plugin_single_t single;
310            xvid_plugin_2pass1_t pass1;
311            xvid_plugin_2pass2_t pass2;
312        int i;
313    
314        /* destroy previously created codec */
315            if(codec->ehandle) {
316                  xvid_encore(codec->ehandle, XVID_ENC_DESTROY, NULL, NULL);                  xvid_encore(codec->ehandle, XVID_ENC_DESTROY, NULL, NULL);
317                  codec->ehandle = NULL;                  codec->ehandle = NULL;
318          }          }
# Line 295  Line 320 
320      memset(&init, 0, sizeof(init));      memset(&init, 0, sizeof(init));
321          init.version = XVID_VERSION;          init.version = XVID_VERSION;
322          init.cpu_flags = codec->config.cpu;          init.cpu_flags = codec->config.cpu;
323        init.debug = codec->config.debug;
324          xvid_global(0, XVID_GBL_INIT, &init, NULL);          xvid_global(0, XVID_GBL_INIT, &init, NULL);
325    
326          memset(&create, 0, sizeof(create));          memset(&create, 0, sizeof(create));
327          create.version = XVID_VERSION;          create.version = XVID_VERSION;
328    
329        /* zones */
330        create.zones = malloc(sizeof(xvid_enc_zone_t) * codec->config.num_zones);
331        create.num_zones = codec->config.num_zones;
332        for (i=0; i < create.num_zones; i++) {
333            create.zones[i].frame = codec->config.zones[i].frame;
334            if (codec->config.zones[i].mode == RC_ZONE_QUANT) {
335                create.zones[i].mode = XVID_ZONE_QUANT;
336                create.zones[i].increment = codec->config.zones[i].quant;
337            }else{
338                create.zones[i].mode = XVID_ZONE_WEIGHT;
339                create.zones[i].increment = codec->config.zones[i].weight;
340            }
341            create.zones[i].base = 100;
342        }
343    
344        /* plugins */
345            create.plugins = plugins;
346            switch (codec->config.mode)
347            {
348            case RC_MODE_1PASS :
349            memset(&single, 0, sizeof(single));
350                single.version = XVID_VERSION;
351            single.bitrate = codec->config.bitrate * CONFIG_KBPS;
352            single.reaction_delay_factor = codec->config.rc_reaction_delay_factor;
353                    single.averaging_period = codec->config.rc_averaging_period;
354                    single.buffer = codec->config.rc_buffer;
355            plugins[create.num_plugins].func = xvid_plugin_single;
356            plugins[create.num_plugins].param = &single;
357            create.num_plugins++;
358            break;
359    
360            case RC_MODE_2PASS1 :
361            memset(&pass1, 0, sizeof(pass1));
362                pass1.version = XVID_VERSION;
363            pass1.filename = codec->config.stats;
364    
365            plugins[create.num_plugins].func = xvid_plugin_2pass1;
366            plugins[create.num_plugins].param = &pass1;
367            create.num_plugins++;
368                    break;
369    
370            case RC_MODE_2PASS2 :
371            memset(&pass2, 0, sizeof(pass2));
372                pass2.version = XVID_VERSION;
373            if (codec->config.use_2pass_bitrate) {
374                pass2.bitrate = codec->config.bitrate * CONFIG_KBPS;
375            }else{
376                pass2.bitrate = -codec->config.desired_size;    /* kilobytes */
377            }
378                    pass2.filename = codec->config.stats;
379    
380            pass2.keyframe_boost = codec->config.keyframe_boost;   /* keyframe boost percentage: [0..100...]; */
381            pass2.curve_compression_high = codec->config.curve_compression_high;
382            pass2.curve_compression_low = codec->config.curve_compression_low;
383                    pass2.overflow_control_strength = codec->config.overflow_control_strength;
384            pass2.max_overflow_improvement = codec->config.twopass_max_overflow_improvement;
385            pass2.max_overflow_degradation = codec->config.twopass_max_overflow_degradation;
386                pass2.kfreduction = codec->config.kfreduction;
387            pass2.kfthreshold = codec->config.kfthreshold;
388            pass2.container_frame_overhead = 24;    /* AVI */
389    
390            plugins[create.num_plugins].func = xvid_plugin_2pass2;
391            plugins[create.num_plugins].param = &pass2;
392            create.num_plugins++;
393                    break;
394    
395            case RC_MODE_NULL :
396                    return ICERR_OK;
397    
398            default :
399                    break;
400            }
401    
402            if ((profiles[codec->config.profile].flags & PROFILE_ADAPTQUANT) && codec->config.lum_masking) {
403            plugins[create.num_plugins].func = xvid_plugin_lumimasking;
404            plugins[create.num_plugins].param = NULL;
405            create.num_plugins++;
406            }
407    
408        plugins[create.num_plugins].func = vfw_debug;
409        plugins[create.num_plugins].param = NULL;
410        create.num_plugins++;
411    
412        create.profile = profiles[codec->config.profile].id;
413    
414          create.width = lpbiInput->bmiHeader.biWidth;          create.width = lpbiInput->bmiHeader.biWidth;
415          create.height = lpbiInput->bmiHeader.biHeight;          create.height = lpbiInput->bmiHeader.biHeight;
416          create.fincr = codec->fincr;          create.fincr = codec->fincr;
417          create.fbase = codec->fbase;          create.fbase = codec->fbase;
418    
     if (codec->config.packed)  
         create.global |= XVID_PACKED;  
         if (codec->config.dx50bvop)  
                 create.global |= XVID_CLOSED_GOP;  
   
419      create.max_key_interval = codec->config.max_key_interval;      create.max_key_interval = codec->config.max_key_interval;
         /* XXX: param.min_quantizer = codec->config.min_pquant;  
         param.max_quantizer = codec->config.max_pquant; */  
420    
421          create.max_bframes = codec->config.max_bframes;      create.min_quant[0] = codec->config.min_iquant;
422          create.frame_drop_ratio = codec->config.frame_drop_ratio;      create.max_quant[0] = codec->config.max_iquant;
423        create.min_quant[1] = codec->config.min_pquant;
424        create.max_quant[1] = codec->config.max_pquant;
425        create.min_quant[2] = codec->config.min_bquant;
426        create.max_quant[2] = codec->config.max_bquant;
427    
428        if ((profiles[codec->config.profile].flags & PROFILE_BVOP) && codec->config.use_bvop) {
429            create.max_bframes = codec->config.max_bframes;
430          create.bquant_ratio = codec->config.bquant_ratio;          create.bquant_ratio = codec->config.bquant_ratio;
431          create.bquant_offset = codec->config.bquant_offset;          create.bquant_offset = codec->config.bquant_offset;
432    
433            if (codec->config.packed)
434                create.global |= XVID_GLOBAL_PACKED;
435    
436                if (codec->config.closed_gov)
437                        create.global |= XVID_GLOBAL_CLOSED_GOP;
438    
439        }
440    
441            create.frame_drop_ratio = codec->config.frame_drop_ratio;
442    
443          create.num_threads = codec->config.num_threads;          create.num_threads = codec->config.num_threads;
444    
445          switch(xvid_encore(0, XVID_ENC_CREATE, &create, (codec->config.mode==DLG_MODE_CBR)?&rc:NULL ))          switch(xvid_encore(0, XVID_ENC_CREATE, &create, NULL))
446          {          {
447          case XVID_ERR_FAIL :          case XVID_ERR_FAIL :
448                  return ICERR_ERROR;                  return ICERR_ERROR;
# Line 341  Line 461 
461          codec->framenum = 0;          codec->framenum = 0;
462          codec->keyspacing = 0;          codec->keyspacing = 0;
463    
464          codec->twopass.hints = codec->twopass.stats1 = codec->twopass.stats2 = INVALID_HANDLE_VALUE;      if (codec->config.display_status) {
465          codec->twopass.hintstream = NULL;          status_destroy_always(&codec->status);
466            status_create(&codec->status, codec->fincr, codec->fbase);
467        }
468    
469          return ICERR_OK;          return ICERR_OK;
470  }  }
# Line 350  Line 472 
472    
473  LRESULT compress_end(CODEC * codec)  LRESULT compress_end(CODEC * codec)
474  {  {
475          if (codec->ehandle != NULL)      if (codec->ehandle != NULL) {
         {  
476                  xvid_encore(codec->ehandle, XVID_ENC_DESTROY, NULL, NULL);                  xvid_encore(codec->ehandle, XVID_ENC_DESTROY, NULL, NULL);
477                    codec->ehandle = NULL;
                 if (codec->twopass.hints != INVALID_HANDLE_VALUE)  
                 {  
                         CloseHandle(codec->twopass.hints);  
                 }  
                 if (codec->twopass.stats1 != INVALID_HANDLE_VALUE)  
                 {  
                         CloseHandle(codec->twopass.stats1);  
478                  }                  }
479                  if (codec->twopass.stats2 != INVALID_HANDLE_VALUE)  
480                  {      if (codec->config.display_status)
481                          CloseHandle(codec->twopass.stats2);          status_destroy(&codec->status);
482    
483            return ICERR_OK;
484                  }                  }
485                  if (codec->twopass.hintstream != NULL)  
486    
487    static void apply_zone_modifiers(xvid_enc_frame_t * frame, CONFIG * config, int framenum)
488                  {                  {
489                          free(codec->twopass.hintstream);      int i;
                 }  
490    
491                  codec->ehandle = NULL;      for (i=0; i<config->num_zones && config->zones[i].frame <= framenum; i++) ;
492    
493        if (--i < 0) return; /* there are no zones, or we're before the first zone */
494    
495                  codec_2pass_finish(codec);      if (framenum == config->zones[i].frame)
496                    frame->type = config->zones[i].type;
497    
498        if (config->zones[i].greyscale) {
499            frame->vop_flags |= XVID_VOP_GREYSCALE;
500          }          }
501    
502          return ICERR_OK;      if (config->zones[i].chroma_opt) {
503            frame->vop_flags |= XVID_VOP_CHROMAOPT;
504  }  }
505    
506        if ((profiles[config->profile].flags & PROFILE_BVOP) && config->use_bvop) {
507            frame->bframe_threshold = config->zones[i].bvop_threshold;
508        }
509    }
510    
511    
512  LRESULT compress(CODEC * codec, ICCOMPRESS * icc)  LRESULT compress(CODEC * codec, ICCOMPRESS * icc)
513  {  {
514          BITMAPINFOHEADER * inhdr = icc->lpbiInput;          BITMAPINFOHEADER * inhdr = icc->lpbiInput;
# Line 387  Line 517 
517          xvid_enc_stats_t stats;          xvid_enc_stats_t stats;
518          int length;          int length;
519    
         // 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;  
                 }  
         }  
   
520          memset(&frame, 0, sizeof(frame));          memset(&frame, 0, sizeof(frame));
521          frame.version = XVID_VERSION;          frame.version = XVID_VERSION;
522    
# Line 403  Line 524 
524    
525          /* vol stuff */          /* vol stuff */
526    
527          if (codec->config.quant_type != QUANT_MODE_H263)      if ((profiles[codec->config.profile].flags & PROFILE_MPEGQUANT) &&
528            codec->config.quant_type != QUANT_MODE_H263)
529          {          {
530                  frame.vol_flags |= XVID_MPEGQUANT;                  frame.vol_flags |= XVID_VOL_MPEGQUANT;
531    
532                  // 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)  
                 {  
533                          frame.quant_intra_matrix = codec->config.qmatrix_intra;                          frame.quant_intra_matrix = codec->config.qmatrix_intra;
534                          frame.quant_inter_matrix = codec->config.qmatrix_inter;                          frame.quant_inter_matrix = codec->config.qmatrix_inter;
535                  }                  }else{
                 else  
                 {  
536                          frame.quant_intra_matrix = NULL;                          frame.quant_intra_matrix = NULL;
537                          frame.quant_inter_matrix = NULL;                          frame.quant_inter_matrix = NULL;
538                  }                  }
539          }          }
540    
541          if (codec->config.reduced_resolution) {      if ((profiles[codec->config.profile].flags & PROFILE_REDUCED) &&
542                  frame.vol_flags |= XVID_REDUCED_ENABLE;              codec->config.reduced_resolution) {
543                  frame.vop_flags |= XVID_REDUCED;        /* XXX: need auto decion mode */                  frame.vol_flags |= XVID_VOL_REDUCED_ENABLE;
544                    frame.vop_flags |= XVID_VOP_REDUCED;    /* XXX: need auto decion mode */
545          }          }
546    
547          if (codec->config.qpel) {          if ((profiles[codec->config.profile].flags & PROFILE_QPEL) && codec->config.qpel) {
548                  frame.vol_flags |= XVID_QUARTERPEL;                  frame.vol_flags |= XVID_VOL_QUARTERPEL;
549                  frame.motion |= PMV_QUARTERPELREFINE16 | PMV_QUARTERPELREFINE8;                  frame.motion |= XVID_ME_QUARTERPELREFINE16 | XVID_ME_QUARTERPELREFINE8;
550          }          }
551    
552          if (codec->config.gmc)          if ((profiles[codec->config.profile].flags & PROFILE_GMC) && codec->config.gmc) {
553                  frame.vol_flags |= XVID_GMC;                  frame.vol_flags |= XVID_VOL_GMC;
554                    frame.motion |= XVID_ME_GME_REFINE;
555          if (codec->config.interlacing)          }
                 frame.vol_flags |= XVID_INTERLACING;  
   
     /* vop stuff */  
   
         frame.vop_flags |= XVID_HALFPEL;  
         frame.vop_flags |= XVID_HQACPRED;  
556    
557          if (codec->config.debug)          if ((profiles[codec->config.profile].flags & PROFILE_INTERLACE) && codec->config.interlacing)
558                  frame.vop_flags |= XVID_DEBUG;                  frame.vol_flags |= XVID_VOL_INTERLACING;
559    
560          if (codec->config.motion_search > 4)          if (codec->config.display_aspect_ratio !=0) {
561                  frame.vop_flags |= XVID_INTER4V;                  int ar_x;
562                    int ar_y;
563    
564          if (codec->config.lum_masking)                  if (codec->config.display_aspect_ratio == 1) {
565                  frame.vop_flags |= XVID_LUMIMASKING;                          ar_x = 4;
566                            ar_y = 3;
567                    }
568                    if (codec->config.display_aspect_ratio == 2) {
569                            ar_x = 16;
570                            ar_y = 9;
571                    }
572    
573          if (codec->config.chromame)                  /* custom pixel aspect ratio -> calculated from DAR */
574                  frame.vop_flags |= PMV_CHROMA16 + PMV_CHROMA8;                  frame.par = XVID_PAR_EXT;
575                    frame.par_width = (100 * inhdr->biHeight) / ar_y;
576                    frame.par_height= (100 * inhdr->biWidth) / ar_x;
577            } else {
578                    /* assuming defaults for now... */
579                    frame.par = XVID_PAR_11_VGA;
580            }
581    
582          if (codec->config.chroma_opt)      /* vop stuff */
                 frame.vop_flags |= XVID_CHROMAOPT;  
583    
584          check_greyscale_mode(&codec->config, &frame, codec->framenum);          frame.vop_flags |= XVID_VOP_HALFPEL;
585            frame.vop_flags |= XVID_VOP_HQACPRED;
586    
587  /* XXX: hinted me          if (codec->config.vop_debug)
588                    frame.vop_flags |= XVID_VOP_DEBUG;
589    
590  // fix 1pass modes/hinted MV by koepi      if (codec->config.trellis_quant) {
591          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))          frame.vop_flags |= XVID_VOP_TRELLISQUANT;
         {  
                 codec->config.hinted_me = 0;  
592          }          }
 // end of ugly hack  
593    
594          if (codec->config.hinted_me && codec->config.mode == DLG_MODE_2PASS_1)      if (codec->config.motion_search > 4)
595          {                  frame.vop_flags |= XVID_VOP_INTER4V;
                 frame.hint.hintstream = codec->twopass.hintstream;  
                 frame.hint.rawhints = 0;  
                 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;  
596    
597                  frame.hint.hintstream = codec->twopass.hintstream;          if (codec->config.chromame)
598                  frame.hint.rawhints = 0;                  frame.motion |= XVID_ME_CHROMA_PVOP + XVID_ME_CHROMA_BVOP;
                 frame.general |= XVID_HINTEDME_SET;  
599    
600                  if (codec->twopass.hints == INVALID_HANDLE_VALUE)          if (codec->config.cartoon_mode) {
601                  {                  frame.vop_flags |= XVID_VOP_CARTOON;
602                          codec->twopass.hints = CreateFile(codec->config.hintfile, GENERIC_READ, 0, 0, OPEN_EXISTING, 0, 0);                  frame.motion |= XVID_ME_DETECT_STATIC_MOTION;
                         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;  
603                  }                  }
         } */  
604    
605          frame.motion |= pmvfast_presets[codec->config.motion_search];          frame.motion |= pmvfast_presets[codec->config.motion_search];
606    
607          switch (codec->config.vhq_mode)          switch (codec->config.vhq_mode)
608          {          {
609          case VHQ_MODE_DECISION :          case VHQ_MODE_DECISION :
610                  frame.vop_flags |= XVID_MODEDECISION_BITS;                  frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
611                  break;                  break;
612    
613          case VHQ_LIMITED_SEARCH :          case VHQ_LIMITED_SEARCH :
614                  frame.vop_flags |= XVID_MODEDECISION_BITS;                  frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
615                  frame.motion |= HALFPELREFINE16_BITS;                  frame.motion |= XVID_ME_HALFPELREFINE16_RD;
616                  frame.motion |= QUARTERPELREFINE16_BITS;                  frame.motion |= XVID_ME_QUARTERPELREFINE16_RD;
617                  break;                  break;
618    
619          case VHQ_MEDIUM_SEARCH :          case VHQ_MEDIUM_SEARCH :
620                  frame.vop_flags |= XVID_MODEDECISION_BITS;                  frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
621                  frame.motion |= HALFPELREFINE16_BITS;                  frame.motion |= XVID_ME_HALFPELREFINE16_RD;
622                  frame.motion |= HALFPELREFINE8_BITS;                  frame.motion |= XVID_ME_HALFPELREFINE8_RD;
623                  frame.motion |= QUARTERPELREFINE16_BITS;                  frame.motion |= XVID_ME_QUARTERPELREFINE16_RD;
624                  frame.motion |= QUARTERPELREFINE8_BITS;                  frame.motion |= XVID_ME_QUARTERPELREFINE8_RD;
625                  frame.motion |= CHECKPREDICTION_BITS;                  frame.motion |= XVID_ME_CHECKPREDICTION_RD;
626                  break;                  break;
627    
628          case VHQ_WIDE_SEARCH :          case VHQ_WIDE_SEARCH :
629                  frame.vop_flags |= XVID_MODEDECISION_BITS;                  frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
630                  frame.motion |= HALFPELREFINE16_BITS;                  frame.motion |= XVID_ME_HALFPELREFINE16_RD;
631                  frame.motion |= HALFPELREFINE8_BITS;                  frame.motion |= XVID_ME_HALFPELREFINE8_RD;
632                  frame.motion |= QUARTERPELREFINE16_BITS;                  frame.motion |= XVID_ME_QUARTERPELREFINE16_RD;
633                  frame.motion |= QUARTERPELREFINE8_BITS;                  frame.motion |= XVID_ME_QUARTERPELREFINE8_RD;
634                  frame.motion |= CHECKPREDICTION_BITS;                  frame.motion |= XVID_ME_CHECKPREDICTION_RD;
635                  frame.motion |= EXTSEARCH_BITS;                  frame.motion |= XVID_ME_EXTSEARCH_RD;
636                  break;                  break;
637    
638          default :          default :
# Line 549  Line 651 
651          frame.bitstream = icc->lpOutput;          frame.bitstream = icc->lpOutput;
652          frame.length = icc->lpbiOutput->biSizeImage;          frame.length = icc->lpbiOutput->biSizeImage;
653    
654          switch (codec->config.mode)      frame.quant = 0;
         {  
         case DLG_MODE_CBR :  
                 frame.quant = 0;  /* use xvidcore cbr rate control */  
                 break;  
655    
656          case DLG_MODE_VBR_QUAL :      if (codec->config.mode == RC_MODE_NULL) {
         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;  
   
         case DLG_MODE_NULL :  
657                  outhdr->biSizeImage = 0;                  outhdr->biSizeImage = 0;
658                  *icc->lpdwFlags = AVIIF_KEYFRAME;                  *icc->lpdwFlags = AVIIF_KEYFRAME;
659                  return ICERR_OK;                  return ICERR_OK;
   
         default :  
                 DEBUGERR("Invalid encoding mode");  
                 return ICERR_ERROR;  
660          }          }
661    
   
662          // force keyframe spacing in 2-pass 1st pass          // force keyframe spacing in 2-pass 1st pass
663          if (codec->config.motion_search == 0)          if (codec->config.motion_search == 0)
         {  
664                  frame.type = XVID_TYPE_IVOP;                  frame.type = XVID_TYPE_IVOP;
         }  
         else if (codec->keyspacing < codec->config.min_key_interval && codec->framenum)  
         {  
                 DEBUG("current frame forced to p-frame");  
                 frame.type = XVID_TYPE_PVOP;  
         }  
665    
666          frame.bquant = 0;      /* frame-based stuff */
667        apply_zone_modifiers(&frame, &codec->config, codec->framenum);
668    
669        /* call encore */
670    
671          memset(&stats, 0, sizeof(stats));          memset(&stats, 0, sizeof(stats));
672          stats.version = XVID_VERSION;          stats.version = XVID_VERSION;
# Line 621  Line 687 
687                  return ICERR_UNSUPPORTED;                  return ICERR_UNSUPPORTED;
688          }          }
689    
690          {  /* XXX: debug text */      if (codec->config.display_status && stats.type>0) {
691                  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);  
692          }          }
693    
694            DPRINTF("{type=%i len=%i} length=%i", stats.type, stats.length, length);
695    
696      if (length == 0)    /* no encoder output */      if (length == 0)    /* no encoder output */
697          {          {
698                  *icc->lpdwFlags = 0;                  *icc->lpdwFlags = 0;
# Line 646  Line 712 
712    
713          outhdr->biSizeImage = length;          outhdr->biSizeImage = length;
714    
715          if (codec->config.mode == DLG_MODE_2PASS_1 && codec->config.discard1pass)          if (codec->config.mode == RC_MODE_2PASS1 && codec->config.discard1pass)
716              {              {
717                      outhdr->biSizeImage = 0;                      outhdr->biSizeImage = 0;
718              }              }
   
         /* 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;  
                         }  
                 }  
                     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;  
                 }  
719              }              }
         */  
720    
721          if (stats.type > 0)          codec->framenum++;
722              {          codec->keyspacing++;
                   codec_2pass_update(codec, &stats);  
             }  
     }  
   
         ++codec->framenum;  
         ++codec->keyspacing;  
723    
724          return ICERR_OK;          return ICERR_OK;
725  }  }
# Line 737  Line 774 
774    
775          if (get_colorspace(inhdr) != XVID_CSP_NULL) {          if (get_colorspace(inhdr) != XVID_CSP_NULL) {
776                  memcpy(outhdr, inhdr, sizeof(BITMAPINFOHEADER));                  memcpy(outhdr, inhdr, sizeof(BITMAPINFOHEADER));
777                  // XXX: should we set outhdr->biSize ??                  /* XXX: should we set outhdr->biSize ?? */
778                  return ICERR_OK;                  return ICERR_OK;
779          }          }
780          /* --- yv12 --- */          /* --- yv12 --- */
# Line 822  Line 859 
859          {          {
860                  xvid_gbl_convert_t convert;                  xvid_gbl_convert_t convert;
861    
862                  DEBUGFOURCC("input", icd->lpbiInput->biCompression);                  DPRINTF("input=%c%c%c%c output=%c%c%c%c",
863                  DEBUGFOURCC("output", icd->lpbiOutput->biCompression);              icd->lpbiInput->biCompression&0xff,
864                (icd->lpbiInput->biCompression>>8)&0xff,
865                (icd->lpbiInput->biCompression>>16)&0xff,
866                (icd->lpbiInput->biCompression>>24)&0xff,
867                icd->lpbiOutput->biCompression&0xff,
868                (icd->lpbiOutput->biCompression>>8)&0xff,
869                (icd->lpbiOutput->biCompression>>16)&0xff,
870                (icd->lpbiOutput->biCompression>>24)&0xff);
871    
872                  memset(&convert, 0, sizeof(convert));                  memset(&convert, 0, sizeof(convert));
873                  convert.version = XVID_VERSION;                  convert.version = XVID_VERSION;
# Line 893  Line 937 
937          return ICERR_OK;          return ICERR_OK;
938  }  }
939    
 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_GREYSCALE))  // use only if not already in greyscale  
   
                                 frame->vop_flags |= XVID_GREYSCALE;  
   
                 } else {  
   
                         if (!(frame->vop_flags && XVID_GREYSCALE))  // if movie is in greyscale, switch back  
   
                                 frame->vop_flags |= XVID_GREYSCALE;  
   
                 }  
   
         } else {  
   
                 if (config->greyscale)  
   
                 {  
   
                         if ((frame->vop_flags && XVID_GREYSCALE))  // use only if not already in greyscale  
   
                                 frame->vop_flags |= XVID_GREYSCALE;  
   
                 } else {  
   
                         if (!(frame->vop_flags && XVID_GREYSCALE))  // if credits is in greyscale, switch back  
   
                                 frame->vop_flags |= XVID_GREYSCALE;  
   
                 }  
   
         }  
   
         return 0;  
   
 }  
   
 // end of koepi's addition  
   
Legend:
Removed from v.889  
changed lines
  Added in v.1267
No admin address has been configured
 ViewVC Help
Powered by ViewVC 1.0.4