runqueue.py 71 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785
  1. #!/usr/bin/env python
  2. # ex:ts=4:sw=4:sts=4:et
  3. # -*- tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*-
  4. """
  5. BitBake 'RunQueue' implementation
  6. Handles preparation and execution of a queue of tasks
  7. """
  8. # Copyright (C) 2006-2007 Richard Purdie
  9. #
  10. # This program is free software; you can redistribute it and/or modify
  11. # it under the terms of the GNU General Public License version 2 as
  12. # published by the Free Software Foundation.
  13. #
  14. # This program is distributed in the hope that it will be useful,
  15. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  16. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  17. # GNU General Public License for more details.
  18. #
  19. # You should have received a copy of the GNU General Public License along
  20. # with this program; if not, write to the Free Software Foundation, Inc.,
  21. # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  22. import copy
  23. import os
  24. import sys
  25. import signal
  26. import stat
  27. import fcntl
  28. import errno
  29. import logging
  30. import bb
  31. from bb import msg, data, event
  32. from bb import monitordisk
  33. import subprocess
  34. try:
  35. import cPickle as pickle
  36. except ImportError:
  37. import pickle
  38. bblogger = logging.getLogger("BitBake")
  39. logger = logging.getLogger("BitBake.RunQueue")
  40. class RunQueueStats:
  41. """
  42. Holds statistics on the tasks handled by the associated runQueue
  43. """
  44. def __init__(self, total):
  45. self.completed = 0
  46. self.skipped = 0
  47. self.failed = 0
  48. self.active = 0
  49. self.total = total
  50. def copy(self):
  51. obj = self.__class__(self.total)
  52. obj.__dict__.update(self.__dict__)
  53. return obj
  54. def taskFailed(self):
  55. self.active = self.active - 1
  56. self.failed = self.failed + 1
  57. def taskCompleted(self, number = 1):
  58. self.active = self.active - number
  59. self.completed = self.completed + number
  60. def taskSkipped(self, number = 1):
  61. self.active = self.active + number
  62. self.skipped = self.skipped + number
  63. def taskActive(self):
  64. self.active = self.active + 1
  65. # These values indicate the next step due to be run in the
  66. # runQueue state machine
  67. runQueuePrepare = 2
  68. runQueueSceneInit = 3
  69. runQueueSceneRun = 4
  70. runQueueRunInit = 5
  71. runQueueRunning = 6
  72. runQueueFailed = 7
  73. runQueueCleanUp = 8
  74. runQueueComplete = 9
  75. runQueueChildProcess = 10
  76. class RunQueueScheduler(object):
  77. """
  78. Control the order tasks are scheduled in.
  79. """
  80. name = "basic"
  81. def __init__(self, runqueue, rqdata):
  82. """
  83. The default scheduler just returns the first buildable task (the
  84. priority map is sorted by task numer)
  85. """
  86. self.rq = runqueue
  87. self.rqdata = rqdata
  88. numTasks = len(self.rqdata.runq_fnid)
  89. self.prio_map = []
  90. self.prio_map.extend(range(numTasks))
  91. def next_buildable_task(self):
  92. """
  93. Return the id of the first task we find that is buildable
  94. """
  95. for tasknum in xrange(len(self.rqdata.runq_fnid)):
  96. taskid = self.prio_map[tasknum]
  97. if self.rq.runq_running[taskid] == 1:
  98. continue
  99. if self.rq.runq_buildable[taskid] == 1:
  100. fn = self.rqdata.taskData.fn_index[self.rqdata.runq_fnid[taskid]]
  101. taskname = self.rqdata.runq_task[taskid]
  102. stamp = bb.build.stampfile(taskname, self.rqdata.dataCache, fn)
  103. if stamp in self.rq.build_stamps.values():
  104. continue
  105. return taskid
  106. def next(self):
  107. """
  108. Return the id of the task we should build next
  109. """
  110. if self.rq.stats.active < self.rq.number_tasks:
  111. return self.next_buildable_task()
  112. class RunQueueSchedulerSpeed(RunQueueScheduler):
  113. """
  114. A scheduler optimised for speed. The priority map is sorted by task weight,
  115. heavier weighted tasks (tasks needed by the most other tasks) are run first.
  116. """
  117. name = "speed"
  118. def __init__(self, runqueue, rqdata):
  119. """
  120. The priority map is sorted by task weight.
  121. """
  122. self.rq = runqueue
  123. self.rqdata = rqdata
  124. sortweight = sorted(copy.deepcopy(self.rqdata.runq_weight))
  125. copyweight = copy.deepcopy(self.rqdata.runq_weight)
  126. self.prio_map = []
  127. for weight in sortweight:
  128. idx = copyweight.index(weight)
  129. self.prio_map.append(idx)
  130. copyweight[idx] = -1
  131. self.prio_map.reverse()
  132. class RunQueueSchedulerCompletion(RunQueueSchedulerSpeed):
  133. """
  134. A scheduler optimised to complete .bb files are quickly as possible. The
  135. priority map is sorted by task weight, but then reordered so once a given
  136. .bb file starts to build, its completed as quickly as possible. This works
  137. well where disk space is at a premium and classes like OE's rm_work are in
  138. force.
  139. """
  140. name = "completion"
  141. def __init__(self, runqueue, rqdata):
  142. RunQueueSchedulerSpeed.__init__(self, runqueue, rqdata)
  143. #FIXME - whilst this groups all fnids together it does not reorder the
  144. #fnid groups optimally.
  145. basemap = copy.deepcopy(self.prio_map)
  146. self.prio_map = []
  147. while (len(basemap) > 0):
  148. entry = basemap.pop(0)
  149. self.prio_map.append(entry)
  150. fnid = self.rqdata.runq_fnid[entry]
  151. todel = []
  152. for entry in basemap:
  153. entry_fnid = self.rqdata.runq_fnid[entry]
  154. if entry_fnid == fnid:
  155. todel.append(basemap.index(entry))
  156. self.prio_map.append(entry)
  157. todel.reverse()
  158. for idx in todel:
  159. del basemap[idx]
  160. class RunQueueData:
  161. """
  162. BitBake Run Queue implementation
  163. """
  164. def __init__(self, rq, cooker, cfgData, dataCache, taskData, targets):
  165. self.cooker = cooker
  166. self.dataCache = dataCache
  167. self.taskData = taskData
  168. self.targets = targets
  169. self.rq = rq
  170. self.warn_multi_bb = False
  171. self.stampwhitelist = cfgData.getVar("BB_STAMP_WHITELIST", True) or ""
  172. self.multi_provider_whitelist = (cfgData.getVar("MULTI_PROVIDER_WHITELIST", True) or "").split()
  173. self.reset()
  174. def reset(self):
  175. self.runq_fnid = []
  176. self.runq_task = []
  177. self.runq_depends = []
  178. self.runq_revdeps = []
  179. self.runq_hash = []
  180. def runq_depends_names(self, ids):
  181. import re
  182. ret = []
  183. for id in self.runq_depends[ids]:
  184. nam = os.path.basename(self.get_user_idstring(id))
  185. nam = re.sub("_[^,]*,", ",", nam)
  186. ret.extend([nam])
  187. return ret
  188. def get_user_idstring(self, task, task_name_suffix = ""):
  189. fn = self.taskData.fn_index[self.runq_fnid[task]]
  190. taskname = self.runq_task[task] + task_name_suffix
  191. return "%s, %s" % (fn, taskname)
  192. def get_task_id(self, fnid, taskname):
  193. for listid in xrange(len(self.runq_fnid)):
  194. if self.runq_fnid[listid] == fnid and self.runq_task[listid] == taskname:
  195. return listid
  196. return None
  197. def circular_depchains_handler(self, tasks):
  198. """
  199. Some tasks aren't buildable, likely due to circular dependency issues.
  200. Identify the circular dependencies and print them in a user readable format.
  201. """
  202. from copy import deepcopy
  203. valid_chains = []
  204. explored_deps = {}
  205. msgs = []
  206. def chain_reorder(chain):
  207. """
  208. Reorder a dependency chain so the lowest task id is first
  209. """
  210. lowest = 0
  211. new_chain = []
  212. for entry in xrange(len(chain)):
  213. if chain[entry] < chain[lowest]:
  214. lowest = entry
  215. new_chain.extend(chain[lowest:])
  216. new_chain.extend(chain[:lowest])
  217. return new_chain
  218. def chain_compare_equal(chain1, chain2):
  219. """
  220. Compare two dependency chains and see if they're the same
  221. """
  222. if len(chain1) != len(chain2):
  223. return False
  224. for index in xrange(len(chain1)):
  225. if chain1[index] != chain2[index]:
  226. return False
  227. return True
  228. def chain_array_contains(chain, chain_array):
  229. """
  230. Return True if chain_array contains chain
  231. """
  232. for ch in chain_array:
  233. if chain_compare_equal(ch, chain):
  234. return True
  235. return False
  236. def find_chains(taskid, prev_chain):
  237. prev_chain.append(taskid)
  238. total_deps = []
  239. total_deps.extend(self.runq_revdeps[taskid])
  240. for revdep in self.runq_revdeps[taskid]:
  241. if revdep in prev_chain:
  242. idx = prev_chain.index(revdep)
  243. # To prevent duplicates, reorder the chain to start with the lowest taskid
  244. # and search through an array of those we've already printed
  245. chain = prev_chain[idx:]
  246. new_chain = chain_reorder(chain)
  247. if not chain_array_contains(new_chain, valid_chains):
  248. valid_chains.append(new_chain)
  249. msgs.append("Dependency loop #%d found:\n" % len(valid_chains))
  250. for dep in new_chain:
  251. msgs.append(" Task %s (%s) (dependent Tasks %s)\n" % (dep, self.get_user_idstring(dep), self.runq_depends_names(dep)))
  252. msgs.append("\n")
  253. if len(valid_chains) > 10:
  254. msgs.append("Aborted dependency loops search after 10 matches.\n")
  255. return msgs
  256. continue
  257. scan = False
  258. if revdep not in explored_deps:
  259. scan = True
  260. elif revdep in explored_deps[revdep]:
  261. scan = True
  262. else:
  263. for dep in prev_chain:
  264. if dep in explored_deps[revdep]:
  265. scan = True
  266. if scan:
  267. find_chains(revdep, copy.deepcopy(prev_chain))
  268. for dep in explored_deps[revdep]:
  269. if dep not in total_deps:
  270. total_deps.append(dep)
  271. explored_deps[taskid] = total_deps
  272. for task in tasks:
  273. find_chains(task, [])
  274. return msgs
  275. def calculate_task_weights(self, endpoints):
  276. """
  277. Calculate a number representing the "weight" of each task. Heavier weighted tasks
  278. have more dependencies and hence should be executed sooner for maximum speed.
  279. This function also sanity checks the task list finding tasks that are not
  280. possible to execute due to circular dependencies.
  281. """
  282. numTasks = len(self.runq_fnid)
  283. weight = []
  284. deps_left = []
  285. task_done = []
  286. for listid in xrange(numTasks):
  287. task_done.append(False)
  288. weight.append(0)
  289. deps_left.append(len(self.runq_revdeps[listid]))
  290. for listid in endpoints:
  291. weight[listid] = 1
  292. task_done[listid] = True
  293. while True:
  294. next_points = []
  295. for listid in endpoints:
  296. for revdep in self.runq_depends[listid]:
  297. weight[revdep] = weight[revdep] + weight[listid]
  298. deps_left[revdep] = deps_left[revdep] - 1
  299. if deps_left[revdep] == 0:
  300. next_points.append(revdep)
  301. task_done[revdep] = True
  302. endpoints = next_points
  303. if len(next_points) == 0:
  304. break
  305. # Circular dependency sanity check
  306. problem_tasks = []
  307. for task in xrange(numTasks):
  308. if task_done[task] is False or deps_left[task] != 0:
  309. problem_tasks.append(task)
  310. logger.debug(2, "Task %s (%s) is not buildable", task, self.get_user_idstring(task))
  311. logger.debug(2, "(Complete marker was %s and the remaining dependency count was %s)\n", task_done[task], deps_left[task])
  312. if problem_tasks:
  313. message = "Unbuildable tasks were found.\n"
  314. message = message + "These are usually caused by circular dependencies and any circular dependency chains found will be printed below. Increase the debug level to see a list of unbuildable tasks.\n\n"
  315. message = message + "Identifying dependency loops (this may take a short while)...\n"
  316. logger.error(message)
  317. msgs = self.circular_depchains_handler(problem_tasks)
  318. message = "\n"
  319. for msg in msgs:
  320. message = message + msg
  321. bb.msg.fatal("RunQueue", message)
  322. return weight
  323. def prepare(self):
  324. """
  325. Turn a set of taskData into a RunQueue and compute data needed
  326. to optimise the execution order.
  327. """
  328. runq_build = []
  329. recursivetasks = {}
  330. recursivetasksselfref = set()
  331. taskData = self.taskData
  332. if len(taskData.tasks_name) == 0:
  333. # Nothing to do
  334. return 0
  335. logger.info("Preparing runqueue")
  336. # Step A - Work out a list of tasks to run
  337. #
  338. # Taskdata gives us a list of possible providers for every build and run
  339. # target ordered by priority. It also gives information on each of those
  340. # providers.
  341. #
  342. # To create the actual list of tasks to execute we fix the list of
  343. # providers and then resolve the dependencies into task IDs. This
  344. # process is repeated for each type of dependency (tdepends, deptask,
  345. # rdeptast, recrdeptask, idepends).
  346. def add_build_dependencies(depids, tasknames, depends):
  347. for depid in depids:
  348. # Won't be in build_targets if ASSUME_PROVIDED
  349. if depid not in taskData.build_targets:
  350. continue
  351. depdata = taskData.build_targets[depid][0]
  352. if depdata is None:
  353. continue
  354. for taskname in tasknames:
  355. taskid = taskData.gettask_id_fromfnid(depdata, taskname)
  356. if taskid is not None:
  357. depends.add(taskid)
  358. def add_runtime_dependencies(depids, tasknames, depends):
  359. for depid in depids:
  360. if depid not in taskData.run_targets:
  361. continue
  362. depdata = taskData.run_targets[depid][0]
  363. if depdata is None:
  364. continue
  365. for taskname in tasknames:
  366. taskid = taskData.gettask_id_fromfnid(depdata, taskname)
  367. if taskid is not None:
  368. depends.add(taskid)
  369. def add_resolved_dependencies(depids, tasknames, depends):
  370. for depid in depids:
  371. for taskname in tasknames:
  372. taskid = taskData.gettask_id_fromfnid(depid, taskname)
  373. if taskid is not None:
  374. depends.add(taskid)
  375. for task in xrange(len(taskData.tasks_name)):
  376. depends = set()
  377. fnid = taskData.tasks_fnid[task]
  378. fn = taskData.fn_index[fnid]
  379. task_deps = self.dataCache.task_deps[fn]
  380. logger.debug(2, "Processing %s:%s", fn, taskData.tasks_name[task])
  381. if fnid not in taskData.failed_fnids:
  382. # Resolve task internal dependencies
  383. #
  384. # e.g. addtask before X after Y
  385. depends = set(taskData.tasks_tdepends[task])
  386. # Resolve 'deptask' dependencies
  387. #
  388. # e.g. do_sometask[deptask] = "do_someothertask"
  389. # (makes sure sometask runs after someothertask of all DEPENDS)
  390. if 'deptask' in task_deps and taskData.tasks_name[task] in task_deps['deptask']:
  391. tasknames = task_deps['deptask'][taskData.tasks_name[task]].split()
  392. add_build_dependencies(taskData.depids[fnid], tasknames, depends)
  393. # Resolve 'rdeptask' dependencies
  394. #
  395. # e.g. do_sometask[rdeptask] = "do_someothertask"
  396. # (makes sure sometask runs after someothertask of all RDEPENDS)
  397. if 'rdeptask' in task_deps and taskData.tasks_name[task] in task_deps['rdeptask']:
  398. tasknames = task_deps['rdeptask'][taskData.tasks_name[task]].split()
  399. add_runtime_dependencies(taskData.rdepids[fnid], tasknames, depends)
  400. # Resolve inter-task dependencies
  401. #
  402. # e.g. do_sometask[depends] = "targetname:do_someothertask"
  403. # (makes sure sometask runs after targetname's someothertask)
  404. idepends = taskData.tasks_idepends[task]
  405. for (depid, idependtask) in idepends:
  406. if depid in taskData.build_targets and not depid in taskData.failed_deps:
  407. # Won't be in build_targets if ASSUME_PROVIDED
  408. depdata = taskData.build_targets[depid][0]
  409. if depdata is not None:
  410. taskid = taskData.gettask_id_fromfnid(depdata, idependtask)
  411. if taskid is None:
  412. bb.msg.fatal("RunQueue", "Task %s in %s depends upon non-existent task %s in %s" % (taskData.tasks_name[task], fn, idependtask, taskData.fn_index[depdata]))
  413. depends.add(taskid)
  414. irdepends = taskData.tasks_irdepends[task]
  415. for (depid, idependtask) in irdepends:
  416. if depid in taskData.run_targets:
  417. # Won't be in run_targets if ASSUME_PROVIDED
  418. depdata = taskData.run_targets[depid][0]
  419. if depdata is not None:
  420. taskid = taskData.gettask_id_fromfnid(depdata, idependtask)
  421. if taskid is None:
  422. bb.msg.fatal("RunQueue", "Task %s in %s rdepends upon non-existent task %s in %s" % (taskData.tasks_name[task], fn, idependtask, taskData.fn_index[depdata]))
  423. depends.add(taskid)
  424. # Resolve recursive 'recrdeptask' dependencies (Part A)
  425. #
  426. # e.g. do_sometask[recrdeptask] = "do_someothertask"
  427. # (makes sure sometask runs after someothertask of all DEPENDS, RDEPENDS and intertask dependencies, recursively)
  428. # We cover the recursive part of the dependencies below
  429. if 'recrdeptask' in task_deps and taskData.tasks_name[task] in task_deps['recrdeptask']:
  430. tasknames = task_deps['recrdeptask'][taskData.tasks_name[task]].split()
  431. recursivetasks[task] = tasknames
  432. add_build_dependencies(taskData.depids[fnid], tasknames, depends)
  433. add_runtime_dependencies(taskData.rdepids[fnid], tasknames, depends)
  434. if taskData.tasks_name[task] in tasknames:
  435. recursivetasksselfref.add(task)
  436. self.runq_fnid.append(taskData.tasks_fnid[task])
  437. self.runq_task.append(taskData.tasks_name[task])
  438. self.runq_depends.append(depends)
  439. self.runq_revdeps.append(set())
  440. self.runq_hash.append("")
  441. runq_build.append(0)
  442. # Resolve recursive 'recrdeptask' dependencies (Part B)
  443. #
  444. # e.g. do_sometask[recrdeptask] = "do_someothertask"
  445. # (makes sure sometask runs after someothertask of all DEPENDS, RDEPENDS and intertask dependencies, recursively)
  446. # We need to do this separately since we need all of self.runq_depends to be complete before this is processed
  447. extradeps = {}
  448. for task in recursivetasks:
  449. extradeps[task] = set(self.runq_depends[task])
  450. tasknames = recursivetasks[task]
  451. seendeps = set()
  452. seenfnid = []
  453. def generate_recdeps(t):
  454. newdeps = set()
  455. add_resolved_dependencies([taskData.tasks_fnid[t]], tasknames, newdeps)
  456. extradeps[task].update(newdeps)
  457. seendeps.add(t)
  458. newdeps.add(t)
  459. for i in newdeps:
  460. for n in self.runq_depends[i]:
  461. if n not in seendeps:
  462. generate_recdeps(n)
  463. generate_recdeps(task)
  464. # Remove circular references so that do_a[recrdeptask] = "do_a do_b" can work
  465. for task in recursivetasks:
  466. extradeps[task].difference_update(recursivetasksselfref)
  467. for task in xrange(len(taskData.tasks_name)):
  468. # Add in extra dependencies
  469. if task in extradeps:
  470. self.runq_depends[task] = extradeps[task]
  471. # Remove all self references
  472. if task in self.runq_depends[task]:
  473. logger.debug(2, "Task %s (%s %s) contains self reference! %s", task, taskData.fn_index[taskData.tasks_fnid[task]], taskData.tasks_name[task], self.runq_depends[task])
  474. self.runq_depends[task].remove(task)
  475. # Step B - Mark all active tasks
  476. #
  477. # Start with the tasks we were asked to run and mark all dependencies
  478. # as active too. If the task is to be 'forced', clear its stamp. Once
  479. # all active tasks are marked, prune the ones we don't need.
  480. logger.verbose("Marking Active Tasks")
  481. def mark_active(listid, depth):
  482. """
  483. Mark an item as active along with its depends
  484. (calls itself recursively)
  485. """
  486. if runq_build[listid] == 1:
  487. return
  488. runq_build[listid] = 1
  489. depends = self.runq_depends[listid]
  490. for depend in depends:
  491. mark_active(depend, depth+1)
  492. self.target_pairs = []
  493. for target in self.targets:
  494. targetid = taskData.getbuild_id(target[0])
  495. if targetid not in taskData.build_targets:
  496. continue
  497. if targetid in taskData.failed_deps:
  498. continue
  499. fnid = taskData.build_targets[targetid][0]
  500. fn = taskData.fn_index[fnid]
  501. self.target_pairs.append((fn, target[1]))
  502. if fnid in taskData.failed_fnids:
  503. continue
  504. if target[1] not in taskData.tasks_lookup[fnid]:
  505. bb.msg.fatal("RunQueue", "Task %s does not exist for target %s" % (target[1], target[0]))
  506. listid = taskData.tasks_lookup[fnid][target[1]]
  507. mark_active(listid, 1)
  508. # Step C - Prune all inactive tasks
  509. #
  510. # Once all active tasks are marked, prune the ones we don't need.
  511. maps = []
  512. delcount = 0
  513. for listid in xrange(len(self.runq_fnid)):
  514. if runq_build[listid-delcount] == 1:
  515. maps.append(listid-delcount)
  516. else:
  517. del self.runq_fnid[listid-delcount]
  518. del self.runq_task[listid-delcount]
  519. del self.runq_depends[listid-delcount]
  520. del runq_build[listid-delcount]
  521. del self.runq_revdeps[listid-delcount]
  522. del self.runq_hash[listid-delcount]
  523. delcount = delcount + 1
  524. maps.append(-1)
  525. #
  526. # Step D - Sanity checks and computation
  527. #
  528. # Check to make sure we still have tasks to run
  529. if len(self.runq_fnid) == 0:
  530. if not taskData.abort:
  531. bb.msg.fatal("RunQueue", "All buildable tasks have been run but the build is incomplete (--continue mode). Errors for the tasks that failed will have been printed above.")
  532. else:
  533. bb.msg.fatal("RunQueue", "No active tasks and not in --continue mode?! Please report this bug.")
  534. logger.verbose("Pruned %s inactive tasks, %s left", delcount, len(self.runq_fnid))
  535. # Remap the dependencies to account for the deleted tasks
  536. # Check we didn't delete a task we depend on
  537. for listid in xrange(len(self.runq_fnid)):
  538. newdeps = []
  539. origdeps = self.runq_depends[listid]
  540. for origdep in origdeps:
  541. if maps[origdep] == -1:
  542. bb.msg.fatal("RunQueue", "Invalid mapping - Should never happen!")
  543. newdeps.append(maps[origdep])
  544. self.runq_depends[listid] = set(newdeps)
  545. logger.verbose("Assign Weightings")
  546. # Generate a list of reverse dependencies to ease future calculations
  547. for listid in xrange(len(self.runq_fnid)):
  548. for dep in self.runq_depends[listid]:
  549. self.runq_revdeps[dep].add(listid)
  550. # Identify tasks at the end of dependency chains
  551. # Error on circular dependency loops (length two)
  552. endpoints = []
  553. for listid in xrange(len(self.runq_fnid)):
  554. revdeps = self.runq_revdeps[listid]
  555. if len(revdeps) == 0:
  556. endpoints.append(listid)
  557. for dep in revdeps:
  558. if dep in self.runq_depends[listid]:
  559. #self.dump_data(taskData)
  560. bb.msg.fatal("RunQueue", "Task %s (%s) has circular dependency on %s (%s)" % (taskData.fn_index[self.runq_fnid[dep]], self.runq_task[dep], taskData.fn_index[self.runq_fnid[listid]], self.runq_task[listid]))
  561. logger.verbose("Compute totals (have %s endpoint(s))", len(endpoints))
  562. # Calculate task weights
  563. # Check of higher length circular dependencies
  564. self.runq_weight = self.calculate_task_weights(endpoints)
  565. # Sanity Check - Check for multiple tasks building the same provider
  566. prov_list = {}
  567. seen_fn = []
  568. for task in xrange(len(self.runq_fnid)):
  569. fn = taskData.fn_index[self.runq_fnid[task]]
  570. if fn in seen_fn:
  571. continue
  572. seen_fn.append(fn)
  573. for prov in self.dataCache.fn_provides[fn]:
  574. if prov not in prov_list:
  575. prov_list[prov] = [fn]
  576. elif fn not in prov_list[prov]:
  577. prov_list[prov].append(fn)
  578. for prov in prov_list:
  579. if len(prov_list[prov]) > 1 and prov not in self.multi_provider_whitelist:
  580. msg = "Multiple .bb files are due to be built which each provide %s (%s)." % (prov, " ".join(prov_list[prov]))
  581. if self.warn_multi_bb:
  582. logger.warn(msg)
  583. else:
  584. msg += "\n This usually means one provides something the other doesn't and should."
  585. logger.error(msg)
  586. # Create a whitelist usable by the stamp checks
  587. stampfnwhitelist = []
  588. for entry in self.stampwhitelist.split():
  589. entryid = self.taskData.getbuild_id(entry)
  590. if entryid not in self.taskData.build_targets:
  591. continue
  592. fnid = self.taskData.build_targets[entryid][0]
  593. fn = self.taskData.fn_index[fnid]
  594. stampfnwhitelist.append(fn)
  595. self.stampfnwhitelist = stampfnwhitelist
  596. # Iterate over the task list looking for tasks with a 'setscene' function
  597. self.runq_setscene = []
  598. if not self.cooker.configuration.nosetscene:
  599. for task in range(len(self.runq_fnid)):
  600. setscene = taskData.gettask_id(self.taskData.fn_index[self.runq_fnid[task]], self.runq_task[task] + "_setscene", False)
  601. if not setscene:
  602. continue
  603. self.runq_setscene.append(task)
  604. def invalidate_task(fn, taskname, error_nostamp):
  605. taskdep = self.dataCache.task_deps[fn]
  606. if 'nostamp' in taskdep and taskname in taskdep['nostamp']:
  607. if error_nostamp:
  608. bb.fatal("Task %s is marked nostamp, cannot invalidate this task" % taskname)
  609. else:
  610. bb.debug(1, "Task %s is marked nostamp, cannot invalidate this task" % taskname)
  611. else:
  612. logger.verbose("Invalidate task %s, %s", taskname, fn)
  613. bb.parse.siggen.invalidate_task(taskname, self.dataCache, fn)
  614. # Invalidate task if force mode active
  615. if self.cooker.configuration.force:
  616. for (fn, target) in self.target_pairs:
  617. invalidate_task(fn, target, False)
  618. # Invalidate task if invalidate mode active
  619. if self.cooker.configuration.invalidate_stamp:
  620. for (fn, target) in self.target_pairs:
  621. for st in self.cooker.configuration.invalidate_stamp.split(','):
  622. invalidate_task(fn, "do_%s" % st, True)
  623. # Interate over the task list and call into the siggen code
  624. dealtwith = set()
  625. todeal = set(range(len(self.runq_fnid)))
  626. while len(todeal) > 0:
  627. for task in todeal.copy():
  628. if len(self.runq_depends[task] - dealtwith) == 0:
  629. dealtwith.add(task)
  630. todeal.remove(task)
  631. procdep = []
  632. for dep in self.runq_depends[task]:
  633. procdep.append(self.taskData.fn_index[self.runq_fnid[dep]] + "." + self.runq_task[dep])
  634. self.runq_hash[task] = bb.parse.siggen.get_taskhash(self.taskData.fn_index[self.runq_fnid[task]], self.runq_task[task], procdep, self.dataCache)
  635. self.hashes = {}
  636. self.hash_deps = {}
  637. for task in xrange(len(self.runq_fnid)):
  638. identifier = '%s.%s' % (self.taskData.fn_index[self.runq_fnid[task]],
  639. self.runq_task[task])
  640. self.hashes[identifier] = self.runq_hash[task]
  641. deps = []
  642. for dep in self.runq_depends[task]:
  643. depidentifier = '%s.%s' % (self.taskData.fn_index[self.runq_fnid[dep]],
  644. self.runq_task[dep])
  645. deps.append(depidentifier)
  646. self.hash_deps[identifier] = deps
  647. return len(self.runq_fnid)
  648. def dump_data(self, taskQueue):
  649. """
  650. Dump some debug information on the internal data structures
  651. """
  652. logger.debug(3, "run_tasks:")
  653. for task in xrange(len(self.rqdata.runq_task)):
  654. logger.debug(3, " (%s)%s - %s: %s Deps %s RevDeps %s", task,
  655. taskQueue.fn_index[self.rqdata.runq_fnid[task]],
  656. self.rqdata.runq_task[task],
  657. self.rqdata.runq_weight[task],
  658. self.rqdata.runq_depends[task],
  659. self.rqdata.runq_revdeps[task])
  660. logger.debug(3, "sorted_tasks:")
  661. for task1 in xrange(len(self.rqdata.runq_task)):
  662. if task1 in self.prio_map:
  663. task = self.prio_map[task1]
  664. logger.debug(3, " (%s)%s - %s: %s Deps %s RevDeps %s", task,
  665. taskQueue.fn_index[self.rqdata.runq_fnid[task]],
  666. self.rqdata.runq_task[task],
  667. self.rqdata.runq_weight[task],
  668. self.rqdata.runq_depends[task],
  669. self.rqdata.runq_revdeps[task])
  670. class RunQueue:
  671. def __init__(self, cooker, cfgData, dataCache, taskData, targets):
  672. self.cooker = cooker
  673. self.cfgData = cfgData
  674. self.rqdata = RunQueueData(self, cooker, cfgData, dataCache, taskData, targets)
  675. self.stamppolicy = cfgData.getVar("BB_STAMP_POLICY", True) or "perfile"
  676. self.hashvalidate = cfgData.getVar("BB_HASHCHECK_FUNCTION", True) or None
  677. self.setsceneverify = cfgData.getVar("BB_SETSCENE_VERIFY_FUNCTION", True) or None
  678. self.depvalidate = cfgData.getVar("BB_SETSCENE_DEPVALID", True) or None
  679. self.state = runQueuePrepare
  680. # For disk space monitor
  681. self.dm = monitordisk.diskMonitor(cfgData)
  682. self.rqexe = None
  683. def check_stamp_task(self, task, taskname = None, recurse = False, cache = None):
  684. def get_timestamp(f):
  685. try:
  686. if not os.access(f, os.F_OK):
  687. return None
  688. return os.stat(f)[stat.ST_MTIME]
  689. except:
  690. return None
  691. if self.stamppolicy == "perfile":
  692. fulldeptree = False
  693. else:
  694. fulldeptree = True
  695. stampwhitelist = []
  696. if self.stamppolicy == "whitelist":
  697. stampwhitelist = self.rqdata.stampfnwhitelist
  698. fn = self.rqdata.taskData.fn_index[self.rqdata.runq_fnid[task]]
  699. if taskname is None:
  700. taskname = self.rqdata.runq_task[task]
  701. stampfile = bb.build.stampfile(taskname, self.rqdata.dataCache, fn)
  702. # If the stamp is missing its not current
  703. if not os.access(stampfile, os.F_OK):
  704. logger.debug(2, "Stampfile %s not available", stampfile)
  705. return False
  706. # If its a 'nostamp' task, it's not current
  707. taskdep = self.rqdata.dataCache.task_deps[fn]
  708. if 'nostamp' in taskdep and taskname in taskdep['nostamp']:
  709. logger.debug(2, "%s.%s is nostamp\n", fn, taskname)
  710. return False
  711. if taskname != "do_setscene" and taskname.endswith("_setscene"):
  712. return True
  713. if cache is None:
  714. cache = {}
  715. iscurrent = True
  716. t1 = get_timestamp(stampfile)
  717. for dep in self.rqdata.runq_depends[task]:
  718. if iscurrent:
  719. fn2 = self.rqdata.taskData.fn_index[self.rqdata.runq_fnid[dep]]
  720. taskname2 = self.rqdata.runq_task[dep]
  721. stampfile2 = bb.build.stampfile(taskname2, self.rqdata.dataCache, fn2)
  722. stampfile3 = bb.build.stampfile(taskname2 + "_setscene", self.rqdata.dataCache, fn2)
  723. t2 = get_timestamp(stampfile2)
  724. t3 = get_timestamp(stampfile3)
  725. if t3 and t3 > t2:
  726. continue
  727. if fn == fn2 or (fulldeptree and fn2 not in stampwhitelist):
  728. if not t2:
  729. logger.debug(2, 'Stampfile %s does not exist', stampfile2)
  730. iscurrent = False
  731. if t1 < t2:
  732. logger.debug(2, 'Stampfile %s < %s', stampfile, stampfile2)
  733. iscurrent = False
  734. if recurse and iscurrent:
  735. if dep in cache:
  736. iscurrent = cache[dep]
  737. if not iscurrent:
  738. logger.debug(2, 'Stampfile for dependency %s:%s invalid (cached)' % (fn2, taskname2))
  739. else:
  740. iscurrent = self.check_stamp_task(dep, recurse=True, cache=cache)
  741. cache[dep] = iscurrent
  742. if recurse:
  743. cache[task] = iscurrent
  744. return iscurrent
  745. def _execute_runqueue(self):
  746. """
  747. Run the tasks in a queue prepared by rqdata.prepare()
  748. Upon failure, optionally try to recover the build using any alternate providers
  749. (if the abort on failure configuration option isn't set)
  750. """
  751. retval = 0.5
  752. if self.state is runQueuePrepare:
  753. self.rqexe = RunQueueExecuteDummy(self)
  754. if self.rqdata.prepare() == 0:
  755. self.state = runQueueComplete
  756. else:
  757. self.state = runQueueSceneInit
  758. if self.state is runQueueSceneInit:
  759. if self.cooker.configuration.dump_signatures:
  760. self.dump_signatures()
  761. else:
  762. self.rqexe = RunQueueExecuteScenequeue(self)
  763. if self.state in [runQueueSceneRun, runQueueRunning, runQueueCleanUp]:
  764. self.dm.check(self)
  765. if self.state is runQueueSceneRun:
  766. retval = self.rqexe.execute()
  767. if self.state is runQueueRunInit:
  768. logger.info("Executing RunQueue Tasks")
  769. self.rqexe = RunQueueExecuteTasks(self)
  770. self.state = runQueueRunning
  771. if self.state is runQueueRunning:
  772. retval = self.rqexe.execute()
  773. if self.state is runQueueCleanUp:
  774. self.rqexe.finish()
  775. if self.state is runQueueComplete or self.state is runQueueFailed:
  776. if self.rqexe.stats.failed:
  777. logger.info("Tasks Summary: Attempted %d tasks of which %d didn't need to be rerun and %d failed.", self.rqexe.stats.completed + self.rqexe.stats.failed, self.rqexe.stats.skipped, self.rqexe.stats.failed)
  778. else:
  779. # Let's avoid the word "failed" if nothing actually did
  780. logger.info("Tasks Summary: Attempted %d tasks of which %d didn't need to be rerun and all succeeded.", self.rqexe.stats.completed, self.rqexe.stats.skipped)
  781. if self.state is runQueueFailed:
  782. if not self.rqdata.taskData.tryaltconfigs:
  783. raise bb.runqueue.TaskFailure(self.rqexe.failed_fnids)
  784. for fnid in self.rqexe.failed_fnids:
  785. self.rqdata.taskData.fail_fnid(fnid)
  786. self.rqdata.reset()
  787. if self.state is runQueueComplete:
  788. # All done
  789. return False
  790. if self.state is runQueueChildProcess:
  791. print("Child process, eeek, shouldn't happen!")
  792. return False
  793. # Loop
  794. return retval
  795. def execute_runqueue(self):
  796. # Catch unexpected exceptions and ensure we exit when an error occurs, not loop.
  797. try:
  798. return self._execute_runqueue()
  799. except bb.runqueue.TaskFailure:
  800. raise
  801. except SystemExit:
  802. raise
  803. except:
  804. logger.error("An uncaught exception occured in runqueue, please see the failure below:")
  805. try:
  806. self.rqexe.teardown()
  807. except:
  808. pass
  809. self.state = runQueueComplete
  810. raise
  811. def finish_runqueue(self, now = False):
  812. if not self.rqexe:
  813. return
  814. if now:
  815. self.rqexe.finish_now()
  816. else:
  817. self.rqexe.finish()
  818. def dump_signatures(self):
  819. self.state = runQueueComplete
  820. done = set()
  821. bb.note("Reparsing files to collect dependency data")
  822. for task in range(len(self.rqdata.runq_fnid)):
  823. if self.rqdata.runq_fnid[task] not in done:
  824. fn = self.rqdata.taskData.fn_index[self.rqdata.runq_fnid[task]]
  825. the_data = bb.cache.Cache.loadDataFull(fn, self.cooker.collection.get_file_appends(fn), self.cooker.data)
  826. done.add(self.rqdata.runq_fnid[task])
  827. bb.parse.siggen.dump_sigs(self.rqdata.dataCache)
  828. return
  829. class RunQueueExecute:
  830. def __init__(self, rq):
  831. self.rq = rq
  832. self.cooker = rq.cooker
  833. self.cfgData = rq.cfgData
  834. self.rqdata = rq.rqdata
  835. self.number_tasks = int(self.cfgData.getVar("BB_NUMBER_THREADS", True) or 1)
  836. self.scheduler = self.cfgData.getVar("BB_SCHEDULER", True) or "speed"
  837. self.runq_buildable = []
  838. self.runq_running = []
  839. self.runq_complete = []
  840. self.build_stamps = {}
  841. self.failed_fnids = []
  842. self.stampcache = {}
  843. logger.debug(1, "Starting bitbake-worker")
  844. self.worker = subprocess.Popen(["bitbake-worker", "decafbad"], stdout=subprocess.PIPE, stdin=subprocess.PIPE)
  845. bb.utils.nonblockingfd(self.worker.stdout)
  846. self.workerpipe = runQueuePipe(self.worker.stdout, None, self.cfgData, self)
  847. workerdata = {
  848. "taskdeps" : self.rqdata.dataCache.task_deps,
  849. "fakerootenv" : self.rqdata.dataCache.fakerootenv,
  850. "fakerootdirs" : self.rqdata.dataCache.fakerootdirs,
  851. "fakerootnoenv" : self.rqdata.dataCache.fakerootnoenv,
  852. "hashes" : self.rqdata.hashes,
  853. "hash_deps" : self.rqdata.hash_deps,
  854. "sigchecksums" : bb.parse.siggen.file_checksum_values,
  855. "runq_hash" : self.rqdata.runq_hash,
  856. "logdefaultdebug" : bb.msg.loggerDefaultDebugLevel,
  857. "logdefaultverbose" : bb.msg.loggerDefaultVerbose,
  858. "logdefaultverboselogs" : bb.msg.loggerVerboseLogs,
  859. "logdefaultdomain" : bb.msg.loggerDefaultDomains,
  860. }
  861. self.worker.stdin.write("<cookerconfig>" + pickle.dumps(self.cooker.configuration) + "</cookerconfig>")
  862. self.worker.stdin.write("<workerdata>" + pickle.dumps(workerdata) + "</workerdata>")
  863. self.worker.stdin.flush()
  864. def runqueue_process_waitpid(self, task, status):
  865. # self.build_stamps[pid] may not exist when use shared work directory.
  866. if task in self.build_stamps:
  867. del self.build_stamps[task]
  868. if status != 0:
  869. self.task_fail(task, status)
  870. else:
  871. self.task_complete(task)
  872. return True
  873. def finish_now(self):
  874. self.worker.stdin.write("<finishnow></finishnow>")
  875. self.worker.stdin.flush()
  876. self.teardown()
  877. if len(self.failed_fnids) != 0:
  878. self.rq.state = runQueueFailed
  879. return
  880. self.rq.state = runQueueComplete
  881. return
  882. def finish(self):
  883. self.rq.state = runQueueCleanUp
  884. if self.stats.active > 0:
  885. bb.event.fire(runQueueExitWait(self.stats.active), self.cfgData)
  886. self.workerpipe.read()
  887. return
  888. self.teardown()
  889. if len(self.failed_fnids) != 0:
  890. self.rq.state = runQueueFailed
  891. return
  892. self.rq.state = runQueueComplete
  893. return
  894. def check_dependencies(self, task, taskdeps, setscene = False):
  895. if not self.rq.depvalidate:
  896. return False
  897. taskdata = {}
  898. taskdeps.add(task)
  899. for dep in taskdeps:
  900. if setscene:
  901. depid = self.rqdata.runq_setscene[dep]
  902. else:
  903. depid = dep
  904. fn = self.rqdata.taskData.fn_index[self.rqdata.runq_fnid[depid]]
  905. pn = self.rqdata.dataCache.pkg_fn[fn]
  906. taskname = self.rqdata.runq_task[depid]
  907. taskdata[dep] = [pn, taskname, fn]
  908. call = self.rq.depvalidate + "(task, taskdata, notneeded, d)"
  909. locs = { "task" : task, "taskdata" : taskdata, "notneeded" : self.scenequeue_notneeded, "d" : self.cooker.data }
  910. valid = bb.utils.better_eval(call, locs)
  911. return valid
  912. def teardown(self):
  913. logger.debug(1, "Teardown for bitbake-worker")
  914. self.worker.stdin.write("<quit></quit>")
  915. self.worker.stdin.flush()
  916. while self.worker.returncode is None:
  917. self.workerpipe.read()
  918. self.worker.poll()
  919. while self.workerpipe.read():
  920. continue
  921. class RunQueueExecuteDummy(RunQueueExecute):
  922. def __init__(self, rq):
  923. self.rq = rq
  924. self.stats = RunQueueStats(0)
  925. def finish(self):
  926. self.rq.state = runQueueComplete
  927. return
  928. class RunQueueExecuteTasks(RunQueueExecute):
  929. def __init__(self, rq):
  930. RunQueueExecute.__init__(self, rq)
  931. self.stats = RunQueueStats(len(self.rqdata.runq_fnid))
  932. self.stampcache = {}
  933. # Mark initial buildable tasks
  934. for task in xrange(self.stats.total):
  935. self.runq_running.append(0)
  936. self.runq_complete.append(0)
  937. if len(self.rqdata.runq_depends[task]) == 0:
  938. self.runq_buildable.append(1)
  939. else:
  940. self.runq_buildable.append(0)
  941. if len(self.rqdata.runq_revdeps[task]) > 0 and self.rqdata.runq_revdeps[task].issubset(self.rq.scenequeue_covered) and task not in self.rq.scenequeue_notcovered:
  942. self.rq.scenequeue_covered.add(task)
  943. found = True
  944. while found:
  945. found = False
  946. for task in xrange(self.stats.total):
  947. if task in self.rq.scenequeue_covered:
  948. continue
  949. logger.debug(1, 'Considering %s (%s): %s' % (task, self.rqdata.get_user_idstring(task), str(self.rqdata.runq_revdeps[task])))
  950. if len(self.rqdata.runq_revdeps[task]) > 0 and self.rqdata.runq_revdeps[task].issubset(self.rq.scenequeue_covered) and task not in self.rq.scenequeue_notcovered:
  951. found = True
  952. self.rq.scenequeue_covered.add(task)
  953. logger.debug(1, 'Skip list (pre setsceneverify) %s', sorted(self.rq.scenequeue_covered))
  954. # Allow the metadata to elect for setscene tasks to run anyway
  955. covered_remove = set()
  956. if self.rq.setsceneverify:
  957. invalidtasks = []
  958. for task in xrange(len(self.rqdata.runq_task)):
  959. fn = self.rqdata.taskData.fn_index[self.rqdata.runq_fnid[task]]
  960. taskname = self.rqdata.runq_task[task]
  961. taskdep = self.rqdata.dataCache.task_deps[fn]
  962. if 'noexec' in taskdep and taskname in taskdep['noexec']:
  963. continue
  964. if self.rq.check_stamp_task(task, taskname + "_setscene", cache=self.stampcache):
  965. logger.debug(2, 'Setscene stamp current for task %s(%s)', task, self.rqdata.get_user_idstring(task))
  966. continue
  967. if self.rq.check_stamp_task(task, taskname, recurse = True, cache=self.stampcache):
  968. logger.debug(2, 'Normal stamp current for task %s(%s)', task, self.rqdata.get_user_idstring(task))
  969. continue
  970. invalidtasks.append(task)
  971. call = self.rq.setsceneverify + "(covered, tasknames, fnids, fns, d, invalidtasks=invalidtasks)"
  972. call2 = self.rq.setsceneverify + "(covered, tasknames, fnids, fns, d)"
  973. locs = { "covered" : self.rq.scenequeue_covered, "tasknames" : self.rqdata.runq_task, "fnids" : self.rqdata.runq_fnid, "fns" : self.rqdata.taskData.fn_index, "d" : self.cooker.data, "invalidtasks" : invalidtasks }
  974. # Backwards compatibility with older versions without invalidtasks
  975. try:
  976. covered_remove = bb.utils.better_eval(call, locs)
  977. except TypeError:
  978. covered_remove = bb.utils.better_eval(call2, locs)
  979. for task in covered_remove:
  980. fn = self.rqdata.taskData.fn_index[self.rqdata.runq_fnid[task]]
  981. taskname = self.rqdata.runq_task[task] + '_setscene'
  982. bb.build.del_stamp(taskname, self.rqdata.dataCache, fn)
  983. logger.debug(1, 'Not skipping task %s due to setsceneverify', task)
  984. self.rq.scenequeue_covered.remove(task)
  985. logger.debug(1, 'Full skip list %s', self.rq.scenequeue_covered)
  986. event.fire(bb.event.StampUpdate(self.rqdata.target_pairs, self.rqdata.dataCache.stamp), self.cfgData)
  987. schedulers = self.get_schedulers()
  988. for scheduler in schedulers:
  989. if self.scheduler == scheduler.name:
  990. self.sched = scheduler(self, self.rqdata)
  991. logger.debug(1, "Using runqueue scheduler '%s'", scheduler.name)
  992. break
  993. else:
  994. bb.fatal("Invalid scheduler '%s'. Available schedulers: %s" %
  995. (self.scheduler, ", ".join(obj.name for obj in schedulers)))
  996. def get_schedulers(self):
  997. schedulers = set(obj for obj in globals().values()
  998. if type(obj) is type and
  999. issubclass(obj, RunQueueScheduler))
  1000. user_schedulers = self.cfgData.getVar("BB_SCHEDULERS", True)
  1001. if user_schedulers:
  1002. for sched in user_schedulers.split():
  1003. if not "." in sched:
  1004. bb.note("Ignoring scheduler '%s' from BB_SCHEDULERS: not an import" % sched)
  1005. continue
  1006. modname, name = sched.rsplit(".", 1)
  1007. try:
  1008. module = __import__(modname, fromlist=(name,))
  1009. except ImportError as exc:
  1010. logger.critical("Unable to import scheduler '%s' from '%s': %s" % (name, modname, exc))
  1011. raise SystemExit(1)
  1012. else:
  1013. schedulers.add(getattr(module, name))
  1014. return schedulers
  1015. def task_completeoutright(self, task):
  1016. """
  1017. Mark a task as completed
  1018. Look at the reverse dependencies and mark any task with
  1019. completed dependencies as buildable
  1020. """
  1021. self.runq_complete[task] = 1
  1022. for revdep in self.rqdata.runq_revdeps[task]:
  1023. if self.runq_running[revdep] == 1:
  1024. continue
  1025. if self.runq_buildable[revdep] == 1:
  1026. continue
  1027. alldeps = 1
  1028. for dep in self.rqdata.runq_depends[revdep]:
  1029. if self.runq_complete[dep] != 1:
  1030. alldeps = 0
  1031. if alldeps == 1:
  1032. self.runq_buildable[revdep] = 1
  1033. fn = self.rqdata.taskData.fn_index[self.rqdata.runq_fnid[revdep]]
  1034. taskname = self.rqdata.runq_task[revdep]
  1035. logger.debug(1, "Marking task %s (%s, %s) as buildable", revdep, fn, taskname)
  1036. def task_complete(self, task):
  1037. self.stats.taskCompleted()
  1038. bb.event.fire(runQueueTaskCompleted(task, self.stats, self.rq), self.cfgData)
  1039. self.task_completeoutright(task)
  1040. def task_fail(self, task, exitcode):
  1041. """
  1042. Called when a task has failed
  1043. Updates the state engine with the failure
  1044. """
  1045. self.stats.taskFailed()
  1046. fnid = self.rqdata.runq_fnid[task]
  1047. self.failed_fnids.append(fnid)
  1048. bb.event.fire(runQueueTaskFailed(task, self.stats, exitcode, self.rq), self.cfgData)
  1049. if self.rqdata.taskData.abort:
  1050. self.rq.state = runQueueCleanUp
  1051. def task_skip(self, task):
  1052. self.runq_running[task] = 1
  1053. self.runq_buildable[task] = 1
  1054. self.task_completeoutright(task)
  1055. self.stats.taskCompleted()
  1056. self.stats.taskSkipped()
  1057. def execute(self):
  1058. """
  1059. Run the tasks in a queue prepared by rqdata.prepare()
  1060. """
  1061. self.workerpipe.read()
  1062. if self.stats.total == 0:
  1063. # nothing to do
  1064. self.rq.state = runQueueCleanUp
  1065. task = self.sched.next()
  1066. if task is not None:
  1067. fn = self.rqdata.taskData.fn_index[self.rqdata.runq_fnid[task]]
  1068. taskname = self.rqdata.runq_task[task]
  1069. if task in self.rq.scenequeue_covered:
  1070. logger.debug(2, "Setscene covered task %s (%s)", task,
  1071. self.rqdata.get_user_idstring(task))
  1072. self.task_skip(task)
  1073. return True
  1074. if self.rq.check_stamp_task(task, taskname, cache=self.stampcache):
  1075. logger.debug(2, "Stamp current task %s (%s)", task,
  1076. self.rqdata.get_user_idstring(task))
  1077. self.task_skip(task)
  1078. return True
  1079. taskdep = self.rqdata.dataCache.task_deps[fn]
  1080. if 'noexec' in taskdep and taskname in taskdep['noexec']:
  1081. startevent = runQueueTaskStarted(task, self.stats, self.rq,
  1082. noexec=True)
  1083. bb.event.fire(startevent, self.cfgData)
  1084. self.runq_running[task] = 1
  1085. self.stats.taskActive()
  1086. bb.build.make_stamp(taskname, self.rqdata.dataCache, fn)
  1087. self.task_complete(task)
  1088. return True
  1089. else:
  1090. startevent = runQueueTaskStarted(task, self.stats, self.rq)
  1091. bb.event.fire(startevent, self.cfgData)
  1092. self.worker.stdin.write("<runtask>" + pickle.dumps((fn, task, taskname, False, self.cooker.collection.get_file_appends(fn))) + "</runtask>")
  1093. self.worker.stdin.flush()
  1094. self.build_stamps[task] = bb.build.stampfile(taskname, self.rqdata.dataCache, fn)
  1095. self.runq_running[task] = 1
  1096. self.stats.taskActive()
  1097. if self.stats.active < self.number_tasks:
  1098. return True
  1099. if self.stats.active > 0:
  1100. self.workerpipe.read()
  1101. return 0.5
  1102. self.teardown()
  1103. if len(self.failed_fnids) != 0:
  1104. self.rq.state = runQueueFailed
  1105. return True
  1106. # Sanity Checks
  1107. for task in xrange(self.stats.total):
  1108. if self.runq_buildable[task] == 0:
  1109. logger.error("Task %s never buildable!", task)
  1110. if self.runq_running[task] == 0:
  1111. logger.error("Task %s never ran!", task)
  1112. if self.runq_complete[task] == 0:
  1113. logger.error("Task %s never completed!", task)
  1114. self.rq.state = runQueueComplete
  1115. return True
  1116. class RunQueueExecuteScenequeue(RunQueueExecute):
  1117. def __init__(self, rq):
  1118. RunQueueExecute.__init__(self, rq)
  1119. self.scenequeue_covered = set()
  1120. self.scenequeue_notcovered = set()
  1121. self.scenequeue_notneeded = set()
  1122. # If we don't have any setscene functions, skip this step
  1123. if len(self.rqdata.runq_setscene) == 0:
  1124. rq.scenequeue_covered = set()
  1125. self.teardown()
  1126. rq.state = runQueueRunInit
  1127. return
  1128. self.stats = RunQueueStats(len(self.rqdata.runq_setscene))
  1129. sq_revdeps = []
  1130. sq_revdeps_new = []
  1131. sq_revdeps_squash = []
  1132. # We need to construct a dependency graph for the setscene functions. Intermediate
  1133. # dependencies between the setscene tasks only complicate the code. This code
  1134. # therefore aims to collapse the huge runqueue dependency tree into a smaller one
  1135. # only containing the setscene functions.
  1136. for task in xrange(self.stats.total):
  1137. self.runq_running.append(0)
  1138. self.runq_complete.append(0)
  1139. self.runq_buildable.append(0)
  1140. # First process the chains up to the first setscene task.
  1141. endpoints = {}
  1142. for task in xrange(len(self.rqdata.runq_fnid)):
  1143. sq_revdeps.append(copy.copy(self.rqdata.runq_revdeps[task]))
  1144. sq_revdeps_new.append(set())
  1145. if (len(self.rqdata.runq_revdeps[task]) == 0) and task not in self.rqdata.runq_setscene:
  1146. endpoints[task] = set()
  1147. # Secondly process the chains between setscene tasks.
  1148. for task in self.rqdata.runq_setscene:
  1149. for dep in self.rqdata.runq_depends[task]:
  1150. if dep not in endpoints:
  1151. endpoints[dep] = set()
  1152. endpoints[dep].add(task)
  1153. def process_endpoints(endpoints):
  1154. newendpoints = {}
  1155. for point, task in endpoints.items():
  1156. tasks = set()
  1157. if task:
  1158. tasks |= task
  1159. if sq_revdeps_new[point]:
  1160. tasks |= sq_revdeps_new[point]
  1161. sq_revdeps_new[point] = set()
  1162. if point in self.rqdata.runq_setscene:
  1163. sq_revdeps_new[point] = tasks
  1164. for dep in self.rqdata.runq_depends[point]:
  1165. if point in sq_revdeps[dep]:
  1166. sq_revdeps[dep].remove(point)
  1167. if tasks:
  1168. sq_revdeps_new[dep] |= tasks
  1169. if (len(sq_revdeps[dep]) == 0 or len(sq_revdeps_new[dep]) != 0) and dep not in self.rqdata.runq_setscene:
  1170. newendpoints[dep] = task
  1171. if len(newendpoints) != 0:
  1172. process_endpoints(newendpoints)
  1173. process_endpoints(endpoints)
  1174. # Build a list of setscene tasks which as "unskippable"
  1175. # These are direct endpoints referenced by the build
  1176. endpoints2 = {}
  1177. sq_revdeps2 = []
  1178. sq_revdeps_new2 = []
  1179. def process_endpoints2(endpoints):
  1180. newendpoints = {}
  1181. for point, task in endpoints.items():
  1182. tasks = set([point])
  1183. if task:
  1184. tasks |= task
  1185. if sq_revdeps_new2[point]:
  1186. tasks |= sq_revdeps_new2[point]
  1187. sq_revdeps_new2[point] = set()
  1188. if point in self.rqdata.runq_setscene:
  1189. sq_revdeps_new2[point] = tasks
  1190. for dep in self.rqdata.runq_depends[point]:
  1191. if point in sq_revdeps2[dep]:
  1192. sq_revdeps2[dep].remove(point)
  1193. if tasks:
  1194. sq_revdeps_new2[dep] |= tasks
  1195. if (len(sq_revdeps2[dep]) == 0 or len(sq_revdeps_new2[dep]) != 0) and dep not in self.rqdata.runq_setscene:
  1196. newendpoints[dep] = tasks
  1197. if len(newendpoints) != 0:
  1198. process_endpoints2(newendpoints)
  1199. for task in xrange(len(self.rqdata.runq_fnid)):
  1200. sq_revdeps2.append(copy.copy(self.rqdata.runq_revdeps[task]))
  1201. sq_revdeps_new2.append(set())
  1202. if (len(self.rqdata.runq_revdeps[task]) == 0) and task not in self.rqdata.runq_setscene:
  1203. endpoints2[task] = set()
  1204. process_endpoints2(endpoints2)
  1205. self.unskippable = []
  1206. for task in self.rqdata.runq_setscene:
  1207. if sq_revdeps_new2[task]:
  1208. self.unskippable.append(self.rqdata.runq_setscene.index(task))
  1209. for task in xrange(len(self.rqdata.runq_fnid)):
  1210. if task in self.rqdata.runq_setscene:
  1211. deps = set()
  1212. for dep in sq_revdeps_new[task]:
  1213. deps.add(self.rqdata.runq_setscene.index(dep))
  1214. sq_revdeps_squash.append(deps)
  1215. elif len(sq_revdeps_new[task]) != 0:
  1216. bb.msg.fatal("RunQueue", "Something went badly wrong during scenequeue generation, aborting. Please report this problem.")
  1217. # Resolve setscene inter-task dependencies
  1218. # e.g. do_sometask_setscene[depends] = "targetname:do_someothertask_setscene"
  1219. # Note that anything explicitly depended upon will have its reverse dependencies removed to avoid circular dependencies
  1220. for task in self.rqdata.runq_setscene:
  1221. realid = self.rqdata.taskData.gettask_id(self.rqdata.taskData.fn_index[self.rqdata.runq_fnid[task]], self.rqdata.runq_task[task] + "_setscene", False)
  1222. idepends = self.rqdata.taskData.tasks_idepends[realid]
  1223. for (depid, idependtask) in idepends:
  1224. if depid not in self.rqdata.taskData.build_targets:
  1225. continue
  1226. depdata = self.rqdata.taskData.build_targets[depid][0]
  1227. if depdata is None:
  1228. continue
  1229. dep = self.rqdata.taskData.fn_index[depdata]
  1230. taskid = self.rqdata.get_task_id(self.rqdata.taskData.getfn_id(dep), idependtask.replace("_setscene", ""))
  1231. if taskid is None:
  1232. bb.msg.fatal("RunQueue", "Task %s:%s depends upon non-existent task %s:%s" % (self.rqdata.taskData.fn_index[self.rqdata.runq_fnid[realid]], self.rqdata.taskData.tasks_name[realid], dep, idependtask))
  1233. sq_revdeps_squash[self.rqdata.runq_setscene.index(task)].add(self.rqdata.runq_setscene.index(taskid))
  1234. # Have to zero this to avoid circular dependencies
  1235. sq_revdeps_squash[self.rqdata.runq_setscene.index(taskid)] = set()
  1236. #for task in xrange(len(sq_revdeps_squash)):
  1237. # print "Task %s: %s.%s is %s " % (task, self.rqdata.taskData.fn_index[self.rqdata.runq_fnid[self.rqdata.runq_setscene[task]]], self.rqdata.runq_task[self.rqdata.runq_setscene[task]] + "_setscene", sq_revdeps_squash[task])
  1238. self.sq_deps = []
  1239. self.sq_revdeps = sq_revdeps_squash
  1240. self.sq_revdeps2 = copy.deepcopy(self.sq_revdeps)
  1241. for task in xrange(len(self.sq_revdeps)):
  1242. self.sq_deps.append(set())
  1243. for task in xrange(len(self.sq_revdeps)):
  1244. for dep in self.sq_revdeps[task]:
  1245. self.sq_deps[dep].add(task)
  1246. for task in xrange(len(self.sq_revdeps)):
  1247. if len(self.sq_revdeps[task]) == 0:
  1248. self.runq_buildable[task] = 1
  1249. if self.rq.hashvalidate:
  1250. sq_hash = []
  1251. sq_hashfn = []
  1252. sq_fn = []
  1253. sq_taskname = []
  1254. sq_task = []
  1255. noexec = []
  1256. stamppresent = []
  1257. for task in xrange(len(self.sq_revdeps)):
  1258. realtask = self.rqdata.runq_setscene[task]
  1259. fn = self.rqdata.taskData.fn_index[self.rqdata.runq_fnid[realtask]]
  1260. taskname = self.rqdata.runq_task[realtask]
  1261. taskdep = self.rqdata.dataCache.task_deps[fn]
  1262. if 'noexec' in taskdep and taskname in taskdep['noexec']:
  1263. noexec.append(task)
  1264. self.task_skip(task)
  1265. bb.build.make_stamp(taskname + "_setscene", self.rqdata.dataCache, fn)
  1266. continue
  1267. if self.rq.check_stamp_task(realtask, taskname + "_setscene", cache=self.stampcache):
  1268. logger.debug(2, 'Setscene stamp current for task %s(%s)', task, self.rqdata.get_user_idstring(realtask))
  1269. stamppresent.append(task)
  1270. self.task_skip(task)
  1271. continue
  1272. if self.rq.check_stamp_task(realtask, taskname, recurse = True, cache=self.stampcache):
  1273. logger.debug(2, 'Normal stamp current for task %s(%s)', task, self.rqdata.get_user_idstring(realtask))
  1274. stamppresent.append(task)
  1275. self.task_skip(task)
  1276. continue
  1277. sq_fn.append(fn)
  1278. sq_hashfn.append(self.rqdata.dataCache.hashfn[fn])
  1279. sq_hash.append(self.rqdata.runq_hash[realtask])
  1280. sq_taskname.append(taskname)
  1281. sq_task.append(task)
  1282. call = self.rq.hashvalidate + "(sq_fn, sq_task, sq_hash, sq_hashfn, d)"
  1283. locs = { "sq_fn" : sq_fn, "sq_task" : sq_taskname, "sq_hash" : sq_hash, "sq_hashfn" : sq_hashfn, "d" : self.cooker.data }
  1284. valid = bb.utils.better_eval(call, locs)
  1285. valid_new = stamppresent
  1286. for v in valid:
  1287. valid_new.append(sq_task[v])
  1288. for task in xrange(len(self.sq_revdeps)):
  1289. if task not in valid_new and task not in noexec:
  1290. realtask = self.rqdata.runq_setscene[task]
  1291. logger.debug(2, 'No package found, so skipping setscene task %s',
  1292. self.rqdata.get_user_idstring(realtask))
  1293. self.task_failoutright(task)
  1294. logger.info('Executing SetScene Tasks')
  1295. self.rq.state = runQueueSceneRun
  1296. def scenequeue_updatecounters(self, task):
  1297. for dep in self.sq_deps[task]:
  1298. self.sq_revdeps2[dep].remove(task)
  1299. if len(self.sq_revdeps2[dep]) == 0:
  1300. self.runq_buildable[dep] = 1
  1301. def task_completeoutright(self, task):
  1302. """
  1303. Mark a task as completed
  1304. Look at the reverse dependencies and mark any task with
  1305. completed dependencies as buildable
  1306. """
  1307. index = self.rqdata.runq_setscene[task]
  1308. logger.debug(1, 'Found task %s which could be accelerated',
  1309. self.rqdata.get_user_idstring(index))
  1310. self.scenequeue_covered.add(task)
  1311. self.scenequeue_updatecounters(task)
  1312. def task_complete(self, task):
  1313. self.stats.taskCompleted()
  1314. self.task_completeoutright(task)
  1315. def task_fail(self, task, result):
  1316. self.stats.taskFailed()
  1317. bb.event.fire(sceneQueueTaskFailed(task, self.stats, result, self), self.cfgData)
  1318. self.scenequeue_notcovered.add(task)
  1319. self.scenequeue_updatecounters(task)
  1320. def task_failoutright(self, task):
  1321. self.runq_running[task] = 1
  1322. self.runq_buildable[task] = 1
  1323. self.stats.taskCompleted()
  1324. self.stats.taskSkipped()
  1325. index = self.rqdata.runq_setscene[task]
  1326. self.scenequeue_notcovered.add(task)
  1327. self.scenequeue_updatecounters(task)
  1328. def task_skip(self, task):
  1329. self.runq_running[task] = 1
  1330. self.runq_buildable[task] = 1
  1331. self.task_completeoutright(task)
  1332. self.stats.taskCompleted()
  1333. self.stats.taskSkipped()
  1334. def execute(self):
  1335. """
  1336. Run the tasks in a queue prepared by prepare_runqueue
  1337. """
  1338. self.workerpipe.read()
  1339. task = None
  1340. if self.stats.active < self.number_tasks:
  1341. # Find the next setscene to run
  1342. for nexttask in xrange(self.stats.total):
  1343. if self.runq_buildable[nexttask] == 1 and self.runq_running[nexttask] != 1:
  1344. if nexttask in self.unskippable:
  1345. logger.debug(2, "Setscene task %s is unskippable" % self.rqdata.get_user_idstring(self.rqdata.runq_setscene[nexttask]))
  1346. if nexttask not in self.unskippable and len(self.sq_revdeps[nexttask]) > 0 and self.sq_revdeps[nexttask].issubset(self.scenequeue_covered) and self.check_dependencies(nexttask, self.sq_revdeps[nexttask], True):
  1347. logger.debug(2, "Skipping setscene for task %s" % self.rqdata.get_user_idstring(self.rqdata.runq_setscene[nexttask]))
  1348. self.task_skip(nexttask)
  1349. self.scenequeue_notneeded.add(nexttask)
  1350. return True
  1351. task = nexttask
  1352. break
  1353. if task is not None:
  1354. realtask = self.rqdata.runq_setscene[task]
  1355. fn = self.rqdata.taskData.fn_index[self.rqdata.runq_fnid[realtask]]
  1356. taskname = self.rqdata.runq_task[realtask] + "_setscene"
  1357. if self.rq.check_stamp_task(realtask, self.rqdata.runq_task[realtask], recurse = True, cache=self.stampcache):
  1358. logger.debug(2, 'Stamp for underlying task %s(%s) is current, so skipping setscene variant',
  1359. task, self.rqdata.get_user_idstring(realtask))
  1360. self.task_failoutright(task)
  1361. return True
  1362. if self.cooker.configuration.force:
  1363. for target in self.rqdata.target_pairs:
  1364. if target[0] == fn and target[1] == self.rqdata.runq_task[realtask]:
  1365. self.task_failoutright(task)
  1366. return True
  1367. if self.rq.check_stamp_task(realtask, taskname, cache=self.stampcache):
  1368. logger.debug(2, 'Setscene stamp current task %s(%s), so skip it and its dependencies',
  1369. task, self.rqdata.get_user_idstring(realtask))
  1370. self.task_skip(task)
  1371. return True
  1372. startevent = sceneQueueTaskStarted(task, self.stats, self.rq)
  1373. bb.event.fire(startevent, self.cfgData)
  1374. self.worker.stdin.write("<runtask>" + pickle.dumps((fn, realtask, taskname, True, self.cooker.collection.get_file_appends(fn))) + "</runtask>")
  1375. self.worker.stdin.flush()
  1376. self.runq_running[task] = 1
  1377. self.stats.taskActive()
  1378. if self.stats.active < self.number_tasks:
  1379. return True
  1380. if self.stats.active > 0:
  1381. self.workerpipe.read()
  1382. return 0.5
  1383. # Convert scenequeue_covered task numbers into full taskgraph ids
  1384. oldcovered = self.scenequeue_covered
  1385. self.rq.scenequeue_covered = set()
  1386. for task in oldcovered:
  1387. self.rq.scenequeue_covered.add(self.rqdata.runq_setscene[task])
  1388. self.rq.scenequeue_notcovered = set()
  1389. for task in self.scenequeue_notcovered:
  1390. self.rq.scenequeue_notcovered.add(self.rqdata.runq_setscene[task])
  1391. logger.debug(1, 'We can skip tasks %s', sorted(self.rq.scenequeue_covered))
  1392. self.rq.state = runQueueRunInit
  1393. self.teardown()
  1394. return True
  1395. def runqueue_process_waitpid(self, task, status):
  1396. task = self.rq.rqdata.runq_setscene.index(task)
  1397. RunQueueExecute.runqueue_process_waitpid(self, task, status)
  1398. class TaskFailure(Exception):
  1399. """
  1400. Exception raised when a task in a runqueue fails
  1401. """
  1402. def __init__(self, x):
  1403. self.args = x
  1404. class runQueueExitWait(bb.event.Event):
  1405. """
  1406. Event when waiting for task processes to exit
  1407. """
  1408. def __init__(self, remain):
  1409. self.remain = remain
  1410. self.message = "Waiting for %s active tasks to finish" % remain
  1411. bb.event.Event.__init__(self)
  1412. class runQueueEvent(bb.event.Event):
  1413. """
  1414. Base runQueue event class
  1415. """
  1416. def __init__(self, task, stats, rq):
  1417. self.taskid = task
  1418. self.taskstring = rq.rqdata.get_user_idstring(task)
  1419. self.stats = stats.copy()
  1420. bb.event.Event.__init__(self)
  1421. class sceneQueueEvent(runQueueEvent):
  1422. """
  1423. Base sceneQueue event class
  1424. """
  1425. def __init__(self, task, stats, rq, noexec=False):
  1426. runQueueEvent.__init__(self, task, stats, rq)
  1427. realtask = rq.rqdata.runq_setscene[task]
  1428. self.taskstring = rq.rqdata.get_user_idstring(realtask, "_setscene")
  1429. class runQueueTaskStarted(runQueueEvent):
  1430. """
  1431. Event notifing a task was started
  1432. """
  1433. def __init__(self, task, stats, rq, noexec=False):
  1434. runQueueEvent.__init__(self, task, stats, rq)
  1435. self.noexec = noexec
  1436. class sceneQueueTaskStarted(sceneQueueEvent):
  1437. """
  1438. Event notifing a setscene task was started
  1439. """
  1440. def __init__(self, task, stats, rq, noexec=False):
  1441. sceneQueueEvent.__init__(self, task, stats, rq)
  1442. self.noexec = noexec
  1443. class runQueueTaskFailed(runQueueEvent):
  1444. """
  1445. Event notifing a task failed
  1446. """
  1447. def __init__(self, task, stats, exitcode, rq):
  1448. runQueueEvent.__init__(self, task, stats, rq)
  1449. self.exitcode = exitcode
  1450. class sceneQueueTaskFailed(sceneQueueEvent):
  1451. """
  1452. Event notifing a setscene task failed
  1453. """
  1454. def __init__(self, task, stats, exitcode, rq):
  1455. sceneQueueEvent.__init__(self, task, stats, rq)
  1456. self.exitcode = exitcode
  1457. class runQueueTaskCompleted(runQueueEvent):
  1458. """
  1459. Event notifing a task completed
  1460. """
  1461. class runQueuePipe():
  1462. """
  1463. Abstraction for a pipe between a worker thread and the server
  1464. """
  1465. def __init__(self, pipein, pipeout, d, rq):
  1466. self.input = pipein
  1467. if pipeout:
  1468. pipeout.close()
  1469. bb.utils.nonblockingfd(self.input)
  1470. self.queue = ""
  1471. self.d = d
  1472. self.rq = rq
  1473. def setrunqueue(self, rq):
  1474. self.rq = rq
  1475. def read(self):
  1476. start = len(self.queue)
  1477. try:
  1478. self.queue = self.queue + self.input.read(102400)
  1479. except (OSError, IOError) as e:
  1480. if e.errno != errno.EAGAIN:
  1481. raise
  1482. end = len(self.queue)
  1483. found = True
  1484. while found and len(self.queue):
  1485. found = False
  1486. index = self.queue.find("</event>")
  1487. while index != -1 and self.queue.startswith("<event>"):
  1488. event = pickle.loads(self.queue[7:index])
  1489. bb.event.fire_from_worker(event, self.d)
  1490. found = True
  1491. self.queue = self.queue[index+8:]
  1492. index = self.queue.find("</event>")
  1493. index = self.queue.find("</exitcode>")
  1494. while index != -1 and self.queue.startswith("<exitcode>"):
  1495. task, status = pickle.loads(self.queue[10:index])
  1496. self.rq.runqueue_process_waitpid(task, status)
  1497. found = True
  1498. self.queue = self.queue[index+11:]
  1499. index = self.queue.find("</exitcode>")
  1500. return (end > start)
  1501. def close(self):
  1502. while self.read():
  1503. continue
  1504. if len(self.queue) > 0:
  1505. print("Warning, worker left partial message: %s" % self.queue)
  1506. self.input.close()