taskexp.py 11 KB

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