ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/rxvt-unicode/src/rxvtimg.C
Revision: 1.72
Committed: Sat Jun 9 16:21:16 2012 UTC (11 years, 11 months ago) by sf-exg
Content type: text/plain
Branch: MAIN
Changes since 1.71: +7 -16 lines
Log Message:
Simplify.

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 void
320 extract (int32_t cl0, int32_t cl1, int32_t &c, unsigned short &xc)
321 {
322 int32_t x = clamp (c, cl0, cl1);
323 c -= x;
324 xc = x;
325 }
326
327 static bool
328 extract (int32_t cl0, int32_t cl1, int32_t &r, int32_t &g, int32_t &b, int32_t &a, unsigned short &xr, unsigned short &xg, unsigned short &xb, unsigned short &xa)
329 {
330 extract (cl0, cl1, r, xr);
331 extract (cl0, cl1, g, xg);
332 extract (cl0, cl1, b, xb);
333 extract (cl0, cl1, a, xa);
334
335 return xr | xg | xb | xa;
336 }
337
338 void
339 rxvt_img::brightness (int32_t r, int32_t g, int32_t b, int32_t a)
340 {
341 unshare ();
342
343 Display *dpy = s->display->dpy;
344 Picture dst = XRenderCreatePicture (dpy, pm, format, 0, 0);
345
346 // loop should not be needed for brightness, as only -1..1 makes sense
347 //while (r | g | b | a)
348 {
349 unsigned short xr, xg, xb, xa;
350 XRenderColor mask_c;
351
352 if (extract (0, 65535, r, g, b, a, mask_c.red, mask_c.green, mask_c.blue, mask_c.alpha))
353 XRenderFillRectangle (dpy, PictOpAdd, dst, &mask_c, 0, 0, w, h);
354
355 if (extract (-65535, 0, r, g, b, a, mask_c.red, mask_c.green, mask_c.blue, mask_c.alpha))
356 {
357 XRenderColor mask_w = { 65535, 65535, 65535, 65535 };
358 XRenderFillRectangle (dpy, PictOpDifference, dst, &mask_w, 0, 0, w, h);
359 mask_c.red = -mask_c.red; //TODO: verify that doing clamp, assign, and negation does the right thing
360 mask_c.green = -mask_c.green;
361 mask_c.blue = -mask_c.blue;
362 mask_c.alpha = -mask_c.alpha;
363 XRenderFillRectangle (dpy, PictOpAdd, dst, &mask_c, 0, 0, w, h);
364 XRenderFillRectangle (dpy, PictOpDifference, dst, &mask_w, 0, 0, w, h);
365 }
366 }
367
368
369 XRenderFreePicture (dpy, dst);
370 }
371
372 void
373 rxvt_img::contrast (int32_t r, int32_t g, int32_t b, int32_t a)
374 {
375 if (!(s->display->flags & DISPLAY_HAS_RENDER_MUL))
376 {
377 rxvt_warn ("rxvt_img::contrast operation not supported on this display, RENDER extension too old.\n");
378 return;
379 }
380
381 unshare ();
382
383 Display *dpy = s->display->dpy;
384 Picture dst = XRenderCreatePicture (dpy, pm, format, 0, 0);
385
386 XRenderColor mask_c;
387 mask_c.red = r;
388 mask_c.green = g;
389 mask_c.blue = b;
390 mask_c.alpha = a;
391 XRenderFillRectangle (dpy, PictOpMultiply, dst, &mask_c, 0, 0, w, h);
392
393 XRenderFreePicture (dpy, dst);
394 }
395
396 rxvt_img *
397 rxvt_img::clone ()
398 {
399 return new rxvt_img (*this);
400 }
401
402 static XRenderPictFormat *
403 find_alpha_format_for (Display *dpy, XRenderPictFormat *format)
404 {
405 if (format->direct.alphaMask)
406 return format; // already has alpha
407
408 // try to find a suitable alpha format, one bit alpha is enough for our purposes
409 if (format->type == PictTypeDirect)
410 for (int n = 0; XRenderPictFormat *f = XRenderFindFormat (dpy, 0, 0, n); ++n)
411 if (f->direct.alphaMask
412 && f->type == PictTypeDirect
413 && ecb_popcount32 (f->direct.redMask ) >= ecb_popcount32 (format->direct.redMask )
414 && ecb_popcount32 (f->direct.greenMask) >= ecb_popcount32 (format->direct.greenMask)
415 && ecb_popcount32 (f->direct.blueMask ) >= ecb_popcount32 (format->direct.blueMask ))
416 return f;
417
418 // should be a very good fallback
419 return XRenderFindStandardFormat (dpy, PictStandardARGB32);
420 }
421
422 rxvt_img *
423 rxvt_img::reify ()
424 {
425 if (x == 0 && y == 0 && w == ref->w && h == ref->h)
426 return clone ();
427
428 Display *dpy = s->display->dpy;
429
430 // add an alpha channel if...
431 bool alpha = !format->direct.alphaMask // pixmap has none yet
432 && (x || y) // we need one because of non-zero offset
433 && repeat == RepeatNone; // and we have no good pixels to fill with
434
435 rxvt_img *img = new rxvt_img (s, alpha ? find_alpha_format_for (dpy, format) : format, 0, 0, w, h, repeat);
436 img->alloc ();
437
438 Picture src = src_picture ();
439 Picture dst = XRenderCreatePicture (dpy, img->pm, img->format, 0, 0);
440
441 if (alpha)
442 {
443 XRenderColor rc = { 0, 0, 0, 0 };
444 XRenderFillRectangle (dpy, PictOpSrc, dst, &rc, 0, 0, w, h);//TODO: split into four fillrectangles
445 XRenderComposite (dpy, PictOpSrc, src, None, dst, 0, 0, 0, 0, -x, -y, ref->w, ref->h);
446 }
447 else
448 XRenderComposite (dpy, PictOpSrc, src, None, dst, x, y, 0, 0, 0, 0, w, h);
449
450 XRenderFreePicture (dpy, src);
451 XRenderFreePicture (dpy, dst);
452
453 return img;
454 }
455
456 rxvt_img *
457 rxvt_img::sub_rect (int x, int y, int width, int height)
458 {
459 rxvt_img *img = clone ();
460
461 img->x += x;
462 img->y += y;
463
464 if (w != width || h != height)
465 {
466 img->w = width;
467 img->h = height;
468
469 rxvt_img *img2 = img->reify ();
470 delete img;
471 img = img2;
472 }
473
474 return img;
475 }
476
477 rxvt_img *
478 rxvt_img::transform (int new_width, int new_height, double matrix[9])
479 {
480 rxvt_img *img = new rxvt_img (s, format, 0, 0, new_width, new_height, repeat);
481 img->alloc ();
482
483 Display *dpy = s->display->dpy;
484 Picture src = src_picture ();
485 Picture dst = XRenderCreatePicture (dpy, img->pm, img->format, 0, 0);
486
487 XTransform xfrm;
488
489 for (int i = 0; i < 3; ++i)
490 for (int j = 0; j < 3; ++j)
491 xfrm.matrix [i][j] = XDoubleToFixed (matrix [i * 3 + j]);
492
493 xfrm.matrix [0][2] -= XDoubleToFixed (x);//TODO
494 xfrm.matrix [1][2] -= XDoubleToFixed (y);
495
496 XRenderSetPictureFilter (dpy, src, "good", 0, 0);
497 XRenderSetPictureTransform (dpy, src, &xfrm);
498 XRenderComposite (dpy, PictOpSrc, src, None, dst, 0, 0, 0, 0, 0, 0, new_width, new_height);
499
500 XRenderFreePicture (dpy, src);
501 XRenderFreePicture (dpy, dst);
502
503 return img;
504 }
505
506 rxvt_img *
507 rxvt_img::scale (int new_width, int new_height)
508 {
509 if (w == new_width && h == new_height)
510 return clone ();
511
512 double matrix[9] = {
513 w / (double)new_width, 0, 0,
514 0, h / (double)new_height, 0,
515 0, 0, 1
516 };
517
518 int old_repeat_mode = repeat;
519 repeat = RepeatPad; // not right, but xrender can't proeprly scale it seems
520
521 rxvt_img *img = transform (new_width, new_height, matrix);
522
523 repeat = old_repeat_mode;
524 img->repeat = repeat;
525
526 return img;
527 }
528
529 rxvt_img *
530 rxvt_img::rotate (int new_width, int new_height, int x, int y, double phi)
531 {
532 double s = sin (phi);
533 double c = cos (phi);
534
535 double matrix[9] = {
536 c, -s, -c * x + s * y + x,
537 s, c, -s * x - c * y + y,
538 0, 0, 1
539 };
540
541 return transform (new_width, new_height, matrix);
542 }
543
544 rxvt_img *
545 rxvt_img::convert_format (XRenderPictFormat *new_format, const rxvt_color &bg)
546 {
547 if (new_format == format)
548 return clone ();
549
550 rxvt_img *img = new rxvt_img (s, new_format, x, y, w, h, repeat);
551 img->alloc ();
552
553 Display *dpy = s->display->dpy;
554 Picture src = src_picture ();
555 Picture dst = XRenderCreatePicture (dpy, img->pm, new_format, 0, 0);
556 int op = PictOpSrc;
557
558 if (format->direct.alphaMask && !new_format->direct.alphaMask)
559 {
560 // does it have to be that complicated
561 rgba c;
562 bg.get (c);
563
564 XRenderColor rc = { c.r, c.g, c.b, 0xffff };
565 XRenderFillRectangle (dpy, PictOpSrc, dst, &rc, 0, 0, w, h);
566
567 op = PictOpOver;
568 }
569
570 XRenderComposite (dpy, op, src, None, dst, 0, 0, 0, 0, 0, 0, w, h);
571
572 XRenderFreePicture (dpy, src);
573 XRenderFreePicture (dpy, dst);
574
575 return img;
576 }
577
578 rxvt_img *
579 rxvt_img::blend (rxvt_img *img, double factor)
580 {
581 rxvt_img *img2 = clone ();
582 Display *dpy = s->display->dpy;
583 Picture src = img->src_picture ();
584 Picture dst = XRenderCreatePicture (dpy, img2->pm, img2->format, 0, 0);
585
586 Pixmap pixmap = XCreatePixmap (dpy, img->pm, 1, 1, 8);
587 XRenderPictFormat *format = XRenderFindStandardFormat (dpy, PictStandardA8);
588 XRenderPictureAttributes pa;
589 pa.repeat = True;
590 Picture mask = XRenderCreatePicture (dpy, pixmap, format, CPRepeat, &pa);
591 XFreePixmap (dpy, pixmap);
592
593 XRenderColor mask_c;
594
595 mask_c.alpha = float_to_component (factor);
596 mask_c.red =
597 mask_c.green =
598 mask_c.blue = 0;
599 XRenderFillRectangle (dpy, PictOpSrc, mask, &mask_c, 0, 0, 1, 1);
600
601 XRenderComposite (dpy, PictOpOver, src, mask, dst, 0, 0, 0, 0, 0, 0, w, h);
602
603 XRenderFreePicture (dpy, src);
604 XRenderFreePicture (dpy, dst);
605 XRenderFreePicture (dpy, mask);
606
607 return img2;
608 }
609
610 #endif
611