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