Rename modules
Rename modules with better name and move the arduino code to its own folder
This commit is contained in:
89
gcodeextractor/gcode/path.py
Normal file
89
gcodeextractor/gcode/path.py
Normal file
@@ -0,0 +1,89 @@
|
||||
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)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user