2 * AVI Decompressor (VFW decompressors wrapper)
4 * Copyright 2004-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
23 #include "quartz_private.h"
38 #include "wine/unicode.h"
39 #include "wine/debug.h"
41 #include "transform.h"
43 WINE_DEFAULT_DEBUG_CHANNEL(quartz
);
45 static HRESULT
AVIDec_Cleanup(TransformFilterImpl
* pTransformFilter
);
47 typedef struct AVIDecImpl
49 TransformFilterImpl tf
;
51 BITMAPINFOHEADER
* pBihIn
;
52 BITMAPINFOHEADER
* pBihOut
;
55 static HRESULT
AVIDec_ProcessBegin(TransformFilterImpl
* pTransformFilter
)
57 AVIDecImpl
* This
= (AVIDecImpl
*)pTransformFilter
;
60 TRACE("(%p)->()\n", This
);
62 result
= ICDecompressBegin(This
->hvid
, This
->pBihIn
, This
->pBihOut
);
63 if (result
!= ICERR_OK
)
65 ERR("Cannot start processing (%d)\n", result
);
71 static HRESULT
AVIDec_ProcessSampleData(TransformFilterImpl
* pTransformFilter
, IMediaSample
*pSample
)
73 AVIDecImpl
* This
= (AVIDecImpl
*)pTransformFilter
;
77 IMediaSample
* pOutSample
= NULL
;
82 LONGLONG tStart
, tStop
;
83 InputPin
*pin
= (InputPin
*)pTransformFilter
->ppPins
[0];
85 EnterCriticalSection(&This
->tf
.csFilter
);
86 if (This
->tf
.state
== State_Stopped
)
88 LeaveCriticalSection(&This
->tf
.csFilter
);
89 return VFW_E_WRONG_STATE
;
92 if (pin
->end_of_stream
|| pin
->flushing
)
94 LeaveCriticalSection(&This
->tf
.csFilter
);
98 hr
= IMediaSample_GetPointer(pSample
, &pbSrcStream
);
101 ERR("Cannot get pointer to sample data (%x)\n", hr
);
105 cbSrcStream
= IMediaSample_GetActualDataLength(pSample
);
107 TRACE("Sample data ptr = %p, size = %ld\n", pbSrcStream
, (long)cbSrcStream
);
109 hr
= IPin_ConnectionMediaType(This
->tf
.ppPins
[0], &amt
);
111 ERR("Unable to retrieve media type\n");
115 /* Update input size to match sample size */
116 This
->pBihIn
->biSizeImage
= cbSrcStream
;
118 hr
= OutputPin_GetDeliveryBuffer((OutputPin
*)This
->tf
.ppPins
[1], &pOutSample
, NULL
, NULL
, 0);
120 ERR("Unable to get delivery buffer (%x)\n", hr
);
124 hr
= IMediaSample_SetActualDataLength(pOutSample
, 0);
127 hr
= IMediaSample_GetPointer(pOutSample
, &pbDstStream
);
129 ERR("Unable to get pointer to buffer (%x)\n", hr
);
132 cbDstStream
= IMediaSample_GetSize(pOutSample
);
133 if (cbDstStream
< This
->pBihOut
->biSizeImage
) {
134 ERR("Sample size is too small %d < %d\n", cbDstStream
, This
->pBihOut
->biSizeImage
);
139 res
= ICDecompress(This
->hvid
, 0, This
->pBihIn
, pbSrcStream
, This
->pBihOut
, pbDstStream
);
141 ERR("Error occurred during the decompression (%x)\n", res
);
143 IMediaSample_SetActualDataLength(pOutSample
, This
->pBihOut
->biSizeImage
);
145 IMediaSample_SetPreroll(pOutSample
, (IMediaSample_IsPreroll(pSample
) == S_OK
));
146 IMediaSample_SetDiscontinuity(pOutSample
, (IMediaSample_IsDiscontinuity(pSample
) == S_OK
));
147 IMediaSample_SetSyncPoint(pOutSample
, (IMediaSample_IsSyncPoint(pSample
) == S_OK
));
149 if (IMediaSample_GetTime(pSample
, &tStart
, &tStop
) == S_OK
)
150 IMediaSample_SetTime(pOutSample
, &tStart
, &tStop
);
152 IMediaSample_SetTime(pOutSample
, NULL
, NULL
);
154 LeaveCriticalSection(&This
->tf
.csFilter
);
155 hr
= OutputPin_SendSample((OutputPin
*)This
->tf
.ppPins
[1], pOutSample
);
156 if (hr
!= S_OK
&& hr
!= VFW_E_NOT_CONNECTED
)
157 ERR("Error sending sample (%x)\n", hr
);
158 IMediaSample_Release(pOutSample
);
163 IMediaSample_Release(pOutSample
);
165 LeaveCriticalSection(&This
->tf
.csFilter
);
169 static HRESULT
AVIDec_ProcessEnd(TransformFilterImpl
* pTransformFilter
)
171 AVIDecImpl
* This
= (AVIDecImpl
*)pTransformFilter
;
174 TRACE("(%p)->()\n", This
);
179 result
= ICDecompressEnd(This
->hvid
);
180 if (result
!= ICERR_OK
)
182 ERR("Cannot stop processing (%d)\n", result
);
188 static HRESULT
AVIDec_ConnectInput(TransformFilterImpl
* pTransformFilter
, const AM_MEDIA_TYPE
* pmt
)
190 AVIDecImpl
* This
= (AVIDecImpl
*)pTransformFilter
;
191 HRESULT hr
= VFW_E_TYPE_NOT_ACCEPTED
;
193 TRACE("(%p)->(%p)\n", This
, pmt
);
195 AVIDec_Cleanup(pTransformFilter
);
197 /* Check root (GUID w/o FOURCC) */
198 if ((IsEqualIID(&pmt
->majortype
, &MEDIATYPE_Video
)) &&
199 (!memcmp(((const char *)&pmt
->subtype
)+4, ((const char *)&MEDIATYPE_Video
)+4, sizeof(GUID
)-4)))
201 VIDEOINFOHEADER
*format1
= (VIDEOINFOHEADER
*)pmt
->pbFormat
;
202 VIDEOINFOHEADER2
*format2
= (VIDEOINFOHEADER2
*)pmt
->pbFormat
;
203 BITMAPINFOHEADER
*bmi
;
205 if (IsEqualIID(&pmt
->formattype
, &FORMAT_VideoInfo
))
206 bmi
= &format1
->bmiHeader
;
207 else if (IsEqualIID(&pmt
->formattype
, &FORMAT_VideoInfo2
))
208 bmi
= &format2
->bmiHeader
;
211 TRACE("Fourcc: %s\n", debugstr_an((char *)&pmt
->subtype
.Data1
, 4));
213 This
->hvid
= ICLocate(pmt
->majortype
.Data1
, pmt
->subtype
.Data1
, bmi
, NULL
, ICMODE_DECOMPRESS
);
216 AM_MEDIA_TYPE
* outpmt
= &This
->tf
.pmt
;
217 const CLSID
* outsubtype
;
219 DWORD output_depth
= bmi
->biBitCount
;
221 FreeMediaType(outpmt
);
223 switch(bmi
->biBitCount
)
225 case 32: outsubtype
= &MEDIASUBTYPE_RGB32
; break;
226 case 24: outsubtype
= &MEDIASUBTYPE_RGB24
; break;
227 case 16: outsubtype
= &MEDIASUBTYPE_RGB565
; break;
228 case 8: outsubtype
= &MEDIASUBTYPE_RGB8
; break;
230 WARN("Non standard input depth %d, forced output depth to 32\n", bmi
->biBitCount
);
231 outsubtype
= &MEDIASUBTYPE_RGB32
;
236 /* Copy bitmap header from media type to 1 for input and 1 for output */
237 bih_size
= bmi
->biSize
+ bmi
->biClrUsed
* 4;
238 This
->pBihIn
= CoTaskMemAlloc(bih_size
);
244 This
->pBihOut
= CoTaskMemAlloc(bih_size
);
250 memcpy(This
->pBihIn
, bmi
, bih_size
);
251 memcpy(This
->pBihOut
, bmi
, bih_size
);
253 /* Update output format as non compressed bitmap */
254 This
->pBihOut
->biCompression
= 0;
255 This
->pBihOut
->biBitCount
= output_depth
;
256 This
->pBihOut
->biSizeImage
= This
->pBihOut
->biWidth
* This
->pBihOut
->biHeight
* This
->pBihOut
->biBitCount
/ 8;
257 TRACE("Size: %u\n", This
->pBihIn
->biSize
);
258 result
= ICDecompressQuery(This
->hvid
, This
->pBihIn
, This
->pBihOut
);
259 if (result
!= ICERR_OK
)
261 ERR("Unable to found a suitable output format (%d)\n", result
);
265 /* Update output media type */
266 CopyMediaType(outpmt
, pmt
);
267 outpmt
->subtype
= *outsubtype
;
269 if (IsEqualIID(&pmt
->formattype
, &FORMAT_VideoInfo
))
270 memcpy(&(((VIDEOINFOHEADER
*)outpmt
->pbFormat
)->bmiHeader
), This
->pBihOut
, This
->pBihOut
->biSize
);
271 else if (IsEqualIID(&pmt
->formattype
, &FORMAT_VideoInfo2
))
272 memcpy(&(((VIDEOINFOHEADER2
*)outpmt
->pbFormat
)->bmiHeader
), This
->pBihOut
, This
->pBihOut
->biSize
);
276 /* Update buffer size of media samples in output */
277 ((OutputPin
*)This
->tf
.ppPins
[1])->allocProps
.cbBuffer
= This
->pBihOut
->biSizeImage
;
279 TRACE("Connection accepted\n");
282 TRACE("Unable to find a suitable VFW decompressor\n");
286 AVIDec_Cleanup(pTransformFilter
);
288 TRACE("Connection refused\n");
292 static HRESULT
AVIDec_Cleanup(TransformFilterImpl
* pTransformFilter
)
294 AVIDecImpl
* This
= (AVIDecImpl
*)pTransformFilter
;
296 TRACE("(%p)->()\n", This
);
301 CoTaskMemFree(This
->pBihIn
);
303 CoTaskMemFree(This
->pBihOut
);
307 This
->pBihOut
= NULL
;
312 static const TransformFuncsTable AVIDec_FuncsTable
= {
314 AVIDec_ProcessSampleData
,
321 HRESULT
AVIDec_create(IUnknown
* pUnkOuter
, LPVOID
* ppv
)
326 TRACE("(%p, %p)\n", pUnkOuter
, ppv
);
331 return CLASS_E_NOAGGREGATION
;
333 /* Note: This memory is managed by the transform filter once created */
334 This
= CoTaskMemAlloc(sizeof(AVIDecImpl
));
338 This
->pBihOut
= NULL
;
340 hr
= TransformFilter_Create(&(This
->tf
), &CLSID_AVIDec
, &AVIDec_FuncsTable
, NULL
, NULL
, NULL
);