dlls/windowscodecs/colortransform.c

   1 /*
   2  * Copyright 2013 Hans Leidekker 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
  29 #include "wincodecs_private.h"
  30
  31 #include "wine/debug.h"
  32
  33 WINE_DEFAULT_DEBUG_CHANNEL(wincodecs);
  34
  35 typedef struct ColorTransform {
  36     IWICColorTransform IWICColorTransform_iface;
  37     LONG ref;
  38     IWICBitmapSource *dst;
  39 } ColorTransform;
  40
  41 static inline ColorTransform *impl_from_IWICColorTransform(IWICColorTransform *iface)
  42 {
  43     return CONTAINING_RECORD(iface, ColorTransform, IWICColorTransform_iface);
  44 }
  45
  46 static HRESULT WINAPI ColorTransform_QueryInterface(IWICColorTransform *iface, REFIID iid,
  47     void **ppv)
  48 {
  49     ColorTransform *This = impl_from_IWICColorTransform(iface);
  50     TRACE("(%p,%s,%p)\n", iface, debugstr_guid(iid), ppv);
  51
  52     if (!ppv) return E_INVALIDARG;
  53
  54     if (IsEqualIID(&IID_IUnknown, iid) ||
  55         IsEqualIID(&IID_IWICBitmapSource, iid) ||
  56         IsEqualIID(&IID_IWICColorTransform, iid))
  57     {
  58         *ppv = &This->IWICColorTransform_iface;
  59     }
  60     else
  61     {
  62         *ppv = NULL;
  63         return E_NOINTERFACE;
  64     }
  65
  66     IUnknown_AddRef((IUnknown*)*ppv);
  67     return S_OK;
  68 }
  69
  70 static ULONG WINAPI ColorTransform_AddRef(IWICColorTransform *iface)
  71 {
  72     ColorTransform *This = impl_from_IWICColorTransform(iface);
  73     ULONG ref = InterlockedIncrement(&This->ref);
  74
  75     TRACE("(%p) refcount=%u\n", iface, ref);
  76
  77     return ref;
  78 }
  79
  80 static ULONG WINAPI ColorTransform_Release(IWICColorTransform *iface)
  81 {
  82     ColorTransform *This = impl_from_IWICColorTransform(iface);
  83     ULONG ref = InterlockedDecrement(&This->ref);
  84
  85     TRACE("(%p) refcount=%u\n", iface, ref);
  86
  87     if (ref == 0)
  88     {
  89         if (This->dst) IWICBitmapSource_Release(This->dst);
  90         HeapFree(GetProcessHeap(), 0, This);
  91     }
  92
  93     return ref;
  94 }
  95
  96 static HRESULT WINAPI ColorTransform_GetSize(IWICColorTransform *iface,
  97     UINT *puiWidth, UINT *puiHeight)
  98 {
  99     ColorTransform *This = impl_from_IWICColorTransform(iface);
 100     TRACE("(%p,%p,%p)\n", iface, puiWidth, puiHeight);
 101
 102     return IWICBitmapSource_GetSize(This->dst, puiWidth, puiHeight);
 103 }
 104
 105 static HRESULT WINAPI ColorTransform_GetPixelFormat(IWICColorTransform *iface,
 106     WICPixelFormatGUID *pPixelFormat)
 107 {
 108     ColorTransform *This = impl_from_IWICColorTransform(iface);
 109     TRACE("(%p,%p)\n", iface, pPixelFormat);
 110
 111     return IWICBitmapSource_GetPixelFormat(This->dst, pPixelFormat);
 112 }
 113
 114 static HRESULT WINAPI ColorTransform_GetResolution(IWICColorTransform *iface,
 115     double *pDpiX, double *pDpiY)
 116 {
 117     ColorTransform *This = impl_from_IWICColorTransform(iface);
 118     TRACE("(%p,%p,%p)\n", iface, pDpiX, pDpiY);
 119
 120     return IWICBitmapSource_GetResolution(This->dst, pDpiX, pDpiY);
 121 }
 122
 123 static HRESULT WINAPI ColorTransform_CopyPalette(IWICColorTransform *iface,
 124     IWICPalette *pIPalette)
 125 {
 126     ColorTransform *This = impl_from_IWICColorTransform(iface);
 127     TRACE("(%p,%p)\n", iface, pIPalette);
 128
 129     return IWICBitmapSource_CopyPalette(This->dst, pIPalette);
 130 }
 131
 132 static HRESULT WINAPI ColorTransform_CopyPixels(IWICColorTransform *iface,
 133     const WICRect *prc, UINT cbStride, UINT cbBufferSize, BYTE *pbBuffer)
 134 {
 135     ColorTransform *This = impl_from_IWICColorTransform(iface);
 136     TRACE("(%p,%p,%u,%u,%p)\n", iface, prc, cbStride, cbBufferSize, pbBuffer);
 137
 138     return IWICBitmapSource_CopyPixels(This->dst, prc, cbStride, cbBufferSize, pbBuffer);
 139 }
 140
 141 static HRESULT WINAPI ColorTransform_Initialize(IWICColorTransform *iface,
 142     IWICBitmapSource *pIBitmapSource, IWICColorContext *pIContextSource,
 143     IWICColorContext *pIContextDest, REFWICPixelFormatGUID pixelFmtDest)
 144 {
 145     ColorTransform *This = impl_from_IWICColorTransform(iface);
 146     IWICBitmapSource *dst;
 147     HRESULT hr;
 148
 149     TRACE("(%p,%p,%p,%p,%s)\n", iface, pIBitmapSource, pIContextSource,
 150           pIContextDest, debugstr_guid(pixelFmtDest));
 151
 152     FIXME("ignoring color contexts\n");
 153
 154     hr = WICConvertBitmapSource(pixelFmtDest, pIBitmapSource, &dst);
 155     if (FAILED(hr)) return hr;
 156
 157     if (This->dst) IWICBitmapSource_Release(This->dst);
 158     This->dst = dst;
 159     return S_OK;
 160 }
 161
 162 static const IWICColorTransformVtbl ColorTransform_Vtbl = {
 163     ColorTransform_QueryInterface,
 164     ColorTransform_AddRef,
 165     ColorTransform_Release,
 166     ColorTransform_GetSize,
 167     ColorTransform_GetPixelFormat,
 168     ColorTransform_GetResolution,
 169     ColorTransform_CopyPalette,
 170     ColorTransform_CopyPixels,
 171     ColorTransform_Initialize
 172 };
 173
 174 HRESULT ColorTransform_Create(IWICColorTransform **colortransform)
 175 {
 176     ColorTransform *This;
 177
 178     if (!colortransform) return E_INVALIDARG;
 179
 180     This = HeapAlloc(GetProcessHeap(), 0, sizeof(ColorTransform));
 181     if (!This) return E_OUTOFMEMORY;
 182
 183     This->IWICColorTransform_iface.lpVtbl = &ColorTransform_Vtbl;
 184     This->ref = 1;
 185     This->dst = NULL;
 186
 187     *colortransform = &This->IWICColorTransform_iface;
 188
 189     return S_OK;
 190 }