yading@10: /*
yading@10:  * DirectShow capture interface
yading@10:  * Copyright (c) 2010 Ramiro Polla
yading@10:  *
yading@10:  * This file is part of FFmpeg.
yading@10:  *
yading@10:  * FFmpeg is free software; you can redistribute it and/or
yading@10:  * modify it under the terms of the GNU Lesser General Public
yading@10:  * License as published by the Free Software Foundation; either
yading@10:  * version 2.1 of the License, or (at your option) any later version.
yading@10:  *
yading@10:  * FFmpeg is distributed in the hope that it will be useful,
yading@10:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
yading@10:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
yading@10:  * Lesser General Public License for more details.
yading@10:  *
yading@10:  * You should have received a copy of the GNU Lesser General Public
yading@10:  * License along with FFmpeg; if not, write to the Free Software
yading@10:  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
yading@10:  */
yading@10: 
yading@10: #include "dshow_capture.h"
yading@10: 
yading@10: #include <stddef.h>
yading@10: #define imemoffset offsetof(libAVPin, imemvtbl)
yading@10: 
yading@10: DECLARE_QUERYINTERFACE(libAVPin,
yading@10:     { {&IID_IUnknown,0}, {&IID_IPin,0}, {&IID_IMemInputPin,imemoffset} })
yading@10: DECLARE_ADDREF(libAVPin)
yading@10: DECLARE_RELEASE(libAVPin)
yading@10: 
yading@10: long WINAPI
yading@10: libAVPin_Connect(libAVPin *this, IPin *pin, const AM_MEDIA_TYPE *type)
yading@10: {
yading@10:     dshowdebug("libAVPin_Connect(%p, %p, %p)\n", this, pin, type);
yading@10:     /* Input pins receive connections. */
yading@10:     return S_FALSE;
yading@10: }
yading@10: long WINAPI
yading@10: libAVPin_ReceiveConnection(libAVPin *this, IPin *pin,
yading@10:                            const AM_MEDIA_TYPE *type)
yading@10: {
yading@10:     enum dshowDeviceType devtype = this->filter->type;
yading@10:     dshowdebug("libAVPin_ReceiveConnection(%p)\n", this);
yading@10: 
yading@10:     if (!pin)
yading@10:         return E_POINTER;
yading@10:     if (this->connectedto)
yading@10:         return VFW_E_ALREADY_CONNECTED;
yading@10: 
yading@10:     ff_print_AM_MEDIA_TYPE(type);
yading@10:     if (devtype == VideoDevice) {
yading@10:         if (!IsEqualGUID(&type->majortype, &MEDIATYPE_Video))
yading@10:             return VFW_E_TYPE_NOT_ACCEPTED;
yading@10:     } else {
yading@10:         if (!IsEqualGUID(&type->majortype, &MEDIATYPE_Audio))
yading@10:             return VFW_E_TYPE_NOT_ACCEPTED;
yading@10:     }
yading@10: 
yading@10:     IPin_AddRef(pin);
yading@10:     this->connectedto = pin;
yading@10: 
yading@10:     ff_copy_dshow_media_type(&this->type, type);
yading@10: 
yading@10:     return S_OK;
yading@10: }
yading@10: long WINAPI
yading@10: libAVPin_Disconnect(libAVPin *this)
yading@10: {
yading@10:     dshowdebug("libAVPin_Disconnect(%p)\n", this);
yading@10: 
yading@10:     if (this->filter->state != State_Stopped)
yading@10:         return VFW_E_NOT_STOPPED;
yading@10:     if (!this->connectedto)
yading@10:         return S_FALSE;
yading@10:     IPin_Release(this->connectedto);
yading@10:     this->connectedto = NULL;
yading@10: 
yading@10:     return S_OK;
yading@10: }
yading@10: long WINAPI
yading@10: libAVPin_ConnectedTo(libAVPin *this, IPin **pin)
yading@10: {
yading@10:     dshowdebug("libAVPin_ConnectedTo(%p)\n", this);
yading@10: 
yading@10:     if (!pin)
yading@10:         return E_POINTER;
yading@10:     if (!this->connectedto)
yading@10:         return VFW_E_NOT_CONNECTED;
yading@10:     IPin_AddRef(this->connectedto);
yading@10:     *pin = this->connectedto;
yading@10: 
yading@10:     return S_OK;
yading@10: }
yading@10: long WINAPI
yading@10: libAVPin_ConnectionMediaType(libAVPin *this, AM_MEDIA_TYPE *type)
yading@10: {
yading@10:     dshowdebug("libAVPin_ConnectionMediaType(%p)\n", this);
yading@10: 
yading@10:     if (!type)
yading@10:         return E_POINTER;
yading@10:     if (!this->connectedto)
yading@10:         return VFW_E_NOT_CONNECTED;
yading@10: 
yading@10:     return ff_copy_dshow_media_type(type, &this->type);
yading@10: }
yading@10: long WINAPI
yading@10: libAVPin_QueryPinInfo(libAVPin *this, PIN_INFO *info)
yading@10: {
yading@10:     dshowdebug("libAVPin_QueryPinInfo(%p)\n", this);
yading@10: 
yading@10:     if (!info)
yading@10:         return E_POINTER;
yading@10: 
yading@10:     if (this->filter)
yading@10:         libAVFilter_AddRef(this->filter);
yading@10: 
yading@10:     info->pFilter = (IBaseFilter *) this->filter;
yading@10:     info->dir     = PINDIR_INPUT;
yading@10:     wcscpy(info->achName, L"Capture");
yading@10: 
yading@10:     return S_OK;
yading@10: }
yading@10: long WINAPI
yading@10: libAVPin_QueryDirection(libAVPin *this, PIN_DIRECTION *dir)
yading@10: {
yading@10:     dshowdebug("libAVPin_QueryDirection(%p)\n", this);
yading@10:     if (!dir)
yading@10:         return E_POINTER;
yading@10:     *dir = PINDIR_INPUT;
yading@10:     return S_OK;
yading@10: }
yading@10: long WINAPI
yading@10: libAVPin_QueryId(libAVPin *this, wchar_t **id)
yading@10: {
yading@10:     dshowdebug("libAVPin_QueryId(%p)\n", this);
yading@10: 
yading@10:     if (!id)
yading@10:         return E_POINTER;
yading@10: 
yading@10:     *id = wcsdup(L"libAV Pin");
yading@10: 
yading@10:     return S_OK;
yading@10: }
yading@10: long WINAPI
yading@10: libAVPin_QueryAccept(libAVPin *this, const AM_MEDIA_TYPE *type)
yading@10: {
yading@10:     dshowdebug("libAVPin_QueryAccept(%p)\n", this);
yading@10:     return S_FALSE;
yading@10: }
yading@10: long WINAPI
yading@10: libAVPin_EnumMediaTypes(libAVPin *this, IEnumMediaTypes **enumtypes)
yading@10: {
yading@10:     const AM_MEDIA_TYPE *type = NULL;
yading@10:     libAVEnumMediaTypes *new;
yading@10:     dshowdebug("libAVPin_EnumMediaTypes(%p)\n", this);
yading@10: 
yading@10:     if (!enumtypes)
yading@10:         return E_POINTER;
yading@10:     new = libAVEnumMediaTypes_Create(type);
yading@10:     if (!new)
yading@10:         return E_OUTOFMEMORY;
yading@10: 
yading@10:     *enumtypes = (IEnumMediaTypes *) new;
yading@10:     return S_OK;
yading@10: }
yading@10: long WINAPI
yading@10: libAVPin_QueryInternalConnections(libAVPin *this, IPin **pin,
yading@10:                                   unsigned long *npin)
yading@10: {
yading@10:     dshowdebug("libAVPin_QueryInternalConnections(%p)\n", this);
yading@10:     return E_NOTIMPL;
yading@10: }
yading@10: long WINAPI
yading@10: libAVPin_EndOfStream(libAVPin *this)
yading@10: {
yading@10:     dshowdebug("libAVPin_EndOfStream(%p)\n", this);
yading@10:     /* I don't care. */
yading@10:     return S_OK;
yading@10: }
yading@10: long WINAPI
yading@10: libAVPin_BeginFlush(libAVPin *this)
yading@10: {
yading@10:     dshowdebug("libAVPin_BeginFlush(%p)\n", this);
yading@10:     /* I don't care. */
yading@10:     return S_OK;
yading@10: }
yading@10: long WINAPI
yading@10: libAVPin_EndFlush(libAVPin *this)
yading@10: {
yading@10:     dshowdebug("libAVPin_EndFlush(%p)\n", this);
yading@10:     /* I don't care. */
yading@10:     return S_OK;
yading@10: }
yading@10: long WINAPI
yading@10: libAVPin_NewSegment(libAVPin *this, REFERENCE_TIME start, REFERENCE_TIME stop,
yading@10:                     double rate)
yading@10: {
yading@10:     dshowdebug("libAVPin_NewSegment(%p)\n", this);
yading@10:     /* I don't care. */
yading@10:     return S_OK;
yading@10: }
yading@10: 
yading@10: static int
yading@10: libAVPin_Setup(libAVPin *this, libAVFilter *filter)
yading@10: {
yading@10:     IPinVtbl *vtbl = this->vtbl;
yading@10:     IMemInputPinVtbl *imemvtbl;
yading@10: 
yading@10:     if (!filter)
yading@10:         return 0;
yading@10: 
yading@10:     imemvtbl = av_malloc(sizeof(IMemInputPinVtbl));
yading@10:     if (!imemvtbl)
yading@10:         return 0;
yading@10: 
yading@10:     SETVTBL(imemvtbl, libAVMemInputPin, QueryInterface);
yading@10:     SETVTBL(imemvtbl, libAVMemInputPin, AddRef);
yading@10:     SETVTBL(imemvtbl, libAVMemInputPin, Release);
yading@10:     SETVTBL(imemvtbl, libAVMemInputPin, GetAllocator);
yading@10:     SETVTBL(imemvtbl, libAVMemInputPin, NotifyAllocator);
yading@10:     SETVTBL(imemvtbl, libAVMemInputPin, GetAllocatorRequirements);
yading@10:     SETVTBL(imemvtbl, libAVMemInputPin, Receive);
yading@10:     SETVTBL(imemvtbl, libAVMemInputPin, ReceiveMultiple);
yading@10:     SETVTBL(imemvtbl, libAVMemInputPin, ReceiveCanBlock);
yading@10: 
yading@10:     this->imemvtbl = imemvtbl;
yading@10: 
yading@10:     SETVTBL(vtbl, libAVPin, QueryInterface);
yading@10:     SETVTBL(vtbl, libAVPin, AddRef);
yading@10:     SETVTBL(vtbl, libAVPin, Release);
yading@10:     SETVTBL(vtbl, libAVPin, Connect);
yading@10:     SETVTBL(vtbl, libAVPin, ReceiveConnection);
yading@10:     SETVTBL(vtbl, libAVPin, Disconnect);
yading@10:     SETVTBL(vtbl, libAVPin, ConnectedTo);
yading@10:     SETVTBL(vtbl, libAVPin, ConnectionMediaType);
yading@10:     SETVTBL(vtbl, libAVPin, QueryPinInfo);
yading@10:     SETVTBL(vtbl, libAVPin, QueryDirection);
yading@10:     SETVTBL(vtbl, libAVPin, QueryId);
yading@10:     SETVTBL(vtbl, libAVPin, QueryAccept);
yading@10:     SETVTBL(vtbl, libAVPin, EnumMediaTypes);
yading@10:     SETVTBL(vtbl, libAVPin, QueryInternalConnections);
yading@10:     SETVTBL(vtbl, libAVPin, EndOfStream);
yading@10:     SETVTBL(vtbl, libAVPin, BeginFlush);
yading@10:     SETVTBL(vtbl, libAVPin, EndFlush);
yading@10:     SETVTBL(vtbl, libAVPin, NewSegment);
yading@10: 
yading@10:     this->filter = filter;
yading@10: 
yading@10:     return 1;
yading@10: }
yading@10: DECLARE_CREATE(libAVPin, libAVPin_Setup(this, filter), libAVFilter *filter)
yading@10: DECLARE_DESTROY(libAVPin, nothing)
yading@10: 
yading@10: /*****************************************************************************
yading@10:  * libAVMemInputPin
yading@10:  ****************************************************************************/
yading@10: long WINAPI
yading@10: libAVMemInputPin_QueryInterface(libAVMemInputPin *this, const GUID *riid,
yading@10:                                 void **ppvObject)
yading@10: {
yading@10:     libAVPin *pin = (libAVPin *) ((uint8_t *) this - imemoffset);
yading@10:     dshowdebug("libAVMemInputPin_QueryInterface(%p)\n", this);
yading@10:     return libAVPin_QueryInterface(pin, riid, ppvObject);
yading@10: }
yading@10: unsigned long WINAPI
yading@10: libAVMemInputPin_AddRef(libAVMemInputPin *this)
yading@10: {
yading@10:     libAVPin *pin = (libAVPin *) ((uint8_t *) this - imemoffset);
yading@10:     dshowdebug("libAVMemInputPin_AddRef(%p)\n", this);
yading@10:     return libAVPin_AddRef(pin);
yading@10: }
yading@10: unsigned long WINAPI
yading@10: libAVMemInputPin_Release(libAVMemInputPin *this)
yading@10: {
yading@10:     libAVPin *pin = (libAVPin *) ((uint8_t *) this - imemoffset);
yading@10:     dshowdebug("libAVMemInputPin_Release(%p)\n", this);
yading@10:     return libAVPin_Release(pin);
yading@10: }
yading@10: long WINAPI
yading@10: libAVMemInputPin_GetAllocator(libAVMemInputPin *this, IMemAllocator **alloc)
yading@10: {
yading@10:     dshowdebug("libAVMemInputPin_GetAllocator(%p)\n", this);
yading@10:     return VFW_E_NO_ALLOCATOR;
yading@10: }
yading@10: long WINAPI
yading@10: libAVMemInputPin_NotifyAllocator(libAVMemInputPin *this, IMemAllocator *alloc,
yading@10:                                  BOOL rdwr)
yading@10: {
yading@10:     dshowdebug("libAVMemInputPin_NotifyAllocator(%p)\n", this);
yading@10:     return S_OK;
yading@10: }
yading@10: long WINAPI
yading@10: libAVMemInputPin_GetAllocatorRequirements(libAVMemInputPin *this,
yading@10:                                           ALLOCATOR_PROPERTIES *props)
yading@10: {
yading@10:     dshowdebug("libAVMemInputPin_GetAllocatorRequirements(%p)\n", this);
yading@10:     return E_NOTIMPL;
yading@10: }
yading@10: long WINAPI
yading@10: libAVMemInputPin_Receive(libAVMemInputPin *this, IMediaSample *sample)
yading@10: {
yading@10:     libAVPin *pin = (libAVPin *) ((uint8_t *) this - imemoffset);
yading@10:     enum dshowDeviceType devtype = pin->filter->type;
yading@10:     void *priv_data;
yading@10:     uint8_t *buf;
yading@10:     int buf_size;
yading@10:     int index;
yading@10:     int64_t curtime;
yading@10: 
yading@10:     dshowdebug("libAVMemInputPin_Receive(%p)\n", this);
yading@10: 
yading@10:     if (!sample)
yading@10:         return E_POINTER;
yading@10: 
yading@10:     if (devtype == VideoDevice) {
yading@10:         /* PTS from video devices is unreliable. */
yading@10:         IReferenceClock *clock = pin->filter->clock;
yading@10:         IReferenceClock_GetTime(clock, &curtime);
yading@10:     } else {
yading@10:         int64_t dummy;
yading@10:         IMediaSample_GetTime(sample, &curtime, &dummy);
yading@10:         curtime += pin->filter->start_time;
yading@10:     }
yading@10: 
yading@10:     buf_size = IMediaSample_GetActualDataLength(sample);
yading@10:     IMediaSample_GetPointer(sample, &buf);
yading@10:     priv_data = pin->filter->priv_data;
yading@10:     index = pin->filter->stream_index;
yading@10: 
yading@10:     pin->filter->callback(priv_data, index, buf, buf_size, curtime);
yading@10: 
yading@10:     return S_OK;
yading@10: }
yading@10: long WINAPI
yading@10: libAVMemInputPin_ReceiveMultiple(libAVMemInputPin *this,
yading@10:                                  IMediaSample **samples, long n, long *nproc)
yading@10: {
yading@10:     int i;
yading@10:     dshowdebug("libAVMemInputPin_ReceiveMultiple(%p)\n", this);
yading@10: 
yading@10:     for (i = 0; i < n; i++)
yading@10:         libAVMemInputPin_Receive(this, samples[i]);
yading@10: 
yading@10:     *nproc = n;
yading@10:     return S_OK;
yading@10: }
yading@10: long WINAPI
yading@10: libAVMemInputPin_ReceiveCanBlock(libAVMemInputPin *this)
yading@10: {
yading@10:     dshowdebug("libAVMemInputPin_ReceiveCanBlock(%p)\n", this);
yading@10:     /* I swear I will not block. */
yading@10:     return S_FALSE;
yading@10: }
yading@10: 
yading@10: void
yading@10: libAVMemInputPin_Destroy(libAVMemInputPin *this)
yading@10: {
yading@10:     libAVPin *pin = (libAVPin *) ((uint8_t *) this - imemoffset);
yading@10:     dshowdebug("libAVMemInputPin_Destroy(%p)\n", this);
yading@10:     libAVPin_Destroy(pin);
yading@10: }