dlls/windowscodecs/main.c

   1 /*
   2  * Copyright 2009 Vincent Povirk 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
  20 #define COBJMACROS
  21 #include "config.h"
  22
  23 #include <stdarg.h>
  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 extern BOOL WINAPI WIC_DllMain(HINSTANCE, DWORD, LPVOID) DECLSPEC_HIDDEN;
  36
  37 BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
  38 {
  39
  40     switch (fdwReason)
  41     {
  42         case DLL_PROCESS_ATTACH:
  43             DisableThreadLibraryCalls(hinstDLL);
  44             break;
  45     }
  46
  47     return WIC_DllMain(hinstDLL, fdwReason, lpvReserved);
  48 }
  49
  50 HRESULT WINAPI DllCanUnloadNow(void)
  51 {
  52     return S_FALSE;
  53 }
  54
  55 HRESULT copy_pixels(UINT bpp, const BYTE *srcbuffer,
  56     UINT srcwidth, UINT srcheight, INT srcstride,
  57     const WICRect *rc, UINT dststride, UINT dstbuffersize, BYTE *dstbuffer)
  58 {
  59     UINT bytesperrow;
  60     UINT row_offset; /* number of bits into the source rows where the data starts */
  61     WICRect rect;
  62
  63     if (!rc)
  64     {
  65         rect.X = 0;
  66         rect.Y = 0;
  67         rect.Width = srcwidth;
  68         rect.Height = srcheight;
  69         rc = &rect;
  70     }
  71     else
  72     {
  73         if (rc->X < 0 || rc->Y < 0 || rc->X+rc->Width > srcwidth || rc->Y+rc->Height > srcheight)
  74             return E_INVALIDARG;
  75     }
  76
  77     bytesperrow = ((bpp * rc->Width)+7)/8;
  78
  79     if (dststride < bytesperrow)
  80         return E_INVALIDARG;
  81
  82     if ((dststride * (rc->Height-1)) + ((rc->Width * bpp) + 7)/8 > dstbuffersize)
  83         return E_INVALIDARG;
  84
  85     /* if the whole bitmap is copied and the buffer format matches then it's a matter of a single memcpy */
  86     if (rc->X == 0 && rc->Y == 0 && rc->Width == srcwidth && rc->Height == srcheight &&
  87         srcstride == dststride && srcstride == bytesperrow)
  88     {
  89         memcpy(dstbuffer, srcbuffer, srcstride * srcheight);
  90         return S_OK;
  91     }
  92
  93     row_offset = rc->X * bpp;
  94
  95     if (row_offset % 8 == 0)
  96     {
  97         /* everything lines up on a byte boundary */
  98         INT row;
  99         const BYTE *src;
 100         BYTE *dst;
 101
 102         src = srcbuffer + (row_offset / 8) + srcstride * rc->Y;
 103         dst = dstbuffer;
 104         for (row=0; row < rc->Height; row++)
 105         {
 106             memcpy(dst, src, bytesperrow);
 107             src += srcstride;
 108             dst += dststride;
 109         }
 110         return S_OK;
 111     }
 112     else
 113     {
 114         /* we have to do a weird bitwise copy. eww. */
 115         FIXME("cannot reliably copy bitmap data if bpp < 8\n");
 116         return E_FAIL;
 117     }
 118 }
 119
 120 HRESULT configure_write_source(IWICBitmapFrameEncode *iface,
 121     IWICBitmapSource *source, const WICRect *prc,
 122     const WICPixelFormatGUID *format,
 123     INT width, INT height, double xres, double yres)
 124 {
 125     HRESULT hr=S_OK;
 126     WICPixelFormatGUID src_format, dst_format;
 127
 128     if (width == 0 || height == 0)
 129         return WINCODEC_ERR_WRONGSTATE;
 130
 131     hr = IWICBitmapSource_GetPixelFormat(source, &src_format);
 132     if (FAILED(hr)) return hr;
 133
 134     if (!format)
 135     {
 136         dst_format = src_format;
 137
 138         hr = IWICBitmapFrameEncode_SetPixelFormat(iface, &dst_format);
 139         if (FAILED(hr)) return hr;
 140
 141         format = &dst_format;
 142     }
 143
 144     if (!IsEqualGUID(&src_format, format))
 145     {
 146         /* FIXME: should use WICConvertBitmapSource to convert */
 147         ERR("format %s unsupported\n", debugstr_guid(&src_format));
 148         return E_FAIL;
 149     }
 150
 151     if (xres == 0.0 || yres == 0.0)
 152     {
 153         hr = IWICBitmapSource_GetResolution(source, &xres, &yres);
 154         if (FAILED(hr)) return hr;
 155         hr = IWICBitmapFrameEncode_SetResolution(iface, xres, yres);
 156         if (FAILED(hr)) return hr;
 157     }
 158
 159     return hr;
 160 }
 161
 162 HRESULT write_source(IWICBitmapFrameEncode *iface,
 163     IWICBitmapSource *source, const WICRect *prc,
 164     const WICPixelFormatGUID *format, UINT bpp,
 165     INT width, INT height)
 166 {
 167     HRESULT hr=S_OK;
 168     WICRect rc;
 169     UINT stride;
 170     BYTE* pixeldata;
 171
 172     if (!prc)
 173     {
 174         UINT src_width, src_height;
 175         hr = IWICBitmapSource_GetSize(source, &src_width, &src_height);
 176         if (FAILED(hr)) return hr;
 177         rc.X = 0;
 178         rc.Y = 0;
 179         rc.Width = src_width;
 180         rc.Height = src_height;
 181         prc = &rc;
 182     }
 183
 184     if (prc->Width != width || prc->Height <= 0)
 185         return E_INVALIDARG;
 186
 187     stride = (bpp * width + 7)/8;
 188
 189     pixeldata = HeapAlloc(GetProcessHeap(), 0, stride * prc->Height);
 190     if (!pixeldata) return E_OUTOFMEMORY;
 191
 192     hr = IWICBitmapSource_CopyPixels(source, prc, stride,
 193         stride*prc->Height, pixeldata);
 194
 195     if (SUCCEEDED(hr))
 196     {
 197         hr = IWICBitmapFrameEncode_WritePixels(iface, prc->Height, stride,
 198             stride*prc->Height, pixeldata);
 199     }
 200
 201     HeapFree(GetProcessHeap(), 0, pixeldata);
 202
 203     return hr;
 204 }
 205
 206 void reverse_bgr8(UINT bytesperpixel, LPBYTE bits, UINT width, UINT height, INT stride)
 207 {
 208     UINT x, y;
 209     BYTE *pixel, temp;
 210
 211     for (y=0; y<height; y++)
 212     {
 213         pixel = bits + stride * y;
 214
 215         for (x=0; x<width; x++)
 216         {
 217             temp = pixel[2];
 218             pixel[2] = pixel[0];
 219             pixel[0] = temp;
 220             pixel += bytesperpixel;
 221         }
 222     }
 223 }
 224
 225 HRESULT get_pixelformat_bpp(const GUID *pixelformat, UINT *bpp)
 226 {
 227     HRESULT hr;
 228     IWICComponentInfo *info;
 229     IWICPixelFormatInfo *formatinfo;
 230
 231     hr = CreateComponentInfo(pixelformat, &info);
 232     if (SUCCEEDED(hr))
 233     {
 234         hr = IWICComponentInfo_QueryInterface(info, &IID_IWICPixelFormatInfo, (void**)&formatinfo);
 235
 236         if (SUCCEEDED(hr))
 237         {
 238             hr = IWICPixelFormatInfo_GetBitsPerPixel(formatinfo, bpp);
 239
 240             IWICPixelFormatInfo_Release(formatinfo);
 241         }
 242
 243         IWICComponentInfo_Release(info);
 244     }
 245
 246     return hr;
 247 }