mirror of
https://github.com/mpv-player/mpv
synced 2024-12-14 10:55:43 +00:00
9f4b0cfaef
git-svn-id: svn://svn.mplayerhq.hu/mplayer/trunk@1074 b3059339-0415-0410-9bf9-f77b7e298cf2
453 lines
12 KiB
C
453 lines
12 KiB
C
/********************************************************
|
|
|
|
DirectShow Video decoder implementation
|
|
Copyright 2000 Eugene Kuznetsov (divx@euro.ru)
|
|
Converted C++ --> C :) by A'rpi/ESP-team
|
|
|
|
*********************************************************/
|
|
|
|
//#include <config.h>
|
|
|
|
//#include "DS_VideoDecoder.h"
|
|
//#include <string.h>
|
|
#include <stdlib.h>
|
|
#include <except.h>
|
|
#define __MODULE__ "DirectShow_VideoDecoder"
|
|
|
|
|
|
#include <errno.h>
|
|
#ifdef HAVE_MALLOC_H
|
|
#include <malloc.h>
|
|
#endif
|
|
//#include <loader.h>
|
|
//#include <wine/winbase.h>
|
|
#include <stdio.h>
|
|
#include <unistd.h>
|
|
#include <fcntl.h>
|
|
#include <strstream>
|
|
#include <dlfcn.h>
|
|
#include <sys/types.h>
|
|
#include <sys/mman.h>
|
|
|
|
#include <registry.h>
|
|
#include <wine/winreg.h>
|
|
|
|
#include "guids.h"
|
|
#include "interfaces.h"
|
|
#include "DS_Filter.h"
|
|
|
|
#include "BitmapInfo.h"
|
|
|
|
#include <string>
|
|
#include <default.h>
|
|
|
|
#include "DS_VideoDec.h"
|
|
|
|
|
|
using namespace std;
|
|
extern "C" char* def_path;
|
|
|
|
static char** m_destptr=0;
|
|
|
|
static DS_Filter* dsf=0;
|
|
|
|
static AM_MEDIA_TYPE m_sOurType, m_sDestType;
|
|
static VIDEOINFOHEADER m_sVhdr;
|
|
static VIDEOINFOHEADER *m_sVhdr2;
|
|
static void* m_pCust;
|
|
|
|
static BITMAPINFOHEADER m_bh;//format of input data
|
|
static BitmapInfo m_decoder;//format of decoder output
|
|
static BitmapInfo m_obh; //format of returned frames
|
|
// CImage* m_outFrame;
|
|
|
|
// int m_iState=0;
|
|
|
|
extern "C" int DS_VideoDecoder_Open(char* dllname, GUID* guid, BITMAPINFOHEADER* format, int flip,char** d_ptr)
|
|
// :IVideoDecoder(info), m_sVhdr2(0)
|
|
{
|
|
|
|
m_destptr=d_ptr;
|
|
|
|
//m_outFrame=0;
|
|
//decpos = 0;
|
|
//playpos = 0;
|
|
//realtime = 0;
|
|
|
|
try
|
|
{
|
|
m_bh=*format;
|
|
memset(&m_obh, 0, sizeof(m_obh));
|
|
m_obh.biSize=sizeof(m_obh);
|
|
|
|
memset(&m_sVhdr, 0, sizeof(m_sVhdr));
|
|
m_sVhdr.bmiHeader=m_bh;
|
|
m_sVhdr.rcSource.left=m_sVhdr.rcSource.top=0;
|
|
m_sVhdr.rcSource.right=m_sVhdr.bmiHeader.biWidth;
|
|
m_sVhdr.rcSource.bottom=m_sVhdr.bmiHeader.biHeight;
|
|
m_sVhdr.rcTarget=m_sVhdr.rcSource;
|
|
m_sOurType.majortype=MEDIATYPE_Video;
|
|
|
|
m_sOurType.subtype=MEDIATYPE_Video;
|
|
m_sOurType.subtype.f1=m_sVhdr.bmiHeader.biCompression;
|
|
m_sOurType.formattype=FORMAT_VideoInfo;
|
|
m_sOurType.bFixedSizeSamples=false;
|
|
m_sOurType.bTemporalCompression=true;
|
|
m_sOurType.pUnk=0;
|
|
m_sOurType.cbFormat=sizeof(m_sVhdr);
|
|
m_sOurType.pbFormat=(char*)&m_sVhdr;
|
|
|
|
m_sVhdr2=(VIDEOINFOHEADER*)(new char[sizeof(VIDEOINFOHEADER)+12]);
|
|
*m_sVhdr2=m_sVhdr;
|
|
m_sVhdr2->bmiHeader.biCompression=0;
|
|
m_sVhdr2->bmiHeader.biBitCount=24;
|
|
|
|
memset(&m_sDestType, 0, sizeof(m_sDestType));
|
|
m_sDestType.majortype=MEDIATYPE_Video;
|
|
m_sDestType.subtype=MEDIASUBTYPE_RGB24;
|
|
m_sDestType.formattype=FORMAT_VideoInfo;
|
|
m_sDestType.bFixedSizeSamples=true;
|
|
m_sDestType.bTemporalCompression=false;
|
|
m_sDestType.lSampleSize=abs(m_sVhdr2->bmiHeader.biWidth*m_sVhdr2->bmiHeader.biHeight*
|
|
((m_sVhdr2->bmiHeader.biBitCount+7)/8));
|
|
m_sVhdr2->bmiHeader.biSizeImage=m_sDestType.lSampleSize;
|
|
m_sDestType.pUnk=0;
|
|
m_sDestType.cbFormat=sizeof(VIDEOINFOHEADER);
|
|
m_sDestType.pbFormat=(char*)m_sVhdr2;
|
|
|
|
m_obh=m_bh;
|
|
m_obh.setBits(24);
|
|
|
|
HRESULT result;
|
|
|
|
dsf=new DS_Filter();
|
|
dsf->Create(dllname, guid, &m_sOurType, &m_sDestType);
|
|
|
|
if(!flip)
|
|
{
|
|
m_sVhdr2->bmiHeader.biHeight*=-1;
|
|
m_obh.biHeight*=-1;
|
|
result=dsf->m_pOutputPin->vt->QueryAccept(dsf->m_pOutputPin, &m_sDestType);
|
|
if(result){
|
|
printf("DShow: Decoder does not support upside-down frames");
|
|
m_obh.biHeight*=-1;
|
|
}
|
|
}
|
|
|
|
#if 0
|
|
m_sVhdr2->bmiHeader.biBitCount=16;
|
|
m_sVhdr2->bmiHeader.biCompression=fccYUY2;
|
|
m_sDestType.subtype=MEDIASUBTYPE_YUY2;
|
|
result=dsf->m_pOutputPin->vt->QueryAccept(dsf->m_pOutputPin, &m_sDestType);
|
|
// if(!result) caps=(CAPS)(caps | CAP_YUY2);
|
|
#endif
|
|
|
|
m_sVhdr2->bmiHeader.biBitCount=24;
|
|
m_sVhdr2->bmiHeader.biCompression=0;
|
|
m_sDestType.subtype=MEDIASUBTYPE_RGB24;
|
|
m_decoder=m_obh;
|
|
//qual = 0-1;
|
|
}
|
|
catch(FatalError& error)
|
|
{
|
|
delete[] m_sVhdr2;
|
|
return 1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
extern "C" void DS_VideoDecoder_Start(){
|
|
if(dsf->m_iState!=1) return;
|
|
dsf->Start();
|
|
|
|
ALLOCATOR_PROPERTIES props, props1;
|
|
props.cBuffers=1;
|
|
props.cbBuffer=1024*1024; //m_sDestType.lSampleSize;//don't know how to do this correctly
|
|
props.cbAlign=props.cbPrefix=0;
|
|
dsf->m_pAll->vt->SetProperties(dsf->m_pAll, &props, &props1);
|
|
dsf->m_pAll->vt->Commit(dsf->m_pAll);
|
|
|
|
// m_outFrame=new CImage(&m_decoder,(unsigned char *)malloc(m_sDestType.lSampleSize),false);
|
|
//m_outFrame=new CImage(&m_decoder, 0, false);
|
|
// printf("Datap %x\n",m_outFrame->getaddr());
|
|
|
|
|
|
// dsf->m_pOurOutput->SetFramePointer((char **)m_outFrame->getaddr()); //!FIXME!
|
|
dsf->m_pOurOutput->SetFramePointer(m_destptr); //!FIXME!
|
|
|
|
// filling = realtime;
|
|
|
|
dsf->m_iState=2;
|
|
return;
|
|
}
|
|
|
|
extern "C" void DS_VideoDecoder_Stop(){
|
|
if(dsf->m_iState!=2) return;
|
|
dsf->Stop();
|
|
// dsf->m_pOurOutput->SetFramePointer(0);
|
|
// free(m_outFrame->data());
|
|
//m_outFrame->release();//just in case
|
|
//m_outFrame=0;
|
|
// FlushCache();
|
|
dsf->m_iState=1;
|
|
return;
|
|
}
|
|
|
|
extern "C" void DS_VideoDecoder_Restart(){
|
|
if(dsf->m_iState!=2) return;
|
|
|
|
dsf->Stop();
|
|
dsf->Start();
|
|
|
|
ALLOCATOR_PROPERTIES props, props1;
|
|
props.cBuffers=1;
|
|
props.cbBuffer=m_sDestType.lSampleSize;//don't know how to do this correctly
|
|
props.cbAlign=props.cbPrefix=0;
|
|
dsf->m_pAll->vt->SetProperties(dsf->m_pAll, &props, &props1);
|
|
dsf->m_pAll->vt->Commit(dsf->m_pAll);
|
|
}
|
|
|
|
extern "C" void DS_VideoDecoder_Close(){
|
|
if(dsf->m_iState==0) return;
|
|
if(dsf->m_iState==2) DS_VideoDecoder_Stop();
|
|
delete[] m_sVhdr2;
|
|
// delete m_outFrame;
|
|
}
|
|
|
|
extern "C" int DS_VideoDecoder_DecodeFrame(char* src, int size, int is_keyframe, int render){
|
|
|
|
if(!size)return 0;
|
|
|
|
m_bh.biSizeImage=size;
|
|
|
|
IMediaSample* sample=0;
|
|
//printf("GetBuffer... (m_pAll=%X) ",dsf->m_pAll);fflush(stdout);
|
|
dsf->m_pAll->vt->GetBuffer(dsf->m_pAll, &sample, 0, 0, 0);
|
|
//printf("OK!\n");
|
|
if(!sample)
|
|
{
|
|
Debug cerr<<"ERROR: null sample"<<endl;
|
|
return -1;
|
|
}
|
|
char* ptr;
|
|
//printf("GetPtr...");fflush(stdout);
|
|
sample->vt->GetPointer(sample, (BYTE **)&ptr);
|
|
//printf("OK!\n");
|
|
memcpy(ptr, src, size);
|
|
//printf("memcpy OK!\n");
|
|
sample->vt->SetActualDataLength(sample, size);
|
|
//printf("SetActualDataLength OK!\n");
|
|
sample->vt->SetSyncPoint(sample, is_keyframe);
|
|
//printf("SetSyncPoint OK!\n");
|
|
sample->vt->SetPreroll(sample, !render);
|
|
// sample->vt->SetMediaType(sample, &m_sOurType);
|
|
int result=dsf->m_pImp->vt->Receive(dsf->m_pImp, sample);
|
|
if(result)
|
|
printf("Error putting data into input pin %x\n", result);
|
|
|
|
sample->vt->Release((IUnknown*)sample);
|
|
|
|
return 0;
|
|
}
|
|
|
|
extern "C" int DS_VideoDecoder_SetDestFmt(int bits, int csp){
|
|
if(dsf->m_iState==0) return -1;
|
|
// if(!CImage::supported(csp, bits)) return -1;
|
|
HRESULT result;
|
|
// BitmapInfo temp=m_obh;
|
|
if(csp==0)
|
|
{
|
|
switch(bits)
|
|
{
|
|
case 15:
|
|
m_sDestType.subtype=MEDIASUBTYPE_RGB555;
|
|
break;
|
|
case 16:
|
|
m_sDestType.subtype=MEDIASUBTYPE_RGB565;
|
|
break;
|
|
case 24:
|
|
m_sDestType.subtype=MEDIASUBTYPE_RGB24;
|
|
break;
|
|
case 32:
|
|
m_sDestType.subtype=MEDIASUBTYPE_RGB32;
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
m_obh.setBits(bits);
|
|
// .biSizeImage=abs(temp.biWidth*temp.biHeight*((temp.biBitCount+7)/8));
|
|
}
|
|
else
|
|
{
|
|
m_obh.setSpace(csp,bits);
|
|
switch(csp)
|
|
{
|
|
case fccYUY2:
|
|
m_sDestType.subtype=MEDIASUBTYPE_YUY2;
|
|
printf("DShow: using YUY2 colorspace\n");
|
|
break;
|
|
case fccYV12:
|
|
m_sDestType.subtype=MEDIASUBTYPE_YV12;
|
|
printf("DShow: using YV12 colorspace\n");
|
|
break;
|
|
case fccIYUV:
|
|
m_sDestType.subtype=MEDIASUBTYPE_IYUV;
|
|
printf("DShow: using IYUV colorspace\n");
|
|
break;
|
|
case fccUYVY:
|
|
m_sDestType.subtype=MEDIASUBTYPE_UYVY;
|
|
printf("DShow: using UYVY colorspace\n");
|
|
break;
|
|
case fccYVYU:
|
|
m_sDestType.subtype=MEDIASUBTYPE_YVYU;
|
|
printf("DShow: using YVYU colorspace\n");
|
|
break;
|
|
}
|
|
}
|
|
|
|
m_sDestType.lSampleSize=m_obh.biSizeImage;
|
|
memcpy(&(m_sVhdr2->bmiHeader), &m_obh, sizeof(m_obh));
|
|
m_sVhdr2->bmiHeader.biSize=sizeof(BITMAPINFOHEADER);
|
|
if(m_sVhdr2->bmiHeader.biCompression==3)
|
|
m_sDestType.cbFormat=sizeof(VIDEOINFOHEADER)+12;
|
|
else
|
|
m_sDestType.cbFormat=sizeof(VIDEOINFOHEADER);
|
|
|
|
result=dsf->m_pOutputPin->vt->QueryAccept(dsf->m_pOutputPin, &m_sDestType);
|
|
|
|
if(result!=0)
|
|
{
|
|
if(csp)
|
|
cerr<<"Warning: unsupported color space"<<endl;
|
|
else
|
|
cerr<<"Warning: unsupported bit depth"<<endl;
|
|
|
|
m_sDestType.lSampleSize=m_decoder.biSizeImage;
|
|
memcpy(&(m_sVhdr2->bmiHeader), &m_decoder, sizeof(m_decoder));
|
|
m_sVhdr2->bmiHeader.biSize=sizeof(BITMAPINFOHEADER);
|
|
if(m_sVhdr2->bmiHeader.biCompression==3)
|
|
m_sDestType.cbFormat=sizeof(VIDEOINFOHEADER)+12;
|
|
else
|
|
m_sDestType.cbFormat=sizeof(VIDEOINFOHEADER);
|
|
return 1;
|
|
}
|
|
|
|
m_decoder=m_obh;
|
|
// m_obh=temp;
|
|
// if(csp)
|
|
// m_obh.biBitCount=BitmapInfo::BitCount(csp);
|
|
m_bh.biBitCount=bits;
|
|
if(dsf->m_iState>0)
|
|
{
|
|
int old_state=dsf->m_iState;
|
|
if(dsf->m_iState==2) DS_VideoDecoder_Stop();
|
|
dsf->m_pInputPin->vt->Disconnect(dsf->m_pInputPin);
|
|
dsf->m_pOutputPin->vt->Disconnect(dsf->m_pOutputPin);
|
|
dsf->m_pOurOutput->SetNewFormat(m_sDestType);
|
|
result=dsf->m_pInputPin->vt->ReceiveConnection(dsf->m_pInputPin, dsf->m_pOurInput, &m_sOurType);
|
|
if(result)
|
|
{
|
|
cerr<<"Error reconnecting input pin "<<hex<<result<<dec<<endl;
|
|
return -1;
|
|
}
|
|
result=dsf->m_pOutputPin->vt->ReceiveConnection(dsf->m_pOutputPin,
|
|
dsf->m_pOurOutput, &m_sDestType);
|
|
if(result)
|
|
{
|
|
cerr<<"Error reconnecting output pin "<<hex<<result<<dec<<endl;
|
|
return -1;
|
|
}
|
|
if(old_state==2) DS_VideoDecoder_Start();
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
|
|
extern "C" int DS_SetValue_DivX(char* name, int value){
|
|
int temp;
|
|
if(dsf->m_iState!=2) return VFW_E_NOT_RUNNING;
|
|
// brightness 87
|
|
// contrast 74
|
|
// hue 23
|
|
// saturation 20
|
|
// post process mode 0
|
|
// get1 0x01
|
|
// get2 10
|
|
// get3=set2 86
|
|
// get4=set3 73
|
|
// get5=set4 19
|
|
// get6=set5 23
|
|
printf("DivX setting: %s = %d\n",name,value);
|
|
|
|
IHidden* hidden=(IHidden*)((int)dsf->m_pFilter+0xb8);
|
|
if(strcmp(name, "Brightness")==0)
|
|
return hidden->vt->SetSmth2(hidden, value, 0);
|
|
if(strcmp(name, "Contrast")==0)
|
|
return hidden->vt->SetSmth3(hidden, value, 0);
|
|
if(strcmp(name, "Hue")==0)
|
|
return hidden->vt->SetSmth5(hidden, value, 0);
|
|
if(strcmp(name, "Saturation")==0)
|
|
return hidden->vt->SetSmth4(hidden, value, 0);
|
|
if(strcmp(name, "Quality")==0)
|
|
return hidden->vt->SetSmth(hidden, value, 0);
|
|
|
|
printf("Invalid setting!\n");
|
|
return -200;
|
|
}
|
|
|
|
extern "C" int DS_SetAttr_DivX(char* attribute, int value){
|
|
int result, status, newkey, count;
|
|
if(strcmp(attribute, "Quality")==0){
|
|
char* keyname="SOFTWARE\\Microsoft\\Scrunch";
|
|
result=RegCreateKeyExA(HKEY_CURRENT_USER, keyname, 0, 0, 0, 0, 0, &newkey, &status);
|
|
if(result!=0)
|
|
{
|
|
printf("VideoDecoder::SetExtAttr: registry failure\n");
|
|
return -1;
|
|
}
|
|
result=RegSetValueExA(newkey, "Current Post Process Mode", 0, REG_DWORD, &value, 4);
|
|
if(result!=0)
|
|
{
|
|
printf("VideoDecoder::SetExtAttr: error writing value\n");
|
|
return -1;
|
|
}
|
|
value=-1;
|
|
result=RegSetValueExA(newkey, "Force Post Process Mode", 0, REG_DWORD, &value, 4);
|
|
if(result!=0)
|
|
{
|
|
printf("VideoDecoder::SetExtAttr: error writing value\n");
|
|
return -1;
|
|
}
|
|
RegCloseKey(newkey);
|
|
return 0;
|
|
}
|
|
|
|
if(
|
|
(strcmp(attribute, "Saturation")==0) ||
|
|
(strcmp(attribute, "Hue")==0) ||
|
|
(strcmp(attribute, "Contrast")==0) ||
|
|
(strcmp(attribute, "Brightness")==0)
|
|
)
|
|
{
|
|
char* keyname="SOFTWARE\\Microsoft\\Scrunch\\Video";
|
|
result=RegCreateKeyExA(HKEY_CURRENT_USER, keyname, 0, 0, 0, 0, 0, &newkey, &status);
|
|
if(result!=0)
|
|
{
|
|
printf("VideoDecoder::SetExtAttr: registry failure\n");
|
|
return -1;
|
|
}
|
|
result=RegSetValueExA(newkey, attribute, 0, REG_DWORD, &value, 4);
|
|
if(result!=0)
|
|
{
|
|
printf("VideoDecoder::SetExtAttr: error writing value\n");
|
|
return -1;
|
|
}
|
|
RegCloseKey(newkey);
|
|
return 0;
|
|
}
|
|
|
|
printf("Unknown attribute!\n");
|
|
return -200;
|
|
}
|
|
|