dlls/windowscodecs/colorcontext.c

   1 /*
   2  * Copyright 2012 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 #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 ColorContext {
  37     IWICColorContext IWICColorContext_iface;
  38     LONG ref;
  39     WICColorContextType type;
  40     BYTE *profile;
  41     UINT profile_len;
  42     UINT exif_color_space;
  43 } ColorContext;
  44
  45 static inline ColorContext *impl_from_IWICColorContext(IWICColorContext *iface)
  46 {
  47     return CONTAINING_RECORD(iface, ColorContext, IWICColorContext_iface);
  48 }
  49
  50 static HRESULT WINAPI ColorContext_QueryInterface(IWICColorContext *iface, REFIID iid,
  51     void **ppv)
  52 {
  53     ColorContext *This = impl_from_IWICColorContext(iface);
  54     TRACE("(%p,%s,%p)\n", iface, debugstr_guid(iid), ppv);
  55
  56     if (!ppv) return E_INVALIDARG;
  57
  58     if (IsEqualIID(&IID_IUnknown, iid) ||
  59         IsEqualIID(&IID_IWICColorContext, iid))
  60     {
  61         *ppv = &This->IWICColorContext_iface;
  62     }
  63     else
  64     {
  65         *ppv = NULL;
  66         return E_NOINTERFACE;
  67     }
  68
  69     IUnknown_AddRef((IUnknown*)*ppv);
  70     return S_OK;
  71 }
  72
  73 static ULONG WINAPI ColorContext_AddRef(IWICColorContext *iface)
  74 {
  75     ColorContext *This = impl_from_IWICColorContext(iface);
  76     ULONG ref = InterlockedIncrement(&This->ref);
  77
  78     TRACE("(%p) refcount=%u\n", iface, ref);
  79
  80     return ref;
  81 }
  82
  83 static ULONG WINAPI ColorContext_Release(IWICColorContext *iface)
  84 {
  85     ColorContext *This = impl_from_IWICColorContext(iface);
  86     ULONG ref = InterlockedDecrement(&This->ref);
  87
  88     TRACE("(%p) refcount=%u\n", iface, ref);
  89
  90     if (ref == 0)
  91     {
  92         HeapFree(GetProcessHeap(), 0, This->profile);
  93         HeapFree(GetProcessHeap(), 0, This);
  94     }
  95
  96     return ref;
  97 }
  98
  99 static HRESULT load_profile(const WCHAR *filename, BYTE **profile, UINT *len)
 100 {
 101     HANDLE handle;
 102     DWORD count;
 103     LARGE_INTEGER size;
 104     BOOL ret;
 105
 106     *len = 0;
 107     *profile = NULL;
 108     handle = CreateFileW(filename, GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, NULL);
 109     if (handle == INVALID_HANDLE_VALUE) return HRESULT_FROM_WIN32(GetLastError());
 110
 111     if (!(GetFileSizeEx(handle, &size)))
 112     {
 113         CloseHandle(handle);
 114         return HRESULT_FROM_WIN32(GetLastError());
 115     }
 116     if (size.u.HighPart)
 117     {
 118         WARN("file too large\n");
 119         CloseHandle(handle);
 120         return E_FAIL;
 121     }
 122     if (!(*profile = HeapAlloc(GetProcessHeap(), 0, size.u.LowPart)))
 123     {
 124         CloseHandle(handle);
 125         return E_OUTOFMEMORY;
 126     }
 127     ret = ReadFile(handle, *profile, size.u.LowPart, &count, NULL);
 128     CloseHandle(handle);
 129     if (!ret) {
 130         HeapFree (GetProcessHeap(),0,*profile);
 131         *profile = NULL;
 132         return HRESULT_FROM_WIN32(GetLastError());
 133     }
 134     if (count != size.u.LowPart) {
 135         HeapFree (GetProcessHeap(),0,*profile);
 136         *profile = NULL;
 137         return E_FAIL;
 138     }
 139     *len = count;
 140     return S_OK;
 141 }
 142
 143 static HRESULT WINAPI ColorContext_InitializeFromFilename(IWICColorContext *iface,
 144     LPCWSTR wzFilename)
 145 {
 146     ColorContext *This = impl_from_IWICColorContext(iface);
 147     BYTE *profile;
 148     UINT len;
 149     HRESULT hr;
 150     TRACE("(%p,%s)\n", iface, debugstr_w(wzFilename));
 151
 152     if (This->type != WICColorContextUninitialized && This->type != WICColorContextProfile)
 153         return WINCODEC_ERR_WRONGSTATE;
 154
 155     if (!wzFilename) return E_INVALIDARG;
 156
 157     hr = load_profile(wzFilename, &profile, &len);
 158     if (FAILED(hr)) return hr;
 159
 160     HeapFree(GetProcessHeap(), 0, This->profile);
 161     This->profile = profile;
 162     This->profile_len = len;
 163     This->type = WICColorContextProfile;
 164
 165     return S_OK;
 166 }
 167
 168 static HRESULT WINAPI ColorContext_InitializeFromMemory(IWICColorContext *iface,
 169     const BYTE *pbBuffer, UINT cbBufferSize)
 170 {
 171     ColorContext *This = impl_from_IWICColorContext(iface);
 172     BYTE *profile;
 173     TRACE("(%p,%p,%u)\n", iface, pbBuffer, cbBufferSize);
 174
 175     if (This->type != WICColorContextUninitialized && This->type != WICColorContextProfile)
 176         return WINCODEC_ERR_WRONGSTATE;
 177
 178     if (!(profile = HeapAlloc(GetProcessHeap(), 0, cbBufferSize))) return E_OUTOFMEMORY;
 179     memcpy(profile, pbBuffer, cbBufferSize);
 180
 181     HeapFree(GetProcessHeap(), 0, This->profile);
 182     This->profile = profile;
 183     This->profile_len = cbBufferSize;
 184     This->type = WICColorContextProfile;
 185
 186     return S_OK;
 187 }
 188
 189 static HRESULT WINAPI ColorContext_InitializeFromExifColorSpace(IWICColorContext *iface,
 190     UINT value)
 191 {
 192     ColorContext *This = impl_from_IWICColorContext(iface);
 193     TRACE("(%p,%u)\n", iface, value);
 194
 195     if (This->type != WICColorContextUninitialized && This->type != WICColorContextExifColorSpace)
 196         return WINCODEC_ERR_WRONGSTATE;
 197
 198     This->exif_color_space = value;
 199     This->type = WICColorContextExifColorSpace;
 200
 201     return S_OK;
 202 }
 203
 204 static HRESULT WINAPI ColorContext_GetType(IWICColorContext *iface,
 205     WICColorContextType *pType)
 206 {
 207     ColorContext *This = impl_from_IWICColorContext(iface);
 208     TRACE("(%p,%p)\n", iface, pType);
 209
 210     if (!pType) return E_INVALIDARG;
 211
 212     *pType = This->type;
 213     return S_OK;
 214 }
 215
 216 static HRESULT WINAPI ColorContext_GetProfileBytes(IWICColorContext *iface,
 217     UINT cbBuffer, BYTE *pbBuffer, UINT *pcbActual)
 218 {
 219     ColorContext *This = impl_from_IWICColorContext(iface);
 220     TRACE("(%p,%u,%p,%p)\n", iface, cbBuffer, pbBuffer, pcbActual);
 221
 222     if (This->type != WICColorContextProfile)
 223         return WINCODEC_ERR_NOTINITIALIZED;
 224
 225     if (!pcbActual) return E_INVALIDARG;
 226
 227     if (cbBuffer >= This->profile_len && pbBuffer)
 228         memcpy(pbBuffer, This->profile, This->profile_len);
 229
 230     *pcbActual = This->profile_len;
 231
 232     return S_OK;
 233 }
 234
 235 static HRESULT WINAPI ColorContext_GetExifColorSpace(IWICColorContext *iface,
 236     UINT *pValue)
 237 {
 238     ColorContext *This = impl_from_IWICColorContext(iface);
 239     TRACE("(%p,%p)\n", iface, pValue);
 240
 241     if (!pValue) return E_INVALIDARG;
 242
 243     *pValue = This->exif_color_space;
 244     return S_OK;
 245 }
 246
 247 static const IWICColorContextVtbl ColorContext_Vtbl = {
 248     ColorContext_QueryInterface,
 249     ColorContext_AddRef,
 250     ColorContext_Release,
 251     ColorContext_InitializeFromFilename,
 252     ColorContext_InitializeFromMemory,
 253     ColorContext_InitializeFromExifColorSpace,
 254     ColorContext_GetType,
 255     ColorContext_GetProfileBytes,
 256     ColorContext_GetExifColorSpace
 257 };
 258
 259 HRESULT ColorContext_Create(IWICColorContext **colorcontext)
 260 {
 261     ColorContext *This;
 262
 263     if (!colorcontext) return E_INVALIDARG;
 264
 265     This = HeapAlloc(GetProcessHeap(), 0, sizeof(ColorContext));
 266     if (!This) return E_OUTOFMEMORY;
 267
 268     This->IWICColorContext_iface.lpVtbl = &ColorContext_Vtbl;
 269     This->ref = 1;
 270     This->type = 0;
 271     This->profile = NULL;
 272     This->profile_len = 0;
 273     This->exif_color_space = ~0u;
 274
 275     *colorcontext = &This->IWICColorContext_iface;
 276
 277     return S_OK;
 278 }