Thuban/UI/classifier.py

# Copyright (c) 2001, 2003 by Intevation GmbH
# Authors:
# Jonathan Coles <[email protected]>
#
# This program is free software under the GPL (>=v2)
# Read the file COPYING coming with Thuban for details.

"""Dialog for classifying how layers are displayed"""

__version__ = "$Revision$"

import copy

from Thuban.Model.table import FIELDTYPE_INT, FIELDTYPE_DOUBLE, \
     FIELDTYPE_STRING

from wxPython.wx import *
from wxPython.grid import *

from Thuban import _
from Thuban.common import *
from Thuban.UI.common import *

from Thuban.Model.classification import *

from Thuban.Model.color import Color

from Thuban.Model.layer import Layer, SHAPETYPE_ARC, SHAPETYPE_POLYGON, SHAPETYPE_POINT

from Thuban.UI.classgen import ClassGenDialog, ClassGenerator

from dialogs import NonModalDialog

# widget id's
ID_PROPERTY_SELECT = 4010
ID_CLASS_TABLE = 40011


# table columns
COL_VISIBLE = 0
COL_SYMBOL  = 1
COL_VALUE   = 2
COL_LABEL   = 3
NUM_COLS    = 4

# indices into the client data lists in Classifier.fields
FIELD_CLASS = 0
FIELD_TYPE = 1
FIELD_NAME = 2

#
# this is a silly work around to ensure that the table that is
# passed into SetTable is the same that is returned by GetTable
#
import weakref
class ClassGrid(wxGrid):


    def __init__(self, parent, classifier):
        """Constructor.

        parent -- the parent window

        clazz -- the working classification that this grid should
                 use for display. 
        """

        wxGrid.__init__(self, parent, ID_CLASS_TABLE)

        self.classifier = classifier

        self.currentSelection = []

        EVT_GRID_CELL_LEFT_DCLICK(self, self._OnCellDClick)
        EVT_GRID_RANGE_SELECT(self, self._OnSelectedRange)
        EVT_GRID_SELECT_CELL(self, self._OnSelectedCell)
        EVT_GRID_COL_SIZE(self, self._OnCellResize)
        EVT_GRID_ROW_SIZE(self, self._OnCellResize)

    #def GetCellAttr(self, row, col):
        #print "GetCellAttr ", row, col
        #wxGrid.GetCellAttr(self, row, col)

    def CreateTable(self, clazz, shapeType, group = None):

        assert isinstance(clazz, Classification)

        table = self.GetTable()
        if table is None:
            w = self.GetDefaultColSize() * 3 + self.GetDefaultRowLabelSize() 
            h = self.GetDefaultRowSize() * 4 + self.GetDefaultColLabelSize()
            self.SetDimensions(-1, -1, w, h)
            self.SetSizeHints(w, h, -1, -1)
            table = ClassTable(self)
            self.SetTable(table, True)


        self.SetSelectionMode(wxGrid.wxGridSelectRows)
        self.ClearSelection()

        table.Reset(clazz, shapeType, group)

    def GetCurrentSelection(self):
        """Return the currently highlighted rows as an increasing list
           of row numbers."""
        sel = copy.copy(self.currentSelection)
        sel.sort()
        return sel

    def GetSelectedRows(self):
        return self.GetCurrentSelection()

    #def SetCellRenderer(self, row, col, renderer):
        #raise ValueError(_("Must not allow setting of renderer in ClassGrid!"))

    #
    # [Set|Get]Table is taken from http://wiki.wxpython.org
    # they are needed as a work around to ensure that the table
    # that is passed to SetTable is the one that is returned 
    # by GetTable.
    #
    def SetTable(self, object, *attributes): 
        self.tableRef = weakref.ref(object) 
        return wxGrid.SetTable(self, object, *attributes) 

    def GetTable(self): 
        try:
            return self.tableRef() 
        except:
            return None

    def DeleteSelectedRows(self):
        """Deletes all highlighted rows.
  
        If only one row is highlighted then after it is deleted the
        row that was below the deleted row is highlighted."""

        sel = self.GetCurrentSelection()

        # nothing to do
        if len(sel) == 0: return 

        # if only one thing is selected check if it is the default
        # data row, because we can't remove that
        if len(sel) == 1:
            #group = self.GetTable().GetValueAsCustom(sel[0], COL_SYMBOL, None)
            group = self.GetTable().GetClassGroup(sel[0])
            if isinstance(group, ClassGroupDefault):
                wxMessageDialog(self, 
                                "The Default group cannot be removed.",
                                style = wxOK | wxICON_EXCLAMATION).ShowModal()
                return
        

        self.ClearSelection()

        # we need to remove things from the bottom up so we don't
        # change the indexes of rows that will be deleted next
        sel.reverse()

        #
        # actually remove the rows
        #
        table = self.GetTable()
        for row in sel:
            table.DeleteRows(row)

        #
        # if there was only one row selected highlight the row
        # that was directly below it, or move up one if the
        # deleted row was the last row.
        #
        if len(sel) == 1:
            r = sel[0]
            if r > self.GetNumberRows() - 1:
                r = self.GetNumberRows() - 1
            self.SelectRow(r)
        

    def SelectGroup(self, group, makeVisible = True):
        if group is None: return

        assert isinstance(group, ClassGroup) 

        table = self.GetTable()

        assert table is not None 

        for i in range(table.GetNumberRows()):
            g = table.GetClassGroup(i)
            if g is group:
                self.SelectRow(i)
                if makeVisible:
                    self.MakeCellVisible(i, 0)
                break

