mirror of https://github.com/lalbornoz/roar.git
38 lines
1.8 KiB
Python
38 lines
1.8 KiB
Python
#!/usr/bin/env python3
|
|
#
|
|
# ToolFill.py
|
|
# Copyright (c) 2018, 2019 Lucio Andrés Illanes Albornoz <lucio@lucioillanes.de>
|
|
#
|
|
|
|
from Tool import Tool
|
|
|
|
class ToolFill(Tool):
|
|
name = "Fill"
|
|
|
|
#
|
|
# onMouseEvent(self, event, atPoint, brushColours, brushSize, isDragging, isLeftDown, isRightDown, dispatchFn, eventDc, viewRect)
|
|
def onMouseEvent(self, event, atPoint, brushColours, brushSize, isDragging, isLeftDown, isRightDown, dispatchFn, eventDc, viewRect):
|
|
pointStack, pointsDone = [list(atPoint)], []
|
|
testColour = self.parentCanvas.canvas.map[atPoint[1]][atPoint[0]][0:2]
|
|
if isLeftDown or isRightDown:
|
|
if isRightDown:
|
|
brushColours = [brushColours[1], brushColours[0]]
|
|
while len(pointStack) > 0:
|
|
point = pointStack.pop()
|
|
pointCell = self.parentCanvas.canvas.map[point[1]][point[0]]
|
|
if (pointCell[0:2] == testColour) \
|
|
or ((pointCell[3] == " ") and (pointCell[1] == testColour[1])):
|
|
if not point in pointsDone:
|
|
dispatchFn(eventDc, False, [*point, brushColours[0], brushColours[0], 0, " "], viewRect)
|
|
if point[0] > 0:
|
|
pointStack.append([point[0] - 1, point[1]])
|
|
if point[0] < (self.parentCanvas.canvas.size[0] - 1):
|
|
pointStack.append([point[0] + 1, point[1]])
|
|
if point[1] > 0:
|
|
pointStack.append([point[0], point[1] - 1])
|
|
if point[1] < (self.parentCanvas.canvas.size[1] - 1):
|
|
pointStack.append([point[0], point[1] + 1])
|
|
pointsDone += [point]
|
|
|
|
# vim:expandtab foldmethod=marker sw=4 ts=4 tw=120
|