[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 1028, Fri May 16 17:15:26 2003 UTC
# Line 49  Line 49 
49    
50  #include <windows.h>  #include <windows.h>
51  #include <vfw.h>  #include <vfw.h>
52    #include "vfwext.h"
53    
54    #include <xvid.h>
55    #include "debug.h"
56  #include "codec.h"  #include "codec.h"
 #include "2pass.h"  
57    
58  int pmvfast_presets[7] = {  
59    static const int pmvfast_presets[7] = {
60          0, 0, 0, 0,          0, 0, 0, 0,
61          0 | XVID_ME_HALFPELREFINE16 | 0,          0 | XVID_ME_HALFPELREFINE16 | 0,
62          0 | XVID_ME_HALFPELREFINE16 | 0 |          0 | XVID_ME_HALFPELREFINE16 | 0 |
# Line 61  Line 64 
64          XVID_ME_HALFPELREFINE8 | 0 | XVID_ME_USESQUARES16          XVID_ME_HALFPELREFINE8 | 0 | XVID_ME_USESQUARES16
65  };  };
66    
67    
68    
69  /*      return xvid compatbile colorspace,  /*      return xvid compatbile colorspace,
70          or XVID_CSP_NULL if failure          or XVID_CSP_NULL if failure
71  */  */
# Line 75  Line 80 
80          case BI_RGB :          case BI_RGB :
81                  if (hdr->biBitCount == 16)                  if (hdr->biBitCount == 16)
82                  {                  {
83                          DEBUG("RGB16 (RGB555)");                          DPRINTF("RGB16 (RGB555)");
84                          return rgb_flip | XVID_CSP_RGB555;                          return rgb_flip | XVID_CSP_RGB555;
85                  }                  }
86                  if (hdr->biBitCount == 24)                  if (hdr->biBitCount == 24)
87                  {                  {
88                          DEBUG("RGB24");                          DPRINTF("RGB24");
89                          return rgb_flip | XVID_CSP_BGR;                          return rgb_flip | XVID_CSP_BGR;
90                  }                  }
91                  if (hdr->biBitCount == 32)                  if (hdr->biBitCount == 32)
92                  {                  {
93                          DEBUG("RGB32");                          DPRINTF("RGB32");
94                          return rgb_flip | XVID_CSP_BGRA;                          return rgb_flip | XVID_CSP_BGRA;
95                  }                  }
96    
97                  DEBUG1("unsupported BI_RGB biBitCount", hdr->biBitCount);                  DPRINTF("unsupported BI_RGB biBitCount=%i", hdr->biBitCount);
98                  return XVID_CSP_NULL;                  return XVID_CSP_NULL;
99    
100          case BI_BITFIELDS :          case BI_BITFIELDS :
# Line 102  Line 107 
107                                  hdr4->bV4GreenMask == 0x3e0 &&                                  hdr4->bV4GreenMask == 0x3e0 &&
108                                  hdr4->bV4BlueMask == 0x1f)                                  hdr4->bV4BlueMask == 0x1f)
109                          {                          {
110                                  DEBUG("RGB555");                                  DPRINTF("RGB555");
111                                  return rgb_flip | XVID_CSP_RGB555;                                  return rgb_flip | XVID_CSP_RGB555;
112                          }                          }
113    
# Line 111  Line 116 
116                                  hdr4->bV4GreenMask == 0x7e0 &&                                  hdr4->bV4GreenMask == 0x7e0 &&
117                                  hdr4->bV4BlueMask == 0x1f)                                  hdr4->bV4BlueMask == 0x1f)
118                          {                          {
119                                  DEBUG("RGB565");                                  DPRINTF("RGB565");
120                                  return rgb_flip | XVID_CSP_RGB565;                                  return rgb_flip | XVID_CSP_RGB565;
121                          }                          }
122    
123                          DEBUG("unsupported BI_BITFIELDS mode");                          DPRINTF("unsupported BI_BITFIELDS mode");
124                          return XVID_CSP_NULL;                          return XVID_CSP_NULL;
125                  }                  }
126    
127                  DEBUG("unsupported BI_BITFIELDS/BITMAPHEADER combination");                  DPRINTF("unsupported BI_BITFIELDS/BITMAPHEADER combination");
128                  return XVID_CSP_NULL;                  return XVID_CSP_NULL;
129    
130          case FOURCC_I420 :          case FOURCC_I420 :
131          case FOURCC_IYUV :          case FOURCC_IYUV :
132                  DEBUG("IYUY");                  DPRINTF("IYUY");
133                  return XVID_CSP_I420;                  return XVID_CSP_I420;
134    
135          case FOURCC_YV12 :          case FOURCC_YV12 :
136                  DEBUG("YV12");                  DPRINTF("YV12");
137                  return XVID_CSP_YV12;                  return XVID_CSP_YV12;
138    
139          case FOURCC_YUYV :          case FOURCC_YUYV :
140          case FOURCC_YUY2 :          case FOURCC_YUY2 :
141                  DEBUG("YUY2");                  DPRINTF("YUY2");
142                  return XVID_CSP_YUY2;                  return XVID_CSP_YUY2;
143    
144          case FOURCC_YVYU :          case FOURCC_YVYU :
145                  DEBUG("YVYU");                  DPRINTF("YVYU");
146                  return XVID_CSP_YVYU;                  return XVID_CSP_YVYU;
147    
148          case FOURCC_UYVY :          case FOURCC_UYVY :
149                  DEBUG("UYVY");                  DPRINTF("UYVY");
150                  return XVID_CSP_UYVY;                  return XVID_CSP_UYVY;
151    
152          default :          default :
153                  DEBUGFOURCC("unsupported colorspace", hdr->biCompression);                  DPRINTF("unsupported colorspace %c%c%c%c",
154                hdr->biCompression&0xff,
155                (hdr->biCompression>>8)&0xff,
156                (hdr->biCompression>>16)&0xff,
157                (hdr->biCompression>>24)&0xff);
158                  return XVID_CSP_NULL;                  return XVID_CSP_NULL;
159          }          }
160  }  }
# Line 161  Line 170 
170          BITMAPINFOHEADER * inhdr = &lpbiInput->bmiHeader;          BITMAPINFOHEADER * inhdr = &lpbiInput->bmiHeader;
171          BITMAPINFOHEADER * outhdr = &lpbiOutput->bmiHeader;          BITMAPINFOHEADER * outhdr = &lpbiOutput->bmiHeader;
172    
173        /* VFWEXT detection */
174        if (inhdr->biCompression == VFWEXT_FOURCC) {
175            return (ICM_USER+0x0fff);
176        }
177    
178          if (get_colorspace(inhdr) == XVID_CSP_NULL)          if (get_colorspace(inhdr) == XVID_CSP_NULL)
179          {          {
180                  return ICERR_BADFORMAT;                  return ICERR_BADFORMAT;
# Line 229  Line 243 
243    
244  LRESULT compress_frames_info(CODEC * codec, ICCOMPRESSFRAMES * icf)  LRESULT compress_frames_info(CODEC * codec, ICCOMPRESSFRAMES * icf)
245  {  {
246          // DEBUG2("frate fscale", codec->frate, codec->fscale);      //DPRINTF("%i %i", icf->lStartFrame, icf->lFrameCount);
247          codec->fincr = icf->dwScale;          codec->fincr = icf->dwScale;
248          codec->fbase = icf->dwRate;          codec->fbase = icf->dwRate;
249          return ICERR_OK;          return ICERR_OK;
250  }  }
251    
252    
253    const char type2char(int type)
254    {
255        if (type==XVID_TYPE_IVOP)
256            return 'I';
257        if (type==XVID_TYPE_PVOP)
258            return 'P';
259        if (type==XVID_TYPE_BVOP)
260            return 'B';
261        return 'S';
262    }
263    
264    int vfw_debug(void *handle,
265                             int opt,
266                             void *param1,
267                             void *param2)
268    {
269            switch (opt) {
270            case XVID_PLG_INFO:
271            case XVID_PLG_CREATE:
272            case XVID_PLG_DESTROY:
273            case XVID_PLG_BEFORE:
274                    return 0;
275    
276            case XVID_PLG_AFTER:
277                    {
278                            xvid_plg_data_t *data = (xvid_plg_data_t *) param1;
279    
280                            DPRINTF("[%5i]   type=%c   Q:%2i   length:%6i",
281                                       data->frame_num,
282                       type2char(data->type),
283                       data->quant,
284                       data->length);
285                            return 0;
286                    }
287            }
288    
289            return XVID_ERR_FAIL;
290    }
291    
292    
293    
294  LRESULT compress_begin(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)  LRESULT compress_begin(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)
295  {  {
296          xvid_gbl_init_t init;          xvid_gbl_init_t init;
297          xvid_enc_create_t create;          xvid_enc_create_t create;
298          xvid_enc_rc_t rc;      xvid_enc_plugin_t plugins[3];
299      xvid_enc_plugin_t plugins[1];          xvid_plugin_single_t single;
300            xvid_plugin_2pass1_t pass1;
301            xvid_plugin_2pass2_t pass2;
302        int i;
303    
304          memset(&rc, 0, sizeof(rc));      /* destroy previously created codec */
305          rc.version = XVID_VERSION;          if(codec->ehandle) {
306                    xvid_encore(codec->ehandle, XVID_ENC_DESTROY, NULL, NULL);
307                    codec->ehandle = NULL;
308            }
309    
310        memset(&init, 0, sizeof(init));
311            init.version = XVID_VERSION;
312            init.cpu_flags = codec->config.cpu;
313            xvid_global(0, XVID_GBL_INIT, &init, NULL);
314    
315            memset(&create, 0, sizeof(create));
316            create.version = XVID_VERSION;
317    
318        // zones
319        create.zones = malloc(sizeof(xvid_enc_zone_t) * codec->config.num_zones);
320        create.num_zones = codec->config.num_zones;
321        for (i=0; i < create.num_zones; i++) {
322            create.zones[i].frame = codec->config.zones[i].frame;
323            if (create.zones[i].mode = RC_ZONE_QUANT) {
324                create.zones[i].mode = XVID_ZONE_QUANT;
325                create.zones[i].increment = codec->config.zones[i].quant;
326            }else{
327                create.zones[i].mode = XVID_ZONE_WEIGHT;
328                create.zones[i].increment = codec->config.zones[i].weight;
329            }
330            create.zones[i].base = 100;
331        }
332    
333        // plugins
334            create.plugins = plugins;
335          switch (codec->config.mode)          switch (codec->config.mode)
336          {          {
337          case DLG_MODE_CBR :          case RC_MODE_1PASS :
338                  //rc.type = XVID_RC_CBR;          memset(&single, 0, sizeof(single));
339                  rc.min_iquant = codec->config.min_iquant;              single.version = XVID_VERSION;
340                  rc.max_iquant = codec->config.max_iquant;          single.bitrate = codec->config.bitrate;
341                  rc.min_pquant = codec->config.min_pquant;          single.reaction_delay_factor = codec->config.rc_reaction_delay_factor;
342                  rc.max_pquant = codec->config.max_pquant;                  single.averaging_period = codec->config.rc_averaging_period;
343                  rc.min_bquant = 0;      /* XXX: todo */                  single.buffer = codec->config.rc_buffer;
344                  rc.max_bquant = 0;      /* XXX: todo */          plugins[create.num_plugins].func = xvid_plugin_single;
345                  rc.data.cbr.bitrate = codec->config.rc_bitrate;          plugins[create.num_plugins].param = &single;
346          rc.data.cbr.reaction_delay_factor = codec->config.rc_reaction_delay_factor;          create.num_plugins++;
                 rc.data.cbr.averaging_period = codec->config.rc_averaging_period;  
                 rc.data.cbr.buffer = codec->config.rc_buffer;  
                 break;  
347    
348          case DLG_MODE_VBR_QUAL :          case RC_MODE_2PASS1 :
349                  codec->config.fquant = 0;          memset(&pass1, 0, sizeof(pass1));
350                  //param.rc_bitrate = 0;              pass1.version = XVID_VERSION;
351                  break;          pass1.filename = codec->config.stats;
352    
353          case DLG_MODE_VBR_QUANT :          plugins[create.num_plugins].func = xvid_plugin_2pass1;
354                  codec->config.fquant = (float) codec->config.quant;          plugins[create.num_plugins].param = &pass1;
355                  //param.rc_bitrate = 0;          create.num_plugins++;
356                  break;                  break;
357    
358          case DLG_MODE_2PASS_1 :          case RC_MODE_2PASS2 :
359          case DLG_MODE_2PASS_2_INT :          memset(&pass2, 0, sizeof(pass2));
360          case DLG_MODE_2PASS_2_EXT :              pass2.version = XVID_VERSION;
361                  //param.rc_bitrate = 0;          pass2.bitrate = codec->config.bitrate;
362                  codec->twopass.max_framesize = (int)((double)codec->config.twopass_max_bitrate / 8.0 / ((double)codec->fbase / (double)codec->fincr));                  pass2.filename = codec->config.stats;
363    
364            plugins[create.num_plugins].func = xvid_plugin_2pass2;
365            plugins[create.num_plugins].param = &pass2;
366            create.num_plugins++;
367                  break;                  break;
368    
369          case DLG_MODE_NULL :          case RC_MODE_NULL :
370                  return ICERR_OK;                  return ICERR_OK;
371    
372          default :          default :
373                  break;                  break;
374          }          }
375    
376      /* destroy previously created codec */          if ((profiles[codec->config.profile].flags & PROFILE_ADAPTQUANT) && codec->config.lum_masking) {
377          if(codec->ehandle)          plugins[create.num_plugins].func = xvid_plugin_lumimasking;
378          {          plugins[create.num_plugins].param = NULL;
379                  xvid_encore(codec->ehandle, XVID_ENC_DESTROY, NULL, NULL);          create.num_plugins++;
                 codec->ehandle = NULL;  
380          }          }
381    
382      memset(&init, 0, sizeof(init));      plugins[create.num_plugins].func = vfw_debug;
383          init.version = XVID_VERSION;      plugins[create.num_plugins].param = NULL;
384          init.cpu_flags = codec->config.cpu;      create.num_plugins++;
         xvid_global(0, XVID_GBL_INIT, &init, NULL);  
385    
386          memset(&create, 0, sizeof(create));      create.profile = profiles[codec->config.profile].id;
         create.version = XVID_VERSION;  
387    
388          create.width = lpbiInput->bmiHeader.biWidth;          create.width = lpbiInput->bmiHeader.biWidth;
389          create.height = lpbiInput->bmiHeader.biHeight;          create.height = lpbiInput->bmiHeader.biHeight;
390          create.fincr = codec->fincr;          create.fincr = codec->fincr;
391          create.fbase = codec->fbase;          create.fbase = codec->fbase;
392    
393      create.plugins = plugins;      create.max_key_interval = codec->config.max_key_interval;
394      create.num_plugins = 0;  
395          if (codec->config.lum_masking) {      create.min_quant[0] = codec->config.min_iquant;
396          plugins[create.num_plugins].func = xvid_plugin_lumimasking;      create.max_quant[0] = codec->config.max_iquant;
397          plugins[create.num_plugins].param = NULL;      create.min_quant[1] = codec->config.min_pquant;
398          create.num_plugins++;      create.max_quant[1] = codec->config.max_pquant;
399      }      create.min_quant[2] = codec->config.min_bquant;
400        create.max_quant[2] = codec->config.max_bquant;
401    
402        if ((profiles[codec->config.profile].flags & PROFILE_BVOP) && codec->config.use_bvop) {
403            create.max_bframes = codec->config.max_bframes;
404                create.bquant_ratio = codec->config.bquant_ratio;
405                create.bquant_offset = codec->config.bquant_offset;
406    
407      if (codec->config.packed)      if (codec->config.packed)
408          create.global |= XVID_GLOBAL_PACKED;          create.global |= XVID_GLOBAL_PACKED;
409          if (codec->config.dx50bvop)  
410                if (codec->config.closed_gov)
411                  create.global |= XVID_GLOBAL_CLOSED_GOP;                  create.global |= XVID_GLOBAL_CLOSED_GOP;
412    
413      create.max_key_interval = codec->config.max_key_interval;      }
         /* XXX: param.min_quantizer = codec->config.min_pquant;  
         param.max_quantizer = codec->config.max_pquant; */  
414    
         create.max_bframes = codec->config.max_bframes;  
415          create.frame_drop_ratio = codec->config.frame_drop_ratio;          create.frame_drop_ratio = codec->config.frame_drop_ratio;
416    
         create.bquant_ratio = codec->config.bquant_ratio;  
         create.bquant_offset = codec->config.bquant_offset;  
   
417          create.num_threads = codec->config.num_threads;          create.num_threads = codec->config.num_threads;
418    
419          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))
420          {          {
421          case XVID_ERR_FAIL :          case XVID_ERR_FAIL :
422                  return ICERR_ERROR;                  return ICERR_ERROR;
# Line 350  Line 435 
435          codec->framenum = 0;          codec->framenum = 0;
436          codec->keyspacing = 0;          codec->keyspacing = 0;
437    
         codec->twopass.hints = codec->twopass.stats1 = codec->twopass.stats2 = INVALID_HANDLE_VALUE;  
         codec->twopass.hintstream = NULL;  
   
438          return ICERR_OK;          return ICERR_OK;
439  }  }
440    
# Line 362  Line 444 
444          if (codec->ehandle != NULL)          if (codec->ehandle != NULL)
445          {          {
446                  xvid_encore(codec->ehandle, XVID_ENC_DESTROY, NULL, NULL);                  xvid_encore(codec->ehandle, XVID_ENC_DESTROY, NULL, NULL);
447                    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);  
448                  }                  }
449                  if (codec->twopass.stats2 != INVALID_HANDLE_VALUE)  
450                  {          return ICERR_OK;
                         CloseHandle(codec->twopass.stats2);  
451                  }                  }
452                  if (codec->twopass.hintstream != NULL)  
453    
454    static void apply_zone_modifiers(xvid_enc_frame_t * frame, CONFIG * config, int framenum)
455                  {                  {
456                          free(codec->twopass.hintstream);      int i;
                 }  
457    
458                  codec->ehandle = NULL;      for (i=0; i<config->num_zones && config->zones[i].frame <= framenum; i++) ;
459        i--;
460    
461                  codec_2pass_finish(codec);      if (config->zones[i].greyscale) {
462            frame->vop_flags |= XVID_VOP_GREYSCALE;
463          }          }
464    
465          return ICERR_OK;      if (config->zones[i].chroma_opt) {
466            frame->vop_flags |= XVID_VOP_CHROMAOPT;
467        }
468    
469        if ((profiles[config->profile].flags & PROFILE_BVOP) && config->use_bvop) {
470            frame->bframe_threshold = config->zones[i].bvop_threshold;
471        }
472  }  }
473    
474    
475  LRESULT compress(CODEC * codec, ICCOMPRESS * icc)  LRESULT compress(CODEC * codec, ICCOMPRESS * icc)
476  {  {
477          BITMAPINFOHEADER * inhdr = icc->lpbiInput;          BITMAPINFOHEADER * inhdr = icc->lpbiInput;
# Line 396  Line 480 
480          xvid_enc_stats_t stats;          xvid_enc_stats_t stats;
481          int length;          int length;
482    
         // 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;  
                 }  
         }  
   