#
# XXX: This isn't working, and there is no way to deselect rows wxPython!
#
#   def DeselectRow(self, row):
#       self.ProcessEvent(
#           wxGridRangeSelectEvent(-1, 
#                                  wxEVT_GRID_RANGE_SELECT,
#                                  self,
#                                  (row, row), (row, row),
#                                  sel = False))

    def _OnCellDClick(self, event):
        """Handle a double click on a cell."""

        r = event.GetRow()
        c = event.GetCol()

        if c == COL_SYMBOL:
            self.classifier.EditSymbol(r)
        else:
            event.Skip()

    #
    # _OnSelectedRange() and _OnSelectedCell() were borrowed
    # from http://wiki.wxpython.org to keep track of which
    # cells are currently highlighted
    #
    def _OnSelectedRange(self, event): 
        """Internal update to the selection tracking list""" 
        if event.Selecting(): 
            for index in range( event.GetTopRow(), event.GetBottomRow()+1): 
                if index not in self.currentSelection: 
                    self.currentSelection.append( index ) 
        else: 
            for index in range( event.GetTopRow(), event.GetBottomRow()+1): 
                while index in self.currentSelection: 
                    self.currentSelection.remove( index ) 
        #self.ConfigureForSelection() 

        event.Skip() 

    def _OnSelectedCell( self, event ): 
        """Internal update to the selection tracking list""" 
        self.currentSelection = [ event.GetRow() ] 
        #self.ConfigureForSelection() 
        event.Skip() 

    def _OnCellResize(self, event):
        self.FitInside()

