trunk/rdesktop/xwin.c

/*
   rdesktop: A Remote Desktop Protocol client.
   User interface services - X Window System
   Copyright (C) Matthew Chapman 1999-2001

   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 <X11/Xlib.h>
#include <X11/Xutil.h>
#include <X11/XKBlib.h>
#include <time.h>
#include <errno.h>
#include "rdesktop.h"

extern char keymapname[16];
extern int keylayout;
extern int width;
extern int height;
extern BOOL sendmotion;
extern BOOL fullscreen;

Display *display;
XkbDescPtr xkb;
static int x_socket;
static Window wnd;
static GC gc;
static Visual *visual;
static int depth;
static int bpp;

/* endianness */
static BOOL host_be;
static BOOL xserver_be;

/* software backing store */
static BOOL ownbackstore;
static Pixmap backstore;

/* needed to keep track of the modifiers */
static unsigned int key_modifier_state = 0;
static unsigned int key_down_state = 0;

#define DShift1Mask   (1<<0)
#define DShift2Mask   (1<<1)
#define DControl1Mask (1<<2)
#define DControl2Mask (1<<3)
#define DMod1Mask     (1<<4)
#define DMod2Mask     (1<<5)

#define FILL_RECTANGLE(x,y,cx,cy)\
{ \
        XFillRectangle(display, wnd, gc, x, y, cx, cy); \
        if (ownbackstore) \
                XFillRectangle(display, backstore, gc, x, y, cx, cy); \
}

/* colour maps */
static BOOL owncolmap;
static Colormap xcolmap;
static uint32 white;
static uint32 *colmap;

#define TRANSLATE(col)          ( owncolmap ? col : translate_colour(colmap[col]) )
#define SET_FOREGROUND(col)     XSetForeground(display, gc, TRANSLATE(col));
#define SET_BACKGROUND(col)     XSetBackground(display, gc, TRANSLATE(col));

static int rop2_map[] = {
        GXclear,                /* 0 */
        GXnor,                  /* DPon */
        GXandInverted,          /* DPna */
        GXcopyInverted,         /* Pn */
        GXandReverse,           /* PDna */
        GXinvert,               /* Dn */
        GXxor,                  /* DPx */
        GXnand,                 /* DPan */
        GXand,                  /* DPa */
        GXequiv,                /* DPxn */
        GXnoop,                 /* D */
        GXorInverted,           /* DPno */
        GXcopy,                 /* P */
        GXorReverse,            /* PDno */
        GXor,                   /* DPo */
        GXset                   /* 1 */
};

#define SET_FUNCTION(rop2)      { if (rop2 != ROP2_COPY) XSetFunction(display, gc, rop2_map[rop2]); }
#define RESET_FUNCTION(rop2)    { if (rop2 != ROP2_COPY) XSetFunction(display, gc, GXcopy); }

void xwin_release_modifiers(XKeyEvent* ev, uint32 ev_time, uint32 scancode);
void xwin_press_modifiers(XKeyEvent* ev, uint32 ev_time, uint32 scancode);

static void
translate8(uint8 *data, uint8 *out, uint8 *end)
{
        while (out < end)
                *(out++) = (uint8)colmap[*(data++)];
}

static void
translate16(uint8 *data, uint16 *out, uint16 *end)
{
        while (out < end)
                *(out++) = (uint16)colmap[*(data++)];
}

/* little endian - conversion happens when colourmap is built */
static void
translate24(uint8 *data, uint8 *out, uint8 *end)
{
        uint32 value;

        while (out < end)
        {
                value = colmap[*(data++)];
                *(out++) = value;
                *(out++) = value >> 8;
                *(out++) = value >> 16;
        }
}

static void
translate32(uint8 *data, uint32 *out, uint32 *end)
{
        while (out < end)
                *(out++) = colmap[*(data++)];
}

static uint8 *
translate_image(int width, int height, uint8 *data)
{
        int size = width * height * bpp/8;
        uint8 *out = xmalloc(size);
        uint8 *end = out + size;

        switch (bpp)
        {
                case 8:
                        translate8(data, out, end);
                        break;

                case 16:
                        translate16(data, (uint16 *)out, (uint16 *)end);
                        break;

                case 24:
                        translate24(data, out, end);
                        break;

                case 32:
                        translate32(data, (uint32 *)out, (uint32 *)end);
                        break;
        }

        return out;
}

