Skip to content

Commit

Permalink
axis_twist_compensation: Applied suggested changes and made a small i…
Browse files Browse the repository at this point in the history 
…mprovement for better code readability.

igned-off-by: Jorge Apaza Merma <[email protected]>
  • Loading branch information
@yochiwarez
yochiwarez committed Oct 18, 2024
1 parent e93f077 commit ea06b2f
Showing 1 changed file with 45 additions and 36 deletions.
81 changes: 45 additions & 36 deletions klippy/extras/axis_twist_compensation.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -23,9 +23,9 @@ def __init__(self, config):
self.horizontal_move_z = config.getfloat('horizontal_move_z',
DEFAULT_HORIZONTAL_MOVE_Z)
self.speed = config.getfloat('speed', DEFAULT_SPEED)
self.calibrate_start_x = config.getfloat('calibrate_start_x')
self.calibrate_end_x = config.getfloat('calibrate_end_x')
self.calibrate_y = config.getfloat('calibrate_y')
self.calibrate_start_x = config.getfloat('calibrate_start_x', default=None)
self.calibrate_end_x = config.getfloat('calibrate_end_x', default=None)
self.calibrate_y = config.getfloat('calibrate_y', default=None)
self.z_compensations = config.getlists('z_compensations',
default=[], parser=float)
self.compensation_start_x = config.getfloat('compensation_start_x',
Expand Down Expand Up @@ -113,8 +113,8 @@ def __init__(self, compensation, config):
compensation.calibrate_y)
self.y_start_point = (compensation.calibrate_x,
compensation.calibrate_start_y)
self.y_end_point = (compensation.calibrate_end_y,
compensation.calibrate_x)
self.y_end_point = (compensation.calibrate_x,
compensation.calibrate_end_y)
self.results = None
self.current_point_index = None
self.gcmd = None
Expand Down Expand Up @@ -164,10 +164,34 @@ def cmd_AXIS_TWIST_COMPENSATION_CALIBRATE(self, gcmd):
# clear the current config
self.compensation.clear_compensations()

# calculate some values
# calculate the points to put the probe at, returned as a list of tuples
nozzle_points = []

if axis == 'X':

if not all([
self.x_start_point[0],
self.x_end_point[0],
self.x_start_point[1]
]):
raise self.gcmd.error(
"""AXIS_TWIST_COMPENSATION for X axis requires
calibrate_start_x, calibrate_end_x and calibrate_y
to be defined
"""
)

start_point = self.x_start_point
end_point = self.x_end_point

x_axis_range = end_point[0] - start_point[0]
interval_dist = x_axis_range / (sample_count - 1)

for i in range(sample_count):
x = start_point[0] + i * interval_dist
y = start_point[1]
nozzle_points.append((x, y))

elif axis == 'Y':

if not all([
Expand All @@ -182,32 +206,21 @@ def cmd_AXIS_TWIST_COMPENSATION_CALIBRATE(self, gcmd):
"""
)

start_point = (self.y_start_point[1], self.y_start_point[0])
start_point = self.y_start_point
end_point = self.y_end_point
else:
raise self.gcmd.error(
"AXIS_TWIST_COMPENSATION_CALIBRATE: "
"Invalid axis.")

axis_range = end_point[0] - start_point[0]

interval_dist = axis_range / (sample_count - 1)

# calculate the points to put the probe at, returned as a list of tuples
nozzle_points = []
y_axis_range = end_point[1] - start_point[1]
interval_dist = y_axis_range / (sample_count - 1)

if axis == 'X':
for i in range(sample_count):
x = start_point[0] + i * interval_dist
y = start_point[1]
nozzle_points.append((x, y))

elif axis == 'Y':
for i in range(sample_count):
x = start_point[1]
y = start_point[0] + i * interval_dist
x = start_point[0]
y = start_point[1] + i * interval_dist
nozzle_points.append((x, y))

else:
raise self.gcmd.error(
"AXIS_TWIST_COMPENSATION_CALIBRATE: "
"Invalid axis.")

probe_points = self._calculate_probe_points(
nozzle_points, self.probe_x_offset, self.probe_y_offset)
Expand Down Expand Up @@ -289,20 +302,17 @@ def cmd_AXIS_TWIST_COMPENSATION_AUTOCALIBRATE(self, gcmd):
# clear the current config
self.compensation.clear_compensations()

# calculate the points to put the probe at, returned as a list of tuples
nozzle_points = []

min_x = self.x_start_point[0]
max_x = self.x_end_point[0]
min_y = self.y_start_point[1]
max_y = self.y_end_point[0]
max_y = self.y_end_point[1]

# calculate x positions
interval_x = (max_x - min_x) / (sample_count - 1)
xps = [min_x + interval_x * i for i in range(sample_count)]

# Calculate points array
spcy = (max_y - min_y) / (sample_count - 1)
interval_y = (max_y - min_y) / (sample_count - 1)
flip = False

points = []
Expand All @@ -312,7 +322,7 @@ def cmd_AXIS_TWIST_COMPENSATION_AUTOCALIBRATE(self, gcmd):
idx = j
else:
idx = sample_count -1 - j
points.append([xps[i], min_y + spcy * idx ])
points.append([xps[i], min_y + interval_y * idx ])
flip = not flip


Expand Down Expand Up @@ -346,8 +356,7 @@ def cmd_AXIS_TWIST_COMPENSATION_AUTOCALIBRATE(self, gcmd):
configfile.set(self.configname, 'compensation_start_y',
self.y_start_point[1])
configfile.set(self.configname, 'compensation_end_y',
self.y_end_point[0])

self.y_end_point[1])

self.gcmd.respond_info(
"AXIS_TWIST_COMPENSATION_AUTOCALIBRATE: Calibration complete: ")
Expand Down Expand Up @@ -475,11 +484,11 @@ def _finalize_calibration(self):
configfile.set(self.configname, 'compensation_start_y',
self.y_start_point[1])
configfile.set(self.configname, 'compensation_end_y',
self.y_end_point[0])
self.y_end_point[1])

self.compensation.zy_compensations = self.results
self.compensation.compensation_start_y = self.y_start_point[1]
self.compensation.compensation_end_y = self.y_end_point[0]
self.compensation.compensation_end_y = self.y_end_point[1]

self.gcode.respond_info(
"AXIS_TWIST_COMPENSATION state has been saved "
Expand Down

0 comments on commit ea06b2f

Please sign in to comment.