class ClassTable(wxPyGridTableBase):
    """Represents the underlying data structure for the grid."""

    __col_labels = [_("Visible"), _("Symbol"), _("Value"), _("Label")]


    def __init__(self, view = None):
    #def __init__(self, clazz, shapeType, view = None):
        """Constructor.

        shapeType -- the type of shape that the layer uses

        view -- a wxGrid object that uses this class for its table
        """

        wxPyGridTableBase.__init__(self)

        assert len(ClassTable.__col_labels) == NUM_COLS

        self.clazz = None
        self.__colAttr = {}

        self.SetView(view)

    def Reset(self, clazz, shapeType, group = None):
        """Reset the table with the given data.

        This is necessary because wxWindows does not allow a grid's
        table to change once it has been intially set and so we
        need a way of modifying the data.

        clazz -- the working classification that this table should
                 use for display. This may be different from the
                 classification in the layer.

        shapeType -- the type of shape that the layer uses
        """

        assert isinstance(clazz, Classification) 

        self.GetView().BeginBatch()

        self.fieldType = clazz.GetFieldType()
        self.shapeType = shapeType

        self.SetClassification(clazz, group)
        self.__Modified(-1)

        self.__colAttr = {}

        attr = wxGridCellAttr()
        attr.SetEditor(wxGridCellBoolEditor())
        attr.SetRenderer(wxGridCellBoolRenderer())
        attr.SetAlignment(wxALIGN_CENTER, wxALIGN_CENTER)
        self.__colAttr[COL_VISIBLE] = attr

        attr = wxGridCellAttr()
        attr.SetRenderer(ClassRenderer(self.shapeType))
        attr.SetReadOnly()
        self.__colAttr[COL_SYMBOL] = attr

        self.GetView().EndBatch()
        self.GetView().FitInside()

    def GetClassification(self):
        return self.clazz

    def SetClassification(self, clazz, group = None):

        self.GetView().BeginBatch()

        old_len = self.GetNumberRows()

        row = -1
        self.clazz = clazz
 
        self.__NotifyRowChanges(old_len, self.GetNumberRows())

        #
        # XXX: this is dead code at the moment
        #
        if row > -1:
            self.GetView().ClearSelection()
            self.GetView().SelectRow(row)
            self.GetView().MakeCellVisible(row, 0)

        self.__Modified()


        self.GetView().EndBatch()
        self.GetView().FitInside()

    def __NotifyRowChanges(self, curRows, newRows):
        #
        # silly message processing for updates to the number of
        # rows and columns
        #
        if newRows > curRows:
            msg = wxGridTableMessage(self,
                        wxGRIDTABLE_NOTIFY_ROWS_APPENDED,
                        newRows - curRows)    # how many
            self.GetView().ProcessTableMessage(msg)
            self.GetView().FitInside()
        elif newRows < curRows:
            msg = wxGridTableMessage(self,
                        wxGRIDTABLE_NOTIFY_ROWS_DELETED,
                        curRows,              # position
                        curRows - newRows)    # how many
            self.GetView().ProcessTableMessage(msg)
            self.GetView().FitInside()


    def __SetRow(self, row, group):
        """Set a row's data to that of the group.

        The table is considered modified after this operation.

        row -- if row is < 0 'group' is inserted at the top of the table
               if row is >= GetNumberRows() or None 'group' is append to 
                    the end of the table.
               otherwise 'group' replaces row 'row'
        """

        # either append or replace
        if row is None or row >= self.GetNumberRows():
            self.clazz.AppendGroup(group)
        elif row < 0:
            self.clazz.InsertGroup(0, group)
        else:
            if row == 0:
                self.clazz.SetDefaultGroup(group)
            else:
                self.clazz.ReplaceGroup(row - 1, group)

        self.__Modified()

    def GetColLabelValue(self, col):
        """Return the label for the given column."""
        return self.__col_labels[col]

    def GetRowLabelValue(self, row):
        """Return the label for the given row."""

        if row == 0:
            return _("Default")
        else:
            group = self.clazz.GetGroup(row - 1)
            if isinstance(group, ClassGroupDefault):   return _("Default")
            if isinstance(group, ClassGroupSingleton): return _("Singleton")
            if isinstance(group, ClassGroupRange):     return _("Range")
            if isinstance(group, ClassGroupMap):       return _("Map")

        assert False # shouldn't get here
        return _("")

    def GetNumberRows(self):
        """Return the number of rows."""
        if self.clazz is None:
            return 0

        return self.clazz.GetNumGroups() + 1 # +1 for default group

    def GetNumberCols(self):
        """Return the number of columns."""
        return NUM_COLS

    def IsEmptyCell(self, row, col):
        """Determine if a cell is empty. This is always false."""
        return False

    def GetValue(self, row, col):
        """Return the object that is used to represent the given
           cell coordinates. This may not be a string."""
        return self.GetValueAsCustom(row, col, None)

    def SetValue(self, row, col, value):
        """Assign 'value' to the cell specified by 'row' and 'col'.

        The table is considered modified after this operation.
        """

        self.SetValueAsCustom(row, col, None, value)
       
    def GetValueAsCustom(self, row, col, typeName):
        """Return the object that is used to represent the given
           cell coordinates. This may not be a string.
 
        typeName -- unused, but needed to overload wxPyGridTableBase
        """

        if row == 0:
            group = self.clazz.GetDefaultGroup()
        else:
            group = self.clazz.GetGroup(row - 1)


        if col == COL_VISIBLE:
            return group.IsVisible()

        if col == COL_SYMBOL:
            return group.GetProperties()

        if col == COL_LABEL:
            return group.GetLabel()

        # col must be COL_VALUE
        assert col == COL_VALUE 

        if isinstance(group, ClassGroupDefault):
            return _("DEFAULT")
        elif isinstance(group, ClassGroupSingleton):
            return group.GetValue()
        elif isinstance(group, ClassGroupRange):
            return _("%s - %s") % (group.GetMin(), group.GetMax())

        assert(False) # shouldn't get here
        return None

    def __ParseInput(self, value):
        """Try to determine what kind of input value is 
           (string, number, or range)

        Returns a tuple of length one if there is a single
        value, or of length two if it is a range.
        """

        type = self.fieldType

        if type == FIELDTYPE_STRING:
            return (value,)
        elif type in (FIELDTYPE_INT, FIELDTYPE_DOUBLE):

            if type == FIELDTYPE_INT:
                conv = lambda p: int(float(p))
            else:
                conv = lambda p: p

            #
            # first try to take the input as a single number
            # if there's an exception try to break it into
            # a range seperated by a '-'. take care to ignore
            # a leading '-' as that could be for a negative number.
            # then try to parse the individual parts. if there
            # is an exception here, let it pass up to the calling
            # function.
            #
            try:
                return (conv(Str2Num(value)),)
            except ValueError:
                i = value.find('-')
                if i == 0:
                    i = value.find('-', 1)

                return (conv(Str2Num(value[:i])), conv(Str2Num(value[i+1:])))

        assert False  # shouldn't get here
        return (0,)
            

    def SetValueAsCustom(self, row, col, typeName, value):
        """Set the cell specified by 'row' and 'col' to 'value'.

        If column represents the value column, the input is parsed
        to determine if a string, number, or range was entered.
        A new ClassGroup may be created if the type of data changes.

        The table is considered modified after this operation.

        typeName -- unused, but needed to overload wxPyGridTableBase
        """

        assert col >= 0 and col < self.GetNumberCols() 
        assert row >= 0 and row < self.GetNumberRows() 

        if row == 0:
            group = self.clazz.GetDefaultGroup()
        else:
            group = self.clazz.GetGroup(row - 1)

        mod = True # assume the data will change

        if col == COL_VISIBLE:
            group.SetVisible(value)
        elif col == COL_SYMBOL:
            group.SetProperties(value)
        elif col == COL_LABEL:
            group.SetLabel(value)
        elif col == COL_VALUE:
            if isinstance(group, ClassGroupDefault):
                # not allowed to modify the default value 
                pass
            elif isinstance(group, ClassGroupMap):
                # something special
                pass
            else: # SINGLETON, RANGE
                try:
                    dataInfo = self.__ParseInput(value)
                except ValueError:
                    # bad input, ignore the request
                    mod = False
                else:

                    changed = False
                    ngroup = group
                    props = group.GetProperties()

                    #
                    # try to update the values, which may include
                    # changing the underlying group type if the
                    # group was a singleton and a range was entered
                    #
                    if len(dataInfo) == 1:
                        if not isinstance(group, ClassGroupSingleton):
                            ngroup = ClassGroupSingleton(prop = props)
                            changed = True
                        ngroup.SetValue(dataInfo[0])
                    elif len(dataInfo) == 2:
                        if not isinstance(group, ClassGroupRange):
                            ngroup = ClassGroupRange(prop = props)
                            changed = True
                        ngroup.SetRange(dataInfo[0], dataInfo[1])
                    else:
                        assert False
                        pass

                    if changed:
                        ngroup.SetLabel(group.GetLabel())
                        self.SetClassGroup(row, ngroup)
        else:
            assert False # shouldn't be here
            pass

        if mod:
            self.__Modified()
            self.GetView().Refresh()

    def GetAttr(self, row, col, someExtraParameter):
        """Returns the cell attributes"""

        return self.__colAttr.get(col, wxGridCellAttr()).Clone()

    def GetClassGroup(self, row):
        """Return the ClassGroup object representing row 'row'."""

        #return self.GetValueAsCustom(row, COL_SYMBOL, None)
        if row == 0:
            return self.clazz.GetDefaultGroup()
        else:
            return self.clazz.GetGroup(row - 1)

    def SetClassGroup(self, row, group):
        self.__SetRow(row, group)
        self.GetView().Refresh()

    def __Modified(self, mod = True):
        """Adjust the modified flag.

        mod -- if -1 set the modified flag to False, otherwise perform
               an 'or' operation with the current value of the flag and
               'mod'
        """

        if mod == -1:
            self.modified = False
        else:
            self.modified = mod or self.modified

    def IsModified(self):
        """True if this table is considered modified."""
        return self.modified

    def DeleteRows(self, pos, numRows = 1):
        """Deletes 'numRows' beginning at row 'pos'. 

        The row representing the default group is not removed.

        The table is considered modified if any rows are removed.
        """

        assert pos >= 0 
        old_len = self.GetNumberRows() 
        for row in range(pos, pos - numRows, -1):
            group = self.GetClassGroup(row)
            if row != 0:
                self.clazz.RemoveGroup(row - 1)
                self.__Modified()
    
        if self.IsModified():
            self.__NotifyRowChanges(old_len, self.GetNumberRows())

    def AppendRows(self, numRows = 1):
        """Append 'numRows' empty rows to the end of the table.

        The table is considered modified if any rows are appended.
        """

        old_len = self.GetNumberRows()
        for i in range(numRows):
            np = ClassGroupSingleton()
            self.__SetRow(None, np)

        if self.IsModified():
            self.__NotifyRowChanges(old_len, self.GetNumberRows())