#define BSWAP16(x) { x = (((x & 0xff) << 8) | (x >> 8)); }
#define BSWAP24(x) { x = (((x & 0xff) << 16) | (x >> 16) | ((x >> 8) & 0xff00)); }
#define BSWAP32(x) { x = (((x & 0xff00ff) << 8) | ((x >> 8) & 0xff00ff)); \
                        x = (x << 16) | (x >> 16); }

static uint32
translate_colour(uint32 colour)
{
        switch (bpp)
        {
                case 16:
                        if (host_be != xserver_be)
                                BSWAP16(colour);
                        break;

                case 24:
                        if (xserver_be)
                                BSWAP24(colour);
                        break;

                case 32:
                        if (host_be != xserver_be)
                                BSWAP32(colour);
                        break;
        }

        return colour;
}

BOOL
ui_create_window(char *title)
{
        XSetWindowAttributes attribs;
        XClassHint *classhints;
        XSizeHints *sizehints;
        unsigned long input_mask;
        XPixmapFormatValues *pfm;
        Screen *screen;
        uint16 test;
        int i;
        
        int xkb_minor, xkb_major;
        int xkb_event, xkb_error, xkb_reason;

        /* compare compiletime libs with runtime libs. */
        xkb_major = XkbMajorVersion;
        xkb_minor = XkbMinorVersion;
        if( XkbLibraryVersion( &xkb_major, &xkb_minor ) == False )
        {
                error("please re-compile rdesktop\ncompile time version of xkb is not compatible with\nyour runtime version of the library\n");
                return False;
        }


        /* XKB is the 'new' keyboard handler in x.. ( the xkb code in Xfree86 originates from SGI, years 1993 and 1995 from what I could tell. )
         * it makes it possible for people with disabilities to use rdesktop, stickykeys, bouncekeys etc. VERY MUCH useful.
         * XFree86 has had support for it since it's earliest incarnation. I believe it is a reasonable dependency.
         */
        display = XkbOpenDisplay( NULL, &xkb_event, &xkb_error, &xkb_major, &xkb_minor, &xkb_reason );
        switch(xkb_reason)
        {
                case XkbOD_BadLibraryVersion:
                        error("XkbOD_BadLibraryVersion: XKB extensions in server and the library rdesktop is linked against aren't compatible with each other.\n");
                        break;
                case XkbOD_ConnectionRefused:
                        error("XkbOD_ConnectionRefused\n");
                        break;
                case XkbOD_BadServerVersion:
                        error("XkbOD_BadServerVersion\n");
                        break;
                case XkbOD_NonXkbServer:
                        error("XkbOD_NonXkbServer: XKB extension not present in server\nupdate your X server.\n");
                        break;
                case XkbOD_Success:
                        DEBUG("XkbOD_Success: Connection established with display\n");
                        break;
        }

        if (display == NULL)
        {
                error("Failed to open display\n");
                return False;
        }

        x_socket = ConnectionNumber(display);
        screen = DefaultScreenOfDisplay(display);
        visual = DefaultVisualOfScreen(screen);
        depth = DefaultDepthOfScreen(screen);
        
        pfm = XListPixmapFormats(display, &i);
        if (pfm != NULL)
        {
                /* Use maximum bpp for this depth - this is generally
                   desirable, e.g. 24 bits->32 bits. */
                while (i--)
                {
                        if ((pfm[i].depth == depth)
                            && (pfm[i].bits_per_pixel > bpp))
                        {
                                bpp = pfm[i].bits_per_pixel;
                        }
                }
                XFree(pfm);
        }

        if (bpp < 8)
        {
                error("Less than 8 bpp not currently supported.\n");
                XCloseDisplay(display);
                return False;
        }

        if (depth <= 8)
                owncolmap = True;
        else
                xcolmap = DefaultColormapOfScreen(screen);

        test = 1;
        host_be = !(BOOL)(*(uint8 *)(&test));
        xserver_be = (ImageByteOrder(display) == MSBFirst);

        white = WhitePixelOfScreen(screen);
        attribs.background_pixel = BlackPixelOfScreen(screen);
        attribs.backing_store = DoesBackingStore(screen);

        if (attribs.backing_store == NotUseful)
                ownbackstore = True;

        if (fullscreen)
        {
                attribs.override_redirect = True;
                width = WidthOfScreen(screen);
                height = HeightOfScreen(screen);
        }
        else
        {
                attribs.override_redirect = False;
        }

        width = (width + 3) & ~3; /* make width a multiple of 32 bits */

        wnd = XCreateWindow(display, RootWindowOfScreen(screen),
                            0, 0, width, height, 0, CopyFromParent,
                            InputOutput, CopyFromParent,
                            CWBackingStore | CWBackPixel | CWOverrideRedirect,
                            &attribs);

        XStoreName(display, wnd, title);

        classhints = XAllocClassHint();
        if (classhints != NULL)
        {
                classhints->res_name = classhints->res_class = "rdesktop";
                XSetClassHint(display, wnd, classhints);
                XFree(classhints);
        }

        sizehints = XAllocSizeHints();
        if (sizehints)
        {
                sizehints->flags = PMinSize | PMaxSize;
                sizehints->min_width = sizehints->max_width = width;
                sizehints->min_height = sizehints->max_height = height;
                XSetWMNormalHints(display, wnd, sizehints);
                XFree(sizehints);
        }

        xkeymap_init();

        input_mask = KeyPressMask | KeyReleaseMask |
                         ButtonPressMask | ButtonReleaseMask |
                         EnterWindowMask | LeaveWindowMask | KeymapStateMask;
        if (sendmotion)
                input_mask |= PointerMotionMask;

        if (ownbackstore)
                input_mask |= ExposureMask;

        XSelectInput(display, wnd, input_mask);
        gc = XCreateGC(display, wnd, 0, NULL);

        if (ownbackstore)
                backstore = XCreatePixmap(display, wnd, width, height, depth);

        XMapWindow(display, wnd);

        /* TODO: error texts... make them friendly. */
        xkb = XkbGetKeyboard(display, XkbAllComponentsMask, XkbUseCoreKbd);
        if ((int)xkb == BadAlloc || xkb == NULL)
        {
                        error( "XkbGetKeyboard failed.\n");
                        exit(0);
        }

        /* TODO: error texts... make them friendly. */
        if( XkbSelectEvents(display, xkb->device_spec, XkbAllEventsMask, XkbAllEventsMask) == False )
        {
                        error( "XkbSelectEvents failed.\n");
                        exit(0);
        }

        return True;
}

