1 |
#include <math.h> |
2 |
#include "../config.h" |
3 |
#include "rxvt.h" |
4 |
|
5 |
#if HAVE_IMG |
6 |
|
7 |
#define float_to_component(d) ((d) * 65535.99) |
8 |
|
9 |
rxvt_img::rxvt_img (rxvt_screen *screen, XRenderPictFormat *format, int width, int height) |
10 |
: s(screen), w(width), h(height), format(format), shared(false) |
11 |
{ |
12 |
pm = XCreatePixmap (s->display->dpy, s->display->root, w, h, format->depth); |
13 |
} |
14 |
|
15 |
rxvt_img::rxvt_img (rxvt_screen *screen, XRenderPictFormat *format, int width, int height, Pixmap pixmap) |
16 |
: s(screen), pm(pixmap), w(width), h(height), format(format), shared(false) |
17 |
{ |
18 |
} |
19 |
|
20 |
rxvt_img * |
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 |
rxvt_img::new_from_file (rxvt_screen *s, const char *filename) |
54 |
{ |
55 |
GError *err = 0; |
56 |
GdkPixbuf *pb = gdk_pixbuf_new_from_file (filename, &err); |
57 |
|
58 |
if (!pb) |
59 |
rxvt_fatal ("rxvt_img::new_from_file: %s\n", err->message); |
60 |
|
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 |
img->render_pixbuf (pb, 0, 0, img->w, img->h, 0, 0); |
69 |
|
70 |
return img; |
71 |
} |
72 |
|
73 |
rxvt_img::~rxvt_img () |
74 |
{ |
75 |
if (!shared) |
76 |
XFreePixmap (s->display->dpy, pm); |
77 |
} |
78 |
|
79 |
void |
80 |
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 |
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 |
XFreeGC (s->display->dpy, gc); |
101 |
} |
102 |
|
103 |
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 |
rxvt_img * |
125 |
rxvt_img::blur (int rh, int rv) |
126 |
{ |
127 |
if (!(s->display->flags & DISPLAY_HAS_RENDER_CONV)) |
128 |
return clone (); |
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 |
rxvt_img *img = new rxvt_img (s, format, w, h); |
135 |
|
136 |
XRenderPictureAttributes pa; |
137 |
|
138 |
pa.repeat = RepeatPad; |
139 |
Picture src = XRenderCreatePicture (dpy, pm , format, CPRepeat, &pa); |
140 |
Picture dst = XRenderCreatePicture (dpy, img->pm, format, CPRepeat, &pa); |
141 |
|
142 |
Pixmap tmp_pm = XCreatePixmap (dpy, pm, w, h, format->depth); |
143 |
Picture tmp = XRenderCreatePicture (dpy, tmp_pm , format, CPRepeat, &pa); |
144 |
XFreePixmap (dpy, tmp_pm); |
145 |
|
146 |
if (kernel && params) |
147 |
{ |
148 |
size = rh * 2 + 1; |
149 |
get_gaussian_kernel (rh, size, kernel, params); |
150 |
|
151 |
XRenderSetPictureFilter (dpy, src, FilterConvolution, params, size+2); |
152 |
XRenderComposite (dpy, |
153 |
PictOpSrc, |
154 |
src, |
155 |
None, |
156 |
tmp, |
157 |
0, 0, |
158 |
0, 0, |
159 |
0, 0, |
160 |
w, h); |
161 |
|
162 |
size = rv * 2 + 1; |
163 |
get_gaussian_kernel (rv, size, kernel, params); |
164 |
::swap (params[0], params[1]); |
165 |
|
166 |
XRenderSetPictureFilter (dpy, src, FilterConvolution, params, size+2); |
167 |
XRenderComposite (dpy, |
168 |
PictOpSrc, |
169 |
tmp, |
170 |
None, |
171 |
dst, |
172 |
0, 0, |
173 |
0, 0, |
174 |
0, 0, |
175 |
w, h); |
176 |
} |
177 |
|
178 |
free (kernel); |
179 |
free (params); |
180 |
XRenderFreePicture (dpy, src); |
181 |
XRenderFreePicture (dpy, dst); |
182 |
XRenderFreePicture (dpy, tmp); |
183 |
|
184 |
return img; |
185 |
} |
186 |
|
187 |
static Picture |
188 |
create_xrender_mask (Display *dpy, Drawable drawable, Bool argb) |
189 |
{ |
190 |
Pixmap pixmap = XCreatePixmap (dpy, drawable, 1, 1, argb ? 32 : 8); |
191 |
|
192 |
XRenderPictFormat *format = XRenderFindStandardFormat (dpy, argb ? PictStandardARGB32 : PictStandardA8); |
193 |
XRenderPictureAttributes pa; |
194 |
pa.repeat = True; |
195 |
Picture mask = XRenderCreatePicture (dpy, pixmap, format, CPRepeat, &pa); |
196 |
|
197 |
XFreePixmap (dpy, pixmap); |
198 |
|
199 |
return mask; |
200 |
} |
201 |
|
202 |
void |
203 |
rxvt_img::brightness (double r, double g, double b, double a) |
204 |
{ |
205 |
Display *dpy = s->display->dpy; |
206 |
Picture src = create_xrender_mask (dpy, pm, True); |
207 |
Picture dst = XRenderCreatePicture (dpy, pm, format, 0, 0); |
208 |
|
209 |
XRenderColor mask_c; |
210 |
mask_c.red = float_to_component (r); |
211 |
mask_c.green = float_to_component (g); |
212 |
mask_c.blue = float_to_component (b); |
213 |
mask_c.alpha = float_to_component (a); |
214 |
XRenderFillRectangle (dpy, PictOpSrc, src, &mask_c, 0, 0, 1, 1); |
215 |
|
216 |
XRenderComposite (dpy, PictOpAdd, src, None, dst, 0, 0, 0, 0, 0, 0, w, h); |
217 |
|
218 |
XRenderFreePicture (dpy, src); |
219 |
XRenderFreePicture (dpy, dst); |
220 |
} |
221 |
|
222 |
void |
223 |
rxvt_img::contrast (double r, double g, double b, double a) |
224 |
{ |
225 |
if (!(s->display->flags & DISPLAY_HAS_RENDER_MUL)) |
226 |
return; |
227 |
|
228 |
Display *dpy = s->display->dpy; |
229 |
Picture src = create_xrender_mask (dpy, pm, True); |
230 |
Picture dst = XRenderCreatePicture (dpy, pm, format, 0, 0); |
231 |
|
232 |
XRenderColor mask_c; |
233 |
mask_c.red = float_to_component (r); |
234 |
mask_c.green = float_to_component (g); |
235 |
mask_c.blue = float_to_component (b); |
236 |
mask_c.alpha = float_to_component (a); |
237 |
XRenderFillRectangle (dpy, PictOpSrc, src, &mask_c, 0, 0, 1, 1); |
238 |
|
239 |
XRenderComposite (dpy, PictOpMultiply, src, None, dst, 0, 0, 0, 0, 0, 0, w, h); |
240 |
|
241 |
XRenderFreePicture (dpy, src); |
242 |
XRenderFreePicture (dpy, dst); |
243 |
} |
244 |
|
245 |
bool |
246 |
rxvt_img::render_pixbuf (GdkPixbuf *pixbuf, int src_x, int src_y, int width, int height, int dst_x, int dst_y) |
247 |
{ |
248 |
Display *dpy = s->display->dpy; |
249 |
|
250 |
if (s->visual->c_class != TrueColor) |
251 |
return false; |
252 |
|
253 |
uint32_t red_mask, green_mask, blue_mask, alpha_mask; |
254 |
|
255 |
red_mask = (uint32_t)format->direct.redMask << format->direct.red; |
256 |
green_mask = (uint32_t)format->direct.greenMask << format->direct.green; |
257 |
blue_mask = (uint32_t)format->direct.blueMask << format->direct.blue; |
258 |
alpha_mask = (uint32_t)format->direct.alphaMask << format->direct.alpha; |
259 |
|
260 |
int width_r = ecb_popcount32 (red_mask); |
261 |
int width_g = ecb_popcount32 (green_mask); |
262 |
int width_b = ecb_popcount32 (blue_mask); |
263 |
int width_a = ecb_popcount32 (alpha_mask); |
264 |
|
265 |
if (width_r > 8 || width_g > 8 || width_b > 8 || width_a > 8) |
266 |
return false; |
267 |
|
268 |
int sh_r = ecb_ctz32 (red_mask); |
269 |
int sh_g = ecb_ctz32 (green_mask); |
270 |
int sh_b = ecb_ctz32 (blue_mask); |
271 |
int sh_a = ecb_ctz32 (alpha_mask); |
272 |
|
273 |
if (width > 32767 || height > 32767) |
274 |
return false; |
275 |
|
276 |
XImage *ximage = XCreateImage (dpy, s->visual, format->depth, ZPixmap, 0, 0, |
277 |
width, height, 32, 0); |
278 |
if (!ximage) |
279 |
return false; |
280 |
|
281 |
if (height > INT_MAX / ximage->bytes_per_line |
282 |
|| !(ximage->data = (char *)malloc (height * ximage->bytes_per_line))) |
283 |
{ |
284 |
XDestroyImage (ximage); |
285 |
return false; |
286 |
} |
287 |
|
288 |
GC gc = XCreateGC (dpy, pm, 0, 0); |
289 |
|
290 |
ximage->byte_order = ecb_big_endian () ? MSBFirst : LSBFirst; |
291 |
|
292 |
int rowstride = gdk_pixbuf_get_rowstride (pixbuf); |
293 |
int channels = gdk_pixbuf_get_n_channels (pixbuf); |
294 |
unsigned char *row = gdk_pixbuf_get_pixels (pixbuf) + src_y * rowstride + src_x * channels; |
295 |
char *line = ximage->data; |
296 |
|
297 |
rgba c (0, 0, 0); |
298 |
|
299 |
if (channels == 4 && alpha_mask == 0) |
300 |
{ |
301 |
//pix_colors[Color_bg].get (c); |
302 |
//TODO |
303 |
c.r = 0xffff; c.g = 0xc0c0; c.b = 0xcbcb;//D |
304 |
c.r >>= 8; |
305 |
c.g >>= 8; |
306 |
c.b >>= 8; |
307 |
} |
308 |
|
309 |
for (int y = 0; y < height; y++) |
310 |
{ |
311 |
for (int x = 0; x < width; x++) |
312 |
{ |
313 |
unsigned char *pixel = row + x * channels; |
314 |
uint32_t value; |
315 |
unsigned char r, g, b, a; |
316 |
|
317 |
if (channels == 4) |
318 |
{ |
319 |
a = pixel[3]; |
320 |
r = (pixel[0] * a + c.r * (0xff - a)) / 0xff; |
321 |
g = (pixel[1] * a + c.g * (0xff - a)) / 0xff; |
322 |
b = (pixel[2] * a + c.b * (0xff - a)) / 0xff; |
323 |
} |
324 |
else |
325 |
{ |
326 |
a = 0xff; |
327 |
r = pixel[0]; |
328 |
g = pixel[1]; |
329 |
b = pixel[2]; |
330 |
} |
331 |
|
332 |
value = ((r >> (8 - width_r)) << sh_r) |
333 |
| ((g >> (8 - width_g)) << sh_g) |
334 |
| ((b >> (8 - width_b)) << sh_b) |
335 |
| ((a >> (8 - width_a)) << sh_a); |
336 |
|
337 |
if (ximage->bits_per_pixel == 32) |
338 |
((uint32_t *)line)[x] = value; |
339 |
else |
340 |
XPutPixel (ximage, x, y, value); |
341 |
} |
342 |
|
343 |
row += rowstride; |
344 |
line += ximage->bytes_per_line; |
345 |
} |
346 |
|
347 |
XPutImage (dpy, pm, gc, ximage, 0, 0, dst_x, dst_y, width, height); |
348 |
XDestroyImage (ximage); |
349 |
XFreeGC (dpy, gc); |
350 |
|
351 |
return true; |
352 |
} |
353 |
|
354 |
rxvt_img * |
355 |
rxvt_img::clone () |
356 |
{ |
357 |
rxvt_img *img = new rxvt_img (s, format, w, h); |
358 |
|
359 |
GC gc = XCreateGC (s->display->dpy, pm, 0, 0); |
360 |
XCopyArea (s->display->dpy, pm, img->pm, gc, 0, 0, w, h, 0, 0); |
361 |
XFreeGC (s->display->dpy, gc); |
362 |
|
363 |
return img; |
364 |
} |
365 |
|
366 |
rxvt_img * |
367 |
rxvt_img::sub_rect (int x, int y, int width, int height, int repeat) |
368 |
{ |
369 |
rxvt_img *img = new rxvt_img (s, format, width, height); |
370 |
|
371 |
Display *dpy = s->display->dpy; |
372 |
XRenderPictureAttributes pa; |
373 |
pa.repeat = repeat; |
374 |
Picture src = XRenderCreatePicture (dpy, pm, format, CPRepeat, &pa); |
375 |
Picture dst = XRenderCreatePicture (dpy, img->pm, img->format, 0, 0); |
376 |
|
377 |
XRenderComposite (dpy, PictOpSrc, src, None, dst, x, y, 0, 0, 0, 0, width, height); |
378 |
|
379 |
XRenderFreePicture (dpy, src); |
380 |
XRenderFreePicture (dpy, dst); |
381 |
|
382 |
return img; |
383 |
} |
384 |
|
385 |
rxvt_img * |
386 |
rxvt_img::transform (int new_width, int new_height, double matrix[9], int repeat) |
387 |
{ |
388 |
rxvt_img *img = new rxvt_img (s, format, new_width, new_height); |
389 |
|
390 |
Display *dpy = s->display->dpy; |
391 |
XRenderPictureAttributes pa; |
392 |
pa.repeat = repeat; |
393 |
Picture src = XRenderCreatePicture (dpy, pm, format, CPRepeat, &pa); |
394 |
Picture dst = XRenderCreatePicture (dpy, img->pm, img->format, 0, 0); |
395 |
|
396 |
XTransform xfrm; |
397 |
|
398 |
for (int i = 0; i < 3; ++i) |
399 |
for (int j = 0; j < 3; ++j) |
400 |
xfrm.matrix [i][j] = XDoubleToFixed (matrix [i * 3 + j]); |
401 |
|
402 |
XRenderSetPictureFilter (dpy, src, "good", 0, 0); |
403 |
XRenderSetPictureTransform (dpy, src, &xfrm); |
404 |
XRenderComposite (dpy, PictOpSrc, src, None, dst, 0, 0, 0, 0, 0, 0, new_width, new_height); |
405 |
|
406 |
XRenderFreePicture (dpy, src); |
407 |
XRenderFreePicture (dpy, dst); |
408 |
|
409 |
return img; |
410 |
} |
411 |
|
412 |
rxvt_img * |
413 |
rxvt_img::scale (int new_width, int new_height) |
414 |
{ |
415 |
double matrix[9] = { |
416 |
w / (double)new_width, 0, 0, |
417 |
0, h / (double)new_height, 0, |
418 |
0, 0, 1 |
419 |
}; |
420 |
|
421 |
return transform (new_width, new_height, matrix); |
422 |
} |
423 |
|
424 |
rxvt_img * |
425 |
rxvt_img::rotate (int new_width, int new_height, int x, int y, double phi, int repeat) |
426 |
{ |
427 |
double s = sin (phi); |
428 |
double c = cos (phi); |
429 |
|
430 |
double matrix[9] = { |
431 |
c, -s, -c * x + s * y + x, |
432 |
s, c, -s * x - c * y + y, |
433 |
0, 0, 1 |
434 |
}; |
435 |
|
436 |
return transform (new_width, new_height, matrix, repeat); |
437 |
} |
438 |
|
439 |
rxvt_img * |
440 |
rxvt_img::convert_to (XRenderPictFormat *new_format) |
441 |
{ |
442 |
rxvt_img *img = new rxvt_img (s, new_format, w, h); |
443 |
|
444 |
Display *dpy = s->display->dpy; |
445 |
Picture src = XRenderCreatePicture (dpy, pm, format, 0, 0); |
446 |
Picture dst = XRenderCreatePicture (dpy, img->pm, new_format, 0, 0); |
447 |
|
448 |
XRenderComposite (dpy, PictOpSrc, src, None, dst, 0, 0, 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::blend (rxvt_img *img, double factor) |
458 |
{ |
459 |
rxvt_img *img2 = clone (); |
460 |
Display *dpy = s->display->dpy; |
461 |
Picture src = XRenderCreatePicture (dpy, img->pm, img->format, 0, 0); |
462 |
Picture dst = XRenderCreatePicture (dpy, img2->pm, img2->format, 0, 0); |
463 |
Picture mask = create_xrender_mask (dpy, img->pm, False); |
464 |
|
465 |
XRenderColor mask_c; |
466 |
|
467 |
mask_c.alpha = float_to_component (factor); |
468 |
mask_c.red = |
469 |
mask_c.green = |
470 |
mask_c.blue = 0; |
471 |
XRenderFillRectangle (dpy, PictOpSrc, mask, &mask_c, 0, 0, 1, 1); |
472 |
|
473 |
XRenderComposite (dpy, PictOpOver, src, mask, dst, 0, 0, 0, 0, 0, 0, w, h); |
474 |
|
475 |
XRenderFreePicture (dpy, src); |
476 |
XRenderFreePicture (dpy, dst); |
477 |
XRenderFreePicture (dpy, mask); |
478 |
|
479 |
return img2; |
480 |
} |
481 |
|
482 |
#endif |
483 |
|