[svn] / branches / dev-api-4 / xvidcore / examples / xvid_encraw.c Repository:
ViewVC logotype

Diff of /branches/dev-api-4/xvidcore/examples/xvid_encraw.c

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

revision 944, Tue Mar 25 22:53:57 2003 UTC revision 1073, Wed Jun 25 23:23:21 2003 UTC
# Line 3  Line 3 
3   *  XVID MPEG-4 VIDEO CODEC   *  XVID MPEG-4 VIDEO CODEC
4   *  - Console based test application  -   *  - Console based test application  -
5   *   *
6   *  Copyright(C) 2002-2003 Christoph Lampert   *  Copyright(C) 2002-2003 Christoph Lampert <gruel@web.de>
7     *               2002-2003 Edouard Gomez <ed.gomez@free.fr>
8     *               2003      Peter Ross <pross@xvid.org>
9   *   *
10   *  This program is free software; you can redistribute it and/or modify   *  This program is free software; you can redistribute it and/or modify
11   *  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 19  Line 21 
21   *  along with this program; if not, write to the Free Software   *  along with this program; if not, write to the Free Software
22   *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA   *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
23   *   *
24   * $Id: xvid_encraw.c,v 1.11.2.13 2003-03-25 22:53:57 edgomez Exp $   * $Id: xvid_encraw.c,v 1.11.2.28 2003-06-25 23:23:21 edgomez Exp $
25   *   *
26   ****************************************************************************/   ****************************************************************************/
27    
# Line 32  Line 34 
34   *  The program is plain C and needs no libraries except for libxvidcore,   *  The program is plain C and needs no libraries except for libxvidcore,
35   *  and maths-lib.   *  and maths-lib.
36   *   *
37     *  Use ./xvid_encraw -help for a list of options
38     *
39   ************************************************************************/   ************************************************************************/
40    
41  #include <stdio.h>  #include <stdio.h>
# Line 46  Line 50 
50    
51  #include "xvid.h"  #include "xvid.h"
52    
53    #undef READ_PNM
54    
55  /*****************************************************************************  /*****************************************************************************
56   *                            Quality presets   *                            Quality presets
57   ****************************************************************************/   ****************************************************************************/
   
