164 lines
3.7 KiB
C++
164 lines
3.7 KiB
C++
#include "mkv_mp4v_decoder.h"
|
|
#include "../Winamp/wa_ipc.h" // for YV12_PLANES
|
|
#include <mmsystem.h>
|
|
#include <assert.h>
|
|
|
|
int MKVDecoderCreator::CreateVideoDecoder(const char *codec_id, const nsmkv::TrackEntryData *track_entry_data, const nsmkv::VideoData *video_data, ifc_mkvvideodecoder **decoder)
|
|
{
|
|
if (!strcmp(codec_id, "V_MPEG4/ISO/ASP")
|
|
|| !strcmp(codec_id, "V_MPEG4/ISO/SP"))
|
|
{
|
|
VIDEOINFOHEADER header = {0, };
|
|
|
|
header.bmiHeader.biHeight = (LONG)video_data->pixel_height;
|
|
header.bmiHeader.biWidth = (LONG)video_data->pixel_width;
|
|
header.bmiHeader.biCompression = 'v4pm';
|
|
|
|
MFTDecoder *ctx = new MFTDecoder;
|
|
if (!ctx) {
|
|
return CREATEDECODER_FAILURE;
|
|
}
|
|
HRESULT hr = ctx->Open(&header);
|
|
if (FAILED(hr)) {
|
|
delete ctx;
|
|
return CREATEDECODER_FAILURE;
|
|
}
|
|
|
|
if (ctx)
|
|
{
|
|
if (track_entry_data->codec_private && track_entry_data->codec_private_len) {
|
|
// mkv stores headers up to first VOP in codec_private
|
|
hr = ctx->Feed(track_entry_data->codec_private, track_entry_data->codec_private_len, 0);
|
|
if (FAILED(hr)) {
|
|
delete ctx;
|
|
return CREATEDECODER_FAILURE;
|
|
}
|
|
}
|
|
*decoder = new MKVMP4V(ctx, video_data);
|
|
return CREATEDECODER_SUCCESS;
|
|
}
|
|
else
|
|
{
|
|
return CREATEDECODER_FAILURE;
|
|
}
|
|
}
|
|
else if (!strcmp(codec_id, "V_MS/VFW/FOURCC"))
|
|
{
|
|
if (track_entry_data->codec_private && track_entry_data->codec_private_len)
|
|
{
|
|
const BITMAPINFOHEADER *header = (const BITMAPINFOHEADER *)track_entry_data->codec_private;
|
|
if (header->biCompression == 'DIVX'
|
|
|| header->biCompression == '05XD')
|
|
{
|
|
if (track_entry_data->codec_private_len < 40) {
|
|
return CREATEDECODER_FAILURE;
|
|
}
|
|
|
|
VIDEOINFOHEADER video_header = {0, };
|
|
memcpy(&video_header.bmiHeader, header, 40);
|
|
assert(track_entry_data->codec_private_len == 40);
|
|
|
|
MFTDecoder *ctx = new MFTDecoder;
|
|
if (!ctx) {
|
|
return CREATEDECODER_FAILURE;
|
|
}
|
|
if (FAILED(ctx->Open(&video_header))) {
|
|
delete ctx;
|
|
return CREATEDECODER_FAILURE;
|
|
}
|
|
|
|
*decoder = new MKVMP4V(ctx, video_data);
|
|
return CREATEDECODER_SUCCESS;
|
|
}
|
|
}
|
|
return CREATEDECODER_NOT_MINE;
|
|
}
|
|
else
|
|
{
|
|
return CREATEDECODER_NOT_MINE;
|
|
}
|
|
}
|
|
|
|
|
|
#define CBCLASS MKVDecoderCreator
|
|
START_DISPATCH;
|
|
CB(CREATE_VIDEO_DECODER, CreateVideoDecoder)
|
|
END_DISPATCH;
|
|
#undef CBCLASS
|
|
|
|
MKVMP4V::MKVMP4V(MFTDecoder *decoder, const nsmkv::VideoData *video_data) : decoder(decoder), video_data(video_data)
|
|
{
|
|
|
|
}
|
|
|
|
MKVMP4V::~MKVMP4V()
|
|
{
|
|
delete decoder;
|
|
}
|
|
|
|
int MKVMP4V::GetOutputProperties(int *x, int *y, int *color_format, double *aspect_ratio)
|
|
{
|
|
UINT width, height;
|
|
bool local_flip=false;
|
|
if (SUCCEEDED(decoder->GetOutputFormat(&width, &height, &local_flip, aspect_ratio))) {
|
|
*x = width;
|
|
*y = height;
|
|
*color_format = '21VY';
|
|
return MKV_SUCCESS;
|
|
}
|
|
return MKV_FAILURE;
|
|
}
|
|
|
|
int MKVMP4V::DecodeBlock(const void *inputBuffer, size_t inputBufferBytes, uint64_t timestamp)
|
|
{
|
|
HRESULT hr;
|
|
hr=decoder->Feed(inputBuffer, inputBufferBytes, timestamp);
|
|
return MKV_SUCCESS;
|
|
}
|
|
|
|
void MKVMP4V::Flush()
|
|
{
|
|
if (decoder) {
|
|
decoder->Flush();
|
|
}
|
|
}
|
|
|
|
int MKVMP4V::GetPicture(void **data, void **decoder_data, uint64_t *timestamp)
|
|
{
|
|
if (SUCCEEDED(decoder->GetFrame((YV12_PLANES **)data, decoder_data, timestamp))) {
|
|
return MKV_SUCCESS;
|
|
} else {
|
|
return MKV_FAILURE;
|
|
}
|
|
}
|
|
|
|
void MKVMP4V::FreePicture(void *data, void *decoder_data)
|
|
{
|
|
if (decoder) {
|
|
decoder->FreeFrame((YV12_PLANES *)data, decoder_data);
|
|
}
|
|
}
|
|
|
|
void MKVMP4V::HurryUp(int state)
|
|
{
|
|
// if (decoder)
|
|
// MPEG4Video_HurryUp(decoder, state);
|
|
}
|
|
|
|
void MKVMP4V::Close()
|
|
{
|
|
delete this;
|
|
}
|
|
|
|
#define CBCLASS MKVMP4V
|
|
START_DISPATCH;
|
|
CB(GET_OUTPUT_PROPERTIES, GetOutputProperties)
|
|
CB(DECODE_BLOCK, DecodeBlock)
|
|
VCB(FLUSH, Flush)
|
|
CB(GET_PICTURE, GetPicture)
|
|
VCB(FREE_PICTURE, FreePicture)
|
|
VCB(HURRY_UP, HurryUp)
|
|
VCB(CLOSE, Close)
|
|
END_DISPATCH;
|
|
#undef CBCLASS
|