path: root/src/mapdata.py

from common import *
import weakref
import mapfile

TILE_SIZE = (24,24)
MAP_SIZE_IN_TILES = (512,512)
MAP_SIZE = (MAP_SIZE_IN_TILES[0] * TILE_SIZE[0], MAP_SIZE_IN_TILES[1] * TILE_SIZE[1])

@mapfile.dumpable('layer')
class KPLayer(object):
	__dump_attribs__ = ('name', '_visible')

	def __repr__(self):
		return "<KPLayer %r>" % self.name

	def __init__(self):
		self.name = ''
		self._visible = True
	
	@property
	def visible(self):
		return self._visible

	@visible.setter
	def visible(self, value):
		if self._visible == value:
			return
		self._visible = value

		self._visibilityChanged(value)
	
	def _visibilityChanged(self, value):
		pass

	def setActivated(self, value, listToUse=None):
		flag1 = QtGui.QGraphicsItem.ItemIsSelectable
		flag2 = QtGui.QGraphicsItem.ItemIsMovable

		if listToUse is None:
			listToUse = self.objects

		for obj in listToUse:
			item = obj.qtItem
			if item:
				item.setFlag(flag1, value)
				item.setFlag(flag2, value)


@mapfile.dumpable('object')
class KPObject(object):
	__dump_attribs__ = ('position', 'size', 'tileset')

	def _load(self, mapObj, src):
		self.kind = mapObj.loadedTilesets[self.tileset].objects[src['kind']]
		self.updateCache()
	
	def _dump(self, mapObj, dest):
		dest['kind'] = mapObj.loadedTilesets[self.tileset].objects.index(self.kind)

	def __init__(self):
		self.position = (0,0)
		self.size = (1,1)
		self.kind = 0
		self.cache = []
		self.tileset = None
		self.qtItem = None
	
	def updateCache(self):
		self.cache = self.kind.render(self.size)


@mapfile.dumpable('tile_layer')
class KPTileLayer(KPLayer):
	__dump_attribs__ = KPLayer.__dump_attribs__ + ('tileset', 'objects')

	def _load(self, mapObj, src):
		self.updateCache()

	def __repr__(self):
		return "<KPTileLayer %r with %r>" % (self.name, self.tileset)

	def __init__(self):
		KPLayer.__init__(self)
		self.tileset = ''
		self.objects = []
		self.cache = ['DUMMY_FLAG']
		self.updateCache()

		self.icon = KP.icon('LayerTile')
	
	def _visibilityChanged(self, value):
		for obj in self.objects:
			item = obj.qtItem
			if item:
				item.setVisible(value)
	
	def updateCache(self):
		if len(self.objects) == 0:
			if len(self.cache) != 0:
				self.cache = []
				self.cacheBasePos = (0,0)
				self.cacheSize = (0,0)
			return

		x1, x2 = MAP_SIZE_IN_TILES[0] - 1, 0
		y1, y2 = MAP_SIZE_IN_TILES[1] - 1, 0

		for obj in self.objects:
			x, y = obj.position
			w, h = obj.size
			right, bottom = (x+w-1), (y+h-1)

			if x < x1:
				x1 = x
			if y < y1:
				y1 = y
			if right > x2:
				x2 = right
			if bottom > y2:
				y2 = bottom


		# create the cache
		# I was going to just resize it, but setting every tile to -1
		# in Python would probably be slower than creating a new one ...
		size = (x2 - x1 + 1, y2 - y1 + 1)
		width, height = size

		cache = [[-1 for i in xrange(width)] for j in xrange(height)]
		self.cache = cache

		self.cacheBasePos = (x1, y1)
		self.cacheSize = size

		# now generate the thing
		for obj in self.objects:
			oX, oY = obj.position
			baseX = oX - x1
			y = oY - y1

			for row in obj.cache:
				destRow = cache[y]
				x = baseX
				for tile in row:
					destRow[x] = tile
					x += 1
				y += 1


