git.oblomov.eu Git - wine/blob - dlls/d3dx9_36/surface.c

   1 /*
   2  * Copyright (C) 2009-2010 Tony Wasserka
   3  * Copyright (C) 2012 Józef Kucia
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2.1 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General Public
  16  * License along with this library; if not, write to the Free Software
  17  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  18  *
  19  */
  20
  21 #include "wine/debug.h"
  22 #include "wine/unicode.h"
  23 #include "d3dx9_36_private.h"
  24
  25 #include "initguid.h"
  26 #include "ole2.h"
  27 #include "wincodec.h"
  28
  29 WINE_DEFAULT_DEBUG_CHANNEL(d3dx);
  30
  31
  32 /* Wine-specific WIC GUIDs */
  33 DEFINE_GUID(GUID_WineContainerFormatTga, 0x0c44fda1,0xa5c5,0x4298,0x96,0x85,0x47,0x3f,0xc1,0x7c,0xd3,0x22);
  34
  35 static const struct
  36 {
  37     const GUID *wic_guid;
  38     D3DFORMAT d3dformat;
  39 } wic_pixel_formats[] = {
  40     { &GUID_WICPixelFormat8bppIndexed, D3DFMT_L8 },
  41     { &GUID_WICPixelFormat1bppIndexed, D3DFMT_L8 },
  42     { &GUID_WICPixelFormat4bppIndexed, D3DFMT_L8 },
  43     { &GUID_WICPixelFormat16bppBGR555, D3DFMT_X1R5G5B5 },
  44     { &GUID_WICPixelFormat16bppBGR565, D3DFMT_R5G6B5 },
  45     { &GUID_WICPixelFormat24bppBGR, D3DFMT_R8G8B8 },
  46     { &GUID_WICPixelFormat32bppBGR, D3DFMT_X8R8G8B8 },
  47     { &GUID_WICPixelFormat32bppBGRA, D3DFMT_A8R8G8B8 }
  48 };
  49
  50 static D3DFORMAT wic_guid_to_d3dformat(const GUID *guid)
  51 {
  52     int i;
  53
  54     for (i = 0; i < sizeof(wic_pixel_formats) / sizeof(wic_pixel_formats[0]); i++)
  55     {
  56         if (IsEqualGUID(wic_pixel_formats[i].wic_guid, guid))
  57             return wic_pixel_formats[i].d3dformat;
  58     }
  59
  60     return D3DFMT_UNKNOWN;
  61 }
  62
  63 static const GUID *d3dformat_to_wic_guid(D3DFORMAT format)
  64 {
  65     int i;
  66
  67     for (i = 0; i < sizeof(wic_pixel_formats) / sizeof(wic_pixel_formats[0]); i++)
  68     {
  69         if (wic_pixel_formats[i].d3dformat == format)
  70             return wic_pixel_formats[i].wic_guid;
  71     }
  72
  73     return NULL;
  74 }
  75
  76 /* dds_header.flags */
  77 #define DDS_CAPS 0x1
  78 #define DDS_HEIGHT 0x2
  79 #define DDS_WIDTH 0x2
  80 #define DDS_PITCH 0x8
  81 #define DDS_PIXELFORMAT 0x1000
  82 #define DDS_MIPMAPCOUNT 0x20000
  83 #define DDS_LINEARSIZE 0x80000
  84 #define DDS_DEPTH 0x800000
  85
  86 /* dds_header.caps */
  87 #define DDS_CAPS_COMPLEX 0x8
  88 #define DDS_CAPS_TEXTURE 0x1000
  89 #define DDS_CAPS_MIPMAP 0x400000
  90
  91 /* dds_header.caps2 */
  92 #define DDS_CAPS2_CUBEMAP 0x200
  93 #define DDS_CAPS2_CUBEMAP_POSITIVEX 0x400
  94 #define DDS_CAPS2_CUBEMAP_NEGATIVEX 0x800
  95 #define DDS_CAPS2_CUBEMAP_POSITIVEY 0x1000
  96 #define DDS_CAPS2_CUBEMAP_NEGATIVEY 0x2000
  97 #define DDS_CAPS2_CUBEMAP_POSITIVEZ 0x4000
  98 #define DDS_CAPS2_CUBEMAP_NEGATIVEZ 0x8000
  99 #define DDS_CAPS2_CUBEMAP_ALL_FACES ( DDS_CAPS2_CUBEMAP_POSITIVEX | DDS_CAPS2_CUBEMAP_NEGATIVEX \
 100                                     | DDS_CAPS2_CUBEMAP_POSITIVEY | DDS_CAPS2_CUBEMAP_NEGATIVEY \
 101                                     | DDS_CAPS2_CUBEMAP_POSITIVEZ | DDS_CAPS2_CUBEMAP_NEGATIVEZ )
 102 #define DDS_CAPS2_VOLUME 0x200000
 103
 104 /* dds_pixel_format.flags */
 105 #define DDS_PF_ALPHA 0x1
 106 #define DDS_PF_ALPHA_ONLY 0x2
 107 #define DDS_PF_FOURCC 0x4
 108 #define DDS_PF_RGB 0x40
 109 #define DDS_PF_YUV 0x200
 110 #define DDS_PF_LUMINANCE 0x20000
 111 #define DDS_PF_BUMPDUDV 0x80000
 112
 113 struct dds_pixel_format
 114 {
 115     DWORD size;
 116     DWORD flags;
 117     DWORD fourcc;
 118     DWORD bpp;
 119     DWORD rmask;
 120     DWORD gmask;
 121     DWORD bmask;
 122     DWORD amask;
 123 };
 124
 125 struct dds_header
 126 {
 127     DWORD signature;
 128     DWORD size;
 129     DWORD flags;
 130     DWORD height;
 131     DWORD width;
 132     DWORD pitch_or_linear_size;
 133     DWORD depth;
 134     DWORD miplevels;
 135     DWORD reserved[11];
 136     struct dds_pixel_format pixel_format;
 137     DWORD caps;
 138     DWORD caps2;
 139     DWORD caps3;
 140     DWORD caps4;
 141     DWORD reserved2;
 142 };
 143
 144 static D3DFORMAT dds_fourcc_to_d3dformat(DWORD fourcc)
 145 {
 146     int i;
 147     static const DWORD known_fourcc[] = {
 148         MAKEFOURCC('U','Y','V','Y'),
 149         MAKEFOURCC('Y','U','Y','2'),
 150         MAKEFOURCC('R','G','B','G'),
 151         MAKEFOURCC('G','R','G','B'),
 152         MAKEFOURCC('D','X','T','1'),
 153         MAKEFOURCC('D','X','T','2'),
 154         MAKEFOURCC('D','X','T','3'),
 155         MAKEFOURCC('D','X','T','4'),
 156         MAKEFOURCC('D','X','T','5')
 157     };
 158
 159     for (i = 0; i < sizeof(known_fourcc) / sizeof(known_fourcc[0]); i++)
 160     {
 161         if (known_fourcc[i] == fourcc)
 162             return fourcc;
 163     }
 164
 165     WARN("Unknown FourCC %#x\n", fourcc);
 166     return D3DFMT_UNKNOWN;
 167 }
 168
 169 static D3DFORMAT dds_rgb_to_d3dformat(const struct dds_pixel_format *pixel_format)
 170 {
 171     int i;
 172     static const struct {
 173         DWORD bpp;
 174         DWORD rmask;
 175         DWORD gmask;
 176         DWORD bmask;
 177         DWORD amask;
 178         D3DFORMAT format;
 179     } rgb_pixel_formats[] = {
 180         { 8, 0xe0, 0x1c, 0x03, 0, D3DFMT_R3G3B2 },
 181         { 16, 0xf800, 0x07e0, 0x001f, 0x0000, D3DFMT_R5G6B5 },
 182         { 16, 0x7c00, 0x03e0, 0x001f, 0x8000, D3DFMT_A1R5G5B5 },
 183         { 16, 0x7c00, 0x03e0, 0x001f, 0x0000, D3DFMT_X1R5G5B5 },
 184         { 16, 0x0f00, 0x00f0, 0x000f, 0xf000, D3DFMT_A4R4G4B4 },
 185         { 16, 0x0f00, 0x00f0, 0x000f, 0x0000, D3DFMT_X4R4G4B4 },
 186         { 16, 0x00e0, 0x001c, 0x0003, 0xff00, D3DFMT_A8R3G3B2 },
 187         { 24, 0xff0000, 0x00ff00, 0x0000ff, 0x000000, D3DFMT_R8G8B8 },
 188         { 32, 0x00ff0000, 0x0000ff00, 0x000000ff, 0xff000000, D3DFMT_A8R8G8B8 },
 189         { 32, 0x00ff0000, 0x0000ff00, 0x000000ff, 0x00000000, D3DFMT_X8R8G8B8 },
 190         { 32, 0x3ff00000, 0x000ffc00, 0x000003ff, 0xc0000000, D3DFMT_A2B10G10R10 },
 191         { 32, 0x000003ff, 0x000ffc00, 0x3ff00000, 0xc0000000, D3DFMT_A2R10G10B10 },
 192         { 32, 0x0000ffff, 0xffff0000, 0x00000000, 0x00000000, D3DFMT_G16R16 },
 193     };
 194
 195     for (i = 0; i < sizeof(rgb_pixel_formats) / sizeof(rgb_pixel_formats[0]); i++)
 196     {
 197         if (rgb_pixel_formats[i].bpp == pixel_format->bpp
 198             && rgb_pixel_formats[i].rmask == pixel_format->rmask
 199             && rgb_pixel_formats[i].gmask == pixel_format->gmask
 200             && rgb_pixel_formats[i].bmask == pixel_format->bmask)
 201         {
 202             if ((pixel_format->flags & DDS_PF_ALPHA) && rgb_pixel_formats[i].amask == pixel_format->amask)
 203                 return rgb_pixel_formats[i].format;
 204             if (rgb_pixel_formats[i].amask == 0)
 205                 return rgb_pixel_formats[i].format;
 206         }
 207     }
 208
 209     WARN("Unknown RGB pixel format (%#x, %#x, %#x, %#x)\n",
 210         pixel_format->rmask, pixel_format->gmask, pixel_format->bmask, pixel_format->amask);
 211     return D3DFMT_UNKNOWN;
 212 }
 213
 214 static D3DFORMAT dds_luminance_to_d3dformat(const struct dds_pixel_format *pixel_format)
 215 {
 216     if (pixel_format->bpp == 8)
 217     {
 218         if (pixel_format->rmask == 0xff)
 219             return D3DFMT_L8;
 220         if ((pixel_format->flags & DDS_PF_ALPHA) && pixel_format->rmask == 0x0f && pixel_format->amask == 0xf0)
 221             return D3DFMT_A4L4;
 222     }
 223     if (pixel_format->bpp == 16)
 224     {
 225         if (pixel_format->rmask == 0xffff)
 226             return D3DFMT_L16;
 227         if ((pixel_format->flags & DDS_PF_ALPHA) && pixel_format->rmask == 0x00ff && pixel_format->amask == 0xff00)
 228             return D3DFMT_A8L8;
 229     }
 230
 231     WARN("Unknown luminance pixel format (bpp %#x, l %#x, a %#x)\n",
 232         pixel_format->bpp, pixel_format->rmask, pixel_format->amask);
 233     return D3DFMT_UNKNOWN;
 234 }
 235
 236 static D3DFORMAT dds_alpha_to_d3dformat(const struct dds_pixel_format *pixel_format)
 237 {
 238     if (pixel_format->bpp == 8 && pixel_format->amask == 0xff)
 239         return D3DFMT_A8;
 240
 241     WARN("Unknown Alpha pixel format (%#x, %#x)\n", pixel_format->bpp, pixel_format->rmask);
 242     return D3DFMT_UNKNOWN;
 243 }
 244
 245 static D3DFORMAT dds_bump_to_d3dformat(const struct dds_pixel_format *pixel_format)
 246 {
 247     if (pixel_format->bpp == 16 && pixel_format->rmask == 0x00ff && pixel_format->gmask == 0xff00)
 248         return D3DFMT_V8U8;
 249     if (pixel_format->bpp == 32 && pixel_format->rmask == 0x0000ffff && pixel_format->gmask == 0xffff0000)
 250         return D3DFMT_V16U16;
 251
 252     WARN("Unknown bump pixel format (%#x, %#x, %#x, %#x, %#x)\n", pixel_format->bpp,
 253         pixel_format->rmask, pixel_format->gmask, pixel_format->bmask, pixel_format->amask);
 254     return D3DFMT_UNKNOWN;
 255 }
 256
 257 static D3DFORMAT dds_pixel_format_to_d3dformat(const struct dds_pixel_format *pixel_format)
 258 {
 259     if (pixel_format->flags & DDS_PF_FOURCC)
 260         return dds_fourcc_to_d3dformat(pixel_format->fourcc);
 261     if (pixel_format->flags & DDS_PF_RGB)
 262         return dds_rgb_to_d3dformat(pixel_format);
 263     if (pixel_format->flags & DDS_PF_LUMINANCE)
 264         return dds_luminance_to_d3dformat(pixel_format);
 265     if (pixel_format->flags & DDS_PF_ALPHA_ONLY)
 266         return dds_alpha_to_d3dformat(pixel_format);
 267     if (pixel_format->flags & DDS_PF_BUMPDUDV)
 268         return dds_bump_to_d3dformat(pixel_format);
 269
 270     WARN("Unknown pixel format (flags %#x, fourcc %#x, bpp %#x, r %#x, g %#x, b %#x, a %#x)\n",
 271         pixel_format->flags, pixel_format->fourcc, pixel_format->bpp,
 272         pixel_format->rmask, pixel_format->gmask, pixel_format->bmask, pixel_format->amask);
 273     return D3DFMT_UNKNOWN;
 274 }
 275
 276 static HRESULT calculate_dds_surface_size(const D3DXIMAGE_INFO *img_info,
 277     UINT width, UINT height, UINT *pitch, UINT *size)
 278 {
 279     const PixelFormatDesc *format_desc = get_format_info(img_info->Format);
 280     if (format_desc->format == D3DFMT_UNKNOWN)
 281         return E_NOTIMPL;
 282
 283     if (format_desc->block_width != 1 || format_desc->block_height != 1)
 284     {
 285         *pitch = format_desc->block_byte_count
 286             * max(1, (width + format_desc->block_width - 1) / format_desc->block_width);
 287         *size = *pitch
 288             * max(1, (height + format_desc->block_height - 1) / format_desc->block_height);
 289     }
 290     else
 291     {
 292         *pitch = width * format_desc->bytes_per_pixel;
 293         *size = *pitch * height;
 294     }
 295
 296     return D3D_OK;
 297 }
 298
 299 /************************************************************
 300 * get_image_info_from_dds
 301 *
 302 * Fills a D3DXIMAGE_INFO structure with information
 303 * about a DDS file stored in the memory.
 304 *
 305 * PARAMS
 306 *   buffer  [I] pointer to DDS data
 307 *   length  [I] size of DDS data
 308 *   info    [O] pointer to D3DXIMAGE_INFO structure
 309 *
 310 * RETURNS
 311 *   Success: D3D_OK
 312 *   Failure: D3DXERR_INVALIDDATA
 313 *
 314 */
 315 static HRESULT get_image_info_from_dds(const void *buffer, UINT length, D3DXIMAGE_INFO *info)
 316 {
 317    UINT i;
 318    UINT faces = 0;
 319    UINT width, height;
 320    const struct dds_header *header = buffer;
 321    UINT expected_length = 0;
 322
 323    if (length < sizeof(*header) || !info)
 324        return D3DXERR_INVALIDDATA;
 325
 326    if (header->pixel_format.size != sizeof(header->pixel_format))
 327        return D3DXERR_INVALIDDATA;
 328
 329    info->Width = header->width;
 330    info->Height = header->height;
 331    info->Depth = 1;
 332    info->MipLevels = (header->flags & DDS_MIPMAPCOUNT) ?  header->miplevels : 1;
 333
 334    info->Format = dds_pixel_format_to_d3dformat(&header->pixel_format);
 335    if (info->Format == D3DFMT_UNKNOWN)
 336        return D3DXERR_INVALIDDATA;
 337
 338    TRACE("Pixel format is %#x\n", info->Format);
 339
 340    if (header->caps2 & DDS_CAPS2_VOLUME)
 341    {
 342        info->Depth = header->depth;
 343        info->ResourceType = D3DRTYPE_VOLUMETEXTURE;
 344    }
 345    else if (header->caps2 & DDS_CAPS2_CUBEMAP)
 346    {
 347        DWORD face;
 348        for (face = DDS_CAPS2_CUBEMAP_POSITIVEX; face <= DDS_CAPS2_CUBEMAP_NEGATIVEZ; face <<= 1)
 349        {
 350            if (header->caps2 & face)
 351                faces++;
 352        }
 353        info->ResourceType = D3DRTYPE_CUBETEXTURE;
 354    }
 355    else
 356    {
 357        faces = 1;
 358        info->ResourceType = D3DRTYPE_TEXTURE;
 359    }
 360
 361    /* calculate the expected length */
 362    width = info->Width;
 363    height = info->Height;
 364    for (i = 0; i < info->MipLevels; i++)
 365    {
 366        UINT pitch, size = 0;
 367        calculate_dds_surface_size(info, width, height, &pitch, &size);
 368        expected_length += size;
 369        width = max(1, width / 2);
 370        height = max(1, height / 2);
 371    }
 372
 373    expected_length *= faces;
 374    expected_length += sizeof(*header);
 375    if (length < expected_length)
 376    {
 377        WARN("File is too short %u, expected at least %u bytes\n", length, expected_length);
 378        return D3DXERR_INVALIDDATA;
 379    }
 380
 381    info->ImageFileFormat = D3DXIFF_DDS;
 382
 383    return D3D_OK;
 384 }
 385
 386 static HRESULT load_surface_from_dds(IDirect3DSurface9 *dst_surface, const PALETTEENTRY *dst_palette,
 387     const RECT *dst_rect, const void *src_data, const RECT *src_rect, DWORD filter, D3DCOLOR color_key,
 388     const D3DXIMAGE_INFO *src_info)
 389 {
 390     UINT size;
 391     UINT src_pitch;
 392     const struct dds_header *header = src_data;
 393     const BYTE *pixels = (BYTE *)(header + 1);
 394
 395     if (src_info->ResourceType != D3DRTYPE_TEXTURE)
 396         return D3DXERR_INVALIDDATA;
 397
 398     if (FAILED(calculate_dds_surface_size(src_info, src_info->Width, src_info->Height, &src_pitch, &size)))
 399         return E_NOTIMPL;
 400
 401     return D3DXLoadSurfaceFromMemory(dst_surface, dst_palette, dst_rect, pixels, src_info->Format,
 402         src_pitch, NULL, src_rect, filter, color_key);
 403 }
 404
 405 HRESULT load_texture_from_dds(IDirect3DTexture9 *texture, const void *src_data, const PALETTEENTRY *palette,
 406     DWORD filter, D3DCOLOR color_key, const D3DXIMAGE_INFO *src_info)
 407
 408 {
 409     HRESULT hr;
 410     RECT src_rect;
 411     UINT src_pitch;
 412     UINT mip_level;
 413     UINT mip_levels;
 414     UINT mip_level_size;
 415     UINT width, height;
 416     IDirect3DSurface9 *surface;
 417     const struct dds_header *header = src_data;
 418     const BYTE *pixels = (BYTE *)(header + 1);
 419
 420     if (src_info->ResourceType != D3DRTYPE_TEXTURE)
 421         return D3DXERR_INVALIDDATA;
 422
 423     width = src_info->Width;
 424     height = src_info->Height;
 425     mip_levels = min(src_info->MipLevels, IDirect3DTexture9_GetLevelCount(texture));
 426     for (mip_level = 0; mip_level < mip_levels; mip_level++)
 427     {
 428         hr = calculate_dds_surface_size(src_info, width, height, &src_pitch, &mip_level_size);
 429         if (FAILED(hr)) return hr;
 430
 431         SetRect(&src_rect, 0, 0, width, height);
 432
 433         IDirect3DTexture9_GetSurfaceLevel(texture, mip_level, &surface);
 434         hr = D3DXLoadSurfaceFromMemory(surface, palette, NULL, pixels, src_info->Format, src_pitch,
 435             NULL, &src_rect, filter, color_key);
 436         IDirect3DSurface9_Release(surface);
 437         if (FAILED(hr)) return hr;
 438
 439         pixels += mip_level_size;
 440         width = max(1, width / 2);
 441         height = max(1, height / 2);
 442     }
 443
 444     return D3D_OK;
 445 }
 446
 447 HRESULT load_cube_texture_from_dds(IDirect3DCubeTexture9 *cube_texture, const void *src_data,
 448     const PALETTEENTRY *palette, DWORD filter, DWORD color_key, const D3DXIMAGE_INFO *src_info)
 449 {
 450     HRESULT hr;
 451     int face;
 452     int mip_level;
 453     UINT size;
 454     RECT src_rect;
 455     UINT src_pitch;
 456     UINT mip_levels;
 457     UINT mip_level_size;
 458     IDirect3DSurface9 *surface;
 459     const struct dds_header *header = src_data;
 460     const BYTE *pixels = (BYTE *)(header + 1);
 461
 462     if (src_info->ResourceType != D3DRTYPE_CUBETEXTURE)
 463         return D3DXERR_INVALIDDATA;
 464
 465     if ((header->caps2 & DDS_CAPS2_CUBEMAP_ALL_FACES) != DDS_CAPS2_CUBEMAP_ALL_FACES)
 466     {
 467         WARN("Only full cubemaps are supported\n");
 468         return D3DXERR_INVALIDDATA;
 469     }
 470
 471     mip_levels = min(src_info->MipLevels, IDirect3DCubeTexture9_GetLevelCount(cube_texture));
 472     for (face = D3DCUBEMAP_FACE_POSITIVE_X; face <= D3DCUBEMAP_FACE_NEGATIVE_Z; face++)
 473     {
 474         size = src_info->Width;
 475         for (mip_level = 0; mip_level < src_info->MipLevels; mip_level++)
 476         {
 477             hr = calculate_dds_surface_size(src_info, size, size, &src_pitch, &mip_level_size);
 478             if (FAILED(hr)) return hr;
 479
 480             /* if texture has fewer mip levels than DDS file, skip excessive mip levels */
 481             if (mip_level < mip_levels)
 482             {
 483                 SetRect(&src_rect, 0, 0, size, size);
 484
 485                 IDirect3DCubeTexture9_GetCubeMapSurface(cube_texture, face, mip_level, &surface);
 486                 hr = D3DXLoadSurfaceFromMemory(surface, palette, NULL, pixels, src_info->Format, src_pitch,
 487                     NULL, &src_rect, filter, color_key);
 488                 IDirect3DSurface9_Release(surface);
 489                 if (FAILED(hr)) return hr;
 490             }
 491
 492             pixels += mip_level_size;
 493             size = max(1, size / 2);
 494         }
 495     }
 496
 497     return D3D_OK;
 498 }
 499
 500 /************************************************************
 501  * D3DXGetImageInfoFromFileInMemory
 502  *
 503  * Fills a D3DXIMAGE_INFO structure with info about an image
 504  *
 505  * PARAMS
 506  *   data     [I] pointer to the image file data
 507  *   datasize [I] size of the passed data
 508  *   info     [O] pointer to the destination structure
 509  *
 510  * RETURNS
 511  *   Success: D3D_OK, if info is not NULL and data and datasize make up a valid image file or
 512  *                    if info is NULL and data and datasize are not NULL
 513  *   Failure: D3DXERR_INVALIDDATA, if data is no valid image file and datasize and info are not NULL
 514  *            D3DERR_INVALIDCALL, if data is NULL or
 515  *                                if datasize is 0
 516  *
 517  * NOTES
 518  *   datasize may be bigger than the actual file size
 519  *
 520  */
 521 HRESULT WINAPI D3DXGetImageInfoFromFileInMemory(LPCVOID data, UINT datasize, D3DXIMAGE_INFO *info)
 522 {
 523     IWICImagingFactory *factory;
 524     IWICBitmapDecoder *decoder = NULL;
 525     IWICStream *stream;
 526     HRESULT hr;
 527     HRESULT initresult;
 528
 529     TRACE("(%p, %d, %p)\n", data, datasize, info);
 530
 531     if (!data || !datasize)
 532         return D3DERR_INVALIDCALL;
 533
 534     if (!info)
 535         return D3D_OK;
 536
 537     if ((datasize >= 4) && !strncmp(data, "DDS ", 4)) {
 538         TRACE("File type is DDS\n");
 539         return get_image_info_from_dds(data, datasize, info);
 540     }
 541
 542     initresult = CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
 543
 544     hr = CoCreateInstance(&CLSID_WICImagingFactory, NULL, CLSCTX_INPROC_SERVER, &IID_IWICImagingFactory, (void**)&factory);
 545
 546     if (SUCCEEDED(hr)) {
 547         IWICImagingFactory_CreateStream(factory, &stream);
 548         IWICStream_InitializeFromMemory(stream, (BYTE*)data, datasize);
 549         hr = IWICImagingFactory_CreateDecoderFromStream(factory, (IStream*)stream, NULL, 0, &decoder);
 550         IStream_Release(stream);
 551         IWICImagingFactory_Release(factory);
 552     }
 553
 554     if (FAILED(hr)) {
 555         if ((datasize >= 2) && (!strncmp(data, "P3", 2) || !strncmp(data, "P6", 2)))
 556             FIXME("File type PPM is not supported yet\n");
 557         else if ((datasize >= 2) && !strncmp(data, "BM", 2))
 558             FIXME("File type DIB is not supported yet\n");
 559         else if ((datasize >= 10) && !strncmp(data, "#?RADIANCE", 10))
 560             FIXME("File type HDR is not supported yet\n");
 561         else if ((datasize >= 2) && (!strncmp(data, "PF", 2) || !strncmp(data, "Pf", 2)))
 562             FIXME("File type PFM is not supported yet\n");
 563     }
 564
 565     if (SUCCEEDED(hr)) {
 566         GUID container_format;
 567         UINT frame_count;
 568
 569         hr = IWICBitmapDecoder_GetContainerFormat(decoder, &container_format);
 570         if (SUCCEEDED(hr)) {
 571             if (IsEqualGUID(&container_format, &GUID_ContainerFormatBmp)) {
 572                 TRACE("File type is BMP\n");
 573                 info->ImageFileFormat = D3DXIFF_BMP;
 574             } else if (IsEqualGUID(&container_format, &GUID_ContainerFormatPng)) {
 575                 TRACE("File type is PNG\n");
 576                 info->ImageFileFormat = D3DXIFF_PNG;
 577             } else if(IsEqualGUID(&container_format, &GUID_ContainerFormatJpeg)) {
 578                 TRACE("File type is JPG\n");
 579                 info->ImageFileFormat = D3DXIFF_JPG;
 580             } else if(IsEqualGUID(&container_format, &GUID_WineContainerFormatTga)) {
 581                 TRACE("File type is TGA\n");
 582                 info->ImageFileFormat = D3DXIFF_TGA;
 583             } else {
 584                 WARN("Unsupported image file format %s\n", debugstr_guid(&container_format));
 585                 hr = D3DXERR_INVALIDDATA;
 586             }
 587         }
 588
 589         if (SUCCEEDED(hr))
 590             hr = IWICBitmapDecoder_GetFrameCount(decoder, &frame_count);
 591         if (SUCCEEDED(hr) && !frame_count)
 592             hr = D3DXERR_INVALIDDATA;
 593
 594         if (SUCCEEDED(hr)) {
 595             IWICBitmapFrameDecode *frame = NULL;
 596
 597             hr = IWICBitmapDecoder_GetFrame(decoder, 0, &frame);
 598
 599             if (SUCCEEDED(hr))
 600                 hr = IWICBitmapFrameDecode_GetSize(frame, &info->Width, &info->Height);
 601
 602             if (SUCCEEDED(hr)) {
 603                 WICPixelFormatGUID pixel_format;
 604
 605                 hr = IWICBitmapFrameDecode_GetPixelFormat(frame, &pixel_format);
 606                 if (SUCCEEDED(hr)) {
 607                     info->Format = wic_guid_to_d3dformat(&pixel_format);
 608                     if (info->Format == D3DFMT_UNKNOWN) {
 609                         WARN("Unsupported pixel format %s\n", debugstr_guid(&pixel_format));
 610                         hr = D3DXERR_INVALIDDATA;
 611                     }
 612                 }
 613             }
 614
 615             if (frame)
 616                  IWICBitmapFrameDecode_Release(frame);
 617
 618             info->Depth = 1;
 619             info->MipLevels = 1;
 620             info->ResourceType = D3DRTYPE_TEXTURE;
 621         }
 622     }
 623
 624     if (decoder)
 625         IWICBitmapDecoder_Release(decoder);
 626
 627     if (SUCCEEDED(initresult))
 628         CoUninitialize();
 629
 630     if (FAILED(hr)) {
 631         TRACE("Invalid or unsupported image file\n");
 632         return D3DXERR_INVALIDDATA;
 633     }
 634
 635     return D3D_OK;
 636 }
 637
 638 /************************************************************
 639  * D3DXGetImageInfoFromFile
 640  *
 641  * RETURNS
 642  *   Success: D3D_OK, if we successfully load a valid image file or
 643  *                    if we successfully load a file which is no valid image and info is NULL
 644  *   Failure: D3DXERR_INVALIDDATA, if we fail to load file or
 645  *                                 if file is not a valid image file and info is not NULL
 646  *            D3DERR_INVALIDCALL, if file is NULL
 647  *
 648  */
 649 HRESULT WINAPI D3DXGetImageInfoFromFileA(LPCSTR file, D3DXIMAGE_INFO *info)
 650 {
 651     LPWSTR widename;
 652     HRESULT hr;
 653     int strlength;
 654
 655     TRACE("(%s, %p): relay\n", debugstr_a(file), info);
 656
 657     if( !file ) return D3DERR_INVALIDCALL;
 658
 659     strlength = MultiByteToWideChar(CP_ACP, 0, file, -1, NULL, 0);
 660     widename = HeapAlloc(GetProcessHeap(), 0, strlength * sizeof(*widename));
 661     MultiByteToWideChar(CP_ACP, 0, file, -1, widename, strlength);
 662
 663     hr = D3DXGetImageInfoFromFileW(widename, info);
 664     HeapFree(GetProcessHeap(), 0, widename);
 665
 666     return hr;
 667 }
 668
 669 HRESULT WINAPI D3DXGetImageInfoFromFileW(LPCWSTR file, D3DXIMAGE_INFO *info)
 670 {
 671     HRESULT hr;
 672     DWORD size;
 673     LPVOID buffer;
 674
 675     TRACE("(%s, %p): relay\n", debugstr_w(file), info);
 676
 677     if( !file ) return D3DERR_INVALIDCALL;
 678
 679     hr = map_view_of_file(file, &buffer, &size);
 680     if(FAILED(hr)) return D3DXERR_INVALIDDATA;
 681
 682     hr = D3DXGetImageInfoFromFileInMemory(buffer, size, info);
 683     UnmapViewOfFile(buffer);
 684
 685     return hr;
 686 }
 687
 688 /************************************************************
 689  * D3DXGetImageInfoFromResource
 690  *
 691  * RETURNS
 692  *   Success: D3D_OK, if resource is a valid image file
 693  *   Failure: D3DXERR_INVALIDDATA, if resource is no valid image file or NULL or
 694  *                                 if we fail to load resource
 695  *
 696  */
 697 HRESULT WINAPI D3DXGetImageInfoFromResourceA(HMODULE module, LPCSTR resource, D3DXIMAGE_INFO *info)
 698 {
 699     HRSRC resinfo;
 700
 701     TRACE("(%p, %s, %p)\n", module, debugstr_a(resource), info);
 702
 703     resinfo = FindResourceA(module, resource, (LPCSTR)RT_RCDATA);
 704     if(resinfo) {
 705         LPVOID buffer;
 706         HRESULT hr;
 707         DWORD size;
 708
 709         hr = load_resource_into_memory(module, resinfo, &buffer, &size);
 710         if(FAILED(hr)) return D3DXERR_INVALIDDATA;
 711         return D3DXGetImageInfoFromFileInMemory(buffer, size, info);
 712     }
 713
 714     resinfo = FindResourceA(module, resource, (LPCSTR)RT_BITMAP);
 715     if(resinfo) {
 716         FIXME("Implement loading bitmaps from resource type RT_BITMAP\n");
 717         return E_NOTIMPL;
 718     }
 719     return D3DXERR_INVALIDDATA;
 720 }
 721
 722 HRESULT WINAPI D3DXGetImageInfoFromResourceW(HMODULE module, LPCWSTR resource, D3DXIMAGE_INFO *info)
 723 {
 724     HRSRC resinfo;
 725
 726     TRACE("(%p, %s, %p)\n", module, debugstr_w(resource), info);
 727
 728     resinfo = FindResourceW(module, resource, (LPCWSTR)RT_RCDATA);
 729     if(resinfo) {
 730         LPVOID buffer;
 731         HRESULT hr;
 732         DWORD size;
 733
 734         hr = load_resource_into_memory(module, resinfo, &buffer, &size);
 735         if(FAILED(hr)) return D3DXERR_INVALIDDATA;
 736         return D3DXGetImageInfoFromFileInMemory(buffer, size, info);
 737     }
 738
 739     resinfo = FindResourceW(module, resource, (LPCWSTR)RT_BITMAP);
 740     if(resinfo) {
 741         FIXME("Implement loading bitmaps from resource type RT_BITMAP\n");
 742         return E_NOTIMPL;
 743     }
 744     return D3DXERR_INVALIDDATA;
 745 }
 746
 747 /************************************************************
 748  * D3DXLoadSurfaceFromFileInMemory
 749  *
 750  * Loads data from a given buffer into a surface and fills a given
 751  * D3DXIMAGE_INFO structure with info about the source data.
 752  *
 753  * PARAMS
 754  *   pDestSurface [I] pointer to the surface
 755  *   pDestPalette [I] palette to use
 756  *   pDestRect    [I] to be filled area of the surface
 757  *   pSrcData     [I] pointer to the source data
 758  *   SrcDataSize  [I] size of the source data in bytes
 759  *   pSrcRect     [I] area of the source data to load
 760  *   dwFilter     [I] filter to apply on stretching
 761  *   Colorkey     [I] colorkey
 762  *   pSrcInfo     [O] pointer to a D3DXIMAGE_INFO structure
 763  *
 764  * RETURNS
 765  *   Success: D3D_OK
 766  *   Failure: D3DERR_INVALIDCALL, if pDestSurface or pSrcData or SrcDataSize are NULL
 767  *            D3DXERR_INVALIDDATA, if pSrcData is no valid image file
 768  *
 769  */
 770 HRESULT WINAPI D3DXLoadSurfaceFromFileInMemory(IDirect3DSurface9 *pDestSurface,
 771         const PALETTEENTRY *pDestPalette, const RECT *pDestRect, const void *pSrcData, UINT SrcDataSize,
 772         const RECT *pSrcRect, DWORD dwFilter, D3DCOLOR Colorkey, D3DXIMAGE_INFO *pSrcInfo)
 773 {
 774     D3DXIMAGE_INFO imginfo;
 775     HRESULT hr;
 776
 777     IWICImagingFactory *factory;
 778     IWICBitmapDecoder *decoder;
 779     IWICBitmapFrameDecode *bitmapframe;
 780     IWICStream *stream;
 781
 782     const PixelFormatDesc *formatdesc;
 783     WICRect wicrect;
 784     RECT rect;
 785
 786     TRACE("dst_surface %p, dst_palette %p, dst_rect %s, src_data %p, src_data_size %u, "
 787             "src_rect %s, filter %#x, color_key 0x%08x, src_info %p.\n",
 788             pDestSurface, pDestPalette, wine_dbgstr_rect(pDestRect), pSrcData, SrcDataSize,
 789             wine_dbgstr_rect(pSrcRect), dwFilter, Colorkey, pSrcInfo);
 790
 791     if (!pDestSurface || !pSrcData || !SrcDataSize)
 792         return D3DERR_INVALIDCALL;
 793
 794     hr = D3DXGetImageInfoFromFileInMemory(pSrcData, SrcDataSize, &imginfo);
 795
 796     if (FAILED(hr))
 797         return hr;
 798
 799     if (pSrcRect)
 800     {
 801         wicrect.X = pSrcRect->left;
 802         wicrect.Y = pSrcRect->top;
 803         wicrect.Width = pSrcRect->right - pSrcRect->left;
 804         wicrect.Height = pSrcRect->bottom - pSrcRect->top;
 805     }
 806     else
 807     {
 808         wicrect.X = 0;
 809         wicrect.Y = 0;
 810         wicrect.Width = imginfo.Width;
 811         wicrect.Height = imginfo.Height;
 812     }
 813
 814     SetRect(&rect, 0, 0, wicrect.Width, wicrect.Height);
 815
 816     if (imginfo.ImageFileFormat == D3DXIFF_DDS)
 817     {
 818         hr = load_surface_from_dds(pDestSurface, pDestPalette, pDestRect, pSrcData, &rect,
 819             dwFilter, Colorkey, &imginfo);
 820         if (SUCCEEDED(hr) && pSrcInfo)
 821             *pSrcInfo = imginfo;
 822         return hr;
 823     }
 824
 825     CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
 826
 827     if (FAILED(CoCreateInstance(&CLSID_WICImagingFactory, NULL, CLSCTX_INPROC_SERVER, &IID_IWICImagingFactory, (void**)&factory)))
 828         goto cleanup_err;
 829
 830     if (FAILED(IWICImagingFactory_CreateStream(factory, &stream)))
 831     {
 832         IWICImagingFactory_Release(factory);
 833         goto cleanup_err;
 834     }
 835
 836     IWICStream_InitializeFromMemory(stream, (BYTE*)pSrcData, SrcDataSize);
 837
 838     hr = IWICImagingFactory_CreateDecoderFromStream(factory, (IStream*)stream, NULL, 0, &decoder);
 839
 840     IStream_Release(stream);
 841     IWICImagingFactory_Release(factory);
 842
 843     if (FAILED(hr))
 844         goto cleanup_err;
 845
 846     hr = IWICBitmapDecoder_GetFrame(decoder, 0, &bitmapframe);
 847
 848     if (FAILED(hr))
 849         goto cleanup_bmp;
 850
 851     formatdesc = get_format_info(imginfo.Format);
 852
 853     if (formatdesc->format == D3DFMT_UNKNOWN)
 854     {
 855         FIXME("Unsupported pixel format\n");
 856         hr = D3DXERR_INVALIDDATA;
 857     }
 858     else
 859     {
 860         BYTE *buffer;
 861         DWORD pitch;
 862
 863         pitch = formatdesc->bytes_per_pixel * wicrect.Width;
 864         buffer = HeapAlloc(GetProcessHeap(), 0, pitch * wicrect.Height);
 865
 866         hr = IWICBitmapFrameDecode_CopyPixels(bitmapframe, &wicrect, pitch,
 867                                               pitch * wicrect.Height, buffer);
 868
 869         if (SUCCEEDED(hr))
 870         {
 871             hr = D3DXLoadSurfaceFromMemory(pDestSurface, pDestPalette, pDestRect,
 872                                            buffer, imginfo.Format, pitch,
 873                                            NULL, &rect, dwFilter, Colorkey);
 874         }
 875
 876         HeapFree(GetProcessHeap(), 0, buffer);
 877     }
 878
 879     IWICBitmapFrameDecode_Release(bitmapframe);
 880
 881 cleanup_bmp:
 882     IWICBitmapDecoder_Release(decoder);
 883
 884 cleanup_err:
 885     CoUninitialize();
 886
 887     if (FAILED(hr))
 888         return D3DXERR_INVALIDDATA;
 889
 890     if (pSrcInfo)
 891         *pSrcInfo = imginfo;
 892
 893     return D3D_OK;
 894 }
 895
 896 HRESULT WINAPI D3DXLoadSurfaceFromFileA(IDirect3DSurface9 *dst_surface,
 897         const PALETTEENTRY *dst_palette, const RECT *dst_rect, const char *src_file,
 898         const RECT *src_rect, DWORD filter, D3DCOLOR color_key, D3DXIMAGE_INFO *src_info)
 899 {
 900     LPWSTR pWidename;
 901     HRESULT hr;
 902     int strlength;
 903
 904     TRACE("dst_surface %p, dst_palette %p, dst_rect %s, src_file %s, "
 905             "src_rect %s, filter %#x, color_key 0x%08x, src_info %p.\n",
 906             dst_surface, dst_palette, wine_dbgstr_rect(dst_rect), debugstr_a(src_file),
 907             wine_dbgstr_rect(src_rect), filter, color_key, src_info);
 908
 909     if (!src_file || !dst_surface)
 910         return D3DERR_INVALIDCALL;
 911
 912     strlength = MultiByteToWideChar(CP_ACP, 0, src_file, -1, NULL, 0);
 913     pWidename = HeapAlloc(GetProcessHeap(), 0, strlength * sizeof(*pWidename));
 914     MultiByteToWideChar(CP_ACP, 0, src_file, -1, pWidename, strlength);
 915
 916     hr = D3DXLoadSurfaceFromFileW(dst_surface, dst_palette, dst_rect,
 917             pWidename, src_rect, filter, color_key, src_info);
 918     HeapFree(GetProcessHeap(), 0, pWidename);
 919
 920     return hr;
 921 }
 922
 923 HRESULT WINAPI D3DXLoadSurfaceFromFileW(IDirect3DSurface9 *dst_surface,
 924         const PALETTEENTRY *dst_palette, const RECT *dst_rect, const WCHAR *src_file,
 925         const RECT *src_rect, DWORD filter, D3DCOLOR color_key, D3DXIMAGE_INFO *src_info)
 926 {
 927     UINT data_size;
 928     void *data;
 929     HRESULT hr;
 930
 931     TRACE("dst_surface %p, dst_palette %p, dst_rect %s, src_file %s, "
 932             "src_rect %s, filter %#x, color_key 0x%08x, src_info %p.\n",
 933             dst_surface, dst_palette, wine_dbgstr_rect(dst_rect), debugstr_w(src_file),
 934             wine_dbgstr_rect(src_rect), filter, color_key, src_info);
 935
 936     if (!src_file || !dst_surface)
 937         return D3DERR_INVALIDCALL;
 938
 939     if (FAILED(map_view_of_file(src_file, &data, &data_size)))
 940         return D3DXERR_INVALIDDATA;
 941
 942     hr = D3DXLoadSurfaceFromFileInMemory(dst_surface, dst_palette, dst_rect,
 943             data, data_size, src_rect, filter, color_key, src_info);
 944     UnmapViewOfFile(data);
 945
 946     return hr;
 947 }
 948
 949 HRESULT WINAPI D3DXLoadSurfaceFromResourceA(IDirect3DSurface9 *dst_surface,
 950         const PALETTEENTRY *dst_palette, const RECT *dst_rect, HMODULE src_module, const char *resource,
 951         const RECT *src_rect, DWORD filter, D3DCOLOR color_key, D3DXIMAGE_INFO *src_info)
 952 {
 953     HRSRC hResInfo;
 954
 955     TRACE("dst_surface %p, dst_palette %p, dst_rect %s, src_module %p, resource %s, "
 956             "src_rect %s, filter %#x, color_key 0x%08x, src_info %p.\n",
 957             dst_surface, dst_palette, wine_dbgstr_rect(dst_rect), src_module, debugstr_a(resource),
 958             wine_dbgstr_rect(src_rect), filter, color_key, src_info);
 959
 960     if (!dst_surface)
 961         return D3DERR_INVALIDCALL;
 962
 963     if ((hResInfo = FindResourceA(src_module, resource, (const char *)RT_RCDATA)))
 964     {
 965         UINT data_size;
 966         void *data;
 967
 968         if (FAILED(load_resource_into_memory(src_module, hResInfo, &data, &data_size)))
 969             return D3DXERR_INVALIDDATA;
 970
 971         return D3DXLoadSurfaceFromFileInMemory(dst_surface, dst_palette, dst_rect,
 972                 data, data_size, src_rect, filter, color_key, src_info);
 973     }
 974
 975     if ((hResInfo = FindResourceA(src_module, resource, (const char *)RT_BITMAP)))
 976     {
 977         FIXME("Implement loading bitmaps from resource type RT_BITMAP.\n");
 978         return E_NOTIMPL;
 979     }
 980
 981     return D3DXERR_INVALIDDATA;
 982 }
 983
 984 HRESULT WINAPI D3DXLoadSurfaceFromResourceW(IDirect3DSurface9 *dst_surface,
 985         const PALETTEENTRY *dst_palette, const RECT *dst_rect, HMODULE src_module, const WCHAR *resource,
 986         const RECT *src_rect, DWORD filter, D3DCOLOR color_key, D3DXIMAGE_INFO *src_info)
 987 {
 988     HRSRC hResInfo;
 989
 990     TRACE("dst_surface %p, dst_palette %p, dst_rect %s, src_module %p, resource %s, "
 991             "src_rect %s, filter %#x, color_key 0x%08x, src_info %p.\n",
 992             dst_surface, dst_palette, wine_dbgstr_rect(dst_rect), src_module, debugstr_w(resource),
 993             wine_dbgstr_rect(src_rect), filter, color_key, src_info);
 994
 995     if (!dst_surface)
 996         return D3DERR_INVALIDCALL;
 997
 998     if ((hResInfo = FindResourceW(src_module, resource, (const WCHAR *)RT_RCDATA)))
 999     {
1000         UINT data_size;
1001         void *data;
1002
1003         if (FAILED(load_resource_into_memory(src_module, hResInfo, &data, &data_size)))
1004             return D3DXERR_INVALIDDATA;
1005
1006         return D3DXLoadSurfaceFromFileInMemory(dst_surface, dst_palette, dst_rect,
1007                 data, data_size, src_rect, filter, color_key, src_info);
1008     }
1009
1010     if ((hResInfo = FindResourceW(src_module, resource, (const WCHAR *)RT_BITMAP)))
1011     {
1012         FIXME("Implement loading bitmaps from resource type RT_BITMAP.\n");
1013         return E_NOTIMPL;
1014     }
1015
1016     return D3DXERR_INVALIDDATA;
1017 }
1018
1019
1020 /************************************************************
1021  * helper functions for D3DXLoadSurfaceFromMemory
1022  */
1023 struct argb_conversion_info
1024 {
1025     CONST PixelFormatDesc *srcformat;
1026     CONST PixelFormatDesc *destformat;
1027     DWORD srcshift[4], destshift[4];
1028     DWORD srcmask[4], destmask[4];
1029     BOOL process_channel[4];
1030     DWORD channelmask;
1031 };
1032
1033 static void init_argb_conversion_info(CONST PixelFormatDesc *srcformat, CONST PixelFormatDesc *destformat, struct argb_conversion_info *info)
1034 {
1035     UINT i;
1036     ZeroMemory(info->process_channel, 4 * sizeof(BOOL));
1037     info->channelmask = 0;
1038
1039     info->srcformat  =  srcformat;
1040     info->destformat = destformat;
1041
1042     for(i = 0;i < 4;i++) {
1043         /* srcshift is used to extract the _relevant_ components */
1044         info->srcshift[i]  =  srcformat->shift[i] + max( srcformat->bits[i] - destformat->bits[i], 0);
1045
1046         /* destshift is used to move the components to the correct position */
1047         info->destshift[i] = destformat->shift[i] + max(destformat->bits[i] -  srcformat->bits[i], 0);
1048
1049         info->srcmask[i]  = ((1 <<  srcformat->bits[i]) - 1) <<  srcformat->shift[i];
1050         info->destmask[i] = ((1 << destformat->bits[i]) - 1) << destformat->shift[i];
1051
1052         /* channelmask specifies bits which aren't used in the source format but in the destination one */
1053         if(destformat->bits[i]) {
1054             if(srcformat->bits[i]) info->process_channel[i] = TRUE;
1055             else info->channelmask |= info->destmask[i];
1056         }
1057     }
1058 }
1059
1060 static DWORD dword_from_bytes(CONST BYTE *src, UINT bytes_per_pixel)
1061 {
1062     DWORD ret = 0;
1063     static BOOL fixme_once;
1064
1065     if(bytes_per_pixel > sizeof(DWORD)) {
1066         if(!fixme_once++) FIXME("Unsupported image: %u bytes per pixel\n", bytes_per_pixel);
1067         bytes_per_pixel = sizeof(DWORD);
1068     }
1069
1070     memcpy(&ret, src, bytes_per_pixel);
1071     return ret;
1072 }
1073
1074 static void dword_to_bytes(BYTE *dst, DWORD dword, UINT bytes_per_pixel)
1075 {
1076     static BOOL fixme_once;
1077
1078     if(bytes_per_pixel > sizeof(DWORD)) {
1079         if(!fixme_once++) FIXME("Unsupported image: %u bytes per pixel\n", bytes_per_pixel);
1080         ZeroMemory(dst, bytes_per_pixel);
1081         bytes_per_pixel = sizeof(DWORD);
1082     }
1083
1084     memcpy(dst, &dword, bytes_per_pixel);
1085 }
1086
1087 /************************************************************
1088  * get_relevant_argb_components
1089  *
1090  * Extracts the relevant components from the source color and
1091  * drops the less significant bits if they aren't used by the destination format.
1092  */
1093 static void get_relevant_argb_components(CONST struct argb_conversion_info *info, CONST DWORD col, DWORD *out)
1094 {
1095     UINT i = 0;
1096     for(;i < 4;i++)
1097         if(info->process_channel[i])
1098             out[i] = (col & info->srcmask[i]) >> info->srcshift[i];
1099 }
1100
1101 /************************************************************
1102  * make_argb_color
1103  *
1104  * Recombines the output of get_relevant_argb_components and converts
1105  * it to the destination format.
1106  */
1107 static DWORD make_argb_color(CONST struct argb_conversion_info *info, CONST DWORD *in)
1108 {
1109     UINT i;
1110     DWORD val = 0;
1111
1112     for(i = 0;i < 4;i++) {
1113         if(info->process_channel[i]) {
1114             /* necessary to make sure that e.g. an X4R4G4B4 white maps to an R8G8B8 white instead of 0xf0f0f0 */
1115             signed int shift;
1116             for(shift = info->destshift[i]; shift > info->destformat->shift[i]; shift -= info->srcformat->bits[i]) val |= in[i] << shift;
1117             val |= (in[i] >> (info->destformat->shift[i] - shift)) << info->destformat->shift[i];
1118         }
1119     }
1120     val |= info->channelmask;   /* new channels are set to their maximal value */
1121     return val;
1122 }
1123
1124 static void format_to_vec4(const PixelFormatDesc *format, const DWORD *src, struct vec4 *dst)
1125 {
1126     DWORD mask;
1127
1128     if (format->bits[1])
1129     {
1130         mask = (1 << format->bits[1]) - 1;
1131         dst->x = (float)((*src >> format->shift[1]) & mask) / mask;
1132     }
1133     else
1134         dst->x = 1.0f;
1135
1136     if (format->bits[2])
1137     {
1138         mask = (1 << format->bits[2]) - 1;
1139         dst->y = (float)((*src >> format->shift[2]) & mask) / mask;
1140     }
1141     else
1142         dst->y = 1.0f;
1143
1144     if (format->bits[3])
1145     {
1146         mask = (1 << format->bits[3]) - 1;
1147         dst->z = (float)((*src >> format->shift[3]) & mask) / mask;
1148     }
1149     else
1150         dst->z = 1.0f;
1151
1152     if (format->bits[0])
1153     {
1154         mask = (1 << format->bits[0]) - 1;
1155         dst->w = (float)((*src >> format->shift[0]) & mask) / mask;
1156     }
1157     else
1158         dst->w = 1.0f;
1159 }
1160
1161 static void format_from_vec4(const PixelFormatDesc *format, const struct vec4 *src, DWORD *dst)
1162 {
1163     *dst = 0;
1164
1165     if (format->bits[1])
1166         *dst |= (DWORD)(src->x * ((1 << format->bits[1]) - 1) + 0.5f) << format->shift[1];
1167     if (format->bits[2])
1168         *dst |= (DWORD)(src->y * ((1 << format->bits[2]) - 1) + 0.5f) << format->shift[2];
1169     if (format->bits[3])
1170         *dst |= (DWORD)(src->z * ((1 << format->bits[3]) - 1) + 0.5f) << format->shift[3];
1171     if (format->bits[0])
1172         *dst |= (DWORD)(src->w * ((1 << format->bits[0]) - 1) + 0.5f) << format->shift[0];
1173 }
1174
1175 /************************************************************
1176  * copy_simple_data
1177  *
1178  * Copies the source buffer to the destination buffer, performing
1179  * any necessary format conversion and color keying.
1180  * Pixels outsize the source rect are blacked out.
1181  * Works only for ARGB formats with 1 - 4 bytes per pixel.
1182  */
1183 static void copy_simple_data(const BYTE *src, UINT srcpitch, SIZE src_size, const PixelFormatDesc *srcformat,
1184         BYTE *dest, UINT destpitch, SIZE dst_size, const PixelFormatDesc *destformat, D3DCOLOR colorkey)
1185 {
1186     struct argb_conversion_info conv_info, ck_conv_info;
1187     const PixelFormatDesc *ck_format = NULL;
1188     DWORD channels[4], pixel;
1189     UINT minwidth, minheight;
1190     UINT x, y;
1191
1192     ZeroMemory(channels, sizeof(channels));
1193     init_argb_conversion_info(srcformat, destformat, &conv_info);
1194
1195     minwidth = (src_size.cx < dst_size.cx) ? src_size.cx : dst_size.cx;
1196     minheight = (src_size.cy < dst_size.cy) ? src_size.cy : dst_size.cy;
1197
1198     if (colorkey)
1199     {
1200         /* Color keys are always represented in D3DFMT_A8R8G8B8 format. */
1201         ck_format = get_format_info(D3DFMT_A8R8G8B8);
1202         init_argb_conversion_info(srcformat, ck_format, &ck_conv_info);
1203     }
1204
1205     for(y = 0;y < minheight;y++) {
1206         const BYTE *srcptr = src + y * srcpitch;
1207         BYTE *destptr = dest + y * destpitch;
1208         DWORD val;
1209
1210         for(x = 0;x < minwidth;x++) {
1211             /* extract source color components */
1212             pixel = dword_from_bytes(srcptr, srcformat->bytes_per_pixel);
1213
1214             if (!srcformat->to_rgba && !destformat->from_rgba)
1215             {
1216                 get_relevant_argb_components(&conv_info, pixel, channels);
1217                 val = make_argb_color(&conv_info, channels);
1218
1219                 if (colorkey)
1220                 {
1221                     get_relevant_argb_components(&ck_conv_info, pixel, channels);
1222                     pixel = make_argb_color(&ck_conv_info, channels);
1223                     if (pixel == colorkey)
1224                         val &= ~conv_info.destmask[0];
1225                 }
1226             }
1227             else
1228             {
1229                 struct vec4 color, tmp;
1230
1231                 format_to_vec4(srcformat, &pixel, &color);
1232                 if (srcformat->to_rgba)
1233                     srcformat->to_rgba(&color, &tmp);
1234                 else
1235                     tmp = color;
1236
1237                 if (ck_format)
1238                 {
1239                     format_from_vec4(ck_format, &tmp, &pixel);
1240                     if (pixel == colorkey)
1241                         tmp.w = 0.0f;
1242                 }
1243
1244                 if (destformat->from_rgba)
1245                     destformat->from_rgba(&tmp, &color);
1246                 else
1247                     color = tmp;
1248
1249                 format_from_vec4(destformat, &color, &val);
1250             }
1251
1252             dword_to_bytes(destptr, val, destformat->bytes_per_pixel);
1253             srcptr  +=  srcformat->bytes_per_pixel;
1254             destptr += destformat->bytes_per_pixel;
1255         }
1256
1257         if (src_size.cx < dst_size.cx) /* black out remaining pixels */
1258             memset(destptr, 0, destformat->bytes_per_pixel * (dst_size.cx - src_size.cx));
1259     }
1260     if (src_size.cy < dst_size.cy) /* black out remaining pixels */
1261         memset(dest + src_size.cy * destpitch, 0, destpitch * (dst_size.cy - src_size.cy));
1262 }
1263
1264 /************************************************************
1265  * point_filter_simple_data
1266  *
1267  * Copies the source buffer to the destination buffer, performing
1268  * any necessary format conversion, color keying and stretching
1269  * using a point filter.
1270  * Works only for ARGB formats with 1 - 4 bytes per pixel.
1271  */
1272 static void point_filter_simple_data(const BYTE *src, UINT srcpitch, SIZE src_size, const PixelFormatDesc *srcformat,
1273         BYTE *dest, UINT destpitch, SIZE dst_size, const PixelFormatDesc *destformat, D3DCOLOR colorkey)
1274 {
1275     struct argb_conversion_info conv_info, ck_conv_info;
1276     const PixelFormatDesc *ck_format = NULL;
1277     DWORD channels[4], pixel;
1278
1279     UINT x, y;
1280
1281     ZeroMemory(channels, sizeof(channels));
1282     init_argb_conversion_info(srcformat, destformat, &conv_info);
1283
1284     if (colorkey)
1285     {
1286         /* Color keys are always represented in D3DFMT_A8R8G8B8 format. */
1287         ck_format = get_format_info(D3DFMT_A8R8G8B8);
1288         init_argb_conversion_info(srcformat, ck_format, &ck_conv_info);
1289     }
1290
1291     for (y = 0; y < dst_size.cy; ++y)
1292     {
1293         BYTE *destptr = dest + y * destpitch;
1294         const BYTE *bufptr = src + srcpitch * (y * src_size.cy / dst_size.cy);
1295
1296         for (x = 0; x < dst_size.cx; ++x)
1297         {
1298             const BYTE *srcptr = bufptr + (x * src_size.cx / dst_size.cx) * srcformat->bytes_per_pixel;
1299             DWORD val;
1300
1301             /* extract source color components */
1302             pixel = dword_from_bytes(srcptr, srcformat->bytes_per_pixel);
1303
1304             if (!srcformat->to_rgba && !destformat->from_rgba)
1305             {
1306                 get_relevant_argb_components(&conv_info, pixel, channels);
1307                 val = make_argb_color(&conv_info, channels);
1308
1309                 if (colorkey)
1310                 {
1311                     get_relevant_argb_components(&ck_conv_info, pixel, channels);
1312                     pixel = make_argb_color(&ck_conv_info, channels);
1313                     if (pixel == colorkey)
1314                         val &= ~conv_info.destmask[0];
1315                 }
1316             }
1317             else
1318             {
1319                 struct vec4 color, tmp;
1320
1321                 format_to_vec4(srcformat, &pixel, &color);
1322                 if (srcformat->to_rgba)
1323                     srcformat->to_rgba(&color, &tmp);
1324                 else
1325                     tmp = color;
1326
1327                 if (ck_format)
1328                 {
1329                     format_from_vec4(ck_format, &tmp, &pixel);
1330                     if (pixel == colorkey)
1331                         tmp.w = 0.0f;
1332                 }
1333
1334                 if (destformat->from_rgba)
1335                     destformat->from_rgba(&tmp, &color);
1336                 else
1337                     color = tmp;
1338
1339                 format_from_vec4(destformat, &color, &val);
1340             }
1341
1342             dword_to_bytes(destptr, val, destformat->bytes_per_pixel);
1343             destptr += destformat->bytes_per_pixel;
1344         }
1345     }
1346 }
1347
1348 /************************************************************
1349  * D3DXLoadSurfaceFromMemory
1350  *
1351  * Loads data from a given memory chunk into a surface,
1352  * applying any of the specified filters.
1353  *
1354  * PARAMS
1355  *   pDestSurface [I] pointer to the surface
1356  *   pDestPalette [I] palette to use
1357  *   pDestRect    [I] to be filled area of the surface
1358  *   pSrcMemory   [I] pointer to the source data
1359  *   SrcFormat    [I] format of the source pixel data
1360  *   SrcPitch     [I] number of bytes in a row
1361  *   pSrcPalette  [I] palette used in the source image
1362  *   pSrcRect     [I] area of the source data to load
1363  *   dwFilter     [I] filter to apply on stretching
1364  *   Colorkey     [I] colorkey
1365  *
1366  * RETURNS
1367  *   Success: D3D_OK, if we successfully load the pixel data into our surface or
1368  *                    if pSrcMemory is NULL but the other parameters are valid
1369  *   Failure: D3DERR_INVALIDCALL, if pDestSurface, SrcPitch or pSrcRect are NULL or
1370  *                                if SrcFormat is an invalid format (other than D3DFMT_UNKNOWN) or
1371  *                                if DestRect is invalid
1372  *            D3DXERR_INVALIDDATA, if we fail to lock pDestSurface
1373  *            E_FAIL, if SrcFormat is D3DFMT_UNKNOWN or the dimensions of pSrcRect are invalid
1374  *
1375  * NOTES
1376  *   pSrcRect specifies the dimensions of the source data;
1377  *   negative values for pSrcRect are allowed as we're only looking at the width and height anyway.
1378  *
1379  */
1380 HRESULT WINAPI D3DXLoadSurfaceFromMemory(IDirect3DSurface9 *dst_surface,
1381         const PALETTEENTRY *dst_palette, const RECT *dst_rect, const void *src_memory,
1382         D3DFORMAT src_format, UINT src_pitch, const PALETTEENTRY *src_palette, const RECT *src_rect,
1383         DWORD filter, D3DCOLOR color_key)
1384 {
1385     CONST PixelFormatDesc *srcformatdesc, *destformatdesc;
1386     D3DSURFACE_DESC surfdesc;
1387     D3DLOCKED_RECT lockrect;
1388     SIZE src_size, dst_size;
1389     HRESULT hr;
1390
1391     TRACE("(%p, %p, %s, %p, %#x, %u, %p, %s %#x, 0x%08x)\n",
1392             dst_surface, dst_palette, wine_dbgstr_rect(dst_rect), src_memory, src_format,
1393             src_pitch, src_palette, wine_dbgstr_rect(src_rect), filter, color_key);
1394
1395     if (!dst_surface || !src_memory || !src_rect)
1396         return D3DERR_INVALIDCALL;
1397     if (src_format == D3DFMT_UNKNOWN
1398             || src_rect->left >= src_rect->right
1399             || src_rect->top >= src_rect->bottom)
1400         return E_FAIL;
1401
1402     if (filter == D3DX_DEFAULT)
1403         filter = D3DX_FILTER_TRIANGLE | D3DX_FILTER_DITHER;
1404
1405     IDirect3DSurface9_GetDesc(dst_surface, &surfdesc);
1406
1407     src_size.cx = src_rect->right - src_rect->left;
1408     src_size.cy = src_rect->bottom - src_rect->top;
1409     if (!dst_rect)
1410     {
1411         dst_size.cx = surfdesc.Width;
1412         dst_size.cy = surfdesc.Height;
1413     }
1414     else
1415     {
1416         if (dst_rect->left > dst_rect->right || dst_rect->right > surfdesc.Width)
1417             return D3DERR_INVALIDCALL;
1418         if (dst_rect->top > dst_rect->bottom || dst_rect->bottom > surfdesc.Height)
1419             return D3DERR_INVALIDCALL;
1420         if (dst_rect->left < 0 || dst_rect->top < 0)
1421             return D3DERR_INVALIDCALL;
1422         dst_size.cx = dst_rect->right - dst_rect->left;
1423         dst_size.cy = dst_rect->bottom - dst_rect->top;
1424         if (!dst_size.cx || !dst_size.cy)
1425             return D3D_OK;
1426     }
1427
1428     srcformatdesc = get_format_info(src_format);
1429     if (srcformatdesc->type == FORMAT_UNKNOWN)
1430         return E_NOTIMPL;
1431
1432     destformatdesc = get_format_info(surfdesc.Format);
1433     if (destformatdesc->type == FORMAT_UNKNOWN)
1434         return E_NOTIMPL;
1435
1436     if (src_format == surfdesc.Format
1437             && dst_size.cx == src_size.cx
1438             && dst_size.cy == src_size.cy) /* Simple copy. */
1439     {
1440         UINT row_block_count = ((src_size.cx + srcformatdesc->block_width - 1) / srcformatdesc->block_width);
1441         UINT row_count = (src_size.cy + srcformatdesc->block_height - 1) / srcformatdesc->block_height;
1442         const BYTE *src_addr;
1443         BYTE *dst_addr;
1444         UINT row;
1445
1446         if (src_rect->left & (srcformatdesc->block_width - 1)
1447                 || src_rect->top & (srcformatdesc->block_height - 1)
1448                 || (src_rect->right & (srcformatdesc->block_width - 1)
1449                     && src_size.cx != surfdesc.Width)
1450                 || (src_rect->bottom & (srcformatdesc->block_height - 1)
1451                     && src_size.cy != surfdesc.Height))
1452         {
1453             WARN("Source rect %s is misaligned.\n", wine_dbgstr_rect(src_rect));
1454             return D3DXERR_INVALIDDATA;
1455         }
1456
1457         if (FAILED(hr = IDirect3DSurface9_LockRect(dst_surface, &lockrect, dst_rect, 0)))
1458             return D3DXERR_INVALIDDATA;
1459
1460         src_addr = src_memory;
1461         src_addr += (src_rect->top / srcformatdesc->block_height) * src_pitch;
1462         src_addr += (src_rect->left / srcformatdesc->block_width) * srcformatdesc->block_byte_count;
1463         dst_addr = lockrect.pBits;
1464
1465         for (row = 0; row < row_count; ++row)
1466         {
1467             memcpy(dst_addr, src_addr, row_block_count * srcformatdesc->block_byte_count);
1468             src_addr += src_pitch;
1469             dst_addr += lockrect.Pitch;
1470         }
1471
1472         IDirect3DSurface9_UnlockRect(dst_surface);
1473     }
1474     else /* Stretching or format conversion. */
1475     {
1476         if (srcformatdesc->bytes_per_pixel > 4)
1477             return E_NOTIMPL;
1478         if (destformatdesc->bytes_per_pixel > 4)
1479             return E_NOTIMPL;
1480         if (srcformatdesc->block_height != 1 || srcformatdesc->block_width != 1)
1481             return E_NOTIMPL;
1482         if (destformatdesc->block_height != 1 || destformatdesc->block_width != 1)
1483             return E_NOTIMPL;
1484
1485         if (FAILED(hr = IDirect3DSurface9_LockRect(dst_surface, &lockrect, dst_rect, 0)))
1486             return D3DXERR_INVALIDDATA;
1487
1488         if ((filter & 0xf) == D3DX_FILTER_NONE)
1489         {
1490             copy_simple_data(src_memory, src_pitch, src_size, srcformatdesc,
1491                     lockrect.pBits, lockrect.Pitch, dst_size, destformatdesc, color_key);
1492         }
1493         else /* if ((filter & 0xf) == D3DX_FILTER_POINT) */
1494         {
1495             if ((filter & 0xf) != D3DX_FILTER_POINT)
1496                 FIXME("Unhandled filter %#x.\n", filter);
1497
1498             /* Always apply a point filter until D3DX_FILTER_LINEAR,
1499              * D3DX_FILTER_TRIANGLE and D3DX_FILTER_BOX are implemented. */
1500             point_filter_simple_data(src_memory, src_pitch, src_size, srcformatdesc,
1501                     lockrect.pBits, lockrect.Pitch, dst_size, destformatdesc, color_key);
1502         }
1503
1504         IDirect3DSurface9_UnlockRect(dst_surface);
1505     }
1506
1507     return D3D_OK;
1508 }
1509
1510 /************************************************************
1511  * D3DXLoadSurfaceFromSurface
1512  *
1513  * Copies the contents from one surface to another, performing any required
1514  * format conversion, resizing or filtering.
1515  *
1516  * PARAMS
1517  *   pDestSurface [I] pointer to the destination surface
1518  *   pDestPalette [I] palette to use
1519  *   pDestRect    [I] to be filled area of the surface
1520  *   pSrcSurface  [I] pointer to the source surface
1521  *   pSrcPalette  [I] palette used for the source surface
1522  *   pSrcRect     [I] area of the source data to load
1523  *   dwFilter     [I] filter to apply on resizing
1524  *   Colorkey     [I] any ARGB value or 0 to disable color-keying
1525  *
1526  * RETURNS
1527  *   Success: D3D_OK
1528  *   Failure: D3DERR_INVALIDCALL, if pDestSurface or pSrcSurface are NULL
1529  *            D3DXERR_INVALIDDATA, if one of the surfaces is not lockable
1530  *
1531  */
1532 HRESULT WINAPI D3DXLoadSurfaceFromSurface(IDirect3DSurface9 *dst_surface,
1533         const PALETTEENTRY *dst_palette, const RECT *dst_rect, IDirect3DSurface9 *src_surface,
1534         const PALETTEENTRY *src_palette, const RECT *src_rect, DWORD filter, D3DCOLOR color_key)
1535 {
1536     RECT rect;
1537     D3DLOCKED_RECT lock;
1538     D3DSURFACE_DESC SrcDesc;
1539     HRESULT hr;
1540
1541     TRACE("dst_surface %p, dst_palette %p, dst_rect %s, src_surface %p, "
1542             "src_palette %p, src_rect %s, filter %#x, color_key 0x%08x.\n",
1543             dst_surface, dst_palette, wine_dbgstr_rect(dst_rect), src_surface,
1544             src_palette, wine_dbgstr_rect(src_rect), filter, color_key);
1545
1546     if (!dst_surface || !src_surface)
1547         return D3DERR_INVALIDCALL;
1548
1549     IDirect3DSurface9_GetDesc(src_surface, &SrcDesc);
1550
1551     if (!src_rect)
1552         SetRect(&rect, 0, 0, SrcDesc.Width, SrcDesc.Height);
1553     else
1554         rect = *src_rect;
1555
1556     if (FAILED(IDirect3DSurface9_LockRect(src_surface, &lock, NULL, D3DLOCK_READONLY)))
1557         return D3DXERR_INVALIDDATA;
1558
1559     hr = D3DXLoadSurfaceFromMemory(dst_surface, dst_palette, dst_rect,
1560             lock.pBits, SrcDesc.Format, lock.Pitch, src_palette, &rect, filter, color_key);
1561
1562     IDirect3DSurface9_UnlockRect(src_surface);
1563
1564     return hr;
1565 }
1566
1567
1568 HRESULT WINAPI D3DXSaveSurfaceToFileA(const char *dst_filename, D3DXIMAGE_FILEFORMAT file_format,
1569         IDirect3DSurface9 *src_surface, const PALETTEENTRY *src_palette, const RECT *src_rect)
1570 {
1571     int len;
1572     WCHAR *filename;
1573     HRESULT hr;
1574     ID3DXBuffer *buffer;
1575
1576     TRACE("(%s, %#x, %p, %p, %s): relay\n",
1577             wine_dbgstr_a(dst_filename), file_format, src_surface, src_palette, wine_dbgstr_rect(src_rect));
1578
1579     if (!dst_filename) return D3DERR_INVALIDCALL;
1580
1581     len = MultiByteToWideChar(CP_ACP, 0, dst_filename, -1, NULL, 0);
1582     filename = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
1583     if (!filename) return E_OUTOFMEMORY;
1584     MultiByteToWideChar(CP_ACP, 0, dst_filename, -1, filename, len);
1585
1586     hr = D3DXSaveSurfaceToFileInMemory(&buffer, file_format, src_surface, src_palette, src_rect);
1587     if (SUCCEEDED(hr))
1588     {
1589         hr = write_buffer_to_file(filename, buffer);
1590         ID3DXBuffer_Release(buffer);
1591     }
1592
1593     HeapFree(GetProcessHeap(), 0, filename);
1594     return hr;
1595 }
1596
1597 HRESULT WINAPI D3DXSaveSurfaceToFileW(const WCHAR *dst_filename, D3DXIMAGE_FILEFORMAT file_format,
1598         IDirect3DSurface9 *src_surface, const PALETTEENTRY *src_palette, const RECT *src_rect)
1599 {
1600     HRESULT hr;
1601     ID3DXBuffer *buffer;
1602
1603     TRACE("(%s, %#x, %p, %p, %s): relay\n",
1604         wine_dbgstr_w(dst_filename), file_format, src_surface, src_palette, wine_dbgstr_rect(src_rect));
1605
1606     if (!dst_filename) return D3DERR_INVALIDCALL;
1607
1608     hr = D3DXSaveSurfaceToFileInMemory(&buffer, file_format, src_surface, src_palette, src_rect);
1609     if (SUCCEEDED(hr))
1610     {
1611         hr = write_buffer_to_file(dst_filename, buffer);
1612         ID3DXBuffer_Release(buffer);
1613     }
1614
1615     return hr;
1616 }
1617
1618 HRESULT WINAPI D3DXSaveSurfaceToFileInMemory(ID3DXBuffer **dst_buffer, D3DXIMAGE_FILEFORMAT file_format,
1619         IDirect3DSurface9 *src_surface, const PALETTEENTRY *src_palette, const RECT *src_rect)
1620 {
1621     IWICBitmapEncoder *encoder = NULL;
1622     IWICBitmapFrameEncode *frame = NULL;
1623     IPropertyBag2 *encoder_options = NULL;
1624     IStream *stream = NULL;
1625     HRESULT hr;
1626     HRESULT initresult;
1627     const CLSID *encoder_clsid;
1628     const GUID *pixel_format_guid;
1629     WICPixelFormatGUID wic_pixel_format;
1630     D3DFORMAT d3d_pixel_format;
1631     D3DSURFACE_DESC src_surface_desc;
1632     D3DLOCKED_RECT locked_rect;
1633     int width, height;
1634     STATSTG stream_stats;
1635     HGLOBAL stream_hglobal;
1636     ID3DXBuffer *buffer;
1637     DWORD size;
1638
1639     TRACE("(%p, %#x, %p, %p, %s)\n",
1640         dst_buffer, file_format, src_surface, src_palette, wine_dbgstr_rect(src_rect));
1641
1642     if (!dst_buffer || !src_surface) return D3DERR_INVALIDCALL;
1643
1644     if (src_palette)
1645     {
1646         FIXME("Saving surfaces with palettized pixel formats not implemented yet\n");
1647         return D3DERR_INVALIDCALL;
1648     }
1649
1650     switch (file_format)
1651     {
1652         case D3DXIFF_BMP:
1653             encoder_clsid = &CLSID_WICBmpEncoder;
1654             break;
1655         case D3DXIFF_PNG:
1656             encoder_clsid = &CLSID_WICPngEncoder;
1657             break;
1658         case D3DXIFF_JPG:
1659             encoder_clsid = &CLSID_WICJpegEncoder;
1660             break;
1661         case D3DXIFF_DDS:
1662         case D3DXIFF_DIB:
1663         case D3DXIFF_HDR:
1664         case D3DXIFF_PFM:
1665         case D3DXIFF_TGA:
1666         case D3DXIFF_PPM:
1667             FIXME("File format %#x is not supported yet\n", file_format);
1668             return E_NOTIMPL;
1669         default:
1670             return D3DERR_INVALIDCALL;
1671     }
1672
1673     IDirect3DSurface9_GetDesc(src_surface, &src_surface_desc);
1674     if (src_rect)
1675     {
1676         if (src_rect->left == src_rect->right || src_rect->top == src_rect->bottom)
1677         {
1678             WARN("Invalid rectangle with 0 area\n");
1679             return D3DXCreateBuffer(64, dst_buffer);
1680         }
1681         if (src_rect->left < 0 || src_rect->top < 0)
1682             return D3DERR_INVALIDCALL;
1683         if (src_rect->left > src_rect->right || src_rect->top > src_rect->bottom)
1684             return D3DERR_INVALIDCALL;
1685         if (src_rect->right > src_surface_desc.Width || src_rect->bottom > src_surface_desc.Height)
1686             return D3DERR_INVALIDCALL;
1687
1688         width = src_rect->right - src_rect->left;
1689         height = src_rect->bottom - src_rect->top;
1690     }
1691     else
1692     {
1693         width = src_surface_desc.Width;
1694         height = src_surface_desc.Height;
1695     }
1696
1697     initresult = CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
1698
1699     hr = CoCreateInstance(encoder_clsid, NULL, CLSCTX_INPROC_SERVER,
1700         &IID_IWICBitmapEncoder, (void **)&encoder);
1701     if (FAILED(hr)) goto cleanup_err;
1702
1703     hr = CreateStreamOnHGlobal(NULL, TRUE, &stream);
1704     if (FAILED(hr)) goto cleanup_err;
1705
1706     hr = IWICBitmapEncoder_Initialize(encoder, stream, WICBitmapEncoderNoCache);
1707     if (FAILED(hr)) goto cleanup_err;
1708
1709     hr = IWICBitmapEncoder_CreateNewFrame(encoder, &frame, &encoder_options);
1710     if (FAILED(hr)) goto cleanup_err;
1711
1712     hr = IWICBitmapFrameEncode_Initialize(frame, encoder_options);
1713     if (FAILED(hr)) goto cleanup_err;
1714
1715     hr = IWICBitmapFrameEncode_SetSize(frame, width, height);
1716     if (FAILED(hr)) goto cleanup_err;
1717
1718     pixel_format_guid = d3dformat_to_wic_guid(src_surface_desc.Format);
1719     if (!pixel_format_guid)
1720     {
1721         FIXME("Pixel format %#x is not supported yet\n", src_surface_desc.Format);
1722         hr = E_NOTIMPL;
1723         goto cleanup;
1724     }
1725
1726     memcpy(&wic_pixel_format, pixel_format_guid, sizeof(GUID));
1727     hr = IWICBitmapFrameEncode_SetPixelFormat(frame, &wic_pixel_format);
1728     d3d_pixel_format = wic_guid_to_d3dformat(&wic_pixel_format);
1729     if (SUCCEEDED(hr) && d3d_pixel_format != D3DFMT_UNKNOWN)
1730     {
1731         TRACE("Using pixel format %s %#x\n", debugstr_guid(&wic_pixel_format), d3d_pixel_format);
1732
1733         if (src_surface_desc.Format == d3d_pixel_format) /* Simple copy */
1734         {
1735             hr = IDirect3DSurface9_LockRect(src_surface, &locked_rect, src_rect, D3DLOCK_READONLY);
1736             if (SUCCEEDED(hr))
1737             {
1738                 IWICBitmapFrameEncode_WritePixels(frame, height,
1739                     locked_rect.Pitch, height * locked_rect.Pitch, locked_rect.pBits);
1740                 IDirect3DSurface9_UnlockRect(src_surface);
1741             }
1742         }
1743         else /* Pixel format conversion */
1744         {
1745             const PixelFormatDesc *src_format_desc, *dst_format_desc;
1746             SIZE size;
1747             DWORD dst_pitch;
1748             void *dst_data;
1749
1750             src_format_desc = get_format_info(src_surface_desc.Format);
1751             dst_format_desc = get_format_info(d3d_pixel_format);
1752             if (src_format_desc->format == D3DFMT_UNKNOWN || dst_format_desc->format == D3DFMT_UNKNOWN)
1753             {
1754                 FIXME("Unsupported pixel format conversion %#x -> %#x\n",
1755                     src_surface_desc.Format, d3d_pixel_format);
1756                 hr = E_NOTIMPL;
1757                 goto cleanup;
1758             }
1759
1760             if (src_format_desc->bytes_per_pixel > 4
1761                 || dst_format_desc->bytes_per_pixel > 4
1762                 || src_format_desc->block_height != 1 || src_format_desc->block_width != 1
1763                 || dst_format_desc->block_height != 1 || dst_format_desc->block_width != 1)
1764             {
1765                 hr = E_NOTIMPL;
1766                 goto cleanup;
1767             }
1768
1769             size.cx = width;
1770             size.cy = height;
1771             dst_pitch = width * dst_format_desc->bytes_per_pixel;
1772             dst_data = HeapAlloc(GetProcessHeap(), 0, dst_pitch * height);
1773             if (!dst_data)
1774             {
1775                 hr = E_OUTOFMEMORY;
1776                 goto cleanup;
1777             }
1778
1779             hr = IDirect3DSurface9_LockRect(src_surface, &locked_rect, src_rect, D3DLOCK_READONLY);
1780             if (SUCCEEDED(hr))
1781             {
1782                 copy_simple_data(locked_rect.pBits, locked_rect.Pitch, size, src_format_desc,
1783                     dst_data, dst_pitch, size, dst_format_desc, 0);
1784                 IDirect3DSurface9_UnlockRect(src_surface);
1785             }
1786
1787             IWICBitmapFrameEncode_WritePixels(frame, height, dst_pitch, dst_pitch * height, dst_data);
1788             HeapFree(GetProcessHeap(), 0, dst_data);
1789         }
1790
1791         hr = IWICBitmapFrameEncode_Commit(frame);
1792         if (SUCCEEDED(hr)) hr = IWICBitmapEncoder_Commit(encoder);
1793     }
1794     else WARN("Unsupported pixel format %#x\n", src_surface_desc.Format);
1795
1796     /* copy data from stream to ID3DXBuffer */
1797     hr = IStream_Stat(stream, &stream_stats, STATFLAG_NONAME);
1798     if (FAILED(hr)) goto cleanup_err;
1799
1800     if (stream_stats.cbSize.u.HighPart != 0)
1801     {
1802         hr = D3DXERR_INVALIDDATA;
1803         goto cleanup;
1804     }
1805     size = stream_stats.cbSize.u.LowPart;
1806
1807     hr = D3DXCreateBuffer(size, &buffer);
1808     if (FAILED(hr)) goto cleanup;
1809
1810     hr = GetHGlobalFromStream(stream, &stream_hglobal);
1811     if (SUCCEEDED(hr))
1812     {
1813         void *buffer_pointer = ID3DXBuffer_GetBufferPointer(buffer);
1814         void *stream_data = GlobalLock(stream_hglobal);
1815         memcpy(buffer_pointer, stream_data, size);
1816         GlobalUnlock(stream_hglobal);
1817         *dst_buffer = buffer;
1818     }
1819     else ID3DXBuffer_Release(buffer);
1820
1821 cleanup_err:
1822     if (FAILED(hr) && hr != E_OUTOFMEMORY)
1823         hr = D3DERR_INVALIDCALL;
1824
1825 cleanup:
1826     if (stream) IStream_Release(stream);
1827
1828     if (frame) IWICBitmapFrameEncode_Release(frame);
1829     if (encoder_options) IPropertyBag2_Release(encoder_options);
1830
1831     if (encoder) IWICBitmapEncoder_Release(encoder);
1832
1833     if (SUCCEEDED(initresult)) CoUninitialize();
1834
1835     return hr;
1836 }