| /* |
| * Copyright 2009 Vincent Povirk for CodeWeavers |
| * |
| * This library is free software; you can redistribute it and/or |
| * modify it under the terms of the GNU Lesser General Public |
| * License as published by the Free Software Foundation; either |
| * version 2.1 of the License, or (at your option) any later version. |
| * |
| * This library is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| * Lesser General Public License for more details. |
| * |
| * You should have received a copy of the GNU Lesser General Public |
| * License along with this library; if not, write to the Free Software |
| * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA |
| */ |
| |
| #include "config.h" |
| |
| #include <stdarg.h> |
| |
| #define COBJMACROS |
| |
| #include "windef.h" |
| #include "winbase.h" |
| #include "winreg.h" |
| #include "objbase.h" |
| #include "wincodec.h" |
| |
| #include "wincodecs_private.h" |
| |
| #include "wine/debug.h" |
| |
| WINE_DEFAULT_DEBUG_CHANNEL(wincodecs); |
| |
| typedef struct { |
| IWICPalette IWICPalette_iface; |
| LONG ref; |
| UINT count; |
| WICColor *colors; |
| WICBitmapPaletteType type; |
| CRITICAL_SECTION lock; /* must be held when count, colors, or type is accessed */ |
| } PaletteImpl; |
| |
| static inline PaletteImpl *impl_from_IWICPalette(IWICPalette *iface) |
| { |
| return CONTAINING_RECORD(iface, PaletteImpl, IWICPalette_iface); |
| } |
| |
| static HRESULT WINAPI PaletteImpl_QueryInterface(IWICPalette *iface, REFIID iid, |
| void **ppv) |
| { |
| PaletteImpl *This = impl_from_IWICPalette(iface); |
| TRACE("(%p,%s,%p)\n", iface, debugstr_guid(iid), ppv); |
| |
| if (!ppv) return E_INVALIDARG; |
| |
| if (IsEqualIID(&IID_IUnknown, iid) || IsEqualIID(&IID_IWICPalette, iid)) |
| { |
| *ppv = This; |
| } |
| else |
| { |
| *ppv = NULL; |
| return E_NOINTERFACE; |
| } |
| |
| IUnknown_AddRef((IUnknown*)*ppv); |
| return S_OK; |
| } |
| |
| static ULONG WINAPI PaletteImpl_AddRef(IWICPalette *iface) |
| { |
| PaletteImpl *This = impl_from_IWICPalette(iface); |
| ULONG ref = InterlockedIncrement(&This->ref); |
| |
| TRACE("(%p) refcount=%u\n", iface, ref); |
| |
| return ref; |
| } |
| |
| static ULONG WINAPI PaletteImpl_Release(IWICPalette *iface) |
| { |
| PaletteImpl *This = impl_from_IWICPalette(iface); |
| ULONG ref = InterlockedDecrement(&This->ref); |
| |
| TRACE("(%p) refcount=%u\n", iface, ref); |
| |
| if (ref == 0) |
| { |
| This->lock.DebugInfo->Spare[0] = 0; |
| DeleteCriticalSection(&This->lock); |
| HeapFree(GetProcessHeap(), 0, This->colors); |
| HeapFree(GetProcessHeap(), 0, This); |
| } |
| |
| return ref; |
| } |
| |
| static HRESULT WINAPI PaletteImpl_InitializePredefined(IWICPalette *iface, |
| WICBitmapPaletteType ePaletteType, BOOL fAddTransparentColor) |
| { |
| FIXME("(%p,%u,%i): stub\n", iface, ePaletteType, fAddTransparentColor); |
| return E_NOTIMPL; |
| } |
| |
| static HRESULT WINAPI PaletteImpl_InitializeCustom(IWICPalette *iface, |
| WICColor *pColors, UINT colorCount) |
| { |
| PaletteImpl *This = impl_from_IWICPalette(iface); |
| WICColor *new_colors; |
| |
| TRACE("(%p,%p,%u)\n", iface, pColors, colorCount); |
| |
| if (colorCount == 0) |
| { |
| new_colors = NULL; |
| } |
| else |
| { |
| if (!pColors) return E_INVALIDARG; |
| new_colors = HeapAlloc(GetProcessHeap(), 0, sizeof(WICColor) * colorCount); |
| if (!new_colors) return E_OUTOFMEMORY; |
| memcpy(new_colors, pColors, sizeof(WICColor) * colorCount); |
| } |
| |
| EnterCriticalSection(&This->lock); |
| HeapFree(GetProcessHeap(), 0, This->colors); |
| This->colors = new_colors; |
| This->count = colorCount; |
| This->type = WICBitmapPaletteTypeCustom; |
| LeaveCriticalSection(&This->lock); |
| |
| return S_OK; |
| } |
| |
| static HRESULT WINAPI PaletteImpl_InitializeFromBitmap(IWICPalette *iface, |
| IWICBitmapSource *pISurface, UINT colorCount, BOOL fAddTransparentColor) |
| { |
| FIXME("(%p,%p,%u,%i): stub\n", iface, pISurface, colorCount, fAddTransparentColor); |
| return E_NOTIMPL; |
| } |
| |
| static HRESULT WINAPI PaletteImpl_InitializeFromPalette(IWICPalette *iface, |
| IWICPalette *pIPalette) |
| { |
| FIXME("(%p,%p): stub\n", iface, pIPalette); |
| return E_NOTIMPL; |
| } |
| |
| static HRESULT WINAPI PaletteImpl_GetType(IWICPalette *iface, |
| WICBitmapPaletteType *pePaletteType) |
| { |
| PaletteImpl *This = impl_from_IWICPalette(iface); |
| |
| TRACE("(%p,%p)\n", iface, pePaletteType); |
| |
| if (!pePaletteType) return E_INVALIDARG; |
| |
| EnterCriticalSection(&This->lock); |
| *pePaletteType = This->type; |
| LeaveCriticalSection(&This->lock); |
| |
| return S_OK; |
| } |
| |
| static HRESULT WINAPI PaletteImpl_GetColorCount(IWICPalette *iface, UINT *pcCount) |
| { |
| PaletteImpl *This = impl_from_IWICPalette(iface); |
| |
| TRACE("(%p,%p)\n", iface, pcCount); |
| |
| if (!pcCount) return E_INVALIDARG; |
| |
| EnterCriticalSection(&This->lock); |
| *pcCount = This->count; |
| LeaveCriticalSection(&This->lock); |
| |
| return S_OK; |
| } |
| |
| static HRESULT WINAPI PaletteImpl_GetColors(IWICPalette *iface, UINT colorCount, |
| WICColor *pColors, UINT *pcActualColors) |
| { |
| PaletteImpl *This = impl_from_IWICPalette(iface); |
| |
| TRACE("(%p,%i,%p,%p)\n", iface, colorCount, pColors, pcActualColors); |
| |
| if (!pColors || !pcActualColors) return E_INVALIDARG; |
| |
| EnterCriticalSection(&This->lock); |
| |
| if (This->count < colorCount) colorCount = This->count; |
| |
| memcpy(pColors, This->colors, sizeof(WICColor) * colorCount); |
| |
| *pcActualColors = colorCount; |
| |
| LeaveCriticalSection(&This->lock); |
| |
| return S_OK; |
| } |
| |
| static HRESULT WINAPI PaletteImpl_IsBlackWhite(IWICPalette *iface, BOOL *pfIsBlackWhite) |
| { |
| PaletteImpl *This = impl_from_IWICPalette(iface); |
| |
| TRACE("(%p,%p)\n", iface, pfIsBlackWhite); |
| |
| if (!pfIsBlackWhite) return E_INVALIDARG; |
| |
| EnterCriticalSection(&This->lock); |
| if (This->type == WICBitmapPaletteTypeFixedBW) |
| *pfIsBlackWhite = TRUE; |
| else |
| *pfIsBlackWhite = FALSE; |
| LeaveCriticalSection(&This->lock); |
| |
| return S_OK; |
| } |
| |
| static HRESULT WINAPI PaletteImpl_IsGrayscale(IWICPalette *iface, BOOL *pfIsGrayscale) |
| { |
| PaletteImpl *This = impl_from_IWICPalette(iface); |
| |
| TRACE("(%p,%p)\n", iface, pfIsGrayscale); |
| |
| if (!pfIsGrayscale) return E_INVALIDARG; |
| |
| EnterCriticalSection(&This->lock); |
| switch(This->type) |
| { |
| case WICBitmapPaletteTypeFixedBW: |
| case WICBitmapPaletteTypeFixedGray4: |
| case WICBitmapPaletteTypeFixedGray16: |
| case WICBitmapPaletteTypeFixedGray256: |
| *pfIsGrayscale = TRUE; |
| break; |
| default: |
| *pfIsGrayscale = FALSE; |
| } |
| LeaveCriticalSection(&This->lock); |
| |
| return S_OK; |
| } |
| |
| static HRESULT WINAPI PaletteImpl_HasAlpha(IWICPalette *iface, BOOL *pfHasAlpha) |
| { |
| PaletteImpl *This = impl_from_IWICPalette(iface); |
| int i; |
| |
| TRACE("(%p,%p)\n", iface, pfHasAlpha); |
| |
| if (!pfHasAlpha) return E_INVALIDARG; |
| |
| *pfHasAlpha = FALSE; |
| |
| EnterCriticalSection(&This->lock); |
| for (i=0; i<This->count; i++) |
| if ((This->colors[i]&0xff000000) != 0xff000000) |
| { |
| *pfHasAlpha = TRUE; |
| break; |
| } |
| LeaveCriticalSection(&This->lock); |
| |
| return S_OK; |
| } |
| |
| static const IWICPaletteVtbl PaletteImpl_Vtbl = { |
| PaletteImpl_QueryInterface, |
| PaletteImpl_AddRef, |
| PaletteImpl_Release, |
| PaletteImpl_InitializePredefined, |
| PaletteImpl_InitializeCustom, |
| PaletteImpl_InitializeFromBitmap, |
| PaletteImpl_InitializeFromPalette, |
| PaletteImpl_GetType, |
| PaletteImpl_GetColorCount, |
| PaletteImpl_GetColors, |
| PaletteImpl_IsBlackWhite, |
| PaletteImpl_IsGrayscale, |
| PaletteImpl_HasAlpha |
| }; |
| |
| HRESULT PaletteImpl_Create(IWICPalette **palette) |
| { |
| PaletteImpl *This; |
| |
| This = HeapAlloc(GetProcessHeap(), 0, sizeof(PaletteImpl)); |
| if (!This) return E_OUTOFMEMORY; |
| |
| This->IWICPalette_iface.lpVtbl = &PaletteImpl_Vtbl; |
| This->ref = 1; |
| This->count = 0; |
| This->colors = NULL; |
| This->type = WICBitmapPaletteTypeCustom; |
| InitializeCriticalSection(&This->lock); |
| This->lock.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": PaletteImpl.lock"); |
| |
| *palette = (IWICPalette*)This; |
| |
| return S_OK; |
| } |