@mapfile.dumpable('associate_layer')
class KPPathTileLayer(KPLayer):
	__dump_attribs__ = KPLayer.__dump_attribs__ + ('tileset', 'objects', 'doodads')

	def _dump(self, mapObj, dest):
		if isinstance(self.associate, KPNode):
			dest['associatedRef'] = ('node', mapObj.refNode(self.associate))
		else:
			dest['associatedRef'] = ('path', mapObj.refPath(self.associate))

	def _load(self, mapObj, src):
		assocType, assocID = src['associatedRef']

		if assocType == 'path':
			self.associate = mapObj.derefPath(assocID)
		else:
			self.associate = mapObj.derefNode(assocID)

		self.updateCache()

	def __repr__(self):
		return "<KPPathTileLayer with %r connected to %r>" % (self.tileset, self.associate)

	def __init__(self, pathnode=None):
		KPLayer.__init__(self)

		self.cache = ['DUMMY_FLAG']

		if pathnode is None:
			return

		self.tileset = KP.map.pathNodeTileset
		self.objects = []
		self.doodads = []
		self.associate = pathnode

		self.updateCache()

	def _visibilityChanged(self, value):
		for obj in self.objects:
			item = obj.qtItem
			if item:
				item.setVisible(value)
	
		for obj in self.doodads:
			item = obj.qtItem
			if item:
				item.setVisible(value)

	def setActivated(self, value, listToUse=None):
		flag1 = QtGui.QGraphicsItem.ItemIsSelectable
		flag2 = QtGui.QGraphicsItem.ItemIsMovable

		if listToUse is None:
			listToUse = self.objects + self.doodads

		for obj in listToUse:
			item = obj.qtItem
			if item:
				item.setFlag(flag1, value)
				item.setFlag(flag2, value)

	def updateCache(self):
		if len(self.objects) == 0:
			if len(self.cache) != 0:
				self.cache = []
				self.cacheBasePos = (0,0)
				self.cacheSize = (0,0)
			return

		x1, x2 = MAP_SIZE_IN_TILES[0] - 1, 0
		y1, y2 = MAP_SIZE_IN_TILES[1] - 1, 0

		for obj in self.objects:
			x, y = obj.position
			w, h = obj.size
			right, bottom = (x+w-1), (y+h-1)

			if x < x1:
				x1 = x
			if y < y1:
				y1 = y
			if right > x2:
				x2 = right
			if bottom > y2:
				y2 = bottom


		# create the cache
		# I was going to just resize it, but setting every tile to -1
		# in Python would probably be slower than creating a new one ...
		size = (x2 - x1 + 1, y2 - y1 + 1)
		width, height = size

		cache = [[-1 for i in xrange(width)] for j in xrange(height)]
		self.cache = cache

		self.cacheBasePos = (x1, y1)
		self.cacheSize = size

		# now generate the thing
		for obj in self.objects:
			oX, oY = obj.position
			baseX = oX - x1
			y = oY - y1

			for row in obj.cache:
				destRow = cache[y]
				x = baseX
				for tile in row:
					destRow[x] = tile
					x += 1
				y += 1


@mapfile.dumpable('doodad')
class KPDoodad(object):
	__dump_attribs__ = ('position', 'size', 'angle', 'animations')

	def _dump(self, mapObj, dest):
		dest['sourceRef'] = mapObj.refDoodad(self.source)
	
	def _load(self, mapObj, src):
		self.source = mapObj.derefDoodad(src['sourceRef'])

	def __init__(self):
		self.position = [0,0]
		self.size = [0,0]
		self.angle = 0
		self.source = None
		self.animations = []
		self.timelines = None

	def setDefaultSize(self):
		pixmap = self.source[1]
		self.size = [pixmap.width(), pixmap.height()]

	def cleanUpAnimations(self):
		myTimelines = self.timelines
		if myTimelines is None: return

		timelineList = KP.mapScene.timeLines

		for timeline in myTimelines:
			try:
				timelineList.remove(timeline)
			except ValueError:
				pass

		self.timelines = None

	def setupAnimations(self):
		self.cleanUpAnimations()

		timelineList = KP.mapScene.timeLines
		myTimelines = []

		for anim in self.animations:
			Loop, Curve, Frames, Type, StartVal, EndVal = anim

			Timeline = QtCore.QTimeLine()

			# Interpolate the correct modifier
			if Curve == "Linear":
				Timeline.setCurveShape(3)
			elif Curve == "Sinusoidial":
				Timeline.setCurveShape(4)
			elif Curve == "Cosinoidial":
				Timeline.setCurveShape(5)

			Timeline.setFrameRange(StartVal, EndVal)

			if Loop == "Contiguous":
				Timeline.setLoopCount(1)
			elif Loop == "Loop":
				Timeline.setLoopCount(0) # Dollars *holds pinky to corner of mouth*
			elif Loop == "Reversible Loop":
				Timeline.setLoopCount(1)
				Timeline.finished.connect(Timeline.toggleDirection)
				Timeline.finished.connect(Timeline.start)

			Timeline.setDuration(Frames/60.0*1000) # Wii goes at 60 frames per second

			timelineList.append(Timeline)
			myTimelines.append(Timeline)

		self.timelines = myTimelines


