dlls/windowscodecs/icnsencoder.c

   1 /*
   2  * Copyright 2010 Damjan Jovanovic
   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 #if 0
  20 #pragma makedep unix
  21 #endif
  22
  23 #include "config.h"
  24 #include "wine/port.h"
  25
  26 #include <stdarg.h>
  27
  28 #ifdef HAVE_APPLICATIONSERVICES_APPLICATIONSERVICES_H
  29 #define GetCurrentProcess GetCurrentProcess_Mac
  30 #define GetCurrentThread GetCurrentThread_Mac
  31 #define LoadResource LoadResource_Mac
  32 #define AnimatePalette AnimatePalette_Mac
  33 #define EqualRgn EqualRgn_Mac
  34 #define FillRgn FillRgn_Mac
  35 #define FrameRgn FrameRgn_Mac
  36 #define GetPixel GetPixel_Mac
  37 #define InvertRgn InvertRgn_Mac
  38 #define LineTo LineTo_Mac
  39 #define OffsetRgn OffsetRgn_Mac
  40 #define PaintRgn PaintRgn_Mac
  41 #define Polygon Polygon_Mac
  42 #define ResizePalette ResizePalette_Mac
  43 #define SetRectRgn SetRectRgn_Mac
  44 #define EqualRect EqualRect_Mac
  45 #define FillRect FillRect_Mac
  46 #define FrameRect FrameRect_Mac
  47 #define GetCursor GetCursor_Mac
  48 #define InvertRect InvertRect_Mac
  49 #define OffsetRect OffsetRect_Mac
  50 #define PtInRect PtInRect_Mac
  51 #define SetCursor SetCursor_Mac
  52 #define SetRect SetRect_Mac
  53 #define ShowCursor ShowCursor_Mac
  54 #define UnionRect UnionRect_Mac
  55 #include <ApplicationServices/ApplicationServices.h>
  56 #undef GetCurrentProcess
  57 #undef GetCurrentThread
  58 #undef LoadResource
  59 #undef AnimatePalette
  60 #undef EqualRgn
  61 #undef FillRgn
  62 #undef FrameRgn
  63 #undef GetPixel
  64 #undef InvertRgn
  65 #undef LineTo
  66 #undef OffsetRgn
  67 #undef PaintRgn
  68 #undef Polygon
  69 #undef ResizePalette
  70 #undef SetRectRgn
  71 #undef EqualRect
  72 #undef FillRect
  73 #undef FrameRect
  74 #undef GetCursor
  75 #undef InvertRect
  76 #undef OffsetRect
  77 #undef PtInRect
  78 #undef SetCursor
  79 #undef SetRect
  80 #undef ShowCursor
  81 #undef UnionRect
  82 #endif
  83
  84 #include "ntstatus.h"
  85 #define WIN32_NO_STATUS
  86 #include "windef.h"
  87 #include "winternl.h"
  88 #include "winbase.h"
  89 #include "objbase.h"
  90
  91 #include "wincodecs_private.h"
  92
  93 #include "wine/debug.h"
  94
  95 WINE_DEFAULT_DEBUG_CHANNEL(wincodecs);
  96
  97 #if defined(HAVE_APPLICATIONSERVICES_APPLICATIONSERVICES_H) && \
  98     MAC_OS_X_VERSION_MAX_ALLOWED > MAC_OS_X_VERSION_10_4
  99
 100 typedef struct IcnsEncoder {
 101     struct encoder encoder;
 102     IStream *stream;
 103     IconFamilyHandle icns_family;
 104     struct encoder_frame frame;
 105     OSType icns_type;
 106     BYTE* icns_image;
 107     int lines_written;
 108 } IcnsEncoder;
 109
 110 static inline IcnsEncoder *impl_from_encoder(struct encoder *iface)
 111 {
 112     return CONTAINING_RECORD(iface, IcnsEncoder, encoder);
 113 }
 114
 115 void CDECL IcnsEncoder_destroy(struct encoder *iface)
 116 {
 117     IcnsEncoder *This = impl_from_encoder(iface);
 118
 119     free(This->icns_image);
 120
 121     if (This->icns_family)
 122         DisposeHandle((Handle)This->icns_family);
 123
 124     RtlFreeHeap(GetProcessHeap(), 0, This);
 125 }
 126
 127 HRESULT CDECL IcnsEncoder_get_supported_format(struct encoder *iface,
 128     GUID *pixel_format, DWORD *bpp, BOOL *indexed)
 129 {
 130     *pixel_format = GUID_WICPixelFormat32bppBGRA;
 131     *bpp = 32;
 132     *indexed = FALSE;
 133     return S_OK;
 134 }
 135
 136 HRESULT CDECL IcnsEncoder_create_frame(struct encoder *iface, const struct encoder_frame *frame)
 137 {
 138     IcnsEncoder *This = impl_from_encoder(iface);
 139
 140     This->frame = *frame;
 141
 142     switch (frame->width)
 143     {
 144         case 16:  This->icns_type = kIconServices16PixelDataARGB;  break;
 145         case 32:  This->icns_type = kIconServices32PixelDataARGB;  break;
 146         case 48:  This->icns_type = kIconServices48PixelDataARGB;  break;
 147         case 128: This->icns_type = kIconServices128PixelDataARGB; break;
 148         case 256: This->icns_type = kIconServices256PixelDataARGB; break;
 149         case 512: This->icns_type = kIconServices512PixelDataARGB; break;
 150         default:
 151             ERR("cannot generate ICNS icon from %dx%d image\n", frame->width, frame->height);
 152             return E_INVALIDARG;
 153     }
 154     This->icns_image = malloc(frame->width * frame->height * 4);
 155     if (!This->icns_image)
 156     {
 157         WARN("failed to allocate image buffer\n");
 158         return E_FAIL;
 159     }
 160     This->lines_written = 0;
 161
 162     return S_OK;
 163 }
 164
 165 static HRESULT CDECL IcnsEncoder_write_lines(struct encoder* iface,
 166     BYTE *data, DWORD line_count, DWORD stride)
 167 {
 168     IcnsEncoder *This = impl_from_encoder(iface);
 169     UINT i;
 170
 171     for (i = 0; i < line_count; i++)
 172     {
 173         BYTE *src_row, *dst_row;
 174         UINT j;
 175         src_row = data + stride * i;
 176         dst_row = This->icns_image + (This->lines_written + i)*(This->frame.width*4);
 177         /* swap bgr -> rgb */
 178         for (j = 0; j < This->frame.width*4; j += 4)
 179         {
 180             dst_row[j] = src_row[j+3];
 181             dst_row[j+1] = src_row[j+2];
 182             dst_row[j+2] = src_row[j+1];
 183             dst_row[j+3] = src_row[j];
 184         }
 185     }
 186     This->lines_written += line_count;
 187
 188     return S_OK;
 189 }
 190
 191 static HRESULT CDECL IcnsEncoder_commit_frame(struct encoder *iface)
 192 {
 193     IcnsEncoder *This = impl_from_encoder(iface);
 194     Handle handle;
 195     OSErr ret;
 196
 197     ret = PtrToHand(This->icns_image, &handle, This->frame.width * This->frame.height * 4);
 198     if (ret != noErr || !handle)
 199     {
 200         WARN("PtrToHand failed with error %d\n", ret);
 201         return E_FAIL;
 202     }
 203
 204     ret = SetIconFamilyData(This->icns_family, This->icns_type, handle);
 205     DisposeHandle(handle);
 206
 207     if (ret != noErr)
 208         {
 209         WARN("SetIconFamilyData failed for image with error %d\n", ret);
 210         return E_FAIL;
 211         }
 212
 213     free(This->icns_image);
 214     This->icns_image = NULL;
 215
 216     return S_OK;
 217 }
 218
 219 static HRESULT CDECL IcnsEncoder_initialize(struct encoder *iface, IStream *stream)
 220 {
 221     IcnsEncoder *This = impl_from_encoder(iface);
 222
 223     This->icns_family = (IconFamilyHandle)NewHandle(0);
 224     if (!This->icns_family)
 225     {
 226         WARN("error creating icns family\n");
 227         return E_FAIL;
 228     }
 229
 230     This->stream = stream;
 231
 232     return S_OK;
 233 }
 234
 235 static HRESULT CDECL IcnsEncoder_commit_file(struct encoder *iface)
 236 {
 237     IcnsEncoder *This = impl_from_encoder(iface);
 238     size_t buffer_size;
 239     HRESULT hr = S_OK;
 240     ULONG byteswritten;
 241
 242     buffer_size = GetHandleSize((Handle)This->icns_family);
 243     hr = stream_write(This->stream, *This->icns_family, buffer_size, &byteswritten);
 244     if (FAILED(hr) || byteswritten != buffer_size)
 245     {
 246         WARN("writing file failed, hr = 0x%08X\n", hr);
 247         return E_FAIL;
 248     }
 249
 250     return S_OK;
 251 }
 252
 253 static const struct encoder_funcs IcnsEncoder_vtable = {
 254     IcnsEncoder_initialize,
 255     IcnsEncoder_get_supported_format,
 256     IcnsEncoder_create_frame,
 257     IcnsEncoder_write_lines,
 258     IcnsEncoder_commit_frame,
 259     IcnsEncoder_commit_file,
 260     IcnsEncoder_destroy
 261 };
 262
 263 HRESULT CDECL icns_encoder_create(struct encoder_info *info, struct encoder **result)
 264 {
 265     IcnsEncoder *This;
 266
 267     TRACE("\n");
 268
 269     This = RtlAllocateHeap(GetProcessHeap(), 0, sizeof(IcnsEncoder));
 270     if (!This) return E_OUTOFMEMORY;
 271
 272     This->encoder.vtable = &IcnsEncoder_vtable;
 273     This->stream = NULL;
 274     This->icns_family = NULL;
 275     This->icns_image = NULL;
 276
 277     *result = &This->encoder;
 278     info->flags = ENCODER_FLAGS_MULTI_FRAME|ENCODER_FLAGS_ICNS_SIZE;
 279     info->container_format = GUID_WineContainerFormatIcns;
 280     info->clsid = CLSID_WICIcnsEncoder;
 281     info->encoder_options[0] = ENCODER_OPTION_END;
 282
 283     return S_OK;
 284 }
 285
 286 #else /* !defined(HAVE_APPLICATIONSERVICES_APPLICATIONSERVICES_H) ||
 287          MAC_OS_X_VERSION_MAX_ALLOWED <= MAC_OS_X_VERSION_10_4 */
 288
 289 HRESULT CDECL icns_encoder_create(struct encoder_info *info, struct encoder **result)
 290 {
 291     ERR("Trying to save ICNS picture, but ICNS support is not compiled in.\n");
 292     return E_FAIL;
 293 }
 294
 295 #endif