def draggable(name: str, was_down: bool, width: float, height: float) -> Tuple[str, bool]:
with style({im.STYLE_WINDOW_PADDING: (0,0)}), \
child(name=name+"_frame", border=True,
width=width, height=height):
prev_cursor_screen_pos = im.get_cursor_screen_position()
_ = im.invisible_button(name+"_invisible_button", width, height)
# _ = im.button("{},{}".format(width,height), width, height)
is_down = im.is_item_active()
status = {
(False, False): 'idle',
(False, True) : 'pressed',
(True, True) : 'held',
(True, False): 'released',
}[(was_down, is_down)]
im.set_cursor_screen_position(prev_cursor_screen_pos)
# The window takes no inputs, so it's okay that we
# draw the button first - clicks just pass through the window
with style({im.STYLE_WINDOW_PADDING: (0,0)}), \
child(name=name+"_content", flags=im.WINDOW_NO_INPUTS,
width=width, height=height):
yield (status, is_down)
return True
def invisible_button(label, width, height, tag=None):
""" Invisible button with a given width and height.
Can be interacted with as a normal button, including clicking and dragging.
Returns `(label, None)` on click, or `(tag, None)` if tag is specified.
"""
while not imgui.invisible_button(label, width, height):
yield
return tag if tag is not None else label, None
def render(self, play):
imgui.begin("sim")
dl = imgui.get_window_draw_list()
meterwidth = max(np.max(self.track[:, 0]), np.max(self.lm[:, 0]))
if play:
if imgui.button("pause"):
play = False
else:
if imgui.button("play"):
play = True
# sliders
_, xy = imgui.slider_float2("car xy", self.car.x, -self.car.y, 0,
meterwidth)
self.car.x, self.car.y = xy[0], -xy[1]
_, self.car.theta = imgui.slider_float("car theta", self.car.theta, 0,
2 * np.pi)
imgui.slider_float("car v", self.car.v, 0, 10)
imgui.slider_float2("controls", self.u, self.k, -1, 1)
_, self.lanewidth = imgui.slider_float("lane width", self.lanewidth,
0.1, 2.0)
# render overview
pos = imgui.get_cursor_screen_pos()
siz = imgui.get_content_region_available()
if siz[1] <= 0:
siz = [400, 400]
imgui.invisible_button("overview", siz[0], siz[1])
origin = pos
scale = siz[0] / (meterwidth * 1.1)
conecolor = imgui.get_color_u32_rgba(1, 0.7, 0, 1)
for lm in self.lm:
dl.add_rect_filled(origin[0] + lm[0] * scale - 2,
origin[1] - lm[1] * scale - 2,
origin[0] + lm[0] * scale + 2,
origin[1] - lm[1] * scale + 2, conecolor, 3)
trackcolor = imgui.get_color_u32_rgba(1, 1, 0.3, 0.5)
for i in range(0, len(self.track), 2):
j = (i + 1) % len(self.track)
ti = self.track[i] * scale
tj = self.track[j] * scale
dl.add_line(origin[0] + ti[0], origin[1] - ti[1],
origin[0] + tj[0], origin[1] - tj[1], trackcolor, 1.5)
for i in range(0, len(self.track)):
j = (i + 1) % len(self.track)
ti = self.track[i] * scale
tj = self.track[j] * scale
Li = ti[2:4] * self.lanewidth
Lj = tj[2:4] * self.lanewidth
dl.add_line(origin[0] + ti[0] + Li[0], origin[1] - ti[1] - Li[1],
origin[0] + tj[0] + Lj[0], origin[1] - tj[1] - Lj[1],
trackcolor, 3)
dl.add_line(origin[0] + ti[0] - Li[0], origin[1] - ti[1] + Li[1],
origin[0] + tj[0] - Lj[0], origin[1] - tj[1] + Lj[1],
trackcolor, 3)
mx = scale * self.car.x
my = scale * self.car.y
# draw forward projections
for traj in self.shots:
for i in range(len(traj) - 1):
dl.add_line(origin[0] + traj[i, 0] * scale,
origin[1] - traj[i, 1] * scale,
origin[0] + traj[i + 1, 0] * scale,
origin[1] - traj[i + 1, 1] * scale,
imgui.get_color_u32_rgba(0, 1, 1, 1), 1)
# draw car + velocity
vxy = scale * self.car.v * .1 * \
np.array([np.cos(self.car.theta), np.sin(self.car.theta)])
dl.add_line(origin[0] + mx, origin[1] - my, origin[0] + mx + vxy[0],
origin[1] - my - vxy[1],
imgui.get_color_u32_rgba(0, 1, 0, 1), 1)
imgui.end()
return play
def render_graphs(self):
imgui.begin("graphs")
i = self.i
dl = imgui.get_window_draw_list()
mi = max(0, i - 30 * 3)
temp = self.controlstate[mi:i + 1, 3].copy()
imgui.plot_lines("velocity", temp)
temp = self.controlstate[mi:i + 1, 8].copy()
imgui.plot_lines("target v", temp)
temp = self.controls[mi:i + 1, 0].copy()
imgui.plot_lines("control v", temp)
temp = self.controls[mi:i + 1, 1].copy()
imgui.plot_lines("control steer", temp)
temp = self.controlstate[mi:i + 1, 9].copy()
imgui.plot_lines("target w", temp)
temp = self.controlstate[mi:i + 1, 4].copy()
imgui.plot_lines("yaw rate", temp)
# live variables
maxv = int(np.ceil(np.max(self.controlstate[:, 3]) * 1.1))
imgui.slider_float("wheel v", self.controlstate[i, 3], 0, maxv)
imgui.slider_float("target v", self.controlstate[i, 8], 0, maxv)
lv = 0
if i > 0:
dx = self.controlstate[i, 0] - self.controlstate[i - 1, 0]
dy = self.controlstate[i, 1] - self.controlstate[i - 1, 1]
lv = np.sqrt(dx**2 + dy**2) * 30.0
imgui.slider_float("localized v", lv, 0, maxv)
imgui.slider_float("control motor", self.controls[i, 0] / 127., -1, 1)
imgui.slider_float("control steer", self.controls[i, 1] / 127., -1, 1)
# for yaw rate and curvature, set the limits backwards
# so that turning right is to the right
maxw = int(np.ceil(np.max(np.abs(self.controlstate[:, 4])) * 1.1))
imgui.slider_float("yaw rate", self.controlstate[i, 4], maxw, -maxw)
imgui.slider_float("target w", self.controlstate[i, 9], maxw, -maxw)
v = self.controlstate[i, 3]
if v > 0.5:
k = self.controlstate[i, 4] / v
else:
k = 0
imgui.slider_float("curvature", k, 2, -2)
targetK = self.controlstate[self.i, 7]
imgui.slider_float("target k", targetK, 2, -2)
# render overview
pos = imgui.get_cursor_screen_pos()
siz = imgui.get_content_region_available()
if siz[1] == 0:
siz = [400, 300]
imgui.invisible_button("overview", siz[0], siz[0] * 0.7)
origin = pos
meterwidth = max(np.max(self.track[:, 0]), np.max(self.lm[:, 0]))
scale = siz[0] / (meterwidth * 1.1)
conecolor = imgui.get_color_u32_rgba(1, 0.7, 0, 1)
for lm in self.lm:
dl.add_rect_filled(origin[0] + lm[0] * scale - 2,
origin[1] - lm[1] * scale - 2,
origin[0] + lm[0] * scale + 2,
origin[1] - lm[1] * scale + 2, conecolor, 3)
trackcolor = imgui.get_color_u32_rgba(1, 1, 0.3, 0.5)
for i in range(0, len(self.track), 2):
j = (i + 1) % len(self.track)
ti = self.track[i] * scale
tj = self.track[j] * scale
dl.add_line(origin[0] + ti[0], origin[1] - ti[1],
origin[0] + tj[0], origin[1] - tj[1], trackcolor, 1.5)
particlecolor = imgui.get_color_u32_rgba(1, 1, 1, 0.3)
dl.add_rect(pos[0], pos[1], pos[0] + siz[0], pos[1] + siz[1],
particlecolor, 1)
if 'particles' in self.frame:
ps = self.frame['particles']
for p in ps:
dl.add_rect_filled(origin[0] + p[0] * scale,
origin[1] - p[1] * scale,
origin[0] + p[0] * scale + 1,
origin[1] - p[1] * scale + 1, particlecolor)
# also draw a mean velocity vector
mxy = scale * np.mean(ps[:, :2], axis=0)
vxy = scale * v * .1 * np.array(
[np.mean(np.cos(ps[:, 3])),
np.mean(np.sin(ps[:, 3]))])
dl.add_line(origin[0] + mxy[0], origin[1] - mxy[1],
origin[0] + mxy[0] + vxy[0],
origin[1] - mxy[1] - vxy[1],
imgui.get_color_u32_rgba(0, 1, 0, 1), 1)
else:
x = self.controlstate[self.i, 0]
y = self.controlstate[self.i, 1]
theta = self.controlstate[self.i, 2]
imgui.slider_float("x", x, 0, 20)
imgui.slider_float("y", y, -10, 0)
imgui.slider_float("theta", theta % (2 * np.pi), -np.pi, np.pi)
v = 0.3 * self.controlstate[self.i, 3]
S, C = np.sin(theta), np.cos(theta)
dl.add_rect_filled(origin[0] + x * scale - 3,
origin[1] - y * scale - 3,
origin[0] + scale * x + 3,
origin[1] - scale * y + 3,
imgui.get_color_u32_rgba(0, 1, 0, 1), 1)
dl.add_line(origin[0] + x * scale, origin[1] - y * scale,
origin[0] + scale * (x + v * C),
origin[1] - scale * (y + v * S),
imgui.get_color_u32_rgba(0, 1, 0, 1), 1)
targetaccel = self.controlstate[self.i][12:14] / 9.8
accel = self.carstate[self.i][2]
ox, oy = origin[0] + scale * 3, origin[1] + scale * 9
for i in range(100):
t0 = i * 2 * np.pi / 100
t1 = (i + 1) * 2 * np.pi / 100
dl.add_line(ox + 2 * scale * np.cos(t0),
oy - 2 * scale * np.sin(t0),
ox + 2 * scale * np.cos(t1),
oy - 2 * scale * np.sin(t1),
imgui.get_color_u32_rgba(0.7, 0.7, 0.7, 1), 1)
dl.add_line(ox + scale * np.cos(t0), oy - scale * np.sin(t0),
ox + scale * np.cos(t1), oy - scale * np.sin(t1),
imgui.get_color_u32_rgba(0.7, 0.7, 0.7, 1), 1)
dl.add_line(ox, oy, ox + scale * accel[1], oy - scale * accel[0],
imgui.get_color_u32_rgba(0.3, 1, 0.5, 1), 3)
dl.add_rect(ox - scale * targetaccel[1] - 2,
oy + scale * targetaccel[0] - 2,
ox - scale * targetaccel[1] + 2,
oy + scale * targetaccel[0] + 2,
imgui.get_color_u32_rgba(0.0, 1, 1, 1), 1)
imgui.end()
def main():
global width_library
global width_shematic
global width_context
global height_window
global previous_key_callback
global selected_link
global selected_node
global iggraph
global node_library
global debug_is_mouse_dragging
global show_debug_window
# states -------------------------
scrolling = imgui.Vec2(0, 0)
iggraph = IGGraph(node_library)
# iggraph.reset()
node_hovered_in_scene = -1
parameter_link_start = None
selected_parameter = None
io_hovered = None
io_anchors_width_not_hovered = 10
io_anchors_width_hovered = 15
right_splitter_is_active = False
left_splitter_is_active = False
image_width = 0
image_height = 0
image_texture = None
# states -------------------------
imgui.create_context()
window = impl_glfw_init()
impl = GlfwRenderer(window)
io = imgui.get_io()
previous_key_callback = glfw.set_key_callback(window,key_event)
init_textures()
assign_parameter_colors(node_library)
while not glfw.window_should_close(window):
glfw.poll_events()
impl.process_inputs()
imgui.new_frame()
if imgui.begin_main_menu_bar():
if imgui.begin_menu("File", True):
clicked_new, selected_new = imgui.menu_item(
"New", 'Cmd+N', False, True
)
if clicked_new:
iggraph = IGGraph(node_library)
clicked_load, selected_load = imgui.menu_item(
"Load", 'Cmd+L', False, True
)
if clicked_load:
root = tk.Tk()
root.withdraw()
filename = filedialog.askopenfilename()
if filename :
f=open(filename)
iggraph.from_json(json.load(f))
f.close()
clicked_save, selected_save = imgui.menu_item(
"Save", 'Cmd+S', False, True
)
if clicked_save:
iggraph.reset()
graph_json = iggraph.to_json()
text2save = json.dumps(graph_json, indent=4, sort_keys=True)
root = tk.Tk()
root.withdraw()
f = filedialog.asksaveasfile(mode='w', defaultextension=".json")
if f is None: # asksaveasfile return `None` if dialog closed with "cancel".
return
f.write(text2save)
f.close()
clicked_quit, selected_quit = imgui.menu_item(
"Quit", 'Cmd+Q', False, True
)
if clicked_quit:
exit(0)
imgui.end_menu()
if imgui.begin_menu("Debug", True):
show_debug_window_clicked, show_debug_window_selected = imgui.menu_item(
"Show Debug window", 'Cmd+D', show_debug_window, True
)
if show_debug_window_clicked:
show_debug_window = not show_debug_window
catch_exceptions_clicked, catch_exceptions_selected = imgui.menu_item(
"Catch Exceptions", '', iggraph.catch_exceptions, True
)
if catch_exceptions_clicked:
iggraph.catch_exceptions = not iggraph.catch_exceptions
imgui.separator()
imgui.menu_item(
"Examples", '', False, False
)
show_example_mosaic_clicked, show_example_mosaic_selected = imgui.menu_item(
"Mosaic", '', False, True
)
if show_example_mosaic_clicked:
example_mosaic(iggraph)
imgui.end_menu()
imgui.end_main_menu_bar()
height_window = io.display_size.y - 18 # imgui.get_cursor_pos_y()
imgui.push_style_var(imgui.STYLE_ITEM_SPACING, imgui.Vec2(0,0))
imgui.set_next_window_size(io.display_size.x, height_window)
imgui.set_next_window_position(0, 18)
imgui.push_style_var(imgui.STYLE_WINDOW_ROUNDING, 0)
imgui.begin("Splitter test", False, imgui.WINDOW_NO_TITLE_BAR | imgui.WINDOW_NO_RESIZE | imgui.WINDOW_NO_MOVE | imgui.WINDOW_NO_BRING_TO_FRONT_ON_FOCUS)
imgui.pop_style_var()
imgui.pop_style_var()
width_shematic = io.display_size.x - separator_width - width_context - separator_width - width_library
# ==============================================================================
# Library
# ==============================================================================
imgui.push_style_var(imgui.STYLE_CHILD_BORDERSIZE, 0)
imgui.begin_child("Library", width_library, 0, True)
for node_name in node_library.nodes:
if imgui.button(node_name, width_library):
iggraph.create_node(node_name)
imgui.end_child()
imgui.pop_style_var()
imgui.same_line()
imgui.button("left_splitter", separator_width, height_window - 20)
left_splitter_is_active = imgui.is_item_active()
if (left_splitter_is_active):
width_library += io.mouse_delta.x
scrolling = imgui.Vec2(scrolling.x - io.mouse_delta.x, scrolling.y)
if (imgui.is_item_hovered()):
imgui.set_mouse_cursor(imgui.MOUSE_CURSOR_RESIZE_EW)
imgui.same_line()
# ==============================================================================
# Shematic
# ==============================================================================
imgui.push_style_var(imgui.STYLE_CHILD_BORDERSIZE, 0)
imgui.begin_child("shematic", width_shematic, 0, True)
if show_inputs_popup(iggraph):
imgui.open_popup("Outputs")
show_outputs_popup(iggraph)
# create our child canvas
if iggraph.get_state() == iggraph.STATE_IDLE:
imgui.text("status: edit | ")
elif iggraph.get_state() == iggraph.STATE_RUNNING:
imgui.text("status: run | ")
imgui.same_line()
if iggraph.get_state() == iggraph.STATE_IDLE:
if imgui.button("run"):
#TODO if not running
iggraph.set_state(iggraph.STATE_RUNNING)
iggraph.prepare_to_run()
imgui.open_popup("User Input")
imgui.same_line()
if imgui.button("run one step"):
iggraph.set_state(iggraph.STATE_RUNNING)
iggraph.prepare_to_run()
iggraph.run_one_step()
elif iggraph.get_state() == iggraph.STATE_RUNNING:
if imgui.button("stop running"):
iggraph.reset()
iggraph.set_state(iggraph.STATE_IDLE)
imgui.same_line()
if imgui.button("run one step"):
iggraph.set_state(iggraph.STATE_RUNNING)
iggraph.run_one_step()
# imgui.same_line(imgui.get_window_width() - 100)
imgui.push_style_var(imgui.STYLE_FRAME_PADDING, imgui.Vec2(1, 1))
imgui.push_style_var(imgui.STYLE_WINDOW_PADDING, imgui.Vec2(0, 0))
imgui.begin_child("scrolling_region", 0, 0, True, imgui.WINDOW_NO_SCROLLBAR | imgui.WINDOW_NO_MOVE)
imgui.pop_style_var()
imgui.pop_style_var()
imgui.push_item_width(120.0)
offset = add(imgui.get_cursor_screen_pos(), scrolling)
draw_list = imgui.get_window_draw_list()
# Display links
draw_list.channels_split(2)
draw_list.channels_set_current(0)
for link in iggraph.links:
draw_link_param_to_param(draw_list, offset, link.output_parameter, link.input_parameter, selected_link == link)
# Display nodes
parameter_link_end = None
one_parameter_hovered = False
one_node_moving_active = False
for node in iggraph.nodes:
imgui.push_id(str(node.id))
node_rect_min = add(offset, node.pos)
draw_list.channels_set_current(1) # foreground
old_any_active = imgui.is_any_item_active()
#display node content first
# todo
test = add(node_rect_min, NODE_WINDOW_PADDING)
imgui.set_cursor_screen_position(add(node_rect_min, NODE_WINDOW_PADDING))
imgui.begin_group()
imgui.text("")
imgui.text(node.name)
imgui.text("")
imgui.end_group()
# save size
node_widgets_active = False # (not old_any_active and imgui.is_any_item_active())
node.size = add( add( imgui.get_item_rect_size(), NODE_WINDOW_PADDING) , NODE_WINDOW_PADDING)
node_rect_max = add(node.size, node_rect_min)
#display node box
draw_list.channels_set_current(0) # background
imgui.set_cursor_screen_position(node_rect_min)
imgui.invisible_button(str(node.id), node.size.x, node.size.y)
if imgui.is_item_hovered():
node_hovered_in_scene = node.id
else:
node_hovered_in_scene = None
node_moving_active = imgui.is_item_active()
use_hovered_color = node_hovered_in_scene or selected_node == node
draw_list.add_rect_filled(node_rect_min.x, node_rect_min.y, node_rect_max.x, node_rect_max.y, get_node_color(node, iggraph, use_hovered_color), 5)
if node_hovered_in_scene and iggraph.is_error(node):
imgui.begin_tooltip()
imgui.text(iggraph.error_nodes[node])
imgui.end_tooltip()
# input parameters
for parameter_name in node.inputs:
parameter = node.inputs[parameter_name]
center = node.get_intput_slot_pos(parameter)
center_with_offset = add(offset, center)
if io_hovered == parameter:
io_anchors_width = io_anchors_width_hovered
else:
io_anchors_width = io_anchors_width_not_hovered
imgui.set_cursor_pos(imgui.Vec2(center.x-io_anchors_width/2, center.y-io_anchors_width/2))
imgui.push_id(str(str(node.id) + "input" + parameter.id))
if (imgui.invisible_button("input", io_anchors_width, io_anchors_width)):
selected_parameter = parameter
# imgui.is_item_hovered() does not work when dragging
is_hovering = ((io.mouse_pos.x-offset.x>center.x-io_anchors_width/2) and
(io.mouse_pos.x-offset.x<center.x+io_anchors_width/2) and
(io.mouse_pos.y-offset.y>center.y-io_anchors_width/2) and
(io.mouse_pos.y-offset.y<center.y+io_anchors_width/2))
if is_hovering:
io_hovered = parameter
one_parameter_hovered = True
imgui.begin_tooltip()
imgui.text(parameter_name)
imgui.end_tooltip()
if is_hovering and imgui.is_mouse_released(0):
parameter_link_end = parameter
imgui.pop_id()
draw_list.add_circle_filled(center_with_offset.x, center_with_offset.y, io_anchors_width/2, get_parameter_color(parameter))
# output parameters
for parameter_name in node.outputs:
parameter = node.outputs[parameter_name]
center = node.get_output_slot_pos(parameter)
center_with_offset = add(offset, center)
if io_hovered == parameter:
io_anchors_width = io_anchors_width_hovered
else:
io_anchors_width = io_anchors_width_not_hovered
imgui.set_cursor_pos(imgui.Vec2(center.x-io_anchors_width/2, center.y-io_anchors_width/2))
imgui.push_id(str(str(node.id) + "output" + parameter.id))
if (imgui.invisible_button("output", io_anchors_width, io_anchors_width)):
selected_parameter = parameter
is_hovering = ((io.mouse_pos.x-offset.x>center.x-io_anchors_width/2) and
(io.mouse_pos.x-offset.x<center.x+io_anchors_width/2) and
(io.mouse_pos.y-offset.y>center.y-io_anchors_width/2) and
(io.mouse_pos.y-offset.y<center.y+io_anchors_width/2))
if is_hovering:
io_hovered = parameter
one_parameter_hovered = True
imgui.begin_tooltip()
imgui.text(parameter_name)
imgui.end_tooltip()
draw_list.add_circle_filled(center_with_offset.x, center_with_offset.y, io_anchors_width/2, get_parameter_color(parameter))
imgui.pop_id()
# cannot use imgui.is_item_active, seems buggy with the scroll
if is_hovering and imgui.is_mouse_down(0):
parameter_link_start = parameter
if node_widgets_active or node_moving_active:
selected_node = node
one_node_moving_active = True
if node_moving_active and imgui.is_mouse_dragging(0) and node.id==selected_node.id:
node.pos = add(node.pos, io.mouse_delta)
debug_is_mouse_dragging = imgui.is_mouse_dragging(0)
imgui.pop_id()
draw_list.channels_merge()
if not one_parameter_hovered:
io_hovered = None
# scrolling
mouse_is_in_schematic = (io.mouse_pos.x > width_library) and\
(io.mouse_pos.x < (width_library + width_shematic)) and\
(io.mouse_pos.y < height_window) and\
(io.mouse_pos.y > 18)
if not one_node_moving_active and\
not parameter_link_start and\
imgui.is_mouse_dragging(0) and\
mouse_is_in_schematic and\
not left_splitter_is_active and\
not right_splitter_is_active and\
imgui.is_window_focused():
scroll_offset = imgui.Vec2(io.mouse_delta.x, io.mouse_delta.y)
scrolling = add(scrolling, scroll_offset)
# link creation
if parameter_link_start and parameter_link_end:
iggraph.add_link(parameter_link_start, parameter_link_end)
# creating link
elif parameter_link_start and imgui.is_mouse_dragging(0):
draw_link_param_to_point(draw_list, offset, parameter_link_start, io.mouse_pos.x, io.mouse_pos.y, True)
# mouse release
if imgui.is_mouse_released(0):
parameter_link_start = None
imgui.pop_item_width()
imgui.end_child()
# mouse click on the scene
if imgui.is_mouse_clicked(0):
mouse_pos = imgui.get_mouse_pos()
local_mouse_pos = imgui.Vec2(mouse_pos.x - offset.x, mouse_pos.y - offset.y)
selected_link = None
for link in iggraph.links:
start_node = link.output_parameter.owner
start_pos = start_node.get_output_slot_pos(link.output_parameter)
end_node = link.input_parameter.owner
end_pos = end_node.get_intput_slot_pos(link.input_parameter)
distance_mouse_start = math.sqrt(((local_mouse_pos.x-start_pos.x)**2) + ((local_mouse_pos.y-start_pos.y)**2))
distance_mouse_end = math.sqrt(((local_mouse_pos.x-end_pos.x)**2) + ((local_mouse_pos.y-end_pos.y)**2))
distance_start_end = math.sqrt(((start_pos.x-end_pos.x)**2) + ((start_pos.y-end_pos.y)**2))
if ((distance_mouse_start + distance_mouse_end) - distance_start_end) < 0.1:
selected_link = link
imgui.end_child()
imgui.pop_style_var()
imgui.same_line()
imgui.button("right_splitter", separator_width, height_window - 20)
right_splitter_is_active = imgui.is_item_active()
if (right_splitter_is_active):
width_context -= io.mouse_delta.x
if (imgui.is_item_hovered()):
imgui.set_mouse_cursor(imgui.MOUSE_CURSOR_RESIZE_EW)
# ==============================================================================
# Context
# ==============================================================================
imgui.same_line()
imgui.push_style_var(imgui.STYLE_CHILD_BORDERSIZE, 0)
imgui.begin_child("child3", width_context, 0, True);
if selected_node:
if selected_node.handle_dynamic_parameters():
if imgui.button("add parameter"):
selected_node.add_dynamic_parameter()
if imgui.tree_node("Inputs"):
for parameter_name in selected_node.inputs:
parameter = selected_node.inputs[parameter_name]
if imgui.tree_node(parameter.id):
display_parameter(parameter, True)
imgui.tree_pop()
imgui.tree_pop()
if imgui.tree_node("Output"):
for parameter_name in selected_node.outputs:
parameter = selected_node.outputs[parameter_name]
if imgui.tree_node(parameter.id):
display_parameter(parameter, False)
imgui.tree_pop()
imgui.tree_pop()
imgui.end_child()
imgui.pop_style_var()
#
imgui.end()
# ==============================================================================
# Debug Window
# ==============================================================================
if show_debug_window:
debug_window_expanded, show_debug_window = imgui.begin("Debug", True)
if parameter_link_start:
imgui.text("parameter_link_start: " + parameter_link_start.id)
else:
imgui.text("parameter_link_start: " + "None")
if selected_parameter:
imgui.text("selected_parameter: " + selected_parameter.id)
else:
imgui.text("selected_parameter: " + "None")
imgui.text("is_mouse_dragging: " + str(debug_is_mouse_dragging))
imgui.text("mouse: (" + str(io.mouse_pos.x) + ", " + str(io.mouse_pos.y) + ")")
imgui.end()
gl.glClearColor(1., 1., 1., 1)
gl.glClear(gl.GL_COLOR_BUFFER_BIT)
imgui.render()
impl.render(imgui.get_draw_data())
glfw.swap_buffers(window)
impl.shutdown()
glfw.terminate()
def pan_zoom(name,
state,
width=None,
height=None,
content_gen=None,
drag_tag=None,
down_tag=None,
hover_tag=None):
""" Create the Pan & Zoom widget, serving as a container for a thing given by `content_gen`.
This widget is mostly not used directly. Instead, a special-purpose wrapper can be used,
such as `concur.extra_widgets.image.image`, or `concur.extra_widgets.frame.frame`.
Args:
name: widget name. Also serves as an event identifier
state: `PanZoom` object representing state.
width: widget width in pixels. If `None`, it fills the available space.
height: widget height in pixels. If `None`, it fills the available space.
content_gen: Widget generator to display inside the pan-zoom widget. All events are passed
through as a return tuple member. This is a function that returns a Concur widget, and
takes two keyword arguments:
* **tf**: `concur.extra_widgets.pan_zoom.TF` object with transformation information
* **event_gen**: This is an event generator which returns the drag, down, and hover events when they occur.
It is up to the widget to do the necessary transformations
using the `TF` object, and optionally react to the events created by `event_gen`.
drag_tag: The event tag for LMB drag events. If `None`, no drag events are fired.
down_tag: The event tag for LMB down events. If `None`, no down events are fired.
Useful in combination with `drag_tag` to indicate when the dragging started.
hover_tag: The event tag for mouse hover when no mouse buttons are down.
If `None`, no hover events are fired. Events are fired only when the mouse is moving.
Returns:
To ease integration with other widgets, this widget returns events in a special format, `(name, state, event)`.
`state` or `event` may be `None` in case `state` didn't change, or there is no `event`.
If `drag_tag`, or `down_tag` arguments are used, these are fired in the `event` member of the return tuple.
"""
assert content_gen is not None
tf, content = None, None
event_queue = queue.Queue()
while True:
assert not state.keep_aspect or not state.fix_axis, \
"Can't fix axis and keep_aspect at the same time."
# Dynamically rescale width and height if they weren't specified
if width is None:
w = imgui.get_content_region_available()[0]
else:
w = width
if height is None:
h = imgui.get_content_region_available()[1]
else:
h = height
frame_w = max(1, w - state.margins[0] + state.margins[2])
frame_h = max(1, h - state.margins[1] + state.margins[3])
w = max(1, w)
h = max(1, h)
zoom_x = frame_w / (state.right - state.left)
zoom_y = frame_h / (state.bottom - state.top)
left, right = state.left, state.right
top, bottom = state.top, state.bottom
if state.keep_aspect:
aspect = float(state.keep_aspect)
assert state.keep_aspect > 0, "Negative aspect ratio is not supported."
if abs(zoom_x) > abs(zoom_y) * aspect:
zoom_x = np.sign(zoom_x) * abs(zoom_y) * aspect
center_x = (left + right) / 2
left = center_x - frame_w / zoom_x / 2
right = center_x + frame_w / zoom_x / 2
if abs(zoom_y) > abs(zoom_x) / aspect:
zoom_y = np.sign(zoom_y) * abs(zoom_x) / aspect
center_y = (top + bottom) / 2
top = center_y - frame_h / zoom_y / 2
bottom = center_y + frame_h / zoom_y / 2
origin = imgui.get_cursor_screen_pos()
# needs to be before is_window_hovered
imgui.begin_child("Pan-zoom",
w,
h,
False,
flags=imgui.WINDOW_NO_SCROLLBAR
| imgui.WINDOW_NO_SCROLL_WITH_MOUSE)
# Interaction
st = copy.deepcopy(state)
io = imgui.get_io()
is_window_hovered = imgui.is_window_hovered()
if (imgui.is_mouse_clicked(1)
or imgui.is_mouse_clicked(2)) and is_window_hovered:
st.is_rmb_dragged = True
if not (io.mouse_down[1] or io.mouse_down[2]):
st.is_rmb_dragged = False
delta = io.mouse_delta
# Pan
if st.is_rmb_dragged and (delta[0] or delta[1]):
if st.fix_axis != 'x':
st.left -= delta[0] / zoom_x
st.right -= delta[0] / zoom_x
if st.fix_axis != 'y':
st.top -= delta[1] / zoom_y
st.bottom -= delta[1] / zoom_y
# Zoom
if (imgui.is_window_hovered() or st.is_rmb_dragged) and io.mouse_wheel:
factor = 1.3**io.mouse_wheel
if st.fix_axis != 'x':
mx_rel = (io.mouse_pos[0] - origin[0] -
state.margins[0]) / frame_w * 2 - 1
mx = mx_rel * (right - left) / (st.right - st.left) / 2 + 0.5
wi = st.right - st.left
st.left = st.left + wi * mx - wi / factor * mx
st.right = st.right - wi * (1 - mx) + wi / factor * (1 - mx)
if st.fix_axis != 'y':
my_rel = (io.mouse_pos[1] - origin[1] -
state.margins[1]) / frame_h * 2 - 1
my = my_rel * (bottom - top) / (st.bottom - st.top) / 2 + 0.5
hi = st.bottom - st.top
st.top = st.top + hi * my - hi / factor * my
st.bottom = st.bottom - hi * (1 - my) + hi / factor * (1 - my)
# Get screen-space bounds disregarding the margins
left_s = left - st.margins[0] / zoom_x
top_s = top - st.margins[1] / zoom_y
right_s = right - st.margins[2] / zoom_x
bottom_s = bottom - st.margins[3] / zoom_y
s2c = np.array([ # Screen to Content
[1 / zoom_x, 0, left_s - origin[0] / zoom_x],
[0, 1 / zoom_y, top_s - origin[1] / zoom_y]
])
c2s = np.array([ # Content to Screen
[zoom_x, 0, origin[0] - left_s * zoom_x],
[0, zoom_y, origin[1] - top_s * zoom_y]
])
view_s = [origin[0], origin[1], origin[0] + w, origin[1] + h]
view_c = [left_s, top_s, right_s, bottom_s]
content_value = None
content_returned = False
new_tf = TF(c2s, s2c, view_c, view_s)
if down_tag \
and not st.is_rmb_dragged \
and imgui.is_mouse_clicked() \
and is_window_hovered:
event_queue.put(
(down_tag, new_tf.inv_transform(np.array([[*io.mouse_pos]
]))[0]))
if hover_tag \
and not st.is_rmb_dragged \
and not any(imgui.is_mouse_down(i) for i in [0,1,2]) \
and is_window_hovered:
new_tf.hovered = True
if (delta[0] or delta[1]):
event_queue.put(
(hover_tag,
new_tf.inv_transform(np.array([[*io.mouse_pos]]))[0]))
if drag_tag \
and not st.is_rmb_dragged:
imgui.invisible_button("", w, h)
dx, dy = io.mouse_delta
if imgui.is_item_active() and (dx or dy):
event_queue.put(
(drag_tag,
np.array([new_tf.s2c[0, 0] * dx, new_tf.s2c[1, 1] * dy])))
imgui.set_item_allow_overlap()
try:
from concur.core import lift
if tf is None or tf != new_tf:
tf = new_tf
w, h = tf.view_s[2] - tf.view_s[0], tf.view_s[3] - tf.view_s[1]
content = content_gen(tf=tf,
event_gen=lambda: listen(event_queue))
next(content)
except StopIteration as e:
content_value = e.value
content_returned = True
finally:
imgui.end_child()
changed = st != state
if changed or content_returned:
return name, (st if changed else None, content_value)
yield
