/*
 * screen.c - screen management
 *
 * Copyright © 2007 Julien Danjou <julien@danjou.info>
 *
 * 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.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 */

#include "util.h"
#include "screen.h"
#include "tag.h"
#include "layout.h"
#include "statusbar.h"

/** Get screens info
 * \param disp Display ref
 * \param screen Screen number
 * \param statusbar statusbar
 * \return ScreenInfo struct array with all screens info
 */
ScreenInfo *
get_screen_info(Display *disp, int screen, Statusbar *statusbar, Padding *padding)
{
    int i, screen_number = 0;
    ScreenInfo *si;

    if(XineramaIsActive(disp))
        si = XineramaQueryScreens(disp, &screen_number);
    else
    {
        /* emulate Xinerama info but only fill the screen we want */
        si = p_new(ScreenInfo, screen + 1);
        si[screen].width = DisplayWidth(disp, screen);
        si[screen].height = DisplayHeight(disp, screen);
        si[screen].x_org = 0;
        si[screen].y_org = 0;
        screen_number = screen + 1;
    }

     /* make padding corrections */
    if(padding)
    {
        si[screen].x_org+=padding->left;
        si[screen].y_org+=padding->top;
        si[screen].width-=padding->left+padding->right;
        si[screen].height-=padding->top+padding->bottom;
    }

    if(statusbar)
        for(i = 0; i < screen_number; i++)
            switch(statusbar->position)
            {
              case BarTop:
                si[i].y_org += statusbar->height;
              case BarBot:
                si[i].height -= statusbar->height;
                break;
              case BarLeft:
                si[i].x_org += statusbar->height;
              case BarRight:
                si[i].width -=  statusbar->height;
                break;
            }

    return si;
}

/** Get display info
 * \param disp Display ref
 * \param screen Screen number
 * \param statusbar the statusbar
 * \return ScreenInfo struct pointer with all display info
 */
ScreenInfo *
get_display_info(Display *disp, int screen, Statusbar *statusbar, Padding *padding)
{
    ScreenInfo *si;

    si = p_new(ScreenInfo, 1);

    si->x_org = 0;
    si->y_org = statusbar && statusbar->position == BarTop ? statusbar->height : 0;
    si->width = DisplayWidth(disp, screen);
    si->height = DisplayHeight(disp, screen) -
        (statusbar && (statusbar->position == BarTop || statusbar->position == BarBot) ? statusbar->height : 0);

	/* make padding corrections */
	if(padding)
	{
		si[screen].x_org+=padding->left;
		si[screen].y_org+=padding->top;
		si[screen].width-=padding->left+padding->right;
		si[screen].height-=padding->top+padding->bottom;
	}

    return si;
}

/** Return the Xinerama screen number where the coordinates belongs to
 * \param disp Display ref
 * \param x x coordinate of the window
 * \param y y coordinate of the window
 * \return screen number or DefaultScreen of disp on no match
 */
int
get_screen_bycoord(Display *disp, int x, int y)
{
    ScreenInfo *si;
    int i;

    /* don't waste our time */
    if(!XineramaIsActive(disp))
        return DefaultScreen(disp);

    si = get_screen_info(disp, 0, NULL, NULL);

    for(i = 0; i < get_screen_count(disp); i++)
        if((x < 0 || (x >= si[i].x_org && x < si[i].x_org + si[i].width))
           && (y < 0 || (y >= si[i].y_org && y < si[i].y_org + si[i].height)))
        {
            p_delete(&si);
            return i;
        }

    p_delete(&si);
    return DefaultScreen(disp);
}

/** Return the actual screen count
 * \param disp Display ref
 * \return the number of screen available
 */
int
get_screen_count(Display *disp)
{
    int screen_number;

    if(XineramaIsActive(disp))
        XineramaQueryScreens(disp, &screen_number);
    else
        return ScreenCount(disp);

    return screen_number;
}

/** This returns the real X screen number for a logical
 * screen if Xinerama is active.
 * \param disp Display ref
 * \param screen the logical screen
 * \return the X screen
 */
int
get_phys_screen(Display *disp, int screen)
{
    if(XineramaIsActive(disp))
        return DefaultScreen(disp);
    return screen;
}

/** Move a client to a virtual screen
 * \param c the client
 * \param acf_new the awesome_config for the new screen
 * \param doresize set to True if we also move the client to the new x_org and
 *         y_org of the new screen
 */
