import os
from wx.lib.buttons import GenBitmapButton

class MultiStateButton(GenBitmapButton):
    """
    A bitmap button that has more than one possible bitmap in its default state.
    These bitmaps are passed in as a list of names, which are also the names of the PNG bitmaps.
    foo = MultiStateButton(view, 100, multibitmaps=["MailState", "NoMailState"])
    will create a button 
    """
    kBitmapFileExtensions = ["png", "gif", "jpg", "tiff", "psd"]
    def __init__(self, parent, ID, label="",
                 pos = wx.DefaultPosition, size = wx.DefaultSize,
                 style = 0, validator = wx.DefaultValidator,
                 name = "multistatebutton", multibitmaps = ()):
        self.bits = {}
        firstBitmap = self.AddStates(multibitmaps)
        # assert firstBitmap is not None
        print "Using firstBitmap of " + firstBitmap
        super(MultiStateButton, self).__init__(parent, ID, firstBitmap, pos, size, style, validator, name)

    def AddStates(self, multibitmaps):
        firstFound = None
        for name in multibitmaps:
            found = False
            for ext in MultiStateButton.kBitmapFileExtensions:
                try:
                    img = wx.Image(('%s.%s' % (os.path.join('/Users/rae/work/osaf/rae-button/', name), ext)))
                    self.bits[name] = img.ConvertToBitmap()
                    if not found:
                        firstFound = name
                        found = True
                    break
                except IOError:
                    pass
            if not found:
                ##Reid
                # should raise an error here somehow
                pass
        return firstFound

    def SetState(self, state):
        bitmap = self.bits.get(state)
        if bitmap != None:
            self.SetBitmapLabel(bitmap)