ID_CLASSIFY_OK = 4001
ID_CLASSIFY_REVERT = 4002
ID_CLASSIFY_ADD = 4003
ID_CLASSIFY_GENCLASS = 4004
ID_CLASSIFY_REMOVE = 4005
ID_CLASSIFY_MOVEUP = 4006
ID_CLASSIFY_MOVEDOWN = 4007
ID_CLASSIFY_TRY = 4008
ID_CLASSIFY_EDITSYM = 4009
ID_CLASSIFY_CLOSE = 4010

BTN_ADD = 0
BTN_EDIT = 1
BTN_GEN = 2
BTN_UP = 3
BTN_DOWN = 4
BTN_RM = 5

class Classifier(NonModalDialog):

    type2string = {None:             _("None"),
                   FIELDTYPE_STRING: _("Text"),
                   FIELDTYPE_INT:    _("Integer"),
                   FIELDTYPE_DOUBLE: _("Decimal")}

    def __init__(self, parent, name, layer, group = None):
        NonModalDialog.__init__(self, parent, name,
                                _("Classifier: %s") % layer.Title())

        panel = wxPanel(self, -1, size=(100, 100))

        self.layer = layer

        self.originalClass = self.layer.GetClassification()
        field = self.originalClass.GetField()
        fieldType = self.originalClass.GetFieldType()

        self.genDlg = None

        topBox = wxBoxSizer(wxVERTICAL)
        panelBox = wxBoxSizer(wxVERTICAL)

        #panelBox.Add(wxStaticText(panel, -1, _("Layer: %s") % layer.Title()),
            #0, wxALIGN_LEFT | wxALL, 4) 
        panelBox.Add(wxStaticText(panel, -1, 
                                _("Layer Type: %s") % layer.ShapeType()),
            0, wxALIGN_LEFT | wxALL, 4) 


        #
        # make field combo box
        #
        self.fields = wxComboBox(panel, ID_PROPERTY_SELECT, "",
                                     style = wxCB_READONLY)

        self.num_cols = layer.table.field_count()
        # just assume the first field in case one hasn't been
        # specified in the file.
        self.__cur_field = 0 

        self.fields.Append("<None>")
        self.fields.SetClientData(0, None)

        for i in range(self.num_cols):
            type, name, len, decc = layer.table.field_info(i)
            self.fields.Append(name)

            if name == field:
                self.__cur_field = i + 1
                self.fields.SetClientData(i + 1, 
                                          copy.deepcopy(self.originalClass))
            else:
                self.fields.SetClientData(i + 1, None)


        ###########

        self.fieldTypeText = wxStaticText(panel, -1, "")
        panelBox.Add(self.fieldTypeText, 0, 
                     wxGROW | wxALIGN_LEFT | wxALL | wxADJUST_MINSIZE, 4) 

        propertyBox = wxBoxSizer(wxHORIZONTAL)
        propertyBox.Add(wxStaticText(panel, -1, _("Field: ")),
            0, wxALIGN_LEFT | wxALL, 4)
        propertyBox.Add(self.fields, 1, wxGROW|wxALL, 4)
        EVT_COMBOBOX(self, ID_PROPERTY_SELECT, self._OnFieldSelect)

        panelBox.Add(propertyBox, 0, wxGROW, 4)


        #
        # Control Box 
        #
        controlBox = wxBoxSizer(wxHORIZONTAL)


        ###########
        #
        # Control buttons:
        #
        self.controlButtons = []

        controlButtonBox = wxBoxSizer(wxVERTICAL)

        button = wxButton(panel, ID_CLASSIFY_GENCLASS, _("Generate Class"))
        controlButtonBox.Add(button, 0, wxGROW | wxALL, 4)
        self.controlButtons.append(button)

        button = wxButton(panel, ID_CLASSIFY_ADD, _("Add"))
        controlButtonBox.Add(button, 0, wxGROW | wxALL, 4)
        self.controlButtons.append(button)

        button = wxButton(panel, ID_CLASSIFY_MOVEUP, _("Move Up"))
        controlButtonBox.Add(button, 0, wxGROW | wxALL, 4)
        self.controlButtons.append(button)

        button = wxButton(panel, ID_CLASSIFY_MOVEDOWN, _("Move Down"))
        controlButtonBox.Add(button, 0, wxGROW | wxALL, 4)
        self.controlButtons.append(button)

        button = wxButton(panel, ID_CLASSIFY_EDITSYM, _("Edit Symbol"))
        controlButtonBox.Add(button, 0, wxGROW | wxALL, 4)
        self.controlButtons.append(button)

        controlButtonBox.Add(60, 20, 0, wxGROW | wxALL | wxALIGN_BOTTOM, 4)

        button = wxButton(panel, ID_CLASSIFY_REMOVE, _("Remove"))
        controlButtonBox.Add(button, 0, wxGROW | wxALL | wxALIGN_BOTTOM, 4)
        self.controlButtons.append(button)


        ###########
        #
        # Classification data table
        #

        self.classGrid = ClassGrid(panel, self)
