wine/dlls/windowscodecs/tests/bitmap.c

438 lines
17 KiB
C

/*
* Copyright 2012 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 <stdarg.h>
#include <math.h>
#define COBJMACROS
#include "windef.h"
#include "objbase.h"
#include "wincodec.h"
#include "wine/test.h"
static IWICImagingFactory *factory;
static void test_createbitmap(void)
{
HRESULT hr;
IWICBitmap *bitmap;
IWICPalette *palette;
IWICBitmapLock *lock, *lock2;
WICBitmapPaletteType palettetype;
int i;
WICRect rc;
const BYTE bitmap_data[27] = {
128,128,255, 128,128,128, 128,255,128,
128,128,128, 128,128,128, 255,255,255,
255,128,128, 255,255,255, 255,255,255};
BYTE returned_data[27] = {0};
BYTE *lock_buffer=NULL, *base_lock_buffer=NULL;
UINT lock_buffer_size=0;
UINT lock_buffer_stride=0;
WICPixelFormatGUID pixelformat = {0};
UINT width=0, height=0;
double dpix=10.0, dpiy=10.0;
int can_lock_null = 1;
hr = IWICImagingFactory_CreateBitmap(factory, 3, 3, &GUID_WICPixelFormat24bppBGR,
WICBitmapCacheOnLoad, &bitmap);
ok(hr == S_OK, "IWICImagingFactory_CreateBitmap failed hr=%x\n", hr);
if (FAILED(hr))
return;
hr = IWICImagingFactory_CreatePalette(factory, &palette);
ok(hr == S_OK, "IWICImagingFactory_CreatePalette failed hr=%x\n", hr);
/* Palette is unavailable until explicitly set */
hr = IWICBitmap_CopyPalette(bitmap, palette);
ok(hr == WINCODEC_ERR_PALETTEUNAVAILABLE, "IWICBitmap_CopyPalette failed hr=%x\n", hr);
hr = IWICPalette_InitializePredefined(palette, WICBitmapPaletteTypeFixedGray256, FALSE);
ok(hr == S_OK, "IWICPalette_InitializePredefined failed hr=%x\n", hr);
hr = IWICBitmap_SetPalette(bitmap, palette);
ok(hr == S_OK, "IWICBitmap_SetPalette failed hr=%x\n", hr);
hr = IWICPalette_InitializePredefined(palette, WICBitmapPaletteTypeFixedGray4, FALSE);
ok(hr == S_OK, "IWICPalette_InitializePredefined failed hr=%x\n", hr);
hr = IWICBitmap_CopyPalette(bitmap, palette);
ok(hr == S_OK, "IWICBitmap_CopyPalette failed hr=%x\n", hr);
hr = IWICPalette_GetType(palette, &palettetype);
ok(hr == S_OK, "IWICPalette_GetType failed hr=%x\n", hr);
ok(palettetype == WICBitmapPaletteTypeFixedGray256,
"expected WICBitmapPaletteTypeFixedGray256, got %x\n", palettetype);
IWICPalette_Release(palette);
/* pixel data is initially zeroed */
hr = IWICBitmap_CopyPixels(bitmap, NULL, 9, 27, returned_data);
ok(hr == S_OK, "IWICBitmap_CopyPixels failed hr=%x\n", hr);
for (i=0; i<27; i++)
ok(returned_data[i] == 0, "returned_data[%i] == %i\n", i, returned_data[i]);
/* Invalid lock rects */
rc.X = rc.Y = 0;
rc.Width = 4;
rc.Height = 3;
hr = IWICBitmap_Lock(bitmap, &rc, WICBitmapLockRead, &lock);
ok(hr == E_INVALIDARG, "IWICBitmap_Lock failed hr=%x\n", hr);
if (SUCCEEDED(hr)) IWICBitmapLock_Release(lock);
rc.Width = 3;
rc.Height = 4;
hr = IWICBitmap_Lock(bitmap, &rc, WICBitmapLockRead, &lock);
ok(hr == E_INVALIDARG, "IWICBitmap_Lock failed hr=%x\n", hr);
if (SUCCEEDED(hr)) IWICBitmapLock_Release(lock);
rc.Height = 3;
rc.X = 4;
hr = IWICBitmap_Lock(bitmap, &rc, WICBitmapLockRead, &lock);
ok(hr == E_INVALIDARG, "IWICBitmap_Lock failed hr=%x\n", hr);
if (SUCCEEDED(hr)) IWICBitmapLock_Release(lock);
rc.X = 0;
rc.Y = 4;
hr = IWICBitmap_Lock(bitmap, &rc, WICBitmapLockRead, &lock);
ok(hr == E_INVALIDARG, "IWICBitmap_Lock failed hr=%x\n", hr);
if (SUCCEEDED(hr)) IWICBitmapLock_Release(lock);
/* NULL lock rect */
hr = IWICBitmap_Lock(bitmap, NULL, WICBitmapLockRead, &lock);
ok(hr == S_OK || broken(hr == E_INVALIDARG) /* winxp */, "IWICBitmap_Lock failed hr=%x\n", hr);
if (SUCCEEDED(hr))
{
/* entire bitmap is locked */
hr = IWICBitmapLock_GetSize(lock, &width, &height);
ok(hr == S_OK, "IWICBitmapLock_GetSize failed hr=%x\n", hr);
ok(width == 3, "got %d, expected 3\n", width);
ok(height == 3, "got %d, expected 3\n", height);
IWICBitmapLock_Release(lock);
}
else
can_lock_null = 0;
/* lock with a valid rect */
rc.Y = 0;
hr = IWICBitmap_Lock(bitmap, &rc, WICBitmapLockRead, &lock);
ok(hr == S_OK, "IWICBitmap_Lock failed hr=%x\n", hr);
if (SUCCEEDED(hr))
{
hr = IWICBitmapLock_GetStride(lock, &lock_buffer_stride);
ok(hr == S_OK, "IWICBitmapLock_GetStride failed hr=%x\n", hr);
/* stride is divisible by 4 */
ok(lock_buffer_stride == 12, "got %i, expected 12\n", lock_buffer_stride);
hr = IWICBitmapLock_GetDataPointer(lock, &lock_buffer_size, &lock_buffer);
ok(hr == S_OK, "IWICBitmapLock_GetDataPointer failed hr=%x\n", hr);
/* buffer size does not include padding from the last row */
ok(lock_buffer_size == 33, "got %i, expected 33\n", lock_buffer_size);
ok(lock_buffer != NULL, "got NULL data pointer\n");
base_lock_buffer = lock_buffer;
hr = IWICBitmapLock_GetPixelFormat(lock, &pixelformat);
ok(hr == S_OK, "IWICBitmapLock_GetPixelFormat failed hr=%x\n", hr);
ok(IsEqualGUID(&pixelformat, &GUID_WICPixelFormat24bppBGR), "unexpected pixel format\n");
hr = IWICBitmapLock_GetSize(lock, &width, &height);
ok(hr == S_OK, "IWICBitmapLock_GetSize failed hr=%x\n", hr);
ok(width == 3, "got %d, expected 3\n", width);
ok(height == 3, "got %d, expected 3\n", height);
/* We can have multiple simultaneous read locks */
hr = IWICBitmap_Lock(bitmap, &rc, WICBitmapLockRead, &lock2);
ok(hr == S_OK, "IWICBitmap_Lock failed hr=%x\n", hr);
if (SUCCEEDED(hr))
{
hr = IWICBitmapLock_GetDataPointer(lock2, &lock_buffer_size, &lock_buffer);
ok(hr == S_OK, "IWICBitmapLock_GetDataPointer failed hr=%x\n", hr);
ok(lock_buffer_size == 33, "got %i, expected 33\n", lock_buffer_size);
ok(lock_buffer == base_lock_buffer, "got %p, expected %p\n", lock_buffer, base_lock_buffer);
IWICBitmapLock_Release(lock2);
}
if (can_lock_null) /* this hangs on xp/vista */
{
/* But not a read and a write lock */
hr = IWICBitmap_Lock(bitmap, &rc, WICBitmapLockWrite, &lock2);
ok(hr == WINCODEC_ERR_ALREADYLOCKED, "IWICBitmap_Lock failed hr=%x\n", hr);
}
/* But we don't need a write lock to write */
if (base_lock_buffer)
{
for (i=0; i<3; i++)
memcpy(base_lock_buffer + lock_buffer_stride*i, bitmap_data + i*9, 9);
}
IWICBitmapLock_Release(lock);
}
/* test that the data we wrote is returned by CopyPixels */
hr = IWICBitmap_CopyPixels(bitmap, NULL, 9, 27, returned_data);
ok(hr == S_OK, "IWICBitmap_CopyPixels failed hr=%x\n", hr);
for (i=0; i<27; i++)
ok(returned_data[i] == bitmap_data[i], "returned_data[%i] == %i\n", i, returned_data[i]);
/* try a valid partial rect, and write mode */
rc.X = 2;
rc.Y = 0;
rc.Width = 1;
rc.Height = 2;
hr = IWICBitmap_Lock(bitmap, &rc, WICBitmapLockWrite, &lock);
ok(hr == S_OK, "IWICBitmap_Lock failed hr=%x\n", hr);
if (SUCCEEDED(hr))
{
if (can_lock_null) /* this hangs on xp/vista */
{
/* Can't lock again while locked for writing */
hr = IWICBitmap_Lock(bitmap, &rc, WICBitmapLockWrite, &lock2);
ok(hr == WINCODEC_ERR_ALREADYLOCKED, "IWICBitmap_Lock failed hr=%x\n", hr);
hr = IWICBitmap_Lock(bitmap, &rc, WICBitmapLockRead, &lock2);
ok(hr == WINCODEC_ERR_ALREADYLOCKED, "IWICBitmap_Lock failed hr=%x\n", hr);
}
hr = IWICBitmapLock_GetStride(lock, &lock_buffer_stride);
ok(hr == S_OK, "IWICBitmapLock_GetStride failed hr=%x\n", hr);
ok(lock_buffer_stride == 12, "got %i, expected 12\n", lock_buffer_stride);
hr = IWICBitmapLock_GetDataPointer(lock, &lock_buffer_size, &lock_buffer);
ok(hr == S_OK, "IWICBitmapLock_GetDataPointer failed hr=%x\n", hr);
ok(lock_buffer_size == 15, "got %i, expected 15\n", lock_buffer_size);
ok(lock_buffer == base_lock_buffer+6, "got %p, expected %p+6\n", lock_buffer, base_lock_buffer);
hr = IWICBitmapLock_GetPixelFormat(lock, &pixelformat);
ok(hr == S_OK, "IWICBitmapLock_GetPixelFormat failed hr=%x\n", hr);
ok(IsEqualGUID(&pixelformat, &GUID_WICPixelFormat24bppBGR), "unexpected pixel format\n");
hr = IWICBitmapLock_GetSize(lock, &width, &height);
ok(hr == S_OK, "IWICBitmapLock_GetSize failed hr=%x\n", hr);
ok(width == 1, "got %d, expected 1\n", width);
ok(height == 2, "got %d, expected 2\n", height);
IWICBitmapLock_Release(lock);
}
hr = IWICBitmap_GetPixelFormat(bitmap, &pixelformat);
ok(hr == S_OK, "IWICBitmap_GetPixelFormat failed hr=%x\n", hr);
ok(IsEqualGUID(&pixelformat, &GUID_WICPixelFormat24bppBGR), "unexpected pixel format\n");
hr = IWICBitmap_GetResolution(bitmap, &dpix, &dpiy);
ok(hr == S_OK, "IWICBitmap_GetResolution failed hr=%x\n", hr);
ok(dpix == 0.0, "got %f, expected 0.0\n", dpix);
ok(dpiy == 0.0, "got %f, expected 0.0\n", dpiy);
hr = IWICBitmap_SetResolution(bitmap, 12.0, 34.0);
ok(hr == S_OK, "IWICBitmap_SetResolution failed hr=%x\n", hr);
hr = IWICBitmap_GetResolution(bitmap, &dpix, &dpiy);
ok(hr == S_OK, "IWICBitmap_GetResolution failed hr=%x\n", hr);
ok(dpix == 12.0, "got %f, expected 12.0\n", dpix);
ok(dpiy == 34.0, "got %f, expected 34.0\n", dpiy);
hr = IWICBitmap_GetSize(bitmap, &width, &height);
ok(hr == S_OK, "IWICBitmap_GetSize failed hr=%x\n", hr);
ok(width == 3, "got %d, expected 3\n", width);
ok(height == 3, "got %d, expected 3\n", height);
IWICBitmap_Release(bitmap);
}
static void test_createbitmapfromsource(void)
{
HRESULT hr;
IWICBitmap *bitmap, *bitmap2;
IWICPalette *palette;
IWICBitmapLock *lock;
int i;
WICRect rc;
const BYTE bitmap_data[27] = {
128,128,255, 128,128,128, 128,255,128,
128,128,128, 128,128,128, 255,255,255,
255,128,128, 255,255,255, 255,255,255};
BYTE returned_data[27] = {0};
BYTE *lock_buffer=NULL;
UINT lock_buffer_stride=0;
UINT lock_buffer_size=0;
WICPixelFormatGUID pixelformat = {0};
UINT width=0, height=0;
double dpix=10.0, dpiy=10.0;
UINT count;
WICBitmapPaletteType palette_type;
hr = IWICImagingFactory_CreateBitmap(factory, 3, 3, &GUID_WICPixelFormat24bppBGR,
WICBitmapCacheOnLoad, &bitmap);
ok(hr == S_OK, "IWICImagingFactory_CreateBitmap failed hr=%x\n", hr);
if (FAILED(hr))
return;
hr = IWICImagingFactory_CreatePalette(factory, &palette);
ok(hr == S_OK, "IWICImagingFactory_CreatePalette failed hr=%x\n", hr);
hr = IWICPalette_InitializePredefined(palette, WICBitmapPaletteTypeFixedGray256, FALSE);
ok(hr == S_OK, "IWICPalette_InitializePredefined failed hr=%x\n", hr);
hr = IWICBitmap_SetPalette(bitmap, palette);
ok(hr == S_OK, "IWICBitmap_SetPalette failed hr=%x\n", hr);
IWICPalette_Release(palette);
rc.X = rc.Y = 0;
rc.Width = 3;
rc.Height = 3;
hr = IWICBitmap_Lock(bitmap, &rc, WICBitmapLockWrite, &lock);
ok(hr == S_OK, "IWICBitmap_Lock failed hr=%x\n", hr);
if (SUCCEEDED(hr))
{
hr = IWICBitmapLock_GetStride(lock, &lock_buffer_stride);
ok(hr == S_OK, "IWICBitmapLock_GetStride failed hr=%x\n", hr);
ok(lock_buffer_stride == 12, "got %i, expected 12\n", lock_buffer_stride);
hr = IWICBitmapLock_GetDataPointer(lock, &lock_buffer_size, &lock_buffer);
ok(hr == S_OK, "IWICBitmapLock_GetDataPointer failed hr=%x\n", hr);
ok(lock_buffer_size == 33, "got %i, expected 33\n", lock_buffer_size);
ok(lock_buffer != NULL, "got NULL data pointer\n");
for (i=0; i<3; i++)
memcpy(lock_buffer + lock_buffer_stride*i, bitmap_data + i*9, 9);
IWICBitmapLock_Release(lock);
}
hr = IWICBitmap_SetResolution(bitmap, 12.0, 34.0);
ok(hr == S_OK, "IWICBitmap_SetResolution failed hr=%x\n", hr);
hr = IWICImagingFactory_CreateBitmapFromSource(factory, (IWICBitmapSource*)bitmap,
WICBitmapCacheOnLoad, &bitmap2);
ok(hr == S_OK, "IWICImagingFactory_CreateBitmapFromSource failed hr=%x\n", hr);
IWICBitmap_Release(bitmap);
if (FAILED(hr)) return;
hr = IWICImagingFactory_CreatePalette(factory, &palette);
ok(hr == S_OK, "IWICImagingFactory_CreatePalette failed hr=%x\n", hr);
/* palette isn't copied for non-indexed formats? */
hr = IWICBitmap_CopyPalette(bitmap2, palette);
ok(hr == WINCODEC_ERR_PALETTEUNAVAILABLE, "IWICBitmap_CopyPalette failed hr=%x\n", hr);
IWICPalette_Release(palette);
hr = IWICBitmap_CopyPixels(bitmap2, NULL, 9, 27, returned_data);
ok(hr == S_OK, "IWICBitmap_CopyPixels failed hr=%x\n", hr);
for (i=0; i<27; i++)
ok(returned_data[i] == bitmap_data[i], "returned_data[%i] == %i\n", i, returned_data[i]);
hr = IWICBitmap_GetPixelFormat(bitmap2, &pixelformat);
ok(hr == S_OK, "IWICBitmap_GetPixelFormat failed hr=%x\n", hr);
ok(IsEqualGUID(&pixelformat, &GUID_WICPixelFormat24bppBGR), "unexpected pixel format\n");
hr = IWICBitmap_GetResolution(bitmap2, &dpix, &dpiy);
ok(hr == S_OK, "IWICBitmap_GetResolution failed hr=%x\n", hr);
ok(dpix == 12.0, "got %f, expected 12.0\n", dpix);
ok(dpiy == 34.0, "got %f, expected 34.0\n", dpiy);
hr = IWICBitmap_GetSize(bitmap2, &width, &height);
ok(hr == S_OK, "IWICBitmap_GetSize failed hr=%x\n", hr);
ok(width == 3, "got %d, expected 3\n", width);
ok(height == 3, "got %d, expected 3\n", height);
IWICBitmap_Release(bitmap2);
/* Ensure palette is copied for indexed formats */
hr = IWICImagingFactory_CreateBitmap(factory, 3, 3, &GUID_WICPixelFormat4bppIndexed,
WICBitmapCacheOnLoad, &bitmap);
ok(hr == S_OK, "IWICImagingFactory_CreateBitmap failed hr=%x\n", hr);
hr = IWICImagingFactory_CreatePalette(factory, &palette);
ok(hr == S_OK, "IWICImagingFactory_CreatePalette failed hr=%x\n", hr);
hr = IWICPalette_InitializePredefined(palette, WICBitmapPaletteTypeFixedGray256, FALSE);
ok(hr == S_OK, "IWICPalette_InitializePredefined failed hr=%x\n", hr);
hr = IWICBitmap_SetPalette(bitmap, palette);
ok(hr == S_OK, "IWICBitmap_SetPalette failed hr=%x\n", hr);
IWICPalette_Release(palette);
hr = IWICImagingFactory_CreateBitmapFromSource(factory, (IWICBitmapSource*)bitmap,
WICBitmapCacheOnLoad, &bitmap2);
ok(hr == S_OK, "IWICImagingFactory_CreateBitmapFromSource failed hr=%x\n", hr);
IWICBitmap_Release(bitmap);
hr = IWICImagingFactory_CreatePalette(factory, &palette);
ok(hr == S_OK, "IWICImagingFactory_CreatePalette failed hr=%x\n", hr);
hr = IWICBitmap_CopyPalette(bitmap2, palette);
ok(hr == S_OK, "IWICBitmap_CopyPalette failed hr=%x\n", hr);
hr = IWICPalette_GetColorCount(palette, &count);
ok(hr == S_OK, "IWICPalette_GetColorCount failed hr=%x\n", hr);
ok(count == 256, "unexpected count %d\n", count);
hr = IWICPalette_GetType(palette, &palette_type);
ok(hr == S_OK, "IWICPalette_GetType failed hr=%x\n", hr);
ok(palette_type == WICBitmapPaletteTypeFixedGray256, "unexpected palette type %d\n", palette_type);
IWICPalette_Release(palette);
hr = IWICBitmap_GetPixelFormat(bitmap2, &pixelformat);
ok(hr == S_OK, "IWICBitmap_GetPixelFormat failed hr=%x\n", hr);
ok(IsEqualGUID(&pixelformat, &GUID_WICPixelFormat4bppIndexed), "unexpected pixel format\n");
hr = IWICBitmap_GetSize(bitmap2, &width, &height);
ok(hr == S_OK, "IWICBitmap_GetSize failed hr=%x\n", hr);
ok(width == 3, "got %d, expected 3\n", width);
ok(height == 3, "got %d, expected 3\n", height);
IWICBitmap_Release(bitmap2);
}
START_TEST(bitmap)
{
HRESULT hr;
CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
hr = CoCreateInstance(&CLSID_WICImagingFactory, NULL, CLSCTX_INPROC_SERVER,
&IID_IWICImagingFactory, (void**)&factory);
ok(SUCCEEDED(hr), "CoCreateInstance failed, hr=%x\n", hr);
test_createbitmap();
test_createbitmapfromsource();
IWICImagingFactory_Release(factory);
CoUninitialize();
}