[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 920, Sat Mar 15 14:32:56 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.6 2003-03-15 14:32:56 suxen_drol Exp $   * $Id: xvid_encraw.c,v 1.11.2.11 2003-03-23 04:01:32 suxen_drol Exp $
23   *   *
24   ****************************************************************************/   ****************************************************************************/
25    
# Line 66  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 75  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 87  Line 87 
87    
88  static int   ARG_STATS = 0;  static int   ARG_STATS = 0;
89  static int   ARG_DUMP = 0;  static int   ARG_DUMP = 0;
90  static int   ARG_BITRATE = 900;  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 104  Line 104 
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;  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 136  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);
# Line 154  Line 158 
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 192  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) {
                 else if (strcmp("-p", argv[i]) == 0) {  
217              ARG_PACKED = 1;              ARG_PACKED = 1;
218                  }                  } else if (strcmp("-bquant_ratio", argv[i]) == 0 && i < argc - 1) {
                 else if (strcmp("-bqr", 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) {
                 else if (strcmp("-s", argv[i]) == 0) {  
234                          ARG_STATS = 1;                          ARG_STATS = 1;
235                  }                  } else if (strcmp("-dump", argv[i]) == 0) {
                 else if (strcmp("-dump", argv[i]) == 0) {  
236                          ARG_DUMP = 1;                          ARG_DUMP = 1;
237                  }                  } else if (strcmp("-lumimasking", argv[i]) == 0) {
238                  else if (strcmp("-t", argv[i]) == 0 && i < argc - 1 ) {                          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) {
                 else if (strcmp("-m", argv[i]) == 0) {  
249                          ARG_SAVEMPEGSTREAM = 1;                          ARG_SAVEMPEGSTREAM = 1;
250                  }                  } else if (strcmp("-debug", argv[i]) == 0) {
251                  else if (strcmp("-o", argv[i]) == 0 && i < argc - 1 ) {                          ARG_DEBUG = 1;
252                    } 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 273  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 282  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 299  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 311  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 332  Line 322 
322    
323    
324          result = enc_init(use_assembler);          result = enc_init(use_assembler);
325          if (result)          if (result) {
         {  
326                  fprintf(stderr, "Encore INIT problem, return value %d\n", result);                  fprintf(stderr, "Encore INIT problem, return value %d\n", result);
327                  goto release_all;                  goto release_all;
328          }          }
# Line 349  Line 338 
338                          goto release_all;                          goto release_all;
339                  }                  }
340    
341          }          } else {
         else {  
342                  out_file = NULL;                  out_file = NULL;
343          }          }
344    
# Line 368  Line 356 
356          do {          do {
357    
358                  char *type;                  char *type;
359                  int stats[3];                  int sse[3];
360    
361          if (input_num >= ARG_MAXFRAMENR)                  if (input_num >= ARG_MAXFRAMENR) {
         {  
362              result = 1;              result = 1;
363          }          }
364    
365          if (!result)                  if (!result) {
         {  
366                      if(ARG_INPUTTYPE) {                      if(ARG_INPUTTYPE) {
367                              /* read PGM data (YUV-format) */                              /* read PGM data (YUV-format) */
368                              result = read_pgmdata(in_file, in_buffer);                              result = read_pgmdata(in_file, in_buffer);
# Line 391  Line 377 
377   ****************************************************************************/   ****************************************************************************/
378    
379                  enctime = msecond();                  enctime = msecond();
380          m4v_size = enc_main(!result?in_buffer:0, mp4_buffer, &key, &stats_type, &stats_quant, &stats_length, 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    
385                  /* Write the Frame statistics */                  /* Write the Frame statistics */
386    
387                  printf("%5d: key=%i, time(ms)=%6.1f, length=%7d",                  printf("%5d: key=%i, time=%6.0f, length=%7d",
388              !result?input_num:-1,              !result?input_num:-1,
389                             key,                             key,
390                             (float)enctime,                             (float)enctime,
# Line 422  Line 410 
410                          break;                          break;
411                  }                  }
412    
413              printf(" | type=%s quant=%2d, length=%7d", type, stats_quant, stats_length);                          printf(" | type=%s, quant=%2d, length=%7d", type, stats_quant,
414                                       stats_length);
415    
416    #define SSE2PSNR(sse, width, height) ((!(sse))?0.0f : 48.131f - 10*(float)log10((float)(sse)/((float)((width)*(height)))))
417    
418                            if (ARG_STATS) {
419                                    printf(", psnr y = %2.2f, psnr u = %2.2f, psnr v = %2.2f",
420                                               SSE2PSNR(sse[0], XDIM, YDIM),
421                                               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");                  printf("\n");
430    
431          if (m4v_size < 0) {          if (m4v_size < 0) {
# Line 439  Line 440 
440   *                       Save stream to file   *                       Save stream to file
441   ****************************************************************************/   ****************************************************************************/
442    
443                  if (m4v_size>0 && 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, output_num);                                  sprintf(filename, "%sframe%05d.m4v", filepath, output_num);
# Line 449  Line 449 
449                                  fclose(out_file);                                  fclose(out_file);
450                                  out_file = NULL;                                  out_file = NULL;
451                  output_num++;                  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 472  Line 471 
471   *         Calculate totals and averages for output, print results   *         Calculate totals and averages for output, print results
472   ****************************************************************************/   ****************************************************************************/
473    
474            printf("Tot: enctime(ms) =%7.2f,               length(bytes) = %7d\n",
475                       totalenctime, (int) totalsize);
476    
477      if (input_num > 0) {      if (input_num > 0) {
478              totalsize    /= input_num;              totalsize    /= input_num;
479              totalenctime /= input_num;              totalenctime /= input_num;
# Line 490  Line 492 
492    
493   release_all:   release_all:
494    
495          if (enc_handle)          if (enc_handle) {
         {  
496                  result = enc_stop();                  result = enc_stop();
497                  if (result)                  if (result)
498                          fprintf(stderr, "Encore RELEASE problem return value %d\n", result);                          fprintf(stderr, "Encore RELEASE problem return value %d\n",
499                                            result);
500          }          }
501    
502          if(in_file)          if(in_file)
# Line 521  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 538  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, " -p             : packed mode\n");          fprintf(stderr, "Output options:\n");
558          fprintf(stderr, " -bqr integer   : bframe quantizer ratio (default=150)\n");          fprintf(stderr, " -dump    : save decoder output\n");
559          fprintf(stderr, " -bqo integer   : bframe quantizer offset (default=100)\n");          fprintf(stderr, " -save    : save mpeg4 raw stream\n");
560          fprintf(stderr, " -f float       : target framerate (>0)\n");          fprintf(stderr, " -o string: output filename\n");
561          fprintf(stderr, " -i string      : input filename (default=stdin)\n");          fprintf(stderr, "\n");
562          fprintf(stderr, " -s             : print stats about encoded frames\n");          fprintf(stderr, "Bitrate options:\n");
563          fprintf(stderr, " -t integer     : input data type (yuv=0, pgm=1)\n");          fprintf(stderr, " -bitrate       integer: target bitrate (>0 | default=900kbit)\n");
         fprintf(stderr, " -n integer     : number of frames to encode\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");  
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 575  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 586  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 600  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 612  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 631  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 646  Line 660 
660    
661  /* sample plugin */  /* sample plugin */
662    
663  int rawenc_debug(void * handle, int opt, void * param1, void * param2)  int
664  {  rawenc_debug(void *handle,
665      switch(opt)                           int opt,
666                             void *param1,
667                             void *param2)
668      {      {
669            switch (opt) {
670      case XVID_PLG_INFO :      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 :      case XVID_PLG_CREATE :
679      case XVID_PLG_DESTROY :      case XVID_PLG_DESTROY :
680      case XVID_PLG_BEFORE :      case XVID_PLG_BEFORE :
# Line 659  Line 683 
683      case XVID_PLG_AFTER :      case XVID_PLG_AFTER :
684         {         {
685         xvid_plg_data_t * data = (xvid_plg_data_t*)param1;         xvid_plg_data_t * data = (xvid_plg_data_t*)param1;
686         printf("type=%i, quant=%i, length=%i\n", data->type, data->quant, data->length);                          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;         return 0;
697         }         }
698      }      }
# Line 670  Line 703 
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_enc_plugin_t plugins[2];      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 697  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 721  Line 755 
755    
756      xvid_enc_create.plugins = plugins;      xvid_enc_create.plugins = plugins;
757      xvid_enc_create.num_plugins = 0;      xvid_enc_create.num_plugins = 0;
758      if (ARG_STATS) {  
759          plugins[xvid_enc_create.num_plugins].func = xvid_plugin_psnr;      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;          plugins[xvid_enc_create.num_plugins].param = NULL;
782          xvid_enc_create.num_plugins++;          xvid_enc_create.num_plugins++;
783      }      }
784    
785      if (ARG_DUMP) {      if (ARG_DUMP) {
786          plugins[xvid_enc_create.num_plugins].func = xvid_plugin_dump;          plugins[xvid_enc_create.num_plugins].func = xvid_plugin_dump;
787          plugins[xvid_enc_create.num_plugins].param = NULL;          plugins[xvid_enc_create.num_plugins].param = NULL;
788          xvid_enc_create.num_plugins++;          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 758  Line 820 
820          /* Global encoder options */          /* Global encoder options */
821          xvid_enc_create.global = 0;          xvid_enc_create.global = 0;
822      if (ARG_PACKED) xvid_enc_create.global |= XVID_PACKED;      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 768  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 778  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                      int * key,                      int * key,
849                                          int *stats_type,                                          int *stats_type,
850                      int *stats_quant,                      int *stats_quant,
851                      int *stats_length,                      int *stats_length,
852                                          int stats[3])                   int sse[3])
853  {  {
854          int ret;          int ret;
855    
# Line 813  Line 878 
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 822  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 832  Line 898 
898          xvid_enc_frame.quant_inter_matrix = NULL;          xvid_enc_frame.quant_inter_matrix = NULL;
899    
900          /* Encode the frame */          /* Encode the frame */
901          ret = xvid_encore(enc_handle, XVID_ENC_ENCODE, &xvid_enc_frame, &xvid_enc_stats);          ret =
902                    xvid_encore(enc_handle, XVID_ENC_ENCODE, &xvid_enc_frame,
903                                            &xvid_enc_stats);
904    
905      *key = (xvid_enc_frame.out_flags & XVID_KEYFRAME);      *key = (xvid_enc_frame.out_flags & XVID_KEYFRAME);
906          *stats_type = xvid_enc_stats.type;          *stats_type = xvid_enc_stats.type;
907      *stats_quant = xvid_enc_stats.quant;      *stats_quant = xvid_enc_stats.quant;
908      *stats_length = xvid_enc_stats.length;      *stats_length = xvid_enc_stats.length;
909          stats[0]   = xvid_enc_stats.sse_y;          sse[0] = xvid_enc_stats.sse_y;
910          stats[1]   = xvid_enc_stats.sse_u;          sse[1] = xvid_enc_stats.sse_u;
911          stats[2]   = xvid_enc_stats.sse_v;          sse[2] = xvid_enc_stats.sse_v;
912    
913          return(ret);          return(ret);
914  }  }
Legend:
Removed from v.920  
changed lines
  Added in v.938
No admin address has been configured
 ViewVC Help
Powered by ViewVC 1.0.4