/[ascend]/trunk/pygtk/gtkbrowser.py
ViewVC logotype

Contents of /trunk/pygtk/gtkbrowser.py

Parent Directory Parent Directory | Revision Log Revision Log


Revision 905 - (show annotations) (download) (as text)
Thu Oct 26 00:58:49 2006 UTC (16 years, 3 months ago) by johnpye
File MIME type: text/x-python
File size: 34323 byte(s)
Improved check-for-updates a little bit
1 #!/usr/bin/env python
2
3 import sys
4 def print_loading_status(status,msg=None):
5 sys.stderr.write("\r \r")
6 if msg!=None:
7 sys.stderr.write(msg+"\n")
8 sys.stderr.write(status+"...\r")
9 sys.stderr.flush()
10
11 try:
12 #print_loading_status("Loading PSYCO")
13 #try:
14 # import psyco
15 # psyco.full()
16 # print "Running with PSYCO optimisation..."
17 #except ImportError:
18 # pass
19
20
21 print_loading_status("Loading python standard libraries")
22
23 import re
24 import urlparse
25 import optparse
26 import platform
27 import sys
28 import os.path
29
30 if platform.system() != "Windows":
31 import dl
32 # This sets the flags for dlopen used by python so that the symbols in the
33 # ascend library are made available to libraries dlopened within ASCEND:
34 sys.setdlopenflags(dl.RTLD_GLOBAL|dl.RTLD_NOW)
35
36 print_loading_status("Loading LIBASCEND/ascpy")
37 import ascpy
38
39 print_loading_status("Loading PyGTK, glade, pango")
40
41 import pygtk
42 pygtk.require('2.0')
43 import gtk
44 import gtk.glade
45 import pango
46
47 print_loading_status("Loading python matplotlib")
48 try:
49 import matplotlib
50 matplotlib.use('GTKAgg')
51
52 try:
53 print_loading_status("Trying python numarray")
54 import numarray
55 matplotlib.rcParams['numerix'] = 'numarray'
56 print_loading_status("","Using python module numarray")
57 except ImportError:
58 try:
59 print_loading_status("Trying python numpy")
60 import numpy
61 matplotlib.rcParams['numerix'] = 'numpy'
62 print_loading_status("","Using python module numpy")
63 except ImportError:
64 try:
65 print_loading_status("Trying python Numeric")
66 import Numeric
67 matplotlib.rcParams['numerix'] = 'Numeric'
68 print_loading_status("","Using python module Numeric")
69 except ImportError:
70 print_loading_status("","FAILED TO LOAD A NUMERIC MODULE FOR PYTHON")
71
72 except ImportError,e:
73 print_loading_status("","FAILED TO LOAD MATPLOTLIB")
74 raise RuntimeError("Failed to load MATPLOTLIB (is it installed?). Details:"+str(e))
75
76 print_loading_status("Loading IPython")
77 import console;
78 if not console.have_ipython:
79 print_loading_status("","IPython couldn't be loaded")
80
81 print_loading_status("Loading ASCEND python modules")
82 from preferences import * # loading/saving of .ini options
83 from solverparameters import * # 'solver parameters' window
84 from help import * # viewing help files
85 from incidencematrix import * # incidence/sparsity matrix matplotlib window
86 from observer import * # observer tab support
87 from properties import * # solver_var properties dialog
88 from varentry import * # for inputting of variables with units
89 from diagnose import * # for diagnosing block non-convergence
90 from solverreporter import * # solver status reporting
91 from modelview import * # model browser
92 from integrator import * # integrator dialog
93 from infodialog import * # general-purpose textual information dialog
94 from versioncheck import * # version check (contacts ascend.cruncher2.dyndns.org)
95 import config
96
97 except RuntimeError, e:
98 print "ASCEND had problems starting up. Please report the following"
99 print "error message at http://mantis.cruncher2.dyndns.org/."
100 print "\n\nFull error message:",str(e)
101 print "\n\nPress ENTER to close this window."
102 sys.stdout.flush()
103 sys.stdin.readline();
104 sys.exit();
105
106 except ImportError, e:
107 print "\n\n------------------ ERROR ---------------------"
108 print "ASCEND had problems importing required models."
109 print "\nPlease ensure you have all the runtime prerequisites installed."
110 print "Please then report a bug if you continue to have problems."
111 print "\nFull error message:",str(e)
112 if platform.system()=="Windows":
113 print "\nYou will also need to report the contents of any popup error"
114 print "messages from Windows if any were shown."
115 print "\n\nPress ENTER to close this window."
116 sys.stdout.flush()
117 sys.stdin.readline();
118 sys.exit();
119
120 print_loading_status("Starting GUI")
121
122 # This is my first ever GUI code so please be nice :)
123 # But I *have* at least read
124 # http://www.joelonsoftware.com/uibook/chapters/fog0000000057.html
125 # and leafed through
126 # http://developer.gnome.org/projects/gup/hig/
127
128 # The fancy tree-view gizmo is the GtkTreeView object. See the article
129 # http://aspn.activestate.com/ASPN/Cookbook/Python/Recipe/300304
130 # for the original source code on which my implementation was based.
131
132 ESCAPE_KEY = 65307
133
134 HELP_ROOT = None
135
136 #======================================
137 # Browser is the main ASCEND library/model browser window
138
139 class Browser:
140
141 # ---------------------------------
142 # SETUP
143
144 def __init__(self,librarypath=None,assetspath=None):
145
146 if assetspath==None:
147 assetspath=config.PYGTK_ASSETS
148
149 #--------
150 # load the file referenced in the command line, if any
151
152 print_loading_status("Parsing options","CONFIG = %s"%config.VERSION)
153
154 parser = optparse.OptionParser(usage="%prog [[-m typename] file]", version="gtkbrowser $rev$" )
155 # add options here if we want
156
157 parser.add_option("-m", "--model"
158 ,action="store", type="string", dest="model"
159 ,help="specify the model to instantiate upon loading modules")
160
161 parser.add_option("--pygtk-assets"
162 ,action="store", type="string", dest="assets_dir"
163 ,help="override the configuration value for the location of assets"\
164 +" required by PyGTK for the ASCEND GUI, optional"
165 ,default=assetspath
166 )
167
168 parser.add_option("--library"
169 ,action="store", type="string", dest="library_path"
170 ,help="override the configuration value for the library path"
171 ,default=librarypath
172 )
173
174 parser.add_option("--no-auto-sim"
175 ,action="store_false", dest="auto_sim"
176 ,help="disable auto-instantiation of MODEL named as the file stem"
177 ,default=True
178 )
179
180 (self.options, args) = parser.parse_args()
181
182 #print "OPTIONS_______________:",self.options
183
184 self.assets_dir = self.options.assets_dir
185
186 self.observers = []
187 self.clip = None
188
189 #--------
190 # load up the preferences ini file
191
192 print_loading_status("Loading preferences")
193
194 self.prefs = Preferences()
195
196 _prefpath = self.prefs.getStringPref("Directories","librarypath",None)
197 _preffileopenpath = self.prefs.getStringPref("Directories","fileopenpath",None)
198
199 #--------
200 # set up library path and the path to use for File->Open dialogs
201
202 if self.options.library_path != None:
203 _path = os.path.abspath(self.options.library_path)
204 _pathsrc = "command line options"
205 # when a special path is specified, use that as the file-open location
206 self.fileopenpath = _path
207 else:
208 if _prefpath:
209 _path = _prefpath
210 _pathsrc = "user preferences"
211 else:
212 _path = config.LIBRARY_PATH
213 _pathsrc = "default (config.py)"
214
215 if _preffileopenpath:
216 self.fileopenpath = _preffileopenpath
217 else:
218 self.fileopenpath = _path
219
220 #--------
221 # Create the ASCXX 'Library' object
222
223 print_loading_status("Creating ASCEND 'Library' object","PATH = "+_path+" FROM "+_pathsrc)
224 self.library = ascpy.Library(_path)
225
226 self.sim = None
227
228 #-------------------
229 # Set up the window and main widget actions
230
231 self.glade_file = os.path.join(self.assets_dir,config.GLADE_FILE)
232
233 print_loading_status("Setting up windows") #,"GLADE_FILE = %s" % self.glade_file)
234
235 glade = gtk.glade.XML(self.glade_file,"browserwin")
236
237 self.window = glade.get_widget("browserwin")
238
239
240 if not self.window:
241 raise RuntimeError("Couldn't load window from glade file")
242
243 _display = self.window.get_screen().get_display().get_name()
244 _geom=self.prefs.getGeometrySizePosition(_display,"browserwin")
245 if _geom:
246 self.window.resize(_geom[0],_geom[1])
247 self.window.move(_geom[2],_geom[3])
248
249 self.window.connect("delete_event", self.delete_event)
250
251 self.browserpaned=glade.get_widget("browserpaned")
252 _geom2=self.prefs.getGeometryValue(_display,"browserpaned")
253 if _geom2:
254 self.browserpaned.set_position(_geom2)
255
256 self.openbutton=glade.get_widget("openbutton")
257 self.openbutton.connect("clicked",self.open_click)
258
259 self.reloadbutton=glade.get_widget("reloadbutton")
260 self.reloadbutton.connect("clicked",self.reload_click)
261
262 self.solvebutton=glade.get_widget("solvebutton")
263 self.solvebutton.connect("clicked",self.solve_click)
264
265 self.integratebutton=glade.get_widget("integratebutton")
266 self.integratebutton.connect("clicked",self.integrate_click)
267
268 self.checkbutton=glade.get_widget("checkbutton")
269 self.checkbutton.connect("clicked",self.check_click)
270
271 self.autotoggle=glade.get_widget("autotoggle")
272 self.automenu = glade.get_widget("automenu")
273 self.autotoggle.connect("toggled",self.auto_toggle)
274
275 self.methodrunbutton=glade.get_widget("methodrunbutton")
276 self.methodrunbutton.connect("clicked",self.methodrun_click)
277
278 self.methodsel=glade.get_widget("methodsel")
279
280 self.maintabs = glade.get_widget("maintabs")
281
282 self.statusbar = glade.get_widget("statusbar")
283
284 self.menu = glade.get_widget("browsermenu")
285
286 self.show_solving_popup=glade.get_widget("show_solving_popup")
287 self.show_solving_popup.set_active(self.prefs.getBoolPref("SolverReporter","show_popup",True))
288 self.close_on_converged=glade.get_widget("close_on_converged")
289 self.close_on_converged.set_active(self.prefs.getBoolPref("SolverReporter","close_on_converged",True))
290 self.close_on_nonconverged=glade.get_widget("close_on_nonconverged")
291 self.close_on_nonconverged.set_active(self.prefs.getBoolPref("SolverReporter","close_on_nonconverged",True))
292 self.solver_engine=glade.get_widget("solver_engine")
293
294 self.use_relation_sharing=glade.get_widget("use_relation_sharing")
295 self.use_relation_sharing.set_active(self.prefs.getBoolPref("Compiler","use_relation_sharing",True))
296
297 glade.signal_autoconnect(self)
298
299 #-------
300 # Status icons
301
302 self.fixedimg = gtk.Image()
303 self.fixedimg.set_from_file(os.path.join(self.options.assets_dir,'locked.png'))
304
305 self.iconstatusunknown = None
306 self.iconfixed = self.fixedimg.get_pixbuf()
307 self.iconsolved = self.window.render_icon(gtk.STOCK_YES,gtk.ICON_SIZE_MENU)
308 self.iconactive = self.window.render_icon(gtk.STOCK_NO,gtk.ICON_SIZE_MENU)
309 self.iconunsolved = None
310
311 self.statusicons={
312 ascpy.ASCXX_VAR_STATUS_UNKNOWN: self.iconstatusunknown
313 ,ascpy.ASCXX_VAR_FIXED: self.iconfixed
314 ,ascpy.ASCXX_VAR_SOLVED: self.iconsolved
315 ,ascpy.ASCXX_VAR_ACTIVE: self.iconactive
316 ,ascpy.ASCXX_VAR_UNSOLVED: self.iconunsolved
317 }
318
319
320 self.statusmessages={
321 ascpy.ASCXX_VAR_STATUS_UNKNOWN: "Status unknown"
322 ,ascpy.ASCXX_VAR_FIXED: "Fixed"
323 ,ascpy.ASCXX_VAR_SOLVED: "Converged"
324 ,ascpy.ASCXX_VAR_ACTIVE: "Active (unconverged)"
325 ,ascpy.ASCXX_VAR_UNSOLVED: "Not yet visited"
326 }
327
328 #-------------------
329 # waitwin
330
331 self.waitwin = gtk.gdk.Window(self.window.window,
332 gtk.gdk.screen_width(),
333 gtk.gdk.screen_height(),
334 gtk.gdk.WINDOW_CHILD,
335 0,
336 gtk.gdk.INPUT_ONLY)
337
338 _cursor = gtk.gdk.Cursor(gtk.gdk.WATCH)
339 self.waitwin.set_cursor(_cursor)
340
341 #-------------------
342 # pixbufs to be used in the error listing
343
344 self.iconok = self.window.render_icon(gtk.STOCK_YES,gtk.ICON_SIZE_MENU)
345 self.iconinfo = self.window.render_icon(gtk.STOCK_DIALOG_INFO,gtk.ICON_SIZE_MENU)
346 self.iconwarning = self.window.render_icon(gtk.STOCK_DIALOG_WARNING,gtk.ICON_SIZE_MENU)
347 self.iconerror = self.window.render_icon(gtk.STOCK_DIALOG_ERROR,gtk.ICON_SIZE_MENU)
348
349 #--------------------
350 # pixbufs for solver_var status
351
352 #--------------------
353 # set up the error view
354
355 self.errorview = glade.get_widget("errorview")
356 errstorecolstypes = [gtk.gdk.Pixbuf,str,str,str,int]
357 self.errorstore = gtk.TreeStore(*errstorecolstypes)
358 errtitles = ["","Location","Message"];
359 self.errorview.set_model(self.errorstore)
360 self.errcols = [ gtk.TreeViewColumn() for _type in errstorecolstypes]
361
362 i = 0
363 for tvcolumn in self.errcols[:len(errtitles)]:
364 tvcolumn.set_title(errtitles[i])
365 self.errorview.append_column(tvcolumn)
366
367 if i>0:
368 _renderer = gtk.CellRendererText()
369 tvcolumn.pack_start(_renderer, True)
370 tvcolumn.add_attribute(_renderer, 'text', i)
371 if(i==2):
372 tvcolumn.add_attribute(_renderer, 'foreground', 3)
373 tvcolumn.add_attribute(_renderer, 'weight', 4)
374 else:
375 _renderer1 = gtk.CellRendererPixbuf()
376 tvcolumn.pack_start(_renderer1, False)
377 tvcolumn.add_attribute(_renderer1, 'pixbuf', int(0))
378
379 i = i + 1
380
381
382 #--------------------
383 # set up the error reporter callback
384 self.reporter = ascpy.getReporter()
385 self.reporter.setPythonErrorCallback(self.error_callback)
386
387
388 #-------
389 # Solver engine list
390
391 _slvlist = ascpy.getSolvers()
392 self.solver_engine_menu = gtk.Menu()
393 self.solver_engine_menu.show()
394 self.solver_engine.set_submenu(self.solver_engine_menu)
395 self.solver_engine_menu_dict = {}
396 _fmi = None
397 for _s in _slvlist:
398 _mi = gtk.RadioMenuItem(_fmi,_s.getName(),False)
399 if _fmi==None:
400 _fmi = _mi
401 _mi.show()
402 _mi.connect('toggled',self.on_select_solver_toggled,_s.getName())
403 self.solver_engine_menu.append(_mi)
404 self.solver_engine_menu_dict[_s.getName()]=_mi
405
406 _pref_solver = self.prefs.getStringPref("Solver","engine","QRSlv")
407 _mi = self.solver_engine_menu_dict.get(_pref_solver)
408 if _mi:
409 _mi.set_active(1)
410
411 #--------
412 # Assign an icon to the main window
413
414 self.icon = None
415 if config.ICON_EXTENSION:
416 _iconpath = ""
417 try:
418 _icon = gtk.Image()
419 _iconpath = os.path.join(self.assets_dir,'ascend'+config.ICON_EXTENSION)
420 _icon.set_from_file(_iconpath)
421 _iconpbuf = _icon.get_pixbuf()
422 self.window.set_icon(_iconpbuf)
423 self.icon = _iconpbuf
424 except Exception, e:
425 print "FAILED TO SET APPLICATION ICON PATH '%s': %s" % (_iconpath,str(e))
426 self.reporter.reportError("FAILED to set application icon '%s': %s"
427 % (_iconpath,str(e))
428 )
429
430 #-------------------
431 # set up the module view
432
433 self.modtank = {}
434 self.moduleview = glade.get_widget("moduleview")
435 modulestorecoltypes = [str, str, int] # bool=can-be-instantiated
436 self.modulestore = gtk.TreeStore(*modulestorecoltypes)
437 moduleviewtitles = ["Module name", "Filename"]
438 self.moduleview.set_model(self.modulestore)
439 self.modcols = [ gtk.TreeViewColumn() for _type in modulestorecoltypes]
440 i = 0
441 for modcol in self.modcols[:len(moduleviewtitles)]:
442 modcol.set_title(moduleviewtitles[i])
443 self.moduleview.append_column(modcol)
444 _renderer = gtk.CellRendererText()
445 modcol.pack_start(_renderer, True)
446 modcol.add_attribute(_renderer, 'text', i)
447 modcol.add_attribute(_renderer,'weight',2)
448 i = i + 1
449 self.moduleview.connect("row-activated", self.module_activated )
450
451 #--------------------
452 # set up the methods combobox
453
454 self.methodstore = gtk.ListStore(str)
455 self.methodsel.set_model(self.methodstore)
456 _methodrenderer = gtk.CellRendererText()
457 self.methodsel.pack_start(_methodrenderer, True)
458 self.methodsel.add_attribute(_methodrenderer, 'text',0)
459
460 #--------
461 # set up the instance browser view
462
463 self.modelview = ModelView(self, glade)
464
465 #--------
466 # set up the tabs
467 self.tabs = {}
468 self.activetab = None # most recent observer tab
469
470 #--------
471 # set the state of the 'auto' toggle
472
473 self.is_auto = self.prefs.getBoolPref("Browser","auto_solve",True)
474 self.autotoggle.set_active(self.is_auto)
475 self.automenu.set_active(self.is_auto)
476
477 #--------
478 # tell libascend about this 'browser' object
479
480 print dir(ascpy.Registry())
481 ascpy.Registry().set("browser",self)
482
483 #--------
484 # options
485
486 if(len(args)==1):
487 self.do_open(args[0])
488
489 print "Options: ",self.options
490
491 _model = None
492 if self.options.model:
493 _model = self.options.model
494 print "MODEL: '%s'" % _model
495 elif self.options.auto_sim:
496 _head, _tail = os.path.split(args[0])
497 if(_tail):
498 _model, _ext = os.path.splitext(_tail)
499
500 if _model:
501 try:
502 _t=self.library.findType(_model)
503 try:
504 self.do_sim(_t)
505 if not self.options.model:
506 self.reporter.reportNote("Instantiated self-titled model '%s'" %_model)
507 except RuntimeError, e:
508 self.reporter.reportError("Failed to create instance of '%s': %s"
509 %(_model, str(e))
510 );
511 except RuntimeError, e:
512 if self.options.model:
513 self.reporter.reportError("Unknown model type '%s': %s"
514 %(_model, str(e))
515 );
516
517 def run(self):
518 self.window.show()
519 print_loading_status("ASCEND is now running")
520 gtk.main()
521
522 # ------------------
523 # SOLVER LIST
524
525 def set_solver(self,solvername):
526 """ this sets the active solver in the GUI, which is the default applied to newly instantiated models """
527 self.solver = ascpy.Solver(solvername)
528 self.prefs.setStringPref("Solver","engine",solvername)
529 self.reporter.reportNote("Set solver engine to '%s'" % solvername)
530
531 # --------------------------------------------
532 # MAJOR GUI COMMANDS
533
534 def on_fix_variable_activate(self,*args):
535 self.modelview.on_fix_variable_activate(*args)
536
537 def on_free_variable_activate(self,*args):
538 self.modelview.on_free_variable_activate(*args)
539
540 def on_select_solver_toggled(self,widget,solvername):
541 if widget.get_active():
542 self.set_solver(solvername)
543
544 def do_open(self,filename):
545 # TODO does the user want to lose their work?
546 # TODO do we need to chdir?
547
548 _context = self.statusbar.get_context_id("do_open")
549
550 self.errorstore.clear()
551 self.modelview.clear()
552
553 # self.library.clear()
554
555 print "Filename =",filename
556 self.statusbar.push(_context,"Loading '"+filename+"'")
557 self.library.load(filename)
558 print "Statusbar =",self.statusbar
559 try:
560 self.statusbar.pop(_context)
561 except TypeError,e:
562 print "For some reason, a type error (context=%s,filename=%s): %s" % (_context,filename,e)
563
564 self.filename = filename
565
566 # Load the current list of modules into self.modules
567 self.modtank = {}
568 self.modulestore.clear()
569 modules = self.library.getModules()
570 #self.library.listModules()
571 try:
572 _lll=len(modules)
573 except:
574 _msg = "UNABLE TO ACCESS MODULES LIST. This is bad.\n"+\
575 "Check your SWIG configuration (check for warnings during build)."
576
577 self.reporter.reportError(_msg)
578 raise RuntimeError(_msg)
579
580 for m in reversed(modules):
581 _n = str( m.getName() )
582 _f = str( m.getFilename() )
583 #print "ADDING ROW name %s, file = %s" % (_n, _f)
584 _r = self.modulestore.append(None, [ _n, _f, pango.WEIGHT_NORMAL ])
585 for t in self.library.getModuleTypes(m):
586 _n = t.getName()
587 _hasparams = t.hasParameters()
588 if _hasparams:
589 _w = pango.WEIGHT_NORMAL
590 else:
591 _w = pango.WEIGHT_BOLD
592
593 #print "ADDING TYPE %s" % _n
594 _piter = self.modulestore.append(_r , [ _n, "", _w ])
595 _path = self.modulestore.get_path(_piter)
596 self.modtank[_path]=t
597
598 #print "DONE ADDING MODULES"
599
600 self.sim = None;
601 self.maintabs.set_current_page(0);
602
603 # See http://www.daa.com.au/pipermail/pygtk/2005-October/011303.html
604 # for details on how the 'wait cursor' is done.
605 def start_waiting(self, message):
606 self.waitcontext = self.statusbar.get_context_id("waiting")
607 self.statusbar.push(self.waitcontext,message)
608
609 if self.waitwin:
610 self.waitwin.show()
611
612 while gtk.events_pending():
613 gtk.main_iteration()
614
615 def stop_waiting(self):
616 if self.waitwin:
617 self.statusbar.pop(self.waitcontext)
618 self.waitwin.hide()
619
620 def do_sim(self, type_object):
621 self.sim = None;
622 # TODO: clear out old simulation first!
623
624 print "DO_SIM(%s)" % str(type_object.getName())
625 self.start_waiting("Compiling...")
626
627 try:
628 _v = self.prefs.getBoolPref("Compiler","use_relation_sharing",True)
629 ascpy.getCompiler().setUseRelationSharing(_v)
630
631 self.sim = type_object.getSimulation(str(type_object.getName())+"_sim")
632
633 self.reporter.reportNote("SIMULATION ASSIGNED")
634 except RuntimeError, e:
635 self.stop_waiting()
636 self.reporter.reportError(str(e))
637 return
638
639 print "...DONE 'getSimulation'"
640 self.stop_waiting()
641
642 self.start_waiting("Building simulation...")
643 print "BUILDING SIMULATION"
644
645 try:
646 self.sim.build()
647 except RuntimeError, e:
648 self.stop_waiting()
649 self.reporter.reportError(str(e))
650 return
651
652 print "DONE BUILDING"
653 self.stop_waiting()
654
655 # get method names and load them into the GUI
656 self.methodstore.clear()
657 _methods = self.sim.getType().getMethods()
658 _activemethod = None;
659 for _m in _methods:
660 _i = self.methodstore.append([_m.getName()])
661 if _m.getName()=="on_load":
662 self.methodsel.set_active_iter(_i)
663
664 self.modelview.setSimulation(self.sim)
665
666 # set the active solver on the simulation
667 self.sim.setSolver(self.solver)
668
669 # run the 'on_load' method
670 self.start_waiting("Running default method...")
671
672 try:
673 print "SIMULATION CREATED, RUNNING DEFAULT METHOD NOW..."
674 self.sim.runDefaultMethod()
675 except RuntimeError, e:
676 self.stop_waiting()
677 self.reporter.reportError(str(e))
678 return
679
680 self.stop_waiting()
681
682 self.modelview.refreshtree()
683
684 def do_solve_if_auto(self):
685 if self.is_auto:
686 self.sim.checkInstance()
687 self.do_solve()
688 else:
689 self.sim.processVarStatus()
690 self.modelview.refreshtree()
691
692 self.sync_observers()
693
694 def do_solve(self):
695 if not self.sim:
696 self.reporter.reportError("No model selected yet")
697 return
698
699 self.start_waiting("Solving with %s..." % self.solver.getName())
700
701 if self.prefs.getBoolPref("SolverReporter","show_popup",True):
702 reporter = PopupSolverReporter(self,self.sim.getNumVars())
703 else:
704 reporter = SimpleSolverReporter(self)
705
706 self.sim.solve(self.solver,reporter)
707
708 self.stop_waiting()
709
710 self.sim.processVarStatus()
711 self.modelview.refreshtree()
712
713 def do_integrate(self):
714 if not self.sim:
715 self.reporter.reportError("No model selected yet")
716 return
717 integwin = IntegratorWindow(self,self.sim)
718 _integratorreporter = integwin.run()
719 if _integratorreporter!=None:
720 _integratorreporter.run()
721 self.sim.processVarStatus()
722 self.modelview.refreshtree()
723
724
725 def do_check(self):
726 if not self.sim:
727 self.reporter.reportError("No simulation yet")
728 return
729
730 self.start_waiting("Checking system...")
731
732 try:
733 self.sim.checkInstance()
734 self.reporter.reportWarning("System instance check run, check above for error (if any).")
735 # the above gives output but doesn't throw errors or return a status.
736 # ... this is a problem (at the C level)
737
738 status = self.sim.checkDoF()
739 if status==ascpy.ASCXX_DOF_UNDERSPECIFIED:
740 self.on_show_fixable_variables_activate(None)
741 elif status==ascpy.ASCXX_DOF_OVERSPECIFIED:
742 self.on_show_freeable_variables_activate(None)
743 elif status==ascpy.ASCXX_DOF_STRUCT_SINGULAR:
744 if not self.sim.checkStructuralSingularity():
745 sing = self.sim.getSingularityInfo()
746 title = "Structural singularity"
747 text = title
748 text += "\n\nThe singularity can be reduced by freeing the following variables:"
749 msgs = {
750 "The singularity can be reduced by freeing the following variables" : sing.freeablevars
751 ,"Relations involved in the structural singularity" : sing.rels
752 ,"Variables involved in the structural singularity" : sing.vars
753 }
754 for k,v in msgs.iteritems():
755 text+="\n\n%s:" % k
756 if len(v):
757 _l = [j.getName() for j in v]
758 _l.sort()
759 text+= "\n\t" + "\n\t".join(_l)
760 else:
761 text += "\nnone"
762
763 _dialog = InfoDialog(self,self.window,text,title)
764 _dialog.run()
765
766 self.reporter.reportNote("System DoF check OK")
767
768 except RuntimeError, e:
769 self.stop_waiting()
770 self.reporter.reportError(str(e))
771 return
772
773 self.stop_waiting()
774
775 self.modelview.refreshtree()
776
777 def do_method(self,method):
778 if not self.sim:
779 self.reporter.reportError("No model selected yet")
780
781 self.sim.run(method)
782 self.modelview.refreshtree()
783
784 def do_quit(self):
785 print_loading_status("Saving window location")
786 self.reporter.clearPythonErrorCallback()
787
788 _w,_h = self.window.get_size()
789 _t,_l = self.window.get_position()
790 _display = self.window.get_screen().get_display().get_name()
791 self.prefs.setGeometrySizePosition(_display,"browserwin",_w,_h,_t,_l );
792
793 _p = self.browserpaned.get_position()
794 self.prefs.setGeometryValue(_display,"browserpaned",_p);
795
796 print_loading_status("Saving current directory")
797 self.prefs.setStringPref("Directories","fileopenpath",self.fileopenpath)
798
799 self.prefs.setBoolPref("Browser","auto_solve",self.is_auto)
800
801 print_loading_status("Saving preferences")
802 # causes prefs to be saved unless they are still being used elsewher
803 del(self.prefs)
804
805 print_loading_status("Closing down GTK")
806 gtk.main_quit()
807
808 print_loading_status("Clearing error callback")
809 self.reporter.clearPythonErrorCallback()
810
811 print_loading_status("Quitting")
812 return False
813
814 def on_tools_sparsity_click(self,*args):
815
816 self.reporter.reportNote("Preparing incidence matrix...")
817 _im = self.sim.getIncidenceMatrix();
818
819 self.reporter.reportNote("Plotting incidence matrix...")
820
821 _sp = IncidenceMatrixWindow(_im);
822 _sp.run();
823
824 def on_diagnose_blocks_click(self,*args):
825 try:
826 _bl = self.sim.getActiveBlock()
827 except RuntimeError, e:
828 self.reporter.reportError(str(e))
829 return
830 _db = DiagnoseWindow(self,_bl)
831 _db.run();
832
833 def on_add_observer_click(self,*args):
834 self.create_observer()
835
836 def on_keep_observed_click(self,*args):
837 print "KEEPING..."
838 if len(self.observers) <= 0:
839 self.reporter.reportError("No observer defined!")
840 return
841 self.tabs[self.currentobservertab].do_add_row()
842
843 def on_copy_observer_matrix_click(self,*args):
844 if self.clip == None:
845 self.clip = gtk.Clipboard()
846
847 if len(self.observers) <= 0:
848 self.reporter.reportError("No observer defined!")
849 return
850 self.tabs[self.currentobservertab].copy_to_clipboard(self.clip)
851
852 def on_use_relation_sharing_toggle(self,checkmenuitem,*args):
853 _v = checkmenuitem.get_active()
854 self.prefs.setBoolPref("Compiler","use_relation_sharing",_v)
855 self.reporter.reportNote("Relation sharing set to "+str(_v))
856
857 def on_show_solving_popup_toggle(self,checkmenuitem,*args):
858 _v = checkmenuitem.get_active()
859 self.prefs.setBoolPref("SolverReporter","show_popup",_v)
860 print "SET TO",_v
861
862 def on_close_on_converged_toggle(self,checkmenuitem,*args):
863 _v = checkmenuitem.get_active()
864 self.prefs.setBoolPref("SolverReporter","close_on_converged",_v)
865
866 def on_close_on_nonconverged_toggle(self,checkmenuitem,*args):
867 _v = checkmenuitem.get_active()
868 self.prefs.setBoolPref("SolverReporter","close_on_nonconverged",_v)
869
870 def on_show_variables_near_bounds_activate(self,*args):
871 _epsilon = 1e-4;
872 text = "Variables Near Bounds"
873 title=text;
874 text += "\n"
875 _vars = self.sim.getVariablesNearBounds(_epsilon)
876 if len(_vars):
877 for _v in _vars:
878 text += "\n%s"%_v.getName()
879 else:
880 text +="\nnone"
881 _dialog = InfoDialog(self,self.window,text,title)
882 _dialog.run()
883
884 # --------------------------------------------
885 # MODULE LIST
886
887 def module_activated(self, treeview, path, column, *args):
888 modules = self.library.getModules()
889 print "PATH",path
890 if len(path)==1:
891 self.reporter.reportNote("Launching of external editor not yet implemented")
892 elif len(path)==2:
893 if(self.modtank.has_key(path)):
894 _type = self.modtank[path];
895 self.reporter.reportNote("Creating simulation for type %s" % str(_type.getName()) )
896 self.do_sim(_type)
897 else:
898 self.reporter.reportError("Didn't find type corresponding to row")
899
900 # ----------------------------------
901 # ERROR PANEL
902
903 def get_error_row_data(self,sev,filename,line,msg):
904 _sevicon = {
905 0: self.iconok
906 ,1: self.iconinfo
907 ,2: self.iconwarning
908 ,3: self.iconerror
909 ,4: self.iconinfo
910 ,5: self.iconwarning
911 ,6: self.iconerror
912 }[sev]
913
914 _fontweight = pango.WEIGHT_NORMAL
915 if sev==6:
916 _fontweight = pango.WEIGHT_BOLD
917
918 _fgcolor = "black"
919 if sev==4:
920 _fgcolor = "#888800"
921 elif sev==5:
922 _fgcolor = "#884400"
923 elif sev==6:
924 _fgcolor = "#880000"
925 elif sev==0:
926 _fgcolor = BROWSER_FIXED_COLOR
927
928 if not filename and not line:
929 _fileline = ""
930 else:
931 if(len(filename) > 25):
932 filename = "..."+filename[-22:]
933 _fileline = filename + ":" + str(line)
934
935 _res = [_sevicon,_fileline,msg.rstrip(),_fgcolor,_fontweight]
936 #print _res
937 return _res
938
939 def error_callback(self,sev,filename,line,msg):
940 try:
941 pos = self.errorstore.append(None, self.get_error_row_data(sev, filename,line,msg))
942 path = self.errorstore.get_path(pos)
943 col = self.errorview.get_column(3)
944 self.errorview.scroll_to_cell(path,col)
945 except Exception,e:
946 print "UNABLE TO DISPLAY ERROR MESSAGE '%s'"%msg
947
948 return 0;
949
950 # --------------------------------
951 # BUTTON METHODS
952
953 def open_click(self,*args):
954 #print_loading_status("CURRENT FILEOPENPATH is",self.fileopenpath)
955 dialog = gtk.FileChooserDialog("Open ASCEND model...",
956 self.window,
957 gtk.FILE_CHOOSER_ACTION_OPEN,
958 (gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL, gtk.STOCK_OPEN, gtk.RESPONSE_OK)
959 )
960 dialog.set_current_folder(self.fileopenpath)
961 dialog.set_default_response(gtk.RESPONSE_OK)
962 dialog.set_transient_for(self.window)
963 dialog.set_modal(True)
964
965 filter = gtk.FileFilter()
966 filter.set_name("*.a4c, *.a4l")
967 filter.add_pattern("*.[Aa]4[Cc]")
968 filter.add_pattern("*.[Aa]4[Ll]")
969 dialog.add_filter(filter)
970
971 filter = gtk.FileFilter()
972 filter.set_name("All files")
973 filter.add_pattern("*")
974 dialog.add_filter(filter)
975
976 response = dialog.run()
977 _filename = dialog.get_filename()
978 print "\nFILENAME SELECTED:",_filename
979
980 _path = dialog.get_current_folder()
981 if _path:
982 self.fileopenpath = _path
983
984 dialog.hide()
985
986 if response == gtk.RESPONSE_OK:
987 self.reporter.reportNote("File %s selected." % dialog.get_filename() )
988 self.library.clear()
989 self.do_open( _filename)
990
991 def reload_click(self,*args):
992 _type = None
993 if(self.sim):
994 _type = self.sim.getType().getName().toString();
995
996 self.library.clear()
997 self.do_open(self.filename)
998
999 if _type:
1000 _t = self.library.findType(_type)
1001 self.do_sim(_t)
1002
1003 def props_activate(self,widget,*args):
1004 return self.modelview.props_activate(self,widget,*args)
1005
1006 def observe_activate(self,widget,*args):
1007 return self.modelview.observe_activate(self,widget,*args)
1008
1009 def solve_click(self,*args):
1010 #self.reporter.reportError("Solving simulation '" + self.sim.getName().toString() +"'...")
1011 self.do_solve()
1012
1013 def console_click(self,*args):
1014 try:
1015 console.start(self)
1016 except RuntimeError,e:
1017 self.reporter.reportError("Unable to start console: "+str(e));
1018
1019 def integrate_click(self,*args):
1020 self.do_integrate()
1021
1022 def check_click(self,*args):
1023 self.do_check()
1024 #self.reporter.reportError("CHECK clicked")
1025
1026 def preferences_click(self,*args):
1027 if not self.sim:
1028 self.reporter.reportError("No simulation created yet!");
1029
1030 _paramswin = SolverParametersWindow(self)
1031 _paramswin.show()
1032
1033 def methodrun_click(self,*args):
1034 _sel = self.methodsel.get_active_text()
1035 if _sel:
1036 _method = None
1037 _methods = self.sim.getType().getMethods()
1038 for _m in _methods:
1039 if _m.getName()==_sel:
1040 _method = _m
1041 if not _method:
1042 self.reporter.reportError("Method is not valid")
1043 return
1044 self.do_method(_method)
1045 else:
1046 self.reporter.reportError("No method selected")
1047
1048 def auto_toggle(self,button,*args):
1049 self.is_auto = button.get_active()
1050 if hasattr(self,'automenu'):
1051 self.automenu.set_active(self.is_auto)
1052 else:
1053 raise RuntimeError("no automenu")
1054
1055 #if self.is_auto:
1056 # self.reporter.reportSuccess("Auto mode is now ON")
1057 #else:
1058 # self.reporter.reportSuccess("Auto mode is now OFF")
1059
1060 def on_file_quit_click(self,*args):
1061 self.do_quit()
1062
1063 def on_tools_auto_toggle(self,checkmenuitem,*args):
1064 self.is_auto = checkmenuitem.get_active()
1065 self.autotoggle.set_active(self.is_auto)
1066
1067 def on_help_about_click(self,*args):
1068 _xml = gtk.glade.XML(self.glade_file,"aboutdialog")
1069 _about = _xml.get_widget("aboutdialog")
1070 _about.set_position(gtk.WIN_POS_CENTER_ON_PARENT)
1071 _about.set_transient_for(self.window);
1072 _about.set_version(config.VERSION)
1073 _about.run()
1074 _about.destroy()
1075
1076 def on_help_contents_click(self,*args):
1077 _help = Help(HELP_ROOT)
1078 _help.run()
1079
1080 def on_help_check_for_updates_click(self,*args):
1081 v = VersionCheck()
1082 title = "Check for updates"
1083 text = "Your version is %s\n" % config.VERSION
1084 try:
1085 v.check()
1086 text += "Latest version is %s\n" % v.latest
1087 if v.info:
1088 text += "Get more info at %s\n" % v.info
1089 if v.download:
1090 text += "Download from %s\n" % v.download
1091 except Exception, e:
1092 text += "\nUnable to check version\n"
1093 text += str(e)
1094
1095 _dialog = InfoDialog(self,self.window,text,title)
1096 _dialog.run()
1097
1098 def on_show_fixable_variables_activate(self,*args):
1099 v = self.sim.getFixableVariables()
1100 text = "Fixable Variables"
1101 title = text
1102 text += "\n"
1103 if len(v):
1104 for var in v:
1105 text += "\n%s"%var
1106 else:
1107 text += "\nnone"
1108 _dialog = InfoDialog(self,self.window,text,title)
1109 _dialog.run()
1110
1111 def on_show_freeable_variables_activate(self,*args):
1112 v = self.sim.getFreeableVariables()
1113
1114 text = "Freeable Variables"
1115 title = text
1116 text += "\n"
1117 if len(v):
1118 for var in v:
1119 text += "\n%s" % var
1120 else:
1121 text += "\nnone"
1122 _dialog = InfoDialog(self,self.window,text,title)
1123 _dialog.run()
1124
1125 def on_show_external_functions_activate(self,*args):
1126 v = self.library.getExtMethods()
1127 text = "External Functions"
1128 title = text
1129 text +="\nHere is the list of external functions currently present in"
1130 text +=" the Library:"
1131
1132 if len(v):
1133 for ext in v:
1134 text += "\n\n%s (%d inputs, %d outputs):" % \
1135 (ext.getName(), ext.getNumInputs(), ext.getNumOutputs())
1136 text += "\n%s" % ext.getHelp()
1137 else:
1138 text +="\n\nNone"
1139 _dialog = InfoDialog(self,self.window,text,title)
1140 _dialog.run()
1141
1142 def on_maintabs_switch_page(self,notebook,page,pagenum):
1143 print("Page switched to %d" % pagenum)
1144 if pagenum in self.tabs.keys():
1145 self.currentobservertab = pagenum
1146
1147 def create_observer(self,name=None):
1148 _xml = gtk.glade.XML(self.glade_file,"observervbox");
1149 _label = gtk.Label();
1150 _tab = self.maintabs.append_page(_xml.get_widget("observervbox"),_label);
1151 _obs = ObserverTab(xml=_xml, name=name, browser=self, tab=_tab)
1152 _label.set_text(_obs.name)
1153 self.observers.append(_obs)
1154 self.tabs[_tab] = _obs
1155 self.currentobservertab = _tab
1156 return _obs
1157
1158 def sync_observers(self):
1159 for _o in self.observers:
1160 _o.sync()
1161
1162 def delete_event(self, widget, event):
1163 self.do_quit()
1164 return False
1165
1166 def observe(self,instance):
1167 if len(self.observers) ==0:
1168 self.create_observer()
1169 _observer = self.tabs[self.currentobservertab]
1170 _observer.add_instance(instance)
1171
1172 if __name__ == "__main__":
1173 b = Browser();
1174 b.run()

john.pye@anu.edu.au
ViewVC Help
Powered by ViewVC 1.1.22