| |
| |
| |
| |
| |
| |
| |
|
|
| import contextlib |
| import imgui |
|
|
| |
|
|
| def set_default_style(color_scheme='dark', spacing=9, indent=23, scrollbar=27): |
| s = imgui.get_style() |
| s.window_padding = [spacing, spacing] |
| s.item_spacing = [spacing, spacing] |
| s.item_inner_spacing = [spacing, spacing] |
| s.columns_min_spacing = spacing |
| s.indent_spacing = indent |
| s.scrollbar_size = scrollbar |
| s.frame_padding = [4, 3] |
| s.window_border_size = 1 |
| s.child_border_size = 1 |
| s.popup_border_size = 1 |
| s.frame_border_size = 1 |
| s.window_rounding = 0 |
| s.child_rounding = 0 |
| s.popup_rounding = 3 |
| s.frame_rounding = 3 |
| s.scrollbar_rounding = 3 |
| s.grab_rounding = 3 |
|
|
| getattr(imgui, f'style_colors_{color_scheme}')(s) |
| c0 = s.colors[imgui.COLOR_MENUBAR_BACKGROUND] |
| c1 = s.colors[imgui.COLOR_FRAME_BACKGROUND] |
| s.colors[imgui.COLOR_POPUP_BACKGROUND] = [x * 0.7 + y * 0.3 for x, y in zip(c0, c1)][:3] + [1] |
|
|
| |
|
|
| @contextlib.contextmanager |
| def grayed_out(cond=True): |
| if cond: |
| s = imgui.get_style() |
| text = s.colors[imgui.COLOR_TEXT_DISABLED] |
| grab = s.colors[imgui.COLOR_SCROLLBAR_GRAB] |
| back = s.colors[imgui.COLOR_MENUBAR_BACKGROUND] |
| imgui.push_style_color(imgui.COLOR_TEXT, *text) |
| imgui.push_style_color(imgui.COLOR_CHECK_MARK, *grab) |
| imgui.push_style_color(imgui.COLOR_SLIDER_GRAB, *grab) |
| imgui.push_style_color(imgui.COLOR_SLIDER_GRAB_ACTIVE, *grab) |
| imgui.push_style_color(imgui.COLOR_FRAME_BACKGROUND, *back) |
| imgui.push_style_color(imgui.COLOR_FRAME_BACKGROUND_HOVERED, *back) |
| imgui.push_style_color(imgui.COLOR_FRAME_BACKGROUND_ACTIVE, *back) |
| imgui.push_style_color(imgui.COLOR_BUTTON, *back) |
| imgui.push_style_color(imgui.COLOR_BUTTON_HOVERED, *back) |
| imgui.push_style_color(imgui.COLOR_BUTTON_ACTIVE, *back) |
| imgui.push_style_color(imgui.COLOR_HEADER, *back) |
| imgui.push_style_color(imgui.COLOR_HEADER_HOVERED, *back) |
| imgui.push_style_color(imgui.COLOR_HEADER_ACTIVE, *back) |
| imgui.push_style_color(imgui.COLOR_POPUP_BACKGROUND, *back) |
| yield |
| imgui.pop_style_color(14) |
| else: |
| yield |
|
|
| |
|
|
| @contextlib.contextmanager |
| def item_width(width=None): |
| if width is not None: |
| imgui.push_item_width(width) |
| yield |
| imgui.pop_item_width() |
| else: |
| yield |
|
|
| |
|
|
| def scoped_by_object_id(method): |
| def decorator(self, *args, **kwargs): |
| imgui.push_id(str(id(self))) |
| res = method(self, *args, **kwargs) |
| imgui.pop_id() |
| return res |
| return decorator |
|
|
| |
|
|
| def button(label, width=0, enabled=True): |
| with grayed_out(not enabled): |
| clicked = imgui.button(label, width=width) |
| clicked = clicked and enabled |
| return clicked |
|
|
| |
|
|
| def collapsing_header(text, visible=None, flags=0, default=False, enabled=True, show=True): |
| expanded = False |
| if show: |
| if default: |
| flags |= imgui.TREE_NODE_DEFAULT_OPEN |
| if not enabled: |
| flags |= imgui.TREE_NODE_LEAF |
| with grayed_out(not enabled): |
| expanded, visible = imgui.collapsing_header(text, visible=visible, flags=flags) |
| expanded = expanded and enabled |
| return expanded, visible |
|
|
| |
|
|
| def popup_button(label, width=0, enabled=True): |
| if button(label, width, enabled): |
| imgui.open_popup(label) |
| opened = imgui.begin_popup(label) |
| return opened |
|
|
| |
|
|
| def input_text(label, value, buffer_length, flags, width=None, help_text=''): |
| old_value = value |
| color = list(imgui.get_style().colors[imgui.COLOR_TEXT]) |
| if value == '': |
| color[-1] *= 0.5 |
| with item_width(width): |
| imgui.push_style_color(imgui.COLOR_TEXT, *color) |
| value = value if value != '' else help_text |
| changed, value = imgui.input_text(label, value, buffer_length, flags) |
| value = value if value != help_text else '' |
| imgui.pop_style_color(1) |
| if not flags & imgui.INPUT_TEXT_ENTER_RETURNS_TRUE: |
| changed = (value != old_value) |
| return changed, value |
|
|
| |
|
|
| def drag_previous_control(enabled=True): |
| dragging = False |
| dx = 0 |
| dy = 0 |
| if imgui.begin_drag_drop_source(imgui.DRAG_DROP_SOURCE_NO_PREVIEW_TOOLTIP): |
| if enabled: |
| dragging = True |
| dx, dy = imgui.get_mouse_drag_delta() |
| imgui.reset_mouse_drag_delta() |
| imgui.end_drag_drop_source() |
| return dragging, dx, dy |
|
|
| |
|
|
| def drag_button(label, width=0, enabled=True): |
| clicked = button(label, width=width, enabled=enabled) |
| dragging, dx, dy = drag_previous_control(enabled=enabled) |
| return clicked, dragging, dx, dy |
|
|
| |
|
|
| def drag_hidden_window(label, x, y, width, height, enabled=True): |
| imgui.push_style_color(imgui.COLOR_WINDOW_BACKGROUND, 0, 0, 0, 0) |
| imgui.push_style_color(imgui.COLOR_BORDER, 0, 0, 0, 0) |
| imgui.set_next_window_position(x, y) |
| imgui.set_next_window_size(width, height) |
| imgui.begin(label, closable=False, flags=(imgui.WINDOW_NO_TITLE_BAR | imgui.WINDOW_NO_RESIZE | imgui.WINDOW_NO_MOVE)) |
| dragging, dx, dy = drag_previous_control(enabled=enabled) |
| imgui.end() |
| imgui.pop_style_color(2) |
| return dragging, dx, dy |
|
|
| |
|
|
| def click_hidden_window(label, x, y, width, height, img_w, img_h, enabled=True): |
| imgui.push_style_color(imgui.COLOR_WINDOW_BACKGROUND, 0, 0, 0, 0) |
| imgui.push_style_color(imgui.COLOR_BORDER, 0, 0, 0, 0) |
| imgui.set_next_window_position(x, y) |
| imgui.set_next_window_size(width, height) |
| imgui.begin(label, closable=False, flags=(imgui.WINDOW_NO_TITLE_BAR | imgui.WINDOW_NO_RESIZE | imgui.WINDOW_NO_MOVE)) |
| clicked, down = False, False |
| img_x, img_y = 0, 0 |
| if imgui.is_mouse_down(): |
| posx, posy = imgui.get_mouse_pos() |
| if posx >= x and posx < x + width and posy >= y and posy < y + height: |
| if imgui.is_mouse_clicked(): |
| clicked = True |
| down = True |
| img_x = round((posx - x) / (width - 1) * (img_w - 1)) |
| img_y = round((posy - y) / (height - 1) * (img_h - 1)) |
| imgui.end() |
| imgui.pop_style_color(2) |
| return clicked, down, img_x, img_y |
|
|
| |
|
|
