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.13 2003-03-25 22:53:57 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 | |
XVID_ME_HALFPELREFINE16 | XVID_ME_HALFPELREFINE8, /* 4 */ |
60 |
PMV_USESQUARES16, |
XVID_ME_HALFPELREFINE16 | XVID_ME_HALFPELREFINE8 | XVID_ME_EXTSEARCH16 | |
61 |
PMV_HALFPELREFINE16 | PMV_HALFPELREFINE8 | PMV_EXTSEARCH16 | |
XVID_ME_USESQUARES16, /* 5 */ |
62 |
PMV_USESQUARES16 | PMV_CHROMA16 | PMV_CHROMA8, |
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_VOP_DYNAMIC_BFRAMES | XVID_VOP_HALFPEL | XVID_VOP_INTER4V | XVID_VOP_HQACPRED, /* 5 */ |
83 |
XVID_MODEDECISION_BITS |
XVID_VOP_DYNAMIC_BFRAMES | XVID_VOP_HALFPEL | XVID_VOP_HQACPRED | /* 6 */ |
84 |
|
XVID_VOP_MODEDECISION_BITS |
85 |
}; |
}; |
86 |
|
|
87 |
/***************************************************************************** |
/***************************************************************************** |
97 |
static int ARG_BITRATE = 0; |
static int ARG_BITRATE = 0; |
98 |
static char *ARG_PASS1 = 0; |
static char *ARG_PASS1 = 0; |
99 |
static char *ARG_PASS2 = 0; |
static char *ARG_PASS2 = 0; |
100 |
|
static int ARG_PASS2_BITRATE = 0; |
101 |
static float ARG_QUANTI = 0.0f; |
static float ARG_QUANTI = 0.0f; |
102 |
static int ARG_QUALITY = 5; |
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; |
221 |
ARG_PASS1 = argv[i]; |
ARG_PASS1 = argv[i]; |
222 |
} else if (strcmp("-pass2", argv[i]) == 0 && i < argc - 2) { |
} else if (strcmp("-pass2", argv[i]) == 0 && i < argc - 2) { |
223 |
i++; |
i++; |
|
ARG_PASS1 = argv[i]; |
|
|
i++; |
|
224 |
ARG_PASS2 = argv[i]; |
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]); |
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 |
} |
} |
431 |
SSE2PSNR(sse[0], XDIM, YDIM), SSE2PSNR(sse[1], XDIM / 2, |
SSE2PSNR(sse[0], XDIM, YDIM), SSE2PSNR(sse[1], XDIM / 2, |
432 |
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 |
} |
} |
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 |
/***************************************************************************** |
/***************************************************************************** |
590 |
fprintf(stderr, " -framerate float : target framerate (>0 | default=25.0)\n"); |
fprintf(stderr, " -framerate float : target framerate (>0 | default=25.0)\n"); |
591 |
fprintf(stderr, " -bitrate integer : bitrate -- for CBR/VBR pass2\n"); |
fprintf(stderr, " -bitrate integer : bitrate -- for CBR/VBR pass2\n"); |
592 |
fprintf(stderr, " -quant float : quantizer -- for \"Fixed\" quantizer RC\n"); |
fprintf(stderr, " -quant float : quantizer -- for \"Fixed\" quantizer RC\n"); |
593 |
fprintf(stderr, " -pass1 filename : stats filename\n"); |
fprintf(stderr, " -pass1 filename : output stats filename\n"); |
594 |
fprintf(stderr, " -pass2 filename1 filename2 : first pass stats and scaled stats filename\n"); |
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"); |
784 |
xvid_enc_create.num_plugins = 0; |
xvid_enc_create.num_plugins = 0; |
785 |
|
|
786 |
if (ARG_BITRATE) { |
if (ARG_BITRATE) { |
|
cbr.version = XVID_VERSION; |
|
787 |
memset(&cbr, 0, sizeof(xvid_plugin_cbr_t)); |
memset(&cbr, 0, sizeof(xvid_plugin_cbr_t)); |
788 |
|
cbr.version = XVID_VERSION; |
789 |
cbr.bitrate = ARG_BITRATE; |
cbr.bitrate = ARG_BITRATE; |
790 |
|
|
791 |
plugins[xvid_enc_create.num_plugins].func = xvid_plugin_cbr; |
plugins[xvid_enc_create.num_plugins].func = xvid_plugin_cbr; |
794 |
} |
} |
795 |
|
|
796 |
if (ARG_QUANTI) { |
if (ARG_QUANTI) { |
797 |
|
memset(&rcfixed, 0, sizeof(xvid_plugin_fixed_t)); |
798 |
rcfixed.version = XVID_VERSION; |
rcfixed.version = XVID_VERSION; |
799 |
/* We will use a 1/10 precision, just to make sure it works */ |
/* We will use a 1/10 precision, just to make sure it works */ |
800 |
rcfixed.quant_base = 10; |
rcfixed.quant_base = 10; |
805 |
xvid_enc_create.num_plugins++; |
xvid_enc_create.num_plugins++; |
806 |
} |
} |
807 |
|
|
808 |
if (ARG_PASS1 && ARG_PASS2) { |
if (ARG_PASS2) { |
|
rc2pass2.version = XVID_VERSION; |
|
809 |
memset(&rc2pass2, 0, sizeof(xvid_plugin_2pass2_t)); |
memset(&rc2pass2, 0, sizeof(xvid_plugin_2pass2_t)); |
810 |
rc2pass2.filename1 = ARG_PASS1; |
rc2pass2.version = XVID_VERSION; |
811 |
rc2pass2.filename2 = ARG_PASS2; |
rc2pass2.filename = ARG_PASS2; |
812 |
|
rc2pass2.bitrate = ARG_PASS2_BITRATE; |
813 |
|
|
814 |
plugins[xvid_enc_create.num_plugins].func = xvid_plugin_2pass2; |
plugins[xvid_enc_create.num_plugins].func = xvid_plugin_2pass2; |
815 |
plugins[xvid_enc_create.num_plugins].param = &rc2pass2; |
plugins[xvid_enc_create.num_plugins].param = &rc2pass2; |
816 |
xvid_enc_create.num_plugins++; |
xvid_enc_create.num_plugins++; |
817 |
} else if (ARG_PASS1) { |
} |
818 |
rc2pass1.version = XVID_VERSION; |
|
819 |
|
if (ARG_PASS1) { |
820 |
memset(&rc2pass1, 0, sizeof(xvid_plugin_2pass1_t)); |
memset(&rc2pass1, 0, sizeof(xvid_plugin_2pass1_t)); |
821 |
|
rc2pass1.version = XVID_VERSION; |
822 |
rc2pass1.filename = ARG_PASS1; |
rc2pass1.filename = ARG_PASS1; |
823 |
|
|
824 |
plugins[xvid_enc_create.num_plugins].func = xvid_plugin_2pass1; |
plugins[xvid_enc_create.num_plugins].func = xvid_plugin_2pass1; |
871 |
xvid_enc_create.global = 0; |
xvid_enc_create.global = 0; |
872 |
|
|
873 |
if (ARG_PACKED) |
if (ARG_PACKED) |
874 |
xvid_enc_create.global |=XVID_PACKED; |
xvid_enc_create.global |=XVID_GLOBAL_PACKED; |
875 |
|
|
876 |
if (ARG_STATS) |
if (ARG_STATS) |
877 |
xvid_enc_create.global |=XVID_EXTRASTATS_ENABLE; |
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]; |