1fded56b37
git-svn-id: https://svn.wxwidgets.org/svn/wx/wxWidgets/trunk@21593 c3d73ce0-8a6f-49c7-b76d-6d57e0e08775
452 lines
15 KiB
Python
452 lines
15 KiB
Python
from wxPython.wx import *
|
|
from wxPython.grid import *
|
|
import images
|
|
|
|
class MegaTable(wxPyGridTableBase):
|
|
"""
|
|
A custom wxGrid Table using user supplied data
|
|
"""
|
|
def __init__(self, data, colnames, plugins):
|
|
"""data is a list of the form
|
|
[(rowname, dictionary),
|
|
dictionary.get(colname, None) returns the data for column
|
|
colname
|
|
"""
|
|
# The base class must be initialized *first*
|
|
wxPyGridTableBase.__init__(self)
|
|
self.data = data
|
|
self.colnames = colnames
|
|
self.plugins = plugins or {}
|
|
# XXX
|
|
# we need to store the row length and collength to
|
|
# see if the table has changed size
|
|
self._rows = self.GetNumberRows()
|
|
self._cols = self.GetNumberCols()
|
|
|
|
def GetNumberCols(self):
|
|
return len(self.colnames)
|
|
|
|
def GetNumberRows(self):
|
|
return len(self.data)
|
|
|
|
def GetColLabelValue(self, col):
|
|
return self.colnames[col]
|
|
|
|
def GetRowLabelValues(self, row):
|
|
return self.data[row][0]
|
|
|
|
def GetValue(self, row, col):
|
|
return str(self.data[row][1].get(self.GetColLabelValue(col), ""))
|
|
|
|
def GetRawValue(self, row, col):
|
|
return self.data[row][1].get(self.GetColLabelValue(col), "")
|
|
|
|
def SetValue(self, row, col, value):
|
|
self.data[row][1][self.GetColLabelValue(col)] = value
|
|
|
|
def ResetView(self, grid):
|
|
"""
|
|
(wxGrid) -> Reset the grid view. Call this to
|
|
update the grid if rows and columns have been added or deleted
|
|
"""
|
|
grid.BeginBatch()
|
|
for current, new, delmsg, addmsg in [
|
|
(self._rows, self.GetNumberRows(), wxGRIDTABLE_NOTIFY_ROWS_DELETED, wxGRIDTABLE_NOTIFY_ROWS_APPENDED),
|
|
(self._cols, self.GetNumberCols(), wxGRIDTABLE_NOTIFY_COLS_DELETED, wxGRIDTABLE_NOTIFY_COLS_APPENDED),
|
|
]:
|
|
if new < current:
|
|
msg = wxGridTableMessage(self,delmsg,new,current-new)
|
|
grid.ProcessTableMessage(msg)
|
|
elif new > current:
|
|
msg = wxGridTableMessage(self,addmsg,new-current)
|
|
grid.ProcessTableMessage(msg)
|
|
self.UpdateValues(grid)
|
|
grid.EndBatch()
|
|
|
|
self._rows = self.GetNumberRows()
|
|
self._cols = self.GetNumberCols()
|
|
# update the column rendering plugins
|
|
self._updateColAttrs(grid)
|
|
|
|
# update the scrollbars and the displayed part of the grid
|
|
grid.AdjustScrollbars()
|
|
grid.ForceRefresh()
|
|
|
|
|
|
def UpdateValues(self, grid):
|
|
"""Update all displayed values"""
|
|
# This sends an event to the grid table to update all of the values
|
|
msg = wxGridTableMessage(self, wxGRIDTABLE_REQUEST_VIEW_GET_VALUES)
|
|
grid.ProcessTableMessage(msg)
|
|
|
|
def _updateColAttrs(self, grid):
|
|
"""
|
|
wxGrid -> update the column attributes to add the
|
|
appropriate renderer given the column name. (renderers
|
|
are stored in the self.plugins dictionary)
|
|
|
|
Otherwise default to the default renderer.
|
|
"""
|
|
col = 0
|
|
for colname in self.colnames:
|
|
attr = wxGridCellAttr()
|
|
if colname in self.plugins:
|
|
renderer = self.plugins[colname](self)
|
|
if renderer.colSize:
|
|
grid.SetColSize(col, renderer.colSize)
|
|
if renderer.rowSize:
|
|
grid.SetDefaultRowSize(renderer.rowSize)
|
|
attr.SetReadOnly(true)
|
|
attr.SetRenderer(renderer)
|
|
grid.SetColAttr(col, attr)
|
|
col += 1
|
|
|
|
# ------------------------------------------------------
|
|
# begin the added code to manipulate the table (non wx related)
|
|
def AppendRow(self, row):
|
|
entry = {}
|
|
for name in self.colnames:
|
|
entry[name] = "Appended_%i"%row
|
|
# XXX Hack
|
|
# entry["A"] can only be between 1..4
|
|
entry["A"] = random.choice(range(4))
|
|
self.data.insert(row, ["Append_%i"%row, entry])
|
|
|
|
def DeleteCols(self, cols):
|
|
"""
|
|
cols -> delete the columns from the dataset
|
|
cols hold the column indices
|
|
"""
|
|
# we'll cheat here and just remove the name from the
|
|
# list of column names. The data will remain but
|
|
# it won't be shown
|
|
deleteCount = 0
|
|
cols = cols[:]
|
|
cols.sort()
|
|
for i in cols:
|
|
self.colnames.pop(i-deleteCount)
|
|
# we need to advance the delete count
|
|
# to make sure we delete the right columns
|
|
deleteCount += 1
|
|
if not len(self.colnames):
|
|
self.data = []
|
|
|
|
def DeleteRows(self, rows):
|
|
"""
|
|
rows -> delete the rows from the dataset
|
|
rows hold the row indices
|
|
"""
|
|
deleteCount = 0
|
|
rows = rows[:]
|
|
rows.sort()
|
|
for i in rows:
|
|
self.data.pop(i-deleteCount)
|
|
# we need to advance the delete count
|
|
# to make sure we delete the right rows
|
|
deleteCount += 1
|
|
|
|
def SortColumn(self, col):
|
|
"""
|
|
col -> sort the data based on the column indexed by col
|
|
"""
|
|
name = self.colnames[col]
|
|
_data = []
|
|
for row in self.data:
|
|
rowname, entry = row
|
|
_data.append((entry.get(name, None), row))
|
|
|
|
_data.sort()
|
|
self.data = []
|
|
for sortvalue, row in _data:
|
|
self.data.append(row)
|
|
|
|
# end table manipulation code
|
|
# ----------------------------------------------------------
|
|
|
|
|
|
# --------------------------------------------------------------------
|
|
# Sample wxGrid renderers
|
|
|
|
class MegaImageRenderer(wxPyGridCellRenderer):
|
|
def __init__(self, table):
|
|
"""
|
|
Image Renderer Test. This just places an image in a cell
|
|
based on the row index. There are N choices and the
|
|
choice is made by choice[row%N]
|
|
"""
|
|
wxPyGridCellRenderer.__init__(self)
|
|
self.table = table
|
|
self._choices = [images.getSmilesBitmap,
|
|
images.getMondrianBitmap,
|
|
images.get_10s_Bitmap,
|
|
images.get_01c_Bitmap]
|
|
|
|
|
|
self.colSize = None
|
|
self.rowSize = None
|
|
|
|
def Draw(self, grid, attr, dc, rect, row, col, isSelected):
|
|
choice = self.table.GetRawValue(row, col)
|
|
bmp = self._choices[ choice % len(self._choices)]()
|
|
image = wxMemoryDC()
|
|
image.SelectObject(bmp)
|
|
|
|
# clear the background
|
|
dc.SetBackgroundMode(wxSOLID)
|
|
if isSelected:
|
|
dc.SetBrush(wxBrush(wxBLUE, wxSOLID))
|
|
dc.SetPen(wxPen(wxBLUE, 1, wxSOLID))
|
|
else:
|
|
dc.SetBrush(wxBrush(wxWHITE, wxSOLID))
|
|
dc.SetPen(wxPen(wxWHITE, 1, wxSOLID))
|
|
dc.DrawRectangle(rect.x, rect.y, rect.width, rect.height)
|
|
|
|
# copy the image but only to the size of the grid cell
|
|
width, height = bmp.GetWidth(), bmp.GetHeight()
|
|
if width > rect.width-2:
|
|
width = rect.width-2
|
|
|
|
if height > rect.height-2:
|
|
height = rect.height-2
|
|
|
|
dc.Blit(rect.x+1, rect.y+1, width, height,
|
|
image,
|
|
0, 0, wxCOPY, True)
|
|
|
|
|
|
class MegaFontRenderer(wxPyGridCellRenderer):
|
|
def __init__(self, table, color="blue", font="ARIAL", fontsize=8):
|
|
"""Render data in the specified color and font and fontsize"""
|
|
wxPyGridCellRenderer.__init__(self)
|
|
self.table = table
|
|
self.color = color
|
|
self.font = wxFont(fontsize, wxDEFAULT, wxNORMAL, wxNORMAL,
|
|
0, font)
|
|
self.selectedBrush = wxBrush("blue",
|
|
wxSOLID)
|
|
self.normalBrush = wxBrush(wxWHITE, wxSOLID)
|
|
self.colSize = None
|
|
self.rowSize = 50
|
|
|
|
def Draw(self, grid, attr, dc, rect, row, col, isSelected):
|
|
# Here we draw text in a grid cell using various fonts
|
|
# and colors. We have to set the clipping region on
|
|
# the grid's DC, otherwise the text will spill over
|
|
# to the next cell
|
|
dc.SetClippingRect(rect)
|
|
|
|
# clear the background
|
|
dc.SetBackgroundMode(wxSOLID)
|
|
if isSelected:
|
|
dc.SetBrush(wxBrush(wxBLUE, wxSOLID))
|
|
dc.SetPen(wxPen(wxBLUE, 1, wxSOLID))
|
|
else:
|
|
dc.SetBrush(wxBrush(wxWHITE, wxSOLID))
|
|
dc.SetPen(wxPen(wxWHITE, 1, wxSOLID))
|
|
dc.DrawRectangle(rect.x, rect.y, rect.width, rect.height)
|
|
|
|
text = self.table.GetValue(row, col)
|
|
dc.SetBackgroundMode(wxSOLID)
|
|
|
|
# change the text background based on whether the grid is selected
|
|
# or not
|
|
if isSelected:
|
|
dc.SetBrush(self.selectedBrush)
|
|
dc.SetTextBackground("blue")
|
|
else:
|
|
dc.SetBrush(self.normalBrush)
|
|
dc.SetTextBackground("white")
|
|
|
|
dc.SetTextForeground(self.color)
|
|
dc.SetFont(self.font)
|
|
dc.DrawText(text, rect.x+1, rect.y+1)
|
|
|
|
# Okay, now for the advanced class :)
|
|
# Let's add three dots "..."
|
|
# to indicate that that there is more text to be read
|
|
# when the text is larger than the grid cell
|
|
|
|
width, height = dc.GetTextExtent(text)
|
|
if width > rect.width-2:
|
|
width, height = dc.GetTextExtent("...")
|
|
x = rect.x+1 + rect.width-2 - width
|
|
dc.DrawRectangle(x, rect.y+1, width+1, height)
|
|
dc.DrawText("...", x, rect.y+1)
|
|
|
|
dc.DestroyClippingRegion()
|
|
|
|
|
|
# --------------------------------------------------------------------
|
|
# Sample Grid using a specialized table and renderers that can
|
|
# be plugged in based on column names
|
|
|
|
class MegaGrid(wxGrid):
|
|
def __init__(self, parent, data, colnames, plugins=None):
|
|
"""parent, data, colnames, plugins=None
|
|
Initialize a grid using the data defined in data and colnames
|
|
(see MegaTable for a description of the data format)
|
|
plugins is a dictionary of columnName -> column renderers.
|
|
"""
|
|
|
|
# The base class must be initialized *first*
|
|
wxGrid.__init__(self, parent, -1)
|
|
self._table = MegaTable(data, colnames, plugins)
|
|
self.SetTable(self._table)
|
|
self._plugins = plugins
|
|
|
|
EVT_GRID_LABEL_RIGHT_CLICK(self, self.OnLabelRightClicked)
|
|
|
|
def Reset(self):
|
|
"""reset the view based on the data in the table. Call
|
|
this when rows are added or destroyed"""
|
|
self._table.ResetView(self)
|
|
|
|
def OnLabelRightClicked(self, evt):
|
|
# Did we click on a row or a column?
|
|
row, col = evt.GetRow(), evt.GetCol()
|
|
if row == -1: self.colPopup(col, evt)
|
|
elif col == -1: self.rowPopup(row, evt)
|
|
|
|
def rowPopup(self, row, evt):
|
|
"""(row, evt) -> display a popup menu when a row label is right clicked"""
|
|
appendID = wxNewId()
|
|
deleteID = wxNewId()
|
|
x = self.GetRowSize(row)/2
|
|
if not self.GetSelectedRows():
|
|
self.SelectRow(row)
|
|
menu = wxMenu()
|
|
xo, yo = evt.GetPosition()
|
|
menu.Append(appendID, "Append Row")
|
|
menu.Append(deleteID, "Delete Row(s)")
|
|
|
|
def append(event, self=self, row=row):
|
|
self._table.AppendRow(row)
|
|
self.Reset()
|
|
|
|
def delete(event, self=self, row=row):
|
|
rows = self.GetSelectedRows()
|
|
self._table.DeleteRows(rows)
|
|
self.Reset()
|
|
|
|
EVT_MENU(self, appendID, append)
|
|
EVT_MENU(self, deleteID, delete)
|
|
self.PopupMenu(menu, wxPoint(x, yo))
|
|
menu.Destroy()
|
|
|
|
def colPopup(self, col, evt):
|
|
"""(col, evt) -> display a popup menu when a column label is
|
|
right clicked"""
|
|
x = self.GetColSize(col)/2
|
|
menu = wxMenu()
|
|
id1 = wxNewId()
|
|
sortID = wxNewId()
|
|
|
|
xo, yo = evt.GetPosition()
|
|
self.SelectCol(col)
|
|
cols = self.GetSelectedCols()
|
|
self.Refresh()
|
|
menu.Append(id1, "Delete Col(s)")
|
|
menu.Append(sortID, "Sort Column")
|
|
|
|
def delete(event, self=self, col=col):
|
|
cols = self.GetSelectedCols()
|
|
self._table.DeleteCols(cols)
|
|
self.Reset()
|
|
|
|
def sort(event, self=self, col=col):
|
|
self._table.SortColumn(col)
|
|
self.Reset()
|
|
|
|
EVT_MENU(self, id1, delete)
|
|
if len(cols) == 1:
|
|
EVT_MENU(self, sortID, sort)
|
|
self.PopupMenu(menu, wxPoint(xo, 0))
|
|
menu.Destroy()
|
|
|
|
# -----------------------------------------------------------------
|
|
# Test data
|
|
# data is in the form
|
|
# [rowname, dictionary]
|
|
# where dictionary.get(colname, None) -> returns the value for the cell
|
|
#
|
|
# the colname must also be supplied
|
|
import random
|
|
colnames = ["Row", "This", "Is", "A", "Test"]
|
|
|
|
data = []
|
|
for row in range(1000):
|
|
d = {}
|
|
for name in ["This", "Test", "Is"]:
|
|
d[name] = random.random()
|
|
d["Row"] = len(data)
|
|
# XXX
|
|
# the "A" column can only be between one and 4
|
|
d["A"] = random.choice(range(4))
|
|
data.append((str(row), d))
|
|
|
|
class MegaFontRendererFactory:
|
|
def __init__(self, color, font, fontsize):
|
|
"""
|
|
(color, font, fontsize) -> set of a factory to generate
|
|
renderers when called.
|
|
func = MegaFontRenderFactory(color, font, fontsize)
|
|
renderer = func(table)
|
|
"""
|
|
self.color = color
|
|
self.font = font
|
|
self.fontsize = fontsize
|
|
|
|
def __call__(self, table):
|
|
return MegaFontRenderer(table, self.color, self.font, self.fontsize)
|
|
|
|
|
|
#---------------------------------------------------------------------------
|
|
|
|
class TestFrame(wxFrame):
|
|
def __init__(self, parent, plugins={"This":MegaFontRendererFactory("red", "ARIAL", 8),
|
|
"A":MegaImageRenderer,
|
|
"Test":MegaFontRendererFactory("orange", "TIMES", 24),}):
|
|
wxFrame.__init__(self, parent, -1,
|
|
"Test Frame", size=(640,480))
|
|
|
|
grid = MegaGrid(self, data, colnames, plugins)
|
|
grid.Reset()
|
|
|
|
|
|
#---------------------------------------------------------------------------
|
|
|
|
def runTest(frame, nb, log):
|
|
win = TestFrame(frame)
|
|
frame.otherWin = win
|
|
win.Show(True)
|
|
|
|
|
|
|
|
overview = """Mega Grid Example
|
|
|
|
This example attempts to show many examples and tricks of
|
|
using a virtual grid object. Hopefully the source isn't too jumbled.
|
|
|
|
Features:
|
|
1) Uses a virtual grid
|
|
2) Columns and rows have popup menus (right click on labels)
|
|
3) Columns and rows can be deleted (i.e. table can be
|
|
resized)
|
|
4) Dynamic renderers. Renderers are plugins based on
|
|
column header name. Shows a simple Font Renderer and
|
|
an Image Renderer.
|
|
|
|
Look for XXX in the code to indicate some workarounds for non-obvious
|
|
behavior and various hacks.
|
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
import sys,os
|
|
import run
|
|
run.main(['', os.path.basename(sys.argv[0])])
|
|
|