#       self.classGrid = wxGrid(panel, -1)
#       self.classGrid.SetDefaultRenderer(wxGridCellBoolRenderer())
#       self.classGrid.SetDefaultEditor(wxGridCellBoolEditor())
        #self.classGrid.CreateGrid(5, 5)
        #self.classGrid.SetCellEditor(0, 0, wxGridCellBoolEditor())
        #self.classGrid.SetCellRenderer(0, 0, wxGridCellBoolRenderer())
        #print self.classGrid.GetCellEditor(0, 0)
        #print self.classGrid.GetCellRenderer(0, 0)
        #self.classGrid = ClassGrid(panel, self)

        # need these
        self.__SetGridTable(self.__cur_field, group)
        self.fields.SetSelection(self.__cur_field)
        #self.classGrid.SetCellEditor(0, 0, wxGridCellBoolEditor())
        #self.classGrid.SetCellRenderer(0, 0, wxGridCellBoolRenderer())

        # calling __SelectField after creating the classGrid fills in the
        # grid with the correct information
        #self.fields.SetSelection(self.__cur_field)
        #self.__SelectField(self.__cur_field, group = group)

        #self.classGrid.SelectGroup(group)

        controlBox.Add(self.classGrid, 1, wxGROW, 0)


        controlBox.Add(controlButtonBox, 0, wxGROW, 10)
        panelBox.Add(controlBox, 1, wxGROW, 10)

        EVT_BUTTON(self, ID_CLASSIFY_ADD, self._OnAdd)
        EVT_BUTTON(self, ID_CLASSIFY_EDITSYM, self._OnEditSymbol)
        EVT_BUTTON(self, ID_CLASSIFY_REMOVE, self._OnRemove)
        EVT_BUTTON(self, ID_CLASSIFY_GENCLASS, self._OnGenClass)
        EVT_BUTTON(self, ID_CLASSIFY_MOVEUP, self._OnMoveUp)
        EVT_BUTTON(self, ID_CLASSIFY_MOVEDOWN, self._OnMoveDown)

        ###########

        buttonBox = wxBoxSizer(wxHORIZONTAL)
        buttonBox.Add(wxButton(panel, ID_CLASSIFY_TRY, _("Try")),
                      0, wxALL, 4)
        buttonBox.Add(60, 20, 0, wxALL, 4)
        buttonBox.Add(wxButton(panel, ID_CLASSIFY_REVERT, _("Revert")),
                      0, wxALL, 4)
        buttonBox.Add(60, 20, 0, wxALL, 4)
        buttonBox.Add(wxButton(panel, ID_CLASSIFY_OK, _("OK")),
                      0, wxALL, 4)
        buttonBox.Add(60, 20, 0, wxALL, 4)
        buttonBox.Add(wxButton(panel, ID_CLASSIFY_CLOSE, _("Close")),
                      0, wxALL, 4)
        panelBox.Add(buttonBox, 0, wxALIGN_CENTER_HORIZONTAL|wxALIGN_BOTTOM, 0)

        EVT_BUTTON(self, ID_CLASSIFY_OK, self._OnOK)
        EVT_BUTTON(self, ID_CLASSIFY_TRY, self._OnTry)
        EVT_BUTTON(self, ID_CLASSIFY_CLOSE, self._OnCloseBtn)
        EVT_BUTTON(self, ID_CLASSIFY_REVERT, self._OnRevert)

        ###########


        panel.SetAutoLayout(True)
        panel.SetSizer(panelBox)
        panelBox.SetSizeHints(panel) 

        topBox.Add(panel, 1, wxGROW, 0)
        panelBox.SetSizeHints(self)
        self.SetAutoLayout(True)
        self.SetSizer(topBox)

        #self.Fit()
        ######################

        self.haveApplied = False

    def EditSymbol(self, row):
        table = self.classGrid.GetTable()
        prop = table.GetValueAsCustom(row, COL_SYMBOL, None)

        # get a new ClassGroupProperties object and copy the 
        # values over to our current object
        propDlg = SelectPropertiesDialog(NULL, prop, self.layer.ShapeType())

        self.Enable(False)
        if propDlg.ShowModal() == wxID_OK:
            new_prop = propDlg.GetClassGroupProperties()
            table.SetValueAsCustom(row, COL_SYMBOL, None, new_prop)
        self.Enable(True)
        propDlg.Destroy()
        
    def _SetClassification(self, clazz):
        
        self.fields.SetClientData(self.__cur_field, clazz)
        self.classGrid.GetTable().SetClassification(clazz)

    def __BuildClassification(self, fieldIndex, copyClass = False):

#       numRows = self.classGrid.GetNumberRows()
#       assert numRows > 0  # there should always be a default row

#       clazz = Classification()
        if fieldIndex == 0:
            fieldName = None
            fieldType = None
        else:
            fieldName = self.fields.GetString(fieldIndex)
            fieldType = self.layer.GetFieldType(fieldName)

        clazz = self.classGrid.GetTable().GetClassification()

        if copyClass:
            clazz = copy.deepcopy(clazz)

        clazz.SetField(fieldName)
        clazz.SetFieldType(fieldType)


#       table = self.classGrid.GetTable()
#       clazz.SetDefaultGroup(table.GetClassGroup(0))

