[svn] / branches / release-1_3-branch / xvidcore / examples / xvid_encraw.c Repository:
ViewVC logotype

Diff of /branches/release-1_3-branch/xvidcore/examples/xvid_encraw.c

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

trunk/xvidcore/examples/xvid_encraw.c revision 1398, Fri Apr 2 21:29:21 2004 UTC branches/release-1_3-branch/xvidcore/examples/xvid_encraw.c revision 2129, Wed Jul 8 17:32:58 2015 UTC
# Line 6  Line 6 
6   *  Copyright(C) 2002-2003 Christoph Lampert <gruel@web.de>   *  Copyright(C) 2002-2003 Christoph Lampert <gruel@web.de>
7   *               2002-2003 Edouard Gomez <ed.gomez@free.fr>   *               2002-2003 Edouard Gomez <ed.gomez@free.fr>
8   *               2003      Peter Ross <pross@xvid.org>   *               2003      Peter Ross <pross@xvid.org>
9     *               2003-2014 Michael Militzer <isibaar@xvid.org>
10   *   *
11   *  This program is free software; you can redistribute it and/or modify   *  This program is free software; you can redistribute it and/or modify
12   *  it under the terms of the GNU General Public License as published by   *  it under the terms of the GNU General Public License as published by
# Line 21  Line 22 
22   *  along with this program; if not, write to the Free Software   *  along with this program; if not, write to the Free Software
23   *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA   *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
24   *   *
25   * $Id: xvid_encraw.c,v 1.14 2004-04-02 21:29:21 edgomez Exp $   * $Id$
26   *   *
27   ****************************************************************************/   ****************************************************************************/
28    
# Line 42  Line 43 
43  #include <stdlib.h>  #include <stdlib.h>
44  #include <string.h>  #include <string.h>
45  #include <math.h>  #include <math.h>
46    #include <ctype.h>
47  #ifndef WIN32  #ifndef WIN32
48  #include <sys/time.h>  #include <sys/time.h>
49  #else  #else
50    #include <windows.h>
51    #include <vfw.h>
52  #include <time.h>  #include <time.h>
53    #define XVID_AVI_INPUT
54    #define XVID_AVI_OUTPUT
55  #endif  #endif
56    
57  #include "xvid.h"  #include "xvid.h"
58    #include "portab.h" /* for pthread */
59    
60    #ifdef XVID_MKV_OUTPUT
61    #include "matroska.cpp"
62    #endif
63    
64  #undef READ_PNM  #undef READ_PNM
65    
66    //#define USE_APP_LEVEL_THREADING /* Should xvid_encraw app use multi-threading? */
67    
68  /*****************************************************************************  /*****************************************************************************
69   *                            Quality presets   *                            Quality presets
70   ****************************************************************************/   ****************************************************************************/
71    
72    // Equivalent to vfw's pmvfast_presets
73  static const int motion_presets[] = {  static const int motion_presets[] = {
74          /* quality 0 */          /* quality 0 */
75          0,          0,
76    
77          /* quality 1 */          /* quality 1 */
78          XVID_ME_ADVANCEDDIAMOND16,          0,
79    
80          /* quality 2 */          /* quality 2 */
81          XVID_ME_ADVANCEDDIAMOND16 | XVID_ME_HALFPELREFINE16,          0,
82    
83          /* quality 3 */          /* quality 3 */
84          XVID_ME_ADVANCEDDIAMOND16 | XVID_ME_HALFPELREFINE16 |          0,
         XVID_ME_ADVANCEDDIAMOND8 | XVID_ME_HALFPELREFINE8,  
85    
86          /* quality 4 */          /* quality 4 */
87          XVID_ME_ADVANCEDDIAMOND16 | XVID_ME_HALFPELREFINE16 |          0 | XVID_ME_HALFPELREFINE16 | 0,
         XVID_ME_ADVANCEDDIAMOND8 | XVID_ME_HALFPELREFINE8 |  
         XVID_ME_CHROMA_PVOP | XVID_ME_CHROMA_BVOP,  
88    
89          /* quality 5 */          /* quality 5 */
90          XVID_ME_ADVANCEDDIAMOND16 | XVID_ME_HALFPELREFINE16 |          0 | XVID_ME_HALFPELREFINE16 | 0 | XVID_ME_ADVANCEDDIAMOND16,
         XVID_ME_ADVANCEDDIAMOND8 | XVID_ME_HALFPELREFINE8 |  
         XVID_ME_CHROMA_PVOP | XVID_ME_CHROMA_BVOP,  
91    
92          /* quality 6 */          /* quality 6 */
93          XVID_ME_ADVANCEDDIAMOND16 | XVID_ME_HALFPELREFINE16 | XVID_ME_EXTSEARCH16 |          XVID_ME_HALFPELREFINE16 | XVID_ME_EXTSEARCH16 | XVID_ME_HALFPELREFINE8 | 0 | XVID_ME_USESQUARES16
         XVID_ME_ADVANCEDDIAMOND8 | XVID_ME_HALFPELREFINE8 | XVID_ME_EXTSEARCH8 |  
         XVID_ME_CHROMA_PVOP | XVID_ME_CHROMA_BVOP,  
94    
95  };  };
96  #define ME_ELEMENTS (sizeof(motion_presets)/sizeof(motion_presets[0]))  #define ME_ELEMENTS (sizeof(motion_presets)/sizeof(motion_presets[0]))
# Line 96  Line 103 
103          0,          0,
104    
105          /* quality 2 */          /* quality 2 */
106          XVID_VOP_HALFPEL,          0,
107    
108          /* quality 3 */          /* quality 3 */
109          XVID_VOP_HALFPEL | XVID_VOP_INTER4V,          0,
110    
111          /* quality 4 */          /* quality 4 */
112          XVID_VOP_HALFPEL | XVID_VOP_INTER4V,          0,
113    
114          /* quality 5 */          /* quality 5 */
115          XVID_VOP_HALFPEL | XVID_VOP_INTER4V |          XVID_VOP_INTER4V,
         XVID_VOP_TRELLISQUANT,  
116    
117          /* quality 6 */          /* quality 6 */
118          XVID_VOP_HALFPEL | XVID_VOP_INTER4V |          XVID_VOP_INTER4V,
         XVID_VOP_TRELLISQUANT | XVID_VOP_HQACPRED,  
119    
120  };  };
121  #define VOP_ELEMENTS (sizeof(vop_presets)/sizeof(vop_presets[0]))  #define VOP_ELEMENTS (sizeof(vop_presets)/sizeof(vop_presets[0]))
# Line 120  Line 125 
125   ****************************************************************************/   ****************************************************************************/
126    
127  #define MAX_ZONES   64  #define MAX_ZONES   64
128    #define MAX_ENC_INSTANCES 4
129    #define MAX_XDIM 4096
130    #define MAX_YDIM 4096
131    #define DEFAULT_QUANT 400
132    #define DEFAULT_BITRATE 700000 /* bitrate expressed in bps, not kbps */
133    #define DEFAULT_INTERLACING 1  /* 1:BFF, 2:TFF */
134    #define DEFAULT_SSIM 2
135    #define DEFAULT_PROGRESS 10    /* show progress every 10 frames by default */
136    
137  static xvid_enc_zone_t ZONES[MAX_ZONES];  typedef struct
138  static int NUM_ZONES = 0;  {
139            int frame;
140    
141            int type;
142            int mode;
143            int modifier;
144    
145            unsigned int greyscale;
146            unsigned int chroma_opt;
147            unsigned int bvop_threshold;
148            unsigned int cartoon_mode;
149    } zone_t;
150    
151    typedef struct
152    {
153            int count;
154            int size;
155            int quants[32];
156    } frame_stats_t;
157    
158    typedef struct
159    {
160            pthread_t handle;       /* thread's handle */
161    
162            int start_num;          /* begin/end of sequence */
163            int stop_num;
164    
165            char *outfilename;      /* output filename */
166            char *outavifilename;   /* output avi filename */
167            char *outmkvfilename;   /* output mkv filename */
168            char *statsfilename1;   /* pass1 statsfile */
169    
170            int input_num;
171    
172            int totalsize;          /* encoder stats */
173            double totalenctime;
174            float totalPSNR[3];
175            frame_stats_t framestats[7];
176    } enc_sequence_data_t;
177    
178  /* Maximum number of frames to encode */  /* Maximum number of frames to encode */
179  #define ABS_MAXFRAMENR 9999  #define ABS_MAXFRAMENR -1 /* no limit */
180    
181  static int ARG_STATS = 0;  #ifndef READ_PNM
182    #define IMAGE_SIZE(x,y) ((x)*(y)*3/2)
183    #else
184    #define IMAGE_SIZE(x,y) ((x)*(y)*3)
185    #endif
186    
187    #define MAX(A,B) ( ((A)>(B)) ? (A) : (B) )
188    #define SMALL_EPS (1e-10)
189    
190    #define SWAP(a) ( (((a)&0x000000ff)<<24) | (((a)&0x0000ff00)<<8) | \
191                      (((a)&0x00ff0000)>>8)  | (((a)&0xff000000)>>24) )
192    
193    static zone_t ZONES[MAX_ZONES];
194    static  int NUM_ZONES = 0;
195    
196    static  int ARG_NUM_APP_THREADS = 1;
197    static  int ARG_CPU_FLAGS = 0;
198    static  int ARG_SSIM = -1;
199    static  char* ARG_SSIM_PATH = NULL;
200  static int ARG_DUMP = 0;  static int ARG_DUMP = 0;
201  static int ARG_LUMIMASKING = 0;  static int ARG_LUMIMASKING = 0;
202  static int ARG_BITRATE = 0;  static int ARG_BITRATE = 0;
203  static int ARG_SINGLE = 0;  static  int ARG_TARGETSIZE = 0;
204    static  int ARG_SINGLE = 1;
205  static char *ARG_PASS1 = 0;  static char *ARG_PASS1 = 0;
206  static char *ARG_PASS2 = 0;  static char *ARG_PASS2 = 0;
207  static int ARG_QUALITY = ME_ELEMENTS - 1;  //static int ARG_QUALITY = ME_ELEMENTS - 1;
208  static float ARG_FRAMERATE = 25.00f;  static  int ARG_QUALITY = 6;
209    static  float ARG_FRAMERATE = 0.f;
210    static  int ARG_DWRATE = 25;
211    static  int ARG_DWSCALE = 1;
212  static int ARG_MAXFRAMENR = ABS_MAXFRAMENR;  static int ARG_MAXFRAMENR = ABS_MAXFRAMENR;
213  static int ARG_MAXKEYINTERVAL = 0;  static  int ARG_MAXKEYINTERVAL = 300;
214    static  int ARG_STARTFRAMENR = 0;
215  static char *ARG_INPUTFILE = NULL;  static char *ARG_INPUTFILE = NULL;
216  static int ARG_INPUTTYPE = 0;  static int ARG_INPUTTYPE = 0;
217  static int ARG_SAVEMPEGSTREAM = 0;  static int ARG_SAVEMPEGSTREAM = 0;
218  static int ARG_SAVEINDIVIDUAL = 0;  static int ARG_SAVEINDIVIDUAL = 0;
219  static char *ARG_OUTPUTFILE = NULL;  static char *ARG_OUTPUTFILE = NULL;
220    static  char *ARG_AVIOUTPUTFILE = NULL;
221    static  char *ARG_MKVOUTPUTFILE = NULL;
222    static  char *ARG_TIMECODEFILE = NULL;
223  static int XDIM = 0;  static int XDIM = 0;
224  static int YDIM = 0;  static int YDIM = 0;
225  static int ARG_BQRATIO = 150;  static int ARG_BQRATIO = 150;
226  static int ARG_BQOFFSET = 100;  static int ARG_BQOFFSET = 100;
227  static int ARG_MAXBFRAMES = 0;  static  int ARG_MAXBFRAMES = 2;
 static int ARG_PACKED = 0;  
228  static int ARG_DEBUG = 0;  static int ARG_DEBUG = 0;
229  static int ARG_VOPDEBUG = 0;  static  int ARG_QTYPE = 0;
230  static int ARG_GMC = 0;  static  int ARG_QMATRIX = 0;
231  static int ARG_QPEL = 0;  static  int ARG_INTERLACING = 0;
232  static int ARG_CLOSED_GOP = 0;  static  int ARG_VHQMODE = 1;
233    static  int ARG_QMETRIC = 0;
234  #ifndef READ_PNM  static  int ARG_PAR = 1;
235  #define IMAGE_SIZE(x,y) ((x)*(y)*3/2)  static  int ARG_PARHEIGHT;
236  #else  static  int ARG_PARWIDTH;
237  #define IMAGE_SIZE(x,y) ((x)*(y)*3)  static  int ARG_QUANTS[6] = {2, 31, 2, 31, 2, 31};
238  #endif  static  int ARG_FRAMEDROP = 0;
239    static  double ARG_CQ = 0;
240    static  int ARG_FULL1PASS = 0;
241    static  int ARG_REACTION = 16;
242    static  int ARG_AVERAGING = 100;
243    static  int ARG_SMOOTHER = 100;
244    static  int ARG_KBOOST = 10;
245    static  int ARG_KREDUCTION = 20;
246    static  int ARG_KTHRESH = 1;
247    static  int ARG_CHIGH = 0;
248    static  int ARG_CLOW = 0;
249    static  int ARG_OVERSTRENGTH = 5;
250    static  int ARG_OVERIMPROVE = 5;
251    static  int ARG_OVERDEGRADE = 5;
252    static  int ARG_OVERHEAD = 0;
253    static  int ARG_VBVSIZE = 0;
254    static  int ARG_VBVMAXRATE = 0;
255    static  int ARG_VBVPEAKRATE = 0;
256    static  int ARG_THREADS = 0;
257    static  int ARG_SLICES = 1;
258    static  int ARG_PROGRESS = 0;
259    static  int ARG_COLORSPACE = XVID_CSP_YV12;
260            /* the path where to save output */
261    static char filepath[256] = "./";
262    
263  #define MAX(A,B) ( ((A)>(B)) ? (A) : (B) )  /* on/off options */
264  #define SMALL_EPS (1e-10)  static  int ARG_USE_ASSEMBLER = 1;
265    static  int ARG_TURBO         = 0;
266    static  int ARG_BVHQ          = 0;
267    static  int ARG_QPEL          = 0;
268    static  int ARG_GMC           = 0;
269    static  int ARG_PACKED        = 1;
270    static  int ARG_CLOSED_GOP    = 1;
271    static  int ARG_STATS         = 0;
272    static  int ARG_PSNRHVSM      = 0;
273    static  int ARG_VOPDEBUG      = 0;
274    static  int ARG_CHROMAME      = 1;
275    static  int ARG_TRELLIS       = 1;
276    
277  #define SWAP(a) ( (((a)&0x000000ff)<<24) | (((a)&0x0000ff00)<<8) | \  static  unsigned char qmatrix_intra[64];
278                    (((a)&0x00ff0000)>>8)  | (((a)&0xff000000)>>24) )  static  unsigned char qmatrix_inter[64];
279    
280  /****************************************************************************  /****************************************************************************
281   *                     Nasty global vars ;-)   *                     Nasty global vars ;-)
282   ***************************************************************************/   ***************************************************************************/
283    
284  static int i;  static const int height_ratios[] = {1, 1, 11, 11, 11, 33};
285    static const int width_ratios[] = {1, 1, 12, 10, 16, 40};
286    
287  /* the path where to save output */  const char userdata_start_code[] = "\0\0\x01\xb2";
 static char filepath[256] = "./";  
288    
 /* Internal structures (handles) for encoding and decoding */  
 static void *enc_handle = NULL;  
