rxvt-unicode/src/rxvttoolkit.C

/*----------------------------------------------------------------------*
 * File:        rxvttoolkit.C
 *----------------------------------------------------------------------*
 *
 * All portions of code are copyright by their respective author/s.
 * Copyright (c) 2003-2006 Marc Lehmann <pcg@goof.com>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 *----------------------------------------------------------------------*/

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

#include <unistd.h>
#include <fcntl.h>

#include <sys/utsname.h>
#include <sys/socket.h>
#include <sys/un.h>

#if XFT
# include <X11/extensions/Xrender.h>
#endif

const char *const xa_names[] =
{
  "TEXT",
  "COMPOUND_TEXT",
  "UTF8_STRING",
  "MULTIPLE",
  "TARGETS",
  "TIMESTAMP",
  "VT_SELECTION",
  "INCR",
  "WM_PROTOCOLS",
  "WM_DELETE_WINDOW",
  "CLIPBOARD",
  "AVERAGE_WIDTH",
  "WEIGHT_NAME",
  "SLANT",
  "CHARSET_REGISTRY",
  "CHARSET_ENCODING",
#if ENABLE_FRILLS
  "_MOTIF_WM_HINTS",
#endif
#if ENABLE_EWMH
  "_NET_WM_PID",
  "_NET_WM_NAME",
  "_NET_WM_ICON_NAME",
  "_NET_WM_PING",
#endif
#if USE_XIM
  "WM_LOCALE_NAME",
  "XIM_SERVERS",
#endif
#ifdef TRANSPARENT
  "_XROOTPMAP_ID",
  "ESETROOT_PMAP_ID",
#endif
#if ENABLE_XEMBED
  "_XEMBED",
  "_XEMBED_INFO",
#endif
#if !ENABLE_MINIMAL
  "SCREEN_RESOURCES",
  "XDCCC_LINEAR_RGB_CORRECTION",
  "XDCCC_LINEAR_RGB_MATRICES",
  "WM_COLORMAP_WINDOWS",
  "WM_STATE",
  "cursor",
# if USE_XIM
  "TRANSPORT",
  "LOCALES",
  "_XIM_PROTOCOL",
  "_XIM_XCONNECT",
  "_XIM_MOREDATA",
# endif
#endif
};

/////////////////////////////////////////////////////////////////////////////

refcounted::refcounted (const char *id)
{
  this->id = strdup (id);
}

refcounted::~refcounted ()
{
  free (id);
}

template<class T>
T *refcache<T>::get (const char *id)
{
  for (T **i = this->begin (); i < this->end (); ++i)
    {
      if (!strcmp (id, (*i)->id))
        {
          ++(*i)->referenced;
          (*i)->ref_next ();
          return *i;
        }
    }

  T *obj = new T (id);

  if (obj && obj->ref_init ())
    {
      obj->referenced = 1;
      this->push_back (obj);
      return obj;
    }
  else
    {
      delete obj;
      return 0;
    }
}

template<class T>
void refcache<T>::put (T *obj)
{
  if (!obj)
    return;

  if (!--obj->referenced)
    {
      this->erase (find (this->begin (), this->end (), obj));
      delete obj;
    }
}

template<class T>
void refcache<T>::clear ()
{
  while (this->size ())
    put (*this->begin ());
}

/////////////////////////////////////////////////////////////////////////////

#ifdef USE_XIM

static void
#if XIMCB_PROTO_BROKEN
im_destroy_cb (XIC unused1, XPointer client_data, XPointer unused3)
#else
im_destroy_cb (XIM unused1, XPointer client_data, XPointer unused3)
#endif
{
  rxvt_xim *xim = (rxvt_xim *)client_data;
  rxvt_display *display = xim->display;

  xim->xim = 0;

  display->xims.erase (find (display->xims.begin (), display->xims.end (), xim));
  display->im_change_cb ();
}

bool
rxvt_xim::ref_init ()
{
  display = GET_R->display; //HACK: TODO

  xim = XOpenIM (display->dpy, 0, 0, 0);

  if (!xim)
    return false;

  XIMCallback ximcallback;
  ximcallback.client_data = (XPointer)this;
  ximcallback.callback = im_destroy_cb;

  XSetIMValues (xim, XNDestroyCallback, &ximcallback, 0);

  return true;
}

rxvt_xim::~rxvt_xim ()
{
  if (xim)
    XCloseIM (xim);
}

#endif

/////////////////////////////////////////////////////////////////////////////

#if XFT
rxvt_drawable::~rxvt_drawable ()
{
  if (xftdrawable)
    XftDrawDestroy (xftdrawable);
}

rxvt_drawable::operator XftDraw *()
{
  if (!xftdrawable)
    xftdrawable = XftDrawCreate (screen->dpy, drawable, screen->visual, screen->cmap);

  return xftdrawable;
}
#endif

/////////////////////////////////////////////////////////////////////////////

#if XFT

// not strictly necessary as it is only used with superclass of zero_initialised
rxvt_screen::rxvt_screen ()
: scratch_area (0)
{
}

rxvt_drawable &rxvt_screen::scratch_drawable (int w, int h)
{
  // it's actually faster to re-allocate every time. don't ask me
  // why, but its likely no big deal there are no roundtrips
  // (I think/hope).
  if (!scratch_area || w > scratch_w || h > scratch_h || 1/*D*/)
    {
      if (scratch_area)
        {
          XFreePixmap (dpy, scratch_area->drawable);
          delete scratch_area;
        }

      Pixmap pm = XCreatePixmap (dpy, RootWindowOfScreen (ScreenOfDisplay (dpy, display->screen)),
                                 scratch_w = w, scratch_h = h, depth);

      scratch_area = new rxvt_drawable (this, pm);
    }

  return *scratch_area;
}

#endif

void
rxvt_screen::set (rxvt_display *disp)
{
  display = disp;
  dpy     = disp->dpy;

  Screen *screen = ScreenOfDisplay (dpy, disp->screen);

  depth   = DefaultDepthOfScreen    (screen);
  visual  = DefaultVisualOfScreen   (screen);
  cmap    = DefaultColormapOfScreen (screen);
}

