Laurent@814: #!/usr/bin/env python
Laurent@814: # -*- coding: utf-8 -*-
Laurent@814: 
Laurent@814: #This library is free software; you can redistribute it and/or
Laurent@814: #modify it under the terms of the GNU General Public
Laurent@814: #License as published by the Free Software Foundation; either
Laurent@814: #version 2.1 of the License, or (at your option) any later version.
Laurent@814: #
Laurent@814: #This library is distributed in the hope that it will be useful,
Laurent@814: #but WITHOUT ANY WARRANTY; without even the implied warranty of
Laurent@814: #MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
Laurent@814: #General Public License for more details.
Laurent@814: #
Laurent@814: #You should have received a copy of the GNU General Public
Laurent@814: #License along with this library; if not, write to the Free Software
Laurent@814: #Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
Laurent@814: 
Laurent@814: import wx
Laurent@814: import wx.grid
Laurent@814: 
Laurent@814: class CustomTable(wx.grid.PyGridTableBase):
Laurent@814:     
Laurent@814:     """
Laurent@814:     A custom wx.grid.Grid Table using user supplied data
Laurent@814:     """
Laurent@814:     def __init__(self, parent, data, colnames):
Laurent@814:         # The base class must be initialized *first*
Laurent@814:         wx.grid.PyGridTableBase.__init__(self)
Laurent@814:         self.data = data
Laurent@814:         self.colnames = colnames
Laurent@814:         self.Highlights = {}
Laurent@814:         self.Parent = parent
Laurent@814:         # XXX
Laurent@814:         # we need to store the row length and collength to
Laurent@814:         # see if the table has changed size
Laurent@814:         self._rows = self.GetNumberRows()
Laurent@814:         self._cols = self.GetNumberCols()
Laurent@814:     
Laurent@814:     def GetNumberCols(self):
Laurent@814:         return len(self.colnames)
Laurent@814:         
Laurent@814:     def GetNumberRows(self):
Laurent@814:         return len(self.data)
Laurent@814: 
Laurent@814:     def GetColLabelValue(self, col, translate=True):
Laurent@814:         if col < len(self.colnames):
Laurent@814:             if translate:
Laurent@814:                 return _(self.colnames[col])
Laurent@814:             return self.colnames[col]
Laurent@814: 
Laurent@814:     def GetRowLabelValue(self, row, translate=True):
Laurent@814:         return row
Laurent@814: 
Laurent@814:     def GetValue(self, row, col):
Laurent@814:         if row < self.GetNumberRows():
Laurent@814:             return self.data[row].get(self.GetColLabelValue(col, False), "")
Laurent@814:     
Laurent@814:     def SetValue(self, row, col, value):
Laurent@814:         if col < len(self.colnames):
Laurent@814:             self.data[row][self.GetColLabelValue(col, False)] = value
Laurent@814:     
Laurent@814:     def GetValueByName(self, row, colname):
Laurent@814:         if row < self.GetNumberRows():
Laurent@814:             return self.data[row].get(colname)
Laurent@814: 
Laurent@814:     def SetValueByName(self, row, colname, value):
Laurent@814:         if row < self.GetNumberRows():
Laurent@814:             self.data[row][colname] = value
Laurent@814: 
Laurent@814:     def ResetView(self, grid):
Laurent@814:         """
Laurent@814:         (wx.grid.Grid) -> Reset the grid view.   Call this to
Laurent@814:         update the grid if rows and columns have been added or deleted
Laurent@814:         """
Laurent@814:         grid.CloseEditControl()
Laurent@814:         grid.BeginBatch()
Laurent@814:         for current, new, delmsg, addmsg in [
Laurent@814:             (self._rows, self.GetNumberRows(), wx.grid.GRIDTABLE_NOTIFY_ROWS_DELETED, wx.grid.GRIDTABLE_NOTIFY_ROWS_APPENDED),
Laurent@814:             (self._cols, self.GetNumberCols(), wx.grid.GRIDTABLE_NOTIFY_COLS_DELETED, wx.grid.GRIDTABLE_NOTIFY_COLS_APPENDED),
Laurent@814:         ]:
Laurent@814:             if new < current:
Laurent@814:                 msg = wx.grid.GridTableMessage(self,delmsg,new,current-new)
Laurent@814:                 grid.ProcessTableMessage(msg)
Laurent@814:             elif new > current:
Laurent@814:                 msg = wx.grid.GridTableMessage(self,addmsg,new-current)
Laurent@814:                 grid.ProcessTableMessage(msg)
Laurent@814:                 self.UpdateValues(grid)
Laurent@814:         grid.EndBatch()
Laurent@814: 
Laurent@814:         self._rows = self.GetNumberRows()
Laurent@814:         self._cols = self.GetNumberCols()
Laurent@814:         # update the column rendering scheme
Laurent@814:         self._updateColAttrs(grid)
Laurent@814: 
Laurent@814:         # update the scrollbars and the displayed part of the grid
Laurent@814:         grid.AdjustScrollbars()
Laurent@814:         grid.ForceRefresh()
Laurent@814: 
Laurent@814:     def UpdateValues(self, grid):
Laurent@814:         """Update all displayed values"""
Laurent@814:         # This sends an event to the grid table to update all of the values
Laurent@814:         msg = wx.grid.GridTableMessage(self, wx.grid.GRIDTABLE_REQUEST_VIEW_GET_VALUES)
Laurent@814:         grid.ProcessTableMessage(msg)
Laurent@814: 
Laurent@814:     def _updateColAttrs(self, grid):
Laurent@814:         """
Laurent@814:         wx.grid.Grid -> update the column attributes to add the
Laurent@814:         appropriate renderer given the column name.
Laurent@814: 
Laurent@814:         Otherwise default to the default renderer.
Laurent@814:         """
Laurent@814:         for row in range(self.GetNumberRows()):
Laurent@814:             row_highlights = self.Highlights.get(row, {})
Laurent@814:             for col in range(self.GetNumberCols()):
Laurent@814:                 colname = self.GetColLabelValue(col, False)
Laurent@814:                 
Laurent@814:                 grid.SetReadOnly(row, col, True)
Laurent@814:                 grid.SetCellEditor(row, col, None)
Laurent@814:                 grid.SetCellRenderer(row, col, None)
Laurent@814:                 
Laurent@814:                 highlight_colours = row_highlights.get(colname.lower(), [(wx.WHITE, wx.BLACK)])[-1]
Laurent@814:                 grid.SetCellBackgroundColour(row, col, highlight_colours[0])
Laurent@814:                 grid.SetCellTextColour(row, col, highlight_colours[1])
Laurent@814:             self.ResizeRow(grid, row)
Laurent@814:     
Laurent@814:     def ResizeRow(self, grid, row):
Laurent@814:         if wx.Platform == '__WXMSW__':
Laurent@814:             grid.SetRowMinimalHeight(row, 20)
Laurent@814:         else:
Laurent@814:             grid.SetRowMinimalHeight(row, 28)
Laurent@814:         grid.AutoSizeRow(row, False)
Laurent@814:     
Laurent@814:     def SetData(self, data):
Laurent@814:         self.data = data
Laurent@814:     
Laurent@814:     def GetData(self):
Laurent@814:         return self.data
Laurent@814:     
Laurent@814:     def GetCurrentIndex(self):
Laurent@814:         return self.CurrentIndex
Laurent@814:     
Laurent@814:     def SetCurrentIndex(self, index):
Laurent@814:         self.CurrentIndex = index
Laurent@814:     
Laurent@814:     def AppendRow(self, row_content):
Laurent@814:         self.data.append(row_content)
Laurent@814: 
Laurent@814:     def InsertRow(self, index, row_content):
Laurent@814:         self.data.insert(index, row_content)
Laurent@814: 
Laurent@814:     def MoveRow(self, row_index, move):
Laurent@814:         new_index = max(0, min(row_index + move, len(self.data) - 1))
Laurent@814:         if new_index != row_index:
Laurent@814:             self.data.insert(new_index, self.data.pop(row_index))
Laurent@814:         return new_index
Laurent@814: 
Laurent@814:     def RemoveRow(self, row_index):
Laurent@814:         self.data.pop(row_index)
Laurent@814: 
Laurent@814:     def GetRow(self, row_index):
Laurent@814:         return self.data[row_index]
Laurent@814: 
Laurent@814:     def Empty(self):
Laurent@814:         self.data = []
Laurent@814: 
Laurent@814:     def AddHighlight(self, infos, highlight_type):
Laurent@814:         row_highlights = self.Highlights.setdefault(infos[0], {})
Laurent@814:         col_highlights = row_highlights.setdefault(infos[1], [])
Laurent@814:         col_highlights.append(highlight_type)
Laurent@814: 
Laurent@814:     def RemoveHighlight(self, infos, highlight_type):
Laurent@814:         row_highlights = self.Highlights.get(infos[0])
Laurent@814:         if row_highlights is not None:
Laurent@814:             col_highlights = row_highlights.get(infos[1])
Laurent@814:             if col_highlights is not None and highlight_type in col_highlights:
Laurent@814:                 col_highlights.remove(highlight_type)
Laurent@814:             if len(col_highlights) == 0:
Laurent@814:                 row_highlights.pop(infos[1])
Laurent@814: 
Laurent@814:     def ClearHighlights(self, highlight_type=None):
Laurent@814:         if highlight_type is None:
Laurent@814:             self.Highlights = {}
Laurent@814:         else:
Laurent@814:             for row, row_highlights in self.Highlights.iteritems():
Laurent@814:                 row_items = row_highlights.items()
Laurent@814:                 for col, col_highlights in row_items:
Laurent@814:                     if highlight_type in col_highlights:
Laurent@814:                         col_highlights.remove(highlight_type)
Laurent@814:                     if len(col_highlights) == 0:
Laurent@814:                         row_highlights.pop(col)