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

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

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

revision 559, Sat Sep 28 14:27:16 2002 UTC revision 729, Wed Dec 18 21:48:46 2002 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_stat.c,v 1.14 2002-09-28 14:27:15 edgomez Exp $   * $Id: xvid_stat.c,v 1.18 2002-12-18 21:48:46 edgomez Exp $
23   *   *
24   ****************************************************************************/   ****************************************************************************/
25    
# 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)
48   *  -help          : prints this help message   *  -help          : prints this help message
49   *  -quant integer : fixed quantizer (disables -b setting)   *  -quant integer : fixed quantizer (disables -b setting)
50   *  (* means default)   *  (* means default)
# Line 99  Line 100 
100  static int enc_init(int use_assembler);  static int enc_init(int use_assembler);
101  static int enc_stop();  static int enc_stop();
102  static int enc_main(unsigned char* image, unsigned char* bitstream,  static int enc_main(unsigned char* image, unsigned char* bitstream,
103                                          int *streamlength, int* frametype);                                          unsigned char* hints_buffer,
104                                            long *streamlength, long* frametype, long* hints_size);
105    
106  /* Decoder related functions */  /* Decoder related functions */
107  static int dec_stop();  static int dec_stop();
# Line 112  Line 114 
114   ****************************************************************************/   ****************************************************************************/
115    
116  static int const motion_presets[7] = {  static int const motion_presets[7] = {
117          0,                                                        // Q 0          0,                                                        /* Q 0 */
118          PMV_EARLYSTOP16,                                          // Q 1          PMV_EARLYSTOP16,                                          /* Q 1 */
119          PMV_EARLYSTOP16,                                          // Q 2          PMV_EARLYSTOP16,                                          /* Q 2 */
120          PMV_EARLYSTOP16 | PMV_HALFPELREFINE16,                    // Q 3          PMV_EARLYSTOP16 | PMV_HALFPELREFINE16,                    /* Q 3 */
121          PMV_EARLYSTOP16 | PMV_HALFPELREFINE16,                    // Q 4          PMV_EARLYSTOP16 | PMV_HALFPELREFINE16,                    /* Q 4 */
122          PMV_EARLYSTOP16 | PMV_HALFPELREFINE16 | PMV_EARLYSTOP8 |  // Q 5          PMV_EARLYSTOP16 | PMV_HALFPELREFINE16 | PMV_EARLYSTOP8 |  /* Q 5 */
123          PMV_HALFPELREFINE8,          PMV_HALFPELREFINE8,
124          PMV_EARLYSTOP16 | PMV_HALFPELREFINE16 | PMV_EXTSEARCH16 | // Q 6          PMV_EARLYSTOP16 | PMV_HALFPELREFINE16 | PMV_EXTSEARCH16 | /* Q 6 */
125          PMV_USESQUARES16 | PMV_EARLYSTOP8 | PMV_HALFPELREFINE8          PMV_USESQUARES16 | PMV_EARLYSTOP8 | PMV_HALFPELREFINE8
126  };  };
127    
128  static int const general_presets[7] = {  static int const general_presets[7] = {
129          XVID_H263QUANT,                               // Q 0          XVID_H263QUANT,                               /* Q 0 */
130          XVID_MPEGQUANT,                               // Q 1          XVID_MPEGQUANT,                               /* Q 1 */
131          XVID_H263QUANT,                               // Q 2          XVID_H263QUANT,                               /* Q 2 */
132          XVID_H263QUANT | XVID_HALFPEL,                // Q 3          XVID_H263QUANT | XVID_HALFPEL,                /* Q 3 */
133          XVID_H263QUANT | XVID_HALFPEL | XVID_INTER4V, // Q 4          XVID_H263QUANT | XVID_HALFPEL | XVID_INTER4V, /* Q 4 */
134          XVID_H263QUANT | XVID_HALFPEL | XVID_INTER4V, // Q 5          XVID_H263QUANT | XVID_HALFPEL | XVID_INTER4V, /* Q 5 */
135          XVID_H263QUANT | XVID_HALFPEL | XVID_INTER4V  // Q 6          XVID_H263QUANT | XVID_HALFPEL | XVID_INTER4V  /* Q 6 */
136  };  };
137    
138    
# Line 141  Line 143 
143  /* Maximum number of frames to encode */  /* Maximum number of frames to encode */
144  #define ABS_MAXFRAMENR 9999  #define ABS_MAXFRAMENR 9999
145    
146    /* HINTMODEs */
147    #define HINT_MODE_NONE 0
148    #define HINT_MODE_GET  1
149    #define HINT_MODE_SET  2
150    #define HINT_FILE "hints.mv"
151    
152  static int   ARG_BITRATE = 900;  static int   ARG_BITRATE = 900;
153  static int   ARG_QUANTI = 0;  static int   ARG_QUANTI = 0;
154  static int   ARG_QUALITY = 6;  static int   ARG_QUALITY = 6;
# Line 152  Line 160 
160  static int   ARG_INPUTTYPE = 0;  static int   ARG_INPUTTYPE = 0;
161  static int   ARG_SAVEDECOUTPUT = 0;  static int   ARG_SAVEDECOUTPUT = 0;
162  static int   ARG_SAVEMPEGSTREAM = 0;  static int   ARG_SAVEMPEGSTREAM = 0;
163    static int   ARG_HINTMODE = HINT_MODE_NONE;
164  static int   XDIM = 0;  static int   XDIM = 0;
165  static int   YDIM = 0;  static int   YDIM = 0;
166  #define IMAGE_SIZE(x,y) ((x)*(y)*3/2)  #define IMAGE_SIZE(x,y) ((x)*(y)*3/2)
# Line 159  Line 168 
168  #define MAX(A,B) ( ((A)>(B)) ? (A) : (B) )  #define MAX(A,B) ( ((A)>(B)) ? (A) : (B) )
169  #define SMALL_EPS 1e-10  #define SMALL_EPS 1e-10
170    
171    #define LONG_PACK(a,b,c,d) ((long) (((long)(a))<<24) | (((long)(b))<<16) | \
172                                       (((long)(c))<<8)  |((long)(d)))
173    
174    #define SWAP(a) ( (((a)&0x000000ff)<<24) | (((a)&0x0000ff00)<<8) | \
175                      (((a)&0x00ff0000)>>8)  | (((a)&0xff000000)>>24) )
176    
177  /****************************************************************************  /****************************************************************************
178   *                     Nasty global vars ;-)   *                     Nasty global vars ;-)
179   ***************************************************************************/   ***************************************************************************/
# Line 183  Line 198 
198          unsigned char *divx_buffer = NULL;          unsigned char *divx_buffer = NULL;
199          unsigned char *in_buffer = NULL;          unsigned char *in_buffer = NULL;
200          unsigned char *out_buffer = NULL;          unsigned char *out_buffer = NULL;
201            unsigned char *hints_buffer = NULL;
202    
203          double enctime,dectime;          double enctime,dectime;
204          double totalenctime=0.;          double totalenctime=0.;
205          double totaldectime=0.;          double totaldectime=0.;
206    
207          long totalsize=0;          long totalsize=0;
208            long hints_size = 0;
209          int status;          int status;
210            int bigendian = 0;
211    
212          int m4v_size;          long m4v_size = 0;
213          int frame_type[ABS_MAXFRAMENR];          long frame_type[ABS_MAXFRAMENR];
214          int Iframes=0, Pframes=0, use_assembler=0;          int Iframes=0, Pframes=0, use_assembler=0;
215          double framepsnr[ABS_MAXFRAMENR];          double framepsnr[ABS_MAXFRAMENR];
216    
217          double Ipsnr=0.,Imaxpsnr=0.,Iminpsnr=999.,Ivarpsnr=0.;          double Ipsnr=0.,Imaxpsnr=0.,Iminpsnr=999.,Ivarpsnr=0.;
218          double Ppsnr=0.,Pmaxpsnr=0.,Pminpsnr=999.,Pvarpsnr=0.;          double Ppsnr=0.,Pmaxpsnr=0.,Pminpsnr=999.,Pvarpsnr=0.;
         double Bpsnr=0.,Bmaxpsnr=0.,Bminpsnr=999.,Bvarpsnr=0.;  
