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