1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98
|
from PyQt4 import QtCore
class Guide(object):
def __init__(self, follows=None):
self.scaleX = 1.0
self.scaleY = 1.0
if follows is not None:
while follows.nextGuide is not follows.firstGuide:
follows = follows.nextGuide
follows.nextGuide = self
self.prevGuide = follows
self.firstGuide = follows.firstGuide
self.nextGuide = follows.firstGuide
self.startLength = int(follows.startLength + follows.length()) + 1
else:
self.prevGuide = self
self.firstGuide = self
self.nextGuide = self
self.startLength = 0
def setScale(self, scaleX, scaleY, all=True):
self.scaleX = scaleX
self.scaleY = scaleY
if all:
next = self.nextGuide
while next is not self:
next.scaleX = scaleX
next.scaleY = scaleY
next = next.nextGuide
def setFence(self, fence, all=True):
self.fence = fence
if all:
next = self.nextGuide
while next is not self:
next.fence = fence
next = next.nextGuide
def lengthAll(self):
len = self.length()
next = self.nextGuide
while next is not self:
len += next.length()
next = next.nextGuide
return len
def move(self, item, dest, moveSpeed):
walkLine = QtCore.QLineF(item.getGuidedPos(), dest)
if moveSpeed >= 0 and walkLine.length() > moveSpeed:
# The item is too far away from it's destination point so we move
# it towards it instead.
dx = walkLine.dx()
dy = walkLine.dy()
if abs(dx) > abs(dy):
# Walk along x-axis.
if dx != 0:
d = moveSpeed * dy / abs(dx)
if dx > 0:
s = moveSpeed
else:
s = -moveSpeed
dest.setX(item.getGuidedPos().x() + s)
dest.setY(item.getGuidedPos().y() + d)
else:
# Walk along y-axis.
if dy != 0:
d = moveSpeed * dx / abs(dy)
if dy > 0:
s = moveSpeed
else:
s = -moveSpeed
dest.setX(item.getGuidedPos().x() + d)
dest.setY(item.getGuidedPos().y() + s)
item.setGuidedPos(dest)
def startPos(self):
return QtCore.QPointF(0, 0)
def endPos(self):
return QtCore.QPointF(0, 0)
def length(self):
return 1.0
def guide(self, item, moveSpeed):
raise NotImplementedError
|