[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 983, Sat Apr 12 06:58:50 2003 UTC revision 1061, Tue Jun 10 10:07:03 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>  #include <xvid.h>
55  #include "debug.h"  #include "debug.h"
56  #include "codec.h"  #include "codec.h"
57    #include "status.h"
58    
59    
60  static const int pmvfast_presets[7] = {  static const int pmvfast_presets[7] = {
# Line 169  Line 171 
171          BITMAPINFOHEADER * inhdr = &lpbiInput->bmiHeader;          BITMAPINFOHEADER * inhdr = &lpbiInput->bmiHeader;
172          BITMAPINFOHEADER * outhdr = &lpbiOutput->bmiHeader;          BITMAPINFOHEADER * outhdr = &lpbiOutput->bmiHeader;
173    
174        /* VFWEXT detection */
175        if (inhdr->biCompression == VFWEXT_FOURCC) {
176            return (ICM_USER+0x0fff);
177        }
178    
179          if (get_colorspace(inhdr) == XVID_CSP_NULL)          if (get_colorspace(inhdr) == XVID_CSP_NULL)
180          {          {
181                  return ICERR_BADFORMAT;                  return ICERR_BADFORMAT;
# Line 237  Line 244 
244    
245  LRESULT compress_frames_info(CODEC * codec, ICCOMPRESSFRAMES * icf)  LRESULT compress_frames_info(CODEC * codec, ICCOMPRESSFRAMES * icf)
246  {  {
247    #if 0
248        DPRINTF("%i %i", icf->lStartFrame, icf->lFrameCount);
249    #endif
250          codec->fincr = icf->dwScale;          codec->fincr = icf->dwScale;
251          codec->fbase = icf->dwRate;          codec->fbase = icf->dwRate;
252          return ICERR_OK;          return ICERR_OK;
# Line 289  Line 299 
299          xvid_gbl_init_t init;          xvid_gbl_init_t init;
300          xvid_enc_create_t create;          xvid_enc_create_t create;
301      xvid_enc_plugin_t plugins[3];      xvid_enc_plugin_t plugins[3];
302            xvid_plugin_single_t single;
         xvid_plugin_fixed_t     fixed;  
         xvid_plugin_cbr_t cbr;  
303          xvid_plugin_2pass1_t pass1;          xvid_plugin_2pass1_t pass1;
304          xvid_plugin_2pass2_t pass2;          xvid_plugin_2pass2_t pass2;
305        int i;
306    
307      /* destroy previously created codec */      /* destroy previously created codec */
308          if(codec->ehandle) {          if(codec->ehandle) {
# Line 304  Line 313 
313      memset(&init, 0, sizeof(init));      memset(&init, 0, sizeof(init));
314          init.version = XVID_VERSION;          init.version = XVID_VERSION;
315          init.cpu_flags = codec->config.cpu;          init.cpu_flags = codec->config.cpu;
316        init.debug = codec->config.debug;
317          xvid_global(0, XVID_GBL_INIT, &init, NULL);          xvid_global(0, XVID_GBL_INIT, &init, NULL);
318    
319          memset(&create, 0, sizeof(create));          memset(&create, 0, sizeof(create));
320          create.version = XVID_VERSION;          create.version = XVID_VERSION;
         create.plugins = plugins;  
321    
322        /* zones */
323        create.zones = malloc(sizeof(xvid_enc_zone_t) * codec->config.num_zones);
324        create.num_zones = codec->config.num_zones;
325        for (i=0; i < create.num_zones; i++) {
326            create.zones[i].frame = codec->config.zones[i].frame;
327            if (codec->config.zones[i].mode == RC_ZONE_QUANT) {
328                create.zones[i].mode = XVID_ZONE_QUANT;
329                create.zones[i].increment = codec->config.zones[i].quant;
330            }else{
331                create.zones[i].mode = XVID_ZONE_WEIGHT;
332                create.zones[i].increment = codec->config.zones[i].weight;
333            }
334            create.zones[i].base = 100;
335        }
336    
337        /* plugins */
338            create.plugins = plugins;
339          switch (codec->config.mode)          switch (codec->config.mode)
340          {          {
341          case RC_MODE_CBR :          case RC_MODE_1PASS :
342          memset(&cbr, 0, sizeof(cbr));          memset(&single, 0, sizeof(single));
343              cbr.version = XVID_VERSION;              single.version = XVID_VERSION;
344          cbr.bitrate = codec->config.rc_bitrate;          single.bitrate = codec->config.bitrate * CONFIG_KBPS;
345          cbr.reaction_delay_factor = codec->config.rc_reaction_delay_factor;          single.reaction_delay_factor = codec->config.rc_reaction_delay_factor;
346                  cbr.averaging_period = codec->config.rc_averaging_period;                  single.averaging_period = codec->config.rc_averaging_period;
347                  cbr.buffer = codec->config.rc_buffer;                  single.buffer = codec->config.rc_buffer;
348          plugins[create.num_plugins].func = xvid_plugin_cbr;          plugins[create.num_plugins].func = xvid_plugin_single;
349          plugins[create.num_plugins].param = &cbr;          plugins[create.num_plugins].param = &single;
         create.num_plugins++;  
   
         case RC_MODE_FIXED :  
         memset(&fixed, 0, sizeof(fixed));  
             fixed.version = XVID_VERSION;  
         fixed.quant_increment = codec->config.quant;  
         fixed.quant_base = 1;  
         plugins[create.num_plugins].func = xvid_plugin_fixed;  
         plugins[create.num_plugins].param = &fixed;  
350          create.num_plugins++;          create.num_plugins++;
351                  break;                  break;
352    
353          case RC_MODE_2PASS1 :          case RC_MODE_2PASS1 :
354          memset(&pass1, 0, sizeof(pass1));          memset(&pass1, 0, sizeof(pass1));
355              pass1.version = XVID_VERSION;              pass1.version = XVID_VERSION;
356            pass1.filename = codec->config.stats;
357    
358          plugins[create.num_plugins].func = xvid_plugin_2pass1;          plugins[create.num_plugins].func = xvid_plugin_2pass1;
359          plugins[create.num_plugins].param = &pass1;          plugins[create.num_plugins].param = &pass1;
360          create.num_plugins++;          create.num_plugins++;
361                  break;                  break;
362    
363            case RC_MODE_2PASS2 :
         case RC_MODE_2PASS2_INT :  
         case RC_MODE_2PASS2_EXT :  
364          memset(&pass2, 0, sizeof(pass2));          memset(&pass2, 0, sizeof(pass2));
365              pass2.version = XVID_VERSION;              pass2.version = XVID_VERSION;
366          pass2.min_quant[0] = codec->config.min_iquant;          if (codec->config.use_2pass_bitrate) {
367          pass2.max_quant[0] = codec->config.max_iquant;              pass2.bitrate = codec->config.bitrate * CONFIG_KBPS;
368          pass2.min_quant[1] = codec->config.min_pquant;          }else{
369          pass2.max_quant[1] = codec->config.max_pquant;              pass2.bitrate = -codec->config.desired_size;    /* kilobytes */
         //pass2.min_quant[2] = codec->config.min_bquant;  
         //pass2.max_quant[2] = codec->config.max_bquant;  
                 pass2.filename = codec->config.stats;  
                 if (codec->config.mode == RC_MODE_2PASS2_INT) {  
                         pass2.bitrate = 10000;  /* xxx */  
370                  }                  }
371                    pass2.filename = codec->config.stats;
372    
373            pass2.keyframe_boost = codec->config.keyframe_boost;   /* keyframe boost percentage: [0..100...]; */
374            pass2.payback_method = codec->config.bitrate_payback_method;
375            pass2.bitrate_payback_delay = codec->config.bitrate_payback_delay;
376            pass2.curve_compression_high = codec->config.curve_compression_high;
377            pass2.curve_compression_low = codec->config.curve_compression_low;
378            pass2.max_overflow_improvement = codec->config.twopass_max_overflow_improvement;
379            pass2.max_overflow_degradation = codec->config.twopass_max_overflow_degradation;
380            pass2.kftreshold = codec->config.kftreshold;
381                pass2.kfreduction = codec->config.kfreduction;
382            pass2.min_key_interval = codec->config.min_key_interval;
383            pass2.container_frame_overhead = 24;    /* AVI */
384    
385          plugins[create.num_plugins].func = xvid_plugin_2pass2;          plugins[create.num_plugins].func = xvid_plugin_2pass2;
386          plugins[create.num_plugins].param = &pass2;          plugins[create.num_plugins].param = &pass2;
387          create.num_plugins++;          create.num_plugins++;
# Line 369  Line 394 
394                  break;                  break;
395          }          }
396    
397          if (codec->config.lum_masking) {          if ((profiles[codec->config.profile].flags & PROFILE_ADAPTQUANT) && codec->config.lum_masking) {
398          plugins[create.num_plugins].func = xvid_plugin_lumimasking;          plugins[create.num_plugins].func = xvid_plugin_lumimasking;
399          plugins[create.num_plugins].param = NULL;          plugins[create.num_plugins].param = NULL;
400          create.num_plugins++;          create.num_plugins++;
# Line 379  Line 404 
404      plugins[create.num_plugins].param = NULL;      plugins[create.num_plugins].param = NULL;
405      create.num_plugins++;      create.num_plugins++;
406    
407        create.profile = profiles[codec->config.profile].id;
408    
409          create.width = lpbiInput->bmiHeader.biWidth;          create.width = lpbiInput->bmiHeader.biWidth;
410          create.height = lpbiInput->bmiHeader.biHeight;          create.height = lpbiInput->bmiHeader.biHeight;
411          create.fincr = codec->fincr;          create.fincr = codec->fincr;
412          create.fbase = codec->fbase;          create.fbase = codec->fbase;
413    
414        create.max_key_interval = codec->config.max_key_interval;
415    
416        create.min_quant[0] = codec->config.min_iquant;
417        create.max_quant[0] = codec->config.max_iquant;
418        create.min_quant[1] = codec->config.min_pquant;
419        create.max_quant[1] = codec->config.max_pquant;
420        create.min_quant[2] = codec->config.min_bquant;
421        create.max_quant[2] = codec->config.max_bquant;
422    
423        if ((profiles[codec->config.profile].flags & PROFILE_BVOP) && codec->config.use_bvop) {
424            create.max_bframes = codec->config.max_bframes;
425                create.bquant_ratio = codec->config.bquant_ratio;
426                create.bquant_offset = codec->config.bquant_offset;
427    
428      if (codec->config.packed)      if (codec->config.packed)
429          create.global |= XVID_GLOBAL_PACKED;          create.global |= XVID_GLOBAL_PACKED;
430    
431          if (codec->config.closed_gov)          if (codec->config.closed_gov)
432                  create.global |= XVID_GLOBAL_CLOSED_GOP;                  create.global |= XVID_GLOBAL_CLOSED_GOP;
433    
434      create.max_key_interval = codec->config.max_key_interval;      }
         /* XXX: param.min_quantizer = codec->config.min_pquant;  
         param.max_quantizer = codec->config.max_pquant; */  
435    
     if (codec->config.use_bvop)  
         create.max_bframes = codec->config.max_bframes;  
436          create.frame_drop_ratio = codec->config.frame_drop_ratio;          create.frame_drop_ratio = codec->config.frame_drop_ratio;
437    
         create.bquant_ratio = codec->config.bquant_ratio;  
         create.bquant_offset = codec->config.bquant_offset;  
   
438          create.num_threads = codec->config.num_threads;          create.num_threads = codec->config.num_threads;
439    
440          switch(xvid_encore(0, XVID_ENC_CREATE, &create, NULL))          switch(xvid_encore(0, XVID_ENC_CREATE, &create, NULL))
# Line 421  Line 456 
456          codec->framenum = 0;          codec->framenum = 0;
457          codec->keyspacing = 0;          codec->keyspacing = 0;
458    
459        if (codec->config.display_status) {
460            status_destroy_always(&codec->status);
461            status_create(&codec->status, codec->fincr, codec->fbase);
462        }
463    
464          return ICERR_OK;          return ICERR_OK;
465  }  }
466    
467    
468  LRESULT compress_end(CODEC * codec)  LRESULT compress_end(CODEC * codec)
469  {  {
470          if (codec->ehandle != NULL)      if (codec->ehandle != NULL) {
         {  
471                  xvid_encore(codec->ehandle, XVID_ENC_DESTROY, NULL, NULL);                  xvid_encore(codec->ehandle, XVID_ENC_DESTROY, NULL, NULL);
472                  codec->ehandle = NULL;                  codec->ehandle = NULL;
473          }          }
474    
475        if (codec->config.display_status)
476            status_destroy(&codec->status);
477    
478          return ICERR_OK;          return ICERR_OK;
479  }  }
480    
481    
482    static void apply_zone_modifiers(xvid_enc_frame_t * frame, CONFIG * config, int framenum)
483    {
484        int i;
485    
486        for (i=0; i<config->num_zones && config->zones[i].frame <= framenum; i++) ;
487        i--;
488    
489        if (config->zones[i].greyscale) {
490            frame->vop_flags |= XVID_VOP_GREYSCALE;
491        }
492    
493        if (config->zones[i].chroma_opt) {
494            frame->vop_flags |= XVID_VOP_CHROMAOPT;
495        }
496    
497        if ((profiles[config->profile].flags & PROFILE_BVOP) && config->use_bvop) {
498            frame->bframe_threshold = config->zones[i].bvop_threshold;
499        }
500    }
501    
502    
503  LRESULT compress(CODEC * codec, ICCOMPRESS * icc)  LRESULT compress(CODEC * codec, ICCOMPRESS * icc)
504  {  {
505          BITMAPINFOHEADER * inhdr = icc->lpbiInput;          BITMAPINFOHEADER * inhdr = icc->lpbiInput;
# Line 444  Line 508 
508          xvid_enc_stats_t stats;          xvid_enc_stats_t stats;
509          int length;          int length;
510    
         // mpeg2avi yuv bug workaround (2 instances of CODEC)  
   
511          memset(&frame, 0, sizeof(frame));          memset(&frame, 0, sizeof(frame));
512          frame.version = XVID_VERSION;          frame.version = XVID_VERSION;
513    
# Line 453  Line 515 
515    
516          /* vol stuff */          /* vol stuff */
517    
518          if (codec->config.quant_type != QUANT_MODE_H263)      if ((profiles[codec->config.profile].flags & PROFILE_MPEGQUANT) &&
519            codec->config.quant_type != QUANT_MODE_H263)
520          {          {
521                  frame.vol_flags |= XVID_VOL_MPEGQUANT;                  frame.vol_flags |= XVID_VOL_MPEGQUANT;
522    
523                  // 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)  
                 {  
524                          frame.quant_intra_matrix = codec->config.qmatrix_intra;                          frame.quant_intra_matrix = codec->config.qmatrix_intra;
525                          frame.quant_inter_matrix = codec->config.qmatrix_inter;                          frame.quant_inter_matrix = codec->config.qmatrix_inter;
526                  }                  }else{
                 else  
                 {  
527                          frame.quant_intra_matrix = NULL;                          frame.quant_intra_matrix = NULL;
528                          frame.quant_inter_matrix = NULL;                          frame.quant_inter_matrix = NULL;
529                  }                  }
530          }          }
531    
532          if (codec->config.reduced_resolution) {      if ((profiles[codec->config.profile].flags & PROFILE_REDUCED) &&
533                codec->config.reduced_resolution) {
534                  frame.vol_flags |= XVID_VOL_REDUCED_ENABLE;                  frame.vol_flags |= XVID_VOL_REDUCED_ENABLE;
535                  frame.vop_flags |= XVID_VOP_REDUCED;    /* XXX: need auto decion mode */                  frame.vop_flags |= XVID_VOP_REDUCED;    /* XXX: need auto decion mode */
536          }          }
537    
538          if (codec->config.qpel) {          if ((profiles[codec->config.profile].flags & PROFILE_QPEL) && codec->config.qpel) {
539                  frame.vol_flags |= XVID_VOL_QUARTERPEL;                  frame.vol_flags |= XVID_VOL_QUARTERPEL;
540                  frame.motion |= XVID_ME_QUARTERPELREFINE16 | XVID_ME_QUARTERPELREFINE8;                  frame.motion |= XVID_ME_QUARTERPELREFINE16 | XVID_ME_QUARTERPELREFINE8;
541          }          }
542    
543          if (codec->config.gmc)          if ((profiles[codec->config.profile].flags & PROFILE_GMC) && codec->config.gmc)
544                  frame.vol_flags |= XVID_VOL_GMC;                  frame.vol_flags |= XVID_VOL_GMC;
545    
546          if (codec->config.interlacing)          if ((profiles[codec->config.profile].flags & PROFILE_INTERLACE) && codec->config.interlacing)
547                  frame.vol_flags |= XVID_VOL_INTERLACING;                  frame.vol_flags |= XVID_VOL_INTERLACING;
548    
549      /* vop stuff */      /* vop stuff */
# Line 492  Line 551 
551          frame.vop_flags |= XVID_VOP_HALFPEL;          frame.vop_flags |= XVID_VOP_HALFPEL;
552          frame.vop_flags |= XVID_VOP_HQACPRED;          frame.vop_flags |= XVID_VOP_HQACPRED;
553    
554          if (codec->config.debug)          if (codec->config.vop_debug)
555                  frame.vop_flags |= XVID_VOP_DEBUG;                  frame.vop_flags |= XVID_VOP_DEBUG;
556    
557        if (codec->config.trellis_quant) {
558            frame.vop_flags |= XVID_VOP_TRELLISQUANT;
559        }
560    
561          if (codec->config.motion_search > 4)          if (codec->config.motion_search > 4)
562                  frame.vop_flags |= XVID_VOP_INTER4V;                  frame.vop_flags |= XVID_VOP_INTER4V;
563    
564          if (codec->config.chromame)          if (codec->config.chromame)
565                  frame.vop_flags |= XVID_ME_CHROMA16 + XVID_ME_CHROMA8;                  frame.vop_flags |= XVID_ME_CHROMA16 + XVID_ME_CHROMA8;
566    
         if (codec->config.chroma_opt)  
                 frame.vop_flags |= XVID_VOP_CHROMAOPT;  
   
567          frame.motion |= pmvfast_presets[codec->config.motion_search];          frame.motion |= pmvfast_presets[codec->config.motion_search];
568    
569          switch (codec->config.vhq_mode)          switch (codec->config.vhq_mode)
# Line 541  Line 601 
601                  break;                  break;
602          }          }
603    
     frame.bframe_threshold = codec->config.bvop_threshold;  
   
604          frame.input.plane[0] = icc->lpInput;          frame.input.plane[0] = icc->lpInput;
605          frame.input.stride[0] = (((icc->lpbiInput->biWidth * icc->lpbiInput->biBitCount) + 31) & ~31) >> 3;          frame.input.stride[0] = (((icc->lpbiInput->biWidth * icc->lpbiInput->biBitCount) + 31) & ~31) >> 3;
606    
# Line 555  Line 613 
613          frame.bitstream = icc->lpOutput;          frame.bitstream = icc->lpOutput;
614          frame.length = icc->lpbiOutput->biSizeImage;          frame.length = icc->lpbiOutput->biSizeImage;
615    
616          switch (codec->config.mode)      frame.quant = 0;
         {  
         case RC_MODE_CBR :  
                 frame.quant = 0;  /* use xvidcore cbr rate control */  
                 break;  
   
         //case RC_MODE_VBR_QUAL :  
         case RC_MODE_FIXED :  
         case RC_MODE_2PASS1 :  
                 /*if (codec_get_quant(codec, &frame) == ICERR_ERROR)  
                 {  
                         return ICERR_ERROR;  
                 }*/  
                 break;  
   
         case RC_MODE_2PASS2_EXT :  
         case RC_MODE_2PASS2_INT :  
                 /*if (codec_2pass_get_quant(codec, &frame) == ICERR_ERROR)  
                 {  
                         return ICERR_ERROR;  
                 }*/  
                 break;  
617    
618          case RC_MODE_NULL :      if (codec->config.mode == RC_MODE_NULL) {
619                  outhdr->biSizeImage = 0;                  outhdr->biSizeImage = 0;
620                  *icc->lpdwFlags = AVIIF_KEYFRAME;                  *icc->lpdwFlags = AVIIF_KEYFRAME;
621                  return ICERR_OK;                  return ICERR_OK;
   
         default :  
                 DPRINTF("Invalid encoding mode");  
                 return ICERR_ERROR;  
622          }          }
623    
   
624          // force keyframe spacing in 2-pass 1st pass          // force keyframe spacing in 2-pass 1st pass
625          if (codec->config.motion_search == 0)          if (codec->config.motion_search == 0)
626          {          {
# Line 600  Line 632 
632                  frame.type = XVID_TYPE_PVOP;                  frame.type = XVID_TYPE_PVOP;
633          }          }
634    
635        /* frame-based stuff */
636        apply_zone_modifiers(&frame, &codec->config, codec->framenum);
637    
638        /* call encore */
639    
640          memset(&stats, 0, sizeof(stats));          memset(&stats, 0, sizeof(stats));
641          stats.version = XVID_VERSION;          stats.version = XVID_VERSION;
642    
# Line 619  Line 656 
656                  return ICERR_UNSUPPORTED;                  return ICERR_UNSUPPORTED;
657          }          }
658    
659        if (codec->config.display_status && stats.type>0) {
660            status_update(&codec->status, stats.type, stats.length, stats.quant);
661        }
662    
663          DPRINTF("{type=%i len=%i} length=%i", stats.type, stats.length, length);          DPRINTF("{type=%i len=%i} length=%i", stats.type, stats.length, length);
664    
665      if (length == 0)    /* no encoder output */      if (length == 0)    /* no encoder output */
# Line 646  Line 687 
687              }              }
688      }      }
689    
690          ++codec->framenum;          codec->framenum++;
691          ++codec->keyspacing;          codec->keyspacing++;
692    
693          return ICERR_OK;          return ICERR_OK;
694  }  }
# Line 702  Line 743 
743    
744          if (get_colorspace(inhdr) != XVID_CSP_NULL) {          if (get_colorspace(inhdr) != XVID_CSP_NULL) {
745                  memcpy(outhdr, inhdr, sizeof(BITMAPINFOHEADER));                  memcpy(outhdr, inhdr, sizeof(BITMAPINFOHEADER));
746                  // XXX: should we set outhdr->biSize ??                  /* XXX: should we set outhdr->biSize ?? */
747                  return ICERR_OK;                  return ICERR_OK;
748          }          }
749          /* --- yv12 --- */          /* --- yv12 --- */

Legend:
Removed from v.983  
changed lines
  Added in v.1061

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