// 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    HAVE_STRING_H
#  include <string.h>
#endif // HAVE_STRING_H

#include <vector>
#include <algorithm>
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.sloppyFocus()) {
      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();
  screen.saveWorkspaceNames();
}


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

static rectList calcSpace(const Rect &win, const rectList &spaces) {
  rectList result;
  rectList::const_iterator siter;
  for(siter=spaces.begin(); siter!=spaces.end(); ++siter) {
    if(win.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;
}


bool incWidth(const Rect &first, const Rect &second) {
  return first.x() < second.x();
}


bool decWidth(const Rect &first, const Rect &second) {
  if (second.x() == first.x())
     return second.w() < first.w();
  return second.x() < first.x();
}


bool incHeight(const Rect &first, const Rect &second) {
  return first.y() < second.y();
}


bool decHeight(const Rect &first, const Rect &second) {
  if (second.y() == first.y())
     return second.h() < first.h();
  return second.y() < first.y();
}


//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->area().Inflate(screen.getBorderWidth() * 4),
                        spaces);
  
  //Sort the spaces by placement choice
  if (screen.rowPlacementDirection() == BScreen::TopBottom)
     sort(spaces.begin(), spaces.end(), decHeight);
  else
     sort(spaces.begin(), spaces.end(), incHeight);
  if (screen.colPlacementDirection() == BScreen::TopBottom)
     stable_sort(spaces.begin(), spaces.end(), incHeight);
  else
     stable_sort(spaces.begin(), spaces.end(), decHeight);

  //Find first space that fits the window
  best = NULL;
  for (siter=spaces.begin(); siter!=spaces.end(); ++siter) {
    if ((siter->w() >= win_size.w()) && (siter->h() >= win_size.h())) {
      if (best==NULL)
        best = siter;
      else if(siter->w()*siter->h()<best->h()*best->w())
        best = siter;
    }
  }
  if (best != NULL) {
    Point *pt = new Point(best->origin());
    if (screen.colPlacementDirection() != BScreen::TopBottom)
      pt->setY(pt->y() + (best->h() - win_size.h()));
    if (screen.rowPlacementDirection() != BScreen::LeftRight)
      pt->setX(pt->x() + (best->w() - win_size.w()));
    return pt;
  } else
    return NULL; //fall back to cascade
}


Point *Workspace::rowSmartPlacement(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->area().Inflate(screen.getBorderWidth() * 4),
                        spaces);
  
  //Sort the spaces by placement choice
  if (screen.colPlacementDirection() == BScreen::LeftRight)
     sort(spaces.begin(), spaces.end(), incWidth);
  else
     sort(spaces.begin(), spaces.end(), decWidth);
  if (screen.rowPlacementDirection() == BScreen::TopBottom)
     stable_sort(spaces.begin(), spaces.end(), incHeight);
  else
     stable_sort(spaces.begin(), spaces.end(), decHeight);

  //Find first space that fits the window
  best = NULL;
  for (siter=spaces.begin(); siter!=spaces.end(); ++siter)
    if ((siter->w() >= win_size.w()) && (siter->h() >= win_size.h())) {
      best = siter;
      break;
    }

  if (best != NULL) {
    Point *pt = new Point(best->origin());
    if (screen.colPlacementDirection() != BScreen::TopBottom)
      pt->setY(pt->y() + (best->h() - win_size.h()));
    if (screen.rowPlacementDirection() != BScreen::LeftRight)
      pt->setX(pt->x() + (best->w() - win_size.w()));
    return pt;
  } else
    return NULL; //fall back to cascade
}

Point *Workspace::colSmartPlacement(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->area().Inflate(screen.getBorderWidth() * 4),
                        spaces);
  
  //Sort the spaces by placement choice
  if (screen.rowPlacementDirection() == BScreen::LeftRight)
     sort(spaces.begin(),spaces.end(),incWidth);
  else
     sort(spaces.begin(),spaces.end(),decWidth);

  //Find first space that fits the window
  best = NULL;
  for (siter=spaces.begin(); siter!=spaces.end(); ++siter)
    if ((siter->w() >= win_size.w()) && (siter->h() >= win_size.h())) {
      best = siter;
      break;
    }

  if (best != NULL) {
    Point *pt = new Point(best->origin());
    if (screen.colPlacementDirection() != BScreen::TopBottom)
      pt->setY(pt->y() + (best->h() - win_size.h()));
    if (screen.rowPlacementDirection() != BScreen::LeftRight)
      pt->setX(pt->x() + (best->w() - win_size.w()));
    return pt;
  } else
    return NULL; //fall back to cascade
}


