[svn] / branches / release-1_0-branch / xvidcore / examples / xvid_decraw.c Repository:
ViewVC logotype

Diff of /branches/release-1_0-branch/xvidcore/examples/xvid_decraw.c

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

trunk/xvidcore/examples/xvid_decraw.c revision 551, Fri Sep 27 20:58:30 2002 UTC branches/release-1_0-branch/xvidcore/examples/xvid_decraw.c revision 1418, Sat Apr 10 10:05:47 2004 UTC
# Line 3  Line 3 
3   *  XVID MPEG-4 VIDEO CODEC   *  XVID MPEG-4 VIDEO CODEC
4   *  - Console based decoding test application  -   *  - Console based decoding test application  -
5   *   *
6   *  Copyright(C) 2002 Christoph Lampert   *  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 19  Line 20 
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., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA   *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
22   *   *
23   * $Id: xvid_decraw.c,v 1.3 2002-09-27 20:58:30 edgomez Exp $   * $Id: xvid_decraw.c,v 1.10.2.2 2004-04-10 10:05:47 edgomez Exp $
24   *   *
25   ****************************************************************************/   ****************************************************************************/
26    
# Line 31  Line 32 
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.
36   *   *
37   *   gcc xvid_decraw.c -lxvidcore -lm -o xvid_decraw   *  Use ./xvid_decraw -help for a list of options
  *  
  *  You have to specify the image dimensions (until the add the feature  
  *  to read this from the bitstream)  
  *  
  * Usage : xvid_decraw <-w width> <-h height> [OPTIONS]  
  * Options :  
  *  -asm           : use assembly optimizations (default=disabled)  
  *  -w integer     : frame width ([1.2048])  
  *  -h integer     : frame height ([1.2048])  
  *  -i string      : input filename (default=stdin)  
  *  -t integer     : input data type (raw=0, mp4u=1)  
  *  -d boolean     : save decoder output (0 False*, !=0 True)  
  *  -m boolean     : save mpeg4 raw stream to single files (0 False*, !=0 True)  
  *  -help          : This help message  
  * (* means default)  
38   *   *
39   ****************************************************************************/   ****************************************************************************/
40    
# Line 56  Line 42 
42  #include <stdlib.h>  #include <stdlib.h>
43  #include <string.h>  #include <string.h>
44  #include <math.h>  #include <math.h>
45  #ifndef _MSC_VER  #ifndef WIN32
46  #include <sys/time.h>  #include <sys/time.h>
47  #else  #else
48  #include <time.h>  #include <time.h>
# Line 71  Line 57 
57  /* max number of frames */  /* max number of frames */
58  #define ABS_MAXFRAMENR 9999  #define ABS_MAXFRAMENR 9999
59    
60    #define USE_PNM 0
61    #define USE_TGA 1
62    
63  static int XDIM = 0;  static int XDIM = 0;
64  static int YDIM = 0;  static int YDIM = 0;
65  static int ARG_SAVEDECOUTPUT = 0;  static int ARG_SAVEDECOUTPUT = 0;
66  static int ARG_SAVEMPEGSTREAM = 0;  static int ARG_SAVEMPEGSTREAM = 0;
 static int ARG_STREAMTYPE = 0;  
67  static char *ARG_INPUTFILE = NULL;  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] = "./";  static char filepath[256] = "./";
73  static void *dec_handle = NULL;  static void *dec_handle = NULL;
74    
75  # define BUFFER_SIZE 10*XDIM*YDIM  #define BUFFER_SIZE (2*1024*1024)
   
 #define LONG_PACK(a,b,c,d) ((long) (((long)(a))<<24) | (((long)(b))<<16) | \  
                                    (((long)(c))<<8)  |((long)(d)))  
   
 #define SWAP(a) ( (((a)&0x000000ff)<<24) | (((a)&0x0000ff00)<<8) | \  
                   (((a)&0x00ff0000)>>8)  | (((a)&0xff000000)>>24) )  
