[cvs] / xvidcore / examples / xvid_stat.c Repository:
ViewVC logotype

Diff of /xvidcore/examples/xvid_stat.c

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

revision 1.4.2.2, Sun Feb 9 06:46:54 2003 UTC revision 1.11, Sat Sep 21 11:55:53 2002 UTC
# Line 44  Line 44 
44   *  -q integer     : quality ([0..5])   *  -q integer     : quality ([0..5])
45   *  -d boolean     : save decoder output (0 False*, !=0 True)   *  -d boolean     : save decoder output (0 False*, !=0 True)
46   *  -m boolean     : save mpeg4 raw stream (0 False*, !=0 True)   *  -m boolean     : save mpeg4 raw stream (0 False*, !=0 True)
47   *  -mv integer    : Hinted Motion Estimation (0 none, 1 get hints, 2 set hints)   *  -h, -help      : prints this help message
  *  -help          : prints this help message  
48   *  -quant integer : fixed quantizer (disables -b setting)   *  -quant integer : fixed quantizer (disables -b setting)
49   *  (* means default)   *  (* means default)
50   *   *
# Line 66  Line 65 
65  #include <stdlib.h>  #include <stdlib.h>
66  #include <string.h>  #include <string.h>
67  #include <math.h>  #include <math.h>
 #ifndef WIN32  
 #include <sys/time.h>  
 #else  
68  #include <time.h>  #include <time.h>
 #endif  
69    
70  #include "xvid.h"  #include "../src/xvid.h"
71    
72  /****************************************************************************  /****************************************************************************
73   *                               Prototypes   *                               Prototypes
# Line 100  Line 95 
95  static int enc_init(int use_assembler);  static int enc_init(int use_assembler);
96  static int enc_stop();  static int enc_stop();
97  static int enc_main(unsigned char* image, unsigned char* bitstream,  static int enc_main(unsigned char* image, unsigned char* bitstream,
98                                          unsigned char* hints_buffer,                                          int *streamlength, int* frametype);
                                         long *streamlength, long* frametype, long* hints_size);  
99    
100  /* Decoder related functions */  /* Decoder related functions */
101  static int dec_stop();  static int dec_stop();
# Line 114  Line 108 
108   ****************************************************************************/   ****************************************************************************/
109    
110  static int const motion_presets[7] = {  static int const motion_presets[7] = {
111          0,                                                        /* Q 0 */          0,                                                        // Q 0
112          PMV_EARLYSTOP16,                                          /* Q 1 */          PMV_EARLYSTOP16,                                          // Q 1
113          PMV_EARLYSTOP16,                                          /* Q 2 */          PMV_EARLYSTOP16,                                          // Q 2
114          PMV_EARLYSTOP16 | PMV_HALFPELREFINE16,                    /* Q 3 */          PMV_EARLYSTOP16 | PMV_HALFPELREFINE16,                    // Q 3
115          PMV_EARLYSTOP16 | PMV_HALFPELREFINE16,                    /* Q 4 */          PMV_EARLYSTOP16 | PMV_HALFPELREFINE16,                    // Q 4
116          PMV_EARLYSTOP16 | PMV_HALFPELREFINE16 | PMV_EARLYSTOP8 |  /* Q 5 */          PMV_EARLYSTOP16 | PMV_HALFPELREFINE16 | PMV_EARLYSTOP8 |  // Q 5
117          PMV_HALFPELREFINE8,          PMV_HALFPELREFINE8,
118          PMV_EARLYSTOP16 | PMV_HALFPELREFINE16 | PMV_EXTSEARCH16 | /* Q 6 */          PMV_EARLYSTOP16 | PMV_HALFPELREFINE16 | PMV_EXTSEARCH16 | // Q 6
119          PMV_USESQUARES16 | PMV_EARLYSTOP8 | PMV_HALFPELREFINE8          PMV_USESQUARES16 | PMV_EARLYSTOP8 | PMV_HALFPELREFINE8
120  };  };
121    
122  static int const general_presets[7] = {  static int const general_presets[7] = {
123          XVID_H263QUANT,                               /* Q 0 */          XVID_H263QUANT,                               // Q 0
124          XVID_MPEGQUANT,                               /* Q 1 */          XVID_MPEGQUANT,                               // Q 1
125          XVID_H263QUANT,                               /* Q 2 */          XVID_H263QUANT,                               // Q 2
126          XVID_H263QUANT | XVID_HALFPEL,                /* Q 3 */          XVID_H263QUANT | XVID_HALFPEL,                // Q 3
127          XVID_H263QUANT | XVID_HALFPEL | XVID_INTER4V, /* Q 4 */          XVID_H263QUANT | XVID_HALFPEL | XVID_INTER4V, // Q 4
128          XVID_H263QUANT | XVID_HALFPEL | XVID_INTER4V, /* Q 5 */          XVID_H263QUANT | XVID_HALFPEL | XVID_INTER4V, // Q 5
129          XVID_H263QUANT | XVID_HALFPEL | XVID_INTER4V  /* Q 6 */          XVID_H263QUANT | XVID_HALFPEL | XVID_INTER4V  // Q 6
130  };  };
131    
132    
# Line 143  Line 137 
137  /* Maximum number of frames to encode */  /* Maximum number of frames to encode */
138  #define ABS_MAXFRAMENR 9999  #define ABS_MAXFRAMENR 9999
139    
 /* HINTMODEs */  
 #define HINT_MODE_NONE 0  
 #define HINT_MODE_GET  1  
 #define HINT_MODE_SET  2  
 #define HINT_FILE "hints.mv"  
   
