reactos/dll/win32/windowscodecs/scaler.c

386 lines
11 KiB
C
Raw Normal View History

/*
* Copyright 2010 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
*/
#include "wincodecs_private.h"
typedef struct BitmapScaler {
IWICBitmapScaler IWICBitmapScaler_iface;
LONG ref;
IWICBitmapSource *source;
UINT width, height;
UINT src_width, src_height;
WICBitmapInterpolationMode mode;
UINT bpp;
void (*fn_get_required_source_rect)(struct BitmapScaler*,UINT,UINT,WICRect*);
void (*fn_copy_scanline)(struct BitmapScaler*,UINT,UINT,UINT,BYTE**,UINT,UINT,BYTE*);
CRITICAL_SECTION lock; /* must be held when initialized */
} BitmapScaler;
static inline BitmapScaler *impl_from_IWICBitmapScaler(IWICBitmapScaler *iface)
{
return CONTAINING_RECORD(iface, BitmapScaler, IWICBitmapScaler_iface);
}
static HRESULT WINAPI BitmapScaler_QueryInterface(IWICBitmapScaler *iface, REFIID iid,
void **ppv)
{
BitmapScaler *This = impl_from_IWICBitmapScaler(iface);
TRACE("(%p,%s,%p)\n", iface, debugstr_guid(iid), ppv);
if (!ppv) return E_INVALIDARG;
if (IsEqualIID(&IID_IUnknown, iid) ||
IsEqualIID(&IID_IWICBitmapSource, iid) ||
IsEqualIID(&IID_IWICBitmapScaler, iid))
{
*ppv = &This->IWICBitmapScaler_iface;
}
else
{
*ppv = NULL;
return E_NOINTERFACE;
}
IUnknown_AddRef((IUnknown*)*ppv);
return S_OK;
}
static ULONG WINAPI BitmapScaler_AddRef(IWICBitmapScaler *iface)
{
BitmapScaler *This = impl_from_IWICBitmapScaler(iface);
ULONG ref = InterlockedIncrement(&This->ref);
TRACE("(%p) refcount=%u\n", iface, ref);
return ref;
}
static ULONG WINAPI BitmapScaler_Release(IWICBitmapScaler *iface)
{
BitmapScaler *This = impl_from_IWICBitmapScaler(iface);
ULONG ref = InterlockedDecrement(&This->ref);
TRACE("(%p) refcount=%u\n", iface, ref);
if (ref == 0)
{
This->lock.DebugInfo->Spare[0] = 0;
DeleteCriticalSection(&This->lock);
if (This->source) IWICBitmapSource_Release(This->source);
HeapFree(GetProcessHeap(), 0, This);
}
return ref;
}
static HRESULT WINAPI BitmapScaler_GetSize(IWICBitmapScaler *iface,
UINT *puiWidth, UINT *puiHeight)
{
BitmapScaler *This = impl_from_IWICBitmapScaler(iface);
TRACE("(%p,%p,%p)\n", iface, puiWidth, puiHeight);
if (!puiWidth || !puiHeight)
return E_INVALIDARG;
if (!This->source)
return WINCODEC_ERR_WRONGSTATE;
*puiWidth = This->width;
*puiHeight = This->height;
return S_OK;
}
static HRESULT WINAPI BitmapScaler_GetPixelFormat(IWICBitmapScaler *iface,
WICPixelFormatGUID *pPixelFormat)
{
BitmapScaler *This = impl_from_IWICBitmapScaler(iface);
TRACE("(%p,%p)\n", iface, pPixelFormat);
if (!pPixelFormat)
return E_INVALIDARG;
if (!This->source)
return WINCODEC_ERR_WRONGSTATE;
return IWICBitmapSource_GetPixelFormat(This->source, pPixelFormat);
}
static HRESULT WINAPI BitmapScaler_GetResolution(IWICBitmapScaler *iface,
double *pDpiX, double *pDpiY)
{
BitmapScaler *This = impl_from_IWICBitmapScaler(iface);
TRACE("(%p,%p,%p)\n", iface, pDpiX, pDpiY);
if (!pDpiX || !pDpiY)
return E_INVALIDARG;
if (!This->source)
return WINCODEC_ERR_WRONGSTATE;
return IWICBitmapSource_GetResolution(This->source, pDpiX, pDpiY);
}
static HRESULT WINAPI BitmapScaler_CopyPalette(IWICBitmapScaler *iface,
IWICPalette *pIPalette)
{
BitmapScaler *This = impl_from_IWICBitmapScaler(iface);
TRACE("(%p,%p)\n", iface, pIPalette);
if (!pIPalette)
return E_INVALIDARG;
if (!This->source)
return WINCODEC_ERR_WRONGSTATE;
return IWICBitmapSource_CopyPalette(This->source, pIPalette);
}
static void NearestNeighbor_GetRequiredSourceRect(BitmapScaler *This,
UINT x, UINT y, WICRect *src_rect)
{
src_rect->X = x * This->src_width / This->width;
src_rect->Y = y * This->src_height / This->height;
src_rect->Width = src_rect->Height = 1;
}
static void NearestNeighbor_CopyScanline(BitmapScaler *This,
UINT dst_x, UINT dst_y, UINT dst_width,
BYTE **src_data, UINT src_data_x, UINT src_data_y, BYTE *pbBuffer)
{
UINT i;
UINT bytesperpixel = This->bpp/8;
UINT src_x, src_y;
src_y = dst_y * This->src_height / This->height - src_data_y;
for (i=0; i<dst_width; i++)
{
src_x = (dst_x + i) * This->src_width / This->width - src_data_x;
memcpy(pbBuffer + bytesperpixel * i, src_data[src_y] + bytesperpixel * src_x, bytesperpixel);
}
}
static HRESULT WINAPI BitmapScaler_CopyPixels(IWICBitmapScaler *iface,
const WICRect *prc, UINT cbStride, UINT cbBufferSize, BYTE *pbBuffer)
{
BitmapScaler *This = impl_from_IWICBitmapScaler(iface);
HRESULT hr;
WICRect dest_rect;
WICRect src_rect_ul, src_rect_br, src_rect;
BYTE **src_rows;
BYTE *src_bits;
ULONG bytesperrow;
ULONG src_bytesperrow;
ULONG buffer_size;
UINT y;
TRACE("(%p,%p,%u,%u,%p)\n", iface, prc, cbStride, cbBufferSize, pbBuffer);
EnterCriticalSection(&This->lock);
if (!This->source)
{
hr = WINCODEC_ERR_WRONGSTATE;
goto end;
}
if (prc)
dest_rect = *prc;
else
{
dest_rect.X = dest_rect.Y = 0;
dest_rect.Width = This->width;
dest_rect.Height = This->height;
}
if (dest_rect.X < 0 || dest_rect.Y < 0 ||
dest_rect.X+dest_rect.Width > This->width|| dest_rect.Y+dest_rect.Height > This->height)
{
hr = E_INVALIDARG;
goto end;
}
bytesperrow = ((This->bpp * dest_rect.Width)+7)/8;
if (cbStride < bytesperrow)
{
hr = E_INVALIDARG;
goto end;
}
if ((cbStride * dest_rect.Height) > cbBufferSize)
{
hr = E_INVALIDARG;
goto end;
}
/* MSDN recommends calling CopyPixels once for each scanline from top to
* bottom, and claims codecs optimize for this. Ideally, when called in this
* way, we should avoid requesting a scanline from the source more than
* once, by saving the data that will be useful for the next scanline after
* the call returns. The GetRequiredSourceRect/CopyScanline functions are
* designed to make it possible to do this in a generic way, but for now we
* just grab all the data we need in each call. */
This->fn_get_required_source_rect(This, dest_rect.X, dest_rect.Y, &src_rect_ul);
This->fn_get_required_source_rect(This, dest_rect.X+dest_rect.Width-1,
dest_rect.Y+dest_rect.Height-1, &src_rect_br);
src_rect.X = src_rect_ul.X;
src_rect.Y = src_rect_ul.Y;
src_rect.Width = src_rect_br.Width + src_rect_br.X - src_rect_ul.X;
src_rect.Height = src_rect_br.Height + src_rect_br.Y - src_rect_ul.Y;
src_bytesperrow = (src_rect.Width * This->bpp + 7)/8;
buffer_size = src_bytesperrow * src_rect.Height;
src_rows = HeapAlloc(GetProcessHeap(), 0, sizeof(BYTE*) * src_rect.Height);
src_bits = HeapAlloc(GetProcessHeap(), 0, buffer_size);
if (!src_rows || !src_bits)
{
HeapFree(GetProcessHeap(), 0, src_rows);
HeapFree(GetProcessHeap(), 0, src_bits);
hr = E_OUTOFMEMORY;
goto end;
}
for (y=0; y<src_rect.Height; y++)
src_rows[y] = src_bits + y * src_bytesperrow;
hr = IWICBitmapSource_CopyPixels(This->source, &src_rect, src_bytesperrow,
buffer_size, src_bits);
if (SUCCEEDED(hr))
{
for (y=0; y < dest_rect.Height; y++)
{
This->fn_copy_scanline(This, dest_rect.X, dest_rect.Y+y, dest_rect.Width,
src_rows, src_rect.X, src_rect.Y, pbBuffer + cbStride * y);
}
}
HeapFree(GetProcessHeap(), 0, src_rows);
HeapFree(GetProcessHeap(), 0, src_bits);
end:
LeaveCriticalSection(&This->lock);
return hr;
}
static HRESULT WINAPI BitmapScaler_Initialize(IWICBitmapScaler *iface,
IWICBitmapSource *pISource, UINT uiWidth, UINT uiHeight,
WICBitmapInterpolationMode mode)
{
BitmapScaler *This = impl_from_IWICBitmapScaler(iface);
HRESULT hr;
GUID src_pixelformat;
TRACE("(%p,%p,%u,%u,%u)\n", iface, pISource, uiWidth, uiHeight, mode);
EnterCriticalSection(&This->lock);
if (This->source)
{
hr = WINCODEC_ERR_WRONGSTATE;
goto end;
}
This->width = uiWidth;
This->height = uiHeight;
This->mode = mode;
hr = IWICBitmapSource_GetSize(pISource, &This->src_width, &This->src_height);
if (SUCCEEDED(hr))
hr = IWICBitmapSource_GetPixelFormat(pISource, &src_pixelformat);
if (SUCCEEDED(hr))
{
hr = get_pixelformat_bpp(&src_pixelformat, &This->bpp);
}
if (SUCCEEDED(hr))
{
switch (mode)
{
default:
FIXME("unsupported mode %i\n", mode);
/* fall-through */
case WICBitmapInterpolationModeNearestNeighbor:
if ((This->bpp % 8) == 0)
{
IWICBitmapSource_AddRef(pISource);
This->source = pISource;
}
else
{
hr = WICConvertBitmapSource(&GUID_WICPixelFormat32bppBGRA,
pISource, &This->source);
This->bpp = 32;
}
This->fn_get_required_source_rect = NearestNeighbor_GetRequiredSourceRect;
This->fn_copy_scanline = NearestNeighbor_CopyScanline;
break;
}
}
end:
LeaveCriticalSection(&This->lock);
return hr;
}
static const IWICBitmapScalerVtbl BitmapScaler_Vtbl = {
BitmapScaler_QueryInterface,
BitmapScaler_AddRef,
BitmapScaler_Release,
BitmapScaler_GetSize,
BitmapScaler_GetPixelFormat,
BitmapScaler_GetResolution,
BitmapScaler_CopyPalette,
BitmapScaler_CopyPixels,
BitmapScaler_Initialize
};
HRESULT BitmapScaler_Create(IWICBitmapScaler **scaler)
{
BitmapScaler *This;
This = HeapAlloc(GetProcessHeap(), 0, sizeof(BitmapScaler));
if (!This) return E_OUTOFMEMORY;
This->IWICBitmapScaler_iface.lpVtbl = &BitmapScaler_Vtbl;
This->ref = 1;
This->source = NULL;
This->width = 0;
This->height = 0;
This->src_width = 0;
This->src_height = 0;
This->mode = 0;
This->bpp = 0;
InitializeCriticalSection(&This->lock);
This->lock.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": BitmapScaler.lock");
*scaler = &This->IWICBitmapScaler_iface;
return S_OK;
}