76    
77  /*****************************************************************************  /*****************************************************************************
78   *               Local prototypes   *               Local prototypes
# Line 101  Line 85 
85  static int dec_main(unsigned char *istream,  static int dec_main(unsigned char *istream,
86                      unsigned char *ostream,                      unsigned char *ostream,
87                      int istream_size,                      int istream_size,
88                      int *ostream_size);                                          xvid_dec_stats_t *xvid_dec_stats);
89  static int dec_stop();  static int dec_stop();
90  static void usage();  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   *        Main program
106   ****************************************************************************/   ****************************************************************************/
# Line 114  Line 110 
110          unsigned char *mp4_buffer = NULL;          unsigned char *mp4_buffer = NULL;
111          unsigned char *mp4_ptr    = NULL;          unsigned char *mp4_ptr    = NULL;
112          unsigned char *out_buffer = NULL;          unsigned char *out_buffer = NULL;
113          int bigendian = 0;          int useful_bytes;
114            xvid_dec_stats_t xvid_dec_stats;
115    
116          double totaldectime;          double totaldectime;
117    
# Line 130  Line 127 
127          int i;          int i;
128    
129          printf("xvid_decraw - raw mpeg4 bitstream decoder ");          printf("xvid_decraw - raw mpeg4 bitstream decoder ");
130          printf("written by Christoph Lampert 2002\n\n");          printf("written by Christoph Lampert 2002-2003\n\n");
131    
132  /*****************************************************************************  /*****************************************************************************
133   * Command line parsing   * Command line parsing
# Line 140  Line 137 
137    
138                  if (strcmp("-asm", argv[i]) == 0 ) {                  if (strcmp("-asm", argv[i]) == 0 ) {
139                          use_assembler = 1;                          use_assembler = 1;
140                  }                  } else if (strcmp("-d", argv[i]) == 0) {
141                  else if (strcmp("-w", argv[i]) == 0 && i < argc - 1 ) {                          ARG_SAVEDECOUTPUT = 1;
142                          i++;                  } else if (strcmp("-i", argv[i]) == 0 && i < argc - 1 ) {
                         XDIM = atoi(argv[i]);  
                 }  
                 else if (strcmp("-h", argv[i]) == 0 && i < argc - 1 ) {  
                         i++;  
                         YDIM = atoi(argv[i]);  
                 }  
                 else if (strcmp("-d", argv[i]) == 0 && i < argc - 1 ) {  
                         i++;  
                         ARG_SAVEDECOUTPUT = atoi(argv[i]);  
                 }  
                 else if (strcmp("-i", argv[i]) == 0 && i < argc - 1 ) {  
143                          i++;                          i++;
144                          ARG_INPUTFILE = argv[i];                          ARG_INPUTFILE = argv[i];
145                  }                  } else if (strcmp("-m", argv[i]) == 0) {
146                  else if (strcmp("-m", argv[i]) == 0 && i < argc - 1 ) {                          ARG_SAVEMPEGSTREAM = 1;
147                    } else if (strcmp("-c", argv[i]) == 0  && i < argc - 1 ) {
148                          i++;                          i++;
149                          ARG_SAVEMPEGSTREAM = atoi(argv[i]);                          if (strcmp(argv[i], "rgb16") == 0) {
150                                    CSP = XVID_CSP_RGB555;
151                                    BPP = 2;
152                            } else if (strcmp(argv[i], "rgb24") == 0) {
153                                    CSP = XVID_CSP_BGR;
154                                    BPP = 3;
155                            } else if (strcmp(argv[i], "rgb32") == 0) {
156                                    CSP = XVID_CSP_BGRA;
157                                    BPP = 4;
158                            } else if (strcmp(argv[i], "yv12") == 0) {
159                                    CSP = XVID_CSP_YV12;
160                                    BPP = 1;
161                            } else {
162                                    CSP = XVID_CSP_I420;
163                                    BPP = 1;
164                  }                  }
165                  else if (strcmp("-t", argv[i]) == 0 && i < argc - 1 ) {                  } else if (strcmp("-f", argv[i]) == 0 && i < argc -1) {
166                          i++;                          i++;
167                          ARG_STREAMTYPE = atoi(argv[i]);                          if (strcmp(argv[i], "tga") == 0) {
168                                    FORMAT = USE_TGA;
169                            } else {
170                                    FORMAT = USE_PNM;
171                  }                  }
172                  else if (strcmp("-help", argv[i])) {                  } else if (strcmp("-help", argv[i]) == 0) {
173                          usage();                          usage();
174                          return(0);                          return(0);
175                  }                  } else {
                 else {  
176                          usage();                          usage();
177                          exit(-1);                          exit(-1);
178                  }                  }
   
179          }          }
180    
181  /*****************************************************************************  /*****************************************************************************
182   * Values checking   * Values checking
183   ****************************************************************************/   ****************************************************************************/
184    
         if(XDIM <= 0 || XDIM > 2048 || YDIM <= 0 || YDIM > 2048) {  
                 usage();  
                 return -1;  
         }  
   
