X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=tabakrolletjie%2Fwidgets.py;h=2678e72c46cce18b390686ee079dda184ee1e56b;hb=c8191a153de7d9354c781b915e6aa3c41c5fc94b;hp=f34fa35fccc0427477107fae9677b4be91d1cea0;hpb=c99b694a2a9e1237a402c0989d9f55e394c56cfe;p=tabakrolletjie.git

diff --git a/tabakrolletjie/widgets.py b/tabakrolletjie/widgets.py
index f34fa35..2678e72 100644
--- 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,12 +65,19 @@ class TextButton(Button):
 class ImageButton(Button):
 
     def __init__(self, *imgparts, **kwargs):
-        self._img = loader.load_image(*imgparts)
+        transform = kwargs.pop("transform", NullTransform())
+        self._img = kwargs.pop("image", None)
+        if self._img is None:
+            self._img = loader.load_image(*imgparts, transform=transform)
         name = kwargs.get('name')
         pos = kwargs.get('pos')
         padding = kwargs.get('padding', 0)
         super(ImageButton, self).__init__(self._img.get_size(), name,
                                           pos, padding)
 
+    def update_image(self, *imgparts, **kwargs):
+        transform = kwargs.pop("transform", NullTransform())
+        self._img = loader.load_image(*imgparts, transform=transform)
+
     def render(self, surface):
         surface.blit(self._img, self._pos, None)