ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/cvsroot/rxvt-unicode/src/rxvtimg.C
Revision: 1.21
Committed: Tue Jun 5 14:59:44 2012 UTC (12 years, 1 month ago) by sf-exg
Content type: text/plain
Branch: MAIN
Changes since 1.20: +32 -0 lines
Log Message:
Implement rxvt_img::new_from_root.

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 sf-exg 1.6 #define float_to_component(d) ((d) * 65535.99)
8    
9 root 1.3 rxvt_img::rxvt_img (rxvt_screen *screen, XRenderPictFormat *format, int width, int height)
10 root 1.4 : s(screen), w(width), h(height), format(format), shared(false)
11 root 1.1 {
12 root 1.3 pm = XCreatePixmap (s->display->dpy, s->display->root, w, h, format->depth);
13 root 1.1 }
14    
15 root 1.3 rxvt_img::rxvt_img (rxvt_screen *screen, XRenderPictFormat *format, int width, int height, Pixmap pixmap)
16 root 1.4 : s(screen), pm(pixmap), w(width), h(height), format(format), shared(false)
17 root 1.1 {
18     }
19    
20 root 1.4 rxvt_img *
21 sf-exg 1.21 rxvt_img::new_from_root (rxvt_screen *s)
22     {
23     Display *dpy = s->display->dpy;
24     unsigned int root_pm_w, root_pm_h;
25     Pixmap root_pixmap = s->display->get_pixmap_property (s->display->xa[XA_XROOTPMAP_ID]);
26     if (root_pixmap == None)
27     root_pixmap = s->display->get_pixmap_property (s->display->xa[XA_ESETROOT_PMAP_ID]);
28    
29     if (root_pixmap == None)
30     return 0;
31    
32     Window wdummy;
33     int idummy;
34     unsigned int udummy;
35    
36     if (!XGetGeometry (dpy, root_pixmap, &wdummy, &idummy, &idummy, &root_pm_w, &root_pm_h, &udummy, &udummy))
37     return 0;
38    
39     rxvt_img *img = new rxvt_img (
40     s,
41     XRenderFindVisualFormat (dpy, DefaultVisual (dpy, s->display->screen)),
42     root_pm_w,
43     root_pm_h,
44     root_pixmap
45     );
46    
47     img->shared = true;
48    
49     return img;
50     }
51    
52     rxvt_img *
53 root 1.4 rxvt_img::new_from_file (rxvt_screen *s, const char *filename)
54     {
55 root 1.11 GError *err = 0;
56 root 1.4 GdkPixbuf *pb = gdk_pixbuf_new_from_file (filename, &err);
57    
58     if (!pb)
59 root 1.11 rxvt_fatal ("rxvt_img::new_from_file: %s\n", err->message);
60 root 1.4
61     rxvt_img *img = new rxvt_img (
62     s,
63     XRenderFindStandardFormat (s->display->dpy, gdk_pixbuf_get_has_alpha (pb) ? PictStandardARGB32 : PictStandardRGB24),
64     gdk_pixbuf_get_width (pb),
65     gdk_pixbuf_get_height (pb)
66     );
67    
68 root 1.14 img->render_pixbuf (pb, 0, 0, img->w, img->h, 0, 0);
69 root 1.4
70     return img;
71     }
72    
73 root 1.1 rxvt_img::~rxvt_img ()
74     {
75 root 1.4 if (!shared)
76 root 1.1 XFreePixmap (s->display->dpy, pm);
77     }
78    
79     void
80 root 1.11 rxvt_img::unshare ()
81     {
82     if (!shared)
83     return;
84    
85     rxvt_img *img = clone ();
86    
87     ::swap (pm , img->pm);
88     ::swap (shared, img->shared);
89    
90     delete img;
91     }
92    
93     void
94 root 1.1 rxvt_img::fill (const rxvt_color &c)
95     {
96     XGCValues gcv;
97     gcv.foreground = c;
98     GC gc = XCreateGC (s->display->dpy, pm, GCForeground, &gcv);
99     XFillRectangle (s->display->dpy, pm, gc, 0, 0, w, h);
100 root 1.3 XFreeGC (s->display->dpy, gc);
101 root 1.1 }
102    
103 sf-exg 1.6 static void
104     get_gaussian_kernel (int radius, int width, double *kernel, XFixed *params)
105     {
106     double sigma = radius / 2.0;
107     double scale = sqrt (2.0 * M_PI) * sigma;
108     double sum = 0.0;
109    
110     for (int i = 0; i < width; i++)
111     {
112     double x = i - width / 2;
113     kernel[i] = exp (-(x * x) / (2.0 * sigma * sigma)) / scale;
114     sum += kernel[i];
115     }
116    
117     params[0] = XDoubleToFixed (width);
118     params[1] = XDoubleToFixed (1);
119    
120     for (int i = 0; i < width; i++)
121     params[i+2] = XDoubleToFixed (kernel[i] / sum);
122     }
123    
124 root 1.3 void
125     rxvt_img::blur (int rh, int rv)
126     {
127 sf-exg 1.6 if (!(s->display->flags & DISPLAY_HAS_RENDER_CONV))
128     return;
129    
130     Display *dpy = s->display->dpy;
131     int size = max (rh, rv) * 2 + 1;
132     double *kernel = (double *)malloc (size * sizeof (double));
133     XFixed *params = (XFixed *)malloc ((size + 2) * sizeof (XFixed));
134    
135     XRenderPictureAttributes pa;
136    
137     pa.repeat = RepeatPad;
138 root 1.17 Picture src = XRenderCreatePicture (dpy, pm , format, CPRepeat, &pa);
139 sf-exg 1.6 Pixmap tmp = XCreatePixmap (dpy, pm, w, h, format->depth);
140     Picture dst = XRenderCreatePicture (dpy, tmp, format, CPRepeat, &pa);
141     XFreePixmap (dpy, tmp);
142    
143     if (kernel && params)
144     {
145     size = rh * 2 + 1;
146     get_gaussian_kernel (rh, size, kernel, params);
147    
148     XRenderSetPictureFilter (dpy, src, FilterConvolution, params, size+2);
149     XRenderComposite (dpy,
150     PictOpSrc,
151     src,
152     None,
153     dst,
154     0, 0,
155     0, 0,
156     0, 0,
157     w, h);
158    
159     ::swap (src, dst);
160    
161     size = rv * 2 + 1;
162     get_gaussian_kernel (rv, size, kernel, params);
163     ::swap (params[0], params[1]);
164    
165     XRenderSetPictureFilter (dpy, src, FilterConvolution, params, size+2);
166     XRenderComposite (dpy,
167     PictOpSrc,
168     src,
169     None,
170     dst,
171     0, 0,
172     0, 0,
173     0, 0,
174     w, h);
175     }
176    
177     free (kernel);
178     free (params);
179     XRenderFreePicture (dpy, src);
180     XRenderFreePicture (dpy, dst);
181     }
182    
183     static Picture
184     create_xrender_mask (Display *dpy, Drawable drawable, Bool argb)
185     {
186     Pixmap pixmap = XCreatePixmap (dpy, drawable, 1, 1, argb ? 32 : 8);
187    
188     XRenderPictFormat *format = XRenderFindStandardFormat (dpy, argb ? PictStandardARGB32 : PictStandardA8);
189     XRenderPictureAttributes pa;
190     pa.repeat = True;
191     Picture mask = XRenderCreatePicture (dpy, pixmap, format, CPRepeat, &pa);
192    
193     XFreePixmap (dpy, pixmap);
194    
195     return mask;
196 root 1.3 }
197    
198     void
199     rxvt_img::brightness (double r, double g, double b, double a)
200     {
201 sf-exg 1.6 Display *dpy = s->display->dpy;
202     Picture src = create_xrender_mask (dpy, pm, True);
203     Picture dst = XRenderCreatePicture (dpy, pm, format, 0, 0);
204    
205     XRenderColor mask_c;
206     mask_c.red = float_to_component (r);
207     mask_c.green = float_to_component (g);
208     mask_c.blue = float_to_component (b);
209     mask_c.alpha = float_to_component (a);
210     XRenderFillRectangle (dpy, PictOpSrc, src, &mask_c, 0, 0, 1, 1);
211    
212     XRenderComposite (dpy, PictOpAdd, src, None, dst, 0, 0, 0, 0, 0, 0, w, h);
213 root 1.3 }
214    
215     void
216     rxvt_img::contrast (double r, double g, double b, double a)
217 root 1.1 {
218 sf-exg 1.6 if (!(s->display->flags & DISPLAY_HAS_RENDER_MUL))
219     return;
220    
221     Display *dpy = s->display->dpy;
222     Picture src = create_xrender_mask (dpy, pm, True);
223     Picture dst = XRenderCreatePicture (dpy, pm, format, 0, 0);
224    
225     XRenderColor mask_c;
226     mask_c.red = float_to_component (r);
227     mask_c.green = float_to_component (g);
228     mask_c.blue = float_to_component (b);
229     mask_c.alpha = float_to_component (a);
230     XRenderFillRectangle (dpy, PictOpSrc, src, &mask_c, 0, 0, 1, 1);
231    
232     XRenderComposite (dpy, PictOpMultiply, src, None, dst, 0, 0, 0, 0, 0, 0, w, h);
233 root 1.1 }
234    
235 root 1.14 bool
236     rxvt_img::render_pixbuf (GdkPixbuf *pixbuf, int src_x, int src_y, int width, int height, int dst_x, int dst_y)
237 root 1.3 {
238 root 1.14 Display *dpy = s->display->dpy;
239    
240     if (s->visual->c_class != TrueColor)
241     return false;
242    
243     uint32_t red_mask, green_mask, blue_mask, alpha_mask;
244    
245 sf-exg 1.18 red_mask = (uint32_t)format->direct.redMask << format->direct.red;
246     green_mask = (uint32_t)format->direct.greenMask << format->direct.green;
247     blue_mask = (uint32_t)format->direct.blueMask << format->direct.blue;
248     alpha_mask = (uint32_t)format->direct.alphaMask << format->direct.alpha;
249 root 1.14
250     int width_r = ecb_popcount32 (red_mask);
251     int width_g = ecb_popcount32 (green_mask);
252     int width_b = ecb_popcount32 (blue_mask);
253     int width_a = ecb_popcount32 (alpha_mask);
254    
255     if (width_r > 8 || width_g > 8 || width_b > 8 || width_a > 8)
256     return false;
257    
258     int sh_r = ecb_ctz32 (red_mask);
259     int sh_g = ecb_ctz32 (green_mask);
260     int sh_b = ecb_ctz32 (blue_mask);
261     int sh_a = ecb_ctz32 (alpha_mask);
262    
263     if (width > 32767 || height > 32767)
264     return false;
265    
266 sf-exg 1.18 XImage *ximage = XCreateImage (dpy, s->visual, format->depth, ZPixmap, 0, 0,
267 root 1.14 width, height, 32, 0);
268     if (!ximage)
269     return false;
270    
271     if (height > INT_MAX / ximage->bytes_per_line
272     || !(ximage->data = (char *)malloc (height * ximage->bytes_per_line)))
273     {
274     XDestroyImage (ximage);
275     return false;
276     }
277    
278     GC gc = XCreateGC (dpy, pm, 0, 0);
279    
280     ximage->byte_order = ecb_big_endian () ? MSBFirst : LSBFirst;
281    
282     int rowstride = gdk_pixbuf_get_rowstride (pixbuf);
283     int channels = gdk_pixbuf_get_n_channels (pixbuf);
284     unsigned char *row = gdk_pixbuf_get_pixels (pixbuf) + src_y * rowstride + src_x * channels;
285     char *line = ximage->data;
286    
287     rgba c (0, 0, 0);
288    
289     if (channels == 4 && alpha_mask == 0)
290     {
291     //pix_colors[Color_bg].get (c);
292     //TODO
293 root 1.15 c.r = 0xffff; c.g = 0xc0c0; c.b = 0xcbcb;//D
294 root 1.14 c.r >>= 8;
295     c.g >>= 8;
296     c.b >>= 8;
297     }
298    
299     for (int y = 0; y < height; y++)
300     {
301     for (int x = 0; x < width; x++)
302     {
303     unsigned char *pixel = row + x * channels;
304     uint32_t value;
305     unsigned char r, g, b, a;
306    
307     if (channels == 4)
308     {
309     a = pixel[3];
310     r = (pixel[0] * a + c.r * (0xff - a)) / 0xff;
311     g = (pixel[1] * a + c.g * (0xff - a)) / 0xff;
312     b = (pixel[2] * a + c.b * (0xff - a)) / 0xff;
313     }
314     else
315     {
316     a = 0xff;
317     r = pixel[0];
318     g = pixel[1];
319     b = pixel[2];
320     }
321    
322     value = ((r >> (8 - width_r)) << sh_r)
323     | ((g >> (8 - width_g)) << sh_g)
324     | ((b >> (8 - width_b)) << sh_b)
325     | ((a >> (8 - width_a)) << sh_a);
326    
327     if (ximage->bits_per_pixel == 32)
328     ((uint32_t *)line)[x] = value;
329     else
330     XPutPixel (ximage, x, y, value);
331     }
332    
333     row += rowstride;
334     line += ximage->bytes_per_line;
335     }
336    
337     XPutImage (dpy, pm, gc, ximage, 0, 0, dst_x, dst_y, width, height);
338     XDestroyImage (ximage);
339     XFreeGC (dpy, gc);
340    
341     return true;
342 root 1.3 }
343    
344     rxvt_img *
345 root 1.11 rxvt_img::clone ()
346 root 1.3 {
347 root 1.19 rxvt_img *img = new rxvt_img (s, format, w, h);
348    
349 root 1.3 GC gc = XCreateGC (s->display->dpy, pm, 0, 0);
350 root 1.19 XCopyArea (s->display->dpy, pm, img->pm, gc, 0, 0, w, h, 0, 0);
351 root 1.3 XFreeGC (s->display->dpy, gc);
352 root 1.20
353     return img;
354 root 1.3 }
355    
356     rxvt_img *
357 root 1.19 rxvt_img::sub_rect (int x, int y, int width, int height, int repeat)
358     {
359     rxvt_img *img = new rxvt_img (s, format, width, height);
360    
361     Display *dpy = s->display->dpy;
362     XRenderPictureAttributes pa;
363     pa.repeat = repeat;
364     Picture src = XRenderCreatePicture (dpy, pm, format, CPRepeat, &pa);
365     Picture dst = XRenderCreatePicture (dpy, img->pm, img->format, 0, 0);
366    
367     XRenderComposite (dpy, PictOpSrc, src, None, dst, x, y, 0, 0, 0, 0, width, height);
368    
369     XRenderFreePicture (dpy, src);
370     XRenderFreePicture (dpy, dst);
371    
372     return img;
373     }
374    
375     rxvt_img *
376     rxvt_img::transform (int new_width, int new_height, double matrix[9], int repeat)
377 root 1.3 {
378 root 1.12 rxvt_img *img = new rxvt_img (s, format, new_width, new_height);
379    
380     Display *dpy = s->display->dpy;
381 root 1.13 XRenderPictureAttributes pa;
382     pa.repeat = repeat;
383     Picture src = XRenderCreatePicture (dpy, pm, format, CPRepeat, &pa);
384     Picture dst = XRenderCreatePicture (dpy, img->pm, img->format, 0, 0);
385 root 1.12
386     XTransform xfrm;
387    
388     for (int i = 0; i < 3; ++i)
389     for (int j = 0; j < 3; ++j)
390     xfrm.matrix [i][j] = XDoubleToFixed (matrix [i * 3 + j]);
391    
392 root 1.17 XRenderSetPictureFilter (dpy, src, "good", 0, 0);
393 root 1.12 XRenderSetPictureTransform (dpy, src, &xfrm);
394     XRenderComposite (dpy, PictOpSrc, src, None, dst, 0, 0, 0, 0, 0, 0, new_width, new_height);
395    
396     XRenderFreePicture (dpy, src);
397     XRenderFreePicture (dpy, dst);
398    
399     return img;
400 root 1.3 }
401    
402     rxvt_img *
403     rxvt_img::scale (int new_width, int new_height)
404     {
405 root 1.11 double matrix[9] = {
406 root 1.16 w / (double)new_width, 0, 0,
407     0, h / (double)new_height, 0,
408 root 1.11 0, 0, 1
409     };
410    
411 root 1.19 return transform (new_width, new_height, matrix);
412 root 1.3 }
413    
414 sf-exg 1.7 rxvt_img *
415 root 1.19 rxvt_img::rotate (int new_width, int new_height, int x, int y, double phi, int repeat)
416 root 1.17 {
417     double s = sin (phi);
418     double c = cos (phi);
419    
420     double matrix[9] = {
421     c, -s, -c * x + s * y + x,
422     s, c, -s * x - c * y + y,
423     0, 0, 1
424     };
425    
426 root 1.19 return transform (new_width, new_height, matrix, repeat);
427 root 1.17 }
428    
429     rxvt_img *
430 sf-exg 1.7 rxvt_img::convert_to (XRenderPictFormat *new_format)
431     {
432 sf-exg 1.10 rxvt_img *img = new rxvt_img (s, new_format, w, h);
433 root 1.9
434 sf-exg 1.7 Display *dpy = s->display->dpy;
435 root 1.12 Picture src = XRenderCreatePicture (dpy, pm, format, 0, 0);
436 root 1.8 Picture dst = XRenderCreatePicture (dpy, img->pm, new_format, 0, 0);
437 sf-exg 1.7
438     XRenderComposite (dpy, PictOpSrc, src, None, dst, 0, 0, 0, 0, 0, 0, w, h);
439    
440     XRenderFreePicture (dpy, src);
441     XRenderFreePicture (dpy, dst);
442    
443     return img;
444     }
445 root 1.3
446 root 1.1 #endif
447