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