19 |
* along with this program; if not, write to the Free Software |
* along with this program; if not, write to the Free Software |
20 |
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA |
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA |
21 |
* |
* |
22 |
* $Id: xvid_encraw.c,v 1.11.2.9 2003-03-20 14:14:52 edgomez Exp $ |
* $Id: xvid_encraw.c,v 1.11.2.19 2003-04-27 19:53:09 chl Exp $ |
23 |
* |
* |
24 |
****************************************************************************/ |
****************************************************************************/ |
25 |
|
|
52 |
****************************************************************************/ |
****************************************************************************/ |
53 |
|
|
54 |
static xvid_motion_t const motion_presets[] = { |
static xvid_motion_t const motion_presets[] = { |
55 |
0, |
0, /* 0 */ |
56 |
PMV_HALFPELREFINE16, |
XVID_ME_HALFPELREFINE16, /* 1 */ |
57 |
PMV_HALFPELREFINE16, |
XVID_ME_HALFPELREFINE16, /* 2 */ |
58 |
PMV_HALFPELREFINE16 | PMV_HALFPELREFINE8, |
XVID_ME_HALFPELREFINE16 | XVID_ME_HALFPELREFINE8, /* 3 */ |
59 |
PMV_HALFPELREFINE16 | PMV_HALFPELREFINE8 | PMV_EXTSEARCH16 | PMV_USESQUARES16, |
XVID_ME_HALFPELREFINE16 | XVID_ME_HALFPELREFINE8, /* 4 */ |
60 |
PMV_HALFPELREFINE16 | PMV_HALFPELREFINE8 | PMV_EXTSEARCH16 | PMV_USESQUARES16 | PMV_CHROMA16 | PMV_CHROMA8, |
XVID_ME_HALFPELREFINE16 | XVID_ME_HALFPELREFINE8 | XVID_ME_EXTSEARCH16 | |
61 |
|
XVID_ME_USESQUARES16, /* 5 */ |
62 |
|
XVID_ME_HALFPELREFINE16 | XVID_ME_HALFPELREFINE8 | XVID_ME_EXTSEARCH16 | |
63 |
|
XVID_ME_USESQUARES16 | XVID_ME_CHROMA16 | XVID_ME_CHROMA8, /* 6 */ |
64 |
}; |
}; |
65 |
|
|
66 |
static xvid_vol_t const vol_presets[] = { |
static xvid_vol_t const vol_presets[] = { |
67 |
XVID_MPEGQUANT, |
XVID_VOL_MPEGQUANT, /* 0 */ |
68 |
0, |
0, /* 1 */ |
69 |
0, |
0, /* 2 */ |
70 |
XVID_QUARTERPEL, |
0, /* 3 */ |
71 |
XVID_QUARTERPEL | XVID_GMC, |
0, /* 4 */ |
72 |
0 |
XVID_VOL_QUARTERPEL | XVID_VOL_GMC, /* 5 */ |
73 |
|
0 /* 6 */ |
74 |
}; |
}; |
75 |
|
|
76 |
static xvid_vop_t const vop_presets[] = { |
static xvid_vop_t const vop_presets[] = { |
77 |
XVID_DYNAMIC_BFRAMES, |
XVID_VOP_DYNAMIC_BFRAMES, /* 0 */ |
78 |
XVID_DYNAMIC_BFRAMES, |
XVID_VOP_DYNAMIC_BFRAMES, /* 1 */ |
79 |
XVID_DYNAMIC_BFRAMES | XVID_HALFPEL, |
XVID_VOP_DYNAMIC_BFRAMES | XVID_VOP_HALFPEL, /* 2 */ |
80 |
XVID_DYNAMIC_BFRAMES | XVID_HALFPEL | XVID_INTER4V, |
XVID_VOP_DYNAMIC_BFRAMES | XVID_VOP_HALFPEL | XVID_VOP_INTER4V, /* 3 */ |
81 |
XVID_DYNAMIC_BFRAMES | XVID_HALFPEL | XVID_INTER4V | XVID_HQACPRED, |
XVID_VOP_DYNAMIC_BFRAMES | XVID_VOP_HALFPEL | XVID_VOP_INTER4V | XVID_VOP_TRELLISQUANT, /* 4 */ |
82 |
XVID_DYNAMIC_BFRAMES | XVID_HALFPEL | XVID_HQACPRED | XVID_MODEDECISION_BITS |
XVID_VOP_DYNAMIC_BFRAMES | XVID_VOP_HALFPEL | XVID_VOP_INTER4V | XVID_VOP_HQACPRED, /* 5 */ |
83 |
|
XVID_VOP_DYNAMIC_BFRAMES | XVID_VOP_HALFPEL | XVID_VOP_HQACPRED | /* 6 */ |
84 |
|
XVID_VOP_MODEDECISION_BITS |
85 |
}; |
}; |
86 |
|
|
87 |
/***************************************************************************** |
/***************************************************************************** |
94 |
static int ARG_STATS = 0; |
static int ARG_STATS = 0; |
95 |
static int ARG_DUMP = 0; |
static int ARG_DUMP = 0; |
96 |
static int ARG_LUMIMASKING = 0; |
static int ARG_LUMIMASKING = 0; |
97 |
static int ARG_BITRATE = 900; |
static int ARG_BITRATE = 0; |
98 |
static int ARG_QUANTI = 0; |
static char *ARG_PASS1 = 0; |
99 |
static int ARG_QUALITY = 5; |
static char *ARG_PASS2 = 0; |
100 |
|
static int ARG_PASS2_BITRATE = 0; |
101 |
|
static float ARG_QUANTI = 0.0f; |
102 |
|
static int ARG_QUALITY = 3; |
103 |
static float ARG_FRAMERATE = 25.00f; |
static float ARG_FRAMERATE = 25.00f; |
104 |
static int ARG_MAXFRAMENR = ABS_MAXFRAMENR; |
static int ARG_MAXFRAMENR = ABS_MAXFRAMENR; |
105 |
static char *ARG_INPUTFILE = NULL; |
static char *ARG_INPUTFILE = NULL; |
177 |
|
|
178 |
double enctime; |
double enctime; |
179 |
double totalenctime = 0.; |
double totalenctime = 0.; |
180 |
|
float totalPSNR[3] = {0., 0., 0.}; |
181 |
|
|
182 |
int totalsize; |
int totalsize; |
183 |
int result; |
int result; |
216 |
} else if (strcmp("-bitrate", argv[i]) == 0 && i < argc - 1) { |
} else if (strcmp("-bitrate", argv[i]) == 0 && i < argc - 1) { |
217 |
i++; |
i++; |
218 |
ARG_BITRATE = atoi(argv[i]); |
ARG_BITRATE = atoi(argv[i]); |
219 |
|
} else if (strcmp("-pass1", argv[i]) == 0 && i < argc - 1) { |
220 |
|
i++; |
221 |
|
ARG_PASS1 = argv[i]; |
222 |
|
} else if (strcmp("-pass2", argv[i]) == 0 && i < argc - 2) { |
223 |
|
i++; |
224 |
|
ARG_PASS2 = argv[i]; |
225 |
|
i++; |
226 |
|
ARG_PASS2_BITRATE = atoi(argv[i]); |
227 |
} else if (strcmp("-max_bframes", argv[i]) == 0 && i < argc - 1) { |
} else if (strcmp("-max_bframes", argv[i]) == 0 && i < argc - 1) { |
228 |
i++; |
i++; |
229 |
ARG_MAXBFRAMES = atoi(argv[i]); |
ARG_MAXBFRAMES = atoi(argv[i]); |
258 |
ARG_MAXFRAMENR = atoi(argv[i]); |
ARG_MAXFRAMENR = atoi(argv[i]); |
259 |
} else if (strcmp("-quant", argv[i]) == 0 && i < argc - 1) { |
} else if (strcmp("-quant", argv[i]) == 0 && i < argc - 1) { |
260 |
i++; |
i++; |
261 |
ARG_QUANTI = atoi(argv[i]); |
ARG_QUANTI = (float) atof(argv[i]); |
262 |
} else if (strcmp("-save", argv[i]) == 0) { |
} else if (strcmp("-save", argv[i]) == 0) { |
263 |
ARG_SAVEMPEGSTREAM = 1; |
ARG_SAVEMPEGSTREAM = 1; |
264 |
} else if (strcmp("-debug", argv[i]) == 0) { |
} else if (strcmp("-debug", argv[i]) == 0) { |
286 |
ARG_INPUTTYPE = 1; /* pgm */ |
ARG_INPUTTYPE = 1; /* pgm */ |
287 |
} |
} |
288 |
|
|
289 |
if (ARG_QUALITY < 0 || ARG_QUALITY > 5) { |
if (ARG_QUALITY < 0 || ARG_QUALITY > 6) { |
290 |
fprintf(stderr, "Wrong Quality\n"); |
fprintf(stderr, "Wrong Quality\n"); |
291 |
return (-1); |
return (-1); |
292 |
} |
} |
293 |
|
|
|
if (ARG_BITRATE <= 0 && ARG_QUANTI == 0) { |
|
|
fprintf(stderr, "Wrong Bitrate\n"); |
|
|
return (-1); |
|
|
} |
|
|
|
|
294 |
if (ARG_FRAMERATE <= 0) { |
if (ARG_FRAMERATE <= 0) { |
295 |
fprintf(stderr, "Wrong Framerate %s \n", argv[5]); |
fprintf(stderr, "Wrong Framerate %s \n", argv[5]); |
296 |
return (-1); |
return (-1); |
398 |
|
|
399 |
/* Write the Frame statistics */ |
/* Write the Frame statistics */ |
400 |
|
|
401 |
printf("%5d: key=%i, time(ms)=%6.1f, length=%7d", |
printf("%5d: key=%i, time= %6.0f, length= %7d", !result ? input_num : -1, |
402 |
!result ? input_num : -1, |
key, (float) enctime, (int) m4v_size); |
|
key, |
|
|
(float) enctime, |
|
|
(int) m4v_size); |
|
403 |
|
|
404 |
if (stats_type > 0) { /* !XVID_TYPE_NOTHING */ |
if (stats_type > 0) { /* !XVID_TYPE_NOTHING */ |
405 |
|
|
428 |
|
|
429 |
if (ARG_STATS) { |
if (ARG_STATS) { |
430 |
printf(", psnr y = %2.2f, psnr u = %2.2f, psnr v = %2.2f", |
printf(", psnr y = %2.2f, psnr u = %2.2f, psnr v = %2.2f", |
431 |
SSE2PSNR(sse[0], XDIM, YDIM), |
SSE2PSNR(sse[0], XDIM, YDIM), SSE2PSNR(sse[1], XDIM / 2, |
432 |
SSE2PSNR(sse[1], XDIM/2, YDIM/2), |
YDIM / 2), |
433 |
SSE2PSNR(sse[2], XDIM/2, YDIM/2)); |
SSE2PSNR(sse[2], XDIM/2, YDIM/2)); |
|
} |
|
434 |
|
|
435 |
|
totalPSNR[0] += SSE2PSNR(sse[0], XDIM, YDIM); |
436 |
|
totalPSNR[1] += SSE2PSNR(sse[1], XDIM/2, YDIM/2); |
437 |
|
totalPSNR[2] += SSE2PSNR(sse[2], XDIM/2, YDIM/2); |
438 |
} |
} |
439 |
|
|
440 |
|
} |
441 |
#undef SSE2PSNR |
#undef SSE2PSNR |
442 |
|
|
443 |
printf("\n"); |
printf("\n"); |
491 |
if (input_num > 0) { |
if (input_num > 0) { |
492 |
totalsize /= input_num; |
totalsize /= input_num; |
493 |
totalenctime /= input_num; |
totalenctime /= input_num; |
494 |
|
totalPSNR[0] /= input_num; |
495 |
|
totalPSNR[1] /= input_num; |
496 |
|
totalPSNR[2] /= input_num; |
497 |
} else { |
} else { |
498 |
totalsize = -1; |
totalsize = -1; |
499 |
totalenctime = -1; |
totalenctime = -1; |
500 |
} |
} |
501 |
|
|
502 |
printf("Avg: enctime(ms) =%7.2f, fps =%7.2f, length(bytes) = %7d\n", |
printf("Avg: enctime(ms) =%7.2f, fps =%7.2f, length(bytes) = %7d, ", |
503 |
totalenctime, 1000 / totalenctime, (int) totalsize); |
totalenctime, 1000 / totalenctime, (int) totalsize); |
504 |
|
if (ARG_STATS) { |
505 |
|
printf("psnr y = %2.2f, psnr u = %2.2f, psnr v = %2.2f", |
506 |
|
totalPSNR[0],totalPSNR[1],totalPSNR[2]); |
507 |
|
} |
508 |
|
printf("\n"); |
509 |
|
|
510 |
|
|
511 |
/***************************************************************************** |
/***************************************************************************** |
581 |
fprintf(stderr, " -save : save mpeg4 raw stream\n"); |
fprintf(stderr, " -save : save mpeg4 raw stream\n"); |
582 |
fprintf(stderr, " -o string: output filename\n"); |
fprintf(stderr, " -o string: output filename\n"); |
583 |
fprintf(stderr, "\n"); |
fprintf(stderr, "\n"); |
584 |
fprintf(stderr, "Bitrate options:\n"); |
fprintf(stderr, "BFrames options:\n"); |
|
fprintf(stderr, " -bitrate integer: target bitrate (>0 | default=900kbit)\n"); |
|
|
fprintf(stderr, " -quant integer: fixed quantizer (disables -b setting)\n"); |
|
585 |
fprintf(stderr, " -max_bframes integer: max bframes (default=0)\n"); |
fprintf(stderr, " -max_bframes integer: max bframes (default=0)\n"); |
586 |
fprintf(stderr, " -bquant_ratio integer: bframe quantizer ratio (default=150)\n"); |
fprintf(stderr, " -bquant_ratio integer: bframe quantizer ratio (default=150)\n"); |
587 |
fprintf(stderr, " -bquant_offset integer: bframe quantizer offset (default=100)\n"); |
fprintf(stderr, " -bquant_offset integer: bframe quantizer offset (default=100)\n"); |
588 |
fprintf(stderr, " -framerate float : target framerate (>0)\n"); |
fprintf(stderr, "\n"); |
589 |
|
fprintf(stderr, "Rate control options:\n"); |
590 |
|
fprintf(stderr, " -framerate float : target framerate (>0 | default=25.0)\n"); |
591 |
|
fprintf(stderr, " -bitrate integer : bitrate -- for CBR/VBR pass2\n"); |
592 |
|
fprintf(stderr, " -quant float : quantizer -- for \"Fixed\" quantizer RC\n"); |
593 |
|
fprintf(stderr, " -pass1 filename : output stats filename\n"); |
594 |
|
fprintf(stderr, " -pass2 filename bitrate : input stats filename, target bitrate\n"); |
595 |
fprintf(stderr, "\n"); |
fprintf(stderr, "\n"); |
596 |
fprintf(stderr, "Other options\n"); |
fprintf(stderr, "Other options\n"); |
597 |
fprintf(stderr, " -asm : use assembly optmized code\n"); |
fprintf(stderr, " -asm : use assembly optmized code\n"); |
599 |
fprintf(stderr, " -packed : packed mode\n"); |
fprintf(stderr, " -packed : packed mode\n"); |
600 |
fprintf(stderr, " -lumimasking : use lumimasking algorithm\n"); |
fprintf(stderr, " -lumimasking : use lumimasking algorithm\n"); |
601 |
fprintf(stderr, " -stats : print stats about encoded frames\n"); |
fprintf(stderr, " -stats : print stats about encoded frames\n"); |
602 |
fprintf(stderr, " -debug : print all MB quantizers\n"); |
fprintf(stderr, " -debug : print all MB dquants\n"); |
603 |
fprintf(stderr, " -help : prints this help message\n"); |
fprintf(stderr, " -help : prints this help message\n"); |
604 |
} |
} |
605 |
|
|
728 |
|
|
729 |
#define FRAMERATE_INCR 1001 |
#define FRAMERATE_INCR 1001 |
730 |
|
|
731 |
|
|
732 |
/* Initialize encoder for first use, pass all needed parameters to the codec */ |
/* Initialize encoder for first use, pass all needed parameters to the codec */ |
733 |
static int |
static int |
734 |
enc_init(int use_assembler) |
enc_init(int use_assembler) |
735 |
{ |
{ |
736 |
int xerr; |
int xerr; |
737 |
|
xvid_plugin_cbr_t cbr; |
738 |
xvid_enc_plugin_t plugins[2]; |
xvid_plugin_2pass1_t rc2pass1; |
739 |
|
xvid_plugin_2pass2_t rc2pass2; |
740 |
|
xvid_plugin_fixed_t rcfixed; |
741 |
|
xvid_enc_plugin_t plugins[7]; |
742 |
xvid_gbl_init_t xvid_gbl_init; |
xvid_gbl_init_t xvid_gbl_init; |
743 |
xvid_enc_create_t xvid_enc_create; |
xvid_enc_create_t xvid_enc_create; |
744 |
|
|
783 |
xvid_enc_create.plugins = plugins; |
xvid_enc_create.plugins = plugins; |
784 |
xvid_enc_create.num_plugins = 0; |
xvid_enc_create.num_plugins = 0; |
785 |
|
|
786 |
|
if (ARG_BITRATE) { |
787 |
|
memset(&cbr, 0, sizeof(xvid_plugin_cbr_t)); |
788 |
|
cbr.version = XVID_VERSION; |
789 |
|
cbr.bitrate = ARG_BITRATE; |
790 |
|
|
791 |
|
plugins[xvid_enc_create.num_plugins].func = xvid_plugin_cbr; |
792 |
|
plugins[xvid_enc_create.num_plugins].param = &cbr; |
793 |
|
xvid_enc_create.num_plugins++; |
794 |
|
} |
795 |
|
|
796 |
|
if (ARG_QUANTI) { |
797 |
|
memset(&rcfixed, 0, sizeof(xvid_plugin_fixed_t)); |
798 |
|
rcfixed.version = XVID_VERSION; |
799 |
|
/* We will use a 1/10 precision, just to make sure it works */ |
800 |
|
rcfixed.quant_base = 10; |
801 |
|
rcfixed.quant_increment = (int) (ARG_QUANTI * 10); |
802 |
|
|
803 |
|
plugins[xvid_enc_create.num_plugins].func = xvid_plugin_fixed; |
804 |
|
plugins[xvid_enc_create.num_plugins].param = &rcfixed; |
805 |
|
xvid_enc_create.num_plugins++; |
806 |
|
} |
807 |
|
|
808 |
|
if (ARG_PASS2) { |
809 |
|
memset(&rc2pass2, 0, sizeof(xvid_plugin_2pass2_t)); |
810 |
|
rc2pass2.version = XVID_VERSION; |
811 |
|
rc2pass2.filename = ARG_PASS2; |
812 |
|
rc2pass2.bitrate = ARG_PASS2_BITRATE; |
813 |
|
|
814 |
|
plugins[xvid_enc_create.num_plugins].func = xvid_plugin_2pass2; |
815 |
|
plugins[xvid_enc_create.num_plugins].param = &rc2pass2; |
816 |
|
xvid_enc_create.num_plugins++; |
817 |
|
} |
818 |
|
|
819 |
|
if (ARG_PASS1) { |
820 |
|
memset(&rc2pass1, 0, sizeof(xvid_plugin_2pass1_t)); |
821 |
|
rc2pass1.version = XVID_VERSION; |
822 |
|
rc2pass1.filename = ARG_PASS1; |
823 |
|
|
824 |
|
plugins[xvid_enc_create.num_plugins].func = xvid_plugin_2pass1; |
825 |
|
plugins[xvid_enc_create.num_plugins].param = &rc2pass1; |
826 |
|
xvid_enc_create.num_plugins++; |
827 |
|
} |
828 |
|
|
829 |
if (ARG_LUMIMASKING) { |
if (ARG_LUMIMASKING) { |
830 |
plugins[xvid_enc_create.num_plugins].func = xvid_plugin_lumimasking; |
plugins[xvid_enc_create.num_plugins].func = xvid_plugin_lumimasking; |
831 |
plugins[xvid_enc_create.num_plugins].param = NULL; |
plugins[xvid_enc_create.num_plugins].param = NULL; |
869 |
|
|
870 |
/* Global encoder options */ |
/* Global encoder options */ |
871 |
xvid_enc_create.global = 0; |
xvid_enc_create.global = 0; |
872 |
if (ARG_PACKED) xvid_enc_create.global |=XVID_PACKED; |
|
873 |
if (ARG_STATS) xvid_enc_create.global |=XVID_EXTRASTATS_ENABLE; |
if (ARG_PACKED) |
874 |
|
xvid_enc_create.global |=XVID_GLOBAL_PACKED; |
875 |
|
|
876 |
|
if (ARG_STATS) |
877 |
|
xvid_enc_create.global |=XVID_GLOBAL_EXTRASTATS_ENABLE; |
878 |
|
|
879 |
/* I use a small value here, since will not encode whole movies, but short clips */ |
/* I use a small value here, since will not encode whole movies, but short clips */ |
880 |
xerr = xvid_encore(NULL, XVID_ENC_CREATE, &xvid_enc_create, NULL); |
xerr = xvid_encore(NULL, XVID_ENC_CREATE, &xvid_enc_create, NULL); |
933 |
/* Set up core's general features */ |
/* Set up core's general features */ |
934 |
xvid_enc_frame.vol_flags = vol_presets[ARG_QUALITY]; |
xvid_enc_frame.vol_flags = vol_presets[ARG_QUALITY]; |
935 |
if (ARG_STATS) |
if (ARG_STATS) |
936 |
xvid_enc_frame.vol_flags |= XVID_EXTRASTATS; |
xvid_enc_frame.vol_flags |= XVID_VOL_EXTRASTATS; |
937 |
|
|
938 |
/* Set up core's general features */ |
/* Set up core's general features */ |
939 |
xvid_enc_frame.vop_flags = vop_presets[ARG_QUALITY]; |
xvid_enc_frame.vop_flags = vop_presets[ARG_QUALITY]; |
941 |
/* Frame type -- let core decide for us */ |
/* Frame type -- let core decide for us */ |
942 |
xvid_enc_frame.type = XVID_TYPE_AUTO; |
xvid_enc_frame.type = XVID_TYPE_AUTO; |
943 |
|
|
944 |
/* Force the right quantizer */ |
/* Force the right quantizer -- It is internally managed by RC plugins */ |
945 |
xvid_enc_frame.quant = ARG_QUANTI; |
xvid_enc_frame.quant = 0; |
946 |
|
|
947 |
/* Set up motion estimation flags */ |
/* Set up motion estimation flags */ |
948 |
xvid_enc_frame.motion = motion_presets[ARG_QUALITY]; |
xvid_enc_frame.motion = motion_presets[ARG_QUALITY]; |