[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 916, Tue Mar 11 01:08:10 2003 UTC revision 932, Thu Mar 20 14:14:52 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.3 2003-03-11 01:08:10 edgomez Exp $   * $Id: xvid_encraw.c,v 1.11.2.9 2003-03-20 14:14:52 edgomez 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 85  Line 86 
86  #define ABS_MAXFRAMENR 9999  #define ABS_MAXFRAMENR 9999
87    
88  static int   ARG_STATS = 0;  static int   ARG_STATS = 0;
89    static int ARG_DUMP = 0;
90    static int ARG_LUMIMASKING = 0;
91  static int   ARG_BITRATE = 900;  static int   ARG_BITRATE = 900;
92  static int   ARG_QUANTI = 0;  static int   ARG_QUANTI = 0;
93  static int   ARG_QUALITY = 5;  static int   ARG_QUALITY = 5;
 static int   ARG_MINQUANT = 1;  
 static int   ARG_MAXQUANT = 31;  
94  static float ARG_FRAMERATE = 25.00f;  static float ARG_FRAMERATE = 25.00f;
95  static int   ARG_MAXFRAMENR = ABS_MAXFRAMENR;  static int   ARG_MAXFRAMENR = ABS_MAXFRAMENR;
96  static char *ARG_INPUTFILE = NULL;  static char *ARG_INPUTFILE = NULL;
# Line 101  Line 102 
102  static int   ARG_BQRATIO = 150;  static int   ARG_BQRATIO = 150;
103  static int   ARG_BQOFFSET = 100;  static int   ARG_BQOFFSET = 100;
104  static int   ARG_MAXBFRAMES = 0;  static int   ARG_MAXBFRAMES = 0;
105    static int ARG_PACKED = 0;
106    static int ARG_DEBUG = 0;
107    
108  #define IMAGE_SIZE(x,y) ((x)*(y)*3/2)  #define IMAGE_SIZE(x,y) ((x)*(y)*3/2)
109    
110  #define MAX(A,B) ( ((A)>(B)) ? (A) : (B) )  #define MAX(A,B) ( ((A)>(B)) ? (A) : (B) )
# Line 113  Line 117 
117   *                     Nasty global vars ;-)   *                     Nasty global vars ;-)
118   ***************************************************************************/   ***************************************************************************/
119    
120  static int i,filenr = 0;  static int i;
121    
122  /* the path where to save output */  /* the path where to save output */
123  static char filepath[256] = "./";  static char filepath[256] = "./";
# Line 133  Line 137 
137    
138  /* PGM related functions */  /* PGM related functions */
139  static int read_pgmheader(FILE* handle);  static int read_pgmheader(FILE* handle);
140  static int read_pgmdata(FILE* handle, unsigned char *image);  static int read_pgmdata(FILE * handle,
141  static int read_yuvdata(FILE* handle, unsigned char *image);                                                  unsigned char *image);
142    static int read_yuvdata(FILE * handle,
143                                                    unsigned char *image);
144    
145  /* Encoder related functions */  /* Encoder related functions */
146  static int enc_init(int use_assembler);  static int enc_init(int use_assembler);
147  static int enc_stop();  static int enc_stop();
148  static int enc_main(unsigned char* image,  static int enc_main(unsigned char* image,
149                                          unsigned char* bitstream,                                          unsigned char* bitstream,
150                                          long *frametype,                                          int *key,
151                                            int *stats_type,
152                                            int *stats_quant,
153                                            int *stats_length,
154                                          int stats[3]);                                          int stats[3]);
155    
156  /*****************************************************************************  /*****************************************************************************
157   *               Main function   *               Main function
158   ****************************************************************************/   ****************************************************************************/
159    
160  int main(int argc, char *argv[])  int
161    main(int argc,
162             char *argv[])
163  {  {
164    
165          unsigned char *mp4_buffer = NULL;          unsigned char *mp4_buffer = NULL;
# Line 158  Line 169 
169          double enctime;          double enctime;
170          double totalenctime=0.;          double totalenctime=0.;
171    
172          long totalsize;          int totalsize;
173          int status;          int result;
174          long frame_type;          int m4v_size;
175            int key;
176          long m4v_size;          int stats_type;
177            int stats_quant;
178            int stats_length;
179          int use_assembler=0;          int use_assembler=0;
180    
181            int input_num;
182            int output_num;
183    
184          char filename[256];          char filename[256];
185    
186          FILE *in_file = stdin;          FILE *in_file = stdin;
# Line 181  Line 197 
197    
198                  if (strcmp("-asm", argv[i]) == 0 ) {                  if (strcmp("-asm", argv[i]) == 0 ) {
199                          use_assembler = 1;                          use_assembler = 1;
200                  }                  } else if (strcmp("-w", argv[i]) == 0 && i < argc - 1) {
                 else if (strcmp("-w", argv[i]) == 0 && i < argc - 1 ) {  
201                          i++;                          i++;
202                          XDIM = atoi(argv[i]);                          XDIM = atoi(argv[i]);
203                  }                  } else if (strcmp("-h", argv[i]) == 0 && i < argc - 1) {
                 else if (strcmp("-h", argv[i]) == 0 && i < argc - 1 ) {  
204                          i++;                          i++;
205                          YDIM = atoi(argv[i]);                          YDIM = atoi(argv[i]);
206                  }                  } else if (strcmp("-bitrate", argv[i]) == 0 && i < argc - 1) {
                 else if (strcmp("-b", argv[i]) == 0 && i < argc - 1 ) {  
207                          i++;                          i++;
208                          ARG_BITRATE = atoi(argv[i]);                          ARG_BITRATE = atoi(argv[i]);
209                  }                  } else if (strcmp("-max_bframes", argv[i]) == 0 && i < argc - 1) {
                 else if (strcmp("-bn", argv[i]) == 0 && i < argc - 1 ) {  
210                          i++;                          i++;
211                          ARG_MAXBFRAMES = atoi(argv[i]);                          ARG_MAXBFRAMES = atoi(argv[i]);
212                  }                  } else if (strcmp("-packed", argv[i]) == 0) {
213                  else if (strcmp("-bqr", argv[i]) == 0 && i < argc - 1 ) {                          ARG_PACKED = 1;
214                    } else if (strcmp("-bquant_ratio", argv[i]) == 0 && i < argc - 1) {
215                          i++;                          i++;
216                          ARG_BQRATIO = atoi(argv[i]);                          ARG_BQRATIO = atoi(argv[i]);
217                  }                  } else if (strcmp("-bquant_offset", argv[i]) == 0 && i < argc - 1) {
                 else if (strcmp("-bqo", argv[i]) == 0 && i < argc - 1 ) {  
218                          i++;                          i++;
219                          ARG_BQOFFSET = atoi(argv[i]);                          ARG_BQOFFSET = atoi(argv[i]);
220                  }                  } else if (strcmp("-quality", argv[i]) == 0 && i < argc - 1) {
                 else if (strcmp("-q", argv[i]) == 0 && i < argc - 1 ) {  
221                          i++;                          i++;
222                          ARG_QUALITY = atoi(argv[i]);                          ARG_QUALITY = atoi(argv[i]);
223                  }                  } else if (strcmp("-framerate", argv[i]) == 0 && i < argc - 1) {
                 else if (strcmp("-f", argv[i]) == 0 && i < argc - 1 ) {  
224                          i++;                          i++;
225                          ARG_FRAMERATE = (float)atof(argv[i]);                          ARG_FRAMERATE = (float)atof(argv[i]);
226                  }                  } else if (strcmp("-i", argv[i]) == 0 && i < argc - 1) {
                 else if (strcmp("-i", argv[i]) == 0 && i < argc - 1 ) {  
227                          i++;                          i++;
228                          ARG_INPUTFILE = argv[i];                          ARG_INPUTFILE = argv[i];
229                  }                  } else if (strcmp("-stats", argv[i]) == 0) {
                 else if (strcmp("-s", argv[i]) == 0) {  
230                          ARG_STATS = 1;                          ARG_STATS = 1;
231                  }                  } else if (strcmp("-dump", argv[i]) == 0) {
232                  else if (strcmp("-t", argv[i]) == 0 && i < argc - 1 ) {                          ARG_DUMP = 1;
233                    } else if (strcmp("-lumimasking", argv[i]) == 0) {
234                            ARG_LUMIMASKING = 1;
235                    } else if (strcmp("-type", argv[i]) == 0 && i < argc - 1) {
236                          i++;                          i++;
237                          ARG_INPUTTYPE = atoi(argv[i]);                          ARG_INPUTTYPE = atoi(argv[i]);
238                  }                  } else if (strcmp("-nframes", argv[i]) == 0 && i < argc - 1) {
                 else if(strcmp("-n", argv[i]) == 0 && i < argc - 1 ) {  
239                          i++;                          i++;
240                          ARG_MAXFRAMENR  = atoi(argv[i]);                          ARG_MAXFRAMENR  = atoi(argv[i]);
241                  }                  } else if (strcmp("-quant", argv[i]) == 0 && i < argc - 1) {
                 else if (strcmp("-quant", argv[i]) == 0 && i < argc - 1 ) {  
242                          i++;                          i++;
243                          ARG_QUANTI = atoi(argv[i]);                          ARG_QUANTI = atoi(argv[i]);
244                  }                  } else if (strcmp("-save", argv[i]) == 0) {
                 else if (strcmp("-m", argv[i]) == 0) {  
245                          ARG_SAVEMPEGSTREAM = 1;                          ARG_SAVEMPEGSTREAM = 1;
246                  }                  } else if (strcmp("-debug", argv[i]) == 0) {
247                  else if (strcmp("-o", argv[i]) == 0 && i < argc - 1 ) {                          ARG_DEBUG = 1;
248                    } else if (strcmp("-o", argv[i]) == 0 && i < argc - 1) {
249                          i++;                          i++;
250                          ARG_OUTPUTFILE = argv[i];                          ARG_OUTPUTFILE = argv[i];
251                  }                  } else if (strcmp("-help", argv[i])) {
                 else if (strcmp("-help", argv[i])) {  
252                          usage();                          usage();
253                          return(0);                          return(0);
254                  }                  } else {
                 else {  
255                          usage();                          usage();
256                          exit(-1);                          exit(-1);
257                  }                  }
# Line 256  Line 263 
263   ****************************************************************************/   ****************************************************************************/
264    
265          if (XDIM <= 0 || XDIM >= 2048 || YDIM <=0 || YDIM >= 2048 ) {          if (XDIM <= 0 || XDIM >= 2048 || YDIM <=0 || YDIM >= 2048 ) {
266                  fprintf(stderr, "Trying to retreive width and height from PGM header\n");                  fprintf(stderr,
267                                    "Trying to retreive width and height from PGM header\n");
268                  ARG_INPUTTYPE = 1; /* pgm */                  ARG_INPUTTYPE = 1; /* pgm */
269          }          }
270    
# Line 282  Line 290 
290    
291          if ( ARG_INPUTFILE == NULL || strcmp(ARG_INPUTFILE, "stdin") == 0) {          if ( ARG_INPUTFILE == NULL || strcmp(ARG_INPUTFILE, "stdin") == 0) {
292                  in_file = stdin;                  in_file = stdin;
293          }          } else {
         else {  
294    
295                  in_file = fopen(ARG_INPUTFILE, "rb");                  in_file = fopen(ARG_INPUTFILE, "rb");
296                  if (in_file == NULL) {                  if (in_file == NULL) {
# Line 294  Line 301 
301    
302          if (ARG_INPUTTYPE) {          if (ARG_INPUTTYPE) {
303                  if (read_pgmheader(in_file)) {                  if (read_pgmheader(in_file)) {
304                          fprintf(stderr, "Wrong input format, I want YUV encapsulated in PGM\n");                          fprintf(stderr,
305                                            "Wrong input format, I want YUV encapsulated in PGM\n");
306                          return(-1);                          return(-1);
307                  }                  }
308          }          }
# Line 314  Line 322 
322   ****************************************************************************/   ****************************************************************************/
323    
324    
325          status = enc_init(use_assembler);          result = enc_init(use_assembler);
326          if (status)          if (result) {
327          {                  fprintf(stderr, "Encore INIT problem, return value %d\n", result);
                 fprintf(stderr, "Encore INIT problem, return value %d\n", status);  
328                  goto release_all;                  goto release_all;
329          }          }
330    
# Line 332  Line 339 
339                          goto release_all;                          goto release_all;
340                  }                  }
341    
342          }          } else {
         else {  
343                  out_file = NULL;                  out_file = NULL;
344          }          }
345    
# Line 343  Line 349 
349    
350          totalsize = 0;          totalsize = 0;
351    
352            result = 0;
353    
354            input_num = 0;                          /* input frame counter */
355            output_num = 0;                         /* output frame counter */
356    
357          do {          do {
358    
359                  char *type;                  char *type;
360                  int stats[3];                  int sse[3];
361    
362                    if (input_num >= ARG_MAXFRAMENR) {
363                            result = 1;
364                    }
365    
366                    if (!result) {
367                  if(ARG_INPUTTYPE) {                  if(ARG_INPUTTYPE) {
368                          /* read PGM data (YUV-format) */                          /* read PGM data (YUV-format) */
369                          status = read_pgmdata(in_file, in_buffer);                                  result = read_pgmdata(in_file, in_buffer);
370                  } else {                  } else {
371                          /* read raw data (YUV-format) */                          /* read raw data (YUV-format) */
372                          status = read_yuvdata(in_file, in_buffer);                                  result = read_yuvdata(in_file, in_buffer);
373                  }                  }
   
                 if(status) {  
                         /* Couldn't read image, most likely end-of-file */  
                         continue;  
374                  }                  }
375    
376  /*****************************************************************************  /*****************************************************************************
# Line 366  Line 378 
378   ****************************************************************************/   ****************************************************************************/
379    
380                  enctime = msecond();                  enctime = msecond();
381                  m4v_size = enc_main(in_buffer, mp4_buffer, &frame_type, stats);                  m4v_size =
382                            enc_main(!result ? in_buffer : 0, mp4_buffer, &key, &stats_type,
383                                             &stats_quant, &stats_length, sse);
384                  enctime = msecond() - enctime;                  enctime = msecond() - enctime;
385    
                 /* Not coded frames return 0 */  
                 if(m4v_size == 0) goto next_frame;  
   
386                  /* Write the Frame statistics */                  /* Write the Frame statistics */
387                  switch(frame_type) {  
388                    printf("%5d: key=%i, time(ms)=%6.1f, length=%7d",
389                               !result ? input_num : -1,
390                               key,
391                               (float) enctime,
392                               (int) m4v_size);
393    
394                    if (stats_type > 0) {   /* !XVID_TYPE_NOTHING */
395    
396                            switch (stats_type) {
397                  case XVID_TYPE_IVOP:                  case XVID_TYPE_IVOP:
398                          type = "I";                          type = "I";
399                          break;                          break;
# Line 386  Line 406 
406                  case XVID_TYPE_SVOP:                  case XVID_TYPE_SVOP:
407                          type = "S";                          type = "S";
408                          break;                          break;
                 case XVID_TYPE_NOTHING:  
                         type = "N";  
                         break;  
409                  default:                  default:
410                          type = "U";                          type = "U";
411                          break;                          break;
412                  }                  }
413    
414                  printf("Frame %5d: type = %s, enctime(ms) =%6.1f, length(bytes) =%7d",                          printf(" | type=%s, quant=%2d, length=%7d", type, stats_quant,
415                             (int)filenr,                                     stats_length);
416                             type,  
417                             (float)enctime,  #define SSE2PSNR(sse, width, height) ((!(sse))?0.0f : 48.131f - 10*(float)log10((float)(sse)/((float)((width)*(height)))))
                            (int)m4v_size);  
418    
419                  if(ARG_STATS) {                  if(ARG_STATS) {
420                          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",
421                                     (stats[0] == 0)? 0.0f: 48.131f - 10*(float)log10((float)stats[0]/((float)(XDIM)*(YDIM))),                                             SSE2PSNR(sse[0], XDIM, YDIM),
422                                     (stats[1] == 0)? 0.0f: 48.131f - 10*(float)log10((float)stats[1]/((float)(XDIM)*(YDIM)/4)),                                             SSE2PSNR(sse[1], XDIM/2, YDIM/2),
423                                     (stats[2] == 0)? 0.0f: 48.131f - 10*(float)log10((float)stats[2]/((float)(XDIM)*(YDIM)/4)));                                             SSE2PSNR(sse[2], XDIM/2, YDIM/2));
424                  }                  }
425    
426                    }
427    
428    #undef SSE2PSNR
429    
430                  printf("\n");                  printf("\n");
431    
432                    if (m4v_size < 0) {
433                            break;
434                    }
435    
436                  /* Update encoding time stats */                  /* Update encoding time stats */
437                  totalenctime += enctime;                  totalenctime += enctime;
438                  totalsize += m4v_size;                  totalsize += m4v_size;
# Line 417  Line 441 
441   *                       Save stream to file   *                       Save stream to file
442   ****************************************************************************/   ****************************************************************************/
443    
444                  if (ARG_SAVEMPEGSTREAM)                  if (m4v_size > 0 && ARG_SAVEMPEGSTREAM) {
                 {  
445                          /* Save single files */                          /* Save single files */
446                          if (out_file == NULL) {                          if (out_file == NULL) {
447                                  sprintf(filename, "%sframe%05d.m4v", filepath, filenr);                                  sprintf(filename, "%sframe%05d.m4v", filepath, output_num);
448                                  out_file = fopen(filename, "wb");                                  out_file = fopen(filename, "wb");
449                                  fwrite(mp4_buffer, m4v_size, 1, out_file);                                  fwrite(mp4_buffer, m4v_size, 1, out_file);
450                                  fclose(out_file);                                  fclose(out_file);
451                                  out_file = NULL;                                  out_file = NULL;
452                          }                                  output_num++;
453                          else {                          } else {
454    
455                                  /* Write mp4 data */                                  /* Write mp4 data */
456                                  fwrite(mp4_buffer, 1, m4v_size, out_file);                                  fwrite(mp4_buffer, 1, m4v_size, out_file);
# Line 435  Line 458 
458                          }                          }
459                  }                  }
460    
461          next_frame:                  input_num++;
                 /* Read the header if it's pgm stream */  
                 if (ARG_INPUTTYPE)  
                         status = read_pgmheader(in_file);  
462    
463                  filenr++;                  /* Read the header if it's pgm stream */
464                    if (!result && ARG_INPUTTYPE)
465                            result = read_pgmheader(in_file);
466    
467          } while ( (!status) && (filenr<ARG_MAXFRAMENR) );          } while (1);
468    
469    
470    
# Line 450  Line 472 
472   *         Calculate totals and averages for output, print results   *         Calculate totals and averages for output, print results
473   ****************************************************************************/   ****************************************************************************/
474    
475          totalsize    /= filenr;          printf("Tot: enctime(ms) =%7.2f,               length(bytes) = %7d\n",
476          totalenctime /= filenr;                     totalenctime, (int) totalsize);
477    
478            if (input_num > 0) {
479                    totalsize /= input_num;
480                    totalenctime /= input_num;
481            } else {
482                    totalsize = -1;
483                    totalenctime = -1;
484            }
485    
486          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",
487                     totalenctime, 1000/totalenctime, (int)totalsize);                     totalenctime, 1000/totalenctime, (int)totalsize);
# Line 463  Line 493 
493    
494   release_all:   release_all:
495    
496          if (enc_handle)          if (enc_handle) {
497          {                  result = enc_stop();
498                  status = enc_stop();                  if (result)
499                  if (status)                          fprintf(stderr, "Encore RELEASE problem return value %d\n",
500                          fprintf(stderr, "Encore RELEASE problem return value %d\n", status);                                          result);
501          }          }
502    
503          if(in_file)          if(in_file)
# Line 494  Line 524 
524   *****************************************************************************/   *****************************************************************************/
525    
526  /* Return time elapsed time in miliseconds since the program started */  /* Return time elapsed time in miliseconds since the program started */
527  static double msecond()  static double
528    msecond()
529  {  {
530  #ifndef WIN32  #ifndef WIN32
531          struct timeval  tv;          struct timeval  tv;
532    
533          gettimeofday(&tv, 0);          gettimeofday(&tv, 0);
534          return(tv.tv_sec*1.0e3 + tv.tv_usec * 1.0e-3);          return(tv.tv_sec*1.0e3 + tv.tv_usec * 1.0e-3);
535  #else  #else
536          clock_t clk;          clock_t clk;
537    
538          clk = clock();          clk = clock();
539          return(clk * 1000 / CLOCKS_PER_SEC);          return(clk * 1000 / CLOCKS_PER_SEC);
540  #endif  #endif
# Line 511  Line 544 
544   *                             Usage message   *                             Usage message
545   *****************************************************************************/   *****************************************************************************/
546    
547  static void usage()  static void
548    usage()
549  {  {
550            fprintf(stderr, "Usage : xvid_stat [OPTIONS]\n\n");
551          fprintf(stderr, "Usage : xvid_stat [OPTIONS]\n");          fprintf(stderr, "Input options:\n");
552          fprintf(stderr, "Options :\n");          fprintf(stderr, " -i       string : input filename (default=stdin)\n");
553            fprintf(stderr, " -type    integer: input data type (yuv=0, pgm=1)\n");
554          fprintf(stderr, " -w integer     : frame width ([1.2048])\n");          fprintf(stderr, " -w integer     : frame width ([1.2048])\n");
555          fprintf(stderr, " -h integer     : frame height ([1.2048])\n");          fprintf(stderr, " -h integer     : frame height ([1.2048])\n");
556          fprintf(stderr, " -b integer     : target bitrate (>0 | default=900kbit)\n");          fprintf(stderr, " -nframes integer: number of frames to encode\n");
557          fprintf(stderr, " -bn integer    : max bframes (default=0)\n");          fprintf(stderr, "\n");
558          fprintf(stderr, " -bqr integer   : bframe quantizer ratio (default=150)\n");          fprintf(stderr, "Output options:\n");
559          fprintf(stderr, " -bqo integer   : bframe quantizer offset (default=100)\n");          fprintf(stderr, " -dump    : save decoder output\n");
560          fprintf(stderr, " -f float       : target framerate (>0)\n");          fprintf(stderr, " -save    : save mpeg4 raw stream\n");
561          fprintf(stderr, " -i string      : input filename (default=stdin)\n");          fprintf(stderr, " -o string: output filename\n");
562          fprintf(stderr, " -s             : print stats about encoded frames\n");          fprintf(stderr, "\n");
563          fprintf(stderr, " -t integer     : input data type (yuv=0, pgm=1)\n");          fprintf(stderr, "Bitrate options:\n");
564          fprintf(stderr, " -n integer     : number of frames to encode\n");          fprintf(stderr, " -bitrate       integer: target bitrate (>0 | default=900kbit)\n");
         fprintf(stderr, " -q integer     : quality ([0..5])\n");  
         fprintf(stderr, " -d boolean     : save decoder output (0 False*, !=0 True)\n");  
         fprintf(stderr, " -m             : save mpeg4 raw stream\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");  
565          fprintf(stderr, " -quant integer : fixed quantizer (disables -b setting)\n");          fprintf(stderr, " -quant integer : fixed quantizer (disables -b setting)\n");
566          fprintf(stderr, " (* means default)\n");          fprintf(stderr, " -max_bframes   integer: max bframes (default=0)\n");
567            fprintf(stderr, " -bquant_ratio  integer: bframe quantizer ratio (default=150)\n");
568            fprintf(stderr, " -bquant_offset integer: bframe quantizer offset (default=100)\n");
569            fprintf(stderr, " -framerate     float  : target framerate (>0)\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 547  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 558  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 572  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 584  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 603  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 616  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  /* Initialize encoder for first use, pass all needed parameters to the codec */  /* Initialize encoder for first use, pass all needed parameters to the codec */
707  static int enc_init(int use_assembler)  static int
708    enc_init(int use_assembler)
709  {  {
710          int xerr;          int xerr;
711    
712            xvid_enc_plugin_t plugins[2];
713    
714          xvid_gbl_init_t   xvid_gbl_init;          xvid_gbl_init_t   xvid_gbl_init;
715          xvid_enc_create_t xvid_enc_create;          xvid_enc_create_t xvid_enc_create;
716    
# Line 631  Line 719 
719           *----------------------------------------------------------------------*/           *----------------------------------------------------------------------*/
720    
721          /* Set version -- version checking will done by xvidcore*/          /* Set version -- version checking will done by xvidcore*/
722            memset(&xvid_gbl_init, 0, sizeof(xvid_gbl_init));
723          xvid_gbl_init.version = XVID_VERSION;          xvid_gbl_init.version = XVID_VERSION;
724    
725    
726          /* Do we have to enable ASM optimizations ? */          /* Do we have to enable ASM optimizations ? */
727          if(use_assembler) {          if(use_assembler) {
728    
# Line 641  Line 731 
731  #else  #else
732                  xvid_gbl_init.cpu_flags = 0;                  xvid_gbl_init.cpu_flags = 0;
733  #endif  #endif
734          }          } else {
         else {  
735                  xvid_gbl_init.cpu_flags = XVID_CPU_FORCE;                  xvid_gbl_init.cpu_flags = XVID_CPU_FORCE;
736          }          }
737    
# Line 654  Line 743 
743           *----------------------------------------------------------------------*/           *----------------------------------------------------------------------*/
744    
745          /* Version again */          /* Version again */
746            memset(&xvid_enc_create, 0, sizeof(xvid_enc_create));
747          xvid_enc_create.version = XVID_VERSION;          xvid_enc_create.version = XVID_VERSION;
748    
749          /* Width and Height of input frames */          /* Width and Height of input frames */
750          xvid_enc_create.width = XDIM;          xvid_enc_create.width = XDIM;
751          xvid_enc_create.height = YDIM;          xvid_enc_create.height = YDIM;
752    
753            /* init plugins  */
754    
755            xvid_enc_create.plugins = plugins;
756            xvid_enc_create.num_plugins = 0;
757    
758            if (ARG_LUMIMASKING) {
759                    plugins[xvid_enc_create.num_plugins].func = xvid_plugin_lumimasking;
760                    plugins[xvid_enc_create.num_plugins].param = NULL;
761                    xvid_enc_create.num_plugins++;
762            }
763    
764            if (ARG_DUMP) {
765                    plugins[xvid_enc_create.num_plugins].func = xvid_plugin_dump;
766                    plugins[xvid_enc_create.num_plugins].param = NULL;
767                    xvid_enc_create.num_plugins++;
768            }
769    
770            if (ARG_DEBUG) {
771                    plugins[xvid_enc_create.num_plugins].func = rawenc_debug;
772                    plugins[xvid_enc_create.num_plugins].param = NULL;
773                    xvid_enc_create.num_plugins++;
774            }
775    
776          /* No fancy thread tests */          /* No fancy thread tests */
777          xvid_enc_create.num_threads = 0;          xvid_enc_create.num_threads = 0;
778    
# Line 684  Line 797 
797          xvid_enc_create.frame_drop_ratio = 0;          xvid_enc_create.frame_drop_ratio = 0;
798    
799          /* Global encoder options */          /* Global encoder options */
800          xvid_enc_create.global = (ARG_STATS)?XVID_EXTRASTATS_ENABLE:0;          xvid_enc_create.global = 0;
801            if (ARG_PACKED) xvid_enc_create.global |=XVID_PACKED;
802            if (ARG_STATS)  xvid_enc_create.global |=XVID_EXTRASTATS_ENABLE;
803    
804          /* 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 */
805          xerr = xvid_encore(NULL, XVID_ENC_CREATE, &xvid_enc_create, NULL);          xerr = xvid_encore(NULL, XVID_ENC_CREATE, &xvid_enc_create, NULL);
# Line 695  Line 810 
810          return(xerr);          return(xerr);
811  }  }
812    
813  static int enc_stop()  static int
814    enc_stop()
815  {  {
816          int xerr;          int xerr;
817    
# Line 705  Line 821 
821          return(xerr);          return(xerr);
822  }  }
823    
824  static int enc_main(unsigned char* image,  static int
825    enc_main(unsigned char *image,
826                                          unsigned char* bitstream,                                          unsigned char* bitstream,
827                                          long *frametype,                   int *key,
828                                          int stats[3])                   int *stats_type,
829                     int *stats_quant,
830                     int *stats_length,
831                     int sse[3])
832  {  {
833          int ret;          int ret;
834    
835          xvid_enc_frame_t xvid_enc_frame;          xvid_enc_frame_t xvid_enc_frame;
836          xvid_enc_stats_t xvid_enc_stats[2];          xvid_enc_stats_t xvid_enc_stats;
837    
838          /* Version for the frame and the stats */          /* Version for the frame and the stats */
839            memset(&xvid_enc_frame, 0, sizeof(xvid_enc_frame));
840          xvid_enc_frame.version = XVID_VERSION;          xvid_enc_frame.version = XVID_VERSION;
841          xvid_enc_stats[0].version = XVID_VERSION;  
842          xvid_enc_stats[1].version = XVID_VERSION;          memset(&xvid_enc_stats, 0, sizeof(xvid_enc_stats));
843            xvid_enc_stats.version = XVID_VERSION;
844    
845          /* Bind output buffer */          /* Bind output buffer */
846          xvid_enc_frame.bitstream = bitstream;          xvid_enc_frame.bitstream = bitstream;
847          xvid_enc_frame.length = -1;          xvid_enc_frame.length = -1;
848    
849          /* Initialize input image fields */          /* Initialize input image fields */
850            if (image) {
851          xvid_enc_frame.input.plane[0]  = image;          xvid_enc_frame.input.plane[0]  = image;
852          xvid_enc_frame.input.csp       = XVID_CSP_I420;          xvid_enc_frame.input.csp       = XVID_CSP_I420;
853          xvid_enc_frame.input.stride[0] = XDIM;          xvid_enc_frame.input.stride[0] = XDIM;
854            } else {
855                    xvid_enc_frame.input.csp = XVID_CSP_NULL;
856            }
857    
858          /* Set up core's general features */          /* Set up core's general features */
859          xvid_enc_frame.vol_flags = vol_presets[ARG_QUALITY];          xvid_enc_frame.vol_flags = vol_presets[ARG_QUALITY];
860            if (ARG_STATS)
861                    xvid_enc_frame.vol_flags |= XVID_EXTRASTATS;
862    
863          /* Set up core's general features */          /* Set up core's general features */
864          xvid_enc_frame.vop_flags = vop_presets[ARG_QUALITY];          xvid_enc_frame.vop_flags = vop_presets[ARG_QUALITY];
# Line 740  Line 868 
868    
869          /* Force the right quantizer */          /* Force the right quantizer */
870          xvid_enc_frame.quant  = ARG_QUANTI;          xvid_enc_frame.quant  = ARG_QUANTI;
         xvid_enc_frame.bquant = 0;  
871    
872          /* Set up motion estimation flags */          /* Set up motion estimation flags */
873          xvid_enc_frame.motion = motion_presets[ARG_QUALITY];          xvid_enc_frame.motion = motion_presets[ARG_QUALITY];
# Line 749  Line 876 
876          xvid_enc_frame.quant_intra_matrix = NULL;          xvid_enc_frame.quant_intra_matrix = NULL;
877          xvid_enc_frame.quant_inter_matrix = NULL;          xvid_enc_frame.quant_inter_matrix = NULL;
878    
         /* Foll proof */  
         xvid_enc_stats[0].sse_y = 0;  
         xvid_enc_stats[0].sse_v = 0;  
         xvid_enc_stats[0].sse_u = 0;  
   
879          /* Encode the frame */          /* Encode the frame */
880          xvid_enc_frame.vop_flags |= (ARG_STATS)?XVID_EXTRASTATS:0;          ret =
881          ret = xvid_encore(enc_handle, XVID_ENC_ENCODE, &xvid_enc_frame, &xvid_enc_stats);                  xvid_encore(enc_handle, XVID_ENC_ENCODE, &xvid_enc_frame,
882                                            &xvid_enc_stats);
883          *frametype = xvid_enc_stats[0].type;  
884          stats[0]   = xvid_enc_stats[0].sse_y;          *key = (xvid_enc_frame.out_flags & XVID_KEYFRAME);
885          stats[1]   = xvid_enc_stats[0].sse_u;          *stats_type = xvid_enc_stats.type;
886          stats[2]   = xvid_enc_stats[0].sse_v;          *stats_quant = xvid_enc_stats.quant;
887            *stats_length = xvid_enc_stats.length;
888            sse[0] = xvid_enc_stats.sse_y;
889            sse[1] = xvid_enc_stats.sse_u;
890            sse[2] = xvid_enc_stats.sse_v;
891    
892          return(ret);          return(ret);
893  }  }

Legend:
Removed from v.916  
changed lines
  Added in v.932

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