140  static int   ARG_BITRATE = 900;  static int   ARG_BITRATE = 900;
141  static int   ARG_QUANTI = 0;  static int   ARG_QUANTI = 0;
142  static int   ARG_QUALITY = 6;  static int   ARG_QUALITY = 6;
# Line 160  Line 148 
148  static int   ARG_INPUTTYPE = 0;  static int   ARG_INPUTTYPE = 0;
149  static int   ARG_SAVEDECOUTPUT = 0;  static int   ARG_SAVEDECOUTPUT = 0;
150  static int   ARG_SAVEMPEGSTREAM = 0;  static int   ARG_SAVEMPEGSTREAM = 0;
 static int   ARG_HINTMODE = HINT_MODE_NONE;  
151  static int   XDIM = 0;  static int   XDIM = 0;
152  static int   YDIM = 0;  static int   YDIM = 0;
153  #define IMAGE_SIZE(x,y) ((x)*(y)*3/2)  #define IMAGE_SIZE(x,y) ((x)*(y)*3/2)
# Line 168  Line 155 
155  #define MAX(A,B) ( ((A)>(B)) ? (A) : (B) )  #define MAX(A,B) ( ((A)>(B)) ? (A) : (B) )
156  #define SMALL_EPS 1e-10  #define SMALL_EPS 1e-10
157    
 #define LONG_PACK(a,b,c,d) ((long) (((long)(a))<<24) | (((long)(b))<<16) | \  
                                    (((long)(c))<<8)  |((long)(d)))  
   
 #define SWAP(a) ( (((a)&0x000000ff)<<24) | (((a)&0x0000ff00)<<8) | \  
                   (((a)&0x00ff0000)>>8)  | (((a)&0xff000000)>>24) )  
   
158  /****************************************************************************  /****************************************************************************
159   *                     Nasty global vars ;-)   *                     Nasty global vars ;-)
160   ***************************************************************************/   ***************************************************************************/
# Line 198  Line 179 
179          unsigned char *divx_buffer = NULL;          unsigned char *divx_buffer = NULL;
180          unsigned char *in_buffer = NULL;          unsigned char *in_buffer = NULL;
181          unsigned char *out_buffer = NULL;          unsigned char *out_buffer = NULL;
         unsigned char *hints_buffer = NULL;  
182    
183          double enctime,dectime;          double enctime,dectime;
184          double totalenctime=0.;          double totalenctime=0.;
185          double totaldectime=0.;          double totaldectime=0.;
186    
187          long totalsize = 0;          long totalsize = 0;
         long hints_size = 0;  
188          int status;          int status;
         int bigendian = 0;  