#       for i in range(1, numRows):
#           clazz.AppendGroup(table.GetClassGroup(i))

        return clazz

    def __SetGridTable(self, fieldIndex, group = None):

        clazz = self.fields.GetClientData(fieldIndex)

        if clazz is None:
            clazz = Classification()
            clazz.SetDefaultGroup(
                ClassGroupDefault(
                    self.layer.GetClassification().
                               GetDefaultGroup().GetProperties()))

            fieldName = self.fields.GetString(fieldIndex)
            fieldType = self.layer.GetFieldType(fieldName)
            clazz.SetFieldType(fieldType)
                
        self.classGrid.CreateTable(clazz, self.layer.ShapeType(), group)

    def __SetFieldTypeText(self, fieldIndex):
        fieldName = self.fields.GetString(fieldIndex)
        fieldType = self.layer.GetFieldType(fieldName)

        assert Classifier.type2string.has_key(fieldType) 

        text = Classifier.type2string[fieldType]

        self.fieldTypeText.SetLabel(_("Field Type: %s") % text)

    def __SelectField(self, newIndex, oldIndex = -1, group = None):
        """This method assumes that the current selection for the
        combo has already been set by a call to SetSelection().
        """

        assert oldIndex >= -1

        if oldIndex != -1:
            clazz = self.__BuildClassification(oldIndex)
            self.fields.SetClientData(oldIndex, clazz)

        self.__SetGridTable(newIndex, group)

        enabled = newIndex != 0

        for b in self.controlButtons:
            b.Enable(enabled)

        self.__SetFieldTypeText(newIndex)

 
    def _OnEditSymbol(self, event):
        sel = self.classGrid.GetCurrentSelection()

        if len(sel) == 1:
            self.EditSymbol(sel[0])

    def _OnFieldSelect(self, event): 
        index = self.fields.GetSelection()
        self.__SelectField(index, self.__cur_field)
        self.__cur_field = index

    def _OnTry(self, event):
        """Put the data from the table into a new Classification and hand
           it to the layer.
        """

        clazz = self.fields.GetClientData(self.__cur_field)

        #
        # only build the classification if there wasn't one to
        # to begin with or it has been modified
        #
        if clazz is None or self.classGrid.GetTable().IsModified():
            clazz = self.__BuildClassification(self.__cur_field, True)

        self.layer.SetClassification(clazz)

        self.haveApplied = True

    def _OnOK(self, event):
        self._OnTry(event)
        self.Close()

    def _OnCloseBtn(self, event):
        """Close is similar to Cancel except that any changes that were
        made and applied remain applied, but the currently displayed
        classification is discarded.
        """

        self.Close()

    def _OnRevert(self, event):
        """The layer's current classification stays the same."""
        if self.haveApplied:
            self.layer.SetClassification(self.originalClass)

        #self.Close()

    def _OnAdd(self, event): 
        self.classGrid.AppendRows()

    def _OnRemove(self, event):
        self.classGrid.DeleteSelectedRows()

    def _OnGenClass(self, event):

        #if self.genDlg is None:
        self.genDlg = ClassGenDialog(self, self.layer,
                          self.fields.GetString(self.__cur_field))

        EVT_CLOSE(self.genDlg, self._OnGenDialogClose)

        self.fields.Enable(False)
        self.controlButtons[BTN_EDIT].Enable(False)
        self.controlButtons[BTN_GEN].Enable(False)

        self.genDlg.Show()
        #if self.genDlg.ShowModal() == wxID_OK:
        #    clazz = self.genDlg.GetClassification()
        #    self.fields.SetClientData(self.__cur_field, clazz)
        #    self.classGrid.GetTable().SetClassification(clazz)
        #self.Enable(True)
        #self.genDlg.Destroy()

    def _OnGenDialogClose(self, event):
        self.genDlg.Destroy()

        self.fields.Enable(True)
        self.controlButtons[BTN_EDIT].Enable(True)
        self.controlButtons[BTN_GEN].Enable(True)

    def _OnMoveUp(self, event):
        sel = self.classGrid.GetCurrentSelection()

        if len(sel) == 1:
            i = sel[0]
            if i > 1:
                table = self.classGrid.GetTable()
                x = table.GetClassGroup(i - 1)
                y = table.GetClassGroup(i)
                table.SetClassGroup(i - 1, y)
                table.SetClassGroup(i, x)
                self.classGrid.ClearSelection()
                self.classGrid.SelectRow(i - 1)
                self.classGrid.MakeCellVisible(i - 1, 0)

    def _OnMoveDown(self, event):
        sel = self.classGrid.GetCurrentSelection()

        if len(sel) == 1:
            i = sel[0]
            table = self.classGrid.GetTable()
            if 0 < i < table.GetNumberRows() - 1:
                x = table.GetClassGroup(i)
                y = table.GetClassGroup(i + 1)
                table.SetClassGroup(i, y)
                table.SetClassGroup(i + 1, x)
                self.classGrid.ClearSelection()
                self.classGrid.SelectRow(i + 1)
                self.classGrid.MakeCellVisible(i + 1, 0)


ID_SELPROP_OK = 4001
ID_SELPROP_CANCEL = 4002
ID_SELPROP_SPINCTRL = 4002
ID_SELPROP_PREVIEW = 4003
ID_SELPROP_STROKECLR = 4004
ID_SELPROP_FILLCLR = 4005
ID_SELPROP_STROKECLRTRANS = 4006
ID_SELPROP_FILLCLRTRANS = 4007