void
ui_destroy_window()
{
        if( xkb != NULL )
                XkbFreeKeyboard(xkb, XkbAllControlsMask, True);

        if (ownbackstore)
                XFreePixmap(display, backstore);

        XFreeGC(display, gc);
        XDestroyWindow(display, wnd);
        XCloseDisplay(display);
        display = NULL;
}

static void
xwin_process_events()
{
        XkbEvent xkbevent;
        
        KeySym keysym;
        uint8 scancode;
        uint16 button, flags;
        uint32 ev_time;
        uint32 tmpmods;

        while (XCheckMaskEvent(display, ~0, &xkbevent.core))
        {
                ev_time = time(NULL);
                flags = 0;

                switch (xkbevent.type)
                {
                        case KeymapNotify:
                                /* TODO:
                                 * read modifier status at focus in, and update the local masks, and the other end as well..
                                 * if not, we may get out of sync.
                                 * xkbevent.core.xkeymap.key_vector
                                 * char key_vector[32]; 
                                 */
                                break;

                        case KeyRelease:
                                flags = KBD_FLAG_DOWN | KBD_FLAG_UP;
                                /* fall through */

                        case KeyPress:
                                if( XkbTranslateKeyCode(xkb, xkbevent.core.xkey.keycode, xkbevent.core.xkey.state, &tmpmods, &keysym) == False )
                                        break;
                                scancode = xkeymap_translate_key(keysym, xkbevent.core.xkey.keycode, &flags);

                                if (scancode == 0 )
                                        break;

                                /* keep track of the modifiers -- needed for stickykeys... */
                                if( xkbevent.type == KeyPress )
                                        xwin_press_modifiers( &xkbevent.core.xkey, ev_time, scancode );

                                rdp_send_input(ev_time, RDP_INPUT_SCANCODE, flags, scancode, 0);

                                if( xkbevent.type == KeyRelease )
                                        xwin_release_modifiers( &xkbevent.core.xkey, ev_time, scancode );

                                break;

                        case ButtonPress:
                                flags = MOUSE_FLAG_DOWN;
                                /* fall through */

                        case ButtonRelease:
                                button = xkeymap_translate_button(xkbevent.core.xbutton.button);
                                if (button == 0)
                                        break;

                                rdp_send_input(ev_time, RDP_INPUT_MOUSE,
                                               flags | button,
                                               xkbevent.core.xbutton.x,
                                               xkbevent.core.xbutton.y);
                                break;

                        case MotionNotify:
                                rdp_send_input(ev_time, RDP_INPUT_MOUSE,
                                               MOUSE_FLAG_MOVE,
                                               xkbevent.core.xmotion.x,
                                               xkbevent.core.xmotion.y);
                                break;

                        case EnterNotify:
                                XGrabKeyboard(display, wnd, True, GrabModeAsync,
                                              GrabModeAsync, CurrentTime);
                                break;

                        case LeaveNotify:
                                XUngrabKeyboard(display, CurrentTime);
                                break;

                        case Expose:
                                XCopyArea(display, backstore, wnd, gc,
                                          xkbevent.core.xexpose.x, xkbevent.core.xexpose.y,
                                          xkbevent.core.xexpose.width, xkbevent.core.xexpose.height,
                                          xkbevent.core.xexpose.x, xkbevent.core.xexpose.y);
                                break;
                }
        }
}