219    
220          char filename[256];          char filename[256];
221    
222          FILE *filehandle;          FILE *filehandle;
223          FILE *in_file = stdin;          FILE *in_file = stdin;
224            FILE *hints_file = NULL;
225    
226          printf("xvid_stat - XviD core library test program ");          printf("xvid_stat - XviD core library test program ");
227          printf("written by Christoph Lampert 2002\n\n");          printf("written by Christoph Lampert 2002\n\n");
# Line 261  Line 279 
279                          i++;                          i++;
280                          ARG_SAVEMPEGSTREAM = atoi(argv[i]);                          ARG_SAVEMPEGSTREAM = atoi(argv[i]);
281                  }                  }
282                    else if (strcmp("-mv", argv[i]) == 0 && i < argc - 1 ) {
283                            i++;
284                            ARG_HINTMODE = atoi(argv[i]);
285                    }
286                  else if (strcmp("-help", argv[i])) {                  else if (strcmp("-help", argv[i])) {
287                          usage();                          usage();
288                          return(0);                          return(0);
# Line 301  Line 323 
323                  return -1;                  return -1;
324          }          }
325    
326            if ( ARG_HINTMODE != HINT_MODE_NONE &&
327                     ARG_HINTMODE != HINT_MODE_GET &&
328                     ARG_HINTMODE != HINT_MODE_SET)
329                    ARG_HINTMODE = HINT_MODE_NONE;
330    
331            if( ARG_HINTMODE != HINT_MODE_NONE) {
332                    char *rights = "rb";
333    
334                    /*
335                     * If we are getting hints from core, we will have to write them to
336                     * hint file
337                     */
338                    if(ARG_HINTMODE == HINT_MODE_GET)
339                            rights = "w+b";
340    
341                    /* Open the hint file */
342                    hints_file = fopen(HINT_FILE, rights);
343                    if(hints_file == NULL) {
344                            fprintf(stderr, "Error opening input file %s\n", HINT_FILE);
345                            return -1;
346                    }
347    
348                    /* Allocate hint memory space, we will be using rawhints */
349                    /* NB : Hope 1Mb is enough */
350                    if((hints_buffer = malloc(1024*1024)) == NULL) {
351                            fprintf(stderr, "Memory allocation error\n");
352                            return -1;
353                    }
354    
355            }
356    
357          if ( ARG_INPUTFILE == NULL || strcmp(ARG_INPUTFILE, "stdin") == 0) {          if ( ARG_INPUTFILE == NULL || strcmp(ARG_INPUTFILE, "stdin") == 0) {
358                  in_file = stdin;                  in_file = stdin;
359          }          }
# Line 355  Line 408 
408                  goto release_all;                  goto release_all;
409          }          }
410    
411            totalsize = LONG_PACK('M','P','4','U');
412            if(*((char *)(&totalsize)) == 'M')
413                    bigendian = 1;
414            else
415                    bigendian = 0;
416            totalsize = 0;
417    
418  /*****************************************************************************  /*****************************************************************************
419   *                            Main loop   *                            Main loop
# Line 363  Line 422 
422          do {          do {
423    
424                  if (ARG_INPUTTYPE)                  if (ARG_INPUTTYPE)
425                          status = read_pgmdata(in_file, in_buffer);      // read PGM data (YUV-format)                          status = read_pgmdata(in_file, in_buffer);      /* read PGM data (YUV-format) */
426                  else                  else
427                          status = read_yuvdata(in_file, in_buffer);      // read raw data (YUV-format)                          status = read_yuvdata(in_file, in_buffer);      /* read raw data (YUV-format) */
428    
429                  if (status)                  if (status)
430                  {                  {
# Line 391  Line 450 
450   */   */
451    
452  /*****************************************************************************  /*****************************************************************************
453     *                       Read hints from file
454     ****************************************************************************/
455    
456                    if(ARG_HINTMODE == HINT_MODE_SET) {
457                            fread(&hints_size, 1, sizeof(long), hints_file);
458                            hints_size = (!bigendian)?SWAP(hints_size):hints_size;
459                            fread(hints_buffer, 1, hints_size, hints_file);
460                    }
461    
462    /*****************************************************************************
463   *                       Encode and decode this frame   *                       Encode and decode this frame
464   ****************************************************************************/   ****************************************************************************/
465    
466                  enctime = msecond();                  enctime = msecond();
467                  status = enc_main(in_buffer, divx_buffer, &m4v_size, &frame_type[filenr]);                  status = enc_main(in_buffer, divx_buffer, hints_buffer,
468                                                      &m4v_size, &frame_type[filenr], &hints_size);
469                  enctime = msecond() - enctime;                  enctime = msecond() - enctime;
470    
471                  totalenctime += enctime;                  totalenctime += enctime;
# Line 404  Line 474 
474                  printf("Frame %5d: intra %1d, enctime=%6.1f ms, size=%6d bytes ",                  printf("Frame %5d: intra %1d, enctime=%6.1f ms, size=%6d bytes ",
475                             (int)filenr, (int)frame_type[filenr], (float)enctime, (int)m4v_size);                             (int)filenr, (int)frame_type[filenr], (float)enctime, (int)m4v_size);
476    
477    /*****************************************************************************
478     *                       Save hints to file
479     ****************************************************************************/
480    
481                    if(ARG_HINTMODE == HINT_MODE_GET) {
482                            hints_size = (!bigendian)?SWAP(hints_size):hints_size;
483                            fwrite(&hints_size, 1, sizeof(long), hints_file);
484                            hints_size = (!bigendian)?SWAP(hints_size):hints_size;
485                            fwrite(hints_buffer, 1, hints_size, hints_file);
486                    }
487    
488    /*****************************************************************************
489     *                       Save stream to file
490     ****************************************************************************/
491    
492                  if (ARG_SAVEMPEGSTREAM)                  if (ARG_SAVEMPEGSTREAM)
493                  {                  {
494                          sprintf(filename, "%sframe%05d.m4v", filepath, filenr);                          sprintf(filename, "%sframe%05d.m4v", filepath, filenr);
# Line 500  Line 585 
585          printf("Avg. Q%1d %2s ",ARG_QUALITY, (ARG_QUANTI ? " q" : "br"));          printf("Avg. Q%1d %2s ",ARG_QUALITY, (ARG_QUANTI ? " q" : "br"));
586          printf("%04d ",(ARG_QUANTI)?ARG_QUANTI:ARG_BITRATE);          printf("%04d ",(ARG_QUANTI)?ARG_QUANTI:ARG_BITRATE);
587          printf("( %.2f bpp) ", (double)ARG_BITRATE*1000/XDIM/YDIM/ARG_FRAMERATE);          printf("( %.2f bpp) ", (double)ARG_BITRATE*1000/XDIM/YDIM/ARG_FRAMERATE);
588          printf("size %6d ",totalsize);          printf("size %6d ", (int)totalsize);
589          printf("( %4d kbps ",(int)(totalsize*8*ARG_FRAMERATE/1000));          printf("( %4d kbps ",(int)(totalsize*8*ARG_FRAMERATE/1000));
590          printf("/ %.2f bpp) ",(double)totalsize*8/XDIM/YDIM);          printf("/ %.2f bpp) ",(double)totalsize*8/XDIM/YDIM);
591          printf("enc: %6.1f fps, dec: %6.1f fps \n",1000/totalenctime, 1000/totaldectime);          printf("enc: %6.1f fps, dec: %6.1f fps \n",1000/totalenctime, 1000/totaldectime);
# Line 818  Line 903 
903  }  }
904    
905  static int enc_main(unsigned char* image, unsigned char* bitstream,  static int enc_main(unsigned char* image, unsigned char* bitstream,
906                                          int *streamlength, int* frametype)                                          unsigned char* hints_buffer,
907                                            long *streamlength, long* frametype, long* hints_size)
908  {  {
909          int xerr;          int xerr;
910    
# Line 826  Line 912 
912          XVID_ENC_STATS xstats;          XVID_ENC_STATS xstats;
913    
914          xframe.bitstream = bitstream;          xframe.bitstream = bitstream;
915          xframe.length = -1;     // this is written by the routine          xframe.length = -1;     /* this is written by the routine */
916    
917          xframe.image = image;          xframe.image = image;
918          xframe.colorspace = XVID_CSP_YV12;      // defined in <xvid.h>          xframe.colorspace = XVID_CSP_YV12;      /* defined in <xvid.h> */
919    
920          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) */
921    
922          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) */
923    
924          xframe.motion = motion_presets[ARG_QUALITY];          xframe.motion = motion_presets[ARG_QUALITY];
925          xframe.general = general_presets[ARG_QUALITY];          xframe.general = general_presets[ARG_QUALITY];
926          xframe.quant_intra_matrix = xframe.quant_inter_matrix = NULL;          xframe.quant_intra_matrix = xframe.quant_inter_matrix = NULL;
927    
928            xframe.hint.hintstream = hints_buffer;
929    
930            if(ARG_HINTMODE == HINT_MODE_SET) {
931                    xframe.hint.hintlength = *hints_size;
932                    xframe.hint.rawhints = 0;
933                    xframe.general |= XVID_HINTEDME_SET;
934            }
935    
936            if(ARG_HINTMODE == HINT_MODE_GET) {
937                    xframe.hint.rawhints = 0;
938                    xframe.general |= XVID_HINTEDME_GET;
939            }
940    
941          xerr = xvid_encore(enc_handle, XVID_ENC_ENCODE, &xframe, &xstats);          xerr = xvid_encore(enc_handle, XVID_ENC_ENCODE, &xframe, &xstats);
942    
943            if(ARG_HINTMODE == HINT_MODE_GET)
944                    *hints_size = xframe.hint.hintlength;
945    
946          /*          /*
947           * 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
948           * 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 895  Line 997 
997          xframe.length = m4v_size;          xframe.length = m4v_size;
998          xframe.image = out_buffer;          xframe.image = out_buffer;
999          xframe.stride = XDIM;          xframe.stride = XDIM;
1000          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) */
1001    
1002          xerr = xvid_decore(dec_handle, XVID_DEC_DECODE, &xframe, NULL);          xerr = xvid_decore(dec_handle, XVID_DEC_DECODE, &xframe, NULL);
1003    

Legend:
Removed from v.559  
changed lines
  Added in v.729

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