# # BitBake Graphical GTK based Dependency Explorer # # Copyright (C) 2007 Ross Burton # Copyright (C) 2007 - 2008 Richard Purdie # # SPDX-License-Identifier: GPL-2.0-only # import sys import traceback try: import gi gi.require_version('Gtk', '3.0') from gi.repository import Gtk, Gdk, GObject except ValueError: sys.exit("FATAL: Gtk version needs to be 3.0") except ImportError: sys.exit("FATAL: Gtk ui could not load the required gi python module") import threading from xmlrpc import client import bb import bb.event # Package Model (COL_PKG_NAME) = (0) # Dependency Model (TYPE_DEP, TYPE_RDEP) = (0, 1) (COL_DEP_TYPE, COL_DEP_PARENT, COL_DEP_PACKAGE) = (0, 1, 2) class PackageDepView(Gtk.TreeView): def __init__(self, model, dep_type, label): Gtk.TreeView.__init__(self) self.current = None self.dep_type = dep_type self.filter_model = model.filter_new() self.filter_model.set_visible_func(self._filter, data=None) self.set_model(self.filter_model) self.append_column(Gtk.TreeViewColumn(label, Gtk.CellRendererText(), text=COL_DEP_PACKAGE)) def _filter(self, model, iter, data): this_type = model[iter][COL_DEP_TYPE] package = model[iter][COL_DEP_PARENT] if this_type != self.dep_type: return False return package == self.current def set_current_package(self, package): self.current = package self.filter_model.refilter() class PackageReverseDepView(Gtk.TreeView): def __init__(self, model, label): Gtk.TreeView.__init__(self) self.current = None self.filter_model = model.filter_new() self.filter_model.set_visible_func(self._filter) # The introspected API was fixed but we can't rely on a pygobject that hides this. # https://gitlab.gnome.org/GNOME/pygobject/-/commit/9cdbc56fbac4db2de78dc080934b8f0a7efc892a if hasattr(Gtk.TreeModelSort, "new_with_model"): self.sort_model = Gtk.TreeModelSort.new_with_model(self.filter_model) else: self.sort_model = self.filter_model.sort_new_with_model() self.sort_model.set_sort_column_id(COL_DEP_PARENT, Gtk.SortType.ASCENDING) self.set_model(self.sort_model) self.append_column(Gtk.TreeViewColumn(label, Gtk.CellRendererText(), text=COL_DEP_PARENT)) def _filter(self, model, iter, data): package = model[iter][COL_DEP_PACKAGE] return package == self.current def set_current_package(self, package): self.current = package self.filter_model.refilter() class DepExplorer(Gtk.Window): def __init__(self): Gtk.Window.__init__(self) self.set_title("Task Dependency Explorer") self.set_default_size(500, 500) self.connect("delete-event", Gtk.main_quit) # Create the data models self.pkg_model = Gtk.ListStore(GObject.TYPE_STRING) self.pkg_model.set_sort_column_id(COL_PKG_NAME, Gtk.SortType.ASCENDING) self.depends_model = Gtk.ListStore(GObject.TYPE_INT, GObject.TYPE_STRING, GObject.TYPE_STRING) self.depends_model.set_sort_column_id(COL_DEP_PACKAGE, Gtk.SortType.ASCENDING) pane = Gtk.HPaned() pane.set_position(250) self.add(pane) # The master list of packages scrolled = Gtk.ScrolledWindow() scrolled.set_policy(Gtk.PolicyType.AUTOMATIC, Gtk.PolicyType.AUTOMATIC) scrolled.set_shadow_type(Gtk.ShadowType.IN) self.pkg_treeview = Gtk.TreeView(self.pkg_model) self.pkg_treeview.get_selection().connect("changed", self.on_cursor_changed) column = Gtk.TreeViewColumn("Package", Gtk.CellRendererText(), text=COL_PKG_NAME) self.pkg_treeview.append_column(column) scrolled.add(self.pkg_treeview) self.search_entry = Gtk.SearchEntry.new() self.pkg_treeview.set_search_entry(self.search_entry) left_panel = Gtk.VPaned() left_panel.add(self.search_entry) left_panel.add(scrolled) pane.add1(left_panel) box = Gtk.VBox(homogeneous=True, spacing=4) # Task Depends scrolled = Gtk.ScrolledWindow() scrolled.set_policy(Gtk.PolicyType.AUTOMATIC, Gtk.PolicyType.AUTOMATIC) scrolled.set_shadow_type(Gtk.ShadowType.IN) self.dep_treeview = PackageDepView(self.depends_model, TYPE_DEP, "Dependencies") self.dep_treeview.connect("row-activated", self.on_package_activated, COL_DEP_PACKAGE) scrolled.add(self.dep_treeview) box.add(scrolled) pane.add2(box) # Reverse Task Depends scrolled = Gtk.ScrolledWindow() scrolled.set_policy(Gtk.PolicyType.AUTOMATIC, Gtk.PolicyType.AUTOMATIC) scrolled.set_shadow_type(Gtk.ShadowType.IN) self.revdep_treeview = PackageReverseDepView(self.depends_model, "Dependent Tasks") self.revdep_treeview.connect("row-activated", self.on_package_activated, COL_DEP_PARENT) scrolled.add(self.revdep_treeview) box.add(scrolled) pane.add2(box) self.show_all() self.search_entry.grab_focus() def on_package_activated(self, treeview, path, column, data_col): model = treeview.get_model() package = model.get_value(model.get_iter(path), data_col) pkg_path = [] def finder(model, path, iter, needle): package = model.get_value(iter, COL_PKG_NAME) if package == needle: pkg_path.append(path) return True else: return False self.pkg_model.foreach(finder, package) if pkg_path: self.pkg_treeview.get_selection().select_path(pkg_path[0]) self.pkg_treeview.scroll_to_cell(pkg_path[0]) def on_cursor_changed(self, selection): (model, it) = selection.get_selected() if it is None: current_package = None else: current_package = model.get_value(it, COL_PKG_NAME) self.dep_treeview.set_current_package(current_package) self.revdep_treeview.set_current_package(current_package) def parse(self, depgraph): for task in depgraph["tdepends"]: self.pkg_model.insert(0, (task,)) for depend in depgraph["tdepends"][task]: self.depends_model.insert (0, (TYPE_DEP, task, depend)) class gtkthread(threading.Thread): quit = threading.Event() def __init__(self, shutdown): threading.Thread.__init__(self) self.setDaemon(True) self.shutdown = shutdown if not Gtk.init_check()[0]: sys.stderr.write("Gtk+ init failed. Make sure DISPLAY variable is set.\n") gtkthread.quit.set() def run(self): GObject.threads_init() Gdk.threads_init() Gtk.main() gtkthread.quit.set() def main(server, eventHandler, params): shutdown = 0 gtkgui = gtkthread(shutdown) gtkgui.start() try: params.updateToServer(server, os.environ.copy()) params.updateFromServer(server) cmdline = params.parseActions() if not cmdline: print("Nothing to do. Use 'bitbake world' to build everything, or run 'bitbake --help' for usage information.") return 1 if 'msg' in cmdline and cmdline['msg']: print(cmdline['msg']) return 1 cmdline = cmdline['action'] if not cmdline or cmdline[0] != "generateDotGraph": print("This UI requires the -g option") return 1 ret, error = server.runCommand(["generateDepTreeEvent", cmdline[1], cmdline[2]]) if error: print("Error running command '%s': %s" % (cmdline, error)) return 1 elif not ret: print("Error running command '%s': returned %s" % (cmdline, ret)) return 1 except client.Fault as x: print("XMLRPC Fault getting commandline:\n %s" % x) return except Exception as e: print("Exception in startup:\n %s" % traceback.format_exc()) return if gtkthread.quit.isSet(): return Gdk.threads_enter() dep = DepExplorer() bardialog = Gtk.Dialog(parent=dep, flags=Gtk.DialogFlags.MODAL|Gtk.DialogFlags.DESTROY_WITH_PARENT) bardialog.set_default_size(400, 50) box = bardialog.get_content_area() pbar = Gtk.ProgressBar() box.pack_start(pbar, True, True, 0) bardialog.show_all() bardialog.connect("delete-event", Gtk.main_quit) Gdk.threads_leave() progress_total = 0 while True: try: event = eventHandler.waitEvent(0.25) if gtkthread.quit.isSet(): _, error = server.runCommand(["stateForceShutdown"]) if error: print('Unable to cleanly stop: %s' % error) break if event is None: continue if isinstance(event, bb.event.CacheLoadStarted): progress_total = event.total Gdk.threads_enter() bardialog.set_title("Loading Cache") pbar.set_fraction(0.0) Gdk.threads_leave() if isinstance(event, bb.event.CacheLoadProgress): x = event.current Gdk.threads_enter() pbar.set_fraction(x * 1.0 / progress_total) Gdk.threads_leave() continue if isinstance(event, bb.event.CacheLoadCompleted): continue if isinstance(event, bb.event.ParseStarted): progress_total = event.total if progress_total == 0: continue Gdk.threads_enter() pbar.set_fraction(0.0) bardialog.set_title("Processing recipes") Gdk.threads_leave() if isinstance(event, bb.event.ParseProgress): x = event.current Gdk.threads_enter() pbar.set_fraction(x * 1.0 / progress_total) Gdk.threads_leave() continue if isinstance(event, bb.event.ParseCompleted): Gdk.threads_enter() bardialog.set_title("Generating dependency tree") Gdk.threads_leave() continue if isinstance(event, bb.event.DepTreeGenerated): Gdk.threads_enter() bardialog.hide() dep.parse(event._depgraph) Gdk.threads_leave() if isinstance(event, bb.command.CommandCompleted): continue if isinstance(event, bb.event.NoProvider): print(str(event)) _, error = server.runCommand(["stateShutdown"]) if error: print('Unable to cleanly shutdown: %s' % error) break if isinstance(event, bb.command.CommandFailed): print(str(event)) return event.exitcode if isinstance(event, bb.command.CommandExit): return event.exitcode if isinstance(event, bb.cooker.CookerExit): break continue except EnvironmentError as ioerror: # ignore interrupted io if ioerror.args[0] == 4: pass except KeyboardInterrupt: if shutdown == 2: print("\nThird Keyboard Interrupt, exit.\n") break if shutdown == 1: print("\nSecond Keyboard Interrupt, stopping...\n") _, error = server.runCommand(["stateForceShutdown"]) if error: print('Unable to cleanly stop: %s' % error) if shutdown == 0: print("\nKeyboard Interrupt, closing down...\n") _, error = server.runCommand(["stateShutdown"]) if error: print('Unable to cleanly shutdown: %s' % error) shutdown = shutdown + 1 pass