void
xwin_release_modifiers(XKeyEvent* ev, uint32 ev_time, uint32 scancode)
{
        switch (scancode) {
        case 0x2a:
                key_down_state &= ~DShift1Mask;
                break;
        case 0x36:
                key_down_state &= ~DShift2Mask;
                break;
        case 0x1d:
                key_down_state &= ~DControl1Mask;
                break;
        case 0x9d:
                key_down_state &= ~DControl2Mask;
                break;
        case 0x38:
                key_down_state &= ~DMod1Mask;
                break;
        case 0xb8:
                key_down_state &= ~DMod2Mask;
                break;
        }

        if( !(ShiftMask & ev->state) && (key_down_state & DShift1Mask))
        {
                rdp_send_input(ev_time, RDP_INPUT_SCANCODE, KBD_FLAG_UP, 0x2a, 0);
                key_down_state &= ~DShift1Mask;

        }

        if( !(ControlMask & ev->state) && (key_down_state & DControl1Mask))
        {
                rdp_send_input(ev_time, RDP_INPUT_SCANCODE, KBD_FLAG_UP, 0x1d, 0);
                key_down_state &= ~DControl1Mask;

        }
        
        if( !(Mod1Mask & ev->state) && (key_down_state & DMod1Mask))
        {
                rdp_send_input(ev_time, RDP_INPUT_SCANCODE, KBD_FLAG_UP, 0x38, 0);
                key_down_state &= ~DMod1Mask;

        }
        
        if( !(Mod2Mask & ev->state) && (key_down_state & DMod2Mask))
        {
                rdp_send_input(ev_time, RDP_INPUT_SCANCODE, KBD_FLAG_UP, 0xb8, 0);
                key_down_state &= ~DMod2Mask;
        }
}


void
xwin_press_modifiers(XKeyEvent* ev, uint32 ev_time, uint32 scancode)
{
        key_modifier_state = ev->state;

        switch (scancode) {
        case 0x2a:
                key_down_state |= DShift1Mask;
                break;
        case 0x36:
                key_down_state |= DShift2Mask;
                break;
        case 0x1d:
                key_down_state |= DControl1Mask;
                break;
        case 0x9d:
                key_down_state |= DControl2Mask;
                break;
        case 0x38:
                key_down_state |= DMod1Mask;
                break;
        case 0xb8:
                key_down_state |= DMod2Mask;
                break;
        }

        if( (ShiftMask & ev->state) && !((key_down_state & DShift1Mask) || (key_down_state & DShift2Mask)))
        {
                rdp_send_input(ev_time, RDP_INPUT_SCANCODE, KBD_FLAG_DOWN, 0x2a, 0);
                key_down_state |= DShift1Mask;

        }

        if( (ControlMask & ev->state) && !((key_down_state & DControl1Mask) || (key_down_state & DControl2Mask)))
        {
                rdp_send_input(ev_time, RDP_INPUT_SCANCODE, KBD_FLAG_DOWN, 0x1d, 0);
                key_down_state |= DControl1Mask;

        }

        if( (Mod1Mask & ev->state) && !(key_down_state & DMod1Mask))
        {
                rdp_send_input(ev_time, RDP_INPUT_SCANCODE, KBD_FLAG_DOWN, 0x38, 0);
                key_down_state |= DMod1Mask;

        }

        if( (Mod2Mask & ev->state) && !(key_down_state & DMod2Mask))
        {
                rdp_send_input(ev_time, RDP_INPUT_SCANCODE, KBD_FLAG_DOWN, 0xb8, 0);
                key_down_state |= DMod2Mask;

        }
}

