Rename modules
Rename modules with better name and move the arduino code to its own folder
This commit is contained in:
101
gcodeextractor/gcode/GcodeCreator.py
Normal file
101
gcodeextractor/gcode/GcodeCreator.py
Normal file
@@ -0,0 +1,101 @@
|
||||
from gcodeextractor.models.Coordinates import Coordinate
|
||||
|
||||
|
||||
def findEndOfLine(image, direction, line, column, sequence):
|
||||
"""
|
||||
Start from a pixel and finds the end of a line of pixel in the direction specified
|
||||
:param image: image the sequence is created from
|
||||
:param direction: direction to go to find the end of line
|
||||
:param line: line index of the pixel to apply the function
|
||||
:param column: column index of the pixel to apply the function
|
||||
:param sequence: list of coordinates to append the end of line coordinates
|
||||
:return:
|
||||
"""
|
||||
|
||||
distance = 0
|
||||
|
||||
if direction == 0:
|
||||
while (line - distance) >= 0 and image[line - distance][column] == 2:
|
||||
image[line - distance][column] = 3
|
||||
distance += 1
|
||||
sequence.append(Coordinate(column, line - (distance - 1)))
|
||||
image[line - (distance-1)][column] = 2
|
||||
|
||||
elif direction == 1:
|
||||
while (column + distance) < len(image[line]) and image[line][column + distance] == 2:
|
||||
image[line][column + distance] = 3
|
||||
distance += 1
|
||||
sequence.append(Coordinate(column + (distance - 1), line))
|
||||
image[line][column + (distance-1)] = 2
|
||||
|
||||
elif direction == 2:
|
||||
while (line + distance) < len(image) and image[line + distance][column] == 2:
|
||||
image[line + distance][column] = 3
|
||||
distance += 1
|
||||
sequence.append(Coordinate(column, line + (distance - 1)))
|
||||
image[line + (distance-1)][column] = 2
|
||||
|
||||
elif direction == 3:
|
||||
while (column - distance) >= 0 and image[line][column - distance] == 2:
|
||||
image[line][column - distance] = 3
|
||||
distance += 1
|
||||
sequence.append(Coordinate(column - (distance - 1), line))
|
||||
image[line][column - (distance-1)] = 2
|
||||
|
||||
elif direction == -1:
|
||||
image[line][column] = 3
|
||||
|
||||
return distance
|
||||
|
||||
|
||||
def findDirection(image, line, column):
|
||||
"""
|
||||
Looks for a nearby pixel to choose a direction to start a line
|
||||
:param image: image the sequence is created from
|
||||
:param line: line index of the pixel to apply the function
|
||||
:param column: column index of the pixel to apply the function
|
||||
:return: direction of a nearby pixel, up = 0, right = 1, down = 2, left = 3, no pixel = -1. priorities is this order
|
||||
"""
|
||||
|
||||
if line != 0:
|
||||
if image[line-1][column] == 2:
|
||||
return 0
|
||||
|
||||
if column != (len(image[0]) - 1):
|
||||
if image[line][column+1] == 2:
|
||||
return 1
|
||||
|
||||
if line != (len(image)-1):
|
||||
if image[line+1][column] == 2:
|
||||
return 2
|
||||
|
||||
if column != 0:
|
||||
if image[line][column-1] == 2:
|
||||
return 3
|
||||
|
||||
return -1
|
||||
|
||||
|
||||
def createSequence(image):
|
||||
"""
|
||||
Create a sequence of index coordinates from a path matrix
|
||||
:param image: matrix of the path the tool needs to take
|
||||
:return: Array of class coordinates (X , Y)
|
||||
"""
|
||||
width = len(image[0])
|
||||
height = len(image)
|
||||
sequence = []
|
||||
|
||||
for line in range(height):
|
||||
for column in range(width):
|
||||
|
||||
if image[line][column] == 2:
|
||||
sequence.append(Coordinate(-1, -1))
|
||||
sequence.append(Coordinate(column, line))
|
||||
direction = 4
|
||||
while direction > -1:
|
||||
x, y = sequence[-1].getX(), sequence[-1].getY()
|
||||
direction = findDirection(image, y, x)
|
||||
findEndOfLine(image, direction, y, x, sequence)
|
||||
|
||||
return sequence
|
||||
Reference in New Issue
Block a user