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