taskexp.py 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324
  1. #
  2. # BitBake Graphical GTK based Dependency Explorer
  3. #
  4. # Copyright (C) 2007 Ross Burton
  5. # Copyright (C) 2007 - 2008 Richard Purdie
  6. #
  7. # SPDX-License-Identifier: GPL-2.0-only
  8. #
  9. import sys
  10. import gi
  11. gi.require_version('Gtk', '3.0')
  12. from gi.repository import Gtk, Gdk, GObject
  13. import threading
  14. from xmlrpc import client
  15. import bb
  16. import bb.event
  17. # Package Model
  18. (COL_PKG_NAME) = (0)
  19. # Dependency Model
  20. (TYPE_DEP, TYPE_RDEP) = (0, 1)
  21. (COL_DEP_TYPE, COL_DEP_PARENT, COL_DEP_PACKAGE) = (0, 1, 2)
  22. class PackageDepView(Gtk.TreeView):
  23. def __init__(self, model, dep_type, label):
  24. Gtk.TreeView.__init__(self)
  25. self.current = None
  26. self.dep_type = dep_type
  27. self.filter_model = model.filter_new()
  28. self.filter_model.set_visible_func(self._filter, data=None)
  29. self.set_model(self.filter_model)
  30. self.append_column(Gtk.TreeViewColumn(label, Gtk.CellRendererText(), text=COL_DEP_PACKAGE))
  31. def _filter(self, model, iter, data):
  32. this_type = model[iter][COL_DEP_TYPE]
  33. package = model[iter][COL_DEP_PARENT]
  34. if this_type != self.dep_type: return False
  35. return package == self.current
  36. def set_current_package(self, package):
  37. self.current = package
  38. self.filter_model.refilter()
  39. class PackageReverseDepView(Gtk.TreeView):
  40. def __init__(self, model, label):
  41. Gtk.TreeView.__init__(self)
  42. self.current = None
  43. self.filter_model = model.filter_new()
  44. self.filter_model.set_visible_func(self._filter)
  45. self.sort_model = self.filter_model.sort_new_with_model()
  46. self.sort_model.set_sort_column_id(COL_DEP_PARENT, Gtk.SortType.ASCENDING)
  47. self.set_model(self.sort_model)
  48. self.append_column(Gtk.TreeViewColumn(label, Gtk.CellRendererText(), text=COL_DEP_PARENT))
  49. def _filter(self, model, iter, data):
  50. package = model[iter][COL_DEP_PACKAGE]
  51. return package == self.current
  52. def set_current_package(self, package):
  53. self.current = package
  54. self.filter_model.refilter()
  55. class DepExplorer(Gtk.Window):
  56. def __init__(self):
  57. Gtk.Window.__init__(self)
  58. self.set_title("Task Dependency Explorer")
  59. self.set_default_size(500, 500)
  60. self.connect("delete-event", Gtk.main_quit)
  61. # Create the data models
  62. self.pkg_model = Gtk.ListStore(GObject.TYPE_STRING)
  63. self.pkg_model.set_sort_column_id(COL_PKG_NAME, Gtk.SortType.ASCENDING)
  64. self.depends_model = Gtk.ListStore(GObject.TYPE_INT, GObject.TYPE_STRING, GObject.TYPE_STRING)
  65. self.depends_model.set_sort_column_id(COL_DEP_PACKAGE, Gtk.SortType.ASCENDING)
  66. pane = Gtk.HPaned()
  67. pane.set_position(250)
  68. self.add(pane)
  69. # The master list of packages
  70. scrolled = Gtk.ScrolledWindow()
  71. scrolled.set_policy(Gtk.PolicyType.AUTOMATIC, Gtk.PolicyType.AUTOMATIC)
  72. scrolled.set_shadow_type(Gtk.ShadowType.IN)
  73. self.pkg_treeview = Gtk.TreeView(self.pkg_model)
  74. self.pkg_treeview.get_selection().connect("changed", self.on_cursor_changed)
  75. column = Gtk.TreeViewColumn("Package", Gtk.CellRendererText(), text=COL_PKG_NAME)
  76. self.pkg_treeview.append_column(column)
  77. scrolled.add(self.pkg_treeview)
  78. self.search_entry = Gtk.SearchEntry.new()
  79. self.pkg_treeview.set_search_entry(self.search_entry)
  80. left_panel = Gtk.VPaned()
  81. left_panel.add(self.search_entry)
  82. left_panel.add(scrolled)
  83. pane.add1(left_panel)
  84. box = Gtk.VBox(homogeneous=True, spacing=4)
  85. # Task Depends
  86. scrolled = Gtk.ScrolledWindow()
  87. scrolled.set_policy(Gtk.PolicyType.AUTOMATIC, Gtk.PolicyType.AUTOMATIC)
  88. scrolled.set_shadow_type(Gtk.ShadowType.IN)
  89. self.dep_treeview = PackageDepView(self.depends_model, TYPE_DEP, "Dependencies")
  90. self.dep_treeview.connect("row-activated", self.on_package_activated, COL_DEP_PACKAGE)
  91. scrolled.add(self.dep_treeview)
  92. box.add(scrolled)
  93. pane.add2(box)
  94. # Reverse Task Depends
  95. scrolled = Gtk.ScrolledWindow()
  96. scrolled.set_policy(Gtk.PolicyType.AUTOMATIC, Gtk.PolicyType.AUTOMATIC)
  97. scrolled.set_shadow_type(Gtk.ShadowType.IN)
  98. self.revdep_treeview = PackageReverseDepView(self.depends_model, "Dependent Tasks")
  99. self.revdep_treeview.connect("row-activated", self.on_package_activated, COL_DEP_PARENT)
  100. scrolled.add(self.revdep_treeview)
  101. box.add(scrolled)
  102. pane.add2(box)
  103. self.show_all()
  104. self.search_entry.grab_focus()
  105. def on_package_activated(self, treeview, path, column, data_col):
  106. model = treeview.get_model()
  107. package = model.get_value(model.get_iter(path), data_col)
  108. pkg_path = []
  109. def finder(model, path, iter, needle):
  110. package = model.get_value(iter, COL_PKG_NAME)
  111. if package == needle:
  112. pkg_path.append(path)
  113. return True
  114. else:
  115. return False
  116. self.pkg_model.foreach(finder, package)
  117. if pkg_path:
  118. self.pkg_treeview.get_selection().select_path(pkg_path[0])
  119. self.pkg_treeview.scroll_to_cell(pkg_path[0])
  120. def on_cursor_changed(self, selection):
  121. (model, it) = selection.get_selected()
  122. if it is None:
  123. current_package = None
  124. else:
  125. current_package = model.get_value(it, COL_PKG_NAME)
  126. self.dep_treeview.set_current_package(current_package)
  127. self.revdep_treeview.set_current_package(current_package)
  128. def parse(self, depgraph):
  129. for task in depgraph["tdepends"]:
  130. self.pkg_model.insert(0, (task,))
  131. for depend in depgraph["tdepends"][task]:
  132. self.depends_model.insert (0, (TYPE_DEP, task, depend))
  133. class gtkthread(threading.Thread):
  134. quit = threading.Event()
  135. def __init__(self, shutdown):
  136. threading.Thread.__init__(self)
  137. self.setDaemon(True)
  138. self.shutdown = shutdown
  139. if not Gtk.init_check()[0]:
  140. sys.stderr.write("Gtk+ init failed. Make sure DISPLAY variable is set.\n")
  141. gtkthread.quit.set()
  142. def run(self):
  143. GObject.threads_init()
  144. Gdk.threads_init()
  145. Gtk.main()
  146. gtkthread.quit.set()
  147. def main(server, eventHandler, params):
  148. shutdown = 0
  149. gtkgui = gtkthread(shutdown)
  150. gtkgui.start()
  151. try:
  152. params.updateFromServer(server)
  153. cmdline = params.parseActions()
  154. if not cmdline:
  155. print("Nothing to do. Use 'bitbake world' to build everything, or run 'bitbake --help' for usage information.")
  156. return 1
  157. if 'msg' in cmdline and cmdline['msg']:
  158. print(cmdline['msg'])
  159. return 1
  160. cmdline = cmdline['action']
  161. if not cmdline or cmdline[0] != "generateDotGraph":
  162. print("This UI requires the -g option")
  163. return 1
  164. ret, error = server.runCommand(["generateDepTreeEvent", cmdline[1], cmdline[2]])
  165. if error:
  166. print("Error running command '%s': %s" % (cmdline, error))
  167. return 1
  168. elif not ret:
  169. print("Error running command '%s': returned %s" % (cmdline, ret))
  170. return 1
  171. except client.Fault as x:
  172. print("XMLRPC Fault getting commandline:\n %s" % x)
  173. return
  174. if gtkthread.quit.isSet():
  175. return
  176. Gdk.threads_enter()
  177. dep = DepExplorer()
  178. bardialog = Gtk.Dialog(parent=dep,
  179. flags=Gtk.DialogFlags.MODAL|Gtk.DialogFlags.DESTROY_WITH_PARENT)
  180. bardialog.set_default_size(400, 50)
  181. box = bardialog.get_content_area()
  182. pbar = Gtk.ProgressBar()
  183. box.pack_start(pbar, True, True, 0)
  184. bardialog.show_all()
  185. bardialog.connect("delete-event", Gtk.main_quit)
  186. Gdk.threads_leave()
  187. progress_total = 0
  188. while True:
  189. try:
  190. event = eventHandler.waitEvent(0.25)
  191. if gtkthread.quit.isSet():
  192. _, error = server.runCommand(["stateForceShutdown"])
  193. if error:
  194. print('Unable to cleanly stop: %s' % error)
  195. break
  196. if event is None:
  197. continue
  198. if isinstance(event, bb.event.CacheLoadStarted):
  199. progress_total = event.total
  200. Gdk.threads_enter()
  201. bardialog.set_title("Loading Cache")
  202. pbar.set_fraction(0.0)
  203. Gdk.threads_leave()
  204. if isinstance(event, bb.event.CacheLoadProgress):
  205. x = event.current
  206. Gdk.threads_enter()
  207. pbar.set_fraction(x * 1.0 / progress_total)
  208. Gdk.threads_leave()
  209. continue
  210. if isinstance(event, bb.event.CacheLoadCompleted):
  211. continue
  212. if isinstance(event, bb.event.ParseStarted):
  213. progress_total = event.total
  214. if progress_total == 0:
  215. continue
  216. Gdk.threads_enter()
  217. pbar.set_fraction(0.0)
  218. bardialog.set_title("Processing recipes")
  219. Gdk.threads_leave()
  220. if isinstance(event, bb.event.ParseProgress):
  221. x = event.current
  222. Gdk.threads_enter()
  223. pbar.set_fraction(x * 1.0 / progress_total)
  224. Gdk.threads_leave()
  225. continue
  226. if isinstance(event, bb.event.ParseCompleted):
  227. Gdk.threads_enter()
  228. bardialog.set_title("Generating dependency tree")
  229. Gdk.threads_leave()
  230. continue
  231. if isinstance(event, bb.event.DepTreeGenerated):
  232. Gdk.threads_enter()
  233. bardialog.hide()
  234. dep.parse(event._depgraph)
  235. Gdk.threads_leave()
  236. if isinstance(event, bb.command.CommandCompleted):
  237. continue
  238. if isinstance(event, bb.event.NoProvider):
  239. print(str(event))
  240. _, error = server.runCommand(["stateShutdown"])
  241. if error:
  242. print('Unable to cleanly shutdown: %s' % error)
  243. break
  244. if isinstance(event, bb.command.CommandFailed):
  245. print(str(event))
  246. return event.exitcode
  247. if isinstance(event, bb.command.CommandExit):
  248. return event.exitcode
  249. if isinstance(event, bb.cooker.CookerExit):
  250. break
  251. continue
  252. except EnvironmentError as ioerror:
  253. # ignore interrupted io
  254. if ioerror.args[0] == 4:
  255. pass
  256. except KeyboardInterrupt:
  257. if shutdown == 2:
  258. print("\nThird Keyboard Interrupt, exit.\n")
  259. break
  260. if shutdown == 1:
  261. print("\nSecond Keyboard Interrupt, stopping...\n")
  262. _, error = server.runCommand(["stateForceShutdown"])
  263. if error:
  264. print('Unable to cleanly stop: %s' % error)
  265. if shutdown == 0:
  266. print("\nKeyboard Interrupt, closing down...\n")
  267. _, error = server.runCommand(["stateShutdown"])
  268. if error:
  269. print('Unable to cleanly shutdown: %s' % error)
  270. shutdown = shutdown + 1
  271. pass