rxvt-unicode/src/rxvtimg.C

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

#if HAVE_IMG

rxvt_img::rxvt_img (rxvt_screen *screen, XRenderPictFormat *format, int x, int y, int width, int height, int repeat)
: s(screen), x(x), y(y), w(width), h(height), format(format), repeat(repeat),
  pm(0), ref(0)
{
}

rxvt_img::rxvt_img (const rxvt_img &img)
: 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)
{
  ++ref->cnt;
}

rxvt_img *
rxvt_img::new_from_root (rxvt_screen *s)
{
  Display *dpy = s->display->dpy;
  unsigned int root_pm_w, root_pm_h;
  Pixmap root_pixmap = s->display->get_pixmap_property (s->display->xa[XA_XROOTPMAP_ID]);
  if (root_pixmap == None)
    root_pixmap = s->display->get_pixmap_property (s->display->xa[XA_ESETROOT_PMAP_ID]);

  if (root_pixmap == None)
    return 0;

  Window wdummy;
  int idummy;
  unsigned int udummy;

  if (!XGetGeometry (dpy, root_pixmap, &wdummy, &idummy, &idummy, &root_pm_w, &root_pm_h, &udummy, &udummy))
    return 0;

  rxvt_img *img = new rxvt_img (
     s,
     XRenderFindVisualFormat (dpy, DefaultVisual (dpy, s->display->screen)),
     0,
     0,
     root_pm_w,
     root_pm_h
  );

  img->pm = root_pixmap;
  img->ref = new pixref (root_pm_w, root_pm_h);
  img->ref->ours = false;

  return img;
}

# if HAVE_PIXBUF
rxvt_img *
rxvt_img::new_from_pixbuf (rxvt_screen *s, GdkPixbuf *pb)
{
  Display *dpy = s->display->dpy;

  int width  = gdk_pixbuf_get_width  (pb);
  int height = gdk_pixbuf_get_height (pb);

  if (width > 32767 || height > 32767) // well, we *could* upload in chunks
    rxvt_fatal ("rxvt_img::new_from_pixbuf: image too big (maximum size 32768x32768).\n");

  // since we require rgb24/argb32 formats from xrender we assume
  // that both 24 and 32 bpp MUST be supported by any screen that supports xrender
  int depth = gdk_pixbuf_get_has_alpha (pb) ? 32 : 24;

  int byte_order = ecb_big_endian () ? MSBFirst : LSBFirst;

  XImage xi;

  xi.width            = width;
  xi.height           = height;
  xi.xoffset          = 0;
  xi.format           = ZPixmap;
  xi.byte_order       = ImageByteOrder (dpy);
  xi.bitmap_unit      = 0;         //XY only, unused
  xi.bitmap_bit_order = 0;         //XY only, unused
  xi.bitmap_pad       = BitmapPad (dpy);
  xi.depth            = depth;
  xi.bytes_per_line   = 0;
  xi.bits_per_pixel   = 32;         //Z only
  xi.red_mask         = 0x00000000; //Z only, unused
  xi.green_mask       = 0x00000000; //Z only, unused
  xi.blue_mask        = 0x00000000; //Z only, unused
  xi.obdata           = 0;          // probably unused

  bool byte_order_mismatch = byte_order != xi.byte_order;

  if (!XInitImage (&xi))
    rxvt_fatal ("unable to initialise ximage, please report.\n");

  if (height > INT_MAX / xi.bytes_per_line)
    rxvt_fatal ("rxvt_img::new_from_pixbuf: image too big for Xlib.\n");

  xi.data = (char *)rxvt_malloc (height * xi.bytes_per_line);

  int rowstride = gdk_pixbuf_get_rowstride (pb);

  assert (3 + (depth == 32) == gdk_pixbuf_get_n_channels (pb));
  unsigned char *row = gdk_pixbuf_get_pixels (pb);
  char *line = xi.data;

  for (int y = 0; y < height; y++)
    {
      unsigned char *src = row;
      uint32_t      *dst = (uint32_t *)line;

      if (depth == 24)
        for (int x = 0; x < width; x++)
          {
            uint8_t r = *src++;
            uint8_t g = *src++;
            uint8_t b = *src++;

            uint32_t v = (r << 16) | (g << 8) | b;
            
            if (ecb_big_endian () ? !byte_order_mismatch : byte_order_mismatch)
              v = ecb_bswap32 (v);

            *dst++ = v;
          }
      else
        for (int x = 0; x < width; x++)
          {
            uint32_t v = *(uint32_t *)src; src += 4;

            if (ecb_big_endian ())
              v = ecb_bswap32 (v);

            v = ecb_rotl32 (v, 8); // abgr to bgra

            if (!byte_order_mismatch)
              v = ecb_bswap32 (v);

            *dst++ = v;
          }

      row += rowstride;
      line += xi.bytes_per_line;
    }

  rxvt_img *img = new rxvt_img (s, XRenderFindStandardFormat (dpy, depth == 24 ? PictStandardRGB24 : PictStandardARGB32), 0, 0, width, height);
  img->alloc ();

  GC gc = XCreateGC (dpy, img->pm, 0, 0);
  XPutImage (dpy, img->pm, gc, &xi, 0, 0, 0, 0, width, height);
  XFreeGC (dpy, gc);

  free (xi.data);

  return img;
}

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_from_pixbuf (s, pb);

  g_object_unref (pb);

  return img;
}
# endif

