// Workspace.cc for Openbox
// Copyright (c) 2002 - 2002 Ben Jansens (ben@orodu.net)
// Copyright (c) 2001 Sean 'Shaleh' Perry <shaleh@debian.org>
// Copyright (c) 1997 - 2000 Brad Hughes (bhughes@tcac.net)
//
// Permission is hereby granted, free of charge, to any person obtaining a
// copy of this software and associated documentation files (the "Software"),
// to deal in the Software without restriction, including without limitation
// the rights to use, copy, modify, merge, publish, distribute, sublicense,
// and/or sell copies of the Software, and to permit persons to whom the
// Software is furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
// THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
// DEALINGS IN THE SOFTWARE.

// stupid macros needed to access some functions in version 2 of the GNU C
// library
#ifndef   _GNU_SOURCE
#define   _GNU_SOURCE
#endif // _GNU_SOURCE

#ifdef    HAVE_CONFIG_H
#  include "../config.h"
#endif // HAVE_CONFIG_H

#include <X11/Xlib.h>
#include <X11/Xatom.h>

#include "i18n.h"
#include "openbox.h"
#include "Clientmenu.h"
#include "Screen.h"
#include "Toolbar.h"
#include "Window.h"
#include "Workspace.h"
#include "Windowmenu.h"
#include "Geometry.h"
#include "Util.h"

#ifdef    HAVE_STDIO_H
#  include <stdio.h>
#endif // HAVE_STDIO_H

#ifdef    HAVE_STDLIB_H
#  include <stdlib.h>
#endif // HAVE_STDLIB_H

#ifdef    STDC_HEADERS
#  include <string.h>
#endif // STDC_HEADERS

#include <vector>
typedef vector<Rect> rectList;

Workspace::Workspace(BScreen &scrn, int i) : screen(scrn) {

  cascade_x = cascade_y = 32;

  id = i;

  stackingList = new LinkedList<OpenboxWindow>;
  windowList = new LinkedList<OpenboxWindow>;
  clientmenu = new Clientmenu(*this);

  lastfocus = (OpenboxWindow *) 0;

  name = (char *) 0;
  char *tmp = screen.getNameOfWorkspace(id);
  setName(tmp);
}


Workspace::~Workspace(void) {
  delete stackingList;
  delete windowList;
  delete clientmenu;

  if (name)
    delete [] name;
}


const int Workspace::addWindow(OpenboxWindow *w, Bool place) {
  if (! w) return -1;

  if (place) placeWindow(w);

  w->setWorkspace(id);
  w->setWindowNumber(windowList->count());

  stackingList->insert(w, 0);
  windowList->insert(w);

  clientmenu->insert((const char **) w->getTitle());
  clientmenu->update();

  screen.updateNetizenWindowAdd(w->getClientWindow(), id);

  raiseWindow(w);

  return w->getWindowNumber();
}


const int Workspace::removeWindow(OpenboxWindow *w) {
  if (! w) return -1;

  stackingList->remove(w);

  if (w->isFocused()) {
    if (w->isTransient() && w->getTransientFor() &&
	w->getTransientFor()->isVisible()) {
      w->getTransientFor()->setInputFocus();
    } else if (screen.isSloppyFocus()) {
      screen.getOpenbox().setFocusedWindow((OpenboxWindow *) 0);
    } else {
      OpenboxWindow *top = stackingList->first();
      if (! top || ! top->setInputFocus()) {
	screen.getOpenbox().setFocusedWindow((OpenboxWindow *) 0);
	XSetInputFocus(screen.getOpenbox().getXDisplay(),
		       screen.getToolbar()->getWindowID(),
		       RevertToParent, CurrentTime);
      }
    }
  }
  
  if (lastfocus == w)
    lastfocus = (OpenboxWindow *) 0;

  windowList->remove(w->getWindowNumber());
  clientmenu->remove(w->getWindowNumber());
  clientmenu->update();

  screen.updateNetizenWindowDel(w->getClientWindow());

  LinkedListIterator<OpenboxWindow> it(windowList);
  OpenboxWindow *bw = it.current();
  for (int i = 0; bw; it++, i++, bw = it.current())
    bw->setWindowNumber(i);

  return windowList->count();
}


void Workspace::showAll(void) {
  LinkedListIterator<OpenboxWindow> it(stackingList);
  for (OpenboxWindow *bw = it.current(); bw; it++, bw = it.current())
    bw->deiconify(False, False);
}


