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