# -*- coding: utf-8 -*-

# ####################################################################
#  Copyright (C) 2005-2009 by the FIFE team
#  http://www.fifengine.de
#  This file is part of FIFE.
#
#  FIFE is free software; you can redistribute it and/or
#  modify it under the terms of the GNU Lesser General Public
#  License as published by the Free Software Foundation; either
#  version 2.1 of the License, or (at your option) any later version.
#
#  This library 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
#  Lesser General Public License for more details.
#
#  You should have received a copy of the GNU Lesser General Public
#  License along with this library; if not, write to the
#  Free Software Foundation, Inc.,
#  51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
# ####################################################################

import editor
import pdb

import math

from fife import fife
import editor
import events
import undomanager

from fife.extensions.pychan.tools import callbackWithArguments as cbwa

class MapController(object):
	""" MapController provides an interface for editing maps """
	def __init__(self, map):
		
		self._editor = editor.getEditor()
		self._engine = self._editor.getEngine()

		self._camera = None     # currently selected camera
		self._layer = None      # currently selected layer
		self._selection = []	# currently selected cells
		self._map = None
		self._undo = False
		self._undomanager = undomanager.UndoManager()
		undomanager.preUndo.connect(self._startUndo, sender=self._undomanager)
		undomanager.preRedo.connect(self._startUndo, sender=self._undomanager)
		undomanager.postUndo.connect(self._endUndo, sender=self._undomanager)
		undomanager.postRedo.connect(self._endUndo, sender=self._undomanager)
		self.debug = False
		
		self.overwriteInstances = True # Remove instances on cell before placing new instance
		
		if map is not None:
			self.setMap(map.getId())
			
	def cleanUp(self):
		undomanager.preUndo.disconnect(self._startUndo, sender=self._undomanager)
		undomanager.preRedo.disconnect(self._startUndo, sender=self._undomanager)
		undomanager.postUndo.disconnect(self._endUndo, sender=self._undomanager)
		undomanager.postRedo.disconnect(self._endUndo, sender=self._undomanager)
		self._undomanager.clear()
		
		self._editor = None
		self._engine = None
		self._map = None
		self._selection = []
		self._layer = None
		self._camera = None
		
	def setMap(self, mapid):
		""" Set the map to be edited """
		self._camera = None
		self._map = None
		self._layer = None
		self._selection = []

		self._map = self._engine.getModel().getMap(mapid)
		if not self._map.getLayers():
			raise AttributeError('Editor error: map ' + self._map.getId() + ' has no layers. Cannot edit.')

		for cam in self._map.getCameras():
			if cam.getLocationRef().getMap().getId() == self._map.getId():
				self._camera = cam
				break

		self._layer = self._map.getLayers()[0]
		
		gridrenderer = self._camera.getRenderer('GridRenderer')
		gridrenderer.activateAllLayers(self._map)

		blockrenderer = self._camera.getRenderer('BlockingInfoRenderer')
		blockrenderer.activateAllLayers(self._map)

		cellrenderer = self._camera.getRenderer('CellSelectionRenderer')
		cellrenderer.activateAllLayers(self._map)

	def getMap(self):
		return self._map
	
	def selectLayer(self, layerid):
		""" Select layer to be edited """
		self.deselectSelection()
		self._layer = None
		layers = [l for l in self._map.getLayers() if l.getId() == layerid]
		if len(layers) == 1:
			self._layer = layers[0]

	def deselectSelection(self):
		""" Deselects all selected cells """
		if not self._camera: 
			if self.debug: print 'No camera bind yet, cannot select any cell'
			return
		self._selection = []
		fife.CellSelectionRenderer.getInstance(self._camera).reset()
		
	def clearSelection(self):
		""" Removes all instances on selected cells """
		instances = self.getInstancesFromSelection()
		self._undomanager.startGroup("Cleared selection")
		self.removeInstances(instances)
		self._undomanager.endGroup()
		
	def fillSelection(self, object):
		""" Adds an instance of object on each selected cell """
		self._undomanager.startGroup("Filled selection")
		for loc in self._selection:
			self.placeInstance(loc.getLayerCoordinates(), object)
		self._undomanager.endGroup()

	def selectCell(self, screenx, screeny):
		""" Selects a cell at a position on screen """
		if not self._camera: 
			if self.debug: print 'No camera bind yet, cannot select any cell'
			return
		if not self._layer:
			if self.debug: print 'No layer assigned in selectCell'
			return

		mapCoords = self._camera.toMapCoordinates(fife.ScreenPoint(screenx, screeny), False)
		position = self._layer.getCellGrid().toLayerCoordinates(mapCoords)
		
		loc = fife.Location(self._layer)
		loc.setLayerCoordinates(position)
		
		for i in self._selection:
			if loc == i: return
			
		self._selection.append( loc )
		fife.CellSelectionRenderer.getInstance(self._camera).selectLocation(loc)
		
	def deselectCell(self, screenx, screeny):
		""" Deselects a cell at a position on screen """
		if not self._camera: 
			if self.debug: print 'No camera bind yet, cannot select any cell'
			return
		if not self._layer:
			if self.debug: print 'No layer assigned in selectCell'
			return

		mapCoords = self._camera.toMapCoordinates(fife.ScreenPoint(screenx, screeny), False)
		position = self._layer.getCellGrid().toLayerCoordinates(mapCoords)
		
		loc = fife.Location(self._layer)
		loc.setLayerCoordinates(position)
		
		for i in self._selection:
			if loc == i:
				self._selection.remove( loc )
				fife.CellSelectionRenderer.getInstance(self._camera).deselectLocation(loc)
				return
		
		
	def getInstancesFromSelection(self):
		""" Returns all instances in the selected cells """
		instances = []
		
		for loc in self._selection:
			instances.extend(self.getInstancesFromPosition(loc.getLayerCoordinates()))

		return instances

	def getInstancesFromPosition(self, position, layer=None):
		""" Returns all instances on a specified position """
		if not self._layer and not layer:
			if self.debug: print 'No layer assigned in getInstancesFromPosition'
			return
		if not position:
			if self.debug: print 'No position assigned in getInstancesFromPosition'
			return

		if layer:
			loc = fife.Location(layer)
		else:
			loc = fife.Location(self._layer)
		if type(position) == fife.ExactModelCoordinate:
			loc.setExactLayerCoordinates(position)
		else:
			loc.setLayerCoordinates(position)
			
		if layer:
			instances = layer.getInstancesAt(loc)
		else:
			instances = self._layer.getInstancesAt(loc)

		return instances

	def getUndoManager(self):
		""" Returns undo manager """
		return self._undomanager

	def undo(self):
		""" Undo one level """
		self._undomanager.undo()

	def redo(self):
		""" Redo one level """
		self._undomanager.redo()
		
	def _startUndo(self):
		""" Called before a undo operation is performed. Makes sure undo stack does not get corrupted """
		self._undo = True
		
	def _endUndo(self):
		""" Called when a undo operation is done """
		self._undo = False

	def placeInstance(self, position, object, layer=None):
		""" Places an instance of object at position. Any existing instances on position are removed. """
		mname = 'placeInstance'
		if not object:
			if self.debug: print 'No object assigned in %s' % mname
			return
		if not position:
			if self.debug: print 'No position assigned in %s' % mname
			return
		if not self._layer:
			if self.debug: print 'No layer assigned in %s' % mname
			return

		if self.debug: print 'Placing instance of ' + object.getId() + ' at ' + str(position)
		
		# Remove instances from target position
		if not self._undo:
			instances = self.getInstancesFromPosition(position)
			if len(instances) == 1:
				# Check if the only instance at position has the same object
				objectId = instances[0].getObject().getId()
				objectNs = instances[0].getObject().getNamespace()
				if objectId == object.getId() and objectNs == object.getNamespace():
					if self.debug: print "Tried to place duplicate instance"
					return
					
			self._undomanager.startGroup("Placed instance")
			self.removeInstances(instances)

		if layer:
			inst = layer.createInstance(object, position)
		else:
			inst = self._layer.createInstance(object, position)
		fife.InstanceVisual.create(inst)
		
		if not self._undo:
			redocall = cbwa(self.placeInstance, position, object, inst.getLocation().getLayer())
			undocall = cbwa(self.removeInstanceOfObjectAt, position, object, inst.getLocation().getLayer())
			undoobject = undomanager.UndoObject(undocall, redocall, "Placed instance")
			self._undomanager.addAction(undoobject)
			self._undomanager.endGroup()
			
	def removeInstanceOfObjectAt(self, position, object, layer=None):
		""" Removes the first instance of object it can find on position """
		instances = self.getInstancesFromPosition(position, layer)
		for i in instances:
			if i.getObject().getId() == object.getId() and i.getObject().getNamespace() == object.getNamespace():
				self.removeInstances([i],layer)
				return
					
	def removeInstances(self, instances, layer=None):
		""" Removes all provided instances """
		mname = 'removeInstances'
		if not instances:
			if self.debug: print 'No instances assigned in %s' % mname
			return
			
		for i in instances:
			if self.debug: print 'Deleting instance ' + i.getObject().getId() + ' at ' + str(i.getLocation().getExactLayerCoordinates())
			
			if not self._undo:
				object = i.getObject()
				position = i.getLocation().getExactLayerCoordinates()
				undocall = cbwa(self.placeInstance, position, object, i.getLocation().getLayer())
				redocall = cbwa(self.removeInstanceOfObjectAt, position, object, i.getLocation().getLayer())
				undoobject = undomanager.UndoObject(undocall, redocall, "Removed instance")
				self._undomanager.addAction(undoobject)
			
			if layer:
				layer.deleteInstance(i)
			else:
				self._layer.deleteInstance(i)

	def moveInstances(self, instances, moveBy, exact=False, origLoc=None, origFacing=None):
		""" Moves provided instances by moveBy. If exact is false, the instances are
		snapped to closest cell. origLoc and origFacing are only set when an undo/redo
		operation takes place and will have no effect and should not be used under normal
		circumstances."""
		mname = 'moveInstances'
		if not self._layer:
			if self.debug: print 'No layer assigned in %s' % mname
			return
			
		if exact and type(moveBy) != fife.ExactModelCoordinate:
			moveBy = fife.ExactModelCoordinate(float(moveBy.x), float(moveBy.y), float(moveBy.z))
		elif exact is False and type(moveBy) != fife.ModelCoordinate:
			moveBy = fife.ModelCoordinate(int(round(moveBy.x)), int(round(moveBy.y)), int(round(moveBy.z)))
			
		for i in instances:
			loc = i.getLocation()
			f = i.getFacingLocation()
			if exact:
				newCoords = loc.getExactLayerCoordinates() + moveBy
				loc.setExactLayerCoordinates(newCoords)
				f.setExactLayerCoordinates(f.getExactLayerCoordinates() + moveBy)
			else:
				# Move instance and snap to closest cell
				newCoords = loc.getLayerCoordinates() + moveBy
				loc.setLayerCoordinates(newCoords)
				f.setLayerCoordinates(f.getLayerCoordinates() + moveBy)
				
			if not self._undo:
				undocall = cbwa(self.moveInstances, [i], moveBy, exact, i.getLocation(), i.getFacingLocation())
				redocall = cbwa(self.moveInstances, [i], moveBy, exact, i.getLocation(), i.getFacingLocation())
				undoobject = undomanager.UndoObject(undocall, redocall, "Moved instance")
				self._undomanager.addAction(undoobject)
				i.setLocation(loc)
				i.setFacingLocation(f)
				
			else:
				assert(origLoc)
				assert(origFacing)
				i.setLocation(origLoc)
				i.setFacingLocation(origFacing)


	def rotateCounterClockwise(self):
		""" Rotates map counterclockwise by 90 degrees """
		currot = self._camera.getRotation()
		self._camera.setRotation((currot + 270) % 360)
		
	def rotateClockwise(self):
		""" Rotates map clockwise by 90 degrees """
		currot = self._camera.getRotation()
		self._camera.setRotation((currot + 90) % 360)
		
	def getZoom(self):
		""" Returns camera zoom """
		if not self._camera: 
			if self.debug: print 'No camera bind yet, cannot get zoom'
			return 0
		return self._camera.getZoom()
		
	def setZoom(self, zoom):
		""" Sets camera zoom """
		if not self._camera: 
			if self.debug: print 'No camera bind yet, cannot get zoom'
			return
		self._camera.setZoom(zoom)

	def moveCamera(self, screen_x, screen_y):
		""" Move camera (scroll) by screen_x, screen_y """
		if not self._camera: 
			return
			
		coords = self._camera.getLocationRef().getMapCoordinates()
		z = self._camera.getZoom()
		r = self._camera.getRotation()
		if screen_x:
			coords.x -= screen_x / z * math.cos(r / 180.0 * math.pi) / 100;
			coords.y -= screen_x / z * math.sin(r / 180.0 * math.pi) / 100;
		if screen_y:
			coords.x -= screen_y / z * math.sin(-r / 180.0 * math.pi) / 100;
			coords.y -= screen_y / z * math.cos(-r / 180.0 * math.pi) / 100;
		coords = self._camera.getLocationRef().setMapCoordinates(coords)
		self._camera.refresh()