icons!

1 files changed, 8 insertions, 0 deletions

diff --git a/src/mapdata.py b/src/mapdata.py
index 013e308..131c669 100644
--- a/src/mapdata.py
+++ b/src/mapdata.py
@@ -65,6 +65,8 @@ class KPTileLayer(KPLayer):
 		self.objects = []
 		self.cache = ['DUMMY_FLAG']
 		self.updateCache()
+
+		self.icon = KP.icon('LayerTile')
 	
 	def _visibilityChanged(self, value):
 		for obj in self.objects:
@@ -145,6 +147,8 @@ class KPDoodadLayer(KPLayer):
 	def __init__(self):
 		KPLayer.__init__(self)
 		self.objects = []
+
+		self.icon = KP.icon('LayerObjects')
 	
 	def _visibilityChanged(self, value):
 		for obj in self.objects:
@@ -208,6 +212,8 @@ class KPPathLayer(KPLayer):
 		KPLayer.__init__(self)
 		self.nodes = []
 		self.paths = []
+
+		self.icon = KP.icon('LayerPath')
 	
 	def _visibilityChanged(self, value):
 		for objList in (self.nodes, self.paths):
@@ -267,6 +273,8 @@ class KPMap(object):
 
 					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)