Add transform support to cursor and image buttons

[tabakrolletjie.git] / tabakrolletjie / widgets.py

diff --git a/tabakrolletjie/widgets.py b/tabakrolletjie/widgets.py
index f34fa35fccc0427477107fae9677b4be91d1cea0..8ca16e31510c061a261a5cc39200dd01798f30bf 100644 (file)
--- a/tabakrolletjie/widgets.py
+++ b/tabakrolletjie/widgets.py
@@ -6,6 +6,7 @@ import pygame.locals as pgl
 
 from .loader import loader
 from .constants import FONTS
+from .transforms import NullTransform
 
 
 class Button(object):
@@ -64,7 +65,8 @@ class TextButton(Button):
 class ImageButton(Button):
 
     def __init__(self, *imgparts, **kwargs):
-        self._img = loader.load_image(*imgparts)
+        transform = kwargs.pop("transform", NullTransform())
+        self._img = loader.load_image(*imgparts, transform=transform)
         name = kwargs.get('name')
         pos = kwargs.get('pos')
         padding = kwargs.get('padding', 0)