source: src/mlx/gui/gui.py@ 77:cc8b178b8102

Last change on this file since 77:cc8b178b8102 was 77:cc8b178b8102, checked in by István Váradi <ivaradi@…>, 12 years ago

The monitoring window is now separate window
File size: 15.9 KB
RevLine 
[28]1# The main file for the GUI
2
[29]3from statusicon import StatusIcon
[32]4from statusbar import Statusbar
[36]5from update import Updater
[29]6from mlx.gui.common import *
[42]7from mlx.gui.flight import Wizard
[77]8from mlx.gui.monitor import MonitorWindow
[28]9
10import mlx.const as const
11import mlx.fs as fs
12import mlx.flight as flight
13import mlx.logger as logger
14import mlx.acft as acft
[41]15import mlx.web as web
[28]16
17import time
[38]18import threading
19import sys
[28]20
[77]21#------------------------------------------------------------------------------
22
[28]23acftTypes = [ ("Boeing 737-600", const.AIRCRAFT_B736),
24 ("Boeing 737-700", const.AIRCRAFT_B737),
25 ("Boeing 737-800", const.AIRCRAFT_B738),
26 ("Bombardier Dash 8-Q400", const.AIRCRAFT_DH8D),
27 ("Boeing 737-300", const.AIRCRAFT_B733),
28 ("Boeing 737-400", const.AIRCRAFT_B734),
29 ("Boeing 737-500", const.AIRCRAFT_B735),
30 ("Boeing 767-200", const.AIRCRAFT_B762),
31 ("Boeing 767-300", const.AIRCRAFT_B763),
32 ("Bombardier CRJ200", const.AIRCRAFT_CRJ2),
33 ("Fokker 70", const.AIRCRAFT_F70),
34 ("Lisunov Li-2", const.AIRCRAFT_DC3),
35 ("Tupolev Tu-134", const.AIRCRAFT_T134),
36 ("Tupolev Tu-154", const.AIRCRAFT_T154),
37 ("Yakovlev Yak-40", const.AIRCRAFT_YK40) ]
38
[77]39#------------------------------------------------------------------------------
40
[28]41class GUI(fs.ConnectionListener):
42 """The main GUI class."""
[36]43 def __init__(self, programDirectory, config):
[28]44 """Construct the GUI."""
45 gobject.threads_init()
46
[36]47 self._programDirectory = programDirectory
[42]48 self.config = config
[28]49 self._connecting = False
[59]50 self._reconnecting = False
[28]51 self._connected = False
52 self._logger = logger.Logger(output = self)
53 self._flight = None
54 self._simulator = None
[59]55 self._monitoring = False
[38]56
57 self._stdioLock = threading.Lock()
58 self._stdioText = ""
[36]59 self._stdioAfterNewLine = True
[28]60
[41]61 self.webHandler = web.Handler()
62 self.webHandler.start()
63
[38]64 self.toRestart = False
65
[29]66 def build(self, iconDirectory):
[28]67 """Build the GUI."""
[29]68
[36]69 window = gtk.Window()
70 window.set_title("MAVA Logger X " + const.VERSION)
71 window.set_icon_from_file(os.path.join(iconDirectory, "logo.ico"))
72 window.connect("delete-event",
73 lambda a, b: self.hideMainWindow())
74 window.connect("window-state-event", self._handleMainWindowState)
[28]75
76 mainVBox = gtk.VBox()
[36]77 window.add(mainVBox)
[28]78
[42]79 notebook = gtk.Notebook()
80 mainVBox.add(notebook)
81
[46]82 self._wizard = Wizard(self)
83 label = gtk.Label("_Flight")
84 label.set_use_underline(True)
85 label.set_tooltip_text("Flight wizard")
86 notebook.append_page(self._wizard, label)
[42]87
88 logVBox = gtk.VBox()
[46]89 label = gtk.Label("_Log")
90 label.set_use_underline(True)
91 label.set_tooltip_text("Flight log")
92 notebook.append_page(logVBox, label)
[42]93
[28]94 logFrame = self._buildLogFrame()
95 logFrame.set_border_width(8)
[42]96 logVBox.pack_start(logFrame, True, True, 0)
[32]97
98 mainVBox.pack_start(gtk.HSeparator(), False, False, 0)
99
100 self._statusbar = Statusbar()
101 mainVBox.pack_start(self._statusbar, False, False, 0)
102
[46]103 notebook.connect("switch-page", self._notebookPageSwitch)
104
[77]105 self._monitorWindow = MonitorWindow(self, iconDirectory)
106 self._monitorWindowX = None
107 self._monitorWindowY = None
108
[46]109 window.show_all()
110 self._wizard.grabDefault()
[28]111
[36]112 self._mainWindow = window
[29]113
114 self._statusIcon = StatusIcon(iconDirectory, self)
[28]115
[49]116 self._busyCursor = gdk.Cursor(gdk.CursorType.WATCH if pygobject
117 else gdk.WATCH)
118
[59]119 @property
120 def simulator(self):
121 """Get the simulator used by us."""
122 return self._simulator
123
[71]124 @property
125 def flight(self):
126 """Get the flight being performed."""
127 return self._flight
128
[28]129 def run(self):
130 """Run the GUI."""
[42]131 if self.config.autoUpdate:
[38]132 self._updater = Updater(self,
133 self._programDirectory,
[42]134 self.config.updateURL,
[36]135 self._mainWindow)
136 self._updater.start()
137
[28]138 gtk.main()
[36]139
[28]140 if self._flight is not None:
141 simulator = self._flight.simulator
[59]142 if self._monitoring:
143 simulator.stopMonitoring()
144 self._monitoring = False
145 simulator.disconnect()
[28]146
147 def connected(self, fsType, descriptor):
148 """Called when we have connected to the simulator."""
149 self._connected = True
150 self._logger.untimedMessage("Connected to the simulator %s" % (descriptor,))
[59]151 gobject.idle_add(self._handleConnected, fsType, descriptor)
152
153 def _handleConnected(self, fsType, descriptor):
154 """Called when the connection to the simulator has succeeded."""
155 self._statusbar.updateConnection(self._connecting, self._connected)
156 self.endBusy()
157 if not self._reconnecting:
158 self._wizard.connected(fsType, descriptor)
159 self._reconnecting = False
160
161 def connectionFailed(self):
162 """Called when the connection failed."""
163 self._logger.untimedMessage("Connection to the simulator failed")
164 gobject.idle_add(self._connectionFailed)
165
166 def _connectionFailed(self):
167 """Called when the connection failed."""
168 self.endBusy()
169 self._statusbar.updateConnection(self._connecting, self._connected)
170
171 dialog = gtk.MessageDialog(type = MESSAGETYPE_ERROR,
172 message_format =
173 "Cannot connect to the simulator.",
174 parent = self._mainWindow)
175 dialog.format_secondary_markup("Rectify the situation, and press <b>Try again</b> "
176 "to try the connection again, "
177 "or <b>Cancel</b> to cancel the flight.")
178
179 dialog.add_button("_Cancel", 0)
180 dialog.add_button("_Try again", 1)
181 dialog.set_default_response(1)
182
183 result = dialog.run()
184 dialog.hide()
185 if result == 1:
186 self.beginBusy("Connecting to the simulator.")
187 self._simulator.reconnect()
188 else:
189 self._connecting = False
190 self._reconnecting = False
191 self._statusbar.updateConnection(self._connecting, self._connected)
192 self._wizard.connectionFailed()
[28]193
194 def disconnected(self):
195 """Called when we have disconnected from the simulator."""
196 self._connected = False
197 self._logger.untimedMessage("Disconnected from the simulator")
[59]198
199 gobject.idle_add(self._disconnected)
200
201 def _disconnected(self):
202 """Called when we have disconnected from the simulator unexpectedly."""
203 self._statusbar.updateConnection(self._connecting, self._connected)
204
205 dialog = gtk.MessageDialog(type = MESSAGETYPE_ERROR,
206 message_format =
207 "The connection to the simulator failed unexpectedly.",
208 parent = self._mainWindow)
209 dialog.format_secondary_markup("If the simulator has crashed, restart it "
210 "and restore your flight as much as possible "
211 "to the state it was in before the crash.\n"
212 "Then press <b>Reconnect</b> to reconnect.\n\n"
213 "If you want to cancel the flight, press <b>Cancel</b>.")
214
215 dialog.add_button("_Cancel", 0)
216 dialog.add_button("_Reconnect", 1)
217 dialog.set_default_response(1)
218
219 result = dialog.run()
220 dialog.hide()
221 if result == 1:
222 self.beginBusy("Connecting to the simulator.")
223 self._reconnecting = True
224 self._simulator.reconnect()
225 else:
226 self._connecting = False
227 self._reconnecting = False
228 self._statusbar.updateConnection(self._connecting, self._connected)
229 self._wizard.disconnected()
[28]230
231 def write(self, msg):
232 """Write the given message to the log."""
233 gobject.idle_add(self._writeLog, msg)
234
235 def check(self, flight, aircraft, logger, oldState, state):
236 """Update the data."""
[77]237 gobject.idle_add(self._monitorWindow.setData, state)
[28]238
[31]239 def resetFlightStatus(self):
240 """Reset the status of the flight."""
[32]241 self._statusbar.resetFlightStatus()
[31]242 self._statusIcon.resetFlightStatus()
243
244 def setStage(self, stage):
245 """Set the stage of the flight."""
246 gobject.idle_add(self._setStage, stage)
247
248 def _setStage(self, stage):
249 """Set the stage of the flight."""
[32]250 self._statusbar.setStage(stage)
[31]251 self._statusIcon.setStage(stage)
252
253 def setRating(self, rating):
254 """Set the rating of the flight."""
255 gobject.idle_add(self._setRating, rating)
256
257 def _setRating(self, rating):
258 """Set the rating of the flight."""
[32]259 self._statusbar.setRating(rating)
[31]260 self._statusIcon.setRating(rating)
261
262 def setNoGo(self, reason):
263 """Set the rating of the flight to No-Go with the given reason."""
264 gobject.idle_add(self._setNoGo, reason)
265
266 def _setNoGo(self, reason):
267 """Set the rating of the flight."""
[32]268 self._statusbar.setNoGo(reason)
[31]269 self._statusIcon.setNoGo(reason)
270
[29]271 def _handleMainWindowState(self, window, event):
272 """Hande a change in the state of the window"""
273 iconified = gdk.WindowState.ICONIFIED if pygobject \
274 else gdk.WINDOW_STATE_ICONIFIED
275 if (event.changed_mask&iconified)!=0 and (event.new_window_state&iconified)!=0:
[35]276 self.hideMainWindow(savePosition = False)
[29]277
[35]278 def hideMainWindow(self, savePosition = True):
[29]279 """Hide the main window and save its position."""
[35]280 if savePosition:
281 (self._mainWindowX, self._mainWindowY) = \
282 self._mainWindow.get_window().get_root_origin()
283 else:
284 self._mainWindowX = self._mainWindowY = None
[29]285 self._mainWindow.hide()
286 self._statusIcon.mainWindowHidden()
287 return True
288
289 def showMainWindow(self):
290 """Show the main window at its former position."""
[35]291 if self._mainWindowX is not None and self._mainWindowY is not None:
292 self._mainWindow.move(self._mainWindowX, self._mainWindowY)
293
294 self._mainWindow.show()
[29]295 self._mainWindow.deiconify()
[35]296
[29]297 self._statusIcon.mainWindowShown()
298
299 def toggleMainWindow(self):
300 """Toggle the main window."""
301 if self._mainWindow.get_visible():
302 self.hideMainWindow()
303 else:
304 self.showMainWindow()
[36]305
[77]306 def hideMonitorWindow(self, savePosition = True):
307 """Hide the monitor window."""
308 if savePosition:
309 (self._monitorWindowX, self._monitorWindowY) = \
310 self._monitorWindow.get_window().get_root_origin()
311 else:
312 self._monitorWindowX = self._monitorWindowY = None
313 self._monitorWindow.hide()
314 self._statusIcon.monitorWindowHidden()
315 return True
316
317 def showMonitorWindow(self):
318 """Show the monitor window."""
319 if self._monitorWindowX is not None and self._monitorWindowY is not None:
320 self._monitorWindow.move(self._monitorWindowX, self._monitorWindowY)
321 self._monitorWindow.show_all()
322 self._statusIcon.monitorWindowShown()
323
[38]324 def restart(self):
325 """Quit and restart the application."""
326 self.toRestart = True
327 self._quit()
328
329 def flushStdIO(self):
330 """Flush any text to the standard error that could not be logged."""
331 if self._stdioText:
332 sys.__stderr__.write(self._stdioText)
333
[36]334 def writeStdIO(self, text):
335 """Write the given text into standard I/O log."""
[38]336 with self._stdioLock:
337 self._stdioText += text
338
339 gobject.idle_add(self._writeStdIO)
[36]340
[49]341 def beginBusy(self, message):
342 """Begin a period of background processing."""
343 self._mainWindow.get_window().set_cursor(self._busyCursor)
344 self._statusbar.updateBusyState(message)
345
346 def endBusy(self):
347 """End a period of background processing."""
348 self._mainWindow.get_window().set_cursor(None)
349 self._statusbar.updateBusyState(None)
350
[38]351 def _writeStdIO(self):
[36]352 """Perform the real writing."""
[38]353 with self._stdioLock:
354 text = self._stdioText
355 self._stdioText = ""
356 if not text: return
357
[36]358 lines = text.splitlines()
359 if text[-1]=="\n":
360 text = ""
361 else:
362 text = lines[-1]
363 lines = lines[:-1]
364
365 for line in lines:
366 if self._stdioAfterNewLine:
367 line = "[STDIO] " + line
368 self._writeLog(line + "\n")
369 self._stdioAfterNewLine = True
370
371 if text:
372 if self._stdioAfterNewLine:
373 text = "[STDIO] " + text
374 self._writeLog(text)
375 self._stdioAfterNewLine = False
[51]376
[59]377 def connectSimulator(self, aircraftType):
378 """Connect to the simulator for the first time."""
379 self._logger.reset()
380
381 self._flight = flight.Flight(self._logger, self)
382 self._flight.aircraftType = aircraftType
383 self._flight.aircraft = acft.Aircraft.create(self._flight)
384 self._flight.aircraft._checkers.append(self)
385
386 if self._simulator is None:
387 self._simulator = fs.createSimulator(const.SIM_MSFS9, self)
388
389 self._flight.simulator = self._simulator
390
391 self.beginBusy("Connecting to the simulator...")
392 self._statusbar.updateConnection(self._connecting, self._connected)
393
394 self._connecting = True
395 self._simulator.connect(self._flight.aircraft)
396
[70]397 def startMonitoring(self):
398 """Start monitoring."""
399 self._simulator.startMonitoring()
400 self._monitoring = True
401
402 def stopMonitoring(self):
403 """Stop monitoring."""
404 self._simulator.stoptMonitoring()
405 self._monitoring = False
406
[28]407 def _buildLogFrame(self):
408 """Build the frame for the log."""
409 logFrame = gtk.Frame(label = "Log")
410
411 frameAlignment = gtk.Alignment(xscale = 1.0, yscale = 1.0)
412
413 frameAlignment.set_padding(padding_top = 4, padding_bottom = 10,
414 padding_left = 16, padding_right = 16)
415
416 logFrame.add(frameAlignment)
417
418 logScroller = gtk.ScrolledWindow()
419 self._logView = gtk.TextView()
420 self._logView.set_editable(False)
421 logScroller.add(self._logView)
422
423 logBox = gtk.VBox()
424 logBox.pack_start(logScroller, True, True, 0)
425 logBox.set_size_request(-1, 200)
426
427 frameAlignment.add(logBox)
428
429 return logFrame
430
431 def _writeLog(self, msg):
432 """Write the given message to the log."""
433 buffer = self._logView.get_buffer()
434 buffer.insert(buffer.get_end_iter(), msg)
435 self._logView.scroll_mark_onscreen(buffer.get_insert())
436
[38]437 def _quit(self, what = None):
438 """Quit from the application."""
[76]439 dialog = gtk.MessageDialog(type = MESSAGETYPE_QUESTION,
440 buttons = BUTTONSTYPE_YES_NO,
441 message_format =
442 "Are you sure to quit the logger?")
443 result = dialog.run()
444 dialog.hide()
445
446 if result==RESPONSETYPE_YES:
447 self._statusIcon.destroy()
448 return gtk.main_quit()
[38]449
[46]450 def _notebookPageSwitch(self, notebook, page, page_num):
451 """Called when the current page of the notebook has changed."""
452 if page_num==0:
[48]453 gobject.idle_add(self._wizard.grabDefault)
[46]454 else:
455 self._mainWindow.set_default(None)
Note: See TracBrowser for help on using the repository browser.