6 from PyQt4.QtCore import *
7 from PyQt4.QtGui import *
10 from sface.config import config
11 from sface.logwindow import LogWindow
12 from sface.rspecwindow import RSpecWindow
13 from sface.screens.sfascreen import SfaScreen
15 # depending on the platform..
16 # could probably use Qt's resource system but looks overkill for just one file...
17 def locate_image_file (filename):
18 for dir in [ '/usr/share', '/Applications/sface.app/Contents/Resources/sface' ] :
19 for suffix in ['png','jpg']:
20 attempt=os.path.join(dir,'images',"%s.%s"%(filename,suffix))
21 if os.path.isfile(attempt) : return attempt
22 return os.path.join('/could/not/locate/image/file',filename)
24 def load_screens(dirname):
26 for fn in os.listdir(dirname):
27 if not fn.endswith(".py"):
29 modname = fn.rsplit(".py",1)[0]
30 if modname == "sfascreen":
31 # ignore this, it's the base class, not a screen
33 modnames.append(modname)
35 # we want the stock screens to show up in a specific order. plugins can
36 # show up in any order afterward.
38 sort_order = ["mainscreen", "userscreen", "configscreen", "helpscreen"]
40 for modname in sort_order:
41 if modname in modnames:
42 modnames.remove(modname)
43 modnames.insert(0,modname)
45 # import each module and find whatever class(es) is descendant from
46 # SfaScreen within the module. Might be a better idea to just define a
47 # screens=[] variable in each module
50 for modname in modnames:
52 mod = __import__("sface.screens." + modname, fromlist=["modname"])
54 print "Exception while importing screen", modname
58 for object in dir(mod):
59 object = getattr(mod, object)
60 if hasattr(object, "__bases__") and sface.screens.sfascreen.SfaScreen in object.__bases__:
61 screens.append(object)
66 def __init__(self, screens, parent=None):
67 QWidget.__init__(self, parent)
69 self.title = QLabel("", self)
70 scene=QGraphicsScene()
71 pixmap = QPixmap(locate_image_file('graphic-sfa64'))
72 logolabel=QLabel("",self)
73 logolabel.setPixmap(pixmap)
75 hlayout = QHBoxLayout()
76 hlayout.addWidget(logolabel)
77 hlayout.addWidget(self.title)
79 gotolabel=QLabel("Go to: ", self)
80 gotolabel.setAlignment(Qt.AlignRight)
81 hlayout.addWidget(gotolabel)
83 self.screenLabels = []
84 for screen in screens:
85 label = QLabel(screen.getLinkText(), self)
86 label.setAlignment(Qt.AlignRight)
87 self.screenLabels.append(label)
88 hlayout.addWidget(label)
90 self.setLayout(hlayout)
92 def setTitle(self, title):
93 self.title.setText(title)
97 def __init__(self, parent=None):
98 QLabel.__init__(self, "", parent)
99 self.setMaximumWidth(640)
100 self.sliceUpdateDate()
102 def set(self, msg, timeout):
105 QTimer.singleShot(timeout, self.reset)
107 def sliceUpdateDate(self):
108 rspec_file = config.getSliceRSpecFile()
109 if not os.path.exists(rspec_file):
112 creation_time = os.stat(rspec_file).st_ctime
113 last_update = time.ctime(creation_time)
114 self.set("Slice data last refreshed on %s" % last_update, timeout=None)
118 QTimer.singleShot(1500, self.sliceUpdateDate)
121 class MainWindow(QWidget):
122 def __init__(self, parent=None):
123 QWidget.__init__(self, parent)
125 # These are top-level windows and should be initialized with parent set
126 # to our parent. Otherwise, getting a segfault on exit in Ubuntu.
127 self.logWindow = LogWindow(parent)
128 self.rspecWindow = RSpecWindow(parent)
130 self.pix = QLabel(self)
132 screenClasses = load_screens(os.path.dirname(sface.screens.__file__))
133 self.screenWidgets = []
135 self.screens = QStackedWidget(self)
136 for screen in screenClasses:
137 # use a try/catch block to isolate the screen. Third-party plugins
140 screenWidget = screen(self)
141 self.screenWidgets.append(screenWidget)
142 self.screens.addWidget(screenWidget)
144 print "Exception while creating screen", screen.__name__
145 traceback.print_exc()
147 self.screens.addWidget(self.pix)
148 self.next_screen = None
150 self.nav = Nav(self.screenWidgets, self)
152 if self.screenWidgets:
153 self.nav.setTitle(self.screenWidgets[0].getTitleText())
155 self.status = Status(self)
156 self.log = QLabel("<a href='showlog'>Show Log</a>", self)
157 self.rspec = QLabel("<a href='showlog'>Show RSpec</a>", self)
159 hlayout = QHBoxLayout()
160 hlayout.addWidget(self.status)
162 hlayout.addWidget(self.rspec)
163 hlayout.addWidget(self.log)
165 layout = QVBoxLayout()
166 layout.addWidget(self.nav)
167 layout.addWidget(self.screens)
168 layout.addLayout(hlayout)
169 self.setLayout(layout)
170 self.resize(800, 500)
172 for link in self.nav.screenLabels:
173 self.connect(link, SIGNAL('linkActivated(QString)'),
174 self.animateToScreen)
176 self.connect(self.log, SIGNAL('linkActivated(QString)'),
178 self.connect(self.rspec, SIGNAL('linkActivated(QString)'),
179 self.showRSpecWindow)
181 def redirectOutputToLog(self):
182 self.logWindow.redirectOutput()
184 def showLogWindow(self, link):
185 self.logWindow.show()
186 self.logWindow.resize(800, 200)
187 self.logWindow.raise_()
188 self.logWindow.activateWindow()
190 def showRSpecWindow(self, link):
191 self.rspecWindow.show()
192 self.rspecWindow.resize(500, 640)
193 self.rspecWindow.raise_()
194 self.rspecWindow.activateWindow()
197 def animatePixmap(self, y):
200 def animateToScreen(self, link):
201 for screen in self.screenWidgets:
202 if link == screen.name:
203 self.next_screen = screen
205 curr_screen = self.screens.currentWidget()
207 if self.next_screen == curr_screen:
208 self.setStatus("Already showing %s" % curr_screen.getTitleText(), timeout=1000)
211 # This is an optimization to have a smoother animation. We
212 # render the widget into a pixmap and animate that instead of
213 # moving the whole widget around.
214 pixmap = QPixmap(self.screens.size())
215 curr_screen.render(pixmap)
216 self.screens.setCurrentWidget(self.pix)
217 self.pix.setPixmap(pixmap)
219 timeLine = QTimeLine(500, self)
220 timeLine.setFrameRange(0, self.screens.height());
221 self.connect(timeLine, SIGNAL('frameChanged(int)'), self.animatePixmap)
222 self.connect(timeLine, SIGNAL('finished()'), self.toNextScreen)
226 def toNextScreen(self):
227 self.screens.setCurrentWidget(self.next_screen)
228 self.nav.setTitle(self.next_screen.getTitleText())
230 def setStatus(self, msg, timeout):
231 self.status.set(msg, timeout)
233 def nodeSelectionChanged(self, hostname):
234 if self.rspecWindow.isVisible():
235 self.rspecWindow.showNode(hostname)
237 def closeEvent(self, event):
238 # give the screens an opportunity to veto the close
239 for screen in self.screenWidgets:
240 if not screen.canClose():
244 # give the screens an opportunity to close gracefully
245 for screen in self.screenWidgets:
246 screen.mainWindowClose()
248 QWidget.closeEvent(self, event)