[svn] / trunk / xvidcore / examples / xvid_decraw.c Repository:
ViewVC logotype

Diff of /trunk/xvidcore/examples/xvid_decraw.c

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

revision 376, Sat Aug 17 20:03:36 2002 UTC revision 1928, Tue Dec 28 19:19:57 2010 UTC
# Line 1  Line 1 
1  /**************************************************************************  /*****************************************************************************
2   *   *
3   *      XVID MPEG-4 VIDEO CODEC - Example for encoding and decoding   *  XVID MPEG-4 VIDEO CODEC
4     *  - Console based decoding test application  -
5     *
6     *  Copyright(C) 2002-2003 Christoph Lampert
7     *               2002-2003 Edouard Gomez <ed.gomez@free.fr>
8   *   *
9   *      This program is free software; you can redistribute it and/or modify   *      This program is free software; you can redistribute it and/or modify
10   *      it under the terms of the GNU General Public License as published by   *      it under the terms of the GNU General Public License as published by
# Line 14  Line 18 
18   *   *
19   *      You should have received a copy of the GNU General Public License   *      You should have received a copy of the GNU General Public License
20   *      along with this program; if not, write to the Free Software   *      along with this program; if not, write to the Free Software
21   *      Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.   *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
22     *
23     * $Id: xvid_decraw.c,v 1.29 2010-12-28 19:19:43 Isibaar Exp $
24   *   *
25   *************************************************************************/   ****************************************************************************/
26    
27  /************************************************************************  /*****************************************************************************
28   *   *
29   *  Test routine for XviD decoding   *  Application notes :
  *  (C) Christoph Lampert, 2002/08/17  
30   *   *
31   *  An MPEG-4 bitstream is read from stdin and decoded,   *  An MPEG-4 bitstream is read from an input file (or stdin) and decoded,
32   *  the speed for this is measured   *  the speed for this is measured.
33   *   *
34   *  The program is plain C and needs no libraries except for libxvidcore,   *  The program is plain C and needs no libraries except for libxvidcore,
35   *  and maths-lib, so with UN*X you simply compile by   *  and maths-lib.
  *  
  *   gcc xvid_decraw.c -lxvidcore -lm -o xvid_decraw  
36   *   *
37   *  You have to specify the image dimensions (until the add the feature   *  Use ./xvid_decraw -help for a list of options
  *  to read this from the bitstream)  
38   *   *
39   *  Parameters are: xvid_stat XDIM YDIM   ****************************************************************************/
  *  
  *  output and indivual m4v-files are saved, if corresponding flags are set  
  *  
  ************************************************************************/  
