Fix or add some DoGetBestSize's

[wxWidgets.git] / wxPython / wx / lib / stattext.py

diff --git a/wxPython/wx/lib/stattext.py b/wxPython/wx/lib/stattext.py
index 220696621736abc7adb9fed476b06accae5ac196..ec38e165e71de310ce4a1b013bb3b3fbdfc375d3 100644 (file)
--- a/wxPython/wx/lib/stattext.py
+++ b/wxPython/wx/lib/stattext.py
@@ -1,8 +1,146 @@
+#----------------------------------------------------------------------
+# Name:        wx.lib.stattext
+# Purpose:     A generic wxGenStaticText class.  Using this should
+#              eliminate some of the platform differences in wxStaticText,
+#              such as background colours and mouse sensitivity.
+#
+# Author:      Robin Dunn
+#
+# Created:     8-July-2002
+# RCS-ID:      $Id$
+# Copyright:   (c) 2002 by Total Control Software
+# Licence:     wxWindows license
+#----------------------------------------------------------------------
+# 12/12/2003 - Jeff Grimmett (grimmtooth@softhome.net)
+#
+# o 2.5 compatability update.
+# o Untested.
+#
+
+import wx
+
+#----------------------------------------------------------------------
+
+class GenStaticText(wx.PyControl):
+    labelDelta = 1
+
+    def __init__(self, parent, ID, label,
+                 pos = wx.DefaultPosition, size = wx.DefaultSize,
+                 style = 0,
+                 name = "genstattext"):
+        wx.PyControl.__init__(self, parent, ID, pos, size, style|wx.NO_BORDER,
+                             wx.DefaultValidator, name)
+
+        wx.PyControl.SetLabel(self, label) # don't check wx.ST_NO_AUTORESIZE yet
+        self.SetPosition(pos)
+        font = parent.GetFont()
+        if not font.Ok():
+            font = wx.SystemSettings.GetFont(wx.SYS_DEFAULT_GUI_FONT)
+        wx.PyControl.SetFont(self, font) # same here
+
+        self.defBackClr = parent.GetBackgroundColour()
+        if not self.defBackClr.Ok():
+            self.defBackClr = wx.SystemSettings.GetColour(wx.SYS_COLOUR_3DFACE)
+        self.SetBackgroundColour(self.defBackClr)
+
+        clr = parent.GetForegroundColour()
+        if not clr.Ok():
+            clr = wx.SystemSettings.GetColour(wx.SYS_COLOUR_BTNTEXT)
+        self.SetForegroundColour(clr)
+
+        rw, rh = size
+        bw, bh = self.GetBestSize()
+        if rw == -1: rw = bw
+        if rh == -1: rh = bh
+        self.SetSize(wx.Size(rw, rh))
+
+        self.Bind(wx.EVT_ERASE_BACKGROUND, self.OnEraseBackground)
+        self.Bind(wx.EVT_PAINT,            self.OnPaint)
+
+
+    def SetLabel(self, label):
+        """
+        Sets the static text label and updates the control's size to exactly
+        fit the label unless the control has wx.ST_NO_AUTORESIZE flag.
+        """
+        wx.PyControl.SetLabel(self, label)
+        style = self.GetWindowStyleFlag()
+        if not style & wx.ST_NO_AUTORESIZE:
+            self.SetSize(self.GetBestSize())
+        self.Refresh()
+
+
+    def SetFont(self, font):
+        """
+        Sets the static text font and updates the control's size to exactly
+        fit the label unless the control has wx.ST_NO_AUTORESIZE flag.
+        """
+        wx.PyControl.SetFont(self, font)
+        style = self.GetWindowStyleFlag()
+        if not style & wx.ST_NO_AUTORESIZE:
+            self.SetSize(self.GetBestSize())
+        self.Refresh()
+
+
+    def DoGetBestSize(self):
+        """Overridden base class virtual.  Determines the best size of the
+        button based on the label size."""
+        label = self.GetLabel()
+        maxWidth = totalHeight = 0
+        for line in label.split('\n'):
+            if line == '':
+                w, h = self.GetTextExtent('W')  # empty lines have height too
+            else:
+                w, h = self.GetTextExtent(line)
+            totalHeight += h
+            maxWidth = max(maxWidth, w)
+        return wx.Size(maxWidth, totalHeight)
+
+
+    def AcceptsFocus(self):
+        """Overridden base class virtual."""
+        return False
+
+
+    def OnPaint(self, event):
+        dc = wx.BufferedPaintDC(self)
+        #dc = wx.PaintDC(self)
+        width, height = self.GetClientSize()
+        if not width or not height:
+            return
+
+        clr = self.GetBackgroundColour()
+        backBrush = wx.Brush(clr, wx.SOLID)
+        if wx.Platform == "__WXMAC__" and clr == self.defBackClr:
+            # if colour is still the default then use the striped background on Mac
+            backBrush.MacSetTheme(1) # 1 == kThemeBrushDialogBackgroundActive
+        dc.SetBackground(backBrush)
+
+        dc.SetTextForeground(self.GetForegroundColour())
+        dc.Clear()
+        dc.SetFont(self.GetFont())
+        label = self.GetLabel()
+        style = self.GetWindowStyleFlag()
+        x = y = 0
+        for line in label.split('\n'):
+            if line == '':
+                w, h = self.GetTextExtent('W')  # empty lines have height too
+            else:
+                w, h = self.GetTextExtent(line)
+            if style & wx.ALIGN_RIGHT:
+                x = width - w
+            if style & wx.ALIGN_CENTER:
+                x = (width - w)/2
+            dc.DrawText(line, (x, y))
+            y += h
+
+
+    def OnEraseBackground(self, event):
+        pass
+
+
+
+
+#----------------------------------------------------------------------
 
-"""Renamer stub: provides a way to drop the wx prefix from wxPython objects."""
 
-from wx import _rename
-from wxPython.lib import stattext
-_rename(globals(), stattext.__dict__, modulename='lib.stattext')
-del stattext
-del _rename