void
rxvt_img::destroy ()
{
  if (--ref->cnt)
    return;

  if (pm && ref->ours)
    XFreePixmap (s->display->dpy, pm);

  delete ref;
}

rxvt_img::~rxvt_img ()
{
  destroy ();
}

void
rxvt_img::alloc ()
{
  pm = XCreatePixmap (s->display->dpy, s->display->root, w, h, format->depth);
  ref = new pixref (w, h);
}

Picture
rxvt_img::src_picture ()
{
  Display *dpy = s->display->dpy;

  XRenderPictureAttributes pa;
  pa.repeat = repeat;
  Picture pic = XRenderCreatePicture (dpy, pm, format, CPRepeat, &pa);

  return pic;
}

void
rxvt_img::unshare ()
{
  if (ref->cnt == 1 && ref->ours)
    return;

  Pixmap pm2 = XCreatePixmap (s->display->dpy, s->display->root, ref->w, ref->h, format->depth);
  GC gc = XCreateGC (s->display->dpy, pm, 0, 0);
  XCopyArea (s->display->dpy, pm, pm2, gc, 0, 0, ref->w, ref->h, 0, 0);
  XFreeGC (s->display->dpy, gc);

  destroy ();

  pm = pm2;
  ref = new pixref (ref->w, ref->h);
}

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);
}

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

  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));
  rxvt_img *img = new rxvt_img (s, format, x, y, w, h, repeat);
  img->alloc ();

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

  Pixmap tmp_pm = XCreatePixmap (dpy, pm, w, h, format->depth);
  Picture tmp = XRenderCreatePicture (dpy, tmp_pm , format, CPRepeat, &pa);
  XFreePixmap (dpy, tmp_pm);

  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,
                        tmp,
                        0, 0,
                        0, 0,
                        0, 0,
                        w, h);

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

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

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

  return img;
}

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 = RepeatNormal;
  Picture mask = XRenderCreatePicture (dpy, pixmap, format, CPRepeat, &pa);

  XFreePixmap (dpy, pixmap);

  return mask;
}

static void
extract (int32_t cl0, int32_t cl1, int32_t &c, unsigned short &xc)
{
  int32_t x = clamp (c, cl0, cl1);
  c -= x;
  xc = x;
}

static bool
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)
{
  extract (cl0, cl1, r, xr);
  extract (cl0, cl1, g, xg);
  extract (cl0, cl1, b, xb);
  extract (cl0, cl1, a, xa);

  return xr | xg | xb | xa;
}

