rxvt-unicode/src/rxvtimg.C

#include <math.h>
#include "../config.h"
#include "rxvt.h"

#if HAVE_IMG

#define float_to_component(d) ((d) * 65535.99)

rxvt_img::rxvt_img (rxvt_screen *screen, XRenderPictFormat *format, int width, int height)
: s(screen), w(width), h(height), format(format), shared(false)
{
  pm = XCreatePixmap (s->display->dpy, s->display->root, w, h, format->depth);
}

rxvt_img::rxvt_img (rxvt_screen *screen, XRenderPictFormat *format, int width, int height, Pixmap pixmap)
: s(screen), pm(pixmap), w(width), h(height), format(format), shared(false)
{
}

rxvt_img *
rxvt_img::new_from_file (rxvt_screen *s, const char *filename)
{
  GError *err = 0;
  GdkPixbuf *pb = gdk_pixbuf_new_from_file (filename, &err);

  if (!pb)
    rxvt_fatal ("rxvt_img::new_from_file: %s\n", err->message);

  rxvt_img *img = new rxvt_img (
     s,
     XRenderFindStandardFormat (s->display->dpy, gdk_pixbuf_get_has_alpha (pb) ? PictStandardARGB32 : PictStandardRGB24),
     gdk_pixbuf_get_width (pb),
     gdk_pixbuf_get_height (pb)
  );

  img->render (pb, 0, 0, img->w, img->h, 0, 0);

  return img;
}

rxvt_img::~rxvt_img ()
{
  if (!shared)
    XFreePixmap (s->display->dpy, pm);
}

void
rxvt_img::unshare ()
{
  if (!shared)
    return;

  rxvt_img *img = clone ();

  ::swap (pm    , img->pm);
  ::swap (shared, img->shared);

  delete img;
}

void
rxvt_img::fill (const rxvt_color &c)
{
  XGCValues gcv;
  gcv.foreground = c;
  GC gc = XCreateGC (s->display->dpy, pm, GCForeground, &gcv);
  XFillRectangle (s->display->dpy, pm, gc, 0, 0, w, h);
  XFreeGC (s->display->dpy, gc);
}

static void
get_gaussian_kernel (int radius, int width, double *kernel, XFixed *params)
{
  double sigma = radius / 2.0;
  double scale = sqrt (2.0 * M_PI) * sigma;
  double sum = 0.0;

  for (int i = 0; i < width; i++)
    {
      double x = i - width / 2;
      kernel[i] = exp (-(x * x) / (2.0 * sigma * sigma)) / scale;
      sum += kernel[i];
    }

  params[0] = XDoubleToFixed (width);
  params[1] = XDoubleToFixed (1);

  for (int i = 0; i < width; i++)
    params[i+2] = XDoubleToFixed (kernel[i] / sum);
}

void
rxvt_img::blur (int rh, int rv)
{
  if (!(s->display->flags & DISPLAY_HAS_RENDER_CONV))
    return;

  Display *dpy = s->display->dpy;
  int size = max (rh, rv) * 2 + 1;
  double *kernel = (double *)malloc (size * sizeof (double));
  XFixed *params = (XFixed *)malloc ((size + 2) * sizeof (XFixed));

  XRenderPictureAttributes pa;

  pa.repeat = RepeatPad;
  Picture src = XRenderCreatePicture (dpy, pm, format, CPRepeat, &pa);
  Pixmap tmp = XCreatePixmap (dpy, pm, w, h, format->depth);
  Picture dst = XRenderCreatePicture (dpy, tmp, format, CPRepeat, &pa);
  XFreePixmap (dpy, tmp);

  if (kernel && params)
    {
      size = rh * 2 + 1;
      get_gaussian_kernel (rh, size, kernel, params);

      XRenderSetPictureFilter (dpy, src, FilterConvolution, params, size+2);
      XRenderComposite (dpy,
                        PictOpSrc,
                        src,
                        None,
                        dst,
                        0, 0,
                        0, 0,
                        0, 0,
                        w, h);

      ::swap (src, dst);

      size = rv * 2 + 1;
      get_gaussian_kernel (rv, size, kernel, params);
      ::swap (params[0], params[1]);

      XRenderSetPictureFilter (dpy, src, FilterConvolution, params, size+2);
      XRenderComposite (dpy,
                        PictOpSrc,
                        src,
                        None,
                        dst,
                        0, 0,
                        0, 0,
                        0, 0,
                        w, h);
    }

  free (kernel);
  free (params);
  XRenderFreePicture (dpy, src);
  XRenderFreePicture (dpy, dst);
}

