/*

  Simple DirectMedia Layer

  Copyright (C) 1997-2012 Sam Lantinga <slouken@libsdl.org>

  This software is provided 'as-is', without any express or implied

  warranty.  In no event will the authors be held liable for any damages

  arising from the use of this software.

  Permission is granted to anyone to use this software for any purpose,

  including commercial applications, and to alter it and redistribute it

  freely, subject to the following restrictions:

  1. The origin of this software must not be misrepresented; you must not

     claim that you wrote the original software. If you use this software

     in a product, an acknowledgment in the product documentation would be

     appreciated but is not required.

  2. Altered source versions must be plainly marked as such, and must not be

     misrepresented as being the original software.

  3. This notice may not be removed or altered from any source distribution.

*/

#include "SDL_config.h"

#if SDL_VIDEO_DRIVER_X11

#include "SDL_hints.h"

#include "SDL_x11video.h"

/*#define X11MODES_DEBUG*/

static int

get_visualinfo(Display * display, int screen, XVisualInfo * vinfo)

{

    const char *visual_id = SDL_getenv("SDL_VIDEO_X11_VISUALID");

    int depth;

    /* Look for an exact visual, if requested */

    if (visual_id) {

        XVisualInfo *vi, template;

        int nvis;

        SDL_zero(template);

        template.visualid = SDL_strtol(visual_id, NULL, 0);

        vi = XGetVisualInfo(display, VisualIDMask, &template, &nvis);

        if (vi) {

            *vinfo = *vi;

            XFree(vi);

            return 0;

        }

    }

    depth = DefaultDepth(display, screen);

    if ((X11_UseDirectColorVisuals() &&

         XMatchVisualInfo(display, screen, depth, DirectColor, vinfo)) ||

        XMatchVisualInfo(display, screen, depth, TrueColor, vinfo) ||

        XMatchVisualInfo(display, screen, depth, PseudoColor, vinfo) ||

        XMatchVisualInfo(display, screen, depth, StaticColor, vinfo)) {

        return 0;

    }

    return -1;

}

int

X11_GetVisualInfoFromVisual(Display * display, Visual * visual, XVisualInfo * vinfo)

{

    XVisualInfo *vi;

    int nvis;

    vinfo->visualid = XVisualIDFromVisual(visual);

    vi = XGetVisualInfo(display, VisualIDMask, vinfo, &nvis);

    if (vi) {

        *vinfo = *vi;

        XFree(vi);

        return 0;

    }

    return -1;

}

Uint32

X11_GetPixelFormatFromVisualInfo(Display * display, XVisualInfo * vinfo)

{

    if (vinfo->class == DirectColor || vinfo->class == TrueColor) {

        int bpp;

        Uint32 Rmask, Gmask, Bmask, Amask;

        Rmask = vinfo->visual->red_mask;

        Gmask = vinfo->visual->green_mask;

        Bmask = vinfo->visual->blue_mask;

        if (vinfo->depth == 32) {

            Amask = (0xFFFFFFFF & ~(Rmask | Gmask | Bmask));

        } else {

            Amask = 0;

        }

        bpp = vinfo->depth;

        if (bpp == 24) {

            int i, n;

            XPixmapFormatValues *p = XListPixmapFormats(display, &n);

            if (p) {

                for (i = 0; i < n; ++i) {

                    if (p[i].depth == 24) {

                        bpp = p[i].bits_per_pixel;

                        break;

                    }

                }

                XFree(p);

            }

        }

        return SDL_MasksToPixelFormatEnum(bpp, Rmask, Gmask, Bmask, Amask);

    }

    if (vinfo->class == PseudoColor || vinfo->class == StaticColor) {

        switch (vinfo->depth) {

        case 8:

            return SDL_PIXELTYPE_INDEX8;

        case 4:

            if (BitmapBitOrder(display) == LSBFirst) {

                return SDL_PIXELFORMAT_INDEX4LSB;

            } else {

                return SDL_PIXELFORMAT_INDEX4MSB;

            }

            break;

        case 1:

            if (BitmapBitOrder(display) == LSBFirst) {

                return SDL_PIXELFORMAT_INDEX1LSB;

            } else {

                return SDL_PIXELFORMAT_INDEX1MSB;

            }

            break;

        }

    }

    return SDL_PIXELFORMAT_UNKNOWN;

}

