Diff of /xvidcore/src/encoder.c

-revision 1.76.2.37, Mon Jan 13 14:33:24 2003 UTC
+revision 1.134, Sat Dec 18 16:02:00 2010 UTC
 Line 3
   *  XVID MPEG-4 VIDEO CODEC
   *  -  Encoder main module  -
   *
-  *  This program is an implementation of a part of one or more MPEG-4
+  *  Copyright(C) 2002-2010 Michael Militzer <isibaar@xvid.org>
-  *  Video tools as specified in ISO/IEC 14496-2 standard.  Those intending
+  *               2002-2003 Peter Ross <pross@xvid.org>
-  *  to use this software module in hardware or software products are
+  *               2002      Daniel Smith <danielsmith@astroboymail.com>
-  *  advised that its use may infringe existing patents or copyrights, and
-  *  any such use would be at such party's own risk.  The original
-  *  developer of this software module and his/her company, and subsequent
-  *  editors and their companies, will have no liability for use of this
-  *  software or modifications or derivatives thereof.
   *
   *  This program is free software; you can redistribute it and/or modify
   *  it under the terms of the GNU General Public License as published by
-Line 26
+Line 21
   *  along with this program; if not, write to the Free Software
   *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
   *
-  ****************************************************************************/
- /*****************************************************************************
-  *
-  *  History
-  *
-  *  10.07.2002  added BFRAMES_DEC_DEBUG support
-  *              MinChen <chenm001@163.com>
-  *  20.06.2002 bframe patch
-  *  08.05.2002 fix some problem in DEBUG mode;
-  *             MinChen <chenm001@163.com>
-  *  14.04.2002 added FrameCodeB()
-  *
   *  $Id$
   *
   ****************************************************************************/
-Line 56
+Line 38
  #include "image/font.h"
  #include "motion/sad.h"
  #include "motion/motion.h"
+ #include "motion/gmc.h"
  #include "bitstream/cbp.h"
  #include "utils/mbfunctions.h"
  #include "bitstream/bitstream.h"
  #include "bitstream/mbcoding.h"
- #include "utils/ratecontrol.h"
  #include "utils/emms.h"
  #include "bitstream/mbcoding.h"
- #include "quant/adapt_quant.h"
  #include "quant/quant_matrix.h"
  #include "utils/mem_align.h"
- /*****************************************************************************
+ # include "motion/motion_smp.h"
-  * Local macros
-  ****************************************************************************/
- #define ENC_CHECK(X) if(!(X)) return XVID_ERR_FORMAT
- #define SWAP(_T_,A,B)    { _T_ tmp = A; A = B; B = tmp; }
  /*****************************************************************************
   * Local function prototypes
   ****************************************************************************/
  static int FrameCodeI(Encoder * pEnc,
-                                           Bitstream * bs,
+                                           Bitstream * bs);
-                                           uint32_t * pBits);
  static int FrameCodeP(Encoder * pEnc,
-                                           Bitstream * bs,
+                                           Bitstream * bs);
-                                           uint32_t * pBits,
-                                           bool force_inter,
-                                           bool vol_header);
  static void FrameCodeB(Encoder * pEnc,
                                             FRAMEINFO * frame,
-                                            Bitstream * bs,
+                                            Bitstream * bs);
-                                            uint32_t * pBits);
- /*****************************************************************************
-  * Local data
-  ****************************************************************************/
- static int DQtab[4] = {
-         -1, -2, 1, 2
- };
- static int iDQtab[5] = {
-, 0, NO_CHANGE, 2, 3
- };
  /*****************************************************************************
-Line 117
+Line 78
   * and cleaning code.
   *
   * Returned values :
-  *    - XVID_ERR_OK     - no errors
+  *      - 0                             - no errors
   *    - XVID_ERR_MEMORY - the libc could not allocate memory, the function
   *                        cleans the structure before exiting.
   *                        pParam->handle is also set to NULL.
   *
   ****************************************************************************/