void
ui_select(int rdp_socket)
{
        int n = (rdp_socket > x_socket) ? rdp_socket+1 : x_socket+1;
        fd_set rfds;

        FD_ZERO(&rfds);

        while (True)
        {
                FD_ZERO(&rfds);
                FD_SET(rdp_socket, &rfds);
                if (display != NULL)
                {
                        FD_SET(x_socket, &rfds);
                        XFlush(display);
                }

                switch (select(n, &rfds, NULL, NULL, NULL))
                {
                        case -1:
                                error("select: %s\n", strerror(errno));

                        case 0:
                                continue;
                }

                if (FD_ISSET(x_socket, &rfds))
                        xwin_process_events();

                if (FD_ISSET(rdp_socket, &rfds))
                        return;
        }
}

void
ui_move_pointer(int x, int y)
{
        XWarpPointer(display, wnd, wnd, 0, 0, 0, 0, x, y);
}

HBITMAP
ui_create_bitmap(int width, int height, uint8 *data)
{
        XImage *image;
        Pixmap bitmap;
        uint8 *tdata;

        tdata = (owncolmap ? data : translate_image(width, height, data));
        bitmap = XCreatePixmap(display, wnd, width, height, depth);
        image = XCreateImage(display, visual, depth, ZPixmap,
                             0, tdata, width, height, 8, 0);

        XPutImage(display, bitmap, gc, image, 0, 0, 0, 0, width, height);

        XFree(image);
        if (!owncolmap)
                xfree(tdata);
        return (HBITMAP) bitmap;
}

void
ui_paint_bitmap(int x, int y, int cx, int cy,
                int width, int height, uint8 *data)
{
        XImage *image;
        uint8 *tdata;

        tdata = (owncolmap ? data : translate_image(width, height, data));
        image = XCreateImage(display, visual, depth, ZPixmap,
                             0, tdata, width, height, 8, 0);

        if (ownbackstore)
        {
                XPutImage(display, backstore, gc, image, 0, 0, x, y, cx, cy);
                XCopyArea(display, backstore, wnd, gc, x, y, cx, cy, x, y);
        }
        else
        {
                XPutImage(display, wnd, gc, image, 0, 0, x, y, cx, cy);
        }

        XFree(image);
        if (!owncolmap)
                xfree(tdata);
}

void
ui_destroy_bitmap(HBITMAP bmp)
{
        XFreePixmap(display, (Pixmap)bmp);
}

HGLYPH
ui_create_glyph(int width, int height, uint8 *data)
{
        XImage *image;
        Pixmap bitmap;
        int scanline;
        GC gc;

        scanline = (width + 7) / 8;

        bitmap = XCreatePixmap(display, wnd, width, height, 1);
        gc = XCreateGC(display, bitmap, 0, NULL);

        image = XCreateImage(display, visual, 1, ZPixmap, 0,
                             data, width, height, 8, scanline);
        image->byte_order = MSBFirst;
        image->bitmap_bit_order = MSBFirst;
        XInitImage(image);

        XPutImage(display, bitmap, gc, image, 0, 0, 0, 0, width, height);

        XFree(image);
        XFreeGC(display, gc);
        return (HGLYPH)bitmap;
}

void
ui_destroy_glyph(HGLYPH glyph)
{
        XFreePixmap(display, (Pixmap)glyph);
}

HCURSOR
ui_create_cursor(unsigned int x, unsigned int y, int width,
                 int height, uint8 *andmask, uint8 *xormask)
{
        HGLYPH maskglyph, cursorglyph;
        XColor bg, fg;
        Cursor xcursor;
        uint8 *cursor, *pcursor;
        uint8 *mask, *pmask;
        uint8 nextbit;
        int scanline, offset;
        int i, j;

        scanline = (width + 7) / 8;
        offset = scanline * height;

        cursor = xmalloc(offset);
        memset(cursor, 0, offset);

        mask = xmalloc(offset);
        memset(mask, 0, offset);

        /* approximate AND and XOR masks with a monochrome X pointer */
        for (i = 0; i < height; i++)
        {
                offset -= scanline;
                pcursor = &cursor[offset];
                pmask = &mask[offset];

                for (j = 0; j < scanline; j++)
                {
                        for (nextbit = 0x80; nextbit != 0; nextbit >>= 1)
                        {
                                if (xormask[0] || xormask[1] || xormask[2])
                                {
                                        *pcursor |= (~(*andmask) & nextbit);
                                        *pmask |= nextbit;
                                }
                                else
                                {
                                        *pcursor |= ((*andmask) & nextbit);
                                        *pmask |= (~(*andmask) & nextbit);
                                }

                                xormask += 3;
                        }

                        andmask++;
                        pcursor++;
                        pmask++;
                }
        }

        fg.red = fg.blue = fg.green = 0xffff;
        bg.red = bg.blue = bg.green = 0x0000;
        fg.flags = bg.flags = DoRed | DoBlue | DoGreen;

        cursorglyph = ui_create_glyph(width, height, cursor);
        maskglyph = ui_create_glyph(width, height, mask);
        
        xcursor = XCreatePixmapCursor(display, (Pixmap)cursorglyph,
                                (Pixmap)maskglyph, &fg, &bg, x, y);

        ui_destroy_glyph(maskglyph);
        ui_destroy_glyph(cursorglyph);
        xfree(mask);
        xfree(cursor);
        return (HCURSOR)xcursor;
}