void
rxvt_screen::select_visual (int bitdepth)
{
#if XFT
  XVisualInfo vinfo;

  if (XMatchVisualInfo (dpy, display->screen, bitdepth, TrueColor, &vinfo))
    {
      depth  = bitdepth;
      visual = vinfo.visual;
      cmap   = XCreateColormap (dpy, display->root, visual, AllocNone);
    }
#endif
}

void
rxvt_screen::clear ()
{
#if XFT
  if (scratch_area)
    {
      XFreePixmap (dpy, scratch_area->drawable);
      delete scratch_area;
    }
#endif

  if (cmap != DefaultColormapOfScreen (ScreenOfDisplay (dpy, display->screen)))
    XFreeColormap (dpy, cmap);
}

/////////////////////////////////////////////////////////////////////////////

rxvt_display::rxvt_display (const char *id)
: refcounted (id)
, x_ev (this, &rxvt_display::x_cb)
, selection_owner (0)
{
}

XrmDatabase
rxvt_display::get_resources (bool refresh)
{
  char *homedir = (char *)getenv ("HOME");
  char fname[1024];

  /*
   * get resources using the X library function
   */
  char *displayResource, *xe;
  XrmDatabase rdb1, database = 0;

  // for ordering, see for example http://www.faqs.org/faqs/Xt-FAQ/ Subject: 20

  // 6. System wide per application default file.

  /* Add in $XAPPLRESDIR/Rxvt only; not bothering with XUSERFILESEARCHPATH */
  if ((xe = (char *)getenv ("XAPPLRESDIR")))
    {
      snprintf (fname, sizeof (fname), "%s/%s", xe, RESCLASS);

      if ((rdb1 = XrmGetFileDatabase (fname)))
        XrmMergeDatabases (rdb1, &database);
    }

  // 5. User's per application default file.
  // none

  // 4. User's defaults file.
  /* Get any Xserver defaults */
  if (refresh)
    {
      // fucking xlib keeps a copy of the rm string
      Atom actual_type;
      int actual_format;
      unsigned long nitems, nremaining;
      char *val = 0;

#if XLIB_ILLEGAL_ACCESS
      if (dpy->xdefaults)
        XFree (dpy->xdefaults);
#endif

      if (XGetWindowProperty (dpy, RootWindow (dpy, 0), XA_RESOURCE_MANAGER,
                              0L, 100000000L, False,
                              XA_STRING, &actual_type, &actual_format,
                              &nitems, &nremaining,
                              (unsigned char **)&val) == Success
          && actual_type == XA_STRING
          && actual_format == 8)
         displayResource = val;
       else
         {
           displayResource = 0;
           if (val)
             XFree(val);
         }

#if XLIB_ILLEGAL_ACCESS
      dpy->xdefaults = displayResource;
#endif
    }
  else
    displayResource = XResourceManagerString (dpy);

  if (displayResource)
    {
      if ((rdb1 = XrmGetStringDatabase (displayResource)))
        XrmMergeDatabases (rdb1, &database);
    }
  else if (homedir)
    {
      snprintf (fname, sizeof (fname), "%s/.Xdefaults", homedir);

      if ((rdb1 = XrmGetFileDatabase (fname)))
        XrmMergeDatabases (rdb1, &database);
    }

#if !XLIB_ILLEGAL_ACCESS
  if (refresh && displayResource)
    XFree (displayResource);
#endif

  /* Get screen specific resources */
  displayResource = XScreenResourceString (ScreenOfDisplay (dpy, screen));

  if (displayResource)
    {
      if ((rdb1 = XrmGetStringDatabase (displayResource)))
        /* Merge with screen-independent resources */
        XrmMergeDatabases (rdb1, &database);

      XFree (displayResource);
    }

  // 3. User's per host defaults file
  /* Add in XENVIRONMENT file */
  if ((xe = (char *)getenv ("XENVIRONMENT"))
      && (rdb1 = XrmGetFileDatabase (xe)))
    XrmMergeDatabases (rdb1, &database);
  else if (homedir)
    {
      struct utsname un;

      if (!uname (&un))
        {
          snprintf (fname, sizeof (fname), "%s/.Xdefaults-%s", homedir, un.nodename);

          if ((rdb1 = XrmGetFileDatabase (fname)))
            XrmMergeDatabases (rdb1, &database);
        }
    }

  return database;
}

bool rxvt_display::ref_init ()
{
#ifdef LOCAL_X_IS_UNIX
  if (id[0] == ':')
    {
      val = rxvt_malloc (5 + strlen (id) + 1);
      strcpy (val, "unix/");
      strcat (val, id);
      dpy = XOpenDisplay (val);
      free (val);
    }
  else
#endif
    dpy = 0;

  if (!dpy)
    dpy = XOpenDisplay (id);

  if (!dpy)
    return false;

  screen = DefaultScreen     (dpy);
  root   = DefaultRootWindow (dpy);

  assert (sizeof (xa_names) / sizeof (char *) == NUM_XA);
  XInternAtoms (dpy, (char **)xa_names, NUM_XA, False, xa);

  XrmSetDatabase (dpy, get_resources (false));

#ifdef POINTER_BLANK
  XColor blackcolour;
  blackcolour.red   = 0;
  blackcolour.green = 0;
  blackcolour.blue  = 0;
  Font f = XLoadFont (dpy, "fixed");
  blank_cursor = XCreateGlyphCursor (dpy, f, f, ' ', ' ',
                                     &blackcolour, &blackcolour);
  XUnloadFont (dpy, f);
#endif

  int fd = XConnectionNumber (dpy);

  // try to detect wether we have a local connection.
  // assume unix domains socket == local, everything else not
  // TODO: might want to check for inet/127.0.0.1
  is_local = 0;
  sockaddr_un sa;
  socklen_t sl = sizeof (sa);

  if (!getsockname (fd, (sockaddr *)&sa, &sl))
    is_local = sa.sun_family == AF_LOCAL;

  x_ev.start (fd, EVENT_READ);
  fcntl (fd, F_SETFD, FD_CLOEXEC);

  XSelectInput (dpy, root, PropertyChangeMask);

  flush ();

  return true;
}

