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_ADVANCEDDIAMOND16, 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; |
309 |
if (codec->config.debug) param.global |= XVID_GLOBAL_DEBUG; |
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.bquant_offset = codec->config.bquant_offset; |
313 |
param.frame_drop_ratio = codec->config.frame_drop_ratio; |
param.frame_drop_ratio = codec->config.frame_drop_ratio; |
314 |
#endif |
#endif |
315 |
|
|
401 |
if (codec->config.interlacing) |
if (codec->config.interlacing) |
402 |
frame.general |= XVID_INTERLACING; |
frame.general |= XVID_INTERLACING; |
403 |
|
|
404 |
|
if (codec->config.qpel) { |
405 |
|
frame.general |= XVID_QUARTERPEL; |
406 |
|
frame.motion |= PMV_QUARTERPELREFINE16 | PMV_QUARTERPELREFINE8; |
407 |
|
|
408 |
|
} |
409 |
|
|
410 |
|
if (codec->config.gmc) |
411 |
|
frame.general |= XVID_GMC; |
412 |
|
|
413 |
|
if (codec->config.chromame) |
414 |
|
frame.general |= XVID_ME_COLOUR; |
415 |
|
|
416 |
// added by koepi for credits greyscale |
// added by koepi for credits greyscale |
417 |
|
|
459 |
} |
} |
460 |
} |
} |
461 |
|
|
462 |
frame.motion = pmvfast_presets[codec->config.motion_search]; |
frame.motion |= pmvfast_presets[codec->config.motion_search]; |
463 |
|
|
464 |
frame.image = icc->lpInput; |
frame.image = icc->lpInput; |
465 |
|
// dev-api-3 |
466 |
|
frame.stride = (((icc->lpbiInput->biWidth * icc->lpbiInput->biBitCount) + 31) & ~31) >> 3; |
467 |
|
|
468 |
if ((frame.colorspace = get_colorspace(inhdr)) == XVID_CSP_NULL) |
if ((frame.colorspace = get_colorspace(inhdr)) == XVID_CSP_NULL) |
469 |
return ICERR_BADFORMAT; |
return ICERR_BADFORMAT; |
561 |
return ICERR_BADFORMAT; |
return ICERR_BADFORMAT; |
562 |
} |
} |
563 |
|
|
564 |
if (frame.intra) |
if (frame.intra==1) |
565 |
{ |
{ |
566 |
codec->keyspacing = 0; |
codec->keyspacing = 0; |
567 |
*icc->lpdwFlags = AVIIF_KEYFRAME; |
*icc->lpdwFlags = AVIIF_KEYFRAME; |
600 |
} |
} |
601 |
} |
} |
602 |
|
|
603 |
|
//quick fix for delayed frames |
604 |
|
// if (frame.intra != 5) |
605 |
codec_2pass_update(codec, &frame, &stats); |
codec_2pass_update(codec, &frame, &stats); |
606 |
|
|
607 |
++codec->framenum; |
++codec->framenum; |
624 |
return ICERR_ERROR; |
return ICERR_ERROR; |
625 |
} |
} |
626 |
|
|
627 |
if (inhdr->biCompression != FOURCC_XVID && inhdr->biCompression != FOURCC_DIVX) |
if (inhdr->biCompression != FOURCC_XVID && inhdr->biCompression != FOURCC_DIVX && inhdr->biCompression != FOURCC_DX50 && get_colorspace(inhdr) == XVID_CSP_NULL) |
628 |
{ |
{ |
629 |
return ICERR_BADFORMAT; |
return ICERR_BADFORMAT; |
630 |
} |
} |
656 |
return sizeof(BITMAPINFOHEADER); |
return sizeof(BITMAPINFOHEADER); |
657 |
} |
} |
658 |
|
|
659 |
|
/* --- yv12 --- */ |
660 |
|
|
661 |
|
if (get_colorspace(inhdr) != XVID_CSP_NULL) { |
662 |
|
memcpy(outhdr, inhdr, sizeof(BITMAPINFOHEADER)); |
663 |
|
// XXX: should we set outhdr->biSize ?? |
664 |
|
return ICERR_OK; |
665 |
|
} |
666 |
|
/* --- yv12 --- */ |
667 |
|
|
668 |
result = decompress_query(codec, lpbiInput, lpbiOutput); |
result = decompress_query(codec, lpbiInput, lpbiOutput); |
669 |
if (result != ICERR_OK) |
if (result != ICERR_OK) |
670 |
{ |
{ |
736 |
frame.length = icd->lpbiInput->biSizeImage; |
frame.length = icd->lpbiInput->biSizeImage; |
737 |
|
|
738 |
frame.image = icd->lpOutput; |
frame.image = icd->lpOutput; |
739 |
frame.stride = icd->lpbiOutput->biWidth; |
// frame.stride = icd->lpbiOutput->biWidth; |
740 |
|
// dev-api-3: |
741 |
|
frame.stride = (((icd->lpbiOutput->biWidth * icd->lpbiOutput->biBitCount) + 31) & ~31) >> 3; |
742 |
|
|
743 |
|
/* --- yv12 --- */ |
744 |
|
if (icd->lpbiInput->biCompression != FOURCC_XVID && |
745 |
|
icd->lpbiInput->biCompression != FOURCC_DIVX && |
746 |
|
icd->lpbiInput->biCompression != FOURCC_DX50) |
747 |
|
{ |
748 |
|
XVID_INIT_CONVERTINFO convert; |
749 |
|
DEBUGFOURCC("input", icd->lpbiInput->biCompression); |
750 |
|
DEBUGFOURCC("output", icd->lpbiOutput->biCompression); |
751 |
|
convert.input.colorspace = get_colorspace(icd->lpbiInput); |
752 |
|
convert.input.y = icd->lpInput; |
753 |
|
convert.input.y_stride = (((icd->lpbiInput->biWidth *icd->lpbiInput->biBitCount) + 31) & ~31) >> 3; |
754 |
|
convert.output.colorspace = get_colorspace(icd->lpbiOutput); |
755 |
|
convert.output.y = icd->lpOutput; |
756 |
|
convert.output.y_stride = (((icd->lpbiOutput->biWidth *icd->lpbiOutput->biBitCount) + 31) & ~31) >> 3; |
757 |
|
convert.width = icd->lpbiInput->biWidth; |
758 |
|
convert.height = icd->lpbiInput->biHeight; |
759 |
|
convert.interlacing = 0; |
760 |
|
if (convert.input.colorspace == XVID_CSP_NULL || |
761 |
|
convert.output.colorspace == XVID_CSP_NULL || |
762 |
|
xvid_init(NULL, XVID_INIT_CONVERT, &convert, NULL) != XVID_ERR_OK) |
763 |
|
{ |
764 |
|
return ICERR_BADFORMAT; |
765 |
|
} |
766 |
|
return ICERR_OK; |
767 |
|
} |
768 |
|
/* --- yv12 --- */ |
769 |
|
|
770 |
|
|
771 |
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))) |
772 |
{ |
{ |