void
ui_set_cursor(HCURSOR cursor)
{
        XDefineCursor(display, wnd, (Cursor)cursor);
}

void
ui_destroy_cursor(HCURSOR cursor)
{
        XFreeCursor(display, (Cursor)cursor);
}

#define MAKE_XCOLOR(xc,c) \
                (xc)->red   = ((c)->red   << 8) | (c)->red; \
                (xc)->green = ((c)->green << 8) | (c)->green; \
                (xc)->blue  = ((c)->blue  << 8) | (c)->blue; \
                (xc)->flags = DoRed | DoGreen | DoBlue;

HCOLOURMAP
ui_create_colourmap(COLOURMAP *colours)
{
        COLOURENTRY *entry;
        int i, ncolours = colours->ncolours;

        if (owncolmap)
        {
                XColor *xcolours, *xentry;
                Colormap map;

                xcolours = xmalloc(sizeof(XColor) * ncolours);
                for (i = 0; i < ncolours; i++)
                {
                        entry = &colours->colours[i];
                        xentry = &xcolours[i];
                        xentry->pixel = i;
                        MAKE_XCOLOR(xentry, entry);
                }

                map = XCreateColormap(display, wnd, visual, AllocAll);
                XStoreColors(display, map, xcolours, ncolours);

                xfree(xcolours);
                return (HCOLOURMAP)map;
        }
        else
        {
                uint32 *map = xmalloc(sizeof(*colmap) * ncolours);
                XColor xentry;
                uint32 colour;

                for (i = 0; i < ncolours; i++)
                {
                        entry = &colours->colours[i];
                        MAKE_XCOLOR(&xentry, entry);

                        if (XAllocColor(display, xcolmap, &xentry) != 0)
                                colour = xentry.pixel;
                        else
                                colour = white;

                        /* byte swap here to make translate_image faster */
                        map[i] = translate_colour(colour);
                }

                return map;
        }
}

void
ui_destroy_colourmap(HCOLOURMAP map)
{
        if (owncolmap)
                XFreeColormap(display, (Colormap)map);
        else
                xfree(map);
}

void
ui_set_colourmap(HCOLOURMAP map)
{
        if (owncolmap)
                XSetWindowColormap(display, wnd, (Colormap)map);
        else
                colmap = map;
}

void
ui_set_clip(int x, int y, int cx, int cy)
{
        XRectangle rect;

        rect.x = x;
        rect.y = y;
        rect.width = cx;
        rect.height = cy;
        XSetClipRectangles(display, gc, 0, 0, &rect, 1, YXBanded);
}

void
ui_reset_clip()
{
        XRectangle rect;

        rect.x = 0;
        rect.y = 0;
        rect.width = width;
        rect.height = height;
        XSetClipRectangles(display, gc, 0, 0, &rect, 1, YXBanded);
}

void
ui_bell()
{
        XBell(display, 0);
}

void
ui_destblt(uint8 opcode,
           /* dest */ int x, int y, int cx, int cy)
{
        SET_FUNCTION(opcode);
        FILL_RECTANGLE(x, y, cx, cy);
        RESET_FUNCTION(opcode);
}

void
ui_patblt(uint8 opcode,
          /* dest */ int x, int y, int cx, int cy,
          /* brush */ BRUSH *brush, int bgcolour, int fgcolour)
{
        Pixmap fill;

        SET_FUNCTION(opcode);

        switch (brush->style)
        {
                case 0: /* Solid */
                        SET_FOREGROUND(fgcolour);
                        FILL_RECTANGLE(x, y, cx, cy);
                        break;

                case 3: /* Pattern */
                        fill = (Pixmap)ui_create_glyph(8, 8, brush->pattern);

                        SET_FOREGROUND(bgcolour);
                        SET_BACKGROUND(fgcolour);
                        XSetFillStyle(display, gc, FillOpaqueStippled);
                        XSetStipple(display, gc, fill);
                        XSetTSOrigin(display, gc, brush->xorigin, brush->yorigin);

                        FILL_RECTANGLE(x, y, cx, cy);

                        XSetFillStyle(display, gc, FillSolid);
                        ui_destroy_glyph((HGLYPH)fill);
                        break;

                default:
                        unimpl("brush %d\n", brush->style);
        }

        RESET_FUNCTION(opcode);
}

