[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 3, Fri Mar 8 02:46:11 2002 UTC revision 968, Fri Apr 4 10:34:57 2003 UTC
# Line 23  Line 23 
23   *   *
24   *      History:   *      History:
25   *   *
26   *      ... ???   *      12.07.2002      num_threads
27   *      01.12.2001      inital version; (c)2001 peter ross <suxen_drol@hotmail.com>   *      23.06.2002      XVID_CPU_CHKONLY; loading speed up
28     *      25.04.2002      ICDECOMPRESS_PREROLL
29     *      17.04.2002      re-enabled lumi masking for 1st pass
30     *      15.04.2002      updated cbr support
31     *      04.04.2002      separated 2-pass code to 2pass.c
32     *                              interlacing support
33     *                              hinted ME support
34     *      23.03.2002      daniel smith <danielsmith@astroboymail.com>
35     *                              changed inter4v to only be in modes 5 or 6
36     *                              fixed null mode crash ?
37     *                              merged foxer's alternative 2-pass code
38     *                              added DEBUGERR output on errors instead of returning
39     *      16.03.2002      daniel smith <danielsmith@astroboymail.com>
40     *                              changed BITMAPV4HEADER to BITMAPINFOHEADER
41     *                                      - prevents memcpy crash in compress_get_format()
42     *                              credits are processed in external 2pass mode
43     *                              motion search precision = 0 now effective in 2-pass
44     *                              modulated quantization
45     *                              added DX50 fourcc
46     *      01.12.2001      inital version; (c)2001 peter ross <pross@xvid.org>
47   *   *
48   *************************************************************************/   *************************************************************************/
49    
# Line 32  Line 51 
51  #include <vfw.h>  #include <vfw.h>
52    
53  #include "codec.h"  #include "codec.h"
54  #include <math.h>       // fabs()  #include "2pass.h"
55    
56  int pmvfast_presets[7] = {  int pmvfast_presets[7] = {
57          0, PMV_QUICKSTOP16, PMV_EARLYSTOP16, PMV_EARLYSTOP16 | PMV_EARLYSTOP8,          0, PMV_QUICKSTOP16, 0, 0,
58          PMV_EARLYSTOP16 | PMV_HALFPELREFINE16 | PMV_EARLYSTOP8 | PMV_HALFPELDIAMOND8,          0 | PMV_HALFPELREFINE16 | PMV_HALFPELDIAMOND8,
59          PMV_EARLYSTOP16 | PMV_HALFPELREFINE16 | PMV_EARLYSTOP8 | PMV_HALFPELDIAMOND8,          0 | PMV_HALFPELREFINE16 | PMV_HALFPELDIAMOND8 |
60          PMV_EARLYSTOP16 | PMV_HALFPELREFINE16 | PMV_EXTSEARCH16 |          PMV_ADVANCEDDIAMOND16, PMV_HALFPELREFINE16 | PMV_EXTSEARCH16 |
61          PMV_EARLYSTOP8 | PMV_HALFPELREFINE8 | PMV_HALFPELDIAMOND8          PMV_HALFPELREFINE8 | PMV_HALFPELDIAMOND8 | PMV_USESQUARES16
62  };  };
63    
64  /*      return xvid compatbile colorspace,  /*      return xvid compatbile colorspace,
65          or XVID_CSP_NULL if failure          or XVID_CSP_NULL if failure
66  */  */
67    
68  int get_colorspace(BITMAPV4HEADER * hdr)  int get_colorspace(BITMAPINFOHEADER * hdr)
69  {  {
70          if (hdr->bV4Height < 0)          /* rgb only: negative height specifies top down image */
71          {          int rgb_flip = (hdr->biHeight < 0 ? 0 : XVID_CSP_VFLIP);
                 DEBUG("colorspace: inverted input format not supported");  
                 return XVID_CSP_NULL;  
         }  
72    
73          switch(hdr->bV4V4Compression)          switch(hdr->biCompression)
74          {          {
75          case BI_RGB :          case BI_RGB :
76                  if (hdr->bV4BitCount == 16)                  if (hdr->biBitCount == 16)
77                  {                  {
78                          DEBUG("RGB16 (RGB555)");                          DEBUG("RGB16 (RGB555)");
79                          return XVID_CSP_VFLIP | XVID_CSP_RGB555;                          return rgb_flip | XVID_CSP_RGB555;
80                  }                  }
81                  if (hdr->bV4BitCount == 24)                  if (hdr->biBitCount == 24)
82                  {                  {
83                          DEBUG("RGB24");                          DEBUG("RGB24");
84                          return XVID_CSP_VFLIP | XVID_CSP_RGB24;                          return rgb_flip | XVID_CSP_RGB24;
85                  }                  }
86                  if (hdr->bV4BitCount == 32)                  if (hdr->biBitCount == 32)
87                  {                  {
88                          DEBUG("RGB32");                          DEBUG("RGB32");
89                          return XVID_CSP_VFLIP | XVID_CSP_RGB32;                          return rgb_flip | XVID_CSP_RGB32;
90                  }                  }
91    
92                  DEBUG1("BI_RGB unsupported", hdr->bV4BitCount);                  DEBUG1("unsupported BI_RGB biBitCount", hdr->biBitCount);
93                  return XVID_CSP_NULL;                  return XVID_CSP_NULL;
94    
95          case BI_BITFIELDS :          case BI_BITFIELDS :
96                  if(hdr->bV4BitCount == 16 &&                  if (hdr->biSize >= sizeof(BITMAPV4HEADER))
97                          hdr->bV4RedMask == 0x7c00 &&                  {
98                          hdr->bV4GreenMask == 0x3e0 &&                          BITMAPV4HEADER * hdr4 = (BITMAPV4HEADER *)hdr;
99                          hdr->bV4BlueMask == 0x1f)  
100                            if (hdr4->bV4BitCount == 16 &&
101                                    hdr4->bV4RedMask == 0x7c00 &&
102                                    hdr4->bV4GreenMask == 0x3e0 &&
103                                    hdr4->bV4BlueMask == 0x1f)
104                  {                  {
105                          DEBUG("RGB555");                          DEBUG("RGB555");
106                          return XVID_CSP_VFLIP | XVID_CSP_RGB555;                                  return rgb_flip | XVID_CSP_RGB555;
107                  }                  }
108                  if(hdr->bV4BitCount == 16 &&  
109                          hdr->bV4RedMask == 0xf800 &&                          if(hdr4->bV4BitCount == 16 &&
110                          hdr->bV4GreenMask == 0x7e0 &&                                  hdr4->bV4RedMask == 0xf800 &&
111                          hdr->bV4BlueMask == 0x1f)                                  hdr4->bV4GreenMask == 0x7e0 &&
112                                    hdr4->bV4BlueMask == 0x1f)
113                  {                  {
114                          DEBUG("RGB565");                          DEBUG("RGB565");
115                          return XVID_CSP_VFLIP | XVID_CSP_RGB565;                                  return rgb_flip | XVID_CSP_RGB565;
116                            }
117    
118                            DEBUG("unsupported BI_BITFIELDS mode");
119                            return XVID_CSP_NULL;
120                  }                  }
121    
122                  DEBUG1("BI_FIELDS unsupported", hdr->bV4BitCount);                  DEBUG("unsupported BI_BITFIELDS/BITMAPHEADER combination");
123                  return XVID_CSP_NULL;                  return XVID_CSP_NULL;
124    
125          case FOURCC_I420:          case FOURCC_I420:
# Line 108  Line 133 
133    
134          case FOURCC_YUYV :          case FOURCC_YUYV :
135          case FOURCC_YUY2 :          case FOURCC_YUY2 :
         case FOURCC_V422 :  
136                  DEBUG("YUY2");                  DEBUG("YUY2");
137                  return XVID_CSP_YUY2;                  return XVID_CSP_YUY2;
138    
# Line 120  Line 144 
144                  DEBUG("UYVY");                  DEBUG("UYVY");
145                  return XVID_CSP_UYVY;                  return XVID_CSP_UYVY;
146    
147          }          default :
148          DEBUGFOURCC("colorspace: unknown", hdr->bV4V4Compression);                  DEBUGFOURCC("unsupported colorspace", hdr->biCompression);
149          return XVID_CSP_NULL;          return XVID_CSP_NULL;
150  }  }
151    }
152    
153    
154  /* compressor */  /* compressor */
# Line 133  Line 158 
158    
159  LRESULT compress_query(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)  LRESULT compress_query(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)
160  {  {
161          BITMAPV4HEADER * inhdr = (BITMAPV4HEADER *)&lpbiInput->bmiHeader;          BITMAPINFOHEADER * inhdr = &lpbiInput->bmiHeader;
162          BITMAPV4HEADER * outhdr = (BITMAPV4HEADER *)&lpbiOutput->bmiHeader;          BITMAPINFOHEADER * outhdr = &lpbiOutput->bmiHeader;
163    
164          if (get_colorspace(inhdr) == XVID_CSP_NULL)          if (get_colorspace(inhdr) == XVID_CSP_NULL)
165          {          {
# Line 146  Line 171 
171                  return ICERR_OK;                  return ICERR_OK;
172          }          }
173    
174          if (inhdr->bV4Width != outhdr->bV4Width || inhdr->bV4Height != outhdr->bV4Height ||          if (inhdr->biWidth != outhdr->biWidth || inhdr->biHeight != outhdr->biHeight ||
175                  (outhdr->bV4V4Compression != FOURCC_XVID && outhdr->bV4V4Compression != FOURCC_DIVX))                  (outhdr->biCompression != FOURCC_XVID && outhdr->biCompression != FOURCC_DIVX && outhdr->biCompression != FOURCC_DX50))
176          {          {
177                  return ICERR_BADFORMAT;                  return ICERR_BADFORMAT;
178          }          }
# Line 158  Line 183 
183    
184  LRESULT compress_get_format(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)  LRESULT compress_get_format(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)
185  {  {
186          BITMAPV4HEADER * inhdr = (BITMAPV4HEADER *)&lpbiInput->bmiHeader;          BITMAPINFOHEADER * inhdr = &lpbiInput->bmiHeader;
187          BITMAPV4HEADER * outhdr = (BITMAPV4HEADER *)&lpbiOutput->bmiHeader;          BITMAPINFOHEADER * outhdr = &lpbiOutput->bmiHeader;
188    
189          if (get_colorspace(inhdr) == XVID_CSP_NULL)          if (get_colorspace(inhdr) == XVID_CSP_NULL)
190          {          {
# Line 171  Line 196 
196                  return sizeof(BITMAPV4HEADER);                  return sizeof(BITMAPV4HEADER);
197          }          }
198    
199          memcpy(outhdr, inhdr, sizeof(BITMAPV4HEADER));          memcpy(outhdr, inhdr, sizeof(BITMAPINFOHEADER));
200          outhdr->bV4Size = sizeof(BITMAPV4HEADER);          outhdr->biSize = sizeof(BITMAPINFOHEADER);
201          outhdr->bV4BitCount = 24;  // or 16          outhdr->biSizeImage = compress_get_size(codec, lpbiInput, lpbiOutput);
202          outhdr->bV4SizeImage = compress_get_size(codec, lpbiInput, lpbiOutput);          outhdr->biXPelsPerMeter = 0;
203          outhdr->bV4XPelsPerMeter = 0;          outhdr->biYPelsPerMeter = 0;
204          outhdr->bV4YPelsPerMeter = 0;          outhdr->biClrUsed = 0;
205          outhdr->bV4ClrUsed = 0;          outhdr->biClrImportant = 0;
         outhdr->bV4ClrImportant = 0;  
206    
207          if (codec->config.fourcc_used == 0)          if (codec->config.fourcc_used == 0)
208          {          {
209                  outhdr->bV4V4Compression = FOURCC_XVID;                  outhdr->biCompression = FOURCC_XVID;
210            }
211            else if (codec->config.fourcc_used == 1)
212            {
213                    outhdr->biCompression = FOURCC_DIVX;
214          }          }
215          else          else
216          {          {
217                  outhdr->bV4V4Compression = FOURCC_DIVX;                  outhdr->biCompression = FOURCC_DX50;
218          }          }
219    
220          return ICERR_OK;          return ICERR_OK;
# Line 195  Line 223 
223    
224  LRESULT compress_get_size(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)  LRESULT compress_get_size(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)
225  {  {
226          BITMAPV4HEADER * outhdr = (BITMAPV4HEADER *)&lpbiOutput->bmiHeader;          return 2 * lpbiOutput->bmiHeader.biWidth * lpbiOutput->bmiHeader.biHeight * 3;
         return outhdr->bV4Width * outhdr->bV4Height * 3;  
227  }  }
228    
229    
# Line 211  Line 238 
238    
239  LRESULT compress_begin(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)  LRESULT compress_begin(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)
240  {  {
         BITMAPV4HEADER * inhdr = (BITMAPV4HEADER *)&lpbiInput->bmiHeader;  
241          XVID_ENC_PARAM param;          XVID_ENC_PARAM param;
242          XVID_INIT_PARAM init_param;          XVID_INIT_PARAM init_param;
243    
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 :
271                    return ICERR_OK;
272    
273          default :          default :
274                  break;                  break;
275          }          }
# Line 253  Line 285 
285          if((codec->config.cpu & XVID_CPU_FORCE) <= 0)          if((codec->config.cpu & XVID_CPU_FORCE) <= 0)
286                  codec->config.cpu = init_param.cpu_flags;                  codec->config.cpu = init_param.cpu_flags;
287    
288          param.width = inhdr->bV4Width;          param.width = lpbiInput->bmiHeader.biWidth;
289          param.height = inhdr->bV4Height;          param.height = lpbiInput->bmiHeader.biHeight;
290          param.fincr = codec->fincr;          param.fincr = codec->fincr;
291          param.fbase = codec->fbase;          param.fbase = codec->fbase;
292    
293          param.rc_buffersize = codec->config.rc_buffersize;          param.min_quantizer = codec->config.min_pquant;
294            param.max_quantizer = codec->config.max_pquant;
         param.min_quantizer = codec->config.min_quant;  
         param.max_quantizer = codec->config.max_quant;  
295          param.max_key_interval = codec->config.max_key_interval;          param.max_key_interval = codec->config.max_key_interval;
296    
297    #ifdef _SMP
298            param.num_threads = codec->config.num_threads;
299    #endif
300    
301            param.global = 0;
302            if (codec->config.packed) param.global |= XVID_GLOBAL_PACKED;
303            if (codec->config.dx50bvop) param.global |= XVID_GLOBAL_DX50BVOP;
304            if (codec->config.debug) param.global |= XVID_GLOBAL_DEBUG;
305            if (codec->config.reduced_resolution) param.global |= XVID_GLOBAL_REDUCED;
306            param.max_bframes = codec->config.max_bframes;
307            param.bquant_ratio = codec->config.bquant_ratio;
308            param.bquant_offset = codec->config.bquant_offset;
309            param.frame_drop_ratio = codec->config.frame_drop_ratio;
310    
311          switch(xvid_encore(0, XVID_ENC_CREATE, &param, NULL))          switch(xvid_encore(0, XVID_ENC_CREATE, &param, NULL))
312          {          {
313          case XVID_ERR_FAIL :          case XVID_ERR_FAIL :
# Line 280  Line 324 
324          codec->framenum = 0;          codec->framenum = 0;
325          codec->keyspacing = 0;          codec->keyspacing = 0;
326    
327          codec->twopass.stats1 = codec->twopass.stats2 = INVALID_HANDLE_VALUE;          codec->twopass.hints = codec->twopass.stats1 = codec->twopass.stats2 = INVALID_HANDLE_VALUE;
328            codec->twopass.hintstream = NULL;
329    
330          return ICERR_OK;          return ICERR_OK;
331  }  }
# Line 291  Line 336 
336          if (codec->ehandle != NULL)          if (codec->ehandle != NULL)
337          {          {
338                  xvid_encore(codec->ehandle, XVID_ENC_DESTROY, NULL, NULL);                  xvid_encore(codec->ehandle, XVID_ENC_DESTROY, NULL, NULL);
339    
340                    if (codec->twopass.hints != INVALID_HANDLE_VALUE)
341                    {
342                            CloseHandle(codec->twopass.hints);
343                    }
344                  if (codec->twopass.stats1 != INVALID_HANDLE_VALUE)                  if (codec->twopass.stats1 != INVALID_HANDLE_VALUE)
345                  {                  {
346                          CloseHandle(codec->twopass.stats1);                          CloseHandle(codec->twopass.stats1);
# Line 299  Line 349 
349                  {                  {
350                          CloseHandle(codec->twopass.stats2);                          CloseHandle(codec->twopass.stats2);
351                  }                  }
352                    if (codec->twopass.hintstream != NULL)
353                    {
354                            free(codec->twopass.hintstream);
355                    }
356    
357                  codec->ehandle = NULL;                  codec->ehandle = NULL;
358    
359                    codec_2pass_finish(codec);
360          }          }
361    
362          return ICERR_OK;          return ICERR_OK;
# Line 308  Line 365 
365    
366  LRESULT compress(CODEC * codec, ICCOMPRESS * icc)  LRESULT compress(CODEC * codec, ICCOMPRESS * icc)
367  {  {
368          BITMAPV4HEADER * inhdr = (BITMAPV4HEADER *)icc->lpbiInput;          BITMAPINFOHEADER * inhdr = icc->lpbiInput;
369          BITMAPV4HEADER * outhdr = (BITMAPV4HEADER *)icc->lpbiOutput;          BITMAPINFOHEADER * outhdr = icc->lpbiOutput;
370          XVID_ENC_FRAME frame;          XVID_ENC_FRAME frame;
371          XVID_ENC_STATS stats;          XVID_ENC_STATS stats;
372    
# Line 324  Line 381 
381    
382          frame.general = 0;          frame.general = 0;
383          frame.motion = 0;          frame.motion = 0;
384            frame.intra = -1;
         if(codec->config.motion_search == 0)  
                 frame.intra = 1;  
385    
386          frame.general |= XVID_HALFPEL;          frame.general |= XVID_HALFPEL;
387    //      frame.general |= XVID_ME_EPZS;
388    
389            frame.general |= XVID_HQACPRED;
390    
391            frame.bframe_threshold = 255;
392    
393          if(codec->config.motion_search > 3)          if (codec->config.motion_search > 4)
394                  frame.general |= XVID_INTER4V;                  frame.general |= XVID_INTER4V;
395    
396          // we actually need "default/custom" selectbox for both inter/intra          if (codec->config.lum_masking)
397          // this will do for now                  frame.general |= XVID_LUMIMASKING;
398    
399            if (codec->config.interlacing)
400                    frame.general |= XVID_INTERLACING;
401    
402            if (codec->config.qpel) {
403                    frame.general |= XVID_QUARTERPEL;
404                    frame.motion |= PMV_QUARTERPELREFINE16 | PMV_QUARTERPELREFINE8;
405            }
406    
407            if (codec->config.gmc)
408                    frame.general |= XVID_GMC;
409    
410            if (codec->config.chromame)
411                    frame.motion |= PMV_CHROMA16 + PMV_CHROMA8;
412    
413          if ((codec->config.quant_type == QUANT_MODE_CUSTOM) && (codec->config.quant_type == 1))          if (codec->config.reduced_resolution)
414                    frame.general |= XVID_REDUCED;
415    
416            if (codec->config.chroma_opt)
417                    frame.general |= XVID_CHROMAOPT;
418    // added by koepi for credits greyscale
419    
420            check_greyscale_mode(&codec->config, &frame, codec->framenum);
421    
422    // end of koepi's addition
423    
424    
425    // fix 1pass modes/hinted MV by koepi
426            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))
427          {          {
428                  frame.general |= XVID_CUSTOM_QMATRIX;                  codec->config.hinted_me = 0;
                 frame.quant_intra_matrix = codec->config.qmatrix_intra;  
                 frame.quant_inter_matrix = codec->config.qmatrix_inter;  
429          }          }
430          else  // end of ugly hack
431    
432            if (codec->config.hinted_me && codec->config.mode == DLG_MODE_2PASS_1)
433          {          {
434                  frame.quant_intra_matrix = NULL;                  frame.hint.hintstream = codec->twopass.hintstream;
435                  frame.quant_inter_matrix = NULL;                  frame.hint.rawhints = 0;
436                    frame.general |= XVID_HINTEDME_GET;
437          }          }
438            else if (codec->config.hinted_me && (codec->config.mode == DLG_MODE_2PASS_2_EXT || codec->config.mode == DLG_MODE_2PASS_2_INT))
439            {
440                    DWORD read;
441                    DWORD blocksize;
442    
443          if(codec->config.quant_type == 0)                  frame.hint.hintstream = codec->twopass.hintstream;
444                  frame.general |= XVID_H263QUANT;                  frame.hint.rawhints = 0;
445          else                  frame.general |= XVID_HINTEDME_SET;
                 frame.general |= XVID_MPEGQUANT;  
446    
447          if(((codec->config.mode == DLG_MODE_2PASS_1) ? 0 : codec->config.lum_masking) == 1)                  if (codec->twopass.hints == INVALID_HANDLE_VALUE)
448                  frame.general |= XVID_LUMIMASKING;                  {
449                            codec->twopass.hints = CreateFile(codec->config.hintfile, GENERIC_READ, 0, 0, OPEN_EXISTING, 0, 0);
450                            if (codec->twopass.hints == INVALID_HANDLE_VALUE)
451                            {
452                                    DEBUGERR("couldn't open hints file");
453                                    return ICERR_ERROR;
454                            }
455                    }
456                    if (!ReadFile(codec->twopass.hints, &blocksize, sizeof(DWORD), &read, 0) || read != sizeof(DWORD) ||
457                            !ReadFile(codec->twopass.hints, frame.hint.hintstream, blocksize, &read, 0) || read != blocksize)
458                    {
459                            DEBUGERR("couldn't read from hints file");
460                            return ICERR_ERROR;
461                    }
462            }
463    
464            frame.motion |= pmvfast_presets[codec->config.motion_search];
465    
466            switch (codec->config.vhq_mode)
467            {
468            case VHQ_MODE_DECISION :
469                    frame.general |= XVID_MODEDECISION_BITS;
470                    break;
471    
472            case VHQ_LIMITED_SEARCH :
473                    frame.general |= XVID_MODEDECISION_BITS;
474                    frame.motion |= HALFPELREFINE16_BITS;
475                    frame.motion |= QUARTERPELREFINE16_BITS;
476                    break;
477    
478            case VHQ_MEDIUM_SEARCH :
479                    frame.general |= XVID_MODEDECISION_BITS;
480                    frame.motion |= HALFPELREFINE16_BITS;
481                    frame.motion |= HALFPELREFINE8_BITS;
482                    frame.motion |= QUARTERPELREFINE16_BITS;
483                    frame.motion |= QUARTERPELREFINE8_BITS;
484                    frame.motion |= CHECKPREDICTION_BITS;
485                    break;
486    
487            case VHQ_WIDE_SEARCH :
488                    frame.general |= XVID_MODEDECISION_BITS;
489                    frame.motion |= HALFPELREFINE16_BITS;
490                    frame.motion |= HALFPELREFINE8_BITS;
491                    frame.motion |= QUARTERPELREFINE16_BITS;
492                    frame.motion |= QUARTERPELREFINE8_BITS;
493                    frame.motion |= CHECKPREDICTION_BITS;
494                    frame.motion |= EXTSEARCH_BITS;
495                    break;
496    
497          frame.motion = pmvfast_presets[codec->config.motion_search];          default :
498                    break;
499            }
500    
501          frame.image = icc->lpInput;          frame.image = icc->lpInput;
502            frame.stride = (((icc->lpbiInput->biWidth * icc->lpbiInput->biBitCount) + 31) & ~31) >> 3;
503    
504          if ((frame.colorspace = get_colorspace(inhdr)) == XVID_CSP_NULL) {          if ((frame.colorspace = get_colorspace(inhdr)) == XVID_CSP_NULL)
505                  return ICERR_BADFORMAT;                  return ICERR_BADFORMAT;
506          }  
507            if (frame.colorspace == XVID_CSP_I420 || frame.colorspace == XVID_CSP_YV12)
508                    frame.stride = (frame.stride*2)/3;
509    
510          frame.bitstream = icc->lpOutput;          frame.bitstream = icc->lpOutput;
511          frame.length = icc->lpbiOutput->biSizeImage;          frame.length = icc->lpbiOutput->biSizeImage;
         frame.intra = -1;  
512    
513          switch (codec->config.mode)          switch (codec->config.mode)
514          {          {
# Line 391  Line 533 
533                  }                  }
534                  if (codec->config.dummy2pass)                  if (codec->config.dummy2pass)
535                  {                  {
536                          outhdr->bV4SizeImage = codec->twopass.bytes2;                          outhdr->biSizeImage = codec->twopass.bytes2;
537                          *icc->lpdwFlags = (codec->twopass.nns1.quant & NNSTATS_KEYFRAME) ? AVIIF_KEYFRAME : 0;                          *icc->lpdwFlags = (codec->twopass.nns1.quant & NNSTATS_KEYFRAME) ? AVIIF_KEYFRAME : 0;
538                          return ICERR_OK;                          return ICERR_OK;
539                  }                  }
540                  break;                  break;
541    
542          case DLG_MODE_NULL :          case DLG_MODE_NULL :
543                  outhdr->bV4SizeImage = 0;                  outhdr->biSizeImage = 0;
544                  *icc->lpdwFlags = AVIIF_KEYFRAME;                  *icc->lpdwFlags = AVIIF_KEYFRAME;
545                  return ICERR_OK;                  return ICERR_OK;
546    
547          default :          default :
548                  DEBUG("Invalid encoding mode");                  DEBUGERR("Invalid encoding mode");
549                  return ICERR_ERROR;                  return ICERR_ERROR;
550          }          }
551    
552          // force keyframe spacing in 2-pass modes          if (codec->config.quant_type == QUANT_MODE_H263)
553          if ((codec->keyspacing < codec->config.min_key_interval && codec->framenum) &&          {
554                  (codec->config.mode == DLG_MODE_2PASS_1 || codec->config.mode == DLG_MODE_2PASS_2_INT ||                  frame.general |= XVID_H263QUANT;
555                  codec->config.mode == DLG_MODE_2PASS_2_EXT))          }
556            else
557            {
558                    frame.general |= XVID_MPEGQUANT;
559    
560                    // we actually need "default/custom" selectbox for both inter/intra
561                    // this will do for now
562                    if (codec->config.quant_type == QUANT_MODE_CUSTOM)
563                    {
564                            frame.general |= XVID_CUSTOM_QMATRIX;
565                            frame.quant_intra_matrix = codec->config.qmatrix_intra;
566                            frame.quant_inter_matrix = codec->config.qmatrix_inter;
567                    }
568                    else
569                    {
570                            frame.quant_intra_matrix = NULL;
571                            frame.quant_inter_matrix = NULL;
572                    }
573            }
574    
575            // force keyframe spacing in 2-pass 1st pass
576            if (codec->config.motion_search == 0)
577            {
578                    frame.intra = 1;
579            }
580            else if (codec->keyspacing < codec->config.min_key_interval && codec->framenum)
581          {          {
582                  DEBUG("current frame forced to p-frame");                  DEBUG("current frame forced to p-frame");
583                  frame.intra = 0;                  frame.intra = 0;
584          }          }
585    
586            frame.bquant = 0;
587    
588    //      OutputDebugString(" ");
589          switch (xvid_encore(codec->ehandle, XVID_ENC_ENCODE, &frame, &stats))          switch (xvid_encore(codec->ehandle, XVID_ENC_ENCODE, &frame, &stats))
590          {          {
591          case XVID_ERR_FAIL :          case XVID_ERR_FAIL :
# Line 428  Line 598 
598                  return ICERR_BADFORMAT;                  return ICERR_BADFORMAT;
599          }          }
600    
601          if (frame.intra)          if (frame.intra==1)
602          {          {
603                  codec->keyspacing = 0;                  codec->keyspacing = 0;
604                  *icc->lpdwFlags = AVIIF_KEYFRAME;                  *icc->lpdwFlags = AVIIF_KEYFRAME;
# Line 437  Line 607 
607          {          {
608                  *icc->lpdwFlags = 0;                  *icc->lpdwFlags = 0;
609          }          }
610            if (frame.length == 0) { frame.length = 1; *((unsigned char*)frame.bitstream) = 0x7f; }
611    
612            outhdr->biSizeImage = frame.length;
613    
614            if (codec->config.mode == DLG_MODE_2PASS_1 && codec->config.discard1pass)
615            {
616                    outhdr->biSizeImage = 0;
617            }
618    
619          outhdr->bV4SizeImage = frame.length;          if (frame.general & XVID_HINTEDME_GET)
620            {
621                    DWORD wrote;
622                    DWORD blocksize = frame.hint.hintlength;
623    
624          if (codec->config.mode == DLG_MODE_2PASS_1)                  if (codec->twopass.hints == INVALID_HANDLE_VALUE)
625                    {
626                            codec->twopass.hints = CreateFile(codec->config.hintfile, GENERIC_WRITE, 0, 0, CREATE_ALWAYS, 0, 0);
627                            if (codec->twopass.hints        == INVALID_HANDLE_VALUE)
628          {          {
629                  if (codec->config.discard1pass)                                  DEBUGERR("couldn't create hints file");
630                                    return ICERR_ERROR;
631                            }
632                    }
633                    if (!WriteFile(codec->twopass.hints, &frame.hint.hintlength, sizeof(int), &wrote, 0) || wrote != sizeof(int) ||
634                            !WriteFile(codec->twopass.hints, frame.hint.hintstream, blocksize, &wrote, 0) || wrote != blocksize)
635                  {                  {
636                          outhdr->bV4SizeImage = 0;                          DEBUGERR("couldn't write to hints file");
637                            return ICERR_ERROR;
638                  }                  }
639          }          }
640    
641    //quick fix for delayed frames
642    //      if (frame.intra != 5)
643          codec_2pass_update(codec, &frame, &stats);          codec_2pass_update(codec, &frame, &stats);
644    
645          ++codec->framenum;          ++codec->framenum;
# Line 462  Line 654 
654    
655  LRESULT decompress_query(CODEC * codec, BITMAPINFO *lpbiInput, BITMAPINFO *lpbiOutput)  LRESULT decompress_query(CODEC * codec, BITMAPINFO *lpbiInput, BITMAPINFO *lpbiOutput)
656  {  {
657          BITMAPV4HEADER * inhdr = (BITMAPV4HEADER *)&lpbiInput->bmiHeader;          BITMAPINFOHEADER * inhdr = &lpbiInput->bmiHeader;
658          BITMAPV4HEADER * outhdr = (BITMAPV4HEADER *)&lpbiOutput->bmiHeader;          BITMAPINFOHEADER * outhdr = &lpbiOutput->bmiHeader;
659    
660          if (lpbiInput == NULL)          if (lpbiInput == NULL)
661          {          {
662                  return ICERR_ERROR;                  return ICERR_ERROR;
663          }          }
664    
665          if (inhdr->bV4V4Compression != FOURCC_XVID && inhdr->bV4V4Compression != FOURCC_DIVX)          if (inhdr->biCompression != FOURCC_XVID && inhdr->biCompression != FOURCC_DIVX && inhdr->biCompression != FOURCC_DX50 && get_colorspace(inhdr) == XVID_CSP_NULL)
666          {          {
667                  return ICERR_BADFORMAT;                  return ICERR_BADFORMAT;
668          }          }
# Line 480  Line 672 
672                  return ICERR_OK;                  return ICERR_OK;
673          }          }
674    
675          if (inhdr->bV4Width != outhdr->bV4Width ||          if (inhdr->biWidth != outhdr->biWidth ||
676                  inhdr->bV4Height != outhdr->bV4Height ||                  inhdr->biHeight != outhdr->biHeight ||
677                  get_colorspace(outhdr) == XVID_CSP_NULL)                  get_colorspace(outhdr) == XVID_CSP_NULL)
678          {          {
679                  return ICERR_BADFORMAT;                  return ICERR_BADFORMAT;
# Line 493  Line 685 
685    
686  LRESULT decompress_get_format(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)  LRESULT decompress_get_format(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)
687  {  {
688          BITMAPV4HEADER * inhdr = (BITMAPV4HEADER *)&lpbiInput->bmiHeader;          BITMAPINFOHEADER * inhdr = &lpbiInput->bmiHeader;
689          BITMAPV4HEADER * outhdr = (BITMAPV4HEADER *)&lpbiOutput->bmiHeader;          BITMAPINFOHEADER * outhdr = &lpbiOutput->bmiHeader;
690          LRESULT result;          LRESULT result;
691    
692          if (lpbiOutput == NULL)          if (lpbiOutput == NULL)
693          {          {
694                  return sizeof(BITMAPV4HEADER);                  return sizeof(BITMAPINFOHEADER);
695            }
696    
697            /* --- yv12 --- */
698    
699            if (get_colorspace(inhdr) != XVID_CSP_NULL) {
700                    memcpy(outhdr, inhdr, sizeof(BITMAPINFOHEADER));
701                    // XXX: should we set outhdr->biSize ??
702                    return ICERR_OK;
703          }          }
704            /* --- yv12 --- */
705    
706          result = decompress_query(codec, lpbiInput, lpbiOutput);          result = decompress_query(codec, lpbiInput, NULL);
707          if (result != ICERR_OK)          if (result != ICERR_OK)
708          {          {
709                  return result;                  return result;
710          }          }
711    
712          memcpy(outhdr, inhdr, sizeof(BITMAPV4HEADER));          outhdr->biSize = sizeof(BITMAPINFOHEADER);
713          outhdr->bV4Size = sizeof(BITMAPV4HEADER);          outhdr->biWidth = inhdr->biWidth;
714          outhdr->bV4V4Compression = FOURCC_YUY2;          outhdr->biHeight = inhdr->biHeight;
715          outhdr->bV4SizeImage = outhdr->bV4Width * outhdr->bV4Height * outhdr->bV4BitCount;          outhdr->biPlanes = 1;
716          outhdr->bV4XPelsPerMeter = 0;          outhdr->biBitCount = 24;
717          outhdr->bV4YPelsPerMeter = 0;          outhdr->biCompression = BI_RGB; /* sonic foundry vegas video v3 only supports BI_RGB */
718          outhdr->bV4ClrUsed = 0;          outhdr->biSizeImage = outhdr->biWidth * outhdr->biHeight * outhdr->biBitCount;
719          outhdr->bV4ClrImportant = 0;          outhdr->biXPelsPerMeter = 0;
720            outhdr->biYPelsPerMeter = 0;
721            outhdr->biClrUsed = 0;
722            outhdr->biClrImportant = 0;
723    
724          return ICERR_OK;          return ICERR_OK;
725  }  }
# Line 523  Line 727 
727    
728  LRESULT decompress_begin(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)  LRESULT decompress_begin(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)
729  {  {
         BITMAPV4HEADER * inhdr = (BITMAPV4HEADER *)&lpbiInput->bmiHeader;  
         BITMAPV4HEADER * outhdr = (BITMAPV4HEADER *)&lpbiOutput->bmiHeader;  
730          XVID_DEC_PARAM param;          XVID_DEC_PARAM param;
731          XVID_INIT_PARAM init_param;          XVID_INIT_PARAM init_param;
732    
# Line 535  Line 737 
737                  codec->config.cpu = init_param.cpu_flags;                  codec->config.cpu = init_param.cpu_flags;
738          }          }
739    
740          param.width = inhdr->bV4Width;          param.width = lpbiInput->bmiHeader.biWidth;
741          param.height = inhdr->bV4Height;          param.height = lpbiInput->bmiHeader.biHeight;
742    
743          switch(xvid_decore(0, XVID_DEC_CREATE, &param, NULL))          switch(xvid_decore(0, XVID_DEC_CREATE, &param, NULL))
744          {          {
# Line 569  Line 771 
771    
772  LRESULT decompress(CODEC * codec, ICDECOMPRESS * icd)  LRESULT decompress(CODEC * codec, ICDECOMPRESS * icd)
773  {  {
         BITMAPV4HEADER * inhdr = (BITMAPV4HEADER *)icd->lpbiInput;  
         BITMAPV4HEADER * outhdr = (BITMAPV4HEADER *)icd->lpbiOutput;  
774          XVID_DEC_FRAME frame;          XVID_DEC_FRAME frame;
775    
776          frame.bitstream = icd->lpInput;          frame.bitstream = icd->lpInput;
777          frame.length = icd->lpbiInput->biSizeImage;          frame.length = icd->lpbiInput->biSizeImage;
778            frame.general = XVID_DEC_LOWDELAY;      /* force low_delay_default mode */
779            if (codec->config.deblock_y)
780                    frame.general |= XVID_DEC_DEBLOCKY;
781            if (codec->config.deblock_uv)
782                    frame.general |= XVID_DEC_DEBLOCKUV;
783    
784          frame.image = icd->lpOutput;          frame.image = icd->lpOutput;
785          frame.stride = icd->lpbiOutput->biWidth;          frame.stride = (((icd->lpbiOutput->biWidth * icd->lpbiOutput->biBitCount) + 31) & ~31) >> 3;
786    
787            /* --- yv12 --- */
788            if (icd->lpbiInput->biCompression != FOURCC_XVID &&
789                     icd->lpbiInput->biCompression != FOURCC_DIVX &&
790                     icd->lpbiInput->biCompression != FOURCC_DX50)
791            {
792                    XVID_INIT_CONVERTINFO convert;
793                    DEBUGFOURCC("input", icd->lpbiInput->biCompression);
794                    DEBUGFOURCC("output", icd->lpbiOutput->biCompression);
795                    convert.input.colorspace = get_colorspace(icd->lpbiInput);
796                    convert.input.y = icd->lpInput;
797                    convert.input.y_stride = (((icd->lpbiInput->biWidth *icd->lpbiInput->biBitCount) + 31) & ~31) >> 3;
798                    if (convert.input.colorspace == XVID_CSP_I420 || convert.input.colorspace == XVID_CSP_YV12)
799                            convert.input.y_stride = (convert.input.y_stride*2)/3;
800    
801                    convert.output.colorspace = get_colorspace(icd->lpbiOutput);
802                    convert.output.y = icd->lpOutput;
803                    convert.output.y_stride = (((icd->lpbiOutput->biWidth *icd->lpbiOutput->biBitCount) + 31) & ~31) >> 3;
804                    if (convert.output.colorspace == XVID_CSP_I420 || convert.output.colorspace == XVID_CSP_YV12)
805                            convert.output.y_stride = (convert.output.y_stride*2)/3;
806    
807                    convert.width = icd->lpbiInput->biWidth;
808                    convert.height = icd->lpbiInput->biHeight;
809                    convert.interlacing = 0;
810                    if (convert.input.colorspace == XVID_CSP_NULL ||
811                            convert.output.colorspace == XVID_CSP_NULL ||
812                            xvid_init(NULL, XVID_INIT_CONVERT, &convert, NULL) != XVID_ERR_OK)
813                    {
814                             return ICERR_BADFORMAT;
815                    }
816                    return ICERR_OK;
817            }
818            /* --- yv12 --- */
819    
820    
821          if (~((icd->dwFlags & ICDECOMPRESS_HURRYUP) | (icd->dwFlags & ICDECOMPRESS_UPDATE)))          if (~((icd->dwFlags & ICDECOMPRESS_HURRYUP) | (icd->dwFlags & ICDECOMPRESS_UPDATE) | (icd->dwFlags & ICDECOMPRESS_PREROLL)))
822          {          {
823                  if ((frame.colorspace = get_colorspace(outhdr)) == XVID_CSP_NULL)                  if ((frame.colorspace = get_colorspace(icd->lpbiOutput)) == XVID_CSP_NULL)
824                  {                  {
825                          return ICERR_BADFORMAT;                          return ICERR_BADFORMAT;
826                  }                  }
# Line 591  Line 830 
830                  frame.colorspace = XVID_CSP_NULL;                  frame.colorspace = XVID_CSP_NULL;
831          }          }
832    
833            if (frame.colorspace == XVID_CSP_I420 || frame.colorspace == XVID_CSP_YV12)
834                    frame.stride = (frame.stride*2)/3;
835    
836          switch (xvid_decore(codec->dhandle, XVID_DEC_DECODE, &frame, NULL))          switch (xvid_decore(codec->dhandle, XVID_DEC_DECODE, &frame, NULL))
837          {          {
838          case XVID_ERR_FAIL :          case XVID_ERR_FAIL :
# Line 606  Line 848 
848          return ICERR_OK;          return ICERR_OK;
849  }  }
850    
851  int codec_2pass_init(CODEC* codec)  int codec_get_quant(CODEC* codec, XVID_ENC_FRAME* frame)
852  {  {
         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;  
   
853          switch (codec->config.mode)          switch (codec->config.mode)
854          {          {
855          case DLG_MODE_2PASS_1 :          case DLG_MODE_VBR_QUAL :
856                  twopass->stats1 = CreateFile(codec->config.stats1, GENERIC_WRITE, FILE_SHARE_READ, 0, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, 0);                  if (codec_is_in_credits(&codec->config, codec->framenum))
                 if (twopass->stats1 == INVALID_HANDLE_VALUE)  
857                  {                  {
858                          DEBUG("2pass init error - couldn't create stats1");  // added by koepi for credits greyscale
859                          return ICERR_ERROR;  
860                  }                          check_greyscale_mode(&codec->config, frame, codec->framenum);
861                  if (WriteFile(twopass->stats1, &version, sizeof(DWORD), &wrote, 0) == 0 || wrote != sizeof(DWORD))  
862    // end of koepi's addition
863    
864                            switch (codec->config.credits_mode)
865                  {                  {
866                          CloseHandle(twopass->stats1);                          case CREDITS_MODE_RATE :
867                          twopass->stats1 = INVALID_HANDLE_VALUE;                                  frame->quant = codec_get_vbr_quant(&codec->config, codec->config.quality * codec->config.credits_rate / 100);
                         DEBUG("2pass init error - couldn't write to stats1");  
                         return ICERR_ERROR;  
                 }  
868                  break;                  break;
869    
870          case DLG_MODE_2PASS_2_INT :                          case CREDITS_MODE_QUANT :
871          case DLG_MODE_2PASS_2_EXT :                                  frame->quant = codec->config.credits_quant_p;
872                  twopass->stats1 = CreateFile(codec->config.stats1, GENERIC_READ, FILE_SHARE_READ, 0, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0);                                  break;
873                  if (twopass->stats1 == INVALID_HANDLE_VALUE)  
874                  {                          default :
875                          DEBUG("2pass init error - couldn't open stats1");                                  DEBUGERR("Can't use credits size mode in quality mode");
876                          return ICERR_ERROR;                          return ICERR_ERROR;
877                  }                  }
                 if (ReadFile(twopass->stats1, &version, sizeof(DWORD), &read, 0) == 0 || read != sizeof(DWORD))  
                 {  
                         CloseHandle(twopass->stats1);  
                         twopass->stats1 = INVALID_HANDLE_VALUE;  
                         DEBUG("2pass init error - couldn't read from stats1");  
                         return ICERR_ERROR;  
878                  }                  }
879                  if (version != -20)                  else
880                  {                  {
881                          CloseHandle(twopass->stats1);  // added by koepi for credits greyscale
                         twopass->stats1 = INVALID_HANDLE_VALUE;  
                         DEBUG("2pass init error - wrong .stats version");  
                         return ICERR_ERROR;  
                 }  
882    
883                  if (codec->config.mode == DLG_MODE_2PASS_2_EXT)                          check_greyscale_mode(&codec->config, frame, codec->framenum);
                 {  
                         if (twopass->stats2 != INVALID_HANDLE_VALUE)  
                         {  
                                 CloseHandle(twopass->stats2);  
                         }  
884    
885                          twopass->stats2 = CreateFile(codec->config.stats2, GENERIC_READ, FILE_SHARE_READ, 0, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0);  // end of koepi's addition
886    
887                          if (twopass->stats2 == INVALID_HANDLE_VALUE)                          frame->quant = codec_get_vbr_quant(&codec->config, codec->config.quality);
                         {  
                                 CloseHandle(twopass->stats1);  
                                 twopass->stats1 = INVALID_HANDLE_VALUE;  
                                 DEBUG("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;  
                                 DEBUG("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;  
                                 DEBUG("2pass init error - wrong .stats version");  
                                 return ICERR_ERROR;  
888                          }                          }
889                    return ICERR_OK;
890    
891                          while (1)          case DLG_MODE_VBR_QUANT :
892                          {                  if (codec_is_in_credits(&codec->config, codec->framenum))
                                 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))  
893                                  {                                  {
894                                          DWORD err = GetLastError();  // added by koepi for credits greyscale
895    
896                            check_greyscale_mode(&codec->config, frame, codec->framenum);
897    
898    // end of koepi's addition
899    
900                                          if (err == ERROR_HANDLE_EOF || err == ERROR_SUCCESS)                          switch (codec->config.credits_mode)
901                                          {                                          {
902                            case CREDITS_MODE_RATE :
903                                    frame->quant =
904                                            codec->config.max_pquant -
905                                            ((codec->config.max_pquant - codec->config.quant) * codec->config.credits_rate / 100);
906                                                  break;                                                  break;
907                                          }  
908                                          else                          case CREDITS_MODE_QUANT :
909                                          {                                  frame->quant = codec->config.credits_quant_p;
910                                                  CloseHandle(twopass->stats1);                                  break;
911                                                  CloseHandle(twopass->stats2);  
912                                                  twopass->stats1 = INVALID_HANDLE_VALUE;                          default :
913                                                  twopass->stats2 = INVALID_HANDLE_VALUE;                                  DEBUGERR("Can't use credits size mode in quantizer mode");
                                                 DEBUG("2pass init error - incomplete stats1/stats2 record?");  
914                                                  return ICERR_ERROR;                                                  return ICERR_ERROR;
915                                          }                                          }
916                                  }                                  }
917                    else
                                 if (!codec_is_in_credits(&codec->config, frames))  
                                 {  
                                         if (twopass->nns1.quant & NNSTATS_KEYFRAME)  
918                                          {                                          {
919                                                  i_boost_total = twopass->nns2.bytes * codec->config.keyframe_boost / 100;  // added by koepi for credits greyscale
                                                 i_total += twopass->nns2.bytes;  
                                                 ++i_frames;  
                                         }  
920    
921                                          total += twopass->nns2.bytes;                          check_greyscale_mode(&codec->config, frame, codec->framenum);
                                 }  
                                 else  
                                         ++credits_frames;  
922    
923                                  ++frames;  // end of koepi's addition
924    
925                            frame->quant = codec->config.quant;
926                          }                          }
927                    return ICERR_OK;
928    
929                          twopass->movie_curve = ((float)(total + i_boost_total) / total);          case DLG_MODE_2PASS_1 :
930                          twopass->average_frame = ((double)(total - i_total) / (frames - credits_frames - i_frames) / twopass->movie_curve);  // added by koepi for credits greyscale
931    
932                          SetFilePointer(twopass->stats1, sizeof(DWORD), 0, FILE_BEGIN);                  check_greyscale_mode(&codec->config, frame, codec->framenum);
                         SetFilePointer(twopass->stats2, sizeof(DWORD), 0, FILE_BEGIN);  
933    
934                          // perform prepass to compensate for over/undersizing  // end of koepi's addition
                         frames = 0;  
935    
936                          while (1)                  if (codec->config.credits_mode == CREDITS_MODE_QUANT)
                         {  
                                 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))  
937                                  {                                  {
938                                          DWORD err = GetLastError();                          if (codec_is_in_credits(&codec->config, codec->framenum))
   
                                         if (err == ERROR_HANDLE_EOF || err == ERROR_SUCCESS)  
939                                          {                                          {
940                                                  break;                                  frame->quant = codec->config.credits_quant_p;
941                                          }                                          }
942                                          else                                          else
943                                          {                                          {
944                                                  CloseHandle(twopass->stats1);                                  frame->quant = 2;
                                                 CloseHandle(twopass->stats2);  
                                                 twopass->stats1 = INVALID_HANDLE_VALUE;  
                                                 twopass->stats2 = INVALID_HANDLE_VALUE;  
                                                 DEBUG("2pass init error - incomplete stats1/stats2 record?");  
                                                 return ICERR_ERROR;  
                                         }  
945                                  }                                  }
   
                                 if (!codec_is_in_credits(&codec->config, frames) &&  
                                         !(twopass->nns1.quant & NNSTATS_KEYFRAME))  
                                 {  
                                         double dbytes = twopass->nns2.bytes / twopass->movie_curve;  
                                         total1 += dbytes;  
   
                                         if (dbytes > twopass->average_frame)  
                                         {  
                                                 total2 += ((double)dbytes + (twopass->average_frame - dbytes) *  
                                                         codec->config.curve_compression_high / 100.0);  
946                                          }                                          }
947                                          else                                          else
948                                          {                                          {
949                                                  total2 += ((double)dbytes + (twopass->average_frame - dbytes) *                          frame->quant = 2;
                                                         codec->config.curve_compression_low / 100.0);  
                                         }  
950                                  }                                  }
951                    return ICERR_OK;
952    
953                                  ++frames;          default:
954                    DEBUGERR("get quant: invalid mode");
955                    return ICERR_ERROR;
956                          }                          }
   
                         twopass->curve_comp_scale = total1 / total2;  
   
                         {  
                                 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);  
957                          }                          }
958    
                         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();  
959    
960                                          if (err == ERROR_HANDLE_EOF || err == ERROR_SUCCESS)  int codec_is_in_credits(CONFIG* config, int framenum)
961                                          {                                          {
962                                                  break;          if (config->credits_start)
                                         }  
                                         else  
963                                          {                                          {
964                                                  CloseHandle(twopass->stats1);                  if (framenum >= config->credits_start_begin &&
965                                                  twopass->stats1 = INVALID_HANDLE_VALUE;                          framenum <= config->credits_start_end)
966                                                  DEBUG("2pass init error - incomplete stats2 record?");                  {
967                                                  return ICERR_ERROR;                          return CREDITS_START;
968                                          }                                          }
969                                  }                                  }
970    
971                                  if (codec_is_in_credits(&codec->config, frames) == CREDITS_START)          if (config->credits_end)
972                                  {                                  {
973                                          start += twopass->nns1.bytes;                  if (framenum >= config->credits_end_begin &&
974                                          ++credits_frames;                          framenum <= config->credits_end_end)
                                 }  
                                 else if (codec_is_in_credits(&codec->config, frames) == CREDITS_END)  
975                                  {                                  {
976                                          end += twopass->nns1.bytes;                          return CREDITS_END;
                                         ++credits_frames;  
977                                  }                                  }
                                 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;  
978                                  }                                  }
979    
980                                  total += twopass->nns1.bytes;          return 0;
   
                                 ++frames;  
981                          }                          }
982    
                         // compensate for avi frame overhead  
                         desired -= frames * 24;  
983    
984                          switch (codec->config.credits_mode)  int codec_get_vbr_quant(CONFIG* config, int quality)
985                          {                          {
986                          case CREDITS_MODE_RATE :          static float fquant_running = 0;
987            static int my_quality = -1;
988                                  // credits curve = (total / desired_size) * (100 / credits_rate)          int     quant;
                                 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;  
989    
990                          case CREDITS_MODE_QUANT :          // if quality changes, recalculate fquant (credits)
991            if (quality != my_quality)
                                 // 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;  
   
                         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;  
                                                 DEBUG("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 (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;  
   
                         {  
                                 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);  
                 }  
   
                 twopass->overflow = 0;  
   
                 break;  
         }  
   
         return ICERR_OK;  
 }  
   
   
 int codec_get_quant(CODEC* codec, XVID_ENC_FRAME* frame)  
 {  
         switch (codec->config.mode)  
         {  
         case DLG_MODE_VBR_QUAL :  
                 if (codec_is_in_credits(&codec->config, codec->framenum))  
                 {  
                         switch (codec->config.credits_mode)  
                         {  
                         case CREDITS_MODE_RATE :  
                                 frame->quant = codec_get_vbr_quant(&codec->config, codec->config.quality * codec->config.credits_rate / 100);  
                                 break;  
   
                         case CREDITS_MODE_QUANT :  
                                 frame->quant = codec->config.credits_quant_p;  
                                 break;  
   
                         default :  
                                 DEBUG("Can't use credits size mode in quality mode");  
                                 return ICERR_ERROR;  
                         }  
                 }  
                 else  
                 {  
                         frame->quant = codec_get_vbr_quant(&codec->config, codec->config.quality);  
                 }  
                 return ICERR_OK;  
   
         case DLG_MODE_VBR_QUANT :  
                 if (codec_is_in_credits(&codec->config, codec->framenum))  
                 {  
                         switch (codec->config.credits_mode)  
                         {  
                         case CREDITS_MODE_RATE :  
                                 frame->quant =  
                                         codec->config.max_quant -  
                                         ((codec->config.max_quant - codec->config.quant) * codec->config.credits_rate / 100);  
                                 break;  
   
                         case CREDITS_MODE_QUANT :  
                                 frame->quant = codec->config.credits_quant_p;  
                                 break;  
   
                         default :  
                                 DEBUG("Can't use credits size mode in quantizer mode");  
                                 return ICERR_ERROR;  
                         }  
                 }  
                 else  
                 {  
                         frame->quant = codec->config.quant;  
                 }  
                 return ICERR_OK;  
   
         case DLG_MODE_2PASS_1 :  
                 if (codec->config.credits_mode == CREDITS_MODE_QUANT)  
                 {  
                         if (codec_is_in_credits(&codec->config, codec->framenum))  
                         {  
                                 frame->quant = codec->config.credits_quant_p;  
                         }  
                         else  
                         {  
                                 frame->quant = 2;  
                         }  
                 }  
                 else  
                 {  
                         frame->quant = 2;  
                 }  
                 return ICERR_OK;  
   
         default:  
                 DEBUG("get quant: invalid mode");  
                 return ICERR_ERROR;  
         }  
 }  
   
   
 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))  
         {  
                 DEBUG("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))  
                 {  
                         DEBUG("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))  
992                          {                          {
993                                  bytes2 = (int)curve_comp_error;                  config->fquant = 0;
                         }  
994                  }                  }
995    
996                  curve_comp_error -= bytes2;          my_quality = quality;
997    
998                  if ((codec->config.curve_compression_high + codec->config.curve_compression_low) &&          // desired quantiser = (maxQ-minQ)/100 * (100-qual) + minQ
999                          !frame->intra)          if (!config->fquant)
                 {  
                         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  
1000                          {                          {
1001                                  curve_temp = twopass->curve_comp_scale *                  config->fquant =
1002                                          ((double)dbytes + (twopass->average_frame - dbytes) *                          ((float) (config->max_pquant - config->min_pquant) / 100) *
1003                                          codec->config.curve_compression_low / 100.0);                          (100 - quality) +
1004                          }                          (float) config->min_pquant;
1005    
1006                          bytes2 += ((int)curve_temp);                  fquant_running = config->fquant;
                         curve_comp_error += curve_temp - ((int)curve_temp);  
                 }  
                 else  
                 {  
                         curve_comp_error += dbytes - ((int)dbytes);  
                         bytes2 += ((int)dbytes);  
1007                  }                  }
1008    
1009                  // cap bytes2 to first pass size, lowers number of quant=1 frames          if (fquant_running < config->min_pquant)
                 if (bytes2 > bytes1)  
1010                  {                  {
1011                          curve_comp_error += bytes2 - bytes1;                  fquant_running = (float) config->min_pquant;
                         bytes2 = bytes1;  
1012                  }                  }
1013                  else if (bytes2 < 1)          else if(fquant_running > config->max_pquant)
1014                  {                  {
1015                          curve_comp_error += --bytes2;                  fquant_running = (float) config->max_pquant;
                         bytes2 = 1;  
                 }  
1016          }          }
1017    
1018          twopass->desired_bytes2 = bytes2;          quant = (int) fquant_running;
   
         // 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;  
         }  
1019    
1020          // Foxer: make sure overflow doesn't run away          // add error between fquant and quant to fquant_running
1021          if (overflow > bytes2 * 6 / 10)          fquant_running += config->fquant - quant;
         {  
                 bytes2 += (overflow <= bytes2) ? bytes2 * 6 / 10 : overflow * 6 / 10;  
         }  
         else if (overflow < bytes2 * -6 / 10)  
         {  
                 bytes2 += bytes2 * -6 / 10;  
         }  
         else  
         {  
                 bytes2 += overflow;  
         }  
1022    
1023          if (bytes2 < 1)          return quant;
         {  
                 bytes2 = 1;  
1024          }          }
1025    
1026          twopass->bytes1 = bytes1;  // added by koepi for credits greyscale
         twopass->bytes2 = bytes2;  
1027    
1028          // very 'simple' quant<->filesize relationship  int check_greyscale_mode(CONFIG* config, XVID_ENC_FRAME* frame, int framenum)
         frame->quant = ((twopass->nns1.quant & ~NNSTATS_KEYFRAME) * bytes1) / bytes2;  
1029    
         if (frame->quant < 1)  
1030          {          {
                 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;  
1031    
1032                  if (quant_error[frame->quant] >= 1.0)          if ((codec_is_in_credits(config, framenum)) && (config->mode!=DLG_MODE_CBR))
                 {  
                         quant_error[frame->quant] -= 1.0;  
                         ++frame->quant;  
                 }  
         }  
1033    
         // we're done with credits  
         if (codec_is_in_credits(&codec->config, codec->framenum))  
1034          {          {
                 return ICERR_OK;  
         }  
1035    
1036          if (frame->intra)                  if (config->credits_greyscale)
         {  
                 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_quant)  
                 {  
                         frame->quant = codec->config.max_quant;  
                 }  
                 if (frame->quant < codec->config.min_quant)  
                 {  
                         frame->quant = codec->config.min_quant;  
                 }  
1037    
                 // subsequent frame quants can only be +- 2  
                 if (last_quant)  
                 {  
                         if (frame->quant > last_quant + 2)  
1038                          {                          {
                                 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");  
                         }  
                 }  
         }  
1039    
1040          last_quant = frame->quant;                          if ((frame->general && XVID_GREYSCALE))  // use only if not already in greyscale
1041    
1042          return ICERR_OK;                                  frame->general |= XVID_GREYSCALE;
 }  
1043    
1044                    } else {
1045    
1046  int codec_2pass_update(CODEC* codec, XVID_ENC_FRAME* frame, XVID_ENC_STATS* stats)                          if (!(frame->general && XVID_GREYSCALE))  // if movie is in greyscale, switch back
 {  
         static __int64 total_size;  
1047    
1048          NNSTATS nns1;                                  frame->general |= XVID_GREYSCALE;
         DWORD wrote;  
1049    
         if (codec->framenum == 0)  
         {  
                 total_size = 0;  
1050          }          }
1051    
1052          switch (codec->config.mode)          } else {
         {  
         case DLG_MODE_2PASS_1 :  
                 nns1.bytes = frame->length;     // total bytes  
                 nns1.dd_v = stats->hlength;     // header bytes  
1053    
1054                  nns1.dd_u = nns1.dd_y = 0;                  if (config->greyscale)
                 nns1.dk_v = nns1.dk_u = nns1.dk_y = 0;  
                 nns1.md_u = nns1.md_y = 0;  
                 nns1.mk_u = nns1.mk_y = 0;  
1055    
                 nns1.quant = stats->quant;  
                 if (frame->intra)  
1056                  {                  {
                         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;  
1057    
1058                  total_size += frame->length;                          if ((frame->general && XVID_GREYSCALE))  // use only if not already in greyscale
                 DEBUG1ST(frame->length, (int)total_size/1024, frame->intra, frame->quant, stats->kblks, stats->mblks)  
1059    
1060                  if (WriteFile(codec->twopass.stats1, &nns1, sizeof(NNSTATS), &wrote, 0) == 0 || wrote != sizeof(NNSTATS))                                  frame->general |= XVID_GREYSCALE;
                 {  
                         DEBUG("stats1: WriteFile error");  
                         return ICERR_ERROR;  
                 }  
                 break;  
1061    
1062          case DLG_MODE_2PASS_2_INT :                  } else {
         case DLG_MODE_2PASS_2_EXT :  
                 codec->twopass.overflow += codec->twopass.desired_bytes2 - frame->length;  
                 DEBUG2ND(frame->quant, 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;  
 }  
1063    
1064                            if (!(frame->general && XVID_GREYSCALE))  // if credits is in greyscale, switch back
1065    
1066  int codec_is_in_credits(CONFIG* config, int framenum)                                  frame->general |= XVID_GREYSCALE;
 {  
         if (config->mode == DLG_MODE_2PASS_2_EXT)  
         {  
                 return 0;  
         }  
1067    
         if (config->credits_start)  
         {  
                 if (framenum >= config->credits_start_begin &&  
                         framenum <= config->credits_start_end)  
                 {  
                         return CREDITS_START;  
                 }  
1068          }          }
1069    
         if (config->credits_end)  
         {  
                 if (framenum >= config->credits_end_begin &&  
                         framenum <= config->credits_end_end)  
                 {  
                         return CREDITS_END;  
                 }  
1070          }          }
1071    
1072          return 0;          return 0;
 }  
   
1073    
 int codec_get_vbr_quant(CONFIG* config, int quality)  
 {  
         static float fquant_running = 0;  
         static int my_quality = -1;  
         int     quant;  
   
         // if quality changes, recalculate fquant (credits)  
         if (quality != my_quality)  
         {  
                 config->fquant = 0;  
1074          }          }
1075    
1076          my_quality = quality;  // end of koepi's addition
   
         // desired quantiser = (maxQ-minQ)/100 * (100-qual) + minQ  
         if (!config->fquant)  
         {  
                 config->fquant =  
                         ((float) (config->max_quant - config->min_quant) / 100) *  
                         (100 - quality) +  
                         (float) config->min_quant;  
   
                 fquant_running = config->fquant;  
         }  
   
         if (fquant_running < config->min_quant)  
         {  
                 fquant_running = (float) config->min_quant;  
         }  
         else if(fquant_running > config->max_quant)  
         {  
                 fquant_running = (float) config->max_quant;  
         }  
   
         quant = (int) fquant_running;  
   
         // add error between fquant and quant to fquant_running  
         fquant_running += config->fquant - quant;  
   
         return quant;  
 }  
1077    
Legend:
Removed from v.3  
changed lines
  Added in v.968
No admin address has been configured
 ViewVC Help
Powered by ViewVC 1.0.4