Merge branch 'master' of ctpug.org.za:koperkapel

[koperkapel.git] / koperkapel / scenes / base.py

diff --git a/koperkapel/scenes/base.py b/koperkapel/scenes/base.py
index fd53f817b9cc6b3fade5a0b617f42a90e1f1e84f..481850b2533153afa5ccbb12ea090aae5995bcf9 100644 (file)
--- a/koperkapel/scenes/base.py
+++ b/koperkapel/scenes/base.py
@@ -116,10 +116,22 @@ class Layer:
         self.name = name
         self.actors = []
 
+    def __iter__(self):
+        return iter(self.actors)
+
+    def __getitem__(self, i):
+        return self.actors[i]
+
+    def __len__(self):
+        return len(self.actors)
+
     def add(self, actor):
         self.actors.append(actor)
         return actor
 
+    def clear(self):
+        self.actors.clear()
+
     def remove(self, actor):
         self.actors.remove(actor)
         return actor
@@ -154,7 +166,7 @@ class Actors:
 
     def draw(self, screen):
         for lvl, name in self._ordered_layers:
-            for actor in self._layers[name].actors:
+            for actor in self._layers[name]:
                 actor.draw()  # TODO: allow an option screen to be passed in