Pull to refresh

Пример использования WxPython для создания нодового интерфейса. Часть 3: Продолжаем добавлять фичи + обработка клавиатуры

Reading time 4 min
Views 7.6K
В небольшом цикле статей будет описано использование WxPython для решения вполне конкретной задачи по разработке пользовательского интерфейса, да еще и то, как сделать это решение универсальным. Туториал этот расчитан на тех, кто уже начал изучать эту библиотеку и хочет увидеть что-то более сложное и целостное, чем простейшие примеры (хотя начнется все с относительно простых вещей).

В прошлой части мы реализовали обработку событий мыши, а теперь мы добавим еще одну фичу и реализуем обработку событий от клавиатуры.

Часть 1: Учимся рисовать
Часть 2: Обработка событий мыши
Часть 3: Продолжаем добавлять фичи + обработка клавиатуры
Часть 4: Реализуем Drag&Drop
Часть 5: Соединяем ноды

Кому интересно, добро пожаловать под кат…


7. Выделение объектов


В прошлый раз мы научились перемещать ноды, теперь пора учиться их удалять, да и не только их, но и любые объекты на канвасе. Правда для удаления, надо сначала указать, что мы хотим удалить, т.е. выделить объект, так что начнем мы не с удаления, а с выделения.
Для выделения обычно все используют одинарный клик, но, насколько я знаю, wxPython не имеет специального события одинарного клика, так что будем отслеживать его сами. Сделаем мы это простейшим способом: если пользователь надапил на левую кнопку мыши и отпустил ее над одним и тем же пикселем, значит это клик. Для этого нам надо запомнить, где пользователь нажал на кнопку (мы сохраним это в «self._lastLeftDownPos»), а при отпускании сравним и сохраним выделенный объект:
    def OnMouseLeftUp(self, evt):
        #Selection
        if (self._lastLeftDownPos 
                and self._lastLeftDownPos[0] == evt.GetPosition()[0] 
                and self._lastLeftDownPos[1] == evt.GetPosition()[1] 
                and self._objectUnderCursor 
                and self._objectUnderCursor.selectable):
            self._selectedObject = self._objectUnderCursor
            
        self.Render()

Ну и чтобы пользователь видел, что же он там выделил, в код рендеринга канваса добавим следующий фрагмент:
        if self._selectedObject:
            gc.PushState()
            self._selectedObject.RenderSelection(gc)
            gc.PopState()

Все? Нет не все, внимательный читатель заметил, что мы проверяли, является ли объект selectable и вызывали новый метод RenderSelection, который должен быть у всех выделяемых объектов. Поэтому, по образу и подобию перемещения, мы добавим новый базовый класс от которого будут наследоваться все объекты, которые могут быть выделены:
class SelectableObject(CanvasObject):
    def __init__(self, **kwargs):
        super(SelectableObject, self).__init__(**kwargs)
        self.selectable = True

    def RenderSelection(self, gc):
        """
        RenderHighlighting method should draw an 
        object with a selection border around it.
        """
        raise NotImplementedError()

Кстати, раз уж наших базовых классов становится больше и все становится сложнее, теперь их конструкторы принимают сколько угодно параметров и передают их вверх по иерархии наследования в виде словаря, что и делает в данном случае строка «super(SelectableObject, self).__init__(**kwargs)».
Осталось лишь добавить к ноде вот такой код:
    def RenderSelection(self, gc):
        gc.SetBrush(wx.Brush('#888888', wx.TRANSPARENT))
        gc.SetPen(wx.Pen('#CC0000', 3, wx.DOT))
        gc.DrawRectangle(self.position[0]-2, 
                         self.position[1]-2, 
                         self.boundingBoxDimensions[0]+4, 
                         self.boundingBoxDimensions[1]+4)

который рендерит выделение вокруг объекта и можно увидеть такой результат, если кликнуть на одну из нод:

Текущий код, как и всегда, можно найти в соответствующем коммите на GitHub'е.

8. Удаление объектов


Вот мы и добрались до удаления объектов. Для удаления нам надо отследить, когда пользователь нажимает кнопку Delete и собственно удалить элемент, если он удаляемый.
Для начала нам потребуется обработчик события нажатия на кнопки клавиатуры, который мы сделаем зарегистрируем так:
        self.Bind(wx.EVT_KEY_DOWN, self.OnKeyPress)

А сам обработчик будет выполнять такой код:
    def OnKeyPress(self, evt):
        if evt.GetKeyCode() == wx.WXK_DELETE:
            if self._selectedObject and self._selectedObject.deletable:
                self._selectedObject.Delete()
                if self._selectedObject in self._canvasObjects:
                    self._canvasObjects.remove(self._selectedObject)
                self._selectedObject = None
        else: 
            evt.Skip()

        #Update object under cursor                
        pos = self.CalcUnscrolledPosition(evt.GetPosition()).Get()
        self._objectUnderCursor = self.FindObjectUnderPoint(pos)
            
        self.Render()

Тут мы, в случае прихода события о нажатии кнопки с кодом wx.WXK_DELETE, проверяем, выделен ли объект и удаляемый ли он. Если есть такой, мы его красиво уведомляем о том, что пора бы ему на покой (вызовом Delete) и удалением из списка объектов на канвасе. evt.Skip() говорит системе, что надо вызывать остальные обработчики этого события зарегистрированные в других местах.

Теперь, уже традиционно, осталось добавить базовый класс для удаляемых объектов:
class DeletableObject(CanvasObject):
    def __init__(self, **kwargs):
        super(DeletableObject, self).__init__(**kwargs)
        self.deletable = True
        
    def Delete(self):
        """
        Delete method is called when an object is deleted from a canvas.
        """
        raise NotImplementedError()

И можно наслаждаться удалением объектов с канваса. Правда не очень долго, так как добавление объектов будет только в следующей части:)
Текущая версия кода живет в этом коммите на GitHub'е.

PS: Об опечатках пишите в личку.
Tags:
Hubs:
+19
Comments 0
Comments Leave a comment

Articles