xcopy: Add support for /C (Continue if errors).
[wine/testsucceed.git] / dlls / quartz / transform.c
blob06625230dcb0f7cf2b41b6118bfe347936828621
1 /*
2 * Transform Filter (Base for decoders, etc...)
4 * Copyright 2005 Christian Costa
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
21 #include "config.h"
23 #include "quartz_private.h"
24 #include "control_private.h"
25 #include "pin.h"
27 #include "uuids.h"
28 #include "aviriff.h"
29 #include "mmreg.h"
30 #include "vfwmsgs.h"
31 #include "amvideo.h"
32 #include "windef.h"
33 #include "winbase.h"
34 #include "dshow.h"
35 #include "strmif.h"
36 #include "vfwmsgs.h"
37 #include "evcode.h"
38 #include "vfw.h"
40 #include <assert.h>
42 #include "wine/unicode.h"
43 #include "wine/debug.h"
45 #include "transform.h"
47 WINE_DEFAULT_DEBUG_CHANNEL(quartz);
49 static const WCHAR wcsInputPinName[] = {'i','n','p','u','t',' ','p','i','n',0};
50 static const WCHAR wcsOutputPinName[] = {'o','u','t','p','u','t',' ','p','i','n',0};
52 static const IBaseFilterVtbl TransformFilter_Vtbl;
53 static const IPinVtbl TransformFilter_InputPin_Vtbl;
54 static const IMemInputPinVtbl MemInputPin_Vtbl;
55 static const IPinVtbl TransformFilter_OutputPin_Vtbl;
57 static HRESULT TransformFilter_Sample(LPVOID iface, IMediaSample * pSample)
59 TransformFilterImpl *This = (TransformFilterImpl *)iface;
60 LPBYTE pbSrcStream = NULL;
61 long cbSrcStream = 0;
62 REFERENCE_TIME tStart, tStop;
63 HRESULT hr;
65 TRACE("%p %p\n", iface, pSample);
67 hr = IMediaSample_GetPointer(pSample, &pbSrcStream);
68 if (FAILED(hr))
70 ERR("Cannot get pointer to sample data (%x)\n", hr);
71 return hr;
74 hr = IMediaSample_GetTime(pSample, &tStart, &tStop);
75 if (FAILED(hr))
76 ERR("Cannot get sample time (%x)\n", hr);
78 cbSrcStream = IMediaSample_GetActualDataLength(pSample);
80 TRACE("Sample data ptr = %p, size = %ld\n", pbSrcStream, cbSrcStream);
82 #if 0 /* For debugging purpose */
84 int i;
85 for(i = 0; i < cbSrcStream; i++)
87 if ((i!=0) && !(i%16))
88 TRACE("\n");
89 TRACE("%02x ", pbSrcStream[i]);
91 TRACE("\n");
93 #endif
95 return This->pFuncsTable->pfnProcessSampleData(This, pbSrcStream, cbSrcStream);
98 static HRESULT TransformFilter_Input_QueryAccept(LPVOID iface, const AM_MEDIA_TYPE * pmt)
100 TransformFilterImpl* This = (TransformFilterImpl*)iface;
101 TRACE("%p\n", iface);
102 dump_AM_MEDIA_TYPE(pmt);
104 return This->pFuncsTable->pfnConnectInput(This, pmt);
108 static HRESULT TransformFilter_Output_QueryAccept(LPVOID iface, const AM_MEDIA_TYPE * pmt)
110 TransformFilterImpl* pTransformFilter = (TransformFilterImpl*)iface;
111 AM_MEDIA_TYPE* outpmt = &((OutputPin*)pTransformFilter->ppPins[1])->pin.mtCurrent;
112 TRACE("%p\n", iface);
114 if (IsEqualIID(&pmt->majortype, &outpmt->majortype) && IsEqualIID(&pmt->subtype, &outpmt->subtype))
115 return S_OK;
116 return S_FALSE;
119 static HRESULT TransformFilter_InputPin_Construct(const PIN_INFO * pPinInfo, SAMPLEPROC pSampleProc, LPVOID pUserData, QUERYACCEPTPROC pQueryAccept, LPCRITICAL_SECTION pCritSec, IPin ** ppPin)
121 InputPin * pPinImpl;
123 *ppPin = NULL;
125 if (pPinInfo->dir != PINDIR_INPUT)
127 ERR("Pin direction(%x) != PINDIR_INPUT\n", pPinInfo->dir);
128 return E_INVALIDARG;
131 pPinImpl = CoTaskMemAlloc(sizeof(*pPinImpl));
133 if (!pPinImpl)
134 return E_OUTOFMEMORY;
136 if (SUCCEEDED(InputPin_Init(pPinInfo, pSampleProc, pUserData, pQueryAccept, pCritSec, pPinImpl)))
138 pPinImpl->pin.lpVtbl = &TransformFilter_InputPin_Vtbl;
139 pPinImpl->lpVtblMemInput = &MemInputPin_Vtbl;
141 *ppPin = (IPin *)(&pPinImpl->pin.lpVtbl);
142 return S_OK;
144 return E_FAIL;
147 static HRESULT TransformFilter_OutputPin_Construct(const PIN_INFO * pPinInfo, ALLOCATOR_PROPERTIES *props, LPVOID pUserData, QUERYACCEPTPROC pQueryAccept, LPCRITICAL_SECTION pCritSec, IPin ** ppPin)
149 OutputPin * pPinImpl;
151 *ppPin = NULL;
153 if (pPinInfo->dir != PINDIR_OUTPUT)
155 ERR("Pin direction(%x) != PINDIR_OUTPUT\n", pPinInfo->dir);
156 return E_INVALIDARG;
159 pPinImpl = CoTaskMemAlloc(sizeof(*pPinImpl));
161 if (!pPinImpl)
162 return E_OUTOFMEMORY;
164 if (SUCCEEDED(OutputPin_Init(pPinInfo, props, pUserData, pQueryAccept, pCritSec, pPinImpl)))
166 pPinImpl->pin.lpVtbl = &TransformFilter_OutputPin_Vtbl;
168 *ppPin = (IPin *)(&pPinImpl->pin.lpVtbl);
169 return S_OK;
171 return E_FAIL;
174 HRESULT TransformFilter_Create(TransformFilterImpl* pTransformFilter, const CLSID* pClsid, const TransformFuncsTable* pFuncsTable)
176 HRESULT hr;
177 PIN_INFO piInput;
178 PIN_INFO piOutput;
180 /* pTransformFilter is already allocated */
181 pTransformFilter->clsid = *pClsid;
182 pTransformFilter->pFuncsTable = pFuncsTable;
184 pTransformFilter->lpVtbl = &TransformFilter_Vtbl;
186 pTransformFilter->refCount = 1;
187 InitializeCriticalSection(&pTransformFilter->csFilter);
188 pTransformFilter->csFilter.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": TransformFilterImpl.csFilter");
189 pTransformFilter->state = State_Stopped;
190 pTransformFilter->pClock = NULL;
191 ZeroMemory(&pTransformFilter->filterInfo, sizeof(FILTER_INFO));
193 pTransformFilter->ppPins = CoTaskMemAlloc(2 * sizeof(IPin *));
195 /* construct input pin */
196 piInput.dir = PINDIR_INPUT;
197 piInput.pFilter = (IBaseFilter *)pTransformFilter;
198 lstrcpynW(piInput.achName, wcsInputPinName, sizeof(piInput.achName) / sizeof(piInput.achName[0]));
199 piOutput.dir = PINDIR_OUTPUT;
200 piOutput.pFilter = (IBaseFilter *)pTransformFilter;
201 lstrcpynW(piOutput.achName, wcsOutputPinName, sizeof(piOutput.achName) / sizeof(piOutput.achName[0]));
203 hr = TransformFilter_InputPin_Construct(&piInput, TransformFilter_Sample, pTransformFilter, TransformFilter_Input_QueryAccept, &pTransformFilter->csFilter, &pTransformFilter->ppPins[0]);
205 if (SUCCEEDED(hr))
207 ALLOCATOR_PROPERTIES props;
208 props.cbAlign = 1;
209 props.cbPrefix = 0;
210 props.cbBuffer = 0; /* Will be updated at connection time */
211 props.cBuffers = 2;
213 hr = TransformFilter_OutputPin_Construct(&piOutput, &props, pTransformFilter, TransformFilter_Output_QueryAccept, &pTransformFilter->csFilter, &pTransformFilter->ppPins[1]);
215 if (FAILED(hr))
216 ERR("Cannot create output pin (%x)\n", hr);
218 else
220 CoTaskMemFree(pTransformFilter->ppPins);
221 pTransformFilter->csFilter.DebugInfo->Spare[0] = 0;
222 DeleteCriticalSection(&pTransformFilter->csFilter);
223 CoTaskMemFree(pTransformFilter);
226 return hr;
229 static HRESULT WINAPI TransformFilter_QueryInterface(IBaseFilter * iface, REFIID riid, LPVOID * ppv)
231 TransformFilterImpl *This = (TransformFilterImpl *)iface;
232 TRACE("(%p/%p)->(%s, %p)\n", This, iface, qzdebugstr_guid(riid), ppv);
234 *ppv = NULL;
236 if (IsEqualIID(riid, &IID_IUnknown))
237 *ppv = (LPVOID)This;
238 else if (IsEqualIID(riid, &IID_IPersist))
239 *ppv = (LPVOID)This;
240 else if (IsEqualIID(riid, &IID_IMediaFilter))
241 *ppv = (LPVOID)This;
242 else if (IsEqualIID(riid, &IID_IBaseFilter))
243 *ppv = (LPVOID)This;
245 if (*ppv)
247 IUnknown_AddRef((IUnknown *)(*ppv));
248 return S_OK;
251 FIXME("No interface for %s!\n", qzdebugstr_guid(riid));
253 return E_NOINTERFACE;
256 static ULONG WINAPI TransformFilter_AddRef(IBaseFilter * iface)
258 TransformFilterImpl *This = (TransformFilterImpl *)iface;
259 ULONG refCount = InterlockedIncrement(&This->refCount);
261 TRACE("(%p/%p)->() AddRef from %d\n", This, iface, refCount - 1);
263 return refCount;
266 static ULONG WINAPI TransformFilter_Release(IBaseFilter * iface)
268 TransformFilterImpl *This = (TransformFilterImpl *)iface;
269 ULONG refCount = InterlockedDecrement(&This->refCount);
271 TRACE("(%p/%p)->() Release from %d\n", This, iface, refCount + 1);
273 if (!refCount)
275 ULONG i;
277 if (This->pClock)
278 IReferenceClock_Release(This->pClock);
280 for (i = 0; i < 2; i++)
282 IPin *pConnectedTo;
284 if (SUCCEEDED(IPin_ConnectedTo(This->ppPins[i], &pConnectedTo)))
286 IPin_Disconnect(pConnectedTo);
287 IPin_Release(pConnectedTo);
289 IPin_Disconnect(This->ppPins[i]);
291 IPin_Release(This->ppPins[i]);
294 CoTaskMemFree(This->ppPins);
295 This->lpVtbl = NULL;
297 This->pFuncsTable->pfnCleanup(This);
299 This->csFilter.DebugInfo->Spare[0] = 0;
300 DeleteCriticalSection(&This->csFilter);
302 TRACE("Destroying transform filter\n");
303 CoTaskMemFree(This);
305 return 0;
307 else
308 return refCount;
311 /** IPersist methods **/
313 static HRESULT WINAPI TransformFilter_GetClassID(IBaseFilter * iface, CLSID * pClsid)
315 TransformFilterImpl *This = (TransformFilterImpl *)iface;
317 TRACE("(%p/%p)->(%p)\n", This, iface, pClsid);
319 *pClsid = This->clsid;
321 return S_OK;
324 /** IMediaFilter methods **/
326 static HRESULT WINAPI TransformFilter_Stop(IBaseFilter * iface)
328 TransformFilterImpl *This = (TransformFilterImpl *)iface;
330 TRACE("(%p/%p)\n", This, iface);
332 EnterCriticalSection(&This->csFilter);
334 This->state = State_Stopped;
335 if (This->pFuncsTable->pfnProcessEnd)
336 This->pFuncsTable->pfnProcessEnd(This);
338 LeaveCriticalSection(&This->csFilter);
340 return S_OK;
343 static HRESULT WINAPI TransformFilter_Pause(IBaseFilter * iface)
345 TransformFilterImpl *This = (TransformFilterImpl *)iface;
347 TRACE("(%p/%p)->()\n", This, iface);
349 EnterCriticalSection(&This->csFilter);
351 This->state = State_Paused;
353 LeaveCriticalSection(&This->csFilter);
355 return S_OK;
358 static HRESULT WINAPI TransformFilter_Run(IBaseFilter * iface, REFERENCE_TIME tStart)
360 HRESULT hr = S_OK;
361 TransformFilterImpl *This = (TransformFilterImpl *)iface;
363 TRACE("(%p/%p)->(%s)\n", This, iface, wine_dbgstr_longlong(tStart));
365 EnterCriticalSection(&This->csFilter);
367 This->rtStreamStart = tStart;
368 This->state = State_Running;
369 OutputPin_CommitAllocator((OutputPin *)This->ppPins[1]);
370 if (This->pFuncsTable->pfnProcessBegin)
371 This->pFuncsTable->pfnProcessBegin(This);
373 LeaveCriticalSection(&This->csFilter);
375 return hr;
378 static HRESULT WINAPI TransformFilter_GetState(IBaseFilter * iface, DWORD dwMilliSecsTimeout, FILTER_STATE *pState)
380 TransformFilterImpl *This = (TransformFilterImpl *)iface;
382 TRACE("(%p/%p)->(%d, %p)\n", This, iface, dwMilliSecsTimeout, pState);
384 EnterCriticalSection(&This->csFilter);
386 *pState = This->state;
388 LeaveCriticalSection(&This->csFilter);
390 return S_OK;
393 static HRESULT WINAPI TransformFilter_SetSyncSource(IBaseFilter * iface, IReferenceClock *pClock)
395 TransformFilterImpl *This = (TransformFilterImpl *)iface;
397 TRACE("(%p/%p)->(%p)\n", This, iface, pClock);
399 EnterCriticalSection(&This->csFilter);
401 if (This->pClock)
402 IReferenceClock_Release(This->pClock);
403 This->pClock = pClock;
404 if (This->pClock)
405 IReferenceClock_AddRef(This->pClock);
407 LeaveCriticalSection(&This->csFilter);
409 return S_OK;
412 static HRESULT WINAPI TransformFilter_GetSyncSource(IBaseFilter * iface, IReferenceClock **ppClock)
414 TransformFilterImpl *This = (TransformFilterImpl *)iface;
416 TRACE("(%p/%p)->(%p)\n", This, iface, ppClock);
418 EnterCriticalSection(&This->csFilter);
420 *ppClock = This->pClock;
421 IReferenceClock_AddRef(This->pClock);
423 LeaveCriticalSection(&This->csFilter);
425 return S_OK;
428 /** IBaseFilter implementation **/
430 static HRESULT WINAPI TransformFilter_EnumPins(IBaseFilter * iface, IEnumPins **ppEnum)
432 ENUMPINDETAILS epd;
433 TransformFilterImpl *This = (TransformFilterImpl *)iface;
435 TRACE("(%p/%p)->(%p)\n", This, iface, ppEnum);
437 epd.cPins = 2; /* input and output pins */
438 epd.ppPins = This->ppPins;
439 return IEnumPinsImpl_Construct(&epd, ppEnum);
442 static HRESULT WINAPI TransformFilter_FindPin(IBaseFilter * iface, LPCWSTR Id, IPin **ppPin)
444 TransformFilterImpl *This = (TransformFilterImpl *)iface;
446 TRACE("(%p/%p)->(%p,%p)\n", This, iface, debugstr_w(Id), ppPin);
448 return E_NOTIMPL;
451 static HRESULT WINAPI TransformFilter_QueryFilterInfo(IBaseFilter * iface, FILTER_INFO *pInfo)
453 TransformFilterImpl *This = (TransformFilterImpl *)iface;
455 TRACE("(%p/%p)->(%p)\n", This, iface, pInfo);
457 strcpyW(pInfo->achName, This->filterInfo.achName);
458 pInfo->pGraph = This->filterInfo.pGraph;
460 if (pInfo->pGraph)
461 IFilterGraph_AddRef(pInfo->pGraph);
463 return S_OK;
466 static HRESULT WINAPI TransformFilter_JoinFilterGraph(IBaseFilter * iface, IFilterGraph *pGraph, LPCWSTR pName)
468 HRESULT hr = S_OK;
469 TransformFilterImpl *This = (TransformFilterImpl *)iface;
471 TRACE("(%p/%p)->(%p, %s)\n", This, iface, pGraph, debugstr_w(pName));
473 EnterCriticalSection(&This->csFilter);
475 if (pName)
476 strcpyW(This->filterInfo.achName, pName);
477 else
478 *This->filterInfo.achName = '\0';
479 This->filterInfo.pGraph = pGraph; /* NOTE: do NOT increase ref. count */
481 LeaveCriticalSection(&This->csFilter);
483 return hr;
486 static HRESULT WINAPI TransformFilter_QueryVendorInfo(IBaseFilter * iface, LPWSTR *pVendorInfo)
488 TransformFilterImpl *This = (TransformFilterImpl *)iface;
489 TRACE("(%p/%p)->(%p)\n", This, iface, pVendorInfo);
490 return E_NOTIMPL;
493 static const IBaseFilterVtbl TransformFilter_Vtbl =
495 TransformFilter_QueryInterface,
496 TransformFilter_AddRef,
497 TransformFilter_Release,
498 TransformFilter_GetClassID,
499 TransformFilter_Stop,
500 TransformFilter_Pause,
501 TransformFilter_Run,
502 TransformFilter_GetState,
503 TransformFilter_SetSyncSource,
504 TransformFilter_GetSyncSource,
505 TransformFilter_EnumPins,
506 TransformFilter_FindPin,
507 TransformFilter_QueryFilterInfo,
508 TransformFilter_JoinFilterGraph,
509 TransformFilter_QueryVendorInfo
512 static HRESULT WINAPI TransformFilter_InputPin_EndOfStream(IPin * iface)
514 InputPin* This = (InputPin*) iface;
515 TransformFilterImpl* pTransform;
516 IPin* ppin;
517 HRESULT hr;
519 TRACE("(%p)->()\n", iface);
521 /* Since we process samples synchronously, just forward notification downstream */
522 pTransform = (TransformFilterImpl*)This->pin.pinInfo.pFilter;
523 if (!pTransform)
524 hr = E_FAIL;
525 else
526 hr = IPin_ConnectedTo(pTransform->ppPins[1], &ppin);
527 if (SUCCEEDED(hr))
529 hr = IPin_EndOfStream(ppin);
530 IPin_Release(ppin);
533 if (FAILED(hr))
534 ERR("%x\n", hr);
535 return hr;
538 static const IPinVtbl TransformFilter_InputPin_Vtbl =
540 InputPin_QueryInterface,
541 IPinImpl_AddRef,
542 InputPin_Release,
543 InputPin_Connect,
544 InputPin_ReceiveConnection,
545 IPinImpl_Disconnect,
546 IPinImpl_ConnectedTo,
547 IPinImpl_ConnectionMediaType,
548 IPinImpl_QueryPinInfo,
549 IPinImpl_QueryDirection,
550 IPinImpl_QueryId,
551 IPinImpl_QueryAccept,
552 IPinImpl_EnumMediaTypes,
553 IPinImpl_QueryInternalConnections,
554 TransformFilter_InputPin_EndOfStream,
555 InputPin_BeginFlush,
556 InputPin_EndFlush,
557 InputPin_NewSegment
560 static HRESULT WINAPI TransformFilter_Output_EnumMediaTypes(IPin * iface, IEnumMediaTypes ** ppEnum)
562 IPinImpl *This = (IPinImpl *)iface;
563 ENUMMEDIADETAILS emd;
565 TRACE("(%p/%p)->(%p)\n", This, iface, ppEnum);
567 emd.cMediaTypes = 1;
568 emd.pMediaTypes = &This->mtCurrent;
570 return IEnumMediaTypesImpl_Construct(&emd, ppEnum);
573 static HRESULT WINAPI TransformFilter_Output_Disconnect(IPin * iface)
575 OutputPin* This = (OutputPin*)iface;
576 HRESULT hr;
577 TransformFilterImpl* pTransformFilter = (TransformFilterImpl*)This->pin.pinInfo.pFilter;
579 TRACE("(%p/%p)->()\n", This, iface);
581 hr = OutputPin_Disconnect(iface);
583 if (hr == S_OK)
585 pTransformFilter->pFuncsTable->pfnCleanup(pTransformFilter);
588 return hr;
591 static const IPinVtbl TransformFilter_OutputPin_Vtbl =
593 OutputPin_QueryInterface,
594 IPinImpl_AddRef,
595 OutputPin_Release,
596 OutputPin_Connect,
597 OutputPin_ReceiveConnection,
598 TransformFilter_Output_Disconnect,
599 IPinImpl_ConnectedTo,
600 IPinImpl_ConnectionMediaType,
601 IPinImpl_QueryPinInfo,
602 IPinImpl_QueryDirection,
603 IPinImpl_QueryId,
604 IPinImpl_QueryAccept,
605 TransformFilter_Output_EnumMediaTypes,
606 IPinImpl_QueryInternalConnections,
607 OutputPin_EndOfStream,
608 OutputPin_BeginFlush,
609 OutputPin_EndFlush,
610 OutputPin_NewSegment
613 static const IMemInputPinVtbl MemInputPin_Vtbl =
615 MemInputPin_QueryInterface,
616 MemInputPin_AddRef,
617 MemInputPin_Release,
618 MemInputPin_GetAllocator,
619 MemInputPin_NotifyAllocator,
620 MemInputPin_GetAllocatorRequirements,
621 MemInputPin_Receive,
622 MemInputPin_ReceiveMultiple,
623 MemInputPin_ReceiveCanBlock