/* Global for the error handler */

int vm_event, vm_error = -1;

#if SDL_VIDEO_DRIVER_X11_XINERAMA

static SDL_bool

CheckXinerama(Display * display, int *major, int *minor)

{

    int event_base = 0;

    int error_base = 0;

    const char *env;

    /* Default the extension not available */

    *major = *minor = 0;

    /* Allow environment override */

    env = SDL_GetHint(SDL_HINT_VIDEO_X11_XINERAMA);

    if (env && !SDL_atoi(env)) {

#ifdef X11MODES_DEBUG

        printf("Xinerama disabled due to hint\n");

#endif

        return SDL_FALSE;

    }

    if (!SDL_X11_HAVE_XINERAMA) {

#ifdef X11MODES_DEBUG

        printf("Xinerama support not available\n");

#endif

        return SDL_FALSE;

    }

    /* Query the extension version */

    if (!XineramaQueryExtension(display, &event_base, &error_base) ||

        !XineramaQueryVersion(display, major, minor) ||

        !XineramaIsActive(display)) {

#ifdef X11MODES_DEBUG

        printf("Xinerama not active on the display\n");

#endif

        return SDL_FALSE;

    }

#ifdef X11MODES_DEBUG

    printf("Xinerama available at version %d.%d!\n", *major, *minor);

#endif

    return SDL_TRUE;

}

#endif /* SDL_VIDEO_DRIVER_X11_XINERAMA */

#if SDL_VIDEO_DRIVER_X11_XRANDR

static SDL_bool

CheckXRandR(Display * display, int *major, int *minor)

{

    const char *env;

    /* Default the extension not available */

    *major = *minor = 0;

    /* Allow environment override */

    env = SDL_GetHint(SDL_HINT_VIDEO_X11_XRANDR);

    if (env && !SDL_atoi(env)) {

#ifdef X11MODES_DEBUG

        printf("XRandR disabled due to hint\n");

#endif

        return SDL_FALSE;

    }

    if (!SDL_X11_HAVE_XRANDR) {

#ifdef X11MODES_DEBUG

        printf("XRandR support not available\n");

#endif

        return SDL_FALSE;

    }

    /* Query the extension version */

    if (!XRRQueryVersion(display, major, minor)) {

#ifdef X11MODES_DEBUG

        printf("XRandR not active on the display\n");

#endif

        return SDL_FALSE;

    }

#ifdef X11MODES_DEBUG

    printf("XRandR available at version %d.%d!\n", *major, *minor);

#endif

    return SDL_TRUE;

}

#define XRANDR_ROTATION_LEFT    (1 << 1)

#define XRANDR_ROTATION_RIGHT   (1 << 3)

static int

CalculateXRandRRefreshRate(const XRRModeInfo *info)

{

    return (info->hTotal

            && info->vTotal) ? (1000 * info->dotClock / (info->hTotal *

                                                         info->vTotal)) : 0;

}

static SDL_bool

SetXRandRModeInfo(Display *display, XRRScreenResources *res, XRROutputInfo *output_info,

                  RRMode modeID, SDL_DisplayMode *mode)

{

    int i;

    for (i = 0; i < res->nmode; ++i) {

        if (res->modes[i].id == modeID) {

            XRRCrtcInfo *crtc;

            Rotation rotation = 0;

            const XRRModeInfo *info = &res->modes[i];

            crtc = XRRGetCrtcInfo(display, res, output_info->crtc);

            if (crtc) {

                rotation = crtc->rotation;

                XRRFreeCrtcInfo(crtc);

            }

            if (rotation & (XRANDR_ROTATION_LEFT|XRANDR_ROTATION_RIGHT)) {

                mode->w = info->height;

                mode->h = info->width;

            } else {

                mode->w = info->width;

                mode->h = info->height;

            }

            mode->refresh_rate = CalculateXRandRRefreshRate(info);

            ((SDL_DisplayModeData*)mode->driverdata)->xrandr_mode = modeID;

#ifdef X11MODES_DEBUG

            printf("XRandR mode %d: %dx%d@%dHz\n", modeID, mode->w, mode->h, mode->refresh_rate);

#endif

            return SDL_TRUE;

        }

    }

    return SDL_FALSE;

}

