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