gdi32: Don't free bitmap bits when changing the owner, we may still have pointers...
[wine] / dlls / wineps.drv / bitmap.c
1 /*
2  *      PostScript driver bitmap functions
3  *
4  * Copyright 1998  Huw D M Davies
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19  */
20
21 #include <assert.h>
22 #include <stdlib.h>
23
24 #include "psdrv.h"
25 #include "winbase.h"
26 #include "wine/debug.h"
27
28 WINE_DEFAULT_DEBUG_CHANNEL(psdrv);
29
30
31 /* Return the width of a DIB bitmap in bytes. DIB bitmap data is 32-bit aligned. */
32 static inline int get_dib_width_bytes( int width, int depth )
33 {
34     return ((width * depth + 31) / 8) & ~3;
35 }
36
37
38 /***************************************************************************
39  *                PSDRV_WriteImageHeader
40  *
41  * Helper for PSDRV_PutImage
42  *
43  * BUGS
44  *  Uses level 2 PostScript
45  */
46
47 static BOOL PSDRV_WriteImageHeader(PHYSDEV dev, const BITMAPINFO *info, INT xDst,
48                                    INT yDst, INT widthDst, INT heightDst,
49                                    INT widthSrc, INT heightSrc)
50 {
51     switch(info->bmiHeader.biBitCount)
52     {
53     case 1:
54     case 4:
55     case 8:
56         PSDRV_WriteIndexColorSpaceBegin(dev, (1 << info->bmiHeader.biBitCount) - 1);
57         PSDRV_WriteRGBQUAD(dev, info->bmiColors, 1 << info->bmiHeader.biBitCount);
58         PSDRV_WriteIndexColorSpaceEnd(dev);
59         break;
60
61     case 16:
62     case 24:
63     case 32:
64       {
65         PSCOLOR pscol;
66         pscol.type = PSCOLOR_RGB;
67         pscol.value.rgb.r = pscol.value.rgb.g = pscol.value.rgb.b = 0.0;
68         PSDRV_WriteSetColor(dev, &pscol);
69         break;
70       }
71     }
72
73     PSDRV_WriteImage(dev, info->bmiHeader.biBitCount, xDst, yDst,
74                      widthDst, heightDst, widthSrc, heightSrc, FALSE, info->bmiHeader.biHeight < 0);
75     return TRUE;
76 }
77
78
79 /***************************************************************************
80  *                PSDRV_WriteImageMaskHeader
81  *
82  * Helper for PSDRV_PutImage
83  *
84  * We use the imagemask operator for 1bpp bitmaps since the output
85  * takes much less time for the printer to render.
86  *
87  * BUGS
88  *  Uses level 2 PostScript
89  */
90
91 static BOOL PSDRV_WriteImageMaskHeader(PHYSDEV dev, const BITMAPINFO *info, INT xDst,
92                                        INT yDst, INT widthDst, INT heightDst,
93                                        INT widthSrc, INT heightSrc)
94 {
95     PSCOLOR bkgnd, foregnd;
96
97     assert(info->bmiHeader.biBitCount == 1);
98
99     /* We'll write the mask with -ve polarity so that 
100        the foregnd color corresponds to a bit equal to
101        0 in the bitmap.
102     */
103     PSDRV_CreateColor(dev, &foregnd, RGB(info->bmiColors[0].rgbRed,
104                                          info->bmiColors[0].rgbGreen,
105                                          info->bmiColors[0].rgbBlue) );
106     PSDRV_CreateColor(dev, &bkgnd, RGB(info->bmiColors[1].rgbRed,
107                                        info->bmiColors[1].rgbGreen,
108                                        info->bmiColors[1].rgbBlue) );
109
110     PSDRV_WriteGSave(dev);
111     PSDRV_WriteNewPath(dev);
112     PSDRV_WriteRectangle(dev, xDst, yDst, widthDst, heightDst);
113     PSDRV_WriteSetColor(dev, &bkgnd);
114     PSDRV_WriteFill(dev);
115     PSDRV_WriteGRestore(dev);
116
117     PSDRV_WriteSetColor(dev, &foregnd);
118     PSDRV_WriteImage(dev, 1, xDst, yDst, widthDst, heightDst,
119                      widthSrc, heightSrc, TRUE, info->bmiHeader.biHeight < 0);
120
121     return TRUE;
122 }
123
124 static inline DWORD max_rle_size(DWORD size)
125 {
126     return size + (size + 127) / 128 + 1;
127 }
128
129 static inline DWORD max_ascii85_size(DWORD size)
130 {
131     return (size + 3) / 4 * 5;
132 }
133
134 static void free_heap_bits( struct gdi_image_bits *bits )
135 {
136     HeapFree( GetProcessHeap(), 0, bits->ptr );
137 }
138
139 /***************************************************************************
140  *                PSDRV_WriteImageBits
141  */
142 static void PSDRV_WriteImageBits( PHYSDEV dev, const BITMAPINFO *info, INT xDst, INT yDst,
143                                   INT widthDst, INT heightDst, INT widthSrc, INT heightSrc,
144                                   void *bits, DWORD size )
145 {
146     BYTE *rle, *ascii85;
147     DWORD rle_len, ascii85_len;
148
149     if (info->bmiHeader.biBitCount == 1)
150         /* Use imagemask rather than image */
151         PSDRV_WriteImageMaskHeader(dev, info, xDst, yDst, widthDst, heightDst,
152                                    widthSrc, heightSrc);
153     else
154         PSDRV_WriteImageHeader(dev, info, xDst, yDst, widthDst, heightDst,
155                                widthSrc, heightSrc);
156
157     rle = HeapAlloc(GetProcessHeap(), 0, max_rle_size(size));
158     rle_len = RLE_encode(bits, size, rle);
159     ascii85 = HeapAlloc(GetProcessHeap(), 0, max_ascii85_size(rle_len));
160     ascii85_len = ASCII85_encode(rle, rle_len, ascii85);
161     HeapFree(GetProcessHeap(), 0, rle);
162     PSDRV_WriteData(dev, ascii85, ascii85_len);
163     PSDRV_WriteSpool(dev, "~>\n", 3);
164     HeapFree(GetProcessHeap(), 0, ascii85);
165 }
166
167 /***********************************************************************
168  *           PSDRV_PutImage
169  */
170 DWORD PSDRV_PutImage( PHYSDEV dev, HBITMAP hbitmap, HRGN clip, BITMAPINFO *info,
171                       const struct gdi_image_bits *bits, struct bitblt_coords *src,
172                       struct bitblt_coords *dst, DWORD rop )
173 {
174     int src_stride, dst_stride, size, x, y, width, height, bit_offset;
175     int dst_x, dst_y, dst_width, dst_height;
176     unsigned char *src_ptr, *dst_ptr;
177     struct gdi_image_bits dst_bits;
178
179     if (hbitmap) return ERROR_NOT_SUPPORTED;
180
181     if (info->bmiHeader.biPlanes != 1) goto update_format;
182     if (info->bmiHeader.biCompression != BI_RGB) goto update_format;
183     if (info->bmiHeader.biBitCount == 16 || info->bmiHeader.biBitCount == 32) goto update_format;
184     if (!bits) return ERROR_SUCCESS;  /* just querying the format */
185
186     TRACE( "bpp %u %s -> %s\n", info->bmiHeader.biBitCount, wine_dbgstr_rect(&src->visrect),
187            wine_dbgstr_rect(&dst->visrect) );
188
189     width = src->visrect.right - src->visrect.left;
190     height = src->visrect.bottom - src->visrect.top;
191     src_stride = get_dib_width_bytes( info->bmiHeader.biWidth, info->bmiHeader.biBitCount );
192     dst_stride = (width * info->bmiHeader.biBitCount + 7) / 8;
193
194     src_ptr = bits->ptr;
195     if (info->bmiHeader.biHeight > 0)
196         src_ptr += (info->bmiHeader.biHeight - src->visrect.bottom) * src_stride;
197     else
198         src_ptr += src->visrect.top * src_stride;
199     bit_offset = src->visrect.left * info->bmiHeader.biBitCount;
200     src_ptr += bit_offset / 8;
201     bit_offset &= 7;
202     if (bit_offset) FIXME( "pos %s not supported\n", wine_dbgstr_rect(&src->visrect) );
203     size = height * dst_stride;
204
205     if (src_stride != dst_stride || (info->bmiHeader.biBitCount == 24 && !bits->is_copy))
206     {
207         if (!(dst_bits.ptr = HeapAlloc( GetProcessHeap(), 0, size ))) return ERROR_OUTOFMEMORY;
208         dst_bits.is_copy = TRUE;
209         dst_bits.free = free_heap_bits;
210     }
211     else
212     {
213         dst_bits.ptr = src_ptr;
214         dst_bits.is_copy = bits->is_copy;
215         dst_bits.free = NULL;
216     }
217     dst_ptr = dst_bits.ptr;
218
219     switch (info->bmiHeader.biBitCount)
220     {
221     case 1:
222     case 4:
223     case 8:
224         if (src_stride != dst_stride)
225             for (y = 0; y < height; y++, src_ptr += src_stride, dst_ptr += dst_stride)
226                 memcpy( dst_ptr, src_ptr, dst_stride );
227         break;
228     case 24:
229         if (dst_ptr != src_ptr)
230             for (y = 0; y < height; y++, src_ptr += src_stride, dst_ptr += dst_stride)
231                 for (x = 0; x < width; x++)
232                 {
233                     dst_ptr[x * 3]     = src_ptr[x * 3 + 2];
234                     dst_ptr[x * 3 + 1] = src_ptr[x * 3 + 1];
235                     dst_ptr[x * 3 + 2] = src_ptr[x * 3];
236                 }
237         else  /* swap R and B in place */
238             for (y = 0; y < height; y++, src_ptr += src_stride, dst_ptr += dst_stride)
239                 for (x = 0; x < width; x++)
240                 {
241                     unsigned char tmp = dst_ptr[x * 3];
242                     dst_ptr[x * 3] = dst_ptr[x * 3 + 2];
243                     dst_ptr[x * 3 + 2] = tmp;
244                 }
245         break;
246     }
247
248     dst_x = dst->visrect.left;
249     dst_y = dst->visrect.top,
250     dst_width = dst->visrect.right - dst->visrect.left;
251     dst_height = dst->visrect.bottom - dst->visrect.top;
252     if (src->width * dst->width < 0)
253     {
254         dst_x += dst_width;
255         dst_width = -dst_width;
256     }
257     if (src->height * dst->height < 0)
258     {
259         dst_y += dst_height;
260         dst_height = -dst_height;
261     }
262
263     PSDRV_SetClip(dev);
264     PSDRV_WriteGSave(dev);
265     if (clip) PSDRV_AddClip( dev, clip );
266     PSDRV_WriteImageBits( dev, info, dst_x, dst_y, dst_width, dst_height,
267                           width, height, dst_bits.ptr, size );
268     PSDRV_WriteGRestore(dev);
269     PSDRV_ResetClip(dev);
270     if (dst_bits.free) dst_bits.free( &dst_bits );
271     return ERROR_SUCCESS;
272
273 update_format:
274     info->bmiHeader.biPlanes = 1;
275     if (info->bmiHeader.biBitCount > 8) info->bmiHeader.biBitCount = 24;
276     info->bmiHeader.biCompression = BI_RGB;
277     return ERROR_BAD_FORMAT;
278 }