#endif /* SDL_VIDEO_DRIVER_X11_XRANDR */

#if SDL_VIDEO_DRIVER_X11_XVIDMODE

static SDL_bool

CheckVidMode(Display * display, int *major, int *minor)

{

    const char *env;

    /* Default the extension not available */

    *major = *minor = 0;

    /* Allow environment override */

    env = SDL_GetHint(SDL_HINT_VIDEO_X11_XVIDMODE);

    if (env && !SDL_atoi(env)) {

#ifdef X11MODES_DEBUG

        printf("XVidMode disabled due to hint\n");

#endif

        return SDL_FALSE;

    }

    if (!SDL_X11_HAVE_XVIDMODE) {

#ifdef X11MODES_DEBUG

        printf("XVidMode support not available\n");

#endif

        return SDL_FALSE;

    }

    /* Query the extension version */

    vm_error = -1;

    if (!XF86VidModeQueryExtension(display, &vm_event, &vm_error)

        || !XF86VidModeQueryVersion(display, major, minor)) {

#ifdef X11MODES_DEBUG

        printf("XVidMode not active on the display\n");

#endif

        return SDL_FALSE;

    }

#ifdef X11MODES_DEBUG

    printf("XVidMode available at version %d.%d!\n", *major, *minor);

#endif

    return SDL_TRUE;

}

static

Bool XF86VidModeGetModeInfo(Display * dpy, int scr,

                                       XF86VidModeModeInfo* info)

{

    Bool retval;

    int dotclock;

    XF86VidModeModeLine l;

    SDL_zerop(info);

    SDL_zero(l);

    retval = XF86VidModeGetModeLine(dpy, scr, &dotclock, &l);

    info->dotclock = dotclock;

    info->hdisplay = l.hdisplay;

    info->hsyncstart = l.hsyncstart;

    info->hsyncend = l.hsyncend;

    info->htotal = l.htotal;

    info->hskew = l.hskew;

    info->vdisplay = l.vdisplay;

    info->vsyncstart = l.vsyncstart;

    info->vsyncend = l.vsyncend;

    info->vtotal = l.vtotal;

    info->flags = l.flags;

    info->privsize = l.privsize;

    info->private = l.private;

    return retval;

}

static int

CalculateXVidModeRefreshRate(const XF86VidModeModeInfo * info)

{

    return (info->htotal

            && info->vtotal) ? (1000 * info->dotclock / (info->htotal *

                                                         info->vtotal)) : 0;

}

SDL_bool

SetXVidModeModeInfo(const XF86VidModeModeInfo *info, SDL_DisplayMode *mode)

{

    mode->w = info->hdisplay;

    mode->h = info->vdisplay;

    mode->refresh_rate = CalculateXVidModeRefreshRate(info);

    ((SDL_DisplayModeData*)mode->driverdata)->vm_mode = *info;

    return SDL_TRUE;

}

#endif /* SDL_VIDEO_DRIVER_X11_XVIDMODE */

int

X11_InitModes(_THIS)

