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