483          memset(&frame, 0, sizeof(frame));          memset(&frame, 0, sizeof(frame));
484          frame.version = XVID_VERSION;          frame.version = XVID_VERSION;
485    
# Line 412  Line 487 
487    
488          /* vol stuff */          /* vol stuff */
489    
490          if (codec->config.quant_type != QUANT_MODE_H263)      if ((profiles[codec->config.profile].flags & PROFILE_MPEGQUANT) &&
491            codec->config.quant_type != QUANT_MODE_H263)
492          {          {
493                  frame.vol_flags |= XVID_VOL_MPEGQUANT;                  frame.vol_flags |= XVID_VOL_MPEGQUANT;
494    
495                  // 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)  
                 {  
496                          frame.quant_intra_matrix = codec->config.qmatrix_intra;                          frame.quant_intra_matrix = codec->config.qmatrix_intra;
497                          frame.quant_inter_matrix = codec->config.qmatrix_inter;                          frame.quant_inter_matrix = codec->config.qmatrix_inter;
498                  }                  }else{
                 else  
                 {  
499                          frame.quant_intra_matrix = NULL;                          frame.quant_intra_matrix = NULL;
500                          frame.quant_inter_matrix = NULL;                          frame.quant_inter_matrix = NULL;
501                  }                  }
502          }          }
503    
504          if (codec->config.reduced_resolution) {      if ((profiles[codec->config.profile].flags & PROFILE_REDUCED) &&
505                codec->config.reduced_resolution) {
506                  frame.vol_flags |= XVID_VOL_REDUCED_ENABLE;                  frame.vol_flags |= XVID_VOL_REDUCED_ENABLE;
507                  frame.vop_flags |= XVID_VOP_REDUCED;    /* XXX: need auto decion mode */                  frame.vop_flags |= XVID_VOP_REDUCED;    /* XXX: need auto decion mode */
508          }          }
509    
510          if (codec->config.qpel) {          if ((profiles[codec->config.profile].flags & PROFILE_QPEL) && codec->config.qpel) {
511                  frame.vol_flags |= XVID_VOL_QUARTERPEL;                  frame.vol_flags |= XVID_VOL_QUARTERPEL;
512                  frame.motion |= XVID_ME_QUARTERPELREFINE16 | XVID_ME_QUARTERPELREFINE8;                  frame.motion |= XVID_ME_QUARTERPELREFINE16 | XVID_ME_QUARTERPELREFINE8;
513          }          }
514    
515          if (codec->config.gmc)          if ((profiles[codec->config.profile].flags & PROFILE_GMC) && codec->config.gmc)
516                  frame.vol_flags |= XVID_VOL_GMC;                  frame.vol_flags |= XVID_VOL_GMC;
517    
518          if (codec->config.interlacing)          if ((profiles[codec->config.profile].flags & PROFILE_INTERLACE) && codec->config.interlacing)
519                  frame.vol_flags |= XVID_VOL_INTERLACING;                  frame.vol_flags |= XVID_VOL_INTERLACING;
520    
521      /* vop stuff */      /* vop stuff */
# Line 454  Line 526 
526          if (codec->config.debug)          if (codec->config.debug)
527                  frame.vop_flags |= XVID_VOP_DEBUG;                  frame.vop_flags |= XVID_VOP_DEBUG;
528    
529        if (codec->config.trellis_quant) {
530            frame.vop_flags |= XVID_VOP_TRELLISQUANT;
531        }
532    
533          if (codec->config.motion_search > 4)          if (codec->config.motion_search > 4)
534                  frame.vop_flags |= XVID_VOP_INTER4V;                  frame.vop_flags |= XVID_VOP_INTER4V;
535    
536          if (codec->config.chromame)          if (codec->config.chromame)
537                  frame.vop_flags |= XVID_ME_CHROMA16 + XVID_ME_CHROMA8;                  frame.vop_flags |= XVID_ME_CHROMA16 + XVID_ME_CHROMA8;
538    
         if (codec->config.chroma_opt)  
                 frame.vop_flags |= XVID_VOP_CHROMAOPT;  
   
         check_greyscale_mode(&codec->config, &frame, codec->framenum);  
   
 /* XXX: hinted me  
   
 // 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;  
         }  
 // end of ugly hack  
   
         if (codec->config.hinted_me && codec->config.mode == DLG_MODE_2PASS_1)  
         {  
                 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;  
   
                 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;  
                 }  
         } */  
   
