[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 1382, Mon Mar 22 22:36:25 2004 UTC revision 1501, Sat Jul 17 11:42:46 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.10 2004-03-22 22:36:23 edgomez Exp $   * $Id: xvid_decraw.c,v 1.16 2004-07-17 11:42:46 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 74  Line 79 
79   ****************************************************************************/   ****************************************************************************/
80    
81  static double msecond();  static double msecond();
82  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);  
83  static int dec_main(unsigned char *istream,  static int dec_main(unsigned char *istream,
84                                          unsigned char *ostream,                                          unsigned char *ostream,
85                                          int istream_size,                                          int istream_size,
86                                          xvid_dec_stats_t *xvid_dec_stats);                                          xvid_dec_stats_t *xvid_dec_stats);
87  static int dec_stop();  static int dec_stop();
88  static void usage();  static void usage();
89    static int write_image(char *prefix, unsigned char *image);
90    static int write_pnm(char *filename, unsigned char *image);
91    static int write_tga(char *filename, unsigned char *image);
92    
93  const char * type2str(int type)  const char * type2str(int type)
94  {  {
# Line 114  Line 119 
119          int status;          int status;
120    
121          int use_assembler = 0;          int use_assembler = 0;
122            int debug_level = 0;
123    
124          char filename[256];          char filename[256];
125    
# Line 132  Line 138 
138    
139                  if (strcmp("-asm", argv[i]) == 0 ) {                  if (strcmp("-asm", argv[i]) == 0 ) {
140                          use_assembler = 1;                          use_assembler = 1;
141                    } else if (strcmp("-debug", argv[i]) == 0 && i < argc - 1 ) {
142                            i++;
143                            if (sscanf(argv[i], "0x%x", &debug_level) != 1) {
144                                    debug_level = atoi(argv[i]);
145                            }
146                  } else if (strcmp("-d", argv[i]) == 0) {                  } else if (strcmp("-d", argv[i]) == 0) {
147                          ARG_SAVEDECOUTPUT = 1;                          ARG_SAVEDECOUTPUT = 1;
148                  } else if (strcmp("-i", argv[i]) == 0 && i < argc - 1 ) {                  } else if (strcmp("-i", argv[i]) == 0 && i < argc - 1 ) {
# Line 139  Line 150 
150                          ARG_INPUTFILE = argv[i];                          ARG_INPUTFILE = argv[i];
151                  } else if (strcmp("-m", argv[i]) == 0) {                  } else if (strcmp("-m", argv[i]) == 0) {
152                          ARG_SAVEMPEGSTREAM = 1;                          ARG_SAVEMPEGSTREAM = 1;
153                    } else if (strcmp("-c", argv[i]) == 0  && i < argc - 1 ) {
154                            i++;
155                            if (strcmp(argv[i], "rgb16") == 0) {
156                                    CSP = XVID_CSP_RGB555;
157                                    BPP = 2;
158                            } else if (strcmp(argv[i], "rgb24") == 0) {
159                                    CSP = XVID_CSP_BGR;
160                                    BPP = 3;
161                            } else if (strcmp(argv[i], "rgb32") == 0) {
162                                    CSP = XVID_CSP_BGRA;
163                                    BPP = 4;
164                            } else if (strcmp(argv[i], "yv12") == 0) {
165                                    CSP = XVID_CSP_YV12;
166                                    BPP = 1;
167                            } else {
168                                    CSP = XVID_CSP_I420;
169                                    BPP = 1;
170                            }
171                    } else if (strcmp("-f", argv[i]) == 0 && i < argc -1) {
172                            i++;
173                            if (strcmp(argv[i], "tga") == 0) {
174                                    FORMAT = USE_TGA;
175                            } else {
176                                    FORMAT = USE_PNM;
177                            }
178                  } else if (strcmp("-help", argv[i]) == 0) {                  } else if (strcmp("-help", argv[i]) == 0) {
179                          usage();                          usage();
180                          return(0);                          return(0);
# Line 148  Line 184 
184                  }                  }
185          }          }
186    
187    #if defined(_MSC_VER)
188            if (ARG_INPUTFILE==NULL) {
189                    fprintf(stderr, "Warning: MSVC build does not read EOF correctly from stdin. Use the -i switch.\n\n");
190            }
191    #endif
192    
193  /*****************************************************************************  /*****************************************************************************
194   * Values checking   * Values checking
195   ****************************************************************************/   ****************************************************************************/
# Line 164  Line 206 
206                  }                  }
207          }          }
208    
209            /* PNM/PGM format can't handle 16/32 bit data */
210            if (BPP != 1 && BPP != 3 && FORMAT == USE_PNM) {
211                    FORMAT = USE_TGA;
212            }
213    
214  /*****************************************************************************  /*****************************************************************************
215   *        Memory allocation   *        Memory allocation
216   ****************************************************************************/   ****************************************************************************/
# Line 178  Line 225 
225   *        XviD PART  Start   *        XviD PART  Start
226   ****************************************************************************/   ****************************************************************************/
227    
228          status = dec_init(use_assembler);          status = dec_init(use_assembler, debug_level);
229          if (status) {          if (status) {
230                  fprintf(stderr,                  fprintf(stderr,
231                                  "Decore INIT problem, return value %d\n", status);                                  "Decore INIT problem, return value %d\n", status);
# Line 299  Line 346 
346    
347                  /* Save output frame if required */                  /* Save output frame if required */
348                  if (ARG_SAVEDECOUTPUT) {                  if (ARG_SAVEDECOUTPUT) {
349                          sprintf(filename, "%sdec%05d.pgm", filepath, filenr);                          sprintf(filename, "%sdec%05d", filepath, filenr);
350                          if(write_pgm(filename,out_buffer)) {                          if(write_image(filename, out_buffer)) {
351                                  fprintf(stderr,                                  fprintf(stderr,
352                                                  "Error writing decoded PGM frame %s\n",                                                  "Error writing decoded frame %s\n",
353                                                  filename);                                                  filename);
354                          }                          }
355                  }                  }
# Line 340  Line 387 
387    
388                  /* Save output frame if required */                  /* Save output frame if required */
389                  if (ARG_SAVEDECOUTPUT) {                  if (ARG_SAVEDECOUTPUT) {
390                          sprintf(filename, "%sdec%05d.pgm", filepath, filenr);                          sprintf(filename, "%sdec%05d", filepath, filenr);
391                          if(write_pgm(filename, out_buffer)) {                          if(write_image(filename, out_buffer)) {
392                                  fprintf(stderr,                                  fprintf(stderr,
393                                                  "Error writing decoded PGM frame %s\n",                                                  "Error writing decoded frame %s\n",
394                                                  filename);                                                  filename);
395                          }                          }
396                  }                  }
# Line 356  Line 403 
403   *     Calculate totals and averages for output, print results   *     Calculate totals and averages for output, print results
404   ****************************************************************************/   ****************************************************************************/
405    
406            if (filenr>0) {
407          totalsize    /= filenr;          totalsize    /= filenr;
408          totaldectime /= filenr;          totaldectime /= filenr;
   
409          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",
410                     totaldectime, 1000/totaldectime, (int)totalsize);                     totaldectime, 1000/totaldectime, (int)totalsize);
411            }else{
412                    printf("Nothing was decoded!\n");
413            }
414    
415  /*****************************************************************************  /*****************************************************************************
416   *      XviD PART  Stop   *      XviD PART  Stop
# Line 390  Line 440 
440          fprintf(stderr, "Usage : xvid_decraw [OPTIONS]\n");          fprintf(stderr, "Usage : xvid_decraw [OPTIONS]\n");
441          fprintf(stderr, "Options :\n");          fprintf(stderr, "Options :\n");
442          fprintf(stderr, " -asm           : use assembly optimizations (default=disabled)\n");          fprintf(stderr, " -asm           : use assembly optimizations (default=disabled)\n");
443            fprintf(stderr, " -debug         : debug level (debug=0)\n");
444          fprintf(stderr, " -i string      : input filename (default=stdin)\n");          fprintf(stderr, " -i string      : input filename (default=stdin)\n");
445          fprintf(stderr, " -d             : save decoder output\n");          fprintf(stderr, " -d             : save decoder output\n");
446            fprintf(stderr, " -c csp         : choose colorspace output (rgb16, rgb24, rgb32, yv12, i420)\n");
447            fprintf(stderr, " -f format      : choose output file format (tga, pnm, pgm)\n");
448          fprintf(stderr, " -m             : save mpeg4 raw stream to individual files\n");          fprintf(stderr, " -m             : save mpeg4 raw stream to individual files\n");
449          fprintf(stderr, " -help          : This help message\n");          fprintf(stderr, " -help          : This help message\n");
450          fprintf(stderr, " (* means default)\n");          fprintf(stderr, " (* means default)\n");
# Line 421  Line 474 
474   *              output functions   *              output functions
475   ****************************************************************************/   ****************************************************************************/
476    
477  static int  static int write_image(char *prefix, unsigned char *image)
 write_pgm(char *filename,  
                   unsigned char *image)  
