[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 851, Sat Feb 15 15:22:19 2003 UTC branches/release-1_0-branch/xvidcore/examples/xvid_decraw.c revision 1430, Thu Apr 15 10:04:55 2004 UTC
# Line 4  Line 4 
4   *  - Console based decoding test application  -   *  - Console based decoding test application  -
5   *   *
6   *  Copyright(C) 2002-2003 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.6 2003-02-15 15:22:17 edgomez Exp $   * $Id: xvid_decraw.c,v 1.10.2.3 2004-04-15 10:04:55 suxen_drol 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 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;
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)
76    
77  /*****************************************************************************  /*****************************************************************************
78   *               Local prototypes   *               Local prototypes
# Line 94  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);
                                         int *isframe);  
89  static int dec_stop();  static int dec_stop();
90  static void usage();  static void usage();
91    int write_image(char *prefix, unsigned char *image);
92    int write_pnm(char *filename, unsigned char *image);
93    int write_tga(char *filename, unsigned char *image);
94    
95    const char * type2str(int type)
96    {
97        if (type==XVID_TYPE_IVOP)
98            return "I";
99        if (type==XVID_TYPE_PVOP)
100            return "P";
101        if (type==XVID_TYPE_BVOP)
102            return "B";
103        return "S";
104    }
105    
106  /*****************************************************************************  /*****************************************************************************
107   *        Main program   *        Main program
# Line 108  Line 112 
112          unsigned char *mp4_buffer = NULL;          unsigned char *mp4_buffer = NULL;
113          unsigned char *mp4_ptr    = NULL;          unsigned char *mp4_ptr    = NULL;
114          unsigned char *out_buffer = NULL;          unsigned char *out_buffer = NULL;
115          int bigendian = 0;          int useful_bytes;
116          int still_left_in_buffer;          xvid_dec_stats_t xvid_dec_stats;
         int delayed_frames;  
117    
118          double totaldectime;          double totaldectime;
119    
# Line 136  Line 139 
139    
140                  if (strcmp("-asm", argv[i]) == 0 ) {                  if (strcmp("-asm", argv[i]) == 0 ) {
141                          use_assembler = 1;                          use_assembler = 1;
142                  }                  } else if (strcmp("-d", argv[i]) == 0) {
143                  else if (strcmp("-w", argv[i]) == 0 && i < argc - 1 ) {                          ARG_SAVEDECOUTPUT = 1;
144                          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 ) {  
145                          i++;                          i++;
146                          ARG_INPUTFILE = argv[i];                          ARG_INPUTFILE = argv[i];
147                    } else if (strcmp("-m", argv[i]) == 0) {
148                            ARG_SAVEMPEGSTREAM = 1;
149                    } else if (strcmp("-c", argv[i]) == 0  && i < argc - 1 ) {
150                            i++;
151                            if (strcmp(argv[i], "rgb16") == 0) {
152                                    CSP = XVID_CSP_RGB555;
153                                    BPP = 2;
154                            } else if (strcmp(argv[i], "rgb24") == 0) {
155                                    CSP = XVID_CSP_BGR;
156                                    BPP = 3;
157                            } else if (strcmp(argv[i], "rgb32") == 0) {
158                                    CSP = XVID_CSP_BGRA;
159                                    BPP = 4;
160                            } else if (strcmp(argv[i], "yv12") == 0) {
161                                    CSP = XVID_CSP_YV12;
162                                    BPP = 1;
163                            } else {
164                                    CSP = XVID_CSP_I420;
165                                    BPP = 1;
166                  }                  }
167                  else if (strcmp("-m", argv[i]) == 0 && i < argc - 1 ) {                  } else if (strcmp("-f", argv[i]) == 0 && i < argc -1) {
168                          i++;                          i++;
169                          ARG_SAVEMPEGSTREAM = atoi(argv[i]);                          if (strcmp(argv[i], "tga") == 0) {
170                                    FORMAT = USE_TGA;
171                            } else {
172                                    FORMAT = USE_PNM;
173                  }                  }
174                  else if (strcmp("-help", argv[i])) {                  } else if (strcmp("-help", argv[i]) == 0) {
175                          usage();                          usage();
176                          return(0);                          return(0);
177                  }                  } else {
                 else {  
178                          usage();                          usage();
179                          exit(-1);                          exit(-1);
180                  }                  }
181            }
182    
183    #if defined(_MSC_VER)
184            if (ARG_INPUTFILE==NULL) {
185                    fprintf(stderr, "Warning: MSVC build does not read EOF correctly from stdin. Use the -i switch.\n\n");
186          }          }
187    #endif
188    
189  /*****************************************************************************  /*****************************************************************************
190   * Values checking   * Values checking
191   ****************************************************************************/   ****************************************************************************/
192    
         if(XDIM <= 0 || XDIM > 2048 || YDIM <= 0 || YDIM > 2048) {  
                 usage();  
                 return -1;  
         }  
   