58  static xvid_motion_t const motion_presets[] = {  static xvid_motion_t const motion_presets[] = {
59            /* quality 0 */
60          0,          0,
61          PMV_HALFPELREFINE16,  
62          PMV_HALFPELREFINE16,          /* quality 1 */
63          PMV_HALFPELREFINE16 | PMV_HALFPELREFINE8,          XVID_ME_ADVANCEDDIAMOND16,
64          PMV_HALFPELREFINE16 | PMV_HALFPELREFINE8 | PMV_EXTSEARCH16 |  
65                  PMV_USESQUARES16,          /* quality 2 */
66          PMV_HALFPELREFINE16 | PMV_HALFPELREFINE8 | PMV_EXTSEARCH16 |          XVID_ME_ADVANCEDDIAMOND16 | XVID_ME_HALFPELREFINE16,
67                  PMV_USESQUARES16 | PMV_CHROMA16 | PMV_CHROMA8,  
68            /* quality 3 */
69            XVID_ME_ADVANCEDDIAMOND16 | XVID_ME_HALFPELREFINE16 |
70            XVID_ME_ADVANCEDDIAMOND8 | XVID_ME_HALFPELREFINE8,
71    
72            /* quality 4 */
73            XVID_ME_ADVANCEDDIAMOND16 | XVID_ME_HALFPELREFINE16 |
74            XVID_ME_ADVANCEDDIAMOND8 | XVID_ME_HALFPELREFINE8 |
75            XVID_ME_CHROMA16 | XVID_ME_CHROMA8,
76    
77            /* quality 5 */
78            XVID_ME_ADVANCEDDIAMOND16 | XVID_ME_HALFPELREFINE16 |
79            XVID_ME_ADVANCEDDIAMOND8 | XVID_ME_HALFPELREFINE8 |
80            XVID_ME_CHROMA16 | XVID_ME_CHROMA8,
81    
82            /* quality 6 */
83            XVID_ME_ADVANCEDDIAMOND16 | XVID_ME_HALFPELREFINE16 | XVID_ME_EXTSEARCH16 |
84            XVID_ME_ADVANCEDDIAMOND8 | XVID_ME_HALFPELREFINE8 | XVID_ME_EXTSEARCH8 |
85            XVID_ME_CHROMA16 | XVID_ME_CHROMA8 ,
86    
87  };  };
88    #define ME_ELEMENTS (sizeof(motion_presets)/sizeof(motion_presets[0]))
89    
90  static xvid_vol_t const vol_presets[] = {  static xvid_vop_t const vop_presets[] = {
91          XVID_MPEGQUANT,          /* quality 0 */
92          0,          0,
93    
94            /* quality 1 */
95          0,          0,
         XVID_QUARTERPEL,  
         XVID_QUARTERPEL | XVID_GMC,  
         0  
 };  
96    
97  static xvid_vop_t const vop_presets[] = {          /* quality 2 */
98          XVID_DYNAMIC_BFRAMES,          XVID_VOP_HALFPEL,
99          XVID_DYNAMIC_BFRAMES,  
100          XVID_DYNAMIC_BFRAMES | XVID_HALFPEL,          /* quality 3 */
101          XVID_DYNAMIC_BFRAMES | XVID_HALFPEL | XVID_INTER4V,          XVID_VOP_HALFPEL | XVID_VOP_INTER4V,
102          XVID_DYNAMIC_BFRAMES | XVID_HALFPEL | XVID_INTER4V | XVID_HQACPRED,  
103          XVID_DYNAMIC_BFRAMES | XVID_HALFPEL | XVID_HQACPRED |          /* quality 4 */
104                  XVID_MODEDECISION_BITS          XVID_VOP_HALFPEL | XVID_VOP_INTER4V,
105    
106            /* quality 5 */
107            XVID_VOP_HALFPEL | XVID_VOP_INTER4V |
108            XVID_VOP_TRELLISQUANT,
109    
110            /* quality 6 */
111            XVID_VOP_HALFPEL | XVID_VOP_INTER4V |
112            XVID_VOP_TRELLISQUANT | XVID_VOP_HQACPRED,
113    
114  };  };
115    #define VOP_ELEMENTS (sizeof(vop_presets)/sizeof(vop_presets[0]))
116    
117  /*****************************************************************************  /*****************************************************************************
118   *                     Command line global variables   *                     Command line global variables
119   ****************************************************************************/   ****************************************************************************/
120    
121    #define MAX_ZONES   64
122    
123    static xvid_enc_zone_t ZONES[MAX_ZONES];
124    static int NUM_ZONES = 0;
125    
126  /* Maximum number of frames to encode */  /* Maximum number of frames to encode */
127  #define ABS_MAXFRAMENR 9999  #define ABS_MAXFRAMENR 9999
128    
# Line 92  Line 130 
130  static int ARG_DUMP = 0;  static int ARG_DUMP = 0;
131  static int ARG_LUMIMASKING = 0;  static int ARG_LUMIMASKING = 0;
132  static int ARG_BITRATE = 0;  static int ARG_BITRATE = 0;
133    static int ARG_SINGLE = 0;
134  static char *ARG_PASS1 = 0;  static char *ARG_PASS1 = 0;
135  static char *ARG_PASS2 = 0;  static char *ARG_PASS2 = 0;
136  static float ARG_QUANTI = 0.0f;  static int ARG_QUALITY = ME_ELEMENTS - 1;
 static int ARG_QUALITY = 5;  