static Picture
create_xrender_mask (Display *dpy, Drawable drawable, Bool argb)
{
  Pixmap pixmap = XCreatePixmap (dpy, drawable, 1, 1, argb ? 32 : 8);

  XRenderPictFormat *format = XRenderFindStandardFormat (dpy, argb ? PictStandardARGB32 : PictStandardA8);
  XRenderPictureAttributes pa;
  pa.repeat = True;
  Picture mask = XRenderCreatePicture (dpy, pixmap, format, CPRepeat, &pa);

  XFreePixmap (dpy, pixmap);

  return mask;
}

void
rxvt_img::brightness (double r, double g, double b, double a)
{
  Display *dpy = s->display->dpy;
  Picture src = create_xrender_mask (dpy, pm, True);
  Picture dst = XRenderCreatePicture (dpy, pm, format, 0, 0);

  XRenderColor mask_c;
  mask_c.red   = float_to_component (r);
  mask_c.green = float_to_component (g);
  mask_c.blue  = float_to_component (b);
  mask_c.alpha = float_to_component (a);
  XRenderFillRectangle (dpy, PictOpSrc, src, &mask_c, 0, 0, 1, 1);

  XRenderComposite (dpy, PictOpAdd, src, None, dst, 0, 0, 0, 0, 0, 0, w, h);
}

void
rxvt_img::contrast (double r, double g, double b, double a)
{
  if (!(s->display->flags & DISPLAY_HAS_RENDER_MUL))
    return;

  Display *dpy = s->display->dpy;
  Picture src = create_xrender_mask (dpy, pm, True);
  Picture dst = XRenderCreatePicture (dpy, pm, format, 0, 0);

  XRenderColor mask_c;
  mask_c.red   = float_to_component (r);
  mask_c.green = float_to_component (g);
  mask_c.blue  = float_to_component (b);
  mask_c.alpha = float_to_component (a);
  XRenderFillRectangle (dpy, PictOpSrc, src, &mask_c, 0, 0, 1, 1);

  XRenderComposite (dpy, PictOpMultiply, src, None, dst, 0, 0, 0, 0, 0, 0, w, h);
}

void
rxvt_img::render (GdkPixbuf *pixbuf, int src_x, int src_y, int width, int height, int dst_x, int dst_y)
{
  //TODO
}

rxvt_img *
rxvt_img::clone ()
{
  GC gc = XCreateGC (s->display->dpy, pm, 0, 0);
  Pixmap pm2 = XCreatePixmap (s->display->dpy, pm, w, h, format->depth);
  XCopyArea (s->display->dpy, pm, pm2, gc, 0, 0, w, h, 0, 0);
  XFreeGC (s->display->dpy, gc);
  return new rxvt_img (s, format, w, h, pm2);
}

rxvt_img *
rxvt_img::transform (int new_width, int new_height, int repeat, double matrix[9])
{
  rxvt_img *img = new rxvt_img (s, format, new_width, new_height);

  Display *dpy = s->display->dpy;
  XRenderPictureAttributes pa;
  pa.repeat = repeat;
  Picture src = XRenderCreatePicture (dpy,      pm,      format, CPRepeat, &pa);
  Picture dst = XRenderCreatePicture (dpy, img->pm, img->format,        0,   0);

  XTransform xfrm;

  for (int i = 0; i < 3; ++i)
    for (int j = 0; j < 3; ++j)
      xfrm.matrix [i][j] = XDoubleToFixed (matrix [i * 3 + j]);

  XRenderSetPictureTransform (dpy, src, &xfrm);
  XRenderComposite (dpy, PictOpSrc, src, None, dst, 0, 0, 0, 0, 0, 0, new_width, new_height);

  XRenderFreePicture (dpy, src);
  XRenderFreePicture (dpy, dst);

  return img;
}

rxvt_img *
rxvt_img::scale (int new_width, int new_height)
{
  double matrix[9] = {
    new_width  / (double)w, 0, 0,
    0, new_height / (double)h, 0,
    0,                      0, 1
  };

  return transform (new_width, new_height, RepeatNormal, matrix);
}

rxvt_img *
rxvt_img::convert_to (XRenderPictFormat *new_format)
{
  rxvt_img *img = new rxvt_img (s, new_format, w, h);

  Display *dpy = s->display->dpy;
  Picture src = XRenderCreatePicture (dpy,      pm,     format, 0, 0);
  Picture dst = XRenderCreatePicture (dpy, img->pm, new_format, 0, 0);

  XRenderComposite (dpy, PictOpSrc, src, None, dst, 0, 0, 0, 0, 0, 0, w, h);

  XRenderFreePicture (dpy, src);
  XRenderFreePicture (dpy, dst);

  return img;
}

#endif