193          if ( ARG_INPUTFILE == NULL || strcmp(ARG_INPUTFILE, "stdin") == 0) {          if ( ARG_INPUTFILE == NULL || strcmp(ARG_INPUTFILE, "stdin") == 0) {
194                  in_file = stdin;                  in_file = stdin;
195          }          }
# Line 185  Line 198 
198                  in_file = fopen(ARG_INPUTFILE, "rb");                  in_file = fopen(ARG_INPUTFILE, "rb");
199                  if (in_file == NULL) {                  if (in_file == NULL) {
200                          fprintf(stderr, "Error opening input file %s\n", ARG_INPUTFILE);                          fprintf(stderr, "Error opening input file %s\n", ARG_INPUTFILE);
201                          return -1;                          return(-1);
202                  }                  }
203          }          }
204    
205            /* PNM/PGM format can't handle 16/32 bit data */
206            if (BPP != 1 && BPP != 3 && FORMAT == USE_PNM) {
207                    FORMAT = USE_TGA;
208            }
209    
210  /*****************************************************************************  /*****************************************************************************
211   *        Memory allocation   *        Memory allocation
212   ****************************************************************************/   ****************************************************************************/
# Line 199  Line 217 
217          if (!mp4_buffer)          if (!mp4_buffer)
218                  goto free_all_memory;                  goto free_all_memory;
219    
         /* Memory for frame output */  
         out_buffer = (unsigned char *) malloc(XDIM*YDIM*4);  
         if (!out_buffer)  
                 goto free_all_memory;  
   
   
220  /*****************************************************************************  /*****************************************************************************
221   *        XviD PART  Start   *        XviD PART  Start
222   ****************************************************************************/   ****************************************************************************/
# Line 222  Line 234 
234   ****************************************************************************/   ****************************************************************************/
235    
236          /* Fill the buffer */          /* Fill the buffer */
237          still_left_in_buffer = fread(mp4_buffer, 1, BUFFER_SIZE, in_file);          useful_bytes = fread(mp4_buffer, 1, BUFFER_SIZE, in_file);
238    
239          totaldectime = 0;          totaldectime = 0;
240          totalsize = 0;          totalsize = 0;
241          filenr = 0;          filenr = 0;
         delayed_frames = 0;  
