1/* 2 * DirectShow capture interface 3 * Copyright (c) 2010 Ramiro Polla 4 * 5 * This file is part of FFmpeg. 6 * 7 * FFmpeg is free software; you can redistribute it and/or 8 * modify it under the terms of the GNU Lesser General Public 9 * License as published by the Free Software Foundation; either 10 * version 2.1 of the License, or (at your option) any later version. 11 * 12 * FFmpeg is distributed in the hope that it will be useful, 13 * but WITHOUT ANY WARRANTY; without even the implied warranty of 14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 15 * Lesser General Public License for more details. 16 * 17 * You should have received a copy of the GNU Lesser General Public 18 * License along with FFmpeg; if not, write to the Free Software 19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA 20 */ 21 22#include "dshow_capture.h" 23 24#include <stddef.h> 25#define imemoffset offsetof(libAVPin, imemvtbl) 26 27DECLARE_QUERYINTERFACE(libAVPin, 28 { {&IID_IUnknown,0}, {&IID_IPin,0}, {&IID_IMemInputPin,imemoffset} }) 29DECLARE_ADDREF(libAVPin) 30DECLARE_RELEASE(libAVPin) 31 32long WINAPI 33libAVPin_Connect(libAVPin *this, IPin *pin, const AM_MEDIA_TYPE *type) 34{ 35 dshowdebug("libAVPin_Connect(%p, %p, %p)\n", this, pin, type); 36 /* Input pins receive connections. */ 37 return S_FALSE; 38} 39long WINAPI 40libAVPin_ReceiveConnection(libAVPin *this, IPin *pin, 41 const AM_MEDIA_TYPE *type) 42{ 43 enum dshowDeviceType devtype = this->filter->type; 44 dshowdebug("libAVPin_ReceiveConnection(%p)\n", this); 45 46 if (!pin) 47 return E_POINTER; 48 if (this->connectedto) 49 return VFW_E_ALREADY_CONNECTED; 50 51 ff_print_AM_MEDIA_TYPE(type); 52 if (devtype == VideoDevice) { 53 if (!IsEqualGUID(&type->majortype, &MEDIATYPE_Video)) 54 return VFW_E_TYPE_NOT_ACCEPTED; 55 } else { 56 if (!IsEqualGUID(&type->majortype, &MEDIATYPE_Audio)) 57 return VFW_E_TYPE_NOT_ACCEPTED; 58 } 59 60 IPin_AddRef(pin); 61 this->connectedto = pin; 62 63 ff_copy_dshow_media_type(&this->type, type); 64 65 return S_OK; 66} 67long WINAPI 68libAVPin_Disconnect(libAVPin *this) 69{ 70 dshowdebug("libAVPin_Disconnect(%p)\n", this); 71 72 if (this->filter->state != State_Stopped) 73 return VFW_E_NOT_STOPPED; 74 if (!this->connectedto) 75 return S_FALSE; 76 IPin_Release(this->connectedto); 77 this->connectedto = NULL; 78 79 return S_OK; 80} 81long WINAPI 82libAVPin_ConnectedTo(libAVPin *this, IPin **pin) 83{ 84 dshowdebug("libAVPin_ConnectedTo(%p)\n", this); 85 86 if (!pin) 87 return E_POINTER; 88 if (!this->connectedto) 89 return VFW_E_NOT_CONNECTED; 90 IPin_AddRef(this->connectedto); 91 *pin = this->connectedto; 92 93 return S_OK; 94} 95long WINAPI 96libAVPin_ConnectionMediaType(libAVPin *this, AM_MEDIA_TYPE *type) 97{ 98 dshowdebug("libAVPin_ConnectionMediaType(%p)\n", this); 99 100 if (!type) 101 return E_POINTER; 102 if (!this->connectedto) 103 return VFW_E_NOT_CONNECTED; 104 105 return ff_copy_dshow_media_type(type, &this->type); 106} 107long WINAPI 108libAVPin_QueryPinInfo(libAVPin *this, PIN_INFO *info) 109{ 110 dshowdebug("libAVPin_QueryPinInfo(%p)\n", this); 111 112 if (!info) 113 return E_POINTER; 114 115 if (this->filter) 116 libAVFilter_AddRef(this->filter); 117 118 info->pFilter = (IBaseFilter *) this->filter; 119 info->dir = PINDIR_INPUT; 120 wcscpy(info->achName, L"Capture"); 121 122 return S_OK; 123} 124long WINAPI 125libAVPin_QueryDirection(libAVPin *this, PIN_DIRECTION *dir) 126{ 127 dshowdebug("libAVPin_QueryDirection(%p)\n", this); 128 if (!dir) 129 return E_POINTER; 130 *dir = PINDIR_INPUT; 131 return S_OK; 132} 133long WINAPI 134libAVPin_QueryId(libAVPin *this, wchar_t **id) 135{ 136 dshowdebug("libAVPin_QueryId(%p)\n", this); 137 138 if (!id) 139 return E_POINTER; 140 141 *id = wcsdup(L"libAV Pin"); 142 143 return S_OK; 144} 145long WINAPI 146libAVPin_QueryAccept(libAVPin *this, const AM_MEDIA_TYPE *type) 147{ 148 dshowdebug("libAVPin_QueryAccept(%p)\n", this); 149 return S_FALSE; 150} 151long WINAPI 152libAVPin_EnumMediaTypes(libAVPin *this, IEnumMediaTypes **enumtypes) 153{ 154 const AM_MEDIA_TYPE *type = NULL; 155 libAVEnumMediaTypes *new; 156 dshowdebug("libAVPin_EnumMediaTypes(%p)\n", this); 157 158 if (!enumtypes) 159 return E_POINTER; 160 new = libAVEnumMediaTypes_Create(type); 161 if (!new) 162 return E_OUTOFMEMORY; 163 164 *enumtypes = (IEnumMediaTypes *) new; 165 return S_OK; 166} 167long WINAPI 168libAVPin_QueryInternalConnections(libAVPin *this, IPin **pin, 169 unsigned long *npin) 170{ 171 dshowdebug("libAVPin_QueryInternalConnections(%p)\n", this); 172 return E_NOTIMPL; 173} 174long WINAPI 175libAVPin_EndOfStream(libAVPin *this) 176{ 177 dshowdebug("libAVPin_EndOfStream(%p)\n", this); 178 /* I don't care. */ 179 return S_OK; 180} 181long WINAPI 182libAVPin_BeginFlush(libAVPin *this) 183{ 184 dshowdebug("libAVPin_BeginFlush(%p)\n", this); 185 /* I don't care. */ 186 return S_OK; 187} 188long WINAPI 189libAVPin_EndFlush(libAVPin *this) 190{ 191 dshowdebug("libAVPin_EndFlush(%p)\n", this); 192 /* I don't care. */ 193 return S_OK; 194} 195long WINAPI 196libAVPin_NewSegment(libAVPin *this, REFERENCE_TIME start, REFERENCE_TIME stop, 197 double rate) 198{ 199 dshowdebug("libAVPin_NewSegment(%p)\n", this); 200 /* I don't care. */ 201 return S_OK; 202} 203 204static int 205libAVPin_Setup(libAVPin *this, libAVFilter *filter) 206{ 207 IPinVtbl *vtbl = this->vtbl; 208 IMemInputPinVtbl *imemvtbl; 209 210 if (!filter) 211 return 0; 212 213 imemvtbl = av_malloc(sizeof(IMemInputPinVtbl)); 214 if (!imemvtbl) 215 return 0; 216 217 SETVTBL(imemvtbl, libAVMemInputPin, QueryInterface); 218 SETVTBL(imemvtbl, libAVMemInputPin, AddRef); 219 SETVTBL(imemvtbl, libAVMemInputPin, Release); 220 SETVTBL(imemvtbl, libAVMemInputPin, GetAllocator); 221 SETVTBL(imemvtbl, libAVMemInputPin, NotifyAllocator); 222 SETVTBL(imemvtbl, libAVMemInputPin, GetAllocatorRequirements); 223 SETVTBL(imemvtbl, libAVMemInputPin, Receive); 224 SETVTBL(imemvtbl, libAVMemInputPin, ReceiveMultiple); 225 SETVTBL(imemvtbl, libAVMemInputPin, ReceiveCanBlock); 226 227 this->imemvtbl = imemvtbl; 228 229 SETVTBL(vtbl, libAVPin, QueryInterface); 230 SETVTBL(vtbl, libAVPin, AddRef); 231 SETVTBL(vtbl, libAVPin, Release); 232 SETVTBL(vtbl, libAVPin, Connect); 233 SETVTBL(vtbl, libAVPin, ReceiveConnection); 234 SETVTBL(vtbl, libAVPin, Disconnect); 235 SETVTBL(vtbl, libAVPin, ConnectedTo); 236 SETVTBL(vtbl, libAVPin, ConnectionMediaType); 237 SETVTBL(vtbl, libAVPin, QueryPinInfo); 238 SETVTBL(vtbl, libAVPin, QueryDirection); 239 SETVTBL(vtbl, libAVPin, QueryId); 240 SETVTBL(vtbl, libAVPin, QueryAccept); 241 SETVTBL(vtbl, libAVPin, EnumMediaTypes); 242 SETVTBL(vtbl, libAVPin, QueryInternalConnections); 243 SETVTBL(vtbl, libAVPin, EndOfStream); 244 SETVTBL(vtbl, libAVPin, BeginFlush); 245 SETVTBL(vtbl, libAVPin, EndFlush); 246 SETVTBL(vtbl, libAVPin, NewSegment); 247 248 this->filter = filter; 249 250 return 1; 251} 252DECLARE_CREATE(libAVPin, libAVPin_Setup(this, filter), libAVFilter *filter) 253DECLARE_DESTROY(libAVPin, nothing) 254 255/***************************************************************************** 256 * libAVMemInputPin 257 ****************************************************************************/ 258long WINAPI 259libAVMemInputPin_QueryInterface(libAVMemInputPin *this, const GUID *riid, 260 void **ppvObject) 261{ 262 libAVPin *pin = (libAVPin *) ((uint8_t *) this - imemoffset); 263 dshowdebug("libAVMemInputPin_QueryInterface(%p)\n", this); 264 return libAVPin_QueryInterface(pin, riid, ppvObject); 265} 266unsigned long WINAPI 267libAVMemInputPin_AddRef(libAVMemInputPin *this) 268{ 269 libAVPin *pin = (libAVPin *) ((uint8_t *) this - imemoffset); 270 dshowdebug("libAVMemInputPin_AddRef(%p)\n", this); 271 return libAVPin_AddRef(pin); 272} 273unsigned long WINAPI 274libAVMemInputPin_Release(libAVMemInputPin *this) 275{ 276 libAVPin *pin = (libAVPin *) ((uint8_t *) this - imemoffset); 277 dshowdebug("libAVMemInputPin_Release(%p)\n", this); 278 return libAVPin_Release(pin); 279} 280long WINAPI 281libAVMemInputPin_GetAllocator(libAVMemInputPin *this, IMemAllocator **alloc) 282{ 283 dshowdebug("libAVMemInputPin_GetAllocator(%p)\n", this); 284 return VFW_E_NO_ALLOCATOR; 285} 286long WINAPI 287libAVMemInputPin_NotifyAllocator(libAVMemInputPin *this, IMemAllocator *alloc, 288 BOOL rdwr) 289{ 290 dshowdebug("libAVMemInputPin_NotifyAllocator(%p)\n", this); 291 return S_OK; 292} 293long WINAPI 294libAVMemInputPin_GetAllocatorRequirements(libAVMemInputPin *this, 295 ALLOCATOR_PROPERTIES *props) 296{ 297 dshowdebug("libAVMemInputPin_GetAllocatorRequirements(%p)\n", this); 298 return E_NOTIMPL; 299} 300long WINAPI 301libAVMemInputPin_Receive(libAVMemInputPin *this, IMediaSample *sample) 302{ 303 libAVPin *pin = (libAVPin *) ((uint8_t *) this - imemoffset); 304 enum dshowDeviceType devtype = pin->filter->type; 305 void *priv_data; 306 uint8_t *buf; 307 int buf_size; 308 int index; 309 int64_t curtime; 310 311 dshowdebug("libAVMemInputPin_Receive(%p)\n", this); 312 313 if (!sample) 314 return E_POINTER; 315 316 if (devtype == VideoDevice) { 317 /* PTS from video devices is unreliable. */ 318 IReferenceClock *clock = pin->filter->clock; 319 IReferenceClock_GetTime(clock, &curtime); 320 } else { 321 int64_t dummy; 322 IMediaSample_GetTime(sample, &curtime, &dummy); 323 curtime += pin->filter->start_time; 324 } 325 326 buf_size = IMediaSample_GetActualDataLength(sample); 327 IMediaSample_GetPointer(sample, &buf); 328 priv_data = pin->filter->priv_data; 329 index = pin->filter->stream_index; 330 331 pin->filter->callback(priv_data, index, buf, buf_size, curtime, devtype); 332 333 return S_OK; 334} 335long WINAPI 336libAVMemInputPin_ReceiveMultiple(libAVMemInputPin *this, 337 IMediaSample **samples, long n, long *nproc) 338{ 339 int i; 340 dshowdebug("libAVMemInputPin_ReceiveMultiple(%p)\n", this); 341 342 for (i = 0; i < n; i++) 343 libAVMemInputPin_Receive(this, samples[i]); 344 345 *nproc = n; 346 return S_OK; 347} 348long WINAPI 349libAVMemInputPin_ReceiveCanBlock(libAVMemInputPin *this) 350{ 351 dshowdebug("libAVMemInputPin_ReceiveCanBlock(%p)\n", this); 352 /* I swear I will not block. */ 353 return S_FALSE; 354} 355 356void 357libAVMemInputPin_Destroy(libAVMemInputPin *this) 358{ 359 libAVPin *pin = (libAVPin *) ((uint8_t *) this - imemoffset); 360 dshowdebug("libAVMemInputPin_Destroy(%p)\n", this); 361 libAVPin_Destroy(pin); 362} 363