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_USESQUARES16 |
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 */ |
172 |
} |
} |
173 |
|
|
174 |
if (inhdr->biWidth != outhdr->biWidth || inhdr->biHeight != outhdr->biHeight || |
if (inhdr->biWidth != outhdr->biWidth || inhdr->biHeight != outhdr->biHeight || |
175 |
(outhdr->biCompression != FOURCC_XVID && outhdr->biCompression != FOURCC_DIVX)) |
(outhdr->biCompression != FOURCC_XVID && outhdr->biCompression != FOURCC_DIVX && outhdr->biCompression != FOURCC_DX50)) |
176 |
{ |
{ |
177 |
return ICERR_BADFORMAT; |
return ICERR_BADFORMAT; |
178 |
} |
} |
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; |
390 |
frame.general |= XVID_HALFPEL; |
frame.general |= XVID_HALFPEL; |
391 |
// frame.general |= XVID_ME_EPZS; |
// 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; |
396 |
|
|
400 |
if (codec->config.interlacing) |
if (codec->config.interlacing) |
401 |
frame.general |= XVID_INTERLACING; |
frame.general |= XVID_INTERLACING; |
402 |
|
|
403 |
|
|
404 |
|
|
405 |
// added by koepi for credits greyscale |
// added by koepi for credits greyscale |
406 |
|
|
407 |
check_greyscale_mode(&codec->config, &frame, codec->framenum); |
check_greyscale_mode(&codec->config, &frame, codec->framenum); |
408 |
|
|
409 |
// end of koepi's addition |
// end of koepi's addition |
410 |
|
|
411 |
|
|
412 |
// fix 1pass modes/hinted MV by koepi |
// fix 1pass modes/hinted MV by koepi |
413 |
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)) |
414 |
{ |
{ |
451 |
frame.motion = pmvfast_presets[codec->config.motion_search]; |
frame.motion = pmvfast_presets[codec->config.motion_search]; |
452 |
|
|
453 |
frame.image = icc->lpInput; |
frame.image = icc->lpInput; |
454 |
|
// dev-api-3 |
455 |
|
frame.stride = (((icc->lpbiInput->biWidth * icc->lpbiInput->biBitCount) + 31) & ~31) >> 3; |
456 |
|
|
457 |
if ((frame.colorspace = get_colorspace(inhdr)) == XVID_CSP_NULL) |
if ((frame.colorspace = get_colorspace(inhdr)) == XVID_CSP_NULL) |
458 |
return ICERR_BADFORMAT; |
return ICERR_BADFORMAT; |
611 |
return ICERR_ERROR; |
return ICERR_ERROR; |
612 |
} |
} |
613 |
|
|
614 |
if (inhdr->biCompression != FOURCC_XVID && inhdr->biCompression != FOURCC_DIVX) |
if (inhdr->biCompression != FOURCC_XVID && |
615 |
|
inhdr->biCompression != FOURCC_DIVX && |
616 |
|
inhdr->biCompression != FOURCC_DX50 && |
617 |
|
get_colorspace(inhdr) == XVID_CSP_NULL) |
618 |
{ |
{ |
619 |
return ICERR_BADFORMAT; |
return ICERR_BADFORMAT; |
620 |
} |
} |
646 |
return sizeof(BITMAPINFOHEADER); |
return sizeof(BITMAPINFOHEADER); |
647 |
} |
} |
648 |
|
|
649 |
|
/* --- yv12 --- */ |
650 |
|
if (get_colorspace(inhdr) != XVID_CSP_NULL) { |
651 |
|
memcpy(outhdr, inhdr, sizeof(BITMAPINFOHEADER)); |
652 |
|
// XXX: should we set outhdr->biSize ?? |
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 |
{ |
{ |
725 |
frame.length = icd->lpbiInput->biSizeImage; |
frame.length = icd->lpbiInput->biSizeImage; |
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: |
730 |
|
frame.stride = (((icd->lpbiOutput->biWidth * icd->lpbiOutput->biBitCount) + 31) & ~31) >> 3; |
731 |
|
|
732 |
|
/* --- yv12 --- */ |
733 |
|
if (icd->lpbiInput->biCompression != FOURCC_XVID && |
734 |
|
icd->lpbiInput->biCompression != FOURCC_DIVX && |
735 |
|
icd->lpbiInput->biCompression != FOURCC_DX50) |
736 |
|
{ |
737 |
|
XVID_INIT_CONVERTINFO convert; |
738 |
|
|
739 |
|
DEBUGFOURCC("input", icd->lpbiInput->biCompression); |
740 |
|
DEBUGFOURCC("output", icd->lpbiOutput->biCompression); |
741 |
|
|
742 |
|
convert.input.colorspace = get_colorspace(icd->lpbiInput); |
743 |
|
convert.input.y = icd->lpInput; |
744 |
|
convert.input.y_stride = (((icd->lpbiInput->biWidth * icd->lpbiInput->biBitCount) + 31) & ~31) >> 3; |
745 |
|
|
746 |
|
convert.output.colorspace = get_colorspace(icd->lpbiOutput); |
747 |
|
convert.output.y = icd->lpOutput; |
748 |
|
convert.output.y_stride = (((icd->lpbiOutput->biWidth * icd->lpbiOutput->biBitCount) + 31) & ~31) >> 3; |
749 |
|
|
750 |
|
convert.width = icd->lpbiInput->biWidth; |
751 |
|
convert.height = icd->lpbiInput->biHeight; |
752 |
|
convert.interlacing = 0; |
753 |
|
|
754 |
|
if (convert.input.colorspace == XVID_CSP_NULL || |
755 |
|
convert.output.colorspace == XVID_CSP_NULL || |
756 |
|
xvid_init(NULL, XVID_INIT_CONVERT, &convert, NULL) != XVID_ERR_OK) |
757 |
|
{ |
758 |
|
return ICERR_BADFORMAT; |
759 |
|
} |
760 |
|
|
761 |
|
return ICERR_OK; |
762 |
|
} |
763 |
|
/* --- yv12 --- */ |
764 |
|
|
765 |
|
|
766 |
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))) |
767 |
{ |
{ |
798 |
if (codec_is_in_credits(&codec->config, codec->framenum)) |
if (codec_is_in_credits(&codec->config, codec->framenum)) |
799 |
{ |
{ |
800 |
// added by koepi for credits greyscale |
// added by koepi for credits greyscale |
801 |
|
|
802 |
check_greyscale_mode(&codec->config, frame, codec->framenum); |
check_greyscale_mode(&codec->config, frame, codec->framenum); |
803 |
|
|
804 |
// end of koepi's addition |
// end of koepi's addition |
805 |
|
|
806 |
switch (codec->config.credits_mode) |
switch (codec->config.credits_mode) |
807 |
{ |
{ |
808 |
case CREDITS_MODE_RATE : |
case CREDITS_MODE_RATE : |
821 |
else |
else |
822 |
{ |
{ |
823 |
// added by koepi for credits greyscale |
// added by koepi for credits greyscale |
824 |
|
|
825 |
check_greyscale_mode(&codec->config, frame, codec->framenum); |
check_greyscale_mode(&codec->config, frame, codec->framenum); |
826 |
|
|
827 |
// end of koepi's addition |
// end of koepi's addition |
828 |
|
|
829 |
frame->quant = codec_get_vbr_quant(&codec->config, codec->config.quality); |
frame->quant = codec_get_vbr_quant(&codec->config, codec->config.quality); |
830 |
} |
} |
831 |
return ICERR_OK; |
return ICERR_OK; |
834 |
if (codec_is_in_credits(&codec->config, codec->framenum)) |
if (codec_is_in_credits(&codec->config, codec->framenum)) |
835 |
{ |
{ |
836 |
// added by koepi for credits greyscale |
// added by koepi for credits greyscale |
837 |
|
|
838 |
check_greyscale_mode(&codec->config, frame, codec->framenum); |
check_greyscale_mode(&codec->config, frame, codec->framenum); |
839 |
|
|
840 |
// end of koepi's addition |
// end of koepi's addition |
841 |
|
|
842 |
switch (codec->config.credits_mode) |
switch (codec->config.credits_mode) |
843 |
{ |
{ |
844 |
case CREDITS_MODE_RATE : |
case CREDITS_MODE_RATE : |
859 |
else |
else |
860 |
{ |
{ |
861 |
// added by koepi for credits greyscale |
// added by koepi for credits greyscale |
862 |
|
|
863 |
check_greyscale_mode(&codec->config, frame, codec->framenum); |
check_greyscale_mode(&codec->config, frame, codec->framenum); |
864 |
|
|
865 |
// end of koepi's addition |
// end of koepi's addition |
866 |
|
|
867 |
frame->quant = codec->config.quant; |
frame->quant = codec->config.quant; |
868 |
} |
} |
869 |
return ICERR_OK; |
return ICERR_OK; |
870 |
|
|
871 |
case DLG_MODE_2PASS_1 : |
case DLG_MODE_2PASS_1 : |
872 |
// added by koepi for credits greyscale |
// added by koepi for credits greyscale |
873 |
|
|
874 |
check_greyscale_mode(&codec->config, frame, codec->framenum); |
check_greyscale_mode(&codec->config, frame, codec->framenum); |
875 |
|
|
876 |
// end of koepi's addition |
// end of koepi's addition |
877 |
|
|
878 |
if (codec->config.credits_mode == CREDITS_MODE_QUANT) |
if (codec->config.credits_mode == CREDITS_MODE_QUANT) |
879 |
{ |
{ |
880 |
if (codec_is_in_credits(&codec->config, codec->framenum)) |
if (codec_is_in_credits(&codec->config, codec->framenum)) |
966 |
} |
} |
967 |
|
|
968 |
// added by koepi for credits greyscale |
// added by koepi for credits greyscale |
969 |
|
|
970 |
int check_greyscale_mode(CONFIG* config, XVID_ENC_FRAME* frame, int framenum) |
int check_greyscale_mode(CONFIG* config, XVID_ENC_FRAME* frame, int framenum) |
971 |
|
|
972 |
{ |
{ |
973 |
|
|
974 |
if ((codec_is_in_credits(config, framenum)) && (config->mode!=DLG_MODE_CBR)) |
if ((codec_is_in_credits(config, framenum)) && (config->mode!=DLG_MODE_CBR)) |
975 |
|
|
976 |
{ |
{ |
977 |
|
|
978 |
if (config->credits_greyscale) |
if (config->credits_greyscale) |
979 |
|
|
980 |
{ |
{ |
981 |
|
|
982 |
if ((frame->general && XVID_GREYSCALE)) // use only if not already in greyscale |
if ((frame->general && XVID_GREYSCALE)) // use only if not already in greyscale |
983 |
|
|
984 |
frame->general |= XVID_GREYSCALE; |
frame->general |= XVID_GREYSCALE; |
985 |
|
|
986 |
} else { |
} else { |
987 |
|
|
988 |
if (!(frame->general && XVID_GREYSCALE)) // if movie is in greyscale, switch back |
if (!(frame->general && XVID_GREYSCALE)) // if movie is in greyscale, switch back |
989 |
|
|
990 |
frame->general |= XVID_GREYSCALE; |
frame->general |= XVID_GREYSCALE; |
991 |
|
|
992 |
} |
} |
993 |
|
|
994 |
} else { |
} else { |
995 |
|
|
996 |
if (config->greyscale) |
if (config->greyscale) |
997 |
|
|
998 |
{ |
{ |
999 |
|
|
1000 |
if ((frame->general && XVID_GREYSCALE)) // use only if not already in greyscale |
if ((frame->general && XVID_GREYSCALE)) // use only if not already in greyscale |
1001 |
|
|
1002 |
frame->general |= XVID_GREYSCALE; |
frame->general |= XVID_GREYSCALE; |
1003 |
|
|
1004 |
} else { |
} else { |
1005 |
|
|
1006 |
if (!(frame->general && XVID_GREYSCALE)) // if credits is in greyscale, switch back |
if (!(frame->general && XVID_GREYSCALE)) // if credits is in greyscale, switch back |
1007 |
|
|
1008 |
frame->general |= XVID_GREYSCALE; |
frame->general |= XVID_GREYSCALE; |
1009 |
|
|
1010 |
} |
} |
1011 |
|
|
1012 |
} |
} |
1013 |
|
|
1014 |
return 0; |
return 0; |
1015 |
|
|
1016 |
} |
} |
1017 |
|
|
1018 |
// end of koepi's addition |
// end of koepi's addition |
1019 |
|
|