void
rxvt_display::ref_next ()
{
  // TODO: somehow check wether the database files/resources changed
  // before affording re-loading/parsing
  XrmDestroyDatabase (XrmGetDatabase (dpy));
  XrmSetDatabase (dpy, get_resources (true));
}

rxvt_display::~rxvt_display ()
{
  if (!dpy)
    return;

#ifdef POINTER_BLANK
  XFreeCursor (dpy, blank_cursor);
#endif
  x_ev.stop ();
#ifdef USE_XIM
  xims.clear ();
#endif
  XCloseDisplay (dpy);
}

#ifdef USE_XIM
void rxvt_display::im_change_cb ()
{
  for (im_watcher **i = imw.begin (); i != imw.end (); ++i)
    (*i)->call ();
}

void rxvt_display::im_change_check ()
{
  // try to only call im_change_cb when a new input method
  // registers, as xlib crashes due to a race otherwise.
  Atom actual_type, *atoms;
  int actual_format;
  unsigned long nitems, bytes_after;

  if (XGetWindowProperty (dpy, root, xa[XA_XIM_SERVERS], 0L, 1000000L,
                          False, XA_ATOM, &actual_type, &actual_format,
                          &nitems, &bytes_after, (unsigned char **)&atoms)
      != Success )
    return;

  if (actual_type == XA_ATOM  && actual_format == 32)
    for (int i = 0; i < nitems; i++)
      if (XGetSelectionOwner (dpy, atoms[i]))
        {
          im_change_cb ();
          break;
        }

  XFree (atoms);
}
#endif

void rxvt_display::x_cb (io_watcher &w, short revents)
{
  do
    {
      XEvent xev;
      XNextEvent (dpy, &xev);

#ifdef USE_XIM
      if (!XFilterEvent (&xev, None))
        {
          if (xev.type == PropertyNotify
              && xev.xany.window == root
              && xev.xproperty.atom == xa[XA_XIM_SERVERS])
            im_change_check ();
#endif
          for (int i = xw.size (); i--; )
            {
              if (!xw[i])
                xw.erase_unordered (i);
              else if (xw[i]->window == xev.xany.window)
                xw[i]->call (xev);
            }
#ifdef USE_XIM
        }
#endif
    }
  while (XEventsQueued (dpy, QueuedAlready));

  XFlush (dpy);
}

void rxvt_display::flush ()
{
  if (XEventsQueued (dpy, QueuedAlready))
    x_cb (x_ev, EVENT_READ);

  XFlush (dpy);
}

void rxvt_display::reg (xevent_watcher *w)
{
  xw.push_back (w);
  w->active = xw.size ();
}

void rxvt_display::unreg (xevent_watcher *w)
{
  if (w->active)
    xw[w->active - 1] = 0;
}

void rxvt_display::set_selection_owner (rxvt_term *owner)
{
  if (selection_owner && selection_owner != owner)
    selection_owner->selection_clear ();

  selection_owner = owner;
}

#ifdef USE_XIM

void rxvt_display::reg (im_watcher *w)
{
  imw.push_back (w);
}

void rxvt_display::unreg (im_watcher *w)
{
  imw.erase (find (imw.begin (), imw.end (), w));
}

rxvt_xim *rxvt_display::get_xim (const char *locale, const char *modifiers)
{
  char *id;
  int l, m;

  l = strlen (locale);
  m = strlen (modifiers);

  if (!(id = (char *)malloc (l + m + 2)))
    return 0;

  memcpy (id, locale, l); id[l] = '\n';
  memcpy (id + l + 1, modifiers, m); id[l + m + 1] = 0;

  rxvt_xim *xim = xims.get (id);

  free (id);

  return xim;
}

void rxvt_display::put_xim (rxvt_xim *xim)
{
# if XLIB_IS_RACEFREE
  xims.put (xim);
# endif
}

#endif

Atom rxvt_display::atom (const char *name)
{
  return XInternAtom (dpy, name, False);
}

/////////////////////////////////////////////////////////////////////////////

template class refcache<rxvt_display>;
refcache<rxvt_display> displays;

/////////////////////////////////////////////////////////////////////////////
//

static unsigned int
insert_component (unsigned int value, unsigned int mask, unsigned int shift)
{
  return (value * (mask + 1) >> 16) << shift;
}
 
bool
rxvt_color::alloc (rxvt_screen *screen, const rgba &color)
{
#if XFT
  XRenderPictFormat *format;

  // FUCKING Xft gets it wrong, of course, so work around it.
  // Transparency users should eat shit and die, and then
  // XRenderQueryPictIndexValues themselves plenty.
  if ((screen->visual->c_class == TrueColor)
      && (format = XRenderFindVisualFormat (screen->dpy, screen->visual)))
    {
      // the fun lies in doing everything manually...
      c.color.red   = color.r;
      c.color.green = color.g;
      c.color.blue  = color.b;
      c.color.alpha = color.a;

      c.pixel = insert_component (color.r, format->direct.redMask  , format->direct.red  )
              | insert_component (color.g, format->direct.greenMask, format->direct.green)
              | insert_component (color.b, format->direct.blueMask , format->direct.blue )
              | insert_component (color.a, format->direct.alphaMask, format->direct.alpha);

      return true;
    }
  else
    {
      XRenderColor d;

      d.red   = color.r;
      d.green = color.g;
      d.blue  = color.b;
      d.alpha = color.a;

      return XftColorAllocValue (screen->dpy, screen->visual, screen->cmap, &d, &c);
    }
#else
  c.red   = color.r;
  c.green = color.g;
  c.blue  = color.b;

  if (screen->visual->c_class == TrueColor)
    {
      c.pixel = (color.r * (screen->visual->red_mask   >> ctz (screen->visual->red_mask  ))
                         / rgba::MAX_CC) << ctz (screen->visual->red_mask  )
              | (color.g * (screen->visual->green_mask >> ctz (screen->visual->green_mask))
                         / rgba::MAX_CC) << ctz (screen->visual->green_mask)
              | (color.b * (screen->visual->blue_mask  >> ctz (screen->visual->blue_mask ))
                         / rgba::MAX_CC) << ctz (screen->visual->blue_mask );

      return true;
    }
  else if (XAllocColor (screen->dpy, screen->cmap, &c))
    return true;
  else
    c.pixel = (color.r + color.g + color.b) > 128*3
            ? WhitePixelOfScreen (DefaultScreenOfDisplay (screen->dpy))
            : BlackPixelOfScreen (DefaultScreenOfDisplay (screen->dpy));
#endif

  return false;
}

