[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 913, Sun Mar 9 16:45:24 2003 UTC revision 938, Sun Mar 23 04:03:01 2003 UTC
# Line 19  Line 19 
19   *  along with this program; if not, write to the Free Software   *  along with this program; if not, write to the Free Software
20   *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA   *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
21   *   *
22   * $Id: xvid_encraw.c,v 1.11.2.2 2003-03-09 16:45:24 edgomez Exp $   * $Id: xvid_encraw.c,v 1.11.2.11 2003-03-23 04:01:32 suxen_drol Exp $
23   *   *
24   ****************************************************************************/   ****************************************************************************/
25    
26  /*****************************************************************************  /*****************************************************************************
27   *  Application notes :   *  Application notes :
28   *   *
29   *  A sequence of YUV pics in PGM file format is encoded and decoded   *  A sequence of raw YUV I420 pics or YUV I420 PGM file format is encoded
30   *  The speed is measured and PSNR of decoded picture is calculated.   *  The speed is measured and frames' PSNR are taken from core.
31   *   *
32   *  The program is plain C and needs no libraries except for libxvidcore,   *  The program is plain C and needs no libraries except for libxvidcore,
33   *  and maths-lib.   *  and maths-lib.
# Line 46  Line 46 
46    
47  #include "xvid.h"  #include "xvid.h"
48    
49    
50  /*****************************************************************************  /*****************************************************************************
51   *                            Quality presets   *                            Quality presets
52   ****************************************************************************/   ****************************************************************************/
# Line 65  Line 66 
66          0,          0,
67          XVID_QUARTERPEL,          XVID_QUARTERPEL,
68          XVID_QUARTERPEL | XVID_GMC,          XVID_QUARTERPEL | XVID_GMC,
69          XVID_QUARTERPEL | XVID_GMC          0
70  };  };
71    
72  static xvid_vop_t const vop_presets[] = {  static xvid_vop_t const vop_presets[] = {
# Line 74  Line 75 
75          XVID_DYNAMIC_BFRAMES | XVID_HALFPEL,          XVID_DYNAMIC_BFRAMES | XVID_HALFPEL,
76          XVID_DYNAMIC_BFRAMES | XVID_HALFPEL | XVID_INTER4V,          XVID_DYNAMIC_BFRAMES | XVID_HALFPEL | XVID_INTER4V,
77          XVID_DYNAMIC_BFRAMES | XVID_HALFPEL | XVID_INTER4V | XVID_HQACPRED,          XVID_DYNAMIC_BFRAMES | XVID_HALFPEL | XVID_INTER4V | XVID_HQACPRED,
78          XVID_DYNAMIC_BFRAMES | XVID_HALFPEL | XVID_INTER4V | XVID_HQACPRED | XVID_MODEDECISION_BITS          XVID_DYNAMIC_BFRAMES | XVID_HALFPEL | XVID_HQACPRED | XVID_MODEDECISION_BITS
79  };  };
80    
81  /*****************************************************************************  /*****************************************************************************
# Line 84  Line 85 
85  /* Maximum number of frames to encode */  /* Maximum number of frames to encode */
86  #define ABS_MAXFRAMENR 9999  #define ABS_MAXFRAMENR 9999
87    
88  static int   ARG_BITRATE = 900;  static int ARG_STATS = 0;
89    static int ARG_DUMP = 0;
90    static int ARG_LUMIMASKING = 0;
91    static int ARG_BITRATE = 0;
92    static char * ARG_PASS1 = 0;
93  static int   ARG_QUANTI = 0;  static int   ARG_QUANTI = 0;
94  static int   ARG_QUALITY = 5;  static int   ARG_QUALITY = 5;
 static int   ARG_MINQUANT = 1;  
 static int   ARG_MAXQUANT = 31;  
