[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 1523, Mon Jul 26 19:31:30 2004 UTC revision 1881, Fri Jan 8 10:03:09 2010 UTC
# Line 20  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.18 2004-07-26 19:31:30 edgomez Exp $   * $Id: xvid_decraw.c,v 1.26 2010-01-08 10:03:09 Isibaar Exp $
24   *   *
25   ****************************************************************************/   ****************************************************************************/
26    
# Line 54  Line 54 
54   *               Global vars in module and constants   *               Global vars in module and constants
55   ****************************************************************************/   ****************************************************************************/
56    
 /* max number of frames */  
 #define ABS_MAXFRAMENR 9999  
   
57  #define USE_PNM 0  #define USE_PNM 0
58  #define USE_TGA 1  #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;
# Line 74  Line 72 
72    
73  #define BUFFER_SIZE (2*1024*1024)  #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   ****************************************************************************/   ****************************************************************************/
# Line 86  Line 88 
88                                          xvid_dec_stats_t *xvid_dec_stats);                                          xvid_dec_stats_t *xvid_dec_stats);
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);  static int write_image(char *prefix, unsigned char *image, int filenr);
92  static int write_pnm(char *filename, unsigned char *image);  static int write_pnm(char *filename, unsigned char *image);
93  static int write_tga(char *filename, unsigned char *image);  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)  const char * type2str(int type)
97  {  {
# Line 111  Line 114 
114          unsigned char *mp4_ptr    = NULL;          unsigned char *mp4_ptr    = NULL;
115          unsigned char *out_buffer = NULL;          unsigned char *out_buffer = NULL;
116          int useful_bytes;          int useful_bytes;
117            int chunk;
118          xvid_dec_stats_t xvid_dec_stats;          xvid_dec_stats_t xvid_dec_stats;
119    
120          double totaldectime;          double totaldectime;
# Line 172  Line 176 
176                          i++;                          i++;
177                          if (strcmp(argv[i], "tga") == 0) {                          if (strcmp(argv[i], "tga") == 0) {
178                                  FORMAT = USE_TGA;                                  FORMAT = USE_TGA;
179                            } else if (strcmp(argv[i], "yuv") == 0) {
180                                    FORMAT = USE_YUV;
181                          } else {                          } else {
182                                  FORMAT = USE_PNM;                                  FORMAT = USE_PNM;
183                          }                          }
# Line 210  Line 216 
216          if (BPP != 1 && BPP != 3 && FORMAT == USE_PNM) {          if (BPP != 1 && BPP != 3 && FORMAT == USE_PNM) {
217                  FORMAT = USE_TGA;                  FORMAT = USE_TGA;
218          }          }
219            if (BPP != 1 && FORMAT == USE_YUV) {
220                    FORMAT = USE_TGA;
221            }
222    
223  /*****************************************************************************  /*****************************************************************************
224   *        Memory allocation   *        Memory allocation
# Line 244  Line 253 
253          totalsize = 0;          totalsize = 0;
254          filenr = 0;          filenr = 0;
255          mp4_ptr = mp4_buffer;          mp4_ptr = mp4_buffer;
256            chunk = 0;
257    
258          do {          do {
259                  int used_bytes = 0;                  int used_bytes = 0;
# Line 264  Line 274 
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)) {
                                 break;  
   
278                          useful_bytes += fread(mp4_buffer + already_in_buffer,                          useful_bytes += fread(mp4_buffer + already_in_buffer,
279                                                                    1, BUFFER_SIZE - already_in_buffer,                                                                    1, BUFFER_SIZE - already_in_buffer,
280                                                                    in_file);                                                                    in_file);
281                            }
282                  }                  }
283    
284    
# Line 302  Line 310 
310    
311                                          fprintf(stderr, "Resized frame buffer to %dx%d\n", XDIM, YDIM);                                          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 */
# Line 313  Line 337 
337                                  totalsize += used_bytes;                                  totalsize += used_bytes;
338                          }                          }
339    
340                  }while(xvid_dec_stats.type <= 0 && useful_bytes > 0);                          if (display_buffer_bytes) {
341                                    printf("Data chunk %d: %d bytes consumed, %d bytes in buffer\n", chunk++, used_bytes, useful_bytes);
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                  /* Check if there is a negative number of useful bytes left in buffer
346                   * This means we went too far */                   * This means we went too far */
# Line 324  Line 351 
351                  totaldectime += dectime;                  totaldectime += dectime;
352    
353    
354                    if (!display_buffer_bytes) {
355          printf("Frame %5d: type = %s, dectime(ms) =%6.1f, length(bytes) =%7d\n",          printf("Frame %5d: type = %s, dectime(ms) =%6.1f, length(bytes) =%7d\n",
356                             filenr, type2str(xvid_dec_stats.type), dectime, used_bytes);                             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 346  Line 375 
375    
376                  /* Save output frame if required */                  /* Save output frame if required */
377                  if (ARG_SAVEDECOUTPUT) {                  if (ARG_SAVEDECOUTPUT) {
378                          sprintf(filename, "%sdec%05d", filepath, filenr);                          sprintf(filename, "%sdec", filepath);
379                          if(write_image(filename, out_buffer)) {  
380                            if(write_image(filename, out_buffer, filenr)) {
381                                  fprintf(stderr,                                  fprintf(stderr,
382                                                  "Error writing decoded frame %s\n",                                                  "Error writing decoded frame %s\n",
383                                                  filename);                                                  filename);
# Line 356  Line 386 
386    
387                  filenr++;                  filenr++;
388    
389          } while ( (status>=0) && (filenr<ABS_MAXFRAMENR));          } while (useful_bytes>MIN_USEFUL_BYTES || !feof(in_file));
390    
391            useful_bytes = 0; /* Empty buffer */
392    
393  /*****************************************************************************  /*****************************************************************************
394   *     Flush decoder buffers   *     Flush decoder buffers
# Line 372  Line 404 
404                      dectime = msecond();                      dectime = msecond();
405                      used_bytes = dec_main(NULL, out_buffer, -1, &xvid_dec_stats);                      used_bytes = dec_main(NULL, out_buffer, -1, &xvid_dec_stats);
406                      dectime = msecond() - dectime;                      dectime = msecond() - dectime;
407                            if (display_buffer_bytes) {
408                                    printf("Data chunk %d: %d bytes consumed, %d bytes in buffer\n", chunk++, used_bytes, useful_bytes);
409                            }
410          }while(used_bytes>=0 && xvid_dec_stats.type <= 0);          }while(used_bytes>=0 && xvid_dec_stats.type <= 0);
411    
412          if (used_bytes < 0) {   /* XVID_ERR_END */          if (used_bytes < 0) {   /* XVID_ERR_END */
# Line 382  Line 417 
417                  totaldectime += dectime;                  totaldectime += dectime;
418    
419                  /* Prints some decoding stats */                  /* Prints some decoding stats */
420                    if (!display_buffer_bytes) {
421          printf("Frame %5d: type = %s, dectime(ms) =%6.1f, length(bytes) =%7d\n",          printf("Frame %5d: type = %s, dectime(ms) =%6.1f, length(bytes) =%7d\n",
422                             filenr, type2str(xvid_dec_stats.type), dectime, used_bytes);                             filenr, type2str(xvid_dec_stats.type), dectime, used_bytes);
423                    }
424    
425                  /* Save output frame if required */                  /* Save output frame if required */
426                  if (ARG_SAVEDECOUTPUT) {                  if (ARG_SAVEDECOUTPUT) {
427                          sprintf(filename, "%sdec%05d", filepath, filenr);                          sprintf(filename, "%sdec", filepath);
428                          if(write_image(filename, out_buffer)) {  
429                            if(write_image(filename, out_buffer, filenr)) {
430                                  fprintf(stderr,                                  fprintf(stderr,
431                                                  "Error writing decoded frame %s\n",                                                  "Error writing decoded frame %s\n",
432                                                  filename);                                                  filename);
# Line 444  Line 482 
482          fprintf(stderr, " -i string      : input filename (default=stdin)\n");          fprintf(stderr, " -i string      : input filename (default=stdin)\n");
483          fprintf(stderr, " -d             : save decoder output\n");          fprintf(stderr, " -d             : save decoder output\n");
484          fprintf(stderr, " -c csp         : choose colorspace output (rgb16, rgb24, rgb32, yv12, i420)\n");          fprintf(stderr, " -c csp         : choose colorspace output (rgb16, rgb24, rgb32, yv12, i420)\n");
485          fprintf(stderr, " -f format      : choose output file format (tga, pnm, pgm)\n");          fprintf(stderr, " -f format      : choose output file format (tga, pnm, pgm, yuv)\n");
486          fprintf(stderr, " -m             : save mpeg4 raw stream to individual files\n");          fprintf(stderr, " -m             : save mpeg4 raw stream to individual files\n");
487          fprintf(stderr, " -help          : This help message\n");          fprintf(stderr, " -help          : This help message\n");
488          fprintf(stderr, " (* means default)\n");          fprintf(stderr, " (* means default)\n");
# Line 466  Line 504 
504  #else  #else
505          clock_t clk;          clock_t clk;
506          clk = clock();          clk = clock();
507          return(clk * 1000 / CLOCKS_PER_SEC);          return(clk * 1000.0 / CLOCKS_PER_SEC);
508  #endif  #endif
509  }  }
510    
# Line 474  Line 512 
512   *              output functions   *              output functions
513   ****************************************************************************/   ****************************************************************************/
514    
515  static int write_image(char *prefix, unsigned char *image)  static int write_image(char *prefix, unsigned char *image, int filenr)
516  {  {
517          char filename[1024];          char filename[1024];
518          char *ext;          char *ext;
# Line 484  Line 522 
522                  ext = "pgm";                  ext = "pgm";
523          } else if (FORMAT == USE_PNM && BPP == 3) {          } else if (FORMAT == USE_PNM && BPP == 3) {
524                  ext = "pnm";                  ext = "pnm";
525            } else if (FORMAT == USE_YUV) {
526                    ext = "yuv";
527          } else if (FORMAT == USE_TGA) {          } else if (FORMAT == USE_TGA) {
528                  ext = "tga";                  ext = "tga";
529          } else {          } else {
# Line 491  Line 531 
531                  exit(-1);                  exit(-1);
532          }          }
533    
534            if (FORMAT == USE_YUV) {
535          sprintf(filename, "%s.%s", prefix, ext);          sprintf(filename, "%s.%s", prefix, ext);
536    
537                    if (!filenr) {
538                            FILE *fp = fopen(filename, "wb");
539                            fclose(fp);
540                    }
541            } else
542                    sprintf(filename, "%s%05d.%s", prefix, filenr, ext);
543    
544          if (FORMAT == USE_PNM) {          if (FORMAT == USE_PNM) {
545                  ret = write_pnm(filename, image);                  ret = write_pnm(filename, image);
546            } else if (FORMAT == USE_YUV) {
547                    ret = write_yuv(filename, image);
548          } else {          } else {
549                  ret = write_tga(filename, image);                  ret = write_tga(filename, image);
550          }          }
# Line 545  Line 595 
595  #else  #else
596          {          {
597                  int i;                  int i;
598                  for (i=0; i<width*height*BPP;i+=BPP) {                  for (i=0; i<XDIM*YDIM*BPP;i+=BPP) {
599                          if (BPP == 1) {                          if (BPP == 1) {
600                                  fputc(image+i, f);                                  fputc(*(image+i), f);
601                          } else if (BPP == 2) {                          } else if (BPP == 2) {
602                                  fputc(image+i+1, f);                                  fputc(*(image+i+1), f);
603                                  fputc(image+i+0, f);                                  fputc(*(image+i+0), f);
604                          } else if (BPP == 3) {                          } else if (BPP == 3) {
605                                  fputc(image+i+2, f);                                  fputc(*(image+i+2), f);
606                                  fputc(image+i+1, f);                                  fputc(*(image+i+1), f);
607                                  fputc(image+i+0, f);                                  fputc(*(image+i+0), f);
608                          } else if (BPP == 4) {                          } else if (BPP == 4) {
609                                  fputc(image+i+3, f);                                  fputc(*(image+i+3), f);
610                                  fputc(image+i+2, f);                                  fputc(*(image+i+2), f);
611                                  fputc(image+i+1, f);                                  fputc(*(image+i+1), f);
612                                  fputc(image+i+0, f);                                  fputc(*(image+i+0), f);
613                          }                          }
614                  }                  }
615          }          }
# Line 594  Line 644 
644    
645          if (BPP == 1) {          if (BPP == 1) {
646                  int i;                  int i;
647                  fprintf(f, "P5\n#xvid\n%i %i\n255\n", XDIM, YDIM*3/2);                  fprintf(f, "P5\n%i %i\n255\n", XDIM, YDIM*3/2);
648    
649                  fwrite(image, 1, XDIM*YDIM, f);                  fwrite(image, 1, XDIM*YDIM, f);
650    
# Line 623  Line 673 
673          return 0;          return 0;
674  }  }
675    
676    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            fwrite(image, 1, 3*XDIM*YDIM/2, f);
686    
687            fclose(f);
688    
689            return 0;
690    }
691    
692  /*****************************************************************************  /*****************************************************************************
693   * Routines for decoding: init decoder, use, and stop decoder   * Routines for decoding: init decoder, use, and stop decoder
694   ****************************************************************************/   ****************************************************************************/
Legend:
Removed from v.1523  
changed lines
  Added in v.1881
No admin address has been configured
 ViewVC Help
Powered by ViewVC 1.0.4