[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 1678, Wed Feb 15 19:16:39 2006 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.24 2006-02-15 19:16:39 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    
60    static int XDIM = 0;
61    static int YDIM = 0;
62    static int ARG_SAVEDECOUTPUT = 0;
63    static int ARG_SAVEMPEGSTREAM = 0;
64    static char *ARG_INPUTFILE = NULL;
65    static int CSP = XVID_CSP_I420;
66    static int BPP = 1;
67    static int FORMAT = USE_PNM;
68    
69    static char filepath[256] = "./";
70    static void *dec_handle = NULL;
71    
72    #define BUFFER_SIZE (2*1024*1024)
73    
74    static const int display_buffer_bytes = 0;
75    
76    #define MIN_USEFUL_BYTES 1
77    
78    /*****************************************************************************
79     *               Local prototypes
80     ****************************************************************************/
81    
82    static double msecond();
83    static int dec_init(int use_assembler, int debug_level);
84    static int dec_main(unsigned char *istream,
85                                            unsigned char *ostream,
86                                            int istream_size,
87                                            xvid_dec_stats_t *xvid_dec_stats);
88    static int dec_stop();
89    static void usage();
90    static int write_image(char *prefix, unsigned char *image);
91    static int write_pnm(char *filename, unsigned char *image);
92    static int write_tga(char *filename, unsigned char *image);
93    
94    const char * type2str(int type)
95    {
96        if (type==XVID_TYPE_IVOP)
97            return "I";
98        if (type==XVID_TYPE_PVOP)
99            return "P";
100        if (type==XVID_TYPE_BVOP)
101            return "B";
102        return "S";
103    }
104    
105    /*****************************************************************************
106     *        Main program
107     ****************************************************************************/
108    
109  int XDIM=720;  int main(int argc, char *argv[])
110  int YDIM=576;  {
111  int i,filenr = 0;          unsigned char *mp4_buffer = NULL;
112            unsigned char *mp4_ptr    = NULL;
113            unsigned char *out_buffer = NULL;
114            int useful_bytes;
115            int chunk;
116            xvid_dec_stats_t xvid_dec_stats;
117    
118  int save_dec_flag = 1;          // save decompressed bytestream?          double totaldectime;
 int save_m4v_flag = 0;          // save bytestream itself?  
119    
120  char filepath[256] = "./";      // the path where to save output          long totalsize;
121            int status;
122    
123  void *dec_handle = NULL;                // handle for decoding          int use_assembler = 0;
124            int debug_level = 0;
125    
126  /*********************************************************************/          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                                                    */  
 /*                                                                   */  
 /*********************************************************************/  
127    
128  double msecond()          FILE *in_file;
129  /* return the current time in seconds(!)  */          int filenr;
130  {          int i;
131          struct timeval  tv;  
132          gettimeofday(&tv, 0);          printf("xvid_decraw - raw mpeg4 bitstream decoder ");
133          return tv.tv_sec + tv.tv_usec * 1.0e-6;          printf("written by Christoph Lampert 2002-2003\n\n");
134    
135    /*****************************************************************************
136     * Command line parsing
137     ****************************************************************************/
138    
139            for (i=1; i< argc; i++) {
140    
141                    if (strcmp("-asm", argv[i]) == 0 ) {
142                            use_assembler = 1;
143                    } else if (strcmp("-debug", argv[i]) == 0 && i < argc - 1 ) {
144                            i++;
145                            if (sscanf(argv[i], "0x%x", &debug_level) != 1) {
146                                    debug_level = atoi(argv[i]);
147                            }
148                    } else if (strcmp("-d", argv[i]) == 0) {
149                            ARG_SAVEDECOUTPUT = 1;
150                    } else if (strcmp("-i", argv[i]) == 0 && i < argc - 1 ) {
151                            i++;
152                            ARG_INPUTFILE = argv[i];
153                    } else if (strcmp("-m", argv[i]) == 0) {
154                            ARG_SAVEMPEGSTREAM = 1;
155                    } else if (strcmp("-c", argv[i]) == 0  && i < argc - 1 ) {
156                            i++;
157                            if (strcmp(argv[i], "rgb16") == 0) {
158                                    CSP = XVID_CSP_RGB555;
159                                    BPP = 2;
160                            } else if (strcmp(argv[i], "rgb24") == 0) {
161                                    CSP = XVID_CSP_BGR;
162                                    BPP = 3;
163                            } else if (strcmp(argv[i], "rgb32") == 0) {
164                                    CSP = XVID_CSP_BGRA;
165                                    BPP = 4;
166                            } else if (strcmp(argv[i], "yv12") == 0) {
167                                    CSP = XVID_CSP_YV12;
168                                    BPP = 1;
169                            } else {
170                                    CSP = XVID_CSP_I420;
171                                    BPP = 1;
172                            }
173                    } else if (strcmp("-f", argv[i]) == 0 && i < argc -1) {
174                            i++;
175                            if (strcmp(argv[i], "tga") == 0) {
176                                    FORMAT = USE_TGA;
177                            } else {
178                                    FORMAT = USE_PNM;
179                            }
180                    } else if (strcmp("-help", argv[i]) == 0) {
181                            usage();
182                            return(0);
183                    } else {
184                            usage();
185                            exit(-1);
186                    }
187            }
188    
189    #if defined(_MSC_VER)
190            if (ARG_INPUTFILE==NULL) {
191                    fprintf(stderr, "Warning: MSVC build does not read EOF correctly from stdin. Use the -i switch.\n\n");
192  }  }
193    #endif
194    
195    /*****************************************************************************
196     * Values checking
197     ****************************************************************************/
198    
199  /*********************************************************************/          if ( ARG_INPUTFILE == NULL || strcmp(ARG_INPUTFILE, "stdin") == 0) {
200  /*                    input and output functions                         */                  in_file = stdin;
201  /*                                                                   */          }
202  /* the are small and simple routines for writing image               */          else {
 /* image. It's just for convenience, again nothing specific to XviD  */  
 /*                                                                   */  
 /*********************************************************************/  
203    
204  int write_pgm(char *filename, unsigned char *image)                  in_file = fopen(ARG_INPUTFILE, "rb");
205  {                  if (in_file == NULL) {
206          FILE *filehandle;                          fprintf(stderr, "Error opening input file %s\n", ARG_INPUTFILE);
207          filehandle=fopen(filename,"wb");                          return(-1);
208          if (filehandle)                  }
209          {          }
                 fprintf(filehandle,"P5\n\n");           //  
                 fprintf(filehandle,"%d %d 255\n",XDIM,YDIM*3/2);  
                 fwrite(image,XDIM,YDIM*3/2,filehandle);  
210    
211                  fclose(filehandle);          /* PNM/PGM format can't handle 16/32 bit data */
212                  return 0;          if (BPP != 1 && BPP != 3 && FORMAT == USE_PNM) {
213                    FORMAT = USE_TGA;
214          }          }
215          else  
216                  return 1;  /*****************************************************************************
217     *        Memory allocation
218     ****************************************************************************/
219    
220            /* Memory for encoded mp4 stream */
221            mp4_buffer = (unsigned char *) malloc(BUFFER_SIZE);
222            mp4_ptr = mp4_buffer;
223            if (!mp4_buffer)
224                    goto free_all_memory;
225    
226    /*****************************************************************************
227     *        XviD PART  Start
228     ****************************************************************************/
229    
230            status = dec_init(use_assembler, debug_level);
231            if (status) {
232                    fprintf(stderr,
233                                    "Decore INIT problem, return value %d\n", status);
234                    goto release_all;
235  }  }
236    
237    
238  int write_ppm(char *filename, unsigned char *image)  /*****************************************************************************
239  {   *                               Main loop
240          FILE *filehandle;   ****************************************************************************/
241          filehandle=fopen(filename,"wb");  
242          if (filehandle)          /* Fill the buffer */
243          {          useful_bytes = fread(mp4_buffer, 1, BUFFER_SIZE, in_file);
244                  fprintf(filehandle,"P6\n\n");           //  
245                  fprintf(filehandle,"%d %d 255\n",XDIM,YDIM);          totaldectime = 0;
246                  fwrite(image,XDIM,YDIM*3,filehandle);          totalsize = 0;
247            filenr = 0;
248            mp4_ptr = mp4_buffer;
249            chunk = 0;
250    
251                  fclose(filehandle);          do {
252                  return 0;                  int used_bytes = 0;
253                    double dectime;
254    
255                    /*
256                     * If the buffer is half empty or there are no more bytes in it
257                     * then fill it.
258                     */
259                    if (mp4_ptr > mp4_buffer + BUFFER_SIZE/2) {
260                            int already_in_buffer = (mp4_buffer + BUFFER_SIZE - mp4_ptr);
261    
262                            /* Move data if needed */
263                            if (already_in_buffer > 0)
264                                    memcpy(mp4_buffer, mp4_ptr, already_in_buffer);
265    
266                            /* Update mp4_ptr */
267                            mp4_ptr = mp4_buffer;
268    
269                            /* read new data */
270                if(!feof(in_file)) {
271                                    useful_bytes += fread(mp4_buffer + already_in_buffer,
272                                                                              1, BUFFER_SIZE - already_in_buffer,
273                                                                              in_file);
274          }          }
         else  
                 return 1;  
275  }  }
276    
 /*********************************************************************/  
 /* Routines for decoding: init encoder, frame step, release encoder  */  
 /*********************************************************************/  
277    
278  int dec_init(int use_assembler) /* init decoder before first run */                  /* This loop is needed to handle VOL/NVOP reading */
279  {                  do {
         int xerr;  
   
         XVID_INIT_PARAM xinit;  
         XVID_DEC_PARAM xparam;  
280    
281                  if(use_assembler)                          /* Decode frame */
282  #ifdef ARCH_IA64                          dectime = msecond();
283                          xinit.cpu_flags = XVID_CPU_FORCE | XVID_CPU_IA64;                          used_bytes = dec_main(mp4_ptr, out_buffer, useful_bytes, &xvid_dec_stats);
284  #else                          dectime = msecond() - dectime;
285                          xinit.cpu_flags = 0;  
286  #endif                          /* Resize image buffer if needed */
287                  else                          if(xvid_dec_stats.type == XVID_TYPE_VOL) {
288                          xinit.cpu_flags = XVID_CPU_FORCE;  
289                                    /* Check if old buffer is smaller */
290                                    if(XDIM*YDIM < xvid_dec_stats.data.vol.width*xvid_dec_stats.data.vol.height) {
291    
292                                            /* Copy new witdh and new height from the vol structure */
293                                            XDIM = xvid_dec_stats.data.vol.width;
294                                            YDIM = xvid_dec_stats.data.vol.height;
295    
296                                            /* Free old output buffer*/
297                                            if(out_buffer) free(out_buffer);
298    
299                                            /* Allocate the new buffer */
300                                            out_buffer = (unsigned char*)malloc(XDIM*YDIM*4);
301                                            if(out_buffer == NULL)
302                                                    goto free_all_memory;
303    
304          xvid_init(NULL, 0, &xinit, NULL);                                          fprintf(stderr, "Resized frame buffer to %dx%d\n", XDIM, YDIM);
305          xparam.width = XDIM;                                  }
         xparam.height = YDIM;  
306    
307          xerr = xvid_decore(NULL, XVID_DEC_CREATE, &xparam, NULL);                                  /* Save individual mpeg4 stream if required */
308          dec_handle = xparam.handle;                                  if(ARG_SAVEMPEGSTREAM) {
309                                            FILE *filehandle = NULL;
310    
311          return xerr;                                          sprintf(filename, "%svolhdr.m4v", filepath);
312                                            filehandle = fopen(filename, "wb");
313                                            if(!filehandle) {
314                                                    fprintf(stderr,
315                                                                    "Error writing vol header mpeg4 stream to file %s\n",
316                                                                    filename);
317                                            } else {
318                                                    fwrite(mp4_ptr, 1, used_bytes, filehandle);
319                                                    fclose(filehandle);
320                                            }
321                                    }
322  }  }
323    
324  int dec_main(unsigned char *m4v_buffer, unsigned char *out_buffer,int *m4v_size)                          /* Update buffer pointers */
325  {       /* decode one frame  */                          if(used_bytes > 0) {
326                                    mp4_ptr += used_bytes;
327                                    useful_bytes -= used_bytes;
328    
329          int xerr;                                  /* Total size */
330          XVID_DEC_FRAME xframe;                                  totalsize += used_bytes;
331                            }
332    
333          xframe.bitstream = m4v_buffer;                          if (display_buffer_bytes) {
334          xframe.length = 9999;                                  printf("Data chunk %d: %d bytes consumed, %d bytes in buffer\n", chunk++, used_bytes, useful_bytes);
335              xframe.image = out_buffer;                          }
336          xframe.stride = XDIM;                  } while (xvid_dec_stats.type <= 0 && useful_bytes > MIN_USEFUL_BYTES);
             xframe.colorspace = XVID_CSP_YV12;  
337    
338                  do {                  /* Check if there is a negative number of useful bytes left in buffer
339                  xerr = xvid_decore(dec_handle, XVID_DEC_DECODE, &xframe, NULL);                   * This means we went too far */
340            if(useful_bytes < 0)
341                break;
342    
343                          *m4v_size = xframe.length;          /* Updated data - Count only usefull decode time */
344                  xframe.bitstream += *m4v_size;                  totaldectime += dectime;
345    
                 } while (*m4v_size<7);  /* this skips N-VOPs */  
346    
347          return xerr;                  if (!display_buffer_bytes) {
348                            printf("Frame %5d: type = %s, dectime(ms) =%6.1f, length(bytes) =%7d\n",
349                                            filenr, type2str(xvid_dec_stats.type), dectime, used_bytes);
350  }  }
351    
352  int dec_stop()  /* close decoder to release resources */                  /* Save individual mpeg4 stream if required */
353  {                  if(ARG_SAVEMPEGSTREAM) {
354          int xerr;                          FILE *filehandle = NULL;
         xerr = xvid_decore(dec_handle, XVID_DEC_DESTROY, NULL, NULL);  
355    
356          return xerr;                          sprintf(filename, "%sframe%05d.m4v", filepath, filenr);
357                            filehandle = fopen(filename, "wb");
358                            if(!filehandle) {
359                                    fprintf(stderr,
360                                                    "Error writing single mpeg4 stream to file %s\n",
361                                                    filename);
362                            }
363                            else {
364                                    fwrite(mp4_ptr-used_bytes, 1, used_bytes, filehandle);
365                                    fclose(filehandle);
366                            }
367  }  }
368    
369                    /* Save output frame if required */
370                    if (ARG_SAVEDECOUTPUT) {
371                            sprintf(filename, "%sdec%05d", filepath, filenr);
372                            if(write_image(filename, out_buffer)) {
373                                    fprintf(stderr,
374                                                    "Error writing decoded frame %s\n",
375                                                    filename);
376                            }
377                    }
378    
379  /*********************************************************************/                  filenr++;
 /*                          Main program                             */  
 /*********************************************************************/  
380    
381  int main(int argc, char *argv[])          } while (useful_bytes>MIN_USEFUL_BYTES || !feof(in_file));
 {  
   unsigned char *divx_buffer = NULL;  
   unsigned char *divx_ptr = NULL;  
   unsigned char *out_buffer = NULL;  
382    
383    double dectime;          useful_bytes = 0; /* Empty buffer */
   double totaldectime=0.;  
384    
385    long totalsize=0;  /*****************************************************************************
386    int status;   *     Flush decoder buffers
387     ****************************************************************************/
388    
389    int m4v_size;          do {
   int frame_type[ABS_MAXFRAMENR];  
   int use_assembler=1;  
390    
391    char filename[256];                  /* Fake vars */
392                    int used_bytes;
393                    double dectime;
394    
395    FILE *filehandle;          do {
396                        dectime = msecond();
397                        used_bytes = dec_main(NULL, out_buffer, -1, &xvid_dec_stats);
398                        dectime = msecond() - dectime;
399                            if (display_buffer_bytes) {
400                                    printf("Data chunk %d: %d bytes consumed, %d bytes in buffer\n", chunk++, used_bytes, useful_bytes);
401                            }
402            } while(used_bytes>=0 && xvid_dec_stats.type <= 0);
403    
404          if (argc>=3)          if (used_bytes < 0) {   /* XVID_ERR_END */
405          {       XDIM = atoi(argv[1]);              break;
                 YDIM = atoi(argv[2]);  
                 if ( (XDIM <= 0) || (XDIM >= 2048) || (YDIM <=0) || (YDIM >= 2048) )  
                 {       fprintf(stderr,"Wrong frames size %d %d, trying PGM \n",XDIM, YDIM);  
406                  }                  }
407    
408                    /* Updated data - Count only usefull decode time */
409                    totaldectime += dectime;
410    
411                    /* Prints some decoding stats */
412                    if (!display_buffer_bytes) {
413                            printf("Frame %5d: type = %s, dectime(ms) =%6.1f, length(bytes) =%7d\n",
414                                            filenr, type2str(xvid_dec_stats.type), dectime, used_bytes);
415          }          }
         if (argc>=4 && !strcmp(argv[3],"noasm"))  
           use_assembler = 0;  
416    
417                    /* Save output frame if required */
418                    if (ARG_SAVEDECOUTPUT) {
419                            sprintf(filename, "%sdec%05d", filepath, filenr);
420                            if(write_image(filename, out_buffer)) {
421                                    fprintf(stderr,
422                                                    "Error writing decoded frame %s\n",
423                                                    filename);
424                            }
425                    }
426    
427  /* allocate memory */                  filenr++;
428    
429    divx_buffer = (unsigned char *) malloc(10*XDIM*YDIM);          }while(1);
   // this should really be enough memory!  
   if (!divx_buffer)  
     goto free_all_memory;  
   divx_ptr = divx_buffer+10*XDIM*YDIM;  
430    
431    out_buffer = (unsigned char *) malloc(4*XDIM*YDIM);   /* YUV needs less */  /*****************************************************************************
432    if (!out_buffer)   *     Calculate totals and averages for output, print results
433      goto free_all_memory;   ****************************************************************************/
434    
435            if (filenr>0) {
436                    totalsize    /= filenr;
437                    totaldectime /= filenr;
438                    printf("Avg: dectime(ms) =%7.2f, fps =%7.2f, length(bytes) =%7d\n",
439                               totaldectime, 1000/totaldectime, (int)totalsize);
440            }else{
441                    printf("Nothing was decoded!\n");
442            }
443    
444  /*********************************************************************/  /*****************************************************************************
445  /*                         XviD PART  Start                          */   *      XviD PART  Stop
446  /*********************************************************************/   ****************************************************************************/
447    
448          status = dec_init(use_assembler);   release_all:
449            if (dec_handle) {
450                    status = dec_stop();
451          if (status)          if (status)
452          {                          fprintf(stderr, "decore RELEASE problem return value %d\n", status);
                 printf("Decore INIT problem, return value %d\n", status);  
                 goto release_all;  
453          }          }
454    
455     free_all_memory:
456            free(out_buffer);
457            free(mp4_buffer);
458    
459            return(0);
460    }
461    
462  /*********************************************************************/  /*****************************************************************************
463  /*                               Main loop                           */   *               Usage function
464  /*********************************************************************/   ****************************************************************************/
465    
466    do  static void usage()
467      {      {
468    
469          if (divx_ptr > divx_buffer+5*XDIM*YDIM) /* buffer more than half empty */          fprintf(stderr, "Usage : xvid_decraw [OPTIONS]\n");
470          {       int rest=(divx_buffer+10*XDIM*YDIM-divx_ptr);          fprintf(stderr, "Options :\n");
471                  if (rest)          fprintf(stderr, " -asm           : use assembly optimizations (default=disabled)\n");
472                          memcpy(divx_buffer, divx_ptr, rest);          fprintf(stderr, " -debug         : debug level (debug=0)\n");
473                  divx_ptr = divx_buffer;          fprintf(stderr, " -i string      : input filename (default=stdin)\n");
474                  fread(divx_buffer+rest, 1, 5*XDIM*YDIM, stdin); /* read new data */          fprintf(stderr, " -d             : save decoder output\n");
475            fprintf(stderr, " -c csp         : choose colorspace output (rgb16, rgb24, rgb32, yv12, i420)\n");
476            fprintf(stderr, " -f format      : choose output file format (tga, pnm, pgm)\n");
477            fprintf(stderr, " -m             : save mpeg4 raw stream to individual files\n");
478            fprintf(stderr, " -help          : This help message\n");
479            fprintf(stderr, " (* means default)\n");
480    
481          }          }
482    
483          dectime = -msecond();  /*****************************************************************************
484          status = dec_main(divx_ptr, out_buffer, &m4v_size);   *               "helper" functions
485     ****************************************************************************/
486    
487          if (status)  /* return the current time in milli seconds */
488    static double
489    msecond()
490          {          {
491                  break;  #ifndef WIN32
492            struct timeval  tv;
493            gettimeofday(&tv, 0);
494            return((double)tv.tv_sec*1.0e3 + (double)tv.tv_usec*1.0e-3);
495    #else
496            clock_t clk;
497            clk = clock();
498            return(clk * 1000.0 / CLOCKS_PER_SEC);
499    #endif
500          }          }
         dectime += msecond();  
         divx_ptr += m4v_size;  
501    
502          totalsize += m4v_size;  /*****************************************************************************
503     *              output functions
504     ****************************************************************************/
505    
506    static int write_image(char *prefix, unsigned char *image)
507    {
508            char filename[1024];
509            char *ext;
510            int ret;
511    
512            if (FORMAT == USE_PNM && BPP == 1) {
513                    ext = "pgm";
514            } else if (FORMAT == USE_PNM && BPP == 3) {
515                    ext = "pnm";
516            } else if (FORMAT == USE_TGA) {
517                    ext = "tga";
518            } else {
519                    fprintf(stderr, "Bug: should not reach this path code -- please report to xvid-devel@xvid.org with command line options used");
520                    exit(-1);
521            }
522    
523            sprintf(filename, "%s.%s", prefix, ext);
524    
525            if (FORMAT == USE_PNM) {
526                    ret = write_pnm(filename, image);
527            } else {
528                    ret = write_tga(filename, image);
529            }
530    
531            return(ret);
532    }
533    
534    static int write_tga(char *filename, unsigned char *image)
535    {
536            FILE * f;
537            char hdr[18];
538    
539            f = fopen(filename, "wb");
540            if ( f == NULL) {
541                    return -1;
542            }
543    
544            hdr[0]  = 0; /* ID length */
545            hdr[1]  = 0; /* Color map type */
546            hdr[2]  = (BPP>1)?2:3; /* Uncompressed true color (2) or greymap (3) */
547            hdr[3]  = 0; /* Color map specification (not used) */
548            hdr[4]  = 0; /* Color map specification (not used) */
549            hdr[5]  = 0; /* Color map specification (not used) */
550            hdr[6]  = 0; /* Color map specification (not used) */
551            hdr[7]  = 0; /* Color map specification (not used) */
552            hdr[8]  = 0; /* LSB X origin */
553            hdr[9]  = 0; /* MSB X origin */
554            hdr[10] = 0; /* LSB Y origin */
555            hdr[11] = 0; /* MSB Y origin */
556            hdr[12] = (XDIM>>0)&0xff; /* LSB Width */
557            hdr[13] = (XDIM>>8)&0xff; /* MSB Width */
558            if (BPP > 1) {
559                    hdr[14] = (YDIM>>0)&0xff; /* LSB Height */
560                    hdr[15] = (YDIM>>8)&0xff; /* MSB Height */
561            } else {
562                    hdr[14] = ((YDIM*3)>>1)&0xff; /* LSB Height */
563                    hdr[15] = ((YDIM*3)>>9)&0xff; /* MSB Height */
564            }
565            hdr[16] = BPP*8;
566            hdr[17] = 0x00 | (1<<5) /* Up to down */ | (0<<4); /* Image descriptor */
567    
568            /* Write header */
569            fwrite(hdr, 1, sizeof(hdr), f);
570    
571    #ifdef ARCH_IS_LITTLE_ENDIAN
572            /* write first plane */
573            fwrite(image, 1, XDIM*YDIM*BPP, f);
574    #else
575            {
576                    int i;
577                    for (i=0; i<XDIM*YDIM*BPP;i+=BPP) {
578                            if (BPP == 1) {
579                                    fputc(*(image+i), f);
580                            } else if (BPP == 2) {
581                                    fputc(*(image+i+1), f);
582                                    fputc(*(image+i+0), f);
583                            } else if (BPP == 3) {
584                                    fputc(*(image+i+2), f);
585                                    fputc(*(image+i+1), f);
586                                    fputc(*(image+i+0), f);
587                            } else if (BPP == 4) {
588                                    fputc(*(image+i+3), f);
589                                    fputc(*(image+i+2), f);
590                                    fputc(*(image+i+1), f);
591                                    fputc(*(image+i+0), f);
592                            }
593                    }
594            }
595    #endif
596    
597          printf("Frame %5d: dectime =%6.1f ms length=%7d bytes \n",          /* Write Y and V planes for YUV formats */
598                          filenr, dectime*1000, m4v_size);          if (BPP == 1) {
599                    int i;
600    
601          if (save_m4v_flag)                  /* Write the two chrominance planes */
602          {                  for (i=0; i<YDIM/2; i++) {
603                  sprintf(filename, "%sframe%05d.m4v", filepath, filenr);                          fwrite(image+XDIM*YDIM + i*XDIM/2, 1, XDIM/2, f);
604                  filehandle = fopen(filename, "wb");                          fwrite(image+5*XDIM*YDIM/4 + i*XDIM/2, 1, XDIM/2, f);
605                  fwrite(divx_buffer, m4v_size, 1, filehandle);                  }
                 fclose(filehandle);  
606          }          }
         totaldectime += dectime;  
607    
608    
609  /*********************************************************************/          /* Close the file */
610  /*        analyse the decoded frame and compare to original          */          fclose(f);
611  /*********************************************************************/  
612            return(0);
613    }
614    
615          if (save_dec_flag)  static int write_pnm(char *filename, unsigned char *image)
616          {          {
617                  sprintf(filename, "%sdec%05d.pgm", filepath, filenr);          FILE * f;
618                  write_pgm(filename,out_buffer);  
619            f = fopen(filename, "wb");
620            if ( f == NULL) {
621                    return -1;
622          }          }
623    
624          filenr++;          if (BPP == 1) {
625                    int i;
626                    fprintf(f, "P5\n%i %i\n255\n", XDIM, YDIM*3/2);
627    
628     } while ( (status>=0) && (filenr<ABS_MAXFRAMENR) );                  fwrite(image, 1, XDIM*YDIM, f);
629    
630                    for (i=0; i<YDIM/2;i++) {
631                            fwrite(image+XDIM*YDIM + i*XDIM/2, 1, XDIM/2, f);
632                            fwrite(image+5*XDIM*YDIM/4 + i*XDIM/2, 1, XDIM/2, f);
633                    }
634            } else if (BPP == 3) {
635                    int i;
636                    fprintf(f, "P6\n#xvid\n%i %i\n255\n", XDIM, YDIM);
637                    for (i=0; i<XDIM*YDIM*3; i+=3) {
638    #ifdef ARCH_IS_LITTLE_ENDIAN
639                            fputc(image[i+2], f);
640                            fputc(image[i+1], f);
641                            fputc(image[i+0], f);
642    #else
643                            fputc(image[i+0], f);
644                            fputc(image[i+1], f);
645                            fputc(image[i+2], f);
646    #endif
647                    }
648            }
649    
650  /*********************************************************************/          fclose(f);
 /*     calculate totals and averages for output, print results       */  
 /*********************************************************************/  
651    
652          totalsize    /= filenr;          return 0;
653          totaldectime /= filenr;  }
654    
655          fprintf(stderr,"Avg: dectime %5.2f ms, %5.2f fps, filesize =%d\n",  /*****************************************************************************
656                  1000*totaldectime, 1./totaldectime, totalsize);   * Routines for decoding: init decoder, use, and stop decoder
657     ****************************************************************************/
658    
659  /*********************************************************************/  /* init decoder before first run */
660  /*                         XviD PART  Stop                           */  static int
661  /*********************************************************************/  dec_init(int use_assembler, int debug_level)
662    {
663            int ret;
664    
665  release_all:          xvid_gbl_init_t   xvid_gbl_init;
666            xvid_dec_create_t xvid_dec_create;
667    
668            /* Reset the structure with zeros */
669            memset(&xvid_gbl_init, 0, sizeof(xvid_gbl_init_t));
670            memset(&xvid_dec_create, 0, sizeof(xvid_dec_create_t));
671    
672            /*------------------------------------------------------------------------
673             * XviD core initialization
674             *----------------------------------------------------------------------*/
675    
676            /* Version */
677            xvid_gbl_init.version = XVID_VERSION;
678    
679            /* Assembly setting */
680            if(use_assembler)
681    #ifdef ARCH_IS_IA64
682                    xvid_gbl_init.cpu_flags = XVID_CPU_FORCE | XVID_CPU_IA64;
683    #else
684            xvid_gbl_init.cpu_flags = 0;
685    #endif
686            else
687                    xvid_gbl_init.cpu_flags = XVID_CPU_FORCE;
688    
689            xvid_gbl_init.debug = debug_level;
690    
691            xvid_global(NULL, 0, &xvid_gbl_init, NULL);
692    
693            /*------------------------------------------------------------------------
694             * XviD encoder initialization
695             *----------------------------------------------------------------------*/
696    
697            /* Version */
698            xvid_dec_create.version = XVID_VERSION;
699    
700            /*
701             * Image dimensions -- set to 0, xvidcore will resize when ever it is
702             * needed
703             */
704            xvid_dec_create.width = 0;
705            xvid_dec_create.height = 0;
706    
707          if (dec_handle)          ret = xvid_decore(NULL, XVID_DEC_CREATE, &xvid_dec_create, NULL);
708    
709            dec_handle = xvid_dec_create.handle;
710    
711            return(ret);
712    }
713    
714    /* decode one frame  */
715    static int
716    dec_main(unsigned char *istream,
717                     unsigned char *ostream,
718                     int istream_size,
719                     xvid_dec_stats_t *xvid_dec_stats)
720          {          {
721                  status = dec_stop();  
722                  if (status)          int ret;
723                          printf("Decore RELEASE problem return value %d\n", status);  
724            xvid_dec_frame_t xvid_dec_frame;
725    
726            /* Reset all structures */
727            memset(&xvid_dec_frame, 0, sizeof(xvid_dec_frame_t));
728            memset(xvid_dec_stats, 0, sizeof(xvid_dec_stats_t));
729    
730            /* Set version */
731            xvid_dec_frame.version = XVID_VERSION;
732            xvid_dec_stats->version = XVID_VERSION;
733    
734            /* No general flags to set */
735            xvid_dec_frame.general          = 0;
736    
737            /* Input stream */
738            xvid_dec_frame.bitstream        = istream;
739            xvid_dec_frame.length           = istream_size;
740    
741            /* Output frame structure */
742            xvid_dec_frame.output.plane[0]  = ostream;
743            xvid_dec_frame.output.stride[0] = XDIM*BPP;
744            xvid_dec_frame.output.csp = CSP;
745    
746            ret = xvid_decore(dec_handle, XVID_DEC_DECODE, &xvid_dec_frame, xvid_dec_stats);
747    
748            return(ret);
749          }          }
750    
751  free_all_memory:  /* close decoder to release resources */
752          free(out_buffer);  static int
753          free(divx_buffer);  dec_stop()
754    {
755            int ret;
756    
757    return 0;          ret = xvid_decore(dec_handle, XVID_DEC_DESTROY, NULL, NULL);
758    
759            return(ret);
760  }  }
Legend:
Removed from v.376  
changed lines
  Added in v.1678
No admin address has been configured
 ViewVC Help
Powered by ViewVC 1.0.4