Added C-C++ interface in DS_VideoDec.h

git-svn-id: svn://svn.mplayerhq.hu/mplayer/trunk@170 b3059339-0415-0410-9bf9-f77b7e298cf2

4 files changed, 487 insertions, 4 deletions

diff --git a/loader/dshow/DS_VideoDec.c b/loader/dshow/DS_VideoDec.c
new file mode 100644
index 0000000000..299f7ceae4
--- /dev/null
+++ b/loader/dshow/DS_VideoDec.c
@@ -0,0 +1,385 @@
+/********************************************************
+
+	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 "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;
+
+	if(!flip)
+	{
+	    m_sVhdr2->bmiHeader.biHeight*=-1;
+	    m_obh.biHeight*=-1;
+//	    result=m_pOutputPin->vt->QueryAccept(m_pOutputPin, &m_sDestType);
+//	    if(result)
+//		throw FATAL("Decoder does not support upside-down frames");
+	}	
+
+        dsf=new DS_Filter();
+        
+	dsf->Create(dllname, guid, &m_sOurType, &m_sDestType);
+
+	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);
+
+	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)
+	    Debug 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);
+	switch(csp)
+	{
+	    case fccYUY2:
+		m_sDestType.subtype=MEDIASUBTYPE_YUY2;
+		break;
+	    case fccYV12:
+		m_sDestType.subtype=MEDIASUBTYPE_YV12;
+		break;
+	    case fccIYUV:
+		m_sDestType.subtype=MEDIASUBTYPE_IYUV;
+		break;
+	    case fccUYVY:
+		m_sDestType.subtype=MEDIASUBTYPE_UYVY;
+		break;
+	    case fccYVYU:
+		m_sDestType.subtype=MEDIASUBTYPE_YVYU;
+		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;
+}
+
+
diff --git a/loader/dshow/DS_VideoDec.h b/loader/dshow/DS_VideoDec.h
new file mode 100644
index 0000000000..c46d34155a
--- /dev/null
+++ b/loader/dshow/DS_VideoDec.h
@@ -0,0 +1,31 @@
+/********************************************************
+
+	DirectShow Video decoder implementation
+	Copyright 2000 Eugene Kuznetsov  (divx@euro.ru)
+        Converted  C++ --> C  :) by A'rpi/ESP-team
+
+*********************************************************/
+
+#ifdef __cplusplus
+extern "C" {
+#endif /* defined(__cplusplus) */
+
+int DS_VideoDecoder_Open(char* dllname, GUID* guid, BITMAPINFOHEADER* format, int flip,char** d_ptr);
+
+void DS_VideoDecoder_Start();
+
+void DS_VideoDecoder_Stop();
+
+void DS_VideoDecoder_Restart();
+
+void DS_VideoDecoder_Close();
+
+int DS_VideoDecoder_DecodeFrame(char* src, int size, int is_keyframe, int render);
+
+int DS_VideoDecoder_SetDestFmt(int bits, int csp);
+
+int DS_SetValue_DivX(char* name, int value);
+
+#ifdef __cplusplus
+} /* extern "C" */
+#endif /* defined(__cplusplus) */
diff --git a/loader/dshow/Makefile b/loader/dshow/Makefile
index 3970cfcfd4..82db334d1d 100644
--- a/loader/dshow/Makefile
+++ b/loader/dshow/Makefile
@@ -3,8 +3,8 @@ LIBNAME = libDS_Filter.a
 
 include ../../config.mak
 
-SRCS	= DS_Filter.cpp allocator.cpp cmediasample.cpp guids.cpp inputpin.cpp outputpin.cpp
-OBJS	= DS_Filter.o allocator.o cmediasample.o guids.o inputpin.o outputpin.o
+SRCS	= DS_VideoDec.cpp DS_Filter.cpp allocator.cpp cmediasample.cpp guids.cpp inputpin.cpp outputpin.cpp
+OBJS	= DS_VideoDec.o DS_Filter.o allocator.o cmediasample.o guids.o inputpin.o outputpin.o
 
 INCLUDE = -I. -I.. -I../wine
 CFLAGS  = $(OPTFLAGS) $(INCLUDE)
@@ -19,8 +19,8 @@ CFLAGS  = $(OPTFLAGS) $(INCLUDE)
 $(LIBNAME):	.depend $(OBJS)
 	$(AR) r $(LIBNAME) $(OBJS)
 
-test:   test.cpp
-	$(CC) test.cpp $(CFLAGS) -o test -L. -lDS_Filter -L.. -lloader -ldl -lpthread -lstdc++
+test:   test.c $(LIBNAME)
+	$(CC) test.c $(CFLAGS) -o test -L. -lDS_Filter -L.. -lloader -ldl -lpthread -lstdc++
 
 all:	$(LIBNAME)
 
diff --git a/loader/dshow/test.c b/loader/dshow/test.c
new file mode 100644
index 0000000000..9a98444132
--- /dev/null
+++ b/loader/dshow/test.c
@@ -0,0 +1,67 @@
+#include <stdio.h>
+#include <stdlib.h>
+
+#include "formats.h"
+#include "com.h"
+
+#include "DS_VideoDec.h"
+
+int main(int argc,char* argv[]){
+    FILE *f;
+    FILE *f2;
+    BITMAPINFOHEADER bih;
+    int len;
+    char *src;
+    char *dst=0;
+    GUID CLSID_DivxDecompressorCF={0x82CCd3E0, 0xF71A, 0x11D0,
+    { 0x9f, 0xe5, 0x00, 0x60, 0x97, 0x78, 0xaa, 0xaa}};
+
+    f=fopen("test.divx","rb");
+    
+    fread(&bih,sizeof(BITMAPINFOHEADER),1,f);
+    printf("frame dim: %d x %d \n",bih.biWidth,bih.biHeight);
+
+    src=(char*)malloc(512000);
+    len=fread(src,1,512000,f);
+    printf("frame len = %d\n",len);
+
+    DS_VideoDecoder_Open("divx_c32.ax", &CLSID_DivxDecompressorCF, &bih, 0, &dst);
+
+//    DS_VideoDecoder_SetDestFmt(16,fccYUY2);
+    DS_VideoDecoder_SetDestFmt(24,0);
+
+    DS_VideoDecoder_Start();
+
+    printf("DivX setting result = %d\n", DS_SetValue_DivX("Quality",100) );
+    printf("DivX setting result = %d\n", DS_SetValue_DivX("Brightness",60) );
+
+    DS_VideoDecoder_DecodeFrame(src, len, 1, 1);
+
+#if 0
+    f2=fopen("test.yuy2","wb");
+    fwrite(dst,bih.biWidth*bih.biHeight*2,1,f2);
+    fclose(f2);
+#endif    
+
+      { unsigned char raw_head[32];
+        FILE *f=fopen("test.raw","wb");
+
+        strcpy((char*)raw_head,"mhwanh");
+        raw_head[7]=4;
+        raw_head[8]=bih.biWidth>>8;
+        raw_head[9]=bih.biWidth&0xFF;
+        raw_head[10]=bih.biHeight>>8;
+        raw_head[11]=bih.biHeight&0xFF;
+        raw_head[12]=raw_head[13]=0; // 24bit
+        raw_head[14]=1;raw_head[15]=0x2C;
+        raw_head[16]=1;raw_head[17]=0x2C;
+        memset(raw_head+18,0,32-18);
+        fwrite(raw_head,32,1,f);
+        
+        fwrite(dst,bih.biWidth*bih.biHeight*3,1,f);
+        fclose(f);
+      }
+
+
+return 0;
+}