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.21 2003-05-12 00:06:49 edgomez Exp $ |
* $Id: xvid_encraw.c,v 1.11.2.25 2003-05-17 13:21:26 suxen_drol Exp $ |
23 |
* |
* |
24 |
****************************************************************************/ |
****************************************************************************/ |
25 |
|
|
87 |
0, |
0, |
88 |
|
|
89 |
/* quality 1 */ |
/* quality 1 */ |
90 |
XVID_VOP_DYNAMIC_BFRAMES, |
0, |
91 |
|
|
92 |
/* quality 2 */ |
/* quality 2 */ |
93 |
XVID_VOP_DYNAMIC_BFRAMES | XVID_VOP_HALFPEL, |
XVID_VOP_HALFPEL, |
94 |
|
|
95 |
/* quality 3 */ |
/* quality 3 */ |
96 |
XVID_VOP_DYNAMIC_BFRAMES | XVID_VOP_HALFPEL | |
XVID_VOP_HALFPEL | XVID_VOP_INTER4V, |
|
XVID_VOP_INTER4V, |
|
97 |
|
|
98 |
/* quality 4 */ |
/* quality 4 */ |
99 |
XVID_VOP_DYNAMIC_BFRAMES | XVID_VOP_HALFPEL | |
XVID_VOP_HALFPEL | XVID_VOP_INTER4V, |
|
XVID_VOP_INTER4V, |
|
100 |
|
|
101 |
/* quality 5 */ |
/* quality 5 */ |
102 |
XVID_VOP_DYNAMIC_BFRAMES | XVID_VOP_HALFPEL | |
XVID_VOP_HALFPEL | XVID_VOP_INTER4V | |
103 |
XVID_VOP_INTER4V | XVID_VOP_TRELLISQUANT, |
XVID_VOP_TRELLISQUANT, |
104 |
|
|
105 |
/* quality 6 */ |
/* quality 6 */ |
106 |
XVID_VOP_DYNAMIC_BFRAMES | XVID_VOP_HALFPEL | |
XVID_VOP_HALFPEL | XVID_VOP_INTER4V | |
107 |
XVID_VOP_INTER4V | XVID_VOP_TRELLISQUANT | |
XVID_VOP_TRELLISQUANT | XVID_VOP_HQACPRED, |
|
XVID_VOP_HQACPRED, |
|
108 |
|
|
109 |
}; |
}; |
110 |
#define VOP_ELEMENTS (sizeof(vop_presets)/sizeof(vop_presets[0])) |
#define VOP_ELEMENTS (sizeof(vop_presets)/sizeof(vop_presets[0])) |
113 |
* Command line global variables |
* Command line global variables |
114 |
****************************************************************************/ |
****************************************************************************/ |
115 |
|
|
116 |
|
#define MAX_ZONES 64 |
117 |
|
|
118 |
|
static xvid_enc_zone_t ZONES[MAX_ZONES]; |
119 |
|
static int NUM_ZONES = 0; |
120 |
|
|
121 |
/* Maximum number of frames to encode */ |
/* Maximum number of frames to encode */ |
122 |
#define ABS_MAXFRAMENR 9999 |
#define ABS_MAXFRAMENR 9999 |
123 |
|
|
125 |
static int ARG_DUMP = 0; |
static int ARG_DUMP = 0; |
126 |
static int ARG_LUMIMASKING = 0; |
static int ARG_LUMIMASKING = 0; |
127 |
static int ARG_BITRATE = 0; |
static int ARG_BITRATE = 0; |
128 |
|
static int ARG_SINGLE = 0; |
129 |
static char *ARG_PASS1 = 0; |
static char *ARG_PASS1 = 0; |
130 |
static char *ARG_PASS2 = 0; |
static char *ARG_PASS2 = 0; |
131 |
static int ARG_PASS2_BITRATE = 0; |
static int ARG_QUALITY = ME_ELEMENTS - 1; |
|
static float ARG_QUANTI = 0.0f; |
|
|
static int ARG_QUALITY = 3; |
|
132 |
static float ARG_FRAMERATE = 25.00f; |
static float ARG_FRAMERATE = 25.00f; |
133 |
static int ARG_MAXFRAMENR = ABS_MAXFRAMENR; |
static int ARG_MAXFRAMENR = ABS_MAXFRAMENR; |
134 |
|
static int ARG_MAXKEYINTERVAL = 0; |
135 |
static char *ARG_INPUTFILE = NULL; |
static char *ARG_INPUTFILE = NULL; |
136 |
static int ARG_INPUTTYPE = 0; |
static int ARG_INPUTTYPE = 0; |
137 |
static int ARG_SAVEMPEGSTREAM = 0; |
static int ARG_SAVEMPEGSTREAM = 0; |
143 |
static int ARG_MAXBFRAMES = 0; |
static int ARG_MAXBFRAMES = 0; |
144 |
static int ARG_PACKED = 0; |
static int ARG_PACKED = 0; |
145 |
static int ARG_DEBUG = 0; |
static int ARG_DEBUG = 0; |
146 |
|
static int ARG_VOPDEBUG = 0; |
147 |
|
|
148 |
#define IMAGE_SIZE(x,y) ((x)*(y)*3/2) |
#define IMAGE_SIZE(x,y) ((x)*(y)*3/2) |
149 |
|
|
253 |
} else if (strcmp("-bitrate", argv[i]) == 0 && i < argc - 1) { |
} else if (strcmp("-bitrate", argv[i]) == 0 && i < argc - 1) { |
254 |
i++; |
i++; |
255 |
ARG_BITRATE = atoi(argv[i]); |
ARG_BITRATE = atoi(argv[i]); |
256 |
|
} else if (strcmp("-single", argv[i]) == 0) { |
257 |
|
ARG_SINGLE = 1; |
258 |
} else if (strcmp("-pass1", argv[i]) == 0 && i < argc - 1) { |
} else if (strcmp("-pass1", argv[i]) == 0 && i < argc - 1) { |
259 |
i++; |
i++; |
260 |
ARG_PASS1 = argv[i]; |
ARG_PASS1 = argv[i]; |
261 |
} else if (strcmp("-pass2", argv[i]) == 0 && i < argc - 2) { |
} else if (strcmp("-pass2", argv[i]) == 0 && i < argc - 1) { |
262 |
i++; |
i++; |
263 |
ARG_PASS2 = argv[i]; |
ARG_PASS2 = argv[i]; |
|
i++; |
|
|
ARG_PASS2_BITRATE = atoi(argv[i]); |
|
264 |
} else if (strcmp("-max_bframes", argv[i]) == 0 && i < argc - 1) { |
} else if (strcmp("-max_bframes", argv[i]) == 0 && i < argc - 1) { |
265 |
i++; |
i++; |
266 |
ARG_MAXBFRAMES = atoi(argv[i]); |
ARG_MAXBFRAMES = atoi(argv[i]); |
272 |
} else if (strcmp("-bquant_offset", argv[i]) == 0 && i < argc - 1) { |
} else if (strcmp("-bquant_offset", argv[i]) == 0 && i < argc - 1) { |
273 |
i++; |
i++; |
274 |
ARG_BQOFFSET = atoi(argv[i]); |
ARG_BQOFFSET = atoi(argv[i]); |
275 |
|
|
276 |
|
} else if ((strcmp("-zq", argv[i]) == 0 || strcmp("-zw", argv[i]) == 0) && i < argc - 2) { |
277 |
|
|
278 |
|
if (NUM_ZONES >= MAX_ZONES) { |
279 |
|
fprintf(stderr,"warning: too many zones; zone ignored\n"); |
280 |
|
continue; |
281 |
|
} |
282 |
|
ZONES[NUM_ZONES].mode = strcmp("-zq", argv[i])==0 ? XVID_ZONE_QUANT : XVID_ZONE_WEIGHT; |
283 |
|
i++; |
284 |
|
ZONES[NUM_ZONES].frame = atoi(argv[i]); |
285 |
|
i++; |
286 |
|
ZONES[NUM_ZONES].increment = (int)(atof(argv[i]) * 100); |
287 |
|
ZONES[NUM_ZONES].base = 100; |
288 |
|
NUM_ZONES++; |
289 |
|
|
290 |
} else if (strcmp("-quality", argv[i]) == 0 && i < argc - 1) { |
} else if (strcmp("-quality", argv[i]) == 0 && i < argc - 1) { |
291 |
i++; |
i++; |
292 |
ARG_QUALITY = atoi(argv[i]); |
ARG_QUALITY = atoi(argv[i]); |
293 |
} else if (strcmp("-framerate", argv[i]) == 0 && i < argc - 1) { |
} else if (strcmp("-framerate", argv[i]) == 0 && i < argc - 1) { |
294 |
i++; |
i++; |
295 |
ARG_FRAMERATE = (float) atof(argv[i]); |
ARG_FRAMERATE = (float) atof(argv[i]); |
296 |
|
} else if (strcmp("-max_key_interval", argv[i]) == 0 && i < argc - 1) { |
297 |
|
i++; |
298 |
|
ARG_MAXKEYINTERVAL = atoi(argv[i]); |
299 |
} else if (strcmp("-i", argv[i]) == 0 && i < argc - 1) { |
} else if (strcmp("-i", argv[i]) == 0 && i < argc - 1) { |
300 |
i++; |
i++; |
301 |
ARG_INPUTFILE = argv[i]; |
ARG_INPUTFILE = argv[i]; |
311 |
} else if (strcmp("-nframes", argv[i]) == 0 && i < argc - 1) { |
} else if (strcmp("-nframes", argv[i]) == 0 && i < argc - 1) { |
312 |
i++; |
i++; |
313 |
ARG_MAXFRAMENR = atoi(argv[i]); |
ARG_MAXFRAMENR = atoi(argv[i]); |
|
} else if (strcmp("-quant", argv[i]) == 0 && i < argc - 1) { |
|
|
i++; |
|
|
ARG_QUANTI = (float) atof(argv[i]); |
|
314 |
} else if (strcmp("-save", argv[i]) == 0) { |
} else if (strcmp("-save", argv[i]) == 0) { |
315 |
ARG_SAVEMPEGSTREAM = 1; |
ARG_SAVEMPEGSTREAM = 1; |
316 |
} else if (strcmp("-debug", argv[i]) == 0) { |
} else if (strcmp("-debug", argv[i]) == 0) { |
317 |
ARG_DEBUG = 1; |
i++; |
318 |
|
if (sscanf(argv[i],"0x%x", &ARG_DEBUG) || sscanf(argv[i],"%d", &ARG_DEBUG)) ; |
319 |
} else if (strcmp("-o", argv[i]) == 0 && i < argc - 1) { |
} else if (strcmp("-o", argv[i]) == 0 && i < argc - 1) { |
320 |
i++; |
i++; |
321 |
ARG_OUTPUTFILE = argv[i]; |
ARG_OUTPUTFILE = argv[i]; |
322 |
|
} else if (strcmp("-vop_debug", argv[i]) == 0) { |
323 |
|
ARG_VOPDEBUG = 1; |
324 |
|
|
325 |
} else if (strcmp("-help", argv[i])) { |
} else if (strcmp("-help", argv[i])) { |
326 |
usage(); |
usage(); |
327 |
return (0); |
return (0); |
455 |
|
|
456 |
/* Write the Frame statistics */ |
/* Write the Frame statistics */ |
457 |
|
|
458 |
printf("%5d: key=%i, time= %6.0f, length= %7d", !result ? input_num : -1, |
printf("%5d: key=%i, time= %6.0f, len= %7d", !result ? input_num : -1, |
459 |
key, (float) enctime, (int) m4v_size); |
key, (float) enctime, (int) m4v_size); |
460 |
|
|
461 |
if (stats_type > 0) { /* !XVID_TYPE_NOTHING */ |
if (stats_type > 0) { /* !XVID_TYPE_NOTHING */ |
478 |
break; |
break; |
479 |
} |
} |
480 |
|
|
481 |
printf(" | type=%s, quant= %2d, length= %7d", type, stats_quant, |
printf(" | type=%s, quant= %2d, len= %7d", type, stats_quant, |
482 |
stats_length); |
stats_length); |
483 |
|
|
484 |
#define SSE2PSNR(sse, width, height) ((!(sse))?0.0f : 48.131f - 10*(float)log10((float)(sse)/((float)((width)*(height))))) |
#define SSE2PSNR(sse, width, height) ((!(sse))?0.0f : 48.131f - 10*(float)log10((float)(sse)/((float)((width)*(height))))) |
645 |
fprintf(stderr, "\n"); |
fprintf(stderr, "\n"); |
646 |
fprintf(stderr, "Rate control options:\n"); |
fprintf(stderr, "Rate control options:\n"); |
647 |
fprintf(stderr, " -framerate float : target framerate (>0 | default=25.0)\n"); |
fprintf(stderr, " -framerate float : target framerate (>0 | default=25.0)\n"); |
648 |
fprintf(stderr, " -bitrate integer : bitrate -- for CBR/VBR pass2\n"); |
fprintf(stderr, " -bitrate integer : target bitrate\n"); |
649 |
fprintf(stderr, " -quant float : quantizer -- for \"Fixed\" quantizer RC\n"); |
fprintf(stderr, " -single : single pass mode\n"); |
650 |
fprintf(stderr, " -pass1 filename : output stats filename\n"); |
fprintf(stderr, " -pass1 filename : twopass mode (first pass)\n"); |
651 |
fprintf(stderr, " -pass2 filename bitrate : input stats filename, target bitrate\n"); |
fprintf(stderr, " -pass2 filename : twopass mode (2nd pass)\n"); |
652 |
|
fprintf(stderr, " -zq starting_frame float : bitrate zone; quant\n"); |
653 |
|
fprintf(stderr, " -zw starting_frame float : bitrate zone; weight\n"); |
654 |
|
fprintf(stderr, " -max_key_interval integer : maximum keyframe interval\n"); |
655 |
fprintf(stderr, "\n"); |
fprintf(stderr, "\n"); |
656 |
fprintf(stderr, "Other options\n"); |
fprintf(stderr, "Other options\n"); |
657 |
fprintf(stderr, " -asm : use assembly optmized code\n"); |
fprintf(stderr, " -asm : use assembly optmized code\n"); |
661 |
fprintf(stderr, " -stats : print stats about encoded frames\n"); |
fprintf(stderr, " -stats : print stats about encoded frames\n"); |
662 |
fprintf(stderr, " -debug : print all MB dquants\n"); |
fprintf(stderr, " -debug : print all MB dquants\n"); |
663 |
fprintf(stderr, " -help : prints this help message\n"); |
fprintf(stderr, " -help : prints this help message\n"); |
664 |
|
fprintf(stderr, "\n"); |
665 |
|
fprintf(stderr, "NB: You can define %d zones repeating the -z[qw] option as many times as needed.\n", MAX_ZONES); |
666 |
|
fprintf(stderr, "\n"); |
667 |
} |
} |
668 |
|
|
669 |
/***************************************************************************** |
/***************************************************************************** |
797 |
enc_init(int use_assembler) |
enc_init(int use_assembler) |
798 |
{ |
{ |
799 |
int xerr; |
int xerr; |
800 |
xvid_plugin_cbr_t cbr; |
//xvid_plugin_cbr_t cbr; |
801 |
|
xvid_plugin_single_t single; |
802 |
xvid_plugin_2pass1_t rc2pass1; |
xvid_plugin_2pass1_t rc2pass1; |
803 |
xvid_plugin_2pass2_t rc2pass2; |
xvid_plugin_2pass2_t rc2pass2; |
804 |
xvid_plugin_fixed_t rcfixed; |
//xvid_plugin_fixed_t rcfixed; |
805 |
xvid_enc_plugin_t plugins[7]; |
xvid_enc_plugin_t plugins[7]; |
806 |
xvid_gbl_init_t xvid_gbl_init; |
xvid_gbl_init_t xvid_gbl_init; |
807 |
xvid_enc_create_t xvid_enc_create; |
xvid_enc_create_t xvid_enc_create; |
813 |
/* Set version -- version checking will done by xvidcore */ |
/* Set version -- version checking will done by xvidcore */ |
814 |
memset(&xvid_gbl_init, 0, sizeof(xvid_gbl_init)); |
memset(&xvid_gbl_init, 0, sizeof(xvid_gbl_init)); |
815 |
xvid_gbl_init.version = XVID_VERSION; |
xvid_gbl_init.version = XVID_VERSION; |
816 |
|
printf("0x%x\n", ARG_DEBUG); |
817 |
|
xvid_gbl_init.debug = ARG_DEBUG; |
818 |
|
|
819 |
|
|
820 |
/* Do we have to enable ASM optimizations ? */ |
/* Do we have to enable ASM optimizations ? */ |
845 |
xvid_enc_create.height = YDIM; |
xvid_enc_create.height = YDIM; |
846 |
|
|
847 |
/* init plugins */ |
/* init plugins */ |
848 |
|
xvid_enc_create.zones = ZONES; |
849 |
|
xvid_enc_create.num_zones = NUM_ZONES; |
850 |
|
|
851 |
xvid_enc_create.plugins = plugins; |
xvid_enc_create.plugins = plugins; |
852 |
xvid_enc_create.num_plugins = 0; |
xvid_enc_create.num_plugins = 0; |
853 |
|
|
854 |
if (ARG_BITRATE) { |
if (ARG_SINGLE) { |
855 |
memset(&cbr, 0, sizeof(xvid_plugin_cbr_t)); |
memset(&single, 0, sizeof(xvid_plugin_single_t)); |
856 |
cbr.version = XVID_VERSION; |
single.version = XVID_VERSION; |
857 |
cbr.bitrate = ARG_BITRATE; |
single.bitrate = ARG_BITRATE; |
|
|
|
|
plugins[xvid_enc_create.num_plugins].func = xvid_plugin_cbr; |
|
|
plugins[xvid_enc_create.num_plugins].param = &cbr; |
|
|
xvid_enc_create.num_plugins++; |
|
|
} |
|
|
|
|
|
if (ARG_QUANTI) { |
|
|
memset(&rcfixed, 0, sizeof(xvid_plugin_fixed_t)); |
|
|
rcfixed.version = XVID_VERSION; |
|
|
/* We will use a 1/10 precision, just to make sure it works */ |
|
|
rcfixed.quant_base = 10; |
|
|
rcfixed.quant_increment = (int) (ARG_QUANTI * 10); |
|
858 |
|
|
859 |
plugins[xvid_enc_create.num_plugins].func = xvid_plugin_fixed; |
plugins[xvid_enc_create.num_plugins].func = xvid_plugin_single; |
860 |
plugins[xvid_enc_create.num_plugins].param = &rcfixed; |
plugins[xvid_enc_create.num_plugins].param = &single; |
861 |
xvid_enc_create.num_plugins++; |
xvid_enc_create.num_plugins++; |
862 |
} |
} |
863 |
|
|
865 |
memset(&rc2pass2, 0, sizeof(xvid_plugin_2pass2_t)); |
memset(&rc2pass2, 0, sizeof(xvid_plugin_2pass2_t)); |
866 |
rc2pass2.version = XVID_VERSION; |
rc2pass2.version = XVID_VERSION; |
867 |
rc2pass2.filename = ARG_PASS2; |
rc2pass2.filename = ARG_PASS2; |
868 |
rc2pass2.bitrate = ARG_PASS2_BITRATE; |
rc2pass2.bitrate = ARG_BITRATE; |
869 |
|
|
870 |
plugins[xvid_enc_create.num_plugins].func = xvid_plugin_2pass2; |
plugins[xvid_enc_create.num_plugins].func = xvid_plugin_2pass2; |
871 |
plugins[xvid_enc_create.num_plugins].param = &rc2pass2; |
plugins[xvid_enc_create.num_plugins].param = &rc2pass2; |
894 |
xvid_enc_create.num_plugins++; |
xvid_enc_create.num_plugins++; |
895 |
} |
} |
896 |
|
|
897 |
if (ARG_DEBUG) { |
/* if (ARG_DEBUG) { |
898 |
plugins[xvid_enc_create.num_plugins].func = rawenc_debug; |
plugins[xvid_enc_create.num_plugins].func = rawenc_debug; |
899 |
plugins[xvid_enc_create.num_plugins].param = NULL; |
plugins[xvid_enc_create.num_plugins].param = NULL; |
900 |
xvid_enc_create.num_plugins++; |
xvid_enc_create.num_plugins++; |
901 |
} |
}*/ |
902 |
|
|
903 |
/* No fancy thread tests */ |
/* No fancy thread tests */ |
904 |
xvid_enc_create.num_threads = 0; |
xvid_enc_create.num_threads = 0; |
913 |
} |
} |
914 |
|
|
915 |
/* Maximum key frame interval */ |
/* Maximum key frame interval */ |
916 |
|
|
917 |
|
if (ARG_MAXKEYINTERVAL > 0) { |
918 |
|
xvid_enc_create.max_key_interval = ARG_MAXKEYINTERVAL; |
919 |
|
}else { |
920 |
xvid_enc_create.max_key_interval = (int) ARG_FRAMERATE *10; |
xvid_enc_create.max_key_interval = (int) ARG_FRAMERATE *10; |
921 |
|
} |
922 |
|
|
923 |
/* Bframes settings */ |
/* Bframes settings */ |
924 |
xvid_enc_create.max_bframes = ARG_MAXBFRAMES; |
xvid_enc_create.max_bframes = ARG_MAXBFRAMES; |
998 |
|
|
999 |
/* Set up core's general features */ |
/* Set up core's general features */ |
1000 |
xvid_enc_frame.vop_flags = vop_presets[ARG_QUALITY]; |
xvid_enc_frame.vop_flags = vop_presets[ARG_QUALITY]; |
1001 |
|
if (ARG_VOPDEBUG) { |
1002 |
|
xvid_enc_frame.vop_flags |= XVID_VOP_DEBUG; |
1003 |
|
} |
1004 |
|
|
1005 |
/* Frame type -- let core decide for us */ |
/* Frame type -- let core decide for us */ |
1006 |
xvid_enc_frame.type = XVID_TYPE_AUTO; |
xvid_enc_frame.type = XVID_TYPE_AUTO; |