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.16 2003-04-14 20:19:00 edgomez 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; |
212 |
} else if (strcmp("-bitrate", argv[i]) == 0 && i < argc - 1) { |
} else if (strcmp("-bitrate", argv[i]) == 0 && i < argc - 1) { |
213 |
i++; |
i++; |
214 |
ARG_BITRATE = atoi(argv[i]); |
ARG_BITRATE = atoi(argv[i]); |
215 |
|
} else if (strcmp("-pass1", argv[i]) == 0 && i < argc - 1) { |
216 |
|
i++; |
217 |
|
ARG_PASS1 = argv[i]; |
218 |
|
} else if (strcmp("-pass2", argv[i]) == 0 && i < argc - 2) { |
219 |
|
i++; |
220 |
|
ARG_PASS2 = argv[i]; |
221 |
|
i++; |
222 |
|
ARG_PASS2_BITRATE = atoi(argv[i]); |
223 |
} else if (strcmp("-max_bframes", argv[i]) == 0 && i < argc - 1) { |
} else if (strcmp("-max_bframes", argv[i]) == 0 && i < argc - 1) { |
224 |
i++; |
i++; |
225 |
ARG_MAXBFRAMES = atoi(argv[i]); |
ARG_MAXBFRAMES = atoi(argv[i]); |
254 |
ARG_MAXFRAMENR = atoi(argv[i]); |
ARG_MAXFRAMENR = atoi(argv[i]); |
255 |
} else if (strcmp("-quant", argv[i]) == 0 && i < argc - 1) { |
} else if (strcmp("-quant", argv[i]) == 0 && i < argc - 1) { |
256 |
i++; |
i++; |
257 |
ARG_QUANTI = atoi(argv[i]); |
ARG_QUANTI = (float) atof(argv[i]); |
258 |
} else if (strcmp("-save", argv[i]) == 0) { |
} else if (strcmp("-save", argv[i]) == 0) { |
259 |
ARG_SAVEMPEGSTREAM = 1; |
ARG_SAVEMPEGSTREAM = 1; |
260 |
} else if (strcmp("-debug", argv[i]) == 0) { |
} else if (strcmp("-debug", argv[i]) == 0) { |
287 |
return (-1); |
return (-1); |
288 |
} |
} |
289 |
|
|
|
if (ARG_BITRATE <= 0 && ARG_QUANTI == 0) { |
|
|
fprintf(stderr, "Wrong Bitrate\n"); |
|
|
return (-1); |
|
|
} |
|
|
|
|
290 |
if (ARG_FRAMERATE <= 0) { |
if (ARG_FRAMERATE <= 0) { |
291 |
fprintf(stderr, "Wrong Framerate %s \n", argv[5]); |
fprintf(stderr, "Wrong Framerate %s \n", argv[5]); |
292 |
return (-1); |
return (-1); |
394 |
|
|
395 |
/* Write the Frame statistics */ |
/* Write the Frame statistics */ |
396 |
|
|
397 |
printf("%5d: key=%i, time(ms)=%6.1f, length=%7d", |
printf("%5d: key=%i, time=%6.0f, length=%7d", !result ? input_num : -1, |
398 |
!result ? input_num : -1, |
key, (float) enctime, (int) m4v_size); |
|
key, |
|
|
(float) enctime, |
|
|
(int) m4v_size); |
|
399 |
|
|
400 |
if (stats_type > 0) { /* !XVID_TYPE_NOTHING */ |
if (stats_type > 0) { /* !XVID_TYPE_NOTHING */ |
401 |
|
|
424 |
|
|
425 |
if (ARG_STATS) { |
if (ARG_STATS) { |
426 |
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", |
427 |
SSE2PSNR(sse[0], XDIM, YDIM), |
SSE2PSNR(sse[0], XDIM, YDIM), SSE2PSNR(sse[1], XDIM / 2, |
428 |
SSE2PSNR(sse[1], XDIM/2, YDIM/2), |
YDIM / 2), |
429 |
SSE2PSNR(sse[2], XDIM/2, YDIM/2)); |
SSE2PSNR(sse[2], XDIM/2, YDIM/2)); |
430 |
} |
} |
431 |
|
|
432 |
} |
} |
|
|
|
433 |
#undef SSE2PSNR |
#undef SSE2PSNR |
434 |
|
|
435 |
printf("\n"); |
printf("\n"); |
565 |
fprintf(stderr, " -save : save mpeg4 raw stream\n"); |
fprintf(stderr, " -save : save mpeg4 raw stream\n"); |
566 |
fprintf(stderr, " -o string: output filename\n"); |
fprintf(stderr, " -o string: output filename\n"); |
567 |
fprintf(stderr, "\n"); |
fprintf(stderr, "\n"); |
568 |
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"); |
|
569 |
fprintf(stderr, " -max_bframes integer: max bframes (default=0)\n"); |
fprintf(stderr, " -max_bframes integer: max bframes (default=0)\n"); |
570 |
fprintf(stderr, " -bquant_ratio integer: bframe quantizer ratio (default=150)\n"); |
fprintf(stderr, " -bquant_ratio integer: bframe quantizer ratio (default=150)\n"); |
571 |
fprintf(stderr, " -bquant_offset integer: bframe quantizer offset (default=100)\n"); |
fprintf(stderr, " -bquant_offset integer: bframe quantizer offset (default=100)\n"); |
572 |
fprintf(stderr, " -framerate float : target framerate (>0)\n"); |
fprintf(stderr, "\n"); |
573 |
|
fprintf(stderr, "Rate control options:\n"); |
574 |
|
fprintf(stderr, " -framerate float : target framerate (>0 | default=25.0)\n"); |
575 |
|
fprintf(stderr, " -bitrate integer : bitrate -- for CBR/VBR pass2\n"); |
576 |
|
fprintf(stderr, " -quant float : quantizer -- for \"Fixed\" quantizer RC\n"); |
577 |
|
fprintf(stderr, " -pass1 filename : output stats filename\n"); |
578 |
|
fprintf(stderr, " -pass2 filename bitrate : input stats filename, target bitrate\n"); |
579 |
fprintf(stderr, "\n"); |
fprintf(stderr, "\n"); |
580 |
fprintf(stderr, "Other options\n"); |
fprintf(stderr, "Other options\n"); |
581 |
fprintf(stderr, " -asm : use assembly optmized code\n"); |
fprintf(stderr, " -asm : use assembly optmized code\n"); |
583 |
fprintf(stderr, " -packed : packed mode\n"); |
fprintf(stderr, " -packed : packed mode\n"); |
584 |
fprintf(stderr, " -lumimasking : use lumimasking algorithm\n"); |
fprintf(stderr, " -lumimasking : use lumimasking algorithm\n"); |
585 |
fprintf(stderr, " -stats : print stats about encoded frames\n"); |
fprintf(stderr, " -stats : print stats about encoded frames\n"); |
586 |
fprintf(stderr, " -debug : print all MB quantizers\n"); |
fprintf(stderr, " -debug : print all MB dquants\n"); |
587 |
fprintf(stderr, " -help : prints this help message\n"); |
fprintf(stderr, " -help : prints this help message\n"); |
588 |
} |
} |
589 |
|
|
712 |
|
|
713 |
#define FRAMERATE_INCR 1001 |
#define FRAMERATE_INCR 1001 |
714 |
|
|
715 |
|
|
716 |
/* Initialize encoder for first use, pass all needed parameters to the codec */ |
/* Initialize encoder for first use, pass all needed parameters to the codec */ |
717 |
static int |
static int |
718 |
enc_init(int use_assembler) |
enc_init(int use_assembler) |
719 |
{ |
{ |
720 |
int xerr; |
int xerr; |
721 |
|
xvid_plugin_cbr_t cbr; |
722 |
xvid_enc_plugin_t plugins[3]; |
xvid_plugin_2pass1_t rc2pass1; |
723 |
|
xvid_plugin_2pass2_t rc2pass2; |
724 |
|
xvid_plugin_fixed_t rcfixed; |
725 |
|
xvid_enc_plugin_t plugins[7]; |
726 |
xvid_gbl_init_t xvid_gbl_init; |
xvid_gbl_init_t xvid_gbl_init; |
727 |
xvid_enc_create_t xvid_enc_create; |
xvid_enc_create_t xvid_enc_create; |
728 |
|
|
767 |
xvid_enc_create.plugins = plugins; |
xvid_enc_create.plugins = plugins; |
768 |
xvid_enc_create.num_plugins = 0; |
xvid_enc_create.num_plugins = 0; |
769 |
|
|
770 |
|
if (ARG_BITRATE) { |
771 |
|
memset(&cbr, 0, sizeof(xvid_plugin_cbr_t)); |
772 |
|
cbr.version = XVID_VERSION; |
773 |
|
cbr.bitrate = ARG_BITRATE; |
774 |
|
|
775 |
|
plugins[xvid_enc_create.num_plugins].func = xvid_plugin_cbr; |
776 |
|
plugins[xvid_enc_create.num_plugins].param = &cbr; |
777 |
|
xvid_enc_create.num_plugins++; |
778 |
|
} |
779 |
|
|
780 |
|
if (ARG_QUANTI) { |
781 |
|
memset(&rcfixed, 0, sizeof(xvid_plugin_fixed_t)); |
782 |
|
rcfixed.version = XVID_VERSION; |
783 |
|
/* We will use a 1/10 precision, just to make sure it works */ |
784 |
|
rcfixed.quant_base = 10; |
785 |
|
rcfixed.quant_increment = (int) (ARG_QUANTI * 10); |
786 |
|
|
787 |
|
plugins[xvid_enc_create.num_plugins].func = xvid_plugin_fixed; |
788 |
|
plugins[xvid_enc_create.num_plugins].param = &rcfixed; |
789 |
|
xvid_enc_create.num_plugins++; |
790 |
|
} |
791 |
|
|
792 |
|
if (ARG_PASS2) { |
793 |
|
memset(&rc2pass2, 0, sizeof(xvid_plugin_2pass2_t)); |
794 |
|
rc2pass2.version = XVID_VERSION; |
795 |
|
rc2pass2.filename = ARG_PASS2; |
796 |
|
rc2pass2.bitrate = ARG_PASS2_BITRATE; |
797 |
|
|
798 |
|
plugins[xvid_enc_create.num_plugins].func = xvid_plugin_2pass2; |
799 |
|
plugins[xvid_enc_create.num_plugins].param = &rc2pass2; |
800 |
|
xvid_enc_create.num_plugins++; |
801 |
|
} |
802 |
|
|
803 |
|
if (ARG_PASS1) { |
804 |
|
memset(&rc2pass1, 0, sizeof(xvid_plugin_2pass1_t)); |
805 |
|
rc2pass1.version = XVID_VERSION; |
806 |
|
rc2pass1.filename = ARG_PASS1; |
807 |
|
|
808 |
|
plugins[xvid_enc_create.num_plugins].func = xvid_plugin_2pass1; |
809 |
|
plugins[xvid_enc_create.num_plugins].param = &rc2pass1; |
810 |
|
xvid_enc_create.num_plugins++; |
811 |
|
} |
812 |
|
|
813 |
if (ARG_LUMIMASKING) { |
if (ARG_LUMIMASKING) { |
814 |
plugins[xvid_enc_create.num_plugins].func = xvid_plugin_lumimasking; |
plugins[xvid_enc_create.num_plugins].func = xvid_plugin_lumimasking; |
815 |
plugins[xvid_enc_create.num_plugins].param = NULL; |
plugins[xvid_enc_create.num_plugins].param = NULL; |
853 |
|
|
854 |
/* Global encoder options */ |
/* Global encoder options */ |
855 |
xvid_enc_create.global = 0; |
xvid_enc_create.global = 0; |
856 |
if (ARG_PACKED) xvid_enc_create.global |=XVID_PACKED; |
|
857 |
if (ARG_STATS) xvid_enc_create.global |=XVID_EXTRASTATS_ENABLE; |
if (ARG_PACKED) |
858 |
|
xvid_enc_create.global |=XVID_GLOBAL_PACKED; |
859 |
|
|
860 |
|
if (ARG_STATS) |
861 |
|
xvid_enc_create.global |=XVID_GLOBAL_EXTRASTATS_ENABLE; |
862 |
|
|
863 |
/* 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 */ |
864 |
xerr = xvid_encore(NULL, XVID_ENC_CREATE, &xvid_enc_create, NULL); |
xerr = xvid_encore(NULL, XVID_ENC_CREATE, &xvid_enc_create, NULL); |
917 |
/* Set up core's general features */ |
/* Set up core's general features */ |
918 |
xvid_enc_frame.vol_flags = vol_presets[ARG_QUALITY]; |
xvid_enc_frame.vol_flags = vol_presets[ARG_QUALITY]; |
919 |
if (ARG_STATS) |
if (ARG_STATS) |
920 |
xvid_enc_frame.vol_flags |= XVID_EXTRASTATS; |
xvid_enc_frame.vol_flags |= XVID_VOL_EXTRASTATS; |
921 |
|
|
922 |
/* Set up core's general features */ |
/* Set up core's general features */ |
923 |
xvid_enc_frame.vop_flags = vop_presets[ARG_QUALITY]; |
xvid_enc_frame.vop_flags = vop_presets[ARG_QUALITY]; |
925 |
/* Frame type -- let core decide for us */ |
/* Frame type -- let core decide for us */ |
926 |
xvid_enc_frame.type = XVID_TYPE_AUTO; |
xvid_enc_frame.type = XVID_TYPE_AUTO; |
927 |
|
|
928 |
/* Force the right quantizer */ |
/* Force the right quantizer -- It is internally managed by RC plugins */ |
929 |
xvid_enc_frame.quant = ARG_QUANTI; |
xvid_enc_frame.quant = 0; |
930 |
|
|
931 |
/* Set up motion estimation flags */ |
/* Set up motion estimation flags */ |
932 |
xvid_enc_frame.motion = motion_presets[ARG_QUALITY]; |
xvid_enc_frame.motion = motion_presets[ARG_QUALITY]; |