d3dx9_36: Add DIB file support to D3DXSaveSurfaceToFileInMemory.
[wine] / 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     unsigned 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     unsigned 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     unsigned 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 const struct {
170     DWORD bpp;
171     DWORD rmask;
172     DWORD gmask;
173     DWORD bmask;
174     DWORD amask;
175     D3DFORMAT format;
176 } rgb_pixel_formats[] = {
177     { 8, 0xe0, 0x1c, 0x03, 0, D3DFMT_R3G3B2 },
178     { 16, 0xf800, 0x07e0, 0x001f, 0x0000, D3DFMT_R5G6B5 },
179     { 16, 0x7c00, 0x03e0, 0x001f, 0x8000, D3DFMT_A1R5G5B5 },
180     { 16, 0x7c00, 0x03e0, 0x001f, 0x0000, D3DFMT_X1R5G5B5 },
181     { 16, 0x0f00, 0x00f0, 0x000f, 0xf000, D3DFMT_A4R4G4B4 },
182     { 16, 0x0f00, 0x00f0, 0x000f, 0x0000, D3DFMT_X4R4G4B4 },
183     { 16, 0x00e0, 0x001c, 0x0003, 0xff00, D3DFMT_A8R3G3B2 },
184     { 24, 0xff0000, 0x00ff00, 0x0000ff, 0x000000, D3DFMT_R8G8B8 },
185     { 32, 0x00ff0000, 0x0000ff00, 0x000000ff, 0xff000000, D3DFMT_A8R8G8B8 },
186     { 32, 0x00ff0000, 0x0000ff00, 0x000000ff, 0x00000000, D3DFMT_X8R8G8B8 },
187     { 32, 0x3ff00000, 0x000ffc00, 0x000003ff, 0xc0000000, D3DFMT_A2B10G10R10 },
188     { 32, 0x000003ff, 0x000ffc00, 0x3ff00000, 0xc0000000, D3DFMT_A2R10G10B10 },
189     { 32, 0x0000ffff, 0xffff0000, 0x00000000, 0x00000000, D3DFMT_G16R16 },
190     { 32, 0x000000ff, 0x0000ff00, 0x00ff0000, 0xff000000, D3DFMT_A8B8G8R8 },
191     { 32, 0x000000ff, 0x0000ff00, 0x00ff0000, 0x00000000, D3DFMT_X8B8G8R8 },
192 };
193
194 static D3DFORMAT dds_rgb_to_d3dformat(const struct dds_pixel_format *pixel_format)
195 {
196     unsigned int i;
197
198     for (i = 0; i < sizeof(rgb_pixel_formats) / sizeof(rgb_pixel_formats[0]); i++)
199     {
200         if (rgb_pixel_formats[i].bpp == pixel_format->bpp
201             && rgb_pixel_formats[i].rmask == pixel_format->rmask
202             && rgb_pixel_formats[i].gmask == pixel_format->gmask
203             && rgb_pixel_formats[i].bmask == pixel_format->bmask)
204         {
205             if ((pixel_format->flags & DDS_PF_ALPHA) && rgb_pixel_formats[i].amask == pixel_format->amask)
206                 return rgb_pixel_formats[i].format;
207             if (rgb_pixel_formats[i].amask == 0)
208                 return rgb_pixel_formats[i].format;
209         }
210     }
211
212     WARN("Unknown RGB pixel format (%#x, %#x, %#x, %#x)\n",
213         pixel_format->rmask, pixel_format->gmask, pixel_format->bmask, pixel_format->amask);
214     return D3DFMT_UNKNOWN;
215 }
216
217 static D3DFORMAT dds_luminance_to_d3dformat(const struct dds_pixel_format *pixel_format)
218 {
219     if (pixel_format->bpp == 8)
220     {
221         if (pixel_format->rmask == 0xff)
222             return D3DFMT_L8;
223         if ((pixel_format->flags & DDS_PF_ALPHA) && pixel_format->rmask == 0x0f && pixel_format->amask == 0xf0)
224             return D3DFMT_A4L4;
225     }
226     if (pixel_format->bpp == 16)
227     {
228         if (pixel_format->rmask == 0xffff)
229             return D3DFMT_L16;
230         if ((pixel_format->flags & DDS_PF_ALPHA) && pixel_format->rmask == 0x00ff && pixel_format->amask == 0xff00)
231             return D3DFMT_A8L8;
232     }
233
234     WARN("Unknown luminance pixel format (bpp %#x, l %#x, a %#x)\n",
235         pixel_format->bpp, pixel_format->rmask, pixel_format->amask);
236     return D3DFMT_UNKNOWN;
237 }
238
239 static D3DFORMAT dds_alpha_to_d3dformat(const struct dds_pixel_format *pixel_format)
240 {
241     if (pixel_format->bpp == 8 && pixel_format->amask == 0xff)
242         return D3DFMT_A8;
243
244     WARN("Unknown Alpha pixel format (%#x, %#x)\n", pixel_format->bpp, pixel_format->rmask);
245     return D3DFMT_UNKNOWN;
246 }
247
248 static D3DFORMAT dds_bump_to_d3dformat(const struct dds_pixel_format *pixel_format)
249 {
250     if (pixel_format->bpp == 16 && pixel_format->rmask == 0x00ff && pixel_format->gmask == 0xff00)
251         return D3DFMT_V8U8;
252     if (pixel_format->bpp == 32 && pixel_format->rmask == 0x0000ffff && pixel_format->gmask == 0xffff0000)
253         return D3DFMT_V16U16;
254
255     WARN("Unknown bump pixel format (%#x, %#x, %#x, %#x, %#x)\n", pixel_format->bpp,
256         pixel_format->rmask, pixel_format->gmask, pixel_format->bmask, pixel_format->amask);
257     return D3DFMT_UNKNOWN;
258 }
259
260 static D3DFORMAT dds_pixel_format_to_d3dformat(const struct dds_pixel_format *pixel_format)
261 {
262     if (pixel_format->flags & DDS_PF_FOURCC)
263         return dds_fourcc_to_d3dformat(pixel_format->fourcc);
264     if (pixel_format->flags & DDS_PF_RGB)
265         return dds_rgb_to_d3dformat(pixel_format);
266     if (pixel_format->flags & DDS_PF_LUMINANCE)
267         return dds_luminance_to_d3dformat(pixel_format);
268     if (pixel_format->flags & DDS_PF_ALPHA_ONLY)
269         return dds_alpha_to_d3dformat(pixel_format);
270     if (pixel_format->flags & DDS_PF_BUMPDUDV)
271         return dds_bump_to_d3dformat(pixel_format);
272
273     WARN("Unknown pixel format (flags %#x, fourcc %#x, bpp %#x, r %#x, g %#x, b %#x, a %#x)\n",
274         pixel_format->flags, pixel_format->fourcc, pixel_format->bpp,
275         pixel_format->rmask, pixel_format->gmask, pixel_format->bmask, pixel_format->amask);
276     return D3DFMT_UNKNOWN;
277 }
278
279 static HRESULT d3dformat_to_dds_pixel_format(struct dds_pixel_format *pixel_format, D3DFORMAT d3dformat)
280 {
281     unsigned int i;
282
283     memset(pixel_format, 0, sizeof(*pixel_format));
284
285     pixel_format->size = sizeof(*pixel_format);
286
287     for (i = 0; i < sizeof(rgb_pixel_formats) / sizeof(rgb_pixel_formats[0]); i++)
288     {
289         if (rgb_pixel_formats[i].format == d3dformat)
290         {
291             pixel_format->flags |= DDS_PF_RGB;
292             pixel_format->bpp = rgb_pixel_formats[i].bpp;
293             pixel_format->rmask = rgb_pixel_formats[i].rmask;
294             pixel_format->gmask = rgb_pixel_formats[i].gmask;
295             pixel_format->bmask = rgb_pixel_formats[i].bmask;
296             pixel_format->amask = rgb_pixel_formats[i].amask;
297             if (pixel_format->amask) pixel_format->flags |= DDS_PF_ALPHA;
298             return D3D_OK;
299         }
300     }
301
302     WARN("Unknown pixel format %#x\n", d3dformat);
303     return E_NOTIMPL;
304 }
305
306 static HRESULT calculate_dds_surface_size(D3DFORMAT format, UINT width, UINT height,
307     UINT *pitch, UINT *size)
308 {
309     const struct pixel_format_desc *format_desc = get_format_info(format);
310     if (format_desc->type == FORMAT_UNKNOWN)
311         return E_NOTIMPL;
312
313     if (format_desc->block_width != 1 || format_desc->block_height != 1)
314     {
315         *pitch = format_desc->block_byte_count
316             * max(1, (width + format_desc->block_width - 1) / format_desc->block_width);
317         *size = *pitch
318             * max(1, (height + format_desc->block_height - 1) / format_desc->block_height);
319     }
320     else
321     {
322         *pitch = width * format_desc->bytes_per_pixel;
323         *size = *pitch * height;
324     }
325
326     return D3D_OK;
327 }
328
329 static UINT calculate_dds_file_size(D3DFORMAT format, UINT width, UINT height, UINT depth,
330     UINT miplevels, UINT faces)
331 {
332     UINT i, file_size = 0;
333
334     for (i = 0; i < miplevels; i++)
335     {
336         UINT pitch, size = 0;
337         calculate_dds_surface_size(format, width, height, &pitch, &size);
338         size *= depth;
339         file_size += size;
340         width = max(1, width / 2);
341         height = max(1, height / 2);
342         depth = max(1, depth / 2);
343     }
344
345     file_size *= faces;
346     file_size += sizeof(struct dds_header);
347     return file_size;
348 }
349
350 /************************************************************
351 * get_image_info_from_dds
352 *
353 * Fills a D3DXIMAGE_INFO structure with information
354 * about a DDS file stored in the memory.
355 *
356 * PARAMS
357 *   buffer  [I] pointer to DDS data
358 *   length  [I] size of DDS data
359 *   info    [O] pointer to D3DXIMAGE_INFO structure
360 *
361 * RETURNS
362 *   Success: D3D_OK
363 *   Failure: D3DXERR_INVALIDDATA
364 *
365 */
366 static HRESULT get_image_info_from_dds(const void *buffer, UINT length, D3DXIMAGE_INFO *info)
367 {
368     UINT faces = 1;
369     UINT expected_length;
370     const struct dds_header *header = buffer;
371
372     if (length < sizeof(*header) || !info)
373         return D3DXERR_INVALIDDATA;
374
375     if (header->pixel_format.size != sizeof(header->pixel_format))
376         return D3DXERR_INVALIDDATA;
377
378     info->Width = header->width;
379     info->Height = header->height;
380     info->Depth = 1;
381     info->MipLevels = (header->flags & DDS_MIPMAPCOUNT) ?  header->miplevels : 1;
382
383     info->Format = dds_pixel_format_to_d3dformat(&header->pixel_format);
384     if (info->Format == D3DFMT_UNKNOWN)
385         return D3DXERR_INVALIDDATA;
386
387     TRACE("Pixel format is %#x\n", info->Format);
388
389     if (header->caps2 & DDS_CAPS2_VOLUME)
390     {
391         info->Depth = header->depth;
392         info->ResourceType = D3DRTYPE_VOLUMETEXTURE;
393     }
394     else if (header->caps2 & DDS_CAPS2_CUBEMAP)
395     {
396         DWORD face;
397         faces = 0;
398         for (face = DDS_CAPS2_CUBEMAP_POSITIVEX; face <= DDS_CAPS2_CUBEMAP_NEGATIVEZ; face <<= 1)
399         {
400             if (header->caps2 & face)
401                 faces++;
402         }
403         info->ResourceType = D3DRTYPE_CUBETEXTURE;
404     }
405     else
406     {
407         info->ResourceType = D3DRTYPE_TEXTURE;
408     }
409
410     expected_length = calculate_dds_file_size(info->Format, info->Width, info->Height, info->Depth,
411         info->MipLevels, faces);
412     if (length < expected_length)
413     {
414         WARN("File is too short %u, expected at least %u bytes\n", length, expected_length);
415         return D3DXERR_INVALIDDATA;
416     }
417
418     info->ImageFileFormat = D3DXIFF_DDS;
419     return D3D_OK;
420 }
421
422 static HRESULT load_surface_from_dds(IDirect3DSurface9 *dst_surface, const PALETTEENTRY *dst_palette,
423     const RECT *dst_rect, const void *src_data, const RECT *src_rect, DWORD filter, D3DCOLOR color_key,
424     const D3DXIMAGE_INFO *src_info)
425 {
426     UINT size;
427     UINT src_pitch;
428     const struct dds_header *header = src_data;
429     const BYTE *pixels = (BYTE *)(header + 1);
430
431     if (src_info->ResourceType != D3DRTYPE_TEXTURE)
432         return D3DXERR_INVALIDDATA;
433
434     if (FAILED(calculate_dds_surface_size(src_info->Format, src_info->Width, src_info->Height, &src_pitch, &size)))
435         return E_NOTIMPL;
436
437     return D3DXLoadSurfaceFromMemory(dst_surface, dst_palette, dst_rect, pixels, src_info->Format,
438         src_pitch, NULL, src_rect, filter, color_key);
439 }
440
441 static HRESULT save_dds_surface_to_memory(ID3DXBuffer **dst_buffer, IDirect3DSurface9 *src_surface, const RECT *src_rect)
442 {
443     HRESULT hr;
444     UINT dst_pitch, surface_size, file_size;
445     D3DSURFACE_DESC src_desc;
446     D3DLOCKED_RECT locked_rect;
447     ID3DXBuffer *buffer;
448     struct dds_header *header;
449     BYTE *pixels;
450     struct volume volume;
451     const struct pixel_format_desc *pixel_format;
452
453     if (src_rect)
454     {
455         FIXME("Saving a part of a surface to a DDS file is not implemented yet\n");
456         return E_NOTIMPL;
457     }
458
459     hr = IDirect3DSurface9_GetDesc(src_surface, &src_desc);
460     if (FAILED(hr)) return hr;
461
462     pixel_format = get_format_info(src_desc.Format);
463     if (pixel_format->type == FORMAT_UNKNOWN) return E_NOTIMPL;
464
465     file_size = calculate_dds_file_size(src_desc.Format, src_desc.Width, src_desc.Height, 1, 1, 1);
466
467     hr = calculate_dds_surface_size(src_desc.Format, src_desc.Width, src_desc.Height, &dst_pitch, &surface_size);
468     if (FAILED(hr)) return hr;
469
470     hr = D3DXCreateBuffer(file_size, &buffer);
471     if (FAILED(hr)) return hr;
472
473     header = ID3DXBuffer_GetBufferPointer(buffer);
474     pixels = (BYTE *)(header + 1);
475
476     memset(header, 0, sizeof(*header));
477     header->signature = MAKEFOURCC('D','D','S',' ');
478     header->size = sizeof(*header);
479     header->flags = DDS_CAPS | DDS_HEIGHT | DDS_WIDTH | DDS_PITCH | DDS_PIXELFORMAT | DDS_MIPMAPCOUNT;
480     header->height = src_desc.Height;
481     header->width = src_desc.Width;
482     header->pitch_or_linear_size = dst_pitch;
483     header->depth = 1;
484     header->miplevels = 1;
485     header->caps = DDS_CAPS_TEXTURE;
486     hr = d3dformat_to_dds_pixel_format(&header->pixel_format, src_desc.Format);
487     if (FAILED(hr))
488     {
489         ID3DXBuffer_Release(buffer);
490         return hr;
491     }
492
493     hr = IDirect3DSurface9_LockRect(src_surface, &locked_rect, NULL, D3DLOCK_READONLY);
494     if (FAILED(hr))
495     {
496         ID3DXBuffer_Release(buffer);
497         return hr;
498     }
499
500     volume.width = src_desc.Width;
501     volume.height = src_desc.Height;
502     volume.depth = 1;
503     copy_pixels(locked_rect.pBits, locked_rect.Pitch, 0, pixels, dst_pitch, 0,
504         &volume, pixel_format);
505
506     IDirect3DSurface9_UnlockRect(src_surface);
507
508     *dst_buffer = buffer;
509     return D3D_OK;
510 }
511
512 HRESULT load_volume_from_dds(IDirect3DVolume9 *dst_volume, const PALETTEENTRY *dst_palette,
513     const D3DBOX *dst_box, const void *src_data, const D3DBOX *src_box, DWORD filter, D3DCOLOR color_key,
514     const D3DXIMAGE_INFO *src_info)
515 {
516     UINT row_pitch, slice_pitch;
517     const struct dds_header *header = src_data;
518     const BYTE *pixels = (BYTE *)(header + 1);
519
520     if (src_info->ResourceType != D3DRTYPE_VOLUMETEXTURE)
521         return D3DXERR_INVALIDDATA;
522
523     if (FAILED(calculate_dds_surface_size(src_info->Format, src_info->Width, src_info->Height, &row_pitch, &slice_pitch)))
524         return E_NOTIMPL;
525
526     return D3DXLoadVolumeFromMemory(dst_volume, dst_palette, dst_box, pixels, src_info->Format,
527         row_pitch, slice_pitch, NULL, src_box, filter, color_key);
528 }
529
530 HRESULT load_texture_from_dds(IDirect3DTexture9 *texture, const void *src_data, const PALETTEENTRY *palette,
531     DWORD filter, D3DCOLOR color_key, const D3DXIMAGE_INFO *src_info)
532
533 {
534     HRESULT hr;
535     RECT src_rect;
536     UINT src_pitch;
537     UINT mip_level;
538     UINT mip_levels;
539     UINT mip_level_size;
540     UINT width, height;
541     IDirect3DSurface9 *surface;
542     const struct dds_header *header = src_data;
543     const BYTE *pixels = (BYTE *)(header + 1);
544
545     /* Loading a cube texture as a simple texture is also supported (only first face texture is taken) */
546     if ((src_info->ResourceType != D3DRTYPE_TEXTURE) && (src_info->ResourceType != D3DRTYPE_CUBETEXTURE))
547         return D3DXERR_INVALIDDATA;
548
549     width = src_info->Width;
550     height = src_info->Height;
551     mip_levels = min(src_info->MipLevels, IDirect3DTexture9_GetLevelCount(texture));
552     for (mip_level = 0; mip_level < mip_levels; mip_level++)
553     {
554         hr = calculate_dds_surface_size(src_info->Format, width, height, &src_pitch, &mip_level_size);
555         if (FAILED(hr)) return hr;
556
557         SetRect(&src_rect, 0, 0, width, height);
558
559         IDirect3DTexture9_GetSurfaceLevel(texture, mip_level, &surface);
560         hr = D3DXLoadSurfaceFromMemory(surface, palette, NULL, pixels, src_info->Format, src_pitch,
561             NULL, &src_rect, filter, color_key);
562         IDirect3DSurface9_Release(surface);
563         if (FAILED(hr)) return hr;
564
565         pixels += mip_level_size;
566         width = max(1, width / 2);
567         height = max(1, height / 2);
568     }
569
570     return D3D_OK;
571 }
572
573 HRESULT load_cube_texture_from_dds(IDirect3DCubeTexture9 *cube_texture, const void *src_data,
574     const PALETTEENTRY *palette, DWORD filter, DWORD color_key, const D3DXIMAGE_INFO *src_info)
575 {
576     HRESULT hr;
577     int face;
578     UINT mip_level;
579     UINT size;
580     RECT src_rect;
581     UINT src_pitch;
582     UINT mip_levels;
583     UINT mip_level_size;
584     IDirect3DSurface9 *surface;
585     const struct dds_header *header = src_data;
586     const BYTE *pixels = (BYTE *)(header + 1);
587
588     if (src_info->ResourceType != D3DRTYPE_CUBETEXTURE)
589         return D3DXERR_INVALIDDATA;
590
591     if ((header->caps2 & DDS_CAPS2_CUBEMAP_ALL_FACES) != DDS_CAPS2_CUBEMAP_ALL_FACES)
592     {
593         WARN("Only full cubemaps are supported\n");
594         return D3DXERR_INVALIDDATA;
595     }
596
597     mip_levels = min(src_info->MipLevels, IDirect3DCubeTexture9_GetLevelCount(cube_texture));
598     for (face = D3DCUBEMAP_FACE_POSITIVE_X; face <= D3DCUBEMAP_FACE_NEGATIVE_Z; face++)
599     {
600         size = src_info->Width;
601         for (mip_level = 0; mip_level < src_info->MipLevels; mip_level++)
602         {
603             hr = calculate_dds_surface_size(src_info->Format, size, size, &src_pitch, &mip_level_size);
604             if (FAILED(hr)) return hr;
605
606             /* if texture has fewer mip levels than DDS file, skip excessive mip levels */
607             if (mip_level < mip_levels)
608             {
609                 SetRect(&src_rect, 0, 0, size, size);
610
611                 IDirect3DCubeTexture9_GetCubeMapSurface(cube_texture, face, mip_level, &surface);
612                 hr = D3DXLoadSurfaceFromMemory(surface, palette, NULL, pixels, src_info->Format, src_pitch,
613                     NULL, &src_rect, filter, color_key);
614                 IDirect3DSurface9_Release(surface);
615                 if (FAILED(hr)) return hr;
616             }
617
618             pixels += mip_level_size;
619             size = max(1, size / 2);
620         }
621     }
622
623     return D3D_OK;
624 }
625
626 HRESULT load_volume_texture_from_dds(IDirect3DVolumeTexture9 *volume_texture, const void *src_data,
627     const PALETTEENTRY *palette, DWORD filter, DWORD color_key, const D3DXIMAGE_INFO *src_info)
628 {
629     HRESULT hr;
630     UINT mip_level;
631     UINT mip_levels;
632     UINT src_slice_pitch;
633     UINT src_row_pitch;
634     D3DBOX src_box;
635     UINT width, height, depth;
636     IDirect3DVolume9 *volume;
637     const struct dds_header *header = src_data;
638     const BYTE *pixels = (BYTE *)(header + 1);
639
640     if (src_info->ResourceType != D3DRTYPE_VOLUMETEXTURE)
641         return D3DXERR_INVALIDDATA;
642
643     width = src_info->Width;
644     height = src_info->Height;
645     depth = src_info->Depth;
646     mip_levels = min(src_info->MipLevels, IDirect3DVolumeTexture9_GetLevelCount(volume_texture));
647
648     for (mip_level = 0; mip_level < mip_levels; mip_level++)
649     {
650         hr = calculate_dds_surface_size(src_info->Format, width, height, &src_row_pitch, &src_slice_pitch);
651         if (FAILED(hr)) return hr;
652
653         hr = IDirect3DVolumeTexture9_GetVolumeLevel(volume_texture, mip_level, &volume);
654         if (FAILED(hr)) return hr;
655
656         src_box.Left = 0;
657         src_box.Top = 0;
658         src_box.Right = width;
659         src_box.Bottom = height;
660         src_box.Front = 0;
661         src_box.Back = depth;
662
663         hr = D3DXLoadVolumeFromMemory(volume, palette, NULL, pixels, src_info->Format,
664             src_row_pitch, src_slice_pitch, NULL, &src_box, filter, color_key);
665
666         IDirect3DVolume9_Release(volume);
667         if (FAILED(hr)) return hr;
668
669         pixels += depth * src_slice_pitch;
670         width = max(1, width / 2);
671         height = max(1, height / 2);
672         depth = max(1, depth / 2);
673     }
674
675     return D3D_OK;
676 }
677
678 static BOOL convert_dib_to_bmp(void **data, UINT *size)
679 {
680     ULONG header_size;
681     ULONG count = 0;
682     ULONG offset;
683     BITMAPFILEHEADER *header;
684     BYTE *new_data;
685     UINT new_size;
686
687     if ((*size < 4) || (*size < (header_size = *(ULONG*)*data)))
688         return FALSE;
689
690     if ((header_size == sizeof(BITMAPINFOHEADER)) ||
691         (header_size == sizeof(BITMAPV4HEADER)) ||
692         (header_size == sizeof(BITMAPV5HEADER)) ||
693         (header_size == 64 /* sizeof(BITMAPCOREHEADER2) */))
694     {
695         /* All structures begin with the same memory layout as BITMAPINFOHEADER */
696         BITMAPINFOHEADER *info_header = (BITMAPINFOHEADER*)*data;
697         count = info_header->biClrUsed;
698
699         if (!count && info_header->biBitCount <= 8)
700             count = 1 << info_header->biBitCount;
701
702         offset = sizeof(BITMAPFILEHEADER) + header_size + sizeof(RGBQUAD) * count;
703
704         /* For BITMAPINFOHEADER with BI_BITFIELDS compression, there are 3 additional color masks after header */
705         if ((info_header->biSize == sizeof(BITMAPINFOHEADER)) && (info_header->biCompression == BI_BITFIELDS))
706             offset += 3 * sizeof(DWORD);
707     }
708     else if (header_size == sizeof(BITMAPCOREHEADER))
709     {
710         BITMAPCOREHEADER *core_header = (BITMAPCOREHEADER*)*data;
711
712         if (core_header->bcBitCount <= 8)
713             count = 1 << core_header->bcBitCount;
714
715         offset = sizeof(BITMAPFILEHEADER) + header_size + sizeof(RGBTRIPLE) * count;
716     }
717     else
718     {
719         return FALSE;
720     }
721
722     TRACE("Converting DIB file to BMP\n");
723
724     new_size = *size + sizeof(BITMAPFILEHEADER);
725     new_data = HeapAlloc(GetProcessHeap(), 0, new_size);
726     CopyMemory(new_data + sizeof(BITMAPFILEHEADER), *data, *size);
727
728     /* Add BMP header */
729     header = (BITMAPFILEHEADER*)new_data;
730     header->bfType = 0x4d42; /* BM */
731     header->bfSize = new_size;
732     header->bfReserved1 = 0;
733     header->bfReserved2 = 0;
734     header->bfOffBits = offset;
735
736     /* Update input data */
737     *data = new_data;
738     *size = new_size;
739
740     return TRUE;
741 }
742
743 /************************************************************
744  * D3DXGetImageInfoFromFileInMemory
745  *
746  * Fills a D3DXIMAGE_INFO structure with info about an image
747  *
748  * PARAMS
749  *   data     [I] pointer to the image file data
750  *   datasize [I] size of the passed data
751  *   info     [O] pointer to the destination structure
752  *
753  * RETURNS
754  *   Success: D3D_OK, if info is not NULL and data and datasize make up a valid image file or
755  *                    if info is NULL and data and datasize are not NULL
756  *   Failure: D3DXERR_INVALIDDATA, if data is no valid image file and datasize and info are not NULL
757  *            D3DERR_INVALIDCALL, if data is NULL or
758  *                                if datasize is 0
759  *
760  * NOTES
761  *   datasize may be bigger than the actual file size
762  *
763  */
764 HRESULT WINAPI D3DXGetImageInfoFromFileInMemory(const void *data, UINT datasize, D3DXIMAGE_INFO *info)
765 {
766     IWICImagingFactory *factory;
767     IWICBitmapDecoder *decoder = NULL;
768     IWICStream *stream;
769     HRESULT hr;
770     HRESULT initresult;
771     BOOL dib;
772
773     TRACE("(%p, %d, %p)\n", data, datasize, info);
774
775     if (!data || !datasize)
776         return D3DERR_INVALIDCALL;
777
778     if (!info)
779         return D3D_OK;
780
781     if ((datasize >= 4) && !strncmp(data, "DDS ", 4)) {
782         TRACE("File type is DDS\n");
783         return get_image_info_from_dds(data, datasize, info);
784     }
785
786     /* In case of DIB file, convert it to BMP */
787     dib = convert_dib_to_bmp((void**)&data, &datasize);
788
789     initresult = CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
790
791     hr = CoCreateInstance(&CLSID_WICImagingFactory, NULL, CLSCTX_INPROC_SERVER, &IID_IWICImagingFactory, (void**)&factory);
792
793     if (SUCCEEDED(hr)) {
794         IWICImagingFactory_CreateStream(factory, &stream);
795         IWICStream_InitializeFromMemory(stream, (BYTE*)data, datasize);
796         hr = IWICImagingFactory_CreateDecoderFromStream(factory, (IStream*)stream, NULL, 0, &decoder);
797         IWICStream_Release(stream);
798         IWICImagingFactory_Release(factory);
799     }
800
801     if (FAILED(hr)) {
802         if ((datasize >= 2) && (!strncmp(data, "P3", 2) || !strncmp(data, "P6", 2)))
803             FIXME("File type PPM is not supported yet\n");
804         else if ((datasize >= 10) && !strncmp(data, "#?RADIANCE", 10))
805             FIXME("File type HDR is not supported yet\n");
806         else if ((datasize >= 2) && (!strncmp(data, "PF", 2) || !strncmp(data, "Pf", 2)))
807             FIXME("File type PFM is not supported yet\n");
808     }
809
810     if (SUCCEEDED(hr)) {
811         GUID container_format;
812         UINT frame_count;
813
814         hr = IWICBitmapDecoder_GetContainerFormat(decoder, &container_format);
815         if (SUCCEEDED(hr)) {
816             if (IsEqualGUID(&container_format, &GUID_ContainerFormatBmp)) {
817                 if (dib) {
818                     TRACE("File type is DIB\n");
819                     info->ImageFileFormat = D3DXIFF_DIB;
820                 } else {
821                     TRACE("File type is BMP\n");
822                     info->ImageFileFormat = D3DXIFF_BMP;
823                 }
824             } else if (IsEqualGUID(&container_format, &GUID_ContainerFormatPng)) {
825                 TRACE("File type is PNG\n");
826                 info->ImageFileFormat = D3DXIFF_PNG;
827             } else if(IsEqualGUID(&container_format, &GUID_ContainerFormatJpeg)) {
828                 TRACE("File type is JPG\n");
829                 info->ImageFileFormat = D3DXIFF_JPG;
830             } else if(IsEqualGUID(&container_format, &GUID_WineContainerFormatTga)) {
831                 TRACE("File type is TGA\n");
832                 info->ImageFileFormat = D3DXIFF_TGA;
833             } else {
834                 WARN("Unsupported image file format %s\n", debugstr_guid(&container_format));
835                 hr = D3DXERR_INVALIDDATA;
836             }
837         }
838
839         if (SUCCEEDED(hr))
840             hr = IWICBitmapDecoder_GetFrameCount(decoder, &frame_count);
841         if (SUCCEEDED(hr) && !frame_count)
842             hr = D3DXERR_INVALIDDATA;
843
844         if (SUCCEEDED(hr)) {
845             IWICBitmapFrameDecode *frame = NULL;
846
847             hr = IWICBitmapDecoder_GetFrame(decoder, 0, &frame);
848
849             if (SUCCEEDED(hr))
850                 hr = IWICBitmapFrameDecode_GetSize(frame, &info->Width, &info->Height);
851
852             if (SUCCEEDED(hr)) {
853                 WICPixelFormatGUID pixel_format;
854
855                 hr = IWICBitmapFrameDecode_GetPixelFormat(frame, &pixel_format);
856                 if (SUCCEEDED(hr)) {
857                     info->Format = wic_guid_to_d3dformat(&pixel_format);
858                     if (info->Format == D3DFMT_UNKNOWN) {
859                         WARN("Unsupported pixel format %s\n", debugstr_guid(&pixel_format));
860                         hr = D3DXERR_INVALIDDATA;
861                     }
862                 }
863             }
864
865             if (frame)
866                  IWICBitmapFrameDecode_Release(frame);
867
868             info->Depth = 1;
869             info->MipLevels = 1;
870             info->ResourceType = D3DRTYPE_TEXTURE;
871         }
872     }
873
874     if (decoder)
875         IWICBitmapDecoder_Release(decoder);
876
877     if (SUCCEEDED(initresult))
878         CoUninitialize();
879
880     if (dib)
881         HeapFree(GetProcessHeap(), 0, (void*)data);
882
883     if (FAILED(hr)) {
884         TRACE("Invalid or unsupported image file\n");
885         return D3DXERR_INVALIDDATA;
886     }
887
888     return D3D_OK;
889 }
890
891 /************************************************************
892  * D3DXGetImageInfoFromFile
893  *
894  * RETURNS
895  *   Success: D3D_OK, if we successfully load a valid image file or
896  *                    if we successfully load a file which is no valid image and info is NULL
897  *   Failure: D3DXERR_INVALIDDATA, if we fail to load file or
898  *                                 if file is not a valid image file and info is not NULL
899  *            D3DERR_INVALIDCALL, if file is NULL
900  *
901  */
902 HRESULT WINAPI D3DXGetImageInfoFromFileA(LPCSTR file, D3DXIMAGE_INFO *info)
903 {
904     LPWSTR widename;
905     HRESULT hr;
906     int strlength;
907
908     TRACE("(%s, %p): relay\n", debugstr_a(file), info);
909
910     if( !file ) return D3DERR_INVALIDCALL;
911
912     strlength = MultiByteToWideChar(CP_ACP, 0, file, -1, NULL, 0);
913     widename = HeapAlloc(GetProcessHeap(), 0, strlength * sizeof(*widename));
914     MultiByteToWideChar(CP_ACP, 0, file, -1, widename, strlength);
915
916     hr = D3DXGetImageInfoFromFileW(widename, info);
917     HeapFree(GetProcessHeap(), 0, widename);
918
919     return hr;
920 }
921
922 HRESULT WINAPI D3DXGetImageInfoFromFileW(LPCWSTR file, D3DXIMAGE_INFO *info)
923 {
924     HRESULT hr;
925     DWORD size;
926     LPVOID buffer;
927
928     TRACE("(%s, %p): relay\n", debugstr_w(file), info);
929
930     if( !file ) return D3DERR_INVALIDCALL;
931
932     hr = map_view_of_file(file, &buffer, &size);
933     if(FAILED(hr)) return D3DXERR_INVALIDDATA;
934
935     hr = D3DXGetImageInfoFromFileInMemory(buffer, size, info);
936     UnmapViewOfFile(buffer);
937
938     return hr;
939 }
940
941 /************************************************************
942  * D3DXGetImageInfoFromResource
943  *
944  * RETURNS
945  *   Success: D3D_OK, if resource is a valid image file
946  *   Failure: D3DXERR_INVALIDDATA, if resource is no valid image file or NULL or
947  *                                 if we fail to load resource
948  *
949  */
950 HRESULT WINAPI D3DXGetImageInfoFromResourceA(HMODULE module, LPCSTR resource, D3DXIMAGE_INFO *info)
951 {
952     HRSRC resinfo;
953
954     TRACE("(%p, %s, %p)\n", module, debugstr_a(resource), info);
955
956     resinfo = FindResourceA(module, resource, (const char *)RT_RCDATA);
957     if (!resinfo) /* Try loading the resource as bitmap data (which is in DIB format D3DXIFF_DIB) */
958         resinfo = FindResourceA(module, resource, (const char *)RT_BITMAP);
959
960     if (resinfo)
961     {
962         LPVOID buffer;
963         HRESULT hr;
964         DWORD size;
965
966         hr = load_resource_into_memory(module, resinfo, &buffer, &size);
967         if(FAILED(hr)) return D3DXERR_INVALIDDATA;
968         return D3DXGetImageInfoFromFileInMemory(buffer, size, info);
969     }
970
971     return D3DXERR_INVALIDDATA;
972 }
973
974 HRESULT WINAPI D3DXGetImageInfoFromResourceW(HMODULE module, LPCWSTR resource, D3DXIMAGE_INFO *info)
975 {
976     HRSRC resinfo;
977
978     TRACE("(%p, %s, %p)\n", module, debugstr_w(resource), info);
979
980     resinfo = FindResourceW(module, resource, (const WCHAR *)RT_RCDATA);
981     if (!resinfo) /* Try loading the resource as bitmap data (which is in DIB format D3DXIFF_DIB) */
982         resinfo = FindResourceW(module, resource, (const WCHAR *)RT_BITMAP);
983
984     if (resinfo)
985     {
986         LPVOID buffer;
987         HRESULT hr;
988         DWORD size;
989
990         hr = load_resource_into_memory(module, resinfo, &buffer, &size);
991         if(FAILED(hr)) return D3DXERR_INVALIDDATA;
992         return D3DXGetImageInfoFromFileInMemory(buffer, size, info);
993     }
994
995     return D3DXERR_INVALIDDATA;
996 }
997
998 /************************************************************
999  * D3DXLoadSurfaceFromFileInMemory
1000  *
1001  * Loads data from a given buffer into a surface and fills a given
1002  * D3DXIMAGE_INFO structure with info about the source data.
1003  *
1004  * PARAMS
1005  *   pDestSurface [I] pointer to the surface
1006  *   pDestPalette [I] palette to use
1007  *   pDestRect    [I] to be filled area of the surface
1008  *   pSrcData     [I] pointer to the source data
1009  *   SrcDataSize  [I] size of the source data in bytes
1010  *   pSrcRect     [I] area of the source data to load
1011  *   dwFilter     [I] filter to apply on stretching
1012  *   Colorkey     [I] colorkey
1013  *   pSrcInfo     [O] pointer to a D3DXIMAGE_INFO structure
1014  *
1015  * RETURNS
1016  *   Success: D3D_OK
1017  *   Failure: D3DERR_INVALIDCALL, if pDestSurface or pSrcData or SrcDataSize are NULL
1018  *            D3DXERR_INVALIDDATA, if pSrcData is no valid image file
1019  *
1020  */
1021 HRESULT WINAPI D3DXLoadSurfaceFromFileInMemory(IDirect3DSurface9 *pDestSurface,
1022         const PALETTEENTRY *pDestPalette, const RECT *pDestRect, const void *pSrcData, UINT SrcDataSize,
1023         const RECT *pSrcRect, DWORD dwFilter, D3DCOLOR Colorkey, D3DXIMAGE_INFO *pSrcInfo)
1024 {
1025     D3DXIMAGE_INFO imginfo;
1026     HRESULT hr;
1027
1028     IWICImagingFactory *factory;
1029     IWICBitmapDecoder *decoder;
1030     IWICBitmapFrameDecode *bitmapframe;
1031     IWICStream *stream;
1032
1033     const struct pixel_format_desc *formatdesc;
1034     WICRect wicrect;
1035     RECT rect;
1036
1037     TRACE("dst_surface %p, dst_palette %p, dst_rect %s, src_data %p, src_data_size %u, "
1038             "src_rect %s, filter %#x, color_key 0x%08x, src_info %p.\n",
1039             pDestSurface, pDestPalette, wine_dbgstr_rect(pDestRect), pSrcData, SrcDataSize,
1040             wine_dbgstr_rect(pSrcRect), dwFilter, Colorkey, pSrcInfo);
1041
1042     if (!pDestSurface || !pSrcData || !SrcDataSize)
1043         return D3DERR_INVALIDCALL;
1044
1045     hr = D3DXGetImageInfoFromFileInMemory(pSrcData, SrcDataSize, &imginfo);
1046
1047     if (FAILED(hr))
1048         return hr;
1049
1050     if (pSrcRect)
1051     {
1052         wicrect.X = pSrcRect->left;
1053         wicrect.Y = pSrcRect->top;
1054         wicrect.Width = pSrcRect->right - pSrcRect->left;
1055         wicrect.Height = pSrcRect->bottom - pSrcRect->top;
1056     }
1057     else
1058     {
1059         wicrect.X = 0;
1060         wicrect.Y = 0;
1061         wicrect.Width = imginfo.Width;
1062         wicrect.Height = imginfo.Height;
1063     }
1064
1065     SetRect(&rect, 0, 0, wicrect.Width, wicrect.Height);
1066
1067     if (imginfo.ImageFileFormat == D3DXIFF_DDS)
1068     {
1069         hr = load_surface_from_dds(pDestSurface, pDestPalette, pDestRect, pSrcData, &rect,
1070             dwFilter, Colorkey, &imginfo);
1071         if (SUCCEEDED(hr) && pSrcInfo)
1072             *pSrcInfo = imginfo;
1073         return hr;
1074     }
1075
1076     if (imginfo.ImageFileFormat == D3DXIFF_DIB)
1077         convert_dib_to_bmp((void**)&pSrcData, &SrcDataSize);
1078
1079     CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
1080
1081     if (FAILED(CoCreateInstance(&CLSID_WICImagingFactory, NULL, CLSCTX_INPROC_SERVER, &IID_IWICImagingFactory, (void**)&factory)))
1082         goto cleanup_err;
1083
1084     if (FAILED(IWICImagingFactory_CreateStream(factory, &stream)))
1085     {
1086         IWICImagingFactory_Release(factory);
1087         goto cleanup_err;
1088     }
1089
1090     IWICStream_InitializeFromMemory(stream, (BYTE*)pSrcData, SrcDataSize);
1091
1092     hr = IWICImagingFactory_CreateDecoderFromStream(factory, (IStream*)stream, NULL, 0, &decoder);
1093
1094     IWICStream_Release(stream);
1095     IWICImagingFactory_Release(factory);
1096
1097     if (FAILED(hr))
1098         goto cleanup_err;
1099
1100     hr = IWICBitmapDecoder_GetFrame(decoder, 0, &bitmapframe);
1101
1102     if (FAILED(hr))
1103         goto cleanup_bmp;
1104
1105     formatdesc = get_format_info(imginfo.Format);
1106
1107     if (formatdesc->type == FORMAT_UNKNOWN)
1108     {
1109         FIXME("Unsupported pixel format\n");
1110         hr = D3DXERR_INVALIDDATA;
1111     }
1112     else
1113     {
1114         BYTE *buffer;
1115         DWORD pitch;
1116
1117         pitch = formatdesc->bytes_per_pixel * wicrect.Width;
1118         buffer = HeapAlloc(GetProcessHeap(), 0, pitch * wicrect.Height);
1119
1120         hr = IWICBitmapFrameDecode_CopyPixels(bitmapframe, &wicrect, pitch,
1121                                               pitch * wicrect.Height, buffer);
1122
1123         if (SUCCEEDED(hr))
1124         {
1125             hr = D3DXLoadSurfaceFromMemory(pDestSurface, pDestPalette, pDestRect,
1126                                            buffer, imginfo.Format, pitch,
1127                                            NULL, &rect, dwFilter, Colorkey);
1128         }
1129
1130         HeapFree(GetProcessHeap(), 0, buffer);
1131     }
1132
1133     IWICBitmapFrameDecode_Release(bitmapframe);
1134
1135 cleanup_bmp:
1136     IWICBitmapDecoder_Release(decoder);
1137
1138 cleanup_err:
1139     CoUninitialize();
1140
1141     if (imginfo.ImageFileFormat == D3DXIFF_DIB)
1142         HeapFree(GetProcessHeap(), 0, (void*)pSrcData);
1143
1144     if (FAILED(hr))
1145         return D3DXERR_INVALIDDATA;
1146
1147     if (pSrcInfo)
1148         *pSrcInfo = imginfo;
1149
1150     return D3D_OK;
1151 }
1152
1153 HRESULT WINAPI D3DXLoadSurfaceFromFileA(IDirect3DSurface9 *dst_surface,
1154         const PALETTEENTRY *dst_palette, const RECT *dst_rect, const char *src_file,
1155         const RECT *src_rect, DWORD filter, D3DCOLOR color_key, D3DXIMAGE_INFO *src_info)
1156 {
1157     LPWSTR pWidename;
1158     HRESULT hr;
1159     int strlength;
1160
1161     TRACE("dst_surface %p, dst_palette %p, dst_rect %s, src_file %s, "
1162             "src_rect %s, filter %#x, color_key 0x%08x, src_info %p.\n",
1163             dst_surface, dst_palette, wine_dbgstr_rect(dst_rect), debugstr_a(src_file),
1164             wine_dbgstr_rect(src_rect), filter, color_key, src_info);
1165
1166     if (!src_file || !dst_surface)
1167         return D3DERR_INVALIDCALL;
1168
1169     strlength = MultiByteToWideChar(CP_ACP, 0, src_file, -1, NULL, 0);
1170     pWidename = HeapAlloc(GetProcessHeap(), 0, strlength * sizeof(*pWidename));
1171     MultiByteToWideChar(CP_ACP, 0, src_file, -1, pWidename, strlength);
1172
1173     hr = D3DXLoadSurfaceFromFileW(dst_surface, dst_palette, dst_rect,
1174             pWidename, src_rect, filter, color_key, src_info);
1175     HeapFree(GetProcessHeap(), 0, pWidename);
1176
1177     return hr;
1178 }
1179
1180 HRESULT WINAPI D3DXLoadSurfaceFromFileW(IDirect3DSurface9 *dst_surface,
1181         const PALETTEENTRY *dst_palette, const RECT *dst_rect, const WCHAR *src_file,
1182         const RECT *src_rect, DWORD filter, D3DCOLOR color_key, D3DXIMAGE_INFO *src_info)
1183 {
1184     UINT data_size;
1185     void *data;
1186     HRESULT hr;
1187
1188     TRACE("dst_surface %p, dst_palette %p, dst_rect %s, src_file %s, "
1189             "src_rect %s, filter %#x, color_key 0x%08x, src_info %p.\n",
1190             dst_surface, dst_palette, wine_dbgstr_rect(dst_rect), debugstr_w(src_file),
1191             wine_dbgstr_rect(src_rect), filter, color_key, src_info);
1192
1193     if (!src_file || !dst_surface)
1194         return D3DERR_INVALIDCALL;
1195
1196     if (FAILED(map_view_of_file(src_file, &data, &data_size)))
1197         return D3DXERR_INVALIDDATA;
1198
1199     hr = D3DXLoadSurfaceFromFileInMemory(dst_surface, dst_palette, dst_rect,
1200             data, data_size, src_rect, filter, color_key, src_info);
1201     UnmapViewOfFile(data);
1202
1203     return hr;
1204 }
1205
1206 HRESULT WINAPI D3DXLoadSurfaceFromResourceA(IDirect3DSurface9 *dst_surface,
1207         const PALETTEENTRY *dst_palette, const RECT *dst_rect, HMODULE src_module, const char *resource,
1208         const RECT *src_rect, DWORD filter, D3DCOLOR color_key, D3DXIMAGE_INFO *src_info)
1209 {
1210     HRSRC hResInfo;
1211
1212     TRACE("dst_surface %p, dst_palette %p, dst_rect %s, src_module %p, resource %s, "
1213             "src_rect %s, filter %#x, color_key 0x%08x, src_info %p.\n",
1214             dst_surface, dst_palette, wine_dbgstr_rect(dst_rect), src_module, debugstr_a(resource),
1215             wine_dbgstr_rect(src_rect), filter, color_key, src_info);
1216
1217     if (!dst_surface)
1218         return D3DERR_INVALIDCALL;
1219
1220     hResInfo = FindResourceA(src_module, resource, (const char *)RT_RCDATA);
1221     if (!hResInfo) /* Try loading the resource as bitmap data (which is in DIB format D3DXIFF_DIB) */
1222         hResInfo = FindResourceA(src_module, resource, (const char *)RT_BITMAP);
1223
1224     if (hResInfo)
1225     {
1226         UINT data_size;
1227         void *data;
1228
1229         if (FAILED(load_resource_into_memory(src_module, hResInfo, &data, &data_size)))
1230             return D3DXERR_INVALIDDATA;
1231
1232         return D3DXLoadSurfaceFromFileInMemory(dst_surface, dst_palette, dst_rect,
1233                 data, data_size, src_rect, filter, color_key, src_info);
1234     }
1235
1236     return D3DXERR_INVALIDDATA;
1237 }
1238
1239 HRESULT WINAPI D3DXLoadSurfaceFromResourceW(IDirect3DSurface9 *dst_surface,
1240         const PALETTEENTRY *dst_palette, const RECT *dst_rect, HMODULE src_module, const WCHAR *resource,
1241         const RECT *src_rect, DWORD filter, D3DCOLOR color_key, D3DXIMAGE_INFO *src_info)
1242 {
1243     HRSRC hResInfo;
1244
1245     TRACE("dst_surface %p, dst_palette %p, dst_rect %s, src_module %p, resource %s, "
1246             "src_rect %s, filter %#x, color_key 0x%08x, src_info %p.\n",
1247             dst_surface, dst_palette, wine_dbgstr_rect(dst_rect), src_module, debugstr_w(resource),
1248             wine_dbgstr_rect(src_rect), filter, color_key, src_info);
1249
1250     if (!dst_surface)
1251         return D3DERR_INVALIDCALL;
1252
1253     hResInfo = FindResourceW(src_module, resource, (const WCHAR *)RT_RCDATA);
1254     if (!hResInfo) /* Try loading the resource as bitmap data (which is in DIB format D3DXIFF_DIB) */
1255         hResInfo = FindResourceW(src_module, resource, (const WCHAR *)RT_BITMAP);
1256
1257     if (hResInfo)
1258     {
1259         UINT data_size;
1260         void *data;
1261
1262         if (FAILED(load_resource_into_memory(src_module, hResInfo, &data, &data_size)))
1263             return D3DXERR_INVALIDDATA;
1264
1265         return D3DXLoadSurfaceFromFileInMemory(dst_surface, dst_palette, dst_rect,
1266                 data, data_size, src_rect, filter, color_key, src_info);
1267     }
1268
1269     return D3DXERR_INVALIDDATA;
1270 }
1271
1272
1273 /************************************************************
1274  * helper functions for D3DXLoadSurfaceFromMemory
1275  */
1276 struct argb_conversion_info
1277 {
1278     const struct pixel_format_desc *srcformat;
1279     const struct pixel_format_desc *destformat;
1280     DWORD srcshift[4], destshift[4];
1281     DWORD srcmask[4], destmask[4];
1282     BOOL process_channel[4];
1283     DWORD channelmask;
1284 };
1285
1286 static void init_argb_conversion_info(const struct pixel_format_desc *srcformat, const struct pixel_format_desc *destformat, struct argb_conversion_info *info)
1287 {
1288     UINT i;
1289     ZeroMemory(info->process_channel, 4 * sizeof(BOOL));
1290     info->channelmask = 0;
1291
1292     info->srcformat  =  srcformat;
1293     info->destformat = destformat;
1294
1295     for(i = 0;i < 4;i++) {
1296         /* srcshift is used to extract the _relevant_ components */
1297         info->srcshift[i]  =  srcformat->shift[i] + max( srcformat->bits[i] - destformat->bits[i], 0);
1298
1299         /* destshift is used to move the components to the correct position */
1300         info->destshift[i] = destformat->shift[i] + max(destformat->bits[i] -  srcformat->bits[i], 0);
1301
1302         info->srcmask[i]  = ((1 <<  srcformat->bits[i]) - 1) <<  srcformat->shift[i];
1303         info->destmask[i] = ((1 << destformat->bits[i]) - 1) << destformat->shift[i];
1304
1305         /* channelmask specifies bits which aren't used in the source format but in the destination one */
1306         if(destformat->bits[i]) {
1307             if(srcformat->bits[i]) info->process_channel[i] = TRUE;
1308             else info->channelmask |= info->destmask[i];
1309         }
1310     }
1311 }
1312
1313 static DWORD dword_from_bytes(CONST BYTE *src, UINT bytes_per_pixel)
1314 {
1315     DWORD ret = 0;
1316     static BOOL fixme_once;
1317
1318     if(bytes_per_pixel > sizeof(DWORD)) {
1319         if(!fixme_once++) FIXME("Unsupported image: %u bytes per pixel\n", bytes_per_pixel);
1320         bytes_per_pixel = sizeof(DWORD);
1321     }
1322
1323     memcpy(&ret, src, bytes_per_pixel);
1324     return ret;
1325 }
1326
1327 static void dword_to_bytes(BYTE *dst, DWORD dword, UINT bytes_per_pixel)
1328 {
1329     static BOOL fixme_once;
1330
1331     if(bytes_per_pixel > sizeof(DWORD)) {
1332         if(!fixme_once++) FIXME("Unsupported image: %u bytes per pixel\n", bytes_per_pixel);
1333         ZeroMemory(dst, bytes_per_pixel);
1334         bytes_per_pixel = sizeof(DWORD);
1335     }
1336
1337     memcpy(dst, &dword, bytes_per_pixel);
1338 }
1339
1340 /************************************************************
1341  * get_relevant_argb_components
1342  *
1343  * Extracts the relevant components from the source color and
1344  * drops the less significant bits if they aren't used by the destination format.
1345  */
1346 static void get_relevant_argb_components(CONST struct argb_conversion_info *info, CONST DWORD col, DWORD *out)
1347 {
1348     UINT i = 0;
1349     for(;i < 4;i++)
1350         if(info->process_channel[i])
1351             out[i] = (col & info->srcmask[i]) >> info->srcshift[i];
1352 }
1353
1354 /************************************************************
1355  * make_argb_color
1356  *
1357  * Recombines the output of get_relevant_argb_components and converts
1358  * it to the destination format.
1359  */
1360 static DWORD make_argb_color(CONST struct argb_conversion_info *info, CONST DWORD *in)
1361 {
1362     UINT i;
1363     DWORD val = 0;
1364
1365     for(i = 0;i < 4;i++) {
1366         if(info->process_channel[i]) {
1367             /* necessary to make sure that e.g. an X4R4G4B4 white maps to an R8G8B8 white instead of 0xf0f0f0 */
1368             signed int shift;
1369             for(shift = info->destshift[i]; shift > info->destformat->shift[i]; shift -= info->srcformat->bits[i]) val |= in[i] << shift;
1370             val |= (in[i] >> (info->destformat->shift[i] - shift)) << info->destformat->shift[i];
1371         }
1372     }
1373     val |= info->channelmask;   /* new channels are set to their maximal value */
1374     return val;
1375 }
1376
1377 static void format_to_vec4(const struct pixel_format_desc *format, const DWORD *src, struct vec4 *dst)
1378 {
1379     DWORD mask;
1380
1381     if (format->bits[1])
1382     {
1383         mask = (1 << format->bits[1]) - 1;
1384         dst->x = (float)((*src >> format->shift[1]) & mask) / mask;
1385     }
1386     else
1387         dst->x = 1.0f;
1388
1389     if (format->bits[2])
1390     {
1391         mask = (1 << format->bits[2]) - 1;
1392         dst->y = (float)((*src >> format->shift[2]) & mask) / mask;
1393     }
1394     else
1395         dst->y = 1.0f;
1396
1397     if (format->bits[3])
1398     {
1399         mask = (1 << format->bits[3]) - 1;
1400         dst->z = (float)((*src >> format->shift[3]) & mask) / mask;
1401     }
1402     else
1403         dst->z = 1.0f;
1404
1405     if (format->bits[0])
1406     {
1407         mask = (1 << format->bits[0]) - 1;
1408         dst->w = (float)((*src >> format->shift[0]) & mask) / mask;
1409     }
1410     else
1411         dst->w = 1.0f;
1412 }
1413
1414 static void format_from_vec4(const struct pixel_format_desc *format, const struct vec4 *src, DWORD *dst)
1415 {
1416     *dst = 0;
1417
1418     if (format->bits[1])
1419         *dst |= (DWORD)(src->x * ((1 << format->bits[1]) - 1) + 0.5f) << format->shift[1];
1420     if (format->bits[2])
1421         *dst |= (DWORD)(src->y * ((1 << format->bits[2]) - 1) + 0.5f) << format->shift[2];
1422     if (format->bits[3])
1423         *dst |= (DWORD)(src->z * ((1 << format->bits[3]) - 1) + 0.5f) << format->shift[3];
1424     if (format->bits[0])
1425         *dst |= (DWORD)(src->w * ((1 << format->bits[0]) - 1) + 0.5f) << format->shift[0];
1426 }
1427
1428 /************************************************************
1429  * copy_pixels
1430  *
1431  * Copies the source buffer to the destination buffer.
1432  * Works for any pixel format.
1433  * The source and the destination must be block-aligned.
1434  */
1435 void copy_pixels(const BYTE *src, UINT src_row_pitch, UINT src_slice_pitch,
1436         BYTE *dst, UINT dst_row_pitch, UINT dst_slice_pitch, const struct volume *size,
1437         const struct pixel_format_desc *format)
1438 {
1439     UINT row, slice;
1440     BYTE *dst_addr;
1441     const BYTE *src_addr;
1442     UINT row_block_count = (size->width + format->block_width - 1) / format->block_width;
1443     UINT row_count = (size->height + format->block_height - 1) / format->block_height;
1444
1445     for (slice = 0; slice < size->depth; slice++)
1446     {
1447         src_addr = src + slice * src_slice_pitch;
1448         dst_addr = dst + slice * dst_slice_pitch;
1449
1450         for (row = 0; row < row_count; row++)
1451         {
1452             memcpy(dst_addr, src_addr, row_block_count * format->block_byte_count);
1453             src_addr += src_row_pitch;
1454             dst_addr += dst_row_pitch;
1455         }
1456     }
1457 }
1458
1459 /************************************************************
1460  * convert_argb_pixels
1461  *
1462  * Copies the source buffer to the destination buffer, performing
1463  * any necessary format conversion and color keying.
1464  * Pixels outsize the source rect are blacked out.
1465  * Works only for ARGB formats with 1 - 4 bytes per pixel.
1466  */
1467 void convert_argb_pixels(const BYTE *src, UINT src_row_pitch, UINT src_slice_pitch, const struct volume *src_size,
1468         const struct pixel_format_desc *src_format, BYTE *dst, UINT dst_row_pitch, UINT dst_slice_pitch,
1469         const struct volume *dst_size, const struct pixel_format_desc *dst_format, D3DCOLOR color_key)
1470 {
1471     struct argb_conversion_info conv_info, ck_conv_info;
1472     const struct pixel_format_desc *ck_format = NULL;
1473     DWORD channels[4], pixel;
1474     UINT min_width, min_height, min_depth;
1475     UINT x, y, z;
1476
1477     ZeroMemory(channels, sizeof(channels));
1478     init_argb_conversion_info(src_format, dst_format, &conv_info);
1479
1480     min_width = min(src_size->width, dst_size->width);
1481     min_height = min(src_size->height, dst_size->height);
1482     min_depth = min(src_size->depth, dst_size->depth);
1483
1484     if (color_key)
1485     {
1486         /* Color keys are always represented in D3DFMT_A8R8G8B8 format. */
1487         ck_format = get_format_info(D3DFMT_A8R8G8B8);
1488         init_argb_conversion_info(src_format, ck_format, &ck_conv_info);
1489     }
1490
1491     for (z = 0; z < min_depth; z++) {
1492         const BYTE *src_slice_ptr = src + z * src_slice_pitch;
1493         BYTE *dst_slice_ptr = dst + z * dst_slice_pitch;
1494
1495         for (y = 0; y < min_height; y++) {
1496             const BYTE *src_ptr = src_slice_ptr + y * src_row_pitch;
1497             BYTE *dst_ptr = dst_slice_ptr + y * dst_row_pitch;
1498             DWORD val;
1499
1500             for (x = 0; x < min_width; x++) {
1501                 /* extract source color components */
1502                 pixel = dword_from_bytes(src_ptr, src_format->bytes_per_pixel);
1503
1504                 if (!src_format->to_rgba && !dst_format->from_rgba)
1505                 {
1506                     get_relevant_argb_components(&conv_info, pixel, channels);
1507                     val = make_argb_color(&conv_info, channels);
1508
1509                     if (color_key)
1510                     {
1511                         get_relevant_argb_components(&ck_conv_info, pixel, channels);
1512                         pixel = make_argb_color(&ck_conv_info, channels);
1513                         if (pixel == color_key)
1514                             val &= ~conv_info.destmask[0];
1515                     }
1516                 }
1517                 else
1518                 {
1519                     struct vec4 color, tmp;
1520
1521                     format_to_vec4(src_format, &pixel, &color);
1522                     if (src_format->to_rgba)
1523                         src_format->to_rgba(&color, &tmp);
1524                     else
1525                         tmp = color;
1526
1527                     if (ck_format)
1528                     {
1529                         format_from_vec4(ck_format, &tmp, &pixel);
1530                         if (pixel == color_key)
1531                             tmp.w = 0.0f;
1532                     }
1533
1534                     if (dst_format->from_rgba)
1535                         dst_format->from_rgba(&tmp, &color);
1536                     else
1537                         color = tmp;
1538
1539                     format_from_vec4(dst_format, &color, &val);
1540                 }
1541
1542                 dword_to_bytes(dst_ptr, val, dst_format->bytes_per_pixel);
1543                 src_ptr  +=  src_format->bytes_per_pixel;
1544                 dst_ptr += dst_format->bytes_per_pixel;
1545             }
1546
1547             if (src_size->width < dst_size->width) /* black out remaining pixels */
1548                 memset(dst_ptr, 0, dst_format->bytes_per_pixel * (dst_size->width - src_size->width));
1549         }
1550
1551         if (src_size->height < dst_size->height) /* black out remaining pixels */
1552             memset(dst + src_size->height * dst_row_pitch, 0, dst_row_pitch * (dst_size->height - src_size->height));
1553     }
1554     if (src_size->depth < dst_size->depth) /* black out remaining pixels */
1555         memset(dst + src_size->depth * dst_slice_pitch, 0, dst_slice_pitch * (dst_size->depth - src_size->depth));
1556 }
1557
1558 /************************************************************
1559  * point_filter_argb_pixels
1560  *
1561  * Copies the source buffer to the destination buffer, performing
1562  * any necessary format conversion, color keying and stretching
1563  * using a point filter.
1564  * Works only for ARGB formats with 1 - 4 bytes per pixel.
1565  */
1566 void point_filter_argb_pixels(const BYTE *src, UINT src_row_pitch, UINT src_slice_pitch, const struct volume *src_size,
1567         const struct pixel_format_desc *src_format, BYTE *dst, UINT dst_row_pitch, UINT dst_slice_pitch,
1568         const struct volume *dst_size, const struct pixel_format_desc *dst_format, D3DCOLOR color_key)
1569 {
1570     struct argb_conversion_info conv_info, ck_conv_info;
1571     const struct pixel_format_desc *ck_format = NULL;
1572     DWORD channels[4], pixel;
1573     UINT x, y, z;
1574
1575     ZeroMemory(channels, sizeof(channels));
1576     init_argb_conversion_info(src_format, dst_format, &conv_info);
1577
1578     if (color_key)
1579     {
1580         /* Color keys are always represented in D3DFMT_A8R8G8B8 format. */
1581         ck_format = get_format_info(D3DFMT_A8R8G8B8);
1582         init_argb_conversion_info(src_format, ck_format, &ck_conv_info);
1583     }
1584
1585     for (z = 0; z < dst_size->depth; z++)
1586     {
1587         BYTE *dst_slice_ptr = dst + z * dst_slice_pitch;
1588         const BYTE *src_slice_ptr = src + src_slice_pitch * (z * src_size->depth / dst_size->depth);
1589
1590         for (y = 0; y < dst_size->height; y++)
1591         {
1592             BYTE *dst_ptr = dst_slice_ptr + y * dst_row_pitch;
1593             const BYTE *src_row_ptr = src_slice_ptr + src_row_pitch * (y * src_size->height / dst_size->height);
1594
1595             for (x = 0; x < dst_size->width; x++)
1596             {
1597                 const BYTE *src_ptr = src_row_ptr + (x * src_size->width / dst_size->width) * src_format->bytes_per_pixel;
1598                 DWORD val;
1599
1600                 /* extract source color components */
1601                 pixel = dword_from_bytes(src_ptr, src_format->bytes_per_pixel);
1602
1603                 if (!src_format->to_rgba && !dst_format->from_rgba)
1604                 {
1605                     get_relevant_argb_components(&conv_info, pixel, channels);
1606                     val = make_argb_color(&conv_info, channels);
1607
1608                     if (color_key)
1609                     {
1610                         get_relevant_argb_components(&ck_conv_info, pixel, channels);
1611                         pixel = make_argb_color(&ck_conv_info, channels);
1612                         if (pixel == color_key)
1613                             val &= ~conv_info.destmask[0];
1614                     }
1615                 }
1616                 else
1617                 {
1618                     struct vec4 color, tmp;
1619
1620                     format_to_vec4(src_format, &pixel, &color);
1621                     if (src_format->to_rgba)
1622                         src_format->to_rgba(&color, &tmp);
1623                     else
1624                         tmp = color;
1625
1626                     if (ck_format)
1627                     {
1628                         format_from_vec4(ck_format, &tmp, &pixel);
1629                         if (pixel == color_key)
1630                             tmp.w = 0.0f;
1631                     }
1632
1633                     if (dst_format->from_rgba)
1634                         dst_format->from_rgba(&tmp, &color);
1635                     else
1636                         color = tmp;
1637
1638                     format_from_vec4(dst_format, &color, &val);
1639                 }
1640
1641                 dword_to_bytes(dst_ptr, val, dst_format->bytes_per_pixel);
1642                 dst_ptr += dst_format->bytes_per_pixel;
1643             }
1644         }
1645     }
1646 }
1647
1648 /************************************************************
1649  * D3DXLoadSurfaceFromMemory
1650  *
1651  * Loads data from a given memory chunk into a surface,
1652  * applying any of the specified filters.
1653  *
1654  * PARAMS
1655  *   pDestSurface [I] pointer to the surface
1656  *   pDestPalette [I] palette to use
1657  *   pDestRect    [I] to be filled area of the surface
1658  *   pSrcMemory   [I] pointer to the source data
1659  *   SrcFormat    [I] format of the source pixel data
1660  *   SrcPitch     [I] number of bytes in a row
1661  *   pSrcPalette  [I] palette used in the source image
1662  *   pSrcRect     [I] area of the source data to load
1663  *   dwFilter     [I] filter to apply on stretching
1664  *   Colorkey     [I] colorkey
1665  *
1666  * RETURNS
1667  *   Success: D3D_OK, if we successfully load the pixel data into our surface or
1668  *                    if pSrcMemory is NULL but the other parameters are valid
1669  *   Failure: D3DERR_INVALIDCALL, if pDestSurface, SrcPitch or pSrcRect are NULL or
1670  *                                if SrcFormat is an invalid format (other than D3DFMT_UNKNOWN) or
1671  *                                if DestRect is invalid
1672  *            D3DXERR_INVALIDDATA, if we fail to lock pDestSurface
1673  *            E_FAIL, if SrcFormat is D3DFMT_UNKNOWN or the dimensions of pSrcRect are invalid
1674  *
1675  * NOTES
1676  *   pSrcRect specifies the dimensions of the source data;
1677  *   negative values for pSrcRect are allowed as we're only looking at the width and height anyway.
1678  *
1679  */
1680 HRESULT WINAPI D3DXLoadSurfaceFromMemory(IDirect3DSurface9 *dst_surface,
1681         const PALETTEENTRY *dst_palette, const RECT *dst_rect, const void *src_memory,
1682         D3DFORMAT src_format, UINT src_pitch, const PALETTEENTRY *src_palette, const RECT *src_rect,
1683         DWORD filter, D3DCOLOR color_key)
1684 {
1685     const struct pixel_format_desc *srcformatdesc, *destformatdesc;
1686     D3DSURFACE_DESC surfdesc;
1687     D3DLOCKED_RECT lockrect;
1688     struct volume src_size, dst_size;
1689
1690     TRACE("(%p, %p, %s, %p, %#x, %u, %p, %s %#x, 0x%08x)\n",
1691             dst_surface, dst_palette, wine_dbgstr_rect(dst_rect), src_memory, src_format,
1692             src_pitch, src_palette, wine_dbgstr_rect(src_rect), filter, color_key);
1693
1694     if (!dst_surface || !src_memory || !src_rect)
1695     {
1696         WARN("Invalid argument specified.\n");
1697         return D3DERR_INVALIDCALL;
1698     }
1699     if (src_format == D3DFMT_UNKNOWN
1700             || src_rect->left >= src_rect->right
1701             || src_rect->top >= src_rect->bottom)
1702     {
1703         WARN("Invalid src_format or src_rect.\n");
1704         return E_FAIL;
1705     }
1706
1707     if (filter == D3DX_DEFAULT)
1708         filter = D3DX_FILTER_TRIANGLE | D3DX_FILTER_DITHER;
1709
1710     IDirect3DSurface9_GetDesc(dst_surface, &surfdesc);
1711
1712     src_size.width = src_rect->right - src_rect->left;
1713     src_size.height = src_rect->bottom - src_rect->top;
1714     src_size.depth = 1;
1715     if (!dst_rect)
1716     {
1717         dst_size.width = surfdesc.Width;
1718         dst_size.height = surfdesc.Height;
1719     }
1720     else
1721     {
1722         if (dst_rect->left > dst_rect->right || dst_rect->right > surfdesc.Width
1723                 || dst_rect->top > dst_rect->bottom || dst_rect->bottom > surfdesc.Height
1724                 || dst_rect->left < 0 || dst_rect->top < 0)
1725         {
1726             WARN("Invalid dst_rect specified.\n");
1727             return D3DERR_INVALIDCALL;
1728         }
1729         dst_size.width = dst_rect->right - dst_rect->left;
1730         dst_size.height = dst_rect->bottom - dst_rect->top;
1731         if (!dst_size.width || !dst_size.height)
1732             return D3D_OK;
1733     }
1734     dst_size.depth = 1;
1735
1736     srcformatdesc = get_format_info(src_format);
1737     destformatdesc = get_format_info(surfdesc.Format);
1738     if (srcformatdesc->type == FORMAT_UNKNOWN || destformatdesc->type == FORMAT_UNKNOWN)
1739     {
1740         FIXME("Unsupported pixel format conversion %#x -> %#x\n", src_format, surfdesc.Format);
1741         return E_NOTIMPL;
1742     }
1743
1744     if (src_format == surfdesc.Format
1745             && dst_size.width == src_size.width
1746             && dst_size.height == src_size.height
1747             && color_key == 0) /* Simple copy. */
1748     {
1749         if (src_rect->left & (srcformatdesc->block_width - 1)
1750                 || src_rect->top & (srcformatdesc->block_height - 1)
1751                 || (src_rect->right & (srcformatdesc->block_width - 1)
1752                     && src_size.width != surfdesc.Width)
1753                 || (src_rect->bottom & (srcformatdesc->block_height - 1)
1754                     && src_size.height != surfdesc.Height))
1755         {
1756             WARN("Source rect %s is misaligned.\n", wine_dbgstr_rect(src_rect));
1757             return D3DXERR_INVALIDDATA;
1758         }
1759
1760         if (FAILED(IDirect3DSurface9_LockRect(dst_surface, &lockrect, dst_rect, 0)))
1761             return D3DXERR_INVALIDDATA;
1762
1763         copy_pixels(src_memory, src_pitch, 0, lockrect.pBits, lockrect.Pitch, 0,
1764                 &src_size, srcformatdesc);
1765
1766         IDirect3DSurface9_UnlockRect(dst_surface);
1767     }
1768     else /* Stretching or format conversion. */
1769     {
1770         if (srcformatdesc->type != FORMAT_ARGB || destformatdesc->type != FORMAT_ARGB)
1771         {
1772             FIXME("Format conversion missing %#x -> %#x\n", src_format, surfdesc.Format);
1773             return E_NOTIMPL;
1774         }
1775
1776         if (FAILED(IDirect3DSurface9_LockRect(dst_surface, &lockrect, dst_rect, 0)))
1777             return D3DXERR_INVALIDDATA;
1778
1779         if ((filter & 0xf) == D3DX_FILTER_NONE)
1780         {
1781             convert_argb_pixels(src_memory, src_pitch, 0, &src_size, srcformatdesc,
1782                     lockrect.pBits, lockrect.Pitch, 0, &dst_size, destformatdesc, color_key);
1783         }
1784         else /* if ((filter & 0xf) == D3DX_FILTER_POINT) */
1785         {
1786             if ((filter & 0xf) != D3DX_FILTER_POINT)
1787                 FIXME("Unhandled filter %#x.\n", filter);
1788
1789             /* Always apply a point filter until D3DX_FILTER_LINEAR,
1790              * D3DX_FILTER_TRIANGLE and D3DX_FILTER_BOX are implemented. */
1791             point_filter_argb_pixels(src_memory, src_pitch, 0, &src_size, srcformatdesc,
1792                     lockrect.pBits, lockrect.Pitch, 0, &dst_size, destformatdesc, color_key);
1793         }
1794
1795         IDirect3DSurface9_UnlockRect(dst_surface);
1796     }
1797
1798     return D3D_OK;
1799 }
1800
1801 /************************************************************
1802  * D3DXLoadSurfaceFromSurface
1803  *
1804  * Copies the contents from one surface to another, performing any required
1805  * format conversion, resizing or filtering.
1806  *
1807  * PARAMS
1808  *   pDestSurface [I] pointer to the destination surface
1809  *   pDestPalette [I] palette to use
1810  *   pDestRect    [I] to be filled area of the surface
1811  *   pSrcSurface  [I] pointer to the source surface
1812  *   pSrcPalette  [I] palette used for the source surface
1813  *   pSrcRect     [I] area of the source data to load
1814  *   dwFilter     [I] filter to apply on resizing
1815  *   Colorkey     [I] any ARGB value or 0 to disable color-keying
1816  *
1817  * RETURNS
1818  *   Success: D3D_OK
1819  *   Failure: D3DERR_INVALIDCALL, if pDestSurface or pSrcSurface are NULL
1820  *            D3DXERR_INVALIDDATA, if one of the surfaces is not lockable
1821  *
1822  */
1823 HRESULT WINAPI D3DXLoadSurfaceFromSurface(IDirect3DSurface9 *dst_surface,
1824         const PALETTEENTRY *dst_palette, const RECT *dst_rect, IDirect3DSurface9 *src_surface,
1825         const PALETTEENTRY *src_palette, const RECT *src_rect, DWORD filter, D3DCOLOR color_key)
1826 {
1827     RECT rect;
1828     D3DLOCKED_RECT lock;
1829     D3DSURFACE_DESC SrcDesc;
1830     HRESULT hr;
1831
1832     TRACE("dst_surface %p, dst_palette %p, dst_rect %s, src_surface %p, "
1833             "src_palette %p, src_rect %s, filter %#x, color_key 0x%08x.\n",
1834             dst_surface, dst_palette, wine_dbgstr_rect(dst_rect), src_surface,
1835             src_palette, wine_dbgstr_rect(src_rect), filter, color_key);
1836
1837     if (!dst_surface || !src_surface)
1838         return D3DERR_INVALIDCALL;
1839
1840     IDirect3DSurface9_GetDesc(src_surface, &SrcDesc);
1841
1842     if (!src_rect)
1843         SetRect(&rect, 0, 0, SrcDesc.Width, SrcDesc.Height);
1844     else
1845         rect = *src_rect;
1846
1847     if (FAILED(IDirect3DSurface9_LockRect(src_surface, &lock, NULL, D3DLOCK_READONLY)))
1848         return D3DXERR_INVALIDDATA;
1849
1850     hr = D3DXLoadSurfaceFromMemory(dst_surface, dst_palette, dst_rect,
1851             lock.pBits, SrcDesc.Format, lock.Pitch, src_palette, &rect, filter, color_key);
1852
1853     IDirect3DSurface9_UnlockRect(src_surface);
1854
1855     return hr;
1856 }
1857
1858
1859 HRESULT WINAPI D3DXSaveSurfaceToFileA(const char *dst_filename, D3DXIMAGE_FILEFORMAT file_format,
1860         IDirect3DSurface9 *src_surface, const PALETTEENTRY *src_palette, const RECT *src_rect)
1861 {
1862     int len;
1863     WCHAR *filename;
1864     HRESULT hr;
1865     ID3DXBuffer *buffer;
1866
1867     TRACE("(%s, %#x, %p, %p, %s): relay\n",
1868             wine_dbgstr_a(dst_filename), file_format, src_surface, src_palette, wine_dbgstr_rect(src_rect));
1869
1870     if (!dst_filename) return D3DERR_INVALIDCALL;
1871
1872     len = MultiByteToWideChar(CP_ACP, 0, dst_filename, -1, NULL, 0);
1873     filename = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
1874     if (!filename) return E_OUTOFMEMORY;
1875     MultiByteToWideChar(CP_ACP, 0, dst_filename, -1, filename, len);
1876
1877     hr = D3DXSaveSurfaceToFileInMemory(&buffer, file_format, src_surface, src_palette, src_rect);
1878     if (SUCCEEDED(hr))
1879     {
1880         hr = write_buffer_to_file(filename, buffer);
1881         ID3DXBuffer_Release(buffer);
1882     }
1883
1884     HeapFree(GetProcessHeap(), 0, filename);
1885     return hr;
1886 }
1887
1888 HRESULT WINAPI D3DXSaveSurfaceToFileW(const WCHAR *dst_filename, D3DXIMAGE_FILEFORMAT file_format,
1889         IDirect3DSurface9 *src_surface, const PALETTEENTRY *src_palette, const RECT *src_rect)
1890 {
1891     HRESULT hr;
1892     ID3DXBuffer *buffer;
1893
1894     TRACE("(%s, %#x, %p, %p, %s): relay\n",
1895         wine_dbgstr_w(dst_filename), file_format, src_surface, src_palette, wine_dbgstr_rect(src_rect));
1896
1897     if (!dst_filename) return D3DERR_INVALIDCALL;
1898
1899     hr = D3DXSaveSurfaceToFileInMemory(&buffer, file_format, src_surface, src_palette, src_rect);
1900     if (SUCCEEDED(hr))
1901     {
1902         hr = write_buffer_to_file(dst_filename, buffer);
1903         ID3DXBuffer_Release(buffer);
1904     }
1905
1906     return hr;
1907 }
1908
1909 HRESULT WINAPI D3DXSaveSurfaceToFileInMemory(ID3DXBuffer **dst_buffer, D3DXIMAGE_FILEFORMAT file_format,
1910         IDirect3DSurface9 *src_surface, const PALETTEENTRY *src_palette, const RECT *src_rect)
1911 {
1912     IWICBitmapEncoder *encoder = NULL;
1913     IWICBitmapFrameEncode *frame = NULL;
1914     IPropertyBag2 *encoder_options = NULL;
1915     IStream *stream = NULL;
1916     HRESULT hr;
1917     HRESULT initresult;
1918     const CLSID *encoder_clsid;
1919     const GUID *pixel_format_guid;
1920     WICPixelFormatGUID wic_pixel_format;
1921     D3DFORMAT d3d_pixel_format;
1922     D3DSURFACE_DESC src_surface_desc;
1923     D3DLOCKED_RECT locked_rect;
1924     int width, height;
1925     STATSTG stream_stats;
1926     HGLOBAL stream_hglobal;
1927     ID3DXBuffer *buffer;
1928     DWORD size;
1929
1930     TRACE("(%p, %#x, %p, %p, %s)\n",
1931         dst_buffer, file_format, src_surface, src_palette, wine_dbgstr_rect(src_rect));
1932
1933     if (!dst_buffer || !src_surface) return D3DERR_INVALIDCALL;
1934
1935     if (src_palette)
1936     {
1937         FIXME("Saving surfaces with palettized pixel formats not implemented yet\n");
1938         return D3DERR_INVALIDCALL;
1939     }
1940
1941     switch (file_format)
1942     {
1943         case D3DXIFF_BMP:
1944         case D3DXIFF_DIB:
1945             encoder_clsid = &CLSID_WICBmpEncoder;
1946             break;
1947         case D3DXIFF_PNG:
1948             encoder_clsid = &CLSID_WICPngEncoder;
1949             break;
1950         case D3DXIFF_JPG:
1951             encoder_clsid = &CLSID_WICJpegEncoder;
1952             break;
1953         case D3DXIFF_DDS:
1954             return save_dds_surface_to_memory(dst_buffer, src_surface, src_rect);
1955         case D3DXIFF_HDR:
1956         case D3DXIFF_PFM:
1957         case D3DXIFF_TGA:
1958         case D3DXIFF_PPM:
1959             FIXME("File format %#x is not supported yet\n", file_format);
1960             return E_NOTIMPL;
1961         default:
1962             return D3DERR_INVALIDCALL;
1963     }
1964
1965     IDirect3DSurface9_GetDesc(src_surface, &src_surface_desc);
1966     if (src_rect)
1967     {
1968         if (src_rect->left == src_rect->right || src_rect->top == src_rect->bottom)
1969         {
1970             WARN("Invalid rectangle with 0 area\n");
1971             return D3DXCreateBuffer(64, dst_buffer);
1972         }
1973         if (src_rect->left < 0 || src_rect->top < 0)
1974             return D3DERR_INVALIDCALL;
1975         if (src_rect->left > src_rect->right || src_rect->top > src_rect->bottom)
1976             return D3DERR_INVALIDCALL;
1977         if (src_rect->right > src_surface_desc.Width || src_rect->bottom > src_surface_desc.Height)
1978             return D3DERR_INVALIDCALL;
1979
1980         width = src_rect->right - src_rect->left;
1981         height = src_rect->bottom - src_rect->top;
1982     }
1983     else
1984     {
1985         width = src_surface_desc.Width;
1986         height = src_surface_desc.Height;
1987     }
1988
1989     initresult = CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
1990
1991     hr = CoCreateInstance(encoder_clsid, NULL, CLSCTX_INPROC_SERVER,
1992         &IID_IWICBitmapEncoder, (void **)&encoder);
1993     if (FAILED(hr)) goto cleanup_err;
1994
1995     hr = CreateStreamOnHGlobal(NULL, TRUE, &stream);
1996     if (FAILED(hr)) goto cleanup_err;
1997
1998     hr = IWICBitmapEncoder_Initialize(encoder, stream, WICBitmapEncoderNoCache);
1999     if (FAILED(hr)) goto cleanup_err;
2000
2001     hr = IWICBitmapEncoder_CreateNewFrame(encoder, &frame, &encoder_options);
2002     if (FAILED(hr)) goto cleanup_err;
2003
2004     hr = IWICBitmapFrameEncode_Initialize(frame, encoder_options);
2005     if (FAILED(hr)) goto cleanup_err;
2006
2007     hr = IWICBitmapFrameEncode_SetSize(frame, width, height);
2008     if (FAILED(hr)) goto cleanup_err;
2009
2010     pixel_format_guid = d3dformat_to_wic_guid(src_surface_desc.Format);
2011     if (!pixel_format_guid)
2012     {
2013         FIXME("Pixel format %#x is not supported yet\n", src_surface_desc.Format);
2014         hr = E_NOTIMPL;
2015         goto cleanup;
2016     }
2017
2018     memcpy(&wic_pixel_format, pixel_format_guid, sizeof(GUID));
2019     hr = IWICBitmapFrameEncode_SetPixelFormat(frame, &wic_pixel_format);
2020     d3d_pixel_format = wic_guid_to_d3dformat(&wic_pixel_format);
2021     if (SUCCEEDED(hr) && d3d_pixel_format != D3DFMT_UNKNOWN)
2022     {
2023         TRACE("Using pixel format %s %#x\n", debugstr_guid(&wic_pixel_format), d3d_pixel_format);
2024
2025         if (src_surface_desc.Format == d3d_pixel_format) /* Simple copy */
2026         {
2027             hr = IDirect3DSurface9_LockRect(src_surface, &locked_rect, src_rect, D3DLOCK_READONLY);
2028             if (SUCCEEDED(hr))
2029             {
2030                 IWICBitmapFrameEncode_WritePixels(frame, height,
2031                     locked_rect.Pitch, height * locked_rect.Pitch, locked_rect.pBits);
2032                 IDirect3DSurface9_UnlockRect(src_surface);
2033             }
2034         }
2035         else /* Pixel format conversion */
2036         {
2037             const struct pixel_format_desc *src_format_desc, *dst_format_desc;
2038             struct volume size;
2039             DWORD dst_pitch;
2040             void *dst_data;
2041
2042             src_format_desc = get_format_info(src_surface_desc.Format);
2043             dst_format_desc = get_format_info(d3d_pixel_format);
2044             if (src_format_desc->type != FORMAT_ARGB || dst_format_desc->type != FORMAT_ARGB)
2045             {
2046                 FIXME("Unsupported pixel format conversion %#x -> %#x\n",
2047                     src_surface_desc.Format, d3d_pixel_format);
2048                 hr = E_NOTIMPL;
2049                 goto cleanup;
2050             }
2051
2052             size.width = width;
2053             size.height = height;
2054             size.depth = 1;
2055             dst_pitch = width * dst_format_desc->bytes_per_pixel;
2056             dst_data = HeapAlloc(GetProcessHeap(), 0, dst_pitch * height);
2057             if (!dst_data)
2058             {
2059                 hr = E_OUTOFMEMORY;
2060                 goto cleanup;
2061             }
2062
2063             hr = IDirect3DSurface9_LockRect(src_surface, &locked_rect, src_rect, D3DLOCK_READONLY);
2064             if (SUCCEEDED(hr))
2065             {
2066                 convert_argb_pixels(locked_rect.pBits, locked_rect.Pitch, 0, &size, src_format_desc,
2067                     dst_data, dst_pitch, 0, &size, dst_format_desc, 0);
2068                 IDirect3DSurface9_UnlockRect(src_surface);
2069             }
2070
2071             IWICBitmapFrameEncode_WritePixels(frame, height, dst_pitch, dst_pitch * height, dst_data);
2072             HeapFree(GetProcessHeap(), 0, dst_data);
2073         }
2074
2075         hr = IWICBitmapFrameEncode_Commit(frame);
2076         if (SUCCEEDED(hr)) hr = IWICBitmapEncoder_Commit(encoder);
2077     }
2078     else WARN("Unsupported pixel format %#x\n", src_surface_desc.Format);
2079
2080     /* copy data from stream to ID3DXBuffer */
2081     hr = IStream_Stat(stream, &stream_stats, STATFLAG_NONAME);
2082     if (FAILED(hr)) goto cleanup_err;
2083
2084     if (stream_stats.cbSize.u.HighPart != 0)
2085     {
2086         hr = D3DXERR_INVALIDDATA;
2087         goto cleanup;
2088     }
2089     size = stream_stats.cbSize.u.LowPart;
2090
2091     /* Remove BMP header for DIB */
2092     if (file_format == D3DXIFF_DIB)
2093         size -= sizeof(BITMAPFILEHEADER);
2094
2095     hr = D3DXCreateBuffer(size, &buffer);
2096     if (FAILED(hr)) goto cleanup;
2097
2098     hr = GetHGlobalFromStream(stream, &stream_hglobal);
2099     if (SUCCEEDED(hr))
2100     {
2101         void *buffer_pointer = ID3DXBuffer_GetBufferPointer(buffer);
2102         void *stream_data = GlobalLock(stream_hglobal);
2103         /* Remove BMP header for DIB */
2104         if (file_format == D3DXIFF_DIB)
2105             stream_data = (void*)((BYTE*)stream_data + sizeof(BITMAPFILEHEADER));
2106         memcpy(buffer_pointer, stream_data, size);
2107         GlobalUnlock(stream_hglobal);
2108         *dst_buffer = buffer;
2109     }
2110     else ID3DXBuffer_Release(buffer);
2111
2112 cleanup_err:
2113     if (FAILED(hr) && hr != E_OUTOFMEMORY)
2114         hr = D3DERR_INVALIDCALL;
2115
2116 cleanup:
2117     if (stream) IStream_Release(stream);
2118
2119     if (frame) IWICBitmapFrameEncode_Release(frame);
2120     if (encoder_options) IPropertyBag2_Release(encoder_options);
2121
2122     if (encoder) IWICBitmapEncoder_Release(encoder);
2123
2124     if (SUCCEEDED(initresult)) CoUninitialize();
2125
2126     return hr;
2127 }