import logging import gi gi.require_version("Gtk", "3.0") from gi.repository import Gtk, GLib from panels.menu import Panel as MenuPanel from ks_includes.widgets.heatergraph import HeaterGraph from ks_includes.widgets.keypad import Keypad from ks_includes.KlippyGtk import find_widget class Panel(MenuPanel): def __init__(self, screen, title, items=None): super().__init__(screen, title, items) self.left_panel = None self.devices = {} self.graph_update = None self.active_heater = None self.h = self.f = 0 self.main_menu = Gtk.Grid(row_homogeneous=True, column_homogeneous=True, hexpand=True, vexpand=True) if self._screen.vertical_mode: self.labels['menu'] = Gtk.Box(orientation=Gtk.Orientation.HORIZONTAL, spacing=0, homogeneous=False) else: self.labels['menu'] = Gtk.Box(orientation=Gtk.Orientation.VERTICAL, spacing=0, homogeneous=False) logo_image = self._gtk.Image("klipper", self._gtk.content_width * 0.2, self._gtk.content_height * 0.5) self.labels['menu'].pack_start(logo_image, False, False, 80) # temp_Button = self.arrangeMenuItems(items, 1, True) temp_Button = self._gtk.Button("heat-up", _("Temperature")) temp_Button.set_valign(Gtk.Align.CENTER) temp_Button.set_halign(Gtk.Align.CENTER) temp_Button.set_size_request(200, 100) temp_Button.get_style_context().add_class('custom-icon-button') temp_Button.connect('clicked', self.menu_item_clicked, {"panel": "temperature"}) self.labels['menu'].pack_start(temp_Button, True, True, 0) self.labels['da'] = HeaterGraph(self._screen, self._printer, self._gtk.font_size) self.labels['devices'] = self.create_top_panel(self.labels['da']) logging.info("### Making MainMenu") stats = self._printer.get_printer_status_data()["printer"] if stats["temperature_devices"]["count"] > 0 or stats["extruders"]["count"] > 0: self._gtk.reset_temp_color() if self._screen.vertical_mode: self.main_menu.attach(self.labels['devices'], 0, 0, 2, 1) self.main_menu.attach(self.labels['da'], 0, 1, 2, 1) self.main_menu.attach(self.labels['menu'], 0, 2, 2, 1) else: self.main_menu.attach(self.labels['devices'], 0, 0, 2, 1) self.main_menu.attach(self.labels['menu'], 0, 1, 1, 2) self.main_menu.attach(self.labels['da'], 1, 1, 1, 2) self.content.add(self.main_menu) def update_graph_visibility(self, force_hide=False): if self.left_panel is None: logging.info("No left panel") return count = 0 for device in self.devices: visible = self._config.get_config().getboolean(f"graph {self._screen.connected_printer}", device, fallback=True) self.devices[device]['visible'] = visible self.labels['da'].set_showing(device, visible) if visible: count += 1 self.devices[device]['name'].get_style_context().add_class("graph_label") else: self.devices[device]['name'].get_style_context().remove_class("graph_label") if count > 0 and not force_hide: self.labels['da'].queue_draw() self.labels['da'].show() if self.graph_update is None: # This has a high impact on load self.graph_update = GLib.timeout_add_seconds(5, self.update_graph) elif self.labels['da'] in self.left_panel: self.left_panel.remove(self.labels['da']) if self.graph_update is not None: GLib.source_remove(self.graph_update) self.graph_update = None return False def activate(self): if not self._printer.tempstore: self._screen.init_tempstore() self.update_graph_visibility() def deactivate(self): if self.graph_update is not None: GLib.source_remove(self.graph_update) self.graph_update = None def add_device(self, device, graph): logging.info(f"Adding device: {device}") temperature = self._printer.get_stat(device, "temperature") if temperature is None: return False devname = device.split()[1] if len(device.split()) > 1 else device # Support for hiding devices by name if devname.startswith("_"): return False if device.startswith("extruder"): if self._printer.extrudercount > 1: image = f"extruder-{device[8:]}" if device[8:] else "extruder-0" else: image = "extruder" class_name = f"graph_label_{device}" dev_type = "extruder" elif device == "heater_bed": image = "bed" devname = "Heater Bed" class_name = "graph_label_heater_bed" dev_type = "bed" elif device == "heater_generic chamber": self.h += 1 image = "chamber" devname = "Heater Chamber" class_name = f"graph_label_sensor_{self.h}" dev_type = "sensor" elif device.startswith("heater_generic"): self.h += 1 image = "heater" class_name = f"graph_label_sensor_{self.h}" dev_type = "sensor" elif device.startswith("temperature_fan"): self.f += 1 image = "fan" class_name = f"graph_label_fan_{self.f}" dev_type = "fan" elif self._config.get_main_config().getboolean("only_heaters", False): return False else: self.h += 1 image = "heat-up" class_name = f"graph_label_sensor_{self.h}" dev_type = "sensor" rgb = self._gtk.get_temp_color(dev_type) can_target = self._printer.device_has_target(device) graph.add_object(device, "temperatures", rgb, False, False) if can_target: graph.add_object(device, "targets", rgb, False, True) if self._show_heater_power and self._printer.device_has_power(device): graph.add_object(device, "powers", rgb, True, False) # name = self._gtk.Button(image, self.prettify(devname), None, self.bts, Gtk.PositionType.TOP, 1) name = self._gtk.Button() vbox = Gtk.Box(orientation=Gtk.Orientation.VERTICAL, spacing=0) name.add(vbox) target = Gtk.Label() icon = self._gtk.Image(image) temp = Gtk.Label() power = Gtk.Label() vbox.pack_start(target, True, False, 5) vbox.pack_start(icon, True, False, 0) vbox.pack_start(temp, True, False, 0) vbox.pack_start(power, True, False, 0) # name.connect("clicked", self.toggle_visibility, device) # name.set_alignment(0, .5) name.get_style_context().add_class(class_name) visible = self._config.get_config().getboolean(f"graph {self._screen.connected_printer}", device, fallback=True) if visible: name.get_style_context().add_class("graph_label") self.labels['da'].set_showing(device, visible) find_widget(temp, Gtk.Label).set_ellipsize(False) if can_target: name.connect("clicked", self.menu_item_clicked, {"panel": "numpad", 'extra': device}) self.devices[device] = { "class": class_name, "name": name, "temp": temp, "target": target, "power": power, "can_target": can_target, "visible": visible } devices = sorted(self.devices) pos = devices.index(device) + 1 self.labels['devices'].insert_row(pos) self.labels['devices'].attach(name, pos, 0, 1, 1) self.labels['devices'].show_all() return True def toggle_visibility(self, widget, device): self.devices[device]['visible'] ^= True logging.info(f"Graph show {self.devices[device]['visible']}: {device}") section = f"graph {self._screen.connected_printer}" if section not in self._config.get_config().sections(): self._config.get_config().add_section(section) self._config.set(section, f"{device}", f"{self.devices[device]['visible']}") self._config.save_user_config_options() self.update_graph_visibility() def create_top_panel(self, graph): self.labels['devices'] = Gtk.Grid(vexpand=False) self.labels['devices'].get_style_context().add_class('heater-grid') # name = Gtk.Label() # temp = Gtk.Label(label=_("Temp (°C)")) # if self._show_heater_power: # temp.get_style_context().add_class("heater-grid-temp-power") # else: # temp.get_style_context().add_class("heater-grid-temp") # self.labels['devices'].attach(name, 0, 0, 1, 1) # self.labels['devices'].attach(temp, 1, 0, 1, 1) # self.labels['da'] = HeaterGraph(self._screen, self._printer, self._gtk.font_size) scroll = self._gtk.ScrolledWindow(steppers=False) scroll.set_policy(Gtk.PolicyType.AUTOMATIC, Gtk.PolicyType.NEVER) scroll.get_style_context().add_class('heater-list') scroll.add(self.labels['devices']) self.left_panel = Gtk.Box(orientation=Gtk.Orientation.HORIZONTAL) self.left_panel.add(scroll) for d in self._printer.get_temp_devices(): self.add_device(d, graph) return self.left_panel def process_update(self, action, data): if action != "notify_status_update": return for x in self._printer.get_temp_devices(): if x in data: self.update_temp( x, self._printer.get_stat(x, "temperature"), self._printer.get_stat(x, "target"), self._printer.get_stat(x, "power"), ) def update_graph(self): self.labels['da'].queue_draw() return True