comctl32: Store style bits provided with WM_CREATE.
[wine/multimedia.git] / dlls / wineqtdecoder / qtvdecoder.c
blob553a486531bb733fb642fd829a362aa2c650080d
1 /*
2 * QuickTime Toolkit decoder filter for video
4 * Copyright 2010 Aric Stewart, CodeWeavers
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 #define ULONG CoreFoundation_ULONG
24 #define HRESULT CoreFoundation_HRESULT
26 #define LoadResource __carbon_LoadResource
27 #define CompareString __carbon_CompareString
28 #define GetCurrentThread __carbon_GetCurrentThread
29 #define GetCurrentProcess __carbon_GetCurrentProcess
30 #define AnimatePalette __carbon_AnimatePalette
31 #define EqualRgn __carbon_EqualRgn
32 #define FillRgn __carbon_FillRgn
33 #define FrameRgn __carbon_FrameRgn
34 #define GetPixel __carbon_GetPixel
35 #define InvertRgn __carbon_InvertRgn
36 #define LineTo __carbon_LineTo
37 #define OffsetRgn __carbon_OffsetRgn
38 #define PaintRgn __carbon_PaintRgn
39 #define Polygon __carbon_Polygon
40 #define ResizePalette __carbon_ResizePalette
41 #define SetRectRgn __carbon_SetRectRgn
43 #define CheckMenuItem __carbon_CheckMenuItem
44 #define DeleteMenu __carbon_DeleteMenu
45 #define DrawMenuBar __carbon_DrawMenuBar
46 #define EnableMenuItem __carbon_EnableMenuItem
47 #define EqualRect __carbon_EqualRect
48 #define FillRect __carbon_FillRect
49 #define FrameRect __carbon_FrameRect
50 #define GetCursor __carbon_GetCursor
51 #define GetMenu __carbon_GetMenu
52 #define InvertRect __carbon_InvertRect
53 #define IsWindowVisible __carbon_IsWindowVisible
54 #define MoveWindow __carbon_MoveWindow
55 #define OffsetRect __carbon_OffsetRect
56 #define PtInRect __carbon_PtInRect
57 #define SetCursor __carbon_SetCursor
58 #define SetRect __carbon_SetRect
59 #define ShowCursor __carbon_ShowCursor
60 #define ShowWindow __carbon_ShowWindow
61 #define UnionRect __carbon_UnionRect
63 #include <QuickTime/ImageCompression.h>
64 #include <CoreVideo/CVPixelBuffer.h>
66 #undef LoadResource
67 #undef CompareString
68 #undef GetCurrentThread
69 #undef _CDECL
70 #undef DPRINTF
71 #undef GetCurrentProcess
72 #undef AnimatePalette
73 #undef EqualRgn
74 #undef FillRgn
75 #undef FrameRgn
76 #undef GetPixel
77 #undef InvertRgn
78 #undef LineTo
79 #undef OffsetRgn
80 #undef PaintRgn
81 #undef Polygon
82 #undef ResizePalette
83 #undef SetRectRgn
84 #undef CheckMenuItem
85 #undef DeleteMenu
86 #undef DrawMenuBar
87 #undef EnableMenuItem
88 #undef EqualRect
89 #undef FillRect
90 #undef FrameRect
91 #undef GetCursor
92 #undef GetMenu
93 #undef InvertRect
94 #undef IsWindowVisible
95 #undef MoveWindow
96 #undef OffsetRect
97 #undef PtInRect
98 #undef SetCursor
99 #undef SetRect
100 #undef ShowCursor
101 #undef ShowWindow
102 #undef UnionRect
104 #undef ULONG
105 #undef HRESULT
106 #undef DPRINTF
107 #undef STDMETHODCALLTYPE
109 #define COBJMACROS
111 #include "windef.h"
112 #include "winbase.h"
113 #include "wtypes.h"
114 #include "winuser.h"
115 #include "dshow.h"
117 #include "uuids.h"
118 #include "amvideo.h"
119 #include "strmif.h"
120 #include "vfwmsgs.h"
121 #include "vfw.h"
122 #include "dvdmedia.h"
124 #include <assert.h>
126 #include "wine/unicode.h"
127 #include "wine/debug.h"
128 #include "wine/strmbase.h"
130 #include "qtprivate.h"
132 extern CLSID CLSID_QTVDecoder;
134 WINE_DEFAULT_DEBUG_CHANNEL(qtdecoder);
136 typedef struct QTVDecoderImpl
138 TransformFilter tf;
139 IUnknown *seekthru_unk;
141 ImageDescriptionHandle hImageDescription;
142 CFMutableDictionaryRef outputBufferAttributes;
143 ICMDecompressionSessionRef decompressionSession;
144 HRESULT decodeHR;
146 DWORD outputSize;
148 } QTVDecoderImpl;
150 static const IBaseFilterVtbl QTVDecoder_Vtbl;
152 static void trackingCallback(
153 void *decompressionTrackingRefCon,
154 OSStatus result,
155 ICMDecompressionTrackingFlags decompressionTrackingFlags,
156 CVPixelBufferRef pixelBuffer,
157 TimeValue64 displayTime,
158 TimeValue64 displayDuration,
159 ICMValidTimeFlags validTimeFlags,
160 void *reserved,
161 void *sourceFrameRefCon )
163 QTVDecoderImpl *This = (QTVDecoderImpl*)decompressionTrackingRefCon;
164 IMediaSample *pSample = (IMediaSample*)sourceFrameRefCon;
165 HRESULT hr = S_OK;
166 IMediaSample* pOutSample = NULL;
167 LPBYTE pbDstStream;
168 DWORD cbDstStream;
170 if (result != noErr)
172 ERR("Error from Codec, no frame decompressed\n");
173 return;
176 if (!pixelBuffer)
178 ERR("No pixel buffer, no frame decompressed\n");
179 return;
182 EnterCriticalSection(&This->tf.filter.csFilter);
183 hr = BaseOutputPinImpl_GetDeliveryBuffer((BaseOutputPin*)This->tf.ppPins[1], &pOutSample, NULL, NULL, 0);
184 if (FAILED(hr)) {
185 ERR("Unable to get delivery buffer (%x)\n", hr);
186 goto error;
189 hr = IMediaSample_SetActualDataLength(pOutSample, 0);
190 assert(hr == S_OK);
192 hr = IMediaSample_GetPointer(pOutSample, &pbDstStream);
193 if (FAILED(hr)) {
194 ERR("Unable to get pointer to buffer (%x)\n", hr);
195 goto error;
198 cbDstStream = IMediaSample_GetSize(pOutSample);
199 if (cbDstStream < This->outputSize) {
200 ERR("Sample size is too small %d < %d\n", cbDstStream, This->outputSize);
201 hr = E_FAIL;
202 goto error;
205 hr = AccessPixelBufferPixels(pixelBuffer, pbDstStream);
206 if (FAILED(hr))
207 goto error;
209 IMediaSample_SetActualDataLength(pOutSample, This->outputSize);
211 IMediaSample_SetPreroll(pOutSample, (IMediaSample_IsPreroll(pSample) == S_OK));
212 IMediaSample_SetDiscontinuity(pOutSample, (IMediaSample_IsDiscontinuity(pSample) == S_OK));
213 IMediaSample_SetSyncPoint(pOutSample, (IMediaSample_IsSyncPoint(pSample) == S_OK));
215 if (!validTimeFlags)
216 IMediaSample_SetTime(pOutSample, NULL, NULL);
217 else
219 LONGLONG tStart, tStop;
221 if (validTimeFlags & kICMValidTime_DisplayTimeStampIsValid)
222 tStart = displayTime;
223 else
224 tStart = 0;
225 if (validTimeFlags & kICMValidTime_DisplayDurationIsValid)
226 tStop = tStart + displayDuration;
227 else
228 tStop = tStart;
230 IMediaSample_SetTime(pOutSample, &tStart, &tStop);
233 LeaveCriticalSection(&This->tf.filter.csFilter);
234 hr = BaseOutputPinImpl_Deliver((BaseOutputPin*)This->tf.ppPins[1], pOutSample);
235 if (hr != S_OK && hr != VFW_E_NOT_CONNECTED)
236 ERR("Error sending sample (%x)\n", hr);
238 error:
239 if (pOutSample)
240 IMediaSample_Release(pOutSample);
242 This->decodeHR = hr;
245 static HRESULT WINAPI QTVDecoder_StartStreaming(TransformFilter* pTransformFilter)
247 QTVDecoderImpl* This = (QTVDecoderImpl*)pTransformFilter;
248 OSErr err = noErr;
249 ICMDecompressionSessionOptionsRef sessionOptions = NULL;
250 ICMDecompressionTrackingCallbackRecord trackingCallbackRecord;
252 TRACE("(%p)->()\n", This);
254 trackingCallbackRecord.decompressionTrackingCallback = trackingCallback;
255 trackingCallbackRecord.decompressionTrackingRefCon = (void*)This;
257 err = ICMDecompressionSessionCreate(NULL, This->hImageDescription, sessionOptions, This->outputBufferAttributes, &trackingCallbackRecord, &This->decompressionSession);
259 if (err != noErr)
261 ERR("Error with ICMDecompressionSessionCreate %i\n",err);
262 return E_FAIL;
265 return S_OK;
268 static HRESULT WINAPI QTVDecoder_Receive(TransformFilter *tf, IMediaSample *pSample)
270 QTVDecoderImpl* This = (QTVDecoderImpl *)tf;
271 HRESULT hr;
272 DWORD cbSrcStream;
273 LPBYTE pbSrcStream;
275 ICMFrameTimeRecord frameTime = {{0}};
276 TimeValue time = 1;
277 TimeScale timeScale = 1;
278 OSStatus err = noErr;
279 LONGLONG tStart, tStop;
281 EnterCriticalSection(&This->tf.filter.csFilter);
282 hr = IMediaSample_GetPointer(pSample, &pbSrcStream);
283 if (FAILED(hr))
285 ERR("Cannot get pointer to sample data (%x)\n", hr);
286 goto error;
289 cbSrcStream = IMediaSample_GetActualDataLength(pSample);
291 if (IMediaSample_GetTime(pSample, &tStart, &tStop) != S_OK)
292 tStart = tStop = 0;
294 time = tStart;
296 frameTime.recordSize = sizeof(ICMFrameTimeRecord);
297 *(TimeValue64 *)&frameTime.value = tStart;
298 frameTime.scale = 1;
299 frameTime.rate = fixed1;
300 frameTime.duration = tStop - tStart;
301 frameTime.frameNumber = 0;
302 frameTime.flags = icmFrameTimeIsNonScheduledDisplayTime;
304 err = ICMDecompressionSessionDecodeFrame(This->decompressionSession,
305 (UInt8 *)pbSrcStream, cbSrcStream, NULL, &frameTime, pSample);
307 if (err != noErr)
309 ERR("Error with ICMDecompressionSessionDecodeFrame\n");
310 hr = E_FAIL;
311 goto error;
314 ICMDecompressionSessionSetNonScheduledDisplayTime(This->decompressionSession, time, timeScale, 0);
315 ICMDecompressionSessionFlush(This->decompressionSession);
316 hr = This->decodeHR;
318 error:
319 LeaveCriticalSection(&This->tf.filter.csFilter);
320 return hr;
323 static HRESULT WINAPI QTVDecoder_StopStreaming(TransformFilter* pTransformFilter)
325 QTVDecoderImpl* This = (QTVDecoderImpl*)pTransformFilter;
327 TRACE("(%p)->()\n", This);
329 if (This->decompressionSession)
330 ICMDecompressionSessionRelease(This->decompressionSession);
331 This->decompressionSession = NULL;
333 return S_OK;
336 static HRESULT WINAPI QTVDecoder_SetMediaType(TransformFilter *tf, PIN_DIRECTION dir, const AM_MEDIA_TYPE * pmt)
338 QTVDecoderImpl* This = (QTVDecoderImpl*)tf;
339 HRESULT hr = VFW_E_TYPE_NOT_ACCEPTED;
340 OSErr err = noErr;
341 AM_MEDIA_TYPE *outpmt = &This->tf.pmt;
342 CFNumberRef n = NULL;
344 TRACE("(%p)->(%p)\n", This, pmt);
346 if (dir != PINDIR_INPUT)
347 return S_OK;
349 FreeMediaType(outpmt);
350 CopyMediaType(outpmt, pmt);
352 if (This->hImageDescription)
353 DisposeHandle((Handle)This->hImageDescription);
355 This->hImageDescription = (ImageDescriptionHandle)
356 NewHandleClear(sizeof(ImageDescription));
358 if (This->hImageDescription != NULL)
360 (**This->hImageDescription).idSize = sizeof(ImageDescription);
361 (**This->hImageDescription).spatialQuality = codecNormalQuality;
362 (**This->hImageDescription).frameCount = 1;
363 (**This->hImageDescription).clutID = -1;
365 else
367 ERR("Failed to create ImageDescription\n");
368 goto failed;
371 /* Check root (GUID w/o FOURCC) */
372 if ((IsEqualIID(&pmt->majortype, &MEDIATYPE_Video)) &&
373 (!memcmp(((const char *)&pmt->subtype)+4, ((const char *)&MEDIATYPE_Video)+4, sizeof(GUID)-4)))
375 VIDEOINFOHEADER *format1 = (VIDEOINFOHEADER *)outpmt->pbFormat;
376 VIDEOINFOHEADER2 *format2 = (VIDEOINFOHEADER2 *)outpmt->pbFormat;
377 BITMAPINFOHEADER *bmi;
378 OSType fourCC;
379 DecompressorComponent dc;
380 OSType format;
381 DWORD outputWidth, outputHeight, outputDepth;
383 if (IsEqualIID(&pmt->formattype, &FORMAT_VideoInfo))
384 bmi = &format1->bmiHeader;
385 else if (IsEqualIID(&pmt->formattype, &FORMAT_VideoInfo2))
386 bmi = &format2->bmiHeader;
387 else
388 goto failed;
390 TRACE("Fourcc: %s\n", debugstr_an((const char *)&pmt->subtype.Data1, 4));
391 fourCC = ((const char *)&pmt->subtype.Data1)[3] |
392 (((const char *)&pmt->subtype.Data1)[2]<<8) |
393 (((const char *)&pmt->subtype.Data1)[1]<<16) |
394 (((const char *)&pmt->subtype.Data1)[0]<<24);
396 err = FindCodec(fourCC,NULL,NULL,&dc);
397 if (err != noErr || dc == 0x0)
399 TRACE("Codec not found\n");
400 goto failed;
403 outputWidth = bmi->biWidth;
404 outputHeight = bmi->biHeight;
406 (**This->hImageDescription).cType = fourCC;
407 (**This->hImageDescription).width = outputWidth;
408 (**This->hImageDescription).height = outputHeight;
409 (**This->hImageDescription).depth = bmi->biBitCount;
410 (**This->hImageDescription).hRes = 72<<16;
411 (**This->hImageDescription).vRes = 72<<16;
413 if (This->outputBufferAttributes)
414 CFRelease(This->outputBufferAttributes);
416 This->outputBufferAttributes = CFDictionaryCreateMutable(NULL, 0, &kCFTypeDictionaryKeyCallBacks, &kCFTypeDictionaryValueCallBacks);
417 if (!This->outputBufferAttributes)
419 ERR("Failed to create outputBufferAttributes\n");
420 goto failed;
423 n = CFNumberCreate(NULL, kCFNumberIntType, &outputWidth);
424 CFDictionaryAddValue(This->outputBufferAttributes, kCVPixelBufferWidthKey, n);
425 CFRelease(n);
427 n = CFNumberCreate(NULL, kCFNumberIntType, &outputHeight);
428 CFDictionaryAddValue(This->outputBufferAttributes, kCVPixelBufferHeightKey, n);
429 CFRelease(n);
431 /* yes this looks wrong. but 32ARGB is 24 RGB with an alpha channel */
432 format = k32ARGBPixelFormat;
433 n = CFNumberCreate(NULL, kCFNumberIntType, &format);
434 CFDictionaryAddValue(This->outputBufferAttributes, kCVPixelBufferPixelFormatTypeKey, n);
435 CFRelease(n);
437 CFDictionaryAddValue(This->outputBufferAttributes, kCVPixelBufferCGBitmapContextCompatibilityKey, kCFBooleanTrue);
438 CFDictionaryAddValue(This->outputBufferAttributes, kCVPixelBufferCGImageCompatibilityKey, kCFBooleanTrue);
440 outputDepth = 3;
441 This->outputSize = outputWidth * outputHeight * outputDepth;
442 bmi->biCompression = BI_RGB;
443 bmi->biBitCount = 24;
444 outpmt->subtype = MEDIASUBTYPE_RGB24;
445 if (bmi->biHeight > 0)
446 bmi->biHeight = -bmi->biHeight;
448 return S_OK;
451 failed:
452 if (This->hImageDescription)
454 DisposeHandle((Handle)This->hImageDescription);
455 This->hImageDescription = NULL;
457 if (This->outputBufferAttributes)
459 CFRelease(This->outputBufferAttributes);
460 This->outputBufferAttributes = NULL;
463 TRACE("Connection refused\n");
464 return hr;
467 static HRESULT WINAPI QTVDecoder_BreakConnect(TransformFilter *tf, PIN_DIRECTION dir)
469 QTVDecoderImpl *This = (QTVDecoderImpl *)tf;
471 TRACE("(%p)->()\n", This);
473 if (This->hImageDescription)
474 DisposeHandle((Handle)This->hImageDescription);
475 if (This->outputBufferAttributes)
476 CFRelease(This->outputBufferAttributes);
478 This->hImageDescription = NULL;
479 This->outputBufferAttributes = NULL;
481 return S_OK;
484 static HRESULT WINAPI QTVDecoder_DecideBufferSize(TransformFilter *tf, IMemAllocator *pAlloc, ALLOCATOR_PROPERTIES *ppropInputRequest)
486 QTVDecoderImpl *This = (QTVDecoderImpl*)tf;
487 ALLOCATOR_PROPERTIES actual;
489 TRACE("()\n");
491 if (!ppropInputRequest->cbAlign)
492 ppropInputRequest->cbAlign = 1;
494 if (ppropInputRequest->cbBuffer < This->outputSize)
495 ppropInputRequest->cbBuffer = This->outputSize;
497 if (!ppropInputRequest->cBuffers)
498 ppropInputRequest->cBuffers = 1;
500 return IMemAllocator_SetProperties(pAlloc, ppropInputRequest, &actual);
503 static const TransformFilterFuncTable QTVDecoder_FuncsTable = {
504 QTVDecoder_DecideBufferSize,
505 QTVDecoder_StartStreaming,
506 QTVDecoder_Receive,
507 QTVDecoder_StopStreaming,
508 NULL,
509 QTVDecoder_SetMediaType,
510 NULL,
511 QTVDecoder_BreakConnect,
512 NULL,
513 NULL,
514 NULL,
515 NULL
518 IUnknown * CALLBACK QTVDecoder_create(IUnknown * pUnkOuter, HRESULT* phr)
520 HRESULT hr;
521 QTVDecoderImpl * This;
523 TRACE("(%p, %p)\n", pUnkOuter, phr);
525 *phr = S_OK;
527 if (pUnkOuter)
529 *phr = CLASS_E_NOAGGREGATION;
530 return NULL;
533 hr = TransformFilter_Construct(&QTVDecoder_Vtbl, sizeof(QTVDecoderImpl), &CLSID_QTVDecoder, &QTVDecoder_FuncsTable, (IBaseFilter**)&This);
535 if (FAILED(hr))
537 *phr = hr;
538 return NULL;
540 else
542 ISeekingPassThru *passthru;
543 hr = CoCreateInstance(&CLSID_SeekingPassThru, (IUnknown*)This, CLSCTX_INPROC_SERVER, &IID_IUnknown, (void**)&This->seekthru_unk);
544 IUnknown_QueryInterface(This->seekthru_unk, &IID_ISeekingPassThru, (void**)&passthru);
545 ISeekingPassThru_Init(passthru, FALSE, (IPin*)This->tf.ppPins[0]);
546 ISeekingPassThru_Release(passthru);
549 *phr = hr;
550 return (IUnknown*)This;
553 HRESULT WINAPI QTVDecoder_QueryInterface(IBaseFilter * iface, REFIID riid, LPVOID * ppv)
555 HRESULT hr;
556 QTVDecoderImpl *This = (QTVDecoderImpl *)iface;
557 TRACE("(%p/%p)->(%s, %p)\n", This, iface, debugstr_guid(riid), ppv);
559 if (IsEqualIID(riid, &IID_IMediaSeeking))
560 return IUnknown_QueryInterface(This->seekthru_unk, riid, ppv);
562 hr = TransformFilterImpl_QueryInterface(iface, riid, ppv);
564 return hr;
567 static const IBaseFilterVtbl QTVDecoder_Vtbl =
569 QTVDecoder_QueryInterface,
570 BaseFilterImpl_AddRef,
571 TransformFilterImpl_Release,
572 BaseFilterImpl_GetClassID,
573 TransformFilterImpl_Stop,
574 TransformFilterImpl_Pause,
575 TransformFilterImpl_Run,
576 BaseFilterImpl_GetState,
577 BaseFilterImpl_SetSyncSource,
578 BaseFilterImpl_GetSyncSource,
579 BaseFilterImpl_EnumPins,
580 TransformFilterImpl_FindPin,
581 BaseFilterImpl_QueryFilterInfo,
582 BaseFilterImpl_JoinFilterGraph,
583 BaseFilterImpl_QueryVendorInfo