void
rxvt_img::brightness (int32_t r, int32_t g, int32_t b, int32_t a)
{
  unshare ();

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

  // loop should not be needed for brightness, as only -1..1 makes sense
  //while (r | g | b | a)
    {
      unsigned short xr, xg, xb, xa;
      XRenderColor mask_c;

      if (extract (0, 65535, r, g, b, a, mask_c.red, mask_c.green, mask_c.blue, mask_c.alpha))
        XRenderFillRectangle (dpy, PictOpAdd, dst, &mask_c, 0, 0, w, h);

      if (extract (-65535, 0, r, g, b, a, mask_c.red, mask_c.green, mask_c.blue, mask_c.alpha))
        {
          XRenderColor mask_w = { 65535, 65535, 65535, 65535 };
          XRenderFillRectangle (dpy, PictOpDifference, dst, &mask_w, 0, 0, w, h);
          mask_c.red   = -mask_c.red; //TODO: verify that doing clamp, assign, and negation does the right thing
          mask_c.green = -mask_c.green;
          mask_c.blue  = -mask_c.blue;
          mask_c.alpha = -mask_c.alpha;
          XRenderFillRectangle (dpy, PictOpAdd, dst, &mask_c, 0, 0, w, h);
          XRenderFillRectangle (dpy, PictOpDifference, dst, &mask_w, 0, 0, w, h);
        }
    }

  XRenderFreePicture (dpy, dst);
}

void
rxvt_img::contrast (int32_t r, int32_t g, int32_t b, int32_t a)
{
  if (r < 0 || g < 0 || b < 0 || a < 0)
    rxvt_fatal ("rxvt_img::contrast does not support negative values.\n");

  rxvt_img *img = new rxvt_img (s, format, x, y, w, h, repeat);
  img->alloc ();

  {
    rxvt_color empty;
    empty.set (s, rgba (0, 0, 0, 0));
    img->fill (empty);
  }

  // premultiply (yeah, these are not exact, sue me or fix it)
  r = (r * (a >> 8)) >> 8;
  g = (g * (a >> 8)) >> 8;
  b = (b * (a >> 8)) >> 8;

  Display *dpy = s->display->dpy;

  Picture src = src_picture ();
  Picture dst = XRenderCreatePicture (dpy, img->pm, format, 0, 0);
  Picture mul = create_xrender_mask (dpy, pm, True);

  XRenderPictureAttributes pa;
  pa.component_alpha = 1;
  XRenderChangePicture (dpy, mul, CPComponentAlpha, &pa);

  //TODO: this operator does not yet implement some useful contrast
  while (r | g | b | a)
    {
      unsigned short xr, xg, xb, xa;
      XRenderColor mask_c;

      if (extract (0, 65535, r, g, b, a, mask_c.red, mask_c.green, mask_c.blue, mask_c.alpha))
        {
          XRenderFillRectangle (dpy, PictOpSrc, mul, &mask_c, 0, 0, 1, 1);
          XRenderComposite (dpy, PictOpAdd, src, mul, dst, 0, 0, 0, 0, 0, 0, w, h);
        }
    }

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

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

  delete img;
}

rxvt_img *
rxvt_img::clone ()
{
  return new rxvt_img (*this);
}

static XRenderPictFormat *
find_alpha_format_for (Display *dpy, XRenderPictFormat *format)
{
  if (format->direct.alphaMask)
    return format; // already has alpha

  // try to find a suitable alpha format, one bit alpha is enough for our purposes
  if (format->type == PictTypeDirect)
    for (int n = 0; XRenderPictFormat *f = XRenderFindFormat (dpy, 0, 0, n); ++n)
      if (f->direct.alphaMask
          && f->type == PictTypeDirect
          && ecb_popcount32 (f->direct.redMask  ) >= ecb_popcount32 (format->direct.redMask  )
          && ecb_popcount32 (f->direct.greenMask) >= ecb_popcount32 (format->direct.greenMask)
          && ecb_popcount32 (f->direct.blueMask ) >= ecb_popcount32 (format->direct.blueMask ))
        return f;

  // should be a very good fallback
  return XRenderFindStandardFormat (dpy, PictStandardARGB32);
}

