Skip to content

Change XRP example 1 to transpose arrow buttons #6

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
wants to merge 1 commit into
base: develop
Choose a base branch
from
Draft

Change XRP example 1 to transpose arrow buttons #6

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
43 changes: 14 additions & 29 deletions red_vision_examples/xrp_examples/ex01_touch_screen_drive.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -39,16 +39,11 @@
stop_background_color = (0, 0, 255)

def create_ui_image():
# Initialize arrow button images. This could be done with a single image
# that gets transposed and flipped, but ulab's transpose() doesn't support
# the axes argument:
# https://github.com/v923z/micropython-ulab/issues/731
# So we instead create separate images for vertical and horizontal arrows
img_arrow_vertical = np.zeros(button_shape, dtype=np.uint8)
img_arrow_vertical[:, :] = arrow_background_color
img_arrow_horizontal = img_arrow_vertical.copy()
img_arrow_vertical = cv.arrowedLine(
img_arrow_vertical,
# Initialize arrow button image
img_arrow = np.zeros(button_shape, dtype=np.uint8)
img_arrow[:, :] = arrow_background_color
img_arrow = cv.arrowedLine(
img_arrow,
(button_cx, button_cy + arrow_length // 2),
(button_cx, button_cy - arrow_length // 2),
button_color,
Expand All @@ -57,16 +52,6 @@ def create_ui_image():
0,
arrow_tip_length
)
img_arrow_horizontal = cv.arrowedLine(
img_arrow_horizontal,
(button_cx - arrow_length // 2, button_cy),
(button_cx + arrow_length // 2, button_cy),
button_color,
arrow_thickness,
cv.FILLED,
0,
arrow_tip_length
)

# Initialize stop button image
img_button_stop = np.zeros(button_shape, dtype=np.uint8)
Expand All @@ -92,25 +77,25 @@ def create_ui_image():
img_ui[
ui_cy-button_spacing-button_cy:ui_cy-button_spacing+button_cy,
ui_cx-button_cx:ui_cx+button_cx
] = img_arrow_vertical
] = img_arrow

# Draw the backward arrow below the stop button
img_ui[
ui_cy+button_spacing-button_cy:ui_cy+button_spacing+button_cy,
ui_cx-button_cx:ui_cx+button_cx
] = img_arrow_vertical[::-1, :] # Flip the arrow image vertically
# Draw the right arrow to the right of the stop button
] = img_arrow[::-1, :] # Flip the arrow image vertically

# Draw the left arrow to the left of the stop button
img_ui[
ui_cy-button_cy:ui_cy+button_cy,
ui_cx+button_spacing-button_cx:ui_cx+button_spacing+button_cx
] = img_arrow_horizontal
ui_cx-button_spacing-button_cx:ui_cx-button_spacing+button_cx
] = img_arrow.transpose((1, 0, 2))

# Draw the left arrow to the left of the stop button
# Draw the right arrow to the right of the stop button
img_ui[
ui_cy-button_cy:ui_cy+button_cy,
ui_cx-button_spacing-button_cx:ui_cx-button_spacing+button_cx
] = img_arrow_horizontal[:, ::-1] # Flip the arrow image horizontally
ui_cx+button_spacing-button_cx:ui_cx+button_spacing+button_cx
] = img_arrow.transpose((1, 0, 2))[:, ::-1] # Flip the arrow image horizontally

# Return the UI image
return img_ui
Expand Down