189    
190          long m4v_size = 0;          int m4v_size;
191          long frame_type[ABS_MAXFRAMENR];          int frame_type[ABS_MAXFRAMENR];
192          int Iframes=0, Pframes=0, use_assembler=0;          int Iframes=0, Pframes=0, use_assembler=0;
193          double framepsnr[ABS_MAXFRAMENR];          double framepsnr[ABS_MAXFRAMENR];
194    
195          double Ipsnr=0.,Imaxpsnr=0.,Iminpsnr=999.,Ivarpsnr=0.;          double Ipsnr=0.,Imaxpsnr=0.,Iminpsnr=999.,Ivarpsnr=0.;
196          double Ppsnr=0.,Pmaxpsnr=0.,Pminpsnr=999.,Pvarpsnr=0.;          double Ppsnr=0.,Pmaxpsnr=0.,Pminpsnr=999.,Pvarpsnr=0.;
197            double Bpsnr=0.,Bmaxpsnr=0.,Bminpsnr=999.,Bvarpsnr=0.;
198    
199          char filename[256];          char filename[256];
200    
201          FILE *filehandle;          FILE *filehandle;
202          FILE *in_file = stdin;          FILE *in_file = stdin;
         FILE *hints_file = NULL;  
203    
204          printf("xvid_stat - XviD core library test program ");          printf("xvid_stat - XviD core library test program ");
205          printf("written by Christoph Lampert 2002\n\n");          printf("written by Christoph Lampert 2002\n\n");
# Line 279  Line 257 
257                          i++;                          i++;
258                          ARG_SAVEMPEGSTREAM = atoi(argv[i]);                          ARG_SAVEMPEGSTREAM = atoi(argv[i]);
259                  }                  }
260                  else if (strcmp("-mv", argv[i]) == 0 && i < argc - 1 ) {                  else if (strcmp("-h", argv[i]) == 0 || strcmp("-help", argv[i])) {
                         i++;  
                         ARG_HINTMODE = atoi(argv[i]);  
                 }  
                 else if (strcmp("-help", argv[i])) {  
261                          usage();                          usage();
262                          return(0);                          return(0);
263                  }                  }
# Line 299  Line 273 
273   ****************************************************************************/   ****************************************************************************/
274    
275          if (XDIM <= 0 || XDIM >= 2048 || YDIM <=0 || YDIM >= 2048 ) {          if (XDIM <= 0 || XDIM >= 2048 || YDIM <=0 || YDIM >= 2048 ) {
276                  fprintf(stderr, "Trying to retreive width and height from PGM header\n");                  fprintf(stderr,
277                                    "Trying to retreive width and height from PGM header\n",
278                                    XDIM,
279                                    YDIM);
280                  ARG_INPUTTYPE = 1; /* pgm */                  ARG_INPUTTYPE = 1; /* pgm */
281          }          }
282    
# Line 314  Line 291 
291          }          }
292    
293          if ( ARG_FRAMERATE <= 0) {          if ( ARG_FRAMERATE <= 0) {
294                  fprintf(stderr,"Wrong Framerate %s \n",argv[5]);                  fprintf(stderr,"Wrong Fraterate %s \n",argv[5]);
295                  return -1;                  return -1;
296          }          }
297    
# Line 323  Line 300 
300                  return -1;                  return -1;
301          }          }
302    
         if ( ARG_HINTMODE != HINT_MODE_NONE &&  
                  ARG_HINTMODE != HINT_MODE_GET &&  
                  ARG_HINTMODE != HINT_MODE_SET)  
                 ARG_HINTMODE = HINT_MODE_NONE;  
   
         if( ARG_HINTMODE != HINT_MODE_NONE) {  
                 char *rights = "rb";  
   
                 /*  
                  * If we are getting hints from core, we will have to write them to  
                  * hint file  
                  */  
                 if(ARG_HINTMODE == HINT_MODE_GET)  
                         rights = "w+b";  
   
                 /* Open the hint file */  
                 hints_file = fopen(HINT_FILE, rights);  
                 if(hints_file == NULL) {  
                         fprintf(stderr, "Error opening input file %s\n", HINT_FILE);  
                         return -1;  
                 }  
   
                 /* Allocate hint memory space, we will be using rawhints */  
                 /* NB : Hope 1Mb is enough */  
                 if((hints_buffer = malloc(1024*1024)) == NULL) {  
                         fprintf(stderr, "Memory allocation error\n");  
                         return -1;  
                 }  
   
         }  
   
