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

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

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

revision 1387, Mon Mar 22 23:56:55 2004 UTC revision 1418, Sat Apr 10 10:05:47 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.10.2.2 2004-04-10 10:05:47 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 139  Line 144 
144                          ARG_INPUTFILE = argv[i];                          ARG_INPUTFILE = argv[i];
145                  } else if (strcmp("-m", argv[i]) == 0) {                  } else if (strcmp("-m", argv[i]) == 0) {
146                          ARG_SAVEMPEGSTREAM = 1;                          ARG_SAVEMPEGSTREAM = 1;
147                    } else if (strcmp("-c", argv[i]) == 0  && i < argc - 1 ) {
148                            i++;
149                            if (strcmp(argv[i], "rgb16") == 0) {
150                                    CSP = XVID_CSP_RGB555;
151                                    BPP = 2;
152                            } else if (strcmp(argv[i], "rgb24") == 0) {
153                                    CSP = XVID_CSP_BGR;
154                                    BPP = 3;
155                            } else if (strcmp(argv[i], "rgb32") == 0) {
156                                    CSP = XVID_CSP_BGRA;
157                                    BPP = 4;
158                            } else if (strcmp(argv[i], "yv12") == 0) {
159                                    CSP = XVID_CSP_YV12;
160                                    BPP = 1;
161                            } else {
162                                    CSP = XVID_CSP_I420;
163                                    BPP = 1;
164                            }
165                    } else if (strcmp("-f", argv[i]) == 0 && i < argc -1) {
166                            i++;
167                            if (strcmp(argv[i], "tga") == 0) {
168                                    FORMAT = USE_TGA;
169                            } else {
170                                    FORMAT = USE_PNM;
171                            }
172                  } else if (strcmp("-help", argv[i]) == 0) {                  } else if (strcmp("-help", argv[i]) == 0) {
173                          usage();                          usage();
174                          return(0);                          return(0);
# Line 164  Line 194 
194                  }                  }
195          }          }
196    
197            /* PNM/PGM format can't handle 16/32 bit data */
198            if (BPP != 1 && BPP != 3 && FORMAT == USE_PNM) {
199                    FORMAT = USE_TGA;
200            }
201    
202  /*****************************************************************************  /*****************************************************************************
203   *        Memory allocation   *        Memory allocation
204   ****************************************************************************/   ****************************************************************************/
# Line 299  Line 334 
334    
335                  /* Save output frame if required */                  /* Save output frame if required */
336                  if (ARG_SAVEDECOUTPUT) {                  if (ARG_SAVEDECOUTPUT) {
337                          sprintf(filename, "%sdec%05d.pgm", filepath, filenr);                          sprintf(filename, "%sdec%05d", filepath, filenr);
338                          if(write_pgm(filename,out_buffer)) {                          if(write_image(filename, out_buffer)) {
339                                  fprintf(stderr,                                  fprintf(stderr,
340                                                  "Error writing decoded PGM frame %s\n",                                                  "Error writing decoded PGM frame %s\n",
341                                                  filename);                                                  filename);
# Line 340  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.pgm", filepath, filenr);                          sprintf(filename, "%sdec%05d", filepath, filenr);
379                          if(write_pgm(filename, out_buffer)) {                          if(write_image(filename, out_buffer)) {
380                                  fprintf(stderr,                                  fprintf(stderr,
381                                                  "Error writing decoded PGM frame %s\n",                                                  "Error writing decoded PGM frame %s\n",
382                                                  filename);                                                  filename);
# Line 392  Line 427 
427          fprintf(stderr, " -asm           : use assembly optimizations (default=disabled)\n");          fprintf(stderr, " -asm           : use assembly optimizations (default=disabled)\n");
428          fprintf(stderr, " -i string      : input filename (default=stdin)\n");          fprintf(stderr, " -i string      : input filename (default=stdin)\n");
429          fprintf(stderr, " -d             : save decoder output\n");          fprintf(stderr, " -d             : save decoder output\n");
430            fprintf(stderr, " -c csp         : choose colorspace output (rgb16, rgb24, rgb32, yv12, i420)\n");
431            fprintf(stderr, " -f format      : choose output file format (tga, pnm, pgm)\n");
432          fprintf(stderr, " -m             : save mpeg4 raw stream to individual files\n");          fprintf(stderr, " -m             : save mpeg4 raw stream to individual files\n");
433          fprintf(stderr, " -help          : This help message\n");          fprintf(stderr, " -help          : This help message\n");
434          fprintf(stderr, " (* means default)\n");          fprintf(stderr, " (* means default)\n");
# Line 421  Line 458 
458   *              output functions   *              output functions
459   ****************************************************************************/   ****************************************************************************/
460    
461  static int  int write_image(char *prefix, unsigned char *image)
 write_pgm(char *filename,  
                   unsigned char *image)  
462  {  {
463          int loop;          char filename[1024];
464            char *ext;
465            int ret;
466    
467          unsigned char *y = image;          if (FORMAT == USE_PNM && BPP == 1) {
468          unsigned char *u = image + XDIM*YDIM;                  ext = "pgm";
469          unsigned char *v = image + XDIM*YDIM + XDIM/2*YDIM/2;          } else if (FORMAT == USE_PNM && BPP == 3) {
470                    ext = "pnm";
471          FILE *filehandle;          } else if (FORMAT == USE_TGA) {
472          filehandle=fopen(filename,"w+b");                  ext = "tga";
473          if (filehandle) {          } else {
474                    fprintf(stderr, "Bug: should not reach this path code -- please report to xvid-devel@xvid.org with command line options used");
475                    exit(-1);
476            }
477    
478                  /* Write header */          sprintf(filename, "%s.%s", prefix, ext);
                 fprintf(filehandle,"P5\n\n%d %d 255\n", XDIM,YDIM*3/2);  
479    
480                  /* Write Y data */          if (FORMAT == USE_PNM) {
481                  fwrite(y, 1, XDIM*YDIM, filehandle);                  ret = write_pnm(filename, image);
482            } else {
483                    ret = write_tga(filename, image);
484            }
485    
486            return(ret);
487    }
488    
489                  for(loop=0; loop<YDIM/2; loop++)  int write_tga(char *filename, unsigned char *image)
490                  {                  {
491                          /* Write U scanline */          FILE * f;
492                          fwrite(u, 1, XDIM/2, filehandle);          char hdr[18];
493    
494                          /* Write V scanline */          f = fopen(filename, "wb");
495                          fwrite(v, 1, XDIM/2, filehandle);          if ( f == NULL) {
496                    return -1;
497            }
498    
499            hdr[0]  = 0; /* ID length */
500            hdr[1]  = 0; /* Color map type */
501            hdr[2]  = (BPP>1)?2:3; /* Uncompressed true color (2) or greymap (3) */
502            hdr[3]  = 0; /* Color map specification (not used) */
503            hdr[4]  = 0; /* Color map specification (not used) */
504            hdr[5]  = 0; /* Color map specification (not used) */
505            hdr[6]  = 0; /* Color map specification (not used) */
506            hdr[7]  = 0; /* Color map specification (not used) */
507            hdr[8]  = 0; /* LSB X origin */
508            hdr[9]  = 0; /* MSB X origin */
509            hdr[10] = 0; /* LSB Y origin */
510            hdr[11] = 0; /* MSB Y origin */
511            hdr[12] = (XDIM>>0)&0xff; /* LSB Width */
512            hdr[13] = (XDIM>>8)&0xff; /* MSB Width */
513            if (BPP > 1) {
514                    hdr[14] = (YDIM>>0)&0xff; /* LSB Height */
515                    hdr[15] = (YDIM>>8)&0xff; /* MSB Height */
516            } else {
517                    hdr[14] = ((YDIM*3)>>1)&0xff; /* LSB Height */
518                    hdr[15] = ((YDIM*3)>>9)&0xff; /* MSB Height */
519            }
520            hdr[16] = BPP*8;
521            hdr[17] = 0x00 | (1<<5) /* Up to down */ | (0<<4); /* Image descriptor */
522    
523                          /* Update pointers */          /* Write header */
524                          u += XDIM/2;          fwrite(hdr, 1, sizeof(hdr), f);
                         v += XDIM/2;  
525    
526    #ifdef ARCH_IS_LITTLE_ENDIAN
527            /* write first plane */
528            fwrite(image, 1, XDIM*YDIM*BPP, f);
529    #else
530            {
531                    int i;
532                    for (i=0; i<width*height*BPP;i+=BPP) {
533                            if (BPP == 1) {
534                                    fputc(image+i, f);
535                            } else if (BPP == 2) {
536                                    fputc(image+i+1, f);
537                                    fputc(image+i+0, f);
538                            } else if (BPP == 3) {
539                                    fputc(image+i+2, f);
540                                    fputc(image+i+1, f);
541                                    fputc(image+i+0, f);
542                            } else if (BPP == 4) {
543                                    fputc(image+i+3, f);
544                                    fputc(image+i+2, f);
545                                    fputc(image+i+1, f);
546                                    fputc(image+i+0, f);
547                  }                  }
548                    }
549            }
550    #endif
551    
552                  /* Close file */          /* Write Y and V planes for YUV formats */
553                  fclose(filehandle);          if (BPP == 1) {
554                    int i;
555    
556                    /* Write the two chrominance planes */
557                    for (i=0; i<YDIM/2; i++) {
558                            fwrite(image+XDIM*YDIM + i*XDIM/2, 1, XDIM/2, f);
559                            fwrite(image+5*XDIM*YDIM/4 + i*XDIM/2, 1, XDIM/2, f);
560                    }
561            }
562    
563    
564            /* Close the file */
565            fclose(f);
566    
567                  return(0);                  return(0);
568          }          }
569          else  
570                  return(1);  int write_pnm(char *filename, unsigned char *image)
571    {
572            FILE * f;
573    
574            f = fopen(filename, "wb");
575            if ( f == NULL) {
576                    return -1;
577  }  }
578    
579            if (BPP == 1) {
580                    int i;
581                    fprintf(f, "P5\n#xvid\n%i %i\n255\n", XDIM, YDIM*3/2);
582    
583                    fwrite(image, 1, XDIM*YDIM, f);
584    
585                    for (i=0; i<YDIM/2;i++) {
586                            fwrite(image+XDIM*YDIM + i*XDIM/2, 1, XDIM/2, f);
587                            fwrite(image+5*XDIM*YDIM/4 + i*XDIM/2, 1, XDIM/2, f);
588                    }
589            } else if (BPP == 3) {
590                    int i;
591                    fprintf(f, "P6\n#xvid\n%i %i\n255\n", XDIM, YDIM);
592                    for (i=0; i<XDIM*YDIM*3; i+=3) {
593    #ifdef ARCH_IS_LITTLE_ENDIAN
594                            fputc(image[i+2], f);
595                            fputc(image[i+1], f);
596                            fputc(image[i+0], f);
597    #else
598                            fputc(image[i+0], f);
599                            fputc(image[i+1], f);
600                            fputc(image[i+2], f);
601    #endif
602                    }
603            }
604    
605            fclose(f);
606    
607            return 0;
608    }
609  /*****************************************************************************  /*****************************************************************************
610   * Routines for decoding: init decoder, use, and stop decoder   * Routines for decoding: init decoder, use, and stop decoder
611   ****************************************************************************/   ****************************************************************************/
# Line 542  Line 684 
684    
685          /* Output frame structure */          /* Output frame structure */
686          xvid_dec_frame.output.plane[0]  = ostream;          xvid_dec_frame.output.plane[0]  = ostream;
687          xvid_dec_frame.output.stride[0] = XDIM;          xvid_dec_frame.output.stride[0] = XDIM*BPP;
688          xvid_dec_frame.output.csp       = XVID_CSP_I420;          xvid_dec_frame.output.csp = CSP;
689    
690          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);
691    
Legend:
Removed from v.1387  
changed lines
  Added in v.1418
No admin address has been configured
 ViewVC Help
Powered by ViewVC 1.0.4