import wx
from wx.lib.stattext import GenStaticText as StaticText

if wx.Platform == "__WXMAC__":
    from Carbon.Appearance import kThemeBrushDialogBackgroundActive


class EllipticStaticText(StaticText):

    def __init__(self, parent, id=wx.ID_ANY, label='', pos=wx.DefaultPosition, size=wx.DefaultSize,
                 style=0, name="ellipticstatictext"):
        """
        Default class constructor.

        :param `parent`: the L{EllipticStaticText} parent. Must not be ``None``;
        :param `id`: window identifier. A value of -1 indicates a default value;
        :param `label`: the text label;
        :param `pos`: the control position. A value of (-1, -1) indicates a default position,
         chosen by either the windowing system or wxPython, depending on platform;
        :param `size`: the control size. A value of (-1, -1) indicates a default size,
         chosen by either the windowing system or wxPython, depending on platform;
        :param `style`: the static text style;
        :param `name`: the window name.
        """

        StaticText.__init__(self, parent, id, label, pos, size, style, name)

        self.Bind(wx.EVT_SIZE, self.OnSize)
        self.Bind(wx.EVT_PAINT, self.OnPaint)
        self.Bind(wx.EVT_ERASE_BACKGROUND, self.OnEraseBackground)


    def OnSize(self, event):
        """
        Handles the ``wx.EVT_SIZE`` event for L{EllipticStaticText}.

        :param `event`: a `wx.SizeEvent` event to be processed.
        """

        event.Skip()
        self.Refresh()


    def OnEraseBackground(self, event):
        """
        Handles the ``wx.EVT_ERASE_BACKGROUND`` event for L{EllipticStaticText}.

        :param `event`: a `wx.EraseEvent` event to be processed.

        :note: This is intentionally empty to reduce flicker.
        """

        pass


    def OnPaint(self, event):
        """
        Handles the ``wx.EVT_PAINT`` event for L{EllipticStaticText}.

        :param `event`: a `wx.PaintEvent` to be processed.
        """

        dc = wx.BufferedPaintDC(self)        
        width, height = self.GetClientSize()

        if not width or not height:
            return

        clr = self.GetBackgroundColour()

        if wx.Platform == "__WXMAC__":
            # if colour is still the default then use the theme's  background on Mac
            themeColour = wx.MacThemeColour(kThemeBrushDialogBackgroundActive)
            backBrush = wx.Brush(themeColour)
        else:
            backBrush = wx.Brush(clr, wx.SOLID)
            
        dc.SetBackground(backBrush)
        dc.Clear()

        if self.IsEnabled():
            dc.SetTextForeground(self.GetForegroundColour())
        else:
            dc.SetTextForeground(wx.SystemSettings.GetColour(wx.SYS_COLOUR_GRAYTEXT))
            
        dc.SetFont(self.GetFont())

        label = self.GetLabel()
        text = self.ChopText(dc, label, width)
        
        dc.DrawText(text, 0, 0)


    def ChopText(self, dc, text, max_size):
        """
        Chops the input `text` if its size does not fit in `max_size`, by cutting the
        text and adding ellipsis at the end.

        :param `dc`: a `wx.DC` device context;
        :param `text`: the text to chop;
        :param `max_size`: the maximum size in which the text should fit.
        """
        
        # first check if the text fits with no problems
        x, y = dc.GetTextExtent(text)
        
        if x <= max_size:
            return text

        textLen = len(text)
        last_good_length = 0
        
        for i in xrange(textLen, -1, -1):
            s = text[0:i]
            s += "..."

            x, y = dc.GetTextExtent(s)
            last_good_length = i
            
            if x < max_size:
                break

        ret = text[0:last_good_length] + "..."    
        return ret


def Example():

    app = wx.PySimpleApp()
    frame = wx.Frame(None, -1, "EllipticStaticText example ;-)", size=(400, 300))

    panel = wx.Panel(frame, -1)
    sizer = wx.BoxSizer(wx.VERTICAL)

    elliptic = EllipticStaticText(panel, -1, r"F:\myreservoir\re\util\python\hm_evaluation\data\HM_Evaluation_0.9.9.7.exe")
    whitePanel = wx.Panel(panel, -1)
    whitePanel.SetBackgroundColour(wx.WHITE)

    sizer.Add(elliptic, 0, wx.ALL|wx.EXPAND, 10)
    sizer.Add(whitePanel, 1, wx.ALL|wx.EXPAND, 10)

    panel.SetSizer(sizer)
    sizer.Layout()

    frame.CenterOnScreen()
    frame.Show()

    app.MainLoop()


if __name__ == "__main__":

    Example()