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

# Content
1 #include <math.h>
2 #include "../config.h"
3 #include "rxvt.h"
4
5 #if HAVE_IMG
6
7 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 pm(0), ref(0)
10 {
11 }
12
13 rxvt_img::rxvt_img (const rxvt_img &img)
14 : 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 {
16 ++ref->cnt;
17 }
18
19 #if 0
20 rxvt_img::rxvt_img (rxvt_screen *screen, XRenderPictFormat *format, int width, int height, Pixmap pixmap)
21 : s(screen), x(0), y(0), w(width), h(height), format(format), repeat(RepeatNormal), shared(false), pm(pixmap)
22 {
23 }
24 #endif
25
26 rxvt_img *
27 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 0,
49 0,
50 root_pm_w,
51 root_pm_h
52 );
53
54 img->pm = root_pixmap;
55 img->ref = new pixref (root_pm_w, root_pm_h);
56 img->ref->ours = false;
57
58 return img;
59 }
60
61 rxvt_img *
62 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 rxvt_img::new_from_file (rxvt_screen *s, const char *filename)
153 {
154 GError *err = 0;
155 GdkPixbuf *pb = gdk_pixbuf_new_from_file (filename, &err);
156
157 if (!pb)
158 rxvt_fatal ("rxvt_img::new_from_file: %s\n", err->message);
159
160 rxvt_img *img = new_from_pixbuf (s, pb);
161
162 g_object_unref (pb);
163
164 return img;
165 }
166
167 void
168 rxvt_img::destroy ()
169 {
170 if (--ref->cnt)
171 return;
172
173 if (pm && ref->ours)
174 XFreePixmap (s->display->dpy, pm);
175
176 delete ref;
177 }
178
179 rxvt_img::~rxvt_img ()
180 {
181 destroy ();
182 }
183
184 void
185 rxvt_img::alloc ()
186 {
187 pm = XCreatePixmap (s->display->dpy, s->display->root, w, h, format->depth);
188 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 }
202
203 void
204 rxvt_img::unshare ()
205 {
206 if (ref->cnt == 1 && ref->ours)
207 return;
208
209 //TODO: maybe should reify instead
210 Pixmap pm2 = XCreatePixmap (s->display->dpy, s->display->root, ref->w, ref->h, format->depth);
211 GC gc = XCreateGC (s->display->dpy, pm, 0, 0);
212 XCopyArea (s->display->dpy, pm, pm2, gc, 0, 0, ref->w, ref->h, 0, 0);
213 XFreeGC (s->display->dpy, gc);
214
215 destroy ();
216
217 pm = pm2;
218 ref = new pixref (ref->w, ref->h);
219 }
220
221 void
222 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 XFreeGC (s->display->dpy, gc);
229 }
230
231 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 rxvt_img *
253 rxvt_img::blur (int rh, int rv)
254 {
255 if (!(s->display->flags & DISPLAY_HAS_RENDER_CONV))
256 return clone ();
257
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 rxvt_img *img = new rxvt_img (s, format, x, y, w, h, repeat);
263 img->alloc ();
264
265 Picture src = src_picture ();
266
267 XRenderPictureAttributes pa;
268 pa.repeat = RepeatPad;
269 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
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 tmp,
286 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 tmp,
299 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 XRenderFreePicture (dpy, tmp);
312
313 return img;
314 }
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 }
330
331 void
332 rxvt_img::brightness (unsigned short r, unsigned short g, unsigned short b, unsigned short a)
333 {
334 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 mask_c.red = r;
340 mask_c.green = g;
341 mask_c.blue = b;
342 mask_c.alpha = a;
343 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
347 XRenderFreePicture (dpy, src);
348 XRenderFreePicture (dpy, dst);
349 }
350
351 void
352 rxvt_img::contrast (unsigned short r, unsigned short g, unsigned short b, unsigned short a)
353 {
354 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 mask_c.red = r;
363 mask_c.green = g;
364 mask_c.blue = b;
365 mask_c.alpha = a;
366 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
370 XRenderFreePicture (dpy, src);
371 XRenderFreePicture (dpy, dst);
372 }
373
374 rxvt_img *
375 rxvt_img::clone ()
376 {
377 return new rxvt_img (*this);
378 }
379
380 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 rxvt_img *
401 rxvt_img::reify ()
402 {
403 if (x == 0 && y == 0 && w == ref->w && h == ref->h)
404 return clone ();
405
406 Display *dpy = s->display->dpy;
407
408 bool alpha = !format->direct.alphaMask
409 && (x || y)
410 && repeat == RepeatNone;
411
412 rxvt_img *img = new rxvt_img (s, alpha ? find_alpha_format_for (dpy, format) : format, 0, 0, w, h, repeat);
413 img->alloc ();
414
415 Picture src = src_picture ();
416 Picture dst = XRenderCreatePicture (dpy, img->pm, img->format, 0, 0);
417
418 if (alpha)
419 {
420 XRenderColor rc = { 0, 0, 0, 0 };
421 XRenderFillRectangle (dpy, PictOpSrc, dst, &rc, 0, 0, w, h);//TODO: split into four fillrectangles
422 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 XRenderFreePicture (dpy, src);
428 XRenderFreePicture (dpy, dst);
429
430 return img;
431 }
432
433 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
441 if (w != width || h != height)
442 {
443 img->w = width;
444 img->h = height;
445
446 rxvt_img *img2 = img->reify ();
447 delete img;
448 img = img2;
449 }
450
451 return img;
452 }
453
454 rxvt_img *
455 rxvt_img::transform (int new_width, int new_height, double matrix[9])
456 {
457 rxvt_img *img = new rxvt_img (s, format, 0, 0, new_width, new_height, repeat);
458 img->alloc ();
459
460 Display *dpy = s->display->dpy;
461 Picture src = src_picture ();
462 Picture dst = XRenderCreatePicture (dpy, img->pm, img->format, 0, 0);
463
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 #if 0
471 xfrm.matrix [0][2] -= XDoubleToFixed (x);//TODO
472 xfrm.matrix [1][2] -= XDoubleToFixed (y);
473 #endif
474
475 XRenderSetPictureFilter (dpy, src, "good", 0, 0);
476 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 }
484
485 rxvt_img *
486 rxvt_img::scale (int new_width, int new_height)
487 {
488 if (w == new_width && h == new_height)
489 return clone ();
490
491 double matrix[9] = {
492 w / (double)new_width, 0, 0,
493 0, h / (double)new_height, 0,
494 0, 0, 1
495 };
496
497 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 }
507
508 rxvt_img *
509 rxvt_img::rotate (int new_width, int new_height, int x, int y, double phi)
510 {
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 return transform (new_width, new_height, matrix);
521 }
522
523 rxvt_img *
524 rxvt_img::convert_format (XRenderPictFormat *new_format, const rxvt_color &bg)
525 {
526 if (new_format == format)
527 return clone ();
528
529 rxvt_img *img = new rxvt_img (s, new_format, x, y, w, h, repeat);
530 img->alloc ();
531
532 Display *dpy = s->display->dpy;
533 Picture src = src_picture ();
534 Picture dst = XRenderCreatePicture (dpy, img->pm, new_format, 0, 0);
535 int op = PictOpSrc;
536
537 if (format->direct.alphaMask && !new_format->direct.alphaMask)
538 {
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 XRenderFillRectangle (dpy, PictOpSrc, dst, &rc, 0, 0, w, h);
545
546 op = PictOpOver;
547 }
548
549 XRenderComposite (dpy, op, src, None, dst, 0, 0, 0, 0, 0, 0, w, h);
550
551 XRenderFreePicture (dpy, src);
552 XRenderFreePicture (dpy, dst);
553
554 return img;
555 }
556
557 rxvt_img *
558 rxvt_img::blend (rxvt_img *img, double factor)
559 {
560 rxvt_img *img2 = clone ();
561 Display *dpy = s->display->dpy;
562 Picture src = img->src_picture ();
563 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 #endif
584