diff pytouhou/resource/anmwrapper.py @ 97:ac2e5e1c2c3c

Refactor \o/
author Thibaut Girka <thib@sitedethib.com>
date Sun, 04 Sep 2011 23:50:00 +0200
parents
children dbb1a86c0235
line wrap: on
line diff
new file mode 100644
--- /dev/null
+++ b/pytouhou/resource/anmwrapper.py
@@ -0,0 +1,17 @@
+class AnmWrapper(object):
+    def __init__(self, anm_files):
+        self.anm_files = list(anm_files)
+
+
+    def get_sprite(self, sprite_index):
+        for anm in self.anm_files:
+            if sprite_index in anm.sprites:
+                return anm, anm.sprites[sprite_index]
+        raise IndexError
+
+
+    def get_script(self, script_index):
+        for anm in self.anm_files:
+            if script_index in anm.scripts:
+                return anm, anm.scripts[script_index]
+        raise IndexError