/* * Modified for use with MPlayer, detailed changelog at * http://svn.mplayerhq.hu/mplayer/trunk/ * $Id$ */#include "config.h"#if (C_HAS_DIRECTSHOW)#include "allocator.h"#include "com.h"#include "wine/winerror.h"static int AllocatorKeeper = 0;struct _avm_list_t{    struct _avm_list_t* next;    struct _avm_list_t* prev;    void* member;};static inline int avm_list_size(avm_list_t* head){    avm_list_t* it = head;    int i = 0;    if (it)    {	for (;;)	{            i++;	    it = it->next;	    if (it == head)                break;	}    }    return i;}static inline int avm_list_print(avm_list_t* head){    avm_list_t* it = head;    int i = 0;    LOG_MSG("Head: %p", head);    if (it)    {	for (;;)	{	    i++;	    LOG_MSG("%d:  member: %p    next: %p  prev: %p",		   i, it->member, it->next, it->prev);	    it = it->next;	    if (it == head)                break;	}    }    return i;}static inline avm_list_t* avm_list_add_head(avm_list_t* head, void* member){    avm_list_t* n = (avm_list_t*) malloc(sizeof(avm_list_t));    n->member = member;    if (!head)    {	head = n;        head->prev = head;    }    n->prev = head->prev;    head->prev = n;    n->next = head;    return n;}static inline avm_list_t* avm_list_add_tail(avm_list_t* head, void* member){    avm_list_t* n = avm_list_add_head(head, member);    return (!head) ? n : head;}static inline avm_list_t* avm_list_del_head(avm_list_t* head){    avm_list_t* n = 0;    if (head)    {	if (head->next != head)	{	    n = head->next;	    head->prev->next = head->next;	    head->next->prev = head->prev;	}	free(head);    }    return n;}static inline avm_list_t* avm_list_find(avm_list_t* head, void* member){    avm_list_t* it = head;    if (it)    {	for (;;)	{	    if (it->member == member)		return it;	    it = it->next;	    if (it == head)                break;	}    }    return NULL;}static long MemAllocator_CreateAllocator(GUID* clsid, const GUID* iid, void** ppv){    IMemAllocator* p;    int result;    if (!ppv)	return -1;    *ppv = 0;    if (memcmp(clsid, &CLSID_MemoryAllocator, sizeof(GUID)))	return -1;    p = (IMemAllocator*) MemAllocatorCreate();    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){    MemAllocator* me = (MemAllocator*)This;    LOG_MSG("MemAllocator_SetProperties(%p) called", This);    if (!pRequest || !pActual)	return E_INVALIDARG;    if (pRequest->cBuffers<=0 || pRequest->cbBuffer<=0)	return E_FAIL;    if (me->used_list != 0 || me->free_list != 0)	return E_FAIL;    *pActual = *pRequest;    /*       DirectShow DOCS ("Negotiating Allocators" chapter) says that allocator might not       honor the requested properties. Thus, since WMSP audio codecs requests bufer with two       bytes length for unknown reason, we should correct requested value. Otherwise above       codec don't want to load.    */    if (pActual->cbBuffer == 2)        pActual->cbBuffer = 10240; //Enough for WMSP codec    me->props = *pActual;    return 0;}static HRESULT STDCALL MemAllocator_GetProperties(IMemAllocator * This,						  /* [out] */ ALLOCATOR_PROPERTIES *pProps){    LOG_MSG("MemAllocator_GetProperties(%p) called", 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){    MemAllocator* me = (MemAllocator*)This;    int i;    LOG_MSG("MemAllocator_Commit(%p) called", This);    if (((MemAllocator*)This)->props.cbBuffer < 0)	return E_FAIL;    if (me->used_list || me->free_list)	return E_INVALIDARG;    for (i = 0; i < me->props.cBuffers; i++)    {	CMediaSample* sample = CMediaSampleCreate((IMemAllocator*)me,						  me->props.cbBuffer);	if (!sample)            return E_OUTOFMEMORY;	//LOG_MSG("FREEEEEEEEEEEE ADDED %p", sample);	me->free_list = avm_list_add_tail(me->free_list, sample);	//avm_list_print(me->free_list);    }    //LOG_MSG("Added mem %p: lsz: %d  %d  size: %d", me, avm_list_size(me->free_list), me->props.cBuffers, me->props.cbBuffer);    return 0;}static HRESULT STDCALL MemAllocator_Decommit(IMemAllocator * This){    MemAllocator* me=(MemAllocator*)This;    LOG_MSG("MemAllocator_Decommit(%p) called", This);    //LOG_MSG("Deleted mem %p: %d  %d", me, me->free_list.size(), me->used_list.size());    while (me->used_list)    {	me->free_list = avm_list_add_tail(me->free_list,					  (CMediaSample*) me->used_list->member);	me->used_list = avm_list_del_head(me->used_list);    }    while (me->free_list)    {        CMediaSample* sample = (CMediaSample*) me->free_list->member;	//LOG_MSG("****************** Decommiting FREE %p", sample);	//sample->vt->Release((IUnknown*)sample);	CMediaSample_Destroy((CMediaSample*)sample);	me->free_list = avm_list_del_head(me->free_list);    }    return 0;}static HRESULT STDCALL MemAllocator_GetBuffer(IMemAllocator * This,					      /* [out] */ IMediaSample **ppBuffer,					      /* [in] */ REFERENCE_TIME *pStartTime,					      /* [in] */ REFERENCE_TIME *pEndTime,					      /* [in] */ DWORD dwFlags){    MemAllocator* me = (MemAllocator*)This;    CMediaSample* sample;    LOG_MSG("MemAllocator_GetBuffer(%p) called   %d  %d", This,		 avm_list_size(me->used_list), avm_list_size(me->free_list));    if (!me->free_list)    {	LOG_MSG("No samples available");	return E_FAIL;//should block here if no samples are available    }    sample = (CMediaSample*) me->free_list->member;    me->free_list = avm_list_del_head(me->free_list);    me->used_list = avm_list_add_tail(me->used_list, sample);    *ppBuffer = (IMediaSample*) sample;    sample->vt->AddRef((IUnknown*) sample);    if (me->new_pointer)    {	if (me->modified_sample)	    me->modified_sample->ResetPointer(me->modified_sample);	sample->SetPointer(sample, me->new_pointer);	me->modified_sample = sample;	me->new_pointer = 0;    }    return 0;}static HRESULT STDCALL MemAllocator_ReleaseBuffer(IMemAllocator* This,						  /* [in] */ IMediaSample* pBuffer){    avm_list_t* l;    MemAllocator* me = (MemAllocator*)This;    LOG_MSG("MemAllocator_ReleaseBuffer(%p) called   %d  %d", This,		 avm_list_size(me->used_list), avm_list_size(me->free_list));    l = avm_list_find(me->used_list, pBuffer);    if (l)    {	CMediaSample* sample = (CMediaSample*) l->member;	if (me->modified_sample == sample)	{	    me->modified_sample->ResetPointer(me->modified_sample);	    me->modified_sample = 0;	}	me->used_list = avm_list_del_head(me->used_list);	me->free_list = avm_list_add_head(me->free_list, sample);	//LOG_MSG("****************** RELEASED OK %p  %p", me->used_list, me->free_list);	return 0;    }    LOG_MSG("MemAllocator_ReleaseBuffer(%p) releasing unknown buffer!!!! %p", This, pBuffer);    return E_FAIL;}static void MemAllocator_SetPointer(MemAllocator* This, char* pointer){    This->new_pointer = pointer;}static void MemAllocator_ResetPointer(MemAllocator* This){    if (This->modified_sample)    {	This->modified_sample->ResetPointer(This->modified_sample);	This->modified_sample = 0;    }}static void MemAllocator_Destroy(MemAllocator* This){    LOG_MSG("MemAllocator_Destroy(%p) called  (%d, %d)", This, This->refcount, AllocatorKeeper);#ifdef WIN32_LOADER    if (--AllocatorKeeper == 0)	UnregisterComClass(&CLSID_MemoryAllocator, MemAllocator_CreateAllocator);#endif    free(This->vt);    free(This);}IMPLEMENT_IUNKNOWN(MemAllocator)MemAllocator* MemAllocatorCreate(){    MemAllocator* This = (MemAllocator*) malloc(sizeof(MemAllocator));    if (!This)        return NULL;    LOG_MSG("MemAllocatorCreate() called -> %p", This);    This->refcount = 1;    This->props.cBuffers = 1;    This->props.cbBuffer = 65536; /* :/ */    This->props.cbAlign = 1;    This->props.cbPrefix = 0;    This->vt = (IMemAllocator_vt*) malloc(sizeof(IMemAllocator_vt));    if (!This->vt)    {        free(This);	return NULL;    }    This->vt->QueryInterface = MemAllocator_QueryInterface;    This->vt->AddRef = MemAllocator_AddRef;    This->vt->Release = MemAllocator_Release;    This->vt->SetProperties = MemAllocator_SetProperties;    This->vt->GetProperties = MemAllocator_GetProperties;    This->vt->Commit = MemAllocator_Commit;    This->vt->Decommit = MemAllocator_Decommit;    This->vt->GetBuffer = MemAllocator_GetBuffer;    This->vt->ReleaseBuffer = MemAllocator_ReleaseBuffer;    This->SetPointer = MemAllocator_SetPointer;    This->ResetPointer = MemAllocator_ResetPointer;    This->modified_sample = 0;    This->new_pointer = 0;    This->used_list = 0;    This->free_list = 0;    This->interfaces[0]=IID_IUnknown;    This->interfaces[1]=IID_IMemAllocator;#ifdef WIN32_LOADER    if (AllocatorKeeper++ == 0)	RegisterComClass(&CLSID_MemoryAllocator, MemAllocator_CreateAllocator);#endif    return This;}#endif // (C_HAS_DIRECTSHOW)