137  static float ARG_FRAMERATE = 25.00f;  static float ARG_FRAMERATE = 25.00f;
138  static int ARG_MAXFRAMENR = ABS_MAXFRAMENR;  static int ARG_MAXFRAMENR = ABS_MAXFRAMENR;
139    static int ARG_MAXKEYINTERVAL = 0;
140  static char *ARG_INPUTFILE = NULL;  static char *ARG_INPUTFILE = NULL;
141  static int ARG_INPUTTYPE = 0;  static int ARG_INPUTTYPE = 0;
142  static int ARG_SAVEMPEGSTREAM = 0;  static int ARG_SAVEMPEGSTREAM = 0;
# Line 109  Line 148 
148  static int ARG_MAXBFRAMES = 0;  static int ARG_MAXBFRAMES = 0;
149  static int ARG_PACKED = 0;  static int ARG_PACKED = 0;
150  static int ARG_DEBUG = 0;  static int ARG_DEBUG = 0;
151    static int ARG_VOPDEBUG = 0;
152    
153    #ifndef READ_PNM
154  #define IMAGE_SIZE(x,y) ((x)*(y)*3/2)  #define IMAGE_SIZE(x,y) ((x)*(y)*3/2)
155    #else
156    #define IMAGE_SIZE(x,y) ((x)*(y)*3)
157    #endif
158    
159  #define MAX(A,B) ( ((A)>(B)) ? (A) : (B) )  #define MAX(A,B) ( ((A)>(B)) ? (A) : (B) )
160  #define SMALL_EPS (1e-10)  #define SMALL_EPS (1e-10)
# Line 141  Line 185 
185  static double msecond();  static double msecond();
186    
187  /* PGM related functions */  /* PGM related functions */
188    #ifndef READ_PNM
189  static int read_pgmheader(FILE * handle);  static int read_pgmheader(FILE * handle);
190  static int read_pgmdata(FILE * handle,  static int read_pgmdata(FILE * handle,
191                                                  unsigned char *image);                                                  unsigned char *image);
192    #else
193    static int read_pnmheader(FILE * handle);
194    static int read_pnmdata(FILE * handle,
195                                                    unsigned char *image);
196    #endif
197  static int read_yuvdata(FILE * handle,  static int read_yuvdata(FILE * handle,
198                                                  unsigned char *image);                                                  unsigned char *image);
199    
# Line 173  Line 223 
223    
224          double enctime;          double enctime;
225          double totalenctime = 0.;          double totalenctime = 0.;
226            float totalPSNR[3] = {0., 0., 0.};
227    
228          int totalsize;          int totalsize;
229          int result;          int result;
# Line 194  Line 245 
245          printf("xvid_encraw - raw mpeg4 bitstream encoder ");          printf("xvid_encraw - raw mpeg4 bitstream encoder ");
246          printf("written by Christoph Lampert 2002-2003\n\n");          printf("written by Christoph Lampert 2002-2003\n\n");
247    
248            /* Is there a dumb XviD coder ? */
249            if(ME_ELEMENTS != VOP_ELEMENTS) {
250                    fprintf(stderr, "Presets' arrays should have the same number of elements -- Please fill a bug to xvid-devel@xvid.org\n");
251                    return(-1);
252            }
253    
254  /*****************************************************************************  /*****************************************************************************
255   *                            Command line parsing   *                            Command line parsing
256   ****************************************************************************/   ****************************************************************************/
# Line 211  Line 268 
268                  } else if (strcmp("-bitrate", argv[i]) == 0 && i < argc - 1) {                  } else if (strcmp("-bitrate", argv[i]) == 0 && i < argc - 1) {
269                          i++;                          i++;
270                          ARG_BITRATE = atoi(argv[i]);                          ARG_BITRATE = atoi(argv[i]);
271                    } else if (strcmp("-single", argv[i]) == 0) {
272                            ARG_SINGLE = 1;
273                  } else if (strcmp("-pass1", argv[i]) == 0 && i < argc - 1) {                  } else if (strcmp("-pass1", argv[i]) == 0 && i < argc - 1) {
274                          i++;                          i++;
275                          ARG_PASS1 = argv[i];                          ARG_PASS1 = argv[i];
276                  } else if (strcmp("-pass2", argv[i]) == 0 && i < argc - 2) {                  } else if (strcmp("-pass2", argv[i]) == 0 && i < argc - 1) {
                         i++;  
                         ARG_PASS1 = argv[i];  
277                          i++;                          i++;
278                          ARG_PASS2 = argv[i];                          ARG_PASS2 = argv[i];
279                  } else if (strcmp("-max_bframes", argv[i]) == 0 && i < argc - 1) {                  } else if (strcmp("-max_bframes", argv[i]) == 0 && i < argc - 1) {
# Line 230  Line 287 
287                  } else if (strcmp("-bquant_offset", argv[i]) == 0 && i < argc - 1) {                  } else if (strcmp("-bquant_offset", argv[i]) == 0 && i < argc - 1) {
288                          i++;                          i++;
289                          ARG_BQOFFSET = atoi(argv[i]);                          ARG_BQOFFSET = atoi(argv[i]);
290    
291            } else if ((strcmp("-zq", argv[i]) == 0 || strcmp("-zw", argv[i]) == 0) && i < argc - 2) {
292    
293                if (NUM_ZONES >= MAX_ZONES) {
294                    fprintf(stderr,"warning: too many zones; zone ignored\n");
295                    continue;
296                }
297                ZONES[NUM_ZONES].mode = strcmp("-zq", argv[i])==0 ? XVID_ZONE_QUANT : XVID_ZONE_WEIGHT;
298                            i++;
299                ZONES[NUM_ZONES].frame = atoi(argv[i]);
300                i++;
301                ZONES[NUM_ZONES].increment  = (int)(atof(argv[i]) * 100);
302                ZONES[NUM_ZONES].base  = 100;
303                NUM_ZONES++;
304    
305                  } else if (strcmp("-quality", argv[i]) == 0 && i < argc - 1) {                  } else if (strcmp("-quality", argv[i]) == 0 && i < argc - 1) {
306                          i++;                          i++;
307                          ARG_QUALITY = atoi(argv[i]);                          ARG_QUALITY = atoi(argv[i]);
308                  } else if (strcmp("-framerate", argv[i]) == 0 && i < argc - 1) {                  } else if (strcmp("-framerate", argv[i]) == 0 && i < argc - 1) {
309                          i++;                          i++;
310                          ARG_FRAMERATE = (float) atof(argv[i]);                          ARG_FRAMERATE = (float) atof(argv[i]);
311                    } else if (strcmp("-max_key_interval", argv[i]) == 0 && i < argc - 1) {
312                            i++;
313                            ARG_MAXKEYINTERVAL = atoi(argv[i]);
314                  } else if (strcmp("-i", argv[i]) == 0 && i < argc - 1) {                  } else if (strcmp("-i", argv[i]) == 0 && i < argc - 1) {
315                          i++;                          i++;
316                          ARG_INPUTFILE = argv[i];                          ARG_INPUTFILE = argv[i];
# Line 251  Line 326 
326                  } else if (strcmp("-nframes", argv[i]) == 0 && i < argc - 1) {                  } else if (strcmp("-nframes", argv[i]) == 0 && i < argc - 1) {
327                          i++;                          i++;
328                          ARG_MAXFRAMENR = atoi(argv[i]);                          ARG_MAXFRAMENR = atoi(argv[i]);
                 } else if (strcmp("-quant", argv[i]) == 0 && i < argc - 1) {  
                         i++;  
                         ARG_QUANTI = (float) atof(argv[i]);  
329                  } else if (strcmp("-save", argv[i]) == 0) {                  } else if (strcmp("-save", argv[i]) == 0) {
330                          ARG_SAVEMPEGSTREAM = 1;                          ARG_SAVEMPEGSTREAM = 1;
331                  } else if (strcmp("-debug", argv[i]) == 0) {                  } else if (strcmp("-debug", argv[i]) == 0) {
332                          ARG_DEBUG = 1;                          i++;
333                if (sscanf(argv[i],"0x%x", &ARG_DEBUG) || sscanf(argv[i],"%d", &ARG_DEBUG)) ;
334                  } else if (strcmp("-o", argv[i]) == 0 && i < argc - 1) {                  } else if (strcmp("-o", argv[i]) == 0 && i < argc - 1) {
335                          i++;                          i++;
336                          ARG_OUTPUTFILE = argv[i];                          ARG_OUTPUTFILE = argv[i];
337                    } else if (strcmp("-vop_debug", argv[i]) == 0) {
338                            ARG_VOPDEBUG = 1;
339                  } else if (strcmp("-help", argv[i])) {                  } else if (strcmp("-help", argv[i])) {
340                          usage();                          usage();
341                          return (0);                          return (0);
# Line 281  Line 356 
356                  ARG_INPUTTYPE = 1;              /* pgm */                  ARG_INPUTTYPE = 1;              /* pgm */
357          }          }
358    
359          if (ARG_QUALITY < 0 || ARG_QUALITY > 5) {          if (ARG_QUALITY < 0 ) {
360                  fprintf(stderr, "Wrong Quality\n");                  ARG_QUALITY = 0;
361                  return (-1);          } else if (ARG_QUALITY >= ME_ELEMENTS) {
362                    ARG_QUALITY = ME_ELEMENTS - 1;
363          }          }
364    
365          if (ARG_FRAMERATE <= 0) {          if (ARG_FRAMERATE <= 0) {
# Line 308  Line 384 
384          }          }
385    
386          if (ARG_INPUTTYPE) {          if (ARG_INPUTTYPE) {
387    #ifndef READ_PNM
388                  if (read_pgmheader(in_file)) {                  if (read_pgmheader(in_file)) {
389    #else
390                    if (read_pnmheader(in_file)) {
391    #endif
392                          fprintf(stderr,                          fprintf(stderr,
393                                          "Wrong input format, I want YUV encapsulated in PGM\n");                                          "Wrong input format, I want YUV encapsulated in PGM\n");
394                          return (-1);                          return (-1);
# Line 374  Line 454 
454                  if (!result) {                  if (!result) {
455                          if (ARG_INPUTTYPE) {                          if (ARG_INPUTTYPE) {
456                                  /* read PGM data (YUV-format) */                                  /* read PGM data (YUV-format) */
457    #ifndef READ_PNM
458                                  result = read_pgmdata(in_file, in_buffer);                                  result = read_pgmdata(in_file, in_buffer);
459    #else
460                                    result = read_pnmdata(in_file, in_buffer);
461    #endif
462                          } else {                          } else {
463                                  /* read raw data (YUV-format) */                                  /* read raw data (YUV-format) */
464                                  result = read_yuvdata(in_file, in_buffer);                                  result = read_yuvdata(in_file, in_buffer);
# Line 393  Line 477 
477    
478                  /* Write the Frame statistics */                  /* Write the Frame statistics */
479    
480                  printf("%5d: key=%i, time=%6.0f, length=%7d", !result ? input_num : -1,                  printf("%5d: key=%i, time= %6.0f, len= %7d", !result ? input_num : -1,
481                             key, (float) enctime, (int) m4v_size);                             key, (float) enctime, (int) m4v_size);
482    
483                  if (stats_type > 0) {   /* !XVID_TYPE_NOTHING */                  if (stats_type > 0) {   /* !XVID_TYPE_NOTHING */
# Line 416  Line 500 
500                                  break;                                  break;
501                          }                          }
502    
503                          printf(" | type=%s, quant=%2d, length=%7d", type, stats_quant,                          printf(" | type=%s, quant= %2d, len= %7d", type, stats_quant,
504                                     stats_length);                                     stats_length);
505    
506  #define SSE2PSNR(sse, width, height) ((!(sse))?0.0f : 48.131f - 10*(float)log10((float)(sse)/((float)((width)*(height)))))  #define SSE2PSNR(sse, width, height) ((!(sse))?0.0f : 48.131f - 10*(float)log10((float)(sse)/((float)((width)*(height)))))
# Line 426  Line 510 
510                                             SSE2PSNR(sse[0], XDIM, YDIM), SSE2PSNR(sse[1], XDIM / 2,                                             SSE2PSNR(sse[0], XDIM, YDIM), SSE2PSNR(sse[1], XDIM / 2,
511                                                                                                                            YDIM / 2),                                                                                                                            YDIM / 2),
512                                             SSE2PSNR(sse[2], XDIM / 2, YDIM / 2));                                             SSE2PSNR(sse[2], XDIM / 2, YDIM / 2));
513    
514                                    totalPSNR[0] += SSE2PSNR(sse[0], XDIM, YDIM);
515                                    totalPSNR[1] += SSE2PSNR(sse[1], XDIM/2, YDIM/2);
516                                    totalPSNR[2] += SSE2PSNR(sse[2], XDIM/2, YDIM/2);
517                          }                          }
518    
519                  }                  }
# Line 466  Line 554 
554    
555                  /* Read the header if it's pgm stream */                  /* Read the header if it's pgm stream */
556                  if (!result && ARG_INPUTTYPE)                  if (!result && ARG_INPUTTYPE)
557    #ifndef READ_PNM
558                          result = read_pgmheader(in_file);                          result = read_pgmheader(in_file);
559    #else
560                            result = read_pnmheader(in_file);
561    #endif
562          } while (1);          } while (1);
563    
564    
# Line 482  Line 573 
573          if (input_num > 0) {          if (input_num > 0) {
574                  totalsize /= input_num;                  totalsize /= input_num;
575                  totalenctime /= input_num;                  totalenctime /= input_num;
576                    totalPSNR[0] /= input_num;
577                    totalPSNR[1] /= input_num;
578                    totalPSNR[2] /= input_num;
579          } else {          } else {
580                  totalsize = -1;                  totalsize = -1;
581                  totalenctime = -1;                  totalenctime = -1;
582          }          }
583    
584          printf("Avg: enctime(ms) =%7.2f, fps =%7.2f, length(bytes) = %7d\n",          printf("Avg: enctime(ms) =%7.2f, fps =%7.2f, length(bytes) = %7d",
585                     totalenctime, 1000 / totalenctime, (int) totalsize);                     totalenctime, 1000 / totalenctime, (int) totalsize);
586       if (ARG_STATS) {
587           printf(", psnr y = %2.2f, psnr u = %2.2f, psnr v = %2.2f",
588                      totalPSNR[0],totalPSNR[1],totalPSNR[2]);
589            }
590            printf("\n");
591    
592    
593  /*****************************************************************************  /*****************************************************************************
# Line 571  Line 670 
670          fprintf(stderr, "\n");          fprintf(stderr, "\n");
671          fprintf(stderr, "Rate control options:\n");          fprintf(stderr, "Rate control options:\n");
672          fprintf(stderr, " -framerate float               : target framerate (>0 | default=25.0)\n");          fprintf(stderr, " -framerate float               : target framerate (>0 | default=25.0)\n");
673          fprintf(stderr, " -bitrate   integer             : bitrate -- for CBR/VBR pass2\n");          fprintf(stderr, " -bitrate   integer             : target bitrate\n");
674          fprintf(stderr, " -quant     float               : quantizer -- for \"Fixed\" quantizer RC\n");      fprintf(stderr,     " -single                        : single pass mode\n");
675          fprintf(stderr, " -pass1     filename            : stats filename\n");          fprintf(stderr, " -pass1     filename            : twopass mode (first pass)\n");
676          fprintf(stderr, " -pass2     filename1 filename2 : first pass stats and scaled stats filename\n");          fprintf(stderr, " -pass2     filename            : twopass mode (2nd pass)\n");
677            fprintf(stderr, " -zq starting_frame float       : bitrate zone; quant\n");
678            fprintf(stderr, " -zw starting_frame float       : bitrate zone; weight\n");
679        fprintf(stderr, " -max_key_interval integer      : maximum keyframe interval\n");
680          fprintf(stderr, "\n");          fprintf(stderr, "\n");
681          fprintf(stderr, "Other options\n");          fprintf(stderr, "Other options\n");
682          fprintf(stderr, " -asm            : use assembly optmized code\n");          fprintf(stderr, " -asm            : use assembly optmized code\n");
683          fprintf(stderr, " -quality integer: quality ([0..5])\n");          fprintf(stderr, " -quality integer: quality ([0..%d])\n", ME_ELEMENTS - 1);
684          fprintf(stderr, " -packed         : packed mode\n");          fprintf(stderr, " -packed         : packed mode\n");
685          fprintf(stderr, " -lumimasking    : use lumimasking algorithm\n");          fprintf(stderr, " -lumimasking    : use lumimasking algorithm\n");
686          fprintf(stderr, " -stats          : print stats about encoded frames\n");          fprintf(stderr, " -stats          : print stats about encoded frames\n");
687          fprintf(stderr, " -debug          : print all MB dquants\n");          fprintf(stderr, " -debug          : activates xvidcore internal debugging output\n");
688            fprintf(stderr, " -vop_debug      : print some info directly into encoded frames\n");
689          fprintf(stderr, " -help           : prints this help message\n");          fprintf(stderr, " -help           : prints this help message\n");
690            fprintf(stderr, "\n");
691            fprintf(stderr, "NB: You can define %d zones repeating the -z[qw] option as many times as needed.\n", MAX_ZONES);
692            fprintf(stderr, "\n");
693  }  }
694    
695  /*****************************************************************************  /*****************************************************************************
# Line 594  Line 700 
700   *   *
701   *****************************************************************************/   *****************************************************************************/
702    
703    #ifndef READ_PNM
704  static int  static int
705  read_pgmheader(FILE * handle)  read_pgmheader(FILE * handle)
706  {  {
# Line 649  Line 756 
756    
757          return (0);          return (0);
758  }  }
759    #else
760    static int
761    read_pnmheader(FILE * handle)
762    {
763            int bytes, xsize, ysize, depth;
764            char dummy[2];
765    
766            bytes = fread(dummy, 1, 2, handle);
767    
768            if ((bytes < 2) || (dummy[0] != 'P') || (dummy[1] != '6'))
769                    return (1);
770    
771            fscanf(handle, "%d %d %d", &xsize, &ysize, &depth);
772            if ((xsize > 1440) || (ysize > 2880) || (depth != 255)) {
773                    fprintf(stderr, "%d %d %d\n", xsize, ysize, depth);
774                    return (2);
775            }
776    
777            XDIM = xsize;
778            YDIM = ysize;
779    
780            return (0);
781    }
782    
783    static int
784    read_pnmdata(FILE * handle,
785                             unsigned char *image)
786    {
787            int i;
788            char dummy;
789    
790            /* read Y component of picture */
791            fread(image, 1, XDIM * YDIM * 3, handle);
792    
793            /*  I don't know why, but this seems needed */
794            fread(&dummy, 1, 1, handle);
795    
796            return (0);
797    }
798    #endif
799    
800  static int  static int
801  read_yuvdata(FILE * handle,  read_yuvdata(FILE * handle,
# Line 717  Line 864 
864  enc_init(int use_assembler)  enc_init(int use_assembler)
865  {  {
866          int xerr;          int xerr;
867          xvid_plugin_cbr_t cbr;          //xvid_plugin_cbr_t cbr;
868        xvid_plugin_single_t single;
869          xvid_plugin_2pass1_t rc2pass1;          xvid_plugin_2pass1_t rc2pass1;
870          xvid_plugin_2pass2_t rc2pass2;          xvid_plugin_2pass2_t rc2pass2;
871          xvid_plugin_fixed_t rcfixed;          //xvid_plugin_fixed_t rcfixed;
872          xvid_enc_plugin_t plugins[7];          xvid_enc_plugin_t plugins[7];
873          xvid_gbl_init_t xvid_gbl_init;          xvid_gbl_init_t xvid_gbl_init;
874          xvid_enc_create_t xvid_enc_create;          xvid_enc_create_t xvid_enc_create;
# Line 732  Line 880 
880          /* Set version -- version checking will done by xvidcore */          /* Set version -- version checking will done by xvidcore */
881          memset(&xvid_gbl_init, 0, sizeof(xvid_gbl_init));          memset(&xvid_gbl_init, 0, sizeof(xvid_gbl_init));
882          xvid_gbl_init.version = XVID_VERSION;          xvid_gbl_init.version = XVID_VERSION;
883        xvid_gbl_init.debug = ARG_DEBUG;
884    
885    
886          /* Do we have to enable ASM optimizations ? */          /* Do we have to enable ASM optimizations ? */
# Line 762  Line 911 
911          xvid_enc_create.height = YDIM;          xvid_enc_create.height = YDIM;
912    
913          /* init plugins  */          /* init plugins  */
914        xvid_enc_create.zones = ZONES;
915        xvid_enc_create.num_zones = NUM_ZONES;
916    
917          xvid_enc_create.plugins = plugins;          xvid_enc_create.plugins = plugins;
918          xvid_enc_create.num_plugins = 0;          xvid_enc_create.num_plugins = 0;
919    
920          if (ARG_BITRATE) {          if (ARG_SINGLE) {
921                  cbr.version = XVID_VERSION;                  memset(&single, 0, sizeof(xvid_plugin_single_t));
922                  memset(&cbr, 0, sizeof(xvid_plugin_cbr_t));                  single.version = XVID_VERSION;
923                  cbr.bitrate = ARG_BITRATE;                  single.bitrate = ARG_BITRATE;
   
                 plugins[xvid_enc_create.num_plugins].func = xvid_plugin_cbr;  
                 plugins[xvid_enc_create.num_plugins].param = &cbr;  
                 xvid_enc_create.num_plugins++;  
         }  
   
         if (ARG_QUANTI) {  
                 rcfixed.version = XVID_VERSION;  
                 /* We will use a 1/10 precision, just to make sure it works */  
                 rcfixed.quant_base = 10;  
                 rcfixed.quant_increment = (int) (ARG_QUANTI * 10);  
924    
925                  plugins[xvid_enc_create.num_plugins].func = xvid_plugin_fixed;                  plugins[xvid_enc_create.num_plugins].func = xvid_plugin_single;
926                  plugins[xvid_enc_create.num_plugins].param = &rcfixed;                  plugins[xvid_enc_create.num_plugins].param = &single;
927                  xvid_enc_create.num_plugins++;                  xvid_enc_create.num_plugins++;
928          }          }
929    
930          if (ARG_PASS1 && ARG_PASS2) {          if (ARG_PASS2) {
                 rc2pass2.version = XVID_VERSION;  
931                  memset(&rc2pass2, 0, sizeof(xvid_plugin_2pass2_t));                  memset(&rc2pass2, 0, sizeof(xvid_plugin_2pass2_t));
932                  rc2pass2.filename1 = ARG_PASS1;                  rc2pass2.version = XVID_VERSION;
933                  rc2pass2.filename2 = ARG_PASS2;                  rc2pass2.filename = ARG_PASS2;
934                    rc2pass2.bitrate = ARG_BITRATE;
935    
936                  plugins[xvid_enc_create.num_plugins].func = xvid_plugin_2pass2;                  plugins[xvid_enc_create.num_plugins].func = xvid_plugin_2pass2;
937                  plugins[xvid_enc_create.num_plugins].param = &rc2pass2;                  plugins[xvid_enc_create.num_plugins].param = &rc2pass2;
938                  xvid_enc_create.num_plugins++;                  xvid_enc_create.num_plugins++;
939          } else if (ARG_PASS1) {          }
940                  rc2pass1.version = XVID_VERSION;  
941            if (ARG_PASS1) {
942                  memset(&rc2pass1, 0, sizeof(xvid_plugin_2pass1_t));                  memset(&rc2pass1, 0, sizeof(xvid_plugin_2pass1_t));
943                    rc2pass1.version = XVID_VERSION;
944                  rc2pass1.filename = ARG_PASS1;                  rc2pass1.filename = ARG_PASS1;
945    
946                  plugins[xvid_enc_create.num_plugins].func = xvid_plugin_2pass1;                  plugins[xvid_enc_create.num_plugins].func = xvid_plugin_2pass1;
# Line 818  Line 960 
960                  xvid_enc_create.num_plugins++;                  xvid_enc_create.num_plugins++;
961          }          }
962    
963    #if 0
964          if (ARG_DEBUG) {          if (ARG_DEBUG) {
965                  plugins[xvid_enc_create.num_plugins].func = rawenc_debug;                  plugins[xvid_enc_create.num_plugins].func = rawenc_debug;
966                  plugins[xvid_enc_create.num_plugins].param = NULL;                  plugins[xvid_enc_create.num_plugins].param = NULL;
967                  xvid_enc_create.num_plugins++;                  xvid_enc_create.num_plugins++;
968          }          }
969    #endif
970    
971          /* No fancy thread tests */          /* No fancy thread tests */
972          xvid_enc_create.num_threads = 0;          xvid_enc_create.num_threads = 0;
# Line 837  Line 981 
981          }          }
982    
983          /* Maximum key frame interval */          /* Maximum key frame interval */
984        if (ARG_MAXKEYINTERVAL > 0) {
985            xvid_enc_create.max_key_interval = ARG_MAXKEYINTERVAL;
986        }else {
987          xvid_enc_create.max_key_interval = (int) ARG_FRAMERATE *10;          xvid_enc_create.max_key_interval = (int) ARG_FRAMERATE *10;
988        }
989    
990          /* Bframes settings */          /* Bframes settings */
991          xvid_enc_create.max_bframes = ARG_MAXBFRAMES;          xvid_enc_create.max_bframes = ARG_MAXBFRAMES;
# Line 851  Line 999 
999          xvid_enc_create.global = 0;          xvid_enc_create.global = 0;
1000    
1001          if (ARG_PACKED)          if (ARG_PACKED)
1002                  xvid_enc_create.global |=XVID_PACKED;                  xvid_enc_create.global |=XVID_GLOBAL_PACKED;
1003    
1004          if (ARG_STATS)          if (ARG_STATS)
1005                  xvid_enc_create.global |=XVID_EXTRASTATS_ENABLE;                  xvid_enc_create.global |=XVID_GLOBAL_EXTRASTATS_ENABLE;
1006    
1007          /* I use a small value here, since will not encode whole movies, but short clips */          /* I use a small value here, since will not encode whole movies, but short clips */
1008          xerr = xvid_encore(NULL, XVID_ENC_CREATE, &xvid_enc_create, NULL);          xerr = xvid_encore(NULL, XVID_ENC_CREATE, &xvid_enc_create, NULL);
# Line 904  Line 1052 
1052          /* Initialize input image fields */          /* Initialize input image fields */
1053          if (image) {          if (image) {
1054                  xvid_enc_frame.input.plane[0] = image;                  xvid_enc_frame.input.plane[0] = image;
1055    #ifndef READ_PNM
1056                  xvid_enc_frame.input.csp = XVID_CSP_I420;                  xvid_enc_frame.input.csp = XVID_CSP_I420;
1057                  xvid_enc_frame.input.stride[0] = XDIM;                  xvid_enc_frame.input.stride[0] = XDIM;
1058    #else
1059                    xvid_enc_frame.input.csp = XVID_CSP_BGR;
1060                    xvid_enc_frame.input.stride[0] = XDIM*3;
1061    #endif
1062          } else {          } else {
1063                  xvid_enc_frame.input.csp = XVID_CSP_NULL;                  xvid_enc_frame.input.csp = XVID_CSP_NULL;
1064          }          }
1065    
1066          /* Set up core's general features */          /* Set up core's general features */
1067          xvid_enc_frame.vol_flags = vol_presets[ARG_QUALITY];          xvid_enc_frame.vol_flags = 0;
1068          if (ARG_STATS)          if (ARG_STATS)
1069                  xvid_enc_frame.vol_flags |= XVID_EXTRASTATS;                  xvid_enc_frame.vol_flags |= XVID_VOL_EXTRASTATS;
1070    
1071          /* Set up core's general features */          /* Set up core's general features */
1072          xvid_enc_frame.vop_flags = vop_presets[ARG_QUALITY];          xvid_enc_frame.vop_flags = vop_presets[ARG_QUALITY];
1073        if (ARG_VOPDEBUG) {
1074            xvid_enc_frame.vop_flags |= XVID_VOP_DEBUG;
1075        }
1076    
1077          /* Frame type -- let core decide for us */          /* Frame type -- let core decide for us */
1078          xvid_enc_frame.type = XVID_TYPE_AUTO;          xvid_enc_frame.type = XVID_TYPE_AUTO;

Legend:
Removed from v.944  
changed lines
  Added in v.1073

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