wxWidgets/wxPython/wx/lib/ogl/_basic.py
2006-08-19 19:22:23 +00:00

3186 lines
108 KiB
Python

# -*- coding: iso-8859-1 -*-
#----------------------------------------------------------------------------
# Name: basic.py
# Purpose: The basic OGL shapes
#
# 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
#----------------------------------------------------------------------------
import wx
import math
from _oglmisc import *
DragOffsetX = 0.0
DragOffsetY = 0.0
def OGLInitialize():
global WhiteBackgroundPen, WhiteBackgroundBrush, TransparentPen
global BlackForegroundPen, NormalFont
WhiteBackgroundPen = wx.Pen(wx.WHITE, 1, wx.SOLID)
WhiteBackgroundBrush = wx.Brush(wx.WHITE, wx.SOLID)
TransparentPen = wx.Pen(wx.WHITE, 1, wx.TRANSPARENT)
BlackForegroundPen = wx.Pen(wx.BLACK, 1, wx.SOLID)
NormalFont = wx.Font(10, wx.SWISS, wx.NORMAL, wx.NORMAL)
def OGLCleanUp():
pass
class ShapeTextLine(object):
def __init__(self, the_x, the_y, the_line):
self._x = the_x
self._y = the_y
self._line = the_line
def GetX(self):
return self._x
def GetY(self):
return self._y
def SetX(self, x):
self._x = x
def SetY(self, y):
self._y = y
def SetText(self, text):
self._line = text
def GetText(self):
return self._line
class ShapeEvtHandler(object):
def __init__(self, prev = None, shape = None):
self._previousHandler = prev
self._handlerShape = shape
def __del__(self):
pass
def SetShape(self, sh):
self._handlerShape = sh
def GetShape(self):
return self._handlerShape
def SetPreviousHandler(self, handler):
self._previousHandler = handler
def GetPreviousHandler(self):
return self._previousHandler
def OnDelete(self):
if self!=self.GetShape():
del self
def OnDraw(self, dc):
if self._previousHandler:
self._previousHandler.OnDraw(dc)
def OnMoveLinks(self, dc):
if self._previousHandler:
self._previousHandler.OnMoveLinks(dc)
def OnMoveLink(self, dc, moveControlPoints = True):
if self._previousHandler:
self._previousHandler.OnMoveLink(dc, moveControlPoints)
def OnDrawContents(self, dc):
if self._previousHandler:
self._previousHandler.OnDrawContents(dc)
def OnDrawBranches(self, dc, erase = False):
if self._previousHandler:
self._previousHandler.OnDrawBranches(dc, erase = erase)
def OnSize(self, x, y):
if self._previousHandler:
self._previousHandler.OnSize(x, y)
def OnMovePre(self, dc, x, y, old_x, old_y, display = True):
if self._previousHandler:
return self._previousHandler.OnMovePre(dc, x, y, old_x, old_y, display)
else:
return True
def OnMovePost(self, dc, x, y, old_x, old_y, display = True):
if self._previousHandler:
return self._previousHandler.OnMovePost(dc, x, y, old_x, old_y, display)
else:
return True
def OnErase(self, dc):
if self._previousHandler:
self._previousHandler.OnErase(dc)
def OnEraseContents(self, dc):
if self._previousHandler:
self._previousHandler.OnEraseContents(dc)
def OnHighlight(self, dc):
if self._previousHandler:
self._previousHandler.OnHighlight(dc)
def OnLeftClick(self, x, y, keys, attachment):
if self._previousHandler:
self._previousHandler.OnLeftClick(x, y, keys, attachment)
def OnLeftDoubleClick(self, x, y, keys = 0, attachment = 0):
if self._previousHandler:
self._previousHandler.OnLeftDoubleClick(x, y, keys, attachment)
def OnRightClick(self, x, y, keys = 0, attachment = 0):
if self._previousHandler:
self._previousHandler.OnRightClick(x, y, keys, attachment)
def OnDragLeft(self, draw, x, y, keys = 0, attachment = 0):
if self._previousHandler:
self._previousHandler.OnDragLeft(draw, x, y, keys, attachment)
def OnBeginDragLeft(self, x, y, keys = 0, attachment = 0):
if self._previousHandler:
self._previousHandler.OnBeginDragLeft(x, y, keys, attachment)
def OnEndDragLeft(self, x, y, keys = 0, attachment = 0):
if self._previousHandler:
self._previousHandler.OnEndDragLeft(x, y, keys, attachment)
def OnDragRight(self, draw, x, y, keys = 0, attachment = 0):
if self._previousHandler:
self._previousHandler.OnDragRight(draw, x, y, keys, attachment)
def OnBeginDragRight(self, x, y, keys = 0, attachment = 0):
if self._previousHandler:
self._previousHandler.OnBeginDragRight(x, y, keys, attachment)
def OnEndDragRight(self, x, y, keys = 0, attachment = 0):
if self._previousHandler:
self._previousHandler.OnEndDragRight(x, y, keys, attachment)
# Control points ('handles') redirect control to the actual shape,
# to make it easier to override sizing behaviour.
def OnSizingDragLeft(self, pt, draw, x, y, keys = 0, attachment = 0):
if self._previousHandler:
self._previousHandler.OnSizingDragLeft(pt, draw, x, y, keys, attachment)
def OnSizingBeginDragLeft(self, pt, x, y, keys = 0, attachment = 0):
if self._previousHandler:
self._previousHandler.OnSizingBeginDragLeft(pt, x, y, keys, attachment)
def OnSizingEndDragLeft(self, pt, x, y, keys = 0, attachment = 0):
if self._previousHandler:
self._previousHandler.OnSizingEndDragLeft(pt, x, y, keys, attachment)
def OnBeginSize(self, w, h):
pass
def OnEndSize(self, w, h):
pass
def OnDrawOutline(self, dc, x, y, w, h):
if self._previousHandler:
self._previousHandler.OnDrawOutline(dc, x, y, w, h)
def OnDrawControlPoints(self, dc):
if self._previousHandler:
self._previousHandler.OnDrawControlPoints(dc)
def OnEraseControlPoints(self, dc):
if self._previousHandler:
self._previousHandler.OnEraseControlPoints(dc)
# Can override this to prevent or intercept line reordering.
def OnChangeAttachment(self, attachment, line, ordering):
if self._previousHandler:
self._previousHandler.OnChangeAttachment(attachment, line, ordering)
class Shape(ShapeEvtHandler):
"""OGL base class
Shape(canvas = None)
The wxShape is the top-level, abstract object that all other objects
are derived from. All common functionality is represented by wxShape's
members, and overriden members that appear in derived classes and have
behaviour as documented for wxShape, are not documented separately.
"""
GraphicsInSizeToContents = False
def __init__(self, canvas = None):
ShapeEvtHandler.__init__(self)
self._eventHandler = self
self.SetShape(self)
self._id = 0
self._formatted = False
self._canvas = canvas
self._xpos = 0.0
self._ypos = 0.0
self._pen = BlackForegroundPen
self._brush = wx.WHITE_BRUSH
self._font = NormalFont
self._textColour = wx.BLACK
self._textColourName = wx.BLACK
self._visible = False
self._selected = False
self._attachmentMode = ATTACHMENT_MODE_NONE
self._spaceAttachments = True
self._disableLabel = False
self._fixedWidth = False
self._fixedHeight = False
self._drawHandles = True
self._sensitivity = OP_ALL
self._draggable = True
self._parent = None
self._formatMode = FORMAT_CENTRE_HORIZ | FORMAT_CENTRE_VERT
self._shadowMode = SHADOW_NONE
self._shadowOffsetX = 6
self._shadowOffsetY = 6
self._shadowBrush = wx.BLACK_BRUSH
self._textMarginX = 5
self._textMarginY = 5
self._regionName = "0"
self._centreResize = True
self._maintainAspectRatio = False
self._highlighted = False
self._rotation = 0.0
self._branchNeckLength = 10
self._branchStemLength = 10
self._branchSpacing = 10
self._branchStyle = BRANCHING_ATTACHMENT_NORMAL
self._regions = []
self._lines = []
self._controlPoints = []
self._attachmentPoints = []
self._text = []
self._children = []
# Set up a default region. Much of the above will be put into
# the region eventually (the duplication is for compatibility)
region = ShapeRegion()
region.SetName("0")
region.SetFont(NormalFont)
region.SetFormatMode(FORMAT_CENTRE_HORIZ | FORMAT_CENTRE_VERT)
region.SetColour("BLACK")
self._regions.append(region)
def __str__(self):
return "<%s.%s>" % (self.__class__.__module__, self.__class__.__name__)
def GetClassName(self):
return str(self.__class__).split(".")[-1][:-2]
def Delete(self):
"""
Fully disconnect this shape from parents, children, the
canvas, etc.
"""
if self._parent:
self._parent.GetChildren().remove(self)
for child in self.GetChildren():
child.Delete()
self.ClearText()
self.ClearRegions()
self.ClearAttachments()
self._handlerShape = None
if self._canvas:
self.RemoveFromCanvas(self._canvas)
if self.GetEventHandler():
self.GetEventHandler().OnDelete()
self._eventHandler = None
def __del__(self):
ShapeEvtHandler.__del__(self)
def Draggable(self):
"""TRUE if the shape may be dragged by the user."""
return True
def SetShape(self, sh):
self._handlerShape = sh
def GetCanvas(self):
"""Get the internal canvas."""
return self._canvas
def GetBranchStyle(self):
return self._branchStyle
def GetRotation(self):
"""Return the angle of rotation in radians."""
return self._rotation
def SetRotation(self, rotation):
self._rotation = rotation
def SetHighlight(self, hi, recurse = False):
"""Set the highlight for a shape. Shape highlighting is unimplemented."""
self._highlighted = hi
if recurse:
for shape in self._children:
shape.SetHighlight(hi, recurse)
def SetSensitivityFilter(self, sens = OP_ALL, recursive = False):
"""Set the shape to be sensitive or insensitive to specific mouse
operations.
sens is a bitlist of the following:
* OP_CLICK_LEFT
* OP_CLICK_RIGHT
* OP_DRAG_LEFT
* OP_DRAG_RIGHT
* OP_ALL (equivalent to a combination of all the above).
"""
self._draggable = sens & OP_DRAG_LEFT
self._sensitivity = sens
if recursive:
for shape in self._children:
shape.SetSensitivityFilter(sens, True)
def SetDraggable(self, drag, recursive = False):
"""Set the shape to be draggable or not draggable."""
self._draggable = drag
if drag:
self._sensitivity |= OP_DRAG_LEFT
elif self._sensitivity & OP_DRAG_LEFT:
self._sensitivity -= OP_DRAG_LEFT
if recursive:
for shape in self._children:
shape.SetDraggable(drag, True)
def SetDrawHandles(self, drawH):
"""Set the drawHandles flag for this shape and all descendants.
If drawH is TRUE (the default), any handles (control points) will
be drawn. Otherwise, the handles will not be drawn.
"""
self._drawHandles = drawH
for shape in self._children:
shape.SetDrawHandles(drawH)
def SetShadowMode(self, mode, redraw = False):
"""Set the shadow mode (whether a shadow is drawn or not).
mode can be one of the following:
SHADOW_NONE
No shadow (the default).
SHADOW_LEFT
Shadow on the left side.
SHADOW_RIGHT
Shadow on the right side.
"""
if redraw and self.GetCanvas():
dc = wx.ClientDC(self.GetCanvas())
self.GetCanvas().PrepareDC(dc)
self.Erase(dc)
self._shadowMode = mode
self.Draw(dc)
else:
self._shadowMode = mode
def GetShadowMode(self):
"""Return the current shadow mode setting"""
return self._shadowMode
def SetCanvas(self, theCanvas):
"""Identical to Shape.Attach."""
self._canvas = theCanvas
for shape in self._children:
shape.SetCanvas(theCanvas)
def AddToCanvas(self, theCanvas, addAfter = None):
"""Add the shape to the canvas's shape list.
If addAfter is non-NULL, will add the shape after this one.
"""
theCanvas.AddShape(self, addAfter)
lastImage = self
for object in self._children:
object.AddToCanvas(theCanvas, lastImage)
lastImage = object
def InsertInCanvas(self, theCanvas):
"""Insert the shape at the front of the shape list of canvas."""
theCanvas.InsertShape(self)
lastImage = self
for object in self._children:
object.AddToCanvas(theCanvas, lastImage)
lastImage = object
def RemoveFromCanvas(self, theCanvas):
"""Remove the shape from the canvas."""
if self.Selected():
self.Select(False)
self._canvas = None
theCanvas.RemoveShape(self)
for object in self._children:
object.RemoveFromCanvas(theCanvas)
def ClearAttachments(self):
"""Clear internal custom attachment point shapes (of class
wxAttachmentPoint).
"""
self._attachmentPoints = []
def ClearText(self, regionId = 0):
"""Clear the text from the specified text region."""
if regionId == 0:
self._text = ""
if regionId < len(self._regions):
self._regions[regionId].ClearText()
def ClearRegions(self):
"""Clear the ShapeRegions from the shape."""
self._regions = []
def AddRegion(self, region):
"""Add a region to the shape."""
self._regions.append(region)
def SetDefaultRegionSize(self):
"""Set the default region to be consistent with the shape size."""
if not self._regions:
return
w, h = self.GetBoundingBoxMax()
self._regions[0].SetSize(w, h)
def HitTest(self, x, y):
"""Given a point on a canvas, returns TRUE if the point was on the
shape, and returns the nearest attachment point and distance from
the given point and target.
"""
width, height = self.GetBoundingBoxMax()
if abs(width) < 4:
width = 4.0
if abs(height) < 4:
height = 4.0
width += 4 # Allowance for inaccurate mousing
height += 4
left = self._xpos - width / 2.0
top = self._ypos - height / 2.0
right = self._xpos + width / 2.0
bottom = self._ypos + height / 2.0
nearest_attachment = 0
# If within the bounding box, check the attachment points
# within the object.
if x >= left and x <= right and y >= top and y <= bottom:
n = self.GetNumberOfAttachments()
nearest = 999999
# GetAttachmentPosition[Edge] takes a logical attachment position,
# i.e. if it's rotated through 90%, position 0 is East-facing.
for i in range(n):
e = self.GetAttachmentPositionEdge(i)
if e:
xp, yp = e
l = math.sqrt(((xp - x) * (xp - x)) + (yp - y) * (yp - y))
if l < nearest:
nearest = l
nearest_attachment = i
return nearest_attachment, nearest
return False
# Format a text string according to the region size, adding
# strings with positions to region text list
def FormatText(self, dc, s, i = 0):
"""Reformat the given text region; defaults to formatting the
default region.
"""
self.ClearText(i)
if not self._regions:
return
if i > len(self._regions):
return
region = self._regions[i]
region._regionText = s
dc.SetFont(region.GetFont())
w, h = region.GetSize()
stringList = FormatText(dc, s, (w - 2 * self._textMarginX), (h - 2 * self._textMarginY), region.GetFormatMode())
for s in stringList:
line = ShapeTextLine(0.0, 0.0, s)
region.GetFormattedText().append(line)
actualW = w
actualH = h
# Don't try to resize an object with more than one image (this
# case should be dealt with by overriden handlers)
if (region.GetFormatMode() & FORMAT_SIZE_TO_CONTENTS) and \
len(region.GetFormattedText()) and \
len(self._regions) == 1 and \
not Shape.GraphicsInSizeToContents:
actualW, actualH = GetCentredTextExtent(dc, region.GetFormattedText())
if actualW + 2 * self._textMarginX != w or actualH + 2 * self._textMarginY != h:
# If we are a descendant of a composite, must make sure
# the composite gets resized properly
topAncestor = self.GetTopAncestor()
if topAncestor != self:
Shape.GraphicsInSizeToContents = True
composite = topAncestor
composite.Erase(dc)
self.SetSize(actualW + 2 * self._textMarginX, actualH + 2 * self._textMarginY)
self.Move(dc, self._xpos, self._ypos)
composite.CalculateSize()
if composite.Selected():
composite.DeleteControlPoints(dc)
composite.MakeControlPoints()
composite.MakeMandatoryControlPoints()
# Where infinite recursion might happen if we didn't stop it
composite.Draw(dc)
Shape.GraphicsInSizeToContents = False
else:
self.Erase(dc)
self.SetSize(actualW + 2 * self._textMarginX, actualH + 2 * self._textMarginY)
self.Move(dc, self._xpos, self._ypos)
self.EraseContents(dc)
CentreText(dc, region.GetFormattedText(), self._xpos, self._ypos, actualW - 2 * self._textMarginX, actualH - 2 * self._textMarginY, region.GetFormatMode())
self._formatted = True
def Recentre(self, dc):
"""Do recentring (or other formatting) for all the text regions
for this shape.
"""
w, h = self.GetBoundingBoxMin()
for region in self._regions:
CentreText(dc, region.GetFormattedText(), self._xpos, self._ypos, w - 2 * self._textMarginX, h - 2 * self._textMarginY, region.GetFormatMode())
def GetPerimeterPoint(self, x1, y1, x2, y2):
"""Get the point at which the line from (x1, y1) to (x2, y2) hits
the shape. Returns False if the line doesn't hit the perimeter.
"""
return False
def SetPen(self, the_pen):
"""Set the pen for drawing the shape's outline."""
self._pen = the_pen
def SetBrush(self, the_brush):
"""Set the brush for filling the shape's shape."""
self._brush = the_brush
# Get the top - most (non-division) ancestor, or self
def GetTopAncestor(self):
"""Return the top-most ancestor of this shape (the root of
the composite).
"""
if not self.GetParent():
return self
if isinstance(self.GetParent(), DivisionShape):
return self
return self.GetParent().GetTopAncestor()
# Region functions
def SetFont(self, the_font, regionId = 0):
"""Set the font for the specified text region."""
self._font = the_font
if regionId < len(self._regions):
self._regions[regionId].SetFont(the_font)
def GetFont(self, regionId = 0):
"""Get the font for the specified text region."""
if regionId >= len(self._regions):
return None
return self._regions[regionId].GetFont()
def SetFormatMode(self, mode, regionId = 0):
"""Set the format mode of the default text region. The argument
can be a bit list of the following:
FORMAT_NONE
No formatting.
FORMAT_CENTRE_HORIZ
Horizontal centring.
FORMAT_CENTRE_VERT
Vertical centring.
"""
if regionId < len(self._regions):
self._regions[regionId].SetFormatMode(mode)
def GetFormatMode(self, regionId = 0):
if regionId >= len(self._regions):
return 0
return self._regions[regionId].GetFormatMode()
def SetTextColour(self, the_colour, regionId = 0):
"""Set the colour for the specified text region."""
self._textColour = wx.TheColourDatabase.Find(the_colour)
self._textColourName = the_colour
if regionId < len(self._regions):
self._regions[regionId].SetColour(the_colour)
def GetTextColour(self, regionId = 0):
"""Get the colour for the specified text region."""
if regionId >= len(self._regions):
return ""
return self._regions[regionId].GetColour()
def SetRegionName(self, name, regionId = 0):
"""Set the name for this region.
The name for a region is unique within the scope of the whole
composite, whereas a region id is unique only for a single image.
"""
if regionId < len(self._regions):
self._regions[regionId].SetName(name)
def GetRegionName(self, regionId = 0):
"""Get the region's name.
A region's name can be used to uniquely determine a region within
an entire composite image hierarchy. See also Shape.SetRegionName.
"""
if regionId >= len(self._regions):
return ""
return self._regions[regionId].GetName()
def GetRegionId(self, name):
"""Get the region's identifier by name.
This is not unique for within an entire composite, but is unique
for the image.
"""
for i, r in enumerate(self._regions):
if r.GetName() == name:
return i
return -1
# Name all _regions in all subimages recursively
def NameRegions(self, parentName=""):
"""Make unique names for all the regions in a shape or composite shape."""
n = self.GetNumberOfTextRegions()
for i in range(n):
if parentName:
buff = parentName+"."+str(i)
else:
buff = str(i)
self.SetRegionName(buff, i)
for j, child in enumerate(self._children):
if parentName:
buff = parentName+"."+str(j)
else:
buff = str(j)
child.NameRegions(buff)
# Get a region by name, possibly looking recursively into composites
def FindRegion(self, name):
"""Find the actual image ('this' if non-composite) and region id
for the given region name.
"""
id = self.GetRegionId(name)
if id > -1:
return self, id
for child in self._children:
actualImage, regionId = child.FindRegion(name)
if actualImage:
return actualImage, regionId
return None, -1
# Finds all region names for this image (composite or simple).
def FindRegionNames(self):
"""Get a list of all region names for this image (composite or simple)."""
list = []
n = self.GetNumberOfTextRegions()
for i in range(n):
list.append(self.GetRegionName(i))
for child in self._children:
list += child.FindRegionNames()
return list
def AssignNewIds(self):
"""Assign new ids to this image and its children."""
self._id = wx.NewId()
for child in self._children:
child.AssignNewIds()
def OnDraw(self, dc):
pass
def OnMoveLinks(self, dc):
# Want to set the ends of all attached links
# to point to / from this object
for line in self._lines:
line.GetEventHandler().OnMoveLink(dc)
def OnDrawContents(self, dc):
if not self._regions:
return
bound_x, bound_y = self.GetBoundingBoxMin()
if self._pen:
dc.SetPen(self._pen)
for region in self._regions:
if region.GetFont():
dc.SetFont(region.GetFont())
dc.SetTextForeground(region.GetActualColourObject())
dc.SetBackgroundMode(wx.TRANSPARENT)
if not self._formatted:
CentreText(dc, region.GetFormattedText(), self._xpos, self._ypos, bound_x - 2 * self._textMarginX, bound_y - 2 * self._textMarginY, region.GetFormatMode())
self._formatted = True
if not self.GetDisableLabel():
DrawFormattedText(dc, region.GetFormattedText(), self._xpos, self._ypos, bound_x - 2 * self._textMarginX, bound_y - 2 * self._textMarginY, region.GetFormatMode())
def DrawContents(self, dc):
"""Draw the internal graphic of the shape (such as text).
Do not override this function: override OnDrawContents, which
is called by this function.
"""
self.GetEventHandler().OnDrawContents(dc)
def OnSize(self, x, y):
pass
def OnMovePre(self, dc, x, y, old_x, old_y, display = True):
return True
def OnErase(self, dc):
if not self._visible:
return
# Erase links
for line in self._lines:
line.GetEventHandler().OnErase(dc)
self.GetEventHandler().OnEraseContents(dc)
def OnEraseContents(self, dc):
if not self._visible:
return
xp, yp = self.GetX(), self.GetY()
minX, minY = self.GetBoundingBoxMin()
maxX, maxY = self.GetBoundingBoxMax()
topLeftX = xp - maxX / 2.0 - 2
topLeftY = yp - maxY / 2.0 - 2
penWidth = 0
if self._pen:
penWidth = self._pen.GetWidth()
dc.SetPen(self.GetBackgroundPen())
dc.SetBrush(self.GetBackgroundBrush())
dc.DrawRectangle(topLeftX - penWidth, topLeftY - penWidth, maxX + penWidth * 2 + 4, maxY + penWidth * 2 + 4)
def EraseLinks(self, dc, attachment = -1, recurse = False):
"""Erase links attached to this shape, but do not repair damage
caused to other shapes.
"""
if not self._visible:
return
for line in self._lines:
if attachment == -1 or (line.GetTo() == self and line.GetAttachmentTo() == attachment or line.GetFrom() == self and line.GetAttachmentFrom() == attachment):
line.GetEventHandler().OnErase(dc)
if recurse:
for child in self._children:
child.EraseLinks(dc, attachment, recurse)
def DrawLinks(self, dc, attachment = -1, recurse = False):
"""Draws any lines linked to this shape."""
if not self._visible:
return
for line in self._lines:
if attachment == -1 or (line.GetTo() == self and line.GetAttachmentTo() == attachment or line.GetFrom() == self and line.GetAttachmentFrom() == attachment):
line.Draw(dc)
if recurse:
for child in self._children:
child.DrawLinks(dc, attachment, recurse)
# Returns TRUE if pt1 <= pt2 in the sense that one point comes before
# another on an edge of the shape.
# attachmentPoint is the attachment point (= side) in question.
# This is the default, rectangular implementation.
def AttachmentSortTest(self, attachmentPoint, pt1, pt2):
"""Return TRUE if pt1 is less than or equal to pt2, in the sense
that one point comes before another on an edge of the shape.
attachment is the attachment point (side) in question.
This function is used in Shape.MoveLineToNewAttachment to determine
the new line ordering.
"""
physicalAttachment = self.LogicalToPhysicalAttachment(attachmentPoint)
if physicalAttachment in [0, 2]:
return pt1[0] <= pt2[0]
elif physicalAttachment in [1, 3]:
return pt1[1] <= pt2[1]
return False
def MoveLineToNewAttachment(self, dc, to_move, x, y):
"""Move the given line (which must already be attached to the shape)
to a different attachment point on the shape, or a different order
on the same attachment.
Calls Shape.AttachmentSortTest and then
ShapeEvtHandler.OnChangeAttachment.
"""
if self.GetAttachmentMode() == ATTACHMENT_MODE_NONE:
return False
# Is (x, y) on this object? If so, find the new attachment point
# the user has moved the point to
hit = self.HitTest(x, y)
if not hit:
return False
newAttachment, distance = hit
self.EraseLinks(dc)
if to_move.GetTo() == self:
oldAttachment = to_move.GetAttachmentTo()
else:
oldAttachment = to_move.GetAttachmentFrom()
# The links in a new ordering
# First, add all links to the new list
newOrdering = self._lines[:]
# Delete the line object from the list of links; we're going to move
# it to another position in the list
del newOrdering[newOrdering.index(to_move)]
old_x = -99999.9
old_y = -99999.9
found = False
for line in newOrdering:
if line.GetTo() == self and oldAttachment == line.GetAttachmentTo() or \
line.GetFrom() == self and oldAttachment == line.GetAttachmentFrom():
startX, startY, endX, endY = line.GetEnds()
if line.GetTo() == self:
xp = endX
yp = endY
else:
xp = startX
yp = startY
thisPoint = wx.RealPoint(xp, yp)
lastPoint = wx.RealPoint(old_x, old_y)
newPoint = wx.RealPoint(x, y)
if self.AttachmentSortTest(newAttachment, newPoint, thisPoint) and self.AttachmentSortTest(newAttachment, lastPoint, newPoint):
found = True
newOrdering.insert(newOrdering.index(line), to_move)
old_x = xp
old_y = yp
if found:
break
if not found:
newOrdering.append(to_move)
self.GetEventHandler().OnChangeAttachment(newAttachment, to_move, newOrdering)
return True
def OnChangeAttachment(self, attachment, line, ordering):
if line.GetTo() == self:
line.SetAttachmentTo(attachment)
else:
line.SetAttachmentFrom(attachment)
self.ApplyAttachmentOrdering(ordering)
dc = wx.ClientDC(self.GetCanvas())
self.GetCanvas().PrepareDC(dc)
self.MoveLinks(dc)
if not self.GetCanvas().GetQuickEditMode():
self.GetCanvas().Redraw(dc)
# Reorders the lines according to the given list
def ApplyAttachmentOrdering(self, linesToSort):
"""Apply the line ordering in linesToSort to the shape, to reorder
the way lines are attached.
"""
linesStore = self._lines[:]
self._lines = []
for line in linesToSort:
if line in linesStore:
del linesStore[linesStore.index(line)]
self._lines.append(line)
# Now add any lines that haven't been listed in linesToSort
self._lines += linesStore
def SortLines(self, attachment, linesToSort):
""" Reorder the lines coming into the node image at this attachment
position, in the order in which they appear in linesToSort.
Any remaining lines not in the list will be added to the end.
"""
# This is a temporary store of all the lines at this attachment
# point. We'll tick them off as we've processed them.
linesAtThisAttachment = []
for line in self._lines[:]:
if line.GetTo() == self and line.GetAttachmentTo() == attachment or \
line.GetFrom() == self and line.GetAttachmentFrom() == attachment:
linesAtThisAttachment.append(line)
del self._lines[self._lines.index(line)]
for line in linesToSort:
if line in linesAtThisAttachment:
# Done this one
del linesAtThisAttachment[linesAtThisAttachment.index(line)]
self._lines.append(line)
# Now add any lines that haven't been listed in linesToSort
self._lines += linesAtThisAttachment
def OnHighlight(self, dc):
pass
def OnLeftClick(self, x, y, keys = 0, attachment = 0):
if self._sensitivity & OP_CLICK_LEFT != OP_CLICK_LEFT:
if self._parent:
attachment, dist = self._parent.HitTest(x, y)
self._parent.GetEventHandler().OnLeftClick(x, y, keys, attachment)
def OnRightClick(self, x, y, keys = 0, attachment = 0):
if self._sensitivity & OP_CLICK_RIGHT != OP_CLICK_RIGHT:
attachment, dist = self._parent.HitTest(x, y)
self._parent.GetEventHandler().OnRightClick(x, y, keys, attachment)
def OnDragLeft(self, draw, x, y, keys = 0, attachment = 0):
if self._sensitivity & OP_DRAG_LEFT != OP_DRAG_LEFT:
if self._parent:
hit = self._parent.HitTest(x, y)
if hit:
attachment, dist = hit
self._parent.GetEventHandler().OnDragLeft(draw, x, y, keys, attachment)
return
dc = wx.ClientDC(self.GetCanvas())
self.GetCanvas().PrepareDC(dc)
dc.SetLogicalFunction(OGLRBLF)
dottedPen = wx.Pen(wx.Colour(0, 0, 0), 1, wx.DOT)
dc.SetPen(dottedPen)
dc.SetBrush(wx.TRANSPARENT_BRUSH)
xx = x + DragOffsetX
yy = y + DragOffsetY
xx, yy = self._canvas.Snap(xx, yy)
w, h = self.GetBoundingBoxMax()
self.GetEventHandler().OnDrawOutline(dc, xx, yy, w, h)
def OnBeginDragLeft(self, x, y, keys = 0, attachment = 0):
global DragOffsetX, DragOffsetY
if self._sensitivity & OP_DRAG_LEFT != OP_DRAG_LEFT:
if self._parent:
hit = self._parent.HitTest(x, y)
if hit:
attachment, dist = hit
self._parent.GetEventHandler().OnBeginDragLeft(x, y, keys, attachment)
return
DragOffsetX = self._xpos - x
DragOffsetY = self._ypos - y
dc = wx.ClientDC(self.GetCanvas())
self.GetCanvas().PrepareDC(dc)
# New policy: don't erase shape until end of drag.
# self.Erase(dc)
xx = x + DragOffsetX
yy = y + DragOffsetY
xx, yy = self._canvas.Snap(xx, yy)
dc.SetLogicalFunction(OGLRBLF)
dottedPen = wx.Pen(wx.Colour(0, 0, 0), 1, wx.DOT)
dc.SetPen(dottedPen)
dc.SetBrush(wx.TRANSPARENT_BRUSH)
w, h = self.GetBoundingBoxMax()
self.GetEventHandler().OnDrawOutline(dc, xx, yy, w, h)
self._canvas.CaptureMouse()
def OnEndDragLeft(self, x, y, keys = 0, attachment = 0):
if self._canvas.HasCapture():
self._canvas.ReleaseMouse()
if self._sensitivity & OP_DRAG_LEFT != OP_DRAG_LEFT:
if self._parent:
hit = self._parent.HitTest(x, y)
if hit:
attachment, dist = hit
self._parent.GetEventHandler().OnEndDragLeft(x, y, keys, attachment)
return
dc = wx.ClientDC(self.GetCanvas())
self.GetCanvas().PrepareDC(dc)
dc.SetLogicalFunction(wx.COPY)
xx = x + DragOffsetX
yy = y + DragOffsetY
xx, yy = self._canvas.Snap(xx, yy)
# New policy: erase shape at end of drag.
self.Erase(dc)
self.Move(dc, xx, yy)
if self._canvas and not self._canvas.GetQuickEditMode():
self._canvas.Redraw(dc)
def OnDragRight(self, draw, x, y, keys = 0, attachment = 0):
if self._sensitivity & OP_DRAG_RIGHT != OP_DRAG_RIGHT:
if self._parent:
attachment, dist = self._parent.HitTest(x, y)
self._parent.GetEventHandler().OnDragRight(draw, x, y, keys, attachment)
return
def OnBeginDragRight(self, x, y, keys = 0, attachment = 0):
if self._sensitivity & OP_DRAG_RIGHT != OP_DRAG_RIGHT:
if self._parent:
attachment, dist = self._parent.HitTest(x, y)
self._parent.GetEventHandler().OnBeginDragRight(x, y, keys, attachment)
return
def OnEndDragRight(self, x, y, keys = 0, attachment = 0):
if self._sensitivity & OP_DRAG_RIGHT != OP_DRAG_RIGHT:
if self._parent:
attachment, dist = self._parent.HitTest(x, y)
self._parent.GetEventHandler().OnEndDragRight(x, y, keys, attachment)
return
def OnDrawOutline(self, dc, x, y, w, h):
points = [[x - w / 2.0, y - h / 2.0],
[x + w / 2.0, y - h / 2.0],
[x + w / 2.0, y + h / 2.0],
[x - w / 2.0, y + h / 2.0],
[x - w / 2.0, y - h / 2.0],
]
dc.DrawLines(points)
def Attach(self, can):
"""Set the shape's internal canvas pointer to point to the given canvas."""
self._canvas = can
def Detach(self):
"""Disassociates the shape from its canvas."""
self._canvas = None
def Move(self, dc, x, y, display = True):
"""Move the shape to the given position.
Redraw if display is TRUE.
"""
old_x = self._xpos
old_y = self._ypos
if not self.GetEventHandler().OnMovePre(dc, x, y, old_x, old_y, display):
return
self._xpos, self._ypos = x, y
self.ResetControlPoints()
if display:
self.Draw(dc)
self.MoveLinks(dc)
self.GetEventHandler().OnMovePost(dc, x, y, old_x, old_y, display)
def MoveLinks(self, dc):
"""Redraw all the lines attached to the shape."""
self.GetEventHandler().OnMoveLinks(dc)
def Draw(self, dc):
"""Draw the whole shape and any lines attached to it.
Do not override this function: override OnDraw, which is called
by this function.
"""
if self._visible:
self.GetEventHandler().OnDraw(dc)
self.GetEventHandler().OnDrawContents(dc)
self.GetEventHandler().OnDrawControlPoints(dc)
self.GetEventHandler().OnDrawBranches(dc)
def Flash(self):
"""Flash the shape."""
if self.GetCanvas():
dc = wx.ClientDC(self.GetCanvas())
self.GetCanvas().PrepareDC(dc)
dc.SetLogicalFunction(OGLRBLF)
self.Draw(dc)
dc.SetLogicalFunction(wx.COPY)
self.Draw(dc)
def Show(self, show):
"""Set a flag indicating whether the shape should be drawn."""
self._visible = show
for child in self._children:
child.Show(show)
def Erase(self, dc):
"""Erase the shape.
Does not repair damage caused to other shapes.
"""
self.GetEventHandler().OnErase(dc)
self.GetEventHandler().OnEraseControlPoints(dc)
self.GetEventHandler().OnDrawBranches(dc, erase = True)
def EraseContents(self, dc):
"""Erase the shape contents, that is, the area within the shape's
minimum bounding box.
"""
self.GetEventHandler().OnEraseContents(dc)
def AddText(self, string):
"""Add a line of text to the shape's default text region."""
if not self._regions:
return
region = self._regions[0]
#region.ClearText()
new_line = ShapeTextLine(0, 0, string)
text = region.GetFormattedText()
text.append(new_line)
self._formatted = False
def SetSize(self, x, y, recursive = True):
"""Set the shape's size."""
self.SetAttachmentSize(x, y)
self.SetDefaultRegionSize()
def SetAttachmentSize(self, w, h):
width, height = self.GetBoundingBoxMin()
if width == 0:
scaleX = 1.0
else:
scaleX = float(w) / width
if height == 0:
scaleY = 1.0
else:
scaleY = float(h) / height
for point in self._attachmentPoints:
point._x = point._x * scaleX
point._y = point._y * scaleY
# Add line FROM this object
def AddLine(self, line, other, attachFrom = 0, attachTo = 0, positionFrom = -1, positionTo = -1):
"""Add a line between this shape and the given other shape, at the
specified attachment points.
The position in the list of lines at each end can also be specified,
so that the line will be drawn at a particular point on its attachment
point.
"""
if positionFrom == -1:
if not line in self._lines:
self._lines.append(line)
else:
# Don't preserve old ordering if we have new ordering instructions
try:
self._lines.remove(line)
except ValueError:
pass
if positionFrom < len(self._lines):
self._lines.insert(positionFrom, line)
else:
self._lines.append(line)
if positionTo == -1:
if not other in other._lines:
other._lines.append(line)
else:
# Don't preserve old ordering if we have new ordering instructions
try:
other._lines.remove(line)
except ValueError:
pass
if positionTo < len(other._lines):
other._lines.insert(positionTo, line)
else:
other._lines.append(line)
line.SetFrom(self)
line.SetTo(other)
line.SetAttachments(attachFrom, attachTo)
dc = wx.ClientDC(self._canvas)
self._canvas.PrepareDC(dc)
self.MoveLinks(dc)
def RemoveLine(self, line):
"""Remove the given line from the shape's list of attached lines."""
if line.GetFrom() == self:
line.GetTo()._lines.remove(line)
else:
line.GetFrom()._lines.remove(line)
self._lines.remove(line)
# Default - make 6 control points
def MakeControlPoints(self):
"""Make a list of control points (draggable handles) appropriate to
the shape.
"""
maxX, maxY = self.GetBoundingBoxMax()
minX, minY = self.GetBoundingBoxMin()
widthMin = minX + CONTROL_POINT_SIZE + 2
heightMin = minY + CONTROL_POINT_SIZE + 2
# Offsets from main object
top = -heightMin / 2.0
bottom = heightMin / 2.0 + (maxY - minY)
left = -widthMin / 2.0
right = widthMin / 2.0 + (maxX - minX)
control = ControlPoint(self._canvas, self, CONTROL_POINT_SIZE, left, top, CONTROL_POINT_DIAGONAL)
self._canvas.AddShape(control)
self._controlPoints.append(control)
control = ControlPoint(self._canvas, self, CONTROL_POINT_SIZE, 0, top, CONTROL_POINT_VERTICAL)
self._canvas.AddShape(control)
self._controlPoints.append(control)
control = ControlPoint(self._canvas, self, CONTROL_POINT_SIZE, right, top, CONTROL_POINT_DIAGONAL)
self._canvas.AddShape(control)
self._controlPoints.append(control)
control = ControlPoint(self._canvas, self, CONTROL_POINT_SIZE, right, 0, CONTROL_POINT_HORIZONTAL)
self._canvas.AddShape(control)
self._controlPoints.append(control)
control = ControlPoint(self._canvas, self, CONTROL_POINT_SIZE, right, bottom, CONTROL_POINT_DIAGONAL)
self._canvas.AddShape(control)
self._controlPoints.append(control)
control = ControlPoint(self._canvas, self, CONTROL_POINT_SIZE, 0, bottom, CONTROL_POINT_VERTICAL)
self._canvas.AddShape(control)
self._controlPoints.append(control)
control = ControlPoint(self._canvas, self, CONTROL_POINT_SIZE, left, bottom, CONTROL_POINT_DIAGONAL)
self._canvas.AddShape(control)
self._controlPoints.append(control)
control = ControlPoint(self._canvas, self, CONTROL_POINT_SIZE, left, 0, CONTROL_POINT_HORIZONTAL)
self._canvas.AddShape(control)
self._controlPoints.append(control)
def MakeMandatoryControlPoints(self):
"""Make the mandatory control points.
For example, the control point on a dividing line should appear even
if the divided rectangle shape's handles should not appear (because
it is the child of a composite, and children are not resizable).
"""
for child in self._children:
child.MakeMandatoryControlPoints()
def ResetMandatoryControlPoints(self):
"""Reset the mandatory control points."""
for child in self._children:
child.ResetMandatoryControlPoints()
def ResetControlPoints(self):
"""Reset the positions of the control points (for instance when the
shape's shape has changed).
"""
self.ResetMandatoryControlPoints()
if len(self._controlPoints) == 0:
return
maxX, maxY = self.GetBoundingBoxMax()
minX, minY = self.GetBoundingBoxMin()
widthMin = minX + CONTROL_POINT_SIZE + 2
heightMin = minY + CONTROL_POINT_SIZE + 2
# Offsets from main object
top = -heightMin / 2.0
bottom = heightMin / 2.0 + (maxY - minY)
left = -widthMin / 2.0
right = widthMin / 2.0 + (maxX - minX)
self._controlPoints[0]._xoffset = left
self._controlPoints[0]._yoffset = top
self._controlPoints[1]._xoffset = 0
self._controlPoints[1]._yoffset = top
self._controlPoints[2]._xoffset = right
self._controlPoints[2]._yoffset = top
self._controlPoints[3]._xoffset = right
self._controlPoints[3]._yoffset = 0
self._controlPoints[4]._xoffset = right
self._controlPoints[4]._yoffset = bottom
self._controlPoints[5]._xoffset = 0
self._controlPoints[5]._yoffset = bottom
self._controlPoints[6]._xoffset = left
self._controlPoints[6]._yoffset = bottom
self._controlPoints[7]._xoffset = left
self._controlPoints[7]._yoffset = 0
def DeleteControlPoints(self, dc = None):
"""Delete the control points (or handles) for the shape.
Does not redraw the shape.
"""
for control in self._controlPoints[:]:
if dc:
control.GetEventHandler().OnErase(dc)
control.Delete()
self._controlPoints.remove(control)
self._controlPoints = []
# Children of divisions are contained objects,
# so stop here
if not isinstance(self, DivisionShape):
for child in self._children:
child.DeleteControlPoints(dc)
def OnDrawControlPoints(self, dc):
if not self._drawHandles:
return
dc.SetBrush(wx.BLACK_BRUSH)
dc.SetPen(wx.BLACK_PEN)
for control in self._controlPoints:
control.Draw(dc)
# Children of divisions are contained objects,
# so stop here.
# This test bypasses the type facility for speed
# (critical when drawing)
if not isinstance(self, DivisionShape):
for child in self._children:
child.GetEventHandler().OnDrawControlPoints(dc)
def OnEraseControlPoints(self, dc):
for control in self._controlPoints:
control.Erase(dc)
if not isinstance(self, DivisionShape):
for child in self._children:
child.GetEventHandler().OnEraseControlPoints(dc)
def Select(self, select, dc = None):
"""Select or deselect the given shape, drawing or erasing control points
(handles) as necessary.
"""
self._selected = select
if select:
self.MakeControlPoints()
# Children of divisions are contained objects,
# so stop here
if not isinstance(self, DivisionShape):
for child in self._children:
child.MakeMandatoryControlPoints()
if dc:
self.GetEventHandler().OnDrawControlPoints(dc)
else:
self.DeleteControlPoints(dc)
if not isinstance(self, DivisionShape):
for child in self._children:
child.DeleteControlPoints(dc)
def Selected(self):
"""TRUE if the shape is currently selected."""
return self._selected
def AncestorSelected(self):
"""TRUE if the shape's ancestor is currently selected."""
if self._selected:
return True
if not self.GetParent():
return False
return self.GetParent().AncestorSelected()
def GetNumberOfAttachments(self):
"""Get the number of attachment points for this shape."""
# Should return the MAXIMUM attachment point id here,
# so higher-level functions can iterate through all attachments,
# even if they're not contiguous.
if len(self._attachmentPoints) == 0:
return 4
else:
maxN = 3
for point in self._attachmentPoints:
if point._id > maxN:
maxN = point._id
return maxN + 1
def AttachmentIsValid(self, attachment):
"""TRUE if attachment is a valid attachment point."""
if len(self._attachmentPoints) == 0:
return attachment in range(4)
for point in self._attachmentPoints:
if point._id == attachment:
return True
return False
def GetAttachmentPosition(self, attachment, nth = 0, no_arcs = 1, line = None):
"""Get the position at which the given attachment point should be drawn.
If attachment isn't found among the attachment points of the shape,
returns None.
"""
if self._attachmentMode == ATTACHMENT_MODE_NONE:
return self._xpos, self._ypos
elif self._attachmentMode == ATTACHMENT_MODE_BRANCHING:
pt, stemPt = self.GetBranchingAttachmentPoint(attachment, nth)
return pt[0], pt[1]
elif self._attachmentMode == ATTACHMENT_MODE_EDGE:
if len(self._attachmentPoints):
for point in self._attachmentPoints:
if point._id == attachment:
return self._xpos + point._x, self._ypos + point._y
return None
else:
# Assume is rectangular
w, h = self.GetBoundingBoxMax()
top = self._ypos + h / 2.0
bottom = self._ypos - h / 2.0
left = self._xpos - w / 2.0
right = self._xpos + w / 2.0
# wtf?
line and line.IsEnd(self)
physicalAttachment = self.LogicalToPhysicalAttachment(attachment)
# Simplified code
if physicalAttachment == 0:
pt = self.CalcSimpleAttachment((left, bottom), (right, bottom), nth, no_arcs, line)
elif physicalAttachment == 1:
pt = self.CalcSimpleAttachment((right, bottom), (right, top), nth, no_arcs, line)
elif physicalAttachment == 2:
pt = self.CalcSimpleAttachment((left, top), (right, top), nth, no_arcs, line)
elif physicalAttachment == 3:
pt = self.CalcSimpleAttachment((left, bottom), (left, top), nth, no_arcs, line)
else:
return None
return pt[0], pt[1]
return None
def GetBoundingBoxMax(self):
"""Get the maximum bounding box for the shape, taking into account
external features such as shadows.
"""
ww, hh = self.GetBoundingBoxMin()
if self._shadowMode != SHADOW_NONE:
ww += self._shadowOffsetX
hh += self._shadowOffsetY
return ww, hh
def GetBoundingBoxMin(self):
"""Get the minimum bounding box for the shape, that defines the area
available for drawing the contents (such as text).
Must be overridden.
"""
return 0, 0
def HasDescendant(self, image):
"""TRUE if image is a descendant of this composite."""
if image == self:
return True
for child in self._children:
if child.HasDescendant(image):
return True
return False
# Assuming the attachment lies along a vertical or horizontal line,
# calculate the position on that point.
def CalcSimpleAttachment(self, pt1, pt2, nth, noArcs, line):
"""Assuming the attachment lies along a vertical or horizontal line,
calculate the position on that point.
Parameters:
pt1
The first point of the line repesenting the edge of the shape.
pt2
The second point of the line representing the edge of the shape.
nth
The position on the edge (for example there may be 6 lines at
this attachment point, and this may be the 2nd line.
noArcs
The number of lines at this edge.
line
The line shape.
Remarks
This function expects the line to be either vertical or horizontal,
and determines which.
"""
isEnd = line and line.IsEnd(self)
# Are we horizontal or vertical?
isHorizontal = RoughlyEqual(pt1[1], pt2[1])
if isHorizontal:
if pt1[0] > pt2[0]:
firstPoint = pt2
secondPoint = pt1
else:
firstPoint = pt1
secondPoint = pt2
if self._spaceAttachments:
if line and line.GetAlignmentType(isEnd) == LINE_ALIGNMENT_TO_NEXT_HANDLE:
# Align line according to the next handle along
point = line.GetNextControlPoint(self)
if point[0] < firstPoint[0]:
x = firstPoint[0]
elif point[0] > secondPoint[0]:
x = secondPoint[0]
else:
x = point[0]
else:
x = firstPoint[0] + (nth + 1) * (secondPoint[0] - firstPoint[0]) / (noArcs + 1.0)
else:
x = (secondPoint[0] - firstPoint[0]) / 2.0 # Midpoint
y = pt1[1]
else:
assert RoughlyEqual(pt1[0], pt2[0])
if pt1[1] > pt2[1]:
firstPoint = pt2
secondPoint = pt1
else:
firstPoint = pt1
secondPoint = pt2
if self._spaceAttachments:
if line and line.GetAlignmentType(isEnd) == LINE_ALIGNMENT_TO_NEXT_HANDLE:
# Align line according to the next handle along
point = line.GetNextControlPoint(self)
if point[1] < firstPoint[1]:
y = firstPoint[1]
elif point[1] > secondPoint[1]:
y = secondPoint[1]
else:
y = point[1]
else:
y = firstPoint[1] + (nth + 1) * (secondPoint[1] - firstPoint[1]) / (noArcs + 1.0)
else:
y = (secondPoint[1] - firstPoint[1]) / 2.0 # Midpoint
x = pt1[0]
return x, y
# Return the zero-based position in m_lines of line
def GetLinePosition(self, line):
"""Get the zero-based position of line in the list of lines
for this shape.
"""
try:
return self._lines.index(line)
except:
return 0
# |________|
# | <- root
# | <- neck
# shoulder1 ->---------<- shoulder2
# | | | | |
# <- branching attachment point N-1
def GetBranchingAttachmentInfo(self, attachment):
"""Get information about where branching connections go.
Returns FALSE if there are no lines at this attachment.
"""
physicalAttachment = self.LogicalToPhysicalAttachment(attachment)
# Number of lines at this attachment
lineCount = self.GetAttachmentLineCount(attachment)
if not lineCount:
return False
totalBranchLength = self._branchSpacing * (lineCount - 1)
root = self.GetBranchingAttachmentRoot(attachment)
neck = wx.RealPoint()
shoulder1 = wx.RealPoint()
shoulder2 = wx.RealPoint()
# Assume that we have attachment points 0 to 3: top, right, bottom, left
if physicalAttachment == 0:
neck[0] = self.GetX()
neck[1] = root[1] - self._branchNeckLength
shoulder1[0] = root[0] - totalBranchLength / 2.0
shoulder2[0] = root[0] + totalBranchLength / 2.0
shoulder1[1] = neck[1]
shoulder2[1] = neck[1]
elif physicalAttachment == 1:
neck[0] = root[0] + self._branchNeckLength
neck[1] = root[1]
shoulder1[0] = neck[0]
shoulder2[0] = neck[0]
shoulder1[1] = neck[1] - totalBranchLength / 2.0
shoulder1[1] = neck[1] + totalBranchLength / 2.0
elif physicalAttachment == 2:
neck[0] = self.GetX()
neck[1] = root[1] + self._branchNeckLength
shoulder1[0] = root[0] - totalBranchLength / 2.0
shoulder2[0] = root[0] + totalBranchLength / 2.0
shoulder1[1] = neck[1]
shoulder2[1] = neck[1]
elif physicalAttachment == 3:
neck[0] = root[0] - self._branchNeckLength
neck[1] = root[1]
shoulder1[0] = neck[0]
shoulder2[0] = neck[0]
shoulder1[1] = neck[1] - totalBranchLength / 2.0
shoulder2[1] = neck[1] + totalBranchLength / 2.0
else:
raise "Unrecognised attachment point in GetBranchingAttachmentInfo"
return root, neck, shoulder1, shoulder2
def GetBranchingAttachmentPoint(self, attachment, n):
physicalAttachment = self.LogicalToPhysicalAttachment(attachment)
root, neck, shoulder1, shoulder2 = self.GetBranchingAttachmentInfo(attachment)
pt = wx.RealPoint()
stemPt = wx.RealPoint()
if physicalAttachment == 0:
pt[1] = neck[1] - self._branchStemLength
pt[0] = shoulder1[0] + n * self._branchSpacing
stemPt[0] = pt[0]
stemPt[1] = neck[1]
elif physicalAttachment == 2:
pt[1] = neck[1] + self._branchStemLength
pt[0] = shoulder1[0] + n * self._branchStemLength
stemPt[0] = pt[0]
stemPt[1] = neck[1]
elif physicalAttachment == 1:
pt[0] = neck[0] + self._branchStemLength
pt[1] = shoulder1[1] + n * self._branchSpacing
stemPt[0] = neck[0]
stemPt[1] = pt[1]
elif physicalAttachment == 3:
pt[0] = neck[0] - self._branchStemLength
pt[1] = shoulder1[1] + n * self._branchSpacing
stemPt[0] = neck[0]
stemPt[1] = pt[1]
else:
raise "Unrecognised attachment point in GetBranchingAttachmentPoint"
return pt, stemPt
def GetAttachmentLineCount(self, attachment):
"""Get the number of lines at this attachment position."""
count = 0
for lineShape in self._lines:
if lineShape.GetFrom() == self and lineShape.GetAttachmentFrom() == attachment:
count += 1
elif lineShape.GetTo() == self and lineShape.GetAttachmentTo() == attachment:
count += 1
return count
def GetBranchingAttachmentRoot(self, attachment):
"""Get the root point at the given attachment."""
physicalAttachment = self.LogicalToPhysicalAttachment(attachment)
root = wx.RealPoint()
width, height = self.GetBoundingBoxMax()
# Assume that we have attachment points 0 to 3: top, right, bottom, left
if physicalAttachment == 0:
root[0] = self.GetX()
root[1] = self.GetY() - height / 2.0
elif physicalAttachment == 1:
root[0] = self.GetX() + width / 2.0
root[1] = self.GetY()
elif physicalAttachment == 2:
root[0] = self.GetX()
root[1] = self.GetY() + height / 2.0
elif physicalAttachment == 3:
root[0] = self.GetX() - width / 2.0
root[1] = self.GetY()
else:
raise "Unrecognised attachment point in GetBranchingAttachmentRoot"
return root
# Draw or erase the branches (not the actual arcs though)
def OnDrawBranchesAttachment(self, dc, attachment, erase = False):
count = self.GetAttachmentLineCount(attachment)
if count == 0:
return
root, neck, shoulder1, shoulder2 = self.GetBranchingAttachmentInfo(attachment)
if erase:
dc.SetPen(wx.WHITE_PEN)
dc.SetBrush(wx.WHITE_BRUSH)
else:
dc.SetPen(wx.BLACK_PEN)
dc.SetBrush(wx.BLACK_BRUSH)
# Draw neck
dc.DrawLine(root[0], root[1], neck[0], neck[1])
if count > 1:
# Draw shoulder-to-shoulder line
dc.DrawLine(shoulder1[0], shoulder1[1], shoulder2[0], shoulder2[1])
# Draw all the little branches
for i in range(count):
pt, stemPt = self.GetBranchingAttachmentPoint(attachment, i)
dc.DrawLine(stemPt[0], stemPt[1], pt[0], pt[1])
if self.GetBranchStyle() & BRANCHING_ATTACHMENT_BLOB and count > 1:
blobSize = 6.0
dc.DrawEllipse(stemPt[0] - blobSize / 2.0, stemPt[1] - blobSize / 2.0, blobSize, blobSize)
def OnDrawBranches(self, dc, erase = False):
if self._attachmentMode != ATTACHMENT_MODE_BRANCHING:
return
for i in range(self.GetNumberOfAttachments()):
self.OnDrawBranchesAttachment(dc, i, erase)
def GetAttachmentPositionEdge(self, attachment, nth = 0, no_arcs = 1, line = None):
""" Only get the attachment position at the _edge_ of the shape,
ignoring branching mode. This is used e.g. to indicate the edge of
interest, not the point on the attachment branch.
"""
oldMode = self._attachmentMode
# Calculate as if to edge, not branch
if self._attachmentMode == ATTACHMENT_MODE_BRANCHING:
self._attachmentMode = ATTACHMENT_MODE_EDGE
res = self.GetAttachmentPosition(attachment, nth, no_arcs, line)
self._attachmentMode = oldMode
return res
def PhysicalToLogicalAttachment(self, physicalAttachment):
""" Rotate the standard attachment point from physical
(0 is always North) to logical (0 -> 1 if rotated by 90 degrees)
"""
if RoughlyEqual(self.GetRotation(), 0):
i = physicalAttachment
elif RoughlyEqual(self.GetRotation(), math.pi / 2.0):
i = physicalAttachment - 1
elif RoughlyEqual(self.GetRotation(), math.pi):
i = physicalAttachment - 2
elif RoughlyEqual(self.GetRotation(), 3 * math.pi / 2.0):
i = physicalAttachment - 3
else:
# Can't handle -- assume the same
return physicalAttachment
if i < 0:
i += 4
return i
def LogicalToPhysicalAttachment(self, logicalAttachment):
"""Rotate the standard attachment point from logical
to physical (0 is always North).
"""
if RoughlyEqual(self.GetRotation(), 0):
i = logicalAttachment
elif RoughlyEqual(self.GetRotation(), math.pi / 2.0):
i = logicalAttachment + 1
elif RoughlyEqual(self.GetRotation(), math.pi):
i = logicalAttachment + 2
elif RoughlyEqual(self.GetRotation(), 3 * math.pi / 2.0):
i = logicalAttachment + 3
else:
return logicalAttachment
if i > 3:
i -= 4
return i
def Rotate(self, x, y, theta):
"""Rotate about the given axis by the given amount in radians."""
self._rotation = theta
if self._rotation < 0:
self._rotation += 2 * math.pi
elif self._rotation > 2 * math.pi:
self._rotation -= 2 * math.pi
def GetBackgroundPen(self):
"""Return pen of the right colour for the background."""
if self.GetCanvas():
return wx.Pen(self.GetCanvas().GetBackgroundColour(), 1, wx.SOLID)
return WhiteBackgroundPen
def GetBackgroundBrush(self):
"""Return brush of the right colour for the background."""
if self.GetCanvas():
return wx.Brush(self.GetCanvas().GetBackgroundColour(), wx.SOLID)
return WhiteBackgroundBrush
def GetX(self):
"""Get the x position of the centre of the shape."""
return self._xpos
def GetY(self):
"""Get the y position of the centre of the shape."""
return self._ypos
def SetX(self, x):
"""Set the x position of the shape."""
self._xpos = x
def SetY(self, y):
"""Set the y position of the shape."""
self._ypos = y
def GetParent(self):
"""Return the parent of this shape, if it is part of a composite."""
return self._parent
def SetParent(self, p):
self._parent = p
def GetChildren(self):
"""Return the list of children for this shape."""
return self._children
def GetDrawHandles(self):
"""Return the list of drawhandles."""
return self._drawHandles
def GetEventHandler(self):
"""Return the event handler for this shape."""
return self._eventHandler
def SetEventHandler(self, handler):
"""Set the event handler for this shape."""
self._eventHandler = handler
def Recompute(self):
"""Recomputes any constraints associated with the shape.
Normally applicable to CompositeShapes only, but harmless for
other classes of Shape.
"""
return True
def IsHighlighted(self):
"""TRUE if the shape is highlighted. Shape highlighting is unimplemented."""
return self._highlighted
def GetSensitivityFilter(self):
"""Return the sensitivity filter, a bitlist of values.
See Shape.SetSensitivityFilter.
"""
return self._sensitivity
def SetFixedSize(self, x, y):
"""Set the shape to be fixed size."""
self._fixedWidth = x
self._fixedHeight = y
def GetFixedSize(self):
"""Return flags indicating whether the shape is of fixed size in
either direction.
"""
return self._fixedWidth, self._fixedHeight
def GetFixedWidth(self):
"""TRUE if the shape cannot be resized in the horizontal plane."""
return self._fixedWidth
def GetFixedHeight(self):
"""TRUE if the shape cannot be resized in the vertical plane."""
return self._fixedHeight
def SetSpaceAttachments(self, sp):
"""Indicate whether lines should be spaced out evenly at the point
they touch the node (sp = True), or whether they should join at a single
point (sp = False).
"""
self._spaceAttachments = sp
def GetSpaceAttachments(self):
"""Return whether lines should be spaced out evenly at the point they
touch the node (True), or whether they should join at a single point
(False).
"""
return self._spaceAttachments
def SetCentreResize(self, cr):
"""Specify whether the shape is to be resized from the centre (the
centre stands still) or from the corner or side being dragged (the
other corner or side stands still).
"""
self._centreResize = cr
def GetCentreResize(self):
"""TRUE if the shape is to be resized from the centre (the centre stands
still), or FALSE if from the corner or side being dragged (the other
corner or side stands still)
"""
return self._centreResize
def SetMaintainAspectRatio(self, ar):
"""Set whether a shape that resizes should not change the aspect ratio
(width and height should be in the original proportion).
"""
self._maintainAspectRatio = ar
def GetMaintainAspectRatio(self):
"""TRUE if shape keeps aspect ratio during resize."""
return self._maintainAspectRatio
def GetLines(self):
"""Return the list of lines connected to this shape."""
return self._lines
def SetDisableLabel(self, flag):
"""Set flag to TRUE to stop the default region being shown."""
self._disableLabel = flag
def GetDisableLabel(self):
"""TRUE if the default region will not be shown, FALSE otherwise."""
return self._disableLabel
def SetAttachmentMode(self, mode):
"""Set the attachment mode.
If TRUE, attachment points will be significant when drawing lines to
and from this shape.
If FALSE, lines will be drawn as if to the centre of the shape.
"""
self._attachmentMode = mode
def GetAttachmentMode(self):
"""Return the attachment mode.
See Shape.SetAttachmentMode.
"""
return self._attachmentMode
def SetId(self, i):
"""Set the integer identifier for this shape."""
self._id = i
def GetId(self):
"""Return the integer identifier for this shape."""
return self._id
def IsShown(self):
"""TRUE if the shape is in a visible state, FALSE otherwise.
Note that this has nothing to do with whether the window is hidden
or the shape has scrolled off the canvas; it refers to the internal
visibility flag.
"""
return self._visible
def GetPen(self):
"""Return the pen used for drawing the shape's outline."""
return self._pen
def GetBrush(self):
"""Return the brush used for filling the shape."""
return self._brush
def GetNumberOfTextRegions(self):
"""Return the number of text regions for this shape."""
return len(self._regions)
def GetRegions(self):
"""Return the list of ShapeRegions."""
return self._regions
# Control points ('handles') redirect control to the actual shape, to
# make it easier to override sizing behaviour.
def OnSizingDragLeft(self, pt, draw, x, y, keys = 0, attachment = 0):
bound_x, bound_y = self.GetBoundingBoxMin()
dc = wx.ClientDC(self.GetCanvas())
self.GetCanvas().PrepareDC(dc)
dc.SetLogicalFunction(OGLRBLF)
dottedPen = wx.Pen(wx.Colour(0, 0, 0), 1, wx.DOT)
dc.SetPen(dottedPen)
dc.SetBrush(wx.TRANSPARENT_BRUSH)
if self.GetCentreResize():
# Maintain the same centre point
new_width = 2.0 * abs(x - self.GetX())
new_height = 2.0 * abs(y - self.GetY())
# Constrain sizing according to what control point you're dragging
if pt._type == CONTROL_POINT_HORIZONTAL:
if self.GetMaintainAspectRatio():
new_height = bound_y * (new_width / bound_x)
else:
new_height = bound_y
elif pt._type == CONTROL_POINT_VERTICAL:
if self.GetMaintainAspectRatio():
new_width = bound_x * (new_height / bound_y)
else:
new_width = bound_x
elif pt._type == CONTROL_POINT_DIAGONAL and (keys & KEY_SHIFT):
new_height = bound_y * (new_width / bound_x)
if self.GetFixedWidth():
new_width = bound_x
if self.GetFixedHeight():
new_height = bound_y
pt._controlPointDragEndWidth = new_width
pt._controlPointDragEndHeight = new_height
self.GetEventHandler().OnDrawOutline(dc, self.GetX(), self.GetY(), new_width, new_height)
else:
# Don't maintain the same centre point
newX1 = min(pt._controlPointDragStartX, x)
newY1 = min(pt._controlPointDragStartY, y)
newX2 = max(pt._controlPointDragStartX, x)
newY2 = max(pt._controlPointDragStartY, y)
if pt._type == CONTROL_POINT_HORIZONTAL:
newY1 = pt._controlPointDragStartY
newY2 = newY1 + pt._controlPointDragStartHeight
elif pt._type == CONTROL_POINT_VERTICAL:
newX1 = pt._controlPointDragStartX
newX2 = newX1 + pt._controlPointDragStartWidth
elif pt._type == CONTROL_POINT_DIAGONAL and (keys & KEY_SHIFT or self.GetMaintainAspectRatio()):
newH = (newX2 - newX1) * (float(pt._controlPointDragStartHeight) / pt._controlPointDragStartWidth)
if self.GetY() > pt._controlPointDragStartY:
newY2 = newY1 + newH
else:
newY1 = newY2 - newH
newWidth = float(newX2 - newX1)
newHeight = float(newY2 - newY1)
if pt._type == CONTROL_POINT_VERTICAL and self.GetMaintainAspectRatio():
newWidth = bound_x * (newHeight / bound_y)
if pt._type == CONTROL_POINT_HORIZONTAL and self.GetMaintainAspectRatio():
newHeight = bound_y * (newWidth / bound_x)
pt._controlPointDragPosX = newX1 + newWidth / 2.0
pt._controlPointDragPosY = newY1 + newHeight / 2.0
if self.GetFixedWidth():
newWidth = bound_x
if self.GetFixedHeight():
newHeight = bound_y
pt._controlPointDragEndWidth = newWidth
pt._controlPointDragEndHeight = newHeight
self.GetEventHandler().OnDrawOutline(dc, pt._controlPointDragPosX, pt._controlPointDragPosY, newWidth, newHeight)
def OnSizingBeginDragLeft(self, pt, x, y, keys = 0, attachment = 0):
self._canvas.CaptureMouse()
dc = wx.ClientDC(self.GetCanvas())
self.GetCanvas().PrepareDC(dc)
dc.SetLogicalFunction(OGLRBLF)
bound_x, bound_y = self.GetBoundingBoxMin()
self.GetEventHandler().OnBeginSize(bound_x, bound_y)
# Choose the 'opposite corner' of the object as the stationary
# point in case this is non-centring resizing.
if pt.GetX() < self.GetX():
pt._controlPointDragStartX = self.GetX() + bound_x / 2.0
else:
pt._controlPointDragStartX = self.GetX() - bound_x / 2.0
if pt.GetY() < self.GetY():
pt._controlPointDragStartY = self.GetY() + bound_y / 2.0
else:
pt._controlPointDragStartY = self.GetY() - bound_y / 2.0
if pt._type == CONTROL_POINT_HORIZONTAL:
pt._controlPointDragStartY = self.GetY() - bound_y / 2.0
elif pt._type == CONTROL_POINT_VERTICAL:
pt._controlPointDragStartX = self.GetX() - bound_x / 2.0
# We may require the old width and height
pt._controlPointDragStartWidth = bound_x
pt._controlPointDragStartHeight = bound_y
dottedPen = wx.Pen(wx.Colour(0, 0, 0), 1, wx.DOT)
dc.SetPen(dottedPen)
dc.SetBrush(wx.TRANSPARENT_BRUSH)
if self.GetCentreResize():
new_width = 2.0 * abs(x - self.GetX())
new_height = 2.0 * abs(y - self.GetY())
# Constrain sizing according to what control point you're dragging
if pt._type == CONTROL_POINT_HORIZONTAL:
if self.GetMaintainAspectRatio():
new_height = bound_y * (new_width / bound_x)
else:
new_height = bound_y
elif pt._type == CONTROL_POINT_VERTICAL:
if self.GetMaintainAspectRatio():
new_width = bound_x * (new_height / bound_y)
else:
new_width = bound_x
elif pt._type == CONTROL_POINT_DIAGONAL and (keys & KEY_SHIFT):
new_height = bound_y * (new_width / bound_x)
if self.GetFixedWidth():
new_width = bound_x
if self.GetFixedHeight():
new_height = bound_y
pt._controlPointDragEndWidth = new_width
pt._controlPointDragEndHeight = new_height
self.GetEventHandler().OnDrawOutline(dc, self.GetX(), self.GetY(), new_width, new_height)
else:
# Don't maintain the same centre point
newX1 = min(pt._controlPointDragStartX, x)
newY1 = min(pt._controlPointDragStartY, y)
newX2 = max(pt._controlPointDragStartX, x)
newY2 = max(pt._controlPointDragStartY, y)
if pt._type == CONTROL_POINT_HORIZONTAL:
newY1 = pt._controlPointDragStartY
newY2 = newY1 + pt._controlPointDragStartHeight
elif pt._type == CONTROL_POINT_VERTICAL:
newX1 = pt._controlPointDragStartX
newX2 = newX1 + pt._controlPointDragStartWidth
elif pt._type == CONTROL_POINT_DIAGONAL and (keys & KEY_SHIFT or self.GetMaintainAspectRatio()):
newH = (newX2 - newX1) * (float(pt._controlPointDragStartHeight) / pt._controlPointDragStartWidth)
if pt.GetY() > pt._controlPointDragStartY:
newY2 = newY1 + newH
else:
newY1 = newY2 - newH
newWidth = float(newX2 - newX1)
newHeight = float(newY2 - newY1)
if pt._type == CONTROL_POINT_VERTICAL and self.GetMaintainAspectRatio():
newWidth = bound_x * (newHeight / bound_y)
if pt._type == CONTROL_POINT_HORIZONTAL and self.GetMaintainAspectRatio():
newHeight = bound_y * (newWidth / bound_x)
pt._controlPointDragPosX = newX1 + newWidth / 2.0
pt._controlPointDragPosY = newY1 + newHeight / 2.0
if self.GetFixedWidth():
newWidth = bound_x
if self.GetFixedHeight():
newHeight = bound_y
pt._controlPointDragEndWidth = newWidth
pt._controlPointDragEndHeight = newHeight
self.GetEventHandler().OnDrawOutline(dc, pt._controlPointDragPosX, pt._controlPointDragPosY, newWidth, newHeight)
def OnSizingEndDragLeft(self, pt, x, y, keys = 0, attachment = 0):
dc = wx.ClientDC(self.GetCanvas())
self.GetCanvas().PrepareDC(dc)
if self._canvas.HasCapture():
self._canvas.ReleaseMouse()
dc.SetLogicalFunction(wx.COPY)
self.Recompute()
self.ResetControlPoints()
self.Erase(dc)
self.SetSize(pt._controlPointDragEndWidth, pt._controlPointDragEndHeight)
# The next operation could destroy this control point (it does for
# label objects, via formatting the text), so save all values we're
# going to use, or we'll be accessing garbage.
#return
if self.GetCentreResize():
self.Move(dc, self.GetX(), self.GetY())
else:
self.Move(dc, pt._controlPointDragPosX, pt._controlPointDragPosY)
# Recursively redraw links if we have a composite
if len(self.GetChildren()):
self.DrawLinks(dc, -1, True)
width, height = self.GetBoundingBoxMax()
self.GetEventHandler().OnEndSize(width, height)
if not self._canvas.GetQuickEditMode() and pt._eraseObject:
self._canvas.Redraw(dc)
class RectangleShape(Shape):
"""
The wxRectangleShape has rounded or square corners.
Derived from:
Shape
"""
def __init__(self, w = 0.0, h = 0.0):
Shape.__init__(self)
self._width = w
self._height = h
self._cornerRadius = 0.0
self.SetDefaultRegionSize()
def OnDraw(self, dc):
x1 = self._xpos - self._width / 2.0
y1 = self._ypos - self._height / 2.0
if self._shadowMode != SHADOW_NONE:
if self._shadowBrush:
dc.SetBrush(self._shadowBrush)
dc.SetPen(TransparentPen)
if self._cornerRadius:
dc.DrawRoundedRectangle(x1 + self._shadowOffsetX, y1 + self._shadowOffsetY, self._width, self._height, self._cornerRadius)
else:
dc.DrawRectangle(x1 + self._shadowOffsetX, y1 + self._shadowOffsetY, self._width, self._height)
if self._pen:
if self._pen.GetWidth() == 0:
dc.SetPen(TransparentPen)
else:
dc.SetPen(self._pen)
if self._brush:
dc.SetBrush(self._brush)
if self._cornerRadius:
dc.DrawRoundedRectangle(x1, y1, self._width, self._height, self._cornerRadius)
else:
dc.DrawRectangle(x1, y1, self._width, self._height)
def GetBoundingBoxMin(self):
return self._width, self._height
def SetSize(self, x, y, recursive = False):
self.SetAttachmentSize(x, y)
self._width = max(x, 1)
self._height = max(y, 1)
self.SetDefaultRegionSize()
def GetCornerRadius(self):
"""Get the radius of the rectangle's rounded corners."""
return self._cornerRadius
def SetCornerRadius(self, rad):
"""Set the radius of the rectangle's rounded corners.
If the radius is zero, a non-rounded rectangle will be drawn.
If the radius is negative, the value is the proportion of the smaller
dimension of the rectangle.
"""
self._cornerRadius = rad
# Assume (x1, y1) is centre of box (most generally, line end at box)
def GetPerimeterPoint(self, x1, y1, x2, y2):
bound_x, bound_y = self.GetBoundingBoxMax()
return FindEndForBox(bound_x, bound_y, self._xpos, self._ypos, x2, y2)
def GetWidth(self):
return self._width
def GetHeight(self):
return self._height
def SetWidth(self, w):
self._width = w
def SetHeight(self, h):
self._height = h
class PolygonShape(Shape):
"""A PolygonShape's shape is defined by a number of points passed to
the object's constructor. It can be used to create new shapes such as
diamonds and triangles.
"""
def __init__(self):
Shape.__init__(self)
self._points = None
self._originalPoints = None
def Create(self, the_points = None):
"""Takes a list of wx.RealPoints or tuples; each point is an offset
from the centre.
"""
self.ClearPoints()
if not the_points:
self._originalPoints = []
self._points = []
else:
self._originalPoints = the_points
# Duplicate the list of points
self._points = []
for point in the_points:
new_point = wx.Point(point[0], point[1])
self._points.append(new_point)
self.CalculateBoundingBox()
self._originalWidth = self._boundWidth
self._originalHeight = self._boundHeight
self.SetDefaultRegionSize()
def ClearPoints(self):
self._points = []
self._originalPoints = []
# Width and height. Centre of object is centre of box
def GetBoundingBoxMin(self):
return self._boundWidth, self._boundHeight
def GetPoints(self):
"""Return the internal list of polygon vertices."""
return self._points
def GetOriginalPoints(self):
return self._originalPoints
def GetOriginalWidth(self):
return self._originalWidth
def GetOriginalHeight(self):
return self._originalHeight
def SetOriginalWidth(self, w):
self._originalWidth = w
def SetOriginalHeight(self, h):
self._originalHeight = h
def CalculateBoundingBox(self):
# Calculate bounding box at construction (and presumably resize) time
left = 10000
right = -10000
top = 10000
bottom = -10000
for point in self._points:
if point[0] < left:
left = point[0]
if point[0] > right:
right = point[0]
if point[1] < top:
top = point[1]
if point[1] > bottom:
bottom = point[1]
self._boundWidth = right - left
self._boundHeight = bottom - top
def CalculatePolygonCentre(self):
"""Recalculates the centre of the polygon, and
readjusts the point offsets accordingly.
Necessary since the centre of the polygon
is expected to be the real centre of the bounding
box.
"""
left = 10000
right = -10000
top = 10000
bottom = -10000
for point in self._points:
if point[0] < left:
left = point[0]
if point[0] > right:
right = point[0]
if point[1] < top:
top = point[1]
if point[1] > bottom:
bottom = point[1]
bwidth = right - left
bheight = bottom - top
newCentreX = left + bwidth / 2.0
newCentreY = top + bheight / 2.0
for i in range(len(self._points)):
self._points[i] = self._points[i][0] - newCentreX, self._points[i][1] - newCentreY
self._xpos += newCentreX
self._ypos += newCentreY
def HitTest(self, x, y):
# Imagine four lines radiating from this point. If all of these lines
# hit the polygon, we're inside it, otherwise we're not. Obviously
# we'd need more radiating lines to be sure of correct results for
# very strange (concave) shapes.
endPointsX = [x, x + 1000, x, x - 1000]
endPointsY = [y - 1000, y, y + 1000, y]
xpoints = []
ypoints = []
for point in self._points:
xpoints.append(point[0] + self._xpos)
ypoints.append(point[1] + self._ypos)
# We assume it's inside the polygon UNLESS one or more
# lines don't hit the outline.
isContained = True
for i in range(4):
if not PolylineHitTest(xpoints, ypoints, x, y, endPointsX[i], endPointsY[i]):
isContained = False
if not isContained:
return False
nearest_attachment = 0
# If a hit, check the attachment points within the object
nearest = 999999
for i in range(self.GetNumberOfAttachments()):
e = self.GetAttachmentPositionEdge(i)
if e:
xp, yp = e
l = math.sqrt((xp - x) * (xp - x) + (yp - y) * (yp - y))
if l < nearest:
nearest = l
nearest_attachment = i
return nearest_attachment, nearest
# Really need to be able to reset the shape! Otherwise, if the
# points ever go to zero, we've lost it, and can't resize.
def SetSize(self, new_width, new_height, recursive = True):
self.SetAttachmentSize(new_width, new_height)
# Multiply all points by proportion of new size to old size
x_proportion = abs(float(new_width) / self._originalWidth)
y_proportion = abs(float(new_height) / self._originalHeight)
for i in range(max(len(self._points), len(self._originalPoints))):
self._points[i] = wx.Point(self._originalPoints[i][0] * x_proportion, self._originalPoints[i][1] * y_proportion)
self._boundWidth = abs(new_width)
self._boundHeight = abs(new_height)
self.SetDefaultRegionSize()
# Make the original points the same as the working points
def UpdateOriginalPoints(self):
"""If we've changed the shape, must make the original points match the
working points with this function.
"""
self._originalPoints = []
for point in self._points:
original_point = wx.RealPoint(point[0], point[1])
self._originalPoints.append(original_point)
self.CalculateBoundingBox()
self._originalWidth = self._boundWidth
self._originalHeight = self._boundHeight
def AddPolygonPoint(self, pos):
"""Add a control point after the given point."""
try:
firstPoint = self._points[pos]
except ValueError:
firstPoint = self._points[0]
try:
secondPoint = self._points[pos + 1]
except ValueError:
secondPoint = self._points[0]
x = (secondPoint[0] - firstPoint[0]) / 2.0 + firstPoint[0]
y = (secondPoint[1] - firstPoint[1]) / 2.0 + firstPoint[1]
point = wx.RealPoint(x, y)
if pos >= len(self._points) - 1:
self._points.append(point)
else:
self._points.insert(pos + 1, point)
self.UpdateOriginalPoints()
if self._selected:
self.DeleteControlPoints()
self.MakeControlPoints()
def DeletePolygonPoint(self, pos):
"""Delete the given control point."""
if pos < len(self._points):
del self._points[pos]
self.UpdateOriginalPoints()
if self._selected:
self.DeleteControlPoints()
self.MakeControlPoints()
# Assume (x1, y1) is centre of box (most generally, line end at box)
def GetPerimeterPoint(self, x1, y1, x2, y2):
# First check for situation where the line is vertical,
# and we would want to connect to a point on that vertical --
# oglFindEndForPolyline can't cope with this (the arrow
# gets drawn to the wrong place).
if self._attachmentMode == ATTACHMENT_MODE_NONE and x1 == x2:
# Look for the point we'd be connecting to. This is
# a heuristic...
for point in self._points:
if point[0] == 0:
if y2 > y1 and point[1] > 0:
return point[0] + self._xpos, point[1] + self._ypos
elif y2 < y1 and point[1] < 0:
return point[0] + self._xpos, point[1] + self._ypos
xpoints = []
ypoints = []
for point in self._points:
xpoints.append(point[0] + self._xpos)
ypoints.append(point[1] + self._ypos)
return FindEndForPolyline(xpoints, ypoints, x1, y1, x2, y2)
def OnDraw(self, dc):
if self._shadowMode != SHADOW_NONE:
if self._shadowBrush:
dc.SetBrush(self._shadowBrush)
dc.SetPen(TransparentPen)
dc.DrawPolygon(self._points, self._xpos + self._shadowOffsetX, self._ypos, self._shadowOffsetY)
if self._pen:
if self._pen.GetWidth() == 0:
dc.SetPen(TransparentPen)
else:
dc.SetPen(self._pen)
if self._brush:
dc.SetBrush(self._brush)
dc.DrawPolygon(self._points, self._xpos, self._ypos)
def OnDrawOutline(self, dc, x, y, w, h):
dc.SetBrush(wx.TRANSPARENT_BRUSH)
# Multiply all points by proportion of new size to old size
x_proportion = abs(float(w) / self._originalWidth)
y_proportion = abs(float(h) / self._originalHeight)
intPoints = []
for point in self._originalPoints:
intPoints.append(wx.Point(x_proportion * point[0], y_proportion * point[1]))
dc.DrawPolygon(intPoints, x, y)
# Make as many control points as there are vertices
def MakeControlPoints(self):
for point in self._points:
control = PolygonControlPoint(self._canvas, self, CONTROL_POINT_SIZE, point, point[0], point[1])
self._canvas.AddShape(control)
self._controlPoints.append(control)
def ResetControlPoints(self):
for i in range(min(len(self._points), len(self._controlPoints))):
point = self._points[i]
self._controlPoints[i]._xoffset = point[0]
self._controlPoints[i]._yoffset = point[1]
self._controlPoints[i].polygonVertex = point
def GetNumberOfAttachments(self):
maxN = max(len(self._points) - 1, 0)
for point in self._attachmentPoints:
if point._id > maxN:
maxN = point._id
return maxN + 1
def GetAttachmentPosition(self, attachment, nth = 0, no_arcs = 1, line = None):
if self._attachmentMode == ATTACHMENT_MODE_EDGE and self._points and attachment < len(self._points):
point = self._points[0]
return point[0] + self._xpos, point[1] + self._ypos
return Shape.GetAttachmentPosition(self, attachment, nth, no_arcs, line)
def AttachmentIsValid(self, attachment):
if not self._points:
return False
if attachment >= 0 and attachment < len(self._points):
return True
for point in self._attachmentPoints:
if point._id == attachment:
return True
return False
# Rotate about the given axis by the given amount in radians
def Rotate(self, x, y, theta):
actualTheta = theta - self._rotation
# Rotate attachment points
sinTheta = math.sin(actualTheta)
cosTheta = math.cos(actualTheta)
for point in self._attachmentPoints:
x1 = point._x
y1 = point._y
point._x = x1 * cosTheta - y1 * sinTheta + x * (1 - cosTheta) + y * sinTheta
point._y = x1 * sinTheta + y1 * cosTheta + y * (1 - cosTheta) + x * sinTheta
for i in range(len(self._points)):
x1, y1 = self._points[i]
self._points[i] = x1 * cosTheta - y1 * sinTheta + x * (1 - cosTheta) + y * sinTheta, x1 * sinTheta + y1 * cosTheta + y * (1 - cosTheta) + x * sinTheta
for i in range(len(self._originalPoints)):
x1, y1 = self._originalPoints[i]
self._originalPoints[i] = x1 * cosTheta - y1 * sinTheta + x * (1 - cosTheta) + y * sinTheta, x1 * sinTheta + y1 * cosTheta + y * (1 - cosTheta) + x * sinTheta
# Added by Pierre Hjälm. If we don't do this the outline will be
# the wrong size. Hopefully it won't have any ill effects.
self.UpdateOriginalPoints()
self._rotation = theta
self.CalculatePolygonCentre()
self.CalculateBoundingBox()
self.ResetControlPoints()
# Control points ('handles') redirect control to the actual shape, to
# make it easier to override sizing behaviour.
def OnSizingDragLeft(self, pt, draw, x, y, keys = 0, attachment = 0):
dc = wx.ClientDC(self.GetCanvas())
self.GetCanvas().PrepareDC(dc)
dc.SetLogicalFunction(OGLRBLF)
dottedPen = wx.Pen(wx.Colour(0, 0, 0), 1, wx.DOT)
dc.SetPen(dottedPen)
dc.SetBrush(wx.TRANSPARENT_BRUSH)
# Code for CTRL-drag in C++ version commented out
pt.CalculateNewSize(x, y)
self.GetEventHandler().OnDrawOutline(dc, self.GetX(), self.GetY(), pt.GetNewSize()[0], pt.GetNewSize()[1])
def OnSizingBeginDragLeft(self, pt, x, y, keys = 0, attachment = 0):
dc = wx.ClientDC(self.GetCanvas())
self.GetCanvas().PrepareDC(dc)
self.Erase(dc)
dc.SetLogicalFunction(OGLRBLF)
bound_x, bound_y = self.GetBoundingBoxMin()
dist = math.sqrt((x - self.GetX()) * (x - self.GetX()) + (y - self.GetY()) * (y - self.GetY()))
pt._originalDistance = dist
pt._originalSize[0] = bound_x
pt._originalSize[1] = bound_y
if pt._originalDistance == 0:
pt._originalDistance = 0.0001
dottedPen = wx.Pen(wx.Colour(0, 0, 0), 1, wx.DOT)
dc.SetPen(dottedPen)
dc.SetBrush(wx.TRANSPARENT_BRUSH)
# Code for CTRL-drag in C++ version commented out
pt.CalculateNewSize(x, y)
self.GetEventHandler().OnDrawOutline(dc, self.GetX(), self.GetY(), pt.GetNewSize()[0], pt.GetNewSize()[1])
self._canvas.CaptureMouse()
def OnSizingEndDragLeft(self, pt, x, y, keys = 0, attachment = 0):
dc = wx.ClientDC(self.GetCanvas())
self.GetCanvas().PrepareDC(dc)
if self._canvas.HasCapture():
self._canvas.ReleaseMouse()
dc.SetLogicalFunction(wx.COPY)
# If we're changing shape, must reset the original points
if keys & KEY_CTRL:
self.CalculateBoundingBox()
self.CalculatePolygonCentre()
else:
self.SetSize(pt.GetNewSize()[0], pt.GetNewSize()[1])
self.Recompute()
self.ResetControlPoints()
self.Move(dc, self.GetX(), self.GetY())
if not self._canvas.GetQuickEditMode():
self._canvas.Redraw(dc)
class EllipseShape(Shape):
"""The EllipseShape behaves similarly to the RectangleShape but is
elliptical.
Derived from:
wxShape
"""
def __init__(self, w, h):
Shape.__init__(self)
self._width = w
self._height = h
self.SetDefaultRegionSize()
def GetBoundingBoxMin(self):
return self._width, self._height
def GetPerimeterPoint(self, x1, y1, x2, y2):
bound_x, bound_y = self.GetBoundingBoxMax()
return DrawArcToEllipse(self._xpos, self._ypos, bound_x, bound_y, x2, y2, x1, y1)
def GetWidth(self):
return self._width
def GetHeight(self):
return self._height
def SetWidth(self, w):
self._width = w
def SetHeight(self, h):
self._height = h
def OnDraw(self, dc):
if self._shadowMode != SHADOW_NONE:
if self._shadowBrush:
dc.SetBrush(self._shadowBrush)
dc.SetPen(TransparentPen)
dc.DrawEllipse(self._xpos - self.GetWidth() / 2.0 + self._shadowOffsetX,
self._ypos - self.GetHeight() / 2.0 + self._shadowOffsetY,
self.GetWidth(), self.GetHeight())
if self._pen:
if self._pen.GetWidth() == 0:
dc.SetPen(TransparentPen)
else:
dc.SetPen(self._pen)
if self._brush:
dc.SetBrush(self._brush)
dc.DrawEllipse(self._xpos - self.GetWidth() / 2.0, self._ypos - self.GetHeight() / 2.0, self.GetWidth(), self.GetHeight())
def SetSize(self, x, y, recursive = True):
self.SetAttachmentSize(x, y)
self._width = x
self._height = y
self.SetDefaultRegionSize()
def GetNumberOfAttachments(self):
return Shape.GetNumberOfAttachments(self)
# There are 4 attachment points on an ellipse - 0 = top, 1 = right,
# 2 = bottom, 3 = left.
def GetAttachmentPosition(self, attachment, nth = 0, no_arcs = 1, line = None):
if self._attachmentMode == ATTACHMENT_MODE_BRANCHING:
return Shape.GetAttachmentPosition(self, attachment, nth, no_arcs, line)
if self._attachmentMode != ATTACHMENT_MODE_NONE:
top = self._ypos + self._height / 2.0
bottom = self._ypos - self._height / 2.0
left = self._xpos - self._width / 2.0
right = self._xpos + self._width / 2.0
physicalAttachment = self.LogicalToPhysicalAttachment(attachment)
if physicalAttachment == 0:
if self._spaceAttachments:
x = left + (nth + 1) * self._width / (no_arcs + 1.0)
else:
x = self._xpos
y = top
# We now have the point on the bounding box: but get the point
# on the ellipse by imagining a vertical line from
# (x, self._ypos - self._height - 500) to (x, self._ypos) intersecting
# the ellipse.
return DrawArcToEllipse(self._xpos, self._ypos, self._width, self._height, x, self._ypos - self._height - 500, x, self._ypos)
elif physicalAttachment == 1:
x = right
if self._spaceAttachments:
y = bottom + (nth + 1) * self._height / (no_arcs + 1.0)
else:
y = self._ypos
return DrawArcToEllipse(self._xpos, self._ypos, self._width, self._height, self._xpos + self._width + 500, y, self._xpos, y)
elif physicalAttachment == 2:
if self._spaceAttachments:
x = left + (nth + 1) * self._width / (no_arcs + 1.0)
else:
x = self._xpos
y = bottom
return DrawArcToEllipse(self._xpos, self._ypos, self._width, self._height, x, self._ypos + self._height + 500, x, self._ypos)
elif physicalAttachment == 3:
x = left
if self._spaceAttachments:
y = bottom + (nth + 1) * self._height / (no_arcs + 1.0)
else:
y = self._ypos
return DrawArcToEllipse(self._xpos, self._ypos, self._width, self._height, self._xpos - self._width - 500, y, self._xpos, y)
else:
return Shape.GetAttachmentPosition(self, attachment, x, y, nth, no_arcs, line)
else:
return self._xpos, self._ypos
class CircleShape(EllipseShape):
"""An EllipseShape whose width and height are the same."""
def __init__(self, diameter):
EllipseShape.__init__(self, diameter, diameter)
self.SetMaintainAspectRatio(True)
def GetPerimeterPoint(self, x1, y1, x2, y2):
return FindEndForCircle(self._width / 2.0, self._xpos, self._ypos, x2, y2)
class TextShape(RectangleShape):
"""As wxRectangleShape, but only the text is displayed."""
def __init__(self, width, height):
RectangleShape.__init__(self, width, height)
def OnDraw(self, dc):
pass
class ShapeRegion(object):
"""Object region."""
def __init__(self, region = None):
if region:
self._regionText = region._regionText
self._regionName = region._regionName
self._textColour = region._textColour
self._font = region._font
self._minHeight = region._minHeight
self._minWidth = region._minWidth
self._width = region._width
self._height = region._height
self._x = region._x
self._y = region._y
self._regionProportionX = region._regionProportionX
self._regionProportionY = region._regionProportionY
self._formatMode = region._formatMode
self._actualColourObject = region._actualColourObject
self._actualPenObject = None
self._penStyle = region._penStyle
self._penColour = region._penColour
self.ClearText()
for line in region._formattedText:
new_line = ShapeTextLine(line.GetX(), line.GetY(), line.GetText())
self._formattedText.append(new_line)
else:
self._regionText = ""
self._font = NormalFont
self._minHeight = 5.0
self._minWidth = 5.0
self._width = 0.0
self._height = 0.0
self._x = 0.0
self._y = 0.0
self._regionProportionX = -1.0
self._regionProportionY = -1.0
self._formatMode = FORMAT_CENTRE_HORIZ | FORMAT_CENTRE_VERT
self._regionName = ""
self._textColour = "BLACK"
self._penColour = "BLACK"
self._penStyle = wx.SOLID
self._actualColourObject = wx.TheColourDatabase.Find("BLACK")
self._actualPenObject = None
self._formattedText = []
def ClearText(self):
self._formattedText = []
def SetFont(self, f):
self._font = f
def SetMinSize(self, w, h):
self._minWidth = w
self._minHeight = h
def SetSize(self, w, h):
self._width = w
self._height = h
def SetPosition(self, xp, yp):
self._x = xp
self._y = yp
def SetProportions(self, xp, yp):
self._regionProportionX = xp
self._regionProportionY = yp
def SetFormatMode(self, mode):
self._formatMode = mode
def SetColour(self, col):
self._textColour = col
self._actualColourObject = col
def GetActualColourObject(self):
self._actualColourObject = wx.TheColourDatabase.Find(self.GetColour())
return self._actualColourObject
def SetPenColour(self, col):
self._penColour = col
self._actualPenObject = None
# Returns NULL if the pen is invisible
# (different to pen being transparent; indicates that
# region boundary should not be drawn.)
def GetActualPen(self):
if self._actualPenObject:
return self._actualPenObject
if not self._penColour:
return None
if self._penColour=="Invisible":
return None
self._actualPenObject = wx.Pen(self._penColour, 1, self._penStyle)
return self._actualPenObject
def SetText(self, s):
self._regionText = s
def SetName(self, s):
self._regionName = s
def GetText(self):
return self._regionText
def GetFont(self):
return self._font
def GetMinSize(self):
return self._minWidth, self._minHeight
def GetProportion(self):
return self._regionProportionX, self._regionProportionY
def GetSize(self):
return self._width, self._height
def GetPosition(self):
return self._x, self._y
def GetFormatMode(self):
return self._formatMode
def GetName(self):
return self._regionName
def GetColour(self):
return self._textColour
def GetFormattedText(self):
return self._formattedText
def GetPenColour(self):
return self._penColour
def GetPenStyle(self):
return self._penStyle
def SetPenStyle(self, style):
self._penStyle = style
self._actualPenObject = None
def GetWidth(self):
return self._width
def GetHeight(self):
return self._height
class ControlPoint(RectangleShape):
def __init__(self, theCanvas, object, size, the_xoffset, the_yoffset, the_type):
RectangleShape.__init__(self, size, size)
self._canvas = theCanvas
self._shape = object
self._xoffset = the_xoffset
self._yoffset = the_yoffset
self._type = the_type
self.SetPen(BlackForegroundPen)
self.SetBrush(wx.BLACK_BRUSH)
self._oldCursor = None
self._visible = True
self._eraseObject = True
# Don't even attempt to draw any text - waste of time
def OnDrawContents(self, dc):
pass
def OnDraw(self, dc):
self._xpos = self._shape.GetX() + self._xoffset
self._ypos = self._shape.GetY() + self._yoffset
RectangleShape.OnDraw(self, dc)
def OnErase(self, dc):
RectangleShape.OnErase(self, dc)
# Implement resizing of canvas object
def OnDragLeft(self, draw, x, y, keys = 0, attachment = 0):
self._shape.GetEventHandler().OnSizingDragLeft(self, draw, x, y, keys, attachment)
def OnBeginDragLeft(self, x, y, keys = 0, attachment = 0):
self._shape.GetEventHandler().OnSizingBeginDragLeft(self, x, y, keys, attachment)
def OnEndDragLeft(self, x, y, keys = 0, attachment = 0):
self._shape.GetEventHandler().OnSizingEndDragLeft(self, x, y, keys, attachment)
def GetNumberOfAttachments(self):
return 1
def GetAttachmentPosition(self, attachment, nth = 0, no_arcs = 1, line = None):
return self._xpos, self._ypos
def SetEraseObject(self, er):
self._eraseObject = er
class PolygonControlPoint(ControlPoint):
def __init__(self, theCanvas, object, size, vertex, the_xoffset, the_yoffset):
ControlPoint.__init__(self, theCanvas, object, size, the_xoffset, the_yoffset, 0)
self._polygonVertex = vertex
self._originalDistance = 0.0
self._newSize = wx.RealPoint()
self._originalSize = wx.RealPoint()
def GetNewSize(self):
return self._newSize
# Calculate what new size would be, at end of resize
def CalculateNewSize(self, x, y):
bound_x, bound_y = self.GetShape().GetBoundingBoxMax()
dist = math.sqrt((x - self._shape.GetX()) * (x - self._shape.GetX()) + (y - self._shape.GetY()) * (y - self._shape.GetY()))
self._newSize[0] = dist / self._originalDistance * self._originalSize[0]
self._newSize[1] = dist / self._originalDistance * self._originalSize[1]
# Implement resizing polygon or moving the vertex
def OnDragLeft(self, draw, x, y, keys = 0, attachment = 0):
self._shape.GetEventHandler().OnSizingDragLeft(self, draw, x, y, keys, attachment)
def OnBeginDragLeft(self, x, y, keys = 0, attachment = 0):
self._shape.GetEventHandler().OnSizingBeginDragLeft(self, x, y, keys, attachment)
def OnEndDragLeft(self, x, y, keys = 0, attachment = 0):
self._shape.GetEventHandler().OnSizingEndDragLeft(self, x, y, keys, attachment)
from _canvas import *
from _lines import *
from _composit import *