303          if ( ARG_INPUTFILE == NULL || strcmp(ARG_INPUTFILE, "stdin") == 0) {          if ( ARG_INPUTFILE == NULL || strcmp(ARG_INPUTFILE, "stdin") == 0) {
304                  in_file = stdin;                  in_file = stdin;
305          }          }
# Line 408  Line 354 
354                  goto release_all;                  goto release_all;
355          }          }
356    
         totalsize = LONG_PACK('M','P','4','U');  
         if(*((char *)(&totalsize)) == 'M')  
                 bigendian = 1;  
         else  
                 bigendian = 0;  
         totalsize = 0;  
357    
358  /*****************************************************************************  /*****************************************************************************
359   *                            Main loop   *                            Main loop
# Line 422  Line 362 
362          do {          do {
363    
364                  if (ARG_INPUTTYPE)                  if (ARG_INPUTTYPE)
365                          status = read_pgmdata(in_file, in_buffer);      /* read PGM data (YUV-format) */                          status = read_pgmdata(in_file, in_buffer);      // read PGM data (YUV-format)
366                  else                  else
367                          status = read_yuvdata(in_file, in_buffer);      /* read raw data (YUV-format) */                          status = read_yuvdata(in_file, in_buffer);      // read raw data (YUV-format)
368    
369                  if (status)                  if (status)
370                  {                  {
# Line 450  Line 390 
390   */   */
391    
392  /*****************************************************************************  /*****************************************************************************
  *                       Read hints from file  
  ****************************************************************************/  
   
                 if(ARG_HINTMODE == HINT_MODE_SET) {  
                         fread(&hints_size, 1, sizeof(long), hints_file);  
                         hints_size = (!bigendian)?SWAP(hints_size):hints_size;  
                         fread(hints_buffer, 1, hints_size, hints_file);  
                 }  
   
 /*****************************************************************************  
393   *                       Encode and decode this frame   *                       Encode and decode this frame
394   ****************************************************************************/   ****************************************************************************/
395    
396                  enctime = msecond();                  enctime = msecond();
397                  status = enc_main(in_buffer, divx_buffer, hints_buffer,                  status = enc_main(in_buffer, divx_buffer, &m4v_size, &frame_type[filenr]);
                                                   &m4v_size, &frame_type[filenr], &hints_size);  
398                  enctime = msecond() - enctime;                  enctime = msecond() - enctime;
399    
400                  totalenctime += enctime;                  totalenctime += enctime;
# Line 474  Line 403 
403                  printf("Frame %5d: intra %1d, enctime=%6.1f ms, size=%6d bytes ",                  printf("Frame %5d: intra %1d, enctime=%6.1f ms, size=%6d bytes ",
404                             (int)filenr, (int)frame_type[filenr], (float)enctime, (int)m4v_size);                             (int)filenr, (int)frame_type[filenr], (float)enctime, (int)m4v_size);
405    
 /*****************************************************************************  
  *                       Save hints to file  
  ****************************************************************************/  
   
                 if(ARG_HINTMODE == HINT_MODE_GET) {  
                         hints_size = (!bigendian)?SWAP(hints_size):hints_size;  
                         fwrite(&hints_size, 1, sizeof(long), hints_file);  
                         hints_size = (!bigendian)?SWAP(hints_size):hints_size;  
                         fwrite(hints_buffer, 1, hints_size, hints_file);  
                 }  
   
 /*****************************************************************************  
  *                       Save stream to file  
  ****************************************************************************/  
   
406                  if (ARG_SAVEMPEGSTREAM)                  if (ARG_SAVEMPEGSTREAM)
407                  {                  {
408                          sprintf(filename, "%sframe%05d.m4v", filepath, filenr);                          sprintf(filename, "%sframe%05d.m4v", filepath, filenr);
# Line 585  Line 499 
499          printf("Avg. Q%1d %2s ",ARG_QUALITY, (ARG_QUANTI ? " q" : "br"));          printf("Avg. Q%1d %2s ",ARG_QUALITY, (ARG_QUANTI ? " q" : "br"));
500          printf("%04d ",(ARG_QUANTI)?ARG_QUANTI:ARG_BITRATE);          printf("%04d ",(ARG_QUANTI)?ARG_QUANTI:ARG_BITRATE);
501          printf("( %.2f bpp) ", (double)ARG_BITRATE*1000/XDIM/YDIM/ARG_FRAMERATE);          printf("( %.2f bpp) ", (double)ARG_BITRATE*1000/XDIM/YDIM/ARG_FRAMERATE);
502          printf("size %6d ", (int)totalsize);          printf("size %6d ",totalsize);
503          printf("( %4d kbps ",(int)(totalsize*8*ARG_FRAMERATE/1000));          printf("( %4d kbps ",(int)(totalsize*8*ARG_FRAMERATE/1000));
504          printf("/ %.2f bpp) ",(double)totalsize*8/XDIM/YDIM);          printf("/ %.2f bpp) ",(double)totalsize*8/XDIM/YDIM);
505          printf("enc: %6.1f fps, dec: %6.1f fps \n",1000/totalenctime, 1000/totaldectime);          printf("enc: %6.1f fps, dec: %6.1f fps \n",CLOCKS_PER_SEC/(totalenctime*1000), CLOCKS_PER_SEC/(totaldectime*1000));
506          printf("PSNR P(%d): %5.2f ( %5.2f , %5.2f ; %5.4f ) ",Pframes,Ppsnr,Pminpsnr,Pmaxpsnr,sqrt(Pvarpsnr/filenr));          printf("PSNR P(%d): %5.2f ( %5.2f , %5.2f ; %5.4f ) ",Pframes,Ppsnr,Pminpsnr,Pmaxpsnr,sqrt(Pvarpsnr/filenr));
507          printf("I(%d): %5.2f ( %5.2f , %5.2f ; %5.4f ) ",Iframes,Ipsnr,Iminpsnr,Imaxpsnr,sqrt(Ivarpsnr/filenr));          printf("I(%d): %5.2f ( %5.2f , %5.2f ; %5.4f ) ",Iframes,Ipsnr,Iminpsnr,Imaxpsnr,sqrt(Ivarpsnr/filenr));
508          printf("\n");          printf("\n");
# Line 637  Line 551 
551  /* Return time elapsed time in miliseconds since the program started */  /* Return time elapsed time in miliseconds since the program started */
552  static double msecond()  static double msecond()
553  {  {
 #ifndef WIN32  
         struct timeval  tv;  
         gettimeofday(&tv, 0);  
         return tv.tv_sec*1.0e3 + tv.tv_usec * 1.0e-3;  
 #else  
554          clock_t clk;          clock_t clk;
555    
556          clk = clock();          clk = clock();
557    
558          return clk * 1000 / CLOCKS_PER_SEC;          return clk * 1000 / CLOCKS_PER_SEC;
559  #endif  
560  }  }
561    
562    
# Line 717  Line 628 
628          fprintf(stderr, " -q integer     : quality ([0..5])\n");          fprintf(stderr, " -q integer     : quality ([0..5])\n");
629          fprintf(stderr, " -d boolean     : save decoder output (0 False*, !=0 True)\n");          fprintf(stderr, " -d boolean     : save decoder output (0 False*, !=0 True)\n");
630          fprintf(stderr, " -m boolean     : save mpeg4 raw stream (0 False*, !=0 True)\n");          fprintf(stderr, " -m boolean     : save mpeg4 raw stream (0 False*, !=0 True)\n");
631          fprintf(stderr, " -help          : prints this help message\n");          fprintf(stderr, " -h, -help      : prints this help message\n");
632          fprintf(stderr, " -quant integer : fixed quantizer (disables -b setting)\n");          fprintf(stderr, " -quant integer : fixed quantizer (disables -b setting)\n");
633          fprintf(stderr, " (* means default)\n");          fprintf(stderr, " (* means default)\n");
634    
# Line 790  Line 701 
701  static int read_yuvdata(FILE* handle, unsigned char *image)  static int read_yuvdata(FILE* handle, unsigned char *image)
702  {  {
703    
704          if (fread(image, 1, IMAGE_SIZE(XDIM, YDIM), handle) != (unsigned int)IMAGE_SIZE(XDIM, YDIM))          if (fread(image, 1, IMAGE_SIZE(XDIM, YDIM), handle) != IMAGE_SIZE(XDIM, YDIM))
705                  return 1;                  return 1;
706          else          else
707                  return 0;                  return 0;
# Line 903  Line 814 
814  }  }
815    
816  static int enc_main(unsigned char* image, unsigned char* bitstream,  static int enc_main(unsigned char* image, unsigned char* bitstream,
817                                          unsigned char* hints_buffer,                                          int *streamlength, int* frametype)
                                         long *streamlength, long* frametype, long* hints_size)  
818  {  {
819          int xerr;          int xerr;
820    
# Line 912  Line 822 
822          XVID_ENC_STATS xstats;          XVID_ENC_STATS xstats;
823    
824          xframe.bitstream = bitstream;          xframe.bitstream = bitstream;
825          xframe.length = -1;     /* this is written by the routine */          xframe.length = -1;     // this is written by the routine
826    
827          xframe.image = image;          xframe.image = image;
828          xframe.colorspace = XVID_CSP_YV12;      /* defined in <xvid.h> */          xframe.colorspace = XVID_CSP_YV12;      // defined in <xvid.h>
829    
830          xframe.intra = -1; /* let the codec decide between I-frame (1) and P-frame (0) */          xframe.intra = -1; // let the codec decide between I-frame (1) and P-frame (0)
831    
832          xframe.quant = ARG_QUANTI;      /* is quant != 0, use a fixed quant (and ignore bitrate) */          xframe.quant = ARG_QUANTI;      // is quant != 0, use a fixed quant (and ignore bitrate)
833    
834          xframe.motion = motion_presets[ARG_QUALITY];          xframe.motion = motion_presets[ARG_QUALITY];
835          xframe.general = general_presets[ARG_QUALITY];          xframe.general = general_presets[ARG_QUALITY];
836          xframe.quant_intra_matrix = xframe.quant_inter_matrix = NULL;          xframe.quant_intra_matrix = xframe.quant_inter_matrix = NULL;
837    
         xframe.hint.hintstream = hints_buffer;  
   
         if(ARG_HINTMODE == HINT_MODE_SET) {  
                 xframe.hint.hintlength = *hints_size;  
                 xframe.hint.rawhints = 0;  
                 xframe.general |= XVID_HINTEDME_SET;  
         }  
   
         if(ARG_HINTMODE == HINT_MODE_GET) {  
                 xframe.hint.rawhints = 0;  
                 xframe.general |= XVID_HINTEDME_GET;  
         }  
   
838          xerr = xvid_encore(enc_handle, XVID_ENC_ENCODE, &xframe, &xstats);          xerr = xvid_encore(enc_handle, XVID_ENC_ENCODE, &xframe, &xstats);
839    
         if(ARG_HINTMODE == HINT_MODE_GET)  
                 *hints_size = xframe.hint.hintlength;  
   
840          /*          /*
841           * This is statictical data, e.g. for 2-pass. If you are not           * This is statictical data, e.g. for 2-pass. If you are not
842           * interested in any of this, you can use NULL instead of &xstats           * interested in any of this, you can use NULL instead of &xstats
# Line 997  Line 891 
891          xframe.length = m4v_size;          xframe.length = m4v_size;
892          xframe.image = out_buffer;          xframe.image = out_buffer;
893          xframe.stride = XDIM;          xframe.stride = XDIM;
894          xframe.colorspace = XVID_CSP_YV12;             /* XVID_CSP_USER is fastest (no memcopy involved) */          xframe.colorspace = XVID_CSP_YV12;             // XVID_CSP_USER is fastest (no memcopy involved)
895    
896          xerr = xvid_decore(dec_handle, XVID_DEC_DECODE, &xframe, NULL);          xerr = xvid_decore(dec_handle, XVID_DEC_DECODE, &xframe, NULL);
897    
Legend:
Removed from v.1.4.2.2  
changed lines
  Added in v.1.11
No admin address has been configured
 ViewVC Help
Powered by ViewVC 1.0.4