rxvt_img *
rxvt_img::reify ()
{
  if (x == 0 && y == 0 && w == ref->w && h == ref->h)
    return clone ();

  Display *dpy = s->display->dpy;

  // add an alpha channel if...
  bool alpha = !format->direct.alphaMask // pixmap has none yet
               && (x || y)               // we need one because of non-zero offset
               && repeat == RepeatNone;  // and we have no good pixels to fill with

  rxvt_img *img = new rxvt_img (s, alpha ? find_alpha_format_for (dpy, format) : format, 0, 0, w, h, repeat);
  img->alloc ();

  Picture src = src_picture ();
  Picture dst = XRenderCreatePicture (dpy, img->pm, img->format, 0, 0);
  
  if (alpha)
    {
      XRenderColor rc = { 0, 0, 0, 0 };
      XRenderFillRectangle (dpy, PictOpSrc, dst, &rc, 0, 0, w, h);//TODO: split into four fillrectangles
      XRenderComposite (dpy, PictOpSrc, src, None, dst, 0, 0, 0, 0, -x, -y, ref->w, ref->h);
    }
  else
    XRenderComposite (dpy, PictOpSrc, src, None, dst, x, y, 0, 0, 0, 0, w, h);

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

  return img;
}

rxvt_img *
rxvt_img::sub_rect (int x, int y, int width, int height)
{
  rxvt_img *img = clone ();

  img->x += x;
  img->y += y;

  if (w != width || h != height)
    {
      img->w = width;
      img->h = height;

      rxvt_img *img2 = img->reify ();
      delete img;
      img = img2;
    }

  return img;
}

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

  Display *dpy = s->display->dpy;
  Picture src = src_picture ();
  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]);

  xfrm.matrix [0][2] -= XDoubleToFixed (x);//TODO
  xfrm.matrix [1][2] -= XDoubleToFixed (y);

  XRenderSetPictureFilter (dpy, src, "good", 0, 0);
  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)
{
  if (w == new_width && h == new_height)
    return clone ();

  double matrix[9] = {
    w  / (double)new_width, 0, 0,
    0, h / (double)new_height, 0,
    0,                      0, 1
  };

  int old_repeat_mode = repeat;
  repeat = RepeatPad; // not right, but xrender can't proeprly scale it seems

  rxvt_img *img = transform (new_width, new_height, matrix);

  repeat = old_repeat_mode;
  img->repeat = repeat;

  return img;
}

rxvt_img *
rxvt_img::rotate (int new_width, int new_height, int x, int y, double phi)
{
  double s = sin (phi);
  double c = cos (phi);

  double matrix[9] = {
    c, -s, -c * x + s * y + x,
    s,  c, -s * x - c * y + y,
    0,  0,                  1
  };

  return transform (new_width, new_height, matrix);
}

rxvt_img *
rxvt_img::convert_format (XRenderPictFormat *new_format, const rxvt_color &bg)
{
  if (new_format == format)
    return clone ();

  rxvt_img *img = new rxvt_img (s, new_format, x, y, w, h, repeat);
  img->alloc ();

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

  if (format->direct.alphaMask && !new_format->direct.alphaMask)
    {
      // does it have to be that complicated
      rgba c;
      bg.get (c);

      XRenderColor rc = { c.r, c.g, c.b, 0xffff };
      XRenderFillRectangle (dpy, PictOpSrc, dst, &rc, 0, 0, w, h);

      op = PictOpOver;
    }

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

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

  return img;
}

rxvt_img *
rxvt_img::blend (rxvt_img *img, double factor)
{
  rxvt_img *img2 = clone ();
  Display *dpy = s->display->dpy;
  Picture src = img->src_picture ();
  Picture dst = XRenderCreatePicture (dpy, img2->pm, img2->format, 0, 0);

  Pixmap pixmap = XCreatePixmap (dpy, img->pm, 1, 1, 8);
  XRenderPictFormat *format = XRenderFindStandardFormat (dpy, PictStandardA8);
  XRenderPictureAttributes pa;
  pa.repeat = True;
  Picture mask = XRenderCreatePicture (dpy, pixmap, format, CPRepeat, &pa);
  XFreePixmap (dpy, pixmap);

  XRenderColor mask_c;

  mask_c.alpha = float_to_component (factor);
  mask_c.red   =
  mask_c.green =
  mask_c.blue  = 0;
  XRenderFillRectangle (dpy, PictOpSrc, mask, &mask_c, 0, 0, 1, 1);

  XRenderComposite (dpy, PictOpOver, src, mask, dst, 0, 0, 0, 0, 0, 0, w, h);

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

  return img2;
}

#endif

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