mpv/loader/dshow/outputpin.c

484 lines
12 KiB
C

#include "outputpin.h"
#include <string.h>
#include <stdio.h>
#include "allocator.h"
#include "iunk.h"
#define E_NOTIMPL 0x80004001
/*
An object beyond interface IEnumMediaTypes.
Returned by COutputPin through call IPin::EnumMediaTypes().
*/
class CEnumMediaTypes: public IEnumMediaTypes
{
AM_MEDIA_TYPE type;
static GUID interfaces[];
DECLARE_IUNKNOWN(CEnumMediaTypes)
public:
CEnumMediaTypes(const AM_MEDIA_TYPE&);
~CEnumMediaTypes(){delete vt;}
static HRESULT STDCALL Next (
IEnumMediaTypes * This,
/* [in] */ ULONG cMediaTypes,
/* [size_is][out] */ AM_MEDIA_TYPE **ppMediaTypes,
/* [out] */ ULONG *pcFetched);
static HRESULT STDCALL Skip (
IEnumMediaTypes * This,
/* [in] */ ULONG cMediaTypes);
static HRESULT STDCALL Reset (
IEnumMediaTypes * This);
static HRESULT STDCALL Clone (
IEnumMediaTypes * This,
/* [out] */ IEnumMediaTypes **ppEnum);
};
GUID CEnumMediaTypes::interfaces[]=
{
IID_IUnknown,
IID_IEnumMediaTypes,
};
IMPLEMENT_IUNKNOWN(CEnumMediaTypes)
CEnumMediaTypes::CEnumMediaTypes(const AM_MEDIA_TYPE& type)
:refcount(1)
{
this->type=type;
vt=new IEnumMediaTypes_vt;
vt->QueryInterface = QueryInterface;
vt->AddRef = AddRef;
vt->Release = Release;
vt->Next = Next;
vt->Skip = Skip;
vt->Reset = Reset;
vt->Clone = Clone;
}
HRESULT STDCALL CEnumMediaTypes::Next (
IEnumMediaTypes * This,
/* [in] */ ULONG cMediaTypes,
/* [size_is][out] */ AM_MEDIA_TYPE **ppMediaTypes,
/* [out] */ ULONG *pcFetched)
{
AM_MEDIA_TYPE& type=((CEnumMediaTypes*)This)->type;
Debug printf("CEnumMediaTypes::Next() called\n");
if(!ppMediaTypes)return 0x80004003;
if(!pcFetched && (cMediaTypes!=1))return 0x80004003;
if(cMediaTypes<=0)return 0;
if(pcFetched)*pcFetched=1;
ppMediaTypes[0]=(AM_MEDIA_TYPE *)CoTaskMemAlloc(sizeof(AM_MEDIA_TYPE));
memcpy(*ppMediaTypes, &type, sizeof(AM_MEDIA_TYPE));
if(ppMediaTypes[0]->pbFormat)
{
ppMediaTypes[0]->pbFormat=(char *)CoTaskMemAlloc(ppMediaTypes[0]->cbFormat);
memcpy(ppMediaTypes[0]->pbFormat, type.pbFormat, ppMediaTypes[0]->cbFormat);
}
if(cMediaTypes==1)return 0;
return 1;
}
/*
I expect that these methods are unused.
*/
HRESULT STDCALL CEnumMediaTypes::Skip (
IEnumMediaTypes * This,
/* [in] */ ULONG cMediaTypes)
{
Debug printf("CEnumMediaTypes::Skip() called\n");
return E_NOTIMPL;
}
HRESULT STDCALL CEnumMediaTypes::Reset (
IEnumMediaTypes * This)
{
Debug printf("CEnumMediaTypes::Reset() called\n");
return 0;
}
HRESULT STDCALL CEnumMediaTypes::Clone (
IEnumMediaTypes * This,
/* [out] */ IEnumMediaTypes **ppEnum)
{
Debug printf("CEnumMediaTypes::Clone() called\n");
return E_NOTIMPL;
}
/*
Implementation of output pin object.
*/
// Constructor
COutputPin::COutputPin(const AM_MEDIA_TYPE& vh) :refcount(1), type(vh), remote(0), frame_pointer(0), frame_size_pointer(0)
{
IPin::vt=new IPin_vt;
IPin::vt->QueryInterface = QueryInterface;
IPin::vt->AddRef = AddRef;
IPin::vt->Release = Release;
IPin::vt->Connect = Connect;
IPin::vt->ReceiveConnection = ReceiveConnection;
IPin::vt->Disconnect=Disconnect;
IPin::vt->ConnectedTo = ConnectedTo;
IPin::vt->ConnectionMediaType = ConnectionMediaType;
IPin::vt->QueryPinInfo = QueryPinInfo;
IPin::vt->QueryDirection = QueryDirection;
IPin::vt->QueryId = QueryId;
IPin::vt->QueryAccept = QueryAccept;
IPin::vt->EnumMediaTypes = EnumMediaTypes;
IPin::vt->QueryInternalConnections = QueryInternalConnections;
IPin::vt->EndOfStream = EndOfStream;
IPin::vt->BeginFlush = BeginFlush;
IPin::vt->EndFlush = EndFlush;
IPin::vt->NewSegment = NewSegment;
IMemInputPin::vt=new IMemInputPin_vt;
IMemInputPin::vt->QueryInterface = M_QueryInterface;
IMemInputPin::vt->AddRef = M_AddRef;
IMemInputPin::vt->Release = M_Release;
IMemInputPin::vt->GetAllocator = GetAllocator;
IMemInputPin::vt->NotifyAllocator = NotifyAllocator;
IMemInputPin::vt->GetAllocatorRequirements = GetAllocatorRequirements;
IMemInputPin::vt->Receive = Receive;
IMemInputPin::vt->ReceiveMultiple = ReceiveMultiple;
IMemInputPin::vt->ReceiveCanBlock = ReceiveCanBlock;
}
// IPin->IUnknown methods
HRESULT STDCALL COutputPin::QueryInterface(IUnknown* This, GUID* iid, void** ppv)
{
Debug printf("COutputPin::QueryInterface() called\n");
if(!ppv)return 0x80004003;
if(!memcmp(iid, &IID_IUnknown, 16))
{
*ppv=(void*)This;
This->vt->AddRef(This);
return 0;
}
if(!memcmp(iid, &IID_IMemInputPin, 16))
{
*ppv=(void*)(This+1);
This->vt->AddRef(This);
return 0;
}
Debug printf("Unknown interface : %08x-%04x-%04x-%02x%02x-" \
"%02x%02x%02x%02x%02x%02x\n",
iid->f1, iid->f2, iid->f3,
(unsigned char)iid->f4[1], (unsigned char)iid->f4[0],
(unsigned char)iid->f4[2],(unsigned char)iid->f4[3],(unsigned char)iid->f4[4],
(unsigned char)iid->f4[5],(unsigned char)iid->f4[6],(unsigned char)iid->f4[7]);
return 0x80004002;
}
HRESULT STDCALL COutputPin::AddRef(IUnknown* This)
{
Debug printf("COutputPin::AddRef() called\n");
((COutputPin*)This)->refcount++;
return 0;
}
HRESULT STDCALL COutputPin::Release(IUnknown* This)
{
Debug printf("COutputPin::Release() called\n");
if(--((COutputPin*)This)->refcount==0)
delete (COutputPin*)This;
return 0;
}
// IPin methods
HRESULT STDCALL COutputPin::Connect (
IPin * This,
/* [in] */ IPin *pReceivePin,
/* [in] */ /* const */ AM_MEDIA_TYPE *pmt)
{
Debug printf("COutputPin::Connect() called\n");
/*
*pmt=((COutputPin*)This)->type;
if(pmt->cbFormat>0)
{
pmt->pbFormat=CoTaskMemAlloc(pmt->cbFormat);
memcpy(pmt->pbFormat, ((COutputPin*)This)->type.pbFormat, pmt->cbFormat);
}
*/
return E_NOTIMPL;
// if I put return 0; here, it crashes
}
HRESULT STDCALL COutputPin::ReceiveConnection (
IPin * This,
/* [in] */ IPin *pConnector,
/* [in] */ const AM_MEDIA_TYPE *pmt)
{
Debug printf("COutputPin::ReceiveConnection() called\n");
((COutputPin*)This)->remote=pConnector;
return 0;
}
HRESULT STDCALL COutputPin::Disconnect (
IPin * This)
{
Debug printf("COutputPin::Disconnect() called\n");
return 1;
}
HRESULT STDCALL COutputPin::ConnectedTo (
IPin * This,
/* [out] */ IPin **pPin)
{
Debug printf("COutputPin::ConnectedTo() called\n");
if(!pPin)return 0x80004003;
*pPin=((COutputPin*)This)->remote;
return 0;
}
HRESULT STDCALL COutputPin::ConnectionMediaType (
IPin * This,
/* [out] */ AM_MEDIA_TYPE *pmt)
{
Debug printf("CInputPin::ConnectionMediaType() called\n");
if(!pmt)return 0x80004003;
*pmt=((COutputPin*)This)->type;
if(pmt->cbFormat>0)
{
pmt->pbFormat=(char *)CoTaskMemAlloc(pmt->cbFormat);
memcpy(pmt->pbFormat, ((COutputPin*)This)->type.pbFormat, pmt->cbFormat);
}
return 0;
}
HRESULT STDCALL COutputPin::QueryPinInfo (
IPin * This,
/* [out] */ PIN_INFO *pInfo)
{
Debug printf("COutputPin::QueryPinInfo() called\n");
return E_NOTIMPL;
}
HRESULT STDCALL COutputPin::QueryDirection (
IPin * This,
/* [out] */ PIN_DIRECTION *pPinDir)
{
Debug printf("COutputPin::QueryDirection() called\n");
if(!pPinDir)return -1;
*pPinDir=PINDIR_INPUT;
return 0;
}
HRESULT STDCALL COutputPin::QueryId (
IPin * This,
/* [out] */ LPWSTR *Id)
{
Debug printf("COutputPin::QueryId() called\n");
return E_NOTIMPL;
}
HRESULT STDCALL COutputPin::QueryAccept (
IPin * This,
/* [in] */ const AM_MEDIA_TYPE *pmt)
{
Debug printf("COutputPin::QueryAccept() called\n");
return E_NOTIMPL;
}
HRESULT STDCALL COutputPin::EnumMediaTypes (
IPin * This,
/* [out] */ IEnumMediaTypes **ppEnum)
{
Debug printf("COutputPin::EnumMediaTypes() called\n");
if(!ppEnum)return 0x80004003;
*ppEnum=new CEnumMediaTypes(((COutputPin*)This)->type);
return 0;
}
HRESULT STDCALL COutputPin::QueryInternalConnections (
IPin * This,
/* [out] */ IPin **apPin,
/* [out][in] */ ULONG *nPin)
{
Debug printf("COutputPin::QueryInternalConnections() called\n");
return E_NOTIMPL;
}
HRESULT STDCALL COutputPin::EndOfStream (
IPin * This)
{
Debug printf("COutputPin::EndOfStream() called\n");
return E_NOTIMPL;
}
HRESULT STDCALL COutputPin::BeginFlush (
IPin * This)
{
Debug printf("COutputPin::BeginFlush() called\n");
return E_NOTIMPL;
}
HRESULT STDCALL COutputPin::EndFlush (
IPin * This)
{
Debug printf("COutputPin::EndFlush() called\n");
return E_NOTIMPL;
}
HRESULT STDCALL COutputPin::NewSegment (
IPin * This,
/* [in] */ REFERENCE_TIME tStart,
/* [in] */ REFERENCE_TIME tStop,
/* [in] */ double dRate)
{
Debug printf("COutputPin::NewSegment(%ld,%ld,%f) called\n",tStart,tStop,dRate);
return 0;
}
// IMemInputPin->IUnknown methods
HRESULT STDCALL COutputPin::M_QueryInterface(IUnknown* This, GUID* iid, void** ppv)
{
Debug printf("COutputPin::QueryInterface() called\n");
if(!ppv)return 0x80004003;
if(!memcmp(iid, &IID_IUnknown, 16))
{
COutputPin* ptr=(COutputPin*)(This-1);
*ppv=(void*)ptr;
AddRef((IUnknown*)ptr);
return 0;
}
/* if(!memcmp(iid, &IID_IPin, 16))
{
COutputPin* ptr=(COutputPin*)(This-1);
*ppv=(void*)ptr;
AddRef((IUnknown*)ptr);
return 0;
}*/
if(!memcmp(iid, &IID_IMemInputPin, 16))
{
*ppv=(void*)This;
This->vt->AddRef(This);
return 0;
}
Debug printf("Unknown interface : %08x-%04x-%04x-%02x%02x-" \
"%02x%02x%02x%02x%02x%02x\n",
iid->f1, iid->f2, iid->f3,
(unsigned char)iid->f4[1], (unsigned char)iid->f4[0],
(unsigned char)iid->f4[2],(unsigned char)iid->f4[3],(unsigned char)iid->f4[4],
(unsigned char)iid->f4[5],(unsigned char)iid->f4[6],(unsigned char)iid->f4[7]);
return 0x80004002;
}
HRESULT STDCALL COutputPin::M_AddRef(IUnknown* This)
{
Debug printf("COutputPin::AddRef() called\n");
((COutputPin*)(This-1))->refcount++;
return 0;
}
HRESULT STDCALL COutputPin::M_Release(IUnknown* This)
{
Debug printf("COutputPin::Release() called\n");
if(--((COutputPin*)(This-1))->refcount==0)
delete (COutputPin*)This;
return 0;
}
// IMemInputPin methods
HRESULT STDCALL COutputPin::GetAllocator(
IMemInputPin * This,
/* [out] */ IMemAllocator **ppAllocator)
{
Debug printf("COutputPin::GetAllocator(%x,%x) called\n",This->vt,ppAllocator);
*ppAllocator=new MemAllocator;
return 0;
}
HRESULT STDCALL COutputPin::NotifyAllocator(
IMemInputPin * This,
/* [in] */ IMemAllocator *pAllocator,
/* [in] */ int bReadOnly)
{
Debug printf("COutputPin::NotifyAllocator() called\n");
return 0;
}
HRESULT STDCALL COutputPin::GetAllocatorRequirements(
IMemInputPin * This,
/* [out] */ ALLOCATOR_PROPERTIES *pProps)
{
Debug printf("COutputPin::GetAllocatorRequirements() called\n");
return E_NOTIMPL;
}
HRESULT STDCALL COutputPin::Receive(
IMemInputPin * This,
/* [in] */ IMediaSample *pSample)
{
Debug printf("COutputPin::Receive() called\n");
COutputPin& me=*(COutputPin*)This;
if(!pSample)return 0x80004003;
char* pointer;
if(pSample->vt->GetPointer(pSample, (BYTE **)&pointer))
return -1;
int len=pSample->vt->GetActualDataLength(pSample);
if(len==0)len=pSample->vt->GetSize(pSample);//for iv50
//if(me.frame_pointer)memcpy(me.frame_pointer, pointer, len);
*me.frame_pointer=pointer;
if(me.frame_size_pointer)*me.frame_size_pointer=len;
/*
FILE* file=fopen("./uncompr.bmp", "wb");
char head[14]={0x42, 0x4D, 0x36, 0x10, 0x05, 0x00, 0x00, 0x00, 0x00, 0x00, 0x36, 0x00, 0x00, 0x00};
*(int*)(&head[2])=len+0x36;
fwrite(head, 14, 1, file);
fwrite(&((VIDEOINFOHEADER*)me.type.pbFormat)->bmiHeader, sizeof(BITMAPINFOHEADER), 1, file);
fwrite(pointer, len, 1, file);
fclose(file);
*/
// pSample->vt->Release((IUnknown*)pSample);
return 0;
}
HRESULT STDCALL COutputPin::ReceiveMultiple(
IMemInputPin * This,
/* [size_is][in] */ IMediaSample **pSamples,
/* [in] */ long nSamples,
/* [out] */ long *nSamplesProcessed)
{
Debug printf("COutputPin::ReceiveMultiple() called\n");
return E_NOTIMPL;
}
HRESULT STDCALL COutputPin::ReceiveCanBlock(
IMemInputPin * This)
{
Debug printf("COutputPin::ReceiveCanBlock() called\n");
return E_NOTIMPL;
}