[cvs] / xvidcore / examples / xvid_decraw.c Repository:
ViewVC logotype

Diff of /xvidcore/examples/xvid_decraw.c

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

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