40    
41  #include <stdio.h>  #include <stdio.h>
42  #include <stdlib.h>  #include <stdlib.h>
43  #include <math.h>               // needed for log10  #include <string.h>
44  #include <sys/time.h>           // only needed for gettimeofday  #include <math.h>
45    #ifndef WIN32
46    #include <sys/time.h>
47    #else
48    #include <time.h>
49    #endif
50    
51  #include "../src/xvid.h"                /* comes with XviD */  #include "xvid.h"
52    
53  #define ABS_MAXFRAMENR 9999               // max number of frames  /*****************************************************************************
54     *               Global vars in module and constants
55     ****************************************************************************/
56    
57    #define USE_PNM 0
58    #define USE_TGA 1
59    #define USE_YUV 2
60    
61    static int XDIM = 0;
62    static int YDIM = 0;
63    static int ARG_SAVEDECOUTPUT = 0;
64    static int ARG_SAVEMPEGSTREAM = 0;
65    static char *ARG_INPUTFILE = NULL;
66    static int CSP = XVID_CSP_I420;
67    static int BPP = 1;
68    static int FORMAT = USE_PNM;
69    static int POSTPROC = 0;
70    static  int ARG_THREADS = 0;
71    
72    static char filepath[256] = "./";
73    static void *dec_handle = NULL;
74    
75    #define BUFFER_SIZE (2*1024*1024)
76    
77    static const int display_buffer_bytes = 0;
78    
79    #define MIN_USEFUL_BYTES 1
80    
81    /*****************************************************************************
82     *               Local prototypes
83     ****************************************************************************/
84    
85    static double msecond();
86    static int dec_init(int use_assembler, int debug_level);
87    static int dec_main(unsigned char *istream,
88                                            unsigned char *ostream,
89                                            int istream_size,
90                                            xvid_dec_stats_t *xvid_dec_stats);
91    static int dec_stop();
92    static void usage();
93    static int write_image(char *prefix, unsigned char *image, int filenr);
94    static int write_pnm(char *filename, unsigned char *image);
95    static int write_tga(char *filename, unsigned char *image);
96    static int write_yuv(char *filename, unsigned char *image);
97    
98    const char * type2str(int type)
99    {
100        if (type==XVID_TYPE_IVOP)
101            return "I";
102        if (type==XVID_TYPE_PVOP)
103            return "P";
104        if (type==XVID_TYPE_BVOP)
105            return "B";
106        return "S";
107    }
108    
109    /*****************************************************************************
110     *        Main program
111     ****************************************************************************/
112    
113  int XDIM=720;  int main(int argc, char *argv[])
114  int YDIM=576;  {
115  int i,filenr = 0;          unsigned char *mp4_buffer = NULL;
116            unsigned char *mp4_ptr    = NULL;
117            unsigned char *out_buffer = NULL;
118            int useful_bytes;
119            int chunk;
120            xvid_dec_stats_t xvid_dec_stats;
121    
122  int save_dec_flag = 1;          // save decompressed bytestream?          double totaldectime;
 int save_m4v_flag = 0;          // save bytestream itself?  
123    
124  char filepath[256] = "./";      // the path where to save output          long totalsize;
125            int status;
126    
127  void *dec_handle = NULL;                // handle for decoding          int use_assembler = 1;
128            int debug_level = 0;
129    
130  /*********************************************************************/          char filename[256];
 /*                     "statistical" functions                               */  
 /*                                                                   */  
 /*  these are not needed for decoding itself, but for measuring      */  
 /*  time (and maybe later quality), there in nothing specific to     */  
 /*  XviD in these                                                    */  
 /*                                                                   */  
 /*********************************************************************/  
131    
132  double msecond()          FILE *in_file;
133  /* return the current time in seconds(!)  */          int filenr;
134  {          int i;
         struct timeval  tv;  
         gettimeofday(&tv, 0);  
         return tv.tv_sec + tv.tv_usec * 1.0e-6;  
 }  
135    
136            printf("xvid_decraw - raw mpeg4 bitstream decoder ");
137            printf("written by Christoph Lampert\n\n");
138    
139  /*********************************************************************/  /*****************************************************************************
140  /*                    input and output functions                         */   * Command line parsing
141  /*                                                                   */   ****************************************************************************/
 /* the are small and simple routines for writing image               */  
 /* image. It's just for convenience, again nothing specific to XviD  */  
 /*                                                                   */  
 /*********************************************************************/  
142    
143  int write_pgm(char *filename, unsigned char *image)          for (i=1; i< argc; i++) {
 {  
         FILE *filehandle;  
         filehandle=fopen(filename,"wb");  
         if (filehandle)  
         {  
                 fprintf(filehandle,"P5\n\n");           //  
                 fprintf(filehandle,"%d %d 255\n",XDIM,YDIM*3/2);  
                 fwrite(image,XDIM,YDIM*3/2,filehandle);  
144    
145                  fclose(filehandle);                  if (strcmp("-noasm", argv[i]) == 0 ) {
146                  return 0;                          use_assembler = 0;
147                    } else if (strcmp("-debug", argv[i]) == 0 && i < argc - 1 ) {
148                            i++;
149                            if (sscanf(argv[i], "0x%x", &debug_level) != 1) {
150                                    debug_level = atoi(argv[i]);
151                            }
152                    } else if (strcmp("-d", argv[i]) == 0) {
153                            ARG_SAVEDECOUTPUT = 1;
154                    } else if (strcmp("-i", argv[i]) == 0 && i < argc - 1 ) {
155                            i++;
156                            ARG_INPUTFILE = argv[i];
157                    } else if (strcmp("-m", argv[i]) == 0) {
158                            ARG_SAVEMPEGSTREAM = 1;
159                    } else if (strcmp("-c", argv[i]) == 0  && i < argc - 1 ) {
160                            i++;
161                            if (strcmp(argv[i], "rgb16") == 0) {
162                                    CSP = XVID_CSP_RGB555;
163                                    BPP = 2;
164                            } else if (strcmp(argv[i], "rgb24") == 0) {
165                                    CSP = XVID_CSP_BGR;
166                                    BPP = 3;
167                            } else if (strcmp(argv[i], "rgb32") == 0) {
168                                    CSP = XVID_CSP_BGRA;
169                                    BPP = 4;
170                            } else if (strcmp(argv[i], "yv12") == 0) {
171                                    CSP = XVID_CSP_YV12;
172                                    BPP = 1;
173                            } else {
174                                    CSP = XVID_CSP_I420;
175                                    BPP = 1;
176                            }
177                    } else if (strcmp("-postproc", argv[i]) == 0 && i < argc - 1 ) {
178                            i++;
179                            POSTPROC = atoi(argv[i]);
180                            if (POSTPROC < 0) POSTPROC = 0;
181                            if (POSTPROC > 2) POSTPROC = 2;
182                    } else if (strcmp("-f", argv[i]) == 0 && i < argc -1) {
183                            i++;
184                            if (strcmp(argv[i], "tga") == 0) {
185                                    FORMAT = USE_TGA;
186                            } else if (strcmp(argv[i], "yuv") == 0) {
187                                    FORMAT = USE_YUV;
188                            } else {
189                                    FORMAT = USE_PNM;
190                            }
191                    } else if (strcmp("-threads", argv[i]) == 0 && i < argc -1) {
192                            i++;
193                            ARG_THREADS = atoi(argv[i]);
194                    } else if (strcmp("-help", argv[i]) == 0) {
195                            usage();
196                            return(0);
197                    } else {
198                            usage();
199                            exit(-1);
200          }          }
         else  
                 return 1;  
201  }  }
202    
203    #if defined(_MSC_VER)
204            if (ARG_INPUTFILE==NULL) {
205                    fprintf(stderr, "Warning: MSVC build does not read EOF correctly from stdin. Use the -i switch.\n\n");
206            }
207    #endif
208    
209  int write_ppm(char *filename, unsigned char *image)  /*****************************************************************************
210  {   * Values checking
211          FILE *filehandle;   ****************************************************************************/
         filehandle=fopen(filename,"wb");  
         if (filehandle)  
         {  
                 fprintf(filehandle,"P6\n\n");           //  
                 fprintf(filehandle,"%d %d 255\n",XDIM,YDIM);  
                 fwrite(image,XDIM,YDIM*3,filehandle);  
212    
213                  fclose(filehandle);          if ( ARG_INPUTFILE == NULL || strcmp(ARG_INPUTFILE, "stdin") == 0) {
214                  return 0;                  in_file = stdin;
215            }
216            else {
217    
218                    in_file = fopen(ARG_INPUTFILE, "rb");
219                    if (in_file == NULL) {
220                            fprintf(stderr, "Error opening input file %s\n", ARG_INPUTFILE);
221                            return(-1);
222          }          }
         else  
                 return 1;  
223  }  }
224    
225  /*********************************************************************/          /* PNM/PGM format can't handle 16/32 bit data */
226  /* Routines for decoding: init encoder, frame step, release encoder  */          if (BPP != 1 && BPP != 3 && FORMAT == USE_PNM) {
227  /*********************************************************************/                  FORMAT = USE_TGA;
228            }
229            if (BPP != 1 && FORMAT == USE_YUV) {
230                    FORMAT = USE_TGA;
231            }
232    
233  int dec_init(int use_assembler) /* init decoder before first run */  /*****************************************************************************
234  {   *        Memory allocation
235          int xerr;   ****************************************************************************/
236    
237          XVID_INIT_PARAM xinit;          /* Memory for encoded mp4 stream */
238          XVID_DEC_PARAM xparam;          mp4_buffer = (unsigned char *) malloc(BUFFER_SIZE);
239            mp4_ptr = mp4_buffer;
240            if (!mp4_buffer)
241                    goto free_all_memory;
242    
243                  if(use_assembler)  /*****************************************************************************
244  #ifdef ARCH_IA64   *        Xvid PART  Start
245                          xinit.cpu_flags = XVID_CPU_FORCE | XVID_CPU_IA64;   ****************************************************************************/
246  #else  
247                          xinit.cpu_flags = 0;          status = dec_init(use_assembler, debug_level);
248  #endif          if (status) {
249                  else                  fprintf(stderr,
250                          xinit.cpu_flags = XVID_CPU_FORCE;                                  "Decore INIT problem, return value %d\n", status);
251                    goto release_all;
252            }
253    
         xvid_init(NULL, 0, &xinit, NULL);  
         xparam.width = XDIM;  
         xparam.height = YDIM;  
