[svn] / trunk / vfw / src / codec.c Repository:
ViewVC logotype

Diff of /trunk/vfw/src/codec.c

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 61, Sat Mar 23 06:58:56 2002 UTC revision 235, Sun Jun 23 03:59:49 2002 UTC
# Line 23  Line 23 
23   *   *
24   *      History:   *      History:
25   *   *
26     *      23.06.2002      XVID_CPU_CHKONLY; loading speed up
27     *      25.04.2002      ICDECOMPRESS_PREROLL
28     *      17.04.2002      re-enabled lumi masking for 1st pass
29     *      15.04.2002      updated cbr support
30     *      04.04.2002      separated 2-pass code to 2pass.c
31     *                              interlacing support
32     *                              hinted ME support
33   *      23.03.2002      daniel smith <danielsmith@astroboymail.com>   *      23.03.2002      daniel smith <danielsmith@astroboymail.com>
34   *                              changed inter4v to only be in modes 5 or 6   *                              changed inter4v to only be in modes 5 or 6
35   *                              fixed null mode crash ?   *                              fixed null mode crash ?
# Line 43  Line 50 
50  #include <vfw.h>  #include <vfw.h>
51    
52  #include "codec.h"  #include "codec.h"
53  #include <math.h>       // fabs()  #include "2pass.h"
54    
55  int pmvfast_presets[7] = {  int pmvfast_presets[7] = {
56          0, PMV_QUICKSTOP16, PMV_EARLYSTOP16, PMV_EARLYSTOP16 | PMV_EARLYSTOP8,          0, PMV_QUICKSTOP16, PMV_EARLYSTOP16, PMV_EARLYSTOP16 | PMV_EARLYSTOP8,
# Line 212  Line 219 
219    
220  LRESULT compress_get_size(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)  LRESULT compress_get_size(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)
221  {  {
222          return lpbiOutput->bmiHeader.biWidth * lpbiOutput->bmiHeader.biHeight * 3;          return
223    #ifdef BFRAMES
224             2 *
225    #endif
226            lpbiOutput->bmiHeader.biWidth * lpbiOutput->bmiHeader.biHeight * 3;
227  }  }
228    
229    
# Line 233  Line 244 
244          switch (codec->config.mode)          switch (codec->config.mode)
245          {          {
246          case DLG_MODE_CBR :          case DLG_MODE_CBR :
247                  param.bitrate = codec->config.bitrate;                  param.rc_bitrate = codec->config.rc_bitrate;
248                  param.rc_buffersize = codec->config.rc_buffersize;                  param.rc_reaction_delay_factor = codec->config.rc_reaction_delay_factor;
249                    param.rc_averaging_period = codec->config.rc_averaging_period;
250                    param.rc_buffer = codec->config.rc_buffer;
251                  break;                  break;
252    
253          case DLG_MODE_VBR_QUAL :          case DLG_MODE_VBR_QUAL :
254                  codec->config.fquant = 0;                  codec->config.fquant = 0;
255                  param.bitrate = 0;                  param.rc_bitrate = 0;
256                  break;                  break;
257    
258          case DLG_MODE_VBR_QUANT :          case DLG_MODE_VBR_QUANT :
259                  codec->config.fquant = (float) codec->config.quant;                  codec->config.fquant = (float) codec->config.quant;
260                  param.bitrate = 0;                  param.rc_bitrate = 0;
261                  break;                  break;
262    
263          case DLG_MODE_2PASS_1 :          case DLG_MODE_2PASS_1 :
264          case DLG_MODE_2PASS_2_INT :          case DLG_MODE_2PASS_2_INT :
265          case DLG_MODE_2PASS_2_EXT :          case DLG_MODE_2PASS_2_EXT :
266                  param.bitrate = 0;                  param.rc_bitrate = 0;
267                    codec->twopass.max_framesize = (int)((double)codec->config.twopass_max_bitrate / 8.0 / ((double)codec->fbase / (double)codec->fincr));
268                  break;                  break;
269    
270          case DLG_MODE_NULL :          case DLG_MODE_NULL :
# Line 276  Line 290 
290          param.fincr = codec->fincr;          param.fincr = codec->fincr;
291          param.fbase = codec->fbase;          param.fbase = codec->fbase;
292    
         param.rc_buffersize = codec->config.rc_buffersize;  
   
293          param.min_quantizer = codec->config.min_pquant;          param.min_quantizer = codec->config.min_pquant;
294          param.max_quantizer = codec->config.max_pquant;          param.max_quantizer = codec->config.max_pquant;
295          param.max_key_interval = codec->config.max_key_interval;          param.max_key_interval = codec->config.max_key_interval;
296    
297    #ifdef BFRAMES
298            param.global = 0;
299            if (codec->config.packed) param.global |= XVID_GLOBAL_PACKED;
300            if (codec->config.dx50bvop) param.global |= XVID_GLOBAL_DX50BVOP;
301            if (codec->config.debug) param.global |= XVID_GLOBAL_DEBUG;
302            param.max_bframes = codec->config.max_bframes;
303            param.bquant_ratio = codec->config.bquant_ratio;
304    #endif
305    
306          switch(xvid_encore(0, XVID_ENC_CREATE, &param, NULL))          switch(xvid_encore(0, XVID_ENC_CREATE, &param, NULL))
307          {          {
308          case XVID_ERR_FAIL :          case XVID_ERR_FAIL :
# Line 298  Line 319 
319          codec->framenum = 0;          codec->framenum = 0;
320          codec->keyspacing = 0;          codec->keyspacing = 0;
321    
322          codec->twopass.stats1 = codec->twopass.stats2 = INVALID_HANDLE_VALUE;          codec->twopass.hints = codec->twopass.stats1 = codec->twopass.stats2 = INVALID_HANDLE_VALUE;
323            codec->twopass.hintstream = NULL;
324    
325          return ICERR_OK;          return ICERR_OK;
326  }  }
# Line 309  Line 331 
331          if (codec->ehandle != NULL)          if (codec->ehandle != NULL)
332          {          {
333                  xvid_encore(codec->ehandle, XVID_ENC_DESTROY, NULL, NULL);                  xvid_encore(codec->ehandle, XVID_ENC_DESTROY, NULL, NULL);
334    
335                    if (codec->twopass.hints != INVALID_HANDLE_VALUE)
336                    {
337                            CloseHandle(codec->twopass.hints);
338                    }
339                  if (codec->twopass.stats1 != INVALID_HANDLE_VALUE)                  if (codec->twopass.stats1 != INVALID_HANDLE_VALUE)
340                  {                  {
341                          CloseHandle(codec->twopass.stats1);                          CloseHandle(codec->twopass.stats1);
# Line 317  Line 344 
344                  {                  {
345                          CloseHandle(codec->twopass.stats2);                          CloseHandle(codec->twopass.stats2);
346                  }                  }
347                    if (codec->twopass.hintstream != NULL)
348                    {
349                            free(codec->twopass.hintstream);
350                    }
351    
352                  codec->ehandle = NULL;                  codec->ehandle = NULL;
353    
354                    codec_2pass_finish(codec);
355          }          }
356    
357          return ICERR_OK;          return ICERR_OK;
# Line 349  Line 383 
383          if(codec->config.motion_search > 4)          if(codec->config.motion_search > 4)
384                  frame.general |= XVID_INTER4V;                  frame.general |= XVID_INTER4V;
385    
386          if(((codec->config.mode == DLG_MODE_2PASS_1) ? 0 : codec->config.lum_masking) == 1)          if (codec->config.lum_masking)
387                  frame.general |= XVID_LUMIMASKING;                  frame.general |= XVID_LUMIMASKING;
388    
389            if (codec->config.interlacing)
390                    frame.general |= XVID_INTERLACING;
391    // fix 1pass modes/hinted MV by koepi
392            if (codec->config.hinted_me && (codec->config.mode == DLG_MODE_CBR || codec->config.mode == DLG_MODE_VBR_QUAL || codec->config.mode == DLG_MODE_VBR_QUANT))
393            {
394                    codec->config.hinted_me = 0;
395            }
396    // end of ugly hack
397    
398            if (codec->config.hinted_me && codec->config.mode == DLG_MODE_2PASS_1)
399            {
400                    frame.hint.hintstream = codec->twopass.hintstream;
401                    frame.hint.rawhints = 0;
402                    frame.general |= XVID_HINTEDME_GET;
403            }
404            else if (codec->config.hinted_me && (codec->config.mode == DLG_MODE_2PASS_2_EXT || codec->config.mode == DLG_MODE_2PASS_2_INT))
405            {
406                    DWORD read;
407                    DWORD blocksize;
408    
409                    frame.hint.hintstream = codec->twopass.hintstream;
410                    frame.hint.rawhints = 0;
411                    frame.general |= XVID_HINTEDME_SET;
412    
413                    if (codec->twopass.hints == INVALID_HANDLE_VALUE)
414                    {
415                            codec->twopass.hints = CreateFile(codec->config.hintfile, GENERIC_READ, 0, 0, OPEN_EXISTING, 0, 0);
416                            if (codec->twopass.hints == INVALID_HANDLE_VALUE)
417                            {
418                                    DEBUGERR("couldn't open hints file");
419                                    return ICERR_ERROR;
420                            }
421                    }
422                    if (!ReadFile(codec->twopass.hints, &blocksize, sizeof(DWORD), &read, 0) || read != sizeof(DWORD) ||
423                            !ReadFile(codec->twopass.hints, frame.hint.hintstream, blocksize, &read, 0) || read != blocksize)
424                    {
425                            DEBUGERR("couldn't read from hints file");
426                            return ICERR_ERROR;
427                    }
428            }
429    
430          frame.motion = pmvfast_presets[codec->config.motion_search];          frame.motion = pmvfast_presets[codec->config.motion_search];
431    
432          frame.image = icc->lpInput;          frame.image = icc->lpInput;
# Line 429  Line 504 
504          {          {
505                  frame.intra = 1;                  frame.intra = 1;
506          }          }
507          else if ((codec->keyspacing < codec->config.min_key_interval && codec->framenum) &&          else if (codec->keyspacing < codec->config.min_key_interval && codec->framenum)
                 (codec->config.mode == DLG_MODE_2PASS_1))  
508          {          {
509                  DEBUG("current frame forced to p-frame");                  DEBUG("current frame forced to p-frame");
510                  frame.intra = 0;                  frame.intra = 0;
511          }          }
512    
513            OutputDebugString(" ");
514          switch (xvid_encore(codec->ehandle, XVID_ENC_ENCODE, &frame, &stats))          switch (xvid_encore(codec->ehandle, XVID_ENC_ENCODE, &frame, &stats))
515          {          {
516          case XVID_ERR_FAIL :          case XVID_ERR_FAIL :
# Line 465  Line 540 
540                  outhdr->biSizeImage = 0;                  outhdr->biSizeImage = 0;
541          }          }
542    
543            if (frame.general & XVID_HINTEDME_GET)
544            {
545                    DWORD wrote;
546                    DWORD blocksize = frame.hint.hintlength;
547    
548                    if (codec->twopass.hints == INVALID_HANDLE_VALUE)
549                    {
550                            codec->twopass.hints = CreateFile(codec->config.hintfile, GENERIC_WRITE, 0, 0, CREATE_ALWAYS, 0, 0);
551                            if (codec->twopass.hints        == INVALID_HANDLE_VALUE)
552                            {
553                                    DEBUGERR("couldn't create hints file");
554                                    return ICERR_ERROR;
555                            }
556                    }
557                    if (!WriteFile(codec->twopass.hints, &frame.hint.hintlength, sizeof(int), &wrote, 0) || wrote != sizeof(int) ||
558                            !WriteFile(codec->twopass.hints, frame.hint.hintstream, blocksize, &wrote, 0) || wrote != blocksize)
559                    {
560                            DEBUGERR("couldn't write to hints file");
561                            return ICERR_ERROR;
562                    }
563            }
564    
565          codec_2pass_update(codec, &frame, &stats);          codec_2pass_update(codec, &frame, &stats);
566    
567          ++codec->framenum;          ++codec->framenum;
# Line 592  Line 689 
689          frame.image = icd->lpOutput;          frame.image = icd->lpOutput;
690          frame.stride = icd->lpbiOutput->biWidth;          frame.stride = icd->lpbiOutput->biWidth;
691    
692          if (~((icd->dwFlags & ICDECOMPRESS_HURRYUP) | (icd->dwFlags & ICDECOMPRESS_UPDATE)))          if (~((icd->dwFlags & ICDECOMPRESS_HURRYUP) | (icd->dwFlags & ICDECOMPRESS_UPDATE) | (icd->dwFlags & ICDECOMPRESS_PREROLL)))
693          {          {
694                  if ((frame.colorspace = get_colorspace(icd->lpbiOutput)) == XVID_CSP_NULL)                  if ((frame.colorspace = get_colorspace(icd->lpbiOutput)) == XVID_CSP_NULL)
695                  {                  {
# Line 619  Line 716 
716          return ICERR_OK;          return ICERR_OK;
717  }  }
718    
 int codec_2pass_init(CODEC* codec)  
 {  
         TWOPASS *twopass = &codec->twopass;  
         DWORD version = -20;  
         DWORD read, wrote;  
   
         int     frames = 0, credits_frames = 0, i_frames = 0;  
         __int64 total = 0, i_total = 0, i_boost_total = 0, start = 0, end = 0, start_curved = 0, end_curved = 0;  
         __int64 desired = codec->config.desired_size * 1024;  
   
         double total1 = 0.0;  
         double total2 = 0.0;  
   
         switch (codec->config.mode)  
         {  
         case DLG_MODE_2PASS_1 :  
                 twopass->stats1 = CreateFile(codec->config.stats1, GENERIC_WRITE, FILE_SHARE_READ, 0, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, 0);  
                 if (twopass->stats1 == INVALID_HANDLE_VALUE)  
                 {  
                         DEBUGERR("2pass init error - couldn't create stats1");  
                         return ICERR_ERROR;  
                 }  
                 if (WriteFile(twopass->stats1, &version, sizeof(DWORD), &wrote, 0) == 0 || wrote != sizeof(DWORD))  
                 {  
                         CloseHandle(twopass->stats1);  
                         twopass->stats1 = INVALID_HANDLE_VALUE;  
                         DEBUGERR("2pass init error - couldn't write to stats1");  
                         return ICERR_ERROR;  
                 }  
                 break;  
   
         case DLG_MODE_2PASS_2_INT :  
         case DLG_MODE_2PASS_2_EXT :  
                 twopass->stats1 = CreateFile(codec->config.stats1, GENERIC_READ, FILE_SHARE_READ, 0, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0);  
                 if (twopass->stats1 == INVALID_HANDLE_VALUE)  
                 {  
                         DEBUGERR("2pass init error - couldn't open stats1");  
                         return ICERR_ERROR;  
                 }  
                 if (ReadFile(twopass->stats1, &version, sizeof(DWORD), &read, 0) == 0 || read != sizeof(DWORD))  
                 {  
                         CloseHandle(twopass->stats1);  
                         twopass->stats1 = INVALID_HANDLE_VALUE;  
                         DEBUGERR("2pass init error - couldn't read from stats1");  
                         return ICERR_ERROR;  
                 }  
                 if (version != -20)  
                 {  
                         CloseHandle(twopass->stats1);  
                         twopass->stats1 = INVALID_HANDLE_VALUE;  
                         DEBUGERR("2pass init error - wrong .stats version");  
                         return ICERR_ERROR;  
                 }  
   
                 if (codec->config.mode == DLG_MODE_2PASS_2_EXT)  
                 {  
                         if (twopass->stats2 != INVALID_HANDLE_VALUE)  
                         {  
                                 CloseHandle(twopass->stats2);  
                         }  
   
                         twopass->stats2 = CreateFile(codec->config.stats2, GENERIC_READ, FILE_SHARE_READ, 0, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0);  
   
                         if (twopass->stats2 == INVALID_HANDLE_VALUE)  
                         {  
                                 CloseHandle(twopass->stats1);  
                                 twopass->stats1 = INVALID_HANDLE_VALUE;  
                                 DEBUGERR("2pass init error - couldn't open stats2");  
                                 return ICERR_ERROR;  
                         }  
                         if (ReadFile(twopass->stats2, &version, sizeof(DWORD), &read, 0) == 0 || read != sizeof(DWORD))  
                         {  
                                 CloseHandle(twopass->stats1);  
                                 twopass->stats1 = INVALID_HANDLE_VALUE;  
                                 CloseHandle(twopass->stats2);  
                                 twopass->stats2 = INVALID_HANDLE_VALUE;  
                                 DEBUGERR("2pass init error - couldn't read from stats2");  
                                 return ICERR_ERROR;  
                         }  
                         if (version != -20)  
                         {  
                                 CloseHandle(twopass->stats1);  
                                 twopass->stats1 = INVALID_HANDLE_VALUE;  
                                 CloseHandle(twopass->stats2);  
                                 twopass->stats2 = INVALID_HANDLE_VALUE;  
                                 DEBUGERR("2pass init error - wrong .stats version");  
                                 return ICERR_ERROR;  
                         }  
   
                         while (1)  
                         {  
                                 if (!ReadFile(twopass->stats1, &twopass->nns1, sizeof(NNSTATS), &read, NULL) || read != sizeof(NNSTATS) ||  
                                         !ReadFile(twopass->stats2, &twopass->nns2, sizeof(NNSTATS), &read, NULL) || read != sizeof(NNSTATS))  
                                 {  
                                         DWORD err = GetLastError();  
   
                                         if (err == ERROR_HANDLE_EOF || err == ERROR_SUCCESS)  
                                         {  
                                                 break;  
                                         }  
                                         else  
                                         {  
                                                 CloseHandle(twopass->stats1);  
                                                 CloseHandle(twopass->stats2);  
                                                 twopass->stats1 = INVALID_HANDLE_VALUE;  
                                                 twopass->stats2 = INVALID_HANDLE_VALUE;  
                                                 DEBUGERR("2pass init error - incomplete stats1/stats2 record?");  
                                                 return ICERR_ERROR;  
                                         }  
                                 }  
   
                                 if (!codec_is_in_credits(&codec->config, frames))  
                                 {  
                                         if (twopass->nns1.quant & NNSTATS_KEYFRAME)  
                                         {  
                                                 i_boost_total = twopass->nns2.bytes * codec->config.keyframe_boost / 100;  
                                                 i_total += twopass->nns2.bytes;  
                                                 ++i_frames;  
                                         }  
   
                                         total += twopass->nns2.bytes;  
                                 }  
                                 else  
                                         ++credits_frames;  
   
                                 ++frames;  
                         }  
   
                         twopass->movie_curve = ((float)(total + i_boost_total) / total);  
                         twopass->average_frame = ((double)(total - i_total) / (frames - credits_frames - i_frames) / twopass->movie_curve);  
   
                         SetFilePointer(twopass->stats1, sizeof(DWORD), 0, FILE_BEGIN);  
                         SetFilePointer(twopass->stats2, sizeof(DWORD), 0, FILE_BEGIN);  
   
                         // perform prepass to compensate for over/undersizing  
                         frames = 0;  
   
                         if (codec->config.use_alt_curve)  
                         {  
                                 twopass->alt_curve_low = twopass->average_frame - twopass->average_frame * (double)codec->config.alt_curve_low_dist / 100.0;  
                                 twopass->alt_curve_low_diff = twopass->average_frame - twopass->alt_curve_low;  
                                 twopass->alt_curve_high = twopass->average_frame + twopass->average_frame * (double)codec->config.alt_curve_high_dist / 100.0;  
                                 twopass->alt_curve_high_diff = twopass->alt_curve_high - twopass->average_frame;  
                                 if (codec->config.alt_curve_use_auto)  
                                 {  
                                         if (twopass->movie_curve > 1.0f)  
                                         {  
                                                 codec->config.alt_curve_min_rel_qual = (int)(100.0 - (100.0 - 100.0 / twopass->movie_curve) * (double)codec->config.alt_curve_auto_str / 100.0);  
                                                 if (codec->config.alt_curve_min_rel_qual < 20)  
                                                         codec->config.alt_curve_min_rel_qual = 20;  
                                         }  
                                         else  
                                                 codec->config.alt_curve_min_rel_qual = 100;  
                                 }  
                                 twopass->alt_curve_mid_qual = (1.0 + (double)codec->config.alt_curve_min_rel_qual / 100.0) / 2.0;  
                                 twopass->alt_curve_qual_dev = 1.0 - twopass->alt_curve_mid_qual;  
                                 if (codec->config.alt_curve_low_dist > 100)  
                                 {  
                                         switch(codec->config.alt_curve_type)  
                                         {  
                                         case 2: // Sine Curve (high aggressiveness)  
                                                 twopass->alt_curve_qual_dev *= 2.0 / (1.0 +  
                                                         sin(DEG2RAD * (twopass->average_frame * 90.0 / twopass->alt_curve_low_diff)));  
                                                 twopass->alt_curve_mid_qual = 1.0 - twopass->alt_curve_qual_dev *  
                                                         sin(DEG2RAD * (twopass->average_frame * 90.0 / twopass->alt_curve_low_diff));  
                                                 break;  
                                         case 1: // Linear (medium aggressiveness)  
                                                 twopass->alt_curve_qual_dev *= 2.0 / (1.0 +  
                                                         twopass->average_frame / twopass->alt_curve_low_diff);  
                                                 twopass->alt_curve_mid_qual = 1.0 - twopass->alt_curve_qual_dev *  
                                                         twopass->average_frame / twopass->alt_curve_low_diff;  
                                                 break;  
                                         case 0: // Cosine Curve (low aggressiveness)  
                                                 twopass->alt_curve_qual_dev *= 2.0 / (1.0 +  
                                                         (1.0 - cos(DEG2RAD * (twopass->average_frame * 90.0 / twopass->alt_curve_low_diff))));  
                                                 twopass->alt_curve_mid_qual = 1.0 - twopass->alt_curve_qual_dev *  
                                                         (1.0 - cos(DEG2RAD * (twopass->average_frame * 90.0 / twopass->alt_curve_low_diff)));  
                                         }  
                                 }  
                         }  
   
                         while (1)  
                         {  
                                 if (!ReadFile(twopass->stats1, &twopass->nns1, sizeof(NNSTATS), &read, NULL) || read != sizeof(NNSTATS) ||  
                                         !ReadFile(twopass->stats2, &twopass->nns2, sizeof(NNSTATS), &read, NULL) || read != sizeof(NNSTATS))  
                                 {  
                                         DWORD err = GetLastError();  
   
                                         if (err == ERROR_HANDLE_EOF || err == ERROR_SUCCESS)  
                                         {  
                                                 break;  
                                         }  
                                         else  
                                         {  
                                                 CloseHandle(twopass->stats1);  
                                                 CloseHandle(twopass->stats2);  
                                                 twopass->stats1 = INVALID_HANDLE_VALUE;  
                                                 twopass->stats2 = INVALID_HANDLE_VALUE;  
                                                 DEBUGERR("2pass init error - incomplete stats1/stats2 record?");  
                                                 return ICERR_ERROR;  
                                         }  
                                 }  
   
                                 if (!codec_is_in_credits(&codec->config, frames) &&  
                                         !(twopass->nns1.quant & NNSTATS_KEYFRAME))  
                                 {  
                                         double dbytes = twopass->nns2.bytes / twopass->movie_curve;  
                                         total1 += dbytes;  
   
                                         if (codec->config.use_alt_curve)  
                                         {  
                                                 if (dbytes > twopass->average_frame)  
                                                 {  
                                                         if (dbytes >= twopass->alt_curve_high)  
                                                                 total2 += dbytes * (twopass->alt_curve_mid_qual - twopass->alt_curve_qual_dev);  
                                                         else  
                                                         {  
                                                                 switch(codec->config.alt_curve_type)  
                                                                 {  
                                                                 case 2:  
                                                                 total2 += dbytes * (twopass->alt_curve_mid_qual - twopass->alt_curve_qual_dev *  
                                                                         sin(DEG2RAD * ((dbytes - twopass->average_frame) * 90.0 / twopass->alt_curve_high_diff)));  
                                                                         break;  
                                                                 case 1:  
                                                                 total2 += dbytes * (twopass->alt_curve_mid_qual - twopass->alt_curve_qual_dev *  
                                                                         (dbytes - twopass->average_frame) / twopass->alt_curve_high_diff);  
                                                                         break;  
                                                                 case 0:  
                                                                 total2 += dbytes * (twopass->alt_curve_mid_qual - twopass->alt_curve_qual_dev *  
                                                                         (1.0 - cos(DEG2RAD * ((dbytes - twopass->average_frame) * 90.0 / twopass->alt_curve_high_diff))));  
                                                                 }  
                                                         }  
                                                 }  
                                                 else  
                                                 {  
                                                         if (dbytes <= twopass->alt_curve_low)  
                                                                 total2 += dbytes;  
                                                         else  
                                                         {  
                                                                 switch(codec->config.alt_curve_type)  
                                                                 {  
                                                                 case 2:  
                                                                 total2 += dbytes * (twopass->alt_curve_mid_qual - twopass->alt_curve_qual_dev *  
                                                                         sin(DEG2RAD * ((dbytes - twopass->average_frame) * 90.0 / twopass->alt_curve_low_diff)));  
                                                                         break;  
                                                                 case 1:  
                                                                 total2 += dbytes * (twopass->alt_curve_mid_qual - twopass->alt_curve_qual_dev *  
                                                                         (dbytes - twopass->average_frame) / twopass->alt_curve_low_diff);  
                                                                         break;  
                                                                 case 0:  
                                                                 total2 += dbytes * (twopass->alt_curve_mid_qual + twopass->alt_curve_qual_dev *  
                                                                         (1.0 - cos(DEG2RAD * ((dbytes - twopass->average_frame) * 90.0 / twopass->alt_curve_low_diff))));  
                                                                 }  
                                                         }  
                                                 }  
                                         }  
                                         else  
                                         {  
                                                 if (dbytes > twopass->average_frame)  
                                                 {  
                                                         total2 += ((double)dbytes + (twopass->average_frame - dbytes) *  
                                                                 codec->config.curve_compression_high / 100.0);  
                                                 }  
                                                 else  
                                                 {  
                                                         total2 += ((double)dbytes + (twopass->average_frame - dbytes) *  
                                                                 codec->config.curve_compression_low / 100.0);  
                                                 }  
                                         }  
                                 }  
   
                                 ++frames;  
                         }  
   
                         twopass->curve_comp_scale = total1 / total2;  
   
                         if (!codec->config.use_alt_curve)  
                         {  
                                 int asymmetric_average_frame;  
                                 char s[100];  
   
                                 asymmetric_average_frame = (int)(twopass->average_frame * twopass->curve_comp_scale);  
                                 wsprintf(s, "middle frame size for asymmetric curve compression: %i", asymmetric_average_frame);  
                                 DEBUG2P(s);  
                         }  
   
                         SetFilePointer(twopass->stats1, sizeof(DWORD), 0, FILE_BEGIN);  
                         SetFilePointer(twopass->stats2, sizeof(DWORD), 0, FILE_BEGIN);  
                 }  
                 else    // DLG_MODE_2PASS_2_INT  
                 {  
                         while (1)  
                         {  
                                 if (!ReadFile(twopass->stats1, &twopass->nns1, sizeof(NNSTATS), &read, NULL) || read != sizeof(NNSTATS))  
                                 {  
                                         DWORD err = GetLastError();  
   
                                         if (err == ERROR_HANDLE_EOF || err == ERROR_SUCCESS)  
                                         {  
                                                 break;  
                                         }  
                                         else  
                                         {  
                                                 CloseHandle(twopass->stats1);  
                                                 twopass->stats1 = INVALID_HANDLE_VALUE;  
                                                 DEBUGERR("2pass init error - incomplete stats2 record?");  
                                                 return ICERR_ERROR;  
                                         }  
                                 }  
   
                                 if (codec_is_in_credits(&codec->config, frames) == CREDITS_START)  
                                 {  
                                         start += twopass->nns1.bytes;  
                                         ++credits_frames;  
                                 }  
                                 else if (codec_is_in_credits(&codec->config, frames) == CREDITS_END)  
                                 {  
                                         end += twopass->nns1.bytes;  
                                         ++credits_frames;  
                                 }  
                                 else if (twopass->nns1.quant & NNSTATS_KEYFRAME)  
                                 {  
                                         i_total += twopass->nns1.bytes + twopass->nns1.bytes * codec->config.keyframe_boost / 100;  
                                         total += twopass->nns1.bytes * codec->config.keyframe_boost / 100;  
                                         ++i_frames;  
                                 }  
   
                                 total += twopass->nns1.bytes;  
   
                                 ++frames;  
                         }  
   
                         // compensate for avi frame overhead  
                         desired -= frames * 24;  
   
                         switch (codec->config.credits_mode)  
                         {  
                         case CREDITS_MODE_RATE :  
   
                                 // credits curve = (total / desired_size) * (100 / credits_rate)  
                                 twopass->credits_start_curve = twopass->credits_end_curve =  
                                         ((float)total / desired) * ((float)100 / codec->config.credits_rate);  
   
                                 start_curved = (__int64)(start / twopass->credits_start_curve);  
                                 end_curved = (__int64)(end / twopass->credits_end_curve);  
   
                                 // movie curve = (total - credits) / (desired_size - curved credits)  
                                 twopass->movie_curve = (float)  
                                         (total - start - end) /  
                                         (desired - start_curved - end_curved);  
   
                                 break;  
   
                         case CREDITS_MODE_QUANT :  
   
                                 // movie curve = (total - credits) / (desired_size - credits)  
                                 twopass->movie_curve = (float)  
                                         (total - start - end) / (desired - start - end);  
   
                                 // aid the average asymmetric frame calculation below  
                                 start_curved = start;  
                                 end_curved = end;  
   
                                 break;  
   
                         case CREDITS_MODE_SIZE :  
   
                                 // start curve = (start / start desired size)  
                                 twopass->credits_start_curve = (float)  
                                         (start / 1024) / codec->config.credits_start_size;  
   
                                 // end curve = (end / end desired size)  
                                 twopass->credits_end_curve = (float)  
                                         (end / 1024) / codec->config.credits_end_size;  
   
                                 start_curved = (__int64)(start / twopass->credits_start_curve);  
                                 end_curved = (__int64)(end / twopass->credits_end_curve);  
   
                                 // movie curve = (total - credits) / (desired_size - curved credits)  
                                 twopass->movie_curve = (float)  
                                         (total - start - end) /  
                                         (desired - start_curved - end_curved);  
   
                                 break;  
                         }  
   
                         // average frame size = (desired - curved credits - curved keyframes) /  
                         //      (frames - credits frames - keyframes)  
                         twopass->average_frame = (double)  
                                 (desired - start_curved - end_curved - (i_total / twopass->movie_curve)) /  
                                 (frames - credits_frames - i_frames);  
   
                         SetFilePointer(twopass->stats1, sizeof(DWORD), 0, FILE_BEGIN);  
   
                         // perform prepass to compensate for over/undersizing  
                         frames = 0;  
   
                         if (codec->config.use_alt_curve)  
                         {  
                                 twopass->alt_curve_low = twopass->average_frame - twopass->average_frame * (double)codec->config.alt_curve_low_dist / 100.0;  
                                 twopass->alt_curve_low_diff = twopass->average_frame - twopass->alt_curve_low;  
                                 twopass->alt_curve_high = twopass->average_frame + twopass->average_frame * (double)codec->config.alt_curve_high_dist / 100.0;  
                                 twopass->alt_curve_high_diff = twopass->alt_curve_high - twopass->average_frame;  
                                 if (codec->config.alt_curve_use_auto)  
                                 {  
                                         if (twopass->movie_curve > 1.0f)  
                                         {  
                                                 codec->config.alt_curve_min_rel_qual = (int)(100.0 - (100.0 - 100.0 / twopass->movie_curve) * (double)codec->config.alt_curve_auto_str / 100.0);  
                                                 if (codec->config.alt_curve_min_rel_qual < 20)  
                                                         codec->config.alt_curve_min_rel_qual = 20;  
                                         }  
                                         else  
                                                 codec->config.alt_curve_min_rel_qual = 100;  
                                 }  
                                 twopass->alt_curve_mid_qual = (1.0 + (double)codec->config.alt_curve_min_rel_qual / 100.0) / 2.0;  
                                 twopass->alt_curve_qual_dev = 1.0 - twopass->alt_curve_mid_qual;  
                                 if (codec->config.alt_curve_low_dist > 100)  
                                 {  
                                         switch(codec->config.alt_curve_type)  
                                         {  
                                         case 2: // Sine Curve (high aggressiveness)  
                                                 twopass->alt_curve_qual_dev *= 2.0 / (1.0 +  
                                                         sin(DEG2RAD * (twopass->average_frame * 90.0 / twopass->alt_curve_low_diff)));  
                                                 twopass->alt_curve_mid_qual = 1.0 - twopass->alt_curve_qual_dev *  
                                                         sin(DEG2RAD * (twopass->average_frame * 90.0 / twopass->alt_curve_low_diff));  
                                                 break;  
                                         case 1: // Linear (medium aggressiveness)  
                                                 twopass->alt_curve_qual_dev *= 2.0 / (1.0 +  
                                                         twopass->average_frame / twopass->alt_curve_low_diff);  
                                                 twopass->alt_curve_mid_qual = 1.0 - twopass->alt_curve_qual_dev *  
                                                         twopass->average_frame / twopass->alt_curve_low_diff;  
                                                 break;  
                                         case 0: // Cosine Curve (low aggressiveness)  
                                                 twopass->alt_curve_qual_dev *= 2.0 / (1.0 +  
                                                         (1.0 - cos(DEG2RAD * (twopass->average_frame * 90.0 / twopass->alt_curve_low_diff))));  
                                                 twopass->alt_curve_mid_qual = 1.0 - twopass->alt_curve_qual_dev *  
                                                         (1.0 - cos(DEG2RAD * (twopass->average_frame * 90.0 / twopass->alt_curve_low_diff)));  
                                         }  
                                 }  
                         }  
   
                         while (1)  
                         {  
                                 if (!ReadFile(twopass->stats1, &twopass->nns1, sizeof(NNSTATS), &read, NULL) || read != sizeof(NNSTATS))  
                                 {  
                                         DWORD err = GetLastError();  
   
                                         if (err == ERROR_HANDLE_EOF || err == ERROR_SUCCESS)  
                                         {  
                                                 break;  
                                         }  
                                         else  
                                         {  
                                                 CloseHandle(twopass->stats1);  
                                                 twopass->stats1 = INVALID_HANDLE_VALUE;  
                                                 DEBUGERR("2pass init error - incomplete stats2 record?");  
                                                 return ICERR_ERROR;  
                                         }  
                                 }  
   
                                 if (!codec_is_in_credits(&codec->config, frames) &&  
                                         !(twopass->nns1.quant & NNSTATS_KEYFRAME))  
                                 {  
                                         double dbytes = twopass->nns1.bytes / twopass->movie_curve;  
                                         total1 += dbytes;  
   
                                         if (codec->config.use_alt_curve)  
                                         {  
                                                 if (dbytes > twopass->average_frame)  
                                                 {  
                                                         if (dbytes >= twopass->alt_curve_high)  
                                                                 total2 += dbytes * (twopass->alt_curve_mid_qual - twopass->alt_curve_qual_dev);  
                                                         else  
                                                         {  
                                                                 switch(codec->config.alt_curve_type)  
                                                                 {  
                                                                 case 2:  
                                                                 total2 += dbytes * (twopass->alt_curve_mid_qual - twopass->alt_curve_qual_dev *  
                                                                         sin(DEG2RAD * ((dbytes - twopass->average_frame) * 90.0 / twopass->alt_curve_high_diff)));  
                                                                         break;  
                                                                 case 1:  
                                                                 total2 += dbytes * (twopass->alt_curve_mid_qual - twopass->alt_curve_qual_dev *  
                                                                         (dbytes - twopass->average_frame) / twopass->alt_curve_high_diff);  
                                                                         break;  
                                                                 case 0:  
                                                                 total2 += dbytes * (twopass->alt_curve_mid_qual - twopass->alt_curve_qual_dev *  
                                                                         (1.0 - cos(DEG2RAD * ((dbytes - twopass->average_frame) * 90.0 / twopass->alt_curve_high_diff))));  
                                                                 }  
                                                         }  
                                                 }  
                                                 else  
                                                 {  
                                                         if (dbytes <= twopass->alt_curve_low)  
                                                                 total2 += dbytes;  
                                                         else  
                                                         {  
                                                                 switch(codec->config.alt_curve_type)  
                                                                 {  
                                                                 case 2:  
                                                                 total2 += dbytes * (twopass->alt_curve_mid_qual - twopass->alt_curve_qual_dev *  
                                                                         sin(DEG2RAD * ((dbytes - twopass->average_frame) * 90.0 / twopass->alt_curve_low_diff)));  
                                                                         break;  
                                                                 case 1:  
                                                                 total2 += dbytes * (twopass->alt_curve_mid_qual - twopass->alt_curve_qual_dev *  
                                                                         (dbytes - twopass->average_frame) / twopass->alt_curve_low_diff);  
                                                                         break;  
                                                                 case 0:  
                                                                 total2 += dbytes * (twopass->alt_curve_mid_qual + twopass->alt_curve_qual_dev *  
                                                                         (1.0 - cos(DEG2RAD * ((dbytes - twopass->average_frame) * 90.0 / twopass->alt_curve_low_diff))));  
                                                                 }  
                                                         }  
                                                 }  
                                         }  
                                         else  
                                         {  
                                                 if (dbytes > twopass->average_frame)  
                                                 {  
                                                         total2 += ((double)dbytes + (twopass->average_frame - dbytes) *  
                                                                 codec->config.curve_compression_high / 100.0);  
                                                 }  
                                                 else  
                                                 {  
                                                         total2 += ((double)dbytes + (twopass->average_frame - dbytes) *  
                                                                 codec->config.curve_compression_low / 100.0);  
                                                 }  
                                         }  
                                 }  
   
                                 ++frames;  
                         }  
   
                         twopass->curve_comp_scale = total1 / total2;  
   
                         if (!codec->config.use_alt_curve)  
                         {  
                                 int asymmetric_average_frame;  
                                 char s[100];  
   
                                 asymmetric_average_frame = (int)(twopass->average_frame * twopass->curve_comp_scale);  
                                 wsprintf(s, "middle frame size for asymmetric curve compression: %i", asymmetric_average_frame);  
                                 DEBUG2P(s);  
                         }  
   
                         SetFilePointer(twopass->stats1, sizeof(DWORD), 0, FILE_BEGIN);  
                 }  
   
                 if (codec->config.use_alt_curve)  
                 {  
                         if (codec->config.alt_curve_use_auto_bonus_bias)  
                                 codec->config.alt_curve_bonus_bias = codec->config.alt_curve_min_rel_qual;  
   
                         twopass->curve_bias_bonus = (total1 - total2) * (double)codec->config.alt_curve_bonus_bias / 100.0 / (double)(frames - credits_frames - i_frames);  
                         twopass->curve_comp_scale = ((total1 - total2) * (1.0 - (double)codec->config.alt_curve_bonus_bias / 100.0) + total2) / total2;  
                 }  
   
                 twopass->overflow = 0;  
   
                 break;  
         }  
   
         return ICERR_OK;  
 }  
   
   
719  int codec_get_quant(CODEC* codec, XVID_ENC_FRAME* frame)  int codec_get_quant(CODEC* codec, XVID_ENC_FRAME* frame)
720  {  {
721          switch (codec->config.mode)          switch (codec->config.mode)
# Line 1262  Line 795 
795  }  }
796    
797    
 int codec_2pass_get_quant(CODEC* codec, XVID_ENC_FRAME* frame)  
 {  
         static double quant_error[32];  
         static double curve_comp_error;  
         static int last_quant;  
   
         TWOPASS * twopass = &codec->twopass;  
   
         DWORD read;  
         int bytes1, bytes2;  
         int overflow;  
         int credits_pos;  
   
         if (codec->framenum == 0)  
         {  
                 int i;  
   
                 for (i=0 ; i<32 ; ++i)  
                 {  
                         quant_error[i] = 0.0;  
                 }  
   
                 curve_comp_error = 0.0;  
                 last_quant = 0;  
         }  
   
         if (ReadFile(twopass->stats1, &twopass->nns1, sizeof(NNSTATS), &read, 0) == 0 || read != sizeof(NNSTATS))  
         {  
                 DEBUGERR("2ndpass quant: couldn't read from stats1");  
                 return ICERR_ERROR;  
         }  
         if (codec->config.mode == DLG_MODE_2PASS_2_EXT)  
         {  
                 if (ReadFile(twopass->stats2, &twopass->nns2, sizeof(NNSTATS), &read, 0) == 0 || read != sizeof(NNSTATS))  
                 {  
                         DEBUGERR("2ndpass quant: couldn't read from stats2");  
                         return ICERR_ERROR;  
                 }  
         }  
   
         bytes1 = twopass->nns1.bytes;  
         overflow = twopass->overflow / 8;  
   
         // override codec i-frame choice (reenable in credits)  
         frame->intra = (twopass->nns1.quant & NNSTATS_KEYFRAME);  
   
         if (frame->intra)  
         {  
                 overflow = 0;  
         }  
   
         credits_pos = codec_is_in_credits(&codec->config, codec->framenum);  
   
         if (credits_pos)  
         {  
                 if (codec->config.mode == DLG_MODE_2PASS_2_INT)  
                 {  
                         switch (codec->config.credits_mode)  
                         {  
                         case CREDITS_MODE_RATE :  
                         case CREDITS_MODE_SIZE :  
                                 if (credits_pos == CREDITS_START)  
                                 {  
                                         bytes2 = (int)(bytes1 / twopass->credits_start_curve);  
                                 }  
                                 else // CREDITS_END  
                                 {  
                                         bytes2 = (int)(bytes1 / twopass->credits_end_curve);  
                                 }  
   
                                 frame->intra = -1;  
                                 break;  
   
                         case CREDITS_MODE_QUANT :  
                                 if (codec->config.credits_quant_i != codec->config.credits_quant_p)  
                                 {  
                                         frame->quant = frame->intra ?  
                                                 codec->config.credits_quant_i :  
                                                 codec->config.credits_quant_p;  
                                 }  
                                 else  
                                 {  
                                         frame->quant = codec->config.credits_quant_p;  
                                         frame->intra = -1;  
                                 }  
   
                                 twopass->bytes1 = bytes1;  
                                 twopass->bytes2 = bytes1;  
                                 twopass->desired_bytes2 = bytes1;  
                                 return ICERR_OK;  
                         }  
                 }  
                 else    // DLG_MODE_2PASS_2_EXT  
                 {  
                         bytes2 = twopass->nns2.bytes;  
                 }  
         }  
         else    // Foxer: apply curve compression outside credits  
         {  
                 double dbytes, curve_temp;  
   
                 bytes2 = (codec->config.mode == DLG_MODE_2PASS_2_INT) ? bytes1 : twopass->nns2.bytes;  
   
                 if (frame->intra)  
                 {  
                         dbytes = ((int)(bytes2 + bytes2 * codec->config.keyframe_boost / 100)) /  
                                 twopass->movie_curve;  
                 }  
                 else  
                 {  
                         dbytes = bytes2 / twopass->movie_curve;  
                 }  
   
                 // spread the compression error across payback_delay frames  
                 if (codec->config.bitrate_payback_method == 0)  
                 {  
                         bytes2 = (int)(curve_comp_error / codec->config.bitrate_payback_delay);  
                 }  
                 else  
                 {  
                         bytes2 = (int)(curve_comp_error * dbytes /  
                                 twopass->average_frame / codec->config.bitrate_payback_delay);  
   
                         if (labs(bytes2) > fabs(curve_comp_error))  
                         {  
                                 bytes2 = (int)curve_comp_error;  
                         }  
                 }  
   
                 curve_comp_error -= bytes2;  
   
                 if (codec->config.use_alt_curve)  
                 {  
                         if (!frame->intra)  
                         {  
                                 if (dbytes > twopass->average_frame)  
                                 {  
                                         if (dbytes >= twopass->alt_curve_high)  
                                                 curve_temp = dbytes * (twopass->alt_curve_mid_qual - twopass->alt_curve_qual_dev);  
                                         else  
                                         {  
                                                 switch(codec->config.alt_curve_type)  
                                                 {  
                                                 case 2:  
                                                 curve_temp = dbytes * (twopass->alt_curve_mid_qual - twopass->alt_curve_qual_dev *  
                                                         sin(DEG2RAD * ((dbytes - twopass->average_frame) * 90.0 / twopass->alt_curve_high_diff)));  
                                                         break;  
                                                 case 1:  
                                                 curve_temp = dbytes * (twopass->alt_curve_mid_qual - twopass->alt_curve_qual_dev *  
                                                         (dbytes - twopass->average_frame) / twopass->alt_curve_high_diff);  
                                                         break;  
                                                 case 0:  
                                                 curve_temp = dbytes * (twopass->alt_curve_mid_qual - twopass->alt_curve_qual_dev *  
                                                         (1.0 - cos(DEG2RAD * ((dbytes - twopass->average_frame) * 90.0 / twopass->alt_curve_high_diff))));  
                                                 }  
                                         }  
                                 }  
                                 else  
                                 {  
                                         if (dbytes <= twopass->alt_curve_low)  
                                                 curve_temp = dbytes;  
                                         else  
                                         {  
                                                 switch(codec->config.alt_curve_type)  
                                                 {  
                                                 case 2:  
                                                 curve_temp = dbytes * (twopass->alt_curve_mid_qual - twopass->alt_curve_qual_dev *  
                                                         sin(DEG2RAD * ((dbytes - twopass->average_frame) * 90.0 / twopass->alt_curve_low_diff)));  
                                                         break;  
                                                 case 1:  
                                                 curve_temp = dbytes * (twopass->alt_curve_mid_qual - twopass->alt_curve_qual_dev *  
                                                         (dbytes - twopass->average_frame) / twopass->alt_curve_low_diff);  
                                                         break;  
                                                 case 0:  
                                                 curve_temp = dbytes * (twopass->alt_curve_mid_qual + twopass->alt_curve_qual_dev *  
                                                         (1.0 - cos(DEG2RAD * ((dbytes - twopass->average_frame) * 90.0 / twopass->alt_curve_low_diff))));  
                                                 }  
                                         }  
                                 }  
                                 curve_temp = curve_temp * twopass->curve_comp_scale + twopass->curve_bias_bonus;  
   
                                 bytes2 += ((int)curve_temp);  
                                 curve_comp_error += curve_temp - ((int)curve_temp);  
                         }  
                         else  
                         {  
                                 curve_comp_error += dbytes - ((int)dbytes);  
                                 bytes2 += ((int)dbytes);  
                         }  
                 }  
                 else if ((codec->config.curve_compression_high + codec->config.curve_compression_low) &&  
                         !frame->intra)  
                 {  
                         if (dbytes > twopass->average_frame)  
                         {  
                                 curve_temp = twopass->curve_comp_scale *  
                                         ((double)dbytes + (twopass->average_frame - dbytes) *  
                                         codec->config.curve_compression_high / 100.0);  
                         }  
                         else  
                         {  
                                 curve_temp = twopass->curve_comp_scale *  
                                         ((double)dbytes + (twopass->average_frame - dbytes) *  
                                         codec->config.curve_compression_low / 100.0);  
                         }  
   
                         bytes2 += ((int)curve_temp);  
                         curve_comp_error += curve_temp - ((int)curve_temp);  
                 }  
                 else  
                 {  
                         curve_comp_error += dbytes - ((int)dbytes);  
                         bytes2 += ((int)dbytes);  
                 }  
   
                 // cap bytes2 to first pass size, lowers number of quant=1 frames  
                 if (bytes2 > bytes1)  
                 {  
                         curve_comp_error += bytes2 - bytes1;  
                         bytes2 = bytes1;  
                 }  
                 else if (bytes2 < 1)  
                 {  
                         curve_comp_error += --bytes2;  
                         bytes2 = 1;  
                 }  
         }  
   
         twopass->desired_bytes2 = bytes2;  
   
         // Foxer: scale overflow in relation to average size, so smaller frames don't get  
         // too much/little bitrate  
         overflow = (int)((double)overflow * bytes2 / twopass->average_frame);  
   
         // Foxer: reign in overflow with huge frames  
         if (labs(overflow) > labs(twopass->overflow))  
         {  
                 overflow = twopass->overflow;  
         }  
   
         // Foxer: make sure overflow doesn't run away  
         if (overflow > bytes2 * 6 / 10)  
         {  
                 bytes2 += (overflow <= bytes2) ? bytes2 * 6 / 10 : overflow * 6 / 10;  
         }  
         else if (overflow < bytes2 * -6 / 10)  
         {  
                 bytes2 += bytes2 * -6 / 10;  
         }  
         else  
         {  
                 bytes2 += overflow;  
         }  
   
         if (bytes2 < 1)  
         {  
                 bytes2 = 1;  
         }  
   
         twopass->bytes1 = bytes1;  
         twopass->bytes2 = bytes2;  
   
         // very 'simple' quant<->filesize relationship  
         frame->quant = ((twopass->nns1.quant & ~NNSTATS_KEYFRAME) * bytes1) / bytes2;  
   
         if (frame->quant < 1)  
         {  
                 frame->quant = 1;  
         }  
         else if (frame->quant > 31)  
         {  
                 frame->quant = 31;  
         }  
         else if (!frame->intra)  
         {  
                 // Foxer: aid desired quantizer precision by accumulating decision error  
                 quant_error[frame->quant] += ((double)((twopass->nns1.quant & ~NNSTATS_KEYFRAME) *  
                         bytes1) / bytes2) - frame->quant;  
   
                 if (quant_error[frame->quant] >= 1.0)  
                 {  
                         quant_error[frame->quant] -= 1.0;  
                         ++frame->quant;  
                 }  
         }  
   
         // we're done with credits  
         if (codec_is_in_credits(&codec->config, codec->framenum))  
         {  
                 return ICERR_OK;  
         }  
   
         if (frame->intra)  
         {  
                 if (frame->quant < codec->config.min_iquant)  
                 {  
                         frame->quant = codec->config.min_iquant;  
                         DEBUG2P("I-frame quantizer raised");  
                 }  
                 if (frame->quant > codec->config.max_iquant)  
                 {  
                         frame->quant = codec->config.max_iquant;  
                         DEBUG2P("I-frame quantizer lowered");  
                 }  
         }  
         else  
         {  
                 if (frame->quant > codec->config.max_pquant)  
                 {  
                         frame->quant = codec->config.max_pquant;  
                 }  
                 if (frame->quant < codec->config.min_pquant)  
                 {  
                         frame->quant = codec->config.min_pquant;  
                 }  
   
                 // subsequent frame quants can only be +- 2  
                 if (last_quant)  
                 {  
                         if (frame->quant > last_quant + 2)  
                         {  
                                 frame->quant = last_quant + 2;  
                                 DEBUG2P("P-frame quantizer prevented from rising too steeply");  
                         }  
                         if (frame->quant < last_quant - 2)  
                         {  
                                 frame->quant = last_quant - 2;  
                                 DEBUG2P("P-frame quantizer prevented from falling too steeply");  
                         }  
                 }  
         }  
   
         last_quant = frame->quant;  
   
         if (codec->config.quant_type == QUANT_MODE_MOD)  
         {  
                 frame->general |= (frame->quant < 4) ? XVID_MPEGQUANT : XVID_H263QUANT;  
                 frame->general &= (frame->quant < 4) ? ~XVID_H263QUANT : ~XVID_MPEGQUANT;  
         }  
   
         return ICERR_OK;  
 }  
   
   
 int codec_2pass_update(CODEC* codec, XVID_ENC_FRAME* frame, XVID_ENC_STATS* stats)  
 {  
         static __int64 total_size;  
   
         NNSTATS nns1;  
         DWORD wrote;  
         char* quant_type;  
   
         if (codec->framenum == 0)  
         {  
                 total_size = 0;  
         }  
   
         quant_type = (frame->general & XVID_H263QUANT) ? "H.263" :  
                 ((frame->general & XVID_MPEGQUANT) && (frame->general & XVID_CUSTOM_QMATRIX)) ?  
                 "Cust" : "MPEG";  
   
         switch (codec->config.mode)  
         {  
         case DLG_MODE_2PASS_1 :  
                 nns1.bytes = frame->length;     // total bytes  
                 nns1.dd_v = stats->hlength;     // header bytes  
   
                 nns1.dd_u = nns1.dd_y = 0;  
                 nns1.dk_v = nns1.dk_u = nns1.dk_y = 0;  
                 nns1.md_u = nns1.md_y = 0;  
                 nns1.mk_u = nns1.mk_y = 0;  
   
                 nns1.quant = stats->quant;  
                 if (frame->intra)  
                 {  
                         nns1.quant |= NNSTATS_KEYFRAME;  
                 }  
                 nns1.kblk = stats->kblks;  
                 nns1.mblk = stats->mblks;  
                 nns1.ublk = stats->ublks;  
                 nns1.lum_noise[0] = nns1.lum_noise[1] = 1;  
   
                 total_size += frame->length;  
   
                 DEBUG1ST(frame->length, (int)total_size/1024, frame->intra, frame->quant, quant_type, stats->kblks, stats->mblks)  
   
                 if (WriteFile(codec->twopass.stats1, &nns1, sizeof(NNSTATS), &wrote, 0) == 0 || wrote != sizeof(NNSTATS))  
                 {  
                         DEBUGERR("stats1: WriteFile error");  
                         return ICERR_ERROR;  
                 }  
                 break;  
   
         case DLG_MODE_2PASS_2_INT :  
         case DLG_MODE_2PASS_2_EXT :  
                 codec->twopass.overflow += codec->twopass.desired_bytes2 - frame->length;  
                 DEBUG2ND(frame->quant, quant_type, frame->intra, codec->twopass.bytes1, codec->twopass.desired_bytes2, frame->length, codec->twopass.overflow, codec_is_in_credits(&codec->config, codec->framenum))  
                 break;  
   
         default:  
                 break;  
         }  
   
         return ICERR_OK;  
 }  
   
   
798  int codec_is_in_credits(CONFIG* config, int framenum)  int codec_is_in_credits(CONFIG* config, int framenum)
799  {  {
800          if (config->credits_start)          if (config->credits_start)

Legend:
Removed from v.61  
changed lines
  Added in v.235

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