ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/rxvt-unicode/src/rxvtimg.C
Revision: 1.54
Committed: Thu Jun 7 20:24:39 2012 UTC (11 years, 11 months ago) by root
Content type: text/plain
Branch: MAIN
Changes since 1.53: +11 -1 lines
Log Message:
*** empty log message ***

File Contents

# User Rev Content
1 sf-exg 1.6 #include <math.h>
2 root 1.1 #include "../config.h"
3     #include "rxvt.h"
4    
5     #if HAVE_IMG
6    
7 root 1.41 rxvt_img::rxvt_img (rxvt_screen *screen, XRenderPictFormat *format, int x, int y, int width, int height, int repeat)
8     : s(screen), x(x), y(y), w(width), h(height), format(format), repeat(repeat),
9 root 1.34 pm(0), ref(0)
10 root 1.1 {
11     }
12    
13 root 1.32 rxvt_img::rxvt_img (const rxvt_img &img)
14 root 1.34 : s(img.s), x(img.x), y(img.y), w(img.w), h(img.h), format(img.format), repeat(img.repeat), pm(img.pm), ref(img.ref)
15 root 1.32 {
16 root 1.34 ++ref->cnt;
17 root 1.32 }
18    
19     #if 0
20 root 1.3 rxvt_img::rxvt_img (rxvt_screen *screen, XRenderPictFormat *format, int width, int height, Pixmap pixmap)
21 root 1.30 : s(screen), x(0), y(0), w(width), h(height), format(format), repeat(RepeatNormal), shared(false), pm(pixmap)
22 root 1.1 {
23     }
24 root 1.32 #endif
25 root 1.1
26 root 1.4 rxvt_img *
27 sf-exg 1.21 rxvt_img::new_from_root (rxvt_screen *s)
28     {
29     Display *dpy = s->display->dpy;
30     unsigned int root_pm_w, root_pm_h;
31     Pixmap root_pixmap = s->display->get_pixmap_property (s->display->xa[XA_XROOTPMAP_ID]);
32     if (root_pixmap == None)
33     root_pixmap = s->display->get_pixmap_property (s->display->xa[XA_ESETROOT_PMAP_ID]);
34    
35     if (root_pixmap == None)
36     return 0;
37    
38     Window wdummy;
39     int idummy;
40     unsigned int udummy;
41    
42     if (!XGetGeometry (dpy, root_pixmap, &wdummy, &idummy, &idummy, &root_pm_w, &root_pm_h, &udummy, &udummy))
43     return 0;
44    
45     rxvt_img *img = new rxvt_img (
46     s,
47     XRenderFindVisualFormat (dpy, DefaultVisual (dpy, s->display->screen)),
48 root 1.33 0,
49     0,
50 sf-exg 1.21 root_pm_w,
51 root 1.32 root_pm_h
52 sf-exg 1.21 );
53    
54 root 1.32 img->pm = root_pixmap;
55 root 1.34 img->ref = new pixref (root_pm_w, root_pm_h);
56     img->ref->ours = false;
57 sf-exg 1.21
58     return img;
59     }
60    
61     rxvt_img *
62 root 1.47 rxvt_img::new_from_pixbuf (rxvt_screen *s, GdkPixbuf *pb)
63     {
64 root 1.48 Display *dpy = s->display->dpy;
65    
66 root 1.47 int width = gdk_pixbuf_get_width (pb);
67     int height = gdk_pixbuf_get_height (pb);
68    
69     if (width > 32767 || height > 32767) // well, we *could* upload in chunks
70     rxvt_fatal ("rxvt_img::new_from_pixbuf: image too big (maximum size 32768x32768).\n");
71    
72     // since we require rgb24/argb32 formats from xrender we assume
73     // that both 24 and 32 bpp MUST be supported by any screen that supports xrender
74     int depth = gdk_pixbuf_get_has_alpha (pb) ? 32 : 24;
75    
76 root 1.54 int byte_order = ecb_big_endian () ? MSBFirst : LSBFirst;
77    
78 root 1.47 XImage xi;
79    
80     xi.width = width;
81     xi.height = height;
82     xi.xoffset = 0;
83     xi.format = ZPixmap;
84 root 1.54 xi.byte_order = ImageByteOrder (dpy);
85 root 1.51 xi.bitmap_unit = 32; //XY only, unused
86     xi.bitmap_bit_order = LSBFirst; //XY only, unused
87 root 1.48 xi.bitmap_pad = BitmapPad (dpy);
88 root 1.47 xi.depth = depth;
89     xi.bytes_per_line = 0;
90 root 1.51 xi.bits_per_pixel = 32; //Z only
91 root 1.52 xi.red_mask = 0x00000000; //Z only, unused
92     xi.green_mask = 0x00000000; //Z only, unused
93     xi.blue_mask = 0x00000000; //Z only, unused
94     xi.obdata = 0; // probably unused
95 root 1.47
96 root 1.54 bool byte_order_mismatch = byte_order != xi.byte_order;
97    
98 root 1.47 if (!XInitImage (&xi))
99     rxvt_fatal ("unable to initialise ximage, please report.\n");
100    
101     if (height > INT_MAX / xi.bytes_per_line)
102     rxvt_fatal ("rxvt_img::new_from_pixbuf: image too big for Xlib.\n");
103    
104     xi.data = (char *)rxvt_malloc (height * xi.bytes_per_line);
105    
106     int rowstride = gdk_pixbuf_get_rowstride (pb);
107    
108     assert (3 + (depth == 32) == gdk_pixbuf_get_n_channels (pb));
109     unsigned char *row = gdk_pixbuf_get_pixels (pb);
110     char *line = xi.data;
111    
112     for (int y = 0; y < height; y++)
113     {
114 root 1.50 unsigned char *src = row;
115     uint32_t *dst = (uint32_t *)line;
116 root 1.47
117     if (depth == 24)
118     for (int x = 0; x < width; x++)
119     {
120 root 1.50 uint8_t r = *src++;
121     uint8_t g = *src++;
122     uint8_t b = *src++;
123    
124 root 1.53 uint32_t v = (r << 16) | (g << 8) | b;
125 root 1.50
126     if (ecb_big_endian ())
127     v = ecb_bswap32 (v);
128    
129 root 1.54 if (byte_order_mismatch)
130     v = ecb_bswap32 (v);
131    
132 root 1.51 *dst++ = v;
133 root 1.47 }
134     else
135     for (int x = 0; x < width; x++)
136     {
137 root 1.50 uint32_t v = *(uint32_t *)src; src += 4;
138 root 1.51
139 root 1.52 if (ecb_little_endian ())
140 root 1.51 v = ecb_bswap32 (v);
141    
142 root 1.52 v = ecb_rotr32 (v, 8);
143    
144 root 1.54 if (byte_order_mismatch)
145     v = ecb_bswap32 (v);
146    
147 root 1.51 *dst++ = v;
148 root 1.47 }
149    
150     row += rowstride;
151 root 1.50 line += xi.bytes_per_line;
152 root 1.47 }
153    
154     rxvt_img *img = new rxvt_img (s, XRenderFindStandardFormat (dpy, depth == 24 ? PictStandardRGB24 : PictStandardARGB32), 0, 0, width, height);
155     img->alloc ();
156    
157     GC gc = XCreateGC (dpy, img->pm, 0, 0);
158     XPutImage (dpy, img->pm, gc, &xi, 0, 0, 0, 0, width, height);
159     XFreeGC (dpy, gc);
160    
161     free (xi.data);
162    
163     return img;
164     }
165    
166     rxvt_img *
167 root 1.4 rxvt_img::new_from_file (rxvt_screen *s, const char *filename)
168     {
169 root 1.11 GError *err = 0;
170 root 1.4 GdkPixbuf *pb = gdk_pixbuf_new_from_file (filename, &err);
171    
172     if (!pb)
173 root 1.11 rxvt_fatal ("rxvt_img::new_from_file: %s\n", err->message);
174 root 1.4
175 root 1.47 rxvt_img *img = new_from_pixbuf (s, pb);
176 root 1.4
177 sf-exg 1.26 g_object_unref (pb);
178    
179 root 1.4 return img;
180     }
181    
182 root 1.32 void
183     rxvt_img::destroy ()
184     {
185 root 1.34 if (--ref->cnt)
186 root 1.32 return;
187    
188 root 1.34 if (pm && ref->ours)
189 root 1.32 XFreePixmap (s->display->dpy, pm);
190    
191 root 1.34 delete ref;
192 root 1.32 }
193    
194 root 1.1 rxvt_img::~rxvt_img ()
195     {
196 root 1.32 destroy ();
197     }
198    
199     void
200     rxvt_img::alloc ()
201     {
202     pm = XCreatePixmap (s->display->dpy, s->display->root, w, h, format->depth);
203 root 1.34 ref = new pixref (w, h);
204     }
205    
206     Picture
207     rxvt_img::src_picture ()
208     {
209     Display *dpy = s->display->dpy;
210    
211     XRenderPictureAttributes pa;
212     pa.repeat = repeat;
213     Picture pic = XRenderCreatePicture (dpy, pm, format, CPRepeat, &pa);
214    
215     return pic;
216 root 1.1 }
217    
218     void
219 root 1.11 rxvt_img::unshare ()
220     {
221 root 1.34 if (ref->cnt == 1 && ref->ours)
222 root 1.11 return;
223    
224 root 1.34 //TODO: maybe should reify instead
225     Pixmap pm2 = XCreatePixmap (s->display->dpy, s->display->root, ref->w, ref->h, format->depth);
226 root 1.32 GC gc = XCreateGC (s->display->dpy, pm, 0, 0);
227 root 1.34 XCopyArea (s->display->dpy, pm, pm2, gc, 0, 0, ref->w, ref->h, 0, 0);
228 root 1.32 XFreeGC (s->display->dpy, gc);
229 root 1.11
230 root 1.32 destroy ();
231 root 1.11
232 root 1.32 pm = pm2;
233 root 1.34 ref = new pixref (ref->w, ref->h);
234 root 1.11 }
235    
236     void
237 root 1.1 rxvt_img::fill (const rxvt_color &c)
238     {
239     XGCValues gcv;
240     gcv.foreground = c;
241     GC gc = XCreateGC (s->display->dpy, pm, GCForeground, &gcv);
242     XFillRectangle (s->display->dpy, pm, gc, 0, 0, w, h);
243 root 1.3 XFreeGC (s->display->dpy, gc);
244 root 1.1 }
245    
246 sf-exg 1.6 static void
247     get_gaussian_kernel (int radius, int width, double *kernel, XFixed *params)
248     {
249     double sigma = radius / 2.0;
250     double scale = sqrt (2.0 * M_PI) * sigma;
251     double sum = 0.0;
252    
253     for (int i = 0; i < width; i++)
254     {
255     double x = i - width / 2;
256     kernel[i] = exp (-(x * x) / (2.0 * sigma * sigma)) / scale;
257     sum += kernel[i];
258     }
259    
260     params[0] = XDoubleToFixed (width);
261     params[1] = XDoubleToFixed (1);
262    
263     for (int i = 0; i < width; i++)
264     params[i+2] = XDoubleToFixed (kernel[i] / sum);
265     }
266    
267 sf-exg 1.22 rxvt_img *
268 root 1.3 rxvt_img::blur (int rh, int rv)
269     {
270 sf-exg 1.6 if (!(s->display->flags & DISPLAY_HAS_RENDER_CONV))
271 sf-exg 1.22 return clone ();
272 sf-exg 1.6
273     Display *dpy = s->display->dpy;
274     int size = max (rh, rv) * 2 + 1;
275     double *kernel = (double *)malloc (size * sizeof (double));
276     XFixed *params = (XFixed *)malloc ((size + 2) * sizeof (XFixed));
277 root 1.41 rxvt_img *img = new rxvt_img (s, format, x, y, w, h, repeat);
278 root 1.32 img->alloc ();
279 sf-exg 1.6
280 root 1.34 Picture src = src_picture ();
281    
282 sf-exg 1.6 XRenderPictureAttributes pa;
283     pa.repeat = RepeatPad;
284 sf-exg 1.22 Picture dst = XRenderCreatePicture (dpy, img->pm, format, CPRepeat, &pa);
285    
286     Pixmap tmp_pm = XCreatePixmap (dpy, pm, w, h, format->depth);
287     Picture tmp = XRenderCreatePicture (dpy, tmp_pm , format, CPRepeat, &pa);
288     XFreePixmap (dpy, tmp_pm);
289 sf-exg 1.6
290     if (kernel && params)
291     {
292     size = rh * 2 + 1;
293     get_gaussian_kernel (rh, size, kernel, params);
294    
295     XRenderSetPictureFilter (dpy, src, FilterConvolution, params, size+2);
296     XRenderComposite (dpy,
297     PictOpSrc,
298     src,
299     None,
300 sf-exg 1.22 tmp,
301 sf-exg 1.6 0, 0,
302     0, 0,
303     0, 0,
304     w, h);
305    
306     size = rv * 2 + 1;
307     get_gaussian_kernel (rv, size, kernel, params);
308     ::swap (params[0], params[1]);
309    
310     XRenderSetPictureFilter (dpy, src, FilterConvolution, params, size+2);
311     XRenderComposite (dpy,
312     PictOpSrc,
313 sf-exg 1.22 tmp,
314 sf-exg 1.6 None,
315     dst,
316     0, 0,
317     0, 0,
318     0, 0,
319     w, h);
320     }
321    
322     free (kernel);
323     free (params);
324     XRenderFreePicture (dpy, src);
325     XRenderFreePicture (dpy, dst);
326 sf-exg 1.22 XRenderFreePicture (dpy, tmp);
327    
328     return img;
329 sf-exg 1.6 }
330    
331     static Picture
332     create_xrender_mask (Display *dpy, Drawable drawable, Bool argb)
333     {
334     Pixmap pixmap = XCreatePixmap (dpy, drawable, 1, 1, argb ? 32 : 8);
335    
336     XRenderPictFormat *format = XRenderFindStandardFormat (dpy, argb ? PictStandardARGB32 : PictStandardA8);
337     XRenderPictureAttributes pa;
338     pa.repeat = True;
339     Picture mask = XRenderCreatePicture (dpy, pixmap, format, CPRepeat, &pa);
340    
341     XFreePixmap (dpy, pixmap);
342    
343     return mask;
344 root 1.3 }
345    
346     void
347 sf-exg 1.31 rxvt_img::brightness (unsigned short r, unsigned short g, unsigned short b, unsigned short a)
348 root 1.3 {
349 sf-exg 1.6 Display *dpy = s->display->dpy;
350     Picture src = create_xrender_mask (dpy, pm, True);
351     Picture dst = XRenderCreatePicture (dpy, pm, format, 0, 0);
352    
353     XRenderColor mask_c;
354 sf-exg 1.31 mask_c.red = r;
355     mask_c.green = g;
356     mask_c.blue = b;
357     mask_c.alpha = a;
358 sf-exg 1.6 XRenderFillRectangle (dpy, PictOpSrc, src, &mask_c, 0, 0, 1, 1);
359    
360     XRenderComposite (dpy, PictOpAdd, src, None, dst, 0, 0, 0, 0, 0, 0, w, h);
361 sf-exg 1.23
362     XRenderFreePicture (dpy, src);
363     XRenderFreePicture (dpy, dst);
364 root 1.3 }
365    
366     void
367 sf-exg 1.31 rxvt_img::contrast (unsigned short r, unsigned short g, unsigned short b, unsigned short a)
368 root 1.1 {
369 sf-exg 1.6 if (!(s->display->flags & DISPLAY_HAS_RENDER_MUL))
370     return;
371    
372     Display *dpy = s->display->dpy;
373     Picture src = create_xrender_mask (dpy, pm, True);
374     Picture dst = XRenderCreatePicture (dpy, pm, format, 0, 0);
375    
376     XRenderColor mask_c;
377 sf-exg 1.31 mask_c.red = r;
378     mask_c.green = g;
379     mask_c.blue = b;
380     mask_c.alpha = a;
381 sf-exg 1.6 XRenderFillRectangle (dpy, PictOpSrc, src, &mask_c, 0, 0, 1, 1);
382    
383     XRenderComposite (dpy, PictOpMultiply, src, None, dst, 0, 0, 0, 0, 0, 0, w, h);
384 sf-exg 1.23
385     XRenderFreePicture (dpy, src);
386     XRenderFreePicture (dpy, dst);
387 root 1.1 }
388    
389 root 1.3 rxvt_img *
390 root 1.11 rxvt_img::clone ()
391 root 1.3 {
392 root 1.32 return new rxvt_img (*this);
393 root 1.3 }
394    
395 root 1.46 static XRenderPictFormat *
396     find_alpha_format_for (Display *dpy, XRenderPictFormat *format)
397     {
398     if (format->direct.alphaMask)
399     return format; // already has alpha
400    
401     // try to find a suitable alpha format, one bit alpha is enough for our purposes
402     if (format->type == PictTypeDirect)
403     for (int n = 0; XRenderPictFormat *f = XRenderFindFormat (dpy, 0, 0, n); ++n)
404     if (f->direct.alphaMask
405     && f->type == PictTypeDirect
406     && ecb_popcount32 (f->direct.redMask ) >= ecb_popcount32 (format->direct.redMask )
407     && ecb_popcount32 (f->direct.greenMask) >= ecb_popcount32 (format->direct.greenMask)
408     && ecb_popcount32 (f->direct.blueMask ) >= ecb_popcount32 (format->direct.blueMask ))
409     return f;
410    
411     // should be a very good fallback
412     return XRenderFindStandardFormat (dpy, PictStandardARGB32);
413     }
414    
415 root 1.3 rxvt_img *
416 root 1.33 rxvt_img::reify ()
417 root 1.19 {
418 root 1.35 if (x == 0 && y == 0 && w == ref->w && h == ref->h)
419     return clone ();
420    
421 root 1.40 Display *dpy = s->display->dpy;
422    
423     bool alpha = !format->direct.alphaMask
424     && (x || y)
425     && repeat == RepeatNone;
426    
427 root 1.46 rxvt_img *img = new rxvt_img (s, alpha ? find_alpha_format_for (dpy, format) : format, 0, 0, w, h, repeat);
428 root 1.32 img->alloc ();
429 root 1.19
430 root 1.34 Picture src = src_picture ();
431     Picture dst = XRenderCreatePicture (dpy, img->pm, img->format, 0, 0);
432 root 1.33
433 root 1.42 if (alpha)
434     {
435     XRenderColor rc = { 0, 0, 0, 0 };
436 root 1.44 XRenderFillRectangle (dpy, PictOpSrc, dst, &rc, 0, 0, w, h);//TODO: split into four fillrectangles
437 root 1.42 XRenderComposite (dpy, PictOpSrc, src, None, dst, 0, 0, 0, 0, -x, -y, ref->w, ref->h);
438     }
439     else
440     XRenderComposite (dpy, PictOpSrc, src, None, dst, x, y, 0, 0, 0, 0, w, h);
441    
442 root 1.33 XRenderFreePicture (dpy, src);
443     XRenderFreePicture (dpy, dst);
444 root 1.19
445 root 1.33 return img;
446     }
447 root 1.19
448 root 1.33 rxvt_img *
449     rxvt_img::sub_rect (int x, int y, int width, int height)
450     {
451     rxvt_img *img = clone ();
452    
453     img->x += x;
454     img->y += y;
455 root 1.19
456 root 1.37 if (w != width || h != height)
457     {
458     img->w = width;
459     img->h = height;
460    
461 root 1.39 rxvt_img *img2 = img->reify ();
462     delete img;
463     img = img2;
464 root 1.37 }
465 root 1.36
466 root 1.19 return img;
467     }
468    
469     rxvt_img *
470 root 1.30 rxvt_img::transform (int new_width, int new_height, double matrix[9])
471 root 1.3 {
472 root 1.41 rxvt_img *img = new rxvt_img (s, format, 0, 0, new_width, new_height, repeat);
473 root 1.32 img->alloc ();
474 root 1.12
475     Display *dpy = s->display->dpy;
476 root 1.34 Picture src = src_picture ();
477     Picture dst = XRenderCreatePicture (dpy, img->pm, img->format, 0, 0);
478 root 1.12
479     XTransform xfrm;
480    
481     for (int i = 0; i < 3; ++i)
482     for (int j = 0; j < 3; ++j)
483     xfrm.matrix [i][j] = XDoubleToFixed (matrix [i * 3 + j]);
484    
485 root 1.44 #if 0
486     xfrm.matrix [0][2] -= XDoubleToFixed (x);//TODO
487     xfrm.matrix [1][2] -= XDoubleToFixed (y);
488     #endif
489 root 1.33
490 root 1.17 XRenderSetPictureFilter (dpy, src, "good", 0, 0);
491 root 1.12 XRenderSetPictureTransform (dpy, src, &xfrm);
492     XRenderComposite (dpy, PictOpSrc, src, None, dst, 0, 0, 0, 0, 0, 0, new_width, new_height);
493    
494     XRenderFreePicture (dpy, src);
495     XRenderFreePicture (dpy, dst);
496    
497     return img;
498 root 1.3 }
499    
500     rxvt_img *
501     rxvt_img::scale (int new_width, int new_height)
502     {
503 sf-exg 1.43 if (w == new_width && h == new_height)
504     return clone ();
505    
506 root 1.11 double matrix[9] = {
507 root 1.16 w / (double)new_width, 0, 0,
508     0, h / (double)new_height, 0,
509 root 1.11 0, 0, 1
510     };
511    
512 root 1.42 int old_repeat_mode = repeat;
513     repeat = RepeatPad; // not right, but xrender can't proeprly scale it seems
514    
515     rxvt_img *img = transform (new_width, new_height, matrix);
516    
517     repeat = old_repeat_mode;
518     img->repeat = repeat;
519    
520     return img;
521 root 1.3 }
522    
523 sf-exg 1.7 rxvt_img *
524 root 1.30 rxvt_img::rotate (int new_width, int new_height, int x, int y, double phi)
525 root 1.17 {
526     double s = sin (phi);
527     double c = cos (phi);
528    
529     double matrix[9] = {
530     c, -s, -c * x + s * y + x,
531     s, c, -s * x - c * y + y,
532     0, 0, 1
533     };
534    
535 root 1.30 return transform (new_width, new_height, matrix);
536 root 1.17 }
537    
538     rxvt_img *
539 root 1.45 rxvt_img::convert_format (XRenderPictFormat *new_format, const rxvt_color &bg)
540 sf-exg 1.7 {
541 root 1.32 if (new_format == format)
542     return clone ();
543    
544 root 1.45 rxvt_img *img = new rxvt_img (s, new_format, x, y, w, h, repeat);
545 root 1.32 img->alloc ();
546 root 1.9
547 sf-exg 1.7 Display *dpy = s->display->dpy;
548 root 1.34 Picture src = src_picture ();
549 root 1.8 Picture dst = XRenderCreatePicture (dpy, img->pm, new_format, 0, 0);
550 root 1.27 int op = PictOpSrc;
551 sf-exg 1.7
552 root 1.29 if (format->direct.alphaMask && !new_format->direct.alphaMask)
553 root 1.27 {
554     // does it have to be that complicated
555     rgba c;
556     bg.get (c);
557    
558     XRenderColor rc = { c.r, c.g, c.b, 0xffff };
559 root 1.28 XRenderFillRectangle (dpy, PictOpSrc, dst, &rc, 0, 0, w, h);
560 root 1.27
561     op = PictOpOver;
562     }
563    
564     XRenderComposite (dpy, op, src, None, dst, 0, 0, 0, 0, 0, 0, w, h);
565 sf-exg 1.7
566     XRenderFreePicture (dpy, src);
567     XRenderFreePicture (dpy, dst);
568    
569     return img;
570     }
571 root 1.3
572 sf-exg 1.24 rxvt_img *
573     rxvt_img::blend (rxvt_img *img, double factor)
574     {
575     rxvt_img *img2 = clone ();
576     Display *dpy = s->display->dpy;
577 sf-exg 1.38 Picture src = img->src_picture ();
578 sf-exg 1.24 Picture dst = XRenderCreatePicture (dpy, img2->pm, img2->format, 0, 0);
579     Picture mask = create_xrender_mask (dpy, img->pm, False);
580    
581     XRenderColor mask_c;
582    
583     mask_c.alpha = float_to_component (factor);
584     mask_c.red =
585     mask_c.green =
586     mask_c.blue = 0;
587     XRenderFillRectangle (dpy, PictOpSrc, mask, &mask_c, 0, 0, 1, 1);
588    
589     XRenderComposite (dpy, PictOpOver, src, mask, dst, 0, 0, 0, 0, 0, 0, w, h);
590    
591     XRenderFreePicture (dpy, src);
592     XRenderFreePicture (dpy, dst);
593     XRenderFreePicture (dpy, mask);
594    
595     return img2;
596     }
597    
598 root 1.1 #endif
599