@mapfile.dumpable('doodad_layer')
class KPDoodadLayer(KPLayer):
	__dump_attribs__ = KPLayer.__dump_attribs__ + ('objects',)

	def __repr__(self):
		return "<KPDoodadLayer %r>" % self.name

	def __init__(self):
		KPLayer.__init__(self)
		self.objects = []

		self.icon = KP.icon('LayerObjects')
	
	def _visibilityChanged(self, value):
		for obj in self.objects:
			item = obj.qtItem
			if item:
				item.setVisible(value)


@mapfile.dumpable('node')
class KPNode(object):  
	__dump_attribs__ = (
			'position', 'actions', 'level', 'mapChange',
			'transition', 'mapID', 'foreignID')

	def _dump(self, mapObj, dest):
		dest['exitIDs'] = map(mapObj.refPath, self.exits)
	
	def _load(self, mapObj, src):
		self.exitIDs = src['exitIDs']
		# The exits array will be created by KPPathLayer._load after the
		# paths have all been created.

	def __init__(self):
		self.position = (0,0)
		self.actions = []
		self.exits = []
		self.level = None
		self.mapChange = None
		self.transition = 0
		self.mapID = None
		self.foreignID = None
	
	def isStop(self):
		return True if (self.level or self.mapChange or len(self.exits) != 2) else False


@mapfile.dumpable('path')
class KPPath(object):
	__dump_attribs__ = ('unlocks', 'secret', 'animation', 'movementSpeed')

	def _dump(self, mapObj, dest):
		dest['startNodeLink'] = mapObj.refNode(self._startNodeRef())
		dest['endNodeLink'] = mapObj.refNode(self._endNodeRef())
		# dest['linkedLayer'] = mapObj.refLayer(self.linkedLayer)
	
	def _load(self, mapObj, src):
		self._startNodeRef = weakref.ref(mapObj.derefNode(src['startNodeLink']))
		self._endNodeRef = weakref.ref(mapObj.derefNode(src['endNodeLink']))
		# self.linkedLayer = mapObj.derefLayer(src['linkedLayer'])

	def __init__(self, startNode=None, endNode=None, cloneFrom=None):
		if startNode is None and endNode is None:
			# null ctor, ignore this
			# we're probably loaded from a file, so trust
			# that everything is correct ... _load will set it all up
			return

		self._startNodeRef = weakref.ref(startNode)
		self._endNodeRef = weakref.ref(endNode)

		startNode.exits.append(self)
		endNode.exits.append(self)

		self.unlocks = 0 	# 0 = always unlocked, 1 = unlocked from startNode, 2 = unlocked from endNode
		self.secret = 0 	# 0 = unlocks from normal exit, 1 = unlocks from secret exit

		if cloneFrom is None:
			self.animation = 0
		else:
			self.animation = cloneFrom.animation

		self.movementSpeed = 1.0
		# self.linkedLayer = None


	def setStart(self, newStart):
		currentStart = self._startNodeRef()
		if currentStart is not None:
			currentStart.exits.remove(self)

		newStart.exits.append(self)
		self._startNodeRef = weakref.ref(newStart)

	def setEnd(self, newEnd):
		currentEnd = self._endNodeRef()
		if currentEnd is not None:
			currentEnd.exits.remove(self)

		newEnd.exits.append(self)
		self._endNodeRef = weakref.ref(newEnd)


@mapfile.dumpable('path_layer')
class KPPathLayer(KPLayer):
	__dump_attribs__ = KPLayer.__dump_attribs__ + ('nodes', 'paths')

	def _load(self, mapObj, src):
		for node in self.nodes:
			node.exits = map(mapObj.derefPath, node.exitIDs)
			del node.exitIDs

	def __repr__(self):
		return "<KPPathLayer %r>" % self.name

	def __init__(self):
		KPLayer.__init__(self)
		self.nodes = []
		self.paths = []

		self.icon = KP.icon('LayerPath')
	
	def _visibilityChanged(self, value):
		for objList in (self.nodes, self.paths):
			for obj in objList:
				item = obj.qtItem
				if item:
					item.setVisible(value)
	
	def setActivated(self, value):
		KPLayer.setActivated(self, value, self.nodes)

		flag = QtGui.QGraphicsItem.ItemIsSelectable
		for path in self.paths:
			item = path.qtItem
			if item:
				item.setFlag(flag, value)


