/* * Copyright 2009 Vincent Povirk for CodeWeavers * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA */ #define COBJMACROS #include "config.h" #include <stdarg.h> #include "windef.h" #include "winbase.h" #include "objbase.h" #include "wincodec.h" #include "wincodecs_private.h" #include "wine/debug.h" WINE_DEFAULT_DEBUG_CHANNEL(wincodecs); extern BOOL WINAPI WIC_DllMain(HINSTANCE, DWORD, LPVOID) DECLSPEC_HIDDEN; BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved) { switch (fdwReason) { case DLL_PROCESS_ATTACH: DisableThreadLibraryCalls(hinstDLL); break; } return WIC_DllMain(hinstDLL, fdwReason, lpvReserved); } HRESULT WINAPI DllCanUnloadNow(void) { return S_FALSE; } HRESULT copy_pixels(UINT bpp, const BYTE *srcbuffer, UINT srcwidth, UINT srcheight, INT srcstride, const WICRect *rc, UINT dststride, UINT dstbuffersize, BYTE *dstbuffer) { UINT bytesperrow; UINT row_offset; /* number of bits into the source rows where the data starts */ WICRect rect; if (!rc) { rect.X = 0; rect.Y = 0; rect.Width = srcwidth; rect.Height = srcheight; rc = ▭ } else { if (rc->X < 0 || rc->Y < 0 || rc->X+rc->Width > srcwidth || rc->Y+rc->Height > srcheight) return E_INVALIDARG; } bytesperrow = ((bpp * rc->Width)+7)/8; if (dststride < bytesperrow) return E_INVALIDARG; if ((dststride * (rc->Height-1)) + ((rc->Width * bpp) + 7)/8 > dstbuffersize) return E_INVALIDARG; /* if the whole bitmap is copied and the buffer format matches then it's a matter of a single memcpy */ if (rc->X == 0 && rc->Y == 0 && rc->Width == srcwidth && rc->Height == srcheight && srcstride == dststride && srcstride == bytesperrow) { memcpy(dstbuffer, srcbuffer, srcstride * srcheight); return S_OK; } row_offset = rc->X * bpp; if (row_offset % 8 == 0) { /* everything lines up on a byte boundary */ INT row; const BYTE *src; BYTE *dst; src = srcbuffer + (row_offset / 8) + srcstride * rc->Y; dst = dstbuffer; for (row=0; row < rc->Height; row++) { memcpy(dst, src, bytesperrow); src += srcstride; dst += dststride; } return S_OK; } else { /* we have to do a weird bitwise copy. eww. */ FIXME("cannot reliably copy bitmap data if bpp < 8\n"); return E_FAIL; } } HRESULT configure_write_source(IWICBitmapFrameEncode *iface, IWICBitmapSource *source, const WICRect *prc, const WICPixelFormatGUID *format, INT width, INT height, double xres, double yres) { HRESULT hr=S_OK; WICPixelFormatGUID src_format, dst_format; if (width == 0 || height == 0) return WINCODEC_ERR_WRONGSTATE; hr = IWICBitmapSource_GetPixelFormat(source, &src_format); if (FAILED(hr)) return hr; if (!format) { dst_format = src_format; hr = IWICBitmapFrameEncode_SetPixelFormat(iface, &dst_format); if (FAILED(hr)) return hr; format = &dst_format; } if (!IsEqualGUID(&src_format, format)) { /* FIXME: should use WICConvertBitmapSource to convert */ ERR("format %s unsupported\n", debugstr_guid(&src_format)); return E_FAIL; } if (xres == 0.0 || yres == 0.0) { hr = IWICBitmapSource_GetResolution(source, &xres, &yres); if (FAILED(hr)) return hr; hr = IWICBitmapFrameEncode_SetResolution(iface, xres, yres); if (FAILED(hr)) return hr; } return hr; } HRESULT write_source(IWICBitmapFrameEncode *iface, IWICBitmapSource *source, const WICRect *prc, const WICPixelFormatGUID *format, UINT bpp, INT width, INT height) { HRESULT hr=S_OK; WICRect rc; UINT stride; BYTE* pixeldata; if (!prc) { UINT src_width, src_height; hr = IWICBitmapSource_GetSize(source, &src_width, &src_height); if (FAILED(hr)) return hr; rc.X = 0; rc.Y = 0; rc.Width = src_width; rc.Height = src_height; prc = &rc; } if (prc->Width != width || prc->Height <= 0) return E_INVALIDARG; stride = (bpp * width + 7)/8; pixeldata = HeapAlloc(GetProcessHeap(), 0, stride * prc->Height); if (!pixeldata) return E_OUTOFMEMORY; hr = IWICBitmapSource_CopyPixels(source, prc, stride, stride*prc->Height, pixeldata); if (SUCCEEDED(hr)) { hr = IWICBitmapFrameEncode_WritePixels(iface, prc->Height, stride, stride*prc->Height, pixeldata); } HeapFree(GetProcessHeap(), 0, pixeldata); return hr; } void reverse_bgr8(UINT bytesperpixel, LPBYTE bits, UINT width, UINT height, INT stride) { UINT x, y; BYTE *pixel, temp; for (y=0; y<height; y++) { pixel = bits + stride * y; for (x=0; x<width; x++) { temp = pixel[2]; pixel[2] = pixel[0]; pixel[0] = temp; pixel += bytesperpixel; } } } HRESULT get_pixelformat_bpp(const GUID *pixelformat, UINT *bpp) { HRESULT hr; IWICComponentInfo *info; IWICPixelFormatInfo *formatinfo; hr = CreateComponentInfo(pixelformat, &info); if (SUCCEEDED(hr)) { hr = IWICComponentInfo_QueryInterface(info, &IID_IWICPixelFormatInfo, (void**)&formatinfo); if (SUCCEEDED(hr)) { hr = IWICPixelFormatInfo_GetBitsPerPixel(formatinfo, bpp); IWICPixelFormatInfo_Release(formatinfo); } IWICComponentInfo_Release(info); } return hr; }