[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 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>  #include <xvid.h>
55  #include "debug.h"  #include "debug.h"
# Line 169  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 237  Line 243 
243    
244  LRESULT compress_frames_info(CODEC * codec, ICCOMPRESSFRAMES * icf)  LRESULT compress_frames_info(CODEC * codec, ICCOMPRESSFRAMES * icf)
245  {  {
246        //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;
# Line 289  Line 296 
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_plugin_t plugins[3];      xvid_enc_plugin_t plugins[3];
299            xvid_plugin_single_t single;
         xvid_plugin_fixed_t     fixed;  
         xvid_plugin_cbr_t cbr;  
300          xvid_plugin_2pass1_t pass1;          xvid_plugin_2pass1_t pass1;
301          xvid_plugin_2pass2_t pass2;          xvid_plugin_2pass2_t pass2;
302        int i;
303    
304      /* destroy previously created codec */      /* destroy previously created codec */
305          if(codec->ehandle) {          if(codec->ehandle) {
# Line 308  Line 314 
314    
315          memset(&create, 0, sizeof(create));          memset(&create, 0, sizeof(create));
316          create.version = XVID_VERSION;          create.version = XVID_VERSION;
         create.plugins = plugins;  
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 RC_MODE_CBR :          case RC_MODE_1PASS :
338          memset(&cbr, 0, sizeof(cbr));          memset(&single, 0, sizeof(single));
339              cbr.version = XVID_VERSION;              single.version = XVID_VERSION;
340          cbr.bitrate = codec->config.rc_bitrate;          single.bitrate = codec->config.bitrate;
341          cbr.reaction_delay_factor = codec->config.rc_reaction_delay_factor;          single.reaction_delay_factor = codec->config.rc_reaction_delay_factor;
342                  cbr.averaging_period = codec->config.rc_averaging_period;                  single.averaging_period = codec->config.rc_averaging_period;
343                  cbr.buffer = codec->config.rc_buffer;                  single.buffer = codec->config.rc_buffer;
344          plugins[create.num_plugins].func = xvid_plugin_cbr;          plugins[create.num_plugins].func = xvid_plugin_single;
345          plugins[create.num_plugins].param = &cbr;          plugins[create.num_plugins].param = &single;
346          create.num_plugins++;          create.num_plugins++;
347    
         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;  
         create.num_plugins++;  
                 break;  
   
348          case RC_MODE_2PASS1 :          case RC_MODE_2PASS1 :
349          memset(&pass1, 0, sizeof(pass1));          memset(&pass1, 0, sizeof(pass1));
350              pass1.version = XVID_VERSION;              pass1.version = XVID_VERSION;
351            pass1.filename = codec->config.stats;
352    
353          plugins[create.num_plugins].func = xvid_plugin_2pass1;          plugins[create.num_plugins].func = xvid_plugin_2pass1;
354          plugins[create.num_plugins].param = &pass1;          plugins[create.num_plugins].param = &pass1;
355          create.num_plugins++;          create.num_plugins++;
356                  break;                  break;
357    
358            case RC_MODE_2PASS2 :
         case RC_MODE_2PASS2_INT :  
         case RC_MODE_2PASS2_EXT :  
359          memset(&pass2, 0, sizeof(pass2));          memset(&pass2, 0, sizeof(pass2));
360              pass2.version = XVID_VERSION;              pass2.version = XVID_VERSION;
361          pass2.min_quant[0] = codec->config.min_iquant;          pass2.bitrate = codec->config.bitrate;
         pass2.max_quant[0] = codec->config.max_iquant;  
         pass2.min_quant[1] = codec->config.min_pquant;  
         pass2.max_quant[1] = codec->config.max_pquant;  
         //pass2.min_quant[2] = codec->config.min_bquant;  
         //pass2.max_quant[2] = codec->config.max_bquant;  
362                  pass2.filename = codec->config.stats;                  pass2.filename = codec->config.stats;
363                  if (codec->config.mode == RC_MODE_2PASS2_INT) {  
                         pass2.bitrate = 10000;  /* xxx */  
                 }  
364          plugins[create.num_plugins].func = xvid_plugin_2pass2;          plugins[create.num_plugins].func = xvid_plugin_2pass2;
365          plugins[create.num_plugins].param = &pass2;          plugins[create.num_plugins].param = &pass2;
366          create.num_plugins++;          create.num_plugins++;
# Line 369  Line 373 
373                  break;                  break;
374          }          }
375    
376          if (codec->config.lum_masking) {          if ((profiles[codec->config.profile].flags & PROFILE_ADAPTQUANT) && codec->config.lum_masking) {
377          plugins[create.num_plugins].func = xvid_plugin_lumimasking;          plugins[create.num_plugins].func = xvid_plugin_lumimasking;
378          plugins[create.num_plugins].param = NULL;          plugins[create.num_plugins].param = NULL;
379          create.num_plugins++;          create.num_plugins++;
# Line 379  Line 383 
383      plugins[create.num_plugins].param = NULL;      plugins[create.num_plugins].param = NULL;
384      create.num_plugins++;      create.num_plugins++;
385    
386        create.profile = profiles[codec->config.profile].id;
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.max_key_interval = codec->config.max_key_interval;
394    
395        create.min_quant[0] = codec->config.min_iquant;
396        create.max_quant[0] = codec->config.max_iquant;
397        create.min_quant[1] = codec->config.min_pquant;
398        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    
410          if (codec->config.closed_gov)          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    
     if (codec->config.use_bvop)  
         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, NULL))          switch(xvid_encore(0, XVID_ENC_CREATE, &create, NULL))
# Line 436  Line 450 
450          return ICERR_OK;          return ICERR_OK;
451  }  }
452    
453    
454    static void apply_zone_modifiers(xvid_enc_frame_t * frame, CONFIG * config, int framenum)
455    {
456        int i;
457    
458        for (i=0; i<config->num_zones && config->zones[i].frame <= framenum; i++) ;
459        i--;
460    
461        if (config->zones[i].greyscale) {
462            frame->vop_flags |= XVID_VOP_GREYSCALE;
463        }
464    
465        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 444  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)  
   
483          memset(&frame, 0, sizeof(frame));          memset(&frame, 0, sizeof(frame));
484          frame.version = XVID_VERSION;          frame.version = XVID_VERSION;
485    
# Line 453  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 495  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;  
   
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 541  Line 573 
573                  break;                  break;
574          }          }
575    
     frame.bframe_threshold = codec->config.bvop_threshold;  
   
576          frame.input.plane[0] = icc->lpInput;          frame.input.plane[0] = icc->lpInput;
577          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;
578    
# 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 RC_MODE_CBR :  
                 frame.quant = 0;  /* use xvidcore cbr rate control */  
                 break;  
589    
590          //case RC_MODE_VBR_QUAL :      if (codec->config.mode == RC_MODE_NULL) {
         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;  
   
         case 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 :  
                 DPRINTF("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 600  Line 604 
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 646  Line 655 
655              }              }
656      }      }
657    
658          ++codec->framenum;          codec->framenum++;
659          ++codec->keyspacing;          codec->keyspacing++;
660    
661          return ICERR_OK;          return ICERR_OK;
662  }  }

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

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