void
move_client_to_screen(Client *c, awesome_config *awesomeconf, int new_screen, Bool doresize)
{
    int i, old_screen = c->screen;

    /* if the client was focused on an old screen tag, remove it */
    for(i = 0; i < awesomeconf->screens[old_screen].ntags; i++)
        if(awesomeconf->screens[old_screen].tags[i].client_sel == c)
            awesomeconf->screens[old_screen].tags[i].client_sel = NULL;

    /* tag client with new screen tags */
    tag_client_with_current_selected(c, &awesomeconf->screens[new_screen]);

    c->screen = new_screen;

    if(doresize && old_screen != c->screen)
    {
        ScreenInfo *si, *si_old;

        si = get_screen_info(c->display, c->screen, NULL, NULL);
        si_old = get_screen_info(c->display, old_screen, NULL, NULL);

        /* compute new coords in new screen */
        c->rx = (c->rx - si_old[old_screen].x_org) + si[c->screen].x_org;
        c->ry = (c->ry - si_old[old_screen].y_org) + si[c->screen].y_org;
        /* check that new coords are still in the screen */
        if(c->rw > si[c->screen].width)
            c->rw = si[c->screen].width;
        if(c->rh > si[c->screen].height)
            c->rh = si[c->screen].height;
        if(c->rx + c->rw >= si[c->screen].x_org + si[c->screen].width)
            c->rx = si[c->screen].x_org + si[c->screen].width - c->rw - 2 * c->border;
        if(c->ry + c->rh >= si[c->screen].y_org + si[c->screen].height)
            c->ry = si[c->screen].y_org + si[c->screen].height - c->rh - 2 * c->border;

        client_resize(c, c->rx, c->ry, c->rw, c->rh, awesomeconf, True, False);

        p_delete(&si);
        p_delete(&si_old);
    }

    focus(c, True, awesomeconf, c->screen);

    /* redraw statusbar on all screens */
    drawstatusbar(awesomeconf, old_screen);
    drawstatusbar(awesomeconf, new_screen);
}

/** Move mouse pointer to x_org and y_xorg of specified screen
 * \param disp display ref
 * \param screen screen number
 */
static void
move_mouse_pointer_to_screen(Display *disp, int screen)
{
    if(XineramaIsActive(disp))
    {
        ScreenInfo *si = get_screen_info(disp, screen, NULL, NULL);
        XWarpPointer(disp, None, DefaultRootWindow(disp), 0, 0, 0, 0, si[screen].x_org, si[screen].y_org);
        p_delete(&si);
    }
    else
        XWarpPointer(disp, None, RootWindow(disp, screen), 0, 0, 0, 0, 0, 0);
}

void
uicb_screen_focusnext(awesome_config * awesomeconf,
                      int screen,
                      const char *arg __attribute__ ((unused)))
{
    int next_screen = screen + 1 >= get_screen_count(awesomeconf->display) ? 0 : screen + 1;
    Client *sel = get_current_tag(awesomeconf->screens[next_screen].tags,
                                  awesomeconf->screens[next_screen].ntags)->client_sel;

    focus(sel, True, awesomeconf, next_screen - screen);
    move_mouse_pointer_to_screen(awesomeconf->display, next_screen);
}

void
uicb_screen_focusprev(awesome_config * awesomeconf,
                      int screen,
                      const char *arg __attribute__ ((unused)))
{
    int prev_screen = screen - 1 < 0 ? get_screen_count(awesomeconf->display) - 1 : screen - 1;
    Client *sel = get_current_tag(awesomeconf->screens[prev_screen].tags,
                                  awesomeconf->screens[prev_screen].ntags)->client_sel;

    focus(sel, True, awesomeconf, prev_screen - screen);
    move_mouse_pointer_to_screen(awesomeconf->display, prev_screen);
}

/** Move client to a virtual screen (if Xinerama is active)
 * \param awesomeconf awesome config
 * \param arg screen number
 * \ingroup ui_callback
 */
void
uicb_client_movetoscreen(awesome_config * awesomeconf,
                         int screen,
                         const char *arg)
{
    int new_screen, prev_screen;
    Client *sel = get_current_tag(awesomeconf->screens[screen].tags, awesomeconf->screens[screen].ntags)->client_sel;

    if(!sel || !XineramaIsActive(awesomeconf->display))
        return;

    if(arg)
        new_screen = compute_new_value_from_arg(arg, sel->screen);
    else
        new_screen = sel->screen + 1;

    if(new_screen >= get_screen_count(awesomeconf->display))
        new_screen = 0;
    else if(new_screen < 0)
        new_screen = get_screen_count(awesomeconf->display) - 1;

    prev_screen = sel->screen;
    move_client_to_screen(sel, awesomeconf, new_screen, True);
    move_mouse_pointer_to_screen(awesomeconf->display, new_screen);
    arrange(awesomeconf, prev_screen);
    arrange(awesomeconf, new_screen);
}
// vim: filetype=c:expandtab:shiftwidth=4:tabstop=8:softtabstop=4:encoding=utf-8:textwidth=99