23 |
* |
* |
24 |
* History: |
* History: |
25 |
* |
* |
26 |
|
* 12.07.2002 num_threads |
27 |
|
* 23.06.2002 XVID_CPU_CHKONLY; loading speed up |
28 |
* 25.04.2002 ICDECOMPRESS_PREROLL |
* 25.04.2002 ICDECOMPRESS_PREROLL |
29 |
* 17.04.2002 re-enabled lumi masking for 1st pass |
* 17.04.2002 re-enabled lumi masking for 1st pass |
30 |
* 15.04.2002 updated cbr support |
* 15.04.2002 updated cbr support |
43 |
* motion search precision = 0 now effective in 2-pass |
* motion search precision = 0 now effective in 2-pass |
44 |
* modulated quantization |
* modulated quantization |
45 |
* added DX50 fourcc |
* added DX50 fourcc |
46 |
* 01.12.2001 inital version; (c)2001 peter ross <suxen_drol@hotmail.com> |
* 01.12.2001 inital version; (c)2001 peter ross <pross@xvid.org> |
47 |
* |
* |
48 |
*************************************************************************/ |
*************************************************************************/ |
49 |
|
|
54 |
#include "2pass.h" |
#include "2pass.h" |
55 |
|
|
56 |
int pmvfast_presets[7] = { |
int pmvfast_presets[7] = { |
57 |
0, PMV_QUICKSTOP16, PMV_EARLYSTOP16, PMV_EARLYSTOP16 | PMV_EARLYSTOP8, |
0, PMV_QUICKSTOP16, 0, 0, |
58 |
PMV_EARLYSTOP16 | PMV_HALFPELREFINE16 | PMV_EARLYSTOP8 | PMV_HALFPELDIAMOND8, |
0 | PMV_HALFPELREFINE16 | PMV_HALFPELDIAMOND8, |
59 |
PMV_EARLYSTOP16 | PMV_HALFPELREFINE16 | PMV_EARLYSTOP8 | PMV_HALFPELDIAMOND8, |
0 | PMV_HALFPELREFINE16 | PMV_HALFPELDIAMOND8 | |
60 |
PMV_EARLYSTOP16 | PMV_HALFPELREFINE16 | PMV_EXTSEARCH16 | |
PMV_ADVANCEDDIAMOND16, PMV_HALFPELREFINE16 | PMV_EXTSEARCH16 | |
61 |
PMV_EARLYSTOP8 | PMV_HALFPELREFINE8 | PMV_HALFPELDIAMOND8 |
PMV_HALFPELREFINE8 | PMV_HALFPELDIAMOND8 | PMV_USESQUARES16 |
62 |
}; |
}; |
63 |
|
|
64 |
/* return xvid compatbile colorspace, |
/* return xvid compatbile colorspace, |
67 |
|
|
68 |
int get_colorspace(BITMAPINFOHEADER * hdr) |
int get_colorspace(BITMAPINFOHEADER * hdr) |
69 |
{ |
{ |
70 |
if (hdr->biHeight < 0) |
/* rgb only: negative height specifies top down image */ |
71 |
{ |
int rgb_flip = (hdr->biHeight < 0 ? 0 : XVID_CSP_VFLIP); |
|
DEBUGERR("colorspace: inverted input format not supported"); |
|
|
return XVID_CSP_NULL; |
|
|
} |
|
72 |
|
|
73 |
switch(hdr->biCompression) |
switch(hdr->biCompression) |
74 |
{ |
{ |
76 |
if (hdr->biBitCount == 16) |
if (hdr->biBitCount == 16) |
77 |
{ |
{ |
78 |
DEBUG("RGB16 (RGB555)"); |
DEBUG("RGB16 (RGB555)"); |
79 |
return XVID_CSP_VFLIP | XVID_CSP_RGB555; |
return rgb_flip | XVID_CSP_RGB555; |
80 |
} |
} |
81 |
if (hdr->biBitCount == 24) |
if (hdr->biBitCount == 24) |
82 |
{ |
{ |
83 |
DEBUG("RGB24"); |
DEBUG("RGB24"); |
84 |
return XVID_CSP_VFLIP | XVID_CSP_RGB24; |
return rgb_flip | XVID_CSP_RGB24; |
85 |
} |
} |
86 |
if (hdr->biBitCount == 32) |
if (hdr->biBitCount == 32) |
87 |
{ |
{ |
88 |
DEBUG("RGB32"); |
DEBUG("RGB32"); |
89 |
return XVID_CSP_VFLIP | XVID_CSP_RGB32; |
return rgb_flip | XVID_CSP_RGB32; |
90 |
} |
} |
91 |
|
|
92 |
DEBUG1("BI_RGB unsupported", hdr->biBitCount); |
DEBUG1("unsupported BI_RGB biBitCount", hdr->biBitCount); |
93 |
return XVID_CSP_NULL; |
return XVID_CSP_NULL; |
94 |
|
|
95 |
// how do these work in BITMAPINFOHEADER ??? |
case BI_BITFIELDS : |
96 |
/* case BI_BITFIELDS : |
if (hdr->biSize >= sizeof(BITMAPV4HEADER)) |
97 |
if (hdr->biBitCount == 16 |
{ |
98 |
if(hdr->biBitCount == 16 && |
BITMAPV4HEADER * hdr4 = (BITMAPV4HEADER *)hdr; |
99 |
hdr->bV4RedMask == 0x7c00 && |
|
100 |
hdr->bV4GreenMask == 0x3e0 && |
if (hdr4->bV4BitCount == 16 && |
101 |
hdr->bV4BlueMask == 0x1f) |
hdr4->bV4RedMask == 0x7c00 && |
102 |
|
hdr4->bV4GreenMask == 0x3e0 && |
103 |
|
hdr4->bV4BlueMask == 0x1f) |
104 |
{ |
{ |
105 |
DEBUG("RGB555"); |
DEBUG("RGB555"); |
106 |
return XVID_CSP_VFLIP | XVID_CSP_RGB555; |
return rgb_flip | XVID_CSP_RGB555; |
107 |
} |
} |
108 |
if(hdr->bV4BitCount == 16 && |
|
109 |
hdr->bV4RedMask == 0xf800 && |
if(hdr4->bV4BitCount == 16 && |
110 |
hdr->bV4GreenMask == 0x7e0 && |
hdr4->bV4RedMask == 0xf800 && |
111 |
hdr->bV4BlueMask == 0x1f) |
hdr4->bV4GreenMask == 0x7e0 && |
112 |
|
hdr4->bV4BlueMask == 0x1f) |
113 |
{ |
{ |
114 |
DEBUG("RGB565"); |
DEBUG("RGB565"); |
115 |
return XVID_CSP_VFLIP | XVID_CSP_RGB565; |
return rgb_flip | XVID_CSP_RGB565; |
116 |
} |
} |
117 |
|
|
118 |
DEBUG1("BI_FIELDS unsupported", hdr->bV4BitCount); |
DEBUG("unsupported BI_BITFIELDS mode"); |
119 |
return XVID_CSP_NULL; |
return XVID_CSP_NULL; |
120 |
*/ |
} |
121 |
|
|
122 |
|
DEBUG("unsupported BI_BITFIELDS/BITMAPHEADER combination"); |
123 |
|
return XVID_CSP_NULL; |
124 |
|
|
125 |
case FOURCC_I420: |
case FOURCC_I420: |
126 |
case FOURCC_IYUV: |
case FOURCC_IYUV: |
127 |
DEBUG("IYUY"); |
DEBUG("IYUY"); |
133 |
|
|
134 |
case FOURCC_YUYV : |
case FOURCC_YUYV : |
135 |
case FOURCC_YUY2 : |
case FOURCC_YUY2 : |
|
case FOURCC_V422 : |
|
136 |
DEBUG("YUY2"); |
DEBUG("YUY2"); |
137 |
return XVID_CSP_YUY2; |
return XVID_CSP_YUY2; |
138 |
|
|
144 |
DEBUG("UYVY"); |
DEBUG("UYVY"); |
145 |
return XVID_CSP_UYVY; |
return XVID_CSP_UYVY; |
146 |
|
|
147 |
} |
default : |
148 |
DEBUGFOURCC("colorspace: unknown", hdr->biCompression); |
DEBUGFOURCC("unsupported colorspace", hdr->biCompression); |
149 |
return XVID_CSP_NULL; |
return XVID_CSP_NULL; |
150 |
} |
} |
151 |
|
} |
152 |
|
|
153 |
|
|
154 |
/* compressor */ |
/* compressor */ |
198 |
|
|
199 |
memcpy(outhdr, inhdr, sizeof(BITMAPINFOHEADER)); |
memcpy(outhdr, inhdr, sizeof(BITMAPINFOHEADER)); |
200 |
outhdr->biSize = sizeof(BITMAPINFOHEADER); |
outhdr->biSize = sizeof(BITMAPINFOHEADER); |
|
outhdr->biBitCount = 24; // or 16 |
|
201 |
outhdr->biSizeImage = compress_get_size(codec, lpbiInput, lpbiOutput); |
outhdr->biSizeImage = compress_get_size(codec, lpbiInput, lpbiOutput); |
202 |
outhdr->biXPelsPerMeter = 0; |
outhdr->biXPelsPerMeter = 0; |
203 |
outhdr->biYPelsPerMeter = 0; |
outhdr->biYPelsPerMeter = 0; |
223 |
|
|
224 |
LRESULT compress_get_size(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput) |
LRESULT compress_get_size(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput) |
225 |
{ |
{ |
226 |
return lpbiOutput->bmiHeader.biWidth * lpbiOutput->bmiHeader.biHeight * 3; |
return |
227 |
|
#ifdef BFRAMES |
228 |
|
2 * |
229 |
|
#endif |
230 |
|
lpbiOutput->bmiHeader.biWidth * lpbiOutput->bmiHeader.biHeight * 3; |
231 |
} |
} |
232 |
|
|
233 |
|
|
236 |
// DEBUG2("frate fscale", codec->frate, codec->fscale); |
// DEBUG2("frate fscale", codec->frate, codec->fscale); |
237 |
codec->fincr = icf->dwScale; |
codec->fincr = icf->dwScale; |
238 |
codec->fbase = icf->dwRate; |
codec->fbase = icf->dwRate; |
239 |
|
|
240 |
|
if (codec->fbase == 0 || codec->fincr == 0) |
241 |
|
{ |
242 |
|
codec->fbase = 25; |
243 |
|
codec->fincr = 1; |
244 |
|
} |
245 |
|
|
246 |
return ICERR_OK; |
return ICERR_OK; |
247 |
} |
} |
248 |
|
|
305 |
param.max_quantizer = codec->config.max_pquant; |
param.max_quantizer = codec->config.max_pquant; |
306 |
param.max_key_interval = codec->config.max_key_interval; |
param.max_key_interval = codec->config.max_key_interval; |
307 |
|
|
308 |
|
#ifdef _SMP |
309 |
|
param.num_threads = codec->config.num_threads; |
310 |
|
#endif |
311 |
|
|
312 |
|
#ifdef BFRAMES |
313 |
|
param.global = 0; |
314 |
|
if (codec->config.packed) param.global |= XVID_GLOBAL_PACKED; |
315 |
|
if (codec->config.dx50bvop) param.global |= XVID_GLOBAL_DX50BVOP; |
316 |
|
if (codec->config.debug) param.global |= XVID_GLOBAL_DEBUG; |
317 |
|
param.max_bframes = codec->config.max_bframes; |
318 |
|
param.bquant_ratio = codec->config.bquant_ratio; |
319 |
|
param.frame_drop_ratio = codec->config.frame_drop_ratio; |
320 |
|
#endif |
321 |
|
|
322 |
switch(xvid_encore(0, XVID_ENC_CREATE, ¶m, NULL)) |
switch(xvid_encore(0, XVID_ENC_CREATE, ¶m, NULL)) |
323 |
{ |
{ |
324 |
case XVID_ERR_FAIL : |
case XVID_ERR_FAIL : |
395 |
frame.intra = -1; |
frame.intra = -1; |
396 |
|
|
397 |
frame.general |= XVID_HALFPEL; |
frame.general |= XVID_HALFPEL; |
398 |
|
// frame.general |= XVID_ME_EPZS; |
399 |
|
|
400 |
|
|
401 |
if (codec->config.motion_search > 4) |
if (codec->config.motion_search > 4) |
402 |
frame.general |= XVID_INTER4V; |
frame.general |= XVID_INTER4V; |
407 |
if (codec->config.interlacing) |
if (codec->config.interlacing) |
408 |
frame.general |= XVID_INTERLACING; |
frame.general |= XVID_INTERLACING; |
409 |
|
|
410 |
|
|
411 |
|
|
412 |
|
// added by koepi for credits greyscale |
413 |
|
|
414 |
|
check_greyscale_mode(&codec->config, &frame, codec->framenum); |
415 |
|
|
416 |
|
// end of koepi's addition |
417 |
|
|
418 |
|
|
419 |
|
// fix 1pass modes/hinted MV by koepi |
420 |
|
if (codec->config.hinted_me && (codec->config.mode == DLG_MODE_CBR || codec->config.mode == DLG_MODE_VBR_QUAL || codec->config.mode == DLG_MODE_VBR_QUANT)) |
421 |
|
{ |
422 |
|
codec->config.hinted_me = 0; |
423 |
|
} |
424 |
|
// end of ugly hack |
425 |
|
|
426 |
if (codec->config.hinted_me && codec->config.mode == DLG_MODE_2PASS_1) |
if (codec->config.hinted_me && codec->config.mode == DLG_MODE_2PASS_1) |
427 |
{ |
{ |
428 |
frame.hint.hintstream = codec->twopass.hintstream; |
frame.hint.hintstream = codec->twopass.hintstream; |
538 |
frame.intra = 0; |
frame.intra = 0; |
539 |
} |
} |
540 |
|
|
541 |
|
#ifdef BFRAMES |
542 |
|
frame.bquant = 0; |
543 |
|
#endif |
544 |
|
|
545 |
|
// OutputDebugString(" "); |
546 |
switch (xvid_encore(codec->ehandle, XVID_ENC_ENCODE, &frame, &stats)) |
switch (xvid_encore(codec->ehandle, XVID_ENC_ENCODE, &frame, &stats)) |
547 |
{ |
{ |
548 |
case XVID_ERR_FAIL : |
case XVID_ERR_FAIL : |
755 |
case DLG_MODE_VBR_QUAL : |
case DLG_MODE_VBR_QUAL : |
756 |
if (codec_is_in_credits(&codec->config, codec->framenum)) |
if (codec_is_in_credits(&codec->config, codec->framenum)) |
757 |
{ |
{ |
758 |
|
// added by koepi for credits greyscale |
759 |
|
|
760 |
|
check_greyscale_mode(&codec->config, frame, codec->framenum); |
761 |
|
|
762 |
|
// end of koepi's addition |
763 |
|
|
764 |
switch (codec->config.credits_mode) |
switch (codec->config.credits_mode) |
765 |
{ |
{ |
766 |
case CREDITS_MODE_RATE : |
case CREDITS_MODE_RATE : |
778 |
} |
} |
779 |
else |
else |
780 |
{ |
{ |
781 |
|
// added by koepi for credits greyscale |
782 |
|
|
783 |
|
check_greyscale_mode(&codec->config, frame, codec->framenum); |
784 |
|
|
785 |
|
// end of koepi's addition |
786 |
|
|
787 |
frame->quant = codec_get_vbr_quant(&codec->config, codec->config.quality); |
frame->quant = codec_get_vbr_quant(&codec->config, codec->config.quality); |
788 |
} |
} |
789 |
return ICERR_OK; |
return ICERR_OK; |
791 |
case DLG_MODE_VBR_QUANT : |
case DLG_MODE_VBR_QUANT : |
792 |
if (codec_is_in_credits(&codec->config, codec->framenum)) |
if (codec_is_in_credits(&codec->config, codec->framenum)) |
793 |
{ |
{ |
794 |
|
// added by koepi for credits greyscale |
795 |
|
|
796 |
|
check_greyscale_mode(&codec->config, frame, codec->framenum); |
797 |
|
|
798 |
|
// end of koepi's addition |
799 |
|
|
800 |
switch (codec->config.credits_mode) |
switch (codec->config.credits_mode) |
801 |
{ |
{ |
802 |
case CREDITS_MODE_RATE : |
case CREDITS_MODE_RATE : |
816 |
} |
} |
817 |
else |
else |
818 |
{ |
{ |
819 |
|
// added by koepi for credits greyscale |
820 |
|
|
821 |
|
check_greyscale_mode(&codec->config, frame, codec->framenum); |
822 |
|
|
823 |
|
// end of koepi's addition |
824 |
|
|
825 |
frame->quant = codec->config.quant; |
frame->quant = codec->config.quant; |
826 |
} |
} |
827 |
return ICERR_OK; |
return ICERR_OK; |
828 |
|
|
829 |
case DLG_MODE_2PASS_1 : |
case DLG_MODE_2PASS_1 : |
830 |
|
// added by koepi for credits greyscale |
831 |
|
|
832 |
|
check_greyscale_mode(&codec->config, frame, codec->framenum); |
833 |
|
|
834 |
|
// end of koepi's addition |
835 |
|
|
836 |
if (codec->config.credits_mode == CREDITS_MODE_QUANT) |
if (codec->config.credits_mode == CREDITS_MODE_QUANT) |
837 |
{ |
{ |
838 |
if (codec_is_in_credits(&codec->config, codec->framenum)) |
if (codec_is_in_credits(&codec->config, codec->framenum)) |
923 |
return quant; |
return quant; |
924 |
} |
} |
925 |
|
|
926 |
|
// added by koepi for credits greyscale |
927 |
|
|
928 |
|
int check_greyscale_mode(CONFIG* config, XVID_ENC_FRAME* frame, int framenum) |
929 |
|
|
930 |
|
{ |
931 |
|
|
932 |
|
if ((codec_is_in_credits(config, framenum)) && (config->mode!=DLG_MODE_CBR)) |
933 |
|
|
934 |
|
{ |
935 |
|
|
936 |
|
if (config->credits_greyscale) |
937 |
|
|
938 |
|
{ |
939 |
|
|
940 |
|
if ((frame->general && XVID_GREYSCALE)) // use only if not already in greyscale |
941 |
|
|
942 |
|
frame->general |= XVID_GREYSCALE; |
943 |
|
|
944 |
|
} else { |
945 |
|
|
946 |
|
if (!(frame->general && XVID_GREYSCALE)) // if movie is in greyscale, switch back |
947 |
|
|
948 |
|
frame->general |= XVID_GREYSCALE; |
949 |
|
|
950 |
|
} |
951 |
|
|
952 |
|
} else { |
953 |
|
|
954 |
|
if (config->greyscale) |
955 |
|
|
956 |
|
{ |
957 |
|
|
958 |
|
if ((frame->general && XVID_GREYSCALE)) // use only if not already in greyscale |
959 |
|
|
960 |
|
frame->general |= XVID_GREYSCALE; |
961 |
|
|
962 |
|
} else { |
963 |
|
|
964 |
|
if (!(frame->general && XVID_GREYSCALE)) // if credits is in greyscale, switch back |
965 |
|
|
966 |
|
frame->general |= XVID_GREYSCALE; |
967 |
|
|
968 |
|
} |
969 |
|
|
970 |
|
} |
971 |
|
|
972 |
|
return 0; |
973 |
|
|
974 |
|
} |
975 |
|
|
976 |
|
// end of koepi's addition |
977 |
|
|