539          frame.motion |= pmvfast_presets[codec->config.motion_search];          frame.motion |= pmvfast_presets[codec->config.motion_search];
540    
541          switch (codec->config.vhq_mode)          switch (codec->config.vhq_mode)
# Line 555  Line 585 
585          frame.bitstream = icc->lpOutput;          frame.bitstream = icc->lpOutput;
586          frame.length = icc->lpbiOutput->biSizeImage;          frame.length = icc->lpbiOutput->biSizeImage;
587    
588          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;  
589    
590          case DLG_MODE_NULL :      if (codec->config.mode == RC_MODE_NULL) {
591                  outhdr->biSizeImage = 0;                  outhdr->biSizeImage = 0;
592                  *icc->lpdwFlags = AVIIF_KEYFRAME;                  *icc->lpdwFlags = AVIIF_KEYFRAME;
593                  return ICERR_OK;                  return ICERR_OK;
   
         default :  
                 DEBUGERR("Invalid encoding mode");  
                 return ICERR_ERROR;  
594          }          }
595    
   
596          // force keyframe spacing in 2-pass 1st pass          // force keyframe spacing in 2-pass 1st pass
597          if (codec->config.motion_search == 0)          if (codec->config.motion_search == 0)
598          {          {
# Line 602  Line 600 
600          }          }
601          else if (codec->keyspacing < codec->config.min_key_interval && codec->framenum)          else if (codec->keyspacing < codec->config.min_key_interval && codec->framenum)
602          {          {
603                  DEBUG("current frame forced to p-frame");                  DPRINTF("current frame forced to p-frame");
604                  frame.type = XVID_TYPE_PVOP;                  frame.type = XVID_TYPE_PVOP;
605          }          }
606    
607        /* frame-based stuff */
608        apply_zone_modifiers(&frame, &codec->config, codec->framenum);
609    
610        /* call encore */
611    
612          memset(&stats, 0, sizeof(stats));          memset(&stats, 0, sizeof(stats));
613          stats.version = XVID_VERSION;          stats.version = XVID_VERSION;
614    
# Line 625  Line 628 
628                  return ICERR_UNSUPPORTED;                  return ICERR_UNSUPPORTED;
629          }          }
630    
631          {  /* XXX: debug text */          DPRINTF("{type=%i len=%i} length=%i", stats.type, stats.length, length);
                 char tmp[100];  
                 wsprintf(tmp, " {type=%i len=%i} length=%i", stats.type, stats.length, length);  
                 OutputDebugString(tmp);  
         }  
