changeset 237:cbe9dbd80dfb

Add an anmviewer script.
author Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
date Sun, 01 Jan 2012 19:51:34 +0100
parents 741860192b56
children 0e1762b1ab9f
files anmviewer pytouhou/ui/anmrenderer.py setup.py
diffstat 3 files changed, 210 insertions(+), 1 deletions(-) [+]
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/anmviewer	Sun Jan 01 19:51:34 2012 +0100
@@ -0,0 +1,48 @@
+#!/usr/bin/env python
+# -*- encoding: utf-8 -*-
+##
+## Copyright (C) 2011 Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
+##
+## This program is free software; you can redistribute it and/or modify
+## it under the terms of the GNU General Public License as published
+## by the Free Software Foundation; version 3 only.
+##
+## This program 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 General Public License for more details.
+##
+
+import argparse
+import os
+
+import pyximport
+pyximport.install()
+
+from pytouhou.resource.loader import Loader
+from pytouhou.game.sprite import Sprite
+from pytouhou.vm.anmrunner import ANMRunner
+from pytouhou.ui.anmrenderer import ANMRenderer
+
+
+def main(path, data, name, script, sprites):
+    resource_loader = Loader()
+    resource_loader.scan_archives(os.path.join(path, name) for name in data)
+
+    # Get out animation
+    anm_wrapper = resource_loader.get_anm_wrapper((name,))
+    anm = ANMRenderer(resource_loader, anm_wrapper, script, sprites)
+    anm.start()
+
+
+parser = argparse.ArgumentParser(description='Viewer of ANM files, archives containing animations used in Touhou games.')
+
+parser.add_argument('data', metavar='DAT', default=('CM.DAT', 'ST.DAT'), nargs='*', help='Game’s .DAT data files')
+parser.add_argument('-p', '--path', metavar='DIRECTORY', default='.', help='Game directory path.')
+parser.add_argument('--anm', metavar='ANM', required=True, help='Select an ANM')
+parser.add_argument('--script', metavar='SCRIPT', type=int, default=0, help='First script to play')
+parser.add_argument('--sprites', action='store_true', default=False, help='Display sprites instead of scripts.')
+
+args = parser.parse_args()
+
+main(args.path, tuple(args.data), args.anm, args.script, args.sprites)
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/pytouhou/ui/anmrenderer.py	Sun Jan 01 19:51:34 2012 +0100
@@ -0,0 +1,161 @@
+#!/usr/bin/env python
+# -*- encoding: utf-8 -*-
+##
+## Copyright (C) 2011 Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
+##
+## This program is free software; you can redistribute it and/or modify
+## it under the terms of the GNU General Public License as published
+## by the Free Software Foundation; version 3 only.
+##
+## This program 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 General Public License for more details.
+##
+
+import pyglet
+import traceback
+
+from pyglet.gl import (glMatrixMode, glLoadIdentity, glEnable,
+                       glHint, glEnableClientState, glViewport,
+                       gluPerspective, GL_PROJECTION,
+                       GL_TEXTURE_2D, GL_BLEND,
+                       GL_PERSPECTIVE_CORRECTION_HINT, GL_NICEST,
+                       GL_COLOR_ARRAY, GL_VERTEX_ARRAY, GL_TEXTURE_COORD_ARRAY,
+                       glClear, GL_COLOR_BUFFER_BIT)
+
+from pytouhou.game.sprite import Sprite
+from pytouhou.vm.anmrunner import ANMRunner
+
+from pytouhou.utils.helpers import get_logger
+
+from .renderer import Renderer
+
+
+logger = get_logger(__name__)
+
+
+class ANMRenderer(pyglet.window.Window, Renderer):
+    def __init__(self, resource_loader, anm_wrapper, index=0, sprites=False):
+        Renderer.__init__(self, resource_loader)
+
+        width, height = 384, 448
+        pyglet.window.Window.__init__(self, width=width, height=height,
+                                      caption='PyTouhou', resizable=False)
+
+        self._anm_wrapper = anm_wrapper
+        self.anm = anm_wrapper.anm_files[0]
+        self.sprites = sprites
+        if sprites:
+            self.items = self.anm.sprites
+        else:
+            self.items = self.anm.scripts
+        self.load(index)
+
+        self.x = width / 2
+        self.y = height / 2
+
+
+    def start(self, width=384, height=448):
+        if (width, height) != (self.width, self.height):
+            self.set_size(width, height)
+
+        # Initialize OpenGL
+        glEnable(GL_BLEND)
+        glEnable(GL_TEXTURE_2D)
+        glHint(GL_PERSPECTIVE_CORRECTION_HINT, GL_NICEST)
+        glEnableClientState(GL_COLOR_ARRAY)
+        glEnableClientState(GL_VERTEX_ARRAY)
+        glEnableClientState(GL_TEXTURE_COORD_ARRAY)
+
+        # Switch to game projection
+        glMatrixMode(GL_PROJECTION)
+        glLoadIdentity()
+        gluPerspective(30, float(self.width) / float(self.height),
+                       101010101./2010101., 101010101./10101.)
+
+        self.setup_camera(0, 0, 1)
+
+        # Use our own loop to ensure 60 fps
+        pyglet.clock.set_fps_limit(60)
+        while not self.has_exit:
+            pyglet.clock.tick()
+            self.dispatch_events()
+            self.update()
+            self.flip()
+
+
+    def on_resize(self, width, height):
+        glViewport(0, 0, width, height)
+
+
+    def _event_text_symbol(self, ev):
+        # XXX: Ugly workaround to a pyglet bug on X11
+        #TODO: fix that bug in pyglet
+        try:
+            return pyglet.window.Window._event_text_symbol(self, ev)
+        except Exception as exc:
+            logger.warn('Pyglet error: %s', traceback.format_exc(exc))
+            return None, None
+
+
+    def on_key_press(self, symbol, modifiers):
+        if symbol == pyglet.window.key.ESCAPE:
+            self.has_exit = True
+        # XXX: Fullscreen will be enabled the day pyglet stops sucking
+        elif symbol == pyglet.window.key.F11:
+            self.set_fullscreen(not self.fullscreen)
+        elif symbol == pyglet.window.key.W:
+            self.load()
+        elif symbol == pyglet.window.key.X:
+            self.x, self.y = (192, 224) if self.x == 0 else (0, 0)
+        elif symbol == pyglet.window.key.LEFT:
+            self.change(-1)
+        elif symbol == pyglet.window.key.RIGHT:
+            self.change(+1)
+        elif symbol == pyglet.window.key.TAB:
+            self.toggle_sprites()
+        elif symbol >= pyglet.window.key.F1 and symbol <= pyglet.window.key.F12:
+            print (symbol - pyglet.window.key.F1 + (12 if modifiers == pyglet.window.key.MOD_CTRL else 0) + 1)
+            #self._anmrunner.interrupt(symbol - pyglet.window.key.F1 + (12 if modifiers == pyglet.window.key.MOD_CTRL) + 1)
+
+
+    def load(self, index=None):
+        if index is None:
+            index = self.num
+        self._sprite = Sprite()
+        print index
+        if self.sprites:
+            self._sprite.anm, self._sprite.texcoords = self._anm_wrapper.get_sprite(index)
+        else:
+            self._anmrunner = ANMRunner(self._anm_wrapper, index, self._sprite)
+            self._anmrunner.run_frame()
+        self.num = index
+
+
+    def change(self, diff):
+        keys = self.items.keys()
+        keys.sort()
+        index = keys.index(self.num) + diff
+        if index < 0 or index >= len(keys):
+            return
+        item = keys[index]
+        self.load(item)
+
+
+    def toggle_sprites(self):
+        self.sprites = not(self.sprites)
+        if self.sprites:
+            self.items = self.anm.sprites
+        else:
+            self.items = self.anm.scripts
+        self.load()
+
+
+    def update(self):
+        if not self.sprites:
+            self._anmrunner.run_frame()
+
+        glClear(GL_COLOR_BUFFER_BIT)
+        self.render_elements([self])
+
--- a/setup.py	Sun Jan 01 19:47:34 2012 +0100
+++ b/setup.py	Sun Jan 01 19:51:34 2012 +0100
@@ -73,7 +73,7 @@
       license='GPLv3',
       packages=packages,
       ext_modules=extensions,
-      scripts=['scripts/eosd'],
+      scripts=['scripts/eosd', 'scripts/anmviewer'],
       cmdclass={'build_ext': build_ext,
                 'build_scripts': BuildScripts},
       **extra