@mapfile.dumpable('map_root')
class KPMap(object):
	__dump_attribs__ = ('layers', 'associateLayers', 'nextLayerNumber', 'doodadDefinitions', 'pathNodeTileset')

	def _preload(self, src):
		# we need this early so we can use the deref methods!
		for layer in self.layers:
			if isinstance(layer, KPPathLayer):
				self.pathLayer = layer

	def _load(self, mapObj, source):
		self.layerModel.list = self.layers
		self.doodadModel.list = self.doodadDefinitions

	def save(self):
		path = self.filePath
		if path is None:
			raise "no path specified for this map"

		self.associateLayers = KP.mainWindow.pathNodeList.getLayers()

		import mapfile
		dumped = mapfile.dump(self)
		open(path, 'wb').write(dumped)

	def export(self, path):
		from exporter import KPMapExporter
		exp = KPMapExporter(self)
		data = exp.build()
		open(path, 'wb').write(data)

	def __init__(self):
		self.filePath = None

		self.nextLayerNumber = 1

		self.pathLayer = self._createPathLayer()
		self.layers = [self.pathLayer]
		self.associateLayers = []
		self.layerModel = KPMap.LayerModel(self.layers)

		self.doodadDefinitions = []
		self.doodadModel = KPMap.DoodadModel(self.doodadDefinitions)

		self.tilesets = {}
		self.loadedTilesets = {}

		self.loadTilesets()

		try:
			self.pathNodeTileset = self.tilesets.keys()[0]
		except:
			# TODO: This should probably error out or something. Seriously.
			#       It'll ruin the path or object layers if there are NO tilesets.
			self.pathNodeTileset = ''


	# LAYERS
	class LayerModel(QtCore.QAbstractListModel):
		def __init__(self, layerList):
			QtCore.QAbstractListModel.__init__(self)
			self.list = layerList


		def headerData(self, section, orientation, role = Qt.DisplayRole):
			return 'Layer'

		def rowCount(self, parent):
			return len(self.list)

		def data(self, index, role = Qt.DisplayRole):
			try:
				if index.isValid():
					layer = self.list[index.row()]

					if (role == Qt.DisplayRole or role == Qt.EditRole):
						return layer.name
					elif role == Qt.DecorationRole:
						return layer.icon
					elif role == Qt.CheckStateRole:
						return (Qt.Checked if layer.visible else Qt.Unchecked)

			except IndexError:
				pass

			return QtCore.QVariant()

		def flags(self, index):
			if not index.isValid():
				return Qt.ItemIsEnabled
			
			return Qt.ItemIsEditable | Qt.ItemIsUserCheckable \
					| QtCore.QAbstractListModel.flags(self, index)

		def setData(self, index, value, role = Qt.EditRole):
			if index.isValid():
				layer = self.list[index.row()]

				if role == Qt.EditRole:
					value = str(value.toString())
					if len(value) > 0:
						layer.name = value
						self.dataChanged.emit(index, index)
						return True

				elif role == Qt.CheckStateRole:
					layer.visible = value.toBool()
					self.dataChanged.emit(index, index)
					return True

			return False

	def _createPathLayer(self):
		layer = KPPathLayer()
		layer.name = 'Paths'
		return layer

	def createNewTileLayer(self, tilesetName):
		layer = KPTileLayer()
		layer.name = "Tilemap - Layer %d" % self.nextLayerNumber
		self.nextLayerNumber += 1
		layer.tileset = tilesetName
		return layer

	def createNewDoodadLayer(self):
		layer = KPDoodadLayer()
		layer.name = "Doodads - Layer %d" % self.nextLayerNumber
		self.nextLayerNumber += 1
		return layer

	def appendLayer(self, layer):
		return self.insertLayer(len(self.layers), layer)
	
	def insertLayer(self, index, layer):
		self.layerModel.beginInsertRows(QtCore.QModelIndex(), index, index)
		self.layers.insert(index, layer)
		self.layerModel.endInsertRows()
		return index

	def moveLayer(self, fromIndex, toIndex):
		if fromIndex == toIndex:
			return
		if fromIndex < 0 or toIndex < 0:
			return
		if fromIndex >= len(self.layers) or toIndex > len(self.layers):
			return
		if fromIndex == toIndex-1:
			return

		self.layerModel.beginMoveRows(
				QtCore.QModelIndex(), fromIndex, fromIndex,
				QtCore.QModelIndex(), toIndex)

		toMove = self.layers[fromIndex]
		newIndex = ((toIndex > fromIndex) and (toIndex - 1)) or toIndex

		del self.layers[fromIndex]
		self.layers.insert(newIndex, toMove)

		self.layerModel.endMoveRows()

		return newIndex

	def removeLayer(self, layer):
		if layer not in self.layers:
			return

		index = self.layers.index(layer)
		self.layerModel.beginRemoveRows(QtCore.QModelIndex(), index, index)
		del self.layers[index]
		self.layerModel.endRemoveRows()
	

	# DOODADS
	class DoodadModel(QtCore.QAbstractListModel):
		def __init__(self, doodadList):
			QtCore.QAbstractListModel.__init__(self)
			self.list = doodadList


		def headerData(self, section, orientation, role = Qt.DisplayRole):
			return 'Doodad'

		def rowCount(self, parent):
			return len(self.list)

		def data(self, index, role = Qt.DisplayRole):
			try:
				if index.isValid():
					doodad = self.list[index.row()]

					if role == Qt.DecorationRole:
						return QtGui.QIcon(doodad[1])
					elif role == Qt.ToolTipRole:
						return doodad[0]
					elif role == Qt.DisplayRole:
						return doodad[0]

			except IndexError:
				pass

			return QtCore.QVariant()

		def flags(self, index):
			if not index.isValid():
				return Qt.ItemIsEnabled
			return QtCore.QAbstractListModel.flags(self, index)

	def addDoodad(self, title, image):
		doodad = (title, image)
		
		index = len(self.doodadDefinitions)
		self.doodadModel.beginInsertRows(QtCore.QModelIndex(), index, index)
		self.doodadDefinitions.append(doodad)
		self.doodadModel.endInsertRows()

		return doodad
	
	def removeDoodad(self, doodad):
		if doodad not in self.doodadDefinitions:
			raise ValueError

		index = self.doodadDefinitions.index(doodad)
		self.doodadModel.beginRemoveRows(QtCore.QModelIndex(), index, index)
		del self.doodadDefinitions[index]
		self.doodadModel.endRemoveRows()


	# REFERENCES
	def refDoodad(self, doodad):
		return -1 if (doodad is None) else self.doodadDefinitions.index(doodad)
	def derefDoodad(self, ref):
		return self.doodadDefinitions[ref] if (ref >= 0) else None

	def refLayer(self, layer):
		return -1 if (layer is None) else self.layers.index(layer)
	def derefLayer(self, ref):
		return self.layers[ref] if (ref >= 0) else None

	def refPath(self, path):
		return -1 if (path is None) else self.pathLayer.paths.index(path)
	def derefPath(self, ref):
		return self.pathLayer.paths[ref] if (ref >= 0) else None

	def refNode(self, node):
		return -1 if (node is None) else self.pathLayer.nodes.index(node)
	def derefNode(self, ref):
		return self.pathLayer.nodes[ref] if (ref >= 0) else None

	# TILESETS
	def loadTilesets(self):
		import os
		from hashlib import sha256 as sha

		path = os.getcwd() + os.sep + 'Tilesets'

		if not os.path.exists(path):
			os.mkdir(path)

		for file in os.listdir(path):
			name = file[:-4]

			if file[-4:] == '.arc':
				filepath = path + os.sep + file
				
				filehandler = open(filepath)
				data = filehandler.read()
				filehandler.close()
				hash = sha(data).hexdigest()
				
				self.tilesets[name] = {'path': filepath, 'hash': hash}
				self.loadedTilesets[name] = KPTileset.loadFromArc(filepath)


	def reloadTileset(self, name):
		from hashlib import sha256 as sha

		info = self.tilesets[name]

		filehandler = open(info['path'])
		data = filehandler.read()
		filehandler.close()
		hash = sha(data).hexdigest()

		if info['hash'] != hash:
			self.loadedTilesets[name] = KPTileset.loadFromArc(info['path'])

			KP.mapscene.update()
			KP.mapscene.views()[0].update()