mirror of https://github.com/lalbornoz/roar.git
88 lines
3.6 KiB
Python
88 lines
3.6 KiB
Python
#!/usr/bin/env python3
|
|
#
|
|
# MiRCARTToolLine.py -- XXX
|
|
# Copyright (c) 2018 Lucio Andrés Illanes Albornoz <lucio@lucioillanes.de>
|
|
#
|
|
|
|
from MiRCARTTool import MiRCARTTool
|
|
|
|
class MiRCARTToolLine(MiRCARTTool):
|
|
"""XXX"""
|
|
name = "Line"
|
|
toolColours = toolOriginPoint = toolState = None
|
|
|
|
TS_NONE = 0
|
|
TS_ORIGIN = 1
|
|
|
|
# {{{ _pointDelta(self, a, b): XXX
|
|
def _pointDelta(self, a, b):
|
|
return [a2-a1 for a1, a2 in zip(a, b)]
|
|
# }}}
|
|
# {{{ _pointSwap(self, a, b): XXX
|
|
def _pointSwap(self, a, b):
|
|
return [b, a]
|
|
# }}}
|
|
# {{{ _getLine(self, brushColours, brushSize, eventDc, isCursor, originPoint, targetPoint, dispatchFn): XXX
|
|
def _getLine(self, brushColours, brushSize, eventDc, isCursor, originPoint, targetPoint, dispatchFn):
|
|
originPoint = originPoint.copy(); targetPoint = targetPoint.copy();
|
|
pointDelta = self._pointDelta(originPoint, targetPoint)
|
|
lineXSign = 1 if pointDelta[0] > 0 else -1;
|
|
lineYSign = 1 if pointDelta[1] > 0 else -1;
|
|
pointDelta = [abs(a) for a in pointDelta]
|
|
if pointDelta[0] > pointDelta[1]:
|
|
lineXX, lineXY, lineYX, lineYY = lineXSign, 0, 0, lineYSign
|
|
else:
|
|
pointDelta = [pointDelta[1], pointDelta[0]]
|
|
lineXX, lineXY, lineYX, lineYY = 0, lineYSign, lineXSign, 0
|
|
lineD = 2 * pointDelta[1] - pointDelta[0]; lineY = 0;
|
|
for lineX in range(pointDelta[0] + 1):
|
|
for brushStep in range(brushSize[0]):
|
|
patch = [ \
|
|
originPoint[0] + lineX*lineXX + lineY*lineYX + brushStep, \
|
|
originPoint[1] + lineX*lineXY + lineY*lineYY, \
|
|
*brushColours, 0, " "]
|
|
if isCursor:
|
|
dispatchFn(eventDc, False, patch); dispatchFn(eventDc, True, patch);
|
|
else:
|
|
dispatchFn(eventDc, True, patch)
|
|
if lineD > 0:
|
|
lineD -= pointDelta[0]; lineY += 1;
|
|
lineD += pointDelta[1]
|
|
# }}}
|
|
|
|
#
|
|
# onMouseEvent(self, event, atPoint, brushColours, brushSize, isDragging, isLeftDown, isRightDown, dispatchFn, eventDc): XXX
|
|
def onMouseEvent(self, event, atPoint, brushColours, brushSize, isDragging, isLeftDown, isRightDown, dispatchFn, eventDc):
|
|
brushColours = brushColours.copy()
|
|
if isLeftDown:
|
|
brushColours[1] = brushColours[0]
|
|
elif isRightDown:
|
|
brushColours[0] = brushColours[1]
|
|
else:
|
|
brushColours[1] = brushColours[0]
|
|
if self.toolState == self.TS_NONE:
|
|
if isLeftDown or isRightDown:
|
|
self.toolColours = brushColours
|
|
self.toolOriginPoint = list(atPoint)
|
|
self.toolState = self.TS_ORIGIN
|
|
dispatchFn(eventDc, True, [*atPoint, *brushColours, 0, " "])
|
|
elif self.toolState == self.TS_ORIGIN:
|
|
targetPoint = list(atPoint)
|
|
originPoint = self.toolOriginPoint
|
|
self._getLine(self.toolColours, brushSize, \
|
|
eventDc, isLeftDown or isRightDown, \
|
|
originPoint, targetPoint, dispatchFn)
|
|
if isLeftDown or isRightDown:
|
|
self.toolColours = None
|
|
self.toolOriginPoint = None
|
|
self.toolState = self.TS_NONE
|
|
|
|
# __init__(self, *args): initialisation method
|
|
def __init__(self, *args):
|
|
super().__init__(*args)
|
|
self.toolColours = None
|
|
self.toolOriginPoint = None
|
|
self.toolState = self.TS_NONE
|
|
|
|
# vim:expandtab foldmethod=marker sw=4 ts=4 tw=120
|