185          if ( ARG_INPUTFILE == NULL || strcmp(ARG_INPUTFILE, "stdin") == 0) {          if ( ARG_INPUTFILE == NULL || strcmp(ARG_INPUTFILE, "stdin") == 0) {
186                  in_file = stdin;                  in_file = stdin;
187          }          }
# Line 193  Line 190 
190                  in_file = fopen(ARG_INPUTFILE, "rb");                  in_file = fopen(ARG_INPUTFILE, "rb");
191                  if (in_file == NULL) {                  if (in_file == NULL) {
192                          fprintf(stderr, "Error opening input file %s\n", ARG_INPUTFILE);                          fprintf(stderr, "Error opening input file %s\n", ARG_INPUTFILE);
193                          return -1;                          return(-1);
194                    }
195                  }                  }
196    
197            /* PNM/PGM format can't handle 16/32 bit data */
198            if (BPP != 1 && BPP != 3 && FORMAT == USE_PNM) {
199                    FORMAT = USE_TGA;
200          }          }
201    
202  /*****************************************************************************  /*****************************************************************************
# Line 207  Line 209 
209          if (!mp4_buffer)          if (!mp4_buffer)
210                  goto free_all_memory;                  goto free_all_memory;
211    
         /* Memory for frame output */  
         out_buffer = (unsigned char *) malloc(XDIM*YDIM*4);  
         if (!out_buffer)  
                 goto free_all_memory;  
   
   
212  /*****************************************************************************  /*****************************************************************************
213   *        XviD PART  Start   *        XviD PART  Start
214   ****************************************************************************/   ****************************************************************************/
# Line 229  Line 225 
225   *                               Main loop   *                               Main loop
226   ****************************************************************************/   ****************************************************************************/
227    
228          if(ARG_STREAMTYPE) {          /* Fill the buffer */
229            useful_bytes = fread(mp4_buffer, 1, BUFFER_SIZE, in_file);
                 unsigned char header[4];  
   
                 /* MP4U format  : read header */  
                 if(feof(in_file))  
                         goto release_all;  
                 fread(header, 4, 1, in_file);  
   
                 if(header[0] != '2' || header[1] != 'M' ||  
                    header[2] != 'O' || header[3] != 'G') {  
                         fprintf(stderr, "Error, this not a mp4u container file\n");  
                         goto release_all;  
                 }  
   
         }  
   
         totalsize = LONG_PACK('M','P','4','U');  
         mp4_ptr = (unsigned char *)&totalsize;  
         if(*mp4_ptr == 'M')  
                 bigendian = 1;  
         else  
                 bigendian = 0;  
