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

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

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

trunk/xvidcore/examples/xvid_decraw.c revision 1416, Sat Apr 10 04:25:31 2004 UTC branches/release-1_1-branch/xvidcore/examples/xvid_decraw.c revision 1715, Mon Jul 10 15:21:49 2006 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.11 2004-04-10 04:25:31 suxen_drol Exp $   * $Id: xvid_decraw.c,v 1.22.2.1 2006-07-10 15:19:41 Isibaar Exp $
24   *   *
25   ****************************************************************************/   ****************************************************************************/
26    
# Line 54  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;
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 (2*1024*1024)  #define BUFFER_SIZE (2*1024*1024)
73    
74    static const int display_buffer_bytes = 0;
75    
76    #define MIN_USEFUL_BYTES 1
77    
78  /*****************************************************************************  /*****************************************************************************
79   *               Local prototypes   *               Local prototypes
80   ****************************************************************************/   ****************************************************************************/
81    
82  static double msecond();  static double msecond();
 static int write_pgm(char *filename,  
                                          unsigned char *image);  
83  static int dec_init(int use_assembler, int debug_level);  static int dec_init(int use_assembler, int debug_level);
84  static int dec_main(unsigned char *istream,  static int dec_main(unsigned char *istream,
85                                          unsigned char *ostream,                                          unsigned char *ostream,
# Line 83  Line 87 
87                                          xvid_dec_stats_t *xvid_dec_stats);                                          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)  const char * type2str(int type)
95  {  {
# Line 106  Line 112 
112          unsigned char *mp4_ptr    = NULL;          unsigned char *mp4_ptr    = NULL;
113          unsigned char *out_buffer = NULL;          unsigned char *out_buffer = NULL;
114          int useful_bytes;          int useful_bytes;
115            int chunk;
116          xvid_dec_stats_t xvid_dec_stats;          xvid_dec_stats_t xvid_dec_stats;
117    
118          double totaldectime;          double totaldectime;
# Line 145  Line 152 
152                          ARG_INPUTFILE = argv[i];                          ARG_INPUTFILE = argv[i];
153                  } else if (strcmp("-m", argv[i]) == 0) {                  } else if (strcmp("-m", argv[i]) == 0) {
154                          ARG_SAVEMPEGSTREAM = 1;                          ARG_SAVEMPEGSTREAM = 1;
155                    } else if (strcmp("-c", argv[i]) == 0  && i < argc - 1 ) {
156                            i++;
157                            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("-f", argv[i]) == 0 && i < argc -1) {
174                            i++;
175                            if (strcmp(argv[i], "tga") == 0) {
176                                    FORMAT = USE_TGA;
177                            } else {
178                                    FORMAT = USE_PNM;
179                            }
180                  } else if (strcmp("-help", argv[i]) == 0) {                  } else if (strcmp("-help", argv[i]) == 0) {
181                          usage();                          usage();
182                          return(0);                          return(0);
# Line 154  Line 186 
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   ****************************************************************************/   ****************************************************************************/
# Line 170  Line 208 
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 203  Line 246 
246          totalsize = 0;          totalsize = 0;
247          filenr = 0;          filenr = 0;
248          mp4_ptr = mp4_buffer;          mp4_ptr = mp4_buffer;
249            chunk = 0;
250    
251          do {          do {
252                  int used_bytes = 0;                  int used_bytes = 0;
# Line 223  Line 267 
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;  
271    
272                          useful_bytes += fread(mp4_buffer + already_in_buffer,                          useful_bytes += fread(mp4_buffer + already_in_buffer,
273                                                                    1, BUFFER_SIZE - already_in_buffer,                                                                    1, BUFFER_SIZE - already_in_buffer,
274                                                                    in_file);                                                                    in_file);
275                            }
276                  }                  }
277    
278    
# Line 261  Line 304 
304    
305                                          fprintf(stderr, "Resized frame buffer to %dx%d\n", XDIM, YDIM);                                          fprintf(stderr, "Resized frame buffer to %dx%d\n", XDIM, YDIM);
306                                  }                                  }
307    
308                                    /* Save individual mpeg4 stream if required */
309                                    if(ARG_SAVEMPEGSTREAM) {
310                                            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 */                          /* Update buffer pointers */
# Line 272  Line 331 
331                                  totalsize += used_bytes;                                  totalsize += used_bytes;
332                          }                          }
333    
334                  }while(xvid_dec_stats.type <= 0 && useful_bytes > 0);                          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                  /* 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
340                   * This means we went too far */                   * This means we went too far */
# Line 283  Line 345 
345                  totaldectime += dectime;                  totaldectime += dectime;
346    
347    
348                    if (!display_buffer_bytes) {
349          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",
350                             filenr, type2str(xvid_dec_stats.type), dectime, used_bytes);                             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 305  Line 369 
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   *     Flush decoder buffers
# Line 331  Line 397 
397                      dectime = msecond();                      dectime = msecond();
398                      used_bytes = dec_main(NULL, out_buffer, -1, &xvid_dec_stats);                      used_bytes = dec_main(NULL, out_buffer, -1, &xvid_dec_stats);
399                      dectime = msecond() - dectime;                      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);          }while(used_bytes>=0 && xvid_dec_stats.type <= 0);
404    
405          if (used_bytes < 0) {   /* XVID_ERR_END */          if (used_bytes < 0) {   /* XVID_ERR_END */
# Line 341  Line 410 
410                  totaldectime += dectime;                  totaldectime += dectime;
411    
412                  /* Prints some decoding stats */                  /* Prints some decoding stats */
413                    if (!display_buffer_bytes) {
414          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",
415                             filenr, type2str(xvid_dec_stats.type), dectime, used_bytes);                             filenr, type2str(xvid_dec_stats.type), dectime, used_bytes);
416                    }
417    
418                  /* Save output frame if required */                  /* Save output frame if required */
419                  if (ARG_SAVEDECOUTPUT) {                  if (ARG_SAVEDECOUTPUT) {
420                          sprintf(filename, "%sdec%05d.pgm", filepath, filenr);                          sprintf(filename, "%sdec%05d", filepath, filenr);
421                          if(write_pgm(filename, out_buffer)) {                          if(write_image(filename, out_buffer)) {
422                                  fprintf(stderr,                                  fprintf(stderr,
423                                                  "Error writing decoded PGM frame %s\n",                                                  "Error writing decoded frame %s\n",
424                                                  filename);                                                  filename);
425                          }                          }
426                  }                  }
# Line 362  Line 433 
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(ms) =%7.2f, fps =%7.2f, length(bytes) =%7d\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 399  Line 473 
473          fprintf(stderr, " -debug         : debug level (debug=0)\n");          fprintf(stderr, " -debug         : debug level (debug=0)\n");
474          fprintf(stderr, " -i string      : input filename (default=stdin)\n");          fprintf(stderr, " -i string      : input filename (default=stdin)\n");
475          fprintf(stderr, " -d             : save decoder output\n");          fprintf(stderr, " -d             : save decoder output\n");
476            fprintf(stderr, " -c csp         : choose colorspace output (rgb16, rgb24, rgb32, yv12, i420)\n");
477            fprintf(stderr, " -f format      : choose output file format (tga, pnm, pgm)\n");
478          fprintf(stderr, " -m             : save mpeg4 raw stream to individual files\n");          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");
# Line 420  Line 496 
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 428  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          unsigned char *y = image;          if (FORMAT == USE_PNM && BPP == 1) {
514          unsigned char *u = image + XDIM*YDIM;                  ext = "pgm";
515          unsigned char *v = image + XDIM*YDIM + XDIM/2*YDIM/2;          } else if (FORMAT == USE_PNM && BPP == 3) {
516                    ext = "pnm";
517          FILE *filehandle;          } else if (FORMAT == USE_TGA) {
518          filehandle=fopen(filename,"w+b");                  ext = "tga";
519          if (filehandle) {          } 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                  /* Write header */          sprintf(filename, "%s.%s", prefix, ext);
                 fprintf(filehandle,"P5\n\n%d %d 255\n", XDIM,YDIM*3/2);  
525    
526                  /* Write Y data */          if (FORMAT == USE_PNM) {
527                  fwrite(y, 1, XDIM*YDIM, filehandle);                  ret = write_pnm(filename, image);
528            } else {
529                    ret = write_tga(filename, image);
530            }
531    
532                  for(loop=0; loop<YDIM/2; loop++)          return(ret);
533    }
534    
535    static int write_tga(char *filename, unsigned char *image)
536                  {                  {
537                          /* Write U scanline */          FILE * f;
538                          fwrite(u, 1, XDIM/2, filehandle);          char hdr[18];
539    
540                          /* Write V scanline */          f = fopen(filename, "wb");
541                          fwrite(v, 1, XDIM/2, filehandle);          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                          /* Update pointers */          /* Write header */
570                          u += XDIM/2;          fwrite(hdr, 1, sizeof(hdr), f);
                         v += XDIM/2;  
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                  /* Close file */          /* Write Y and V planes for YUV formats */
599                  fclose(filehandle);          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    
609    
610            /* Close the file */
611            fclose(f);
612    
613                  return(0);                  return(0);
614          }          }
615          else  
616                  return(1);  static int write_pnm(char *filename, unsigned char *image)
617    {
618            FILE * f;
619    
620            f = fopen(filename, "wb");
621            if ( f == NULL) {
622                    return -1;
623            }
624    
625            if (BPP == 1) {
626                    int i;
627                    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            fclose(f);
652    
653            return 0;
654  }  }
655    
656  /*****************************************************************************  /*****************************************************************************
# Line 484  Line 666 
666          xvid_gbl_init_t   xvid_gbl_init;          xvid_gbl_init_t   xvid_gbl_init;
667          xvid_dec_create_t xvid_dec_create;          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          /*------------------------------------------------------------------------          /*------------------------------------------------------------------------
674           * XviD core initialization           * XviD core initialization
675           *----------------------------------------------------------------------*/           *----------------------------------------------------------------------*/
# Line 538  Line 724 
724    
725          xvid_dec_frame_t xvid_dec_frame;          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 */          /* Set version */
732          xvid_dec_frame.version = XVID_VERSION;          xvid_dec_frame.version = XVID_VERSION;
733          xvid_dec_stats->version = XVID_VERSION;          xvid_dec_stats->version = XVID_VERSION;
# Line 551  Line 741 
741    
742          /* Output frame structure */          /* Output frame structure */
743          xvid_dec_frame.output.plane[0]  = ostream;          xvid_dec_frame.output.plane[0]  = ostream;
744          xvid_dec_frame.output.stride[0] = XDIM;          xvid_dec_frame.output.stride[0] = XDIM*BPP;
745          xvid_dec_frame.output.csp       = XVID_CSP_I420;          xvid_dec_frame.output.csp = CSP;
746    
747          ret = xvid_decore(dec_handle, XVID_DEC_DECODE, &xvid_dec_frame, xvid_dec_stats);          ret = xvid_decore(dec_handle, XVID_DEC_DECODE, &xvid_dec_frame, xvid_dec_stats);
748    
Legend:
Removed from v.1416  
changed lines
  Added in v.1715
No admin address has been configured
 ViewVC Help
Powered by ViewVC 1.0.4