class SelectPropertiesDialog(wxDialog):

    def __init__(self, parent, prop, shapeType):
        wxDialog.__init__(self, parent, -1, _("Select Properties"),
                          style = wxDEFAULT_DIALOG_STYLE | wxRESIZE_BORDER)

        self.prop = ClassGroupProperties(prop)

        topBox = wxBoxSizer(wxVERTICAL)

        itemBox = wxBoxSizer(wxHORIZONTAL)

        # preview box
        previewBox = wxBoxSizer(wxVERTICAL)
        previewBox.Add(wxStaticText(self, -1, _("Preview:")),
            0, wxALIGN_LEFT | wxALL, 4)

        self.previewWin = ClassGroupPropertiesCtrl(
            self, ID_SELPROP_PREVIEW, self.prop, shapeType, 
            (40, 40), wxSIMPLE_BORDER)

        self.previewWin.AllowEdit(False)

        previewBox.Add(self.previewWin, 1, wxGROW | wxALL, 4)

        itemBox.Add(previewBox, 1, wxALIGN_LEFT | wxALL | wxGROW, 0)

        # control box
        ctrlBox = wxBoxSizer(wxVERTICAL)

        lineColorBox = wxBoxSizer(wxHORIZONTAL)
        lineColorBox.Add(
            wxButton(self, ID_SELPROP_STROKECLR, _("Change Line Color")),
            1, wxALL | wxGROW, 4)
        EVT_BUTTON(self, ID_SELPROP_STROKECLR, self._OnChangeLineColor)

        lineColorBox.Add(
            wxButton(self, ID_SELPROP_STROKECLRTRANS, _("Transparent")),
            1, wxALL | wxGROW, 4)
        EVT_BUTTON(self, ID_SELPROP_STROKECLRTRANS,
                   self._OnChangeLineColorTrans)

        ctrlBox.Add(lineColorBox, 0, 
                    wxALIGN_CENTER_HORIZONTAL | wxALL | wxGROW, 4)

        if shapeType != SHAPETYPE_ARC:
            fillColorBox = wxBoxSizer(wxHORIZONTAL)
            fillColorBox.Add(
                wxButton(self, ID_SELPROP_FILLCLR, _("Change Fill Color")),
                1, wxALL | wxGROW, 4)
            EVT_BUTTON(self, ID_SELPROP_FILLCLR, self._OnChangeFillColor)
            fillColorBox.Add(
                wxButton(self, ID_SELPROP_FILLCLRTRANS, _("Transparent")),
                1, wxALL | wxGROW, 4)
            EVT_BUTTON(self, ID_SELPROP_FILLCLRTRANS,
                       self._OnChangeFillColorTrans)
            ctrlBox.Add(fillColorBox, 0, 
                        wxALIGN_CENTER_HORIZONTAL | wxALL | wxGROW, 4)

        spinBox = wxBoxSizer(wxHORIZONTAL)
        spinBox.Add(wxStaticText(self, -1, _("Line Width: ")),
                0, wxALIGN_LEFT | wxALIGN_CENTER_VERTICAL | wxALL, 4) 
        self.spinCtrl = wxSpinCtrl(self, ID_SELPROP_SPINCTRL, 
                                   min=1, max=10, 
                                   value=str(prop.GetLineWidth()),
                                   initial=prop.GetLineWidth())

        EVT_SPINCTRL(self, ID_SELPROP_SPINCTRL, self._OnSpin)

        spinBox.Add(self.spinCtrl, 0, wxALIGN_LEFT | wxALL, 4) 

        ctrlBox.Add(spinBox, 0, wxALIGN_RIGHT | wxALL, 0)
        itemBox.Add(ctrlBox, 0, wxALIGN_RIGHT | wxALL | wxGROW, 0)
        topBox.Add(itemBox, 1, wxALIGN_LEFT | wxALL | wxGROW, 0)

        #
        # Control buttons:
        #
        buttonBox = wxBoxSizer(wxHORIZONTAL)
        buttonBox.Add(wxButton(self, ID_SELPROP_OK, _("OK")),
                      0, wxALL, 4)
        buttonBox.Add(wxButton(self, ID_SELPROP_CANCEL, _("Cancel")),
                      0, wxALL, 4)
        topBox.Add(buttonBox, 0, wxALIGN_CENTER_HORIZONTAL|wxALIGN_BOTTOM, 10)
                                                                                
        EVT_BUTTON(self, ID_SELPROP_OK, self._OnOK)
        EVT_BUTTON(self, ID_SELPROP_CANCEL, self._OnCancel)
                                                                                
        self.SetAutoLayout(True)
        self.SetSizer(topBox)
        topBox.Fit(self)
        topBox.SetSizeHints(self)

    def _OnOK(self, event):
        self.EndModal(wxID_OK)

    def _OnCancel(self, event):
        self.EndModal(wxID_CANCEL)

    def _OnSpin(self, event):
        self.prop.SetLineWidth(self.spinCtrl.GetValue())
        self.previewWin.Refresh()

    def __GetColor(self, cur):
        dialog = wxColourDialog(self)
        if cur is not Color.Transparent:
            dialog.GetColourData().SetColour(Color2wxColour(cur))

        ret = None
        if dialog.ShowModal() == wxID_OK:
            ret = wxColour2Color(dialog.GetColourData().GetColour())

        dialog.Destroy()

        return ret
        
    def _OnChangeLineColor(self, event):
        clr = self.__GetColor(self.prop.GetLineColor())
        if clr is not None:
            self.prop.SetLineColor(clr)
        self.previewWin.Refresh() # XXX: work around, see ClassDataPreviewer

    def _OnChangeLineColorTrans(self, event):
        self.prop.SetLineColor(Color.Transparent)
        self.previewWin.Refresh() # XXX: work around, see ClassDataPreviewer
        
    def _OnChangeFillColor(self, event):
        clr = self.__GetColor(self.prop.GetFill())
        if clr is not None:
            self.prop.SetFill(clr)
        self.previewWin.Refresh() # XXX: work around, see ClassDataPreviewer

    def _OnChangeFillColorTrans(self, event):
        self.prop.SetFill(Color.Transparent)
        self.previewWin.Refresh() # XXX: work around, see ClassDataPreviewer

    def GetClassGroupProperties(self):
        return self.prop