95  static float ARG_FRAMERATE = 25.00f;  static float ARG_FRAMERATE = 25.00f;
96  static int   ARG_MAXFRAMENR = ABS_MAXFRAMENR;  static int   ARG_MAXFRAMENR = ABS_MAXFRAMENR;
97  static char *ARG_INPUTFILE = NULL;  static char *ARG_INPUTFILE = NULL;
# Line 100  Line 103 
103  static int   ARG_BQRATIO = 150;  static int   ARG_BQRATIO = 150;
104  static int   ARG_BQOFFSET = 100;  static int   ARG_BQOFFSET = 100;
105  static int   ARG_MAXBFRAMES = 0;  static int   ARG_MAXBFRAMES = 0;
106    static int ARG_PACKED = 0;
107    static int ARG_DEBUG = 0;
108    
109  #define IMAGE_SIZE(x,y) ((x)*(y)*3/2)  #define IMAGE_SIZE(x,y) ((x)*(y)*3/2)
110    
111  #define MAX(A,B) ( ((A)>(B)) ? (A) : (B) )  #define MAX(A,B) ( ((A)>(B)) ? (A) : (B) )
# Line 112  Line 118 
118   *                     Nasty global vars ;-)   *                     Nasty global vars ;-)
119   ***************************************************************************/   ***************************************************************************/
120    
121  static int i,filenr = 0;  static int i;
122    
123  /* the path where to save output */  /* the path where to save output */
124  static char filepath[256] = "./";  static char filepath[256] = "./";
# Line 132  Line 138 
138    
139  /* PGM related functions */  /* PGM related functions */
140  static int read_pgmheader(FILE* handle);  static int read_pgmheader(FILE* handle);
141  static int read_pgmdata(FILE* handle, unsigned char *image);  static int read_pgmdata(FILE * handle,
142  static int read_yuvdata(FILE* handle, unsigned char *image);                                                  unsigned char *image);
143    static int read_yuvdata(FILE * handle,
144                                                    unsigned char *image);
145    
146  /* Encoder related functions */  /* Encoder related functions */
147  static int enc_init(int use_assembler);  static int enc_init(int use_assembler);
148  static int enc_stop();  static int enc_stop();
149  static int enc_main(unsigned char* image,  static int enc_main(unsigned char* image,
150                                          unsigned char* bitstream,                                          unsigned char* bitstream,
151                                          long *frametype,                                          int *key,
152                                            int *stats_type,
153                                            int *stats_quant,
154                                            int *stats_length,
155                                          int stats[3]);                                          int stats[3]);
156    
157  /*****************************************************************************  /*****************************************************************************
158   *               Main function   *               Main function
159   ****************************************************************************/   ****************************************************************************/
160    
161  int main(int argc, char *argv[])  int
162    main(int argc,
163             char *argv[])
164  {  {
165    
166          unsigned char *mp4_buffer = NULL;          unsigned char *mp4_buffer = NULL;
# Line 157  Line 170 
170          double enctime;          double enctime;
171          double totalenctime=0.;          double totalenctime=0.;
172    
173          long totalsize;          int totalsize;
174          int status;          int result;
175          long frame_type;          int m4v_size;
176            int key;
177          long m4v_size;          int stats_type;
178            int stats_quant;
179            int stats_length;
180          int use_assembler=0;          int use_assembler=0;
181    
182            int input_num;
183            int output_num;
184    
185          char filename[256];          char filename[256];
186    
187          FILE *in_file = stdin;          FILE *in_file = stdin;
# Line 180  Line 198 
198    
199                  if (strcmp("-asm", argv[i]) == 0 ) {                  if (strcmp("-asm", argv[i]) == 0 ) {
200                          use_assembler = 1;                          use_assembler = 1;
201                  }                  } else if (strcmp("-w", argv[i]) == 0 && i < argc - 1) {
                 else if (strcmp("-w", argv[i]) == 0 && i < argc - 1 ) {  
202                          i++;                          i++;
203                          XDIM = atoi(argv[i]);                          XDIM = atoi(argv[i]);
204                  }                  } else if (strcmp("-h", argv[i]) == 0 && i < argc - 1) {
                 else if (strcmp("-h", argv[i]) == 0 && i < argc - 1 ) {  
205                          i++;                          i++;
206                          YDIM = atoi(argv[i]);                          YDIM = atoi(argv[i]);
207                  }                  } else if (strcmp("-bitrate", argv[i]) == 0 && i < argc - 1) {
                 else if (strcmp("-b", argv[i]) == 0 && i < argc - 1 ) {  
208                          i++;                          i++;
209                          ARG_BITRATE = atoi(argv[i]);                          ARG_BITRATE = atoi(argv[i]);
210                  }                  } else if (strcmp("-pass1", argv[i]) == 0 && i < argc - 1) {
211                  else if (strcmp("-bn", argv[i]) == 0 && i < argc - 1 ) {                          i++;
212                            ARG_PASS1 = argv[i];
213                    } else if (strcmp("-max_bframes", argv[i]) == 0 && i < argc - 1) {
214                          i++;                          i++;
215                          ARG_MAXBFRAMES = atoi(argv[i]);                          ARG_MAXBFRAMES = atoi(argv[i]);
216                  }                  } else if (strcmp("-packed", argv[i]) == 0) {
217                  else if (strcmp("-bqr", argv[i]) == 0 && i < argc - 1 ) {                          ARG_PACKED = 1;
218                    } else if (strcmp("-bquant_ratio", argv[i]) == 0 && i < argc - 1) {
219                          i++;                          i++;
220                          ARG_BQRATIO = atoi(argv[i]);                          ARG_BQRATIO = atoi(argv[i]);
221                  }                  } else if (strcmp("-bquant_offset", argv[i]) == 0 && i < argc - 1) {
                 else if (strcmp("-bqo", argv[i]) == 0 && i < argc - 1 ) {  
222                          i++;                          i++;
223                          ARG_BQOFFSET = atoi(argv[i]);                          ARG_BQOFFSET = atoi(argv[i]);
224                  }                  } else if (strcmp("-quality", argv[i]) == 0 && i < argc - 1) {
                 else if (strcmp("-q", argv[i]) == 0 && i < argc - 1 ) {  
225                          i++;                          i++;
226                          ARG_QUALITY = atoi(argv[i]);                          ARG_QUALITY = atoi(argv[i]);
227                  }                  } else if (strcmp("-framerate", argv[i]) == 0 && i < argc - 1) {
                 else if (strcmp("-f", argv[i]) == 0 && i < argc - 1 ) {  
228                          i++;                          i++;
229                          ARG_FRAMERATE = (float)atof(argv[i]);                          ARG_FRAMERATE = (float)atof(argv[i]);
230                  }                  } else if (strcmp("-i", argv[i]) == 0 && i < argc - 1) {
                 else if (strcmp("-i", argv[i]) == 0 && i < argc - 1 ) {  
231                          i++;                          i++;
232                          ARG_INPUTFILE = argv[i];                          ARG_INPUTFILE = argv[i];
233                  }                  } else if (strcmp("-stats", argv[i]) == 0) {
234                  else if (strcmp("-t", argv[i]) == 0 && i < argc - 1 ) {                          ARG_STATS = 1;
235                    } else if (strcmp("-dump", argv[i]) == 0) {
236                            ARG_DUMP = 1;
237                    } else if (strcmp("-lumimasking", argv[i]) == 0) {
238                            ARG_LUMIMASKING = 1;
239                    } else if (strcmp("-type", argv[i]) == 0 && i < argc - 1) {
240                          i++;                          i++;
241                          ARG_INPUTTYPE = atoi(argv[i]);                          ARG_INPUTTYPE = atoi(argv[i]);
242                  }                  } else if (strcmp("-nframes", argv[i]) == 0 && i < argc - 1) {
                 else if(strcmp("-n", argv[i]) == 0 && i < argc - 1 ) {  
243                          i++;                          i++;
244                          ARG_MAXFRAMENR  = atoi(argv[i]);                          ARG_MAXFRAMENR  = atoi(argv[i]);
245                  }                  } else if (strcmp("-quant", argv[i]) == 0 && i < argc - 1) {
                 else if (strcmp("-quant", argv[i]) == 0 && i < argc - 1 ) {  
246                          i++;                          i++;
247                          ARG_QUANTI = atoi(argv[i]);                          ARG_QUANTI = atoi(argv[i]);
248                  }                  } else if (strcmp("-save", argv[i]) == 0) {
249                  else if (strcmp("-m", argv[i]) == 0 && i < argc - 1 ) {                          ARG_SAVEMPEGSTREAM = 1;
250                          i++;                  } else if (strcmp("-debug", argv[i]) == 0) {
251                          ARG_SAVEMPEGSTREAM = atoi(argv[i]);                          ARG_DEBUG = 1;
252                  }                  } else if (strcmp("-o", argv[i]) == 0 && i < argc - 1) {
                 else if (strcmp("-o", argv[i]) == 0 && i < argc - 1 ) {  
253                          i++;                          i++;
254                          ARG_OUTPUTFILE = argv[i];                          ARG_OUTPUTFILE = argv[i];
255                  }                  } else if (strcmp("-help", argv[i])) {
                 else if (strcmp("-help", argv[i])) {  
256                          usage();                          usage();
257                          return(0);                          return(0);
258                  }                  } else {
                 else {  
259                          usage();                          usage();
260                          exit(-1);                          exit(-1);
261                  }                  }
# Line 253  Line 267 
267   ****************************************************************************/   ****************************************************************************/
268    
269          if (XDIM <= 0 || XDIM >= 2048 || YDIM <=0 || YDIM >= 2048 ) {          if (XDIM <= 0 || XDIM >= 2048 || YDIM <=0 || YDIM >= 2048 ) {
270                  fprintf(stderr, "Trying to retreive width and height from PGM header\n");                  fprintf(stderr,
271                                    "Trying to retreive width and height from PGM header\n");
272                  ARG_INPUTTYPE = 1; /* pgm */                  ARG_INPUTTYPE = 1; /* pgm */
273          }          }
274    
# Line 262  Line 277 
277                  return(-1);                  return(-1);
278          }          }
279    
         if ( ARG_BITRATE <= 0 && ARG_QUANTI == 0) {  
                 fprintf(stderr,"Wrong Bitrate\n");  
                 return(-1);  
         }  
   