289    
290  /*****************************************************************************  /*****************************************************************************
291   *               Local prototypes   *               Local prototypes
# Line 188  Line 296 
296    
297  /* Statistical functions */  /* Statistical functions */
298  static double msecond();  static double msecond();
299    int gcd(int a, int b);
300    int minquant(int quants[32]);
301    int maxquant(int quants[32]);
302    double avgquant(frame_stats_t frame);
303    
304  /* PGM related functions */  /* PGM related functions */
305  #ifndef READ_PNM  #ifndef READ_PNM
# Line 203  Line 315 
315                                                  unsigned char *image);                                                  unsigned char *image);
316    
317  /* Encoder related functions */  /* Encoder related functions */
318  static int enc_init(int use_assembler);  static void enc_gbl(int use_assembler);
319  static int enc_stop();  static int  enc_init(void **enc_handle, char *stats_pass1, int start_num);
320  static int enc_main(unsigned char *image,  static int  enc_info();
321    static int  enc_stop(void *enc_handle);
322    static int  enc_main(void *enc_handle,
323                                             unsigned char *image,
324                                          unsigned char *bitstream,                                          unsigned char *bitstream,
325                                          int *key,                                          int *key,
326                                          int *stats_type,                                          int *stats_type,
327                                          int *stats_quant,                                          int *stats_quant,
328                                          int *stats_length,                                          int *stats_length,
329                                          int stats[3]);                                           int stats[3],
330                                             int framenum);
331    static void encode_sequence(enc_sequence_data_t *h);
332    
333    /* Zone Related Functions */
334    static void apply_zone_modifiers(xvid_enc_frame_t * frame, int framenum);
335    static void prepare_full1pass_zones();
336    static void prepare_cquant_zones();
337    void sort_zones(zone_t * zones, int zone_num, int * sel);
338    
339    
340    void removedivxp(char *buf, int size);
341    
342  /*****************************************************************************  /*****************************************************************************
343   *               Main function   *               Main function
# Line 221  Line 347 
347  main(int argc,  main(int argc,
348           char *argv[])           char *argv[])
349  {  {
   
         unsigned char *mp4_buffer = NULL;  
         unsigned char *in_buffer = NULL;  
         unsigned char *out_buffer = NULL;  
   
         double enctime;  
350          double totalenctime = 0.;          double totalenctime = 0.;
351          float totalPSNR[3] = {0., 0., 0.};          float totalPSNR[3] = {0., 0., 0.};
352    
353          int totalsize;          FILE *statsfile;
354          int result;          frame_stats_t framestats[7];
         int m4v_size;  
         int key;  
         int stats_type;  
         int stats_quant;  
         int stats_length;  
         int use_assembler = 0;  
   
         int input_num;  
         int output_num;  
   
         char filename[256];  
355    
356          FILE *in_file = stdin;          int input_num = 0;
357          FILE *out_file = NULL;          int totalsize = 0;
358            int use_assembler = ARG_USE_ASSEMBLER;
359            int i;
360    
361          printf("xvid_encraw - raw mpeg4 bitstream encoder ");          printf("xvid_encraw - raw mpeg4 bitstream encoder ");
362          printf("written by Christoph Lampert 2002-2003\n\n");          printf("written by Christoph Lampert\n\n");
363    
364          /* Is there a dumb XviD coder ? */          /* Is there a dumb Xvid coder ? */
365          if(ME_ELEMENTS != VOP_ELEMENTS) {          if(ME_ELEMENTS != VOP_ELEMENTS) {
366                  fprintf(stderr, "Presets' arrays should have the same number of elements -- Please fill a bug to xvid-devel@xvid.org\n");                  fprintf(stderr, "Presets' arrays should have the same number of elements -- Please file a bug to xvid-devel@xvid.org\n");
367                  return(-1);                  return(-1);
368          }          }
369    
370            /* Clear framestats */
371            memset(framestats, 0, sizeof(framestats));
372    
373  /*****************************************************************************  /*****************************************************************************
374   *                            Command line parsing   *                            Command line parsing
375   ****************************************************************************/   ****************************************************************************/
# Line 264  Line 378 
378    
379                  if (strcmp("-asm", argv[i]) == 0) {                  if (strcmp("-asm", argv[i]) == 0) {
380                          use_assembler = 1;                          use_assembler = 1;
381                    } else if (strcmp("-noasm", argv[i]) == 0) {
382                            use_assembler = 0;
383                  } else if (strcmp("-w", argv[i]) == 0 && i < argc - 1) {                  } else if (strcmp("-w", argv[i]) == 0 && i < argc - 1) {
384                          i++;                          i++;
385                          XDIM = atoi(argv[i]);                          XDIM = atoi(argv[i]);
386                  } else if (strcmp("-h", argv[i]) == 0 && i < argc - 1) {                  } else if (strcmp("-h", argv[i]) == 0 && i < argc - 1) {
387                          i++;                          i++;
388                          YDIM = atoi(argv[i]);                          YDIM = atoi(argv[i]);
389          } else if (strcmp("-bitrate", argv[i]) == 0 && i < argc - 1) {                  } else if (strcmp("-csp",argv[i]) == 0 && i < argc - 1) {
390                            i++;
391                            if (strcmp(argv[i],"i420") == 0){
392                                    ARG_COLORSPACE = XVID_CSP_I420;
393                            } else if(strcmp(argv[i],"yv12") == 0){
394                                    ARG_COLORSPACE = XVID_CSP_YV12;
395                            } else {
396                                    printf("Invalid colorspace\n");
397                                    return 0;
398                            }
399                    } else if (strcmp("-bitrate", argv[i]) == 0) {
400                            if (i < argc - 1)
401                                    ARG_BITRATE = atoi(argv[i+1]);
402                            if (ARG_BITRATE) {
403                                    i++;
404                                    if (ARG_BITRATE <= 20000)
405                                            /* if given parameter is <= 20000, assume it means kbps */
406                                            ARG_BITRATE *= 1000;
407                            }
408                            else
409                                    ARG_BITRATE = DEFAULT_BITRATE;
410                    } else if (strcmp("-size", argv[i]) == 0 && i < argc - 1) {
411                          i++;                          i++;
412                          ARG_BITRATE = atoi(argv[i]);                          ARG_TARGETSIZE = atoi(argv[i]);
413            } else if (strcmp("-cq", argv[i]) == 0 && i < argc - 1) {
414                            i++;
415                            ARG_CQ = atof(argv[i])*100;
416                  } else if (strcmp("-single", argv[i]) == 0) {                  } else if (strcmp("-single", argv[i]) == 0) {
417                          ARG_SINGLE = 1;                          ARG_SINGLE = 1;
418                  } else if (strcmp("-pass1", argv[i]) == 0 && i < argc - 1) {                          ARG_PASS1 = NULL;
419                            ARG_PASS2 = NULL;
420                    } else if (strcmp("-pass1", argv[i]) == 0) {
421                            ARG_SINGLE = 0;
422                            if ((i < argc - 1) && (*argv[i+1] != '-')) {
423                          i++;                          i++;
424                          ARG_PASS1 = argv[i];                          ARG_PASS1 = argv[i];
425                  } else if (strcmp("-pass2", argv[i]) == 0 && i < argc - 1) {                          } else {
426                                    ARG_PASS1 = "xvid.stats";
427                            }
428                    } else if (strcmp("-full1pass", argv[i]) == 0) {
429                            ARG_FULL1PASS = 1;
430                    } else if (strcmp("-pass2", argv[i]) == 0) {
431                            ARG_SINGLE = 0;
432                            if ((i < argc - 1) && (*argv[i+1] != '-')) {
433                          i++;                          i++;
434                          ARG_PASS2 = argv[i];                          ARG_PASS2 = argv[i];
435                            } else {
436                                    ARG_PASS2 = "xvid.stats";
437                            }
438                  } else if (strcmp("-max_bframes", argv[i]) == 0 && i < argc - 1) {                  } else if (strcmp("-max_bframes", argv[i]) == 0 && i < argc - 1) {
439                          i++;                          i++;
440                          ARG_MAXBFRAMES = atoi(argv[i]);                          ARG_MAXBFRAMES = atoi(argv[i]);
441                    } else if (strcmp("-par", argv[i]) == 0 && i < argc - 1) {
442                            i++;
443                            if (sscanf(argv[i], "%d:%d", &(ARG_PARWIDTH), &(ARG_PARHEIGHT))!=2)
444                                    ARG_PAR = atoi(argv[i]);
445                            else {
446                                    int div;
447                                    ARG_PAR = 0;
448                                    div = gcd(ARG_PARWIDTH, ARG_PARHEIGHT);
449                                    ARG_PARWIDTH /= div;
450                                    ARG_PARHEIGHT /= div;
451                            }
452                    } else if (strcmp("-nopacked", argv[i]) == 0) {
453                            ARG_PACKED = 0;
454                  } else if (strcmp("-packed", argv[i]) == 0) {                  } else if (strcmp("-packed", argv[i]) == 0) {
455                          ARG_PACKED = 1;                          ARG_PACKED = 2;
456                    } else if (strcmp("-nochromame", argv[i]) == 0) {
457                            ARG_CHROMAME = 0;
458                    } else if (strcmp("-chromame", argv[i]) == 0) {
459                            ARG_CHROMAME = 1;
460                    } else if (strcmp("-threads", argv[i]) == 0 && i < argc -1) {
461                            i++;
462                            ARG_THREADS = atoi(argv[i]);
463                    } else if (strcmp("-slices", argv[i]) == 0 && i < argc -1) {
464                            i++;
465                            ARG_SLICES = atoi(argv[i]);
466                  } else if (strcmp("-bquant_ratio", argv[i]) == 0 && i < argc - 1) {                  } else if (strcmp("-bquant_ratio", argv[i]) == 0 && i < argc - 1) {
467                          i++;                          i++;
468                          ARG_BQRATIO = atoi(argv[i]);                          ARG_BQRATIO = atoi(argv[i]);
# Line 293  Line 470 
470                          i++;                          i++;
471                          ARG_BQOFFSET = atoi(argv[i]);                          ARG_BQOFFSET = atoi(argv[i]);
472    
473                    } else if (strcmp("-zones", argv[i]) == 0 && i < argc -1) {
474                            char c;
475                            char *frameoptions, *rem;
476                            int startframe;
477                            char options[40];
478    
479                            i++;
480    
481                            do {
482                                    rem = strrchr(argv[i], '/');
483                                    if (rem==NULL)
484                                            rem=argv[i];
485                                    else {
486                                            *rem = '\0';
487                                            rem++;
488                                    }
489                                    if (sscanf(rem, "%d,%c,%s", &startframe, &c, options)<3) {
490                                            fprintf(stderr, "Zone error, bad parameters %s\n", rem);
491                                            continue;
492                                    }
493                                    if (NUM_ZONES >= MAX_ZONES) {
494                                            fprintf(stderr, "warning: too many zones; zone ignored\n");
495                                            continue;
496                                    }
497                                    memset(&ZONES[NUM_ZONES], 0, sizeof(zone_t));
498    
499                                    ZONES[NUM_ZONES].frame = startframe;
500                                    ZONES[NUM_ZONES].modifier = (int)(atof(options)*100);
501                                    if (toupper(c)=='Q')
502                                            ZONES[NUM_ZONES].mode = XVID_ZONE_QUANT;
503                                    else if (toupper(c)=='W')
504                                            ZONES[NUM_ZONES].mode = XVID_ZONE_WEIGHT;
505                                    else {
506                                            fprintf(stderr, "Bad zone type %c\n", c);
507                                            continue;
508                                    }
509    
510                                    if ((frameoptions=strchr(options, ','))!=NULL) {
511                                            int readchar=0, count;
512                                            frameoptions++;
513                                            while (readchar<(int)strlen(frameoptions)) {
514                                                    if (sscanf(frameoptions+readchar, "%d%n", &(ZONES[NUM_ZONES].bvop_threshold), &count)==1) {
515                                                            readchar += count;
516                                                    }
517                                                    else {
518                                                            if (toupper(frameoptions[readchar])=='K')
519                                                                    ZONES[NUM_ZONES].type = XVID_TYPE_IVOP;
520                                                            else if (toupper(frameoptions[readchar])=='G')
521                                                                    ZONES[NUM_ZONES].greyscale = 1;
522                                                            else if (toupper(frameoptions[readchar])=='O')
523                                                                    ZONES[NUM_ZONES].chroma_opt = 1;
524                                                            else if (toupper(frameoptions[readchar])=='C')
525                                                                    ZONES[NUM_ZONES].cartoon_mode = 1;
526                                                            else {
527                                                                    fprintf(stderr, "Error in zone %s option %c\n", rem, frameoptions[readchar]);
528                                                                    break;
529                                                            }
530                                                            readchar++;
531                                                    }
532                                            }
533                                    }
534                                    NUM_ZONES++;
535                            } while (rem != argv[i]);
536    
537    
538          } else if ((strcmp("-zq", argv[i]) == 0 || strcmp("-zw", argv[i]) == 0) && i < argc - 2) {          } else if ((strcmp("-zq", argv[i]) == 0 || strcmp("-zw", argv[i]) == 0) && i < argc - 2) {
539    
540              if (NUM_ZONES >= MAX_ZONES) {              if (NUM_ZONES >= MAX_ZONES) {
541                  fprintf(stderr,"warning: too many zones; zone ignored\n");                  fprintf(stderr,"warning: too many zones; zone ignored\n");
542                  continue;                  continue;
543              }              }
544              ZONES[NUM_ZONES].mode = strcmp("-zq", argv[i])==0 ? XVID_ZONE_QUANT : XVID_ZONE_WEIGHT;                          memset(&ZONES[NUM_ZONES], 0, sizeof(zone_t));
545                            if (strcmp("-zq", argv[i])== 0) {
546                                    ZONES[NUM_ZONES].mode = XVID_ZONE_QUANT;
547                            }
548                            else {
549                                    ZONES[NUM_ZONES].mode = XVID_ZONE_WEIGHT;
550                            }
551                            ZONES[NUM_ZONES].modifier = (int)(atof(argv[i+2])*100);
552                          i++;                          i++;
553              ZONES[NUM_ZONES].frame = atoi(argv[i]);              ZONES[NUM_ZONES].frame = atoi(argv[i]);
554              i++;              i++;
555              ZONES[NUM_ZONES].increment  = (int)(atof(argv[i]) * 100);                          ZONES[NUM_ZONES].type = XVID_TYPE_AUTO;
556              ZONES[NUM_ZONES].base  = 100;                          ZONES[NUM_ZONES].greyscale = 0;
557              NUM_ZONES++;                          ZONES[NUM_ZONES].chroma_opt = 0;
558                            ZONES[NUM_ZONES].bvop_threshold = 0;
559                            ZONES[NUM_ZONES].cartoon_mode = 0;
560    
561                NUM_ZONES++;
562                  } else if (strcmp("-quality", argv[i]) == 0 && i < argc - 1) {                  } else if (strcmp("-quality", argv[i]) == 0 && i < argc - 1) {
563                          i++;                          i++;
564                          ARG_QUALITY = atoi(argv[i]);                          ARG_QUALITY = atoi(argv[i]);
565                    } else if (strcmp("-start", argv[i]) == 0 && i < argc - 1) {
566                            i++;
567                            ARG_STARTFRAMENR = atoi(argv[i]);
568                    } else if (strcmp("-vhqmode", argv[i]) == 0 && i < argc - 1) {
569                            i++;
570                            ARG_VHQMODE = atoi(argv[i]);
571                    } else if (strcmp("-metric", argv[i]) == 0 && i < argc - 1) {
572                            i++;
573                            ARG_QMETRIC = atoi(argv[i]);
574                  } else if (strcmp("-framerate", argv[i]) == 0 && i < argc - 1) {                  } else if (strcmp("-framerate", argv[i]) == 0 && i < argc - 1) {
575                            int exponent;
576                          i++;                          i++;
577                          ARG_FRAMERATE = (float) atof(argv[i]);                          ARG_FRAMERATE = (float) atof(argv[i]);
578                            exponent = (int) strcspn(argv[i], ".");
579                            if (exponent<(int)strlen(argv[i]))
580                                    exponent=(int)pow(10.0, (int)(strlen(argv[i])-1-exponent));
581                            else
582                                    exponent=1;
583                            ARG_DWRATE = (int)(atof(argv[i])*exponent);
584                            ARG_DWSCALE = exponent;
585                            exponent = gcd(ARG_DWRATE, ARG_DWSCALE);
586                            ARG_DWRATE /= exponent;
587                            ARG_DWSCALE /= exponent;
588                  } else if (strcmp("-max_key_interval", argv[i]) == 0 && i < argc - 1) {                  } else if (strcmp("-max_key_interval", argv[i]) == 0 && i < argc - 1) {
589                          i++;                          i++;
590                          ARG_MAXKEYINTERVAL = atoi(argv[i]);                          ARG_MAXKEYINTERVAL = atoi(argv[i]);
# Line 321  Line 593 
593                          ARG_INPUTFILE = argv[i];                          ARG_INPUTFILE = argv[i];
594                  } else if (strcmp("-stats", argv[i]) == 0) {                  } else if (strcmp("-stats", argv[i]) == 0) {
595                          ARG_STATS = 1;                          ARG_STATS = 1;
596                    } else if (strcmp("-nostats", argv[i]) == 0) {
597                            ARG_STATS = 0;
598                    } else if (strcmp("-ssim", argv[i]) == 0) {
599                            ARG_SSIM = DEFAULT_SSIM;
600                            if ((i < argc - 1) && (*argv[i+1] != '-')) {
601                                    i++;
602                                    ARG_SSIM = atoi(argv[i]);
603                            }
604                    } else if (strcmp("-psnrhvsm", argv[i]) == 0) {
605                            ARG_PSNRHVSM = 1;
606                    } else if (strcmp("-nopsnrhvsm", argv[i]) == 0) {
607                            ARG_PSNRHVSM = 0;
608                    } else if (strcmp("-ssim_file", argv[i]) == 0 && i < argc -1) {
609                            i++;
610                            ARG_SSIM_PATH = argv[i];
611                    } else if (strcmp("-timecode", argv[i]) == 0 && i < argc -1) {
612                            i++;
613                            ARG_TIMECODEFILE = argv[i];
614                  } else if (strcmp("-dump", argv[i]) == 0) {                  } else if (strcmp("-dump", argv[i]) == 0) {
615                          ARG_DUMP = 1;                          ARG_DUMP = 1;
616                  } else if (strcmp("-lumimasking", argv[i]) == 0) {                  } else if (strcmp("-masking", argv[i]) == 0 && i < argc -1) {
617                          ARG_LUMIMASKING = 1;                          i++;
618                            ARG_LUMIMASKING = atoi(argv[i]);
619                  } else if (strcmp("-type", argv[i]) == 0 && i < argc - 1) {                  } else if (strcmp("-type", argv[i]) == 0 && i < argc - 1) {
620                          i++;                          i++;
621                          ARG_INPUTTYPE = atoi(argv[i]);                          ARG_INPUTTYPE = atoi(argv[i]);
622                  } else if (strcmp("-frames", argv[i]) == 0 && i < argc - 1) {                  } else if (strcmp("-frames", argv[i]) == 0 && i < argc - 1) {
623                          i++;                          i++;
624                          ARG_MAXFRAMENR = atoi(argv[i]);                          ARG_MAXFRAMENR = atoi(argv[i]);
625                    } else if (strcmp("-drop", argv[i]) == 0 && i < argc - 1) {
626                            i++;
627                            ARG_FRAMEDROP = atoi(argv[i]);
628                    } else if (strcmp("-imin", argv[i]) == 0 && i < argc - 1) {
629                            i++;
630                            ARG_QUANTS[0] = atoi(argv[i]);
631                    } else if (strcmp("-imax", argv[i]) == 0 && i < argc - 1) {
632                            i++;
633                            ARG_QUANTS[1] = atoi(argv[i]);
634                    } else if (strcmp("-pmin", argv[i]) == 0 && i < argc - 1) {
635                            i++;
636                            ARG_QUANTS[2] = atoi(argv[i]);
637                    } else if (strcmp("-pmax", argv[i]) == 0 && i < argc - 1) {
638                            i++;
639                            ARG_QUANTS[3] = atoi(argv[i]);
640                    } else if (strcmp("-bmin", argv[i]) == 0 && i < argc - 1) {
641                            i++;
642                            ARG_QUANTS[4] = atoi(argv[i]);
643                    } else if (strcmp("-bmax", argv[i]) == 0 && i < argc - 1) {
644                            i++;
645                            ARG_QUANTS[5] = atoi(argv[i]);
646                    } else if (strcmp("-qtype", argv[i]) == 0 && i < argc - 1) {
647                            i++;
648                            ARG_QTYPE = atoi(argv[i]);
649                    } else if (strcmp("-qmatrix", argv[i]) == 0 && i < argc - 1) {
650                            FILE *fp = fopen(argv[++i], "rb");
651                            if (fp == NULL) {
652                                    fprintf(stderr, "Error opening input file %s\n", argv[i]);
653                                    return (-1);
654                            }
655                            fseek(fp, 0, SEEK_END);
656                            if (ftell(fp) != 128) {
657                                    fprintf(stderr, "Unexpected size of input file %s\n", argv[i]);
658                                    return (-1);
659                            }
660    
661                            fseek(fp, 0, SEEK_SET);
662                            fread(qmatrix_intra, 1, 64, fp);
663                            fread(qmatrix_inter, 1, 64, fp);
664    
665                            ARG_QMATRIX = 1;
666                            ARG_QTYPE = 1;
667                  } else if (strcmp("-save", argv[i]) == 0) {                  } else if (strcmp("-save", argv[i]) == 0) {
668                          ARG_SAVEMPEGSTREAM = 1;                          ARG_SAVEMPEGSTREAM = 1;
669                          ARG_SAVEINDIVIDUAL = 1;                          ARG_SAVEINDIVIDUAL = 1;
670                  } else if (strcmp("-debug", argv[i]) == 0) {                  } else if (strcmp("-debug", argv[i]) == 0 && i < argc -1) {
671                          i++;                          i++;
672              if (sscanf(argv[i],"0x%x", &ARG_DEBUG) || sscanf(argv[i],"%d", &ARG_DEBUG)) ;              if (!(sscanf(argv[i],"0x%x", &(ARG_DEBUG))))
673                                    sscanf(argv[i],"%d", &(ARG_DEBUG));
674                  } else if (strcmp("-o", argv[i]) == 0 && i < argc - 1) {                  } else if (strcmp("-o", argv[i]) == 0 && i < argc - 1) {
675                          ARG_SAVEMPEGSTREAM = 1;                          ARG_SAVEMPEGSTREAM = 1;
676                          i++;                          i++;
677                          ARG_OUTPUTFILE = argv[i];                          ARG_OUTPUTFILE = argv[i];
678                    } else if (strcmp("-avi", argv[i]) == 0 && i < argc - 1) {
679    #ifdef XVID_AVI_OUTPUT
680                            ARG_SAVEMPEGSTREAM = 1;
681                            i++;
682                            ARG_AVIOUTPUTFILE = argv[i];
683    #else
684                            fprintf( stderr, "Not compiled with AVI output support.\n");
685                            return(-1);
686    #endif
687                    } else if (strcmp("-mkv", argv[i]) == 0 && i < argc - 1) {
688    #ifdef XVID_MKV_OUTPUT
689                            ARG_SAVEMPEGSTREAM = 1;
690                            i++;
691                            ARG_MKVOUTPUTFILE = argv[i];
692    #else
693                            fprintf(stderr, "Not compiled with MKV output support.\n");
694                            return(-1);
695    #endif
696                  } else if (strcmp("-vop_debug", argv[i]) == 0) {                  } else if (strcmp("-vop_debug", argv[i]) == 0) {
697                          ARG_VOPDEBUG = 1;                          ARG_VOPDEBUG = 1;
698                    } else if (strcmp("-novop_debug", argv[i]) == 0) {
699                            ARG_VOPDEBUG = 0;
700                    } else if (strcmp("-trellis", argv[i]) == 0) {
701                            ARG_TRELLIS = 1;
702                    } else if (strcmp("-notrellis", argv[i]) == 0) {
703                            ARG_TRELLIS = 0;
704                    } else if (strcmp("-bvhq", argv[i]) == 0) {
705                            ARG_BVHQ = 1;
706                    } else if (strcmp("-nobvhq", argv[i]) == 0) {
707                            ARG_BVHQ = 0;
708                  } else if (strcmp("-qpel", argv[i]) == 0) {                  } else if (strcmp("-qpel", argv[i]) == 0) {
709                          ARG_QPEL = 1;                          ARG_QPEL = 1;
710                    } else if (strcmp("-noqpel", argv[i]) == 0) {
711                            ARG_QPEL = 0;
712                    } else if (strcmp("-turbo", argv[i]) == 0) {
713                            ARG_TURBO = 1;
714                    } else if (strcmp("-noturbo", argv[i]) == 0) {
715                            ARG_TURBO = 0;
716                  } else if (strcmp("-gmc", argv[i]) == 0) {                  } else if (strcmp("-gmc", argv[i]) == 0) {
717                          ARG_GMC = 1;                          ARG_GMC = 1;
718                    } else if (strcmp("-nogmc", argv[i]) == 0) {
719                            ARG_GMC = 0;
720                    } else if (strcmp("-interlaced", argv[i]) == 0) {
721                            if ((i < argc - 1) && (*argv[i+1] != '-')) {
722                                    i++;
723                                    ARG_INTERLACING = atoi(argv[i]);
724                            } else {
725                                    ARG_INTERLACING = DEFAULT_INTERLACING;
726                            }
727                    } else if (strcmp("-noclosed_gop", argv[i]) == 0) {
728                            ARG_CLOSED_GOP = 0;
729                  } else if (strcmp("-closed_gop", argv[i]) == 0) {                  } else if (strcmp("-closed_gop", argv[i]) == 0) {
730                          ARG_CLOSED_GOP = 1;                          ARG_CLOSED_GOP = 2;
731                  } else if (strcmp("-help", argv[i])) {                  } else if (strcmp("-vbvsize", argv[i]) == 0 && i < argc -1) {
732                            i++;
733                            ARG_VBVSIZE = atoi(argv[i]);
734                    } else if (strcmp("-vbvmax", argv[i]) == 0 && i < argc -1) {
735                            i++;
736                            ARG_VBVMAXRATE = atoi(argv[i]);
737                    } else if (strcmp("-vbvpeak", argv[i]) == 0 && i < argc -1) {
738                            i++;
739                            ARG_VBVPEAKRATE = atoi(argv[i]);
740                    } else if (strcmp("-reaction", argv[i]) == 0 && i < argc -1) {
741                            i++;
742                            ARG_REACTION = atoi(argv[i]);
743                    } else if (strcmp("-averaging", argv[i]) == 0 && i < argc -1) {
744                            i++;
745                            ARG_AVERAGING = atoi(argv[i]);
746                    } else if (strcmp("-smoother", argv[i]) == 0 && i < argc -1) {
747                            i++;
748                            ARG_SMOOTHER = atoi(argv[i]);
749                    } else if (strcmp("-kboost", argv[i]) == 0 && i < argc -1) {
750                            i++;
751                            ARG_KBOOST = atoi(argv[i]);
752                    } else if (strcmp("-kthresh", argv[i]) == 0 && i < argc -1) {
753                            i++;
754                            ARG_KTHRESH = atoi(argv[i]);
755                    } else if (strcmp("-chigh", argv[i]) == 0 && i < argc -1) {
756                            i++;
757                            ARG_CHIGH = atoi(argv[i]);
758                    } else if (strcmp("-clow", argv[i]) == 0 && i < argc -1) {
759                            i++;
760                            ARG_CLOW = atoi(argv[i]);
761                    } else if (strcmp("-ostrength", argv[i]) == 0 && i < argc -1) {
762                            i++;
763                            ARG_OVERSTRENGTH = atoi(argv[i]);
764                    } else if (strcmp("-oimprove", argv[i]) == 0 && i < argc -1) {
765                            i++;
766                            ARG_OVERIMPROVE = atoi(argv[i]);
767                    } else if (strcmp("-odegrade", argv[i]) == 0 && i < argc -1) {
768                            i++;
769                            ARG_OVERDEGRADE = atoi(argv[i]);
770                    } else if (strcmp("-overhead", argv[i]) == 0 && i < argc -1) {
771                            i++;
772                            ARG_OVERHEAD = atoi(argv[i]);
773                    } else if (strcmp("-kreduction", argv[i]) == 0 && i < argc -1) {
774                            i++;
775                            ARG_KREDUCTION = atoi(argv[i]);
776            } else if (strcmp("-progress", argv[i]) == 0) {
777                            if (i < argc - 1)
778                                    /* in kbps */
779                                    ARG_PROGRESS = atoi(argv[i+1]);
780                            if (ARG_PROGRESS > 0)
781                                    i++;
782                            else
783                                    ARG_PROGRESS = DEFAULT_PROGRESS;
784                    } else if (strcmp("-help", argv[i]) == 0) {
785                        if (i < argc - 1 && strcmp("zones", argv[i+1]) == 0) {
786                                    fprintf(stderr, "Zones options\n\
787    NB: You can define up to %d zones using the -zones option as described below.\n\
788    \n\
789     -zones start,mode,value[,options][/start,mode,value[,options]]...\n\
790    \n\
791     Parameters of a zone use the comma (,) as a delimiter. Multiple zones are\n\
792     separated by a slash (/). The end of each zone is defined by either the start\n\
793     of the following zone or the last frame of the input file.\n\
794    \n\
795     start     : start frame of the zone\n\
796     mode      : weight zone = w, quantizer zone = q\n\
797     value     : depending on mode either the zone's weight or quantizer\n\
798     options   : enable certain encoder features for the zone. Each feature is\n\
799                 represented by a single letter. An integer number stands for\n\
800                 b-frame sensitivity. To enable multiple features at the same time\n\
801                 combine the appropriate symbols without any delimiting characters.\n\
802                 K = begin with keyframe\n\
803                 O = enable chroma optimizer\n\
804                 G = greyscale encoding\n\
805                 C = cartoon mode\n\
806           integer = b-frame sensitivity\n\
807    \n\
808     Example:\n\
809      to create a first zone starting at frame 0 with weight 1.0, all options\n\
810      enabled and b-frame sensitivity -5, and a second zone starting at frame 1000\n\
811      with constant quant 4 and no options enabled you would use the -zones option\n\
812      like this:\n\
813    \n\
814      -zones 0,w,1.0,-5KOGC/1000,q,4\n\n", MAX_ZONES);
815                            } else
816                          usage();                          usage();
817                          return (0);                          return (0);
818                  } else {                  } else {
# Line 363  Line 826 
826   *                            Arguments checking   *                            Arguments checking
827   ****************************************************************************/   ****************************************************************************/
828    
829          if (XDIM <= 0 || XDIM >= 4096 || YDIM <= 0 || YDIM >= 4096) {          if (XDIM <= 0 || XDIM >= MAX_XDIM || YDIM <= 0 || YDIM >= MAX_YDIM) {
830                  fprintf(stderr,                  fprintf(stderr,
831                                  "Trying to retreive width and height from PGM header\n");                                  "Trying to retrieve width and height from input header\n");
832                    if (!ARG_INPUTTYPE)
833                  ARG_INPUTTYPE = 1;              /* pgm */                  ARG_INPUTTYPE = 1;              /* pgm */
834          }          }
835    
836          if (ARG_QUALITY < 0 ) {          if (ARG_QUALITY < 0 ) {
837                  ARG_QUALITY = 0;                  ARG_QUALITY = 0;
838          } else if (ARG_QUALITY >= ME_ELEMENTS) {          } else if (ARG_QUALITY >= (int) ME_ELEMENTS) {
839                  ARG_QUALITY = ME_ELEMENTS - 1;                  ARG_QUALITY = ME_ELEMENTS - 1;
840          }          }
841    
842          if (ARG_FRAMERATE <= 0) {          if (ARG_STARTFRAMENR < 0) {
843                  fprintf(stderr, "Wrong Framerate %s \n", argv[5]);                  fprintf(stderr, "Bad starting frame number %d, cannot be negative\n", ARG_STARTFRAMENR);
844                  return (-1);                  return (-1);
845          }          }
846    
847          if (ARG_MAXFRAMENR <= 0) {          if (ARG_PASS2) {
848                    if (ARG_PASS2 == ARG_PASS1) {
849                            fprintf(stderr, "Can't use the same statsfile for pass1 and pass2: %s\n", ARG_PASS2);
850                            return(-1);
851                    }
852                      statsfile = fopen(ARG_PASS2, "rb");
853                      if (statsfile == NULL) {
854                              fprintf(stderr, "Couldn't open statsfile '%s'!\n", ARG_PASS2);
855                              return (-1);
856                      }
857                      fclose(statsfile);
858            }
859    
860    #ifdef XVID_AVI_OUTPUT
861            if (ARG_AVIOUTPUTFILE == NULL && ARG_PACKED <= 1)
862                    ARG_PACKED = 0;
863    #endif
864    
865            if (ARG_BITRATE < 0) {
866                    fprintf(stderr, "Bad bitrate %d, cannot be negative\n", ARG_BITRATE);
867                    return(-1);
868            }
869    
870            if (NUM_ZONES) {
871                    int i;
872                    sort_zones(ZONES, NUM_ZONES, &i);
873            }
874    
875            if (ARG_PAR > 5) {
876                    fprintf(stderr, "Bad PAR: %d. Must be [1..5] or width:height\n", ARG_PAR);
877                    return(-1);
878            }
879    
880            if (ARG_MAXFRAMENR == 0) {
881                  fprintf(stderr, "Wrong number of frames\n");                  fprintf(stderr, "Wrong number of frames\n");
882                  return (-1);                  return (-1);
883          }          }
884    
885            if (ARG_INPUTFILE != NULL) {
886    #if defined(XVID_AVI_INPUT)
887          if (strcmp(ARG_INPUTFILE+(strlen(ARG_INPUTFILE)-3), "avs")==0 ||
888              strcmp(ARG_INPUTFILE+(strlen(ARG_INPUTFILE)-3), "avi")==0 ||
889                      ARG_INPUTTYPE==2)
890          {
891                      PAVIFILE avi_in = NULL;
892                      PAVISTREAM avi_in_stream = NULL;
893                      PGETFRAME get_frame = NULL;
894                      BITMAPINFOHEADER myBitmapInfoHeader;
895                      AVISTREAMINFO avi_info;
896                      FILE *avi_fp = fopen(ARG_INPUTFILE, "rb");
897    
898                      AVIFileInit();
899    
900                      if (avi_fp == NULL) {
901                              fprintf(stderr, "Couldn't open file '%s'!\n", ARG_INPUTFILE);
902                              return (-1);
903                      }
904                      fclose(avi_fp);
905    
906                      if (AVIFileOpen(&avi_in, ARG_INPUTFILE, OF_READ, NULL) != AVIERR_OK) {
907                              fprintf(stderr, "Can't open avi/avs file %s\n", ARG_INPUTFILE);
908                              AVIFileExit();
909                              return(-1);
910                      }
911    
912                      if (AVIFileGetStream(avi_in, &avi_in_stream, streamtypeVIDEO, 0) != AVIERR_OK) {
913                              fprintf(stderr, "Can't open stream from file '%s'!\n", ARG_INPUTFILE);
914                              AVIFileRelease(avi_in);
915                              AVIFileExit();
916                              return (-1);
917                      }
918    
919                      AVIFileRelease(avi_in);
920    
921                      if(AVIStreamInfo(avi_in_stream, &avi_info, sizeof(AVISTREAMINFO)) != AVIERR_OK) {
922                              fprintf(stderr, "Can't get stream info from file '%s'!\n", ARG_INPUTFILE);
923                              AVIStreamRelease(avi_in_stream);
924                              AVIFileExit();
925                              return (-1);
926                      }
927    
928                  if (avi_info.fccHandler != MAKEFOURCC('Y', 'V', '1', '2')) {
929                              LONG size;
930                              fprintf(stderr, "Non YV12 input colorspace %c%c%c%c! Attempting conversion...\n",
931                                      (char)avi_info.fccHandler%256, (char)(avi_info.fccHandler>>8)%256, (char)(avi_info.fccHandler>>16)%256,
932                                      (char)(avi_info.fccHandler>>24)%256);
933                              size = sizeof(myBitmapInfoHeader);
934                              AVIStreamReadFormat(avi_in_stream, 0, &myBitmapInfoHeader, &size);
935                              if (size==0)
936                                      fprintf(stderr, "AVIStreamReadFormat read 0 bytes.\n");
937                              else {
938                                      fprintf(stderr, "AVIStreamReadFormat read %d bytes.\n", (int)size);
939                                      fprintf(stderr, "width = %d, height = %d, planes = %d\n", (int)myBitmapInfoHeader.biWidth,
940                                              (int)myBitmapInfoHeader.biHeight, myBitmapInfoHeader.biPlanes);
941                                      fprintf(stderr, "Compression = %c%c%c%c, %d\n",
942                                              (char)myBitmapInfoHeader.biCompression%256, (char)(myBitmapInfoHeader.biCompression>>8)%256,
943                                              (char)(myBitmapInfoHeader.biCompression>>16)%256, (char)(myBitmapInfoHeader.biCompression>>24)%256,
944                                              (int)myBitmapInfoHeader.biCompression);
945                                      fprintf(stderr, "Bits Per Pixel = %d\n", myBitmapInfoHeader.biBitCount);
946                                      myBitmapInfoHeader.biCompression = MAKEFOURCC('Y', 'V', '1', '2');
947                                      myBitmapInfoHeader.biBitCount = 12;
948                                      myBitmapInfoHeader.biSizeImage = (myBitmapInfoHeader.biWidth*myBitmapInfoHeader.biHeight)*3/2;
949                                      get_frame = AVIStreamGetFrameOpen(avi_in_stream, &myBitmapInfoHeader);
950                              }
951                              if (get_frame == NULL) {
952                                    AVIStreamRelease(avi_in_stream);
953                                    AVIFileExit();
954                                    return (-1);
955                              }
956                              else {
957                                    unsigned char *temp;
958                                    fprintf(stderr, "AVIStreamGetFrameOpen successful.\n");
959                                    temp = (unsigned char*)AVIStreamGetFrame(get_frame, 0);
960                                    if (temp != NULL) {
961                                            int i;
962                                            for (i = 0; i < (int)((DWORD*)temp)[0]; i++) {
963                                                    fprintf(stderr, "%2d ", temp[i]);
964                                            }
965                                            fprintf(stderr, "\n");
966                                    }
967                              }
968                              if (avi_info.fccHandler == MAKEFOURCC('D', 'I', 'B', ' ')) {
969                                      AVIStreamGetFrameClose(get_frame);
970                                      get_frame = NULL;
971                                      ARG_COLORSPACE = XVID_CSP_BGR | XVID_CSP_VFLIP;
972                              }
973                      }
974    
975              if (ARG_MAXFRAMENR<0)
976                            ARG_MAXFRAMENR = avi_info.dwLength-ARG_STARTFRAMENR;
977                      else
978                            ARG_MAXFRAMENR = min(ARG_MAXFRAMENR, (int) (avi_info.dwLength-ARG_STARTFRAMENR));
979    
980                      XDIM = avi_info.rcFrame.right - avi_info.rcFrame.left;
981                      YDIM = avi_info.rcFrame.bottom - avi_info.rcFrame.top;
982                      if (ARG_FRAMERATE==0) {
983                            ARG_FRAMERATE = (float) avi_info.dwRate / (float) avi_info.dwScale;
984                            ARG_DWRATE = avi_info.dwRate;
985                            ARG_DWSCALE = avi_info.dwScale;
986                      }
987    
988                      ARG_INPUTTYPE = 2;
989    
990                      if (get_frame) AVIStreamGetFrameClose(get_frame);
991                      if (avi_in_stream) AVIStreamRelease(avi_in_stream);
992                      AVIFileExit();
993          }
994          else
995    #endif
996                    {
997                            FILE *in_file = fopen(ARG_INPUTFILE, "rb");
998                            if (in_file == NULL) {
999                                    fprintf(stderr, "Error opening input file %s\n", ARG_INPUTFILE);
1000                                    return (-1);
1001                            }
1002    #ifdef USE_APP_LEVEL_THREADING
1003                            {
1004                                    int pos;
1005                                    fseek(in_file, 0, SEEK_END); /* Determine input size */
1006                                    pos = ftell(in_file);
1007                                    ARG_MAXFRAMENR = pos / IMAGE_SIZE(XDIM, YDIM); /* PGM, header size ?? */
1008                            }
1009    #endif
1010                            fclose(in_file);
1011                    }
1012            }
1013    
1014            if (ARG_FRAMERATE <= 0) {
1015                    ARG_FRAMERATE = 25.00f; /* default value */
1016            }
1017    
1018            if (ARG_TARGETSIZE) {
1019                    if (ARG_MAXFRAMENR <= 0) {
1020                            fprintf(stderr, "Bad target size; number of input frames unknown\n");
1021                            goto release_all;
1022                    } else if (ARG_BITRATE) {
1023                                    fprintf(stderr, "Parameter conflict: Do not specify both -bitrate and -size\n");
1024                                    goto release_all;
1025                    } else
1026                            ARG_BITRATE = (int) (((ARG_TARGETSIZE * 8) / (ARG_MAXFRAMENR / ARG_FRAMERATE)) * 1024);
1027            }
1028    
1029                    /* Set constant quant to default if no bitrate given for single pass */
1030            if (ARG_SINGLE && (!ARG_BITRATE) && (!ARG_CQ))
1031                            ARG_CQ = DEFAULT_QUANT;
1032    
1033                    /* Init xvidcore */
1034        enc_gbl(use_assembler);
1035    
1036    #ifdef USE_APP_LEVEL_THREADING
1037            if (ARG_INPUTFILE == NULL || strcmp(ARG_INPUTFILE, "stdin") == 0 ||
1038                ARG_NUM_APP_THREADS <= 1 || ARG_THREADS != 0 ||
1039                ARG_TIMECODEFILE != NULL || ARG_INPUTTYPE == 1)             /* TODO: PGM input */
1040    #endif /* Spawn just one encoder instance */
1041            {
1042                    enc_sequence_data_t enc_data;
1043                    memset(&enc_data, 0, sizeof(enc_sequence_data_t));
1044    
1045                    if (!ARG_THREADS) ARG_THREADS = ARG_NUM_APP_THREADS;
1046                    ARG_NUM_APP_THREADS = 1;
1047    
1048                    enc_data.outfilename = ARG_OUTPUTFILE;
1049                    enc_data.outavifilename = ARG_AVIOUTPUTFILE;
1050                    enc_data.outmkvfilename = ARG_MKVOUTPUTFILE;
1051                    enc_data.statsfilename1 = ARG_PASS1;
1052                    enc_data.start_num = ARG_STARTFRAMENR;
1053                    enc_data.stop_num = ARG_MAXFRAMENR;
1054    
1055                            /* Encode input */
1056                    encode_sequence(&enc_data);
1057    
1058                            /* Copy back stats */
1059                    input_num = enc_data.input_num;
1060                    totalsize = enc_data.totalsize;
1061                    totalenctime = enc_data.totalenctime;
1062                    for (i=0; i < 3; i++) totalPSNR[i] = enc_data.totalPSNR[i];
1063                    memcpy(framestats, enc_data.framestats, sizeof(framestats));
1064            }
1065    #ifdef USE_APP_LEVEL_THREADING
1066            else { /* Split input into sequences and create multiple encoder instances */
1067                    int k;
1068                    void *status;
1069                    FILE *f_out = NULL, *f_stats = NULL;
1070    
1071                    enc_sequence_data_t enc_data[MAX_ENC_INSTANCES];
1072                    char outfile[MAX_ENC_INSTANCES][256];
1073                    char outavifile[MAX_ENC_INSTANCES][256];
1074                    char outmkvfile[MAX_ENC_INSTANCES][256];
1075                    char statsfilename[MAX_ENC_INSTANCES][256];
1076    
1077                    for (k = 0; k < MAX_ENC_INSTANCES; k++)
1078                            memset(&enc_data[k], 0, sizeof(enc_sequence_data_t));
1079    
1080                            /* Overwrite internal encoder threading */
1081                    if (ARG_NUM_APP_THREADS > MAX_ENC_INSTANCES) {
1082                            ARG_THREADS = (int) (ARG_NUM_APP_THREADS / MAX_ENC_INSTANCES);
1083                            ARG_NUM_APP_THREADS = MAX_ENC_INSTANCES;
1084                    }
1085                    else
1086                            ARG_THREADS = -1;
1087    
1088                    enc_data[0].outfilename = ARG_OUTPUTFILE;
1089                    enc_data[0].outavifilename = ARG_AVIOUTPUTFILE;
1090                    enc_data[0].outmkvfilename = ARG_MKVOUTPUTFILE;
1091                    enc_data[0].statsfilename1 = ARG_PASS1;
1092                    enc_data[0].start_num = ARG_STARTFRAMENR;
1093                    enc_data[0].stop_num = (ARG_MAXFRAMENR-ARG_STARTFRAMENR)/ARG_NUM_APP_THREADS;
1094    
1095                    for (k = 1; k < ARG_NUM_APP_THREADS; k++) {
1096                            if (ARG_OUTPUTFILE) {
1097                                    sprintf(outfile[k], "%s.%03d", ARG_OUTPUTFILE, k);
1098                                    enc_data[k].outfilename = outfile[k];
1099                            }
1100                            else
1101                                    enc_data[k].outfilename = NULL;
1102    
1103                            if (ARG_AVIOUTPUTFILE) {
1104                                    sprintf(outavifile[k], "%.*s_%03d.avi", strlen(ARG_AVIOUTPUTFILE) - 4, ARG_AVIOUTPUTFILE, k);
1105                                    enc_data[k].outavifilename = outavifile[k];
1106                            }
1107                            else
1108                                    enc_data[k].outavifilename = NULL;
1109    
1110                            if (ARG_MKVOUTPUTFILE) {
1111                                    sprintf(outmkvfile[k], "%.*s_%03d.mkv", strlen(ARG_MKVOUTPUTFILE) - 4, ARG_MKVOUTPUTFILE, k);
1112                                    enc_data[k].outmkvfilename = outmkvfile[k];
1113                            }
1114                            else
1115                                    enc_data[k].outmkvfilename = NULL;
1116    
1117                            if (ARG_PASS1) {
1118                                    sprintf(statsfilename[k], "%s.%03d", ARG_PASS1, k);
1119                                    enc_data[k].statsfilename1 = statsfilename[k];
1120                            }
1121                            enc_data[k].start_num = (k*(ARG_MAXFRAMENR-ARG_STARTFRAMENR))/ARG_NUM_APP_THREADS;
1122                            enc_data[k].stop_num = ((k+1)*(ARG_MAXFRAMENR-ARG_STARTFRAMENR))/ARG_NUM_APP_THREADS;
1123                    }
1124    
1125                            /* Start multiple encoder threads in parallel */
1126                    for (k = 1; k < ARG_NUM_APP_THREADS; k++) {
1127                            pthread_create(&enc_data[k].handle, NULL, (void*)encode_sequence, (void*)&enc_data[k]);
1128                    }
1129    
1130                            /* Encode first sequence in this thread */
1131                    encode_sequence(&enc_data[0]);
1132    
1133                            /* Wait until encoder threads have finished */
1134                    for (k = 1; k < ARG_NUM_APP_THREADS; k++) {
1135                            pthread_join(enc_data[k].handle, &status);
1136                    }
1137    
1138                            /* Join encoder stats and encoder output files */
1139                    if (ARG_OUTPUTFILE)
1140                            f_out = fopen(enc_data[0].outfilename, "ab+");
1141                    if (ARG_PASS1)
1142                            f_stats = fopen(enc_data[0].statsfilename1, "ab+");
1143    
1144                    for (k = 0; k < ARG_NUM_APP_THREADS; k++) {
1145                                    /* Join stats */
1146                            input_num += enc_data[k].input_num;
1147                            totalsize += enc_data[k].totalsize;
1148                            totalenctime = MAX(totalenctime, enc_data[k].totalenctime);
1149    
1150                            for (i=0; i < 3; i++) totalPSNR[i] += enc_data[k].totalPSNR[i];
1151                            for (i=0; i < 7; i++) {
1152                                    int l;
1153                                    framestats[i].count += enc_data[k].framestats[i].count;
1154                                    framestats[i].size += enc_data[k].framestats[i].size;
1155                                    for (l=0; l < 32; l++)
1156                                            framestats[i].quants[l] += enc_data[k].framestats[i].quants[l];
1157                            }
1158                                    /* Join output files */
1159                            if ((k > 0) && (f_out != NULL)) {
1160                                    int ch;
1161                                    FILE *f = fopen(enc_data[k].outfilename, "rb");
1162                                    while((ch = fgetc(f)) != EOF) { fputc(ch, f_out); }
1163                                    fclose(f);
1164                                    remove(enc_data[k].outfilename);
1165                            }
1166                                    /* Join first pass stats files */
1167                            if ((k > 0) && (f_stats != NULL)) {
1168                                    char str[256];
1169                                    FILE *f = fopen(enc_data[k].statsfilename1, "r");
1170                                    while(fgets(str, sizeof(str), f) != NULL) {
1171                                            if (str[0] != '#' && strlen(str) > 3)
1172                                                    fputs(str, f_stats);
1173                                    }
1174                                    fclose(f);
1175                                    remove(enc_data[k].statsfilename1);
1176                            }
1177                    }
1178                    if (f_out) fclose(f_out);
1179                    if (f_stats) fclose(f_stats);
1180            }
1181    #endif
1182    
1183    /*****************************************************************************
1184     *         Calculate totals and averages for output, print results
1185     ****************************************************************************/
1186    
1187     printf("\n");
1188            printf("Tot: enctime(ms) =%7.2f,               length(bytes) = %7d\n",
1189                       totalenctime, (int) totalsize);
1190    
1191            if (input_num > 0) {
1192                    totalsize /= input_num;
1193                    totalenctime /= input_num;
1194                    totalPSNR[0] /= input_num;
1195                    totalPSNR[1] /= input_num;
1196                    totalPSNR[2] /= input_num;
1197            } else {
1198                    totalsize = -1;
1199                    totalenctime = -1;
1200            }
1201    
1202            printf("Avg: enctime(ms) =%7.2f, fps =%7.2f, length(bytes) = %7d",
1203                       totalenctime, 1000 / totalenctime, (int) totalsize);
1204       if (ARG_STATS) {
1205           printf(", psnr y = %2.2f, psnr u = %2.2f, psnr v = %2.2f",
1206                      totalPSNR[0],totalPSNR[1],totalPSNR[2]);
1207            }
1208            printf("\n");
1209            if (framestats[XVID_TYPE_IVOP].count) {
1210                    printf("I frames: %6d frames, size = %7d/%7d, quants = %2d / %.2f / %2d\n", \
1211                            framestats[XVID_TYPE_IVOP].count, framestats[XVID_TYPE_IVOP].size/framestats[XVID_TYPE_IVOP].count, \
1212                            framestats[XVID_TYPE_IVOP].size, minquant(framestats[XVID_TYPE_IVOP].quants), \
1213                            avgquant(framestats[XVID_TYPE_IVOP]), maxquant(framestats[XVID_TYPE_IVOP].quants));
1214            }
1215            if (framestats[XVID_TYPE_PVOP].count) {
1216                    printf("P frames: %6d frames, size = %7d/%7d, quants = %2d / %.2f / %2d\n", \
1217                            framestats[XVID_TYPE_PVOP].count, framestats[XVID_TYPE_PVOP].size/framestats[XVID_TYPE_PVOP].count, \
1218                            framestats[XVID_TYPE_PVOP].size, minquant(framestats[XVID_TYPE_PVOP].quants), \
1219                            avgquant(framestats[XVID_TYPE_PVOP]), maxquant(framestats[XVID_TYPE_PVOP].quants));
1220            }
1221            if (framestats[XVID_TYPE_BVOP].count) {
1222                    printf("B frames: %6d frames, size = %7d/%7d, quants = %2d / %.2f / %2d\n", \
1223                            framestats[XVID_TYPE_BVOP].count, framestats[XVID_TYPE_BVOP].size/framestats[XVID_TYPE_BVOP].count, \
1224                            framestats[XVID_TYPE_BVOP].size, minquant(framestats[XVID_TYPE_BVOP].quants), \
1225                            avgquant(framestats[XVID_TYPE_BVOP]), maxquant(framestats[XVID_TYPE_BVOP].quants));
1226            }
1227            if (framestats[XVID_TYPE_SVOP].count) {
1228                    printf("S frames: %6d frames, size = %7d/%7d, quants = %2d / %.2f / %2d\n", \
1229                            framestats[XVID_TYPE_SVOP].count, framestats[XVID_TYPE_SVOP].size/framestats[XVID_TYPE_SVOP].count, \
1230                            framestats[XVID_TYPE_SVOP].size, minquant(framestats[XVID_TYPE_SVOP].quants), \
1231                            avgquant(framestats[XVID_TYPE_SVOP]), maxquant(framestats[XVID_TYPE_SVOP].quants));
1232            }
1233            if (framestats[5].count) {
1234                    printf("N frames: %6d frames, size = %7d/%7d\n", \
1235                            framestats[5].count, framestats[5].size/framestats[5].count, \
1236                            framestats[5].size);
1237            }
1238    
1239    
1240    /*****************************************************************************
1241     *                            Xvid PART  Stop
1242     ****************************************************************************/
1243    
1244      release_all:
1245    
1246            return (0);
1247    }
1248    
1249    /*****************************************************************************
1250     *               Encode a sequence
1251     ****************************************************************************/
1252    
1253    void encode_sequence(enc_sequence_data_t *h) {
1254    
1255            /* Internal structures (handles) for encoding */
1256            void *enc_handle = NULL;
1257    
1258            int start_num = h->start_num;
1259            int stop_num = h->stop_num;
1260            char *outfilename = h->outfilename;
1261            char *outavifilename = h->outavifilename;
1262            char *outmkvfilename = h->outmkvfilename;
1263            float *totalPSNR = h->totalPSNR;
1264    
1265            int input_num;
1266            int totalsize;
1267            double totalenctime = 0.;
1268    
1269            unsigned char *mp4_buffer = NULL;
1270            unsigned char *in_buffer = NULL;
1271            unsigned char *out_buffer = NULL;
1272    
1273            double enctime;
1274    
1275            int result;
1276            int output_num;
1277            int m4v_size;
1278            int key;
1279            int stats_type;
1280            int stats_quant;
1281            int stats_length;
1282            int fakenvop = 0;
1283    
1284            FILE *in_file = stdin;
1285            FILE *out_file = NULL;
1286            FILE *time_file = NULL;
1287    
1288            char filename[256];
1289    
1290    #ifdef XVID_MKV_OUTPUT
1291            PMKVFILE myMKVFile = NULL;
1292            PMKVSTREAM myMKVStream = NULL;
1293            MKVSTREAMINFO myMKVStreamInfo;
1294    #endif
1295    #if defined(XVID_AVI_INPUT)
1296            PAVIFILE avi_in = NULL;
1297            PAVISTREAM avi_in_stream = NULL;
1298            PGETFRAME get_frame = NULL;
1299    #else
1300    #define get_frame NULL
1301    #endif
1302    #if defined(XVID_AVI_OUTPUT)
1303            int avierr;
1304            PAVIFILE myAVIFile = NULL;
1305            PAVISTREAM myAVIStream = NULL;
1306            AVISTREAMINFO myAVIStreamInfo;
1307    #endif
1308    #if defined(XVID_AVI_INPUT) || defined(XVID_AVI_OUTPUT)
1309            BITMAPINFOHEADER myBitmapInfoHeader;
1310            if (ARG_NUM_APP_THREADS > 1)
1311                    CoInitializeEx(0, COINIT_MULTITHREADED);
1312            AVIFileInit();
1313    #endif
1314    
1315          if (ARG_INPUTFILE == NULL || strcmp(ARG_INPUTFILE, "stdin") == 0) {          if (ARG_INPUTFILE == NULL || strcmp(ARG_INPUTFILE, "stdin") == 0) {
1316                  in_file = stdin;                  in_file = stdin;
1317          } else {          } else {
1318    #ifdef XVID_AVI_INPUT
1319          if (strcmp(ARG_INPUTFILE+(strlen(ARG_INPUTFILE)-3), "avs")==0 ||
1320              strcmp(ARG_INPUTFILE+(strlen(ARG_INPUTFILE)-3), "avi")==0 ||
1321                      ARG_INPUTTYPE==2)
1322          {
1323                      AVISTREAMINFO avi_info;
1324                      FILE *avi_fp = fopen(ARG_INPUTFILE, "rb");
1325    
1326                      if (avi_fp == NULL) {
1327                              fprintf(stderr, "Couldn't open file '%s'!\n", ARG_INPUTFILE);
1328                              return;
1329                      }
1330                      fclose(avi_fp);
1331    
1332                      if (AVIFileOpen(&avi_in, ARG_INPUTFILE, OF_READ, NULL) != AVIERR_OK) {
1333                              fprintf(stderr, "Can't open avi/avs file %s\n", ARG_INPUTFILE);
1334                              AVIFileExit();
1335                              return;
1336                      }
1337    
1338                      if (AVIFileGetStream(avi_in, &avi_in_stream, streamtypeVIDEO, 0) != AVIERR_OK) {
1339                              fprintf(stderr, "Can't open stream from file '%s'!\n", ARG_INPUTFILE);
1340                              AVIFileRelease(avi_in);
1341                              AVIFileExit();
1342                              return;
1343                      }
1344    
1345                      AVIFileRelease(avi_in);
1346    
1347                      if(AVIStreamInfo(avi_in_stream, &avi_info, sizeof(AVISTREAMINFO)) != AVIERR_OK) {
1348                              fprintf(stderr, "Can't get stream info from file '%s'!\n", ARG_INPUTFILE);
1349                              AVIStreamRelease(avi_in_stream);
1350                              AVIFileExit();
1351                              return;
1352                      }
1353    
1354                  if (avi_info.fccHandler != MAKEFOURCC('Y', 'V', '1', '2')) {
1355                              LONG size;
1356                              fprintf(stderr, "Non YV12 input colorspace %c%c%c%c! Attempting conversion...\n",
1357                                      (char)avi_info.fccHandler%256, (char)(avi_info.fccHandler>>8)%256, (char)(avi_info.fccHandler>>16)%256,
1358                                      (char)(avi_info.fccHandler>>24)%256);
1359                              size = sizeof(myBitmapInfoHeader);
1360                              AVIStreamReadFormat(avi_in_stream, 0, &myBitmapInfoHeader, &size);
1361                              if (size==0)
1362                                      fprintf(stderr, "AVIStreamReadFormat read 0 bytes.\n");
1363                              else {
1364                                      fprintf(stderr, "AVIStreamReadFormat read %d bytes.\n", (int)size);
1365                                      fprintf(stderr, "width = %d, height = %d, planes = %d\n", (int)myBitmapInfoHeader.biWidth,
1366                                              (int)myBitmapInfoHeader.biHeight, myBitmapInfoHeader.biPlanes);
1367                                      fprintf(stderr, "Compression = %c%c%c%c, %d\n",
1368                                              (char)myBitmapInfoHeader.biCompression%256, (char)(myBitmapInfoHeader.biCompression>>8)%256,
1369                                              (char)(myBitmapInfoHeader.biCompression>>16)%256, (char)(myBitmapInfoHeader.biCompression>>24)%256,
1370                                              (int)myBitmapInfoHeader.biCompression);
1371                                      fprintf(stderr, "Bits Per Pixel = %d\n", myBitmapInfoHeader.biBitCount);
1372                                      myBitmapInfoHeader.biCompression = MAKEFOURCC('Y', 'V', '1', '2');
1373                                      myBitmapInfoHeader.biBitCount = 12;
1374                                      myBitmapInfoHeader.biSizeImage = (myBitmapInfoHeader.biWidth*myBitmapInfoHeader.biHeight)*3/2;
1375                                      get_frame = AVIStreamGetFrameOpen(avi_in_stream, &myBitmapInfoHeader);
1376                              }
1377                              if (get_frame == NULL) {
1378                                    AVIStreamRelease(avi_in_stream);
1379                                    AVIFileExit();
1380                                    return;
1381                              }
1382                              else {
1383                                    unsigned char *temp;
1384                                    fprintf(stderr, "AVIStreamGetFrameOpen successful.\n");
1385                                    temp = (unsigned char*)AVIStreamGetFrame(get_frame, 0);
1386                                    if (temp != NULL) {
1387                                            int i;
1388                                            for (i = 0; i < (int)((DWORD*)temp)[0]; i++) {
1389                                                    fprintf(stderr, "%2d ", temp[i]);
1390                                            }
1391                                            fprintf(stderr, "\n");
1392                                    }
1393                              }
1394                              if (avi_info.fccHandler == MAKEFOURCC('D', 'I', 'B', ' ')) {
1395                                      AVIStreamGetFrameClose(get_frame);
1396                                      get_frame = NULL;
1397                                      ARG_COLORSPACE = XVID_CSP_BGR | XVID_CSP_VFLIP;
1398                              }
1399                      }
1400        }
1401        else
1402    #endif
1403                    {
1404                  in_file = fopen(ARG_INPUTFILE, "rb");                  in_file = fopen(ARG_INPUTFILE, "rb");
1405                  if (in_file == NULL) {                  if (in_file == NULL) {
1406                          fprintf(stderr, "Error opening input file %s\n", ARG_INPUTFILE);                          fprintf(stderr, "Error opening input file %s\n", ARG_INPUTFILE);
1407                          return (-1);                                  return;
1408                            }
1409                    }
1410            }
1411    
1412            // This should be after the avi input opening stuff
1413            if (ARG_TIMECODEFILE != NULL) {
1414                    time_file = fopen(ARG_TIMECODEFILE, "r");
1415                    if (time_file==NULL) {
1416                            fprintf(stderr, "Couldn't open timecode file '%s'!\n", ARG_TIMECODEFILE);
1417                            return;
1418                    }
1419                    else {
1420                            fscanf(time_file, "# timecode format v2\n");
1421                  }                  }
1422          }          }
1423    
1424          if (ARG_INPUTTYPE) {          if (ARG_INPUTTYPE==1) {
1425  #ifndef READ_PNM  #ifndef READ_PNM
1426                  if (read_pgmheader(in_file)) {                  if (read_pgmheader(in_file)) {
1427  #else  #else
# Line 404  Line 1429 
1429  #endif  #endif
1430                          fprintf(stderr,                          fprintf(stderr,
1431                                          "Wrong input format, I want YUV encapsulated in PGM\n");                                          "Wrong input format, I want YUV encapsulated in PGM\n");
1432                          return (-1);                          return;
1433                  }                  }
1434          }          }
1435    
1436            /* Jump to the starting frame */
1437            if (ARG_INPUTTYPE == 0) /* TODO: Other input formats ??? */
1438                    fseek(in_file, start_num*IMAGE_SIZE(XDIM, YDIM), SEEK_SET);
1439    
1440    
1441          /* now we know the sizes, so allocate memory */          /* now we know the sizes, so allocate memory */
1442          in_buffer = (unsigned char *) malloc(IMAGE_SIZE(XDIM, YDIM));          if (get_frame == NULL)
1443            {
1444                    in_buffer = (unsigned char *) malloc(4*XDIM*YDIM);
1445          if (!in_buffer)          if (!in_buffer)
1446                  goto free_all_memory;                  goto free_all_memory;
1447            }
1448    
1449          /* this should really be enough memory ! */          /* this should really be enough memory ! */
1450          mp4_buffer = (unsigned char *) malloc(IMAGE_SIZE(XDIM, YDIM) * 2);          mp4_buffer = (unsigned char *) malloc(IMAGE_SIZE(XDIM, YDIM) * 2);
# Line 419  Line 1452 
1452                  goto free_all_memory;                  goto free_all_memory;
1453    
1454  /*****************************************************************************  /*****************************************************************************
1455   *                            XviD PART  Start   *                            Xvid PART  Start
1456   ****************************************************************************/   ****************************************************************************/
1457    
1458    
1459          result = enc_init(use_assembler);          result = enc_init(&enc_handle, h->statsfilename1, h->start_num);
1460          if (result) {          if (result) {
1461                  fprintf(stderr, "Encore INIT problem, return value %d\n", result);                  fprintf(stderr, "Encore INIT problem, return value %d\n", result);
1462                  goto release_all;                  goto release_all;
# Line 433  Line 1466 
1466   *                            Main loop   *                            Main loop
1467   ****************************************************************************/   ****************************************************************************/
1468    
1469          if (ARG_SAVEMPEGSTREAM && ARG_OUTPUTFILE) {          if (ARG_SAVEMPEGSTREAM) {
1470    
1471                    if (outfilename) {
1472                            if ((out_file = fopen(outfilename, "w+b")) == NULL) {
1473                                    fprintf(stderr, "Error opening output file %s\n", outfilename);
1474                                    goto release_all;
1475                            }
1476                    }
1477    
1478    #ifdef XVID_AVI_OUTPUT
1479                    if (outavifilename != NULL ) {
1480                            {
1481                                    /* Open the .avi output then close it */
1482                                    /* Resets the file size to 0, which AVIFile doesn't seem to do */
1483                                    FILE *scrub;
1484                                    if ((scrub = fopen(outavifilename, "w+b")) == NULL) {
1485                                            fprintf(stderr, "Error opening output file %s\n", outavifilename);
1486                                            goto release_all;
1487                                    }
1488                                    else
1489                                            fclose(scrub);
1490                            }
1491                            memset(&myAVIStreamInfo, 0, sizeof(AVISTREAMINFO));
1492                            myAVIStreamInfo.fccType = streamtypeVIDEO;
1493                            myAVIStreamInfo.fccHandler = MAKEFOURCC('x', 'v', 'i', 'd');
1494                            myAVIStreamInfo.dwScale = ARG_DWSCALE;
1495                            myAVIStreamInfo.dwRate = ARG_DWRATE;
1496                            myAVIStreamInfo.dwLength = (h->stop_num - h->start_num);
1497                            myAVIStreamInfo.dwQuality = 10000;
1498                            SetRect(&myAVIStreamInfo.rcFrame, 0, 0, XDIM, YDIM);
1499    
1500                            if ((avierr=AVIFileOpen(&myAVIFile, outavifilename, OF_CREATE|OF_WRITE, NULL))) {
1501                                    fprintf(stderr, "AVIFileOpen failed opening output file %s, error code %d\n", outavifilename, avierr);
1502                                    goto release_all;
1503                            }
1504    
1505                  if ((out_file = fopen(ARG_OUTPUTFILE, "w+b")) == NULL) {                          if ((avierr=AVIFileCreateStream(myAVIFile, &myAVIStream, &myAVIStreamInfo))) {
1506                          fprintf(stderr, "Error opening output file %s\n", ARG_OUTPUTFILE);                                  fprintf(stderr, "AVIFileCreateStream failed, error code %d\n", avierr);
1507                          goto release_all;                          goto release_all;
1508                  }                  }
1509    
1510                            memset(&myBitmapInfoHeader, 0, sizeof(BITMAPINFOHEADER));
1511                            myBitmapInfoHeader.biHeight = YDIM;
1512                            myBitmapInfoHeader.biWidth = XDIM;
1513                            myBitmapInfoHeader.biPlanes = 1;
1514                            myBitmapInfoHeader.biSize = sizeof(BITMAPINFOHEADER);
1515                            myBitmapInfoHeader.biCompression = MAKEFOURCC('X', 'V', 'I', 'D');
1516                            myBitmapInfoHeader.biBitCount = 12;
1517                            myBitmapInfoHeader.biSizeImage = 6*XDIM*YDIM;
1518                            if ((avierr=AVIStreamSetFormat(myAVIStream, 0, &myBitmapInfoHeader, sizeof(BITMAPINFOHEADER)))) {
1519                                    fprintf(stderr, "AVIStreamSetFormat failed, error code %d\n", avierr);
1520                                    goto release_all;
1521                            }
1522                    }
1523    #endif
1524    #ifdef XVID_MKV_OUTPUT
1525                    if (outmkvfilename != NULL) {
1526                            {
1527                                    /* Open the .mkv output then close it */
1528                                    /* Just to make sure we can write to it */
1529                                    FILE *scrub;
1530                                    if ((scrub = fopen(outmkvfilename, "w+b")) == NULL) {
1531                                            fprintf(stderr, "Error opening output file %s\n", outmkvfilename);
1532                                            goto release_all;
1533                                    }
1534                                    else
1535                                            fclose(scrub);
1536                            }
1537    
1538                            MKVFileOpen(&myMKVFile, outmkvfilename, OF_CREATE|OF_WRITE, NULL);
1539                            if (ARG_PAR) {
1540                                    myMKVStreamInfo.display_height = YDIM*height_ratios[ARG_PAR];
1541                                    myMKVStreamInfo.display_width = XDIM*width_ratios[ARG_PAR];
1542                            }
1543                            else {
1544                                    myMKVStreamInfo.display_height = YDIM*ARG_PARHEIGHT;
1545                                    myMKVStreamInfo.display_width = XDIM*ARG_PARWIDTH;
1546                            }
1547                            myMKVStreamInfo.height = YDIM;
1548                            myMKVStreamInfo.width = XDIM;
1549                            myMKVStreamInfo.framerate = ARG_DWRATE;
1550                            myMKVStreamInfo.framescale = ARG_DWSCALE;
1551                            myMKVStreamInfo.length = (h->stop_num - h->start_num);
1552                            MKVFileCreateStream(myMKVFile, &myMKVStream, &myMKVStreamInfo);
1553                    }
1554    #endif
1555          } else {          } else {
1556                  out_file = NULL;                  out_file = NULL;
1557          }          }
1558    
1559    
1560  /*****************************************************************************  /*****************************************************************************
1561   *                       Encoding loop   *                       Encoding loop
1562   ****************************************************************************/   ****************************************************************************/
# Line 453  Line 1566 
1566          result = 0;          result = 0;
1567    
1568          input_num = 0;                          /* input frame counter */          input_num = 0;                          /* input frame counter */
1569          output_num = 0;                         /* output frame counter */          output_num = start_num;             /* output frame counter */
1570    
1571          do {          do {
1572    
1573                  char *type;                  char *type;
1574                  int sse[3];                  int sse[3];
1575    
1576                  if (input_num >= ARG_MAXFRAMENR) {                  if ((input_num+start_num) >= stop_num && stop_num > 0) {
1577                          result = 1;                          result = 1;
1578                  }                  }
1579    
1580                  if (!result) {                  if (!result) {
1581                          if (ARG_INPUTTYPE) {  #ifdef XVID_AVI_INPUT
1582                            if (ARG_INPUTTYPE==2) {
1583                                    /* read avs/avi data (YUV-format) */
1584                                    if (get_frame != NULL) {
1585                                            in_buffer = (unsigned char*)AVIStreamGetFrame(get_frame, input_num+start_num);
1586                                            if (in_buffer == NULL)
1587                                                    result = 1;
1588                                            else
1589                                                    in_buffer += ((DWORD*)in_buffer)[0];
1590                                    } else {
1591                                            if (AVIStreamRead(avi_in_stream, input_num + start_num, 1, in_buffer, 4 * XDIM*YDIM, NULL, NULL) != AVIERR_OK)
1592                                                    result = 1;
1593                                    }
1594                            } else
1595    #endif
1596                            if (ARG_INPUTTYPE==1) {
1597                                  /* read PGM data (YUV-format) */                                  /* read PGM data (YUV-format) */
1598  #ifndef READ_PNM  #ifndef READ_PNM
1599                                  result = read_pgmdata(in_file, in_buffer);                                  result = read_pgmdata(in_file, in_buffer);
# Line 482  Line 1610 
1610   *                       Encode and decode this frame   *                       Encode and decode this frame
1611   ****************************************************************************/   ****************************************************************************/
1612    
1613                    if ((unsigned int)(input_num+start_num) >= (unsigned int)(stop_num-1) && ARG_MAXBFRAMES) {
1614                            stats_type = XVID_TYPE_PVOP;
1615                    }
1616                    else
1617                            stats_type = XVID_TYPE_AUTO;
1618    
1619                  enctime = msecond();                  enctime = msecond();
1620                  m4v_size =                  m4v_size =
1621                          enc_main(!result ? in_buffer : 0, mp4_buffer, &key, &stats_type,                          enc_main(enc_handle, !result ? in_buffer : 0, mp4_buffer, &key, &stats_type,
1622                                           &stats_quant, &stats_length, sse);                                           &stats_quant, &stats_length, sse, input_num);
1623                  enctime = msecond() - enctime;                  enctime = msecond() - enctime;
1624    
1625                  /* Write the Frame statistics */                  /* Write the Frame statistics */
1626    
                 printf("%5d: key=%i, time= %6.0f, len= %7d", !result ? input_num : -1,  
                            key, (float) enctime, (int) m4v_size);  
   
1627                  if (stats_type > 0) {   /* !XVID_TYPE_NOTHING */                  if (stats_type > 0) {   /* !XVID_TYPE_NOTHING */
   
1628                          switch (stats_type) {                          switch (stats_type) {
1629                          case XVID_TYPE_IVOP:                          case XVID_TYPE_IVOP:
1630                                  type = "I";                                  type = "I";
# Line 504  Line 1634 
1634                                  break;                                  break;
1635                          case XVID_TYPE_BVOP:                          case XVID_TYPE_BVOP:
1636                                  type = "B";                                  type = "B";
1637                                            if (ARG_PACKED)
1638                                                    fakenvop = 1;
1639                                  break;                                  break;
1640                          case XVID_TYPE_SVOP:                          case XVID_TYPE_SVOP:
1641                                  type = "S";                                  type = "S";
# Line 513  Line 1645 
1645                                  break;                                  break;
1646                          }                          }
1647    
1648                            if (stats_length > 8) {
1649                                    h->framestats[stats_type].count++;
1650                                    h->framestats[stats_type].quants[stats_quant]++;
1651                                    h->framestats[stats_type].size += stats_length;
1652                            }
1653                            else {
1654                                    h->framestats[5].count++;
1655                                    h->framestats[5].quants[stats_quant]++;
1656                                    h->framestats[5].size += stats_length;
1657                            }
1658    
1659    #define SSE2PSNR(sse, width, height) ((!(sse))?0.0f : 48.131f - 10*(float)log10((float)(sse)/((float)((width)*(height)))))
1660    
1661                            if (ARG_PROGRESS == 0) {
1662                                    printf("%5d: key=%i, time= %6.0f, len= %7d", !result ? (input_num+start_num) : -1,
1663                                            key, (float) enctime, (int) m4v_size);
1664                          printf(" | type=%s, quant= %2d, len= %7d", type, stats_quant,                          printf(" | type=%s, quant= %2d, len= %7d", type, stats_quant,
1665                                     stats_length);                                     stats_length);
1666    
 #define SSE2PSNR(sse, width, height) ((!(sse))?0.0f : 48.131f - 10*(float)log10((float)(sse)/((float)((width)*(height)))))  
1667    
1668                          if (ARG_STATS) {                          if (ARG_STATS) {
1669                                  printf(", psnr y = %2.2f, psnr u = %2.2f, psnr v = %2.2f",                                  printf(", psnr y = %2.2f, psnr u = %2.2f, psnr v = %2.2f",
1670                                             SSE2PSNR(sse[0], XDIM, YDIM), SSE2PSNR(sse[1], XDIM / 2,                                                  SSE2PSNR(sse[0], XDIM, YDIM), SSE2PSNR(sse[1], XDIM / 2, YDIM / 2),
                                                                                                                           YDIM / 2),  
1671                                             SSE2PSNR(sse[2], XDIM / 2, YDIM / 2));                                             SSE2PSNR(sse[2], XDIM / 2, YDIM / 2));
1672                                    }
1673                                    printf("\n");
1674                            } else {
1675                                    if ((input_num) % ARG_PROGRESS == 1) {
1676                                            if (stop_num > 0) {
1677                                                    fprintf(stderr, "\r%7d frames(%3d%%) encoded, %6.2f fps, Average Bitrate = %5.0fkbps", \
1678                                                            (ARG_NUM_APP_THREADS*input_num), (input_num)*100/(stop_num-start_num), (ARG_NUM_APP_THREADS*input_num)*1000/(totalenctime), \
1679                                                            ((((totalsize)/1000)*ARG_FRAMERATE)*8)/(input_num));
1680                                            } else {
1681                                                    fprintf(stderr, "\r%7d frames encoded, %6.2f fps, Average Bitrate = %5.0fkbps", \
1682                                                            (ARG_NUM_APP_THREADS*input_num), (ARG_NUM_APP_THREADS*input_num)*1000/(totalenctime), \
1683                                                            ((((totalsize)/1000)*ARG_FRAMERATE)*8)/(input_num));
1684                                            }
1685                                    }
1686                            }
1687    
1688                            if (ARG_STATS) {
1689                                  totalPSNR[0] += SSE2PSNR(sse[0], XDIM, YDIM);                                  totalPSNR[0] += SSE2PSNR(sse[0], XDIM, YDIM);
1690                                  totalPSNR[1] += SSE2PSNR(sse[1], XDIM/2, YDIM/2);                                  totalPSNR[1] += SSE2PSNR(sse[1], XDIM/2, YDIM/2);
1691                                  totalPSNR[2] += SSE2PSNR(sse[2], XDIM/2, YDIM/2);                                  totalPSNR[2] += SSE2PSNR(sse[2], XDIM/2, YDIM/2);
1692                          }                          }
   
                 }  
1693  #undef SSE2PSNR  #undef SSE2PSNR
1694                    }
1695    
1696                  printf("\n");                  if (m4v_size < 0)
   
                 if (m4v_size < 0) {  
1697                          break;                          break;
                 }  
1698    
1699                  /* Update encoding time stats */                  /* Update encoding time stats */
1700                  totalenctime += enctime;                  totalenctime += enctime;
# Line 547  Line 1705 
1705   ****************************************************************************/   ****************************************************************************/
1706    
1707                  if (m4v_size > 0 && ARG_SAVEMPEGSTREAM) {                  if (m4v_size > 0 && ARG_SAVEMPEGSTREAM) {
1708                            char timecode[50];
1709    
1710                            if (time_file != NULL) {
1711                                    if (fscanf(time_file, "%s\n", timecode) != 1) {
1712                                            fprintf(stderr, "Error reading timecode file, frame %d\n", output_num);
1713                                            goto release_all;
1714                                    }
1715                            }
1716                            else
1717                                    sprintf(timecode, "%f", ((double)ARG_DWSCALE/ARG_DWRATE)*1000*(output_num-start_num));
1718    
1719                          /* Save single files */                          /* Save single files */
1720                          if (ARG_SAVEINDIVIDUAL) {                          if (ARG_SAVEINDIVIDUAL) {
# Line 555  Line 1723 
1723                                  out = fopen(filename, "w+b");                                  out = fopen(filename, "w+b");
1724                                  fwrite(mp4_buffer, m4v_size, 1, out);                                  fwrite(mp4_buffer, m4v_size, 1, out);
1725                                  fclose(out);                                  fclose(out);
                                 output_num++;  
1726                          }                          }
1727    #ifdef XVID_AVI_OUTPUT
1728                            if (outavifilename && myAVIStream) {
1729                                    int output_frame;
1730    
1731                                    if (time_file == NULL)
1732                                            output_frame = (output_num-start_num);
1733                                    else {
1734                                            output_frame = (int)(atof(timecode)/1000/((double)ARG_DWSCALE/ARG_DWRATE)+.5);
1735                                    }
1736                                    if (AVIStreamWrite(myAVIStream, output_frame, 1, mp4_buffer, m4v_size, key ? AVIIF_KEYFRAME : 0, NULL, NULL)) {
1737                                            fprintf(stderr, "AVIStreamWrite failed writing frame %d\n", output_num);
1738                                            goto release_all;
1739                                    }
1740                            }
1741    #endif
1742    
1743                                    if (key && ARG_PACKED)
1744                                            removedivxp((char*)mp4_buffer, m4v_size);
1745    
1746                          /* Save ES stream */                          /* Save ES stream */
1747                          if (ARG_OUTPUTFILE && out_file)                                  if (outfilename && out_file && !(fakenvop && m4v_size <= 8)) {
1748                                  fwrite(mp4_buffer, 1, m4v_size, out_file);                                  fwrite(mp4_buffer, 1, m4v_size, out_file);
1749                  }                  }
1750    #ifdef XVID_MKV_OUTPUT
1751                                    if (outmkvfilename && myMKVStream) {
1752                                            MKVStreamWrite(myMKVStream, atof(timecode), 1, (ARG_PACKED && fakenvop && (m4v_size <= 8)) ? NULL : mp4_buffer, m4v_size, key ? AVIIF_KEYFRAME : 0, NULL, NULL);
1753                                    }
1754    #endif
1755    
1756                            output_num++;
1757                            if (stats_type != XVID_TYPE_BVOP)
1758                                    fakenvop=0;
1759                    }
1760    
1761                  input_num++;                  if (!result)
1762                            (input_num)++;
1763    
1764                  /* Read the header if it's pgm stream */                  /* Read the header if it's pgm stream */
1765                  if (!result && ARG_INPUTTYPE)                  if (!result && (ARG_INPUTTYPE==1))
1766  #ifndef READ_PNM  #ifndef READ_PNM
1767                          result = read_pgmheader(in_file);                          result = read_pgmheader(in_file);
1768  #else  #else
# Line 575  Line 1771 
1771          } while (1);          } while (1);
1772    
1773    
   
 /*****************************************************************************  
  *         Calculate totals and averages for output, print results  
  ****************************************************************************/  
   
         printf("Tot: enctime(ms) =%7.2f,               length(bytes) = %7d\n",  
                    totalenctime, (int) totalsize);  
   
         if (input_num > 0) {  
                 totalsize /= input_num;  
                 totalenctime /= input_num;  
                 totalPSNR[0] /= input_num;  
                 totalPSNR[1] /= input_num;  
                 totalPSNR[2] /= input_num;  
         } else {  
                 totalsize = -1;  
                 totalenctime = -1;  
         }  
   
         printf("Avg: enctime(ms) =%7.2f, fps =%7.2f, length(bytes) = %7d",  
                    totalenctime, 1000 / totalenctime, (int) totalsize);  
    if (ARG_STATS) {  
        printf(", psnr y = %2.2f, psnr u = %2.2f, psnr v = %2.2f",  
                   totalPSNR[0],totalPSNR[1],totalPSNR[2]);  
         }  
         printf("\n");  
   
   
 /*****************************************************************************  
  *                            XviD PART  Stop  
  ****************************************************************************/  
   
1774    release_all:    release_all:
1775    
1776      h->input_num = input_num;
1777      h->totalenctime = totalenctime;
1778      h->totalsize = totalsize;
1779    
1780    #ifdef XVID_AVI_INPUT
1781            if (get_frame) AVIStreamGetFrameClose(get_frame);
1782            if (avi_in_stream) AVIStreamRelease(avi_in_stream);
1783    #endif
1784    
1785          if (enc_handle) {          if (enc_handle) {
1786                  result = enc_stop();                  result = enc_stop(enc_handle);
1787                  if (result)                  if (result)
1788                          fprintf(stderr, "Encore RELEASE problem return value %d\n",                          fprintf(stderr, "Encore RELEASE problem return value %d\n",
1789                                          result);                                          result);
# Line 620  Line 1793 
1793                  fclose(in_file);                  fclose(in_file);
1794          if (out_file)          if (out_file)
1795                  fclose(out_file);                  fclose(out_file);
1796            if (time_file)
1797                    fclose(time_file);
1798    
1799    #ifdef XVID_AVI_OUTPUT
1800            if (myAVIStream) AVIStreamRelease(myAVIStream);
1801            if (myAVIFile) AVIFileRelease(myAVIFile);
1802    #endif
1803    #ifdef XVID_MKV_OUTPUT
1804            if (myMKVStream) MKVStreamRelease(myMKVStream);
1805            if (myMKVFile) MKVFileRelease(myMKVFile);
1806    #endif
1807    #if defined(XVID_AVI_INPUT) || defined(XVID_AVI_OUTPUT)
1808            AVIFileExit();
1809    #endif
1810    
1811    free_all_memory:    free_all_memory:
1812            if (out_buffer)
1813          free(out_buffer);          free(out_buffer);
1814            if (mp4_buffer)
1815          free(mp4_buffer);          free(mp4_buffer);
1816            if ((in_buffer) && (get_frame == NULL))
1817          free(in_buffer);          free(in_buffer);
   
         return (0);  
   
1818  }  }
1819    
   
1820  /*****************************************************************************  /*****************************************************************************
1821   *                        "statistical" functions   *                        "statistical" functions
1822   *   *
1823   *  these are not needed for encoding or decoding, but for measuring   *  these are not needed for encoding or decoding, but for measuring
1824   *  time and quality, there in nothing specific to XviD in these   *  time and quality, there in nothing specific to Xvid in these
1825   *   *
1826   *****************************************************************************/   *****************************************************************************/
1827    
# Line 652  Line 1838 
1838          clock_t clk;          clock_t clk;
1839    
1840          clk = clock();          clk = clock();
1841          return (clk * 1000 / CLOCKS_PER_SEC);          return (clk * 1000.0 / CLOCKS_PER_SEC);
1842  #endif  #endif
1843  }  }
1844    
1845    int
1846    gcd(int a, int b)
1847    {
1848            int r ;
1849    
1850            if (b > a) {
1851                    r = a;
1852                    a = b;
1853                    b = r;
1854            }
1855    
1856            while ((r = a % b)) {
1857                    a = b;
1858                    b = r;
1859            }
1860            return b;
1861    }
1862    
1863    int minquant(int quants[32])
1864    {
1865            int i = 1;
1866            while (quants[i] == 0) {
1867                    i++;
1868            }
1869            return i;
1870    }
1871    
1872    int maxquant(int quants[32])
1873    {
1874            int i = 31;
1875            while (quants[i] == 0) {
1876                    i--;
1877            }
1878            return i;
1879    }
1880    
1881    double avgquant(frame_stats_t frame)
1882    {
1883            double avg=0;
1884            int i;
1885            for (i=1; i < 32; i++) {
1886                    avg += frame.quants[i]*i;
1887            }
1888            avg /= frame.count;
1889            return avg;
1890    }
1891    
1892  /*****************************************************************************  /*****************************************************************************
1893   *                             Usage message   *                             Usage message
1894   *****************************************************************************/   *****************************************************************************/
# Line 663  Line 1896 
1896  static void  static void
1897  usage()  usage()
1898  {  {
1899          fprintf(stderr, "Usage : xvid_stat [OPTIONS]\n\n");          fprintf(stderr, "xvid_encraw built at %s on %s\n", __TIME__, __DATE__);
1900            fprintf(stderr, "Usage : xvid_encraw [OPTIONS]\n\n");
1901          fprintf(stderr, "Input options:\n");          fprintf(stderr, "Input options:\n");
1902          fprintf(stderr, " -i      string : input filename (default=stdin)\n");          fprintf(stderr, " -i      string  : input filename (stdin)\n");
1903          fprintf(stderr, " -type   integer: input data type (yuv=0, pgm=1)\n");  #ifdef XVID_AVI_INPUT
1904          fprintf(stderr, " -w      integer: frame width ([1.2048])\n");          fprintf(stderr, " -type   integer : input data type (yuv=0, pgm=1, avi/avs=2) (");
1905          fprintf(stderr, " -h      integer: frame height ([1.2048])\n");  #else
1906          fprintf(stderr, " -frames integer: number of frames to encode\n");          fprintf(stderr, " -type   integer : input data type (yuv=0, pgm=1) (");
1907          fprintf(stderr, "\n");  #endif
1908            fprintf(stderr, "%d)\n", ARG_INPUTTYPE);
1909            fprintf(stderr, " -w      integer : frame width ([1.%d])\n", MAX_XDIM);
1910            fprintf(stderr, " -h      integer : frame height ([1.%d])\n", MAX_YDIM);
1911            fprintf(stderr, " -csp    string  : colorspace of raw input file i420%s, yv12%s\n", (ARG_COLORSPACE == XVID_CSP_I420)?" (default)":"", (ARG_COLORSPACE == XVID_CSP_YV12)?" (default)":"");
1912            fprintf(stderr, " -frames integer : number of frames to encode (");
1913            if (ARG_MAXFRAMENR==-1)
1914                    fprintf(stderr, "all)");
1915            else
1916                    fprintf(stderr, "%d)", ARG_MAXFRAMENR);
1917            fprintf(stderr, "\n\n");
1918          fprintf(stderr, "Output options:\n");          fprintf(stderr, "Output options:\n");
1919          fprintf(stderr, " -dump    : save decoder output\n");          fprintf(stderr, " -dump    : save decoder output\n");
1920          fprintf(stderr, " -save    : save an Elementary Stream file per frame\n");          fprintf(stderr, " -save    : save an Elementary Stream file per frame\n");
1921          fprintf(stderr, " -o string: save an Elementary Stream for the complete sequence\n");          fprintf(stderr, " -o string: save an Elementary Stream for the complete sequence\n");
1922          fprintf(stderr, "\n");  #ifdef XVID_AVI_OUTPUT
1923          fprintf(stderr, "BFrames options:\n");          fprintf(stderr, " -avi    string  : save an AVI file for the complete sequence\n");
1924          fprintf(stderr, " -max_bframes   integer: max bframes (default=0)\n");  #endif
1925          fprintf(stderr, " -bquant_ratio  integer: bframe quantizer ratio (default=150)\n");  #ifdef XVID_MKV_OUTPUT
1926          fprintf(stderr, " -bquant_offset integer: bframe quantizer offset (default=100)\n");          fprintf(stderr, " -mkv    string  : save a MKV file for the complete sequence\n");
1927    #endif
1928          fprintf(stderr, "\n");          fprintf(stderr, "\n");
1929          fprintf(stderr, "Rate control options:\n");          fprintf(stderr, "Rate control options:\n");
1930          fprintf(stderr, " -framerate float               : target framerate (>0 | default=25.0)\n");          fprintf(stderr, " -framerate        float      : target framerate (auto)\n");
1931          fprintf(stderr, " -bitrate   integer             : target bitrate\n");          fprintf(stderr, " -bitrate          [integer]  : target bitrate in kbps (%d)\n", DEFAULT_BITRATE/1000);
1932      fprintf(stderr,     " -single                        : single pass mode\n");          fprintf(stderr, " -size             integer    : target size in kilobytes\n");
1933          fprintf(stderr, " -pass1     filename            : twopass mode (first pass)\n");          fprintf(stderr, " -single                      : single pass mode%s\n", (ARG_SINGLE)?" (default)":"");
1934          fprintf(stderr, " -pass2     filename            : twopass mode (2nd pass)\n");          fprintf(stderr, " -cq               float      : single pass constant quantizer\n");
1935            fprintf(stderr, " -pass1            [filename] : twopass mode (first pass)\n");
1936            fprintf(stderr, " -full1pass                   : perform full quality first pass (disabled)\n");
1937            fprintf(stderr, " -pass2            [filename] : twopass mode (2nd pass)\n");
1938            fprintf(stderr, " -max_key_interval integer    : maximum keyframe interval (%d)\n", ARG_MAXKEYINTERVAL);
1939            fprintf(stderr, " -zones            [zones]    : use `-help zones' for usage guidelines\n");
1940          fprintf(stderr, " -zq starting_frame float       : bitrate zone; quant\n");          fprintf(stderr, " -zq starting_frame float       : bitrate zone; quant\n");
1941          fprintf(stderr, " -zw starting_frame float       : bitrate zone; weight\n");          fprintf(stderr, " -zw starting_frame float       : bitrate zone; weight\n");
     fprintf(stderr, " -max_key_interval integer      : maximum keyframe interval\n");  
1942      fprintf(stderr, "\n");      fprintf(stderr, "\n");
1943          fprintf(stderr, "Other options\n");          fprintf(stderr, "Single Pass options:\n");
1944          fprintf(stderr, " -asm            : use assembly optmized code\n");          fprintf(stderr, " -reaction         integer    : reaction delay factor (%d)\n", ARG_REACTION);
1945          fprintf(stderr, " -quality integer: quality ([0..%d])\n", ME_ELEMENTS - 1);          fprintf(stderr, " -averaging        integer    : averaging period (%d)\n", ARG_AVERAGING);
1946          fprintf(stderr, " -qpel           : use quarter pixel ME\n");          fprintf(stderr, " -smoother         integer    : smoothing buffer (%d)\n", ARG_SMOOTHER);
1947          fprintf(stderr, " -gmc            : use global motion compensation\n");          fprintf(stderr, "\n");
1948          fprintf(stderr, " -packed         : packed mode\n");          fprintf(stderr, "Second Pass options:\n");
1949          fprintf(stderr, " -closed_gop     : closed GOP mode\n");          fprintf(stderr, " -kboost           integer    : I frame boost (%d)\n", ARG_KBOOST);
1950          fprintf(stderr, " -lumimasking    : use lumimasking algorithm\n");          fprintf(stderr, " -kthresh          integer    : I frame reduction threshold (%d)\n", ARG_KTHRESH);
1951          fprintf(stderr, " -stats          : print stats about encoded frames\n");          fprintf(stderr, " -kreduction       integer    : I frame reduction amount (%d)\n", ARG_KREDUCTION);
1952          fprintf(stderr, " -debug          : activates xvidcore internal debugging output\n");          fprintf(stderr, " -ostrength        integer    : overflow control strength (%d)\n", ARG_OVERSTRENGTH);
1953          fprintf(stderr, " -vop_debug      : print some info directly into encoded frames\n");          fprintf(stderr, " -oimprove         integer    : max overflow improvement (%d)\n", ARG_OVERIMPROVE);
1954          fprintf(stderr, " -help           : prints this help message\n");          fprintf(stderr, " -odegrade         integer    : max overflow degradation (%d)\n", ARG_OVERDEGRADE);
1955            fprintf(stderr, " -chigh            integer    : high bitrate scenes degradation (%d)\n", ARG_CHIGH);
1956            fprintf(stderr, " -clow             integer    : low bitrate scenes improvement (%d)\n", ARG_CLOW);
1957            fprintf(stderr, " -overhead         integer    : container frame overhead (%d)\n", ARG_OVERHEAD);
1958            fprintf(stderr, " -vbvsize          integer    : use vbv buffer size\n");
1959            fprintf(stderr, " -vbvmax           integer    : vbv max bitrate\n");
1960            fprintf(stderr, " -vbvpeak          integer    : vbv peak bitrate over 1 second\n");
1961            fprintf(stderr, "\n");
1962            fprintf(stderr, "BFrames options:\n");
1963            fprintf(stderr, " -max_bframes      integer    : max bframes (%d)\n", ARG_MAXBFRAMES);
1964            fprintf(stderr, " -bquant_ratio     integer    : bframe quantizer ratio (%d)\n", ARG_BQRATIO);
1965            fprintf(stderr, " -bquant_offset    integer    : bframe quantizer offset (%d)\n", ARG_BQOFFSET);
1966          fprintf(stderr, "\n");          fprintf(stderr, "\n");
1967          fprintf(stderr, "NB: You can define %d zones repeating the -z[qw] option as many times as needed.\n", MAX_ZONES);          fprintf(stderr, "Other options\n");
1968            if (ARG_USE_ASSEMBLER)
1969                    fprintf(stderr, " -noasm                       : do not use assembly optimized code (use)\n");
1970            else
1971                    fprintf(stderr, " -asm                         : use assembly optimized code (don't use)\n");
1972            if (ARG_TURBO)
1973                    fprintf(stderr, " -noturbo                     : do not use turbo presets for higher encoding speed (use)\n");
1974            else
1975                    fprintf(stderr, " -turbo                       : use turbo presets for higher encoding speed (don't use)\n");
1976            fprintf(stderr, " -quality          integer    : quality ([0..%d]) (%d)\n", ME_ELEMENTS - 1, ARG_QUALITY);
1977            fprintf(stderr, " -vhqmode          integer    : level of R-D optimizations ([0..4]) (%d)\n", ARG_VHQMODE);
1978            if (ARG_BVHQ)
1979                    fprintf(stderr, " -nobvhq                      : do not use R-D optimizations for B-frames (use)\n");
1980            else
1981                    fprintf(stderr, " -bvhq                        : use R-D optimizations for B-frames (don't use)\n");
1982            fprintf(stderr, " -metric           integer    : distortion metric for R-D opt (PSNR:0, PSNRHVSM:1) (%d)\n", ARG_QMETRIC);
1983            if (ARG_QPEL)
1984                    fprintf(stderr, " -noqpel                      : do not use quarter pixel ME (use)\n");
1985            else
1986                    fprintf(stderr, " -qpel                        : use quarter pixel ME (don't use)\n");
1987            if (ARG_GMC)
1988                    fprintf(stderr, " -nogmc                       : do not use global motion compensation (use)\n");
1989            else
1990                    fprintf(stderr, " -gmc                         : use global motion compensation (don't use)\n");
1991            fprintf(stderr, " -qtype            integer    : quantization type (H263:0, MPEG4:1) (%d)\n", ARG_QTYPE);
1992            fprintf(stderr, " -qmatrix          filename   : use custom MPEG4 quantization matrix\n");
1993            fprintf(stderr, " -interlaced       [integer]  : interlaced encoding (BFF:1, TFF:2) (%d)\n", DEFAULT_INTERLACING);
1994            if (ARG_PACKED)
1995                    fprintf(stderr, " -nopacked                    : disable packed B-frames mode (enabled)\n");
1996            else
1997                    fprintf(stderr, " -packed                      : enable packed B-frames mode (disabled)\n");
1998            if (ARG_CLOSED_GOP)
1999                    fprintf(stderr, " -noclosed_gop                : disable closed GOP mode (enabled)\n");
2000            else
2001                    fprintf(stderr, " -closed_gop                  : enable closed GOP mode (disabled)\n");
2002            fprintf(stderr, " -masking          [integer]  : HVS masking mode (None:0, Lumi:1, Variance:2) (%d)\n", ARG_LUMIMASKING);
2003            if (ARG_STATS)
2004                    fprintf(stderr, " -nostats                     : do not print stats about encoded frames (print)\n");
2005            else
2006                    fprintf(stderr, " -stats                       : print stats about encoded frames (don't print)\n");
2007            fprintf(stderr, " -ssim             [integer]  : print ssim for every frame (accurate: 0 fast: 4) (%d)\n", DEFAULT_SSIM);
2008            fprintf(stderr, " -ssim_file        filename   : output the ssim stats into a file\n");
2009            if (ARG_PSNRHVSM)
2010                    fprintf(stderr, " -nopsnrhvsm                  : do not print PSNRHVSM metric for every frame (print)\n");
2011            else
2012                    fprintf(stderr, " -psnrhvsm                    : print PSNRHVSM metric for every frame (don't print)\n");
2013            fprintf(stderr, " -debug            integer    : activate xvidcore internal debugging output (don't activate)\n");
2014            if (ARG_VOPDEBUG)
2015                    fprintf(stderr, " -novop_debug                 : do not print debug info directly into encoded frames (print)\n");
2016            else
2017                    fprintf(stderr, " -vop_debug                   : print some info directly into encoded frames (don't print)\n");
2018            if (ARG_CHROMAME)
2019                    fprintf(stderr, " -nochromame                  : disable chroma motion estimation (enabled)\n");
2020            else
2021                    fprintf(stderr, " -chromame                    : enable chroma motion estimation (disabled)\n");
2022            if (ARG_TRELLIS)
2023                    fprintf(stderr, " -notrellis                   : disable trellis quantization (enabled)\n");
2024            else
2025                    fprintf(stderr, " -trellis                     : enable trellis quantization (disabled)\n");
2026            fprintf(stderr, " -imin             integer    : minimum I Quantizer (1..31) (%d)\n", ARG_QUANTS[0]);
2027            fprintf(stderr, " -imax             integer    : maximum I quantizer (1..31) (%d)\n", ARG_QUANTS[1]);
2028            fprintf(stderr, " -bmin             integer    : minimum B Quantizer (1..31) (%d)\n", ARG_QUANTS[4]);
2029            fprintf(stderr, " -bmax             integer    : maximum B quantizer (1..31) (%d)\n", ARG_QUANTS[5]);
2030            fprintf(stderr, " -pmin             integer    : minimum P Quantizer (1..31) (%d)\n", ARG_QUANTS[2]);
2031            fprintf(stderr, " -pmax             integer    : maximum P quantizer (1..31) (%d)\n", ARG_QUANTS[3]);
2032            fprintf(stderr, " -drop             integer    : frame Drop Ratio (0..100) (%d)\n", ARG_FRAMEDROP);
2033            fprintf(stderr, " -start            integer    : starting frame number (%d)\n", ARG_STARTFRAMENR);
2034            fprintf(stderr, " -threads          integer    : number of threads (auto)\n");
2035            fprintf(stderr, " -slices           integer    : number of slices (%d)\n", ARG_SLICES);
2036            fprintf(stderr, " -progress         [integer]  : show progress updates every n frames (%d)\n", DEFAULT_PROGRESS);
2037            fprintf(stderr, " -par       integer[:integer] : set Pixel Aspect Ratio (%d)\n", ARG_PAR);
2038            fprintf(stderr, "                                1 = 1:1\n");
2039            fprintf(stderr, "                                2 = 12:11 (4:3 PAL)\n");
2040            fprintf(stderr, "                                3 = 10:11 (4:3 NTSC)\n");
2041            fprintf(stderr, "                                4 = 16:11 (16:9 PAL)\n");
2042            fprintf(stderr, "                                5 = 40:33 (16:9 NTSC)\n");
2043            fprintf(stderr, "                            other = custom (width:height)\n");
2044            fprintf(stderr, " -help                        : print this help message\n");
2045          fprintf(stderr, "\n");          fprintf(stderr, "\n");
2046            fprintf(stderr, "NB: You can define up to %d zones via both the -zones and the -z[qw] options as needed.\n", MAX_ZONES);
2047  }  }
2048    
2049  /*****************************************************************************  /*****************************************************************************
2050   *                       Input and output functions   *                       Input and output functions
2051   *   *
2052   *      the are small and simple routines to read and write PGM and YUV   *      the are small and simple routines to read and write PGM and YUV
2053   *      image. It's just for convenience, again nothing specific to XviD   *      image. It's just for convenience, again nothing specific to Xvid
2054   *   *
2055   *****************************************************************************/   *****************************************************************************/
2056    
# Line 723  Line 2061 
2061          int bytes, xsize, ysize, depth;          int bytes, xsize, ysize, depth;
2062          char dummy[2];          char dummy[2];
2063    
2064          bytes = fread(dummy, 1, 2, handle);          bytes = (int) fread(dummy, 1, 2, handle);
2065    
2066          if ((bytes < 2) || (dummy[0] != 'P') || (dummy[1] != '5'))          if ((bytes < 2) || (dummy[0] != 'P') || (dummy[1] != '5'))
2067                  return (1);                  return (1);
2068    
2069          fscanf(handle, "%d %d %d", &xsize, &ysize, &depth);          fscanf(handle, "%d %d %d", &xsize, &ysize, &depth);
2070            fread(&dummy, 1, 1, handle);
2071          if ((xsize > 4096) || (ysize > 4096*3/2) || (depth != 255)) {          if ((xsize > 4096) || (ysize > 4096*3/2) || (depth != 255)) {
2072                  fprintf(stderr, "%d %d %d\n", xsize, ysize, depth);                  fprintf(stderr, "%d %d %d\n", xsize, ysize, depth);
2073                  return (2);                  return (2);
# Line 746  Line 2085 
2085                           unsigned char *image)                           unsigned char *image)
2086  {  {
2087          int i;          int i;
         char dummy;  
2088    
2089          unsigned char *y = image;          unsigned char *y = image;
2090          unsigned char *u = image + XDIM * YDIM;          unsigned char *u = image + XDIM * YDIM;
# Line 767  Line 2105 
2105                  v += XDIM / 2;                  v += XDIM / 2;
2106          }          }
2107    
         /*  I don't know why, but this seems needed */  
         fread(&dummy, 1, 1, handle);  
   
2108          return (0);          return (0);
2109  }  }
2110  #else  #else
# Line 831  Line 2166 
2166    
2167  /* sample plugin */  /* sample plugin */
2168    
2169    #if 0
2170  int  int
2171  rawenc_debug(void *handle,  rawenc_debug(void *handle,
2172                           int opt,                           int opt,
# Line 870  Line 2206 
2206    
2207          return XVID_ERR_FAIL;          return XVID_ERR_FAIL;
2208  }  }
2209    #endif
2210    
2211  #define FRAMERATE_INCR 1001  #define FRAMERATE_INCR 1001
2212    
2213    /* Gobal encoder init, once per process */
2214  /* Initialize encoder for first use, pass all needed parameters to the codec */  void
2215  static int  enc_gbl(int use_assembler)
 enc_init(int use_assembler)  
2216  {  {
         int xerr;  
         //xvid_plugin_cbr_t cbr;  
     xvid_plugin_single_t single;  
         xvid_plugin_2pass1_t rc2pass1;  
         xvid_plugin_2pass2_t rc2pass2;  
         //xvid_plugin_fixed_t rcfixed;  
         xvid_enc_plugin_t plugins[7];  
2217          xvid_gbl_init_t xvid_gbl_init;          xvid_gbl_init_t xvid_gbl_init;
         xvid_enc_create_t xvid_enc_create;  
2218    
2219          /*------------------------------------------------------------------------          /*------------------------------------------------------------------------
2220           * XviD core initialization           * Xvid core initialization
2221           *----------------------------------------------------------------------*/           *----------------------------------------------------------------------*/
2222    
2223          /* Set version -- version checking will done by xvidcore */          /* Set version -- version checking will done by xvidcore */
# Line 911  Line 2238 
2238                  xvid_gbl_init.cpu_flags = XVID_CPU_FORCE;                  xvid_gbl_init.cpu_flags = XVID_CPU_FORCE;
2239          }          }
2240    
2241          /* Initialize XviD core -- Should be done once per __process__ */          /* Initialize Xvid core -- Should be done once per __process__ */
2242          xvid_global(NULL, XVID_GBL_INIT, &xvid_gbl_init, NULL);          xvid_global(NULL, XVID_GBL_INIT, &xvid_gbl_init, NULL);
2243        ARG_CPU_FLAGS = xvid_gbl_init.cpu_flags;
2244            enc_info();
2245    }
2246    
2247    /* Initialize encoder for first use, pass all needed parameters to the codec */
2248    static int
2249    enc_init(void **enc_handle, char *stats_pass1, int start_num)
2250    {
2251            int xerr;
2252            //xvid_plugin_cbr_t cbr;
2253        xvid_plugin_single_t single;
2254            xvid_plugin_2pass1_t rc2pass1;
2255            xvid_plugin_2pass2_t rc2pass2;
2256            xvid_plugin_ssim_t ssim;
2257        xvid_plugin_lumimasking_t masking;
2258            //xvid_plugin_fixed_t rcfixed;
2259            xvid_enc_plugin_t plugins[8];
2260            xvid_enc_create_t xvid_enc_create;
2261            int i;
2262    
2263          /*------------------------------------------------------------------------          /*------------------------------------------------------------------------
2264           * XviD encoder initialization           * Xvid encoder initialization
2265           *----------------------------------------------------------------------*/           *----------------------------------------------------------------------*/
2266    
2267          /* Version again */          /* Version again */
# Line 925  Line 2271 
2271          /* Width and Height of input frames */          /* Width and Height of input frames */
2272          xvid_enc_create.width = XDIM;          xvid_enc_create.width = XDIM;
2273          xvid_enc_create.height = YDIM;          xvid_enc_create.height = YDIM;
2274          xvid_enc_create.profile = XVID_PROFILE_AS_L4;          xvid_enc_create.profile = 0xf5; /* Unrestricted */
2275    
2276          /* init plugins  */          /* init plugins  */
2277      xvid_enc_create.zones = ZONES;  //    xvid_enc_create.zones = ZONES;
2278      xvid_enc_create.num_zones = NUM_ZONES;  //    xvid_enc_create.num_zones = NUM_ZONES;
2279    
2280          xvid_enc_create.plugins = plugins;          xvid_enc_create.plugins = plugins;
2281          xvid_enc_create.num_plugins = 0;          xvid_enc_create.num_plugins = 0;
# Line 938  Line 2284 
2284                  memset(&single, 0, sizeof(xvid_plugin_single_t));                  memset(&single, 0, sizeof(xvid_plugin_single_t));
2285                  single.version = XVID_VERSION;                  single.version = XVID_VERSION;
2286                  single.bitrate = ARG_BITRATE;                  single.bitrate = ARG_BITRATE;
2287                    single.reaction_delay_factor = ARG_REACTION;
2288                    single.averaging_period = ARG_AVERAGING;
2289                    single.buffer = ARG_SMOOTHER;
2290    
2291    
2292                  plugins[xvid_enc_create.num_plugins].func = xvid_plugin_single;                  plugins[xvid_enc_create.num_plugins].func = xvid_plugin_single;
2293                  plugins[xvid_enc_create.num_plugins].param = &single;                  plugins[xvid_enc_create.num_plugins].param = &single;
2294                  xvid_enc_create.num_plugins++;                  xvid_enc_create.num_plugins++;
2295                    if (!ARG_BITRATE)
2296                            prepare_cquant_zones();
2297          }          }
2298    
2299          if (ARG_PASS2) {          if (ARG_PASS2) {
# Line 950  Line 2302 
2302                  rc2pass2.filename = ARG_PASS2;                  rc2pass2.filename = ARG_PASS2;
2303                  rc2pass2.bitrate = ARG_BITRATE;                  rc2pass2.bitrate = ARG_BITRATE;
2304    
2305                    rc2pass2.keyframe_boost = ARG_KBOOST;
2306                    rc2pass2.curve_compression_high = ARG_CHIGH;
2307                    rc2pass2.curve_compression_low = ARG_CLOW;
2308                    rc2pass2.overflow_control_strength = ARG_OVERSTRENGTH;
2309                    rc2pass2.max_overflow_improvement = ARG_OVERIMPROVE;
2310                    rc2pass2.max_overflow_degradation = ARG_OVERDEGRADE;
2311                    rc2pass2.kfreduction = ARG_KREDUCTION;
2312                    rc2pass2.kfthreshold = ARG_KTHRESH;
2313                    rc2pass2.container_frame_overhead = ARG_OVERHEAD;
2314    
2315    //              An example of activating VBV could look like this
2316                    rc2pass2.vbv_size     =  ARG_VBVSIZE;
2317                    rc2pass2.vbv_initial  =  (ARG_VBVSIZE*3)/4;
2318                    rc2pass2.vbv_maxrate  =  ARG_VBVMAXRATE;
2319                    rc2pass2.vbv_peakrate =  ARG_VBVPEAKRATE;
2320    
2321    
2322                  plugins[xvid_enc_create.num_plugins].func = xvid_plugin_2pass2;                  plugins[xvid_enc_create.num_plugins].func = xvid_plugin_2pass2;
2323                  plugins[xvid_enc_create.num_plugins].param = &rc2pass2;                  plugins[xvid_enc_create.num_plugins].param = &rc2pass2;
2324                  xvid_enc_create.num_plugins++;                  xvid_enc_create.num_plugins++;
2325          }          }
2326    
2327          if (ARG_PASS1) {          if (stats_pass1) {
2328                  memset(&rc2pass1, 0, sizeof(xvid_plugin_2pass1_t));                  memset(&rc2pass1, 0, sizeof(xvid_plugin_2pass1_t));
2329                  rc2pass1.version = XVID_VERSION;                  rc2pass1.version = XVID_VERSION;
2330                  rc2pass1.filename = ARG_PASS1;                  rc2pass1.filename = stats_pass1;
2331                    if (ARG_FULL1PASS)
2332                            prepare_full1pass_zones();
2333                  plugins[xvid_enc_create.num_plugins].func = xvid_plugin_2pass1;                  plugins[xvid_enc_create.num_plugins].func = xvid_plugin_2pass1;
2334                  plugins[xvid_enc_create.num_plugins].param = &rc2pass1;                  plugins[xvid_enc_create.num_plugins].param = &rc2pass1;
2335                  xvid_enc_create.num_plugins++;                  xvid_enc_create.num_plugins++;
2336          }          }
2337    
2338            /* Zones stuff */
2339            xvid_enc_create.zones = (xvid_enc_zone_t*)malloc(sizeof(xvid_enc_zone_t) * NUM_ZONES);
2340            xvid_enc_create.num_zones = NUM_ZONES;
2341            for (i=0; i < xvid_enc_create.num_zones; i++) {
2342                    xvid_enc_create.zones[i].frame = ZONES[i].frame;
2343                    xvid_enc_create.zones[i].base = 100;
2344                    xvid_enc_create.zones[i].mode = ZONES[i].mode;
2345                    xvid_enc_create.zones[i].increment = ZONES[i].modifier;
2346            }
2347    
2348    
2349          if (ARG_LUMIMASKING) {          if (ARG_LUMIMASKING) {
2350                    memset(&masking, 0, sizeof(xvid_plugin_lumimasking_t));
2351                    masking.method = (ARG_LUMIMASKING==2);
2352                  plugins[xvid_enc_create.num_plugins].func = xvid_plugin_lumimasking;                  plugins[xvid_enc_create.num_plugins].func = xvid_plugin_lumimasking;
2353                  plugins[xvid_enc_create.num_plugins].param = NULL;                  plugins[xvid_enc_create.num_plugins].param = &masking;
2354                  xvid_enc_create.num_plugins++;                  xvid_enc_create.num_plugins++;
2355          }          }
2356    
# Line 977  Line 2360 
2360                  xvid_enc_create.num_plugins++;                  xvid_enc_create.num_plugins++;
2361          }          }
2362    
2363            if (ARG_SSIM>=0 || ARG_SSIM_PATH != NULL) {
2364            memset(&ssim, 0, sizeof(xvid_plugin_ssim_t));
2365    
2366            plugins[xvid_enc_create.num_plugins].func = xvid_plugin_ssim;
2367    
2368                    if( ARG_SSIM >=0){
2369                            ssim.b_printstat = 1;
2370                            ssim.acc = ARG_SSIM;
2371                    } else {
2372                            ssim.b_printstat = 0;
2373                            ssim.acc = 2;
2374                    }
2375    
2376                    if(ARG_SSIM_PATH != NULL){
2377                            ssim.stat_path = ARG_SSIM_PATH;
2378                    }
2379    
2380            ssim.cpu_flags = ARG_CPU_FLAGS;
2381                    ssim.b_visualize = 0;
2382                    plugins[xvid_enc_create.num_plugins].param = &ssim;
2383                    xvid_enc_create.num_plugins++;
2384            }
2385    
2386            if (ARG_PSNRHVSM>0) {
2387            plugins[xvid_enc_create.num_plugins].func = xvid_plugin_psnrhvsm;
2388                    plugins[xvid_enc_create.num_plugins].param = NULL;
2389                    xvid_enc_create.num_plugins++;
2390            }
2391    
2392  #if 0  #if 0
2393          if (ARG_DEBUG) {          if (ARG_DEBUG) {
2394                  plugins[xvid_enc_create.num_plugins].func = rawenc_debug;                  plugins[xvid_enc_create.num_plugins].func = rawenc_debug;
# Line 985  Line 2397 
2397          }          }
2398  #endif  #endif
2399    
2400          /* No fancy thread tests */          xvid_enc_create.num_threads = ARG_THREADS;
2401          xvid_enc_create.num_threads = 0;          xvid_enc_create.num_slices  = ARG_SLICES;
2402    
2403          /* Frame rate - Do some quick float fps = fincr/fbase hack */          /* Frame rate  */
2404          if ((ARG_FRAMERATE - (int) ARG_FRAMERATE) < SMALL_EPS) {          xvid_enc_create.fincr = ARG_DWSCALE;
2405                  xvid_enc_create.fincr = 1;          xvid_enc_create.fbase = ARG_DWRATE;
                 xvid_enc_create.fbase = (int) ARG_FRAMERATE;  
         } else {  
                 xvid_enc_create.fincr = FRAMERATE_INCR;  
                 xvid_enc_create.fbase = (int) (FRAMERATE_INCR * ARG_FRAMERATE);  
         }  