{

    SDL_VideoData *data = (SDL_VideoData *) _this->driverdata;

    int screen, screencount;

#if SDL_VIDEO_DRIVER_X11_XINERAMA

    int xinerama_major, xinerama_minor;

    int use_xinerama = 0;

    XineramaScreenInfo *xinerama = NULL;

#endif

#if SDL_VIDEO_DRIVER_X11_XRANDR

    int xrandr_major, xrandr_minor;

    int use_xrandr = 0;

    XRRScreenResources *res = NULL;

#endif

#if SDL_VIDEO_DRIVER_X11_XVIDMODE

    int vm_major, vm_minor;

    int use_vidmode = 0;

#endif

#if SDL_VIDEO_DRIVER_X11_XINERAMA

    /* Query Xinerama extention

     * NOTE: This works with Nvidia Twinview correctly, but you need version 302.17 (released on June 2012)

     *       or newer of the Nvidia binary drivers

     */

    if (CheckXinerama(data->display, &xinerama_major, &xinerama_minor)) {

        xinerama = XineramaQueryScreens(data->display, &screencount);

        if (xinerama) {

            use_xinerama = xinerama_major * 100 + xinerama_minor;

        }

    }

    if (!xinerama) {

        screencount = ScreenCount(data->display);

    }

#else

    screencount = ScreenCount(data->display);

#endif /* SDL_VIDEO_DRIVER_X11_XINERAMA */

#if SDL_VIDEO_DRIVER_X11_XRANDR

    /* require at least XRandR v1.2 */

    if (CheckXRandR(data->display, &xrandr_major, &xrandr_minor) &&

        (xrandr_major >= 2 || (xrandr_major == 1 && xrandr_minor >= 2))) {

        use_xrandr = xrandr_major * 100 + xrandr_minor;

    }

#endif /* SDL_VIDEO_DRIVER_X11_XRANDR */

#if SDL_VIDEO_DRIVER_X11_XVIDMODE

    if (CheckVidMode(data->display, &vm_major, &vm_minor)) {

        use_vidmode = vm_major * 100 + vm_minor;

    }

#endif /* SDL_VIDEO_DRIVER_X11_XVIDMODE */

    for (screen = 0; screen < screencount; ++screen) {

        XVisualInfo vinfo;

        SDL_VideoDisplay display;

        SDL_DisplayData *displaydata;

        SDL_DisplayMode mode;

        SDL_DisplayModeData *modedata;

        XPixmapFormatValues *pixmapFormats;

        int i, n;

#if SDL_VIDEO_DRIVER_X11_XINERAMA

        if (xinerama) {

            if (get_visualinfo(data->display, 0, &vinfo) < 0) {

                continue;

            }

        } else {

            if (get_visualinfo(data->display, screen, &vinfo) < 0) {

                continue;

            }

        }

#else

        if (get_visualinfo(data->display, screen, &vinfo) < 0) {

            continue;

        }

#endif

        displaydata = (SDL_DisplayData *) SDL_calloc(1, sizeof(*displaydata));

        if (!displaydata) {

            continue;

        }

        mode.format = X11_GetPixelFormatFromVisualInfo(data->display, &vinfo);

        if (SDL_ISPIXELFORMAT_INDEXED(mode.format)) {

            /* We don't support palettized modes now */

            SDL_free(displaydata);

            continue;

        }

#if SDL_VIDEO_DRIVER_X11_XINERAMA

        if (xinerama) {

            mode.w = xinerama[screen].width;

            mode.h = xinerama[screen].height;

        } else {

            mode.w = DisplayWidth(data->display, screen);

            mode.h = DisplayHeight(data->display, screen);

        }

#else

        mode.w = DisplayWidth(data->display, screen);

        mode.h = DisplayHeight(data->display, screen);

#endif

        mode.refresh_rate = 0;

        modedata = (SDL_DisplayModeData *) SDL_calloc(1, sizeof(SDL_DisplayModeData));

        if (!modedata) {

            SDL_free(displaydata);

            continue;

        }

        mode.driverdata = modedata;

#if SDL_VIDEO_DRIVER_X11_XINERAMA

        /* Most of SDL's calls to X11 are unwaware of Xinerama, and to X11 standard calls, when Xinerama is active,

         * there's only one screen available. So we force the screen number to zero and

         * let Xinerama specific code handle specific functionality using displaydata->xinerama_info

         */

        if (use_xinerama) {

            displaydata->screen = 0;

            displaydata->use_xinerama = use_xinerama;

            displaydata->xinerama_info = xinerama[screen];

            displaydata->xinerama_screen = screen;

        }

        else displaydata->screen = screen;

#else

        displaydata->screen = screen;

#endif

        displaydata->visual = vinfo.visual;

        displaydata->depth = vinfo.depth;

        displaydata->scanline_pad = SDL_BYTESPERPIXEL(mode.format) * 8;

        pixmapFormats = XListPixmapFormats(data->display, &n);

        if (pixmapFormats) {

            for (i = 0; i < n; ++i) {

                if (pixmapFormats[i].depth == displaydata->depth) {

                    displaydata->scanline_pad = pixmapFormats[i].scanline_pad;

                    break;

                }

            }

            XFree(pixmapFormats);

        }

#if SDL_VIDEO_DRIVER_X11_XINERAMA