478  {  {
479          int loop;          char filename[1024];
480            char *ext;
481            int ret;
482    
483          unsigned char *y = image;          if (FORMAT == USE_PNM && BPP == 1) {
484          unsigned char *u = image + XDIM*YDIM;                  ext = "pgm";
485          unsigned char *v = image + XDIM*YDIM + XDIM/2*YDIM/2;          } else if (FORMAT == USE_PNM && BPP == 3) {
486                    ext = "pnm";
487          FILE *filehandle;          } else if (FORMAT == USE_TGA) {
488          filehandle=fopen(filename,"w+b");                  ext = "tga";
489          if (filehandle) {          } else {
490                    fprintf(stderr, "Bug: should not reach this path code -- please report to xvid-devel@xvid.org with command line options used");
491                    exit(-1);
492            }
493    
494                  /* Write header */          sprintf(filename, "%s.%s", prefix, ext);
                 fprintf(filehandle,"P5\n\n%d %d 255\n", XDIM,YDIM*3/2);  
495    
496                  /* Write Y data */          if (FORMAT == USE_PNM) {
497                  fwrite(y, 1, XDIM*YDIM, filehandle);                  ret = write_pnm(filename, image);
498            } else {
499                    ret = write_tga(filename, image);
500            }
501    
502            return(ret);
503    }
504    
505                  for(loop=0; loop<YDIM/2; loop++)  static int write_tga(char *filename, unsigned char *image)
506                  {                  {
507                          /* Write U scanline */          FILE * f;
508                          fwrite(u, 1, XDIM/2, filehandle);          char hdr[18];
509    
510                          /* Write V scanline */          f = fopen(filename, "wb");
511                          fwrite(v, 1, XDIM/2, filehandle);          if ( f == NULL) {
512                    return -1;
513            }
514    
515            hdr[0]  = 0; /* ID length */
516            hdr[1]  = 0; /* Color map type */
517            hdr[2]  = (BPP>1)?2:3; /* Uncompressed true color (2) or greymap (3) */
518            hdr[3]  = 0; /* Color map specification (not used) */
519            hdr[4]  = 0; /* Color map specification (not used) */
520            hdr[5]  = 0; /* Color map specification (not used) */
521            hdr[6]  = 0; /* Color map specification (not used) */
522            hdr[7]  = 0; /* Color map specification (not used) */
523            hdr[8]  = 0; /* LSB X origin */
524            hdr[9]  = 0; /* MSB X origin */
525            hdr[10] = 0; /* LSB Y origin */
526            hdr[11] = 0; /* MSB Y origin */
527            hdr[12] = (XDIM>>0)&0xff; /* LSB Width */
528            hdr[13] = (XDIM>>8)&0xff; /* MSB Width */
529            if (BPP > 1) {
530                    hdr[14] = (YDIM>>0)&0xff; /* LSB Height */
531                    hdr[15] = (YDIM>>8)&0xff; /* MSB Height */
532            } else {
533                    hdr[14] = ((YDIM*3)>>1)&0xff; /* LSB Height */
534                    hdr[15] = ((YDIM*3)>>9)&0xff; /* MSB Height */
535            }
536            hdr[16] = BPP*8;
537            hdr[17] = 0x00 | (1<<5) /* Up to down */ | (0<<4); /* Image descriptor */
538    
539                          /* Update pointers */          /* Write header */
540                          u += XDIM/2;          fwrite(hdr, 1, sizeof(hdr), f);
                         v += XDIM/2;  
541    
542    #ifdef ARCH_IS_LITTLE_ENDIAN
543            /* write first plane */
544            fwrite(image, 1, XDIM*YDIM*BPP, f);
545    #else
546            {
547                    int i;
548                    for (i=0; i<width*height*BPP;i+=BPP) {
549                            if (BPP == 1) {
550                                    fputc(image+i, f);
551                            } else if (BPP == 2) {
552                                    fputc(image+i+1, f);
553                                    fputc(image+i+0, f);
554                            } else if (BPP == 3) {
555                                    fputc(image+i+2, f);
556                                    fputc(image+i+1, f);
557                                    fputc(image+i+0, f);
558                            } else if (BPP == 4) {
559                                    fputc(image+i+3, f);
560                                    fputc(image+i+2, f);
561                                    fputc(image+i+1, f);
562                                    fputc(image+i+0, f);
563                            }
564                  }                  }
565            }
566    #endif
567    
568                  /* Close file */          /* Write Y and V planes for YUV formats */
569                  fclose(filehandle);          if (BPP == 1) {
570                    int i;
571    
572                    /* Write the two chrominance planes */
573                    for (i=0; i<YDIM/2; i++) {
574                            fwrite(image+XDIM*YDIM + i*XDIM/2, 1, XDIM/2, f);
575                            fwrite(image+5*XDIM*YDIM/4 + i*XDIM/2, 1, XDIM/2, f);
576                    }
577            }
578    
579    
580            /* Close the file */
581            fclose(f);
582    
583                  return(0);                  return(0);
584          }          }
585          else  
586                  return(1);  static int write_pnm(char *filename, unsigned char *image)
587    {
588            FILE * f;
589    
590            f = fopen(filename, "wb");
591            if ( f == NULL) {
592                    return -1;
593            }
594    
595            if (BPP == 1) {
596                    int i;
597                    fprintf(f, "P5\n#xvid\n%i %i\n255\n", XDIM, YDIM*3/2);
598    
599                    fwrite(image, 1, XDIM*YDIM, f);
600    
601                    for (i=0; i<YDIM/2;i++) {
602                            fwrite(image+XDIM*YDIM + i*XDIM/2, 1, XDIM/2, f);
603                            fwrite(image+5*XDIM*YDIM/4 + i*XDIM/2, 1, XDIM/2, f);
604                    }
605            } else if (BPP == 3) {
606                    int i;
607                    fprintf(f, "P6\n#xvid\n%i %i\n255\n", XDIM, YDIM);
608                    for (i=0; i<XDIM*YDIM*3; i+=3) {
609    #ifdef ARCH_IS_LITTLE_ENDIAN
610                            fputc(image[i+2], f);
611                            fputc(image[i+1], f);
612                            fputc(image[i+0], f);
613    #else
614                            fputc(image[i+0], f);
615                            fputc(image[i+1], f);
616                            fputc(image[i+2], f);
617    #endif
618                    }
619            }
620    
621            fclose(f);
622    
623            return 0;
624  }  }
625    
626  /*****************************************************************************  /*****************************************************************************
# Line 470  Line 629 
629    
630  /* init decoder before first run */  /* init decoder before first run */
631  static int  static int
632  dec_init(int use_assembler)  dec_init(int use_assembler, int debug_level)
633  {  {
634          int ret;          int ret;
635    
636          xvid_gbl_init_t   xvid_gbl_init;          xvid_gbl_init_t   xvid_gbl_init;
637          xvid_dec_create_t xvid_dec_create;          xvid_dec_create_t xvid_dec_create;
638    
639            /* Reset the structure with zeros */
640            memset(&xvid_gbl_init, 0, sizeof(xvid_gbl_init_t));
641            memset(&xvid_dec_create, 0, sizeof(xvid_dec_create_t));
642    
643          /*------------------------------------------------------------------------          /*------------------------------------------------------------------------
644           * XviD core initialization           * XviD core initialization
645           *----------------------------------------------------------------------*/           *----------------------------------------------------------------------*/
# Line 494  Line 657 
657          else          else
658                  xvid_gbl_init.cpu_flags = XVID_CPU_FORCE;                  xvid_gbl_init.cpu_flags = XVID_CPU_FORCE;
659    
660            xvid_gbl_init.debug = debug_level;
661    
662          xvid_global(NULL, 0, &xvid_gbl_init, NULL);          xvid_global(NULL, 0, &xvid_gbl_init, NULL);
663    
664          /*------------------------------------------------------------------------          /*------------------------------------------------------------------------
# Line 542  Line 707 
707    
708          /* Output frame structure */          /* Output frame structure */
709          xvid_dec_frame.output.plane[0]  = ostream;          xvid_dec_frame.output.plane[0]  = ostream;
710          xvid_dec_frame.output.stride[0] = XDIM;          xvid_dec_frame.output.stride[0] = XDIM*BPP;
711          xvid_dec_frame.output.csp       = XVID_CSP_I420;          xvid_dec_frame.output.csp = CSP;
712    
713          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);
714    
Legend:
Removed from v.1382  
changed lines
  Added in v.1501
No admin address has been configured
 ViewVC Help
Powered by ViewVC 1.0.4