void
ui_screenblt(uint8 opcode,
             /* dest */ int x, int y, int cx, int cy,
             /* src */ int srcx, int srcy)
{
        SET_FUNCTION(opcode);
        XCopyArea(display, wnd, wnd, gc, srcx, srcy, cx, cy, x, y);
        if (ownbackstore)
                XCopyArea(display, backstore, backstore, gc, srcx, srcy,
                          cx, cy, x, y);
        RESET_FUNCTION(opcode);
}

void
ui_memblt(uint8 opcode,
          /* dest */ int x, int y, int cx, int cy,
          /* src */ HBITMAP src, int srcx, int srcy)
{
        SET_FUNCTION(opcode);
        XCopyArea(display, (Pixmap)src, wnd, gc, srcx, srcy, cx, cy, x, y);
        if (ownbackstore)
                XCopyArea(display, (Pixmap)src, backstore, gc, srcx, srcy,
                          cx, cy, x, y);
        RESET_FUNCTION(opcode);
}

void
ui_triblt(uint8 opcode,
          /* dest */ int x, int y, int cx, int cy,
          /* src */ HBITMAP src, int srcx, int srcy,
          /* brush */ BRUSH *brush, int bgcolour, int fgcolour)
{
        /* This is potentially difficult to do in general. Until someone
           comes up with a more efficient way of doing it I am using cases. */

        switch (opcode)
        {
                case 0x69:      /* PDSxxn */
                        ui_memblt(ROP2_XOR, x, y, cx, cy, src, srcx, srcy);
                        ui_patblt(ROP2_NXOR, x, y, cx, cy,
                                  brush, bgcolour, fgcolour);
                        break;

                case 0xb8:      /* PSDPxax */
                        ui_patblt(ROP2_XOR, x, y, cx, cy,
                                  brush, bgcolour, fgcolour);
                        ui_memblt(ROP2_AND, x, y, cx, cy, src, srcx, srcy);
                        ui_patblt(ROP2_XOR, x, y, cx, cy,
                                  brush, bgcolour, fgcolour);
                        break;

                case 0xc0:      /* PSa */
                        ui_memblt(ROP2_COPY, x, y, cx, cy, src, srcx, srcy);
                        ui_patblt(ROP2_AND, x, y, cx, cy, brush, bgcolour,
                                  fgcolour);
                        break;

                default:
                        unimpl("triblt 0x%x\n", opcode);
                        ui_memblt(ROP2_COPY, x, y, cx, cy, src, srcx, srcy);
        }
}

void
ui_line(uint8 opcode,
        /* dest */ int startx, int starty, int endx, int endy,
        /* pen */ PEN *pen)
{
        SET_FUNCTION(opcode);
        SET_FOREGROUND(pen->colour);
        XDrawLine(display, wnd, gc, startx, starty, endx, endy);
        if (ownbackstore)
                XDrawLine(display, backstore, gc, startx, starty, endx, endy);
        RESET_FUNCTION(opcode);
}

void
ui_rect(
               /* dest */ int x, int y, int cx, int cy,
               /* brush */ int colour)
{
        SET_FOREGROUND(colour);
        FILL_RECTANGLE(x, y, cx, cy);
}

void
ui_draw_glyph(int mixmode,
              /* dest */ int x, int y, int cx, int cy,
              /* src */ HGLYPH glyph, int srcx, int srcy, int bgcolour,
              int fgcolour)
{
        SET_FOREGROUND(fgcolour);
        SET_BACKGROUND(bgcolour);

        XSetFillStyle(display, gc, (mixmode == MIX_TRANSPARENT)
                      ? FillStippled : FillOpaqueStippled);
        XSetStipple(display, gc, (Pixmap)glyph);
        XSetTSOrigin(display, gc, x, y);

        FILL_RECTANGLE(x, y, cx, cy);

        XSetFillStyle(display, gc, FillSolid);
}