void Workspace::hideAll(void) {
  LinkedList<OpenboxWindow> lst;

  LinkedListIterator<OpenboxWindow> it(stackingList);
  for (OpenboxWindow *bw = it.current(); bw; it++, bw = it.current())
    lst.insert(bw, 0);

  LinkedListIterator<OpenboxWindow> it2(&lst);
  for (OpenboxWindow *bw = it2.current(); bw; it2++, bw = it2.current())
    if (! bw->isStuck())
      bw->withdraw();
}


void Workspace::removeAll(void) {
  LinkedListIterator<OpenboxWindow> it(windowList);
  for (OpenboxWindow *bw = it.current(); bw; it++, bw = it.current())
    bw->iconify();
}


void Workspace::raiseWindow(OpenboxWindow *w) {
  OpenboxWindow *win = (OpenboxWindow *) 0, *bottom = w;

  while (bottom->isTransient() && bottom->getTransientFor())
    bottom = bottom->getTransientFor();

  int i = 1;
  win = bottom;
  while (win->hasTransient() && win->getTransient()) {
    win = win->getTransient();

    i++;
  }

  Window *nstack = new Window[i], *curr = nstack;
  Workspace *wkspc;

  win = bottom;
  while (True) {
    *(curr++) = win->getFrameWindow();
    screen.updateNetizenWindowRaise(win->getClientWindow());

    if (! win->isIconic()) {
      wkspc = screen.getWorkspace(win->getWorkspaceNumber());
      wkspc->stackingList->remove(win);
      wkspc->stackingList->insert(win, 0);
    }

    if (! win->hasTransient() || ! win->getTransient())
      break;

    win = win->getTransient();
  }

  screen.raiseWindows(nstack, i);

  delete [] nstack;
}


void Workspace::lowerWindow(OpenboxWindow *w) {
  OpenboxWindow *win = (OpenboxWindow *) 0, *bottom = w;

  while (bottom->isTransient() && bottom->getTransientFor())
    bottom = bottom->getTransientFor();

  int i = 1;
  win = bottom;
  while (win->hasTransient() && win->getTransient()) {
    win = win->getTransient();

    i++;
  }

  Window *nstack = new Window[i], *curr = nstack;
  Workspace *wkspc;

  while (True) {
    *(curr++) = win->getFrameWindow();
    screen.updateNetizenWindowLower(win->getClientWindow());

    if (! win->isIconic()) {
      wkspc = screen.getWorkspace(win->getWorkspaceNumber());
      wkspc->stackingList->remove(win);
      wkspc->stackingList->insert(win);
    }

    if (! win->getTransientFor())
      break;

    win = win->getTransientFor();
  }

  screen.getOpenbox().grab();

  XLowerWindow(screen.getBaseDisplay().getXDisplay(), *nstack);
  XRestackWindows(screen.getBaseDisplay().getXDisplay(), nstack, i);

  screen.getOpenbox().ungrab();

  delete [] nstack;
}


void Workspace::reconfigure(void) {
  clientmenu->reconfigure();

  LinkedListIterator<OpenboxWindow> it(windowList);
  for (OpenboxWindow *bw = it.current(); bw; it++, bw = it.current()) {
    if (bw->validateClient())
      bw->reconfigure();
  }
}


OpenboxWindow *Workspace::getWindow(int index) {
  if ((index >= 0) && (index < windowList->count()))
    return windowList->find(index);
  else
    return 0;
}


const int Workspace::getCount(void) {
  return windowList->count();
}


void Workspace::update(void) {
  clientmenu->update();
  screen.getToolbar()->redrawWindowLabel(True);
}


Bool Workspace::isCurrent(void) {
  return (id == screen.getCurrentWorkspaceID());
}


Bool Workspace::isLastWindow(OpenboxWindow *w) {
  return (w == windowList->last());
}

void Workspace::setCurrent(void) {
  screen.changeWorkspaceID(id);
}


void Workspace::setName(char *new_name) {
  if (name)
    delete [] name;

  if (new_name) {
    name = bstrdup(new_name);
  } else {
    name = new char[128];
    sprintf(name, i18n->getMessage(WorkspaceSet, WorkspaceDefaultNameFormat,
				   "Workspace %d"), id + 1);
  }
  
  clientmenu->setLabel(name);
  clientmenu->update();
}


void Workspace::shutdown(void) {
  while (windowList->count()) {
    windowList->first()->restore();
    delete windowList->first();
  }
}

