245 lines
6.4 KiB
Python
245 lines
6.4 KiB
Python
#! /usr/bin/env python
|
|
|
|
from Tkinter import *
|
|
from Canvas import Oval, Group, CanvasText
|
|
|
|
|
|
# Fix a bug in Canvas.Group as distributed in Python 1.4. The
|
|
# distributed bind() method is broken. This is what should be used:
|
|
|
|
class Group(Group):
|
|
def bind(self, sequence=None, command=None):
|
|
return self.canvas.tag_bind(self.id, sequence, command)
|
|
|
|
class Object:
|
|
|
|
"""Base class for composite graphical objects.
|
|
|
|
Objects belong to a canvas, and can be moved around on the canvas.
|
|
They also belong to at most one ``pile'' of objects, and can be
|
|
transferred between piles (or removed from their pile).
|
|
|
|
Objects have a canonical ``x, y'' position which is moved when the
|
|
object is moved. Where the object is relative to this position
|
|
depends on the object; for simple objects, it may be their center.
|
|
|
|
Objects have mouse sensitivity. They can be clicked, dragged and
|
|
double-clicked. The behavior may actually determined by the pile
|
|
they are in.
|
|
|
|
All instance attributes are public since the derived class may
|
|
need them.
|
|
|
|
"""
|
|
|
|
def __init__(self, canvas, x=0, y=0, fill='red', text='object'):
|
|
self.canvas = canvas
|
|
self.x = x
|
|
self.y = y
|
|
self.pile = None
|
|
self.group = Group(self.canvas)
|
|
self.createitems(fill, text)
|
|
|
|
def __str__(self):
|
|
return str(self.group)
|
|
|
|
def createitems(self, fill, text):
|
|
self.__oval = Oval(self.canvas,
|
|
self.x-20, self.y-10, self.x+20, self.y+10,
|
|
fill=fill, width=3)
|
|
self.group.addtag_withtag(self.__oval)
|
|
self.__text = CanvasText(self.canvas,
|
|
self.x, self.y, text=text)
|
|
self.group.addtag_withtag(self.__text)
|
|
|
|
def moveby(self, dx, dy):
|
|
if dx == dy == 0:
|
|
return
|
|
self.group.move(dx, dy)
|
|
self.x = self.x + dx
|
|
self.y = self.y + dy
|
|
|
|
def moveto(self, x, y):
|
|
self.moveby(x - self.x, y - self.y)
|
|
|
|
def transfer(self, pile):
|
|
if self.pile:
|
|
self.pile.delete(self)
|
|
self.pile = None
|
|
self.pile = pile
|
|
if self.pile:
|
|
self.pile.add(self)
|
|
|
|
def tkraise(self):
|
|
self.group.tkraise()
|
|
|
|
|
|
class Bottom(Object):
|
|
|
|
"""An object to serve as the bottom of a pile."""
|
|
|
|
def createitems(self, *args):
|
|
self.__oval = Oval(self.canvas,
|
|
self.x-20, self.y-10, self.x+20, self.y+10,
|
|
fill='gray', outline='')
|
|
self.group.addtag_withtag(self.__oval)
|
|
|
|
|
|
class Pile:
|
|
|
|
"""A group of graphical objects."""
|
|
|
|
def __init__(self, canvas, x, y, tag=None):
|
|
self.canvas = canvas
|
|
self.x = x
|
|
self.y = y
|
|
self.objects = []
|
|
self.bottom = Bottom(self.canvas, self.x, self.y)
|
|
self.group = Group(self.canvas, tag=tag)
|
|
self.group.addtag_withtag(self.bottom.group)
|
|
self.bindhandlers()
|
|
|
|
def bindhandlers(self):
|
|
self.group.bind('<1>', self.clickhandler)
|
|
self.group.bind('<Double-1>', self.doubleclickhandler)
|
|
|
|
def add(self, object):
|
|
self.objects.append(object)
|
|
self.group.addtag_withtag(object.group)
|
|
self.position(object)
|
|
|
|
def delete(self, object):
|
|
object.group.dtag(self.group)
|
|
self.objects.remove(object)
|
|
|
|
def position(self, object):
|
|
object.tkraise()
|
|
i = self.objects.index(object)
|
|
object.moveto(self.x + i*4, self.y + i*8)
|
|
|
|
def clickhandler(self, event):
|
|
pass
|
|
|
|
def doubleclickhandler(self, event):
|
|
pass
|
|
|
|
|
|
class MovingPile(Pile):
|
|
|
|
def bindhandlers(self):
|
|
Pile.bindhandlers(self)
|
|
self.group.bind('<B1-Motion>', self.motionhandler)
|
|
self.group.bind('<ButtonRelease-1>', self.releasehandler)
|
|
|
|
movethis = None
|
|
|
|
def clickhandler(self, event):
|
|
tags = self.canvas.gettags('current')
|
|
for i in range(len(self.objects)):
|
|
o = self.objects[i]
|
|
if o.group.tag in tags:
|
|
break
|
|
else:
|
|
self.movethis = None
|
|
return
|
|
self.movethis = self.objects[i:]
|
|
for o in self.movethis:
|
|
o.tkraise()
|
|
self.lastx = event.x
|
|
self.lasty = event.y
|
|
|
|
doubleclickhandler = clickhandler
|
|
|
|
def motionhandler(self, event):
|
|
if not self.movethis:
|
|
return
|
|
dx = event.x - self.lastx
|
|
dy = event.y - self.lasty
|
|
self.lastx = event.x
|
|
self.lasty = event.y
|
|
for o in self.movethis:
|
|
o.moveby(dx, dy)
|
|
|
|
def releasehandler(self, event):
|
|
objects = self.movethis
|
|
if not objects:
|
|
return
|
|
self.movethis = None
|
|
self.finishmove(objects)
|
|
|
|
def finishmove(self, objects):
|
|
for o in objects:
|
|
self.position(o)
|
|
|
|
|
|
class Pile1(MovingPile):
|
|
|
|
x = 50
|
|
y = 50
|
|
tag = 'p1'
|
|
|
|
def __init__(self, demo):
|
|
self.demo = demo
|
|
MovingPile.__init__(self, self.demo.canvas, self.x, self.y, self.tag)
|
|
|
|
def doubleclickhandler(self, event):
|
|
try:
|
|
o = self.objects[-1]
|
|
except IndexError:
|
|
return
|
|
o.transfer(self.other())
|
|
MovingPile.doubleclickhandler(self, event)
|
|
|
|
def other(self):
|
|
return self.demo.p2
|
|
|
|
def finishmove(self, objects):
|
|
o = objects[0]
|
|
p = self.other()
|
|
x, y = o.x, o.y
|
|
if (x-p.x)**2 + (y-p.y)**2 < (x-self.x)**2 + (y-self.y)**2:
|
|
for o in objects:
|
|
o.transfer(p)
|
|
else:
|
|
MovingPile.finishmove(self, objects)
|
|
|
|
class Pile2(Pile1):
|
|
|
|
x = 150
|
|
y = 50
|
|
tag = 'p2'
|
|
|
|
def other(self):
|
|
return self.demo.p1
|
|
|
|
|
|
class Demo:
|
|
|
|
def __init__(self, master):
|
|
self.master = master
|
|
self.canvas = Canvas(master,
|
|
width=200, height=200,
|
|
background='yellow',
|
|
relief=SUNKEN, borderwidth=2)
|
|
self.canvas.pack(expand=1, fill=BOTH)
|
|
self.p1 = Pile1(self)
|
|
self.p2 = Pile2(self)
|
|
o1 = Object(self.canvas, fill='red', text='o1')
|
|
o2 = Object(self.canvas, fill='green', text='o2')
|
|
o3 = Object(self.canvas, fill='light blue', text='o3')
|
|
o1.transfer(self.p1)
|
|
o2.transfer(self.p1)
|
|
o3.transfer(self.p2)
|
|
|
|
|
|
# Main function, run when invoked as a stand-alone Python program.
|
|
|
|
def main():
|
|
root = Tk()
|
|
demo = Demo(root)
|
|
root.protocol('WM_DELETE_WINDOW', root.quit)
|
|
root.mainloop()
|
|
|
|
if __name__ == '__main__':
|
|
main()
|