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