static rectList calcSpace(const OpenboxWindow &win, const rectList &spaces) {
  rectList result;
  rectList::const_iterator siter;
  for(siter=spaces.begin(); siter!=spaces.end(); ++siter) {
    if(win.area().Intersect(*siter)) {
      //Check for space to the left of the window
      if(win.origin().x() > siter->x())
        result.push_back(Rect(siter->x(), siter->y(),
                              win.origin().x() - siter->x() - 1,
                              siter->h()));
      //Check for space above the window
      if(win.origin().y() > siter->y())
        result.push_back(Rect(siter->x(), siter->y(),
                              siter->w(),
                              win.origin().y() - siter->y() - 1));
      //Check for space to the right of the window
      if((win.origin().x()+win.size().w()) <
         (siter->x()+siter->w()))
        result.push_back(Rect(win.origin().x() + win.size().w() + 1,
                              siter->y(),
                              siter->x() + siter->w() -
                              win.origin().x() - win.size().w() - 1,
                              siter->h()));
      //Check for space below the window
      if((win.origin().y()+win.size().h()) <
         (siter->y()+siter->h()))
        result.push_back(Rect(siter->x(),
                              win.origin().y() + win.size().h() + 1,
                              siter->w(),
                              siter->y() + siter->h()-
                              win.origin().y() - win.size().h() - 1));

    }
    else
      result.push_back(*siter);
  }
  return result;
}

//BestFitPlacement finds the smallest free space that fits the window
//to be placed. It currentl ignores whether placement is right to left or top
//to bottom.
Point *Workspace::bestFitPlacement(const Size &win_size, const Rect &space)
{
  const Rect *best;
  rectList spaces;
  LinkedListIterator<OpenboxWindow> it(windowList);
  rectList::const_iterator siter;
  spaces.push_back(space); //initially the entire screen is free
  it.reset();
  
  //Find Free Spaces
  for (OpenboxWindow *cur=it.current(); cur!=NULL; it++, cur=it.current())
     spaces = calcSpace(*cur, spaces);
  
  //Find first space that fits the window
  best = 0;
  for (siter=spaces.begin(); siter!=spaces.end(); ++siter) {
    if ((siter->w() >= win_size.w()) &&
        (siter->h() >= win_size.h()))
      best = siter;
  }

  if (best != 0)
    return new Point(best->origin());
  else
    return new Point(200, 0);
}

inline Point *Workspace::rowSmartPlacement(const Size &win_size,
                                           const Rect &space){
  bool placed=false;
  int test_x, test_y, place_x = 0, place_y = 0;
  int start_pos = 0;
  int change_y =
     ((screen.getColPlacementDirection() == BScreen::TopBottom) ? 1 : -1);
  int change_x =
     ((screen.getRowPlacementDirection() == BScreen::LeftRight) ? 1 : -1);
  int delta_x = 8, delta_y = 8;
  LinkedListIterator<OpenboxWindow> it(windowList);

  test_y = (screen.getColPlacementDirection() == BScreen::TopBottom) ?
    start_pos : screen.size().h() - win_size.h() - start_pos;

  while(!placed &&
        ((screen.getColPlacementDirection() == BScreen::BottomTop) ?
         test_y > 0 : test_y + win_size.h() < (signed) space.h())) {
    test_x = (screen.getRowPlacementDirection() == BScreen::LeftRight) ?
      start_pos : space.w() - win_size.w() - start_pos;
    while (!placed &&
           ((screen.getRowPlacementDirection() == BScreen::RightLeft) ?
            test_x > 0 : test_x + win_size.w() < (signed) space.w())) {
      placed = true;

      it.reset();
      for (OpenboxWindow *curr = it.current(); placed && curr;
           it++, curr = it.current()) {
        int curr_w = curr->size().w() + (screen.getBorderWidth() * 4);
        int curr_h = curr->size().h() + (screen.getBorderWidth() * 4);

        if (curr->origin().x() < test_x + win_size.w() &&
            curr->origin().x() + curr_w > test_x &&
            curr->origin().y() < test_y + win_size.h() &&
            curr->origin().y() + curr_h > test_y) {
          placed = false;
        }
      }

      // Removed code for checking toolbar and slit
      // The space passed in should not include either

      if (placed) {
        place_x = test_x;
        place_y = test_y;

        break;
      }   

      test_x += (change_x * delta_x);
    }

    test_y += (change_y * delta_y);
  }
  return new Point(place_x, place_y);
}