230    
231          totaldectime = 0;          totaldectime = 0;
232          totalsize = 0;          totalsize = 0;
# Line 259  Line 234 
234          mp4_ptr = mp4_buffer;          mp4_ptr = mp4_buffer;
235    
236          do {          do {
   
                 int mp4_size = (mp4_buffer + BUFFER_SIZE - mp4_ptr);  
237                  int used_bytes = 0;                  int used_bytes = 0;
238                  double dectime;                  double dectime;
239    
                 /* Read data from input file */  
                 if(ARG_STREAMTYPE) {  
   
                         /* MP4U container */  
   
                         /* Read stream size first */  
                         if(feof(in_file))  
                                 break;  
                         fread(&mp4_size, sizeof(long), 1, in_file);  
   
                         if(bigendian)  
                                 mp4_size = SWAP(mp4_size);  
   
                         /* Read mp4_size_bytes */  
                         if(feof(in_file))  
                                 break;  
                         fread(mp4_buffer, mp4_size, 1, in_file);  
   
240                          /*                          /*
241                           * When reading mp4u, we don't have to care about buffer                   * If the buffer is half empty or there are no more bytes in it
242                           * filling as we know exactly how much bytes there are in                   * then fill it.
                          * next frame  
243                           */                           */
                         mp4_ptr = mp4_buffer;  
   
                 }  
                 else {  
   
                         /* Real raw stream */  
   
                         /* buffer more than half empty -> Fill it */  
244                          if (mp4_ptr > mp4_buffer + BUFFER_SIZE/2) {                          if (mp4_ptr > mp4_buffer + BUFFER_SIZE/2) {
245                                  int rest = (mp4_buffer + BUFFER_SIZE - mp4_ptr);                          int already_in_buffer = (mp4_buffer + BUFFER_SIZE - mp4_ptr);
246    
247                                  /* Move data if needed */                                  /* Move data if needed */
248                                  if (rest)                          if (already_in_buffer > 0)
249                                          memcpy(mp4_buffer, mp4_ptr, rest);                                  memcpy(mp4_buffer, mp4_ptr, already_in_buffer);
250    
251                                  /* Update mp4_ptr */                                  /* Update mp4_ptr */
252                                  mp4_ptr = mp4_buffer;                                  mp4_ptr = mp4_buffer;
# Line 308  Line 254 
254                                  /* read new data */                                  /* read new data */
255                                  if(feof(in_file))                                  if(feof(in_file))
256                                          break;                                          break;
257                                  fread(mp4_buffer + rest, BUFFER_SIZE - rest, 1, in_file);  
258                          }                          useful_bytes += fread(mp4_buffer + already_in_buffer,
259                                                                      1, BUFFER_SIZE - already_in_buffer,
260                                                                      in_file);
261    
262                  }                  }
263    
264    
265                    /* This loop is needed to handle VOL/NVOP reading */
266                    do {
267    
268                  /* Decode frame */                  /* Decode frame */
269                  dectime = msecond();                  dectime = msecond();
270                  status = dec_main(mp4_ptr, out_buffer, mp4_size, &used_bytes);                          used_bytes = dec_main(mp4_ptr, out_buffer, useful_bytes, &xvid_dec_stats);
271                  dectime = msecond() - dectime;                  dectime = msecond() - dectime;
272    
273                  if (status) {                          /* Resize image buffer if needed */
274                          break;                          if(xvid_dec_stats.type == XVID_TYPE_VOL) {
275    
276                                    /* Check if old buffer is smaller */
277                                    if(XDIM*YDIM < xvid_dec_stats.data.vol.width*xvid_dec_stats.data.vol.height) {
278    
279                                            /* Copy new witdh and new height from the vol structure */
280                                            XDIM = xvid_dec_stats.data.vol.width;
281                                            YDIM = xvid_dec_stats.data.vol.height;
282    
283                                            /* Free old output buffer*/
284                                            if(out_buffer) free(out_buffer);
285    
286                                            /* Allocate the new buffer */
287                                            out_buffer = (unsigned char*)malloc(XDIM*YDIM*4);
288                                            if(out_buffer == NULL)
289                                                    goto free_all_memory;
290    
291                                            fprintf(stderr, "Resized frame buffer to %dx%d\n", XDIM, YDIM);
292                                    }
293                  }                  }
294    
295                  /*                          /* Update buffer pointers */
296                   * Only needed for real raw stream, mp4u uses                          if(used_bytes > 0) {
                  * mp4_ptr = mp4_buffer for each frame  
                  */  
297                  mp4_ptr += used_bytes;                  mp4_ptr += used_bytes;
298                                    useful_bytes -= used_bytes;
299    
300                  /* Updated data */                                  /* Total size */
301                  totalsize += used_bytes;                  totalsize += used_bytes;
302                            }
303    
304                    }while(xvid_dec_stats.type <= 0 && useful_bytes > 0);
305    
306                    /* Check if there is a negative number of useful bytes left in buffer
307                     * This means we went too far */
308            if(useful_bytes < 0)
309                break;
310    
311            /* Updated data - Count only usefull decode time */
312                  totaldectime += dectime;                  totaldectime += dectime;
313    
                 /* Prints some decoding stats */  
                 printf("Frame %5d: dectime =%6.1f ms length=%7d bytes \n",  
                        filenr, dectime, used_bytes);  
314    
315                  /* Save individual mpeg4 strean if required */          printf("Frame %5d: type = %s, dectime(ms) =%6.1f, length(bytes) =%7d\n",
316                               filenr, type2str(xvid_dec_stats.type), dectime, used_bytes);
317    
318                    /* Save individual mpeg4 stream if required */
319                  if (ARG_SAVEMPEGSTREAM) {                  if (ARG_SAVEMPEGSTREAM) {
320                          FILE *filehandle = NULL;                          FILE *filehandle = NULL;
321    
# Line 348  Line 327 
327                                          filename);                                          filename);
328                          }                          }
329                          else {                          else {
330                                  fwrite(mp4_buffer, used_bytes, 1, filehandle);                                  fwrite(mp4_ptr-used_bytes, 1, used_bytes, filehandle);
331                                  fclose(filehandle);                                  fclose(filehandle);
332                          }                          }
333                  }                  }
334    
   
