from wxPython.wx import *
from wxPython.lib.mixins.listctrl import wxColumnSorterMixin, wxListCtrlAutoWidthMixin

import images

class testListCtrl(wxListCtrl, wxListCtrlAutoWidthMixin):
    def __init__(self, parent, ID, pos=wxDefaultPosition,
                 size=wxDefaultSize, style=0):
        wxListCtrl.__init__(self, parent, ID, pos, size, style)
        wxListCtrlAutoWidthMixin.__init__(self)


class testListCtrlPanel(wxPanel, wxColumnSorterMixin):
    def __init__(self, parent, ds, grandparent):
        wxPanel.__init__(self, parent, -1, style=wxWANTS_CHARS)
        self.ds = ds
        self.grandparent = grandparent

        tID = wxNewId()

        self.il = wxImageList(16, 16)

        self.idx1 = self.il.Add(images.getSmilesBitmap())
        self.sm_up = self.il.Add(images.getSmallUpArrowBitmap())
        self.sm_dn = self.il.Add(images.getSmallDnArrowBitmap())

        self.list = testListCtrl(self, tID,
                                 style=wxLC_REPORT | wxSUNKEN_BORDER
                                 | wxLC_VRULES | wxLC_HRULES
                                 )
        self.list.SetImageList(self.il, wxIMAGE_LIST_SMALL)
        
        self.populateList()

        # Now that the list exists we can init the other base class,
        # see wxPython/lib/mixins/listctrl.py
        self.itemDataMap = self.musicdata
        wxColumnSorterMixin.__init__(self, 3)
        self.SortListItems(2, True) # sort by genre

        EVT_SIZE(self, self.OnSize)
        EVT_LIST_ITEM_SELECTED(self, tID, self.OnItemSelected)
        EVT_LIST_ITEM_DESELECTED(self, tID, self.OnItemDeselected)
        EVT_LIST_ITEM_ACTIVATED(self, tID, self.OnItemActivated)
        EVT_LIST_DELETE_ITEM(self, tID, self.OnItemDelete)
        EVT_LIST_COL_CLICK(self, tID, self.OnColClick)
        EVT_LIST_COL_RIGHT_CLICK(self, tID, self.OnColRightClick)
        EVT_LIST_COL_BEGIN_DRAG(self, tID, self.OnColBeginDrag)
        EVT_LIST_COL_DRAGGING(self, tID, self.OnColDragging)
        EVT_LIST_COL_END_DRAG(self, tID, self.OnColEndDrag)

        EVT_LEFT_DCLICK(self.list, self.OnDoubleClick)
        EVT_RIGHT_DOWN(self.list, self.OnRightDown)

        # for wxMSW
        EVT_COMMAND_RIGHT_CLICK(self.list, tID, self.OnRightClick)

        # for wxGTK
        EVT_RIGHT_UP(self.list, self.OnRightClick)

    def populateList(self):
        self.loadData()
        # since we want images on the column header we have to do it the hard way:
        # see the wxPython demo for a simpler way if no images are required,
        # i.e. no columnsortmixing
        info = wxListItem()
        info.m_mask = wxLIST_MASK_TEXT | wxLIST_MASK_IMAGE | wxLIST_MASK_FORMAT
        info.m_image = -1
        info.m_format = 0
        info.m_text = "Artist"
        self.list.InsertColumnInfo(0, info)

        info.m_format = wxLIST_FORMAT_RIGHT
        info.m_text = "Title"
        self.list.InsertColumnInfo(1, info)

        info.m_format = 0
        info.m_text = "Genre"
        self.list.InsertColumnInfo(2, info)

        items = self.musicdata.items()
        for x in range(len(items)):
            key, data = items[x]
            self.list.InsertImageStringItem(x, data[0], self.idx1)
            self.list.SetStringItem(x, 1, data[1])
            self.list.SetStringItem(x, 2, data[2])
            self.list.SetItemData(x, key)

        self.list.SetColumnWidth(0, wxLIST_AUTOSIZE)
        self.list.SetColumnWidth(1, wxLIST_AUTOSIZE)
        self.list.SetColumnWidth(2, 100)

        # show how to select an item
        self.list.SetItemState(5, wxLIST_STATE_SELECTED, wxLIST_STATE_SELECTED)

        # show how to change the colour of a couple items
        item = self.list.GetItem(1)
        item.SetTextColour(wxBLUE)
        self.list.SetItem(item)
        item = self.list.GetItem(4)
        item.SetTextColour(wxRED)
        self.list.SetItem(item)

        self.currentItem = 0

    # Used by the wxColumnSorterMixin, see wxPython/lib/mixins/listctrl.py
    def GetListCtrl(self):
        return self.list

    # Used by the wxColumnSorterMixin, see wxPython/lib/mixins/listctrl.py
    def GetSortImages(self):
        return (self.sm_dn, self.sm_up)

    def OnRightDown(self, event):
        self.x = event.GetX()
        self.y = event.GetY()
        item, flags = self.list.HitTest((self.x, self.y))
        if flags & wxLIST_HITTEST_ONITEM:
            self.list.Select(item)
        event.Skip()

    def getColumnText(self, index, col):
        item = self.list.GetItem(index, col)
        return item.GetText()

    def OnItemSelected(self, event):
        self.currentItem = event.m_itemIndex
        if self.currentItem == 10:
            #event.Veto()  # doesn't work
            # this does
            self.list.SetItemState(10, 0, wxLIST_STATE_SELECTED)
        event.Skip()

    def OnItemDeselected(self, event):
        item = event.GetItem()
        # Show how to reselect something we don't want deselected
        if event.m_itemIndex == 11:
            wxCallAfter(self.list.SetItemState, 11, wxLIST_STATE_SELECTED, wxLIST_STATE_SELECTED)
        event.Skip()

    def OnItemActivated(self, event):
        self.currentItem = event.m_itemIndex
        event.Skip()

    def OnItemDelete(self, event):
        event.Skip()

    def OnColClick(self, event):
        event.Skip()

    def OnColRightClick(self, event):
        item = self.list.GetColumn(event.GetColumn())
        event.Skip()

    def OnColBeginDrag(self, event):
        ## Show how to not allow a column to be resized
        #if event.GetColumn() == 0:
        #    event.Veto()
        event.Skip()

    def OnColDragging(self, event):
        event.Skip()

    def OnColEndDrag(self, event):
        event.Skip()

    def OnDoubleClick(self, event):
        event.Skip()

    def OnRightClick(self, event):
        event.Skip()

        # only do this part the first time
        if not hasattr(self, "popupID1"):
            self.popupID1 = wxNewId()
            self.popupID2 = wxNewId()
            self.popupID3 = wxNewId()
            self.popupID4 = wxNewId()
            self.popupID5 = wxNewId()
            EVT_MENU(self, self.popupID1, self.OnPopupOne)
            EVT_MENU(self, self.popupID2, self.OnPopupTwo)
            EVT_MENU(self, self.popupID3, self.OnPopupThree)
            EVT_MENU(self, self.popupID4, self.OnPopupFour)
            EVT_MENU(self, self.popupID5, self.OnPopupFive)

        # make a menu
        menu = wxMenu()
        # Show how to put an icon in the menu
        item = wxMenuItem(menu, self.popupID1,"One")
        item.SetBitmap(images.getSmilesBitmap())
        menu.AppendItem(item)
        # add some other items
        menu.Append(self.popupID2, "Two")
        menu.Append(self.popupID3, "ClearAll and repopulate")
        menu.Append(self.popupID4, "DeleteAllItems")
        menu.Append(self.popupID5, "GetItem - Show info in Notebook - Page 1")

        # Popup the menu.  If an item is selected then its handler
        # will be called before PopupMenu returns.
        self.PopupMenu(menu, wxPoint(self.x, self.y))
        menu.Destroy()

    def OnPopupOne(self, event):
        print "FindItem:", self.list.FindItem(-1, "Roxette")
        print "FindItemData:", self.list.FindItemData(-1, 11)
        event.Skip()

    def OnPopupTwo(self, event):
        event.Skip()

    def OnPopupThree(self, event):
        self.list.ClearAll()
        wxCallAfter(self.populateList)
        event.Skip()

    def OnPopupFour(self, event):
        self.list.DeleteAllItems()
        event.Skip()

    def OnPopupFive(self, event):
        item = self.list.GetItem(self.currentItem)
        print 'm_text: %s, m_itemId: %s, GetItemData: %s' % (
                        item.m_text,
                        item.m_itemId, 
                        self.list.GetItemData(self.currentItem))
        # update page 1 in notebook
        self.grandparent.panelm1.updateControls(self.ds, (
                                    item.m_text,
                                    item.m_itemId,
                                    self.list.GetItemData(self.currentItem)))
        event.Skip()
        
    def OnSize(self, event):
        # this is essential to ensure that listctrl sizes correctly
        w,h = self.GetClientSizeTuple()
        self.list.SetDimensions(0, 0, w, h)
        event.Skip()

    def loadData(self):
        # create sample for getting it from e.g. kinterbasdb ???
        self.musicdata = {

            1 : ("Bad English", "The Price Of Love", "Rock"),

            2 : ("DNA featuring Suzanne Vega", "Tom's Diner", "Rock"),

            3 : ("George Michael", "Praying For Time", "Rock"),

            4 : ("Gloria Estefan", "Here We Are", "Rock"),

            5 : ("Linda Ronstadt", "Don't Know Much", "Rock"),

            6 : ("Michael Bolton", "How Am I Supposed To Live Without You", "Blues"),

            7 : ("Paul Young", "Oh Girl", "Rock"),

            8 : ("Paula Abdul", "Opposites Attract", "Rock"),

            9 : ("Richard Marx", "Should've Known Better", "Rock"),

            10: ("Rod Stewart", "Forever Young", "Rock"),

            11: ("Roxette", "Dangerous", "Rock"),

            12: ("Sheena Easton", "The Lover In Me", "Rock"),

            13: ("Sinead O'Connor", "Nothing Compares 2 U", "Rock"),

            14: ("Stevie B.", "Because I Love You", "Rock"),

            15: ("Taylor Dayne", "Love Will Lead You Back", "Rock"),

            16: ("The Bangles", "Eternal Flame", "Rock"),

            17: ("Wilson Phillips", "Release Me", "Rock"),

            18: ("Billy Joel", "Blonde Over Blue", "Rock"),

            19: ("Billy Joel", "Famous Last Words", "Rock"),

            20: ("Billy Joel", "Lullabye (Goodnight, My Angel)", "Rock"),

            21: ("Billy Joel", "The River Of Dreams", "Rock"),

            22: ("Billy Joel", "Two Thousand Years", "Rock"),

            23: ("Janet Jackson", "Alright", "Rock"),

            24: ("Janet Jackson", "Black Cat", "Rock"),

            25: ("Janet Jackson", "Come Back To Me", "Rock"),

            26: ("Janet Jackson", "Escapade", "Rock"),

            27: ("Janet Jackson", "Love Will Never Do (Without You)", "Rock"),

            28: ("Janet Jackson", "Miss You Much", "Rock"),

            29: ("Janet Jackson", "Rhythm Nation", "Rock"),

            30: ("Janet Jackson", "State Of The World", "Rock"),

            31: ("Janet Jackson", "The Knowledge", "Rock"),

            32: ("Spyro Gyra", "End of Romanticism", "Jazz"),

            33: ("Spyro Gyra", "Heliopolis", "Jazz"),

            34: ("Spyro Gyra", "Jubilee", "Jazz"),

            35: ("Spyro Gyra", "Little Linda", "Jazz"),

            36: ("Spyro Gyra", "Morning Dance", "Jazz"),

            37: ("Spyro Gyra", "Song for Lorraine", "Jazz"),

            38: ("Yes", "Owner Of A Lonely Heart", "Rock"),

            39: ("Yes", "Rhythm Of Love", "Rock"),

            40: ("Cusco", "Dream Catcher", "New Age"),

            41: ("Cusco", "Geronimos Laughter", "New Age"),

            42: ("Cusco", "Ghost Dance", "New Age"),

            43: ("Blue Man Group", "Drumbone", "New Age"),

            44: ("Blue Man Group", "Endless Column", "New Age"),

            45: ("Blue Man Group", "Klein Mandelbrot", "New Age"),

            46: ("Kenny G", "Silhouette", "Jazz"),

            47: ("Sade", "Smooth Operator", "Jazz"),

            48: ("David Arkenstone", "Papillon (On The Wings Of The Butterfly)", "New Age"),

            49: ("David Arkenstone", "Stepping Stars", "New Age"),

            50: ("David Arkenstone", "Carnation Lily Lily Rose", "New Age"),

            51: ("David Lanz", "Behind The Waterfall", "New Age"),

            52: ("David Lanz", "Cristofori's Dream", "New Age"),

            53: ("David Lanz", "Heartsounds", "New Age"),

            54: ("David Lanz", "Leaves on the Seine", "New Age"),

            }

    