void Workspace::placeWindow(OpenboxWindow *win) {
  ASSERT(win != NULL);

  Bool placed = False;

  const int win_w = win->size().w() + (screen.getBorderWidth() * 4),
    win_h = win->size().h() + (screen.getBorderWidth() * 4),
#ifdef    SLIT
    slit_x = screen.getSlit()->getX() - screen.getBorderWidth(),
    slit_y = screen.getSlit()->getY() - screen.getBorderWidth(),
    slit_w = screen.getSlit()->getWidth() +
      (screen.getBorderWidth() * 4),
    slit_h = screen.getSlit()->getHeight() +
      (screen.getBorderWidth() * 4),
#endif // SLIT
    toolbar_x = screen.getToolbar()->getX() - screen.getBorderWidth(),
    toolbar_y = screen.getToolbar()->getY() - screen.getBorderWidth(),
    toolbar_w = screen.getToolbar()->getWidth() +
      (screen.getBorderWidth() * 4),
    toolbar_h = screen.getToolbar()->getHeight() + 
      (screen.getBorderWidth() * 4),
    start_pos = 0,
    change_y =
      ((screen.getColPlacementDirection() == BScreen::TopBottom) ? 1 : -1),
    change_x =
      ((screen.getRowPlacementDirection() == BScreen::LeftRight) ? 1 : -1),
    delta_x = 8, delta_y = 8;

  int test_x, test_y, place_x = 0, place_y = 0;
  LinkedListIterator<OpenboxWindow> it(windowList);

  Rect space(0, 0,
             screen.size().w(),
             screen.size().h()
            );
  Size window_size(win_w, win_h);

  switch (screen.getPlacementPolicy()) {
  case BScreen::BestFitPlacement: {
    Point *spot = bestFitPlacement(window_size, space);
    if (spot != NULL) {
      place_x=spot->x();
      place_y=spot->y();
      delete spot;
      placed=true;
    }
    break;
  }
  case BScreen::RowSmartPlacement: {
    Point *spot=rowSmartPlacement(window_size, space);
    if (spot != NULL) {
      place_x=spot->x();
      place_y=spot->y();
      delete spot;
      placed=true;
    }
    break;
  }

  case BScreen::ColSmartPlacement: {
    test_x = (screen.getRowPlacementDirection() == BScreen::LeftRight) ?
      start_pos : screen.size().w() - win_w - start_pos;

    while (!placed &&
	   ((screen.getRowPlacementDirection() == BScreen::RightLeft) ?
	    test_x > 0 : test_x + win_w < (signed) screen.size().w())) {
      test_y = (screen.getColPlacementDirection() == BScreen::TopBottom) ?
	start_pos : screen.size().h() - win_h - start_pos;
      
      while (!placed &&
	     ((screen.getColPlacementDirection() == BScreen::BottomTop) ?
	      test_y > 0 : test_y + win_h < (signed) screen.size().h())) {
        placed = True;

        it.reset();
        for (OpenboxWindow *curr = it.current(); placed && curr;
	     it++, curr = it.current()) {
          if (curr->isMaximizedFull()) // fully maximized, ignore it
            continue;
          int curr_w = curr->size().w() + (screen.getBorderWidth() * 4);
          int curr_h =
            ((curr->isShaded()) ? curr->getTitleHeight() : curr->size().h()) +
            (screen.getBorderWidth() * 4);

          if (curr->origin().x() < test_x + win_w &&
              curr->origin().x() + curr_w > test_x &&
              curr->origin().y() < test_y + win_h &&
              curr->origin().y() + curr_h > test_y) {
            placed = False;
	  }
        }

        if (placed &&
	    (toolbar_x < test_x + win_w &&
	     toolbar_x + toolbar_w > test_x &&
	     toolbar_y < test_y + win_h &&
	     toolbar_y + toolbar_h > test_y)
#ifdef    SLIT
	    ||
	    (slit_x < test_x + win_w &&
	     slit_x + slit_w > test_x &&
	     slit_y < test_y + win_h &&
	     slit_y + slit_h > test_y)
#endif // SLIT
	    )
	  placed = False;

	if (placed) {
	  place_x = test_x;
	  place_y = test_y;

	  break;
	}

	test_y += (change_y * delta_y);
      }

      test_x += (change_x * delta_x);
    }

    break;
  }
  } // switch

  if (! placed) {
    if (((unsigned) cascade_x > (screen.size().w() / 2)) ||
	((unsigned) cascade_y > (screen.size().h() / 2)))
      cascade_x = cascade_y = 32;

    place_x = cascade_x;
    place_y = cascade_y;

    cascade_x += win->getTitleHeight();
    cascade_y += win->getTitleHeight();
  }
  
  if (place_x + win_w > (signed) screen.size().w())
    place_x = (((signed) screen.size().w()) - win_w) / 2;
  if (place_y + win_h > (signed) screen.size().h())
    place_y = (((signed) screen.size().h()) - win_h) / 2;

  win->configure(place_x, place_y, win->size().w(), win->size().h());
}