90 lines
2.6 KiB
Python
90 lines
2.6 KiB
Python
def scanHorizontal(image, rTool):
|
|
"""
|
|
:param image: image to apply the scan
|
|
:param rTool: tool radius
|
|
:return: adds 2's where there should be a path for the tool, looking at the image only horizontally
|
|
"""
|
|
|
|
width = len(image[0])
|
|
height = len(image)
|
|
|
|
for line in range(height):
|
|
for column in range(width):
|
|
|
|
if column > 0:
|
|
if image[line][column] == 1 and image[line][column - 1] != 1:
|
|
for px in range(2 * rTool + 1):
|
|
if column - rTool >= 0 and 0 <= line - rTool + px < height:
|
|
if image[line - rTool + px][column - rTool] == 0:
|
|
image[line - rTool + px][column - rTool] = 2
|
|
|
|
if column < width - 1:
|
|
if image[line][column] == 1 and image[line][column + 1] != 1:
|
|
for px in range(2 * rTool + 1):
|
|
if column + rTool < width and 0 <= line - rTool + px < height:
|
|
if image[line - rTool + px][column + rTool] == 0:
|
|
image[line - rTool + px][column + rTool] = 2
|
|
|
|
return image
|
|
|
|
|
|
def scanVertical(image, rTool):
|
|
"""
|
|
:param image: image to apply the scan
|
|
:param rTool: tool radius
|
|
:return: adds 2's where there should be a path for the tool, looking at the image only vertically
|
|
"""
|
|
|
|
width = len(image[0])
|
|
height = len(image)
|
|
|
|
for line in range(height):
|
|
for column in range(width):
|
|
|
|
if line > 0:
|
|
if image[line][column] == 1 and image[line - 1][column] != 1:
|
|
for px in range(2 * rTool + 1):
|
|
if line - rTool >= 0 and 0 <= column - rTool + px < width:
|
|
if image[line - rTool][column - rTool + px] == 0:
|
|
image[line - rTool][column - rTool + px] = 2
|
|
|
|
if line < height - 1:
|
|
if image[line][column] == 1 and image[line + 1][column] != 1:
|
|
for px in range(2 * rTool + 1):
|
|
if line + rTool < height and 0 <= column - rTool + px < width:
|
|
if image[line + rTool][column - rTool + px] == 0:
|
|
image[line + rTool][column - rTool + px] = 2
|
|
|
|
return image
|
|
|
|
def twoRemoving(image, rTool):
|
|
"""
|
|
:param image: image to apply the scan
|
|
:param rTool: tool radius
|
|
:return: removes unnecessary twos to leave a path of only one pixel large
|
|
"""
|
|
|
|
width = len(image[0])
|
|
height = len(image)
|
|
|
|
for line in range(height):
|
|
for column in range(width):
|
|
|
|
if image[line][column] == 1:
|
|
for pixelx in range(1, 2 * rTool):
|
|
for pixely in range(1, 2*rTool):
|
|
if 0 <= line - rTool + pixelx < height and 0 <= column - rTool + pixely < width:
|
|
if image[line - rTool + pixelx][column - rTool + pixely] == 2:
|
|
image[line - rTool + pixelx][column - rTool + pixely] = 0
|
|
|
|
print("image width = " + str(width))
|
|
print("image height = " + str(height))
|
|
|
|
return image
|
|
|
|
|
|
def path(image, rTool):
|
|
return twoRemoving(scanVertical(scanHorizontal(image, rTool), rTool), rTool)
|
|
|
|
|