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; |
298 |
param.max_quantizer = codec->config.max_pquant; |
param.max_quantizer = codec->config.max_pquant; |
299 |
param.max_key_interval = codec->config.max_key_interval; |
param.max_key_interval = codec->config.max_key_interval; |
300 |
|
|
301 |
|
#ifdef _SMP |
302 |
|
param.num_threads = codec->config.num_threads; |
303 |
|
#endif |
304 |
|
|
305 |
#ifdef BFRAMES |
#ifdef BFRAMES |
306 |
param.packed = codec->config.packed; |
param.global = 0; |
307 |
|
if (codec->config.packed) param.global |= XVID_GLOBAL_PACKED; |
308 |
|
if (codec->config.dx50bvop) param.global |= XVID_GLOBAL_DX50BVOP; |
309 |
|
if (codec->config.debug) param.global |= XVID_GLOBAL_DEBUG; |
310 |
param.max_bframes = codec->config.max_bframes; |
param.max_bframes = codec->config.max_bframes; |
311 |
param.bquant_ratio = codec->config.bquant_ratio; |
param.bquant_ratio = codec->config.bquant_ratio; |
312 |
|
param.frame_drop_ratio = codec->config.frame_drop_ratio; |
313 |
#endif |
#endif |
314 |
|
|
315 |
switch(xvid_encore(0, XVID_ENC_CREATE, ¶m, NULL)) |
switch(xvid_encore(0, XVID_ENC_CREATE, ¶m, NULL)) |
388 |
frame.intra = -1; |
frame.intra = -1; |
389 |
|
|
390 |
frame.general |= XVID_HALFPEL; |
frame.general |= XVID_HALFPEL; |
391 |
|
// frame.general |= XVID_ME_EPZS; |
392 |
|
|
393 |
|
|
394 |
if (codec->config.motion_search > 4) |
if (codec->config.motion_search > 4) |
395 |
frame.general |= XVID_INTER4V; |
frame.general |= XVID_INTER4V; |
399 |
|
|
400 |
if (codec->config.interlacing) |
if (codec->config.interlacing) |
401 |
frame.general |= XVID_INTERLACING; |
frame.general |= XVID_INTERLACING; |
402 |
|
|
403 |
|
|
404 |
|
// added by koepi for credits greyscale |
405 |
|
|
406 |
|
check_greyscale_mode(&codec->config, &frame, codec->framenum); |
407 |
|
|
408 |
|
// end of koepi's addition |
409 |
|
|
410 |
|
|
411 |
// fix 1pass modes/hinted MV by koepi |
// fix 1pass modes/hinted MV by koepi |
412 |
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)) |
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)) |
413 |
{ |
{ |
450 |
frame.motion = pmvfast_presets[codec->config.motion_search]; |
frame.motion = pmvfast_presets[codec->config.motion_search]; |
451 |
|
|
452 |
frame.image = icc->lpInput; |
frame.image = icc->lpInput; |
453 |
|
// dev-api-3 frame.stride = (((icc->lpbiInput->biWidth * icc->lpbiInput->biBitCount) + 31) & ~31) >> 3; |
454 |
|
|
455 |
if ((frame.colorspace = get_colorspace(inhdr)) == XVID_CSP_NULL) |
if ((frame.colorspace = get_colorspace(inhdr)) == XVID_CSP_NULL) |
456 |
return ICERR_BADFORMAT; |
return ICERR_BADFORMAT; |
531 |
frame.intra = 0; |
frame.intra = 0; |
532 |
} |
} |
533 |
|
|
534 |
OutputDebugString(" "); |
#ifdef BFRAMES |
535 |
|
frame.bquant = 0; |
536 |
|
#endif |
537 |
|
|
538 |
|
// OutputDebugString(" "); |
539 |
switch (xvid_encore(codec->ehandle, XVID_ENC_ENCODE, &frame, &stats)) |
switch (xvid_encore(codec->ehandle, XVID_ENC_ENCODE, &frame, &stats)) |
540 |
{ |
{ |
541 |
case XVID_ERR_FAIL : |
case XVID_ERR_FAIL : |
608 |
{ |
{ |
609 |
return ICERR_ERROR; |
return ICERR_ERROR; |
610 |
} |
} |
611 |
|
/* --- yv12 --- */ |
612 |
|
if (inhdr->biCompression == FOURCC_YV12) { |
613 |
|
return ICERR_OK; |
614 |
|
} |
615 |
|
/* --- yv12 --- */ |
616 |
|
|
617 |
if (inhdr->biCompression != FOURCC_XVID && inhdr->biCompression != FOURCC_DIVX) |
if (inhdr->biCompression != FOURCC_XVID && inhdr->biCompression != FOURCC_DIVX) |
618 |
{ |
{ |
646 |
return sizeof(BITMAPINFOHEADER); |
return sizeof(BITMAPINFOHEADER); |
647 |
} |
} |
648 |
|
|
649 |
|
/* --- yv12 --- */ |
650 |
|
|
651 |
|
if (lpbiInput->bmiHeader.biCompression == FOURCC_YV12) { |
652 |
|
memcpy(outhdr, inhdr, sizeof(BITMAPINFOHEADER)); |
653 |
|
return ICERR_OK; |
654 |
|
} |
655 |
|
/* --- yv12 --- */ |
656 |
|
|
657 |
result = decompress_query(codec, lpbiInput, lpbiOutput); |
result = decompress_query(codec, lpbiInput, lpbiOutput); |
658 |
if (result != ICERR_OK) |
if (result != ICERR_OK) |
659 |
{ |
{ |
726 |
|
|
727 |
frame.image = icd->lpOutput; |
frame.image = icd->lpOutput; |
728 |
frame.stride = icd->lpbiOutput->biWidth; |
frame.stride = icd->lpbiOutput->biWidth; |
729 |
|
// dev-api-3: frame.stride = (((icd->lpbiOutput->biWidth * icd->lpbiOutput->biBitCount) + 31) & ~31) >> 3; |
730 |
|
|
731 |
|
/* --- yv12 --- */ |
732 |
|
if (icd->lpbiInput->biCompression == FOURCC_YV12) { |
733 |
|
DEBUGFOURCC("output", icd->lpbiOutput->biCompression); |
734 |
|
if (icd->lpbiOutput->biCompression == FOURCC_YV12) { |
735 |
|
memcpy(frame.image,codec->dhandle,icd->lpbiInput->biSizeImage); |
736 |
|
} |
737 |
|
return ICERR_OK; |
738 |
|
} |
739 |
|
/* --- yv12 --- */ |
740 |
|
|
741 |
|
|
742 |
if (~((icd->dwFlags & ICDECOMPRESS_HURRYUP) | (icd->dwFlags & ICDECOMPRESS_UPDATE) | (icd->dwFlags & ICDECOMPRESS_PREROLL))) |
if (~((icd->dwFlags & ICDECOMPRESS_HURRYUP) | (icd->dwFlags & ICDECOMPRESS_UPDATE) | (icd->dwFlags & ICDECOMPRESS_PREROLL))) |
743 |
{ |
{ |
773 |
case DLG_MODE_VBR_QUAL : |
case DLG_MODE_VBR_QUAL : |
774 |
if (codec_is_in_credits(&codec->config, codec->framenum)) |
if (codec_is_in_credits(&codec->config, codec->framenum)) |
775 |
{ |
{ |
776 |
|
// added by koepi for credits greyscale |
777 |
|
|
778 |
|
check_greyscale_mode(&codec->config, frame, codec->framenum); |
779 |
|
|
780 |
|
// end of koepi's addition |
781 |
|
|
782 |
switch (codec->config.credits_mode) |
switch (codec->config.credits_mode) |
783 |
{ |
{ |
784 |
case CREDITS_MODE_RATE : |
case CREDITS_MODE_RATE : |
796 |
} |
} |
797 |
else |
else |
798 |
{ |
{ |
799 |
|
// added by koepi for credits greyscale |
800 |
|
|
801 |
|
check_greyscale_mode(&codec->config, frame, codec->framenum); |
802 |
|
|
803 |
|
// end of koepi's addition |
804 |
|
|
805 |
frame->quant = codec_get_vbr_quant(&codec->config, codec->config.quality); |
frame->quant = codec_get_vbr_quant(&codec->config, codec->config.quality); |
806 |
} |
} |
807 |
return ICERR_OK; |
return ICERR_OK; |
809 |
case DLG_MODE_VBR_QUANT : |
case DLG_MODE_VBR_QUANT : |
810 |
if (codec_is_in_credits(&codec->config, codec->framenum)) |
if (codec_is_in_credits(&codec->config, codec->framenum)) |
811 |
{ |
{ |
812 |
|
// added by koepi for credits greyscale |
813 |
|
|
814 |
|
check_greyscale_mode(&codec->config, frame, codec->framenum); |
815 |
|
|
816 |
|
// end of koepi's addition |
817 |
|
|
818 |
switch (codec->config.credits_mode) |
switch (codec->config.credits_mode) |
819 |
{ |
{ |
820 |
case CREDITS_MODE_RATE : |
case CREDITS_MODE_RATE : |
834 |
} |
} |
835 |
else |
else |
836 |
{ |
{ |
837 |
|
// added by koepi for credits greyscale |
838 |
|
|
839 |
|
check_greyscale_mode(&codec->config, frame, codec->framenum); |
840 |
|
|
841 |
|
// end of koepi's addition |
842 |
|
|
843 |
frame->quant = codec->config.quant; |
frame->quant = codec->config.quant; |
844 |
} |
} |
845 |
return ICERR_OK; |
return ICERR_OK; |
846 |
|
|
847 |
case DLG_MODE_2PASS_1 : |
case DLG_MODE_2PASS_1 : |
848 |
|
// added by koepi for credits greyscale |
849 |
|
|
850 |
|
check_greyscale_mode(&codec->config, frame, codec->framenum); |
851 |
|
|
852 |
|
// end of koepi's addition |
853 |
|
|
854 |
if (codec->config.credits_mode == CREDITS_MODE_QUANT) |
if (codec->config.credits_mode == CREDITS_MODE_QUANT) |
855 |
{ |
{ |
856 |
if (codec_is_in_credits(&codec->config, codec->framenum)) |
if (codec_is_in_credits(&codec->config, codec->framenum)) |
941 |
return quant; |
return quant; |
942 |
} |
} |
943 |
|
|
944 |
|
// added by koepi for credits greyscale |
945 |
|
|
946 |
|
int check_greyscale_mode(CONFIG* config, XVID_ENC_FRAME* frame, int framenum) |
947 |
|
|
948 |
|
{ |
949 |
|
|
950 |
|
if ((codec_is_in_credits(config, framenum)) && (config->mode!=DLG_MODE_CBR)) |
951 |
|
|
952 |
|
{ |
953 |
|
|
954 |
|
if (config->credits_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 movie is in greyscale, switch back |
965 |
|
|
966 |
|
frame->general |= XVID_GREYSCALE; |
967 |
|
|
968 |
|
} |
969 |
|
|
970 |
|
} else { |
971 |
|
|
972 |
|
if (config->greyscale) |
973 |
|
|
974 |
|
{ |
975 |
|
|
976 |
|
if ((frame->general && XVID_GREYSCALE)) // use only if not already in greyscale |
977 |
|
|
978 |
|
frame->general |= XVID_GREYSCALE; |
979 |
|
|
980 |
|
} else { |
981 |
|
|
982 |
|
if (!(frame->general && XVID_GREYSCALE)) // if credits is in greyscale, switch back |
983 |
|
|
984 |
|
frame->general |= XVID_GREYSCALE; |
985 |
|
|
986 |
|
} |
987 |
|
|
988 |
|
} |
989 |
|
|
990 |
|
return 0; |
991 |
|
|
992 |
|
} |
993 |
|
|
994 |
|
// end of koepi's addition |
995 |
|
|