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.10 2003-03-20 17:59:07 edgomez Exp $ |
* $Id: xvid_encraw.c,v 1.11.2.18 2003-04-19 11:26:47 chl Exp $ |
23 |
* |
* |
24 |
****************************************************************************/ |
****************************************************************************/ |
25 |
|
|
53 |
|
|
54 |
static xvid_motion_t const motion_presets[] = { |
static xvid_motion_t const motion_presets[] = { |
55 |
0, |
0, |
56 |
PMV_HALFPELREFINE16, |
XVID_ME_HALFPELREFINE16, |
57 |
PMV_HALFPELREFINE16, |
XVID_ME_HALFPELREFINE16, |
58 |
PMV_HALFPELREFINE16 | PMV_HALFPELREFINE8, |
XVID_ME_HALFPELREFINE16 | XVID_ME_HALFPELREFINE8, |
59 |
PMV_HALFPELREFINE16 | PMV_HALFPELREFINE8 | PMV_EXTSEARCH16 | PMV_USESQUARES16, |
XVID_ME_HALFPELREFINE16 | XVID_ME_HALFPELREFINE8 | XVID_ME_EXTSEARCH16 | |
60 |
PMV_HALFPELREFINE16 | PMV_HALFPELREFINE8 | PMV_EXTSEARCH16 | PMV_USESQUARES16 | PMV_CHROMA16 | PMV_CHROMA8, |
XVID_ME_USESQUARES16, |
61 |
|
XVID_ME_HALFPELREFINE16 | XVID_ME_HALFPELREFINE8 | XVID_ME_EXTSEARCH16 | |
62 |
|
XVID_ME_USESQUARES16 | XVID_ME_CHROMA16 | XVID_ME_CHROMA8, |
63 |
}; |
}; |
64 |
|
|
65 |
static xvid_vol_t const vol_presets[] = { |
static xvid_vol_t const vol_presets[] = { |
66 |
XVID_MPEGQUANT, |
XVID_VOL_MPEGQUANT, |
67 |
0, |
0, |
68 |
0, |
0, |
69 |
XVID_QUARTERPEL, |
XVID_VOL_QUARTERPEL, |
70 |
XVID_QUARTERPEL | XVID_GMC, |
XVID_VOL_QUARTERPEL | XVID_VOL_GMC, |
71 |
0 |
0 |
72 |
}; |
}; |
73 |
|
|
74 |
static xvid_vop_t const vop_presets[] = { |
static xvid_vop_t const vop_presets[] = { |
75 |
XVID_DYNAMIC_BFRAMES, |
XVID_VOP_DYNAMIC_BFRAMES, |
76 |
XVID_DYNAMIC_BFRAMES, |
XVID_VOP_DYNAMIC_BFRAMES, |
77 |
XVID_DYNAMIC_BFRAMES | XVID_HALFPEL, |
XVID_VOP_DYNAMIC_BFRAMES | XVID_VOP_HALFPEL, |
78 |
XVID_DYNAMIC_BFRAMES | XVID_HALFPEL | XVID_INTER4V, |
XVID_VOP_DYNAMIC_BFRAMES | XVID_VOP_HALFPEL | XVID_VOP_INTER4V, |
79 |
XVID_DYNAMIC_BFRAMES | XVID_HALFPEL | XVID_INTER4V | XVID_HQACPRED, |
XVID_VOP_DYNAMIC_BFRAMES | XVID_VOP_HALFPEL | XVID_VOP_INTER4V | XVID_VOP_HQACPRED, |
80 |
XVID_DYNAMIC_BFRAMES | XVID_HALFPEL | XVID_HQACPRED | XVID_MODEDECISION_BITS |
XVID_VOP_DYNAMIC_BFRAMES | XVID_VOP_HALFPEL | XVID_VOP_HQACPRED | |
81 |
|
XVID_VOP_MODEDECISION_BITS |
82 |
}; |
}; |
83 |
|
|
84 |
/***************************************************************************** |
/***************************************************************************** |
91 |
static int ARG_STATS = 0; |
static int ARG_STATS = 0; |
92 |
static int ARG_DUMP = 0; |
static int ARG_DUMP = 0; |
93 |
static int ARG_LUMIMASKING = 0; |
static int ARG_LUMIMASKING = 0; |
94 |
static int ARG_BITRATE = 900; |
static int ARG_BITRATE = 0; |
95 |
static int ARG_QUANTI = 0; |
static char *ARG_PASS1 = 0; |
96 |
|
static char *ARG_PASS2 = 0; |
97 |
|
static int ARG_PASS2_BITRATE = 0; |
98 |
|
static float ARG_QUANTI = 0.0f; |
99 |
static int ARG_QUALITY = 5; |
static int ARG_QUALITY = 5; |
100 |
static float ARG_FRAMERATE = 25.00f; |
static float ARG_FRAMERATE = 25.00f; |
101 |
static int ARG_MAXFRAMENR = ABS_MAXFRAMENR; |
static int ARG_MAXFRAMENR = ABS_MAXFRAMENR; |
174 |
|
|
175 |
double enctime; |
double enctime; |
176 |
double totalenctime = 0.; |
double totalenctime = 0.; |
177 |
|
float totalPSNR[3] = {0., 0., 0.}; |
178 |
|
|
179 |
int totalsize; |
int totalsize; |
180 |
int result; |
int result; |
213 |
} else if (strcmp("-bitrate", argv[i]) == 0 && i < argc - 1) { |
} else if (strcmp("-bitrate", argv[i]) == 0 && i < argc - 1) { |
214 |
i++; |
i++; |
215 |
ARG_BITRATE = atoi(argv[i]); |
ARG_BITRATE = atoi(argv[i]); |
216 |
|
} else if (strcmp("-pass1", argv[i]) == 0 && i < argc - 1) { |
217 |
|
i++; |
218 |
|
ARG_PASS1 = argv[i]; |
219 |
|
} else if (strcmp("-pass2", argv[i]) == 0 && i < argc - 2) { |
220 |
|
i++; |
221 |
|
ARG_PASS2 = argv[i]; |
222 |
|
i++; |
223 |
|
ARG_PASS2_BITRATE = atoi(argv[i]); |
224 |
} else if (strcmp("-max_bframes", argv[i]) == 0 && i < argc - 1) { |
} else if (strcmp("-max_bframes", argv[i]) == 0 && i < argc - 1) { |
225 |
i++; |
i++; |
226 |
ARG_MAXBFRAMES = atoi(argv[i]); |
ARG_MAXBFRAMES = atoi(argv[i]); |
255 |
ARG_MAXFRAMENR = atoi(argv[i]); |
ARG_MAXFRAMENR = atoi(argv[i]); |
256 |
} else if (strcmp("-quant", argv[i]) == 0 && i < argc - 1) { |
} else if (strcmp("-quant", argv[i]) == 0 && i < argc - 1) { |
257 |
i++; |
i++; |
258 |
ARG_QUANTI = atoi(argv[i]); |
ARG_QUANTI = (float) atof(argv[i]); |
259 |
} else if (strcmp("-save", argv[i]) == 0) { |
} else if (strcmp("-save", argv[i]) == 0) { |
260 |
ARG_SAVEMPEGSTREAM = 1; |
ARG_SAVEMPEGSTREAM = 1; |
261 |
} else if (strcmp("-debug", argv[i]) == 0) { |
} else if (strcmp("-debug", argv[i]) == 0) { |
288 |
return (-1); |
return (-1); |
289 |
} |
} |
290 |
|
|
|
if (ARG_BITRATE <= 0 && ARG_QUANTI == 0) { |
|
|
fprintf(stderr, "Wrong Bitrate\n"); |
|
|
return (-1); |
|
|
} |
|
|
|
|
291 |
if (ARG_FRAMERATE <= 0) { |
if (ARG_FRAMERATE <= 0) { |
292 |
fprintf(stderr, "Wrong Framerate %s \n", argv[5]); |
fprintf(stderr, "Wrong Framerate %s \n", argv[5]); |
293 |
return (-1); |
return (-1); |
395 |
|
|
396 |
/* Write the Frame statistics */ |
/* Write the Frame statistics */ |
397 |
|
|
398 |
printf("%5d: key=%i, time(ms)=%6.1f, length=%7d", |
printf("%5d: key=%i, time= %6.0f, length= %7d", !result ? input_num : -1, |
399 |
!result ? input_num : -1, |
key, (float) enctime, (int) m4v_size); |
|
key, |
|
|
(float) enctime, |
|
|
(int) m4v_size); |
|
400 |
|
|
401 |
if (stats_type > 0) { /* !XVID_TYPE_NOTHING */ |
if (stats_type > 0) { /* !XVID_TYPE_NOTHING */ |
402 |
|
|
425 |
|
|
426 |
if (ARG_STATS) { |
if (ARG_STATS) { |
427 |
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", |
428 |
SSE2PSNR(sse[0], XDIM, YDIM), |
SSE2PSNR(sse[0], XDIM, YDIM), SSE2PSNR(sse[1], XDIM / 2, |
429 |
SSE2PSNR(sse[1], XDIM/2, YDIM/2), |
YDIM / 2), |
430 |
SSE2PSNR(sse[2], XDIM/2, YDIM/2)); |
SSE2PSNR(sse[2], XDIM/2, YDIM/2)); |
|
} |
|
431 |
|
|
432 |
|
totalPSNR[0] += SSE2PSNR(sse[0], XDIM, YDIM); |
433 |
|
totalPSNR[1] += SSE2PSNR(sse[1], XDIM/2, YDIM/2); |
434 |
|
totalPSNR[2] += SSE2PSNR(sse[2], XDIM/2, YDIM/2); |
435 |
} |
} |
436 |
|
|
437 |
|
} |
438 |
#undef SSE2PSNR |
#undef SSE2PSNR |
439 |
|
|
440 |
printf("\n"); |
printf("\n"); |
488 |
if (input_num > 0) { |
if (input_num > 0) { |
489 |
totalsize /= input_num; |
totalsize /= input_num; |
490 |
totalenctime /= input_num; |
totalenctime /= input_num; |
491 |
|
totalPSNR[0] /= input_num; |
492 |
|
totalPSNR[1] /= input_num; |
493 |
|
totalPSNR[2] /= input_num; |
494 |
} else { |
} else { |
495 |
totalsize = -1; |
totalsize = -1; |
496 |
totalenctime = -1; |
totalenctime = -1; |
497 |
} |
} |
498 |
|
|
499 |
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, ", |
500 |
totalenctime, 1000 / totalenctime, (int) totalsize); |
totalenctime, 1000 / totalenctime, (int) totalsize); |
501 |
|
if (ARG_STATS) { |
502 |
|
printf("psnr y = %2.2f, psnr u = %2.2f, psnr v = %2.2f", |
503 |
|
totalPSNR[0],totalPSNR[1],totalPSNR[2]); |
504 |
|
} |
505 |
|
printf("\n"); |
506 |
|
|
507 |
|
|
508 |
/***************************************************************************** |
/***************************************************************************** |
578 |
fprintf(stderr, " -save : save mpeg4 raw stream\n"); |
fprintf(stderr, " -save : save mpeg4 raw stream\n"); |
579 |
fprintf(stderr, " -o string: output filename\n"); |
fprintf(stderr, " -o string: output filename\n"); |
580 |
fprintf(stderr, "\n"); |
fprintf(stderr, "\n"); |
581 |
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"); |
|
582 |
fprintf(stderr, " -max_bframes integer: max bframes (default=0)\n"); |
fprintf(stderr, " -max_bframes integer: max bframes (default=0)\n"); |
583 |
fprintf(stderr, " -bquant_ratio integer: bframe quantizer ratio (default=150)\n"); |
fprintf(stderr, " -bquant_ratio integer: bframe quantizer ratio (default=150)\n"); |
584 |
fprintf(stderr, " -bquant_offset integer: bframe quantizer offset (default=100)\n"); |
fprintf(stderr, " -bquant_offset integer: bframe quantizer offset (default=100)\n"); |
585 |
fprintf(stderr, " -framerate float : target framerate (>0)\n"); |
fprintf(stderr, "\n"); |
586 |
|
fprintf(stderr, "Rate control options:\n"); |
587 |
|
fprintf(stderr, " -framerate float : target framerate (>0 | default=25.0)\n"); |
588 |
|
fprintf(stderr, " -bitrate integer : bitrate -- for CBR/VBR pass2\n"); |
589 |
|
fprintf(stderr, " -quant float : quantizer -- for \"Fixed\" quantizer RC\n"); |
590 |
|
fprintf(stderr, " -pass1 filename : output stats filename\n"); |
591 |
|
fprintf(stderr, " -pass2 filename bitrate : input stats filename, target bitrate\n"); |
592 |
fprintf(stderr, "\n"); |
fprintf(stderr, "\n"); |
593 |
fprintf(stderr, "Other options\n"); |
fprintf(stderr, "Other options\n"); |
594 |
fprintf(stderr, " -asm : use assembly optmized code\n"); |
fprintf(stderr, " -asm : use assembly optmized code\n"); |
596 |
fprintf(stderr, " -packed : packed mode\n"); |
fprintf(stderr, " -packed : packed mode\n"); |
597 |
fprintf(stderr, " -lumimasking : use lumimasking algorithm\n"); |
fprintf(stderr, " -lumimasking : use lumimasking algorithm\n"); |
598 |
fprintf(stderr, " -stats : print stats about encoded frames\n"); |
fprintf(stderr, " -stats : print stats about encoded frames\n"); |
599 |
fprintf(stderr, " -debug : print all MB quantizers\n"); |
fprintf(stderr, " -debug : print all MB dquants\n"); |
600 |
fprintf(stderr, " -help : prints this help message\n"); |
fprintf(stderr, " -help : prints this help message\n"); |
601 |
} |
} |
602 |
|
|
725 |
|
|
726 |
#define FRAMERATE_INCR 1001 |
#define FRAMERATE_INCR 1001 |
727 |
|
|
728 |
|
|
729 |
/* Initialize encoder for first use, pass all needed parameters to the codec */ |
/* Initialize encoder for first use, pass all needed parameters to the codec */ |
730 |
static int |
static int |
731 |
enc_init(int use_assembler) |
enc_init(int use_assembler) |
732 |
{ |
{ |
733 |
int xerr; |
int xerr; |
734 |
|
xvid_plugin_cbr_t cbr; |
735 |
xvid_enc_plugin_t plugins[3]; |
xvid_plugin_2pass1_t rc2pass1; |
736 |
|
xvid_plugin_2pass2_t rc2pass2; |
737 |
|
xvid_plugin_fixed_t rcfixed; |
738 |
|
xvid_enc_plugin_t plugins[7]; |
739 |
xvid_gbl_init_t xvid_gbl_init; |
xvid_gbl_init_t xvid_gbl_init; |
740 |
xvid_enc_create_t xvid_enc_create; |
xvid_enc_create_t xvid_enc_create; |
741 |
|
|
780 |
xvid_enc_create.plugins = plugins; |
xvid_enc_create.plugins = plugins; |
781 |
xvid_enc_create.num_plugins = 0; |
xvid_enc_create.num_plugins = 0; |
782 |
|
|
783 |
|
if (ARG_BITRATE) { |
784 |
|
memset(&cbr, 0, sizeof(xvid_plugin_cbr_t)); |
785 |
|
cbr.version = XVID_VERSION; |
786 |
|
cbr.bitrate = ARG_BITRATE; |
787 |
|
|
788 |
|
plugins[xvid_enc_create.num_plugins].func = xvid_plugin_cbr; |
789 |
|
plugins[xvid_enc_create.num_plugins].param = &cbr; |
790 |
|
xvid_enc_create.num_plugins++; |
791 |
|
} |
792 |
|
|
793 |
|
if (ARG_QUANTI) { |
794 |
|
memset(&rcfixed, 0, sizeof(xvid_plugin_fixed_t)); |
795 |
|
rcfixed.version = XVID_VERSION; |
796 |
|
/* We will use a 1/10 precision, just to make sure it works */ |
797 |
|
rcfixed.quant_base = 10; |
798 |
|
rcfixed.quant_increment = (int) (ARG_QUANTI * 10); |
799 |
|
|
800 |
|
plugins[xvid_enc_create.num_plugins].func = xvid_plugin_fixed; |
801 |
|
plugins[xvid_enc_create.num_plugins].param = &rcfixed; |
802 |
|
xvid_enc_create.num_plugins++; |
803 |
|
} |
804 |
|
|
805 |
|
if (ARG_PASS2) { |
806 |
|
memset(&rc2pass2, 0, sizeof(xvid_plugin_2pass2_t)); |
807 |
|
rc2pass2.version = XVID_VERSION; |
808 |
|
rc2pass2.filename = ARG_PASS2; |
809 |
|
rc2pass2.bitrate = ARG_PASS2_BITRATE; |
810 |
|
|
811 |
|
plugins[xvid_enc_create.num_plugins].func = xvid_plugin_2pass2; |
812 |
|
plugins[xvid_enc_create.num_plugins].param = &rc2pass2; |
813 |
|
xvid_enc_create.num_plugins++; |
814 |
|
} |
815 |
|
|
816 |
|
if (ARG_PASS1) { |
817 |
|
memset(&rc2pass1, 0, sizeof(xvid_plugin_2pass1_t)); |
818 |
|
rc2pass1.version = XVID_VERSION; |
819 |
|
rc2pass1.filename = ARG_PASS1; |
820 |
|
|
821 |
|
plugins[xvid_enc_create.num_plugins].func = xvid_plugin_2pass1; |
822 |
|
plugins[xvid_enc_create.num_plugins].param = &rc2pass1; |
823 |
|
xvid_enc_create.num_plugins++; |
824 |
|
} |
825 |
|
|
826 |
if (ARG_LUMIMASKING) { |
if (ARG_LUMIMASKING) { |
827 |
plugins[xvid_enc_create.num_plugins].func = xvid_plugin_lumimasking; |
plugins[xvid_enc_create.num_plugins].func = xvid_plugin_lumimasking; |
828 |
plugins[xvid_enc_create.num_plugins].param = NULL; |
plugins[xvid_enc_create.num_plugins].param = NULL; |
866 |
|
|
867 |
/* Global encoder options */ |
/* Global encoder options */ |
868 |
xvid_enc_create.global = 0; |
xvid_enc_create.global = 0; |
869 |
if (ARG_PACKED) xvid_enc_create.global |=XVID_PACKED; |
|
870 |
if (ARG_STATS) xvid_enc_create.global |=XVID_EXTRASTATS_ENABLE; |
if (ARG_PACKED) |
871 |
|
xvid_enc_create.global |=XVID_GLOBAL_PACKED; |
872 |
|
|
873 |
|
if (ARG_STATS) |
874 |
|
xvid_enc_create.global |=XVID_GLOBAL_EXTRASTATS_ENABLE; |
875 |
|
|
876 |
/* 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 */ |
877 |
xerr = xvid_encore(NULL, XVID_ENC_CREATE, &xvid_enc_create, NULL); |
xerr = xvid_encore(NULL, XVID_ENC_CREATE, &xvid_enc_create, NULL); |
930 |
/* Set up core's general features */ |
/* Set up core's general features */ |
931 |
xvid_enc_frame.vol_flags = vol_presets[ARG_QUALITY]; |
xvid_enc_frame.vol_flags = vol_presets[ARG_QUALITY]; |
932 |
if (ARG_STATS) |
if (ARG_STATS) |
933 |
xvid_enc_frame.vol_flags |= XVID_EXTRASTATS; |
xvid_enc_frame.vol_flags |= XVID_VOL_EXTRASTATS; |
934 |
|
|
935 |
/* Set up core's general features */ |
/* Set up core's general features */ |
936 |
xvid_enc_frame.vop_flags = vop_presets[ARG_QUALITY]; |
xvid_enc_frame.vop_flags = vop_presets[ARG_QUALITY]; |
938 |
/* Frame type -- let core decide for us */ |
/* Frame type -- let core decide for us */ |
939 |
xvid_enc_frame.type = XVID_TYPE_AUTO; |
xvid_enc_frame.type = XVID_TYPE_AUTO; |
940 |
|
|
941 |
/* Force the right quantizer */ |
/* Force the right quantizer -- It is internally managed by RC plugins */ |
942 |
xvid_enc_frame.quant = ARG_QUANTI; |
xvid_enc_frame.quant = 0; |
943 |
|
|
944 |
/* Set up motion estimation flags */ |
/* Set up motion estimation flags */ |
945 |
xvid_enc_frame.motion = motion_presets[ARG_QUALITY]; |
xvid_enc_frame.motion = motion_presets[ARG_QUALITY]; |