[svn] / trunk / xvidcore / examples / xvid_encraw.c Repository:
ViewVC logotype

Diff of /trunk/xvidcore/examples/xvid_encraw.c

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

revision 1622, Wed Jun 22 06:18:14 2005 UTC revision 1676, Tue Jan 17 19:06:25 2006 UTC
# Line 21  Line 21 
21   *  along with this program; if not, write to the Free Software   *  along with this program; if not, write to the Free Software
22   *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA   *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
23   *   *
24   * $Id: xvid_encraw.c,v 1.17 2005-06-22 06:18:14 Skal Exp $   * $Id: xvid_encraw.c,v 1.23 2006-01-17 19:06:25 Isibaar Exp $
25   *   *
26   ****************************************************************************/   ****************************************************************************/
27    
# Line 45  Line 45 
45  #ifndef WIN32  #ifndef WIN32
46  #include <sys/time.h>  #include <sys/time.h>
47  #else  #else
48    #include <windows.h>
49    #include <vfw.h>
50  #include <time.h>  #include <time.h>
51    #define XVID_AVI_INPUT
52  #endif  #endif
53    
54  #include "xvid.h"  #include "xvid.h"
# Line 125  Line 128 
128  static int NUM_ZONES = 0;  static int NUM_ZONES = 0;
129    
130  /* Maximum number of frames to encode */  /* Maximum number of frames to encode */
131  #define ABS_MAXFRAMENR 9999  #define ABS_MAXFRAMENR -1 /* no limit */
132    
133  static int ARG_STATS = 0;  static int ARG_STATS = 0;
134  static int ARG_DUMP = 0;  static int ARG_DUMP = 0;
# Line 152  Line 155 
155  static int ARG_DEBUG = 0;  static int ARG_DEBUG = 0;
156  static int ARG_VOPDEBUG = 0;  static int ARG_VOPDEBUG = 0;
157  static int ARG_GREYSCALE = 0;  static int ARG_GREYSCALE = 0;
158    static int ARG_QTYPE = 0;
159    static int ARG_QMATRIX = 0;
160  static int ARG_GMC = 0;  static int ARG_GMC = 0;
161  static int ARG_INTERLACING = 0;  static int ARG_INTERLACING = 0;
162  static int ARG_QPEL = 0;  static int ARG_QPEL = 0;
163    static int ARG_TURBO = 0;
164    static int ARG_VHQMODE = 0;
165    static int ARG_BVHQ = 0;
166  static int ARG_CLOSED_GOP = 0;  static int ARG_CLOSED_GOP = 0;
167    
168  #ifndef READ_PNM  #ifndef READ_PNM
# Line 181  Line 189 
189  /* Internal structures (handles) for encoding and decoding */  /* Internal structures (handles) for encoding and decoding */
190  static void *enc_handle = NULL;  static void *enc_handle = NULL;
191    
192    static unsigned char qmatrix_intra[64];
193    static unsigned char qmatrix_inter[64];
194    
195    #ifdef XVID_AVI_INPUT
196    static PAVISTREAM avi_stream = NULL;
197    #endif
198    
199  /*****************************************************************************  /*****************************************************************************
200   *               Local prototypes   *               Local prototypes
201   ****************************************************************************/   ****************************************************************************/
# Line 239  Line 254 
254          int stats_type;          int stats_type;
255          int stats_quant;          int stats_quant;
256          int stats_length;          int stats_length;
257          int use_assembler = 0;          int use_assembler = 1;
258    
259          int input_num;          int input_num;
260          int output_num;          int output_num;
# Line 266  Line 281 
281    
282                  if (strcmp("-asm", argv[i]) == 0) {                  if (strcmp("-asm", argv[i]) == 0) {
283                          use_assembler = 1;                          use_assembler = 1;
284                    } else if (strcmp("-noasm", argv[i]) == 0) {
285                            use_assembler = 0;
286                  } else if (strcmp("-w", argv[i]) == 0 && i < argc - 1) {                  } else if (strcmp("-w", argv[i]) == 0 && i < argc - 1) {
287                          i++;                          i++;
288                          XDIM = atoi(argv[i]);                          XDIM = atoi(argv[i]);
# Line 312  Line 329 
329                  } else if (strcmp("-quality", argv[i]) == 0 && i < argc - 1) {                  } else if (strcmp("-quality", argv[i]) == 0 && i < argc - 1) {
330                          i++;                          i++;
331                          ARG_QUALITY = atoi(argv[i]);                          ARG_QUALITY = atoi(argv[i]);
332                    } else if (strcmp("-vhqmode", argv[i]) == 0 && i < argc - 1) {
333                            i++;
334                            ARG_VHQMODE = atoi(argv[i]);
335                  } else if (strcmp("-framerate", argv[i]) == 0 && i < argc - 1) {                  } else if (strcmp("-framerate", argv[i]) == 0 && i < argc - 1) {
336                          i++;                          i++;
337                          ARG_FRAMERATE = (float) atof(argv[i]);                          ARG_FRAMERATE = (float) atof(argv[i]);
# Line 333  Line 353 
353                  } else if (strcmp("-frames", argv[i]) == 0 && i < argc - 1) {                  } else if (strcmp("-frames", argv[i]) == 0 && i < argc - 1) {
354                          i++;                          i++;
355                          ARG_MAXFRAMENR = atoi(argv[i]);                          ARG_MAXFRAMENR = atoi(argv[i]);
356                    } else if (strcmp("-qtype", argv[i]) == 0 && i < argc - 1) {
357                            i++;
358                            ARG_QTYPE = atoi(argv[i]);
359                    } else if (strcmp("-qmatrix", argv[i]) == 0 && i < argc - 1) {
360                            FILE *fp = fopen(argv[++i], "rb");
361                            if (fp == NULL) {
362                                    fprintf(stderr, "Error opening input file %s\n", argv[i]);
363                                    return (-1);
364                            }
365                            fseek(fp, 0, SEEK_END);
366                            if (ftell(fp) != 128) {
367                                    fprintf(stderr, "Unexpected size of input file %s\n", argv[i]);
368                                    return (-1);
369                            }
370    
371                            fseek(fp, 0, SEEK_SET);
372                            fread(qmatrix_intra, 1, 64, fp);
373                            fread(qmatrix_inter, 1, 64, fp);
374    
375                            ARG_QMATRIX = 1;
376                  } else if (strcmp("-save", argv[i]) == 0) {                  } else if (strcmp("-save", argv[i]) == 0) {
377                          ARG_SAVEMPEGSTREAM = 1;                          ARG_SAVEMPEGSTREAM = 1;
378                          ARG_SAVEINDIVIDUAL = 1;                          ARG_SAVEINDIVIDUAL = 1;
# Line 345  Line 385 
385                          ARG_OUTPUTFILE = argv[i];                          ARG_OUTPUTFILE = argv[i];
386                  } else if (strcmp("-vop_debug", argv[i]) == 0) {                  } else if (strcmp("-vop_debug", argv[i]) == 0) {
387                          ARG_VOPDEBUG = 1;                          ARG_VOPDEBUG = 1;
388                  }                  } else if (strcmp("-grey", argv[i]) == 0) {
                 else if (strcmp("-grey", argv[i]) == 0) {  
389                          ARG_GREYSCALE = 1;                          ARG_GREYSCALE = 1;
390                  }                  } else if (strcmp("-bvhq", argv[i]) == 0) {
391                  else if (strcmp("-qpel", argv[i]) == 0) {                          ARG_BVHQ = 1;
392                    } else if (strcmp("-qpel", argv[i]) == 0) {
393                          ARG_QPEL = 1;                          ARG_QPEL = 1;
394                    } else if (strcmp("-turbo", argv[i]) == 0) {
395                            ARG_TURBO = 1;
396                  } else if (strcmp("-gmc", argv[i]) == 0) {                  } else if (strcmp("-gmc", argv[i]) == 0) {
397                          ARG_GMC = 1;                          ARG_GMC = 1;
398                  } else if (strcmp("-interlaced", argv[i]) == 0) {                  } else if (strcmp("-interlaced", argv[i]) == 0) {
# Line 373  Line 415 
415    
416          if (XDIM <= 0 || XDIM >= 4096 || YDIM <= 0 || YDIM >= 4096) {          if (XDIM <= 0 || XDIM >= 4096 || YDIM <= 0 || YDIM >= 4096) {
417                  fprintf(stderr,                  fprintf(stderr,
418                                  "Trying to retreive width and height from PGM header\n");                                  "Trying to retrieve width and height from input header\n");
419                    if (!ARG_INPUTTYPE)
420                  ARG_INPUTTYPE = 1;              /* pgm */                  ARG_INPUTTYPE = 1;              /* pgm */
421          }          }
422    
# Line 388  Line 431 
431                  return (-1);                  return (-1);
432          }          }
433    
434          if (ARG_MAXFRAMENR <= 0) {          if (ARG_MAXFRAMENR == 0) {
435                  fprintf(stderr, "Wrong number of frames\n");                  fprintf(stderr, "Wrong number of frames\n");
436                  return (-1);                  return (-1);
437          }          }
# Line 396  Line 439 
439          if (ARG_INPUTFILE == NULL || strcmp(ARG_INPUTFILE, "stdin") == 0) {          if (ARG_INPUTFILE == NULL || strcmp(ARG_INPUTFILE, "stdin") == 0) {
440                  in_file = stdin;                  in_file = stdin;
441          } else {          } else {
442    #ifdef XVID_AVI_INPUT
443          if (strcmp(ARG_INPUTFILE+(strlen(ARG_INPUTFILE)-3), "avs")==0 ||
444              strcmp(ARG_INPUTFILE+(strlen(ARG_INPUTFILE)-3), "avi")==0 ||
445                      ARG_INPUTTYPE==2)
446          {
447                      AVISTREAMINFO avi_info;
448                      FILE *avi_fp = fopen(ARG_INPUTFILE, "rb");
449                      if (avi_fp == NULL) {
450                              fprintf(stderr, "Couldn't open file '%s'!\n", ARG_INPUTFILE);
451                              return (-1);
452                      }
453                      fclose(avi_fp);
454    
455                      AVIFileInit();
456                      if (AVIStreamOpenFromFile(&avi_stream, ARG_INPUTFILE, streamtypeVIDEO, 0, OF_READ, NULL) != AVIERR_OK) {
457                              fprintf(stderr, "Can't open stream from file '%s'!\n", ARG_INPUTFILE);
458                              AVIFileExit();
459                              return (-1);
460                      }
461    
462                      if(AVIStreamInfo(avi_stream, &avi_info, sizeof(AVISTREAMINFO)) != AVIERR_OK) {
463                              fprintf(stderr, "Can't get stream info from file '%s'!\n", ARG_INPUTFILE);
464                              AVIStreamRelease(avi_stream);
465                              AVIFileExit();
466                              return (-1);
467                      }
468    
469                  if (avi_info.fccHandler != MAKEFOURCC('Y', 'V', '1', '2')) {
470                              fprintf(stderr, "Unsupported input colorspace! Only YV12 is supported!\n");
471                              AVIStreamRelease(avi_stream);
472                          AVIFileExit();
473                  return (-1);
474                      }
475    
476    
477              if (ARG_MAXFRAMENR<0)
478                            ARG_MAXFRAMENR = avi_info.dwLength;
479                      else
480                            ARG_MAXFRAMENR = min(ARG_MAXFRAMENR, avi_info.dwLength);
481    
482                      XDIM = avi_info.rcFrame.right - avi_info.rcFrame.left;
483                      YDIM = avi_info.rcFrame.bottom - avi_info.rcFrame.top;
484                      ARG_FRAMERATE = (float) avi_info.dwRate / (float) avi_info.dwScale;
485    
486                      ARG_INPUTTYPE = 2;
487        }
488        else
489    #endif
490                    {
491                  in_file = fopen(ARG_INPUTFILE, "rb");                  in_file = fopen(ARG_INPUTFILE, "rb");
492                  if (in_file == NULL) {                  if (in_file == NULL) {
493                          fprintf(stderr, "Error opening input file %s\n", ARG_INPUTFILE);                          fprintf(stderr, "Error opening input file %s\n", ARG_INPUTFILE);
494                          return (-1);                          return (-1);
495                  }                  }
496          }          }
497            }
498    
499          if (ARG_INPUTTYPE) {          if (ARG_INPUTTYPE==1) {
500  #ifndef READ_PNM  #ifndef READ_PNM
501                  if (read_pgmheader(in_file)) {                  if (read_pgmheader(in_file)) {
502  #else  #else
# Line 468  Line 560 
560                  char *type;                  char *type;
561                  int sse[3];                  int sse[3];
562    
563                  if (input_num >= ARG_MAXFRAMENR) {                  if (input_num >= ARG_MAXFRAMENR && ARG_MAXFRAMENR > 0) {
564                          result = 1;                          result = 1;
565                  }                  }
566    
567                  if (!result) {                  if (!result) {
568                          if (ARG_INPUTTYPE) {  #ifdef XVID_AVI_INPUT
569                            if (ARG_INPUTTYPE==2) {
570                                    /* read avs/avi data (YUV-format) */
571                                    if(AVIStreamRead(avi_stream, input_num, 1, in_buffer, IMAGE_SIZE(XDIM, YDIM), NULL, NULL ) != AVIERR_OK)
572                                            result = 1;
573                            } else
574    #endif
575                                    if (ARG_INPUTTYPE==1) {
576                                  /* read PGM data (YUV-format) */                                  /* read PGM data (YUV-format) */
577  #ifndef READ_PNM  #ifndef READ_PNM
578                                  result = read_pgmdata(in_file, in_buffer);                                  result = read_pgmdata(in_file, in_buffer);
# Line 574  Line 673 
673                  input_num++;                  input_num++;
674    
675                  /* Read the header if it's pgm stream */                  /* Read the header if it's pgm stream */
676                  if (!result && ARG_INPUTTYPE)                  if (!result && (ARG_INPUTTYPE==1))
677  #ifndef READ_PNM  #ifndef READ_PNM
678                          result = read_pgmheader(in_file);                          result = read_pgmheader(in_file);
679  #else  #else
# Line 617  Line 716 
716    
717    release_all:    release_all:
718    
719    #ifdef XVID_AVI_INPUT
720            if (avi_stream) {
721                    AVIStreamRelease(avi_stream);
722                    AVIFileExit();
723            }
724    #endif
725    
726          if (enc_handle) {          if (enc_handle) {
727                  result = enc_stop();                  result = enc_stop();
728                  if (result)                  if (result)
# Line 660  Line 766 
766          clock_t clk;          clock_t clk;
767    
768          clk = clock();          clk = clock();
769          return (clk * 1000 / CLOCKS_PER_SEC);          return (clk * 1000.0 / CLOCKS_PER_SEC);
770  #endif  #endif
771  }  }
772    
# Line 674  Line 780 
780          fprintf(stderr, "Usage : xvid_stat [OPTIONS]\n\n");          fprintf(stderr, "Usage : xvid_stat [OPTIONS]\n\n");
781          fprintf(stderr, "Input options:\n");          fprintf(stderr, "Input options:\n");
782          fprintf(stderr, " -i      string : input filename (default=stdin)\n");          fprintf(stderr, " -i      string : input filename (default=stdin)\n");
783    #ifdef XVID_AVI_INPUT
784            fprintf(stderr, " -type   integer: input data type (yuv=0, pgm=1, avi/avs=2)\n");
785    #else
786          fprintf(stderr, " -type   integer: input data type (yuv=0, pgm=1)\n");          fprintf(stderr, " -type   integer: input data type (yuv=0, pgm=1)\n");
787    #endif
788          fprintf(stderr, " -w      integer: frame width ([1.2048])\n");          fprintf(stderr, " -w      integer: frame width ([1.2048])\n");
789          fprintf(stderr, " -h      integer: frame height ([1.2048])\n");          fprintf(stderr, " -h      integer: frame height ([1.2048])\n");
790          fprintf(stderr, " -frames integer: number of frames to encode\n");          fprintf(stderr, " -frames integer: number of frames to encode\n");
# Line 700  Line 810 
810      fprintf(stderr, " -max_key_interval integer      : maximum keyframe interval\n");      fprintf(stderr, " -max_key_interval integer      : maximum keyframe interval\n");
811      fprintf(stderr, "\n");      fprintf(stderr, "\n");
812          fprintf(stderr, "Other options\n");          fprintf(stderr, "Other options\n");
813          fprintf(stderr, " -asm            : use assembly optmized code\n");          fprintf(stderr, " -noasm           : do not use assembly optmized code\n");
814            fprintf(stderr, " -turbo           : use turbo presets for higher encoding speed\n");
815          fprintf(stderr, " -quality integer: quality ([0..%d])\n", ME_ELEMENTS - 1);          fprintf(stderr, " -quality integer: quality ([0..%d])\n", ME_ELEMENTS - 1);
816            fprintf(stderr, " -vhqmode integer : level of Rate-Distortion optimizations ([0..4]) (default=0)\n");
817            fprintf(stderr, " -bvhq            : use Rate-Distortion optimizations for B-frames too\n");
818          fprintf(stderr, " -qpel           : use quarter pixel ME\n");          fprintf(stderr, " -qpel           : use quarter pixel ME\n");
819          fprintf(stderr, " -gmc            : use global motion compensation\n");          fprintf(stderr, " -gmc            : use global motion compensation\n");
820            fprintf(stderr, " -qtype   integer : quantization type (H263:0, MPEG4:1) (default=0)\n");
821            fprintf(stderr, " -qmatrix filename: use custom MPEG4 quantization matrix\n");
822          fprintf(stderr, " -interlaced     : use interlaced encoding (this is NOT a deinterlacer!)\n");          fprintf(stderr, " -interlaced     : use interlaced encoding (this is NOT a deinterlacer!)\n");
823          fprintf(stderr, " -packed         : packed mode\n");          fprintf(stderr, " -packed         : packed mode\n");
824          fprintf(stderr, " -closed_gop     : closed GOP mode\n");          fprintf(stderr, " -closed_gop     : closed GOP mode\n");
# Line 1090  Line 1205 
1205          if (image) {          if (image) {
1206                  xvid_enc_frame.input.plane[0] = image;                  xvid_enc_frame.input.plane[0] = image;
1207  #ifndef READ_PNM  #ifndef READ_PNM
1208                    if (ARG_INPUTTYPE==2)
1209                            xvid_enc_frame.input.csp = XVID_CSP_YV12;
1210                    else
1211                  xvid_enc_frame.input.csp = XVID_CSP_I420;                  xvid_enc_frame.input.csp = XVID_CSP_I420;
1212                  xvid_enc_frame.input.stride[0] = XDIM;                  xvid_enc_frame.input.stride[0] = XDIM;
1213  #else  #else
# Line 1104  Line 1222 
1222          xvid_enc_frame.vol_flags = 0;          xvid_enc_frame.vol_flags = 0;
1223          if (ARG_STATS)          if (ARG_STATS)
1224                  xvid_enc_frame.vol_flags |= XVID_VOL_EXTRASTATS;                  xvid_enc_frame.vol_flags |= XVID_VOL_EXTRASTATS;
1225            if (ARG_QTYPE)
1226                    xvid_enc_frame.vol_flags |= XVID_VOL_MPEGQUANT;
1227          if (ARG_QPEL)          if (ARG_QPEL)
1228                  xvid_enc_frame.vol_flags |= XVID_VOL_QUARTERPEL;                  xvid_enc_frame.vol_flags |= XVID_VOL_QUARTERPEL;
1229          if (ARG_GMC)          if (ARG_GMC)
# Line 1139  Line 1259 
1259          if (ARG_QPEL && (xvid_enc_frame.vop_flags & XVID_VOP_INTER4V))          if (ARG_QPEL && (xvid_enc_frame.vop_flags & XVID_VOP_INTER4V))
1260                  xvid_enc_frame.motion |= XVID_ME_QUARTERPELREFINE8;                  xvid_enc_frame.motion |= XVID_ME_QUARTERPELREFINE8;
1261    
1262            if (ARG_TURBO)
1263                    xvid_enc_frame.motion |= XVID_ME_FASTREFINE16 | XVID_ME_FASTREFINE8 |
1264                                                                     XVID_ME_SKIP_DELTASEARCH | XVID_ME_FAST_MODEINTERPOLATE |
1265                                                                     XVID_ME_BFRAME_EARLYSTOP;
1266    
1267            if (ARG_BVHQ)
1268                    xvid_enc_frame.vop_flags |= XVID_VOP_RD_BVOP;
1269    
1270            switch (ARG_VHQMODE) /* this is the same code as for vfw */
1271            {
1272            case 1: /* VHQ_MODE_DECISION */
1273                    xvid_enc_frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
1274                    break;
1275    
1276            case 2: /* VHQ_LIMITED_SEARCH */
1277                    xvid_enc_frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
1278                    xvid_enc_frame.motion |= XVID_ME_HALFPELREFINE16_RD;
1279                    xvid_enc_frame.motion |= XVID_ME_QUARTERPELREFINE16_RD;
1280                    break;
1281    
1282            case 3: /* VHQ_MEDIUM_SEARCH */
1283                    xvid_enc_frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
1284                    xvid_enc_frame.motion |= XVID_ME_HALFPELREFINE16_RD;
1285                    xvid_enc_frame.motion |= XVID_ME_HALFPELREFINE8_RD;
1286                    xvid_enc_frame.motion |= XVID_ME_QUARTERPELREFINE16_RD;
1287                    xvid_enc_frame.motion |= XVID_ME_QUARTERPELREFINE8_RD;
1288                    xvid_enc_frame.motion |= XVID_ME_CHECKPREDICTION_RD;
1289                    break;
1290    
1291            case 4: /* VHQ_WIDE_SEARCH */
1292                    xvid_enc_frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
1293                    xvid_enc_frame.motion |= XVID_ME_HALFPELREFINE16_RD;
1294                    xvid_enc_frame.motion |= XVID_ME_HALFPELREFINE8_RD;
1295                    xvid_enc_frame.motion |= XVID_ME_QUARTERPELREFINE16_RD;
1296                    xvid_enc_frame.motion |= XVID_ME_QUARTERPELREFINE8_RD;
1297                    xvid_enc_frame.motion |= XVID_ME_CHECKPREDICTION_RD;
1298                    xvid_enc_frame.motion |= XVID_ME_EXTSEARCH_RD;
1299                    break;
1300    
1301            default :
1302                    break;
1303            }
1304    
1305            if (ARG_QMATRIX) {
1306                    /* We don't use special matrices */
1307                    xvid_enc_frame.quant_intra_matrix = qmatrix_intra;
1308                    xvid_enc_frame.quant_inter_matrix = qmatrix_inter;
1309            }
1310            else {
1311          /* We don't use special matrices */          /* We don't use special matrices */
1312          xvid_enc_frame.quant_intra_matrix = NULL;          xvid_enc_frame.quant_intra_matrix = NULL;
1313          xvid_enc_frame.quant_inter_matrix = NULL;          xvid_enc_frame.quant_inter_matrix = NULL;
1314            }
1315    
1316          /* Encode the frame */          /* Encode the frame */
1317          ret = xvid_encore(enc_handle, XVID_ENC_ENCODE, &xvid_enc_frame,          ret = xvid_encore(enc_handle, XVID_ENC_ENCODE, &xvid_enc_frame,
Legend:
Removed from v.1622  
changed lines
  Added in v.1676
No admin address has been configured
 ViewVC Help
Powered by ViewVC 1.0.4