interiors: finish base code and tests

opencc/simulation/interior.py

@@ -1,6 +1,7 @@
 # coding: utf-8
 import typing
 
+from PIL.PngImagePlugin import PngImageFile
 from synergine2_xyz.map import TMXMap
 from synergine2_xyz.utils import get_direct_around_positions_of_position
 
@@ -87,3 +88,21 @@ class InteriorManager(object):
                 if where_position in interior and interior not in interiors:
                     interiors.append(interior)
         return interiors
+
+    def update_image_for_interiors(
+        self,
+        image: PngImageFile,
+        interiors: typing.List[typing.List[typing.Tuple[int, int]]],
+        tile_width: int,
+        tile_height: int,
+    ) -> None:
+        # TODO BS 20171213: Optimization can be done: keep in cache modifications on image instead change it entirely
+        pixels = image.load()
+
+        for interior in interiors:
+            for tile_x, tile_y in interior:
+                start_x = tile_x * tile_width
+                start_y = tile_y * tile_height
+                for x in range(start_x, start_x+tile_width):
+                    for y in range(start_y, start_y+tile_height):
+                        pixels[x, y] = (0, 0, 0, 0)

requirements.txt

@@ -9,6 +9,7 @@ py==1.5.2
 pyglet==1.3.0
 pytest==3.3.0
 python-dateutil==2.6.1
+Pillow==4.3.0
 PyYAML==3.12
 redis==2.10.6
 six==1.11.0

tests/gui/test_interiors.py

@@ -1,5 +1,6 @@
 # coding: utf-8
 from synergine2_xyz.map import TMXMap
+from PIL import Image
 
 from opencc.simulation.interior import InteriorManager
 
@@ -112,6 +113,7 @@ def test_interiors_zones__active_zones():
     interiors = manager.get_interiors(where_positions=[(0, 1), (4, 2)])
 
     assert 2 == len(interiors)
+    interiors = sorted(interiors)
 
     assert (0, 1) in interiors[0]
     assert (1, 1) in interiors[0]
@@ -128,3 +130,14 @@ def test_interiors_zones__active_zones():
     assert (3, 3) in interiors[1]
     assert (4, 3) in interiors[1]
     assert 6 == len(interiors[1])
+
+
+def test_interiors_zones__make_image_transparent__just_replace():
+    map_ = TMXMap('tests/fixtures/one_interior.tmx')
+    manager = InteriorManager(map_)
+    interiors = manager.get_interiors()
+    image = Image.open('tests/fixtures/white_40x40.png')
+    after_image_bytes = Image.open('tests/fixtures/white_one_interior_40x40.png').tobytes()
+
+    manager.update_image_for_interiors(image, interiors, 8, 8)
+    assert after_image_bytes == image.tobytes()