335                  /* Save output frame if required */                  /* Save output frame if required */
336                  if (ARG_SAVEDECOUTPUT) {                  if (ARG_SAVEDECOUTPUT) {
337                          sprintf(filename, "%sdec%05d.pgm", filepath, filenr);                          sprintf(filename, "%sdec%05d", filepath, filenr);
338                          if(write_pgm(filename,out_buffer)) {                          if(write_image(filename, out_buffer)) {
339                                  fprintf(stderr,                                  fprintf(stderr,
340                                          "Error writing decoded PGM frame %s\n",                                          "Error writing decoded PGM frame %s\n",
341                                          filename);                                          filename);
# Line 368  Line 346 
346    
347          } while ( (status>=0) && (filenr<ABS_MAXFRAMENR) );          } while ( (status>=0) && (filenr<ABS_MAXFRAMENR) );
348    
349    /*****************************************************************************
350     *     Flush decoder buffers
351     ****************************************************************************/
352    
353            do {
354    
355                    /* Fake vars */
356                    int used_bytes;
357                    double dectime;
358    
359            do {
360                        dectime = msecond();
361                        used_bytes = dec_main(NULL, out_buffer, -1, &xvid_dec_stats);
362                        dectime = msecond() - dectime;
363            }while(used_bytes>=0 && xvid_dec_stats.type <= 0);
364    
365            if (used_bytes < 0) {   /* XVID_ERR_END */
366                break;
367            }
368    
369                    /* Updated data - Count only usefull decode time */
370                    totaldectime += dectime;
371    
372                    /* Prints some decoding stats */
373            printf("Frame %5d: type = %s, dectime(ms) =%6.1f, length(bytes) =%7d\n",
374                               filenr, type2str(xvid_dec_stats.type), dectime, used_bytes);
375    
376                    /* Save output frame if required */
377                    if (ARG_SAVEDECOUTPUT) {
378                            sprintf(filename, "%sdec%05d", filepath, filenr);
379                            if(write_image(filename, out_buffer)) {
380                                    fprintf(stderr,
381                                                    "Error writing decoded PGM frame %s\n",
382                                                    filename);
383                            }
384                    }
385    
386                    filenr++;
387    
388            }while(1);
389    
390  /*****************************************************************************  /*****************************************************************************
391   *     Calculate totals and averages for output, print results   *     Calculate totals and averages for output, print results
# Line 376  Line 394 
394          totalsize    /= filenr;          totalsize    /= filenr;
395          totaldectime /= filenr;          totaldectime /= filenr;
396    
397          printf("Avg: dectime %5.2f ms, %5.2f fps, mp4 stream size =%d\n",          printf("Avg: dectime(ms) =%7.2f, fps =%7.2f, length(bytes) =%7d\n",
398                  totaldectime, 1000/totaldectime, totalsize);                     totaldectime, 1000/totaldectime, (int)totalsize);
399    
400  /*****************************************************************************  /*****************************************************************************
401   *      XviD PART  Stop   *      XviD PART  Stop
402  /****************************************************************************/   ****************************************************************************/
403    
404   release_all:   release_all:
405          if (dec_handle) {          if (dec_handle) {
# Line 394  Line 412 
412          free(out_buffer);          free(out_buffer);
413          free(mp4_buffer);          free(mp4_buffer);
414    
415          return 0;          return(0);
416  }  }
417    
418  /*****************************************************************************  /*****************************************************************************
# Line 404  Line 422 
422  static void usage()  static void usage()
423  {  {
424    
425          fprintf(stderr, "Usage : xvid_decraw <-w width> <-h height> [OPTIONS]\n");          fprintf(stderr, "Usage : xvid_decraw [OPTIONS]\n");
426          fprintf(stderr, "Options :\n");          fprintf(stderr, "Options :\n");
427          fprintf(stderr, " -asm           : use assembly optimizations (default=disabled)\n");          fprintf(stderr, " -asm           : use assembly optimizations (default=disabled)\n");
         fprintf(stderr, " -w integer     : frame width ([1.2048])\n");  
         fprintf(stderr, " -h integer     : frame height ([1.2048])\n");  
428          fprintf(stderr, " -i string      : input filename (default=stdin)\n");          fprintf(stderr, " -i string      : input filename (default=stdin)\n");
429          fprintf(stderr, " -t integer     : input data type (raw=0, mp4u=1)\n");          fprintf(stderr, " -d             : save decoder output\n");
430          fprintf(stderr, " -d boolean     : save decoder output (0 False*, !=0 True)\n");          fprintf(stderr, " -c csp         : choose colorspace output (rgb16, rgb24, rgb32, yv12, i420)\n");
431          fprintf(stderr, " -m boolean     : save mpeg4 raw stream to individual files (0 False*, !=0 True)\n");          fprintf(stderr, " -f format      : choose output file format (tga, pnm, pgm)\n");
432            fprintf(stderr, " -m             : save mpeg4 raw stream to individual files\n");
433          fprintf(stderr, " -help          : This help message\n");          fprintf(stderr, " -help          : This help message\n");
434          fprintf(stderr, " (* means default)\n");          fprintf(stderr, " (* means default)\n");
435    
# Line 426  Line 443 
443  static double  static double
444  msecond()  msecond()
445  {  {
446  #ifndef _MSC_VER  #ifndef WIN32
447          struct timeval  tv;          struct timeval  tv;
448          gettimeofday(&tv, 0);          gettimeofday(&tv, 0);
449          return (double)tv.tv_sec*1.0e3 + (double)tv.tv_usec*1.0e-3;          return((double)tv.tv_sec*1.0e3 + (double)tv.tv_usec*1.0e-3);
450  #else  #else
451          clock_t clk;          clock_t clk;
452          clk = clock();          clk = clock();
453          return clk * 1000 / CLOCKS_PER_SEC;          return(clk * 1000 / CLOCKS_PER_SEC);
454  #endif  #endif
455  }  }
456    
   
457  /*****************************************************************************  /*****************************************************************************
458   *              output functions   *              output functions
459   ****************************************************************************/   ****************************************************************************/
460    
461  static int  int write_image(char *prefix, unsigned char *image)
 write_pgm(char *filename,  
           unsigned char *image)  
462  {  {
463          int loop;          char filename[1024];
464            char *ext;
465            int ret;
466    
467            if (FORMAT == USE_PNM && BPP == 1) {
468                    ext = "pgm";
469            } else if (FORMAT == USE_PNM && BPP == 3) {
470                    ext = "pnm";
471            } else if (FORMAT == USE_TGA) {
472                    ext = "tga";
473            } else {
474                    fprintf(stderr, "Bug: should not reach this path code -- please report to xvid-devel@xvid.org with command line options used");
475                    exit(-1);
476            }
477    
478          unsigned char *y = image;          sprintf(filename, "%s.%s", prefix, ext);
         unsigned char *u = image + XDIM*YDIM;  
         unsigned char *v = image + XDIM*YDIM + XDIM/2*YDIM/2;  
   
         FILE *filehandle;  
         filehandle=fopen(filename,"w+b");  
         if (filehandle) {  
479    
480                  /* Write header */          if (FORMAT == USE_PNM) {
481                  fprintf(filehandle,"P5\n\n%d %d 255\n", XDIM,YDIM*3/2);                  ret = write_pnm(filename, image);
482            } else {
483                    ret = write_tga(filename, image);
484            }
485    
486                  /* Write Y data */          return(ret);
487                  fwrite(y, 1, XDIM*YDIM, filehandle);  }
488    
489                  for(loop=0; loop<YDIM/2; loop++)  int write_tga(char *filename, unsigned char *image)
490                  {                  {
491                          /* Write U scanline */          FILE * f;
492                          fwrite(u, 1, XDIM/2, filehandle);          char hdr[18];
493    
494            f = fopen(filename, "wb");
495            if ( f == NULL) {
496                    return -1;
497            }
498    
499                          /* Write V scanline */          hdr[0]  = 0; /* ID length */
500                          fwrite(v, 1, XDIM/2, filehandle);          hdr[1]  = 0; /* Color map type */
501            hdr[2]  = (BPP>1)?2:3; /* Uncompressed true color (2) or greymap (3) */
502            hdr[3]  = 0; /* Color map specification (not used) */
503            hdr[4]  = 0; /* Color map specification (not used) */
504            hdr[5]  = 0; /* Color map specification (not used) */
505            hdr[6]  = 0; /* Color map specification (not used) */
506            hdr[7]  = 0; /* Color map specification (not used) */
507            hdr[8]  = 0; /* LSB X origin */
508            hdr[9]  = 0; /* MSB X origin */
509            hdr[10] = 0; /* LSB Y origin */
510            hdr[11] = 0; /* MSB Y origin */
511            hdr[12] = (XDIM>>0)&0xff; /* LSB Width */
512            hdr[13] = (XDIM>>8)&0xff; /* MSB Width */
513            if (BPP > 1) {
514                    hdr[14] = (YDIM>>0)&0xff; /* LSB Height */
515                    hdr[15] = (YDIM>>8)&0xff; /* MSB Height */
516            } else {
517                    hdr[14] = ((YDIM*3)>>1)&0xff; /* LSB Height */
518                    hdr[15] = ((YDIM*3)>>9)&0xff; /* MSB Height */
519            }
520            hdr[16] = BPP*8;
521            hdr[17] = 0x00 | (1<<5) /* Up to down */ | (0<<4); /* Image descriptor */
522    
523                          /* Update pointers */          /* Write header */
524                          u += XDIM/2;          fwrite(hdr, 1, sizeof(hdr), f);
                         v += XDIM/2;  
525    
526    #ifdef ARCH_IS_LITTLE_ENDIAN
527            /* write first plane */
528            fwrite(image, 1, XDIM*YDIM*BPP, f);
529    #else
530            {
531                    int i;
532                    for (i=0; i<width*height*BPP;i+=BPP) {
533                            if (BPP == 1) {
534                                    fputc(image+i, f);
535                            } else if (BPP == 2) {
536                                    fputc(image+i+1, f);
537                                    fputc(image+i+0, f);
538                            } else if (BPP == 3) {
539                                    fputc(image+i+2, f);
540                                    fputc(image+i+1, f);
541                                    fputc(image+i+0, f);
542                            } else if (BPP == 4) {
543                                    fputc(image+i+3, f);
544                                    fputc(image+i+2, f);
545                                    fputc(image+i+1, f);
546                                    fputc(image+i+0, f);
547                            }
548                    }
549                  }                  }
550    #endif
551    
552                  /* Close file */          /* Write Y and V planes for YUV formats */
553                  fclose(filehandle);          if (BPP == 1) {
554                    int i;
555    
556                  return 0;                  /* Write the two chrominance planes */
557                    for (i=0; i<YDIM/2; i++) {
558                            fwrite(image+XDIM*YDIM + i*XDIM/2, 1, XDIM/2, f);
559                            fwrite(image+5*XDIM*YDIM/4 + i*XDIM/2, 1, XDIM/2, f);
560          }          }
         else  
                 return 1;  
561  }  }
562    
563    
564            /* Close the file */
565            fclose(f);
566    
567            return(0);
568    }
569    
570    int write_pnm(char *filename, unsigned char *image)
571    {
572            FILE * f;
573    
574            f = fopen(filename, "wb");
575            if ( f == NULL) {
576                    return -1;
577            }
578    
579            if (BPP == 1) {
580                    int i;
581                    fprintf(f, "P5\n#xvid\n%i %i\n255\n", XDIM, YDIM*3/2);
582    
583                    fwrite(image, 1, XDIM*YDIM, f);
584    
585                    for (i=0; i<YDIM/2;i++) {
586                            fwrite(image+XDIM*YDIM + i*XDIM/2, 1, XDIM/2, f);
587                            fwrite(image+5*XDIM*YDIM/4 + i*XDIM/2, 1, XDIM/2, f);
588                    }
589            } else if (BPP == 3) {
590                    int i;
591                    fprintf(f, "P6\n#xvid\n%i %i\n255\n", XDIM, YDIM);
592                    for (i=0; i<XDIM*YDIM*3; i+=3) {
593    #ifdef ARCH_IS_LITTLE_ENDIAN
594                            fputc(image[i+2], f);
595                            fputc(image[i+1], f);
596                            fputc(image[i+0], f);
597    #else
598                            fputc(image[i+0], f);
599                            fputc(image[i+1], f);
600                            fputc(image[i+2], f);
601    #endif
602                    }
603            }
604    
605            fclose(f);
606    
607            return 0;
608    }
609  /*****************************************************************************  /*****************************************************************************
610   * Routines for decoding: init decoder, use, and stop decoder   * Routines for decoding: init decoder, use, and stop decoder
611   ****************************************************************************/   ****************************************************************************/
# Line 493  Line 614 
614  static int  static int
615  dec_init(int use_assembler)  dec_init(int use_assembler)
616  {  {
617          int xerr;          int ret;
618    
619            xvid_gbl_init_t   xvid_gbl_init;
620            xvid_dec_create_t xvid_dec_create;
621    
622          XVID_INIT_PARAM xinit;          /*------------------------------------------------------------------------
623          XVID_DEC_PARAM xparam;           * XviD core initialization
624             *----------------------------------------------------------------------*/
625    
626            /* Version */
627            xvid_gbl_init.version = XVID_VERSION;
628    
629            /* Assembly setting */
630                  if(use_assembler)                  if(use_assembler)
631  #ifdef ARCH_IA64  #ifdef ARCH_IS_IA64
632                          xinit.cpu_flags = XVID_CPU_FORCE | XVID_CPU_IA64;                  xvid_gbl_init.cpu_flags = XVID_CPU_FORCE | XVID_CPU_IA64;
633  #else  #else
634                          xinit.cpu_flags = 0;          xvid_gbl_init.cpu_flags = 0;
635  #endif  #endif
636                  else                  else
637                          xinit.cpu_flags = XVID_CPU_FORCE;                  xvid_gbl_init.cpu_flags = XVID_CPU_FORCE;
638    
639            xvid_global(NULL, 0, &xvid_gbl_init, NULL);
640    
641            /*------------------------------------------------------------------------
642             * XviD encoder initialization
643             *----------------------------------------------------------------------*/
644    
645            /* Version */
646            xvid_dec_create.version = XVID_VERSION;
647    
648            /*
649             * Image dimensions -- set to 0, xvidcore will resize when ever it is
650             * needed
651             */
652            xvid_dec_create.width = 0;
653            xvid_dec_create.height = 0;
654    
655          xvid_init(NULL, 0, &xinit, NULL);          ret = xvid_decore(NULL, XVID_DEC_CREATE, &xvid_dec_create, NULL);
         xparam.width = XDIM;  
         xparam.height = YDIM;  
656    
657          xerr = xvid_decore(NULL, XVID_DEC_CREATE, &xparam, NULL);          dec_handle = xvid_dec_create.handle;
         dec_handle = xparam.handle;  
658    
659          return xerr;          return(ret);
660  }  }
661    
662  /* decode one frame  */  /* decode one frame  */
# Line 522  Line 664 
664  dec_main(unsigned char *istream,  dec_main(unsigned char *istream,
665           unsigned char *ostream,           unsigned char *ostream,
666           int istream_size,           int istream_size,
667           int *ostream_size)                   xvid_dec_stats_t *xvid_dec_stats)
668  {  {
669    
670          int xerr;          int ret;
671          XVID_DEC_FRAME xframe;  
672            xvid_dec_frame_t xvid_dec_frame;
673    
674            /* Set version */
675            xvid_dec_frame.version = XVID_VERSION;
676            xvid_dec_stats->version = XVID_VERSION;
677    
678            /* No general flags to set */
679            xvid_dec_frame.general          = 0;
680    
681          xframe.bitstream = istream;          /* Input stream */
682          xframe.length = istream_size;          xvid_dec_frame.bitstream        = istream;
683          xframe.image = ostream;          xvid_dec_frame.length           = istream_size;
         xframe.stride = XDIM;  
         xframe.colorspace = XVID_CSP_YV12;  
684    
685          xerr = xvid_decore(dec_handle, XVID_DEC_DECODE, &xframe, NULL);          /* Output frame structure */
686            xvid_dec_frame.output.plane[0]  = ostream;
687            xvid_dec_frame.output.stride[0] = XDIM*BPP;
688            xvid_dec_frame.output.csp = CSP;
689    
690          *ostream_size = xframe.length;          ret = xvid_decore(dec_handle, XVID_DEC_DECODE, &xvid_dec_frame, xvid_dec_stats);
691    
692          return xerr;          return(ret);
693  }  }
694    
695  /* close decoder to release resources */  /* close decoder to release resources */
696  static int  static int
697  dec_stop()  dec_stop()
698  {  {
699          int xerr;          int ret;
700    
701          xerr = xvid_decore(dec_handle, XVID_DEC_DESTROY, NULL, NULL);          ret = xvid_decore(dec_handle, XVID_DEC_DESTROY, NULL, NULL);
702    
703          return xerr;          return(ret);
704  }  }
Legend:
Removed from v.551  
changed lines
  Added in v.1418
No admin address has been configured
 ViewVC Help
Powered by ViewVC 1.0.4