2406    
2407          /* Maximum key frame interval */          /* Maximum key frame interval */
2408      if (ARG_MAXKEYINTERVAL > 0) {      if (ARG_MAXKEYINTERVAL > 0) {
# Line 1004  Line 2411 
2411              xvid_enc_create.max_key_interval = (int) ARG_FRAMERATE *10;              xvid_enc_create.max_key_interval = (int) ARG_FRAMERATE *10;
2412      }      }
2413    
2414            xvid_enc_create.min_quant[0]=ARG_QUANTS[0];
2415            xvid_enc_create.min_quant[1]=ARG_QUANTS[2];
2416            xvid_enc_create.min_quant[2]=ARG_QUANTS[4];
2417            xvid_enc_create.max_quant[0]=ARG_QUANTS[1];
2418            xvid_enc_create.max_quant[1]=ARG_QUANTS[3];
2419            xvid_enc_create.max_quant[2]=ARG_QUANTS[5];
2420    
2421          /* Bframes settings */          /* Bframes settings */
2422          xvid_enc_create.max_bframes = ARG_MAXBFRAMES;          xvid_enc_create.max_bframes = ARG_MAXBFRAMES;
2423          xvid_enc_create.bquant_ratio = ARG_BQRATIO;          xvid_enc_create.bquant_ratio = ARG_BQRATIO;
2424          xvid_enc_create.bquant_offset = ARG_BQOFFSET;          xvid_enc_create.bquant_offset = ARG_BQOFFSET;
2425    
2426          /* Dropping ratio frame -- we don't need that */          /* Frame drop ratio */
2427          xvid_enc_create.frame_drop_ratio = 0;          xvid_enc_create.frame_drop_ratio = ARG_FRAMEDROP;
2428    
2429            /* Start frame number */
2430            xvid_enc_create.start_frame_num = start_num;
2431    
2432          /* Global encoder options */          /* Global encoder options */
2433          xvid_enc_create.global = 0;          xvid_enc_create.global = 0;
# Line 1028  Line 2445 
2445          xerr = xvid_encore(NULL, XVID_ENC_CREATE, &xvid_enc_create, NULL);          xerr = xvid_encore(NULL, XVID_ENC_CREATE, &xvid_enc_create, NULL);
2446    
2447          /* Retrieve the encoder instance from the structure */          /* Retrieve the encoder instance from the structure */
2448          enc_handle = xvid_enc_create.handle;          *enc_handle = xvid_enc_create.handle;
2449    
2450            free(xvid_enc_create.zones);
2451    
2452          return (xerr);          return (xerr);
2453  }  }
2454    
2455  static int  static int
2456  enc_stop()  enc_info()
2457    {
2458            xvid_gbl_info_t xvid_gbl_info;
2459            int ret;
2460    
2461            memset(&xvid_gbl_info, 0, sizeof(xvid_gbl_info));
2462            xvid_gbl_info.version = XVID_VERSION;
2463            ret = xvid_global(NULL, XVID_GBL_INFO, &xvid_gbl_info, NULL);
2464            if (xvid_gbl_info.build != NULL) {
2465                    fprintf(stderr, "xvidcore build version: %s\n", xvid_gbl_info.build);
2466            }
2467            fprintf(stderr, "Bitstream version: %d.%d.%d\n", XVID_VERSION_MAJOR(xvid_gbl_info.actual_version), XVID_VERSION_MINOR(xvid_gbl_info.actual_version), XVID_VERSION_PATCH(xvid_gbl_info.actual_version));
2468            fprintf(stderr, "Detected CPU flags: ");
2469            if (xvid_gbl_info.cpu_flags & XVID_CPU_ASM)
2470                    fprintf(stderr, "ASM ");
2471            if (xvid_gbl_info.cpu_flags & XVID_CPU_MMX)
2472                    fprintf(stderr, "MMX ");
2473            if (xvid_gbl_info.cpu_flags & XVID_CPU_MMXEXT)
2474                    fprintf(stderr, "MMXEXT ");
2475            if (xvid_gbl_info.cpu_flags & XVID_CPU_SSE)
2476                    fprintf(stderr, "SSE ");
2477            if (xvid_gbl_info.cpu_flags & XVID_CPU_SSE2)
2478                    fprintf(stderr, "SSE2 ");
2479            if (xvid_gbl_info.cpu_flags & XVID_CPU_SSE3)
2480                    fprintf(stderr, "SSE3 ");
2481            if (xvid_gbl_info.cpu_flags & XVID_CPU_SSE41)
2482                    fprintf(stderr, "SSE41 ");
2483        if (xvid_gbl_info.cpu_flags & XVID_CPU_3DNOW)
2484                    fprintf(stderr, "3DNOW ");
2485            if (xvid_gbl_info.cpu_flags & XVID_CPU_3DNOWEXT)
2486                    fprintf(stderr, "3DNOWEXT ");
2487            if (xvid_gbl_info.cpu_flags & XVID_CPU_TSC)
2488                    fprintf(stderr, "TSC ");
2489            fprintf(stderr, "\n");
2490            fprintf(stderr, "Detected %d cpus,", xvid_gbl_info.num_threads);
2491            ARG_NUM_APP_THREADS = xvid_gbl_info.num_threads;
2492            fprintf(stderr, " using %d threads.\n", (!ARG_THREADS) ? ARG_NUM_APP_THREADS : ARG_THREADS);
2493            return ret;
2494    }
2495    
2496    static int
2497    enc_stop(void *enc_handle)
2498  {  {
2499          int xerr;          int xerr;
2500    
# Line 1045  Line 2505 
2505  }  }
2506    
2507  static int  static int
2508  enc_main(unsigned char *image,  enc_main(void *enc_handle,
2509                     unsigned char *image,
2510                   unsigned char *bitstream,                   unsigned char *bitstream,
2511                   int *key,                   int *key,
2512                   int *stats_type,                   int *stats_type,
2513                   int *stats_quant,                   int *stats_quant,
2514                   int *stats_length,                   int *stats_length,
2515                   int sse[3])                   int sse[3],
2516                     int framenum)
2517  {  {
2518          int ret;          int ret;
2519    
# Line 1073  Line 2535 
2535          if (image) {          if (image) {
2536                  xvid_enc_frame.input.plane[0] = image;                  xvid_enc_frame.input.plane[0] = image;
2537  #ifndef READ_PNM  #ifndef READ_PNM
2538                  xvid_enc_frame.input.csp = XVID_CSP_I420;                  xvid_enc_frame.input.csp = ARG_COLORSPACE;
2539                  xvid_enc_frame.input.stride[0] = XDIM;                  xvid_enc_frame.input.stride[0] = XDIM;
2540  #else  #else
2541                  xvid_enc_frame.input.csp = XVID_CSP_BGR;                  xvid_enc_frame.input.csp = XVID_CSP_BGR;
# Line 1087  Line 2549 
2549          xvid_enc_frame.vol_flags = 0;          xvid_enc_frame.vol_flags = 0;
2550          if (ARG_STATS)          if (ARG_STATS)
2551                  xvid_enc_frame.vol_flags |= XVID_VOL_EXTRASTATS;                  xvid_enc_frame.vol_flags |= XVID_VOL_EXTRASTATS;
2552          if (ARG_QPEL)          if (ARG_QTYPE) {
2553                    xvid_enc_frame.vol_flags |= XVID_VOL_MPEGQUANT;
2554                    if (ARG_QMATRIX) {
2555                            xvid_enc_frame.quant_intra_matrix = qmatrix_intra;
2556                            xvid_enc_frame.quant_inter_matrix = qmatrix_inter;
2557                    }
2558                    else {
2559                            /* We don't use special matrices */
2560                            xvid_enc_frame.quant_intra_matrix = NULL;
2561                            xvid_enc_frame.quant_inter_matrix = NULL;
2562                    }
2563            }
2564    
2565            if (ARG_PAR)
2566                    xvid_enc_frame.par = ARG_PAR;
2567            else {
2568                    xvid_enc_frame.par = XVID_PAR_EXT;
2569                    xvid_enc_frame.par_width = ARG_PARWIDTH;
2570                    xvid_enc_frame.par_height = ARG_PARHEIGHT;
2571            }
2572    
2573    
2574            if (ARG_QPEL) {
2575                  xvid_enc_frame.vol_flags |= XVID_VOL_QUARTERPEL;                  xvid_enc_frame.vol_flags |= XVID_VOL_QUARTERPEL;
2576          if (ARG_GMC)                  xvid_enc_frame.motion |= XVID_ME_QUARTERPELREFINE16 | XVID_ME_QUARTERPELREFINE8;
2577            }
2578            if (ARG_GMC) {
2579                  xvid_enc_frame.vol_flags |= XVID_VOL_GMC;                  xvid_enc_frame.vol_flags |= XVID_VOL_GMC;
2580                    xvid_enc_frame.motion |= XVID_ME_GME_REFINE;
2581            }
2582    
2583          /* Set up core's general features */          /* Set up core's general features */
2584          xvid_enc_frame.vop_flags = vop_presets[ARG_QUALITY];          xvid_enc_frame.vop_flags = vop_presets[ARG_QUALITY];
2585    
2586            if (ARG_INTERLACING) {
2587                    xvid_enc_frame.vol_flags |= XVID_VOL_INTERLACING;
2588                    if (ARG_INTERLACING == 2)
2589                            xvid_enc_frame.vop_flags |= XVID_VOP_TOPFIELDFIRST;
2590            }
2591    
2592            xvid_enc_frame.vop_flags |= XVID_VOP_HALFPEL;
2593            xvid_enc_frame.vop_flags |= XVID_VOP_HQACPRED;
2594    
2595      if (ARG_VOPDEBUG) {      if (ARG_VOPDEBUG) {
2596          xvid_enc_frame.vop_flags |= XVID_VOP_DEBUG;          xvid_enc_frame.vop_flags |= XVID_VOP_DEBUG;
2597      }      }
2598    
2599          /* Frame type -- let core decide for us */      if (ARG_TRELLIS) {
2600          xvid_enc_frame.type = XVID_TYPE_AUTO;          xvid_enc_frame.vop_flags |= XVID_VOP_TRELLISQUANT;
2601        }
2602    
2603            /* Frame type -- taken from function call parameter */
2604            /* Sometimes we might want to force the last frame to be a P Frame */
2605            xvid_enc_frame.type = *stats_type;
2606    
2607          /* Force the right quantizer -- It is internally managed by RC plugins */          /* Force the right quantizer -- It is internally managed by RC plugins */
2608          xvid_enc_frame.quant = 0;          xvid_enc_frame.quant = 0;
2609    
2610            if (ARG_CHROMAME)
2611                    xvid_enc_frame.motion |= XVID_ME_CHROMA_PVOP + XVID_ME_CHROMA_BVOP;
2612    
2613          /* Set up motion estimation flags */          /* Set up motion estimation flags */
2614          xvid_enc_frame.motion = motion_presets[ARG_QUALITY];          xvid_enc_frame.motion |= motion_presets[ARG_QUALITY];
2615    
2616          if (ARG_GMC)          if (ARG_TURBO)
2617                  xvid_enc_frame.motion |= XVID_ME_GME_REFINE;                  xvid_enc_frame.motion |= XVID_ME_FASTREFINE16 | XVID_ME_FASTREFINE8 |
2618                                                                     XVID_ME_SKIP_DELTASEARCH | XVID_ME_FAST_MODEINTERPOLATE |
2619                                                                     XVID_ME_BFRAME_EARLYSTOP;
2620    
2621          if (ARG_QPEL)          if (ARG_BVHQ)
2622                  xvid_enc_frame.motion |= XVID_ME_QUARTERPELREFINE16;                  xvid_enc_frame.vop_flags |= XVID_VOP_RD_BVOP;
2623          if (ARG_QPEL && (xvid_enc_frame.vop_flags & XVID_VOP_INTER4V))  
2624                  xvid_enc_frame.motion |= XVID_ME_QUARTERPELREFINE8;          if (ARG_QMETRIC == 1)
2625                    xvid_enc_frame.vop_flags |= XVID_VOP_RD_PSNRHVSM;
2626    
2627            switch (ARG_VHQMODE) /* this is the same code as for vfw */
2628            {
2629            case 1: /* VHQ_MODE_DECISION */
2630                    xvid_enc_frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
2631                    break;
2632    
2633            case 2: /* VHQ_LIMITED_SEARCH */
2634                    xvid_enc_frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
2635                    xvid_enc_frame.motion |= XVID_ME_HALFPELREFINE16_RD;
2636                    xvid_enc_frame.motion |= XVID_ME_QUARTERPELREFINE16_RD;
2637                    break;
2638    
2639            case 3: /* VHQ_MEDIUM_SEARCH */
2640                    xvid_enc_frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
2641                    xvid_enc_frame.motion |= XVID_ME_HALFPELREFINE16_RD;
2642                    xvid_enc_frame.motion |= XVID_ME_HALFPELREFINE8_RD;
2643                    xvid_enc_frame.motion |= XVID_ME_QUARTERPELREFINE16_RD;
2644                    xvid_enc_frame.motion |= XVID_ME_QUARTERPELREFINE8_RD;
2645                    xvid_enc_frame.motion |= XVID_ME_CHECKPREDICTION_RD;
2646                    break;
2647    
2648            case 4: /* VHQ_WIDE_SEARCH */
2649                    xvid_enc_frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
2650                    xvid_enc_frame.motion |= XVID_ME_HALFPELREFINE16_RD;
2651                    xvid_enc_frame.motion |= XVID_ME_HALFPELREFINE8_RD;
2652                    xvid_enc_frame.motion |= XVID_ME_QUARTERPELREFINE16_RD;
2653                    xvid_enc_frame.motion |= XVID_ME_QUARTERPELREFINE8_RD;
2654                    xvid_enc_frame.motion |= XVID_ME_CHECKPREDICTION_RD;
2655                    xvid_enc_frame.motion |= XVID_ME_EXTSEARCH_RD;
2656                    break;
2657    
2658            default :
2659                    break;
2660            }
2661    
2662            /* Not sure what this does */
2663            // force keyframe spacing in 2-pass 1st pass
2664            if (ARG_QUALITY == 0)
2665                    xvid_enc_frame.type = XVID_TYPE_IVOP;
2666    
2667            /* frame-based stuff */
2668            apply_zone_modifiers(&xvid_enc_frame, framenum);
2669    
         /* We don't use special matrices */  
         xvid_enc_frame.quant_intra_matrix = NULL;  
         xvid_enc_frame.quant_inter_matrix = NULL;  
2670    
2671          /* Encode the frame */          /* Encode the frame */
2672          ret = xvid_encore(enc_handle, XVID_ENC_ENCODE, &xvid_enc_frame,          ret = xvid_encore(enc_handle, XVID_ENC_ENCODE, &xvid_enc_frame,
# Line 1134  Line 2682 
2682    
2683          return (ret);          return (ret);
2684  }  }
2685    
2686    void
2687    sort_zones(zone_t * zones, int zone_num, int * sel)
2688    {
2689            int i, j;
2690            zone_t tmp;
2691            for (i = 0; i < zone_num; i++) {
2692                    int cur = i;
2693                    int min_f = zones[i].frame;
2694                    for (j = i + 1; j < zone_num; j++) {
2695                            if (zones[j].frame < min_f) {
2696                                    min_f = zones[j].frame;
2697                                    cur = j;
2698                            }
2699                    }
2700                    if (cur != i) {
2701                            tmp = zones[i];
2702                            zones[i] = zones[cur];
2703                            zones[cur] = tmp;
2704                            if (i == *sel) *sel = cur;
2705                            else if (cur == *sel) *sel = i;
2706                    }
2707            }
2708    }
2709    
2710    /* constant-quant zones for fixed quant encoding */
2711    static void
2712    prepare_cquant_zones() {
2713    
2714            int i = 0;
2715            if (NUM_ZONES == 0 || ZONES[0].frame != 0) {
2716                    /* first zone does not start at frame 0 or doesn't exist */
2717    
2718                    if (NUM_ZONES >= MAX_ZONES) NUM_ZONES--; /* we sacrifice last zone */
2719    
2720                    ZONES[NUM_ZONES].frame = 0;
2721                    ZONES[NUM_ZONES].mode = XVID_ZONE_QUANT;
2722                    ZONES[NUM_ZONES].modifier = (int) ARG_CQ;
2723                    ZONES[NUM_ZONES].type = XVID_TYPE_AUTO;
2724                    ZONES[NUM_ZONES].greyscale = 0;
2725                    ZONES[NUM_ZONES].chroma_opt = 0;
2726                    ZONES[NUM_ZONES].bvop_threshold = 0;
2727                    ZONES[NUM_ZONES].cartoon_mode = 0;
2728                    NUM_ZONES++;
2729    
2730                    sort_zones(ZONES, NUM_ZONES, &i);
2731            }
2732    
2733            /* step 2: let's change all weight zones into quant zones */
2734    
2735            for(i = 0; i < NUM_ZONES; i++)
2736                    if (ZONES[i].mode == XVID_ZONE_WEIGHT) {
2737                            ZONES[i].mode = XVID_ZONE_QUANT;
2738                            ZONES[i].modifier = (int) ((100*ARG_CQ) / ZONES[i].modifier);
2739                    }
2740    }
2741    
2742    /* full first pass zones */
2743    static void
2744    prepare_full1pass_zones() {
2745    
2746            int i = 0;
2747            if (NUM_ZONES == 0 || ZONES[0].frame != 0) {
2748                    /* first zone does not start at frame 0 or doesn't exist */
2749    
2750                    if (NUM_ZONES >= MAX_ZONES) NUM_ZONES--; /* we sacrifice last zone */
2751    
2752                    ZONES[NUM_ZONES].frame = 0;
2753                    ZONES[NUM_ZONES].mode = XVID_ZONE_QUANT;
2754                    ZONES[NUM_ZONES].modifier = 200;
2755                    ZONES[NUM_ZONES].type = XVID_TYPE_AUTO;
2756                    ZONES[NUM_ZONES].greyscale = 0;
2757                    ZONES[NUM_ZONES].chroma_opt = 0;
2758                    ZONES[NUM_ZONES].bvop_threshold = 0;
2759                    ZONES[NUM_ZONES].cartoon_mode = 0;
2760                    NUM_ZONES++;
2761    
2762                    sort_zones(ZONES, NUM_ZONES, &i);
2763            }
2764    
2765            /* step 2: let's change all weight zones into quant zones */
2766    
2767            for(i = 0; i < NUM_ZONES; i++)
2768                    if (ZONES[i].mode == XVID_ZONE_WEIGHT) {
2769                            ZONES[i].mode = XVID_ZONE_QUANT;
2770                            ZONES[i].modifier = 200;
2771                    }
2772    }
2773    
2774    static void apply_zone_modifiers(xvid_enc_frame_t * frame, int framenum)
2775    {
2776            int i;
2777    
2778            for (i=0; i<NUM_ZONES && ZONES[i].frame <= framenum; i++) ;
2779    
2780            if (--i < 0) return; /* there are no zones, or we're before the first zone */
2781    
2782            if (framenum == ZONES[i].frame)
2783                    frame->type = ZONES[i].type;
2784    
2785            if (ZONES[i].greyscale) {
2786                    frame->vop_flags |= XVID_VOP_GREYSCALE;
2787            }
2788    
2789            if (ZONES[i].chroma_opt) {
2790                    frame->vop_flags |= XVID_VOP_CHROMAOPT;
2791            }
2792    
2793            if (ZONES[i].cartoon_mode) {
2794                    frame->vop_flags |= XVID_VOP_CARTOON;
2795                    frame->motion |= XVID_ME_DETECT_STATIC_MOTION;
2796            }
2797    
2798            if (ARG_MAXBFRAMES) {
2799                    frame->bframe_threshold = ZONES[i].bvop_threshold;
2800            }
2801    }
2802    
2803    void removedivxp(char *buf, int bufsize) {
2804            int i;
2805            char* userdata;
2806    
2807            for (i=0; i <= (int)(bufsize-sizeof(userdata_start_code)); i++) {
2808                    if (memcmp((void*)userdata_start_code, (void*)(buf+i), strlen(userdata_start_code))==0) {
2809                            if ((userdata = strstr(buf+i+4, "DivX"))!=NULL) {
2810                                    userdata[strlen(userdata)-1] = '\0';
2811                                    return;
2812                            }
2813                    }
2814            }
2815    }

Legend:
Removed from v.1398  
changed lines
  Added in v.2129

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