actor crawling with weapon image

opencombat/gui/actor.py

@@ -6,6 +6,8 @@ from synergine2.config import Config
 
 from synergine2.simulation import Subject
 from synergine2_cocos2d.actor import Actor
+
+from opencombat.exception import UnknownWeapon
 from opencombat.gui.animation import ANIMATION_WALK
 from opencombat.gui.animation import ANIMATION_CRAWL
 from opencombat.gui.weapon import WeaponImageApplier
@@ -19,6 +21,7 @@ FLAG_COLORS = {
 }
 
 MAN_STAND_UP = 'MAN_STAND_UP'
+MAN_CRAWLING = 'MAN_CRAWLING'
 
 
 class BaseActor(Actor):
@@ -28,12 +31,13 @@ class BaseActor(Actor):
         config: Config,
         subject: Subject,
     ) -> None:
+        self._mode = MAN_STAND_UP
         self.weapon_image_applier = WeaponImageApplier(config, self)
         super().__init__(image_path, subject=subject, config=config)
 
     @property
     def mode(self) -> str:
-        return MAN_STAND_UP
+        return self._mode
 
     @property
     def weapons(self) -> typing.List[str]:
@@ -50,6 +54,25 @@ class BaseActor(Actor):
             )
         ]
 
+    def get_animation_appliable_images(
+        self,
+        animation_name: str,
+        animation_position: int,
+    ) -> typing.List[Image.Image]:
+        if not self.weapons:
+            return []
+
+        try:
+            return [
+                self.weapon_image_applier.get_animation_image_for_weapon(
+                    animation_name,
+                    self.weapons[0],
+                    animation_position,
+                )
+            ]
+        except UnknownWeapon:
+            return []
+
     def firing(self) -> None:
         pass
 

opencombat/gui/weapon.py

@@ -29,9 +29,23 @@ class WeaponImageApplier(ImageApplier):
 
     def get_images_scheme(self) -> typing.Dict[str, typing.Dict[str, str]]:
         from opencombat.gui.actor import MAN_STAND_UP
+        from opencombat.gui.actor import MAN_CRAWLING
         return {
             MAN_STAND_UP: {
-                RIFFLE: 'actors/man_weap1.png',
+                RIFFLE: [
+                    'actors/man_weap1.png'
+                ],
+            },
+            # FIXME NOW
+            # MAN_CRAWLING: {
+            'CRAWL': {
+                RIFFLE: [
+                    'actors/man_c1_weap1.png',
+                    'actors/man_c2_weap1.png',
+                    'actors/man_c3_weap1.png',
+                    'actors/man_c4_weap1.png',
+                ],
+
             }
         }
 
@@ -39,7 +53,24 @@ class WeaponImageApplier(ImageApplier):
         try:
             # FIXME Cache
             image_file_path = self.path_manager.path(
-                self._images_scheme[mode][weapon_type],
+                self._images_scheme[mode][weapon_type][0],
+            )
+            return Image.open(image_file_path)
+        except KeyError:
+            raise UnknownWeapon(
+                'Unknown weapon "{}" for mode "{}"'.format(weapon_type, mode),
+            )
+
+    def get_animation_image_for_weapon(
+        self,
+        mode: str,
+        weapon_type: str,
+        animation_position: int,
+    ) -> Image.Image:
+        try:
+            # FIXME Cache
+            image_file_path = self.path_manager.path(
+                self._images_scheme[mode][weapon_type][animation_position],
             )
             return Image.open(image_file_path)
         except KeyError: