mirror of
https://github.com/mpv-player/mpv
synced 2024-12-14 02:45:43 +00:00
4f3776f018
git-svn-id: svn://svn.mplayerhq.hu/mplayer/trunk@2070 b3059339-0415-0410-9bf9-f77b7e298cf2
179 lines
4.8 KiB
C
179 lines
4.8 KiB
C
#include "allocator.h"
|
|
#include "com.h"
|
|
#include "wine/winerror.h"
|
|
#include <stdio.h>
|
|
|
|
//#undef Debug
|
|
//#define Debug
|
|
|
|
using namespace std;
|
|
|
|
class AllocatorKeeper
|
|
{
|
|
public:
|
|
AllocatorKeeper()
|
|
{
|
|
RegisterComClass(&CLSID_MemoryAllocator, MemAllocator::CreateAllocator);
|
|
}
|
|
~AllocatorKeeper()
|
|
{
|
|
UnregisterComClass(&CLSID_MemoryAllocator, MemAllocator::CreateAllocator);
|
|
}
|
|
};
|
|
static AllocatorKeeper keeper;
|
|
|
|
|
|
GUID MemAllocator::interfaces[]=
|
|
{
|
|
IID_IUnknown,
|
|
IID_IMemAllocator,
|
|
};
|
|
|
|
IMPLEMENT_IUNKNOWN(MemAllocator)
|
|
|
|
long MemAllocator::CreateAllocator(GUID* clsid, GUID* iid, void** ppv)
|
|
{
|
|
if(!ppv)return -1;
|
|
*ppv=0;
|
|
if(memcmp(clsid, &CLSID_MemoryAllocator, sizeof(GUID)))
|
|
return -1;
|
|
|
|
IMemAllocator* p=new MemAllocator;
|
|
int result=p->vt->QueryInterface((IUnknown*)p, iid, ppv);
|
|
p->vt->Release((IUnknown*)p);
|
|
return result;
|
|
}
|
|
|
|
|
|
static HRESULT STDCALL MemAllocator_SetProperties(IMemAllocator * This,
|
|
/* [in] */ ALLOCATOR_PROPERTIES *pRequest,
|
|
/* [out] */ ALLOCATOR_PROPERTIES *pActual)
|
|
{
|
|
Debug printf("MemAllocator_SetProperties() called\n");
|
|
if (!pRequest || !pActual)
|
|
return E_INVALIDARG;
|
|
if (pRequest->cBuffers<=0 || pRequest->cbBuffer<=0)
|
|
return E_FAIL;
|
|
MemAllocator* me = (MemAllocator*)This;
|
|
if (me->used_list.size() || me->free_list.size())
|
|
return E_FAIL;
|
|
me->props = *pRequest;
|
|
*pActual = *pRequest;
|
|
return 0;
|
|
}
|
|
|
|
static HRESULT STDCALL MemAllocator_GetProperties(IMemAllocator * This,
|
|
/* [out] */ ALLOCATOR_PROPERTIES *pProps)
|
|
{
|
|
Debug printf("MemAllocator_GetProperties(%p) called\n", This);
|
|
if (!pProps)
|
|
return E_INVALIDARG;
|
|
if (((MemAllocator*)This)->props.cbBuffer<0)
|
|
return E_FAIL;
|
|
*pProps=((MemAllocator*)This)->props;
|
|
return 0;
|
|
}
|
|
|
|
static HRESULT STDCALL MemAllocator_Commit(IMemAllocator * This)
|
|
{
|
|
Debug printf("MemAllocator_Commit(%p) called\n", This);
|
|
MemAllocator* me=(MemAllocator*)This;
|
|
if (((MemAllocator*)This)->props.cbBuffer < 0)
|
|
return E_FAIL;
|
|
if (me->used_list.size() || me->free_list.size())
|
|
return E_INVALIDARG;
|
|
for(int i = 0; i<me->props.cBuffers; i++)
|
|
me->free_list.push_back(new CMediaSample(me, me->props.cbBuffer));
|
|
|
|
return 0;
|
|
}
|
|
|
|
static HRESULT STDCALL MemAllocator_Decommit(IMemAllocator * This)
|
|
{
|
|
Debug printf("MemAllocator_Decommit(%p) called\n", This);
|
|
MemAllocator* me=(MemAllocator*)This;
|
|
list<CMediaSample*>::iterator it;
|
|
for(it=me->free_list.begin(); it!=me->free_list.end(); it++)
|
|
delete *it;
|
|
for(it=me->used_list.begin(); it!=me->used_list.end(); it++)
|
|
delete *it;
|
|
me->free_list.clear();
|
|
me->used_list.clear();
|
|
return 0;
|
|
}
|
|
|
|
static HRESULT STDCALL MemAllocator_GetBuffer(IMemAllocator * This,
|
|
/* [out] */ IMediaSample **ppBuffer,
|
|
/* [in] */ REFERENCE_TIME *pStartTime,
|
|
/* [in] */ REFERENCE_TIME *pEndTime,
|
|
/* [in] */ DWORD dwFlags)
|
|
{
|
|
Debug printf("MemAllocator_GetBuffer(%p) called\n", This);
|
|
MemAllocator* me = (MemAllocator*)This;
|
|
if (me->free_list.size() == 0)
|
|
{
|
|
Debug printf("No samples available\n");
|
|
return E_FAIL;//should block here if no samples are available
|
|
}
|
|
list<CMediaSample*>::iterator it = me->free_list.begin();
|
|
me->used_list.push_back(*it);
|
|
*ppBuffer = *it;
|
|
(*ppBuffer)->vt->AddRef((IUnknown*)*ppBuffer);
|
|
if (me->new_pointer)
|
|
{
|
|
if(me->modified_sample)
|
|
me->modified_sample->ResetPointer();
|
|
(*it)->SetPointer(me->new_pointer);
|
|
me->modified_sample=*it;
|
|
me->new_pointer = 0;
|
|
}
|
|
me->free_list.remove(*it);
|
|
return 0;
|
|
}
|
|
|
|
static HRESULT STDCALL MemAllocator_ReleaseBuffer(IMemAllocator * This,
|
|
/* [in] */ IMediaSample *pBuffer)
|
|
{
|
|
Debug printf("MemAllocator_ReleaseBuffer(%p) called\n", This);
|
|
MemAllocator* me = (MemAllocator*)This;
|
|
list<CMediaSample*>::iterator it;
|
|
for (it = me->used_list.begin(); it != me->used_list.end(); it++)
|
|
if (*it == pBuffer)
|
|
{
|
|
me->used_list.erase(it);
|
|
me->free_list.push_back((CMediaSample*)pBuffer);
|
|
return 0;
|
|
}
|
|
Debug printf("Releasing unknown buffer\n");
|
|
return E_FAIL;
|
|
}
|
|
|
|
MemAllocator::MemAllocator()
|
|
{
|
|
Debug printf("MemAllocator::MemAllocator() called\n");
|
|
vt = new IMemAllocator_vt;
|
|
vt->QueryInterface = QueryInterface;
|
|
vt->AddRef = AddRef;
|
|
vt->Release = Release;
|
|
vt->SetProperties = MemAllocator_SetProperties;
|
|
vt->GetProperties = MemAllocator_GetProperties;
|
|
vt->Commit = MemAllocator_Commit;
|
|
vt->Decommit = MemAllocator_Decommit;
|
|
vt->GetBuffer = MemAllocator_GetBuffer;
|
|
vt->ReleaseBuffer = MemAllocator_ReleaseBuffer;
|
|
|
|
refcount = 1;
|
|
props.cBuffers = 1;
|
|
props.cbBuffer = 65536; /* :/ */
|
|
props.cbAlign = props.cbPrefix = 0;
|
|
|
|
new_pointer=0;
|
|
modified_sample=0;
|
|
}
|
|
|
|
MemAllocator::~MemAllocator()
|
|
{
|
|
Debug printf("MemAllocator::~MemAllocator() called\n");
|
|
delete vt;
|
|
}
|