#define DO_GLYPH(ttext,idx) \
{\
  glyph = cache_get_font (font, ttext[idx]);\
  if (!(flags & TEXT2_IMPLICIT_X))\
    {\
      xyoffset = ttext[++idx];\
      if ((xyoffset & 0x80))\
        {\
          if (flags & TEXT2_VERTICAL) \
            y += ttext[++idx] | (ttext[++idx] << 8);\
          else\
            x += ttext[++idx] | (ttext[++idx] << 8);\
        }\
      else\
        {\
          if (flags & TEXT2_VERTICAL) \
            y += xyoffset;\
          else\
            x += xyoffset;\
        }\
    }\
  if (glyph != NULL)\
    {\
      ui_draw_glyph (mixmode, x + (short) glyph->offset,\
                     y + (short) glyph->baseline,\
                     glyph->width, glyph->height,\
                     glyph->pixmap, 0, 0, bgcolour, fgcolour);\
      if (flags & TEXT2_IMPLICIT_X)\
        x += glyph->width;\
    }\
}

void
ui_draw_text(uint8 font, uint8 flags, int mixmode, int x, int y,
             int clipx, int clipy, int clipcx, int clipcy, int boxx,
             int boxy, int boxcx, int boxcy, int bgcolour,
             int fgcolour, uint8 * text, uint8 length)
{
        FONTGLYPH *glyph;
        int i, j, xyoffset;
        DATABLOB *entry;

        SET_FOREGROUND(bgcolour);

        if (boxcx > 1)
        {
                FILL_RECTANGLE(boxx, boxy, boxcx, boxcy);
        }
        else if (mixmode == MIX_OPAQUE)
        {
                FILL_RECTANGLE(clipx, clipy, clipcx, clipcy);
        }

        /* Paint text, character by character */
        for (i = 0; i < length;) {
                switch (text[i]) {
                case 0xff:
                        if (i + 2 < length)
                                cache_put_text(text[i + 1], text, text[i + 2]);
                        else {
                                error("this shouldn't be happening\n");
                                break;
                        }
                        /* this will move pointer from start to first character after FF command */
                        length -= i + 3;
                        text = &(text[i + 3]);
                        i = 0;
                        break;

                case 0xfe:
                        entry = cache_get_text(text[i + 1]);
                        if (entry != NULL) {
                                if ((((uint8 *) (entry->data))[1] == 0)
                                    && (!(flags & TEXT2_IMPLICIT_X))) {
                                        if (flags & TEXT2_VERTICAL)
                                                y += text[i + 2];
                                        else
                                                x += text[i + 2];
                                }
                                if (i + 2 < length)
                                        i += 3;
                                else
                                        i += 2;
                                length -= i;
                                /* this will move pointer from start to first character after FE command */
                                text = &(text[i]);
                                i = 0;
                                for (j = 0; j < entry->size; j++)
                                        DO_GLYPH(((uint8 *) (entry->data)), j);
                        }
                        break;

                default:
                        DO_GLYPH(text, i);
                        i++;
                        break;
                }
        }


}

void
ui_desktop_save(uint32 offset, int x, int y, int cx, int cy)
{
        Pixmap pix;
        XImage *image;

        if (ownbackstore)
        {
                image = XGetImage(display, backstore, x, y, cx, cy, AllPlanes,
                                  ZPixmap);
        }
        else
        {
                pix = XCreatePixmap(display, wnd, cx, cy, depth);
                XCopyArea(display, wnd, pix, gc, x, y, cx, cy, 0, 0);
                image = XGetImage(display, pix, 0, 0, cx, cy, AllPlanes,
                                  ZPixmap);
                XFreePixmap(display, pix);
        }

        offset *= bpp/8;
        cache_put_desktop(offset, cx, cy, image->bytes_per_line,
                          bpp/8, (uint8 *)image->data);

        XDestroyImage(image);
}

void
ui_desktop_restore(uint32 offset, int x, int y, int cx, int cy)
{
        XImage *image;
        uint8 *data;

        offset *= bpp/8;
        data = cache_get_desktop(offset, cx, cy, bpp/8);
        if (data == NULL)
                return;

        image = XCreateImage(display, visual, depth, ZPixmap,
                             0, data, cx, cy, BitmapPad(display),
                             cx * bpp/8);

        if (ownbackstore)
        {
                XPutImage(display, backstore, gc, image, 0, 0, x, y, cx, cy);
                XCopyArea(display, backstore, wnd, gc, x, y, cx, cy, x, y);
        }
        else
        {
                XPutImage(display, wnd, gc, image, 0, 0, x, y, cx, cy);
        }

        XFree(image);
}