- int
- encoder_create(XVID_ENC_PARAM * pParam)
- {
-         Encoder *pEnc;
-         int i;
-         pParam->handle = NULL;
-         ENC_CHECK(pParam);
-         ENC_CHECK(pParam->width > 0 && pParam->width <= 1920);
-         ENC_CHECK(pParam->height > 0 && pParam->height <= 1280);
-         ENC_CHECK(!(pParam->width % 2));
-         ENC_CHECK(!(pParam->height % 2));
-         /* Fps */
-         if (pParam->fincr <= 0 || pParam->fbase <= 0) {
-                 pParam->fincr = 1;
-                 pParam->fbase = 25;
-         }
          /*
           * Simplify the "fincr/fbase" fraction
-          * (neccessary, since windows supplies us with huge numbers)
           */
+ static int
+ gcd(int a, int b)
+ {
+         int r ;
-         i = pParam->fincr;
+         if (b > a) {
-         while (i > 1) {
+                 r = a;
-                 if (pParam->fincr % i == 0 && pParam->fbase % i == 0) {
+                 a = b;
-                         pParam->fincr /= i;
+                 b = r;
-                         pParam->fbase /= i;
-                         i = pParam->fincr;
-                         continue;
-                 }
-                 i--;
          }
-         if (pParam->fbase > 65535) {
+         while ((r = a % b)) {
-                 float div = (float) pParam->fbase / 65535;
+                 a = b;
+                 b = r;
-                 pParam->fbase = (int) (pParam->fbase / div);
+         }
-                 pParam->fincr = (int) (pParam->fincr / div);
+         return b;
          }
-         /* Bitrate allocator defaults */
+ static void
+ simplify_time(int *inc, int *base)
-         if (pParam->rc_bitrate <= 0)
+ {
-                 pParam->rc_bitrate = 900000;
+         /* common factor */
+         const int s = gcd(*inc, *base);
-         if (pParam->rc_reaction_delay_factor <= 0)
+   *inc  /= s;
-                 pParam->rc_reaction_delay_factor = 16;
+   *base /= s;
-         if (pParam->rc_averaging_period <= 0)
+         if (*base > 65535 || *inc > 65535) {
-                 pParam->rc_averaging_period = 100;
+                 int *biggest;
+                 int *other;
+                 float div;
+                 if (*base > *inc) {
+                         biggest = base;
+                         other = inc;
+                 } else {
+                         biggest = inc;
+                         other = base;
+                 }
-         if (pParam->rc_buffer <= 0)
+                 div = ((float)*biggest)/((float)65535);
-                 pParam->rc_buffer = 100;
+                 *biggest = (unsigned int)(((float)*biggest)/div);
+                 *other = (unsigned int)(((float)*other)/div);
+         }
+ }
-         /* Max and min quantizers */
-         if ((pParam->min_quantizer <= 0) || (pParam->min_quantizer > 31))
+ int
-                 pParam->min_quantizer = 1;
+ enc_create(xvid_enc_create_t * create)
+ {
+         Encoder *pEnc;
+         int n;
-         if ((pParam->max_quantizer <= 0) || (pParam->max_quantizer > 31))
+         if (XVID_VERSION_MAJOR(create->version) != 1) /* v1.x.x */
-                 pParam->max_quantizer = 31;
+                 return XVID_ERR_VERSION;
-         if (pParam->max_quantizer < pParam->min_quantizer)
+         if (create->width%2 || create->height%2)
-                 pParam->max_quantizer = pParam->min_quantizer;
+                 return XVID_ERR_FAIL;
-         /* 1 keyframe each 10 seconds */
+         if (create->width<=0 || create->height<=0)
+                 return XVID_ERR_FAIL;
-         if (pParam->max_key_interval <= 0)
+         /* allocate encoder struct */
-                 pParam->max_key_interval = 10 * pParam->fincr / pParam->fbase;
          pEnc = (Encoder *) xvid_malloc(sizeof(Encoder), CACHE_LINE);
          if (pEnc == NULL)
                  return XVID_ERR_MEMORY;
-         /* Zero the Encoder Structure */
          memset(pEnc, 0, sizeof(Encoder));
-         /* Fill members of Encoder structure */
+         pEnc->mbParam.profile = create->profile;
-         pEnc->mbParam.width = pParam->width;
-         pEnc->mbParam.height = pParam->height;
+         /* global flags */
+         pEnc->mbParam.global_flags = create->global;
+   if ((pEnc->mbParam.global_flags & XVID_GLOBAL_PACKED))
+     pEnc->mbParam.global_flags |= XVID_GLOBAL_DIVX5_USERDATA;
+         /* width, height */
+         pEnc->mbParam.width = create->width;
+         pEnc->mbParam.height = create->height;
          pEnc->mbParam.mb_width = (pEnc->mbParam.width + 15) / 16;
          pEnc->mbParam.mb_height = (pEnc->mbParam.height + 15) / 16;
          pEnc->mbParam.edged_width = 16 * pEnc->mbParam.mb_width + 2 * EDGE_SIZE;
          pEnc->mbParam.edged_height = 16 * pEnc->mbParam.mb_height + 2 * EDGE_SIZE;
-         pEnc->mbParam.fbase = pParam->fbase;
+         /* framerate */
-         pEnc->mbParam.fincr = pParam->fincr;
+         pEnc->mbParam.fincr = MAX(create->fincr, 0);
+         pEnc->mbParam.fbase = create->fincr <= 0 ? 25 : create->fbase;
+         if (pEnc->mbParam.fincr>0)
+                 simplify_time((int*)&pEnc->mbParam.fincr, (int*)&pEnc->mbParam.fbase);
+         /* zones */
+         if(create->num_zones > 0) {
+                 pEnc->num_zones = create->num_zones;
+                 pEnc->zones = xvid_malloc(sizeof(xvid_enc_zone_t) * pEnc->num_zones, CACHE_LINE);
+                 if (pEnc->zones == NULL)
+                         goto xvid_err_memory0;
+                 memcpy(pEnc->zones, create->zones, sizeof(xvid_enc_zone_t) * pEnc->num_zones);
+         } else {
+                 pEnc->num_zones = 0;
+                 pEnc->zones = NULL;
+         }
+         /* plugins */
+         if(create->num_plugins > 0) {
+                 pEnc->num_plugins = create->num_plugins;
+                 pEnc->plugins = xvid_malloc(sizeof(xvid_enc_plugin_t) * pEnc->num_plugins, CACHE_LINE);
+                 if (pEnc->plugins == NULL)
+                         goto xvid_err_memory0;
+         } else {
+                 pEnc->num_plugins = 0;
+                 pEnc->plugins = NULL;
+         }
+         for (n=0; n<pEnc->num_plugins;n++) {
+                 xvid_plg_create_t pcreate;
+                 xvid_plg_info_t pinfo;
+                 memset(&pinfo, 0, sizeof(xvid_plg_info_t));
+                 pinfo.version = XVID_VERSION;
+                 if (create->plugins[n].func(NULL, XVID_PLG_INFO, &pinfo, NULL) >= 0) {
+                         pEnc->mbParam.plugin_flags |= pinfo.flags;
+                 }
-         pEnc->mbParam.m_quant_type = H263_QUANT;
+                 memset(&pcreate, 0, sizeof(xvid_plg_create_t));
+                 pcreate.version = XVID_VERSION;
+                 pcreate.num_zones = pEnc->num_zones;
+                 pcreate.zones = pEnc->zones;
+                 pcreate.width = pEnc->mbParam.width;
+                 pcreate.height = pEnc->mbParam.height;
+                 pcreate.mb_width = pEnc->mbParam.mb_width;
+                 pcreate.mb_height = pEnc->mbParam.mb_height;
+                 pcreate.fincr = pEnc->mbParam.fincr;
+                 pcreate.fbase = pEnc->mbParam.fbase;
+                 pcreate.param = create->plugins[n].param;
-         pEnc->fMvPrevSigma = -1;
+                 pEnc->plugins[n].func = NULL;   /* disable plugins that fail */
+                 if (create->plugins[n].func(NULL, XVID_PLG_CREATE, &pcreate, &pEnc->plugins[n].param) >= 0) {
+                         pEnc->plugins[n].func = create->plugins[n].func;
+                 }
+         }
+         if ((pEnc->mbParam.global_flags & XVID_GLOBAL_EXTRASTATS_ENABLE) ||
+                 (pEnc->mbParam.plugin_flags & XVID_REQPSNR)) {
+                 pEnc->mbParam.plugin_flags |= XVID_REQORIGINAL; /* psnr calculation requires the original */
+         }
-         /* Fill rate control parameters */
+         /* temp dquants */
+         if ((pEnc->mbParam.plugin_flags & XVID_REQDQUANTS)) {
+                 pEnc->temp_dquants = (int *) xvid_malloc(pEnc->mbParam.mb_width *
+                                                 pEnc->mbParam.mb_height * sizeof(int), CACHE_LINE);
+                 if (pEnc->temp_dquants==NULL)
+                         goto xvid_err_memory1a;
+         }
+         /* temp lambdas */
+         if (pEnc->mbParam.plugin_flags & XVID_REQLAMBDA) {
+                 pEnc->temp_lambda = (float *) xvid_malloc(pEnc->mbParam.mb_width *
+                                                 pEnc->mbParam.mb_height * 6 * sizeof(float), CACHE_LINE);
+                 if (pEnc->temp_lambda == NULL)
+                         goto xvid_err_memory1a;
+         }
-         pEnc->bitrate = pParam->rc_bitrate;
+         /* bframes */
+         pEnc->mbParam.max_bframes = MAX(create->max_bframes, 0);
+         pEnc->mbParam.bquant_ratio = MAX(create->bquant_ratio, 0);
+         pEnc->mbParam.bquant_offset = create->bquant_offset;
-         pEnc->iFrameNum = 0;
+         /* min/max quant */
-         pEnc->mbParam.iMaxKeyInterval = pParam->max_key_interval;
+         for (n=0; n<3; n++) {
+                 pEnc->mbParam.min_quant[n] = create->min_quant[n] > 0 ? create->min_quant[n] : 2;
+                 pEnc->mbParam.max_quant[n] = create->max_quant[n] > 0 ? create->max_quant[n] : 31;
+         }
+         /* frame drop ratio */
+         pEnc->mbParam.frame_drop_ratio = MAX(create->frame_drop_ratio, 0);
-         /* try to allocate frame memory */
+         /* max keyframe interval */
+         pEnc->mbParam.iMaxKeyInterval = create->max_key_interval <= 0 ? (10 * (int)pEnc->mbParam.fbase) / (int)pEnc->mbParam.fincr : create->max_key_interval;
+         /* allocate working frame-image memory */
          pEnc->current = xvid_malloc(sizeof(FRAMEINFO), CACHE_LINE);
          pEnc->reference = xvid_malloc(sizeof(FRAMEINFO), CACHE_LINE);
-Line 240
+Line 274
          if (pEnc->current == NULL || pEnc->reference == NULL)
                  goto xvid_err_memory1;
-         /* try to allocate mb memory */
+         /* allocate macroblock memory */
          pEnc->current->mbs =
                  xvid_malloc(sizeof(MACROBLOCK) * pEnc->mbParam.mb_width *
-Line 252
+Line 286
          if (pEnc->current->mbs == NULL || pEnc->reference->mbs == NULL)
                  goto xvid_err_memory2;
-         /* try to allocate image memory */
+         /* allocate quant matrix memory */
+         pEnc->mbParam.mpeg_quant_matrices =
+                 xvid_malloc(sizeof(uint16_t) * 64 * 8, CACHE_LINE);
- #ifdef _DEBUG_PSNR
+         if (pEnc->mbParam.mpeg_quant_matrices == NULL)
+                 goto xvid_err_memory2a;
+         /* allocate interpolation image memory */
+         if ((pEnc->mbParam.plugin_flags & XVID_REQORIGINAL)) {
          image_null(&pEnc->sOriginal);
- #endif
+                 image_null(&pEnc->sOriginal2);
+         }
          image_null(&pEnc->f_refh);
          image_null(&pEnc->f_refv);
-Line 266
+Line 309
          image_null(&pEnc->reference->image);
          image_null(&pEnc->vInterH);
          image_null(&pEnc->vInterV);
-         image_null(&pEnc->vInterVf);
          image_null(&pEnc->vInterHV);
-         image_null(&pEnc->vInterHVf);
- #ifdef _DEBUG_PSNR
+         if ((pEnc->mbParam.plugin_flags & XVID_REQORIGINAL)) {
          if (image_create
                  (&pEnc->sOriginal, pEnc->mbParam.edged_width,
                   pEnc->mbParam.edged_height) < 0)
                  goto xvid_err_memory3;
- #endif
+                 if (image_create
+                         (&pEnc->sOriginal2, pEnc->mbParam.edged_width,
+                          pEnc->mbParam.edged_height) < 0)
+                         goto xvid_err_memory3;
+         }
          if (image_create
                  (&pEnc->f_refh, pEnc->mbParam.edged_width,
-Line 307
+Line 353
                   pEnc->mbParam.edged_height) < 0)
                  goto xvid_err_memory3;
          if (image_create
-                 (&pEnc->vInterVf, pEnc->mbParam.edged_width,
-                  pEnc->mbParam.edged_height) < 0)
-                 goto xvid_err_memory3;
-         if (image_create
                  (&pEnc->vInterHV, pEnc->mbParam.edged_width,
                   pEnc->mbParam.edged_height) < 0)
                  goto xvid_err_memory3;
-         if (image_create
-                 (&pEnc->vInterHVf, pEnc->mbParam.edged_width,
-                  pEnc->mbParam.edged_height) < 0)
-                 goto xvid_err_memory3;
  /* Create full bitplane for GMC, this might be wasteful */
          if (image_create
-Line 325
+Line 363
                   pEnc->mbParam.edged_height) < 0)
                  goto xvid_err_memory3;
+         /* init bframe image buffers */
-         /* B Frames specific init */
+         pEnc->bframenum_head = 0;
+         pEnc->bframenum_tail = 0;
+         pEnc->flush_bframes = 0;
+         pEnc->closed_bframenum = -1;
-         pEnc->mbParam.global = pParam->global;
+         /* B Frames specific init */
-         pEnc->mbParam.max_bframes = pParam->max_bframes;
-         pEnc->mbParam.bquant_ratio = pParam->bquant_ratio;
-         pEnc->mbParam.bquant_offset = pParam->bquant_offset;
-         pEnc->mbParam.frame_drop_ratio = pParam->frame_drop_ratio;
          pEnc->bframes = NULL;
          if (pEnc->mbParam.max_bframes > 0) {
-                 int n;
                  pEnc->bframes =
                          xvid_malloc(pEnc->mbParam.max_bframes * sizeof(FRAMEINFO *),
-Line 372
+Line 409
                  }
          }
-         pEnc->bframenum_head = 0;
+         /* init incoming frame queue */
-         pEnc->bframenum_tail = 0;
+         pEnc->queue_head = 0;
-         pEnc->flush_bframes = 0;
+         pEnc->queue_tail = 0;
-         pEnc->bframenum_dx50bvop = -1;
+         pEnc->queue_size = 0;
-         pEnc->queue = NULL;
-         if (pEnc->mbParam.max_bframes > 0) {
-                 int n;
                  pEnc->queue =
-                         xvid_malloc(pEnc->mbParam.max_bframes * sizeof(IMAGE),
+                 xvid_malloc((pEnc->mbParam.max_bframes+1) * sizeof(QUEUEINFO),
                                                  CACHE_LINE);
                  if (pEnc->queue == NULL)
                          goto xvid_err_memory4;
-                 for (n = 0; n < pEnc->mbParam.max_bframes; n++)
+         for (n = 0; n < pEnc->mbParam.max_bframes+1; n++)
-                         image_null(&pEnc->queue[n]);
+                 image_null(&pEnc->queue[n].image);
-                 for (n = 0; n < pEnc->mbParam.max_bframes; n++) {
+         for (n = 0; n < pEnc->mbParam.max_bframes+1; n++) {
                          if (image_create
-                                 (&pEnc->queue[n], pEnc->mbParam.edged_width,
+                         (&pEnc->queue[n].image, pEnc->mbParam.edged_width,
                                   pEnc->mbParam.edged_height) < 0)
                                  goto xvid_err_memory5;
-                 }
          }
-         pEnc->queue_head = 0;
+         /* timestamp stuff */
-         pEnc->queue_tail = 0;
-         pEnc->queue_size = 0;
          pEnc->mbParam.m_stamp = 0;
+         pEnc->m_framenum = create->start_frame_num;
-         pEnc->m_framenum = 0;
          pEnc->current->stamp = 0;
          pEnc->reference->stamp = 0;
-         pParam->handle = (void *) pEnc;
+         /* other stuff */
+         pEnc->iFrameNum = 0;
+         pEnc->fMvPrevSigma = -1;
+         /* slices */
+         pEnc->num_slices = MIN(MAX(1, create->num_slices), (int) pEnc->mbParam.mb_height);
+         /* multithreaded stuff */
+         if (create->num_threads > 0) {
+                 int t = MIN(create->num_threads, (int) (pEnc->mbParam.mb_height>>1)); /* at least two rows per thread */
+                 int threads_per_slice = MAX(1, (t / pEnc->num_slices));
+                 int rows_per_thread = (pEnc->mbParam.mb_height + threads_per_slice - 1) / threads_per_slice;
+                 pEnc->num_threads = t;
+                 pEnc->smpData = xvid_malloc(t*sizeof(SMPData), CACHE_LINE);
+                 if (!pEnc->smpData)
+                         goto xvid_err_nosmp;
+                 /* tmp bitstream buffer for slice coding */
+                 pEnc->smpData[0].tmp_buffer = xvid_malloc(16*pEnc->mbParam.edged_width*pEnc->mbParam.mb_height*sizeof(uint8_t), CACHE_LINE);
+                 if (! pEnc->smpData[0].tmp_buffer) goto xvid_err_nosmp;
+                 for (n = 0; n < t; n++) {
+                         int s = MIN(pEnc->num_threads, pEnc->num_slices);
+                         pEnc->smpData[n].complete_count_self =
+                                 xvid_malloc(rows_per_thread * sizeof(int), CACHE_LINE);
+                         if (!pEnc->smpData[n].complete_count_self)
+                                 goto xvid_err_nosmp;
+                         if (n > 0 && n < s) {
+                                 pEnc->smpData[n].bs = (Bitstream *) xvid_malloc(sizeof(Bitstream), CACHE_LINE);
+                                 if (!pEnc->smpData[n].bs)
+                                         goto xvid_err_nosmp;
+                                 pEnc->smpData[n].sStat = (Statistics *) xvid_malloc(sizeof(Statistics), CACHE_LINE);
+                                 if (!pEnc->smpData[n].sStat)
+                                         goto xvid_err_nosmp;
+                                 pEnc->smpData[n].tmp_buffer = pEnc->smpData[0].tmp_buffer + 16*(((n-1)*pEnc->mbParam.edged_width*pEnc->mbParam.mb_height)/s);
+                                 BitstreamInit(pEnc->smpData[n].bs, pEnc->smpData[n].tmp_buffer, 0);
+                         }
+                         if (n != 0)
+                                 pEnc->smpData[n].complete_count_above =
+                                         pEnc->smpData[n-1].complete_count_self;
+                 }
+                 pEnc->smpData[0].complete_count_above =
+                         pEnc->smpData[t-1].complete_count_self - 1;
+         } else {
+   xvid_err_nosmp:
+                 /* no SMP */
+                 if (pEnc->smpData) {
+                         if (pEnc->smpData[0].tmp_buffer)
+                                 xvid_free(pEnc->smpData[0].tmp_buffer);
+                 }
+                 else {
+                         pEnc->smpData = xvid_malloc(1*sizeof(SMPData), CACHE_LINE);
+                         if (pEnc->smpData == NULL)
+                                 goto xvid_err_memory5;
+                 }
-         if (pParam->rc_bitrate) {
+                 create->num_threads = 0;
-                 RateControlInit(&pEnc->rate_control, pParam->rc_bitrate,
-                                                 pParam->rc_reaction_delay_factor,
-                                                 pParam->rc_averaging_period, pParam->rc_buffer,
-                                                 pParam->fbase * 1000 / pParam->fincr,
-                                                 pParam->max_quantizer, pParam->min_quantizer);
          }
+         create->handle = (void *) pEnc;
          init_timer();
+         init_mpeg_matrix(pEnc->mbParam.mpeg_quant_matrices);
-         return XVID_ERR_OK;
+         return 0;   /* ok */
          /*
           * We handle all XVID_ERR_MEMORY here, this makes the code lighter
-Line 432
+Line 520
    xvid_err_memory5:
+         for (n = 0; n < pEnc->mbParam.max_bframes+1; n++) {
-         if (pEnc->mbParam.max_bframes > 0) {
+                         image_destroy(&pEnc->queue[n].image, pEnc->mbParam.edged_width,
-                 for (i = 0; i < pEnc->mbParam.max_bframes; i++) {
-                         image_destroy(&pEnc->queue[i], pEnc->mbParam.edged_width,
                                                    pEnc->mbParam.edged_height);
                  }
                  xvid_free(pEnc->queue);
-         }
    xvid_err_memory4:
          if (pEnc->mbParam.max_bframes > 0) {
+                 int i;
                  for (i = 0; i < pEnc->mbParam.max_bframes; i++) {
-Line 453
+Line 539
                          image_destroy(&pEnc->bframes[i]->image, pEnc->mbParam.edged_width,
                                                    pEnc->mbParam.edged_height);
                          xvid_free(pEnc->bframes[i]->mbs);
                          xvid_free(pEnc->bframes[i]);
                  }
                  xvid_free(pEnc->bframes);
          }
    xvid_err_memory3:
- #ifdef _DEBUG_PSNR
+         if ((pEnc->mbParam.plugin_flags & XVID_REQORIGINAL)) {
          image_destroy(&pEnc->sOriginal, pEnc->mbParam.edged_width,
                                    pEnc->mbParam.edged_height);
- #endif
+                 image_destroy(&pEnc->sOriginal2, pEnc->mbParam.edged_width,
+                                           pEnc->mbParam.edged_height);
+         }
          image_destroy(&pEnc->f_refh, pEnc->mbParam.edged_width,
                                    pEnc->mbParam.edged_height);
-Line 484
+Line 570
                                    pEnc->mbParam.edged_height);
          image_destroy(&pEnc->vInterV, pEnc->mbParam.edged_width,
                                    pEnc->mbParam.edged_height);
-         image_destroy(&pEnc->vInterVf, pEnc->mbParam.edged_width,
-                                   pEnc->mbParam.edged_height);
          image_destroy(&pEnc->vInterHV, pEnc->mbParam.edged_width,
                                    pEnc->mbParam.edged_height);
-         image_destroy(&pEnc->vInterHVf, pEnc->mbParam.edged_width,
-                                   pEnc->mbParam.edged_height);
  /* destroy GMC image */
          image_destroy(&pEnc->vGMC, pEnc->mbParam.edged_width,
                                    pEnc->mbParam.edged_height);
+   xvid_err_memory2a:
+         xvid_free(pEnc->mbParam.mpeg_quant_matrices);
    xvid_err_memory2:
          xvid_free(pEnc->current->mbs);
-Line 503
+Line 587
    xvid_err_memory1:
          xvid_free(pEnc->current);
          xvid_free(pEnc->reference);
+   xvid_err_memory1a:
+         if ((pEnc->mbParam.plugin_flags & XVID_REQDQUANTS)) {
+                 xvid_free(pEnc->temp_dquants);
+         }
+         if(pEnc->mbParam.plugin_flags & XVID_REQLAMBDA) {
+                 xvid_free(pEnc->temp_lambda);
+         }
+   xvid_err_memory0:
+         for (n=0; n<pEnc->num_plugins;n++) {
+                 if (pEnc->plugins[n].func) {
+                         pEnc->plugins[n].func(pEnc->plugins[n].param, XVID_PLG_DESTROY, NULL, NULL);
+                 }
+         }
+         xvid_free(pEnc->plugins);
+         xvid_free(pEnc->zones);
          xvid_free(pEnc);
-         pParam->handle = NULL;
+         create->handle = NULL;
          return XVID_ERR_MEMORY;
  }
-Line 514
+Line 618
   * Encoder destruction
   *
   * This function destroy the entire encoder structure created by a previous
-  * successful encoder_create call.
+  * successful enc_create call.
   *
   * Returned values (for now only one returned value) :
-  *    - XVID_ERR_OK     - no errors
+  *      - 0      - no errors
   *
   ****************************************************************************/
  int
- encoder_destroy(Encoder * pEnc)
+ enc_destroy(Encoder * pEnc)
  {
          int i;
-         ENC_CHECK(pEnc);
          /* B Frames specific */
-         if (pEnc->mbParam.max_bframes > 0) {
+         for (i = 0; i < pEnc->mbParam.max_bframes+1; i++) {
+                 image_destroy(&pEnc->queue[i].image, pEnc->mbParam.edged_width,
-                 for (i = 0; i < pEnc->mbParam.max_bframes; i++) {
-                         image_destroy(&pEnc->queue[i], pEnc->mbParam.edged_width,
                                            pEnc->mbParam.edged_height);
                  }
-                 xvid_free(pEnc->queue);
-         }
+         xvid_free(pEnc->queue);
          if (pEnc->mbParam.max_bframes > 0) {
-Line 549
+Line 647
                          image_destroy(&pEnc->bframes[i]->image, pEnc->mbParam.edged_width,
                                            pEnc->mbParam.edged_height);
                          xvid_free(pEnc->bframes[i]->mbs);
                          xvid_free(pEnc->bframes[i]);
                  }
-Line 569
+Line 665
                                    pEnc->mbParam.edged_height);
          image_destroy(&pEnc->vInterV, pEnc->mbParam.edged_width,
                                    pEnc->mbParam.edged_height);
-         image_destroy(&pEnc->vInterVf, pEnc->mbParam.edged_width,
-                                   pEnc->mbParam.edged_height);
          image_destroy(&pEnc->vInterHV, pEnc->mbParam.edged_width,
                                    pEnc->mbParam.edged_height);
-         image_destroy(&pEnc->vInterHVf, pEnc->mbParam.edged_width,
-                                   pEnc->mbParam.edged_height);
          image_destroy(&pEnc->f_refh, pEnc->mbParam.edged_width,
                                    pEnc->mbParam.edged_height);
          image_destroy(&pEnc->f_refv, pEnc->mbParam.edged_width,
                                    pEnc->mbParam.edged_height);
          image_destroy(&pEnc->f_refhv, pEnc->mbParam.edged_width,
                                    pEnc->mbParam.edged_height);
+         image_destroy(&pEnc->vGMC, pEnc->mbParam.edged_width,
+                                   pEnc->mbParam.edged_height);
- #ifdef _DEBUG_PSNR
+         if ((pEnc->mbParam.plugin_flags & XVID_REQORIGINAL)) {
          image_destroy(&pEnc->sOriginal, pEnc->mbParam.edged_width,
                                    pEnc->mbParam.edged_height);
- #endif
+                 image_destroy(&pEnc->sOriginal2, pEnc->mbParam.edged_width,
+                                           pEnc->mbParam.edged_height);
+         }
          /* Encoder structure */
-Line 596
+Line 691
          xvid_free(pEnc->reference->mbs);
          xvid_free(pEnc->reference);
-         xvid_free(pEnc);
+         if ((pEnc->mbParam.plugin_flags & XVID_REQDQUANTS)) {
+                 xvid_free(pEnc->temp_dquants);
+         }
-         return XVID_ERR_OK;
+         if ((pEnc->mbParam.plugin_flags & XVID_REQLAMBDA)) {
+                 xvid_free(pEnc->temp_lambda);
  }
+         if (pEnc->num_plugins>0) {
+                 xvid_plg_destroy_t pdestroy;
+                 memset(&pdestroy, 0, sizeof(xvid_plg_destroy_t));
- static __inline void inc_frame_num(Encoder * pEnc)
+                 pdestroy.version = XVID_VERSION;
- {
+                 pdestroy.num_frames = pEnc->m_framenum;
-         pEnc->current->stamp = pEnc->mbParam.m_stamp;   // first frame is zero
-         pEnc->mbParam.m_stamp += pEnc->mbParam.fincr;
+                 for (i=0; i<pEnc->num_plugins;i++) {
+                         if (pEnc->plugins[i].func) {
+                                 pEnc->plugins[i].func(pEnc->plugins[i].param, XVID_PLG_DESTROY, &pdestroy, NULL);
+                         }
+                 }
+                 xvid_free(pEnc->plugins);
  }
+         xvid_free(pEnc->mbParam.mpeg_quant_matrices);
- static __inline void
+         if (pEnc->num_zones > 0)
- queue_image(Encoder * pEnc, XVID_ENC_FRAME * pFrame)
+                 xvid_free(pEnc->zones);
- {
-         if (pEnc->queue_size >= pEnc->mbParam.max_bframes)
-         {
-                 DPRINTF(DPRINTF_DEBUG,"FATAL: QUEUE FULL");
-                 return;
-         }
-         DPRINTF(DPRINTF_DEBUG,"*** QUEUE bf: head=%i tail=%i   queue: head=%i tail=%i size=%i",
+         if (pEnc->num_threads > 0) {
-                                 pEnc->bframenum_head, pEnc->bframenum_tail,
+                 for (i = 1; i < MAX(1, MIN(pEnc->num_threads, pEnc->num_slices)); i++) {
-                                 pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);
+                         xvid_free(pEnc->smpData[i].bs);
+                         xvid_free(pEnc->smpData[i].sStat);
+                 }
+                 if (pEnc->smpData[0].tmp_buffer) xvid_free(pEnc->smpData[0].tmp_buffer);
+                 for (i = 0; i < pEnc->num_threads; i++)
+                         xvid_free(pEnc->smpData[i].complete_count_self);
+         }
+         xvid_free(pEnc->smpData);
-         start_timer();
+         xvid_free(pEnc);
-         if (image_input
-                 (&pEnc->queue[pEnc->queue_tail], pEnc->mbParam.width, pEnc->mbParam.height,
-                  pEnc->mbParam.edged_width, pFrame->image, pFrame->stride, pFrame->colorspace, pFrame->general & XVID_INTERLACING))
-                 return;
-         stop_conv_timer();
-         pEnc->queue_size++;
+         return 0;  /* ok */
-         pEnc->queue_tail =  (pEnc->queue_tail + 1) % pEnc->mbParam.max_bframes;
  }
- static __inline void
- set_timecodes(FRAMEINFO* pCur,FRAMEINFO *pRef, int32_t time_base)
- {
-                 pCur->ticks = (int32_t)pCur->stamp % time_base;
+ /*
-                 pCur->seconds =  ((int32_t)pCur->stamp / time_base)     - ((int32_t)pRef->stamp / time_base) ;
+   call the plugins
+   */
-                 //HEAVY DEBUG OUTPUT    remove when timecodes prove to be stable
+ static void call_plugins(Encoder * pEnc, FRAMEINFO * frame, IMAGE * original,
+                                                  int opt, int * type, int * quant, xvid_enc_stats_t * stats)
+ {
+         unsigned int i, j, k;
+         xvid_plg_data_t data;
- /*              fprintf(stderr,"WriteVop:   %d - %d \n",
+         /* set data struct */
-                         ((int32_t)pCur->stamp / time_base), ((int32_t)pRef->stamp / time_base));
-                 fprintf(stderr,"set_timecodes: VOP %1d   stamp=%lld ref_stamp=%lld  base=%d\n",
-                         pCur->coding_type, pCur->stamp, pRef->stamp, time_base);
-                 fprintf(stderr,"set_timecodes: VOP %1d   seconds=%d   ticks=%d   (ref-sec=%d  ref-tick=%d)\n",
-                         pCur->coding_type, pCur->seconds, pCur->ticks, pRef->seconds, pRef->ticks);
- */
+         memset(&data, 0, sizeof(xvid_plg_data_t));
+         data.version = XVID_VERSION;
+         /* find zone */
+         for(i=0; i<pEnc->num_zones && pEnc->zones[i].frame<=frame->frame_num; i++) ;
+         data.zone = i>0 ? &pEnc->zones[i-1] : NULL;
+         data.width = pEnc->mbParam.width;
+         data.height = pEnc->mbParam.height;
+         data.mb_width = pEnc->mbParam.mb_width;
+         data.mb_height = pEnc->mbParam.mb_height;
+         data.fincr = frame->fincr;
+         data.fbase = pEnc->mbParam.fbase;
+         data.bquant_ratio = pEnc->mbParam.bquant_ratio;
+         data.bquant_offset = pEnc->mbParam.bquant_offset;
+         for (i=0; i<3; i++) {
+                 data.min_quant[i] = pEnc->mbParam.min_quant[i];
+                 data.max_quant[i] = pEnc->mbParam.max_quant[i];
+         }
+         data.reference.csp = XVID_CSP_PLANAR;
+         data.reference.plane[0] = pEnc->reference->image.y;
+         data.reference.plane[1] = pEnc->reference->image.u;
+         data.reference.plane[2] = pEnc->reference->image.v;
+         data.reference.stride[0] = pEnc->mbParam.edged_width;
+         data.reference.stride[1] = pEnc->mbParam.edged_width/2;
+         data.reference.stride[2] = pEnc->mbParam.edged_width/2;
+         data.current.csp = XVID_CSP_PLANAR;
+         data.current.plane[0] = frame->image.y;
+         data.current.plane[1] = frame->image.u;
+         data.current.plane[2] = frame->image.v;
+         data.current.stride[0] = pEnc->mbParam.edged_width;
+         data.current.stride[1] = pEnc->mbParam.edged_width/2;
+         data.current.stride[2] = pEnc->mbParam.edged_width/2;
+         data.frame_num = frame->frame_num;
+         if (opt == XVID_PLG_BEFORE) {
+                 data.type = *type;
+                 data.quant = *quant;
+                 data.vol_flags = frame->vol_flags;
+                 data.vop_flags = frame->vop_flags;
+                 data.motion_flags = frame->motion_flags;
+         } else if (opt == XVID_PLG_FRAME) {
+                 data.type = coding2type(frame->coding_type);
+                 data.quant = frame->quant;
+                 if ((pEnc->mbParam.plugin_flags & XVID_REQDQUANTS)) {
+                         data.dquant = pEnc->temp_dquants;
+                         data.dquant_stride = pEnc->mbParam.mb_width;
+                         memset(data.dquant, 0, data.mb_width*data.mb_height*sizeof(int));
  }
+                 if(pEnc->mbParam.plugin_flags & XVID_REQLAMBDA) {
+                         int block = 0;
+                         emms();
+                         data.lambda = pEnc->temp_lambda;
+                         for(i = 0;i < pEnc->mbParam.mb_height; i++)
+                                 for(j = 0;j < pEnc->mbParam.mb_width; j++)
+                                         for (k = 0; k < 6; k++)
+                                                 data.lambda[block++] = 1.0f;
+                 }
+         } else { /* XVID_PLG_AFTER */
+                 if ((pEnc->mbParam.plugin_flags & XVID_REQORIGINAL)) {
+                         data.original.csp = XVID_CSP_PLANAR;
+                         data.original.plane[0] = original->y;
+                         data.original.plane[1] = original->u;
+                         data.original.plane[2] = original->v;
+                         data.original.stride[0] = pEnc->mbParam.edged_width;
+                         data.original.stride[1] = pEnc->mbParam.edged_width/2;
+                         data.original.stride[2] = pEnc->mbParam.edged_width/2;
+                 }
+                 if ((frame->vol_flags & XVID_VOL_EXTRASTATS) ||
+                         (pEnc->mbParam.plugin_flags & XVID_REQPSNR)) {
- /* convert pFrame->intra to coding_type */
+                         data.sse_y =
- static int intra2coding_type(int intra)
+                                 plane_sse( original->y, frame->image.y,
- {
+                                                    pEnc->mbParam.edged_width, pEnc->mbParam.width,
-         if (intra < 0)          return -1;
+                                                    pEnc->mbParam.height);
-         if (intra == 1)         return I_VOP;
-         if (intra == 2)         return B_VOP;
-         return P_VOP;
+                         data.sse_u =
+                                 plane_sse( original->u, frame->image.u,
+                                                    pEnc->mbParam.edged_width/2, pEnc->mbParam.width/2,
+                                                    pEnc->mbParam.height/2);
+                         data.sse_v =
+                                 plane_sse( original->v, frame->image.v,
+                                                    pEnc->mbParam.edged_width/2, pEnc->mbParam.width/2,
+                                                    pEnc->mbParam.height/2);
+                 }
+                 data.type = coding2type(frame->coding_type);
+                 data.quant = frame->quant;
+                 if ((pEnc->mbParam.plugin_flags & XVID_REQDQUANTS)) {
+                         data.dquant = pEnc->temp_dquants;
+                         data.dquant_stride = pEnc->mbParam.mb_width;
+                         for (j=0; j<pEnc->mbParam.mb_height; j++)
+                         for (i=0; i<pEnc->mbParam.mb_width; i++) {
+                                 data.dquant[j*data.dquant_stride + i] = frame->mbs[j*pEnc->mbParam.mb_width + i].dquant;
+                         }
  }
+                 data.vol_flags = frame->vol_flags;
+                 data.vop_flags = frame->vop_flags;
+                 data.motion_flags = frame->motion_flags;
+                 data.length = frame->length;
+                 data.kblks = frame->sStat.kblks;
+                 data.mblks = frame->sStat.mblks;
+                 data.ublks = frame->sStat.ublks;
+                 /* New code */
+                 data.stats.type      = coding2type(frame->coding_type);
+                 data.stats.quant     = frame->quant;
+                 data.stats.vol_flags = frame->vol_flags;
+                 data.stats.vop_flags = frame->vop_flags;
+                 data.stats.length    = frame->length;
+                 data.stats.hlength   = frame->length - (frame->sStat.iTextBits / 8);
+                 data.stats.kblks     = frame->sStat.kblks;
+                 data.stats.mblks     = frame->sStat.mblks;
+                 data.stats.ublks     = frame->sStat.ublks;
+                 data.stats.sse_y     = data.sse_y;
+                 data.stats.sse_u     = data.sse_u;
+                 data.stats.sse_v     = data.sse_v;
+                 if (stats)
+                         *stats = data.stats;
+         }
- /*****************************************************************************
+         /* call plugins */
-  * IPB frame encoder entry point
+         for (i=0; i<(unsigned int)pEnc->num_plugins;i++) {
-  *
+                 emms();
-  * Returned values :
+                 if (pEnc->plugins[i].func) {
-  *    - XVID_ERR_OK     - no errors
+                         if (pEnc->plugins[i].func(pEnc->plugins[i].param, opt, &data, NULL) < 0) {
-  *    - XVID_ERR_FORMAT - the image subsystem reported the image had a wrong
+                                 continue;
-  *                        format
+                         }
-  ****************************************************************************/
+                 }
+         }
- int
+         emms();
- encoder_encode_bframes(Encoder * pEnc,
-                            XVID_ENC_FRAME * pFrame,
-                            XVID_ENC_STATS * pResult)
- {
-         uint16_t x, y;
-         Bitstream bs;
-         uint32_t bits;
-         int mode;
-         int input_valid = 1;
+         /* copy modified values back into frame*/
-         int bframes_count = 0;
+         if (opt == XVID_PLG_BEFORE) {
+                 *type = data.type;
+                 *quant = data.quant > 0 ? data.quant : 2;   /* default */
+                 frame->vol_flags = data.vol_flags;
+                 frame->vop_flags = data.vop_flags;
+                 frame->motion_flags = data.motion_flags;
+         } else if (opt == XVID_PLG_FRAME) {
+                 if ((pEnc->mbParam.plugin_flags & XVID_REQDQUANTS)) {
+                         for (j=0; j<pEnc->mbParam.mb_height; j++)
+                         for (i=0; i<pEnc->mbParam.mb_width; i++) {
+                                 frame->mbs[j*pEnc->mbParam.mb_width + i].dquant = data.dquant[j*data.mb_width + i];
+                         }
+                 } else {
+                         for (j=0; j<pEnc->mbParam.mb_height; j++)
+                         for (i=0; i<pEnc->mbParam.mb_width; i++) {
+                                 frame->mbs[j*pEnc->mbParam.mb_width + i].dquant = 0;
+                         }
+                 }
- #ifdef _DEBUG_PSNR
+                 if (pEnc->mbParam.plugin_flags & XVID_REQLAMBDA) {
-         float psnr;
+                         for (j = 0; j < pEnc->mbParam.mb_height; j++)
-         char temp[128];
+                                 for (i = 0; i < pEnc->mbParam.mb_width; i++)
- #endif
+                                         for (k = 0; k < 6; k++) {
+                                                 frame->mbs[j*pEnc->mbParam.mb_width + i].lambda[k] =
+                                                         (int) ((float)(1<<LAMBDA_EXP) * data.lambda[6 * (j * data.mb_width + i) + k]);
+                                         }
+                 } else {
+                         for (j = 0; j<pEnc->mbParam.mb_height; j++)
+                                 for (i = 0; i<pEnc->mbParam.mb_width; i++)
+                                         for (k = 0; k < 6; k++) {
+                                                 frame->mbs[j*pEnc->mbParam.mb_width + i].lambda[k] = 1<<LAMBDA_EXP;
+                                         }
+                 }
-         ENC_CHECK(pEnc);
-         ENC_CHECK(pFrame);
-         ENC_CHECK(pFrame->image);
-         start_global_timer();
+                 frame->mbs[0].quant = data.quant; /* FRAME will not affect the quant in stats */
+         }
-         BitstreamInit(&bs, pFrame->bitstream, 0);
- ipvop_loop:
+ }
-         /*
-          * bframe "flush" code
-          */
-         if ((pFrame->image == NULL || pEnc->flush_bframes)
+ static __inline void inc_frame_num(Encoder * pEnc)
-                 && (pEnc->bframenum_head < pEnc->bframenum_tail)) {
+ {
+         pEnc->current->frame_num = pEnc->m_framenum;
+         pEnc->current->stamp = pEnc->mbParam.m_stamp;   /* first frame is zero */
-                 if (pEnc->flush_bframes == 0) {
+         pEnc->mbParam.m_stamp += pEnc->current->fincr;
-                         /*
+         pEnc->m_framenum++;     /* debug ticker */
-                          * we have reached the end of stream without getting
+ }
-                          * a future reference frame... so encode last final
-                          * frame as a pframe
-                          */
-                         DPRINTF(DPRINTF_DEBUG,"*** BFRAME (final frame) bf: head=%i tail=%i   queue: head=%i tail=%i size=%i",
+ static __inline void dec_frame_num(Encoder * pEnc)
-                                 pEnc->bframenum_head, pEnc->bframenum_tail,
+ {
-                                 pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);
+         pEnc->mbParam.m_stamp -= pEnc->mbParam.fincr;
+         pEnc->m_framenum--;     /* debug ticker */
+ }
-                         pEnc->bframenum_tail--;
+ static __inline void
-                         SWAP(FRAMEINFO *, pEnc->current, pEnc->reference);
+ MBSetDquant(MACROBLOCK * pMB, int x, int y, MBParam * mbParam)
+ {
+         if (pMB->cbp == 0) {
+                 /* we want to code dquant but the quantizer value will not be used yet
+                         let's find out if we can postpone dquant to next MB
+                 */
+                 if (x == mbParam->mb_width-1 && y == mbParam->mb_height-1) {
+                         pMB->dquant = 0; /* it's the last MB of all, the easiest case */
+                         return;
+                 } else {
+                         MACROBLOCK * next = pMB + 1;
+                         const MACROBLOCK * prev = pMB - 1;
+                         if (next->mode != MODE_INTER4V && next->mode != MODE_NOT_CODED)
+                                 /* mode allows dquant change in the future */
+                                 if (abs(next->quant - prev->quant) <= 2) {
+                                         /* quant change is not out of range */
+                                         pMB->quant = prev->quant;
+                                         pMB->dquant = 0;
+                                         next->dquant = next->quant - prev->quant;
+                                         return;
+                                 }
+                 }
+         }
+         /* couldn't skip this dquant */
+         pMB->mode = MODE_INTER_Q;
+ }
-                         SWAP(FRAMEINFO *, pEnc->current, pEnc->bframes[pEnc->bframenum_tail]);
-                         FrameCodeP(pEnc, &bs, &bits, 1, 0);
-                         bframes_count = 0;
-                         BitstreamPadAlways(&bs);
+ static __inline void
-                         pFrame->length = BitstreamLength(&bs);
+ set_timecodes(FRAMEINFO* pCur,FRAMEINFO *pRef, int32_t time_base)
-                         pFrame->intra = 0;
+ {
-                         emms();
+         pCur->ticks = (int32_t)pCur->stamp % time_base;
+         pCur->seconds =  ((int32_t)pCur->stamp / time_base)     - ((int32_t)pRef->stamp / time_base) ;
-                         return XVID_ERR_OK;
+ #if 0   /* HEAVY DEBUG OUTPUT */
+         fprintf(stderr,"WriteVop:   %d - %d \n",
+                         ((int32_t)pCur->stamp / time_base), ((int32_t)pRef->stamp / time_base));
+         fprintf(stderr,"set_timecodes: VOP %1d   stamp=%lld ref_stamp=%lld  base=%d\n",
+                         pCur->coding_type, pCur->stamp, pRef->stamp, time_base);
+         fprintf(stderr,"set_timecodes: VOP %1d   seconds=%d   ticks=%d   (ref-sec=%d  ref-tick=%d)\n",
+                         pCur->coding_type, pCur->seconds, pCur->ticks, pRef->seconds, pRef->ticks);
+ #endif
                  }
+ static void
+ simplify_par(int *par_width, int *par_height)
+ {
-                 DPRINTF(DPRINTF_DEBUG,"*** BFRAME (flush) bf: head=%i tail=%i   queue: head=%i tail=%i size=%i",
+         int _par_width  = (!*par_width)  ? 1 : (*par_width<0)  ? -*par_width:  *par_width;
-                                 pEnc->bframenum_head, pEnc->bframenum_tail,
+         int _par_height = (!*par_height) ? 1 : (*par_height<0) ? -*par_height: *par_height;
-                                 pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);
+         int divisor = gcd(_par_width, _par_height);
-                 FrameCodeB(pEnc, pEnc->bframes[pEnc->bframenum_head], &bs, &bits);
-                 pEnc->bframenum_head++;
-                 BitstreamPadAlways(&bs);
-                 pFrame->length = BitstreamLength(&bs);
-                 pFrame->intra = 2;
-                 if (input_valid)
+         _par_width  /= divisor;
-                         queue_image(pEnc, pFrame);
+         _par_height /= divisor;
+         /* 2^8 precision maximum */
+         if (_par_width>255 || _par_height>255) {
+                 float div;
                  emms();
+                 if (_par_width>_par_height)
+                         div = (float)_par_width/255;
+                 else
+                         div = (float)_par_height/255;
-                 return XVID_ERR_OK;
+                 _par_width  = (int)((float)_par_width/div);
+                 _par_height = (int)((float)_par_height/div);
          }
-         if (pEnc->bframenum_head > 0) {
+         *par_width = _par_width;
-                 pEnc->bframenum_head = pEnc->bframenum_tail = 0;
+         *par_height = _par_height;
-                 /* write an empty marker to the bitstream.
-                    for divx5 decoder compatibility, this marker must consist
-                    of a not-coded p-vop, with a time_base of zero, and time_increment
-                    indentical to the future-referece frame.
-                 */
-                 if ((pEnc->mbParam.global & XVID_GLOBAL_PACKED)) {
-                         int tmp;
-                         DPRINTF(DPRINTF_DEBUG,"*** EMPTY bf: head=%i tail=%i   queue: head=%i tail=%i size=%i",
-                                 pEnc->bframenum_head, pEnc->bframenum_tail,
-                                 pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);
-                         tmp = pEnc->current->seconds;
-                         pEnc->current->seconds = 0; /* force time_base = 0 */
-                         BitstreamWriteVopHeader(&bs, &pEnc->mbParam, pEnc->current, 0);
-                         pEnc->current->seconds = tmp;
-                         BitstreamPadAlways(&bs);
-                         pFrame->length = BitstreamLength(&bs);
-                         pFrame->intra = 4;
-                         if (input_valid)
+         return;
-                                 queue_image(pEnc, pFrame);
-                         emms();
-                         return XVID_ERR_OK;
-                 }
          }
+ /*****************************************************************************
+  * IPB frame encoder entry point
+  *
+  * Returned values :
+  *      - >0                       - output bytes
+  *      - 0                             - no output
+  *      - XVID_ERR_VERSION - wrong version passed to core
+  *      - XVID_ERR_END   - End of stream reached before end of coding
+  *      - XVID_ERR_FORMAT  - the image subsystem reported the image had a wrong
+  *                                               format
+  ****************************************************************************/
- bvop_loop:
-         if (pEnc->bframenum_dx50bvop != -1)
+ int
+ enc_encode(Encoder * pEnc,
+                            xvid_enc_frame_t * xFrame,
+                            xvid_enc_stats_t * stats)
          {
+         xvid_enc_frame_t * frame;
+         int type;
+         Bitstream bs;
-                 SWAP(FRAMEINFO *, pEnc->current, pEnc->reference);
+         if (XVID_VERSION_MAJOR(xFrame->version) != 1 || (stats && XVID_VERSION_MAJOR(stats->version) != 1))     /* v1.x.x */
-                 SWAP(FRAMEINFO *, pEnc->current, pEnc->bframes[pEnc->bframenum_dx50bvop]);
+                 return XVID_ERR_VERSION;
-                 if ((pEnc->mbParam.global & XVID_GLOBAL_DEBUG)) {
+         xFrame->out_flags = 0;
-                         image_printf(&pEnc->current->image, pEnc->mbParam.edged_width, pEnc->mbParam.height, 5, 100, "DX50 IVOP");
-                 }
-                 if (input_valid)
+         start_global_timer();
-                 {
+         BitstreamInit(&bs, xFrame->bitstream, 0);
-                         queue_image(pEnc, pFrame);
-                         input_valid = 0;
-                 }
-         } else if (input_valid) {
-                 SWAP(FRAMEINFO *, pEnc->current, pEnc->reference);
+         /* %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+          * enqueue image to the encoding-queue
+          * %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% */
+         if (xFrame->input.csp != XVID_CSP_NULL)
+         {
+                 QUEUEINFO * q = &pEnc->queue[pEnc->queue_tail];
                  start_timer();
                  if (image_input
-                         (&pEnc->current->image, pEnc->mbParam.width, pEnc->mbParam.height,
+                         (&q->image, pEnc->mbParam.width, pEnc->mbParam.height,
-                         pEnc->mbParam.edged_width, pFrame->image, pFrame->stride, pFrame->colorspace, pFrame->general & XVID_INTERLACING))
+                         pEnc->mbParam.edged_width, (uint8_t**)xFrame->input.plane, xFrame->input.stride,
+                         xFrame->input.csp, xFrame->vol_flags & XVID_VOL_INTERLACING))
                  {
                          emms();
                          return XVID_ERR_FORMAT;
                  }
                  stop_conv_timer();
-                 // queue input frame, and dequue next image
+                 if ((xFrame->vop_flags & XVID_VOP_CHROMAOPT)) {
-                 if (pEnc->queue_size > 0)
+                         image_chroma_optimize(&q->image,
-                 {
+                                 pEnc->mbParam.width, pEnc->mbParam.height, pEnc->mbParam.edged_width);
-                         image_swap(&pEnc->current->image, &pEnc->queue[pEnc->queue_tail]);
-                         if (pEnc->queue_head != pEnc->queue_tail)
-                         {
-                                 image_swap(&pEnc->current->image, &pEnc->queue[pEnc->queue_head]);
                          }
-                         pEnc->queue_head =  (pEnc->queue_head + 1) % pEnc->mbParam.max_bframes;
-                         pEnc->queue_tail =  (pEnc->queue_tail + 1) % pEnc->mbParam.max_bframes;
+                 q->frame = *xFrame;
+                 if (xFrame->quant_intra_matrix)
+                 {
+                         memcpy(q->quant_intra_matrix, xFrame->quant_intra_matrix, 64*sizeof(unsigned char));
+                         q->frame.quant_intra_matrix = q->quant_intra_matrix;
                  }
-         } else if (pEnc->queue_size > 0) {
+                 if (xFrame->quant_inter_matrix)
+                 {
+                         memcpy(q->quant_inter_matrix, xFrame->quant_inter_matrix, 64*sizeof(unsigned char));
+                         q->frame.quant_inter_matrix = q->quant_inter_matrix;
+                 }
-                 SWAP(FRAMEINFO *, pEnc->current, pEnc->reference);
+                 pEnc->queue_tail = (pEnc->queue_tail + 1) % (pEnc->mbParam.max_bframes+1);
+                 pEnc->queue_size++;
+         }
-                 image_swap(&pEnc->current->image, &pEnc->queue[pEnc->queue_head]);
-                 pEnc->queue_head =  (pEnc->queue_head + 1) % pEnc->mbParam.max_bframes;
-                 pEnc->queue_size--;
-         } else {
+         /* %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+          * bframe flush code
+          * %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% */
-                 /* if nothing was encoded, write an 'ignore this frame' flag
+ repeat:
-                    to the bitstream */
-                 if (BitstreamPos(&bs) == 0) {
+         if (pEnc->flush_bframes)
+         {
+                 if (pEnc->bframenum_head < pEnc->bframenum_tail) {
-                         DPRINTF(DPRINTF_DEBUG,"*** SKIP bf: head=%i tail=%i   queue: head=%i tail=%i size=%i",
+                         DPRINTF(XVID_DEBUG_DEBUG,"*** BFRAME (flush) bf: head=%i tail=%i   queue: head=%i tail=%i size=%i\n",
                                  pEnc->bframenum_head, pEnc->bframenum_tail,
                                  pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);
-                 //      BitstreamPutBits(&bs, 0x7f, 8);
+                         if ((pEnc->mbParam.plugin_flags & XVID_REQORIGINAL)) {
-                         pFrame->intra = 5;
+                                 image_copy(&pEnc->sOriginal2, &pEnc->bframes[pEnc->bframenum_head]->image,
+                                                    pEnc->mbParam.edged_width, pEnc->mbParam.height);
                  }
-                 pFrame->length = BitstreamLength(&bs);
+                         FrameCodeB(pEnc, pEnc->bframes[pEnc->bframenum_head], &bs);
-                 emms();
+                         call_plugins(pEnc, pEnc->bframes[pEnc->bframenum_head], &pEnc->sOriginal2, XVID_PLG_AFTER, NULL, NULL, stats);
-                 return XVID_ERR_OK;
+                         pEnc->bframenum_head++;
+                         goto done;
          }
-         pEnc->flush_bframes = 0;
+                 /* write an empty marker to the bitstream.
-         emms();
+                    for divx5 decoder compatibility, this marker must consist
+                    of a not-coded p-vop, with a time_base of zero, and time_increment
+                    indentical to the future-referece frame.
+                 */
-         // only inc frame num, adapt quant, etc. if we havent seen it before
+                 if ((pEnc->mbParam.global_flags & XVID_GLOBAL_PACKED && pEnc->bframenum_tail > 0)) {
-         if (pEnc->bframenum_dx50bvop < 0 )
+                         int tmp;
-         {
+                         int bits;
-                 mode = intra2coding_type(pFrame->intra);
-                 if (pFrame->quant == 0)
-                         pEnc->current->quant = RateControlGetQ(&pEnc->rate_control, 0);
-                 else
-                         pEnc->current->quant = pFrame->quant;
- /*              if (pEnc->current->quant < 1)
+                         DPRINTF(XVID_DEBUG_DEBUG,"*** EMPTY bf: head=%i tail=%i   queue: head=%i tail=%i size=%i\n",
-                         pEnc->current->quant = 1;
+                                 pEnc->bframenum_head, pEnc->bframenum_tail,
+                                 pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);
-                 if (pEnc->current->quant > 31)
+                         bits = BitstreamPos(&bs);
-                         pEnc->current->quant = 31;
- */
-                 pEnc->current->global_flags = pFrame->general;
-                 pEnc->current->motion_flags = pFrame->motion;
-                 /* ToDo : dynamic fcode (in both directions) */
+                         tmp = pEnc->current->seconds;
-                 pEnc->current->fcode = pEnc->mbParam.m_fcode;
+                         pEnc->current->seconds = 0; /* force time_base = 0 */
-                 pEnc->current->bcode = pEnc->mbParam.m_fcode;
-                 inc_frame_num(pEnc);
+                         BitstreamWriteVopHeader(&bs, &pEnc->mbParam, pEnc->current, 0, pEnc->current->quant);
+                         BitstreamPad(&bs);
+                         pEnc->current->seconds = tmp;
- #ifdef _DEBUG_PSNR
+                         /* add the not-coded length to the reference frame size */
-                 image_copy(&pEnc->sOriginal, &pEnc->current->image,
+                         pEnc->current->length += (BitstreamPos(&bs) - bits) / 8;
-                            pEnc->mbParam.edged_width, pEnc->mbParam.height);
+                         call_plugins(pEnc, pEnc->current, &pEnc->sOriginal, XVID_PLG_AFTER, NULL, NULL, stats);
- #endif
-                 emms();
+                         /* flush complete: reset counters */
+                         pEnc->flush_bframes = 0;
+                         pEnc->bframenum_head = pEnc->bframenum_tail = 0;
+                         goto done;
-                 if ((pEnc->mbParam.global & XVID_GLOBAL_DEBUG)) {
+                 }
-                         image_printf(&pEnc->current->image, pEnc->mbParam.edged_width, pEnc->mbParam.height, 5, 5,
-                                 "%i  if:%i  st:%i", pEnc->m_framenum++, pEnc->iFrameNum, pEnc->current->stamp);
+                 /* flush complete: reset counters */
+                 pEnc->flush_bframes = 0;
+                 pEnc->bframenum_head = pEnc->bframenum_tail = 0;
                  }
          /* %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
-          * Luminance masking
+          * dequeue frame from the encoding queue
           * %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% */
-                 if ((pEnc->current->global_flags & XVID_LUMIMASKING)) {
+         if (pEnc->queue_size == 0)              /* empty */
-                         int *temp_dquants =
+         {
-                                 (int *) xvid_malloc(pEnc->mbParam.mb_width *
+                 if (xFrame->input.csp == XVID_CSP_NULL) /* no futher input */
-                                                                 pEnc->mbParam.mb_height * sizeof(int),
+                 {
-                                                                 CACHE_LINE);
-                         pEnc->current->quant =
+                         DPRINTF(XVID_DEBUG_DEBUG,"*** FINISH bf: head=%i tail=%i   queue: head=%i tail=%i size=%i\n",
-                                 adaptive_quantization(pEnc->current->image.y,
+                                 pEnc->bframenum_head, pEnc->bframenum_tail,
-                                                                   pEnc->mbParam.edged_width, temp_dquants,
+                                 pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);
-                                                                   pEnc->current->quant, pEnc->current->quant,
-* pEnc->current->quant,
-                                                                   pEnc->mbParam.mb_width,
-                                                                   pEnc->mbParam.mb_height);
-                         for (y = 0; y < pEnc->mbParam.mb_height; y++) {
+                         if (!(pEnc->mbParam.global_flags & XVID_GLOBAL_PACKED) && pEnc->mbParam.max_bframes > 0) {
+                                 call_plugins(pEnc, pEnc->current, &pEnc->sOriginal, XVID_PLG_AFTER, NULL, NULL, stats);
+                         }
- #define OFFSET(x,y) ((x) + (y)*pEnc->mbParam.mb_width)
+                         /* if the very last frame is to be b-vop, we must change it to a p-vop */
+                         if (pEnc->bframenum_tail > 0) {
-                                 for (x = 0; x < pEnc->mbParam.mb_width; x++) {
+                                 SWAP(FRAMEINFO*, pEnc->current, pEnc->reference);
-                                         MACROBLOCK *pMB = &pEnc->current->mbs[OFFSET(x, y)];
+                                 pEnc->bframenum_tail--;
+                                 SWAP(FRAMEINFO*, pEnc->current, pEnc->bframes[pEnc->bframenum_tail]);
-                                         pMB->dquant = iDQtab[temp_dquants[OFFSET(x, y)] + 2];
+                                 /* convert B-VOP to P-VOP */
-                                 }
+                                 pEnc->current->quant  = 100*pEnc->current->quant - pEnc->mbParam.bquant_offset;
+                                 pEnc->current->quant += pEnc->mbParam.bquant_ratio - 1; /* to avoid rouding issues */
+                                 pEnc->current->quant /= pEnc->mbParam.bquant_ratio;
- #undef OFFSET
+                                 if ((pEnc->mbParam.plugin_flags & XVID_REQORIGINAL)) {
+                                         image_copy(&pEnc->sOriginal, &pEnc->current->image,
+                                                    pEnc->mbParam.edged_width, pEnc->mbParam.height);
                          }
-                         xvid_free(temp_dquants);
+                                 DPRINTF(XVID_DEBUG_DEBUG,"*** PFRAME bf: head=%i tail=%i   queue: head=%i tail=%i size=%i\n",
-                 }
+                                 pEnc->bframenum_head, pEnc->bframenum_tail,
+                                 pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);
+                                 pEnc->mbParam.frame_drop_ratio = -1; /* it must be a coded vop */
-         }
+                                 FrameCodeP(pEnc, &bs);
-         /* %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
-          * ivop/pvop/bvop selection
-          * %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% */
-         pEnc->iFrameNum++;
-         if (pEnc->iFrameNum == 0 || pEnc->bframenum_dx50bvop >= 0 ||
+                                 if ((pEnc->mbParam.global_flags & XVID_GLOBAL_PACKED) && pEnc->bframenum_tail==0) {
-                 (mode < 0 && pEnc->mbParam.iMaxKeyInterval > 0 &&
+                                         call_plugins(pEnc, pEnc->current, &pEnc->sOriginal, XVID_PLG_AFTER, NULL, NULL, stats);
-                         pEnc->iFrameNum >= pEnc->mbParam.iMaxKeyInterval))
-         {
-                 mode = I_VOP;
          }else{
-                 mode = MEanalysis(&pEnc->reference->image, pEnc->current,
+                                         pEnc->flush_bframes = 1;
-                                         &pEnc->mbParam, pEnc->mbParam.iMaxKeyInterval,
+                                         goto done;
-                                         (mode < 0) ? pEnc->iFrameNum : 0,
-                                         bframes_count++);
          }
+                         }
+                         DPRINTF(XVID_DEBUG_DEBUG, "*** END\n");
-         if (mode == I_VOP) {
+                         emms();
-                 /*
+                         return XVID_ERR_END;    /* end of stream reached */
-                  * This will be coded as an Intra Frame
+                 }
-                  */
+                 goto done;      /* nothing to encode yet; encoder lag */
-                 if ((pEnc->current->global_flags & XVID_QUARTERPEL))
-                         pEnc->mbParam.m_quarterpel = 1;
-                 else
-                         pEnc->mbParam.m_quarterpel = 0;
-                 if (pEnc->current->global_flags & XVID_MPEGQUANT) pEnc->mbParam.m_quant_type = MPEG4_QUANT;
-                 if ((pEnc->current->global_flags & XVID_CUSTOM_QMATRIX) > 0) {
-                         if (pFrame->quant_intra_matrix != NULL)
-                                 set_intra_matrix(pFrame->quant_intra_matrix);
-                         if (pFrame->quant_inter_matrix != NULL)
-                                 set_inter_matrix(pFrame->quant_inter_matrix);
                  }
+         /* the current FRAME becomes the reference */
+         SWAP(FRAMEINFO*, pEnc->current, pEnc->reference);
-                 DPRINTF(DPRINTF_DEBUG,"*** IFRAME bf: head=%i tail=%i   queue: head=%i tail=%i size=%i",
+         /* remove frame from encoding-queue (head), and move it into the current */
-                                 pEnc->bframenum_head, pEnc->bframenum_tail,
+         image_swap(&pEnc->current->image, &pEnc->queue[pEnc->queue_head].image);
-                                 pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);
+         frame = &pEnc->queue[pEnc->queue_head].frame;
+         pEnc->queue_head = (pEnc->queue_head + 1) % (pEnc->mbParam.max_bframes+1);
+         pEnc->queue_size--;
-                 if ((pEnc->mbParam.global & XVID_GLOBAL_DEBUG)) {
-                         image_printf(&pEnc->current->image, pEnc->mbParam.edged_width, pEnc->mbParam.height, 5, 200, "IVOP");
-                 }
-                 // when we reach an iframe in DX50BVOP mode, encode the last bframe as a pframe
+         /* %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+          * init pEnc->current fields
+          * %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% */
-                 if ((pEnc->mbParam.global & XVID_GLOBAL_DX50BVOP) && pEnc->bframenum_tail > 0) {
+         pEnc->current->fincr = pEnc->mbParam.fincr>0 ? pEnc->mbParam.fincr : frame->fincr;
+         inc_frame_num(pEnc);
+         pEnc->current->vol_flags = frame->vol_flags;
+         pEnc->current->vop_flags = frame->vop_flags;
+         pEnc->current->motion_flags = frame->motion;
+         pEnc->current->fcode = pEnc->mbParam.m_fcode;
+         pEnc->current->bcode = pEnc->mbParam.m_fcode;
-                         pEnc->bframenum_tail--;
-                         pEnc->bframenum_dx50bvop = pEnc->bframenum_tail;
-                         SWAP(FRAMEINFO *, pEnc->current, pEnc->bframes[pEnc->bframenum_dx50bvop]);
+         if ((xFrame->vop_flags & XVID_VOP_CHROMAOPT)) {
-                         if ((pEnc->mbParam.global & XVID_GLOBAL_DEBUG)) {
+                 image_chroma_optimize(&pEnc->current->image,
-                                 image_printf(&pEnc->current->image, pEnc->mbParam.edged_width, pEnc->mbParam.height, 5, 100, "DX50 BVOP->PVOP");
+                         pEnc->mbParam.width, pEnc->mbParam.height, pEnc->mbParam.edged_width);
                          }
-                         FrameCodeP(pEnc, &bs, &bits, 1, 0);
-                         bframes_count = 0;
-                         pFrame->intra = 0;
+         /* %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+          * frame type & quant selection
+          * %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% */
-                 } else {
+         type = frame->type;
+         pEnc->current->quant = frame->quant;
-                         FrameCodeI(pEnc, &bs, &bits);
+         call_plugins(pEnc, pEnc->current, NULL, XVID_PLG_BEFORE, &type, (int*)&pEnc->current->quant, stats);
-                         bframes_count = 0;
-                         pFrame->intra = 1;
-                         pEnc->bframenum_dx50bvop = -1;
+         if (type > 0){  /* XVID_TYPE_?VOP */
+                 type = type2coding(type);       /* convert XVID_TYPE_?VOP to bitstream coding type */
+         } else{         /* XVID_TYPE_AUTO */
+                 if (pEnc->iFrameNum == 0 || (pEnc->mbParam.iMaxKeyInterval > 0 && pEnc->iFrameNum >= pEnc->mbParam.iMaxKeyInterval)){
+                         pEnc->iFrameNum = 0;
+                         type = I_VOP;
+                 }else{
+                         type = MEanalysis(&pEnc->reference->image, pEnc->current,
+                                                           &pEnc->mbParam, pEnc->mbParam.iMaxKeyInterval,
+                                                           pEnc->iFrameNum, pEnc->bframenum_tail, xFrame->bframe_threshold,
+                                                           (pEnc->bframes) ? pEnc->bframes[pEnc->bframenum_head]->mbs: NULL);
                  }
-                 pEnc->flush_bframes = 1;
-                 if ((pEnc->mbParam.global & XVID_GLOBAL_PACKED) && pEnc->bframenum_tail > 0) {
-                         BitstreamPadAlways(&bs);
-                         input_valid = 0;
-                         goto ipvop_loop;
                  }
-                 /*
+         if (type != I_VOP)
-                  * NB : sequences like "IIBB" decode fine with msfdam but,
+                 pEnc->current->vol_flags = pEnc->mbParam.vol_flags; /* don't allow VOL changes here */
-                  *      go screwy with divx 5.00
-                  */
-         } else if (mode == P_VOP || pEnc->bframenum_tail >= pEnc->mbParam.max_bframes) {
-                 /*
-                  * This will be coded as a Predicted Frame
-                  */
-                 DPRINTF(DPRINTF_DEBUG,"*** PFRAME bf: head=%i tail=%i   queue: head=%i tail=%i size=%i",
+         /* bframes buffer overflow check */
-                                 pEnc->bframenum_head, pEnc->bframenum_tail,
+         if (type == B_VOP && pEnc->bframenum_tail >= pEnc->mbParam.max_bframes) {
-                                 pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);
+                 type = P_VOP;
-                 if ((pEnc->mbParam.global & XVID_GLOBAL_DEBUG)) {
-                         image_printf(&pEnc->current->image, pEnc->mbParam.edged_width, pEnc->mbParam.height, 5, 200, "PVOP");
                  }
-                 FrameCodeP(pEnc, &bs, &bits, 1, 0);
+         pEnc->iFrameNum++;
-                 bframes_count = 0;
-                 pFrame->intra = 0;
-                 pEnc->flush_bframes = 1;
-                 if ((pEnc->mbParam.global & XVID_GLOBAL_PACKED) && (pEnc->bframenum_tail > 0)) {
+         if ((pEnc->current->vop_flags & XVID_VOP_DEBUG)) {
-                         BitstreamPadAlways(&bs);
+                 image_printf(&pEnc->current->image, pEnc->mbParam.edged_width, pEnc->mbParam.height, 5, 5,
-                         input_valid = 0;
+                         "%d  st:%lld  if:%d", pEnc->current->frame_num, pEnc->current->stamp, pEnc->iFrameNum);
-                         goto ipvop_loop;
                  }
-         } else {        /* mode == B_VOP */
+         /* %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
-                 /*
+          * encode this frame as a b-vop
-                  * This will be coded as a Bidirectional Frame
+          * (we dont encode here, rather we store the frame in the bframes queue, to be encoded later)
-                  */
+          * %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% */
+         if (type == B_VOP) {
-                 if ((pEnc->mbParam.global & XVID_GLOBAL_DEBUG)) {
+                 if ((pEnc->current->vop_flags & XVID_VOP_DEBUG)) {
                          image_printf(&pEnc->current->image, pEnc->mbParam.edged_width, pEnc->mbParam.height, 5, 200, "BVOP");
                  }
-                 if (pFrame->bquant < 1) {
+                 if (frame->quant < 1) {
                          pEnc->current->quant = ((((pEnc->reference->quant + pEnc->current->quant) *
                                  pEnc->mbParam.bquant_ratio) / 2) + pEnc->mbParam.bquant_offset)/100;
                  } else {
-                         pEnc->current->quant = pFrame->bquant;
+                         pEnc->current->quant = frame->quant;
                  }
                  if (pEnc->current->quant < 1)
-Line 1072
+Line 1320
                  else if (pEnc->current->quant > 31)
              pEnc->current->quant = 31;
-                 DPRINTF(DPRINTF_DEBUG,"*** BFRAME (store) bf: head=%i tail=%i   queue: head=%i tail=%i size=%i  quant=%i\n",
+                 DPRINTF(XVID_DEBUG_DEBUG,"*** BFRAME (store) bf: head=%i tail=%i   queue: head=%i tail=%i size=%i  quant=%i\n",
                                  pEnc->bframenum_head, pEnc->bframenum_tail,
                                  pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size,pEnc->current->quant);
-Line 1082
+Line 1330
                  pEnc->bframenum_tail++;
- // bframe report by koepi
+                 goto repeat;
-                 pFrame->intra = 2;
-                 pFrame->length = 0;
-                 input_valid = 0;
-                 goto bvop_loop;
-         }
-         BitstreamPadAlways(&bs);
-         pFrame->length = BitstreamLength(&bs);
-         if (pResult) {
-                 pResult->quant = pEnc->current->quant;
-                 pResult->hlength = pFrame->length - (pEnc->current->sStat.iTextBits / 8);
-                 pResult->kblks = pEnc->current->sStat.kblks;
-                 pResult->mblks = pEnc->current->sStat.mblks;
-                 pResult->ublks = pEnc->current->sStat.ublks;
          }
-         emms();
- #ifdef _DEBUG_PSNR
+                 DPRINTF(XVID_DEBUG_DEBUG,"*** XXXXXX bf: head=%i tail=%i   queue: head=%i tail=%i size=%i\n",
-         psnr =
+                                 pEnc->bframenum_head, pEnc->bframenum_tail,
-                 image_psnr(&pEnc->sOriginal, &pEnc->current->image,
+                                 pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);
-                                    pEnc->mbParam.edged_width, pEnc->mbParam.width,
-                                    pEnc->mbParam.height);
-         printf("PSNR: %f\n", psnr);
- //      DEBUG(temp);
- #endif
-         if (pFrame->quant == 0) {
+         /* for unpacked bframes, output the stats for the last encoded frame */
-                 RateControlUpdate(&pEnc->rate_control, pEnc->current->quant,
+         if (!(pEnc->mbParam.global_flags & XVID_GLOBAL_PACKED) && pEnc->mbParam.max_bframes > 0)
-                                                   pFrame->length, pFrame->intra);
+         {
+                 if (pEnc->current->stamp > 0) {
+                         call_plugins(pEnc, pEnc->reference, &pEnc->sOriginal, XVID_PLG_AFTER, NULL, NULL, stats);
+                 }
+         else if (stats) {
+             stats->type = XVID_TYPE_NOTHING;
+         }
          }
-         stop_global_timer();
+         /* %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
-         write_timer();
+          * closed-gop
+          * if the frame prior to an iframe is scheduled as a bframe, we must change it to a pframe
+          * %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% */
-         emms();
+         if (type == I_VOP && (pEnc->mbParam.global_flags & XVID_GLOBAL_CLOSED_GOP) && pEnc->bframenum_tail > 0) {
-         return XVID_ERR_OK;
- }
+                 /* place this frame back on the encoding-queue (head) */
+                 /* we will deal with it next time */
+                 dec_frame_num(pEnc);
+                 pEnc->iFrameNum--;
+                 pEnc->queue_head = (pEnc->queue_head + (pEnc->mbParam.max_bframes+1) - 1) % (pEnc->mbParam.max_bframes+1);
+                 pEnc->queue_size++;
+                 image_swap(&pEnc->current->image, &pEnc->queue[pEnc->queue_head].image);
- /*****************************************************************************
+                 /* grab the last frame from the bframe-queue */
-  * "original" IP frame encoder entry point
-  *
-  * Returned values :
-  *    - XVID_ERR_OK     - no errors
-  *    - XVID_ERR_FORMAT - the image subsystem reported the image had a wrong
-  *                        format
-  ****************************************************************************/
- int
+                 pEnc->bframenum_tail--;
- encoder_encode(Encoder * pEnc,
+                 SWAP(FRAMEINFO*, pEnc->current, pEnc->bframes[pEnc->bframenum_tail]);
-                            XVID_ENC_FRAME * pFrame,
-                            XVID_ENC_STATS * pResult)
- {
-         uint16_t x, y;
-         Bitstream bs;
-         uint32_t bits;
-         uint16_t write_vol_header = 0;
- #ifdef _DEBUG_PSNR
+                 if ((pEnc->current->vop_flags & XVID_VOP_DEBUG)) {
-         float psnr;
+                         image_printf(&pEnc->current->image, pEnc->mbParam.edged_width, pEnc->mbParam.height, 5, 100, "CLOSED GOP BVOP->PVOP");
-         uint8_t temp[128];
+                 }
- #endif
-         start_global_timer();
+                 /* convert B-VOP quant to P-VOP */
+                 pEnc->current->quant  = 100*pEnc->current->quant - pEnc->mbParam.bquant_offset;
+                 pEnc->current->quant += pEnc->mbParam.bquant_ratio - 1; /* to avoid rouding issues */
+                 pEnc->current->quant /= pEnc->mbParam.bquant_ratio;
+                 type = P_VOP;
+         }
-         ENC_CHECK(pEnc);
-         ENC_CHECK(pFrame);
-         ENC_CHECK(pFrame->bitstream);
-         ENC_CHECK(pFrame->image);
-         SWAP(FRAMEINFO *, pEnc->current, pEnc->reference);
+         /* %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+          * encode this frame as an i-vop
+          * %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% */
-         pEnc->current->global_flags = pFrame->general;
+         if (type == I_VOP) {
-         pEnc->current->motion_flags = pFrame->motion;
-         pEnc->mbParam.hint = &pFrame->hint;
-         inc_frame_num(pEnc);
+                 DPRINTF(XVID_DEBUG_DEBUG,"*** IFRAME bf: head=%i tail=%i   queue: head=%i tail=%i size=%i\n",
+                                 pEnc->bframenum_head, pEnc->bframenum_tail,
+                                 pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);
-         /* disable alternate scan flag if interlacing is not enabled */
+                 if ((pEnc->current->vop_flags & XVID_VOP_DEBUG)) {
-         if ((pEnc->current->global_flags & XVID_ALTERNATESCAN) &&
+                         image_printf(&pEnc->current->image, pEnc->mbParam.edged_width, pEnc->mbParam.height, 5, 200, "IVOP");
-                 !(pEnc->current->global_flags & XVID_INTERLACING))
-         {
-                 pEnc->current->global_flags -= XVID_ALTERNATESCAN;
          }
-         start_timer();
+                 pEnc->iFrameNum = 1;
-         if (image_input
-                 (&pEnc->current->image, pEnc->mbParam.width, pEnc->mbParam.height,
-                  pEnc->mbParam.edged_width, pFrame->image, pFrame->stride, pFrame->colorspace, pFrame->general & XVID_INTERLACING) < 0)
-                 return XVID_ERR_FORMAT;
-         stop_conv_timer();
- #ifdef _DEBUG_PSNR
+                 /* ---- update vol flags at IVOP ----------- */
-         image_copy(&pEnc->sOriginal, &pEnc->current->image,
+                 pEnc->mbParam.vol_flags = pEnc->current->vol_flags;
-                            pEnc->mbParam.edged_width, pEnc->mbParam.height);
- #endif
-         emms();
+                 /* Aspect ratio */
+                 switch(frame->par) {
+                 case XVID_PAR_11_VGA:
+                 case XVID_PAR_43_PAL:
+                 case XVID_PAR_43_NTSC:
+                 case XVID_PAR_169_PAL:
+                 case XVID_PAR_169_NTSC:
+                 case XVID_PAR_EXT:
+                         pEnc->mbParam.par = frame->par;
+                         break;
+                 default:
+                         pEnc->mbParam.par = XVID_PAR_11_VGA;
+                         break;
+                 }
-         BitstreamInit(&bs, pFrame->bitstream, 0);
+                 /* For extended PAR only, we try to sanityse/simplify par values */
+                 if (pEnc->mbParam.par == XVID_PAR_EXT) {
+                         pEnc->mbParam.par_width  = frame->par_width;
+                         pEnc->mbParam.par_height = frame->par_height;
+                         simplify_par(&pEnc->mbParam.par_width, &pEnc->mbParam.par_height);
+                 }
-         if (pFrame->quant == 0) {
+                 if ((pEnc->mbParam.vol_flags & XVID_VOL_MPEGQUANT)) {
-                 pEnc->current->quant = RateControlGetQ(&pEnc->rate_control, 0);
+                         if (frame->quant_intra_matrix != NULL)
-         } else {
+                                 set_intra_matrix(pEnc->mbParam.mpeg_quant_matrices, frame->quant_intra_matrix);
-                 pEnc->current->quant = pFrame->quant;
+                         if (frame->quant_inter_matrix != NULL)
+                                 set_inter_matrix(pEnc->mbParam.mpeg_quant_matrices, frame->quant_inter_matrix);
          }
-         if ((pEnc->current->global_flags & XVID_QUARTERPEL))
+                 /* prevent vol/vop misuse */
-                 pEnc->mbParam.m_quarterpel = 1;
-         else
-                 pEnc->mbParam.m_quarterpel = 0;
-         if ((pEnc->current->global_flags & XVID_LUMIMASKING)) {
+                 if (!(pEnc->current->vol_flags & XVID_VOL_INTERLACING))
-                 int *temp_dquants =
+                         pEnc->current->vop_flags &= ~(XVID_VOP_TOPFIELDFIRST|XVID_VOP_ALTERNATESCAN);
-                         (int *) xvid_malloc(pEnc->mbParam.mb_width *
-                                                                 pEnc->mbParam.mb_height * sizeof(int),
-                                                                 CACHE_LINE);
-                 pEnc->current->quant =
+                 /* ^^^------------------------ */
-                         adaptive_quantization(pEnc->current->image.y,
-                                                                   pEnc->mbParam.edged_width, temp_dquants,
-                                                                   pEnc->current->quant, pEnc->current->quant,
-* pEnc->current->quant,
-                                                                   pEnc->mbParam.mb_width,
-                                                                   pEnc->mbParam.mb_height);
-                 for (y = 0; y < pEnc->mbParam.mb_height; y++) {
+                 if ((pEnc->mbParam.plugin_flags & XVID_REQORIGINAL)) {
+                         image_copy(&pEnc->sOriginal, &pEnc->current->image,
+                                    pEnc->mbParam.edged_width, pEnc->mbParam.height);
+                 }
- #define OFFSET(x,y) ((x) + (y)*pEnc->mbParam.mb_width)
+                 FrameCodeI(pEnc, &bs);
+                 xFrame->out_flags |= XVID_KEYFRAME;
-                         for (x = 0; x < pEnc->mbParam.mb_width; x++) {
+         /* %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+          * encode this frame as an p-vop
+          * %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% */
+         } else { /* (type == P_VOP || type == S_VOP) */
-                                 MACROBLOCK *pMB = &pEnc->current->mbs[OFFSET(x, y)];
+                 DPRINTF(XVID_DEBUG_DEBUG,"*** PFRAME bf: head=%i tail=%i   queue: head=%i tail=%i size=%i\n",
+                                 pEnc->bframenum_head, pEnc->bframenum_tail,
+                                 pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size);
-                                 pMB->dquant = iDQtab[temp_dquants[OFFSET(x, y)] + 2];
+                 if ((pEnc->current->vop_flags & XVID_VOP_DEBUG)) {
+                         image_printf(&pEnc->current->image, pEnc->mbParam.edged_width, pEnc->mbParam.height, 5, 200, "PVOP");
                          }
- #undef OFFSET
+                 if ((pEnc->mbParam.plugin_flags & XVID_REQORIGINAL)) {
+                         image_copy(&pEnc->sOriginal, &pEnc->current->image,
+                                    pEnc->mbParam.edged_width, pEnc->mbParam.height);
                  }
-                 xvid_free(temp_dquants);
+                 if ( FrameCodeP(pEnc, &bs) == 0 ) {
+                         /* N-VOP, we mustn't code b-frames yet */
+                         if ((pEnc->mbParam.global_flags & XVID_GLOBAL_PACKED) ||
+                                  pEnc->mbParam.max_bframes == 0)
+                                 call_plugins(pEnc, pEnc->current, &pEnc->sOriginal, XVID_PLG_AFTER, NULL, NULL, stats);
+                         goto done;
+                 }
          }
-         if (pEnc->current->global_flags & XVID_H263QUANT) {
-                 if (pEnc->mbParam.m_quant_type != H263_QUANT)
-                         write_vol_header = 1;
-                 pEnc->mbParam.m_quant_type = H263_QUANT;
-         } else if (pEnc->current->global_flags & XVID_MPEGQUANT) {
-                 int matrix1_changed, matrix2_changed;
-                 matrix1_changed = matrix2_changed = 0;
+         /* %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+          * on next enc_encode call we must flush bframes
+          * %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% */
-                 if (pEnc->mbParam.m_quant_type != MPEG4_QUANT)
+ /*done_flush:*/
-                         write_vol_header = 1;
-                 pEnc->mbParam.m_quant_type = MPEG4_QUANT;
+         pEnc->flush_bframes = 1;
-                 if ((pEnc->current->global_flags & XVID_CUSTOM_QMATRIX) > 0) {
+         /* packed & queued_bframes: dont bother outputting stats here, we do so after the flush */
-                         if (pFrame->quant_intra_matrix != NULL)
+         if ((pEnc->mbParam.global_flags & XVID_GLOBAL_PACKED) && pEnc->bframenum_tail > 0) {
-                                 matrix1_changed = set_intra_matrix(pFrame->quant_intra_matrix);
+                 goto repeat;
-                         if (pFrame->quant_inter_matrix != NULL)
-                                 matrix2_changed = set_inter_matrix(pFrame->quant_inter_matrix);
-                 } else {
-                         matrix1_changed = set_intra_matrix(get_default_intra_matrix());
-                         matrix2_changed = set_inter_matrix(get_default_inter_matrix());
-                 }
-                 if (write_vol_header == 0)
-                         write_vol_header = matrix1_changed | matrix2_changed;
          }
-         if (pFrame->intra < 0) {
+         /* packed or no-bframes or no-bframes-queued: output stats */
-                 if ((pEnc->iFrameNum == 0)
+         if ((pEnc->mbParam.global_flags & XVID_GLOBAL_PACKED) || pEnc->mbParam.max_bframes == 0 ) {
-                         || ((pEnc->mbParam.iMaxKeyInterval > 0)
+                 call_plugins(pEnc, pEnc->current, &pEnc->sOriginal, XVID_PLG_AFTER, NULL, NULL, stats);
-                                 && (pEnc->iFrameNum >= pEnc->mbParam.iMaxKeyInterval))) {
-                         pFrame->intra = FrameCodeI(pEnc, &bs, &bits);
-                 } else {
-                         pFrame->intra = FrameCodeP(pEnc, &bs, &bits, 0, write_vol_header);
-                 }
-         } else {
-                 if (pFrame->intra == 1) {
-                         pFrame->intra = FrameCodeI(pEnc, &bs, &bits);
-                 } else {
-                         pFrame->intra = FrameCodeP(pEnc, &bs, &bits, 1, write_vol_header);
                  }
-         }
+         /* %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
+          * done; return number of bytes consumed
+          * %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% */
- //      BitstreamPutBits(&bs, 0xFFFF, 16);
+ done:
- //      BitstreamPutBits(&bs, 0xFFFF, 16);
-         BitstreamPadAlways(&bs);
-         pFrame->length = BitstreamLength(&bs);
-         if (pResult) {
+         stop_global_timer();
-                 pResult->quant = pEnc->current->quant;
+         write_timer();
-                 pResult->hlength = pFrame->length - (pEnc->current->sStat.iTextBits / 8);
-                 pResult->kblks = pEnc->current->sStat.kblks;
-                 pResult->mblks = pEnc->current->sStat.mblks;
-                 pResult->ublks = pEnc->current->sStat.ublks;
-         }
          emms();
+         return BitstreamLength(&bs);
+ }
-         if (pFrame->quant == 0) {
-                 RateControlUpdate(&pEnc->rate_control, pEnc->current->quant,
-                                                   pFrame->length, pFrame->intra);
-         }
- #ifdef _DEBUG_PSNR
-         psnr =
-                 image_psnr(&pEnc->sOriginal, &pEnc->current->image,
-                                    pEnc->mbParam.edged_width, pEnc->mbParam.width,
-                                    pEnc->mbParam.height);
-         snprintf(temp, 127, "PSNR: %f\n", psnr);
- //      DEBUG(temp);
- #endif
-         pEnc->iFrameNum++;
-         stop_global_timer();
+ static void SetMacroblockQuants(MBParam * const pParam, FRAMEINFO * frame)
-         write_timer();
+ {
+         unsigned int i;
-         return XVID_ERR_OK;
+         MACROBLOCK * pMB = frame->mbs;
+         int quant = frame->mbs[0].quant; /* set by XVID_PLG_FRAME */
+         if (quant > 31)
+                 frame->quant = quant = 31;
+         else if (quant < 1)
+                 frame->quant = quant = 1;
+         for (i = 0; i < pParam->mb_height * pParam->mb_width; i++) {
+                 quant += pMB->dquant;
+                 if (quant > 31)
+                         quant = 31;
+                 else if (quant < 1)
+                         quant = 1;
+                 pMB->quant = quant;
+                 pMB++;
+         }
  }
  static __inline void
- CodeIntraMB(Encoder * pEnc,
+ CodeIntraMB(MACROBLOCK * pMB)
-                         MACROBLOCK * pMB)
  {
          pMB->mode = MODE_INTRA;
          /* zero mv statistics */
-Line 1327
+Line 1539
          pMB->sad8[0] = pMB->sad8[1] = pMB->sad8[2] = pMB->sad8[3] = 0;
          pMB->sad16 = 0;
-         if ((pEnc->current->global_flags & XVID_LUMIMASKING)) {
+         if (pMB->dquant != 0) {
-                 if (pMB->dquant != NO_CHANGE) {
                          pMB->mode = MODE_INTRA_Q;
-                         pEnc->current->quant += DQtab[pMB->dquant];
-                         if (pEnc->current->quant > 31)
-                                 pEnc->current->quant = 31;
-                         if (pEnc->current->quant < 1)
-                                 pEnc->current->quant = 1;
-                 }
          }
-         pMB->quant = pEnc->current->quant;
  }
+ static void
- #define FCODEBITS       3
+ SliceCodeI(SMPData *data)
- #define MODEBITS        5
- void
- HintedMESet(Encoder * pEnc,
-                         int *intra)
  {
-         HINTINFO *hint;
+         Encoder *pEnc = (Encoder *) data->pEnc;
-         Bitstream bs;
+         Bitstream *bs = (Bitstream *) data->bs;
-         int length, high;
-         uint32_t x, y;
-         hint = pEnc->mbParam.hint;
+         uint16_t x, y;
+         int mb_width = pEnc->mbParam.mb_width;
+         int mb_height = pEnc->mbParam.mb_height;
-         if (hint->rawhints) {
+         int bound = 0, num_slices = pEnc->num_slices;
-                 *intra = hint->mvhint.intra;
+         FRAMEINFO *const current = pEnc->current;
-         } else {
-                 BitstreamInit(&bs, hint->hintstream, hint->hintlength);
-                 *intra = BitstreamGetBit(&bs);
-         }
-         if (*intra) {
+         DECLARE_ALIGNED_MATRIX(dct_codes, 6, 64, int16_t, CACHE_LINE);
-                 return;
+         DECLARE_ALIGNED_MATRIX(qcoeff, 6, 64, int16_t, CACHE_LINE);
+         if (data->start_y > 0) { /* write resync marker */
+                 bound = data->start_y*mb_width;
+                 write_video_packet_header(bs, &pEnc->mbParam, current, bound);
          }
-         pEnc->current->fcode =
+         for (y = data->start_y; y < data->stop_y; y++) {
-                 (hint->rawhints) ? hint->mvhint.fcode : BitstreamGetBits(&bs,
+                 int new_bound = mb_width * ((((y*num_slices) / mb_height) * mb_height + (num_slices-1)) / num_slices);
-                                                                                                                                  FCODEBITS);
-         length = pEnc->current->fcode + 5;
-         high = 1 << (length - 1);
-         for (y = 0; y < pEnc->mbParam.mb_height; ++y) {
-                 for (x = 0; x < pEnc->mbParam.mb_width; ++x) {
-                         MACROBLOCK *pMB =
-                                 &pEnc->current->mbs[x + y * pEnc->mbParam.mb_width];
-                         MVBLOCKHINT *bhint =
-                                 &hint->mvhint.block[x + y * pEnc->mbParam.mb_width];
-                         VECTOR pred;
-                         VECTOR tmp;
-                         int vec;
-                         pMB->mode =
-                                 (hint->rawhints) ? bhint->mode : BitstreamGetBits(&bs,
-                                                                                                                                   MODEBITS);
-                         pMB->mode = (pMB->mode == MODE_INTER_Q) ? MODE_INTER : pMB->mode;
-                         pMB->mode = (pMB->mode == MODE_INTRA_Q) ? MODE_INTRA : pMB->mode;
-                         if (pMB->mode == MODE_INTER) {
-                                 tmp.x =
-                                         (hint->rawhints) ? bhint->mvs[0].x : BitstreamGetBits(&bs,
-                                                                                                                                                   length);
-                                 tmp.y =
-                                         (hint->rawhints) ? bhint->mvs[0].y : BitstreamGetBits(&bs,
-                                                                                                                                                   length);
-                                 tmp.x -= (tmp.x >= high) ? high * 2 : 0;
-                                 tmp.y -= (tmp.y >= high) ? high * 2 : 0;
-                                 pred = get_pmv2(pEnc->current->mbs,pEnc->mbParam.mb_width,0,x,y,0);
-                                 for (vec = 0; vec < 4; ++vec) {
-                                         pMB->mvs[vec].x = tmp.x;
-                                         pMB->mvs[vec].y = tmp.y;
-                                         pMB->pmvs[vec].x = pMB->mvs[0].x - pred.x;
-                                         pMB->pmvs[vec].y = pMB->mvs[0].y - pred.y;
-                                 }
-                         } else if (pMB->mode == MODE_INTER4V) {
-                                 for (vec = 0; vec < 4; ++vec) {
-                                         tmp.x =
-                                                 (hint->rawhints) ? bhint->mvs[vec].
-                                                 x : BitstreamGetBits(&bs, length);
-                                         tmp.y =
-                                                 (hint->rawhints) ? bhint->mvs[vec].
-                                                 y : BitstreamGetBits(&bs, length);
-                                         tmp.x -= (tmp.x >= high) ? high * 2 : 0;
-                                         tmp.y -= (tmp.y >= high) ? high * 2 : 0;
-                                         pred = get_pmv2(pEnc->current->mbs,pEnc->mbParam.mb_width,0,x,y,vec);
-                                         pMB->mvs[vec].x = tmp.x;
-                                         pMB->mvs[vec].y = tmp.y;
-                                         pMB->pmvs[vec].x = pMB->mvs[vec].x - pred.x;
-                                         pMB->pmvs[vec].y = pMB->mvs[vec].y - pred.y;
-                                 }
-                         } else                          // intra / stuffing / not_coded
-                         {
-                                 for (vec = 0; vec < 4; ++vec) {
-                                         pMB->mvs[vec].x = pMB->mvs[vec].y = 0;
-                                 }
-                         }
-                         if (pMB->mode == MODE_INTER4V &&
-                                 (pEnc->current->global_flags & XVID_LUMIMASKING)
-                                 && pMB->dquant != NO_CHANGE) {
-                                 pMB->mode = MODE_INTRA;
-                                 for (vec = 0; vec < 4; ++vec) {
+                 if (new_bound > bound) {
-                                         pMB->mvs[vec].x = pMB->mvs[vec].y = 0;
+                         bound = new_bound;
-                                 }
+                         BitstreamPadAlways(bs);
-                         }
+                         write_video_packet_header(bs, &pEnc->mbParam, current, bound);
-                 }
-         }
  }
+                 for (x = 0; x < mb_width; x++) {
+                         MACROBLOCK *pMB = &current->mbs[x + y * mb_width];
- void
+                         CodeIntraMB(pMB);
- HintedMEGet(Encoder * pEnc,
-                         int intra)
- {
-         HINTINFO *hint;
-         Bitstream bs;
-         uint32_t x, y;
-         int length, high;
-         hint = pEnc->mbParam.hint;
+                         MBTransQuantIntra(&pEnc->mbParam, current, pMB, x, y,
+                                                           dct_codes, qcoeff);
-         if (hint->rawhints) {
+                         start_timer();
-                 hint->mvhint.intra = intra;
+                         MBPrediction(current, x, y, mb_width, qcoeff, bound);
-         } else {
+                         stop_prediction_timer();
-                 BitstreamInit(&bs, hint->hintstream, 0);
-                 BitstreamPutBit(&bs, intra);
+                         start_timer();
-         }
+                         MBCoding(current, pMB, qcoeff, bs, data->sStat);
+                         stop_coding_timer();
-         if (intra) {
-                 if (!hint->rawhints) {
-                         BitstreamPadAlways(&bs);
-                         hint->hintlength = BitstreamLength(&bs);
                  }
-                 return;
          }
-         length = pEnc->current->fcode + 5;
+         emms();
-         high = 1 << (length - 1);
+         BitstreamPadAlways(bs);
-         if (hint->rawhints) {
-                 hint->mvhint.fcode = pEnc->current->fcode;
-         } else {
-                 BitstreamPutBits(&bs, pEnc->current->fcode, FCODEBITS);
          }
-         for (y = 0; y < pEnc->mbParam.mb_height; ++y) {
+ static __inline void
-                 for (x = 0; x < pEnc->mbParam.mb_width; ++x) {
+ SerializeBitstreams(Encoder *pEnc, FRAMEINFO *current, Bitstream *bs, int num_threads)
-                         MACROBLOCK *pMB =
+ {
-                                 &pEnc->current->mbs[x + y * pEnc->mbParam.mb_width];
+         int k;
-                         MVBLOCKHINT *bhint =
+         uint32_t pos = BitstreamLength(bs);
-                                 &hint->mvhint.block[x + y * pEnc->mbParam.mb_width];
-                         VECTOR tmp;
-                         if (hint->rawhints) {
+         for (k = 1; k < num_threads; k++) {
-                                 bhint->mode = pMB->mode;
+                 uint32_t len = BitstreamLength(pEnc->smpData[k].bs);
-                         } else {
-                                 BitstreamPutBits(&bs, pMB->mode, MODEBITS);
-                         }
-                         if (pMB->mode == MODE_INTER || pMB->mode == MODE_INTER_Q) {
+                 memcpy((void *)((ptr_t)bs->start + pos),
-                                 tmp.x = pMB->mvs[0].x;
+                            (void *)((ptr_t)pEnc->smpData[k].bs->start), len);
-                                 tmp.y = pMB->mvs[0].y;
-                                 tmp.x += (tmp.x < 0) ? high * 2 : 0;
+                 current->length = pos += len;
-                                 tmp.y += (tmp.y < 0) ? high * 2 : 0;
+                 /* collect stats */
-                                 if (hint->rawhints) {
+                 current->sStat.iTextBits += pEnc->smpData[k].sStat->iTextBits;
-                                         bhint->mvs[0].x = tmp.x;
+                 current->sStat.kblks += pEnc->smpData[k].sStat->kblks;
-                                         bhint->mvs[0].y = tmp.y;
+                 current->sStat.mblks += pEnc->smpData[k].sStat->mblks;
-                                 } else {
+                 current->sStat.ublks += pEnc->smpData[k].sStat->ublks;
-                                         BitstreamPutBits(&bs, tmp.x, length);
+                 current->sStat.iMVBits += pEnc->smpData[k].sStat->iMVBits;
-                                         BitstreamPutBits(&bs, tmp.y, length);
+         }
-                                 }
-                         } else if (pMB->mode == MODE_INTER4V) {
+         if (num_threads > 1) {
-                                 int vec;
+                 uint32_t pos32 = pos>>2;
+                 bs->tail = bs->start + pos32;
-                                 for (vec = 0; vec < 4; ++vec) {
+                 bs->pos = 8*(pos - (pos32<<2));
-                                         tmp.x = pMB->mvs[vec].x;
+                 bs->buf = 0;
-                                         tmp.y = pMB->mvs[vec].y;
-                                         tmp.x += (tmp.x < 0) ? high * 2 : 0;
+                 if (bs->pos > 0) {
-                                         tmp.y += (tmp.y < 0) ? high * 2 : 0;
+                         uint32_t pos8 = bs->pos/8;
+                         memset((void *)((ptr_t)bs->tail+pos8), 0, (4-pos8));
-                                         if (hint->rawhints) {
+                         pos = *bs->tail;
-                                                 bhint->mvs[vec].x = tmp.x;
+ #ifndef ARCH_IS_BIG_ENDIAN
-                                                 bhint->mvs[vec].y = tmp.y;
+                         BSWAP(pos);
-                                         } else {
+ #endif
-                                                 BitstreamPutBits(&bs, tmp.x, length);
+                         bs->buf = pos;
-                                                 BitstreamPutBits(&bs, tmp.y, length);
-                                         }
-                                 }
-                         }
-                 }
          }
-         if (!hint->rawhints) {
-                 BitstreamPad(&bs);
-                 hint->hintlength = BitstreamLength(&bs);
          }
  }
  static int
  FrameCodeI(Encoder * pEnc,
-                    Bitstream * bs,
+                    Bitstream * bs)
-                    uint32_t * pBits)
  {
+         int bits = BitstreamPos(bs);
+         int bound = 0, num_slices = pEnc->num_slices;
+         int num_threads = MAX(1, MIN(pEnc->num_threads, num_slices));
+         int slices_per_thread = (num_slices*1024 / num_threads);
          int mb_width = pEnc->mbParam.mb_width;
          int mb_height = pEnc->mbParam.mb_height;
+         void * status = NULL;
+         uint16_t k;
-         DECLARE_ALIGNED_MATRIX(dct_codes, 6, 64, int16_t, CACHE_LINE);
-         DECLARE_ALIGNED_MATRIX(qcoeff, 6, 64, int16_t, CACHE_LINE);
-         uint16_t x, y;
-         if ((pEnc->current->global_flags & XVID_REDUCED))
-         {
-                 mb_width = (pEnc->mbParam.width + 31) / 32;
-                 mb_height = (pEnc->mbParam.height + 31) / 32;
-                 /* 16x16->8x8 downsample requires 1 additional edge pixel*/
-                 /* XXX: setedges is overkill */
-                 start_timer();
-                 image_setedges(&pEnc->current->image,
-                         pEnc->mbParam.edged_width, pEnc->mbParam.edged_height,
-                         pEnc->mbParam.width, pEnc->mbParam.height);
-                 stop_edges_timer();
-         }
-         pEnc->iFrameNum = 0;
          pEnc->mbParam.m_rounding_type = 1;
          pEnc->current->rounding_type = pEnc->mbParam.m_rounding_type;
-         pEnc->current->quarterpel =  pEnc->mbParam.m_quarterpel;
          pEnc->current->coding_type = I_VOP;
-         BitstreamWriteVolHeader(bs, &pEnc->mbParam, pEnc->current);
+         call_plugins(pEnc, pEnc->current, NULL, XVID_PLG_FRAME, NULL, NULL, NULL);
+         SetMacroblockQuants(&pEnc->mbParam, pEnc->current);
+         BitstreamWriteVolHeader(bs, &pEnc->mbParam, pEnc->current, num_slices);
          set_timecodes(pEnc->current,pEnc->reference,pEnc->mbParam.fbase);
-         BitstreamPadAlways(bs);
+         BitstreamPad(bs);
-         BitstreamWriteVopHeader(bs, &pEnc->mbParam, pEnc->current, 1);
-         *pBits = BitstreamPos(bs);
+         BitstreamWriteVopHeader(bs, &pEnc->mbParam, pEnc->current, 1, pEnc->current->mbs[0].quant);
          pEnc->current->sStat.iTextBits = 0;
-         pEnc->current->sStat.kblks = mb_width * mb_height;
-         pEnc->current->sStat.mblks = pEnc->current->sStat.ublks = 0;
-         for (y = 0; y < mb_height; y++)
+         /* multithreaded intra coding - dispatch threads */
-                 for (x = 0; x < mb_width; x++) {
+         for (k = 0; k < num_threads; k++) {
-                         MACROBLOCK *pMB =
+                 int add = ((slices_per_thread + 512) >> 10);
-                                 &pEnc->current->mbs[x + y * pEnc->mbParam.mb_width];
-                         CodeIntraMB(pEnc, pMB);
+                 slices_per_thread += ((num_slices*1024 / num_threads) - add*1024);
-                         MBTransQuantIntra(&pEnc->mbParam, pEnc->current, pMB, x, y,
+                 pEnc->smpData[k].pEnc = (void *) pEnc;
-                                                           dct_codes, qcoeff);
+                 pEnc->smpData[k].stop_y = (((bound+add) * mb_height + (num_slices-1)) / num_slices);
+                 pEnc->smpData[k].start_y = ((bound * mb_height + (num_slices-1)) / num_slices);
-                         start_timer();
+                 bound += add;
-                         MBPrediction(pEnc->current, x, y, pEnc->mbParam.mb_width, qcoeff);
-                         stop_prediction_timer();
-                         start_timer();
+                 if (k > 0) {
-                         if (pEnc->current->global_flags & XVID_GREYSCALE)
+                         BitstreamReset(pEnc->smpData[k].bs);
-                         {       pMB->cbp &= 0x3C;               /* keep only bits 5-2 */
+                         pEnc->smpData[k].sStat->iTextBits = 0;
-                                 qcoeff[4*64+0]=0;               /* zero, because for INTRA MBs DC value is saved */
-                                 qcoeff[5*64+0]=0;
                          }
-                         MBCoding(pEnc->current, pMB, qcoeff, bs, &pEnc->current->sStat);
-                         stop_coding_timer();
                  }
+         pEnc->smpData[0].bs = bs;
+         pEnc->smpData[0].sStat = &pEnc->current->sStat;
-         if ((pEnc->current->global_flags & XVID_REDUCED))
+         /* create threads */
-         {
+         for (k = 1; k < num_threads; k++) {
-                 image_deblock_rrv(&pEnc->current->image, pEnc->mbParam.edged_width,
+                 pthread_create(&pEnc->smpData[k].handle, NULL,
-                         pEnc->current->mbs, mb_width, mb_height, pEnc->mbParam.mb_width,
+                                (void*)SliceCodeI, (void*)&pEnc->smpData[k]);
-, XVID_DEC_DEBLOCKY|XVID_DEC_DEBLOCKUV);
          }
-         emms();
-         *pBits = BitstreamPos(bs) - *pBits;
-         pEnc->fMvPrevSigma = -1;
-         pEnc->mbParam.m_fcode = 2;
-         if (pEnc->current->global_flags & XVID_HINTEDME_GET) {
+         SliceCodeI(&pEnc->smpData[0]);
-                 HintedMEGet(pEnc, 1);
-         }
-         return 1;                                       // intra
+         /* wait until all threads are finished */
+         for (k = 1; k < num_threads; k++) {
+                 pthread_join(pEnc->smpData[k].handle, &status);
  }
+         pEnc->current->length = BitstreamLength(bs) - (bits/8);
- #define INTRA_THRESHOLD 0.5
+         /* reassemble the pieces together */
- #define BFRAME_SKIP_THRESHHOLD 30
+         SerializeBitstreams(pEnc, pEnc->current, bs, num_threads);
- /* FrameCodeP also handles S(GMC)-VOPs */
+         pEnc->current->sStat.iMVBits = 0;
- static int
+         pEnc->current->sStat.mblks = pEnc->current->sStat.ublks = 0;
- FrameCodeP(Encoder * pEnc,
+         pEnc->current->sStat.kblks = pEnc->mbParam.mb_width * pEnc->mbParam.mb_height;
-                    Bitstream * bs,
-                    uint32_t * pBits,
-                    bool force_inter,
-                    bool vol_header)
- {
-         float fSigma;
-         DECLARE_ALIGNED_MATRIX(dct_codes, 6, 64, int16_t, CACHE_LINE);
-         DECLARE_ALIGNED_MATRIX(qcoeff, 6, 64, int16_t, CACHE_LINE);
-         int mb_width = pEnc->mbParam.mb_width;
-         int mb_height = pEnc->mbParam.mb_height;
-         int iLimit;
-         int x, y, k;
-         int iSearchRange;
-         int bIntra, skip_possible;
-         /* IMAGE *pCurrent = &pEnc->current->image; */
-         IMAGE *pRef = &pEnc->reference->image;
-         if ((pEnc->current->global_flags & XVID_REDUCED))
-         {
-                 mb_width = (pEnc->mbParam.width + 31) / 32;
-                 mb_height = (pEnc->mbParam.height + 31) / 32;
-         }
-         start_timer();
-         image_setedges(pRef, pEnc->mbParam.edged_width, pEnc->mbParam.edged_height,
-                                    pEnc->mbParam.width, pEnc->mbParam.height);
-         stop_edges_timer();
-         pEnc->mbParam.m_rounding_type = 1 - pEnc->mbParam.m_rounding_type;
+         pEnc->fMvPrevSigma = -1;
-         pEnc->current->rounding_type = pEnc->mbParam.m_rounding_type;
+         pEnc->mbParam.m_fcode = 2;
-         pEnc->current->quarterpel =  pEnc->mbParam.m_quarterpel;
-         pEnc->current->fcode = pEnc->mbParam.m_fcode;
-         if (!force_inter)
+         pEnc->current->is_edged = 0; /* not edged */
-                 iLimit = (int)(mb_width * mb_height *  INTRA_THRESHOLD);
+         pEnc->current->is_interpolated = -1; /* not interpolated (fake rounding -1) */
-         else
-                 iLimit = mb_width * mb_height + 1;
-         if ((pEnc->current->global_flags & XVID_HALFPEL)) {
+         return 1;                                       /* intra */
-                 start_timer();
-                 image_interpolate(pRef, &pEnc->vInterH, &pEnc->vInterV,
-                                                   &pEnc->vInterHV, pEnc->mbParam.edged_width,
-                                                   pEnc->mbParam.edged_height,
-                                                   pEnc->mbParam.m_quarterpel,
-                                                   pEnc->current->rounding_type);
-                 stop_inter_timer();
          }
-         pEnc->current->coding_type = P_VOP;
+ static __inline void
+ updateFcode(Statistics * sStat, Encoder * pEnc)
+ {
+         float fSigma;
+         int iSearchRange;
-         start_timer();
+         if (sStat->iMvCount == 0)
-         if (pEnc->current->global_flags & XVID_HINTEDME_SET)
+                 sStat->iMvCount = 1;
-                 HintedMESet(pEnc, &bIntra);
-         else
-                 bIntra =
-                         MotionEstimation(&pEnc->mbParam, pEnc->current, pEnc->reference,
-                          &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV,
-                          iLimit);
-         stop_motion_timer();
+         fSigma = (float) sqrt((float) sStat->iMvSum / sStat->iMvCount);
-         if (bIntra == 1) return FrameCodeI(pEnc, bs, pBits);
+         iSearchRange = 16 << pEnc->mbParam.m_fcode;
-         if ( ( pEnc->current->global_flags & XVID_GMC )
+         if ((3.0 * fSigma > iSearchRange) && (pEnc->mbParam.m_fcode <= 5) )
-                 && ( (pEnc->current->warp.duv[1].x != 0) || (pEnc->current->warp.duv[1].y != 0) ) )
+                 pEnc->mbParam.m_fcode++;
-         {
-                 pEnc->current->coding_type = S_VOP;
-                 generate_GMCparameters( 2, 16, &pEnc->current->warp,
+         else if ((5.0 * fSigma < iSearchRange)
-                                         pEnc->mbParam.width, pEnc->mbParam.height,
+                            && (4.0 * pEnc->fMvPrevSigma < iSearchRange)
-                                         &pEnc->current->gmc_data);
+                            && (pEnc->mbParam.m_fcode >= 2) )
+                 pEnc->mbParam.m_fcode--;
-                 generate_GMCimage(&pEnc->current->gmc_data, &pEnc->reference->image,
-                                 pEnc->mbParam.mb_width, pEnc->mbParam.mb_height,
-                                 pEnc->mbParam.edged_width, pEnc->mbParam.edged_width/2,
-                                 pEnc->mbParam.m_fcode, pEnc->mbParam.m_quarterpel, 0,
-                                 pEnc->current->rounding_type, pEnc->current->mbs, &pEnc->vGMC);
+         pEnc->fMvPrevSigma = fSigma;
          }
-         set_timecodes(pEnc->current,pEnc->reference,pEnc->mbParam.fbase);
+ #define BFRAME_SKIP_THRESHHOLD 30
-         if (vol_header)
-         {       BitstreamWriteVolHeader(bs, &pEnc->mbParam, pEnc->current);
-                 BitstreamPadAlways(bs);
-         }
-         BitstreamWriteVopHeader(bs, &pEnc->mbParam, pEnc->current, 1);
+ static void
+ SliceCodeP(SMPData *data)
+ {
+         Encoder *pEnc = (Encoder *) data->pEnc;
+         Bitstream *bs = (Bitstream *) data->bs;
+         int x, y, k;
+         FRAMEINFO *const current = pEnc->current;
+         FRAMEINFO *const reference = pEnc->reference;
+         MBParam * const pParam = &pEnc->mbParam;
+         IMAGE *pRef = &reference->image;
+         int mb_width = pParam->mb_width;
+         int mb_height = pParam->mb_height;
-         *pBits = BitstreamPos(bs);
+         DECLARE_ALIGNED_MATRIX(dct_codes, 6, 64, int16_t, CACHE_LINE);
+         DECLARE_ALIGNED_MATRIX(qcoeff, 6, 64, int16_t, CACHE_LINE);
-         pEnc->current->sStat.iTextBits = pEnc->current->sStat.iMvSum = pEnc->current->sStat.iMvCount =
+         int bound = 0, num_slices = pEnc->num_slices;
-                 pEnc->current->sStat.kblks = pEnc->current->sStat.mblks = pEnc->current->sStat.ublks = 0;
+         if (data->start_y > 0) { /* write resync marker */
+                 bound = data->start_y*mb_width;
+                 write_video_packet_header(bs, pParam, current, bound);
+         }
-         for (y = 0; y < mb_height; y++) {
+         for (y = data->start_y; y < data->stop_y; y++) {
-                 for (x = 0; x < mb_width; x++) {
+                 int new_bound = mb_width * ((((y*num_slices) / mb_height) * mb_height + (num_slices-1)) / num_slices);
-                         MACROBLOCK *pMB =
-                                 &pEnc->current->mbs[x + y * pEnc->mbParam.mb_width];
- /* Mode decision: Check, if the block should be INTRA / INTER or GMC-coded */
+                 if (new_bound > bound) {
- /* For a start, leave INTRA decision as is, only choose only between INTER/GMC  - gruel, 9.1.2002 */
+                         bound = new_bound;
+                         BitstreamPadAlways(bs);
+                         write_video_packet_header(bs, pParam, current, bound);
+                 }
-                         bIntra = (pMB->mode == MODE_INTRA) || (pMB->mode == MODE_INTRA_Q);
+                 for (x = 0; x < mb_width; x++) {
+                         MACROBLOCK *pMB = &current->mbs[x + y * pParam->mb_width];
+                         int skip_possible;
-                         if (bIntra) {
+                         if (pMB->mode == MODE_INTRA || pMB->mode == MODE_INTRA_Q) {
-                                 CodeIntraMB(pEnc, pMB);
+                                 CodeIntraMB(pMB);
-                                 MBTransQuantIntra(&pEnc->mbParam, pEnc->current, pMB, x, y,
+                                 MBTransQuantIntra(pParam, current, pMB, x, y,
                                                                    dct_codes, qcoeff);
                                  start_timer();
-                                 MBPrediction(pEnc->current, x, y, pEnc->mbParam.mb_width, qcoeff);
+                                 MBPrediction(current, x, y, pParam->mb_width, qcoeff, bound);
                                  stop_prediction_timer();
-                                 pEnc->current->sStat.kblks++;
+                                 data->sStat->kblks++;
-                                 MBCoding(pEnc->current, pMB, qcoeff, bs, &pEnc->current->sStat);
+                                 MBCoding(current, pMB, qcoeff, bs, data->sStat);
                                  stop_coding_timer();
                                  continue;
                          }
-                         if (pEnc->current->coding_type == S_VOP) {
-                                 int32_t iSAD = sad16(pEnc->current->image.y + 16*y*pEnc->mbParam.edged_width + 16*x,
-                                         pEnc->vGMC.y + 16*y*pEnc->mbParam.edged_width + 16*x,
-                                         pEnc->mbParam.edged_width, 65536);
-                                 if (iSAD <= pMB->sad16) {               /* mode decision GMC */
-                                         if (pEnc->mbParam.m_quarterpel)
-                                                 pMB->qmvs[0] = pMB->qmvs[1] = pMB->qmvs[2] = pMB->qmvs[3] = pMB->amv;
-                                         else
-                                                 pMB->mvs[0] = pMB->mvs[1] = pMB->mvs[2] = pMB->mvs[3] = pMB->amv;
-                                         pMB->mode = MODE_INTER;
-                                         pMB->mcsel = 1;
-                                         pMB->sad16 = iSAD;
-                                 } else {
-                                         pMB->mcsel = 0;
-                                 }
-                         } else {
-                                 pMB->mcsel = 0; /* just a precaution */
-                         }
                          start_timer();
-                         MBMotionCompensation(pMB, x, y, &pEnc->reference->image,
+                         MBMotionCompensation(pMB, x, y, &reference->image,
                                                                   &pEnc->vInterH, &pEnc->vInterV,
                                                                   &pEnc->vInterHV, &pEnc->vGMC,
-                                                                  &pEnc->current->image,
+                                                                  &current->image,
-                                                                  dct_codes, pEnc->mbParam.width,
+                                                                  dct_codes, pParam->width,
-                                                                  pEnc->mbParam.height,
+                                                                  pParam->height,
-                                                                  pEnc->mbParam.edged_width,
+                                                                  pParam->edged_width,
-                                                                  pEnc->mbParam.m_quarterpel,
+                                                                  (current->vol_flags & XVID_VOL_QUARTERPEL),
-                                                                  (pEnc->current->global_flags & XVID_REDUCED),
+                                                                  current->rounding_type,
-                                                                  pEnc->current->rounding_type);
+                                                                  data->RefQ);
                          stop_comp_timer();
-                         if ((pEnc->current->global_flags & XVID_LUMIMASKING)) {
-                                 if (pMB->dquant != NO_CHANGE) {
-                                         pMB->mode = MODE_INTER_Q;
-                                         pEnc->current->quant += DQtab[pMB->dquant];
-                                         if (pEnc->current->quant > 31)
-                                                 pEnc->current->quant = 31;
-                                         else if (pEnc->current->quant < 1)
-                                                 pEnc->current->quant = 1;
-                                 }
-                         }
-                         pMB->quant = pEnc->current->quant;
                          pMB->field_pred = 0;
-                         if (pMB->mode != MODE_NOT_CODED)
+                         if (pMB->cbp != 0) {
-                         {       pMB->cbp =
+                                 pMB->cbp = MBTransQuantInter(pParam, current, pMB, x, y,
-                                         MBTransQuantInter(&pEnc->mbParam, pEnc->current, pMB, x, y,
                                                                            dct_codes, qcoeff);
                          }
+                         if (pMB->dquant != 0)
+                                 MBSetDquant(pMB, x, y, pParam);
                          if (pMB->cbp || pMB->mvs[0].x || pMB->mvs[0].y ||
                                     pMB->mvs[1].x || pMB->mvs[1].y || pMB->mvs[2].x ||
                                     pMB->mvs[2].y || pMB->mvs[3].x || pMB->mvs[3].y) {
-                                 pEnc->current->sStat.mblks++;
+                                 data->sStat->mblks++;
                          }  else {
-                                 pEnc->current->sStat.ublks++;
+                                 data->sStat->ublks++;
                          }
                          start_timer();
                          /* Finished processing the MB, now check if to CODE or SKIP */
-                         skip_possible = (pMB->cbp == 0) && (pMB->mode == MODE_INTER) &&
+                         skip_possible = (pMB->cbp == 0) && (pMB->mode == MODE_INTER);
-                                                         (pMB->dquant == NO_CHANGE);
-                         if (pEnc->current->coding_type == S_VOP)
+                         if (current->coding_type == S_VOP)
                                  skip_possible &= (pMB->mcsel == 1);
-                         else if (pEnc->current->coding_type == P_VOP) {
+                         else { /* PVOP */
-                                 if (pEnc->mbParam.m_quarterpel)
+                                 const VECTOR * const mv = (pParam->vol_flags & XVID_VOL_QUARTERPEL) ?
-                                         skip_possible &= ( (pMB->qmvs[0].x == 0) && (pMB->qmvs[0].y == 0) );
+                                                                                 pMB->qmvs : pMB->mvs;
-                                 else
+                                 skip_possible &= ((mv->x|mv->y) == 0);
-                                         skip_possible &= ( (pMB->mvs[0].x == 0) && (pMB->mvs[0].y == 0) );
                          }
                          if ( (pMB->mode == MODE_NOT_CODED) || (skip_possible)) {
  /* This is a candidate for SKIPping, but for P-VOPs check intermediate B-frames first */
-                                 if (pEnc->current->coding_type == P_VOP)        /* special rule for P-VOP's SKIP */
-                                 {
                                          int bSkip = 1;
-                                         for (k=pEnc->bframenum_head; k< pEnc->bframenum_tail; k++)
+                                 if (current->coding_type == P_VOP) {    /* special rule for P-VOP's SKIP */
-                                         {
+                                         for (k = pEnc->bframenum_head; k < pEnc->bframenum_tail; k++) {
                                                  int iSAD;
-                                                 iSAD = sad16(pEnc->reference->image.y + 16*y*pEnc->mbParam.edged_width + 16*x,
+                                                 iSAD = sad16(reference->image.y + 16*y*pParam->edged_width + 16*x,
-                                                                         pEnc->bframes[k]->image.y + 16*y*pEnc->mbParam.edged_width + 16*x,
+                                                                                 pEnc->bframes[k]->image.y + 16*y*pParam->edged_width + 16*x,
-                                                                 pEnc->mbParam.edged_width,BFRAME_SKIP_THRESHHOLD);
+                                                                                 pParam->edged_width, BFRAME_SKIP_THRESHHOLD * pMB->quant);
-                                                 if (iSAD >= BFRAME_SKIP_THRESHHOLD * pMB->quant)
+                                                 if (iSAD >= BFRAME_SKIP_THRESHHOLD * pMB->quant) {
-                                                 {       bSkip = 0;
+                                                         bSkip = 0; /* could not SKIP */
-                                                         break;
+                                                         if (pParam->vol_flags & XVID_VOL_QUARTERPEL) {
-                                                 }
+                                                                 VECTOR predMV = get_qpmv2(current->mbs, pParam->mb_width, bound, x, y, 0);
-                                         }
-                                         if (!bSkip) {   /* no SKIP, but trivial block */
-                                                 if(pEnc->mbParam.m_quarterpel) {
-                                                         VECTOR predMV = get_qpmv2(pEnc->current->mbs, pEnc->mbParam.mb_width, 0, x, y, 0);
                                                          pMB->pmvs[0].x = - predMV.x;
                                                          pMB->pmvs[0].y = - predMV.y;
-                                                 }
+                                                         } else {
-                                                 else {
+                                                                 VECTOR predMV = get_pmv2(current->mbs, pParam->mb_width, bound, x, y, 0);
-                                                         VECTOR predMV = get_pmv2(pEnc->current->mbs, pEnc->mbParam.mb_width, 0, x, y, 0);
                                                          pMB->pmvs[0].x = - predMV.x;
                                                          pMB->pmvs[0].y = - predMV.y;
                                                  }
                                                  pMB->mode = MODE_INTER;
                                                  pMB->cbp = 0;
-                                                 MBCoding(pEnc->current, pMB, qcoeff, bs, &pEnc->current->sStat);
+                                                         break;
-                                                 stop_coding_timer();
+                                                 }
-                                                 continue;       /* next MB */
                                          }
                                  }
-                                 /* do SKIP */
+                                 if (bSkip) {
+                                         /* do SKIP */
                                  pMB->mode = MODE_NOT_CODED;
                                  MBSkip(bs);
                                  stop_coding_timer();
                                  continue;       /* next MB */
                          }
+                         }
                          /* ordinary case: normal coded INTER/INTER4V block */
+                         MBCoding(current, pMB, qcoeff, bs, data->sStat);
+                         stop_coding_timer();
+                 }
+         }
-                         if (pEnc->current->global_flags & XVID_GREYSCALE)
+         BitstreamPadAlways(bs); /* next_start_code() at the end of VideoObjectPlane() */
-                         {       pMB->cbp &= 0x3C;               /* keep only bits 5-2 */
+         emms();
-                                 qcoeff[4*64+0]=0;               /* zero, because DC for INTRA MBs DC value is saved */
-                                 qcoeff[5*64+0]=0;
                          }
-                         if(pEnc->mbParam.m_quarterpel) {
+ /* FrameCodeP also handles S(GMC)-VOPs */
-                                 VECTOR predMV = get_qpmv2(pEnc->current->mbs, pEnc->mbParam.mb_width, 0, x, y, 0);
+ static int
-                                 pMB->pmvs[0].x = pMB->qmvs[0].x - predMV.x;
+ FrameCodeP(Encoder * pEnc, Bitstream * bs)
-                                 pMB->pmvs[0].y = pMB->qmvs[0].y - predMV.y;
+ {
-                                 DPRINTF(DPRINTF_MV,"mv_diff (%i,%i) pred (%i,%i) result (%i,%i)", pMB->pmvs[0].x, pMB->pmvs[0].y, predMV.x, predMV.y, pMB->mvs[0].x, pMB->mvs[0].y);
+         int bits = BitstreamPos(bs);
-                         } else {
-                                 VECTOR predMV = get_pmv2(pEnc->current->mbs, pEnc->mbParam.mb_width, 0, x, y, 0);
+         FRAMEINFO *const current = pEnc->current;
-                                 pMB->pmvs[0].x = pMB->mvs[0].x - predMV.x;
+         FRAMEINFO *const reference = pEnc->reference;
-                                 pMB->pmvs[0].y = pMB->mvs[0].y - predMV.y;
+         MBParam * const pParam = &pEnc->mbParam;
-                                 DPRINTF(DPRINTF_MV,"mv_diff (%i,%i) pred (%i,%i) result (%i,%i)", pMB->pmvs[0].x, pMB->pmvs[0].y, predMV.x, predMV.y, pMB->mvs[0].x, pMB->mvs[0].y);
+         int mb_width = pParam->mb_width;
+         int mb_height = pParam->mb_height;
+         int coded = 1;
+         int k = 0, bound = 0, num_slices = pEnc->num_slices;
+         int num_threads = MAX(1, MIN(pEnc->num_threads, num_slices));
+         void * status = NULL;
+         int slices_per_thread = (num_slices*1024 / num_threads);
+         int threads_per_slice = (pEnc->num_threads*1024 / num_threads);
+         IMAGE *pRef = &reference->image;
+         if (!reference->is_edged) {
+                 start_timer();
+                 image_setedges(pRef, pParam->edged_width, pParam->edged_height,
+                                            pParam->width, pParam->height, 0);
+                 stop_edges_timer();
+                 reference->is_edged = 1;
                          }
+         pParam->m_rounding_type = 1 - pParam->m_rounding_type;
+         current->rounding_type = pParam->m_rounding_type;
+         current->fcode = pParam->m_fcode;
-                         if (pMB->mode == MODE_INTER4V)
+         if ((current->vop_flags & XVID_VOP_HALFPEL)) {
-                         {       int k;
+                 if (reference->is_interpolated != current->rounding_type) {
-                                 for (k=1;k<4;k++)
+                         start_timer();
-                                 {
+                         image_interpolate(pRef->y, pEnc->vInterH.y, pEnc->vInterV.y,
-                                         if(pEnc->mbParam.m_quarterpel) {
+                                                           pEnc->vInterHV.y, pParam->edged_width,
-                                                 VECTOR predMV = get_qpmv2(pEnc->current->mbs, pEnc->mbParam.mb_width, 0, x, y, k);
+                                                           pParam->edged_height,
-                                                 pMB->pmvs[k].x = pMB->qmvs[k].x - predMV.x;
+                                                           (pParam->vol_flags & XVID_VOL_QUARTERPEL),
-                                                 pMB->pmvs[k].y = pMB->qmvs[k].y - predMV.y;
+                                                           current->rounding_type);
-                                 DPRINTF(DPRINTF_MV,"mv_diff (%i,%i) pred (%i,%i) result (%i,%i)", pMB->pmvs[k].x, pMB->pmvs[k].y, predMV.x, predMV.y, pMB->mvs[k].x, pMB->mvs[k].y);
+                         stop_inter_timer();
-                                         } else {
+                         reference->is_interpolated = current->rounding_type;
-                                                 VECTOR predMV = get_pmv2(pEnc->current->mbs, pEnc->mbParam.mb_width, 0, x, y, k);
+                 }
-                                                 pMB->pmvs[k].x = pMB->mvs[k].x - predMV.x;
+         }
-                                                 pMB->pmvs[k].y = pMB->mvs[k].y - predMV.y;
-                                 DPRINTF(DPRINTF_MV,"mv_diff (%i,%i) pred (%i,%i) result (%i,%i)", pMB->pmvs[k].x, pMB->pmvs[k].y, predMV.x, predMV.y, pMB->mvs[k].x, pMB->mvs[k].y);
+         current->sStat.iTextBits = current->sStat.iMvSum = current->sStat.iMvCount =
+                 current->sStat.kblks = current->sStat.mblks = current->sStat.ublks =
+                 current->sStat.iMVBits = 0;
+         current->coding_type = P_VOP;
+         if (current->vop_flags & XVID_VOP_RD_PSNRHVSM) {
+                 image_block_variance(&current->image, pParam->edged_width, current->mbs,
+                                      pParam->mb_width, pParam->mb_height);
                                          }
+         call_plugins(pEnc, pEnc->current, NULL, XVID_PLG_FRAME, NULL, NULL, NULL);
+         SetMacroblockQuants(&pEnc->mbParam, current);
+         start_timer();
+         if (current->vol_flags & XVID_VOL_GMC)  /* GMC only for S(GMC)-VOPs */
+         {       int gmcval;
+                 current->warp = GlobalMotionEst( current->mbs, pParam, current, reference,
+                                                                  &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV, num_slices);
+                 if (current->motion_flags & XVID_ME_GME_REFINE) {
+                         gmcval = GlobalMotionEstRefine(&current->warp,
+                                                                                    current->mbs, pParam,
+                                                                                    current, reference,
+                                                                                    &current->image,
+                                                                                    &reference->image,
+                                                                                    &pEnc->vInterH,
+                                                                                    &pEnc->vInterV,
+                                                                                    &pEnc->vInterHV);
+                 } else {
+                         gmcval = globalSAD(&current->warp, pParam, current->mbs,
+                                                            current,
+                                                            &reference->image,
+                                                            &current->image,
+                                                            pEnc->vGMC.y);
+                 }
+                 gmcval += /*current->quant*/ 2 * (int)(pParam->mb_width*pParam->mb_height);
+                 /* 1st '3': 3 warpoints, 2nd '3': 16th pel res (2<<3) */
+                 generate_GMCparameters( 3, 3, &current->warp,
+                                 pParam->width, pParam->height,
+                                 &current->new_gmc_data);
+                 if ( (gmcval<0) && ( (current->warp.duv[1].x != 0) || (current->warp.duv[1].y != 0) ||
+                          (current->warp.duv[2].x != 0) || (current->warp.duv[2].y != 0) ) )
+                 {
+                         current->coding_type = S_VOP;
+                         generate_GMCimage(&current->new_gmc_data, &reference->image,
+                                 pParam->mb_width, pParam->mb_height,
+                                 pParam->edged_width, pParam->edged_width/2,
+                                 pParam->m_fcode, ((pParam->vol_flags & XVID_VOL_QUARTERPEL)?1:0), 0,
+                                 current->rounding_type, current->mbs, &pEnc->vGMC);
+                 } else {
+                         generate_GMCimage(&current->new_gmc_data, &reference->image,
+                                 pParam->mb_width, pParam->mb_height,
+                                 pParam->edged_width, pParam->edged_width/2,
+                                 pParam->m_fcode, ((pParam->vol_flags & XVID_VOL_QUARTERPEL)?1:0), 0,
+                                 current->rounding_type, current->mbs, NULL);    /* no warping, just AMV */
                                  }
                          }
-                         MBCoding(pEnc->current, pMB, qcoeff, bs, &pEnc->current->sStat);
+         if (pEnc->num_threads > 0) {
-                         stop_coding_timer();
+                 /* multithreaded motion estimation - dispatch threads */
+                 while (k < pEnc->num_threads) {
+                         int i, add_s = (slices_per_thread + 512) >> 10;
+                         int add_t = (threads_per_slice + 512) >> 10;
+                         int start_y = (bound * mb_height + (num_slices-1)) / num_slices;
+                         int stop_y = ((bound+add_s) * mb_height + (num_slices-1)) / num_slices;
+                         int rows_per_thread = (stop_y - start_y + add_t - 1) / add_t;
+                         slices_per_thread += ((num_slices*1024 / num_threads) - add_s*1024);
+                         threads_per_slice += ((pEnc->num_threads*1024 / num_threads) - add_t*1024);
+                         for (i = 0; i < add_t; i++) {
+                                 memset(pEnc->smpData[k+i].complete_count_self, 0, rows_per_thread * sizeof(int));
+                                 pEnc->smpData[k+i].pEnc = (void *) pEnc;
+                                 pEnc->smpData[k+i].y_row = i;
+                                 pEnc->smpData[k+i].y_step = add_t;
+                                 pEnc->smpData[k+i].stop_y = stop_y;
+                                 pEnc->smpData[k+i].start_y = start_y;
+                                 /* todo: sort out temp space once and for all */
+                                 pEnc->smpData[k+i].RefQ = (((k+i)&1) ? pEnc->vInterV.u : pEnc->vInterV.v) +
+*((k+i)>>1)*pParam->edged_width;
                  }
+                         pEnc->smpData[k].complete_count_above =
+                                 pEnc->smpData[k+add_t-1].complete_count_self - 1;
+                         bound += add_s;
+                         k += add_t;
          }
-         if ((pEnc->current->global_flags & XVID_REDUCED))
+                 for (k = 1; k < pEnc->num_threads; k++) {
-         {
+                         pthread_create(&pEnc->smpData[k].handle, NULL,
-                 image_deblock_rrv(&pEnc->current->image, pEnc->mbParam.edged_width,
+                                 (void*)MotionEstimateSMP, (void*)&pEnc->smpData[k]);
-                         pEnc->current->mbs, mb_width, mb_height, pEnc->mbParam.mb_width,
-, XVID_DEC_DEBLOCKY|XVID_DEC_DEBLOCKUV);
          }
-         emms();
+                 MotionEstimateSMP(&pEnc->smpData[0]);
-         if (pEnc->current->global_flags & XVID_HINTEDME_GET) {
+                 for (k = 1; k < pEnc->num_threads; k++) {
-                 HintedMEGet(pEnc, 0);
+                         pthread_join(pEnc->smpData[k].handle, &status);
          }
-         if (pEnc->current->sStat.iMvCount == 0)
+                 current->fcode = 0;
-                 pEnc->current->sStat.iMvCount = 1;
+                 for (k = 0; k < pEnc->num_threads; k++) {
+                         current->sStat.iMvSum += pEnc->smpData[k].mvSum;
+                         current->sStat.iMvCount += pEnc->smpData[k].mvCount;
+                         if (pEnc->smpData[k].minfcode > current->fcode)
+                                 current->fcode = pEnc->smpData[k].minfcode;
+                 }
-         fSigma = (float) sqrt((float) pEnc->current->sStat.iMvSum / pEnc->current->sStat.iMvCount);
+         } else {
-         iSearchRange = 1 << (3 + pEnc->mbParam.m_fcode);
+                 /* regular ME */
+                 MotionEstimation(&pEnc->mbParam, current, reference,
+                                                  &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV,
+                                                  &pEnc->vGMC, 256*4096, num_slices);
-         if ((fSigma > iSearchRange / 3)
-                 && (pEnc->mbParam.m_fcode <= (3 + pEnc->mbParam.m_quarterpel))) // maximum search range 128
-         {
-                 pEnc->mbParam.m_fcode++;
-                 iSearchRange *= 2;
-         } else if ((fSigma < iSearchRange / 6)
-                            && (pEnc->fMvPrevSigma >= 0)
-                            && (pEnc->fMvPrevSigma < iSearchRange / 6)
-                         && (pEnc->mbParam.m_fcode >= (2 + pEnc->mbParam.m_quarterpel))) // minimum search range 16
-         {
-                 pEnc->mbParam.m_fcode--;
-                 iSearchRange /= 2;
          }
-         pEnc->fMvPrevSigma = fSigma;
+         stop_motion_timer();
+         set_timecodes(current,reference,pParam->fbase);
+         BitstreamWriteVopHeader(bs, &pEnc->mbParam, current, 1, current->mbs[0].quant);
+         /* multithreaded inter coding - dispatch threads */
+         bound = 0;
+         slices_per_thread = (num_slices*1024 / num_threads);
+         for (k = 0; k < num_threads; k++) {
+                 int add = ((slices_per_thread + 512) >> 10);
+                 slices_per_thread += ((num_slices*1024 / num_threads) - add*1024);
+                 pEnc->smpData[k].pEnc = (void *) pEnc;
+                 pEnc->smpData[k].stop_y = (((bound+add) * mb_height + (num_slices-1)) / num_slices);
+                 pEnc->smpData[k].start_y = ((bound * mb_height + (num_slices-1)) / num_slices);
+                 pEnc->smpData[k].RefQ = ((k&1) ? pEnc->vInterV.u : pEnc->vInterV.v) + 16*(k>>1)*pParam->edged_width;
+                 bound += add;
+                 if (k > 0) {
+                         pEnc->smpData[k].sStat->iTextBits = pEnc->smpData[k].sStat->kblks =
+                         pEnc->smpData[k].sStat->mblks = pEnc->smpData[k].sStat->ublks =
+                         pEnc->smpData[k].sStat->iMVBits = 0;
+                         BitstreamReset(pEnc->smpData[k].bs);
+                 }
+         }
+         pEnc->smpData[0].bs = bs;
+         pEnc->smpData[0].sStat = &current->sStat;
+         /* create threads */
+         for (k = 1; k < num_threads; k++) {
+                 pthread_create(&pEnc->smpData[k].handle, NULL,
+                         (void*)SliceCodeP, (void*)&pEnc->smpData[k]);
+         }
+         SliceCodeP(&pEnc->smpData[0]);
+         /* wait until all threads are finished */
+         for (k = 1; k < num_threads; k++) {
+                 pthread_join(pEnc->smpData[k].handle, &status);
+         }
+         current->length = BitstreamLength(bs) - (bits/8);
+         /* reassemble the pieces together */
+         SerializeBitstreams(pEnc, pEnc->current, bs, num_threads);
+         updateFcode(&current->sStat, pEnc);
          /* frame drop code */
-         DPRINTF(DPRINTF_DEBUG, "kmu %i %i %i", pEnc->current->sStat.kblks, pEnc->current->sStat.mblks, pEnc->current->sStat.ublks);
+ #if 0
-         if (pEnc->current->sStat.kblks + pEnc->current->sStat.mblks <
+         DPRINTF(XVID_DEBUG_DEBUG, "kmu %i %i %i\n", current->sStat.kblks, current->sStat.mblks, current->sStat.ublks);
-                 (pEnc->mbParam.frame_drop_ratio * mb_width * mb_height) / 100)
+ #endif
+         if (current->sStat.kblks + current->sStat.mblks <
+                 (pParam->frame_drop_ratio * mb_width * mb_height) / 100 &&
+                 ( (pEnc->bframenum_head >= pEnc->bframenum_tail) || !(pEnc->mbParam.global_flags & XVID_GLOBAL_CLOSED_GOP)) )
          {
-                 pEnc->current->sStat.kblks = pEnc->current->sStat.mblks = 0;
+                 current->sStat.kblks = current->sStat.mblks = current->sStat.iTextBits = 0;
-                 pEnc->current->sStat.ublks = mb_width * mb_height;
+                 current->sStat.ublks = mb_width * mb_height;
                  BitstreamReset(bs);
-                 set_timecodes(pEnc->current,pEnc->reference,pEnc->mbParam.fbase);
+                 set_timecodes(current,reference,pParam->fbase);
-                 BitstreamWriteVopHeader(bs, &pEnc->mbParam, pEnc->current, 0);
+                 BitstreamWriteVopHeader(bs, &pEnc->mbParam, current, 0, current->mbs[0].quant);
+                 /* copy reference frame details into the current frame */
+                 current->quant = reference->quant;
+                 current->motion_flags = reference->motion_flags;
+                 current->rounding_type = reference->rounding_type;
+                 current->fcode = reference->fcode;
+                 current->bcode = reference->bcode;
+                 current->stamp = reference->stamp;
+                 image_copy(&current->image, &reference->image, pParam->edged_width, pParam->height);
+                 memcpy(current->mbs, reference->mbs, sizeof(MACROBLOCK) * mb_width * mb_height);
+                 coded = 0;
+                 BitstreamPadAlways(bs); /* next_start_code() at the end of VideoObjectPlane() */
+                 current->length = (BitstreamPos(bs) - bits) / 8;
+         } else {
+                 pEnc->current->is_edged = 0; /* not edged */
+                 pEnc->current->is_interpolated = -1; /* not interpolated (fake rounding -1) */
-                 // copy reference frame details into the current frame
+                 /* what was this frame's interpolated reference will become
-                 pEnc->current->quant = pEnc->reference->quant;
+                         forward (past) reference in b-frame coding */
-                 pEnc->current->motion_flags = pEnc->reference->motion_flags;
-                 pEnc->current->rounding_type = pEnc->reference->rounding_type;
+                 image_swap(&pEnc->vInterH, &pEnc->f_refh);
-                 pEnc->current->quarterpel =  pEnc->reference->quarterpel;
+                 image_swap(&pEnc->vInterV, &pEnc->f_refv);
-                 pEnc->current->fcode = pEnc->reference->fcode;
+                 image_swap(&pEnc->vInterHV, &pEnc->f_refhv);
-                 pEnc->current->bcode = pEnc->reference->bcode;
-                 image_copy(&pEnc->current->image, &pEnc->reference->image, pEnc->mbParam.edged_width, pEnc->mbParam.height);
-                 memcpy(pEnc->current->mbs, pEnc->reference->mbs, sizeof(MACROBLOCK) * mb_width * mb_height);
          }
          /* XXX: debug
          {
                  char s[100];
                  sprintf(s, "\\%05i_cur.pgm", pEnc->m_framenum);
-                 image_dump_yuvpgm(&pEnc->current->image,
+                 image_dump_yuvpgm(&current->image,
-                         pEnc->mbParam.edged_width,
+                         pParam->edged_width,
-                         pEnc->mbParam.width, pEnc->mbParam.height, s);
+                         pParam->width, pParam->height, s);
                  sprintf(s, "\\%05i_ref.pgm", pEnc->m_framenum);
-                 image_dump_yuvpgm(&pEnc->reference->image,
+                 image_dump_yuvpgm(&reference->image,
-                         pEnc->mbParam.edged_width,
+                         pParam->edged_width,
-                         pEnc->mbParam.width, pEnc->mbParam.height, s);
+                         pParam->width, pParam->height, s);
          }
          */
+         return coded;
+ }
+ static void
+ SliceCodeB(SMPData *data)
+ {
+         Encoder *pEnc = (Encoder *) data->pEnc;
+         Bitstream *bs = (Bitstream *) data->bs;
+         DECLARE_ALIGNED_MATRIX(dct_codes, 6, 64, int16_t, CACHE_LINE);
+         DECLARE_ALIGNED_MATRIX(qcoeff, 6, 64, int16_t, CACHE_LINE);
+         int x, y;
+         FRAMEINFO * const frame = (FRAMEINFO * const) data->current;
+         MBParam * const pParam = &pEnc->mbParam;
+         int mb_width = pParam->mb_width;
+         int mb_height = pParam->mb_height;
+         IMAGE *f_ref = &pEnc->reference->image;
+         IMAGE *b_ref = &pEnc->current->image;
+         int bound = data->start_y*mb_width;
+         int num_slices = pEnc->num_slices;
+         if (data->start_y > 0) { /* write resync marker */
+                 write_video_packet_header(bs, pParam, frame, bound);
+         }
+         for (y = data->start_y; y < data->stop_y; y++) {
+                 int new_bound = mb_width * ((((y*num_slices) / mb_height) * mb_height + (num_slices-1)) / num_slices);
+                 if (new_bound > bound) {
+                         bound = new_bound;
+                         BitstreamPadAlways(bs);
+                         write_video_packet_header(bs, pParam, frame, bound);
+                 }
+                 for (x = 0; x < mb_width; x++) {
+                         MACROBLOCK * const mb = &frame->mbs[x + y * pEnc->mbParam.mb_width];
+                         /* decoder ignores mb when refence block is INTER(0,0), CBP=0 */
+                         if (mb->mode == MODE_NOT_CODED) {
+                                 if (pParam->plugin_flags & XVID_REQORIGINAL) {
+                                         MBMotionCompensation(mb, x, y, f_ref, NULL, f_ref, NULL, NULL, &frame->image,
+                                                                                  NULL, 0, 0, pParam->edged_width, 0, 0, data->RefQ);
+                                 }
+                                 continue;
+                         }
+                         mb->quant = frame->quant;
+                         if (mb->cbp != 0 || pParam->plugin_flags & XVID_REQORIGINAL) {
+                                 /* we have to motion-compensate, transfer etc,
+                                         because there might be blocks to code */
+                                 MBMotionCompensationBVOP(pParam, mb, x, y, &frame->image,
+                                                                                  f_ref, &pEnc->f_refh, &pEnc->f_refv,
+                                                                                  &pEnc->f_refhv, b_ref, &pEnc->vInterH,
+                                                                                  &pEnc->vInterV, &pEnc->vInterHV, dct_codes,
+                                                                                  data->RefQ);
+                                 mb->cbp = MBTransQuantInterBVOP(pParam, frame, mb, x, y,  dct_codes, qcoeff);
+                         }
+                         if (mb->mode == MODE_DIRECT_NO4V)
+                                 mb->mode = MODE_DIRECT;
-         *pBits = BitstreamPos(bs) - *pBits;
+                         if (mb->mode == MODE_DIRECT && (mb->cbp | mb->pmvs[3].x | mb->pmvs[3].y) == 0)
+                                 mb->mode = MODE_DIRECT_NONE_MV; /* skipped */
+                         else
+                                 if (frame->vop_flags & XVID_VOP_GREYSCALE)
+                                         /* keep only bits 5-2 -- Chroma blocks will just be skipped by MBCodingBVOP */
+                                         mb->cbp &= 0x3C;
-         return 0;                                       // inter
+                         start_timer();
+                         MBCodingBVOP(frame, mb, qcoeff, frame->fcode, frame->bcode, bs, data->sStat);
+                         stop_coding_timer();
+                 }
  }
+         BitstreamPadAlways(bs); /* next_start_code() at the end of VideoObjectPlane() */
+         emms();
+ }
  static void
  FrameCodeB(Encoder * pEnc,
                     FRAMEINFO * frame,
-                    Bitstream * bs,
+                    Bitstream * bs)
-                    uint32_t * pBits)
  {
-         DECLARE_ALIGNED_MATRIX(dct_codes, 6, 64, int16_t, CACHE_LINE);
+         int bits = BitstreamPos(bs);
-         DECLARE_ALIGNED_MATRIX(qcoeff, 6, 64, int16_t, CACHE_LINE);
+         int k = 0, bound = 0, num_slices = pEnc->num_slices;
-         uint32_t x, y;
+         int num_threads = MAX(1, MIN(pEnc->num_threads, num_slices));
+         void * status = NULL;
+         int slices_per_thread = (num_slices*1024 / num_threads);
+         int threads_per_slice = (pEnc->num_threads*1024 / num_threads);
          IMAGE *f_ref = &pEnc->reference->image;
          IMAGE *b_ref = &pEnc->current->image;
+         MBParam * const pParam = &pEnc->mbParam;
+         int mb_width = pParam->mb_width;
+         int mb_height = pParam->mb_height;
  #ifdef BFRAMES_DEC_DEBUG
          FILE *fp;
          static char first=0;
-Line 2036
+Line 2291
                  fprintf(fp,"Y=%3d   X=%3d   MB=%2d   CBP=%02X\n",y,x,mb->mode,mb->cbp); \
          }
-         pEnc->current->global_flags &= ~XVID_REDUCED;   /* reduced resoltion not yet supported */
          if (!first){
                  fp=fopen("C:\\XVIDDBGE.TXT","w");
          }
  #endif
-         frame->quarterpel =  pEnc->mbParam.m_quarterpel;
+         /* forward  */
+         if (!pEnc->reference->is_edged) {
-         // forward
          image_setedges(f_ref, pEnc->mbParam.edged_width,
                                     pEnc->mbParam.edged_height, pEnc->mbParam.width,
-                                    pEnc->mbParam.height);
+                                            pEnc->mbParam.height, 0);
+                 pEnc->current->is_edged = 1;
+         }
+         if (pEnc->reference->is_interpolated != 0) {
          start_timer();
-         image_interpolate(f_ref, &pEnc->f_refh, &pEnc->f_refv, &pEnc->f_refhv,
+                 image_interpolate(f_ref->y, pEnc->f_refh.y, pEnc->f_refv.y, pEnc->f_refhv.y,
                                            pEnc->mbParam.edged_width, pEnc->mbParam.edged_height,
-                                           pEnc->mbParam.m_quarterpel, 0);
+                                                   (pEnc->mbParam.vol_flags & XVID_VOL_QUARTERPEL), 0);
          stop_inter_timer();
+                 pEnc->reference->is_interpolated = 0;
+         }
-         // backward
+         /* backward */
+         if (!pEnc->current->is_edged) {
          image_setedges(b_ref, pEnc->mbParam.edged_width,
                                     pEnc->mbParam.edged_height, pEnc->mbParam.width,
-                                    pEnc->mbParam.height);
+                                            pEnc->mbParam.height, 0);
+                 pEnc->current->is_edged = 1;
+         }
+         if (pEnc->current->is_interpolated != 0) {
          start_timer();
-         image_interpolate(b_ref, &pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV,
+                 image_interpolate(b_ref->y, pEnc->vInterH.y, pEnc->vInterV.y, pEnc->vInterHV.y,
                                            pEnc->mbParam.edged_width, pEnc->mbParam.edged_height,
-                                           pEnc->mbParam.m_quarterpel, 0);
+                                                 (pEnc->mbParam.vol_flags & XVID_VOL_QUARTERPEL), 0);
          stop_inter_timer();
+                 pEnc->current->is_interpolated = 0;
+         }
+         frame->coding_type = B_VOP;
+         if (pEnc->current->vop_flags & XVID_VOP_RD_PSNRHVSM) {
+                 image_block_variance(&pEnc->current->image, pEnc->mbParam.edged_width, pEnc->current->mbs,
+                                      pEnc->mbParam.mb_width, pEnc->mbParam.mb_height);
+         }
+         call_plugins(pEnc, frame, NULL, XVID_PLG_FRAME, NULL, NULL, NULL);
+         frame->fcode = frame->bcode = pEnc->current->fcode;
          start_timer();
+         if (pEnc->num_threads > 0) {
+                 /* multithreaded motion estimation - dispatch threads */
+                 while (k < pEnc->num_threads) {
+                         int i, add_s = (slices_per_thread + 512) >> 10;
+                         int add_t = (threads_per_slice + 512) >> 10;
+                         int start_y = (bound * mb_height + (num_slices-1)) / num_slices;
+                         int stop_y = ((bound+add_s) * mb_height + (num_slices-1)) / num_slices;
+                         int rows_per_thread = (stop_y - start_y + add_t - 1) / add_t;
+                         slices_per_thread += ((num_slices*1024 / num_threads) - add_s*1024);
+                         threads_per_slice += ((pEnc->num_threads*1024 / num_threads) - add_t*1024);
+                         for (i = 0; i < add_t; i++) {
+                                 memset(pEnc->smpData[k+i].complete_count_self, 0, rows_per_thread * sizeof(int));
+                                 pEnc->smpData[k+i].pEnc = (void *) pEnc;
+                                 pEnc->smpData[k+i].current = frame;
+                                 pEnc->smpData[k+i].y_row = i;
+                                 pEnc->smpData[k+i].y_step = add_t;
+                                 pEnc->smpData[k+i].stop_y = stop_y;
+                                 pEnc->smpData[k+i].start_y = start_y;
+                                 /* todo: sort out temp space once and for all */
+                                 pEnc->smpData[k+i].RefQ = (((k+i)&1) ? pEnc->vInterV.u : pEnc->vInterV.v) +
+*((k+i)>>1)*pParam->edged_width;
+                         }
+                         pEnc->smpData[k].complete_count_above =
+                                 pEnc->smpData[k+add_t-1].complete_count_self - 1;
+                         bound += add_s;
+                         k += add_t;
+                 }
+                 for (k = 1; k < pEnc->num_threads; k++) {
+                         pthread_create(&pEnc->smpData[k].handle, NULL,
+                                 (void*)SMPMotionEstimationBVOP, (void*)&pEnc->smpData[k]);
+                 }
+                 SMPMotionEstimationBVOP(&pEnc->smpData[0]);
+                 for (k = 1; k < pEnc->num_threads; k++) {
+                         pthread_join(pEnc->smpData[k].handle, &status);
+                 }
+                 frame->fcode = frame->bcode = 0;
+                 for (k = 0; k < pEnc->num_threads; k++) {
+                         if (pEnc->smpData[k].minfcode > frame->fcode)
+                                 frame->fcode = pEnc->smpData[k].minfcode;
+                         if (pEnc->smpData[k].minbcode > frame->bcode)
+                                 frame->bcode = pEnc->smpData[k].minbcode;
+                 }
+         } else {
          MotionEstimationBVOP(&pEnc->mbParam, frame,
-                 ((int32_t)(pEnc->current->stamp - frame->stamp)),                               // time_bp
+                                                          ((int32_t)(pEnc->current->stamp - frame->stamp)),                              /* time_bp */
-                 ((int32_t)(pEnc->current->stamp - pEnc->reference->stamp)),     // time_pp
+                                                          ((int32_t)(pEnc->current->stamp - pEnc->reference->stamp)),    /* time_pp */
                          pEnc->reference->mbs, f_ref,
                                                   &pEnc->f_refh, &pEnc->f_refv, &pEnc->f_refhv,
                                                   pEnc->current, b_ref, &pEnc->vInterH,
                                                   &pEnc->vInterV, &pEnc->vInterHV);
+         }
          stop_motion_timer();
-         /*if (test_quant_type(&pEnc->mbParam, pEnc->current))
-            {
-            BitstreamWriteVolHeader(bs, pEnc->mbParam.width, pEnc->mbParam.height, pEnc->mbParam.quant_type);
-            } */
-         frame->coding_type = B_VOP;
          set_timecodes(frame, pEnc->reference,pEnc->mbParam.fbase);
-         BitstreamWriteVopHeader(bs, &pEnc->mbParam, frame, 1);
+         BitstreamWriteVopHeader(bs, &pEnc->mbParam, frame, 1, frame->quant);
-         *pBits = BitstreamPos(bs);
+         /* reset stats */
          frame->sStat.iTextBits = 0;
+         frame->sStat.iMVBits = 0;
          frame->sStat.iMvSum = 0;
          frame->sStat.iMvCount = 0;
          frame->sStat.kblks = frame->sStat.mblks = frame->sStat.ublks = 0;
+         frame->sStat.mblks = pEnc->mbParam.mb_width * pEnc->mbParam.mb_height;
+         frame->sStat.kblks = frame->sStat.ublks = 0;
+         /* multithreaded inter coding - dispatch threads */
+         bound = 0;
+         slices_per_thread = (num_slices*1024 / num_threads);
-         for (y = 0; y < pEnc->mbParam.mb_height; y++) {
+         for (k = 0; k < num_threads; k++) {
-                 for (x = 0; x < pEnc->mbParam.mb_width; x++) {
+                 int add = ((slices_per_thread + 512) >> 10);
-                         MACROBLOCK * const mb = &frame->mbs[x + y * pEnc->mbParam.mb_width];
-                         int direction = pEnc->mbParam.global & XVID_ALTERNATESCAN ? 2 : 0;
-                         // decoder ignores mb when refence block is INTER(0,0), CBP=0
-                         if (mb->mode == MODE_NOT_CODED) {
-                                 //mb->mvs[0].x = mb->mvs[0].y = mb->cbp = 0;
-                                 continue;
-                         }
-                         if (mb->mode != MODE_DIRECT_NONE_MV) {
+                 slices_per_thread += ((num_slices*1024 / num_threads) - add*1024);
-                                 MBMotionCompensationBVOP(&pEnc->mbParam, mb, x, y, &frame->image,
-                                                                          f_ref, &pEnc->f_refh, &pEnc->f_refv,
-                                                                          &pEnc->f_refhv, b_ref, &pEnc->vInterH,
-                                                                          &pEnc->vInterV, &pEnc->vInterHV,
-                                                                          dct_codes);
-                                 if (mb->mode == MODE_DIRECT_NO4V) mb->mode = MODE_DIRECT;
+                 pEnc->smpData[k].pEnc = (void *) pEnc;
-                                 mb->quant = frame->quant;
+                 pEnc->smpData[k].current = frame;
+                 pEnc->smpData[k].stop_y = (((bound+add) * mb_height + (num_slices-1)) / num_slices);
+                 pEnc->smpData[k].start_y = ((bound * mb_height + (num_slices-1)) / num_slices);
+                 bound += add;
-                                 mb->cbp =
+                 /* todo: sort out temp space once and for all */
-                                         MBTransQuantInterBVOP(&pEnc->mbParam, frame, mb, dct_codes, qcoeff);
+                 pEnc->smpData[k].RefQ = ((k&1) ? pEnc->vInterV.u : pEnc->vInterV.v) + 16*(k>>1)*pParam->edged_width;
-                                 if ( (mb->mode == MODE_DIRECT) && (mb->cbp == 0)
+                 if (k > 0) {
-                                         && (mb->pmvs[3].x == 0) && (mb->pmvs[3].y == 0) ) {
+                         BitstreamReset(pEnc->smpData[k].bs);
-                                         mb->mode = MODE_DIRECT_NONE_MV; // skipped
+                         pEnc->smpData[k].sStat->iTextBits = pEnc->smpData[k].sStat->kblks =
+                         pEnc->smpData[k].sStat->mblks = pEnc->smpData[k].sStat->ublks = pEnc->smpData[k].sStat->iMVBits = 0;
                                  }
                          }
- #ifdef BFRAMES_DEC_DEBUG
+         for (k = 1; k < num_threads; k++) {
-         BFRAME_DEBUG
+                 pthread_create(&pEnc->smpData[k].handle, NULL,
- #endif
+                         (void*)SliceCodeB, (void*)&pEnc->smpData[k]);
-                         start_timer();
-                         MBCodingBVOP(mb, qcoeff, frame->fcode, frame->bcode, bs,
-                                                  &frame->sStat, direction);
-                         stop_coding_timer();
-                 }
          }
-         emms();
+         pEnc->smpData[0].bs = bs;
+         pEnc->smpData[0].sStat = &frame->sStat;
+         SliceCodeB(&pEnc->smpData[0]);
+         for (k = 1; k < num_threads; k++) {
+                 pthread_join(pEnc->smpData[k].handle, &status);
+         }
-         // TODO: dynamic fcode/bcode ???
+         frame->length = BitstreamLength(bs) - (bits/8);
-         *pBits = BitstreamPos(bs) - *pBits;
+         /* reassemble the pieces together */
+         SerializeBitstreams(pEnc, frame, bs, num_threads);
  #ifdef BFRAMES_DEC_DEBUG
          if (!first){

 Legend:



Removed from v.1.76.2.37
 


changed lines


 
Added in v.1.134
 Legend:



Removed from v.1.76.2.37
 


changed lines


 
Added in v.1.134
-Removed from v.1.76.2.37
+Added in v.1.134

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