LRESULT decompress(CODEC * codec, ICDECOMPRESS * icd)
{
xvid_dec_frame_t frame;
/* --- yv12 --- */
if (icd->lpbiInput->biCompression != FOURCC_XVID &&
icd->lpbiInput->biCompression != FOURCC_DIVX &&
icd->lpbiInput->biCompression != FOURCC_DX50 &&
icd->lpbiInput->biCompression != FOURCC_MP4V &&
icd->lpbiInput->biCompression != FOURCC_xvid &&
icd->lpbiInput->biCompression != FOURCC_divx &&
icd->lpbiInput->biCompression != FOURCC_dx50 &&
icd->lpbiInput->biCompression != FOURCC_mp4v)
{
xvid_gbl_convert_t convert;
DPRINTF("input=%c%c%c%c output=%c%c%c%c",
icd->lpbiInput->biCompression&0xff,
(icd->lpbiInput->biCompression>>8)&0xff,
(icd->lpbiInput->biCompression>>16)&0xff,
(icd->lpbiInput->biCompression>>24)&0xff,
icd->lpbiOutput->biCompression&0xff,
(icd->lpbiOutput->biCompression>>8)&0xff,
(icd->lpbiOutput->biCompression>>16)&0xff,
(icd->lpbiOutput->biCompression>>24)&0xff);
memset(&convert, 0, sizeof(convert));
convert.version = XVID_VERSION;
convert.input.csp = get_colorspace(icd->lpbiInput);
convert.input.plane[0] = icd->lpInput;
convert.input.stride[0] = CALC_BI_STRIDE(icd->lpbiInput->biWidth, icd->lpbiInput->biBitCount);
if (convert.input.csp == XVID_CSP_I420 || convert.input.csp == XVID_CSP_YV12)
convert.input.stride[0] = (convert.input.stride[0]*2)/3;
convert.output.csp = get_colorspace(icd->lpbiOutput);
convert.output.plane[0] = icd->lpOutput;
convert.output.stride[0] = CALC_BI_STRIDE(icd->lpbiOutput->biWidth, icd->lpbiOutput->biBitCount);
if (convert.output.csp == XVID_CSP_I420 || convert.output.csp == XVID_CSP_YV12)
convert.output.stride[0] = (convert.output.stride[0]*2)/3;
convert.width = icd->lpbiInput->biWidth;
convert.height = icd->lpbiInput->biHeight;
convert.interlacing = 0;
if (convert.input.csp == XVID_CSP_NULL ||
convert.output.csp == XVID_CSP_NULL ||
codec->xvid_global_func(0, XVID_GBL_CONVERT, &convert, NULL) < 0)
{
return ICERR_BADFORMAT;
}
return ICERR_OK;
}
LRESULT decompress_get_format(CODEC * codec, BITMAPINFO * lpbiInput, BITMAPINFO * lpbiOutput)
{
BITMAPINFOHEADER * inhdr = &lpbiInput->bmiHeader;
BITMAPINFOHEADER * outhdr = &lpbiOutput->bmiHeader;
LRESULT result;
if (lpbiOutput == NULL)
{
return sizeof(BITMAPINFOHEADER);
}
/* --- yv12 --- */
if (get_colorspace(inhdr) != XVID_CSP_NULL) {
memcpy(outhdr, inhdr, sizeof(BITMAPINFOHEADER));
/* XXX: should we set outhdr->biSize ?? */
return ICERR_OK;
}
/* --- yv12 --- */
result = decompress_query(codec, lpbiInput, lpbiOutput);
if (result != ICERR_OK)
{
return result;
}
outhdr->biSize = sizeof(BITMAPINFOHEADER);
outhdr->biWidth = inhdr->biWidth;
outhdr->biHeight = inhdr->biHeight;
outhdr->biPlanes = 1;
outhdr->biBitCount = 24;
outhdr->biCompression = BI_RGB; /* sonic foundry vegas video v3 only supports BI_RGB */
outhdr->biSizeImage = outhdr->biHeight * CALC_BI_STRIDE(outhdr->biWidth, outhdr->biBitCount);
outhdr->biXPelsPerMeter = 0;
outhdr->biYPelsPerMeter = 0;
outhdr->biClrUsed = 0;
outhdr->biClrImportant = 0;
return ICERR_OK;
}
LRESULT compress(CODEC * codec, ICCOMPRESS * icc)
{
BITMAPINFOHEADER * inhdr = icc->lpbiInput;
BITMAPINFOHEADER * outhdr = icc->lpbiOutput;
xvid_enc_frame_t frame;
xvid_enc_stats_t stats;
int length;
const quality_t* quality_preset = (codec->config.quality==quality_table_num) ?
&codec->config.quality_user : &quality_table[codec->config.quality];
memset(&frame, 0, sizeof(frame));
frame.version = XVID_VERSION;
frame.type = XVID_TYPE_AUTO;
/* vol stuff */
if ((profiles[codec->config.profile].flags & PROFILE_MPEGQUANT) &&
codec->config.quant_type != QUANT_MODE_H263)
{
frame.vol_flags |= XVID_VOL_MPEGQUANT;
if (codec->config.quant_type == QUANT_MODE_CUSTOM) {
frame.quant_intra_matrix = codec->config.qmatrix_intra;
frame.quant_inter_matrix = codec->config.qmatrix_inter;
}else{
frame.quant_intra_matrix = NULL;
frame.quant_inter_matrix = NULL;
}
}
if ((profiles[codec->config.profile].flags & PROFILE_QPEL) && codec->config.qpel) {
frame.vol_flags |= XVID_VOL_QUARTERPEL;
frame.motion |= XVID_ME_QUARTERPELREFINE16 | XVID_ME_QUARTERPELREFINE8;
}
if ((profiles[codec->config.profile].flags & PROFILE_GMC) && codec->config.gmc) {
frame.vol_flags |= XVID_VOL_GMC;
frame.motion |= XVID_ME_GME_REFINE;
}
if ((profiles[codec->config.profile].flags & PROFILE_INTERLACE) && codec->config.interlacing)
frame.vol_flags |= XVID_VOL_INTERLACING;
/* dxn: force 1:1 picture aspect ration */
if ((profiles[codec->config.profile].flags & PROFILE_EXTRA)) {
frame.par = XVID_PAR_11_VGA;
} else if (codec->config.ar_mode == 0) { /* PAR */
if (codec->config.display_aspect_ratio != 5) {
frame.par = codec->config.display_aspect_ratio + 1;
} else {
frame.par = XVID_PAR_EXT;
frame.par_width = codec->config.par_x;
frame.par_height= codec->config.par_y;
}
} else { /* AR */
/* custom pixel aspect ratio -> calculated from DAR */
frame.par = XVID_PAR_EXT;
frame.par_width = (100 * inhdr->biHeight) / codec->config.ar_y;
frame.par_height= (100 * inhdr->biWidth) / codec->config.ar_x;
}
/* vop stuff */
frame.vop_flags |= XVID_VOP_HALFPEL;
frame.vop_flags |= XVID_VOP_HQACPRED;
if (codec->config.interlacing && codec->config.tff)
frame.vop_flags |= XVID_VOP_TOPFIELDFIRST;
if (codec->config.vop_debug)
frame.vop_flags |= XVID_VOP_DEBUG;
if (quality_preset->trellis_quant) {
frame.vop_flags |= XVID_VOP_TRELLISQUANT;
}
if ((profiles[codec->config.profile].flags & PROFILE_4MV)) {
if (quality_preset->motion_search > 4)
frame.vop_flags |= XVID_VOP_INTER4V;
}
if (quality_preset->chromame)
frame.motion |= XVID_ME_CHROMA_PVOP + XVID_ME_CHROMA_BVOP;
if (quality_preset->turbo)
frame.motion |= XVID_ME_FASTREFINE16 | XVID_ME_FASTREFINE8 |
XVID_ME_SKIP_DELTASEARCH | XVID_ME_FAST_MODEINTERPOLATE |
XVID_ME_BFRAME_EARLYSTOP;
frame.motion |= pmvfast_presets[quality_preset->motion_search];
if (quality_preset->vhq_bframe) frame.vop_flags |= XVID_VOP_RD_BVOP;
switch (quality_preset->vhq_mode)
{
case VHQ_MODE_DECISION :
frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
break;
case VHQ_LIMITED_SEARCH :
frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
frame.motion |= XVID_ME_HALFPELREFINE16_RD;
frame.motion |= XVID_ME_QUARTERPELREFINE16_RD;
break;
case VHQ_MEDIUM_SEARCH :
frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
frame.motion |= XVID_ME_HALFPELREFINE16_RD;
frame.motion |= XVID_ME_HALFPELREFINE8_RD;
frame.motion |= XVID_ME_QUARTERPELREFINE16_RD;
frame.motion |= XVID_ME_QUARTERPELREFINE8_RD;
frame.motion |= XVID_ME_CHECKPREDICTION_RD;
break;
case VHQ_WIDE_SEARCH :
frame.vop_flags |= XVID_VOP_MODEDECISION_RD;
frame.motion |= XVID_ME_HALFPELREFINE16_RD;
frame.motion |= XVID_ME_HALFPELREFINE8_RD;
frame.motion |= XVID_ME_QUARTERPELREFINE16_RD;
frame.motion |= XVID_ME_QUARTERPELREFINE8_RD;
frame.motion |= XVID_ME_CHECKPREDICTION_RD;
frame.motion |= XVID_ME_EXTSEARCH_RD;
break;
default :
break;
}
frame.input.plane[0] = icc->lpInput;
frame.input.stride[0] = CALC_BI_STRIDE(icc->lpbiInput->biWidth, icc->lpbiInput->biBitCount);
if ((frame.input.csp = get_colorspace(inhdr)) == XVID_CSP_NULL)
return ICERR_BADFORMAT;
if (frame.input.csp == XVID_CSP_I420 || frame.input.csp == XVID_CSP_YV12) {
frame.input.stride[0] = (4 * icc->lpbiInput->biWidth + 3) / 4;
frame.input.stride[1] = frame.input.stride[2] = frame.input.stride[0] / 2 ;
}
frame.bitstream = icc->lpOutput;
frame.length = icc->lpbiOutput->biSizeImage;
frame.quant = 0;
if (codec->config.mode == RC_MODE_NULL) {
outhdr->biSizeImage = 0;
*icc->lpdwFlags = AVIIF_KEYFRAME;
return ICERR_OK;
}
// force keyframe spacing in 2-pass 1st pass
if (quality_preset->motion_search == 0)
frame.type = XVID_TYPE_IVOP;
/* frame-based stuff */
apply_zone_modifiers(&frame, &codec->config, codec->framenum);
/* call encore */
memset(&stats, 0, sizeof(stats));
stats.version = XVID_VERSION;
length = codec->xvid_encore_func(codec->ehandle, XVID_ENC_ENCODE, &frame, &stats);
switch (length)
{
case XVID_ERR_FAIL :
return ICERR_ERROR;
case XVID_ERR_MEMORY :
return ICERR_MEMORY;
case XVID_ERR_FORMAT :
return ICERR_BADFORMAT;
case XVID_ERR_VERSION :
return ICERR_UNSUPPORTED;
}
if (codec->config.display_status && stats.type>0) {
status_update(&codec->status, stats.type, stats.length, stats.quant);
}
DPRINTF("{type=%i len=%i} length=%i", stats.type, stats.length, length);
if (length == 0) /* no encoder output */
{
*icc->lpdwFlags = 0;
((char*)icc->lpOutput)[0] = 0x7f; /* virtual dub skip frame */
outhdr->biSizeImage = 1;
}else{
if (frame.out_flags & XVID_KEYFRAME)
{
codec->keyspacing = 0;
*icc->lpdwFlags = AVIIF_KEYFRAME;
}
else
{
*icc->lpdwFlags = 0;
}
outhdr->biSizeImage = length;
if (codec->config.mode == RC_MODE_2PASS1 && codec->config.discard1pass)
{
outhdr->biSizeImage = 0;
}
}
codec->framenum++;
codec->keyspacing++;
return ICERR_OK;
}
HRESULT CXvidDecoder::ChangeColorspace(GUID subtype, GUID formattype, void * format)
{
DWORD biWidth;
if (formattype == FORMAT_VideoInfo)
{
VIDEOINFOHEADER * vih = (VIDEOINFOHEADER * )format;
biWidth = vih->bmiHeader.biWidth;
m_frame.output.stride[0] = CALC_BI_STRIDE(vih->bmiHeader.biWidth, vih->bmiHeader.biBitCount);
rgb_flip = (vih->bmiHeader.biHeight < 0 ? 0 : XVID_CSP_VFLIP);
}
else if (formattype == FORMAT_VideoInfo2)
{
VIDEOINFOHEADER2 * vih2 = (VIDEOINFOHEADER2 * )format;
biWidth = vih2->bmiHeader.biWidth;
m_frame.output.stride[0] = CALC_BI_STRIDE(vih2->bmiHeader.biWidth, vih2->bmiHeader.biBitCount);
rgb_flip = (vih2->bmiHeader.biHeight < 0 ? 0 : XVID_CSP_VFLIP);
}
else
{
return S_FALSE;
}
if (subtype == CLSID_MEDIASUBTYPE_IYUV)
{
DPRINTF("IYUV");
rgb_flip = 0;
m_frame.output.csp = XVID_CSP_I420;
m_frame.output.stride[0] = CALC_BI_STRIDE(biWidth, 8); /* planar format fix */
}
else if (subtype == MEDIASUBTYPE_YV12)
{
DPRINTF("YV12");
rgb_flip = 0;
m_frame.output.csp = XVID_CSP_YV12;
m_frame.output.stride[0] = CALC_BI_STRIDE(biWidth, 8); /* planar format fix */
}
else if (subtype == MEDIASUBTYPE_YUY2)
{
DPRINTF("YUY2");
rgb_flip = 0;
m_frame.output.csp = XVID_CSP_YUY2;
}
else if (subtype == MEDIASUBTYPE_YVYU)
{
DPRINTF("YVYU");
rgb_flip = 0;
m_frame.output.csp = XVID_CSP_YVYU;
}
else if (subtype == MEDIASUBTYPE_UYVY)
{
DPRINTF("UYVY");
rgb_flip = 0;
m_frame.output.csp = XVID_CSP_UYVY;
}
else if (subtype == MEDIASUBTYPE_RGB32)
{
DPRINTF("RGB32");
m_frame.output.csp = rgb_flip | XVID_CSP_BGRA;
}
else if (subtype == MEDIASUBTYPE_RGB24)
{
DPRINTF("RGB24");
m_frame.output.csp = rgb_flip | XVID_CSP_BGR;
}
else if (subtype == MEDIASUBTYPE_RGB555)
{
DPRINTF("RGB555");
m_frame.output.csp = rgb_flip | XVID_CSP_RGB555;
}
else if (subtype == MEDIASUBTYPE_RGB565)
{
DPRINTF("RGB565");
m_frame.output.csp = rgb_flip | XVID_CSP_RGB565;
}
else if (subtype == GUID_NULL)
{
m_frame.output.csp = XVID_CSP_NULL;
}
else
{
return S_FALSE;
}
return S_OK;
}