bool
rxvt_color::set (rxvt_screen *screen, const char *name)
{
  rgba c;
  char eos;
  int skip;

  // parse the nonstandard "[alphapercent]" prefix
  if (1 <= sscanf (name, "[%hd]%n", &c.a, &skip))
    {
      c.a = lerp<int, int, int> (0, rgba::MAX_CC, c.a);
      name += skip;
    }
  else
    c.a = rgba::MAX_CC;

  // parse the non-standard "rgba:rrrr/gggg/bbbb/aaaa" format
  if (strlen (name) != 4+5*4 || 4 != sscanf (name, "rgba:%4hx/%4hx/%4hx/%4hx%c", &c.r, &c.g, &c.b, &c.a, &eos))
    {
      XColor xc, xc_exact;

      if (XParseColor (screen->dpy, screen->cmap, name, &xc))
        {
          c.r = xc.red;
          c.g = xc.green;
          c.b = xc.blue;
        }
      else
        {
          c.r = 0xffff;
          c.g = 0x6969;
          c.b = 0xb4b4;

          rxvt_warn ("unable to parse color '%s', using pink instead.\n", name);
        }
    }

  return set (screen, c);
}

bool
rxvt_color::set (rxvt_screen *screen, const rgba &color)
{
  bool got = alloc (screen, color);

#if !ENABLE_MINIMAL
  int cmap_size = screen->visual->map_entries;

  if (!got
      && screen->visual->c_class == PseudoColor
      && cmap_size < 4096)
    {
      XColor *colors = new XColor [screen->visual->map_entries];

      for (int i = 0; i < cmap_size; i++)
        colors [i].pixel = i;
 
      // many kilobytes transfer per colour, but pseudocolor isn't worth
      // many extra optimisations.
      XQueryColors (screen->dpy, screen->cmap, colors, cmap_size);

      int diff = 0x7fffffffUL;
      XColor *best = colors;

      for (int i = 0; i < cmap_size; i++)
        {
          int d = (squared_diff<int> (color.r >> 2, colors [i].red   >> 2))
                + (squared_diff<int> (color.g >> 2, colors [i].green >> 2))
                + (squared_diff<int> (color.b >> 2, colors [i].blue  >> 2));

          if (d < diff)
            {
              diff = d;
              best = colors + i;
            }
        }

      //rxvt_warn ("could not allocate %04x %04x %04x, getting %04x %04x %04x instead (%d)\n",
      //    color.r, color.g, color.b, best->red, best->green, best->blue, diff);
          
      got = alloc (screen, rgba (best->red, best->green, best->blue));

      delete colors;
    }
#endif

  return got;
}

void
rxvt_color::get (rgba &color)
{
#if XFT
  color.r = c.color.red;
  color.g = c.color.green;
  color.b = c.color.blue;
  color.a = c.color.alpha;
#else
  color.r = c.red;
  color.g = c.green;
  color.b = c.blue;
  color.a = rgba::MAX_CC;
#endif
}

void
rxvt_color::get (XColor &color)
{
  rgba c;
  get (c);

  color.red   = c.r;
  color.green = c.g;
  color.blue  = c.b;
  color.pixel = (Pixel)*this;
}

void 
rxvt_color::free (rxvt_screen *screen)
{
  if (screen->visual->c_class == TrueColor)
    return; // nothing to do

#if XFT
  XftColorFree (screen->dpy, screen->visual, screen->cmap, &c);
#else
  XFreeColors (screen->dpy, screen->cmap, &c.pixel, 1, AllPlanes);
#endif
}

void
rxvt_color::fade (rxvt_screen *screen, int percent, rxvt_color &result, const rgba &to)
{
  rgba c;
  get (c);

  result.set (
    screen,
    rgba (
      lerp (c.r, to.r, percent),
      lerp (c.g, to.g, percent),
      lerp (c.b, to.b, percent),
      lerp (c.a, to.a, percent)
    )
  );
}