254    
255          xerr = xvid_decore(NULL, XVID_DEC_CREATE, &xparam, NULL);  /*****************************************************************************
256          dec_handle = xparam.handle;   *                               Main loop
257     ****************************************************************************/
258    
259            /* Fill the buffer */
260            useful_bytes = (int) fread(mp4_buffer, 1, BUFFER_SIZE, in_file);
261    
262            totaldectime = 0;
263            totalsize = 0;
264            filenr = 0;
265            mp4_ptr = mp4_buffer;
266            chunk = 0;
267    
268          return xerr;          do {
269  }                  int used_bytes = 0;
270                    double dectime;
271    
272                    /*
273                     * If the buffer is half empty or there are no more bytes in it
274                     * then fill it.
275                     */
276                    if (mp4_ptr > mp4_buffer + BUFFER_SIZE/2) {
277                            int already_in_buffer = (int)(mp4_buffer + BUFFER_SIZE - mp4_ptr);
278    
279                            /* Move data if needed */
280                            if (already_in_buffer > 0)
281                                    memcpy(mp4_buffer, mp4_ptr, already_in_buffer);
282    
283  int dec_main(unsigned char *m4v_buffer, unsigned char *out_buffer,int *m4v_size)                          /* Update mp4_ptr */
284  {       /* decode one frame  */                          mp4_ptr = mp4_buffer;
285    
286          int xerr;                          /* read new data */
287          XVID_DEC_FRAME xframe;              if(!feof(in_file)) {
288                                    useful_bytes += (int) fread(mp4_buffer + already_in_buffer,
289                                                                                    1, BUFFER_SIZE - already_in_buffer,
290                                                                                    in_file);
291                            }
292                    }
293    
         xframe.bitstream = m4v_buffer;  
         xframe.length = 9999;  
             xframe.image = out_buffer;  
         xframe.stride = XDIM;  
             xframe.colorspace = XVID_CSP_YV12;  
294    
295                    /* This loop is needed to handle VOL/NVOP reading */
296                  do {                  do {
                 xerr = xvid_decore(dec_handle, XVID_DEC_DECODE, &xframe, NULL);  
297    
298                          *m4v_size = xframe.length;                          /* Decode frame */
299                  xframe.bitstream += *m4v_size;                          dectime = msecond();
300                            used_bytes = dec_main(mp4_ptr, out_buffer, useful_bytes, &xvid_dec_stats);
301                            dectime = msecond() - dectime;
302    
303                            /* Resize image buffer if needed */
304                            if(xvid_dec_stats.type == XVID_TYPE_VOL) {
305    
306                                    /* Check if old buffer is smaller */
307                                    if(XDIM*YDIM < xvid_dec_stats.data.vol.width*xvid_dec_stats.data.vol.height) {
308    
309                                            /* Copy new witdh and new height from the vol structure */
310                                            XDIM = xvid_dec_stats.data.vol.width;
311                                            YDIM = xvid_dec_stats.data.vol.height;
312    
313                                            /* Free old output buffer*/
314                                            if(out_buffer) free(out_buffer);
315    
316                                            /* Allocate the new buffer */
317                                            out_buffer = (unsigned char*)malloc(XDIM*YDIM*4);
318                                            if(out_buffer == NULL)
319                                                    goto free_all_memory;
320    
321                                            fprintf(stderr, "Resized frame buffer to %dx%d\n", XDIM, YDIM);
322                                    }
323    
324                  } while (*m4v_size<7);  /* this skips N-VOPs */                                  /* Save individual mpeg4 stream if required */
325                                    if(ARG_SAVEMPEGSTREAM) {
326                                            FILE *filehandle = NULL;
327    
328          return xerr;                                          sprintf(filename, "%svolhdr.m4v", filepath);
329                                            filehandle = fopen(filename, "wb");
330                                            if(!filehandle) {
331                                                    fprintf(stderr,
332                                                                    "Error writing vol header mpeg4 stream to file %s\n",
333                                                                    filename);
334                                            } else {
335                                                    fwrite(mp4_ptr, 1, used_bytes, filehandle);
336                                                    fclose(filehandle);
337                                            }
338                                    }
339  }  }
340    
341  int dec_stop()  /* close decoder to release resources */                          /* Update buffer pointers */
342  {                          if(used_bytes > 0) {
343          int xerr;                                  mp4_ptr += used_bytes;
344          xerr = xvid_decore(dec_handle, XVID_DEC_DESTROY, NULL, NULL);                                  useful_bytes -= used_bytes;
345    
346          return xerr;                                  /* Total size */
347                                    totalsize += used_bytes;
348  }  }
349    
350                            if (display_buffer_bytes) {
351                                    printf("Data chunk %d: %d bytes consumed, %d bytes in buffer\n", chunk++, used_bytes, useful_bytes);
352                            }
353                    } while (xvid_dec_stats.type <= 0 && useful_bytes > MIN_USEFUL_BYTES);
354    
355  /*********************************************************************/                  /* Check if there is a negative number of useful bytes left in buffer
356  /*                          Main program                             */                   * This means we went too far */
357  /*********************************************************************/          if(useful_bytes < 0)
358                break;
359    
360  int main(int argc, char *argv[])          /* Updated data - Count only usefull decode time */
361  {                  totaldectime += dectime;
   unsigned char *divx_buffer = NULL;  
   unsigned char *divx_ptr = NULL;  
   unsigned char *out_buffer = NULL;  
362    
   double dectime;  
   double totaldectime=0.;  
363    
364    long totalsize=0;                  if (!display_buffer_bytes) {
365    int status;                          printf("Frame %5d: type = %s, dectime(ms) =%6.1f, length(bytes) =%7d\n",
366                                            filenr, type2str(xvid_dec_stats.type), dectime, used_bytes);
367                    }
368    
369    int m4v_size;                  /* Save individual mpeg4 stream if required */
370    int frame_type[ABS_MAXFRAMENR];                  if(ARG_SAVEMPEGSTREAM) {
371    int use_assembler=1;                          FILE *filehandle = NULL;
372    
373    char filename[256];                          sprintf(filename, "%sframe%05d.m4v", filepath, filenr);
374                            filehandle = fopen(filename, "wb");
375                            if(!filehandle) {
376                                    fprintf(stderr,
377                                                    "Error writing single mpeg4 stream to file %s\n",
378                                                    filename);
379                            }
380                            else {
381                                    fwrite(mp4_ptr-used_bytes, 1, used_bytes, filehandle);
382                                    fclose(filehandle);
383                            }
384                    }
385    
386                    /* Save output frame if required */
387                    if (ARG_SAVEDECOUTPUT) {
388                            sprintf(filename, "%sdec", filepath);
389    
390                            if(write_image(filename, out_buffer, filenr)) {
391                                    fprintf(stderr,
392                                                    "Error writing decoded frame %s\n",
393                                                    filename);
394                            }
395                    }
396    
397                    filenr++;
398    
399            } while (useful_bytes>MIN_USEFUL_BYTES || !feof(in_file));
400    
401            useful_bytes = 0; /* Empty buffer */
402    
403    FILE *filehandle;  /*****************************************************************************
404     *     Flush decoder buffers
405     ****************************************************************************/
406    
407          if (argc>=3)          do {
408          {       XDIM = atoi(argv[1]);  
409                  YDIM = atoi(argv[2]);                  /* Fake vars */
410                  if ( (XDIM <= 0) || (XDIM >= 2048) || (YDIM <=0) || (YDIM >= 2048) )                  int used_bytes;
411                  {       fprintf(stderr,"Wrong frames size %d %d, trying PGM \n",XDIM, YDIM);                  double dectime;
412    
413            do {
414                        dectime = msecond();
415                        used_bytes = dec_main(NULL, out_buffer, -1, &xvid_dec_stats);
416                        dectime = msecond() - dectime;
417                            if (display_buffer_bytes) {
418                                    printf("Data chunk %d: %d bytes consumed, %d bytes in buffer\n", chunk++, used_bytes, useful_bytes);
419                  }                  }
420            } while(used_bytes>=0 && xvid_dec_stats.type <= 0);
421    
422            if (used_bytes < 0) {   /* XVID_ERR_END */
423                break;
424          }          }
         if (argc>=4 && !strcmp(argv[3],"noasm"))  
           use_assembler = 0;  
425    
426                    /* Updated data - Count only usefull decode time */
427                    totaldectime += dectime;
428    
429  /* allocate memory */                  /* Prints some decoding stats */
430                    if (!display_buffer_bytes) {
431                            printf("Frame %5d: type = %s, dectime(ms) =%6.1f, length(bytes) =%7d\n",
432                                            filenr, type2str(xvid_dec_stats.type), dectime, used_bytes);
433                    }
434    
435    divx_buffer = (unsigned char *) malloc(10*XDIM*YDIM);                  /* Save output frame if required */
436    // this should really be enough memory!                  if (ARG_SAVEDECOUTPUT) {
437    if (!divx_buffer)                          sprintf(filename, "%sdec", filepath);
438      goto free_all_memory;  
439    divx_ptr = divx_buffer+10*XDIM*YDIM;                          if(write_image(filename, out_buffer, filenr)) {
440                                    fprintf(stderr,
441                                                    "Error writing decoded frame %s\n",
442                                                    filename);
443                            }
444                    }
445    
446    out_buffer = (unsigned char *) malloc(4*XDIM*YDIM);   /* YUV needs less */                  filenr++;
447    if (!out_buffer)  
448      goto free_all_memory;          }while(1);
449    
450    /*****************************************************************************
451     *     Calculate totals and averages for output, print results
452     ****************************************************************************/
453    
454            if (filenr>0) {
455                    totalsize    /= filenr;
456                    totaldectime /= filenr;
457                    printf("Avg: dectime(ms) =%7.2f, fps =%7.2f, length(bytes) =%7d\n",
458                               totaldectime, 1000/totaldectime, (int)totalsize);
459            }else{
460                    printf("Nothing was decoded!\n");
461            }
462    
463  /*********************************************************************/  /*****************************************************************************
464  /*                         XviD PART  Start                          */   *      Xvid PART  Stop
465  /*********************************************************************/   ****************************************************************************/
466    
467          status = dec_init(use_assembler);   release_all:
468            if (dec_handle) {
469                    status = dec_stop();
470          if (status)          if (status)
471          {                          fprintf(stderr, "decore RELEASE problem return value %d\n", status);
                 printf("Decore INIT problem, return value %d\n", status);  
                 goto release_all;  
472          }          }
473    
474     free_all_memory:
475            free(out_buffer);
476            free(mp4_buffer);
477    
478  /*********************************************************************/          return(0);
479  /*                               Main loop                           */  }
480  /*********************************************************************/  
481    /*****************************************************************************
482     *               Usage function
483     ****************************************************************************/
484    
485    do  static void usage()
486      {      {
487    
488          if (divx_ptr > divx_buffer+5*XDIM*YDIM) /* buffer more than half empty */          fprintf(stderr, "Usage : xvid_decraw [OPTIONS]\n");
489          {       int rest=(divx_buffer+10*XDIM*YDIM-divx_ptr);          fprintf(stderr, "Options :\n");
490                  if (rest)          fprintf(stderr, " -noasm         : don't use assembly optimizations (default=enabled)\n");
491                          memcpy(divx_buffer, divx_ptr, rest);          fprintf(stderr, " -debug         : debug level (debug=0)\n");
492                  divx_ptr = divx_buffer;          fprintf(stderr, " -i string      : input filename (default=stdin)\n");
493                  fread(divx_buffer+rest, 1, 5*XDIM*YDIM, stdin); /* read new data */          fprintf(stderr, " -d             : save decoder output\n");
494            fprintf(stderr, " -c csp         : choose colorspace output (rgb16, rgb24, rgb32, yv12, i420)\n");
495            fprintf(stderr, " -f format      : choose output file format (tga, pnm, pgm, yuv)\n");
496            fprintf(stderr, " -postproc      : postprocessing level (0=off, 1=deblock, 2=deblock+dering)\n");
497            fprintf(stderr, " -threads int   : number of threads\n");
498            fprintf(stderr, " -m             : save mpeg4 raw stream to individual files\n");
499            fprintf(stderr, " -help          : This help message\n");
500            fprintf(stderr, " (* means default)\n");
501    
502          }          }
503    
504          dectime = -msecond();  /*****************************************************************************
505          status = dec_main(divx_ptr, out_buffer, &m4v_size);   *               "helper" functions
506     ****************************************************************************/
507    
508          if (status)  /* return the current time in milli seconds */
509    static double
510    msecond()
511          {          {
512                  break;  #ifndef WIN32
513            struct timeval  tv;
514            gettimeofday(&tv, 0);
515            return((double)tv.tv_sec*1.0e3 + (double)tv.tv_usec*1.0e-3);
516    #else
517            clock_t clk;
518            clk = clock();
519            return(clk * 1000.0 / CLOCKS_PER_SEC);
520    #endif
521          }          }
         dectime += msecond();  
         divx_ptr += m4v_size;  
522    
523          totalsize += m4v_size;  /*****************************************************************************
524     *              output functions
525     ****************************************************************************/
526    
527    static int write_image(char *prefix, unsigned char *image, int filenr)
528    {
529            char filename[1024];
530            char *ext;
531            int ret;
532    
533            if (FORMAT == USE_PNM && BPP == 1) {
534                    ext = "pgm";
535            } else if (FORMAT == USE_PNM && BPP == 3) {
536                    ext = "pnm";
537            } else if (FORMAT == USE_YUV) {
538                    ext = "yuv";
539            } else if (FORMAT == USE_TGA) {
540                    ext = "tga";
541            } else {
542                    fprintf(stderr, "Bug: should not reach this path code -- please report to xvid-devel@xvid.org with command line options used");
543                    exit(-1);
544            }
545    
546            if (FORMAT == USE_YUV) {
547                    sprintf(filename, "%s.%s", prefix, ext);
548    
549                    if (!filenr) {
550                            FILE *fp = fopen(filename, "wb");
551                            fclose(fp);
552                    }
553            } else
554                    sprintf(filename, "%s%05d.%s", prefix, filenr, ext);
555    
556            if (FORMAT == USE_PNM) {
557                    ret = write_pnm(filename, image);
558            } else if (FORMAT == USE_YUV) {
559                    ret = write_yuv(filename, image);
560            } else {
561                    ret = write_tga(filename, image);
562            }
563    
564            return(ret);
565    }
566    
567    static int write_tga(char *filename, unsigned char *image)
568    {
569            FILE * f;
570            char hdr[18];
571    
572            f = fopen(filename, "wb");
573            if ( f == NULL) {
574                    return -1;
575            }
576    
577            hdr[0]  = 0; /* ID length */
578            hdr[1]  = 0; /* Color map type */
579            hdr[2]  = (BPP>1)?2:3; /* Uncompressed true color (2) or greymap (3) */
580            hdr[3]  = 0; /* Color map specification (not used) */
581            hdr[4]  = 0; /* Color map specification (not used) */
582            hdr[5]  = 0; /* Color map specification (not used) */
583            hdr[6]  = 0; /* Color map specification (not used) */
584            hdr[7]  = 0; /* Color map specification (not used) */
585            hdr[8]  = 0; /* LSB X origin */
586            hdr[9]  = 0; /* MSB X origin */
587            hdr[10] = 0; /* LSB Y origin */
588            hdr[11] = 0; /* MSB Y origin */
589            hdr[12] = (XDIM>>0)&0xff; /* LSB Width */
590            hdr[13] = (XDIM>>8)&0xff; /* MSB Width */
591            if (BPP > 1) {
592                    hdr[14] = (YDIM>>0)&0xff; /* LSB Height */
593                    hdr[15] = (YDIM>>8)&0xff; /* MSB Height */
594            } else {
595                    hdr[14] = ((YDIM*3)>>1)&0xff; /* LSB Height */
596                    hdr[15] = ((YDIM*3)>>9)&0xff; /* MSB Height */
597            }
598            hdr[16] = BPP*8;
599            hdr[17] = 0x00 | (1<<5) /* Up to down */ | (0<<4); /* Image descriptor */
600    
601            /* Write header */
602            fwrite(hdr, 1, sizeof(hdr), f);
603    
604    #ifdef ARCH_IS_LITTLE_ENDIAN
605            /* write first plane */
606            fwrite(image, 1, XDIM*YDIM*BPP, f);
607    #else
608            {
609                    int i;
610                    for (i=0; i<XDIM*YDIM*BPP;i+=BPP) {
611                            if (BPP == 1) {
612                                    fputc(*(image+i), f);
613                            } else if (BPP == 2) {
614                                    fputc(*(image+i+1), f);
615                                    fputc(*(image+i+0), f);
616                            } else if (BPP == 3) {
617                                    fputc(*(image+i+2), f);
618                                    fputc(*(image+i+1), f);
619                                    fputc(*(image+i+0), f);
620                            } else if (BPP == 4) {
621                                    fputc(*(image+i+3), f);
622                                    fputc(*(image+i+2), f);
623                                    fputc(*(image+i+1), f);
624                                    fputc(*(image+i+0), f);
625                            }
626                    }
627            }
628    #endif
629    
630          printf("Frame %5d: dectime =%6.1f ms length=%7d bytes \n",          /* Write Y and V planes for YUV formats */
631                          filenr, dectime*1000, m4v_size);          if (BPP == 1) {
632                    int i;
633    
634          if (save_m4v_flag)                  /* Write the two chrominance planes */
635                    for (i=0; i<YDIM/2; i++) {
636                            fwrite(image+XDIM*YDIM + i*XDIM/2, 1, XDIM/2, f);
637                            fwrite(image+5*XDIM*YDIM/4 + i*XDIM/2, 1, XDIM/2, f);
638                    }
639            }
640    
641    
642            /* Close the file */
643            fclose(f);
644    
645            return(0);
646    }
647    
648    static int write_pnm(char *filename, unsigned char *image)
649          {          {
650                  sprintf(filename, "%sframe%05d.m4v", filepath, filenr);          FILE * f;
651                  filehandle = fopen(filename, "wb");  
652                  fwrite(divx_buffer, m4v_size, 1, filehandle);          f = fopen(filename, "wb");
653                  fclose(filehandle);          if ( f == NULL) {
654                    return -1;
655          }          }
         totaldectime += dectime;  
656    
657            if (BPP == 1) {
658                    int i;
659                    fprintf(f, "P5\n%i %i\n255\n", XDIM, YDIM*3/2);
660    
661  /*********************************************************************/                  fwrite(image, 1, XDIM*YDIM, f);
 /*        analyse the decoded frame and compare to original          */  
 /*********************************************************************/  
662    
663          if (save_dec_flag)                  for (i=0; i<YDIM/2;i++) {
664                            fwrite(image+XDIM*YDIM + i*XDIM/2, 1, XDIM/2, f);
665                            fwrite(image+5*XDIM*YDIM/4 + i*XDIM/2, 1, XDIM/2, f);
666                    }
667            } else if (BPP == 3) {
668                    int i;
669                    fprintf(f, "P6\n#xvid\n%i %i\n255\n", XDIM, YDIM);
670                    for (i=0; i<XDIM*YDIM*3; i+=3) {
671    #ifdef ARCH_IS_LITTLE_ENDIAN
672                            fputc(image[i+2], f);
673                            fputc(image[i+1], f);
674                            fputc(image[i+0], f);
675    #else
676                            fputc(image[i+0], f);
677                            fputc(image[i+1], f);
678                            fputc(image[i+2], f);
679    #endif
680                    }
681            }
682    
683            fclose(f);
684    
685            return 0;
686    }
687    
688    static int write_yuv(char *filename, unsigned char *image)
689          {          {
690                  sprintf(filename, "%sdec%05d.pgm", filepath, filenr);          FILE * f;
691                  write_pgm(filename,out_buffer);  
692            f = fopen(filename, "ab+");
693            if ( f == NULL) {
694                    return -1;
695          }          }
696    
697          filenr++;          fwrite(image, 1, 3*XDIM*YDIM/2, f);
698    
699     } while ( (status>=0) && (filenr<ABS_MAXFRAMENR) );          fclose(f);
700    
701            return 0;
702    }
703    
704  /*********************************************************************/  /*****************************************************************************
705  /*     calculate totals and averages for output, print results       */   * Routines for decoding: init decoder, use, and stop decoder
706  /*********************************************************************/   ****************************************************************************/
707    
708    /* init decoder before first run */
709    static int
710    dec_init(int use_assembler, int debug_level)
711    {
712            int ret;
713    
714            xvid_gbl_init_t   xvid_gbl_init;
715            xvid_dec_create_t xvid_dec_create;
716            xvid_gbl_info_t   xvid_gbl_info;
717    
718            /* Reset the structure with zeros */
719            memset(&xvid_gbl_init, 0, sizeof(xvid_gbl_init_t));
720            memset(&xvid_dec_create, 0, sizeof(xvid_dec_create_t));
721            memset(&xvid_gbl_info, 0, sizeof(xvid_gbl_info));
722    
723            /*------------------------------------------------------------------------
724             * Xvid core initialization
725             *----------------------------------------------------------------------*/
726    
727            xvid_gbl_info.version = XVID_VERSION;
728            xvid_global(NULL, XVID_GBL_INFO, &xvid_gbl_info, NULL);
729    
730            if (xvid_gbl_info.build != NULL) {
731                    fprintf(stderr, "xvidcore build version: %s\n", xvid_gbl_info.build);
732            }
733            fprintf(stderr, "Bitstream version: %d.%d.%d\n", XVID_VERSION_MAJOR(xvid_gbl_info.actual_version), XVID_VERSION_MINOR(xvid_gbl_info.actual_version), XVID_VERSION_PATCH(xvid_gbl_info.actual_version));
734            fprintf(stderr, "Detected CPU flags: ");
735            if (xvid_gbl_info.cpu_flags & XVID_CPU_ASM)
736                    fprintf(stderr, "ASM ");
737            if (xvid_gbl_info.cpu_flags & XVID_CPU_MMX)
738                    fprintf(stderr, "MMX ");
739            if (xvid_gbl_info.cpu_flags & XVID_CPU_MMXEXT)
740                    fprintf(stderr, "MMXEXT ");
741            if (xvid_gbl_info.cpu_flags & XVID_CPU_SSE)
742                    fprintf(stderr, "SSE ");
743            if (xvid_gbl_info.cpu_flags & XVID_CPU_SSE2)
744                    fprintf(stderr, "SSE2 ");
745            if (xvid_gbl_info.cpu_flags & XVID_CPU_SSE3)
746                    fprintf(stderr, "SSE3 ");
747            if (xvid_gbl_info.cpu_flags & XVID_CPU_SSE41)
748                    fprintf(stderr, "SSE41 ");
749        if (xvid_gbl_info.cpu_flags & XVID_CPU_3DNOW)
750                    fprintf(stderr, "3DNOW ");
751            if (xvid_gbl_info.cpu_flags & XVID_CPU_3DNOWEXT)
752                    fprintf(stderr, "3DNOWEXT ");
753            if (xvid_gbl_info.cpu_flags & XVID_CPU_TSC)
754                    fprintf(stderr, "TSC ");
755            fprintf(stderr, "\n");
756            fprintf(stderr, "Detected %d cpus,", xvid_gbl_info.num_threads);
757            if (!ARG_THREADS) ARG_THREADS = xvid_gbl_info.num_threads;
758            fprintf(stderr, " using %d threads.\n", ARG_THREADS);
759    
760          totalsize    /= filenr;          /* Version */
761          totaldectime /= filenr;          xvid_gbl_init.version = XVID_VERSION;
762    
763          fprintf(stderr,"Avg: dectime %5.2f ms, %5.2f fps, filesize =%d\n",          /* Assembly setting */
764                  1000*totaldectime, 1./totaldectime, totalsize);          if(use_assembler)
765    #ifdef ARCH_IS_IA64
766                    xvid_gbl_init.cpu_flags = XVID_CPU_FORCE | XVID_CPU_IA64;
767    #else
768            xvid_gbl_init.cpu_flags = 0;
769    #endif
770            else
771                    xvid_gbl_init.cpu_flags = XVID_CPU_FORCE;
772    
773  /*********************************************************************/          xvid_gbl_init.debug = debug_level;
 /*                         XviD PART  Stop                           */  
 /*********************************************************************/  
774    
775  release_all:          xvid_global(NULL, 0, &xvid_gbl_init, NULL);
776    
777            /*------------------------------------------------------------------------
778             * Xvid decoder initialization
779             *----------------------------------------------------------------------*/
780    
781            /* Version */
782            xvid_dec_create.version = XVID_VERSION;
783    
784            /*
785             * Image dimensions -- set to 0, xvidcore will resize when ever it is
786             * needed
787             */
788            xvid_dec_create.width = 0;
789            xvid_dec_create.height = 0;
790    
791          if (dec_handle)          xvid_dec_create.num_threads = ARG_THREADS;
792    
793            ret = xvid_decore(NULL, XVID_DEC_CREATE, &xvid_dec_create, NULL);
794    
795            dec_handle = xvid_dec_create.handle;
796    
797            return(ret);
798    }
799    
800    /* decode one frame  */
801    static int
802    dec_main(unsigned char *istream,
803                     unsigned char *ostream,
804                     int istream_size,
805                     xvid_dec_stats_t *xvid_dec_stats)
806          {          {
807                  status = dec_stop();  
808                  if (status)          int ret;
809                          printf("Decore RELEASE problem return value %d\n", status);  
810            xvid_dec_frame_t xvid_dec_frame;
811    
812            /* Reset all structures */
813            memset(&xvid_dec_frame, 0, sizeof(xvid_dec_frame_t));
814            memset(xvid_dec_stats, 0, sizeof(xvid_dec_stats_t));
815    
816            /* Set version */
817            xvid_dec_frame.version = XVID_VERSION;
818            xvid_dec_stats->version = XVID_VERSION;
819    
820            /* No general flags to set */
821            if (POSTPROC == 1)
822                    xvid_dec_frame.general          = XVID_DEBLOCKY | XVID_DEBLOCKUV;
823            else if (POSTPROC==2)
824                    xvid_dec_frame.general          = XVID_DEBLOCKY | XVID_DEBLOCKUV | XVID_DERINGY | XVID_DERINGUV;
825            else
826                    xvid_dec_frame.general          = 0;
827    
828            /* Input stream */
829            xvid_dec_frame.bitstream        = istream;
830            xvid_dec_frame.length           = istream_size;
831    
832            /* Output frame structure */
833            xvid_dec_frame.output.plane[0]  = ostream;
834            xvid_dec_frame.output.stride[0] = XDIM*BPP;
835            xvid_dec_frame.output.csp = CSP;
836    
837            ret = xvid_decore(dec_handle, XVID_DEC_DECODE, &xvid_dec_frame, xvid_dec_stats);
838    
839            return(ret);
840          }          }
841    
842  free_all_memory:  /* close decoder to release resources */
843          free(out_buffer);  static int
844          free(divx_buffer);  dec_stop()
845    {
846            int ret;
847    
848    return 0;          ret = xvid_decore(dec_handle, XVID_DEC_DESTROY, NULL, NULL);
849    
850            return(ret);
851  }  }

Legend:
Removed from v.376  
changed lines
  Added in v.1928

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