temp: caching rendered movie clips

Author: danila-schelkov

system/lib/features/cut_sprites.py

@@ -20,7 +20,7 @@ def render_objects(swf: SupercellSWF, output_folder: Path):
     #     rendered_movie_clip = movie_clip.render(swf)
     #     if sum(rendered_movie_clip.size) >= 2:
     #         clip_name = movie_clip.export_name or movie_clip.id
-    #         rendered_movie_clip.save(f"{export_folder}/movie_clips/{clip_name}.png")
+    #         rendered_movie_clip.save(f"{output_folder}/movie_clips/{clip_name}.png")
     #     else:
     #         # For debug:
     #         # logger.warning(f'MovieClip {movie_clip.id} cannot be rendered.')

system/lib/objects/movie_clip.py

@@ -12,6 +12,9 @@
     from system.lib.swf import SupercellSWF
 
 
+CACHE = {}
+
+
 class MovieClipFrame:
     def __init__(self):
         self._elements_count: int = 0
@@ -110,6 +113,9 @@ def load(self, swf: "SupercellSWF", tag: int):
                 swf.reader.read(frame_length)
 
     def render(self, swf: "SupercellSWF", matrix=None) -> Image.Image:
+        if self in CACHE:
+            return CACHE[self].copy()
+
         matrix_bank = swf.get_matrix_bank(self.matrix_bank_index)
 
         # TODO: make it faster
@@ -137,6 +143,8 @@ def render(self, swf: "SupercellSWF", matrix=None) -> Image.Image:
 
                 image.paste(rendered_shape, (x, y), rendered_shape)
 
+        CACHE[self] = image
+
         return image
 
     def get_sides(self, swf: "SupercellSWF") -> Tuple[float, float, float, float]:
@@ -171,3 +179,7 @@ def get_sides(self, swf: "SupercellSWF") -> Tuple[float, float, float, float]:
                     bottom = max(bottom, shape_bottom)
 
         return left, top, right, bottom
+
+    def get_position(self) -> Tuple[float, float]:
+        left, top, _, _ = self.get_sides()
+        return left, top