1fded56b37
git-svn-id: https://svn.wxwidgets.org/svn/wx/wxWidgets/trunk@21593 c3d73ce0-8a6f-49c7-b76d-6d57e0e08775
293 lines
8.7 KiB
Python
293 lines
8.7 KiB
Python
|
|
from wxPython.wx import *
|
|
from wxPython.ogl import *
|
|
|
|
import images
|
|
|
|
#----------------------------------------------------------------------
|
|
# This creates some pens and brushes that the OGL library uses.
|
|
|
|
wxOGLInitialize()
|
|
|
|
#----------------------------------------------------------------------
|
|
|
|
class DiamondShape(wxPolygonShape):
|
|
def __init__(self, w=0.0, h=0.0):
|
|
wxPolygonShape.__init__(self)
|
|
if w == 0.0:
|
|
w = 60.0
|
|
if h == 0.0:
|
|
h = 60.0
|
|
|
|
## Either wxRealPoints or 2-tuples of floats works.
|
|
|
|
#points = [ wxRealPoint(0.0, -h/2.0),
|
|
# wxRealPoint(w/2.0, 0.0),
|
|
# wxRealPoint(0.0, h/2.0),
|
|
# wxRealPoint(-w/2.0, 0.0),
|
|
# ]
|
|
points = [ (0.0, -h/2.0),
|
|
(w/2.0, 0.0),
|
|
(0.0, h/2.0),
|
|
(-w/2.0, 0.0),
|
|
]
|
|
|
|
self.Create(points)
|
|
|
|
|
|
#----------------------------------------------------------------------
|
|
|
|
class RoundedRectangleShape(wxRectangleShape):
|
|
def __init__(self, w=0.0, h=0.0):
|
|
wxRectangleShape.__init__(self, w, h)
|
|
self.SetCornerRadius(-0.3)
|
|
|
|
|
|
#----------------------------------------------------------------------
|
|
|
|
class DividedShape(wxDividedShape):
|
|
def __init__(self, width, height, canvas):
|
|
wxDividedShape.__init__(self, width, height)
|
|
|
|
region1 = wxShapeRegion()
|
|
region1.SetText('wxDividedShape')
|
|
region1.SetProportions(0.0, 0.2)
|
|
region1.SetFormatMode(FORMAT_CENTRE_HORIZ)
|
|
self.AddRegion(region1)
|
|
|
|
region2 = wxShapeRegion()
|
|
region2.SetText('This is Region number two.')
|
|
region2.SetProportions(0.0, 0.3)
|
|
region2.SetFormatMode(FORMAT_CENTRE_HORIZ|FORMAT_CENTRE_VERT)
|
|
self.AddRegion(region2)
|
|
|
|
region3 = wxShapeRegion()
|
|
region3.SetText('Region 3\nwith embedded\nline breaks')
|
|
region3.SetProportions(0.0, 0.5)
|
|
region3.SetFormatMode(FORMAT_NONE)
|
|
self.AddRegion(region3)
|
|
|
|
self.SetRegionSizes()
|
|
self.ReformatRegions(canvas)
|
|
|
|
|
|
def ReformatRegions(self, canvas=None):
|
|
rnum = 0
|
|
if canvas is None:
|
|
canvas = self.GetCanvas()
|
|
dc = wxClientDC(canvas) # used for measuring
|
|
for region in self.GetRegions():
|
|
text = region.GetText()
|
|
self.FormatText(dc, text, rnum)
|
|
rnum += 1
|
|
|
|
|
|
def OnSizingEndDragLeft(self, pt, x, y, keys, attch):
|
|
self.base_OnSizingEndDragLeft(pt, x, y, keys, attch)
|
|
self.SetRegionSizes()
|
|
self.ReformatRegions()
|
|
self.GetCanvas().Refresh()
|
|
|
|
|
|
#----------------------------------------------------------------------
|
|
|
|
class MyEvtHandler(wxShapeEvtHandler):
|
|
def __init__(self, log, frame):
|
|
wxShapeEvtHandler.__init__(self)
|
|
self.log = log
|
|
self.statbarFrame = frame
|
|
|
|
def UpdateStatusBar(self, shape):
|
|
x,y = shape.GetX(), shape.GetY()
|
|
width, height = shape.GetBoundingBoxMax()
|
|
self.statbarFrame.SetStatusText("Pos: (%d,%d) Size: (%d, %d)" %
|
|
(x, y, width, height))
|
|
|
|
|
|
def OnLeftClick(self, x, y, keys = 0, attachment = 0):
|
|
shape = self.GetShape()
|
|
print shape.__class__, shape.GetClassName()
|
|
canvas = shape.GetCanvas()
|
|
dc = wxClientDC(canvas)
|
|
canvas.PrepareDC(dc)
|
|
|
|
if shape.Selected():
|
|
shape.Select(False, dc)
|
|
canvas.Redraw(dc)
|
|
else:
|
|
redraw = False
|
|
shapeList = canvas.GetDiagram().GetShapeList()
|
|
toUnselect = []
|
|
for s in shapeList:
|
|
if s.Selected():
|
|
# If we unselect it now then some of the objects in
|
|
# shapeList will become invalid (the control points are
|
|
# shapes too!) and bad things will happen...
|
|
toUnselect.append(s)
|
|
|
|
shape.Select(True, dc)
|
|
|
|
if toUnselect:
|
|
for s in toUnselect:
|
|
s.Select(False, dc)
|
|
canvas.Redraw(dc)
|
|
|
|
self.UpdateStatusBar(shape)
|
|
|
|
|
|
def OnEndDragLeft(self, x, y, keys = 0, attachment = 0):
|
|
shape = self.GetShape()
|
|
self.base_OnEndDragLeft(x, y, keys, attachment)
|
|
if not shape.Selected():
|
|
self.OnLeftClick(x, y, keys, attachment)
|
|
self.UpdateStatusBar(shape)
|
|
|
|
|
|
def OnSizingEndDragLeft(self, pt, x, y, keys, attch):
|
|
self.base_OnSizingEndDragLeft(pt, x, y, keys, attch)
|
|
self.UpdateStatusBar(self.GetShape())
|
|
|
|
|
|
def OnMovePost(self, dc, x, y, oldX, oldY, display):
|
|
self.base_OnMovePost(dc, x, y, oldX, oldY, display)
|
|
self.UpdateStatusBar(self.GetShape())
|
|
|
|
|
|
def OnRightClick(self, *dontcare):
|
|
self.log.WriteText("%s\n" % self.GetShape())
|
|
|
|
|
|
#----------------------------------------------------------------------
|
|
|
|
class TestWindow(wxShapeCanvas):
|
|
def __init__(self, parent, log, frame):
|
|
wxShapeCanvas.__init__(self, parent)
|
|
|
|
maxWidth = 1000
|
|
maxHeight = 1000
|
|
self.SetScrollbars(20, 20, maxWidth/20, maxHeight/20)
|
|
|
|
self.log = log
|
|
self.frame = frame
|
|
self.SetBackgroundColour("LIGHT BLUE") #wxWHITE)
|
|
self.diagram = wxDiagram()
|
|
self.SetDiagram(self.diagram)
|
|
self.diagram.SetCanvas(self)
|
|
self.shapes = []
|
|
self.save_gdi = []
|
|
|
|
rRectBrush = wxBrush("MEDIUM TURQUOISE", wxSOLID)
|
|
dsBrush = wxBrush("WHEAT", wxSOLID)
|
|
|
|
self.MyAddShape(wxCircleShape(80), 100, 100, wxPen(wxBLUE, 3), wxGREEN_BRUSH, "Circle")
|
|
self.MyAddShape(wxRectangleShape(85, 50), 305, 60, wxBLACK_PEN, wxLIGHT_GREY_BRUSH, "Rectangle")
|
|
ds = self.MyAddShape(DividedShape(140, 150, self), 495, 145, wxBLACK_PEN, dsBrush, '')
|
|
self.MyAddShape(DiamondShape(90, 90), 345, 235, wxPen(wxBLUE, 3, wxDOT), wxRED_BRUSH, "Polygon")
|
|
self.MyAddShape(RoundedRectangleShape(95,70), 140, 255, wxPen(wxRED, 2), rRectBrush, "Rounded Rect")
|
|
|
|
bmp = images.getTest2Bitmap()
|
|
mask = wxMaskColour(bmp, wxBLUE)
|
|
bmp.SetMask(mask)
|
|
|
|
s = wxBitmapShape()
|
|
s.SetBitmap(bmp)
|
|
self.MyAddShape(s, 225, 150, None, None, "Bitmap")
|
|
|
|
dc = wxClientDC(self)
|
|
self.PrepareDC(dc)
|
|
for x in range(len(self.shapes)):
|
|
fromShape = self.shapes[x]
|
|
if x+1 == len(self.shapes):
|
|
toShape = self.shapes[0]
|
|
else:
|
|
toShape = self.shapes[x+1]
|
|
line = wxLineShape()
|
|
line.SetCanvas(self)
|
|
line.SetPen(wxBLACK_PEN)
|
|
line.SetBrush(wxBLACK_BRUSH)
|
|
line.AddArrow(ARROW_ARROW)
|
|
line.MakeLineControlPoints(2)
|
|
fromShape.AddLine(line, toShape)
|
|
self.diagram.AddShape(line)
|
|
line.Show(True)
|
|
|
|
# for some reason, the shapes have to be moved for the line to show up...
|
|
fromShape.Move(dc, fromShape.GetX(), fromShape.GetY())
|
|
|
|
EVT_WINDOW_DESTROY(self, self.OnDestroy)
|
|
|
|
|
|
def MyAddShape(self, shape, x, y, pen, brush, text):
|
|
shape.SetDraggable(True, True)
|
|
shape.SetCanvas(self)
|
|
shape.SetX(x)
|
|
shape.SetY(y)
|
|
if pen: shape.SetPen(pen)
|
|
if brush: shape.SetBrush(brush)
|
|
if text: shape.AddText(text)
|
|
#shape.SetShadowMode(SHADOW_RIGHT)
|
|
self.diagram.AddShape(shape)
|
|
shape.Show(True)
|
|
|
|
evthandler = MyEvtHandler(self.log, self.frame)
|
|
evthandler.SetShape(shape)
|
|
evthandler.SetPreviousHandler(shape.GetEventHandler())
|
|
shape.SetEventHandler(evthandler)
|
|
|
|
self.shapes.append(shape)
|
|
return shape
|
|
|
|
|
|
def OnDestroy(self, evt):
|
|
# Do some cleanup
|
|
for shape in self.diagram.GetShapeList():
|
|
if shape.GetParent() == None:
|
|
shape.SetCanvas(None)
|
|
shape.Destroy()
|
|
self.diagram.Destroy()
|
|
|
|
|
|
def OnBeginDragLeft(self, x, y, keys):
|
|
self.log.write("OnBeginDragLeft: %s, %s, %s\n" % (x, y, keys))
|
|
|
|
def OnEndDragLeft(self, x, y, keys):
|
|
self.log.write("OnEndDragLeft: %s, %s, %s\n" % (x, y, keys))
|
|
|
|
|
|
#----------------------------------------------------------------------
|
|
|
|
def runTest(frame, nb, log):
|
|
win = TestWindow(nb, log, frame)
|
|
return win
|
|
|
|
#----------------------------------------------------------------------
|
|
|
|
class __Cleanup:
|
|
cleanup = wxOGLCleanUp
|
|
def __del__(self):
|
|
self.cleanup()
|
|
|
|
# when this module gets cleaned up then wxOGLCleanUp() will get called
|
|
__cu = __Cleanup()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
overview = """\
|
|
The Object Graphics Library is a library supporting the creation and
|
|
manipulation of simple and complex graphic images on a canvas.
|
|
|
|
"""
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
import sys,os
|
|
import run
|
|
run.main(['', os.path.basename(sys.argv[0])])
|
|
|