add publics functions on Label and Button class

2 files changed, 29 insertions, 7 deletions

diff --git a/src/api/Button.py b/src/api/Button.py
index b6c0efc..04b4515 100755
--- a/src/api/Button.py
+++ b/src/api/Button.py
@@ -19,9 +19,10 @@ class CButton(CLabel):
         self.active = False

         self.clicked = False

         self.bgColor = (0xCC, 0xCC, 0xCC)

+        self.set_text('Button')

         

     def setCenter(self, aCenter):

-        self.center = aCenter

+        self.set_center(aCenter)

     

     def update(self):

         CLabel.update(self)

diff --git a/src/api/Label.py b/src/api/Label.py
index c762868..bd7577d 100755
--- a/src/api/Label.py
+++ b/src/api/Label.py
@@ -14,19 +14,40 @@ class CLabel(pygame.sprite.Sprite):
     def __init__(self, fontName = "freesansbold.ttf"):

         pygame.sprite.Sprite.__init__(self)

         self.font = pygame.font.Font(fontName, 20)

-        self.text = "Button"

+        self.text = ""

         self.fgColor = ((0x00, 0x00, 0x00))

         self.bgColor = ((0xFF, 0xFF, 0xFF))

-        self.center = (400, 100)

-        self.size = (150, 30)

-

-    def update(self):

+        self.center = (0, 0)

+        self.size = (0, 0)

+        

+    def set_center(self, aCenter):

+        self.center = aCenter

+        

+    def set_size(self, aSize):

+        self.size = aSize

+        self._update_image()

+        

+    def set_fgColor(self, afgColor):

+        self.fgColor = afgColor

+        self._update_image()

+        

+    def set_bgColor(self, abgColor):

+        self.bgColor = abgColor

+        self._update_image()

+        

+    def set_text(self, aText):

+        self.text = aText

+        self._update_image()

+        

+    def _update_image(self):

         self.image = pygame.Surface(self.size)

         self.image.fill(self.bgColor)

         fontSurface = self.font.render(self.text, True, self.fgColor, self.bgColor)

         #center the text

         xPos = (self.image.get_width() - fontSurface.get_width())/2

-        

         self.image.blit(fontSurface, (xPos, 0))

+

+    def update(self):

+        

         self.rect = self.image.get_rect()

         self.rect.center = self.center
\ No newline at end of file