632    
633      if (length == 0)    /* no encoder output */      if (length == 0)    /* no encoder output */
634          {          {
# Line 650  Line 649 
649    
650          outhdr->biSizeImage = length;          outhdr->biSizeImage = length;
651    
652          if (codec->config.mode == DLG_MODE_2PASS_1 && codec->config.discard1pass)          if (codec->config.mode == RC_MODE_2PASS1 && codec->config.discard1pass)
653              {              {
654                      outhdr->biSizeImage = 0;                      outhdr->biSizeImage = 0;
655              }              }
   
         /* 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;  
                 }  
             }  
         */  
   
         if (stats.type > 0)  
             {  
                   codec_2pass_update(codec, &stats);  
             }  
656      }      }
657    
658          ++codec->framenum;          codec->framenum++;
659          ++codec->keyspacing;          codec->keyspacing++;
660    
661          return ICERR_OK;          return ICERR_OK;
662  }  }
# Line 826  Line 796 
796          {          {
797                  xvid_gbl_convert_t convert;                  xvid_gbl_convert_t convert;
798    
799                  DEBUGFOURCC("input", icd->lpbiInput->biCompression);                  DPRINTF("input=%c%c%c%c output=%c%c%c%c",
800                  DEBUGFOURCC("output", icd->lpbiOutput->biCompression);              icd->lpbiInput->biCompression&0xff,
801                (icd->lpbiInput->biCompression>>8)&0xff,
802                (icd->lpbiInput->biCompression>>16)&0xff,
803                (icd->lpbiInput->biCompression>>24)&0xff,
804                icd->lpbiOutput->biCompression&0xff,
805                (icd->lpbiOutput->biCompression>>8)&0xff,
806                (icd->lpbiOutput->biCompression>>16)&0xff,
807                (icd->lpbiOutput->biCompression>>24)&0xff);
808    
809                  memset(&convert, 0, sizeof(convert));                  memset(&convert, 0, sizeof(convert));
810                  convert.version = XVID_VERSION;                  convert.version = XVID_VERSION;
# Line 897  Line 874 
874          return ICERR_OK;          return ICERR_OK;
875  }  }
876    
 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.1028

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