gdiplus: Check for NULL image in GdipImageRotateFlip.
[wine] / dlls / wined3d / palette.c
1 /*              DirectDraw - IDirectPalette base interface
2  *
3  * Copyright 1997-2000 Marcus Meissner
4  * Copyright 2000-2001 TransGaming Technologies Inc.
5  * Copyright 2006 Stefan Dösinger for CodeWeavers
6  *
7  * This library is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * This library is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with this library; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
20  */
21 #include "config.h"
22 #include "wine/port.h"
23 #include "winerror.h"
24 #include "wine/debug.h"
25
26 #include <string.h>
27
28 #include "wined3d_private.h"
29
30 WINE_DEFAULT_DEBUG_CHANNEL(d3d);
31
32 #define SIZE_BITS (WINEDDPCAPS_1BIT | WINEDDPCAPS_2BIT | WINEDDPCAPS_4BIT | WINEDDPCAPS_8BIT)
33
34 ULONG CDECL wined3d_palette_incref(struct wined3d_palette *palette)
35 {
36     ULONG refcount = InterlockedIncrement(&palette->ref);
37
38     TRACE("%p increasing refcount to %u.\n", palette, refcount);
39
40     return refcount;
41 }
42
43 ULONG CDECL wined3d_palette_decref(struct wined3d_palette *palette)
44 {
45     ULONG refcount = InterlockedDecrement(&palette->ref);
46
47     TRACE("%p decreasing refcount to %u.\n", palette, refcount);
48
49     if (!refcount)
50     {
51         DeleteObject(palette->hpal);
52         HeapFree(GetProcessHeap(), 0, palette);
53     }
54
55     return refcount;
56 }
57
58 static WORD wined3d_palette_size(DWORD flags)
59 {
60     switch (flags & SIZE_BITS)
61     {
62         case WINEDDPCAPS_1BIT: return 2;
63         case WINEDDPCAPS_2BIT: return 4;
64         case WINEDDPCAPS_4BIT: return 16;
65         case WINEDDPCAPS_8BIT: return 256;
66         default:
67             FIXME("Unhandled size bits %#x.\n", flags & SIZE_BITS);
68             return 256;
69     }
70 }
71
72 HRESULT CDECL wined3d_palette_get_entries(const struct wined3d_palette *palette,
73         DWORD flags, DWORD start, DWORD count, PALETTEENTRY *entries)
74 {
75     TRACE("palette %p, flags %#x, start %u, count %u, entries %p.\n",
76             palette, flags, start, count, entries);
77
78     if (flags) return WINED3DERR_INVALIDCALL; /* unchecked */
79     if (start + count > wined3d_palette_size(palette->flags))
80         return WINED3DERR_INVALIDCALL;
81
82     if (palette->flags & WINEDDPCAPS_8BITENTRIES)
83     {
84         BYTE *entry = (BYTE *)entries;
85         unsigned int i;
86
87         for (i = start; i < count + start; ++i)
88             *entry++ = palette->palents[i].peRed;
89     }
90     else
91         memcpy(entries, palette->palents + start, count * sizeof(*entries));
92
93     return WINED3D_OK;
94 }
95
96 HRESULT CDECL wined3d_palette_set_entries(struct wined3d_palette *palette,
97         DWORD flags, DWORD start, DWORD count, const PALETTEENTRY *entries)
98 {
99     struct wined3d_resource *resource;
100
101     TRACE("palette %p, flags %#x, start %u, count %u, entries %p.\n",
102             palette, flags, start, count, entries);
103     TRACE("Palette flags: %#x.\n", palette->flags);
104
105     if (palette->flags & WINEDDPCAPS_8BITENTRIES)
106     {
107         const BYTE *entry = (const BYTE *)entries;
108         unsigned int i;
109
110         for (i = start; i < count + start; ++i)
111             palette->palents[i].peRed = *entry++;
112     }
113     else
114     {
115         memcpy(palette->palents + start, entries, count * sizeof(*palette->palents));
116
117         /* When WINEDDCAPS_ALLOW256 isn't set we need to override entry 0 with black and 255 with white */
118         if (!(palette->flags & WINEDDPCAPS_ALLOW256))
119         {
120             TRACE("WINEDDPCAPS_ALLOW256 set, overriding palette entry 0 with black and 255 with white\n");
121             palette->palents[0].peRed = 0;
122             palette->palents[0].peGreen = 0;
123             palette->palents[0].peBlue = 0;
124
125             palette->palents[255].peRed = 255;
126             palette->palents[255].peGreen = 255;
127             palette->palents[255].peBlue = 255;
128         }
129
130         if (palette->hpal)
131             SetPaletteEntries(palette->hpal, start, count, palette->palents + start);
132     }
133
134     /* If the palette is attached to the render target, update all render targets */
135     LIST_FOR_EACH_ENTRY(resource, &palette->device->resources, struct wined3d_resource, resource_list_entry)
136     {
137         if (resource->type == WINED3D_RTYPE_SURFACE)
138         {
139             struct wined3d_surface *surface = surface_from_resource(resource);
140             if (surface->palette == palette)
141                 surface->surface_ops->surface_realize_palette(surface);
142         }
143     }
144
145     return WINED3D_OK;
146 }
147
148 DWORD CDECL wined3d_palette_get_flags(const struct wined3d_palette *palette)
149 {
150     TRACE("palette %p.\n", palette);
151
152     return palette->flags;
153 }
154
155 void * CDECL wined3d_palette_get_parent(const struct wined3d_palette *palette)
156 {
157     TRACE("palette %p.\n", palette);
158
159     return palette->parent;
160 }
161
162 static HRESULT wined3d_palette_init(struct wined3d_palette *palette, struct wined3d_device *device,
163         DWORD flags, const PALETTEENTRY *entries, void *parent)
164 {
165     HRESULT hr;
166
167     palette->ref = 1;
168     palette->parent = parent;
169     palette->device = device;
170     palette->flags = flags;
171
172     palette->palNumEntries = wined3d_palette_size(flags);
173     palette->hpal = CreatePalette((const LOGPALETTE *)&palette->palVersion);
174     if (!palette->hpal)
175     {
176         WARN("Failed to create palette.\n");
177         return E_FAIL;
178     }
179
180     hr = wined3d_palette_set_entries(palette, 0, 0, wined3d_palette_size(flags), entries);
181     if (FAILED(hr))
182     {
183         WARN("Failed to set palette entries, hr %#x.\n", hr);
184         DeleteObject(palette->hpal);
185         return hr;
186     }
187
188     return WINED3D_OK;
189 }
190
191 HRESULT CDECL wined3d_palette_create(struct wined3d_device *device, DWORD flags,
192         const PALETTEENTRY *entries, void *parent, struct wined3d_palette **palette)
193 {
194     struct wined3d_palette *object;
195     HRESULT hr;
196
197     TRACE("device %p, flags %#x, entries %p, palette %p, parent %p.\n",
198             device, flags, entries, palette, parent);
199
200     object = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*object));
201     if (!object)
202     {
203         ERR("Failed to allocate palette memory.\n");
204         return E_OUTOFMEMORY;
205     }
206
207     hr = wined3d_palette_init(object, device, flags, entries, parent);
208     if (FAILED(hr))
209     {
210         WARN("Failed to initialize palette, hr %#x.\n", hr);
211         HeapFree(GetProcessHeap(), 0, object);
212         return hr;
213     }
214
215     TRACE("Created palette %p.\n", object);
216     *palette = object;
217
218     return WINED3D_OK;
219 }