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