X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=tabakrolletjie%2Ftransforms.py;h=8590c1bbd1ee61539921e9d906cfdf96f9fbb890;hb=05a783db47a36c0cc18df7cb625d44a6689aaf56;hp=c5eb05f64e7d85739910eb2f89e1e4f428095070;hpb=c30f1093315596a321a5abe26a9ecfd254d54c2a;p=tabakrolletjie.git

diff --git a/tabakrolletjie/transforms.py b/tabakrolletjie/transforms.py
index c5eb05f..8590c1b 100644
--- a/tabakrolletjie/transforms.py
+++ b/tabakrolletjie/transforms.py
@@ -2,6 +2,10 @@
 
 import pygame.surface
 
+import pygame.locals as pgl
+
+from .constants import COLOURS
+
 
 class Transform(object):
 
@@ -49,3 +53,56 @@ class Overlay(Transform):
         over.fill(self.colour)
         surface.blit(over, (0, 0), None)
         return surface
+
+
+class Multiply(Transform):
+    """ Apply a colour by multiplying. """
+
+    ARGS = ["colour"]
+
+    def apply(self, surface):
+        mult = pygame.surface.Surface(surface.get_size())
+        mult = mult.convert_alpha()
+        mult.fill(self.colour)
+        surface.blit(mult, (0, 0), None, pgl.BLEND_RGBA_MULT)
+        return surface
+
+
+class Alpha(Transform):
+    """ Make translucent. """
+
+    ARGS = ["alpha"]
+
+    def apply(self, surface):
+        surface.fill((255, 255, 255, self.alpha), None, pgl.BLEND_RGBA_MULT)
+        return surface
+
+
+class ColourWedges(Transform):
+    """ Apply colours as wedges. """
+
+    ARGS = ["colours"]
+
+    def apply(self, surface):
+        size = surface.get_width()
+        assert size in (16, 32, 48, 64)
+        fitting_colours = [COLOURS[c] for c in self.colours]
+        ncolour = len(fitting_colours)
+        if ncolour > 3:
+            print "Multicoloured light should not have more than 3 colours"
+            ncolour = 3
+
+        if ncolour == 1:
+            multiply = Multiply(colour=fitting_colours[0])
+            return multiply.apply(surface)
+
+        colour_mult_image = pygame.surface.Surface((size, size))
+        from .loader import loader
+        for i in range(ncolour):
+            sector = loader.load_image(
+                str(size), "light_mask_%d_%d.png" % (ncolour, i + 1),
+                transform=Multiply(colour=fitting_colours[i]))
+            colour_mult_image.blit(sector, (0, 0), None, 0)
+
+        surface.blit(colour_mult_image, (0, 0), None, pgl.BLEND_RGBA_MULT)
+        return surface