blob: 4a3ddea8907cae8149066651ae595213a7ecaeb3 [file] [log] [blame]
Maxwell Henderson7af00982023-02-04 12:42:07 -08001#!/usr/bin/python3
2
3from __future__ import print_function
Maxwell Henderson765567c2023-03-05 14:22:17 -08004# matplotlib overrides fontconfig locations, so it needs to be imported before gtk.
5import matplotlib.pyplot as plt
Maxwell Henderson7af00982023-02-04 12:42:07 -08006import os
7from frc971.control_loops.python import basic_window
8from frc971.control_loops.python.color import Color, palette
9import random
10import gi
milind-u18a901d2023-02-17 21:51:55 -080011import numpy as np
Maxwell Henderson7af00982023-02-04 12:42:07 -080012
13gi.require_version('Gtk', '3.0')
14from gi.repository import Gdk, Gtk
15import cairo
Maxwell Henderson165b17e2023-03-09 17:14:04 -080016from y2023.control_loops.python.graph_tools import to_theta, to_xy, alpha_blend, shift_angles, get_xy
milind-u600738b2023-02-22 14:42:19 -080017from y2023.control_loops.python.graph_tools import l1, l2, joint_center
18from y2023.control_loops.python.graph_tools import DRIVER_CAM_POINTS
19from y2023.control_loops.python import graph_paths
Maxwell Henderson7af00982023-02-04 12:42:07 -080020
Maxwell Henderson83cf6d62023-02-10 20:29:26 -080021from frc971.control_loops.python.basic_window import quit_main_loop, set_color, OverrideMatrix, identity
Maxwell Henderson7af00982023-02-04 12:42:07 -080022
23import shapely
24from shapely.geometry import Polygon
25
Maxwell Hendersonbfaf5ef2023-03-22 13:31:32 -070026from frc971.control_loops.python.constants import *
27
Maxwell Henderson7af00982023-02-04 12:42:07 -080028
Maxwell Henderson83cf6d62023-02-10 20:29:26 -080029def px(cr):
30 return OverrideMatrix(cr, identity)
31
32
33# Draw lines to cr + stroke.
34def draw_lines(cr, lines):
35 cr.move_to(lines[0][0], lines[0][1])
36 for pt in lines[1:]:
37 cr.line_to(pt[0], pt[1])
38 with px(cr):
39 cr.stroke()
40
41
Maxwell Henderson7af00982023-02-04 12:42:07 -080042def draw_px_cross(cr, length_px):
43 """Draws a cross with fixed dimensions in pixel space."""
44 with px(cr):
45 x, y = cr.get_current_point()
46 cr.move_to(x, y - length_px)
47 cr.line_to(x, y + length_px)
48 cr.stroke()
49
50 cr.move_to(x - length_px, y)
51 cr.line_to(x + length_px, y)
52 cr.stroke()
53
54
55def angle_dist_sqr(a1, a2):
56 """Distance between two points in angle space."""
57 return (a1[0] - a2[0])**2 + (a1[1] - a2[1])**2
58
59
60# Find the highest y position that intersects the vertical line defined by x.
61def inter_y(x):
milind-u18a901d2023-02-17 21:51:55 -080062 return np.sqrt((l2 + l1)**2 - (x - joint_center[0])**2) + joint_center[1]
Maxwell Henderson7af00982023-02-04 12:42:07 -080063
64
65# Define min and max l1 angles based on vertical constraints.
66def get_angle(boundary):
milind-u18a901d2023-02-17 21:51:55 -080067 h = np.sqrt((l1)**2 - (boundary - joint_center[0])**2) + joint_center[1]
68 return np.arctan2(h, boundary - joint_center[0])
Maxwell Henderson7af00982023-02-04 12:42:07 -080069
70
Maxwell Henderson7af00982023-02-04 12:42:07 -080071# Rotate a rasterized loop such that it aligns to when the parameters loop
72def rotate_to_jump_point(points):
73 last_pt = points[0]
74 for pt_i in range(1, len(points)):
75 pt = points[pt_i]
76 delta = last_pt[1] - pt[1]
milind-u18a901d2023-02-17 21:51:55 -080077 if abs(delta) > np.pi:
Maxwell Henderson7af00982023-02-04 12:42:07 -080078 return points[pt_i:] + points[:pt_i]
79 last_pt = pt
80 return points
81
82
83# shift points vertically by dy.
84def y_shift(points, dy):
85 return [(x, y + dy) for x, y in points]
86
87
Maxwell Henderson7af00982023-02-04 12:42:07 -080088# Get the closest point to a line from a test pt.
89def get_closest(prev, cur, pt):
90 dx_ang = (cur[0] - prev[0])
91 dy_ang = (cur[1] - prev[1])
92
milind-u18a901d2023-02-17 21:51:55 -080093 d = np.sqrt(dx_ang**2 + dy_ang**2)
Maxwell Henderson7af00982023-02-04 12:42:07 -080094 if (d < 0.000001):
milind-u18a901d2023-02-17 21:51:55 -080095 return prev, np.sqrt((prev[0] - pt[0])**2 + (prev[1] - pt[1])**2)
Maxwell Henderson7af00982023-02-04 12:42:07 -080096
97 pdx = -dy_ang / d
98 pdy = dx_ang / d
99
100 dpx = pt[0] - prev[0]
101 dpy = pt[1] - prev[1]
102
103 alpha = (dx_ang * dpx + dy_ang * dpy) / d / d
104
105 if (alpha < 0):
milind-u18a901d2023-02-17 21:51:55 -0800106 return prev, np.sqrt((prev[0] - pt[0])**2 + (prev[1] - pt[1])**2)
Maxwell Henderson7af00982023-02-04 12:42:07 -0800107 elif (alpha > 1):
milind-u18a901d2023-02-17 21:51:55 -0800108 return cur, np.sqrt((cur[0] - pt[0])**2 + (cur[1] - pt[1])**2)
Maxwell Henderson7af00982023-02-04 12:42:07 -0800109 else:
110 return (alpha_blend(prev[0], cur[0], alpha), alpha_blend(prev[1], cur[1], alpha)), \
111 abs(dpx * pdx + dpy * pdy)
112
113
114def closest_segment(lines, pt):
115 c_pt, c_pt_dist = get_closest(lines[-1], lines[0], pt)
116 for i in range(1, len(lines)):
117 prev = lines[i - 1]
118 cur = lines[i]
119 c_pt_new, c_pt_new_dist = get_closest(prev, cur, pt)
120 if c_pt_new_dist < c_pt_dist:
121 c_pt = c_pt_new
122 c_pt_dist = c_pt_new_dist
123 return c_pt, c_pt_dist
124
125
Maxwell Henderson4b8d02a2023-02-24 15:53:26 -0800126# Defining outline of robot
127class RobotOutline():
128
129 def __init__(self, drivetrain_pos, drivetrain_width, joint_center_radius,
130 joint_tower_pos, joint_tower_width, joint_tower_height,
131 driver_cam_pos, driver_cam_width, driver_cam_height):
132 self.drivetrain_pos = drivetrain_pos
133 self.drivetrain_width = drivetrain_width
134
135 self.joint_center_radius = joint_center_radius
136 self.joint_tower_pos = joint_tower_pos
137 self.joint_tower_width = joint_tower_width
138 self.joint_tower_height = joint_tower_height
139
140 self.driver_cam_pos = driver_cam_pos
141 self.driver_cam_width = driver_cam_width
142 self.driver_cam_height = driver_cam_height
143
144 def draw(self, cr):
145 set_color(cr, palette["BLUE"])
146
147 cr.move_to(self.drivetrain_pos[0], self.drivetrain_pos[1])
148 cr.line_to(self.drivetrain_pos[0] + self.drivetrain_width,
149 self.drivetrain_pos[1])
150
151 with px(cr):
152 cr.stroke()
153
154 # Draw joint center
155 cr.arc(joint_center[0], joint_center[1], self.joint_center_radius, 0,
156 2.0 * np.pi)
157 with px(cr):
158 cr.stroke()
159
160 # Draw joint tower
161 cr.rectangle(self.joint_tower_pos[0], self.joint_tower_pos[1],
162 self.joint_tower_width, self.joint_tower_height)
163
164 with px(cr):
165 cr.stroke()
166
167 # Draw driver cam
168 cr.set_source_rgba(1, 0, 0, 0.5)
169 cr.rectangle(self.driver_cam_pos[0], self.driver_cam_pos[1],
170 self.driver_cam_width, self.driver_cam_height)
171
172 with px(cr):
173 cr.fill()
174
175 def draw_theta(self, cr):
176 # TOOD(Max): add theta mode drawing
177 pass
178
179
180DRIVETRAIN_X = -0.490
181DRIVETRAIN_Y = 0.184
182DRIVETRAIN_WIDTH = 0.980
183
184JOINT_CENTER_RADIUS = 0.173 / 2
185
186JOINT_TOWER_X = -0.252
187JOINT_TOWER_Y = DRIVETRAIN_Y
188JOINT_TOWER_WIDTH = 0.098
189JOINT_TOWER_HEIGHT = 0.864
190
191DRIVER_CAM_X = DRIVER_CAM_POINTS[0][0]
192DRIVER_CAM_Y = DRIVER_CAM_POINTS[0][1]
193DRIVER_CAM_WIDTH = DRIVER_CAM_POINTS[-1][0] - DRIVER_CAM_POINTS[0][0]
194DRIVER_CAM_HEIGHT = DRIVER_CAM_POINTS[-1][1] - DRIVER_CAM_POINTS[0][1]
195
196
Maxwell Henderson0b5f1d12023-03-08 20:01:35 -0800197class SegmentSelector(basic_window.BaseWindow):
198
199 def __init__(self, segments):
200 super(SegmentSelector, self).__init__()
201
202 self.window = Gtk.Window()
203 self.window.set_title("Segment Selector")
204
205 self.segments = segments
206
207 self.segment_store = Gtk.ListStore(int, str)
208
209 for i, segment in enumerate(segments):
210 self.segment_store.append([i, segment.name])
211
212 self.segment_box = Gtk.ComboBox.new_with_model_and_entry(
213 self.segment_store)
214 self.segment_box.connect("changed", self.on_combo_changed)
215 self.segment_box.set_entry_text_column(1)
216
217 self.current_path_index = None
218
219 self.window.add(self.segment_box)
220 self.window.show_all()
221
222 def on_combo_changed(self, combo):
223 iter = combo.get_active_iter()
224
225 if iter is not None:
226 model = combo.get_model()
227 id, name = model[iter][:2]
228 print("Selected: ID=%d, name=%s" % (id, name))
229 self.current_path_index = id
230
231
Maxwell Henderson7af00982023-02-04 12:42:07 -0800232# Create a GTK+ widget on which we will draw using Cairo
Maxwell Hendersonbfaf5ef2023-03-22 13:31:32 -0700233class ArmUi(Gtk.DrawingArea):
Maxwell Henderson7af00982023-02-04 12:42:07 -0800234
Maxwell Henderson0b5f1d12023-03-08 20:01:35 -0800235 def __init__(self, segments):
milind-u18a901d2023-02-17 21:51:55 -0800236 super(ArmUi, self).__init__()
Maxwell Henderson7af00982023-02-04 12:42:07 -0800237
Maxwell Hendersonbfaf5ef2023-03-22 13:31:32 -0700238 self.set_size_request(2 * SCREEN_SIZE, SCREEN_SIZE)
239 self.center = (0, 0)
240 self.shape = (2 * SCREEN_SIZE, SCREEN_SIZE)
Maxwell Henderson7af00982023-02-04 12:42:07 -0800241 self.theta_version = False
Maxwell Hendersonbfaf5ef2023-03-22 13:31:32 -0700242
243 self.init_extents()
244
245 self.connect('draw', self.on_draw)
Maxwell Henderson7af00982023-02-04 12:42:07 -0800246
Austin Schuh9a11ebd2023-02-26 14:16:31 -0800247 self.last_pos = to_xy(*graph_paths.points['Neutral'][:2])
milind-u600738b2023-02-22 14:42:19 -0800248 self.circular_index_select = 1
Maxwell Henderson7af00982023-02-04 12:42:07 -0800249
250 # Extra stuff for drawing lines.
Maxwell Henderson0b5f1d12023-03-08 20:01:35 -0800251 self.segments = segments
Maxwell Henderson7af00982023-02-04 12:42:07 -0800252 self.prev_segment_pt = None
253 self.now_segment_pt = None
254 self.spline_edit = 0
255 self.edit_control1 = True
256
milind-uadd8fa32023-02-24 23:37:36 -0800257 self.joint_thetas = None
258 self.joint_points = None
milind-u18a901d2023-02-17 21:51:55 -0800259 self.fig = plt.figure()
milind-uadd8fa32023-02-24 23:37:36 -0800260 self.axes = [
261 self.fig.add_subplot(3, 1, 1),
262 self.fig.add_subplot(3, 1, 2),
263 self.fig.add_subplot(3, 1, 3)
264 ]
265 self.fig.subplots_adjust(hspace=1.0)
milind-u18a901d2023-02-17 21:51:55 -0800266 plt.show(block=False)
267
Maxwell Hendersonf5c08a52023-02-23 17:28:35 -0800268 self.index = 0
269
Maxwell Henderson4b8d02a2023-02-24 15:53:26 -0800270 self.outline = RobotOutline([DRIVETRAIN_X, DRIVETRAIN_Y],
271 DRIVETRAIN_WIDTH, JOINT_CENTER_RADIUS,
272 [JOINT_TOWER_X, JOINT_TOWER_Y],
273 JOINT_TOWER_WIDTH, JOINT_TOWER_HEIGHT,
274 [DRIVER_CAM_X, DRIVER_CAM_Y],
275 DRIVER_CAM_WIDTH, DRIVER_CAM_HEIGHT)
276
Maxwell Hendersonc088b842023-03-11 07:22:02 -0800277 self.show_indicators = True
Maxwell Henderson4e57e0c2023-03-11 12:25:04 -0800278 # Lets you only view selected path
279 self.view_current = False
Maxwell Hendersonc088b842023-03-11 07:22:02 -0800280
Maxwell Henderson4890c202023-03-21 12:09:04 -0700281 self.editing = True
282
Maxwell Hendersonbfaf5ef2023-03-22 13:31:32 -0700283 self.x_offset = 0
284 self.y_offset = 0
285
Maxwell Henderson7af00982023-02-04 12:42:07 -0800286 def _do_button_press_internal(self, event):
287 o_x = event.x
288 o_y = event.y
Maxwell Hendersonbfaf5ef2023-03-22 13:31:32 -0700289 event.y -= self.y_offset
290 event.x -= self.x_offset
Maxwell Henderson7af00982023-02-04 12:42:07 -0800291 x = event.x - self.window_shape[0] / 2
292 y = self.window_shape[1] / 2 - event.y
293 scale = self.get_current_scale()
294 event.x = x / scale + self.center[0]
295 event.y = y / scale + self.center[1]
296 self.do_button_press(event)
297 event.x = o_x
298 event.y = o_y
299
Maxwell Henderson7af00982023-02-04 12:42:07 -0800300 def _do_configure(self, event):
301 self.window_shape = (event.width, event.height)
302
Maxwell Hendersonbfaf5ef2023-03-22 13:31:32 -0700303 def init_extents(self):
Maxwell Henderson7af00982023-02-04 12:42:07 -0800304 if self.theta_version:
milind-u18a901d2023-02-17 21:51:55 -0800305 self.extents_x_min = -np.pi * 2
306 self.extents_x_max = np.pi * 2
307 self.extents_y_min = -np.pi * 2
308 self.extents_y_max = np.pi * 2
Maxwell Henderson7af00982023-02-04 12:42:07 -0800309 else:
310 self.extents_x_min = -40.0 * 0.0254
311 self.extents_x_max = 40.0 * 0.0254
312 self.extents_y_min = -4.0 * 0.0254
313 self.extents_y_max = 110.0 * 0.0254
314
Maxwell Hendersonbfaf5ef2023-03-22 13:31:32 -0700315 self.center = (0.5 * (self.extents_x_min + self.extents_x_max),
316 0.5 * (self.extents_y_max + self.extents_y_min))
317 self.shape = (1.0 * (self.extents_x_max - self.extents_x_min),
318 1.0 * (self.extents_y_max - self.extents_y_min))
319
320 def get_current_scale(self):
321 w_w, w_h = self.window_shape
322 w, h = self.shape
323 return min((w_w / w), (w_h / h))
324
325 def on_draw(self, widget, event):
326 cr = self.get_window().cairo_create()
327
328 self.window_shape = (self.get_window().get_geometry().width,
329 self.get_window().get_geometry().height)
330
331 cr.save()
332 cr.set_font_size(20)
333 cr.translate(self.window_shape[0] / 2, self.window_shape[1] / 2)
334 scale = self.get_current_scale()
335 cr.scale(scale, -scale)
336 cr.translate(-self.center[0], -self.center[1])
337 cr.reset_clip()
338 self.handle_draw(cr)
339 cr.restore()
340
341 def method_connect(self, event, cb):
342
343 def handler(obj, *args):
344 cb(*args)
345
346 self.window.connect(event, handler)
Maxwell Henderson7af00982023-02-04 12:42:07 -0800347
348 # Handle the expose-event by drawing
349 def handle_draw(self, cr):
Maxwell Henderson7af00982023-02-04 12:42:07 -0800350 # Fill the background color of the window with grey
351 set_color(cr, palette["GREY"])
352 cr.paint()
353
354 # Draw a extents rectangle
355 set_color(cr, palette["WHITE"])
356 cr.rectangle(self.extents_x_min, self.extents_y_min,
357 (self.extents_x_max - self.extents_x_min),
358 self.extents_y_max - self.extents_y_min)
359 cr.fill()
360
Maxwell Henderson93380322023-02-04 16:31:54 -0800361 if self.theta_version:
362 # Draw a filled white rectangle.
363 set_color(cr, palette["WHITE"])
milind-u18a901d2023-02-17 21:51:55 -0800364 cr.rectangle(-np.pi, -np.pi, np.pi * 2.0, np.pi * 2.0)
Maxwell Henderson93380322023-02-04 16:31:54 -0800365 cr.fill()
366
367 set_color(cr, palette["BLUE"])
368 for i in range(-6, 6):
milind-u18a901d2023-02-17 21:51:55 -0800369 cr.move_to(-40, -40 + i * np.pi)
370 cr.line_to(40, 40 + i * np.pi)
Maxwell Henderson93380322023-02-04 16:31:54 -0800371 with px(cr):
372 cr.stroke()
373
Maxwell Henderson93380322023-02-04 16:31:54 -0800374 set_color(cr, Color(0.0, 1.0, 0.2))
375 cr.move_to(self.last_pos[0], self.last_pos[1])
376 draw_px_cross(cr, 5)
377
Maxwell Henderson93380322023-02-04 16:31:54 -0800378 else:
Maxwell Henderson7af00982023-02-04 12:42:07 -0800379 # Draw a filled white rectangle.
380 set_color(cr, palette["WHITE"])
381 cr.rectangle(-2.0, -2.0, 4.0, 4.0)
382 cr.fill()
383
Maxwell Henderson4b8d02a2023-02-24 15:53:26 -0800384 self.outline.draw(cr)
milind-u18a901d2023-02-17 21:51:55 -0800385
386 # Draw max radius
Maxwell Henderson7af00982023-02-04 12:42:07 -0800387 set_color(cr, palette["BLUE"])
milind-u18a901d2023-02-17 21:51:55 -0800388 cr.arc(joint_center[0], joint_center[1], l2 + l1, 0, 2.0 * np.pi)
389 with px(cr):
390 cr.stroke()
391 cr.arc(joint_center[0], joint_center[1], l1 - l2, 0, 2.0 * np.pi)
Maxwell Henderson7af00982023-02-04 12:42:07 -0800392 with px(cr):
393 cr.stroke()
394
milind-u18a901d2023-02-17 21:51:55 -0800395 set_color(cr, Color(0.5, 1.0, 1))
Maxwell Henderson7af00982023-02-04 12:42:07 -0800396
milind-u600738b2023-02-22 14:42:19 -0800397 if not self.theta_version:
398 theta1, theta2 = to_theta(self.last_pos,
399 self.circular_index_select)
400 x, y = joint_center[0], joint_center[1]
401 cr.move_to(x, y)
402
403 x += np.cos(theta1) * l1
404 y += np.sin(theta1) * l1
405 cr.line_to(x, y)
406 x += np.cos(theta2) * l2
407 y += np.sin(theta2) * l2
408 cr.line_to(x, y)
409 with px(cr):
410 cr.stroke()
411
412 cr.move_to(self.last_pos[0], self.last_pos[1])
413 set_color(cr, Color(0.0, 1.0, 0.2))
414 draw_px_cross(cr, 20)
415
416 if self.theta_version:
417 set_color(cr, Color(0.0, 1.0, 0.2))
418 cr.move_to(self.last_pos[0], self.last_pos[1])
419 draw_px_cross(cr, 5)
420
Maxwell Henderson4b8d02a2023-02-24 15:53:26 -0800421 self.outline.draw_theta(cr)
milind-u600738b2023-02-22 14:42:19 -0800422
Maxwell Henderson7af00982023-02-04 12:42:07 -0800423 set_color(cr, Color(0.0, 0.5, 1.0))
Maxwell Henderson4e57e0c2023-03-11 12:25:04 -0800424 if not self.view_current:
425 for i in range(len(self.segments)):
426 color = None
427 if i == self.index:
428 continue
429 color = [0, random.random(), 1]
430 random.shuffle(color)
431 set_color(cr, Color(color[0], color[1], color[2]))
432 self.segments[i].DrawTo(cr, self.theta_version)
Maxwell Henderson165b17e2023-03-09 17:14:04 -0800433
Maxwell Henderson4e57e0c2023-03-11 12:25:04 -0800434 with px(cr):
435 cr.stroke()
Maxwell Henderson7af00982023-02-04 12:42:07 -0800436
Austin Schuhdba32702023-03-04 22:51:05 -0800437 # Draw current spline in black
438 color = [0, 0, 0]
439 set_color(cr, Color(color[0], color[1], color[2]))
440 self.segments[self.index].DrawTo(cr, self.theta_version)
Maxwell Henderson165b17e2023-03-09 17:14:04 -0800441
Maxwell Hendersonc088b842023-03-11 07:22:02 -0800442 with px(cr):
443 cr.stroke()
Maxwell Henderson165b17e2023-03-09 17:14:04 -0800444 control1 = get_xy(self.segments[self.index].control1)
445 control2 = get_xy(self.segments[self.index].control2)
446
447 if self.theta_version:
448 control1 = shift_angles(self.segments[self.index].control1)
449 control2 = shift_angles(self.segments[self.index].control2)
450
Maxwell Hendersonc088b842023-03-11 07:22:02 -0800451 if self.show_indicators:
452 set_color(cr, Color(1.0, 0.0, 1.0))
453 cr.move_to(control1[0] + 0.02, control1[1])
454 cr.arc(control1[0], control1[1], 0.02, 0, 2.0 * np.pi)
455 with px(cr):
456 cr.stroke()
457 set_color(cr, Color(1.0, 0.7, 0.0))
458 cr.move_to(control2[0] + 0.02, control2[1])
459 cr.arc(control2[0], control2[1], 0.02, 0, 2.0 * np.pi)
Maxwell Henderson165b17e2023-03-09 17:14:04 -0800460
Austin Schuhdba32702023-03-04 22:51:05 -0800461 with px(cr):
462 cr.stroke()
463
Maxwell Henderson7af00982023-02-04 12:42:07 -0800464 set_color(cr, Color(0.0, 1.0, 0.5))
milind-u18a901d2023-02-17 21:51:55 -0800465
milind-uadd8fa32023-02-24 23:37:36 -0800466 # Create the plots
467 if self.joint_thetas:
468 if self.joint_points:
469 titles = ["Proximal", "Distal", "Roll joint"]
470 for i in range(len(self.joint_points)):
471 self.axes[i].clear()
472 self.axes[i].plot(self.joint_thetas[0],
473 self.joint_thetas[1][i])
474 self.axes[i].scatter([self.joint_points[i][0]],
475 [self.joint_points[i][1]],
476 s=10,
477 c="red")
478 self.axes[i].set_title(titles[i])
479 plt.title("Joint Angle")
milind-u18a901d2023-02-17 21:51:55 -0800480 plt.xlabel("t (0 to 1)")
481 plt.ylabel("theta (rad)")
482
483 self.fig.canvas.draw()
Maxwell Henderson7af00982023-02-04 12:42:07 -0800484
485 def cur_pt_in_theta(self):
milind-u18a901d2023-02-17 21:51:55 -0800486 if self.theta_version: return self.last_pos
milind-u600738b2023-02-22 14:42:19 -0800487 return to_theta(self.last_pos,
488 self.circular_index_select,
489 cross_point=-np.pi,
490 die=False)
Maxwell Henderson7af00982023-02-04 12:42:07 -0800491
milind-u18a901d2023-02-17 21:51:55 -0800492 def do_motion(self, event):
493 o_x = event.x
494 o_y = event.y
Maxwell Hendersonbfaf5ef2023-03-22 13:31:32 -0700495 event.x -= self.x_offset
496 event.y -= self.y_offset
milind-u18a901d2023-02-17 21:51:55 -0800497 x = event.x - self.window_shape[0] / 2
498 y = self.window_shape[1] / 2 - event.y
499 scale = self.get_current_scale()
500 event.x = x / scale + self.center[0]
501 event.y = y / scale + self.center[1]
502
milind-u52af04c2023-02-25 14:03:08 -0800503 segment = self.segments[self.index]
504 self.joint_thetas = segment.joint_thetas()
milind-u18a901d2023-02-17 21:51:55 -0800505
milind-u52af04c2023-02-25 14:03:08 -0800506 hovered_t = segment.intersection(event)
507 if hovered_t:
508 min_diff = np.inf
509 closest_t = None
510 closest_thetas = None
511 for i in range(len(self.joint_thetas[0])):
512 t = self.joint_thetas[0][i]
513 diff = abs(t - hovered_t)
514 if diff < min_diff:
515 min_diff = diff
516 closest_t = t
517 closest_thetas = [
518 self.joint_thetas[1][0][i], self.joint_thetas[1][1][i],
519 self.joint_thetas[1][2][i]
520 ]
521 self.joint_points = [(closest_t, closest_theta)
522 for closest_theta in closest_thetas]
milind-u18a901d2023-02-17 21:51:55 -0800523
524 event.x = o_x
525 event.y = o_y
526
Maxwell Hendersonbfaf5ef2023-03-22 13:31:32 -0700527 self.queue_draw()
Maxwell Henderson7af00982023-02-04 12:42:07 -0800528
Maxwell Henderson75c70122023-03-22 20:16:47 -0700529 def switch_theta(self):
530 # Toggle between theta and xy renderings
531 if self.theta_version:
532 theta1, theta2 = self.last_pos
533 data = to_xy(theta1, theta2)
534 self.circular_index_select = int(
535 np.floor((theta2 - theta1) / np.pi))
536 self.last_pos = (data[0], data[1])
537 else:
538 self.last_pos = self.cur_pt_in_theta()
539
540 self.theta_version = not self.theta_version
541 self.init_extents()
542
Maxwell Henderson7af00982023-02-04 12:42:07 -0800543 def do_key_press(self, event):
544 keyval = Gdk.keyval_to_lower(event.keyval)
545 print("Gdk.KEY_" + Gdk.keyval_name(keyval))
546 if keyval == Gdk.KEY_q:
547 print("Found q key and exiting.")
548 quit_main_loop()
549 elif keyval == Gdk.KEY_c:
550 # Increment which arm solution we render
551 self.circular_index_select += 1
552 print(self.circular_index_select)
553 elif keyval == Gdk.KEY_v:
554 # Decrement which arm solution we render
555 self.circular_index_select -= 1
556 print(self.circular_index_select)
Maxwell Henderson7af00982023-02-04 12:42:07 -0800557
558 elif keyval == Gdk.KEY_r:
559 self.prev_segment_pt = self.now_segment_pt
560
Maxwell Hendersonf5c08a52023-02-23 17:28:35 -0800561 elif keyval == Gdk.KEY_o:
562 # Only prints current segment
563 print(repr(self.segments[self.index]))
Maxwell Henderson7af00982023-02-04 12:42:07 -0800564 elif keyval == Gdk.KEY_g:
565 # Generate theta points.
566 if self.segments:
Maxwell Hendersonf5c08a52023-02-23 17:28:35 -0800567 print(repr(self.segments[self.index].ToThetaPoints()))
Maxwell Henderson7af00982023-02-04 12:42:07 -0800568 elif keyval == Gdk.KEY_e:
569 best_pt = self.now_segment_pt
570 best_dist = 1e10
571 for segment in self.segments:
572 d = angle_dist_sqr(segment.start, self.now_segment_pt)
573 if (d < best_dist):
574 best_pt = segment.start
575 best_dist = d
576 d = angle_dist_sqr(segment.end, self.now_segment_pt)
577 if (d < best_dist):
578 best_pt = segment.end
579 best_dist = d
580 self.now_segment_pt = best_pt
581
Austin Schuha09e1bb2023-02-25 22:34:41 -0800582 elif keyval == Gdk.KEY_p:
583 if self.index > 0:
584 self.index -= 1
585 else:
586 self.index = len(self.segments) - 1
587 print("Switched to segment:", self.segments[self.index].name)
Austin Schuh9a11ebd2023-02-26 14:16:31 -0800588 self.segments[self.index].Print(graph_paths.points)
Austin Schuha09e1bb2023-02-25 22:34:41 -0800589
Maxwell Hendersonc088b842023-03-11 07:22:02 -0800590 elif keyval == Gdk.KEY_i:
591 self.show_indicators = not self.show_indicators
592
Austin Schuha09e1bb2023-02-25 22:34:41 -0800593 elif keyval == Gdk.KEY_n:
Maxwell Hendersonf5c08a52023-02-23 17:28:35 -0800594 self.index += 1
595 self.index = self.index % len(self.segments)
milind-u2a28c592023-02-24 23:13:04 -0800596 print("Switched to segment:", self.segments[self.index].name)
Austin Schuh9a11ebd2023-02-26 14:16:31 -0800597 self.segments[self.index].Print(graph_paths.points)
Maxwell Hendersonf5c08a52023-02-23 17:28:35 -0800598
Maxwell Henderson4890c202023-03-21 12:09:04 -0700599 elif keyval == Gdk.KEY_d:
600 self.editing = not self.editing
601
Maxwell Henderson4e57e0c2023-03-11 12:25:04 -0800602 elif keyval == Gdk.KEY_l:
603 self.view_current = not self.view_current
604
Maxwell Henderson7af00982023-02-04 12:42:07 -0800605 elif keyval == Gdk.KEY_t:
Maxwell Henderson75c70122023-03-22 20:16:47 -0700606 self.switch_theta()
Maxwell Henderson7af00982023-02-04 12:42:07 -0800607
608 elif keyval == Gdk.KEY_z:
609 self.edit_control1 = not self.edit_control1
610 if self.edit_control1:
Maxwell Hendersonf5c08a52023-02-23 17:28:35 -0800611 self.now_segment_pt = self.segments[self.index].control1
Maxwell Henderson7af00982023-02-04 12:42:07 -0800612 else:
Maxwell Hendersonf5c08a52023-02-23 17:28:35 -0800613 self.now_segment_pt = self.segments[self.index].control2
Maxwell Henderson7af00982023-02-04 12:42:07 -0800614 if not self.theta_version:
615 data = to_xy(self.now_segment_pt[0], self.now_segment_pt[1])
616 self.last_pos = (data[0], data[1])
617 else:
618 self.last_pos = self.now_segment_pt
619
620 print("self.last_pos: ", self.last_pos, " ci: ",
621 self.circular_index_select)
622
Maxwell Hendersonbfaf5ef2023-03-22 13:31:32 -0700623 self.queue_draw()
Maxwell Henderson7af00982023-02-04 12:42:07 -0800624
625 def do_button_press(self, event):
Maxwell Henderson4890c202023-03-21 12:09:04 -0700626
milind-u600738b2023-02-22 14:42:19 -0800627 last_pos = self.last_pos
Maxwell Henderson7af00982023-02-04 12:42:07 -0800628 self.last_pos = (event.x, event.y)
milind-u600738b2023-02-22 14:42:19 -0800629 pt_theta = self.cur_pt_in_theta()
630 if pt_theta is None:
631 self.last_pos = last_pos
632 return
633
634 self.now_segment_pt = np.array(shift_angles(pt_theta))
Maxwell Henderson7af00982023-02-04 12:42:07 -0800635
Maxwell Henderson4890c202023-03-21 12:09:04 -0700636 if self.editing:
637 if self.edit_control1:
638 self.segments[self.index].control1 = self.now_segment_pt
639 else:
640 self.segments[self.index].control2 = self.now_segment_pt
Maxwell Henderson7af00982023-02-04 12:42:07 -0800641
Austin Schuh9a11ebd2023-02-26 14:16:31 -0800642 print('Clicked at theta: np.array([%s, %s])' %
643 (self.now_segment_pt[0], self.now_segment_pt[1]))
Maxwell Henderson7af00982023-02-04 12:42:07 -0800644 if not self.theta_version:
Austin Schuh9a11ebd2023-02-26 14:16:31 -0800645 print(
646 'Clicked at to_theta_with_circular_index(%.3f, %.3f, circular_index=%d)'
647 % (self.last_pos[0], self.last_pos[1],
Maxwell Henderson7af00982023-02-04 12:42:07 -0800648 self.circular_index_select))
649
Austin Schuh9a11ebd2023-02-26 14:16:31 -0800650 self.segments[self.index].Print(graph_paths.points)
Maxwell Henderson7af00982023-02-04 12:42:07 -0800651
Maxwell Hendersonbfaf5ef2023-03-22 13:31:32 -0700652 self.queue_draw()
Maxwell Henderson7af00982023-02-04 12:42:07 -0800653
654
Maxwell Hendersonbfaf5ef2023-03-22 13:31:32 -0700655class Window(Gtk.Window):
656
657 def __init__(self, segments):
658 super().__init__(title="Drawing Area")
659
660 self.segment_store = Gtk.ListStore(int, str)
661
662 for i, segment in enumerate(segments):
663 self.segment_store.append([i, segment.name])
664
665 self.segment_box = Gtk.ComboBox.new_with_model_and_entry(
666 self.segment_store)
667 self.segment_box.connect("changed", self.on_combo_changed)
668 self.segment_box.set_entry_text_column(1)
669
670 self.arm_draw = ArmUi(segments)
671
672 self.arm_draw.y_offset = self.segment_box.get_allocation().width
673
674 print('Starting with segment: ',
675 self.arm_draw.segments[self.arm_draw.index].name)
676 self.arm_draw.segments[self.arm_draw.index].Print(graph_paths.points)
677
678 self.set_events(Gdk.EventMask.BUTTON_PRESS_MASK
679 | Gdk.EventMask.BUTTON_RELEASE_MASK
680 | Gdk.EventMask.POINTER_MOTION_MASK
681 | Gdk.EventMask.SCROLL_MASK
682 | Gdk.EventMask.KEY_PRESS_MASK)
683 self.method_connect('map-event', self.do_map_event)
684 self.method_connect("key-press-event", self.arm_draw.do_key_press)
685 self.method_connect("motion-notify-event", self.arm_draw.do_motion)
686 self.method_connect("button-press-event",
687 self.arm_draw._do_button_press_internal)
688 self.method_connect("configure-event", self.arm_draw._do_configure)
689
690 self.grid = Gtk.Grid()
691 self.add(self.grid)
692
693 self.grid.attach(self.arm_draw, 0, 1, 1, 1)
694
Maxwell Henderson75c70122023-03-22 20:16:47 -0700695 self.isolate_button = Gtk.Button(label="Toggle Path Isolation")
696 self.isolate_button.connect('clicked', self.on_button_click)
697
698 self.theta_button = Gtk.Button(label="Toggle Theta Mode")
699 self.theta_button.connect('clicked', self.on_button_click)
700
701 self.editing_button = Gtk.Button(label="Toggle Editing Mode")
702 self.editing_button.connect('clicked', self.on_button_click)
703
704 self.indicator_button = Gtk.Button(
705 label="Toggle Control Point Indicators")
706 self.indicator_button.connect('clicked', self.on_button_click)
707
708 self.box = Gtk.Box(spacing=6)
709 self.grid.attach(self.box, 0, 0, 1, 1)
710
711 self.box.pack_start(self.segment_box, False, False, 0)
712 self.box.pack_start(self.isolate_button, False, False, 0)
713 self.box.pack_start(self.theta_button, False, False, 0)
714 self.box.pack_start(self.editing_button, False, False, 0)
715 self.box.pack_start(self.indicator_button, False, False, 0)
Maxwell Hendersonbfaf5ef2023-03-22 13:31:32 -0700716
717 def on_combo_changed(self, combo):
718 iter = combo.get_active_iter()
719
720 if iter is not None:
721 model = combo.get_model()
722 id, name = model[iter][:2]
723 print("Selected: ID=%d, name=%s" % (id, name))
724 self.arm_draw.index = id
Maxwell Henderson75c70122023-03-22 20:16:47 -0700725 self.arm_draw.queue_draw()
Maxwell Hendersonbfaf5ef2023-03-22 13:31:32 -0700726
727 def method_connect(self, event, cb):
728
729 def handler(obj, *args):
730 cb(*args)
731
732 self.connect(event, handler)
733
734 def do_map_event(self, event):
Maxwell Henderson75c70122023-03-22 20:16:47 -0700735 self.arm_draw.y_offset = self.box.get_allocation().height
736
737 def on_button_click(self, button):
738 if self.isolate_button == button:
739 self.arm_draw.view_current = not self.arm_draw.view_current
740 elif self.theta_button == button:
741 self.arm_draw.switch_theta()
742 elif self.editing_button == button:
743 self.arm_draw.editing = not self.arm_draw.editing
744 elif self.indicator_button == button:
745 self.arm_draw.show_indicators = not self.arm_draw.show_indicators
746 self.arm_draw.queue_draw()
Maxwell Hendersonbfaf5ef2023-03-22 13:31:32 -0700747
748
749window = Window(graph_paths.segments)
750window.show_all()
Maxwell Henderson7af00982023-02-04 12:42:07 -0800751basic_window.RunApp()