wined3d: CMP supports _SAT.
[wine] / dlls / windowscodecs / fliprotate.c
1 /*
2  * Copyright 2010 Vincent Povirk for CodeWeavers
3  *
4  * This library is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * This library is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with this library; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
17  */
18
19 #include "config.h"
20
21 #include <stdarg.h>
22
23 #define COBJMACROS
24
25 #include "windef.h"
26 #include "winbase.h"
27 #include "objbase.h"
28 #include "wincodec.h"
29
30 #include "wincodecs_private.h"
31
32 #include "wine/debug.h"
33
34 WINE_DEFAULT_DEBUG_CHANNEL(wincodecs);
35
36 typedef struct FlipRotator {
37     IWICBitmapFlipRotator IWICBitmapFlipRotator_iface;
38     LONG ref;
39     IWICBitmapSource *source;
40     int flip_x;
41     int flip_y;
42     int swap_xy;
43     CRITICAL_SECTION lock; /* must be held when initialized */
44 } FlipRotator;
45
46 static inline FlipRotator *impl_from_IWICBitmapFlipRotator(IWICBitmapFlipRotator *iface)
47 {
48     return CONTAINING_RECORD(iface, FlipRotator, IWICBitmapFlipRotator_iface);
49 }
50
51 static HRESULT WINAPI FlipRotator_QueryInterface(IWICBitmapFlipRotator *iface, REFIID iid,
52     void **ppv)
53 {
54     FlipRotator *This = impl_from_IWICBitmapFlipRotator(iface);
55     TRACE("(%p,%s,%p)\n", iface, debugstr_guid(iid), ppv);
56
57     if (!ppv) return E_INVALIDARG;
58
59     if (IsEqualIID(&IID_IUnknown, iid) ||
60         IsEqualIID(&IID_IWICBitmapSource, iid) ||
61         IsEqualIID(&IID_IWICBitmapFlipRotator, iid))
62     {
63         *ppv = This;
64     }
65     else
66     {
67         *ppv = NULL;
68         return E_NOINTERFACE;
69     }
70
71     IUnknown_AddRef((IUnknown*)*ppv);
72     return S_OK;
73 }
74
75 static ULONG WINAPI FlipRotator_AddRef(IWICBitmapFlipRotator *iface)
76 {
77     FlipRotator *This = impl_from_IWICBitmapFlipRotator(iface);
78     ULONG ref = InterlockedIncrement(&This->ref);
79
80     TRACE("(%p) refcount=%u\n", iface, ref);
81
82     return ref;
83 }
84
85 static ULONG WINAPI FlipRotator_Release(IWICBitmapFlipRotator *iface)
86 {
87     FlipRotator *This = impl_from_IWICBitmapFlipRotator(iface);
88     ULONG ref = InterlockedDecrement(&This->ref);
89
90     TRACE("(%p) refcount=%u\n", iface, ref);
91
92     if (ref == 0)
93     {
94         This->lock.DebugInfo->Spare[0] = 0;
95         DeleteCriticalSection(&This->lock);
96         if (This->source) IWICBitmapSource_Release(This->source);
97         HeapFree(GetProcessHeap(), 0, This);
98     }
99
100     return ref;
101 }
102
103 static HRESULT WINAPI FlipRotator_GetSize(IWICBitmapFlipRotator *iface,
104     UINT *puiWidth, UINT *puiHeight)
105 {
106     FlipRotator *This = impl_from_IWICBitmapFlipRotator(iface);
107     TRACE("(%p,%p,%p)\n", iface, puiWidth, puiHeight);
108
109     if (!This->source)
110         return WINCODEC_ERR_WRONGSTATE;
111     else if (This->swap_xy)
112         return IWICBitmapSource_GetSize(This->source, puiHeight, puiWidth);
113     else
114         return IWICBitmapSource_GetSize(This->source, puiWidth, puiHeight);
115 }
116
117 static HRESULT WINAPI FlipRotator_GetPixelFormat(IWICBitmapFlipRotator *iface,
118     WICPixelFormatGUID *pPixelFormat)
119 {
120     FlipRotator *This = impl_from_IWICBitmapFlipRotator(iface);
121     TRACE("(%p,%p)\n", iface, pPixelFormat);
122
123     if (!This->source)
124         return WINCODEC_ERR_WRONGSTATE;
125     else
126         return IWICBitmapSource_GetPixelFormat(This->source, pPixelFormat);
127 }
128
129 static HRESULT WINAPI FlipRotator_GetResolution(IWICBitmapFlipRotator *iface,
130     double *pDpiX, double *pDpiY)
131 {
132     FIXME("(%p,%p,%p): stub\n", iface, pDpiX, pDpiY);
133
134     return E_NOTIMPL;
135 }
136
137 static HRESULT WINAPI FlipRotator_CopyPalette(IWICBitmapFlipRotator *iface,
138     IWICPalette *pIPalette)
139 {
140     FIXME("(%p,%p): stub\n", iface, pIPalette);
141     return E_NOTIMPL;
142 }
143
144 static HRESULT WINAPI FlipRotator_CopyPixels(IWICBitmapFlipRotator *iface,
145     const WICRect *prc, UINT cbStride, UINT cbBufferSize, BYTE *pbBuffer)
146 {
147     FlipRotator *This = impl_from_IWICBitmapFlipRotator(iface);
148     HRESULT hr;
149     UINT y;
150     UINT srcy, srcwidth, srcheight;
151     WICRect rc;
152     WICRect rect;
153
154     TRACE("(%p,%p,%u,%u,%p)\n", iface, prc, cbStride, cbBufferSize, pbBuffer);
155
156     if (!This->source) return WINCODEC_ERR_WRONGSTATE;
157
158     if (This->swap_xy || This->flip_x)
159     {
160         /* This requires knowledge of the pixel format. */
161         FIXME("flipping x and rotating are not implemented\n");
162         return E_NOTIMPL;
163     }
164
165     hr = IWICBitmapSource_GetSize(This->source, &srcwidth, &srcheight);
166     if (FAILED(hr)) return hr;
167
168     if (!prc)
169     {
170         UINT width, height;
171         hr = IWICBitmapSource_GetSize(iface, &width, &height);
172         if (FAILED(hr)) return hr;
173         rect.X = 0;
174         rect.Y = 0;
175         rect.Width = width;
176         rect.Height = height;
177         prc = &rect;
178     }
179
180     for (y=prc->Y; y - prc->Y < prc->Height; y++)
181     {
182         if (This->flip_y)
183             srcy = srcheight - 1 - y;
184         else
185             srcy = y;
186
187         rc.X = prc->X;
188         rc.Y = srcy;
189         rc.Width = prc->Width;
190         rc.Height = 1;
191
192         hr = IWICBitmapSource_CopyPixels(This->source, &rc, cbStride, cbStride,
193             pbBuffer);
194
195         if (FAILED(hr)) break;
196
197         pbBuffer += cbStride;
198     }
199
200     return hr;
201 }
202
203 static HRESULT WINAPI FlipRotator_Initialize(IWICBitmapFlipRotator *iface,
204     IWICBitmapSource *pISource, WICBitmapTransformOptions options)
205 {
206     FlipRotator *This = impl_from_IWICBitmapFlipRotator(iface);
207     HRESULT hr=S_OK;
208
209     TRACE("(%p,%p,%u)\n", iface, pISource, options);
210
211     EnterCriticalSection(&This->lock);
212
213     if (This->source)
214     {
215         hr = WINCODEC_ERR_WRONGSTATE;
216         goto end;
217     }
218
219     if (options&WICBitmapTransformRotate90)
220     {
221         This->swap_xy = 1;
222         This->flip_x = !This->flip_x;
223     }
224
225     if (options&WICBitmapTransformRotate180)
226     {
227         This->flip_x = !This->flip_x;
228         This->flip_y = !This->flip_y;
229     }
230
231     if (options&WICBitmapTransformFlipHorizontal)
232         This->flip_x = !This->flip_x;
233
234     if (options&WICBitmapTransformFlipVertical)
235         This->flip_y = !This->flip_y;
236
237     IWICBitmapSource_AddRef(pISource);
238     This->source = pISource;
239
240 end:
241     LeaveCriticalSection(&This->lock);
242
243     return hr;
244 }
245
246 static const IWICBitmapFlipRotatorVtbl FlipRotator_Vtbl = {
247     FlipRotator_QueryInterface,
248     FlipRotator_AddRef,
249     FlipRotator_Release,
250     FlipRotator_GetSize,
251     FlipRotator_GetPixelFormat,
252     FlipRotator_GetResolution,
253     FlipRotator_CopyPalette,
254     FlipRotator_CopyPixels,
255     FlipRotator_Initialize
256 };
257
258 HRESULT FlipRotator_Create(IWICBitmapFlipRotator **fliprotator)
259 {
260     FlipRotator *This;
261
262     This = HeapAlloc(GetProcessHeap(), 0, sizeof(FlipRotator));
263     if (!This) return E_OUTOFMEMORY;
264
265     This->IWICBitmapFlipRotator_iface.lpVtbl = &FlipRotator_Vtbl;
266     This->ref = 1;
267     This->source = NULL;
268     This->flip_x = 0;
269     This->flip_y = 0;
270     This->swap_xy = 0;
271     InitializeCriticalSection(&This->lock);
272     This->lock.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": FlipRotator.lock");
273
274     *fliprotator = &This->IWICBitmapFlipRotator_iface;
275
276     return S_OK;
277 }