Revision:	1.13
Committed:	Mon Jun 4 15:30:41 2012 UTC (11 years, 11 months ago) by root
Content type:	text/plain
Branch:	MAIN
Changes since 1.12:	+4 -2 lines
Log Message:	* empty log message *
#	Content
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_file (rxvt_screen s, const char filename)
22	{
23	GError *err = 0;
24	GdkPixbuf *pb = gdk_pixbuf_new_from_file (filename, &err);
25
26	if (!pb)
27	rxvt_fatal ("rxvt_img::new_from_file: %s\n", err->message);
28
29	rxvt_img *img = new rxvt_img (
30	s,
31	XRenderFindStandardFormat (s->display->dpy, gdk_pixbuf_get_has_alpha (pb) ? PictStandardARGB32 : PictStandardRGB24),
32	gdk_pixbuf_get_width (pb),
33	gdk_pixbuf_get_height (pb)
34	);
35
36	img->render (pb, 0, 0, img->w, img->h, 0, 0);
37
38	return img;
39	}
40
41	rxvt_img::~rxvt_img ()
42	{
43	if (!shared)
44	XFreePixmap (s->display->dpy, pm);
45	}
46
47	void
48	rxvt_img::unshare ()
49	{
50	if (!shared)
51	return;
52
53	rxvt_img *img = clone ();
54
55	::swap (pm , img->pm);
56	::swap (shared, img->shared);
57
58	delete img;
59	}
60
61	void
62	rxvt_img::fill (const rxvt_color &c)
63	{
64	XGCValues gcv;
65	gcv.foreground = c;
66	GC gc = XCreateGC (s->display->dpy, pm, GCForeground, &gcv);
67	XFillRectangle (s->display->dpy, pm, gc, 0, 0, w, h);
68	XFreeGC (s->display->dpy, gc);
69	}
70
71	static void
72	get_gaussian_kernel (int radius, int width, double kernel, XFixed params)
73	{
74	double sigma = radius / 2.0;
75	double scale = sqrt (2.0 * M_PI) * sigma;
76	double sum = 0.0;
77
78	for (int i = 0; i < width; i++)
79	{
80	double x = i - width / 2;
81	kernel[i] = exp (-(x * x) / (2.0 * sigma * sigma)) / scale;
82	sum += kernel[i];
83	}
84
85	params[0] = XDoubleToFixed (width);
86	params[1] = XDoubleToFixed (1);
87
88	for (int i = 0; i < width; i++)
89	params[i+2] = XDoubleToFixed (kernel[i] / sum);
90	}
91
92	void
93	rxvt_img::blur (int rh, int rv)
94	{
95	if (!(s->display->flags & DISPLAY_HAS_RENDER_CONV))
96	return;
97
98	Display *dpy = s->display->dpy;
99	int size = max (rh, rv) * 2 + 1;
100	double kernel = (double )malloc (size * sizeof (double));
101	XFixed params = (XFixed )malloc ((size + 2) * sizeof (XFixed));
102
103	XRenderPictureAttributes pa;
104
105	pa.repeat = RepeatPad;
106	Picture src = XRenderCreatePicture (dpy, pm, format, CPRepeat, &pa);
107	Pixmap tmp = XCreatePixmap (dpy, pm, w, h, format->depth);
108	Picture dst = XRenderCreatePicture (dpy, tmp, format, CPRepeat, &pa);
109	XFreePixmap (dpy, tmp);
110
111	if (kernel && params)
112	{
113	size = rh * 2 + 1;
114	get_gaussian_kernel (rh, size, kernel, params);
115
116	XRenderSetPictureFilter (dpy, src, FilterConvolution, params, size+2);
117	XRenderComposite (dpy,
118	PictOpSrc,
119	src,
120	None,
121	dst,
122	0, 0,
123	0, 0,
124	0, 0,
125	w, h);
126
127	::swap (src, dst);
128
129	size = rv * 2 + 1;
130	get_gaussian_kernel (rv, size, kernel, params);
131	::swap (params[0], params[1]);
132
133	XRenderSetPictureFilter (dpy, src, FilterConvolution, params, size+2);
134	XRenderComposite (dpy,
135	PictOpSrc,
136	src,
137	None,
138	dst,
139	0, 0,
140	0, 0,
141	0, 0,
142	w, h);
143	}
144
145	free (kernel);
146	free (params);
147	XRenderFreePicture (dpy, src);
148	XRenderFreePicture (dpy, dst);
149	}
150
151	static Picture
152	create_xrender_mask (Display *dpy, Drawable drawable, Bool argb)
153	{
154	Pixmap pixmap = XCreatePixmap (dpy, drawable, 1, 1, argb ? 32 : 8);
155
156	XRenderPictFormat *format = XRenderFindStandardFormat (dpy, argb ? PictStandardARGB32 : PictStandardA8);
157	XRenderPictureAttributes pa;
158	pa.repeat = True;
159	Picture mask = XRenderCreatePicture (dpy, pixmap, format, CPRepeat, &pa);
160
161	XFreePixmap (dpy, pixmap);
162
163	return mask;
164	}
165
166	void
167	rxvt_img::brightness (double r, double g, double b, double a)
168	{
169	Display *dpy = s->display->dpy;
170	Picture src = create_xrender_mask (dpy, pm, True);
171	Picture dst = XRenderCreatePicture (dpy, pm, format, 0, 0);
172
173	XRenderColor mask_c;
174	mask_c.red = float_to_component (r);
175	mask_c.green = float_to_component (g);
176	mask_c.blue = float_to_component (b);
177	mask_c.alpha = float_to_component (a);
178	XRenderFillRectangle (dpy, PictOpSrc, src, &mask_c, 0, 0, 1, 1);
179
180	XRenderComposite (dpy, PictOpAdd, src, None, dst, 0, 0, 0, 0, 0, 0, w, h);
181	}
182
183	void
184	rxvt_img::contrast (double r, double g, double b, double a)
185	{
186	if (!(s->display->flags & DISPLAY_HAS_RENDER_MUL))
187	return;
188
189	Display *dpy = s->display->dpy;
190	Picture src = create_xrender_mask (dpy, pm, True);
191	Picture dst = XRenderCreatePicture (dpy, pm, format, 0, 0);
192
193	XRenderColor mask_c;
194	mask_c.red = float_to_component (r);
195	mask_c.green = float_to_component (g);
196	mask_c.blue = float_to_component (b);
197	mask_c.alpha = float_to_component (a);
198	XRenderFillRectangle (dpy, PictOpSrc, src, &mask_c, 0, 0, 1, 1);
199
200	XRenderComposite (dpy, PictOpMultiply, src, None, dst, 0, 0, 0, 0, 0, 0, w, h);
201	}
202
203	void
204	rxvt_img::render (GdkPixbuf *pixbuf, int src_x, int src_y, int width, int height, int dst_x, int dst_y)
205	{
206	//TODO
207	}
208
209	rxvt_img *
210	rxvt_img::clone ()
211	{
212	GC gc = XCreateGC (s->display->dpy, pm, 0, 0);
213	Pixmap pm2 = XCreatePixmap (s->display->dpy, pm, w, h, format->depth);
214	XCopyArea (s->display->dpy, pm, pm2, gc, 0, 0, w, h, 0, 0);
215	XFreeGC (s->display->dpy, gc);
216	return new rxvt_img (s, format, w, h, pm2);
217	}
218
219	rxvt_img *
220	rxvt_img::transform (int new_width, int new_height, int repeat, double matrix[9])
221	{
222	rxvt_img *img = new rxvt_img (s, format, new_width, new_height);
223
224	Display *dpy = s->display->dpy;
225	XRenderPictureAttributes pa;
226	pa.repeat = repeat;
227	Picture src = XRenderCreatePicture (dpy, pm, format, CPRepeat, &pa);
228	Picture dst = XRenderCreatePicture (dpy, img->pm, img->format, 0, 0);
229
230	XTransform xfrm;
231
232	for (int i = 0; i < 3; ++i)
233	for (int j = 0; j < 3; ++j)
234	xfrm.matrix [i][j] = XDoubleToFixed (matrix [i * 3 + j]);
235
236	XRenderSetPictureTransform (dpy, src, &xfrm);
237	XRenderComposite (dpy, PictOpSrc, src, None, dst, 0, 0, 0, 0, 0, 0, new_width, new_height);
238
239	XRenderFreePicture (dpy, src);
240	XRenderFreePicture (dpy, dst);
241
242	return img;
243	}
244
245	rxvt_img *
246	rxvt_img::scale (int new_width, int new_height)
247	{
248	double matrix[9] = {
249	new_width / (double)w, 0, 0,
250	0, new_height / (double)h, 0,
251	0, 0, 1
252	};
253
254	return transform (new_width, new_height, RepeatNormal, matrix);
255	}
256
257	rxvt_img *
258	rxvt_img::convert_to (XRenderPictFormat *new_format)
259	{
260	rxvt_img *img = new rxvt_img (s, new_format, w, h);
261
262	Display *dpy = s->display->dpy;
263	Picture src = XRenderCreatePicture (dpy, pm, format, 0, 0);
264	Picture dst = XRenderCreatePicture (dpy, img->pm, new_format, 0, 0);
265
266	XRenderComposite (dpy, PictOpSrc, src, None, dst, 0, 0, 0, 0, 0, 0, w, h);
267
268	XRenderFreePicture (dpy, src);
269	XRenderFreePicture (dpy, dst);
270
271	return img;
272	}
273
274	#endif
275