Revision:	1.59
Committed:	Wed Jul 5 20:31:48 2006 UTC (17 years, 10 months ago) by root
Content type:	text/plain
Branch:	MAIN
Changes since 1.58:	+2 -4 lines
Log Message:	* empty log message *
#	Content
1	/----------------------------------------------------------------------
2	* File: rxvttoolkit.C
3	----------------------------------------------------------------------
4	*
5	* All portions of code are copyright by their respective author/s.
6	* Copyright (c) 2003-2006 Marc Lehmann <pcg@goof.com>
7	*
8	* This program is free software; you can redistribute it and/or modify
9	* it under the terms of the GNU General Public License as published by
10	* the Free Software Foundation; either version 2 of the License, or
11	* (at your option) any later version.
12	*
13	* This program is distributed in the hope that it will be useful,
14	* but WITHOUT ANY WARRANTY; without even the implied warranty of
15	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16	* GNU General Public License for more details.
17	*
18	* You should have received a copy of the GNU General Public License
19	* along with this program; if not, write to the Free Software
20	* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
21	----------------------------------------------------------------------/
22
23	#include "../config.h"
24	#include <rxvt.h>
25	#include <rxvttoolkit.h>
26
27	#include <unistd.h>
28	#include <fcntl.h>
29
30	#include <sys/utsname.h>
31	#include <sys/socket.h>
32	#include <sys/un.h>
33
34	#if XFT
35	# include <X11/extensions/Xrender.h>
36	#endif
37
38	const char *const xa_names[] =
39	{
40	"TEXT",
41	"COMPOUND_TEXT",
42	"UTF8_STRING",
43	"MULTIPLE",
44	"TARGETS",
45	"TIMESTAMP",
46	"VT_SELECTION",
47	"INCR",
48	"WM_PROTOCOLS",
49	"WM_DELETE_WINDOW",
50	"CLIPBOARD",
51	"AVERAGE_WIDTH",
52	"WEIGHT_NAME",
53	"SLANT",
54	"CHARSET_REGISTRY",
55	"CHARSET_ENCODING",
56	#if ENABLE_FRILLS
57	"_MOTIF_WM_HINTS",
58	#endif
59	#if ENABLE_EWMH
60	"_NET_WM_PID",
61	"_NET_WM_NAME",
62	"_NET_WM_ICON_NAME",
63	"_NET_WM_PING",
64	#endif
65	#if USE_XIM
66	"WM_LOCALE_NAME",
67	"XIM_SERVERS",
68	#endif
69	#ifdef TRANSPARENT
70	"_XROOTPMAP_ID",
71	"ESETROOT_PMAP_ID",
72	#endif
73	#if ENABLE_XEMBED
74	"_XEMBED",
75	"_XEMBED_INFO",
76	#endif
77	#if !ENABLE_MINIMAL
78	"SCREEN_RESOURCES",
79	"XDCCC_LINEAR_RGB_CORRECTION",
80	"XDCCC_LINEAR_RGB_MATRICES",
81	"WM_COLORMAP_WINDOWS",
82	"WM_STATE",
83	"cursor",
84	# if USE_XIM
85	"TRANSPORT",
86	"LOCALES",
87	"_XIM_PROTOCOL",
88	"_XIM_XCONNECT",
89	"_XIM_MOREDATA",
90	# endif
91	#endif
92	};
93
94	/////////////////////////////////////////////////////////////////////////////
95
96	refcounted::refcounted (const char *id)
97	{
98	this->id = strdup (id);
99	}
100
101	refcounted::~refcounted ()
102	{
103	free (id);
104	}
105
106	template<class T>
107	T refcache<T>::get (const char id)
108	{
109	for (T **i = this->begin (); i < this->end (); ++i)
110	{
111	if (!strcmp (id, (*i)->id))
112	{
113	++(*i)->referenced;
114	(*i)->ref_next ();
115	return *i;
116	}
117	}
118
119	T *obj = new T (id);
120
121	if (obj && obj->ref_init ())
122	{
123	obj->referenced = 1;
124	this->push_back (obj);
125	return obj;
126	}
127	else
128	{
129	delete obj;
130	return 0;
131	}
132	}
133
134	template<class T>
135	void refcache<T>::put (T *obj)
136	{
137	if (!obj)
138	return;
139
140	if (!--obj->referenced)
141	{
142	this->erase (find (this->begin (), this->end (), obj));
143	delete obj;
144	}
145	}
146
147	template<class T>
148	void refcache<T>::clear ()
149	{
150	while (this->size ())
151	put (*this->begin ());
152	}
153
154	/////////////////////////////////////////////////////////////////////////////
155
156	#ifdef USE_XIM
157
158	static void
159	#if XIMCB_PROTO_BROKEN
160	im_destroy_cb (XIC unused1, XPointer client_data, XPointer unused3)
161	#else
162	im_destroy_cb (XIM unused1, XPointer client_data, XPointer unused3)
163	#endif
164	{
165	rxvt_xim xim = (rxvt_xim )client_data;
166	rxvt_display *display = xim->display;
167
168	xim->xim = 0;
169
170	display->xims.erase (find (display->xims.begin (), display->xims.end (), xim));
171	display->im_change_cb ();
172	}
173
174	bool
175	rxvt_xim::ref_init ()
176	{
177	display = GET_R->display; //HACK: TODO
178
179	xim = XOpenIM (display->dpy, 0, 0, 0);
180
181	if (!xim)
182	return false;
183
184	XIMCallback ximcallback;
185	ximcallback.client_data = (XPointer)this;
186	ximcallback.callback = im_destroy_cb;
187
188	XSetIMValues (xim, XNDestroyCallback, &ximcallback, 0);
189
190	return true;
191	}
192
193	rxvt_xim::~rxvt_xim ()
194	{
195	if (xim)
196	XCloseIM (xim);
197	}
198
199	#endif
200
201	/////////////////////////////////////////////////////////////////////////////
202
203	#if XFT
204	rxvt_drawable::~rxvt_drawable ()
205	{
206	if (xftdrawable)
207	XftDrawDestroy (xftdrawable);
208	}
209
210	rxvt_drawable::operator XftDraw *()
211	{
212	if (!xftdrawable)
213	xftdrawable = XftDrawCreate (screen->dpy, drawable, screen->visual, screen->cmap);
214
215	return xftdrawable;
216	}
217	#endif
218
219	/////////////////////////////////////////////////////////////////////////////
220
221	#if XFT
222
223	// not strictly necessary as it is only used with superclass of zero_initialised
224	rxvt_screen::rxvt_screen ()
225	: scratch_area (0)
226	{
227	}
228
229	rxvt_drawable &rxvt_screen::scratch_drawable (int w, int h)
230	{
231	// it's actually faster to re-allocate every time. don't ask me
232	// why, but its likely no big deal there are no roundtrips
233	// (I think/hope).
234	if (!scratch_area \|\| w > scratch_w \|\| h > scratch_h \|\| 1/D/)
235	{
236	if (scratch_area)
237	{
238	XFreePixmap (dpy, scratch_area->drawable);
239	delete scratch_area;
240	}
241
242	Pixmap pm = XCreatePixmap (dpy, RootWindowOfScreen (ScreenOfDisplay (dpy, display->screen)),
243	scratch_w = w, scratch_h = h, depth);
244
245	scratch_area = new rxvt_drawable (this, pm);
246	}
247
248	return *scratch_area;
249	}
250
251	#endif
252
253	void
254	rxvt_screen::set (rxvt_display *disp)
255	{
256	display = disp;
257	dpy = disp->dpy;
258
259	Screen *screen = ScreenOfDisplay (dpy, disp->screen);
260
261	depth = DefaultDepthOfScreen (screen);
262	visual = DefaultVisualOfScreen (screen);
263	cmap = DefaultColormapOfScreen (screen);
264	}
265
266	void
267	rxvt_screen::select_visual (int bitdepth)
268	{
269	#if XFT
270	XVisualInfo vinfo;
271
272	if (XMatchVisualInfo (dpy, display->screen, bitdepth, TrueColor, &vinfo))
273	{
274	depth = bitdepth;
275	visual = vinfo.visual;
276	cmap = XCreateColormap (dpy, display->root, visual, AllocNone);
277	}
278	#endif
279	}
280
281	void
282	rxvt_screen::clear ()
283	{
284	#if XFT
285	if (scratch_area)
286	{
287	XFreePixmap (dpy, scratch_area->drawable);
288	delete scratch_area;
289	}
290	#endif
291
292	if (cmap != DefaultColormapOfScreen (ScreenOfDisplay (dpy, display->screen)))
293	XFreeColormap (dpy, cmap);
294	}
295
296	/////////////////////////////////////////////////////////////////////////////
297
298	rxvt_display::rxvt_display (const char *id)
299	: refcounted (id)
300	, x_ev (this, &rxvt_display::x_cb)
301	, selection_owner (0)
302	{
303	}
304
305	XrmDatabase
306	rxvt_display::get_resources (bool refresh)
307	{
308	char homedir = (char )getenv ("HOME");
309	char fname[1024];
310
311	/*
312	* get resources using the X library function
313	*/
314	char displayResource, xe;
315	XrmDatabase rdb1, database = 0;
316
317	// for ordering, see for example http://www.faqs.org/faqs/Xt-FAQ/ Subject: 20
318
319	// 6. System wide per application default file.
320
321	/* Add in $XAPPLRESDIR/Rxvt only; not bothering with XUSERFILESEARCHPATH */
322	if ((xe = (char *)getenv ("XAPPLRESDIR")))
323	{
324	snprintf (fname, sizeof (fname), "%s/%s", xe, RESCLASS);
325
326	if ((rdb1 = XrmGetFileDatabase (fname)))
327	XrmMergeDatabases (rdb1, &database);
328	}
329
330	// 5. User's per application default file.
331	// none
332
333	// 4. User's defaults file.
334	/* Get any Xserver defaults */
335	if (refresh)
336	{
337	// fucking xlib keeps a copy of the rm string
338	Atom actual_type;
339	int actual_format;
340	unsigned long nitems, nremaining;
341	char *val = 0;
342
343	#if XLIB_ILLEGAL_ACCESS
344	if (dpy->xdefaults)
345	XFree (dpy->xdefaults);
346	#endif
347
348	if (XGetWindowProperty (dpy, RootWindow (dpy, 0), XA_RESOURCE_MANAGER,
349	0L, 100000000L, False,
350	XA_STRING, &actual_type, &actual_format,
351	&nitems, &nremaining,
352	(unsigned char **)&val) == Success
353	&& actual_type == XA_STRING
354	&& actual_format == 8)
355	displayResource = val;
356	else
357	{
358	displayResource = 0;
359	if (val)
360	XFree(val);
361	}
362
363	#if XLIB_ILLEGAL_ACCESS
364	dpy->xdefaults = displayResource;
365	#endif
366	}
367	else
368	displayResource = XResourceManagerString (dpy);
369
370	if (displayResource)
371	{
372	if ((rdb1 = XrmGetStringDatabase (displayResource)))
373	XrmMergeDatabases (rdb1, &database);
374	}
375	else if (homedir)
376	{
377	snprintf (fname, sizeof (fname), "%s/.Xdefaults", homedir);
378
379	if ((rdb1 = XrmGetFileDatabase (fname)))
380	XrmMergeDatabases (rdb1, &database);
381	}
382
383	#if !XLIB_ILLEGAL_ACCESS
384	if (refresh && displayResource)
385	XFree (displayResource);
386	#endif
387
388	/* Get screen specific resources */
389	displayResource = XScreenResourceString (ScreenOfDisplay (dpy, screen));
390
391	if (displayResource)
392	{
393	if ((rdb1 = XrmGetStringDatabase (displayResource)))
394	/* Merge with screen-independent resources */
395	XrmMergeDatabases (rdb1, &database);
396
397	XFree (displayResource);
398	}
399
400	// 3. User's per host defaults file
401	/* Add in XENVIRONMENT file */
402	if ((xe = (char *)getenv ("XENVIRONMENT"))
403	&& (rdb1 = XrmGetFileDatabase (xe)))
404	XrmMergeDatabases (rdb1, &database);
405	else if (homedir)
406	{
407	struct utsname un;
408
409	if (!uname (&un))
410	{
411	snprintf (fname, sizeof (fname), "%s/.Xdefaults-%s", homedir, un.nodename);
412
413	if ((rdb1 = XrmGetFileDatabase (fname)))
414	XrmMergeDatabases (rdb1, &database);
415	}
416	}
417
418	return database;
419	}
420
421	bool rxvt_display::ref_init ()
422	{
423	#ifdef LOCAL_X_IS_UNIX
424	if (id[0] == ':')
425	{
426	val = rxvt_malloc (5 + strlen (id) + 1);
427	strcpy (val, "unix/");
428	strcat (val, id);
429	dpy = XOpenDisplay (val);
430	free (val);
431	}
432	else
433	#endif
434	dpy = 0;
435
436	if (!dpy)
437	dpy = XOpenDisplay (id);
438
439	if (!dpy)
440	return false;
441
442	screen = DefaultScreen (dpy);
443	root = DefaultRootWindow (dpy);
444
445	assert (sizeof (xa_names) / sizeof (char *) == NUM_XA);
446	XInternAtoms (dpy, (char **)xa_names, NUM_XA, False, xa);
447
448	XrmSetDatabase (dpy, get_resources (false));
449
450	#ifdef POINTER_BLANK
451	XColor blackcolour;
452	blackcolour.red = 0;
453	blackcolour.green = 0;
454	blackcolour.blue = 0;
455	Font f = XLoadFont (dpy, "fixed");
456	blank_cursor = XCreateGlyphCursor (dpy, f, f, ' ', ' ',
457	&blackcolour, &blackcolour);
458	XUnloadFont (dpy, f);
459	#endif
460
461	int fd = XConnectionNumber (dpy);
462
463	// try to detect wether we have a local connection.
464	// assume unix domains socket == local, everything else not
465	// TODO: might want to check for inet/127.0.0.1
466	is_local = 0;
467	sockaddr_un sa;
468	socklen_t sl = sizeof (sa);
469
470	if (!getsockname (fd, (sockaddr *)&sa, &sl))
471	is_local = sa.sun_family == AF_LOCAL;
472
473	x_ev.start (fd, EVENT_READ);
474	fcntl (fd, F_SETFD, FD_CLOEXEC);
475
476	XSelectInput (dpy, root, PropertyChangeMask);
477
478	flush ();
479
480	return true;
481	}
482
483	void
484	rxvt_display::ref_next ()
485	{
486	// TODO: somehow check wether the database files/resources changed
487	// before affording re-loading/parsing
488	XrmDestroyDatabase (XrmGetDatabase (dpy));
489	XrmSetDatabase (dpy, get_resources (true));
490	}
491
492	rxvt_display::~rxvt_display ()
493	{
494	if (!dpy)
495	return;
496
497	#ifdef POINTER_BLANK
498	XFreeCursor (dpy, blank_cursor);
499	#endif
500	x_ev.stop ();
501	#ifdef USE_XIM
502	xims.clear ();
503	#endif
504	XCloseDisplay (dpy);
505	}
506
507	#ifdef USE_XIM
508	void rxvt_display::im_change_cb ()
509	{
510	for (im_watcher **i = imw.begin (); i != imw.end (); ++i)
511	(*i)->call ();
512	}
513
514	void rxvt_display::im_change_check ()
515	{
516	// try to only call im_change_cb when a new input method
517	// registers, as xlib crashes due to a race otherwise.
518	Atom actual_type, *atoms;
519	int actual_format;
520	unsigned long nitems, bytes_after;
521
522	if (XGetWindowProperty (dpy, root, xa[XA_XIM_SERVERS], 0L, 1000000L,
523	False, XA_ATOM, &actual_type, &actual_format,
524	&nitems, &bytes_after, (unsigned char **)&atoms)
525	!= Success )
526	return;
527
528	if (actual_type == XA_ATOM && actual_format == 32)
529	for (int i = 0; i < nitems; i++)
530	if (XGetSelectionOwner (dpy, atoms[i]))
531	{
532	im_change_cb ();
533	break;
534	}
535
536	XFree (atoms);
537	}
538	#endif
539
540	void rxvt_display::x_cb (io_watcher &w, short revents)
541	{
542	do
543	{
544	XEvent xev;
545	XNextEvent (dpy, &xev);
546
547	#ifdef USE_XIM
548	if (!XFilterEvent (&xev, None))
549	{
550	if (xev.type == PropertyNotify
551	&& xev.xany.window == root
552	&& xev.xproperty.atom == xa[XA_XIM_SERVERS])
553	im_change_check ();
554	#endif
555	for (int i = xw.size (); i--; )
556	{
557	if (!xw[i])
558	xw.erase_unordered (i);
559	else if (xw[i]->window == xev.xany.window)
560	xw[i]->call (xev);
561	}
562	#ifdef USE_XIM
563	}
564	#endif
565	}
566	while (XEventsQueued (dpy, QueuedAlready));
567
568	XFlush (dpy);
569	}
570
571	void rxvt_display::flush ()
572	{
573	if (XEventsQueued (dpy, QueuedAlready))
574	x_cb (x_ev, EVENT_READ);
575
576	XFlush (dpy);
577	}
578
579	void rxvt_display::reg (xevent_watcher *w)
580	{
581	xw.push_back (w);
582	w->active = xw.size ();
583	}
584
585	void rxvt_display::unreg (xevent_watcher *w)
586	{
587	if (w->active)
588	xw[w->active - 1] = 0;
589	}
590
591	void rxvt_display::set_selection_owner (rxvt_term *owner)
592	{
593	if (selection_owner && selection_owner != owner)
594	selection_owner->selection_clear ();
595
596	selection_owner = owner;
597	}
598
599	#ifdef USE_XIM
600
601	void rxvt_display::reg (im_watcher *w)
602	{
603	imw.push_back (w);
604	}
605
606	void rxvt_display::unreg (im_watcher *w)
607	{
608	imw.erase (find (imw.begin (), imw.end (), w));
609	}
610
611	rxvt_xim rxvt_display::get_xim (const char locale, const char *modifiers)
612	{
613	char *id;
614	int l, m;
615
616	l = strlen (locale);
617	m = strlen (modifiers);
618
619	if (!(id = (char *)malloc (l + m + 2)))
620	return 0;
621
622	memcpy (id, locale, l); id[l] = '\n';
623	memcpy (id + l + 1, modifiers, m); id[l + m + 1] = 0;
624
625	rxvt_xim *xim = xims.get (id);
626
627	free (id);
628
629	return xim;
630	}
631
632	void rxvt_display::put_xim (rxvt_xim *xim)
633	{
634	# if XLIB_IS_RACEFREE
635	xims.put (xim);
636	# endif
637	}
638
639	#endif
640
641	Atom rxvt_display::atom (const char *name)
642	{
643	return XInternAtom (dpy, name, False);
644	}
645
646	/////////////////////////////////////////////////////////////////////////////
647
648	template class refcache<rxvt_display>;
649	refcache<rxvt_display> displays;
650
651	/////////////////////////////////////////////////////////////////////////////
652	//
653
654	static unsigned int
655	insert_component (unsigned int value, unsigned int mask, unsigned int shift)
656	{
657	return (value * (mask + 1) >> 16) << shift;
658	}
659
660	bool
661	rxvt_color::alloc (rxvt_screen *screen, const rgba &color)
662	{
663	#if XFT
664	XRenderPictFormat *format;
665
666	// FUCKING Xft gets it wrong, of course, so work around it.
667	// Transparency users should eat shit and die, and then
668	// XRenderQueryPictIndexValues themselves plenty.
669	if ((screen->visual->c_class == TrueColor)
670	&& (format = XRenderFindVisualFormat (screen->dpy, screen->visual)))
671	{
672	// the fun lies in doing everything manually...
673	c.color.red = color.r;
674	c.color.green = color.g;
675	c.color.blue = color.b;
676	c.color.alpha = color.a;
677
678	c.pixel = insert_component (color.r, format->direct.redMask , format->direct.red )
679	\| insert_component (color.g, format->direct.greenMask, format->direct.green)
680	\| insert_component (color.b, format->direct.blueMask , format->direct.blue )
681	\| insert_component (color.a, format->direct.alphaMask, format->direct.alpha);
682
683	return true;
684	}
685	else
686	{
687	XRenderColor d;
688
689	d.red = color.r;
690	d.green = color.g;
691	d.blue = color.b;
692	d.alpha = color.a;
693
694	return XftColorAllocValue (screen->dpy, screen->visual, screen->cmap, &d, &c);
695	}
696	#else
697	c.red = color.r;
698	c.green = color.g;
699	c.blue = color.b;
700
701	if (screen->visual->c_class == TrueColor)
702	{
703	c.pixel = (color.r * (screen->visual->red_mask >> ctz (screen->visual->red_mask ))
704	/ rgba::MAX_CC) << ctz (screen->visual->red_mask )
705	\| (color.g * (screen->visual->green_mask >> ctz (screen->visual->green_mask))
706	/ rgba::MAX_CC) << ctz (screen->visual->green_mask)
707	\| (color.b * (screen->visual->blue_mask >> ctz (screen->visual->blue_mask ))
708	/ rgba::MAX_CC) << ctz (screen->visual->blue_mask );
709
710	return true;
711	}
712	else if (XAllocColor (screen->dpy, screen->cmap, &c))
713	return true;
714	else
715	c.pixel = (color.r + color.g + color.b) > 128*3
716	? WhitePixelOfScreen (DefaultScreenOfDisplay (screen->dpy))
717	: BlackPixelOfScreen (DefaultScreenOfDisplay (screen->dpy));
718	#endif
719
720	return false;
721	}
722
723	bool
724	rxvt_color::set (rxvt_screen screen, const char name)
725	{
726	rgba c;
727	char eos;
728	int skip;
729
730	// parse the nonstandard "[alphapercent]" prefix
731	if (1 <= sscanf (name, "[%hd]%n", &c.a, &skip))
732	{
733	c.a = lerp<int, int, int> (0, rgba::MAX_CC, c.a);
734	name += skip;
735	}
736	else
737	c.a = rgba::MAX_CC;
738
739	// parse the non-standard "rgba:rrrr/gggg/bbbb/aaaa" format
740	if (strlen (name) != 4+5*4 \|\| 4 != sscanf (name, "rgba:%4hx/%4hx/%4hx/%4hx%c", &c.r, &c.g, &c.b, &c.a, &eos))
741	{
742	XColor xc, xc_exact;
743
744	if (XParseColor (screen->dpy, screen->cmap, name, &xc))
745	{
746	c.r = xc.red;
747	c.g = xc.green;
748	c.b = xc.blue;
749	}
750	else
751	{
752	c.r = 0xffff;
753	c.g = 0x6969;
754	c.b = 0xb4b4;
755
756	rxvt_warn ("unable to parse color '%s', using pink instead.\n", name);
757	}
758	}
759
760	return set (screen, c);
761	}
762
763	bool
764	rxvt_color::set (rxvt_screen *screen, const rgba &color)
765	{
766	bool got = alloc (screen, color);
767
768	#if !ENABLE_MINIMAL
769	int cmap_size = screen->visual->map_entries;
770
771	if (!got
772	&& screen->visual->c_class == PseudoColor
773	&& cmap_size < 4096)
774	{
775	XColor *colors = new XColor [screen->visual->map_entries];
776
777	for (int i = 0; i < cmap_size; i++)
778	colors [i].pixel = i;
779
780	// many kilobytes transfer per colour, but pseudocolor isn't worth
781	// many extra optimisations.
782	XQueryColors (screen->dpy, screen->cmap, colors, cmap_size);
783
784	int diff = 0x7fffffffUL;
785	XColor *best = colors;
786
787	for (int i = 0; i < cmap_size; i++)
788	{
789	int d = (squared_diff<int> (color.r >> 2, colors [i].red >> 2))
790	+ (squared_diff<int> (color.g >> 2, colors [i].green >> 2))
791	+ (squared_diff<int> (color.b >> 2, colors [i].blue >> 2));
792
793	if (d < diff)
794	{
795	diff = d;
796	best = colors + i;
797	}
798	}
799
800	//rxvt_warn ("could not allocate %04x %04x %04x, getting %04x %04x %04x instead (%d)\n",
801	// color.r, color.g, color.b, best->red, best->green, best->blue, diff);
802
803	got = alloc (screen, rgba (best->red, best->green, best->blue));
804
805	delete colors;
806	}
807	#endif
808
809	return got;
810	}
811
812	void
813	rxvt_color::get (rgba &color)
814	{
815	#if XFT
816	color.r = c.color.red;
817	color.g = c.color.green;
818	color.b = c.color.blue;
819	color.a = c.color.alpha;
820	#else
821	color.r = c.red;
822	color.g = c.green;
823	color.b = c.blue;
824	color.a = rgba::MAX_CC;
825	#endif
826	}
827
828	void
829	rxvt_color::get (XColor &color)
830	{
831	rgba c;
832	get (c);
833
834	color.red = c.r;
835	color.green = c.g;
836	color.blue = c.b;
837	color.pixel = (Pixel)*this;
838	}
839
840	void
841	rxvt_color::free (rxvt_screen *screen)
842	{
843	if (screen->visual->c_class == TrueColor)
844	return; // nothing to do
845
846	#if XFT
847	XftColorFree (screen->dpy, screen->visual, screen->cmap, &c);
848	#else
849	XFreeColors (screen->dpy, screen->cmap, &c.pixel, 1, AllPlanes);
850	#endif
851	}
852
853	void
854	rxvt_color::fade (rxvt_screen *screen, int percent, rxvt_color &result, const rgba &to)
855	{
856	rgba c;
857	get (c);
858
859	result.set (
860	screen,
861	rgba (
862	lerp (c.r, to.r, percent),
863	lerp (c.g, to.g, percent),
864	lerp (c.b, to.b, percent),
865	lerp (c.a, to.a, percent)
866	)
867	);
868	}
869