/*
*      screen.c
*      Copyright © 2008 Martin Duquesnoy <xorg62@gmail.com>
*      All rights reserved.
*
*      Redistribution and use in source and binary forms, with or without
*      modification, are permitted provided that the following conditions are
*      met:
*
*      * Redistributions of source code must retain the above copyright
*        notice, this list of conditions and the following disclaimer.
*      * Redistributions in binary form must reproduce the above
*        copyright notice, this list of conditions and the following disclaimer
*        in the documentation and/or other materials provided with the
*        distribution.
*      * Neither the name of the  nor the names of its
*        contributors may be used to endorse or promote products derived from
*        this software without specific prior written permission.
*
*      THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
*      "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
*      LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
*      A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
*      OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
*      SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
*      LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
*      DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
*      THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
*      (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
*      OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/

#include "wmfs.h"

/** Count the screens
 *\return the number of screen
*/
int
screen_count(void)
{
     int n = 0;

     if(XineramaIsActive(dpy))
          XineramaQueryScreens(dpy, &n);
     else
          n = ScreenCount(dpy);

     /* Set _WMFS_SCREEN_COUNT */
     XChangeProperty(dpy, ROOT, net_atom[wmfs_screen_count], XA_CARDINAL, 32,
                     PropModeReplace, (uchar*)&n, 1);

     return n;
}

/** Get screen geometry by number
 *\param s Screen number
 *\return XRectangle struct
 *\todo Set a pure screen geo (without change (- INFOBARH - TBAR etc..))
*/
XRectangle
screen_get_geo(int s)
{
     int n = 0;
     int barpos = tags[selscreen][seltag[selscreen]].barpos;
     XRectangle geo;


     if(XineramaIsActive(dpy))
     {
          XineramaScreenInfo *xsi;

          xsi = XineramaQueryScreens(dpy, &n);
          geo.x = xsi[s].x_org + BORDH;
          if(barpos == IB_Hide || barpos == IB_Bottom)
               geo.y = TBARH;
          else
               geo.y = xsi[s].y_org + INFOBARH + TBARH;
          geo.height = xsi[s].height - INFOBARH - TBARH;
          geo.width = xsi[s].width;

          XFree(xsi);
     }
     else
     {
          geo.x = BORDH;
          if(barpos == IB_Hide || barpos == IB_Bottom)
               geo.y = TBARH;
          else
               geo.y = INFOBARH + TBARH;
          geo.height = MAXH - INFOBARH - TBARH;
          geo.width = MAXW;
     }

     return geo;
}

/** Get the current screen number with
 * coordinated
 *\param geo Geometry for get the screen number
 *\return The screen number
*/
int
screen_get_with_geo(int x, int y)
{
     int i, r = 0;
     int barpos = tags[selscreen][seltag[selscreen]].barpos;
     int yh;

     for(i = 0; i < screen_count(); ++i)
     {
          if(barpos == IB_Hide || barpos == IB_Bottom)
               yh = (sgeo[i].y - TBARH);
          else
               yh = (sgeo[i].y - INFOBARH - TBARH);

          if((x >= sgeo[i].x && x < sgeo[i].x + sgeo[i].width)
             && y >= yh && y < yh + (sgeo[i].height + INFOBARH + TBARH))
               r = i;
     }

     return r;
}

/** Set the selected screen
 *\param screen Number of the wanted selected screen
*/
void
screen_set_sel(int screen)
{
     if(screen < 0 || screen > screen_count() - 1)
          screen = 0;

     client_focus(NULL);
     XWarpPointer(dpy, None, ROOT, 0, 0, 0, 0,
                  sgeo[screen].x + sgeo[screen].width / 2,
                  sgeo[screen].y + sgeo[screen].height / 2);
     selscreen = screen;

     return;
}

/** Get and set the selected screen
 *\return The number of the selected screen
*/
int
screen_get_sel(void)
{
     if(XineramaIsActive(dpy))
     {
          /* Unused variables */
          Window w;
          int d, u, x, y;

          XQueryPointer(dpy, ROOT, &w, &w, &x, &y, &d, &d, (uint *)&u);

          selscreen = screen_get_with_geo(x, y);

          /* Set _WMFS_CURRENT_SCREEN */
          XChangeProperty(dpy, ROOT, net_atom[wmfs_current_screen], XA_CARDINAL, 32,
                          PropModeReplace, (uchar*)&selscreen, 1);
     }

     return selscreen;
}

/** Init screen geo
 */
void
screen_init_geo(void)
{
     int i;

     sgeo = emalloc(screen_count(), sizeof(XRectangle));

     for(i = 0; i < screen_count(); ++i)
          sgeo[i] = screen_get_geo(i);

     ewmh_set_desktop_geometry();
     ewmh_set_workarea();

     return;
}

/** Uicb: screen select
 * \param cmd Screen uicb_t type
*/
void
uicb_screen_select(uicb_t cmd)
{
     screen_set_sel(atoi(cmd));

     return;
}

/** Uicb: screen next
 * \param cmd Screen uicb_t type
*/
void
uicb_screen_next(uicb_t cmd)
{
     selscreen = (selscreen + 1 > screen_count() - 1) ? 0 : selscreen + 1;

     screen_set_sel(selscreen);

     return;
}

/** Uicb: screen prev
 * \param cmd Screen uicb_t type
*/
void
uicb_screen_prev(uicb_t cmd)
{
     selscreen = (selscreen - 1 < 0) ? screen_count() - 1 : selscreen - 1;

     screen_set_sel(selscreen);

     return;
}