Point *const Workspace::cascadePlacement(const OpenboxWindow *const win){
  if (((unsigned) cascade_x > (screen.size().w() / 2)) ||
      ((unsigned) cascade_y > (screen.size().h() / 2)))
    cascade_x = cascade_y = 32;

  cascade_x += win->getTitleHeight();
  cascade_y += win->getTitleHeight();

  return new Point(cascade_x, cascade_y);
}


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

  // the following code is temporary and will be taken care of by Screen in the
  // future (with the NETWM 'strut')
  Rect space(0, 0, screen.size().w(), screen.size().h());

#ifdef    SLIT
  Slit *slit = screen.getSlit();
  Toolbar *toolbar = screen.getToolbar();
  int tbarh = screen.hideToolbar() ? 0 :
    toolbar->getExposedHeight() + screen.getBorderWidth() * 2;
  bool tbartop;
  switch (toolbar->placement()) {
  case Toolbar::TopLeft:
  case Toolbar::TopCenter:
  case Toolbar::TopRight:
    tbartop = true;
    break;
  case Toolbar::BottomLeft:
  case Toolbar::BottomCenter:
  case Toolbar::BottomRight:
    tbartop = false;
    break;
  default:
    ASSERT(false);      // unhandled placement
  }
  if ((slit->direction() == Slit::Horizontal &&
       (slit->placement() == Slit::TopLeft ||
        slit->placement() == Slit::TopRight)) ||
      slit->placement() == Slit::TopCenter) {
    // exclude top
    if (tbartop) {
      space.setY(slit->area().y());
      space.setH(space.h() - space.y());
    } else
      space.setH(space.h() - tbarh);
    space.setY(space.y() + slit->area().h() + screen.getBorderWidth() * 2);
    space.setH(space.h() - (slit->area().h() + screen.getBorderWidth() * 2));
  } else if ((slit->direction() == Slit::Vertical &&
              (slit->placement() == Slit::TopRight ||
               slit->placement() == Slit::BottomRight)) ||
             slit->placement() == Slit::CenterRight) {
    // exclude right
    space.setW(space.w() - (slit->area().w() + screen.getBorderWidth() * 2));
    if (tbartop)
      space.setY(space.y() + tbarh);
    space.setH(space.h() - tbarh);
  } else if ((slit->direction() == Slit::Horizontal &&
              (slit->placement() == Slit::BottomLeft ||
               slit->placement() == Slit::BottomRight)) ||
             slit->placement() == Slit::BottomCenter) {
    // exclude bottom
    space.setH(space.h() - (screen.size().h() - slit->area().y()));
  } else {// if ((slit->direction() == Slit::Vertical &&
    //      (slit->placement() == Slit::TopLeft ||
    //       slit->placement() == Slit::BottomLeft)) ||
    //     slit->placement() == Slit::CenterLeft)
    // exclude left
    space.setX(slit->area().w() + screen.getBorderWidth() * 2);
    space.setW(space.w() - (slit->area().w() + screen.getBorderWidth() * 2));
    if (tbartop)
      space.setY(space.y() + tbarh);
    space.setH(space.h() - tbarh);
  }
#else // !SLIT
  Toolbar *toolbar = screen.getToolbar();
  int tbarh = screen.hideToolbar() ? 0 :
    toolbar->getExposedHeight() + screen.getBorderWidth() * 2;
  switch (toolbar->placement()) {
  case Toolbar::TopLeft:
  case Toolbar::TopCenter:
  case Toolbar::TopRight:
    space.setY(toolbar->getExposedHeight());
    space.setH(space.h() - toolbar->getExposedHeight());
    break;
  case Toolbar::BottomLeft:
  case Toolbar::BottomCenter:
  case Toolbar::BottomRight:
    space.setH(space.h() - tbarh);
    break;
  default:
    ASSERT(false);      // unhandled placement
  }
#endif // SLIT

  const Size window_size(win->area().w()+screen.getBorderWidth() * 4,
                         win->area().h()+screen.getBorderWidth() * 4);
  Point *place = NULL;
  LinkedListIterator<OpenboxWindow> it(windowList);

  switch (screen.placementPolicy()) {
  case BScreen::BestFitPlacement:
    place = bestFitPlacement(window_size, space);
    break;
  case BScreen::RowSmartPlacement:
    place = rowSmartPlacement(window_size, space);
    break;
  case BScreen::ColSmartPlacement:
    place = colSmartPlacement(window_size, space);
    break;
  } // switch

  if (place == NULL)
    place = cascadePlacement(win);
 
  ASSERT(place != NULL);  
  if (place->x() + window_size.w() > (signed) screen.size().w())
    place->setX(((signed) screen.size().w() - window_size.w()) / 2);
  if (place->y() + window_size.h() > (signed) screen.size().h())
    place->setY(((signed) screen.size().h() - window_size.h()) / 2);

  win->configure(place->x(), place->y(), win->area().w(), win->area().h());
  delete place;
}