-# -*- coding: iso-8859-1 -*-
-#----------------------------------------------------------------------------
-# Name: canvas.py
-# Purpose: The canvas class
-#
-# Author: Pierre Hjälm (from C++ original by Julian Smart)
-#
-# Created: 2004-05-08
-# RCS-ID: $Id$
-# Copyright: (c) 2004 Pierre Hjälm - 1998 Julian Smart
-# Licence: wxWindows license
-#----------------------------------------------------------------------------
-
-from __future__ import division
-
-import wx
-from lines import LineShape
-from composit import *
-
-NoDragging, StartDraggingLeft, ContinueDraggingLeft, StartDraggingRight, ContinueDraggingRight = 0, 1, 2, 3, 4
-
-KEY_SHIFT, KEY_CTRL = 1, 2
-
-
-
-# Helper function: True if 'contains' wholly contains 'contained'.
-def WhollyContains(contains, contained):
- xp1, yp1 = contains.GetX(), contains.GetY()
- xp2, yp2 = contained.GetX(), contained.GetY()
-
- w1, h1 = contains.GetBoundingBoxMax()
- w2, h2 = contained.GetBoundingBoxMax()
-
- left1 = xp1-w1 / 2.0
- top1 = yp1-h1 / 2.0
- right1 = xp1 + w1 / 2.0
- bottom1 = yp1 + h1 / 2.0
-
- left2 = xp2-w2 / 2.0
- top2 = yp2-h2 / 2.0
- right2 = xp2 + w2 / 2.0
- bottom2 = yp2 + h2 / 2.0
-
- return ((left1 <= left2) and (top1 <= top2) and (right1 >= right2) and (bottom1 >= bottom2))
-
-
-
-class ShapeCanvas(wx.ScrolledWindow):
- def __init__(self, parent = None, id=-1, pos = wx.DefaultPosition, size = wx.DefaultSize, style = wx.BORDER, name="ShapeCanvas"):
- wx.ScrolledWindow.__init__(self, parent, id, pos, size, style, name)
-
- self._shapeDiagram = None
- self._dragState = NoDragging
- self._draggedShape = None
- self._oldDragX = 0
- self._oldDragY = 0
- self._firstDragX = 0
- self._firstDragY = 0
- self._checkTolerance = True
-
- wx.EVT_PAINT(self, self.OnPaint)
- wx.EVT_MOUSE_EVENTS(self, self.OnMouseEvent)
-
- def SetDiagram(self, diag):
- self._shapeDiagram = diag
-
- def GetDiagram(self):
- return self._shapeDiagram
-
- def OnPaint(self, evt):
- dc = wx.PaintDC(self)
- self.PrepareDC(dc)
-
- dc.SetBackground(wx.Brush(self.GetBackgroundColour(), wx.SOLID))
- dc.Clear()
-
- if self.GetDiagram():
- self.GetDiagram().Redraw(dc)
-
- def OnMouseEvent(self, evt):
- dc = wx.ClientDC(self)
- self.PrepareDC(dc)
-
- x, y = evt.GetLogicalPosition(dc)
-
- keys = 0
- if evt.ShiftDown():
- keys |= KEY_SHIFT
- if evt.ControlDown():
- keys |= KEY_CTRL
-
- dragging = evt.Dragging()
-
- # Check if we're within the tolerance for mouse movements.
- # If we're very close to the position we started dragging
- # from, this may not be an intentional drag at all.
- if dragging:
- dx = abs(dc.LogicalToDeviceX(x-self._firstDragX))
- dy = abs(dc.LogicalToDeviceY(y-self._firstDragY))
- if self._checkTolerance and (dx <= self.GetDiagram().GetMouseTolerance()) and (dy <= self.GetDiagram().GetMouseTolerance()):
- return
- # If we've ignored the tolerance once, then ALWAYS ignore
- # tolerance in this drag, even if we come back within
- # the tolerance range.
- self._checkTolerance = False
-
- # Dragging - note that the effect of dragging is left entirely up
- # to the object, so no movement is done unless explicitly done by
- # object.
- if dragging and self._draggedShape and self._dragState == StartDraggingLeft:
- self._dragState = ContinueDraggingLeft
-
- # If the object isn't m_draggable, transfer message to canvas
- if self._draggedShape.Draggable():
- self._draggedShape.GetEventHandler().OnBeginDragLeft(x, y, keys, self._draggedAttachment)
- else:
- self._draggedShape = None
- self.OnBeginDragLeft(x, y, keys)
-
- self._oldDragX, self._oldDragY = x, y
-
- elif dragging and self._draggedShape and self._dragState == ContinueDraggingLeft:
- # Continue dragging
- self._draggedShape.GetEventHandler().OnDragLeft(False, self._oldDragX, self._oldDragY, keys, self._draggedAttachment)
- self._draggedShape.GetEventHandler().OnDragLeft(True, x, y, keys, self._draggedAttachment)
- self._oldDragX, self._oldDragY = x, y
-
- elif evt.LeftUp and self._draggedShape and self._dragState == ContinueDraggingLeft:
- self._dragState = NoDragging
- self._checkTolerance = True
-
- self._draggedShape.GetEventHandler().OnDragLeft(False, self._oldDragX, self._oldDragY, keys, self._draggedAttachment)
- self._draggedShape.GetEventHandler().OnEndDragLeft(x, y, keys, self._draggedAttachment)
- self._draggedShape = None
-
- elif dragging and self._draggedShape and self._dragState == StartDraggingRight:
- self._dragState = ContinueDraggingRight
- if self._draggedShape.Draggable:
- self._draggedShape.GetEventHandler().OnBeginDragRight(x, y, keys, self._draggedAttachment)
- else:
- self._draggedShape = None
- self.OnBeginDragRight(x, y, keys)
- self._oldDragX, self._oldDragY = x, y
-
- elif dragging and self._draggedShape and self._dragState == ContinueDraggingRight:
- # Continue dragging
- self._draggedShape.GetEventHandler().OnDragRight(False, self._oldDragX, self._oldDragY, keys, self._draggedAttachment)
- self._draggedShape.GetEventHandler().OnDragRight(True, x, y, keys, self._draggedAttachment)
- self._oldDragX, self._oldDragY = x, y
-
- elif evt.RightUp() and self._draggedShape and self._dragState == ContinueDraggingRight:
- self._dragState = NoDragging
- self._checkTolerance = True
-
- self._draggedShape.GetEventHandler().OnDragRight(False, self._oldDragX, self._oldDragY, keys, self._draggedAttachment)
- self._draggedShape.GetEventHandler().OnEndDragRight(x, y, keys, self._draggedAttachment)
- self._draggedShape = None
-
- # All following events sent to canvas, not object
- elif dragging and not self._draggedShape and self._dragState == StartDraggingLeft:
- self._dragState = ContinueDraggingLeft
- self.OnBeginDragLeft(x, y, keys)
- self._oldDragX, self._oldDragY = x, y
-
- elif dragging and not self._draggedShape and self._dragState == ContinueDraggingLeft:
- # Continue dragging
- self.OnDragLeft(False, self._oldDragX, self._oldDragY, keys)
- self.OnDragLeft(True, x, y, keys)
- self._oldDragX, self._oldDragY = x, y
-
- elif evt.LeftUp() and not self._draggedShape and self._dragState == ContinueDraggingLeft:
- self._dragState = NoDragging
- self._checkTolerance = True
-
- self.OnDragLeft(False, self._oldDragX, self._oldDragY, keys)
- self.OnEndDragLeft(x, y, keys)
- self._draggedShape = None
-
- elif dragging and not self._draggedShape and self._dragState == StartDraggingRight:
- self._dragState = ContinueDraggingRight
- self.OnBeginDragRight(x, y, keys)
- self._oldDragX, self._oldDragY = x, y
-
- elif dragging and not self._draggedShape and self._dragState == ContinueDraggingRight:
- # Continue dragging
- self.OnDragRight(False, self._oldDragX, self._oldDragY, keys)
- self.OnDragRight(True, x, y, keys)
- self._oldDragX, self._oldDragY = x, y
-
- elif evt.RightUp() and not self._draggedShape and self._dragState == ContinueDraggingRight:
- self._dragState = NoDragging
- self._checkTolerance = True
-
- self.OnDragRight(False, self._oldDragX, self._oldDragY, keys)
- self.OnEndDragRight(x, y, keys)
- self._draggedShape = None
-
- # Non-dragging events
- elif evt.IsButton():
- self._checkTolerance = True
-
- # Find the nearest object
- attachment = 0
-
- nearest_object, attachment = self.FindShape(x, y)
- if nearest_object: # Object event
- if evt.LeftDown():
- self._draggedShape = nearest_object
- self._draggedAttachment = attachment
- self._dragState = StartDraggingLeft
- self._firstDragX = x
- self._firstDragY = y
-
- elif evt.LeftUp():
- # N.B. Only register a click if the same object was
- # identified for down *and* up.
- if nearest_object == self._draggedShape:
- nearest_object.GetEventHandler().OnLeftClick(x, y, keys, attachment)
- self._draggedShape = None
- self._dragState = NoDragging
-
- elif evt.LeftDClick():
- nearest_object.GetEventHandler().OnLeftDoubleClick(x, y, keys, attachment)
- self._draggedShape = None
- self._dragState = NoDragging
-
- elif evt.RightDown():
- self._draggedShape = nearest_object
- self._draggedAttachment = attachment
- self._dragState = StartDraggingRight
- self._firstDragX = x
- self._firstDragY = y
-
- elif evt.RightUp():
- if nearest_object == self._draggedShape:
- nearest_object.GetEventHandler().OnRightClick(x, y, keys, attachment)
- self._draggedShape = None
- self._dragState = NoDragging
-
- else: # Canvas event
- if evt.LeftDown():
- self._draggedShape = None
- self._dragState = StartDraggingLeft
- self._firstDragX = x
- self._firstDragY = y
-
- elif evt.LeftUp():
- self.OnLeftClick(x, y, keys)
- self._draggedShape = None
- self._dragState = NoDragging
-
- elif evt.RightDown():
- self._draggedShape = None
- self._dragState = StartDraggingRight
- self._firstDragX = x
- self._firstDragY = y
-
- elif evt.RightUp():
- self.OnRightClick(x, y, keys)
- self._draggedShape = None
- self._dragState = NoDragging
-
- def FindShape(self, x, y, info = None, notObject = None):
- nearest = 100000.0
- nearest_attachment = 0
- nearest_object = None
-
- # Go backward through the object list, since we want:
- # (a) to have the control points drawn LAST to overlay
- # the other objects
- # (b) to find the control points FIRST if they exist
-
- for object in self.GetDiagram().GetShapeList()[::-1]:
- # First pass for lines, which might be inside a container, so we
- # want lines to take priority over containers. This first loop
- # could fail if we clickout side a line, so then we'll
- # try other shapes.
- if object.IsShown() and \
- isinstance(object, LineShape) and \
- object.HitTest(x, y) and \
- ((info == None) or isinstance(object, info)) and \
- (not notObject or not notObject.HasDescendant(object)):
- temp_attachment, dist = object.HitTest(x, y)
- # A line is trickier to spot than a normal object.
- # For a line, since it's the diagonal of the box
- # we use for the hit test, we may have several
- # lines in the box and therefore we need to be able
- # to specify the nearest point to the centre of the line
- # as our hit criterion, to give the user some room for
- # manouevre.
- if dist<nearest:
- nearest = dist
- nearest_object = object
- nearest_attachment = temp_attachment
-
- for object in self.GetDiagram().GetShapeList()[::-1]:
- # On second pass, only ever consider non-composites or
- # divisions. If children want to pass up control to
- # the composite, that's up to them.
- if (object.IsShown() and
- (isinstance(object, DivisionShape) or
- not isinstance(object, CompositeShape)) and
- object.HitTest(x, y) and
- (info == None or isinstance(object, info)) and
- (not notObject or not notObject.HasDescendant(object))):
- temp_attachment, dist = object.HitTest(x, y)
- if not isinstance(object, LineShape):
- # If we've hit a container, and we have already
- # found a line in the first pass, then ignore
- # the container in case the line is in the container.
- # Check for division in case line straddles divisions
- # (i.e. is not wholly contained).
- if not nearest_object or not (isinstance(object, DivisionShape) or WhollyContains(object, nearest_object)):
- nearest_object = object
- nearest_attachment = temp_attachment
- break
-
- return nearest_object, nearest_attachment
-
- def AddShape(self, object, addAfter = None):
- self.GetDiagram().AddShape(object, addAfter)
-
- def InsertShape(self, object):
- self.GetDiagram().InsertShape(object)
-
- def RemoveShape(self, object):
- self.GetDiagram().RemoveShape(object)
-
- def GetQuickEditMode(self):
- return self.GetDiagram().GetQuickEditMode()
-
- def Redraw(self, dc):
- self.GetDiagram().Redraw(dc)
-
- def Snap(self, x, y):
- return self.GetDiagram().Snap(x, y)
-
- def OnLeftClick(self, x, y, keys = 0):
- pass
-
- def OnRightClick(self, x, y, keys = 0):
- pass
-
- def OnDragLeft(self, draw, x, y, keys = 0):
- pass
-
- def OnBeginDragLeft(self, x, y, keys = 0):
- pass
-
- def OnEndDragLeft(self, x, y, keys = 0):
- pass
-
- def OnDragRight(self, draw, x, y, keys = 0):
- pass
-
- def OnBeginDragRight(self, x, y, keys = 0):
- pass
-
- def OnEndDragRight(self, x, y, keys = 0):
- pass
projects / wxWidgets.git / blobdiff