242          mp4_ptr = mp4_buffer;          mp4_ptr = mp4_buffer;
243    
244          do {          do {
   
                 int mp4_size = (mp4_buffer + BUFFER_SIZE - mp4_ptr);  
245                  int used_bytes = 0;                  int used_bytes = 0;
246                  double dectime;                  double dectime;
                 int notification;  
247    
248                  /*                  /*
249                   * If the buffer is half empty or there are no more bytes in it                   * If the buffer is half empty or there are no more bytes in it
250                   * then fill it.                   * then fill it.
251                   */                   */
252                  if (mp4_ptr > mp4_buffer + BUFFER_SIZE/2 ||                  if (mp4_ptr > mp4_buffer + BUFFER_SIZE/2) {
253                          still_left_in_buffer <= 0) {                          int already_in_buffer = (mp4_buffer + BUFFER_SIZE - mp4_ptr);
                         int rest = (mp4_buffer + BUFFER_SIZE - mp4_ptr);  
254    
255                          /* Move data if needed */                          /* Move data if needed */
256                          if (rest)                          if (already_in_buffer > 0)
257                                  memcpy(mp4_buffer, mp4_ptr, rest);                                  memcpy(mp4_buffer, mp4_ptr, already_in_buffer);
258    
259                          /* Update mp4_ptr */                          /* Update mp4_ptr */
260                          mp4_ptr = mp4_buffer;                          mp4_ptr = mp4_buffer;
# Line 256  Line 263 
263                          if(feof(in_file))                          if(feof(in_file))
264                                  break;                                  break;
265    
266                          still_left_in_buffer = fread(mp4_buffer + rest,                          useful_bytes += fread(mp4_buffer + already_in_buffer,
267                                                                                   1,                                                                    1, BUFFER_SIZE - already_in_buffer,
                                                                                  BUFFER_SIZE - rest,  
268                                                                                   in_file);                                                                                   in_file);
269    
270                  }                  }
271    
272    
273                  /* This loop flushes N_VOPS (with vop_coded bit set to 0) */                  /* This loop is needed to handle VOL/NVOP reading */
274                  do {                  do {
275    
276                          /* Decode frame */                          /* Decode frame */
277                          dectime = msecond();                          dectime = msecond();
278                          status = dec_main(mp4_ptr, out_buffer, mp4_size, &used_bytes, &notification);                          used_bytes = dec_main(mp4_ptr, out_buffer, useful_bytes, &xvid_dec_stats);
279                          dectime = msecond() - dectime;                          dectime = msecond() - dectime;
280    
281                          if (status) {                          /* Resize image buffer if needed */
282                                  break;                          if(xvid_dec_stats.type == XVID_TYPE_VOL) {
283    
284                                    /* Check if old buffer is smaller */
285                                    if(XDIM*YDIM < xvid_dec_stats.data.vol.width*xvid_dec_stats.data.vol.height) {
286    
287                                            /* Copy new witdh and new height from the vol structure */
288                                            XDIM = xvid_dec_stats.data.vol.width;
289                                            YDIM = xvid_dec_stats.data.vol.height;
290    
291                                            /* Free old output buffer*/
292                                            if(out_buffer) free(out_buffer);
293    
294                                            /* Allocate the new buffer */
295                                            out_buffer = (unsigned char*)malloc(XDIM*YDIM*4);
296                                            if(out_buffer == NULL)
297                                                    goto free_all_memory;
298    
299                                            fprintf(stderr, "Resized frame buffer to %dx%d\n", XDIM, YDIM);
300                                    }
301                          }                          }
302    
303                          /* Update buffer pointers */                          /* Update buffer pointers */
304                            if(used_bytes > 0) {
305                          mp4_ptr += used_bytes;                          mp4_ptr += used_bytes;
306                          still_left_in_buffer -= used_bytes;                                  useful_bytes -= used_bytes;
307    
308                          /* Total size */                          /* Total size */
309                          totalsize += used_bytes;                          totalsize += used_bytes;
310                            }
311    
312                  }while(used_bytes <= 7 && still_left_in_buffer > 0); /* <= 7 bytes is a NVOPS */                  }while(xvid_dec_stats.type <= 0 && useful_bytes > 0);
   
                 /* Negative buffer would mean we went too far */  
                 if(still_left_in_buffer < 0) break;  
313    
314                  /* Skip when decoder is buffering images or decoding VOL information */                  /* Check if there is a negative number of useful bytes left in buffer
315                  if(notification != XVID_DEC_VOP) {                   * This means we went too far */
316                          /* It's a delay only if it notifies NOTHING */          if(useful_bytes < 0)
317                          if(notification == XVID_DEC_NOTHING)              break;
                                 delayed_frames++;  
                         continue;  
                 }  
318    
319                  /* Updated data - Count only usefull decode time */                  /* Updated data - Count only usefull decode time */
320                  totaldectime += dectime;                  totaldectime += dectime;
321    
322                  /* Prints some decoding stats */  
323                  printf("Frame %5d: dectime(ms) =%6.1f, length(bytes) =%7d\n",          printf("Frame %5d: type = %s, dectime(ms) =%6.1f, length(bytes) =%7d\n",
324                             filenr, dectime, used_bytes);                             filenr, type2str(xvid_dec_stats.type), dectime, used_bytes);
325    
326                  /* Save individual mpeg4 stream if required */                  /* Save individual mpeg4 stream if required */
327                  if (ARG_SAVEMPEGSTREAM) {                  if (ARG_SAVEMPEGSTREAM) {
# Line 315  Line 335 
335                                                  filename);                                                  filename);
336                          }                          }
337                          else {                          else {
338                                  fwrite(mp4_buffer, 1, used_bytes, filehandle);                                  fwrite(mp4_ptr-used_bytes, 1, used_bytes, filehandle);
339                                  fclose(filehandle);                                  fclose(filehandle);
340                          }                          }
341                  }                  }
342    
343                  /* Save output frame if required */                  /* Save output frame if required */
344                  if (ARG_SAVEDECOUTPUT) {                  if (ARG_SAVEDECOUTPUT) {
345                          sprintf(filename, "%sdec%05d.pgm", filepath, filenr);                          sprintf(filename, "%sdec%05d", filepath, filenr);
346                          if(write_pgm(filename,out_buffer)) {                          if(write_image(filename, out_buffer)) {
347                                  fprintf(stderr,                                  fprintf(stderr,
348                                                  "Error writing decoded PGM frame %s\n",                                                  "Error writing decoded PGM frame %s\n",
349                                                  filename);                                                  filename);
# Line 337  Line 357 
357  /*****************************************************************************  /*****************************************************************************
358   *     Flush decoder buffers   *     Flush decoder buffers
359   ****************************************************************************/   ****************************************************************************/
360          while(delayed_frames--) {  
361            do {
362    
363                  /* Fake vars */                  /* Fake vars */
364                  int used_bytes, isframe;                  int used_bytes;
365                  double dectime;                  double dectime;
366    
367                  /* Decode frame */          do {
368                  dectime = msecond();                  dectime = msecond();
369                  status = dec_main(NULL, out_buffer, -1, &used_bytes, &isframe);                      used_bytes = dec_main(NULL, out_buffer, -1, &xvid_dec_stats);
370                  dectime = msecond() - dectime;                  dectime = msecond() - dectime;
371            }while(used_bytes>=0 && xvid_dec_stats.type <= 0);
372    
373                  if (status) {          if (used_bytes < 0) {   /* XVID_ERR_END */
374                          break;                          break;
375                  }                  }
376    
# Line 356  Line 378 
378                  totaldectime += dectime;                  totaldectime += dectime;
379    
380                  /* Prints some decoding stats */                  /* Prints some decoding stats */
381                  printf("Frame %5d: dectime(ms) =%6.1f, length(bytes) =%7d\n",          printf("Frame %5d: type = %s, dectime(ms) =%6.1f, length(bytes) =%7d\n",
382                             filenr, dectime, used_bytes);                             filenr, type2str(xvid_dec_stats.type), dectime, used_bytes);
383    
384                  /* Save output frame if required */                  /* Save output frame if required */
385                  if (ARG_SAVEDECOUTPUT) {                  if (ARG_SAVEDECOUTPUT) {
386                          sprintf(filename, "%sdec%05d.pgm", filepath, filenr);                          sprintf(filename, "%sdec%05d", filepath, filenr);
387                          if(write_pgm(filename,out_buffer)) {                          if(write_image(filename, out_buffer)) {
388                                  fprintf(stderr,                                  fprintf(stderr,
389                                                  "Error writing decoded PGM frame %s\n",                                                  "Error writing decoded frame %s\n",
390                                                  filename);                                                  filename);
391                          }                          }
392                  }                  }
393    
394                  filenr++;                  filenr++;
395    
396          }          }while(1);
397    
398  /*****************************************************************************  /*****************************************************************************
399   *     Calculate totals and averages for output, print results   *     Calculate totals and averages for output, print results
400   ****************************************************************************/   ****************************************************************************/
401    
402            if (filenr>0) {
403          totalsize    /= filenr;          totalsize    /= filenr;
404          totaldectime /= filenr;          totaldectime /= filenr;
   
405          printf("Avg: dectime(ms) =%7.2f, fps =%7.2f, length(bytes) =%7d\n",          printf("Avg: dectime(ms) =%7.2f, fps =%7.2f, length(bytes) =%7d\n",
406                  totaldectime, 1000/totaldectime, (int)totalsize);                  totaldectime, 1000/totaldectime, (int)totalsize);
407            }else{
408                    printf("Nothing was decoded!\n");
409            }
410    
411  /*****************************************************************************  /*****************************************************************************
412   *      XviD PART  Stop   *      XviD PART  Stop
# Line 398  Line 423 
423          free(out_buffer);          free(out_buffer);
424          free(mp4_buffer);          free(mp4_buffer);
425    
426          return 0;          return(0);
427  }  }
428    
429  /*****************************************************************************  /*****************************************************************************
# Line 408  Line 433 
433  static void usage()  static void usage()
434  {  {
435    
436          fprintf(stderr, "Usage : xvid_decraw <-w width> <-h height> [OPTIONS]\n");          fprintf(stderr, "Usage : xvid_decraw [OPTIONS]\n");
437          fprintf(stderr, "Options :\n");          fprintf(stderr, "Options :\n");
438          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");  
439          fprintf(stderr, " -i string      : input filename (default=stdin)\n");          fprintf(stderr, " -i string      : input filename (default=stdin)\n");
440          fprintf(stderr, " -t integer     : input data type (raw=0, mp4u=1)\n");          fprintf(stderr, " -d             : save decoder output\n");
441          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");
442          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");
443            fprintf(stderr, " -m             : save mpeg4 raw stream to individual files\n");
444          fprintf(stderr, " -help          : This help message\n");          fprintf(stderr, " -help          : This help message\n");
445          fprintf(stderr, " (* means default)\n");          fprintf(stderr, " (* means default)\n");
446    
# Line 433  Line 457 
457  #ifndef WIN32  #ifndef WIN32
458          struct timeval  tv;          struct timeval  tv;
459          gettimeofday(&tv, 0);          gettimeofday(&tv, 0);
460          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);
461  #else  #else
462          clock_t clk;          clock_t clk;
463          clk = clock();          clk = clock();
464          return clk * 1000 / CLOCKS_PER_SEC;          return(clk * 1000 / CLOCKS_PER_SEC);
465  #endif  #endif
466  }  }
467    
# Line 445  Line 469 
469   *              output functions   *              output functions
470   ****************************************************************************/   ****************************************************************************/
471    
472  static int  int write_image(char *prefix, unsigned char *image)
 write_pgm(char *filename,  
           unsigned char *image)  
473  {  {
474          int loop;          char filename[1024];
475            char *ext;
476            int ret;
477    
478            if (FORMAT == USE_PNM && BPP == 1) {
479                    ext = "pgm";
480            } else if (FORMAT == USE_PNM && BPP == 3) {
481                    ext = "pnm";
482            } else if (FORMAT == USE_TGA) {
483                    ext = "tga";
484            } else {
485                    fprintf(stderr, "Bug: should not reach this path code -- please report to xvid-devel@xvid.org with command line options used");
486                    exit(-1);
487            }
488    
489          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) {  
490    
491                  /* Write header */          if (FORMAT == USE_PNM) {
492                  fprintf(filehandle,"P5\n\n%d %d 255\n", XDIM,YDIM*3/2);                  ret = write_pnm(filename, image);
493            } else {
494                    ret = write_tga(filename, image);
495            }
496    
497                  /* Write Y data */          return(ret);
498                  fwrite(y, 1, XDIM*YDIM, filehandle);  }
499    
500                  for(loop=0; loop<YDIM/2; loop++)  int write_tga(char *filename, unsigned char *image)
501                  {                  {
502                          /* Write U scanline */          FILE * f;
503                          fwrite(u, 1, XDIM/2, filehandle);          char hdr[18];
504    
505                          /* Write V scanline */          f = fopen(filename, "wb");
506                          fwrite(v, 1, XDIM/2, filehandle);          if ( f == NULL) {
507                    return -1;
508            }
509    
510                          /* Update pointers */          hdr[0]  = 0; /* ID length */
511                          u += XDIM/2;          hdr[1]  = 0; /* Color map type */
512                          v += XDIM/2;          hdr[2]  = (BPP>1)?2:3; /* Uncompressed true color (2) or greymap (3) */
513            hdr[3]  = 0; /* Color map specification (not used) */
514            hdr[4]  = 0; /* Color map specification (not used) */
515            hdr[5]  = 0; /* Color map specification (not used) */
516            hdr[6]  = 0; /* Color map specification (not used) */
517            hdr[7]  = 0; /* Color map specification (not used) */
518            hdr[8]  = 0; /* LSB X origin */
519            hdr[9]  = 0; /* MSB X origin */
520            hdr[10] = 0; /* LSB Y origin */
521            hdr[11] = 0; /* MSB Y origin */
522            hdr[12] = (XDIM>>0)&0xff; /* LSB Width */
523            hdr[13] = (XDIM>>8)&0xff; /* MSB Width */
524            if (BPP > 1) {
525                    hdr[14] = (YDIM>>0)&0xff; /* LSB Height */
526                    hdr[15] = (YDIM>>8)&0xff; /* MSB Height */
527            } else {
528                    hdr[14] = ((YDIM*3)>>1)&0xff; /* LSB Height */
529                    hdr[15] = ((YDIM*3)>>9)&0xff; /* MSB Height */
530            }
531            hdr[16] = BPP*8;
532            hdr[17] = 0x00 | (1<<5) /* Up to down */ | (0<<4); /* Image descriptor */
533    
534            /* Write header */
535            fwrite(hdr, 1, sizeof(hdr), f);
536    
537    #ifdef ARCH_IS_LITTLE_ENDIAN
538            /* write first plane */
539            fwrite(image, 1, XDIM*YDIM*BPP, f);
540    #else
541            {
542                    int i;
543                    for (i=0; i<width*height*BPP;i+=BPP) {
544                            if (BPP == 1) {
545                                    fputc(image+i, f);
546                            } else if (BPP == 2) {
547                                    fputc(image+i+1, f);
548                                    fputc(image+i+0, f);
549                            } else if (BPP == 3) {
550                                    fputc(image+i+2, f);
551                                    fputc(image+i+1, f);
552                                    fputc(image+i+0, f);
553                            } else if (BPP == 4) {
554                                    fputc(image+i+3, f);
555                                    fputc(image+i+2, f);
556                                    fputc(image+i+1, f);
557                                    fputc(image+i+0, f);
558                            }
559                    }
560                  }                  }
561    #endif
562    
563                  /* Close file */          /* Write Y and V planes for YUV formats */
564                  fclose(filehandle);          if (BPP == 1) {
565                    int i;
566    
567                  return 0;                  /* Write the two chrominance planes */
568                    for (i=0; i<YDIM/2; i++) {
569                            fwrite(image+XDIM*YDIM + i*XDIM/2, 1, XDIM/2, f);
570                            fwrite(image+5*XDIM*YDIM/4 + i*XDIM/2, 1, XDIM/2, f);
571          }          }
         else  
                 return 1;  
572  }  }
573    
574    
575            /* Close the file */
576            fclose(f);
577    
578            return(0);
579    }
580    
581    int write_pnm(char *filename, unsigned char *image)
582    {
583            FILE * f;
584    
585            f = fopen(filename, "wb");
586            if ( f == NULL) {
587                    return -1;
588            }
589    
590            if (BPP == 1) {
591                    int i;
592                    fprintf(f, "P5\n#xvid\n%i %i\n255\n", XDIM, YDIM*3/2);
593    
594                    fwrite(image, 1, XDIM*YDIM, f);
595    
596                    for (i=0; i<YDIM/2;i++) {
597                            fwrite(image+XDIM*YDIM + i*XDIM/2, 1, XDIM/2, f);
598                            fwrite(image+5*XDIM*YDIM/4 + i*XDIM/2, 1, XDIM/2, f);
599                    }
600            } else if (BPP == 3) {
601                    int i;
602                    fprintf(f, "P6\n#xvid\n%i %i\n255\n", XDIM, YDIM);
603                    for (i=0; i<XDIM*YDIM*3; i+=3) {
604    #ifdef ARCH_IS_LITTLE_ENDIAN
605                            fputc(image[i+2], f);
606                            fputc(image[i+1], f);
607                            fputc(image[i+0], f);
608    #else
609                            fputc(image[i+0], f);
610                            fputc(image[i+1], f);
611                            fputc(image[i+2], f);
612    #endif
613                    }
614            }
615    
616            fclose(f);
617    
618            return 0;
619    }
620  /*****************************************************************************  /*****************************************************************************
621   * Routines for decoding: init decoder, use, and stop decoder   * Routines for decoding: init decoder, use, and stop decoder
622   ****************************************************************************/   ****************************************************************************/
# Line 496  Line 625 
625  static int  static int
626  dec_init(int use_assembler)  dec_init(int use_assembler)
627  {  {
628          int xerr;          int ret;
629    
630          XVID_INIT_PARAM xinit;          xvid_gbl_init_t   xvid_gbl_init;
631          XVID_DEC_PARAM xparam;          xvid_dec_create_t xvid_dec_create;
632    
633            /*------------------------------------------------------------------------
634             * XviD core initialization
635             *----------------------------------------------------------------------*/
636    
637            /* Version */
638            xvid_gbl_init.version = XVID_VERSION;
639    
640            /* Assembly setting */
641                  if(use_assembler)                  if(use_assembler)
642  #ifdef ARCH_IA64  #ifdef ARCH_IS_IA64
643                          xinit.cpu_flags = XVID_CPU_FORCE | XVID_CPU_IA64;                  xvid_gbl_init.cpu_flags = XVID_CPU_FORCE | XVID_CPU_IA64;
644  #else  #else
645                          xinit.cpu_flags = 0;          xvid_gbl_init.cpu_flags = 0;
646  #endif  #endif
647                  else                  else
648                          xinit.cpu_flags = XVID_CPU_FORCE;                  xvid_gbl_init.cpu_flags = XVID_CPU_FORCE;
649    
650          xvid_init(NULL, 0, &xinit, NULL);          xvid_global(NULL, 0, &xvid_gbl_init, NULL);
         xparam.width = XDIM;  
         xparam.height = YDIM;  
651    
652          xerr = xvid_decore(NULL, XVID_DEC_CREATE, &xparam, NULL);          /*------------------------------------------------------------------------
653             * XviD encoder initialization
654             *----------------------------------------------------------------------*/
655    
656          dec_handle = xparam.handle;          /* Version */
657            xvid_dec_create.version = XVID_VERSION;
658    
659          return xerr;          /*
660             * Image dimensions -- set to 0, xvidcore will resize when ever it is
661             * needed
662             */
663            xvid_dec_create.width = 0;
664            xvid_dec_create.height = 0;
665    
666            ret = xvid_decore(NULL, XVID_DEC_CREATE, &xvid_dec_create, NULL);
667    
668            dec_handle = xvid_dec_create.handle;
669    
670            return(ret);
671  }  }
672    
673  /* decode one frame  */  /* decode one frame  */
# Line 526  Line 675 
675  dec_main(unsigned char *istream,  dec_main(unsigned char *istream,
676                   unsigned char *ostream,                   unsigned char *ostream,
677                   int istream_size,                   int istream_size,
678                   int *ostream_size,                   xvid_dec_stats_t *xvid_dec_stats)
                  int *notification)  
679  {  {
680    
681          int xerr;          int ret;
682          XVID_DEC_FRAME xframe;  
683                  XVID_DEC_STATS xstats;          xvid_dec_frame_t xvid_dec_frame;
684    
685            /* Set version */
686            xvid_dec_frame.version = XVID_VERSION;
687            xvid_dec_stats->version = XVID_VERSION;
688    
689          xframe.general    = 0;          /* No general flags to set */
690                  xframe.bitstream  = istream;          xvid_dec_frame.general          = 0;
         xframe.length     = istream_size;  
                 xframe.image      = ostream;  
         xframe.stride     = XDIM;  
                 xframe.colorspace = XVID_CSP_YV12;  
691    
692                  xerr = xvid_decore(dec_handle, XVID_DEC_DECODE, &xframe, &xstats);          /* Input stream */
693            xvid_dec_frame.bitstream        = istream;
694            xvid_dec_frame.length           = istream_size;
695    
696                  *ostream_size = xframe.length;          /* Output frame structure */
697            xvid_dec_frame.output.plane[0]  = ostream;
698            xvid_dec_frame.output.stride[0] = XDIM*BPP;
699            xvid_dec_frame.output.csp = CSP;
700    
701                  *notification = xstats.notify;          ret = xvid_decore(dec_handle, XVID_DEC_DECODE, &xvid_dec_frame, xvid_dec_stats);
702    
703          return xerr;          return(ret);
704  }  }
705    
706  /* close decoder to release resources */  /* close decoder to release resources */
707  static int  static int
708  dec_stop()  dec_stop()
709  {  {
710          int xerr;          int ret;
711    
712          xerr = xvid_decore(dec_handle, XVID_DEC_DESTROY, NULL, NULL);          ret = xvid_decore(dec_handle, XVID_DEC_DESTROY, NULL, NULL);
713    
714          return xerr;          return(ret);
715  }  }
Legend:
Removed from v.851  
changed lines
  Added in v.1430
No admin address has been configured
 ViewVC Help
Powered by ViewVC 1.0.4