[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 1416, Sat Apr 10 04:25:31 2004 UTC revision 1432, Thu Apr 15 19:44:06 2004 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.14 2004-04-15 19:44:05 edgomez Exp $
24   *   *
25   ****************************************************************************/   ****************************************************************************/
26    
# Line 57  Line 57 
57  /* max number of frames */  /* max number of frames */
58  #define ABS_MAXFRAMENR 9999  #define ABS_MAXFRAMENR 9999
59    
60    #define USE_PNM 0
61    #define USE_TGA 1
62    
63  static int XDIM = 0;  static int XDIM = 0;
64  static int YDIM = 0;  static int YDIM = 0;
65  static int ARG_SAVEDECOUTPUT = 0;  static int ARG_SAVEDECOUTPUT = 0;
66  static int ARG_SAVEMPEGSTREAM = 0;  static int ARG_SAVEMPEGSTREAM = 0;
67  static char *ARG_INPUTFILE = NULL;  static char *ARG_INPUTFILE = NULL;
68    static int CSP = XVID_CSP_I420;
69    static int BPP = 1;
70    static int FORMAT = USE_PNM;
71    
72  static char filepath[256] = "./";  static char filepath[256] = "./";
73  static void *dec_handle = NULL;  static void *dec_handle = NULL;
# Line 83  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    int write_image(char *prefix, unsigned char *image);
92    int write_pnm(char *filename, unsigned char *image);
93    int write_tga(char *filename, unsigned char *image);
94    
95  const char * type2str(int type)  const char * type2str(int type)
96  {  {
# 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 305  Line 348 
348    
349                  /* Save output frame if required */                  /* Save output frame if required */
350                  if (ARG_SAVEDECOUTPUT) {                  if (ARG_SAVEDECOUTPUT) {
351                          sprintf(filename, "%sdec%05d.pgm", filepath, filenr);                          sprintf(filename, "%sdec%05d", filepath, filenr);
352                          if(write_pgm(filename,out_buffer)) {                          if(write_image(filename, out_buffer)) {
353                                  fprintf(stderr,                                  fprintf(stderr,
354                                                  "Error writing decoded PGM frame %s\n",                                                  "Error writing decoded PGM frame %s\n",
355                                                  filename);                                                  filename);
# Line 346  Line 389 
389    
390                  /* Save output frame if required */                  /* Save output frame if required */
391                  if (ARG_SAVEDECOUTPUT) {                  if (ARG_SAVEDECOUTPUT) {
392                          sprintf(filename, "%sdec%05d.pgm", filepath, filenr);                          sprintf(filename, "%sdec%05d", filepath, filenr);
393                          if(write_pgm(filename, out_buffer)) {                          if(write_image(filename, out_buffer)) {
394                                  fprintf(stderr,                                  fprintf(stderr,
395                                                  "Error writing decoded PGM frame %s\n",                                                  "Error writing decoded frame %s\n",
396                                                  filename);                                                  filename);
397                          }                          }
398                  }                  }
# Line 362  Line 405 
405   *     Calculate totals and averages for output, print results   *     Calculate totals and averages for output, print results
406   ****************************************************************************/   ****************************************************************************/
407    
408            if (filenr>0) {
409          totalsize    /= filenr;          totalsize    /= filenr;
410          totaldectime /= filenr;          totaldectime /= filenr;
   
411          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",
412                     totaldectime, 1000/totaldectime, (int)totalsize);                     totaldectime, 1000/totaldectime, (int)totalsize);
413            }else{
414                    printf("Nothing was decoded!\n");
415            }
416    
417  /*****************************************************************************  /*****************************************************************************
418   *      XviD PART  Stop   *      XviD PART  Stop
# Line 399  Line 445 
445          fprintf(stderr, " -debug         : debug level (debug=0)\n");          fprintf(stderr, " -debug         : debug level (debug=0)\n");
446          fprintf(stderr, " -i string      : input filename (default=stdin)\n");          fprintf(stderr, " -i string      : input filename (default=stdin)\n");
447          fprintf(stderr, " -d             : save decoder output\n");          fprintf(stderr, " -d             : save decoder output\n");
448            fprintf(stderr, " -c csp         : choose colorspace output (rgb16, rgb24, rgb32, yv12, i420)\n");
449            fprintf(stderr, " -f format      : choose output file format (tga, pnm, pgm)\n");
450          fprintf(stderr, " -m             : save mpeg4 raw stream to individual files\n");          fprintf(stderr, " -m             : save mpeg4 raw stream to individual files\n");
451          fprintf(stderr, " -help          : This help message\n");          fprintf(stderr, " -help          : This help message\n");
452          fprintf(stderr, " (* means default)\n");          fprintf(stderr, " (* means default)\n");
# Line 428  Line 476 
476   *              output functions   *              output functions
477   ****************************************************************************/   ****************************************************************************/
478    
479  static int  int write_image(char *prefix, unsigned char *image)
 write_pgm(char *filename,  
                   unsigned char *image)  
480  {  {
481          int loop;          char filename[1024];
482            char *ext;
483            int ret;
484    
485          unsigned char *y = image;          if (FORMAT == USE_PNM && BPP == 1) {
486          unsigned char *u = image + XDIM*YDIM;                  ext = "pgm";
487          unsigned char *v = image + XDIM*YDIM + XDIM/2*YDIM/2;          } else if (FORMAT == USE_PNM && BPP == 3) {
488                    ext = "pnm";
489          FILE *filehandle;          } else if (FORMAT == USE_TGA) {
490          filehandle=fopen(filename,"w+b");                  ext = "tga";
491          if (filehandle) {          } else {
492                    fprintf(stderr, "Bug: should not reach this path code -- please report to xvid-devel@xvid.org with command line options used");
493                    exit(-1);
494            }
495    
496                  /* Write header */          sprintf(filename, "%s.%s", prefix, ext);
                 fprintf(filehandle,"P5\n\n%d %d 255\n", XDIM,YDIM*3/2);  
497    
498                  /* Write Y data */          if (FORMAT == USE_PNM) {
499                  fwrite(y, 1, XDIM*YDIM, filehandle);                  ret = write_pnm(filename, image);
500            } else {
501                    ret = write_tga(filename, image);
502            }
503    
504                  for(loop=0; loop<YDIM/2; loop++)          return(ret);
505    }
506    
507    int write_tga(char *filename, unsigned char *image)
508                  {                  {
509                          /* Write U scanline */          FILE * f;
510                          fwrite(u, 1, XDIM/2, filehandle);          char hdr[18];
511    
512                          /* Write V scanline */          f = fopen(filename, "wb");
513                          fwrite(v, 1, XDIM/2, filehandle);          if ( f == NULL) {
514                    return -1;
515            }
516    
517            hdr[0]  = 0; /* ID length */
518            hdr[1]  = 0; /* Color map type */
519            hdr[2]  = (BPP>1)?2:3; /* Uncompressed true color (2) or greymap (3) */
520            hdr[3]  = 0; /* Color map specification (not used) */
521            hdr[4]  = 0; /* Color map specification (not used) */
522            hdr[5]  = 0; /* Color map specification (not used) */
523            hdr[6]  = 0; /* Color map specification (not used) */
524            hdr[7]  = 0; /* Color map specification (not used) */
525            hdr[8]  = 0; /* LSB X origin */
526            hdr[9]  = 0; /* MSB X origin */
527            hdr[10] = 0; /* LSB Y origin */
528            hdr[11] = 0; /* MSB Y origin */
529            hdr[12] = (XDIM>>0)&0xff; /* LSB Width */
530            hdr[13] = (XDIM>>8)&0xff; /* MSB Width */
531            if (BPP > 1) {
532                    hdr[14] = (YDIM>>0)&0xff; /* LSB Height */
533                    hdr[15] = (YDIM>>8)&0xff; /* MSB Height */
534            } else {
535                    hdr[14] = ((YDIM*3)>>1)&0xff; /* LSB Height */
536                    hdr[15] = ((YDIM*3)>>9)&0xff; /* MSB Height */
537            }
538            hdr[16] = BPP*8;
539            hdr[17] = 0x00 | (1<<5) /* Up to down */ | (0<<4); /* Image descriptor */
540    
541                          /* Update pointers */          /* Write header */
542                          u += XDIM/2;          fwrite(hdr, 1, sizeof(hdr), f);
                         v += XDIM/2;  
543    
544    #ifdef ARCH_IS_LITTLE_ENDIAN
545            /* write first plane */
546            fwrite(image, 1, XDIM*YDIM*BPP, f);
547    #else
548            {
549                    int i;
550                    for (i=0; i<width*height*BPP;i+=BPP) {
551                            if (BPP == 1) {
552                                    fputc(image+i, f);
553                            } else if (BPP == 2) {
554                                    fputc(image+i+1, f);
555                                    fputc(image+i+0, f);
556                            } else if (BPP == 3) {
557                                    fputc(image+i+2, f);
558                                    fputc(image+i+1, f);
559                                    fputc(image+i+0, f);
560                            } else if (BPP == 4) {
561                                    fputc(image+i+3, f);
562                                    fputc(image+i+2, f);
563                                    fputc(image+i+1, f);
564                                    fputc(image+i+0, f);
565                  }                  }
566                    }
567            }
568    #endif
569    
570                  /* Close file */          /* Write Y and V planes for YUV formats */
571                  fclose(filehandle);          if (BPP == 1) {
572                    int i;
573    
574                    /* Write the two chrominance planes */
575                    for (i=0; i<YDIM/2; i++) {
576                            fwrite(image+XDIM*YDIM + i*XDIM/2, 1, XDIM/2, f);
577                            fwrite(image+5*XDIM*YDIM/4 + i*XDIM/2, 1, XDIM/2, f);
578                    }
579            }
580    
581    
582            /* Close the file */
583            fclose(f);
584    
585                  return(0);                  return(0);
586          }          }
587          else  
588                  return(1);  int write_pnm(char *filename, unsigned char *image)
589    {
590            FILE * f;
591    
592            f = fopen(filename, "wb");
593            if ( f == NULL) {
594                    return -1;
595  }  }
596    
597            if (BPP == 1) {
598                    int i;
599                    fprintf(f, "P5\n#xvid\n%i %i\n255\n", XDIM, YDIM*3/2);
600    
601                    fwrite(image, 1, XDIM*YDIM, f);
602    
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            } else if (BPP == 3) {
608                    int i;
609                    fprintf(f, "P6\n#xvid\n%i %i\n255\n", XDIM, YDIM);
610                    for (i=0; i<XDIM*YDIM*3; i+=3) {
611    #ifdef ARCH_IS_LITTLE_ENDIAN
612                            fputc(image[i+2], f);
613                            fputc(image[i+1], f);
614                            fputc(image[i+0], f);
615    #else
616                            fputc(image[i+0], f);
617                            fputc(image[i+1], f);
618                            fputc(image[i+2], f);
619    #endif
620                    }
621            }
622    
623            fclose(f);
624    
625            return 0;
626    }
627  /*****************************************************************************  /*****************************************************************************
628   * Routines for decoding: init decoder, use, and stop decoder   * Routines for decoding: init decoder, use, and stop decoder
629   ****************************************************************************/   ****************************************************************************/
# Line 551  Line 704 
704    
705          /* Output frame structure */          /* Output frame structure */
706          xvid_dec_frame.output.plane[0]  = ostream;          xvid_dec_frame.output.plane[0]  = ostream;
707          xvid_dec_frame.output.stride[0] = XDIM;          xvid_dec_frame.output.stride[0] = XDIM*BPP;
708          xvid_dec_frame.output.csp       = XVID_CSP_I420;          xvid_dec_frame.output.csp = CSP;
709    
710          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);
711    
Legend:
Removed from v.1416  
changed lines
  Added in v.1432
No admin address has been configured
 ViewVC Help
Powered by ViewVC 1.0.4