class ClassDataPreviewWindow(wxWindow):

    def __init__(self, rect, prop, shapeType, 
                       parent = None, id = -1, size = wxDefaultSize):
        if parent is not None:
            wxWindow.__init__(self, parent, id, (0, 0), size)
            EVT_PAINT(self, self._OnPaint)

        self.rect = rect

        self.prop = prop
        self.shapeType = shapeType
        self.previewer = ClassDataPreviewer()

    def GetProperties():
        return self.prop

    def _OnPaint(self, event):
        dc = wxPaintDC(self)

        # XXX: this doesn't seem to be having an effect:
        dc.DestroyClippingRegion() 

        if self.rect is None:
            w, h = self.GetSize()
            rect = wxRect(0, 0, w, h)
        else:
            rect = self.rect

        self.previewer.Draw(dc, rect, self.prop, self.shapeType)

class ClassDataPreviewer:

    def Draw(self, dc, rect, prop, shapeType):

        assert dc is not None
        assert isinstance(prop, ClassGroupProperties)

        if rect is None:
            x = 0
            y = 0
            w, h = dc.GetSize()
        else:
            x = rect.GetX()
            y = rect.GetY()
            w = rect.GetWidth()
            h = rect.GetHeight()

        stroke = prop.GetLineColor()
        if stroke is Color.Transparent:
            pen = wxTRANSPARENT_PEN
        else:
            pen = wxPen(Color2wxColour(stroke),
                        prop.GetLineWidth(),
                        wxSOLID)

        stroke = prop.GetFill()
        if stroke is Color.Transparent:
            brush = wxTRANSPARENT_BRUSH
        else:
            brush = wxBrush(Color2wxColour(stroke), wxSOLID)

        dc.SetPen(pen)
        dc.SetBrush(brush)

        if shapeType == SHAPETYPE_ARC:
            dc.DrawSpline([wxPoint(x, y + h),
                           wxPoint(x + w/2, y + h/4),
                           wxPoint(x + w/2, y + h/4*3),
                           wxPoint(x + w, y)])

        elif shapeType == SHAPETYPE_POINT:

            dc.DrawCircle(x + w/2, y + h/2,
                          (min(w, h) - prop.GetLineWidth())/2)

        elif shapeType == SHAPETYPE_POLYGON:
            dc.DrawRectangle(x, y, w, h)

class ClassRenderer(wxPyGridCellRenderer):

    def __init__(self, shapeType):
        wxPyGridCellRenderer.__init__(self)
        self.shapeType = shapeType
        self.previewer = ClassDataPreviewer()

    def Draw(self, grid, attr, dc, rect, row, col, isSelected):
        data = grid.GetTable().GetClassGroup(row)

        dc.SetClippingRegion(rect.GetX(), rect.GetY(), 
                             rect.GetWidth(), rect.GetHeight())
        dc.SetPen(wxPen(wxLIGHT_GREY))
        dc.SetBrush(wxBrush(wxLIGHT_GREY, wxSOLID))
        dc.DrawRectangle(rect.GetX(), rect.GetY(), 
                         rect.GetWidth(), rect.GetHeight())

        if not isinstance(data, ClassGroupMap):
            self.previewer.Draw(dc, rect, data.GetProperties(), self.shapeType)

        if isSelected:
            dc.SetPen(wxPen(wxBLACK, 1, wxSOLID))
            dc.SetBrush(wxTRANSPARENT_BRUSH)

            dc.DrawRectangle(rect.GetX(), rect.GetY(), 
                             rect.GetWidth(), rect.GetHeight())

        dc.DestroyClippingRegion()


class ClassGroupPropertiesCtrl(wxWindow, wxControl):

    def __init__(self, parent, id, props, shapeType, 
                 size = wxDefaultSize, style = 0):

        wxWindow.__init__(self, parent, id, size = size, style = style)

        self.SetProperties(props)
        self.SetShapeType(shapeType)
        self.AllowEdit(True)

        EVT_PAINT(self, self._OnPaint)
        EVT_LEFT_DCLICK(self, self._OnLeftDClick) 

        self.previewer = ClassDataPreviewer()

    def _OnPaint(self, event):
        dc = wxPaintDC(self)

        # XXX: this doesn't seem to be having an effect:
        dc.DestroyClippingRegion() 

        w, h = self.GetClientSize()

        self.previewer.Draw(dc, 
                            wxRect(0, 0, w, h), 
                            self.GetProperties(), 
                            self.GetShapeType())


    def GetProperties(self):
        return self.props

    def SetProperties(self, props):
        self.props = props
        self.Refresh()

    def GetShapeType(self):
        return self.shapeType

    def SetShapeType(self, shapeType):
        self.shapeType = shapeType
        self.Refresh()

    def AllowEdit(self, allow):
        self.allowEdit = allow

    def DoEdit(self):
        if not self.allowEdit: return

        propDlg = SelectPropertiesDialog(NULL, 
                                         self.GetProperties(), 
                                         self.GetShapeType())

        if propDlg.ShowModal() == wxID_OK:
            new_prop = propDlg.GetClassGroupProperties()
            self.SetProperties(new_prop)
            self.Refresh()

        propDlg.Destroy()

    def _OnLeftDClick(self, event):
        self.DoEdit()
Name	Value
svn:eol-style	native
svn:keywords	Author Date Id Revision