diff options
author | Richard Purdie <rpurdie@linux.intel.com> | 2010-01-20 18:46:02 +0000 |
---|---|---|
committer | Richard Purdie <rpurdie@linux.intel.com> | 2010-01-20 18:46:02 +0000 |
commit | 22c29d8651668195f72e2f6a8e059d625eb511c3 (patch) | |
tree | dd1dd43f0ec47a9964c8a766eb8b3ad75cf51a64 /bitbake/lib/bb/ui/ncurses.py | |
parent | 1bfd6edef9db9c9175058ae801d1b601e4f15263 (diff) | |
download | poky-22c29d8651668195f72e2f6a8e059d625eb511c3.tar.gz |
bitbake: Switch to bitbake-dev version (bitbake master upstream)
Signed-off-by: Richard Purdie <rpurdie@linux.intel.com>
Diffstat (limited to 'bitbake/lib/bb/ui/ncurses.py')
-rw-r--r-- | bitbake/lib/bb/ui/ncurses.py | 335 |
1 files changed, 335 insertions, 0 deletions
diff --git a/bitbake/lib/bb/ui/ncurses.py b/bitbake/lib/bb/ui/ncurses.py new file mode 100644 index 0000000000..14310dc124 --- /dev/null +++ b/bitbake/lib/bb/ui/ncurses.py | |||
@@ -0,0 +1,335 @@ | |||
1 | # | ||
2 | # BitBake Curses UI Implementation | ||
3 | # | ||
4 | # Implements an ncurses frontend for the BitBake utility. | ||
5 | # | ||
6 | # Copyright (C) 2006 Michael 'Mickey' Lauer | ||
7 | # Copyright (C) 2006-2007 Richard Purdie | ||
8 | # | ||
9 | # This program is free software; you can redistribute it and/or modify | ||
10 | # it under the terms of the GNU General Public License version 2 as | ||
11 | # published by the Free Software Foundation. | ||
12 | # | ||
13 | # This program is distributed in the hope that it will be useful, | ||
14 | # but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
15 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||
16 | # GNU General Public License for more details. | ||
17 | # | ||
18 | # You should have received a copy of the GNU General Public License along | ||
19 | # with this program; if not, write to the Free Software Foundation, Inc., | ||
20 | # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. | ||
21 | |||
22 | """ | ||
23 | We have the following windows: | ||
24 | |||
25 | 1.) Main Window: Shows what we are ultimately building and how far we are. Includes status bar | ||
26 | 2.) Thread Activity Window: Shows one status line for every concurrent bitbake thread. | ||
27 | 3.) Command Line Window: Contains an interactive command line where you can interact w/ Bitbake. | ||
28 | |||
29 | Basic window layout is like that: | ||
30 | |||
31 | |---------------------------------------------------------| | ||
32 | | <Main Window> | <Thread Activity Window> | | ||
33 | | | 0: foo do_compile complete| | ||
34 | | Building Gtk+-2.6.10 | 1: bar do_patch complete | | ||
35 | | Status: 60% | ... | | ||
36 | | | ... | | ||
37 | | | ... | | ||
38 | |---------------------------------------------------------| | ||
39 | |<Command Line Window> | | ||
40 | |>>> which virtual/kernel | | ||
41 | |openzaurus-kernel | | ||
42 | |>>> _ | | ||
43 | |---------------------------------------------------------| | ||
44 | |||
45 | """ | ||
46 | |||
47 | import os, sys, curses, itertools, time | ||
48 | import bb | ||
49 | import xmlrpclib | ||
50 | from bb import ui | ||
51 | from bb.ui import uihelper | ||
52 | |||
53 | parsespin = itertools.cycle( r'|/-\\' ) | ||
54 | |||
55 | X = 0 | ||
56 | Y = 1 | ||
57 | WIDTH = 2 | ||
58 | HEIGHT = 3 | ||
59 | |||
60 | MAXSTATUSLENGTH = 32 | ||
61 | |||
62 | class NCursesUI: | ||
63 | """ | ||
64 | NCurses UI Class | ||
65 | """ | ||
66 | class Window: | ||
67 | """Base Window Class""" | ||
68 | def __init__( self, x, y, width, height, fg=curses.COLOR_BLACK, bg=curses.COLOR_WHITE ): | ||
69 | self.win = curses.newwin( height, width, y, x ) | ||
70 | self.dimensions = ( x, y, width, height ) | ||
71 | """ | ||
72 | if curses.has_colors(): | ||
73 | color = 1 | ||
74 | curses.init_pair( color, fg, bg ) | ||
75 | self.win.bkgdset( ord(' '), curses.color_pair(color) ) | ||
76 | else: | ||
77 | self.win.bkgdset( ord(' '), curses.A_BOLD ) | ||
78 | """ | ||
79 | self.erase() | ||
80 | self.setScrolling() | ||
81 | self.win.noutrefresh() | ||
82 | |||
83 | def erase( self ): | ||
84 | self.win.erase() | ||
85 | |||
86 | def setScrolling( self, b = True ): | ||
87 | self.win.scrollok( b ) | ||
88 | self.win.idlok( b ) | ||
89 | |||
90 | def setBoxed( self ): | ||
91 | self.boxed = True | ||
92 | self.win.box() | ||
93 | self.win.noutrefresh() | ||
94 | |||
95 | def setText( self, x, y, text, *args ): | ||
96 | self.win.addstr( y, x, text, *args ) | ||
97 | self.win.noutrefresh() | ||
98 | |||
99 | def appendText( self, text, *args ): | ||
100 | self.win.addstr( text, *args ) | ||
101 | self.win.noutrefresh() | ||
102 | |||
103 | def drawHline( self, y ): | ||
104 | self.win.hline( y, 0, curses.ACS_HLINE, self.dimensions[WIDTH] ) | ||
105 | self.win.noutrefresh() | ||
106 | |||
107 | class DecoratedWindow( Window ): | ||
108 | """Base class for windows with a box and a title bar""" | ||
109 | def __init__( self, title, x, y, width, height, fg=curses.COLOR_BLACK, bg=curses.COLOR_WHITE ): | ||
110 | NCursesUI.Window.__init__( self, x+1, y+3, width-2, height-4, fg, bg ) | ||
111 | self.decoration = NCursesUI.Window( x, y, width, height, fg, bg ) | ||
112 | self.decoration.setBoxed() | ||
113 | self.decoration.win.hline( 2, 1, curses.ACS_HLINE, width-2 ) | ||
114 | self.setTitle( title ) | ||
115 | |||
116 | def setTitle( self, title ): | ||
117 | self.decoration.setText( 1, 1, title.center( self.dimensions[WIDTH]-2 ), curses.A_BOLD ) | ||
118 | |||
119 | #-------------------------------------------------------------------------# | ||
120 | # class TitleWindow( Window ): | ||
121 | #-------------------------------------------------------------------------# | ||
122 | # """Title Window""" | ||
123 | # def __init__( self, x, y, width, height ): | ||
124 | # NCursesUI.Window.__init__( self, x, y, width, height ) | ||
125 | # version = bb.__version__ | ||
126 | # title = "BitBake %s" % version | ||
127 | # credit = "(C) 2003-2007 Team BitBake" | ||
128 | # #self.win.hline( 2, 1, curses.ACS_HLINE, width-2 ) | ||
129 | # self.win.border() | ||
130 | # self.setText( 1, 1, title.center( self.dimensions[WIDTH]-2 ), curses.A_BOLD ) | ||
131 | # self.setText( 1, 2, credit.center( self.dimensions[WIDTH]-2 ), curses.A_BOLD ) | ||
132 | |||
133 | #-------------------------------------------------------------------------# | ||
134 | class ThreadActivityWindow( DecoratedWindow ): | ||
135 | #-------------------------------------------------------------------------# | ||
136 | """Thread Activity Window""" | ||
137 | def __init__( self, x, y, width, height ): | ||
138 | NCursesUI.DecoratedWindow.__init__( self, "Thread Activity", x, y, width, height ) | ||
139 | |||
140 | def setStatus( self, thread, text ): | ||
141 | line = "%02d: %s" % ( thread, text ) | ||
142 | width = self.dimensions[WIDTH] | ||
143 | if ( len(line) > width ): | ||
144 | line = line[:width-3] + "..." | ||
145 | else: | ||
146 | line = line.ljust( width ) | ||
147 | self.setText( 0, thread, line ) | ||
148 | |||
149 | #-------------------------------------------------------------------------# | ||
150 | class MainWindow( DecoratedWindow ): | ||
151 | #-------------------------------------------------------------------------# | ||
152 | """Main Window""" | ||
153 | def __init__( self, x, y, width, height ): | ||
154 | self.StatusPosition = width - MAXSTATUSLENGTH | ||
155 | NCursesUI.DecoratedWindow.__init__( self, None, x, y, width, height ) | ||
156 | curses.nl() | ||
157 | |||
158 | def setTitle( self, title ): | ||
159 | title = "BitBake %s" % bb.__version__ | ||
160 | self.decoration.setText( 2, 1, title, curses.A_BOLD ) | ||
161 | self.decoration.setText( self.StatusPosition - 8, 1, "Status:", curses.A_BOLD ) | ||
162 | |||
163 | def setStatus(self, status): | ||
164 | while len(status) < MAXSTATUSLENGTH: | ||
165 | status = status + " " | ||
166 | self.decoration.setText( self.StatusPosition, 1, status, curses.A_BOLD ) | ||
167 | |||
168 | |||
169 | #-------------------------------------------------------------------------# | ||
170 | class ShellOutputWindow( DecoratedWindow ): | ||
171 | #-------------------------------------------------------------------------# | ||
172 | """Interactive Command Line Output""" | ||
173 | def __init__( self, x, y, width, height ): | ||
174 | NCursesUI.DecoratedWindow.__init__( self, "Command Line Window", x, y, width, height ) | ||
175 | |||
176 | #-------------------------------------------------------------------------# | ||
177 | class ShellInputWindow( Window ): | ||
178 | #-------------------------------------------------------------------------# | ||
179 | """Interactive Command Line Input""" | ||
180 | def __init__( self, x, y, width, height ): | ||
181 | NCursesUI.Window.__init__( self, x, y, width, height ) | ||
182 | |||
183 | # put that to the top again from curses.textpad import Textbox | ||
184 | # self.textbox = Textbox( self.win ) | ||
185 | # t = threading.Thread() | ||
186 | # t.run = self.textbox.edit | ||
187 | # t.start() | ||
188 | |||
189 | #-------------------------------------------------------------------------# | ||
190 | def main(self, stdscr, server, eventHandler): | ||
191 | #-------------------------------------------------------------------------# | ||
192 | height, width = stdscr.getmaxyx() | ||
193 | |||
194 | # for now split it like that: | ||
195 | # MAIN_y + THREAD_y = 2/3 screen at the top | ||
196 | # MAIN_x = 2/3 left, THREAD_y = 1/3 right | ||
197 | # CLI_y = 1/3 of screen at the bottom | ||
198 | # CLI_x = full | ||
199 | |||
200 | main_left = 0 | ||
201 | main_top = 0 | ||
202 | main_height = ( height / 3 * 2 ) | ||
203 | main_width = ( width / 3 ) * 2 | ||
204 | clo_left = main_left | ||
205 | clo_top = main_top + main_height | ||
206 | clo_height = height - main_height - main_top - 1 | ||
207 | clo_width = width | ||
208 | cli_left = main_left | ||
209 | cli_top = clo_top + clo_height | ||
210 | cli_height = 1 | ||
211 | cli_width = width | ||
212 | thread_left = main_left + main_width | ||
213 | thread_top = main_top | ||
214 | thread_height = main_height | ||
215 | thread_width = width - main_width | ||
216 | |||
217 | #tw = self.TitleWindow( 0, 0, width, main_top ) | ||
218 | mw = self.MainWindow( main_left, main_top, main_width, main_height ) | ||
219 | taw = self.ThreadActivityWindow( thread_left, thread_top, thread_width, thread_height ) | ||
220 | clo = self.ShellOutputWindow( clo_left, clo_top, clo_width, clo_height ) | ||
221 | cli = self.ShellInputWindow( cli_left, cli_top, cli_width, cli_height ) | ||
222 | cli.setText( 0, 0, "BB>" ) | ||
223 | |||
224 | mw.setStatus("Idle") | ||
225 | |||
226 | helper = uihelper.BBUIHelper() | ||
227 | shutdown = 0 | ||
228 | |||
229 | try: | ||
230 | cmdline = server.runCommand(["getCmdLineAction"]) | ||
231 | if not cmdline: | ||
232 | return | ||
233 | ret = server.runCommand(cmdline) | ||
234 | if ret != True: | ||
235 | print "Couldn't get default commandlind! %s" % ret | ||
236 | return | ||
237 | except xmlrpclib.Fault, x: | ||
238 | print "XMLRPC Fault getting commandline:\n %s" % x | ||
239 | return | ||
240 | |||
241 | exitflag = False | ||
242 | while not exitflag: | ||
243 | try: | ||
244 | event = eventHandler.waitEvent(0.25) | ||
245 | if not event: | ||
246 | continue | ||
247 | helper.eventHandler(event) | ||
248 | #mw.appendText("%s\n" % event[0]) | ||
249 | if isinstance(event, bb.build.Task): | ||
250 | mw.appendText("NOTE: %s\n" % event._message) | ||
251 | if isinstance(event, bb.msg.MsgDebug): | ||
252 | mw.appendText('DEBUG: ' + event._message + '\n') | ||
253 | if isinstance(event, bb.msg.MsgNote): | ||
254 | mw.appendText('NOTE: ' + event._message + '\n') | ||
255 | if isinstance(event, bb.msg.MsgWarn): | ||
256 | mw.appendText('WARNING: ' + event._message + '\n') | ||
257 | if isinstance(event, bb.msg.MsgError): | ||
258 | mw.appendText('ERROR: ' + event._message + '\n') | ||
259 | if isinstance(event, bb.msg.MsgFatal): | ||
260 | mw.appendText('FATAL: ' + event._message + '\n') | ||
261 | if isinstance(event, bb.event.ParseProgress): | ||
262 | x = event.sofar | ||
263 | y = event.total | ||
264 | if x == y: | ||
265 | mw.setStatus("Idle") | ||
266 | mw.appendText("Parsing finished. %d cached, %d parsed, %d skipped, %d masked." | ||
267 | % ( event.cached, event.parsed, event.skipped, event.masked )) | ||
268 | else: | ||
269 | mw.setStatus("Parsing: %s (%04d/%04d) [%2d %%]" % ( parsespin.next(), x, y, x*100/y ) ) | ||
270 | # if isinstance(event, bb.build.TaskFailed): | ||
271 | # if event.logfile: | ||
272 | # if data.getVar("BBINCLUDELOGS", d): | ||
273 | # bb.msg.error(bb.msg.domain.Build, "log data follows (%s)" % logfile) | ||
274 | # number_of_lines = data.getVar("BBINCLUDELOGS_LINES", d) | ||
275 | # if number_of_lines: | ||
276 | # os.system('tail -n%s %s' % (number_of_lines, logfile)) | ||
277 | # else: | ||
278 | # f = open(logfile, "r") | ||
279 | # while True: | ||
280 | # l = f.readline() | ||
281 | # if l == '': | ||
282 | # break | ||
283 | # l = l.rstrip() | ||
284 | # print '| %s' % l | ||
285 | # f.close() | ||
286 | # else: | ||
287 | # bb.msg.error(bb.msg.domain.Build, "see log in %s" % logfile) | ||
288 | |||
289 | if isinstance(event, bb.command.CookerCommandCompleted): | ||
290 | exitflag = True | ||
291 | if isinstance(event, bb.command.CookerCommandFailed): | ||
292 | mw.appendText("Command execution failed: %s" % event.error) | ||
293 | time.sleep(2) | ||
294 | exitflag = True | ||
295 | if isinstance(event, bb.cooker.CookerExit): | ||
296 | exitflag = True | ||
297 | |||
298 | if helper.needUpdate: | ||
299 | activetasks, failedtasks = helper.getTasks() | ||
300 | taw.erase() | ||
301 | taw.setText(0, 0, "") | ||
302 | if activetasks: | ||
303 | taw.appendText("Active Tasks:\n") | ||
304 | for task in activetasks: | ||
305 | taw.appendText(task) | ||
306 | if failedtasks: | ||
307 | taw.appendText("Failed Tasks:\n") | ||
308 | for task in failedtasks: | ||
309 | taw.appendText(task) | ||
310 | |||
311 | curses.doupdate() | ||
312 | except KeyboardInterrupt: | ||
313 | if shutdown == 2: | ||
314 | mw.appendText("Third Keyboard Interrupt, exit.\n") | ||
315 | exitflag = True | ||
316 | if shutdown == 1: | ||
317 | mw.appendText("Second Keyboard Interrupt, stopping...\n") | ||
318 | server.runCommand(["stateStop"]) | ||
319 | if shutdown == 0: | ||
320 | mw.appendText("Keyboard Interrupt, closing down...\n") | ||
321 | server.runCommand(["stateShutdown"]) | ||
322 | shutdown = shutdown + 1 | ||
323 | pass | ||
324 | |||
325 | def init(server, eventHandler): | ||
326 | if not os.isatty(sys.stdout.fileno()): | ||
327 | print "FATAL: Unable to run 'ncurses' UI without a TTY." | ||
328 | return | ||
329 | ui = NCursesUI() | ||
330 | try: | ||
331 | curses.wrapper(ui.main, server, eventHandler) | ||
332 | except: | ||
333 | import traceback | ||
334 | traceback.print_exc() | ||
335 | |||