280          if ( ARG_FRAMERATE <= 0) {          if ( ARG_FRAMERATE <= 0) {
281                  fprintf(stderr,"Wrong Framerate %s \n",argv[5]);                  fprintf(stderr,"Wrong Framerate %s \n",argv[5]);
282                  return(-1);                  return(-1);
# Line 279  Line 289 
289    
290          if ( ARG_INPUTFILE == NULL || strcmp(ARG_INPUTFILE, "stdin") == 0) {          if ( ARG_INPUTFILE == NULL || strcmp(ARG_INPUTFILE, "stdin") == 0) {
291                  in_file = stdin;                  in_file = stdin;
292          }          } else {
         else {  
293    
294                  in_file = fopen(ARG_INPUTFILE, "rb");                  in_file = fopen(ARG_INPUTFILE, "rb");
295                  if (in_file == NULL) {                  if (in_file == NULL) {
# Line 291  Line 300 
300    
301          if (ARG_INPUTTYPE) {          if (ARG_INPUTTYPE) {
302                  if (read_pgmheader(in_file)) {                  if (read_pgmheader(in_file)) {
303                          fprintf(stderr, "Wrong input format, I want YUV encapsulated in PGM\n");                          fprintf(stderr,
304                                            "Wrong input format, I want YUV encapsulated in PGM\n");
305                          return(-1);                          return(-1);
306                  }                  }
307          }          }
# Line 311  Line 321 
321   ****************************************************************************/   ****************************************************************************/
322    
323    
324          status = enc_init(use_assembler);          result = enc_init(use_assembler);
325          if (status)          if (result) {
326          {                  fprintf(stderr, "Encore INIT problem, return value %d\n", result);
                 fprintf(stderr, "Encore INIT problem, return value %d\n", status);  
327                  goto release_all;                  goto release_all;
328          }          }
329    
# Line 329  Line 338 
338                          goto release_all;                          goto release_all;
339                  }                  }
340    
341          }          } else {
         else {  
342                  out_file = NULL;                  out_file = NULL;
343          }          }
344    
# Line 340  Line 348 
348    
349          totalsize = 0;          totalsize = 0;
350    
351            result = 0;
352    
353            input_num = 0;                          /* input frame counter */
354            output_num = 0;                         /* output frame counter */
355    
356          do {          do {
357    
358                  char *type;                  char *type;
359                  int stats[3];                  int sse[3];
360    
361                    if (input_num >= ARG_MAXFRAMENR) {
362                            result = 1;
363                    }
364    
365                    if (!result) {
366                  if(ARG_INPUTTYPE) {                  if(ARG_INPUTTYPE) {
367                          /* read PGM data (YUV-format) */                          /* read PGM data (YUV-format) */
368                          status = read_pgmdata(in_file, in_buffer);                                  result = read_pgmdata(in_file, in_buffer);
369                  } else {                  } else {
370                          /* read raw data (YUV-format) */                          /* read raw data (YUV-format) */
371                          status = read_yuvdata(in_file, in_buffer);                                  result = read_yuvdata(in_file, in_buffer);
372                  }                  }
   
                 if(status) {  
                         /* Couldn't read image, most likely end-of-file */  
                         continue;  
373                  }                  }
374    
375  /*****************************************************************************  /*****************************************************************************
# Line 363  Line 377 
377   ****************************************************************************/   ****************************************************************************/
378    
379                  enctime = msecond();                  enctime = msecond();
380                  m4v_size = enc_main(in_buffer, mp4_buffer, &frame_type, stats);                  m4v_size =
381                            enc_main(!result ? in_buffer : 0, mp4_buffer, &key, &stats_type,
382                                             &stats_quant, &stats_length, sse);
383                  enctime = msecond() - enctime;                  enctime = msecond() - enctime;
384    
                 /* Not coded frames return 0 */  
                 if(m4v_size == 0) goto next_frame;  
   
385                  /* Write the Frame statistics */                  /* Write the Frame statistics */
386                  switch(frame_type) {  
387                    printf("%5d: key=%i, time=%6.0f, length=%7d",
388                               !result ? input_num : -1,
389                               key,
390                               (float) enctime,
391                               (int) m4v_size);
392    
393                    if (stats_type > 0) {   /* !XVID_TYPE_NOTHING */
394    
395                            switch (stats_type) {
396                  case XVID_TYPE_IVOP:                  case XVID_TYPE_IVOP:
397                          type = "I";                          type = "I";
398                          break;                          break;
# Line 383  Line 405 
405                  case XVID_TYPE_SVOP:                  case XVID_TYPE_SVOP:
406                          type = "S";                          type = "S";
407                          break;                          break;
                 case XVID_TYPE_NOTHING:  
                         type = "N";  
                         break;  
408                  default:                  default:
409                          type = "U";                          type = "U";
410                          break;                          break;
411                  }                  }
412    
413                  printf("Frame %5d: type = %s, enctime(ms) =%6.1f, length(bytes) =%7d, "                          printf(" | type=%s, quant=%2d, length=%7d", type, stats_quant,
414                             "psnr y = %2.2f, psnr u = %2.2f, psnr v = %2.2f\n",                                     stats_length);
415                             (int)filenr,  
416                             type,  #define SSE2PSNR(sse, width, height) ((!(sse))?0.0f : 48.131f - 10*(float)log10((float)(sse)/((float)((width)*(height)))))
417                             (float)enctime,  
418                             (int)m4v_size,                          if (ARG_STATS) {
419                             (stats[0] == 0)? 0.0f: 48.131f - 10*(float)log10((float)stats[0]/((float)(XDIM)*(YDIM))),                                  printf(", psnr y = %2.2f, psnr u = %2.2f, psnr v = %2.2f",
420                             (stats[1] == 0)? 0.0f: 48.131f - 10*(float)log10((float)stats[1]/((float)(XDIM)*(YDIM)/4)),                                             SSE2PSNR(sse[0], XDIM, YDIM),
421                             (stats[2] == 0)? 0.0f: 48.131f - 10*(float)log10((float)stats[2]/((float)(XDIM)*(YDIM)/4)));                                             SSE2PSNR(sse[1], XDIM/2, YDIM/2),
422                                               SSE2PSNR(sse[2], XDIM/2, YDIM/2));
423                            }
424    
425                    }
426    
427    #undef SSE2PSNR
428    
429                    printf("\n");
430    
431                    if (m4v_size < 0) {
432                            break;
433                    }
434    
435                  /* Update encoding time stats */                  /* Update encoding time stats */
436                  totalenctime += enctime;                  totalenctime += enctime;
# Line 409  Line 440 
440   *                       Save stream to file   *                       Save stream to file
441   ****************************************************************************/   ****************************************************************************/
442    
443                  if (ARG_SAVEMPEGSTREAM)                  if (m4v_size > 0 && ARG_SAVEMPEGSTREAM) {
                 {  
444                          /* Save single files */                          /* Save single files */
445                          if (out_file == NULL) {                          if (out_file == NULL) {
446                                  sprintf(filename, "%sframe%05d.m4v", filepath, filenr);                                  sprintf(filename, "%sframe%05d.m4v", filepath, output_num);
447                                  out_file = fopen(filename, "wb");                                  out_file = fopen(filename, "wb");
448                                  fwrite(mp4_buffer, m4v_size, 1, out_file);                                  fwrite(mp4_buffer, m4v_size, 1, out_file);
449                                  fclose(out_file);                                  fclose(out_file);
450                                  out_file = NULL;                                  out_file = NULL;
451                          }                                  output_num++;
452                          else {                          } else {
453    
454                                  /* Write mp4 data */                                  /* Write mp4 data */
455                                  fwrite(mp4_buffer, 1, m4v_size, out_file);                                  fwrite(mp4_buffer, 1, m4v_size, out_file);
# Line 427  Line 457 
457                          }                          }
458                  }                  }
459    
460          next_frame:                  input_num++;
                 /* Read the header if it's pgm stream */  
                 if (ARG_INPUTTYPE)  
                         status = read_pgmheader(in_file);  
461    
462                  filenr++;                  /* Read the header if it's pgm stream */
463                    if (!result && ARG_INPUTTYPE)
464                            result = read_pgmheader(in_file);
465    
466          } while ( (!status) && (filenr<ARG_MAXFRAMENR) );          } while (1);
467    
468    
469    
# Line 442  Line 471 
471   *         Calculate totals and averages for output, print results   *         Calculate totals and averages for output, print results
472   ****************************************************************************/   ****************************************************************************/
473    
474          totalsize    /= filenr;          printf("Tot: enctime(ms) =%7.2f,               length(bytes) = %7d\n",
475          totalenctime /= filenr;                     totalenctime, (int) totalsize);
476    
477            if (input_num > 0) {
478                    totalsize /= input_num;
479                    totalenctime /= input_num;
480            } else {
481                    totalsize = -1;
482                    totalenctime = -1;
483            }
484    
485          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\n",
486                     totalenctime, 1000/totalenctime, (int)totalsize);                     totalenctime, 1000/totalenctime, (int)totalsize);
# Line 455  Line 492 
492    
493   release_all:   release_all:
494    
495          if (enc_handle)          if (enc_handle) {
496          {                  result = enc_stop();
497                  status = enc_stop();                  if (result)
498                  if (status)                          fprintf(stderr, "Encore RELEASE problem return value %d\n",
499                          fprintf(stderr, "Encore RELEASE problem return value %d\n", status);                                          result);
500          }          }
501    
502          if(in_file)          if(in_file)
# Line 486  Line 523 
523   *****************************************************************************/   *****************************************************************************/
524    
525  /* Return time elapsed time in miliseconds since the program started */  /* Return time elapsed time in miliseconds since the program started */
526  static double msecond()  static double
527    msecond()
528  {  {
529  #ifndef WIN32  #ifndef WIN32
530          struct timeval  tv;          struct timeval  tv;
531    
532          gettimeofday(&tv, 0);          gettimeofday(&tv, 0);
533          return(tv.tv_sec*1.0e3 + tv.tv_usec * 1.0e-3);          return(tv.tv_sec*1.0e3 + tv.tv_usec * 1.0e-3);
534  #else  #else
535          clock_t clk;          clock_t clk;
536    
537          clk = clock();          clk = clock();
538          return(clk * 1000 / CLOCKS_PER_SEC);          return(clk * 1000 / CLOCKS_PER_SEC);
539  #endif  #endif
# Line 503  Line 543 
543   *                             Usage message   *                             Usage message
544   *****************************************************************************/   *****************************************************************************/
545    
546  static void usage()  static void
547    usage()
548  {  {
549            fprintf(stderr, "Usage : xvid_stat [OPTIONS]\n\n");
550          fprintf(stderr, "Usage : xvid_stat [OPTIONS]\n");          fprintf(stderr, "Input options:\n");
551          fprintf(stderr, "Options :\n");          fprintf(stderr, " -i       string : input filename (default=stdin)\n");
552            fprintf(stderr, " -type    integer: input data type (yuv=0, pgm=1)\n");
553          fprintf(stderr, " -w integer     : frame width ([1.2048])\n");          fprintf(stderr, " -w integer     : frame width ([1.2048])\n");
554          fprintf(stderr, " -h integer     : frame height ([1.2048])\n");          fprintf(stderr, " -h integer     : frame height ([1.2048])\n");
555          fprintf(stderr, " -b integer     : target bitrate (>0 | default=900kbit)\n");          fprintf(stderr, " -nframes integer: number of frames to encode\n");
556          fprintf(stderr, " -bn integer    : max bframes (default=0)\n");          fprintf(stderr, "\n");
557          fprintf(stderr, " -bqr integer   : bframe quantizer ratio (default=150)\n");          fprintf(stderr, "Output options:\n");
558          fprintf(stderr, " -bqo integer   : bframe quantizer offset (default=100)\n");          fprintf(stderr, " -dump    : save decoder output\n");
559          fprintf(stderr, " -f float       : target framerate (>0)\n");          fprintf(stderr, " -save    : save mpeg4 raw stream\n");
560          fprintf(stderr, " -i string      : input filename (default=stdin)\n");          fprintf(stderr, " -o string: output filename\n");
561          fprintf(stderr, " -t integer     : input data type (yuv=0, pgm=1)\n");          fprintf(stderr, "\n");
562          fprintf(stderr, " -n integer     : number of frames to encode\n");          fprintf(stderr, "Bitrate options:\n");
563          fprintf(stderr, " -q integer     : quality ([0..5])\n");          fprintf(stderr, " -bitrate       integer: target bitrate (>0 | default=900kbit)\n");
         fprintf(stderr, " -d boolean     : save decoder output (0 False*, !=0 True)\n");  
         fprintf(stderr, " -m boolean     : save mpeg4 raw stream (0 False*, !=0 True)\n");  
         fprintf(stderr, " -o string      : output container filename (only usefull when -m 1 is used) :\n");  
         fprintf(stderr, "                  When this option is not used : one file per encoded frame\n");  
         fprintf(stderr, "                  When this option is used : save to 'string' file\n");  
         fprintf(stderr, " -help          : prints this help message\n");  
564          fprintf(stderr, " -quant integer : fixed quantizer (disables -b setting)\n");          fprintf(stderr, " -quant integer : fixed quantizer (disables -b setting)\n");
565          fprintf(stderr, " (* means default)\n");          fprintf(stderr, " -max_bframes   integer: max bframes (default=0)\n");
566            fprintf(stderr, " -bquant_ratio  integer: bframe quantizer ratio (default=150)\n");
567            fprintf(stderr, " -bquant_offset integer: bframe quantizer offset (default=100)\n");
568            fprintf(stderr, " -framerate     float  : target framerate (>0)\n");
569        fprintf(stderr,     " -pass1         string: fisrt pass stats file\n");
570            fprintf(stderr, "\n");
571            fprintf(stderr, "Other options\n");
572            fprintf(stderr, " -asm            : use assembly optmized code\n");
573            fprintf(stderr, " -quality integer: quality ([0..5])\n");
574            fprintf(stderr, " -packed         : packed mode\n");
575            fprintf(stderr, " -lumimasking    : use lumimasking algorithm\n");
576            fprintf(stderr, " -stats          : print stats about encoded frames\n");
577            fprintf(stderr, " -debug          : print all MB quantizers\n");
578            fprintf(stderr, " -help           : prints this help message\n");
579  }  }
580    
581  /*****************************************************************************  /*****************************************************************************
# Line 538  Line 586 
586   *   *
587   *****************************************************************************/   *****************************************************************************/
588    
589  static int read_pgmheader(FILE* handle)  static int
590    read_pgmheader(FILE * handle)
591  {  {
592          int bytes,xsize,ysize,depth;          int bytes,xsize,ysize,depth;
593          char dummy[2];          char dummy[2];
# Line 549  Line 598 
598                  return(1);                  return(1);
599    
600          fscanf(handle,"%d %d %d",&xsize,&ysize,&depth);          fscanf(handle,"%d %d %d",&xsize,&ysize,&depth);
601          if ( (xsize > 1440) || (ysize > 2880 ) || (depth != 255) )          if ((xsize > 1440) || (ysize > 2880) || (depth != 255)) {
         {  
602                  fprintf(stderr,"%d %d %d\n",xsize,ysize,depth);                  fprintf(stderr,"%d %d %d\n",xsize,ysize,depth);
603                  return(2);                  return(2);
604          }          }
605          if ( (XDIM==0) || (YDIM==0) )          if ((XDIM == 0) || (YDIM == 0)) {
         {  
606                  XDIM=xsize;                  XDIM=xsize;
607                  YDIM=ysize*2/3;                  YDIM=ysize*2/3;
608          }          }
# Line 563  Line 610 
610          return(0);          return(0);
611  }  }
612    
613  static int read_pgmdata(FILE* handle, unsigned char *image)  static int
614    read_pgmdata(FILE * handle,
615                             unsigned char *image)
616  {  {
617          int i;          int i;
618          char dummy;          char dummy;
# Line 575  Line 624 
624          /* read Y component of picture */          /* read Y component of picture */
625          fread(y, 1, XDIM*YDIM, handle);          fread(y, 1, XDIM*YDIM, handle);
626    
627          for (i=0;i<YDIM/2;i++)          for (i = 0; i < YDIM / 2; i++) {
         {  
628                  /* read U */                  /* read U */
629                  fread(u, 1, XDIM/2, handle);                  fread(u, 1, XDIM/2, handle);
630    
# Line 594  Line 642 
642          return(0);          return(0);
643  }  }
644    
645  static int read_yuvdata(FILE* handle, unsigned char *image)  static int
646    read_yuvdata(FILE * handle,
647                             unsigned char *image)
648  {  {
649    
650          if (fread(image, 1, IMAGE_SIZE(XDIM, YDIM), handle) != (unsigned int)IMAGE_SIZE(XDIM, YDIM))          if (fread(image, 1, IMAGE_SIZE(XDIM, YDIM), handle) !=
651                    (unsigned int) IMAGE_SIZE(XDIM, YDIM))
652                  return(1);                  return(1);
653          else          else
654                  return(0);                  return(0);
# Line 607  Line 658 
658   *     Routines for encoding: init encoder, frame step, release encoder   *     Routines for encoding: init encoder, frame step, release encoder
659   ****************************************************************************/   ****************************************************************************/
660    
661    /* sample plugin */
662    
663    int
664    rawenc_debug(void *handle,
665                             int opt,
666                             void *param1,
667                             void *param2)
668    {
669            switch (opt) {
670            case XVID_PLG_INFO:
671                    {
672                            xvid_plg_info_t *info = (xvid_plg_info_t *) param1;
673    
674                            info->flags = XVID_REQDQUANTS;
675                            return 0;
676                    }
677    
678            case XVID_PLG_CREATE:
679            case XVID_PLG_DESTROY:
680            case XVID_PLG_BEFORE:
681                    return 0;
682    
683            case XVID_PLG_AFTER:
684                    {
685                            xvid_plg_data_t *data = (xvid_plg_data_t *) param1;
686                            int i, j;
687    
688                            printf("---[ frame: %5i   quant: %2i   length: %6i ]---\n",
689                                       data->frame_num, data->quant, data->length);
690                            for (j = 0; j < data->mb_height; j++) {
691                                    for (i = 0; i < data->mb_width; i++)
692                                            printf("%2i ", data->dquant[j * data->dquant_stride + i]);
693                                    printf("\n");
694                            }
695    
696                            return 0;
697                    }
698            }
699    
700            return XVID_ERR_FAIL;
701    }
702    
703    
704  #define FRAMERATE_INCR 1001  #define FRAMERATE_INCR 1001
705    
706    
707  /* Initialize encoder for first use, pass all needed parameters to the codec */  /* Initialize encoder for first use, pass all needed parameters to the codec */
708  static int enc_init(int use_assembler)  static int
709    enc_init(int use_assembler)
710  {  {
711          int xerr;          int xerr;
712        xvid_plugin_cbr_t cbr;
713        xvid_plugin_2pass1_t rc2pass1;
714        xvid_enc_plugin_t plugins[5];
715          xvid_gbl_init_t   xvid_gbl_init;          xvid_gbl_init_t   xvid_gbl_init;
716          xvid_enc_create_t xvid_enc_create;          xvid_enc_create_t xvid_enc_create;
717    
# Line 622  Line 720 
720           *----------------------------------------------------------------------*/           *----------------------------------------------------------------------*/
721    
722          /* Set version -- version checking will done by xvidcore*/          /* Set version -- version checking will done by xvidcore*/
723            memset(&xvid_gbl_init, 0, sizeof(xvid_gbl_init));
724          xvid_gbl_init.version = XVID_VERSION;          xvid_gbl_init.version = XVID_VERSION;
725    
726    
727          /* Do we have to enable ASM optimizations ? */          /* Do we have to enable ASM optimizations ? */
728          if(use_assembler) {          if(use_assembler) {
729    
# Line 632  Line 732 
732  #else  #else
733                  xvid_gbl_init.cpu_flags = 0;                  xvid_gbl_init.cpu_flags = 0;
734  #endif  #endif
735          }          } else {
         else {  
736                  xvid_gbl_init.cpu_flags = XVID_CPU_FORCE;                  xvid_gbl_init.cpu_flags = XVID_CPU_FORCE;
737          }          }
738    
# Line 645  Line 744 
744           *----------------------------------------------------------------------*/           *----------------------------------------------------------------------*/
745    
746          /* Version again */          /* Version again */
747            memset(&xvid_enc_create, 0, sizeof(xvid_enc_create));
748          xvid_enc_create.version = XVID_VERSION;          xvid_enc_create.version = XVID_VERSION;
749    
750          /* Width and Height of input frames */          /* Width and Height of input frames */
751          xvid_enc_create.width = XDIM;          xvid_enc_create.width = XDIM;
752          xvid_enc_create.height = YDIM;          xvid_enc_create.height = YDIM;
753    
754            /* init plugins  */
755    
756            xvid_enc_create.plugins = plugins;
757            xvid_enc_create.num_plugins = 0;
758    
759        if (ARG_BITRATE) {
760            cbr.version = XVID_VERSION;
761            memset(&cbr, 0, sizeof(xvid_plugin_cbr_t));
762            cbr.bitrate = ARG_BITRATE;
763    
764            plugins[xvid_enc_create.num_plugins].func = xvid_plugin_cbr;
765                    plugins[xvid_enc_create.num_plugins].param = &cbr;
766                    xvid_enc_create.num_plugins++;
767        }
768    
769        if (ARG_PASS1) {
770            rc2pass1.version = XVID_VERSION;
771            memset(&rc2pass1, 0, sizeof(xvid_plugin_2pass1_t));
772            rc2pass1.filename = ARG_PASS1;
773    
774            plugins[xvid_enc_create.num_plugins].func = xvid_plugin_2pass1;
775                    plugins[xvid_enc_create.num_plugins].param = &rc2pass1;
776                    xvid_enc_create.num_plugins++;
777        }
778    
779            if (ARG_LUMIMASKING) {
780                    plugins[xvid_enc_create.num_plugins].func = xvid_plugin_lumimasking;
781                    plugins[xvid_enc_create.num_plugins].param = NULL;
782                    xvid_enc_create.num_plugins++;
783            }
784    
785            if (ARG_DUMP) {
786                    plugins[xvid_enc_create.num_plugins].func = xvid_plugin_dump;
787                    plugins[xvid_enc_create.num_plugins].param = NULL;
788                    xvid_enc_create.num_plugins++;
789            }
790    
791            if (ARG_DEBUG) {
792                    plugins[xvid_enc_create.num_plugins].func = rawenc_debug;
793                    plugins[xvid_enc_create.num_plugins].param = NULL;
794                    xvid_enc_create.num_plugins++;
795            }
796    
797          /* No fancy thread tests */          /* No fancy thread tests */
798          xvid_enc_create.num_threads = 0;          xvid_enc_create.num_threads = 0;
799    
# Line 675  Line 818 
818          xvid_enc_create.frame_drop_ratio = 0;          xvid_enc_create.frame_drop_ratio = 0;
819    
820          /* Global encoder options */          /* Global encoder options */
821          xvid_enc_create.global = XVID_EXTRASTATS_ENABLE;          xvid_enc_create.global = 0;
822            if (ARG_PACKED) xvid_enc_create.global |=XVID_PACKED;
823            if (ARG_STATS)  xvid_enc_create.global |=XVID_EXTRASTATS_ENABLE;
824    
825          /* 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 */
826          xerr = xvid_encore(NULL, XVID_ENC_CREATE, &xvid_enc_create, NULL);          xerr = xvid_encore(NULL, XVID_ENC_CREATE, &xvid_enc_create, NULL);
# Line 686  Line 831 
831          return(xerr);          return(xerr);
832  }  }
833    
834  static int enc_stop()  static int
835    enc_stop()
836  {  {
837          int xerr;          int xerr;
838    
# Line 696  Line 842 
842          return(xerr);          return(xerr);
843  }  }
844    
845  static int enc_main(unsigned char* image,  static int
846    enc_main(unsigned char *image,
847                                          unsigned char* bitstream,                                          unsigned char* bitstream,
848                                          long *frametype,                   int *key,
849                                          int stats[3])                   int *stats_type,
850                     int *stats_quant,
851                     int *stats_length,
852                     int sse[3])
853  {  {
854          int ret;          int ret;
855    
856          xvid_enc_frame_t xvid_enc_frame;          xvid_enc_frame_t xvid_enc_frame;
857          xvid_enc_stats_t xvid_enc_stats[2];          xvid_enc_stats_t xvid_enc_stats;
858    
859          /* Version for the frame and the stats */          /* Version for the frame and the stats */
860            memset(&xvid_enc_frame, 0, sizeof(xvid_enc_frame));
861          xvid_enc_frame.version = XVID_VERSION;          xvid_enc_frame.version = XVID_VERSION;
862          xvid_enc_stats[0].version = XVID_VERSION;  
863          xvid_enc_stats[1].version = XVID_VERSION;          memset(&xvid_enc_stats, 0, sizeof(xvid_enc_stats));
864            xvid_enc_stats.version = XVID_VERSION;
865    
866          /* Bind output buffer */          /* Bind output buffer */
867          xvid_enc_frame.bitstream = bitstream;          xvid_enc_frame.bitstream = bitstream;
868          xvid_enc_frame.length = -1;          xvid_enc_frame.length = -1;
869    
870          /* Initialize input image fields */          /* Initialize input image fields */
871            if (image) {
872          xvid_enc_frame.input.plane[0]  = image;          xvid_enc_frame.input.plane[0]  = image;
873          xvid_enc_frame.input.csp       = XVID_CSP_I420;          xvid_enc_frame.input.csp       = XVID_CSP_I420;
874          xvid_enc_frame.input.stride[0] = XDIM;          xvid_enc_frame.input.stride[0] = XDIM;
875            } else {
876                    xvid_enc_frame.input.csp = XVID_CSP_NULL;
877            }
878    
879          /* Set up core's general features */          /* Set up core's general features */
880          xvid_enc_frame.vol_flags = vol_presets[ARG_QUALITY];          xvid_enc_frame.vol_flags = vol_presets[ARG_QUALITY];
881            if (ARG_STATS)
882                    xvid_enc_frame.vol_flags |= XVID_EXTRASTATS;
883    
884          /* Set up core's general features */          /* Set up core's general features */
885          xvid_enc_frame.vop_flags = vop_presets[ARG_QUALITY];          xvid_enc_frame.vop_flags = vop_presets[ARG_QUALITY];
# Line 731  Line 889 
889    
890          /* Force the right quantizer */          /* Force the right quantizer */
891          xvid_enc_frame.quant  = ARG_QUANTI;          xvid_enc_frame.quant  = ARG_QUANTI;
         xvid_enc_frame.bquant = 0;  
892    
893          /* Set up motion estimation flags */          /* Set up motion estimation flags */
894          xvid_enc_frame.motion = motion_presets[ARG_QUALITY];          xvid_enc_frame.motion = motion_presets[ARG_QUALITY];
# Line 740  Line 897 
897          xvid_enc_frame.quant_intra_matrix = NULL;          xvid_enc_frame.quant_intra_matrix = NULL;
898          xvid_enc_frame.quant_inter_matrix = NULL;          xvid_enc_frame.quant_inter_matrix = NULL;
899    
         /* Foll proof */  
         xvid_enc_stats[0].sse_y = 0;  
         xvid_enc_stats[0].sse_v = 0;  
         xvid_enc_stats[0].sse_u = 0;  
   
900          /* Encode the frame */          /* Encode the frame */
901          xvid_enc_frame.vop_flags |= XVID_EXTRASTATS;          ret =
902          ret = xvid_encore(enc_handle, XVID_ENC_ENCODE, &xvid_enc_frame, &xvid_enc_stats);                  xvid_encore(enc_handle, XVID_ENC_ENCODE, &xvid_enc_frame,
903                                            &xvid_enc_stats);
904          *frametype = xvid_enc_stats[0].type;  
905          stats[0]   = xvid_enc_stats[0].sse_y;          *key = (xvid_enc_frame.out_flags & XVID_KEYFRAME);
906          stats[1]   = xvid_enc_stats[0].sse_u;          *stats_type = xvid_enc_stats.type;
907          stats[2]   = xvid_enc_stats[0].sse_v;          *stats_quant = xvid_enc_stats.quant;
908            *stats_length = xvid_enc_stats.length;
909            sse[0] = xvid_enc_stats.sse_y;
910            sse[1] = xvid_enc_stats.sse_u;
911            sse[2] = xvid_enc_stats.sse_v;
912    
913          return(ret);          return(ret);
914  }  }
Legend:
Removed from v.913  
changed lines
  Added in v.938
No admin address has been configured
 ViewVC Help
Powered by ViewVC 1.0.4