a744e48aed56e16460601957989197bcb8646e82
[mevemon] / package / src / ui / diablo / gui.py
1 #
2 # mEveMon - A character monitor for EVE Online
3 # Copyright (c) 2010  Ryan and Danny Campbell, and the mEveMon Team
4 #
5 # mEveMon is free software; you can redistribute it and/or modify
6 # it under the terms of the GNU General Public License as published by
7 # the Free Software Foundation; either version 3 of the License, or
8 # (at your option) any later version.
9 #
10 # mEveMon is distributed in the hope that it will be useful,
11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13 # GNU General Public License for more details.
14 #
15 # You should have received a copy of the GNU General Public License
16 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
17 #
18
19 # DIABLO UI: Heavily based on Ry's Fremantle Python code. --danny
20
21 import sys, time
22
23 import gtk
24 import hildon
25 import gobject
26
27 from ui import models
28
29 class BaseUI():
30
31     menu_items = ("Settings", "About", "Refresh")
32
33     def create_menu(self, window):
34         menu = gtk.Menu()
35         for command in self.menu_items:
36             # Create menu entries
37             button = gtk.MenuItem(command)
38
39             if command == "About":
40                 button.connect("activate", self.about_clicked)
41             elif command == "Settings":
42                 button.connect("activate", self.settings_clicked, window)
43             elif command == "Refresh":
44                 button.connect("activate", self.refresh_clicked, window)
45             else:
46                 assert False, command
47
48             # Add entry to the view menu
49             menu.append(button)
50             
51         menu.show_all()
52
53         return menu
54
55     def settings_clicked(self, button, window):
56
57         RESPONSE_NEW, RESPONSE_EDIT, RESPONSE_DELETE = range(3)
58
59         dialog = gtk.Dialog()
60         dialog.set_transient_for(window)
61         dialog.set_title("Settings")
62
63         vbox = dialog.vbox
64
65         acctsLabel = gtk.Label("Accounts:")
66         acctsLabel.set_justify(gtk.JUSTIFY_LEFT)
67
68         vbox.pack_start(acctsLabel, False, False, 1)
69
70         self.accounts_model = models.AccountsModel(self.controller)
71
72         accounts_treeview = gtk.TreeView(model = self.accounts_model)
73         self.add_columns_to_accounts(accounts_treeview)
74         vbox.pack_start(accounts_treeview, False, False, 1)
75
76         # all stock responses are negative, so we can use any positive value
77         new_button = dialog.add_button("New", RESPONSE_NEW)
78         #TODO: get edit button working
79         #edit_button = dialog.add_button("Edit", RESPONSE_EDIT)
80         delete_button = dialog.add_button("Delete", RESPONSE_DELETE)
81         ok_button = dialog.add_button(gtk.STOCK_OK, gtk.RESPONSE_OK)
82         cancel_button = dialog.add_button(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL)
83
84         #TODO: for some reason the scrollbar shows up in the middle of the
85         # dialog. Why?
86         #scrollbar = gtk.VScrollbar()
87         #vbox.add(scrollbar)
88
89         dialog.show_all()
90
91         result = dialog.run()
92         
93         while(result != gtk.RESPONSE_CANCEL):
94             if result == RESPONSE_NEW:
95                 self.new_account_clicked(window)
96             elif result == RESPONSE_EDIT:
97                 # get the selected treeview item and pop up the account_box
98                 self.edit_account(accounts_treeview)
99             elif result == RESPONSE_DELETE:
100                 # get the selected treeview item, and delete the gconf keys
101                 self.delete_account(accounts_treeview)
102             elif result == gtk.RESPONSE_OK:
103                 self.char_model.get_characters()
104                 break
105         
106             result = dialog.run()
107
108         dialog.destroy()
109
110
111
112     def get_selected_item(self, treeview, column):
113         selection = treeview.get_selection()
114         model, miter = selection.get_selected()
115
116         value = model.get_value(miter, column)
117
118         return value
119
120     def edit_account(self, treeview):
121         uid = self.get_selected_item(treeview, 0)
122         # pop up the account dialog
123
124         self.accounts_model.get_accounts()
125
126     def delete_account(self, treeview):
127         uid = self.get_selected_item(treeview, 0)
128         self.controller.remove_account(uid)
129         # refresh model
130         self.accounts_model.get_accounts()
131
132
133     def add_columns_to_accounts(self, treeview):
134         #Column 0 for the treeview
135         renderer = gtk.CellRendererText()
136         column = gtk.TreeViewColumn('Account ID', renderer, text=0)
137         column.set_property("expand", True)
138         treeview.append_column(column)
139         #Column 2 (characters) for the treeview
140         column = gtk.TreeViewColumn('Characters', renderer, 
141                 markup=models.AccountsModel.C_CHARS)
142         column.set_property("expand", True)
143         treeview.append_column(column)
144
145
146     def new_account_clicked(self, window):
147         dialog = gtk.Dialog()
148     
149         #get the vbox to pack all the settings into
150         vbox = dialog.vbox
151     
152         dialog.set_transient_for(window)
153         dialog.set_title("Settings")
154
155         uidLabel = gtk.Label("User ID:")
156         uidLabel.set_justify(gtk.JUSTIFY_LEFT)
157         vbox.add(uidLabel)
158         
159         uidEntry = gtk.Entry()
160         uidEntry.set_property('is_focus', False)
161         
162         vbox.add(uidEntry)
163
164         apiLabel = gtk.Label("API key:")
165         apiLabel.set_justify(gtk.JUSTIFY_LEFT)
166         vbox.add(apiLabel)
167         
168         apiEntry = gtk.Entry()
169         apiEntry.set_property('is_focus', False)
170
171         vbox.add(apiEntry)
172            
173         ok_button = dialog.add_button(gtk.STOCK_OK, gtk.RESPONSE_OK)
174
175         dialog.show_all()
176         result = dialog.run()
177
178         if result == gtk.RESPONSE_OK:
179             self.controller.add_account(uidEntry.get_text(), apiEntry.get_text())
180             self.accounts_model.get_accounts()
181         
182         dialog.destroy()
183
184         return result
185
186     def about_clicked(self, button):
187         dialog = gtk.AboutDialog()
188         dialog.set_website(self.controller.about_website)
189         dialog.set_website_label(self.controller.about_website)
190         dialog.set_name(self.controller.about_name)
191         dialog.set_authors(self.controller.about_authors)
192         dialog.set_comments(self.controller.about_text)
193         dialog.set_version(self.controller.app_version)
194         dialog.run()
195         dialog.destroy()
196
197     def add_label(self, text, box, markup=True, align="left", padding=1):
198         label = gtk.Label(text)
199         if markup:
200             label.set_use_markup(True)
201         if align == "left":
202             label.set_alignment(0, 0.5)
203         box.pack_start(label, False, False, padding)
204
205 class mEveMonUI(BaseUI):
206
207     def __init__(self, controller):
208         self.controller = controller
209         gtk.set_application_name("mEveMon")
210
211         # create the main window
212         win = hildon.Window()
213         win.connect("destroy", self.controller.quit)
214         win.show_all()
215         progress_bar = hildon.hildon_banner_show_progress(win, None, "Loading overview...")
216         progress_bar.set_fraction(0.4)
217
218         # Create menu
219         menu = self.create_menu(win)
220
221         # Attach menu to the window
222         win.set_menu(menu)
223
224         character_win = CharacterSheetUI(self.controller)
225
226         # create the treeview --danny
227         self.char_model = models.CharacterListModel(self.controller)
228         treeview = gtk.TreeView(model = self.char_model)
229         treeview.connect('row-activated', character_win.build_window)
230         treeview.set_model(self.char_model)
231         self.add_columns_to_treeview(treeview)
232
233         # add the treeview with scrollbar --danny
234         win.add_with_scrollbar(treeview)
235         win.show_all()
236
237         progress_bar.set_fraction(1)
238         progress_bar.destroy()
239
240     def add_columns_to_treeview(self, treeview):
241         #Column 0 for the treeview
242         renderer = gtk.CellRendererPixbuf()
243         column = gtk.TreeViewColumn()
244         column.pack_start(renderer, True)
245         column.add_attribute(renderer, "pixbuf", 0)
246         treeview.append_column(column)
247
248         #Column 1 for the treeview
249         renderer = gtk.CellRendererText()
250         column = gtk.TreeViewColumn('Character Name', renderer, text=1)
251         column.set_property("expand", True)
252         treeview.append_column(column)
253  
254     def refresh_clicked(self, button, window):
255         progress_bar = hildon.hildon_banner_show_progress(win, None, "Loading characters...")
256         progress_bar.set_fraction(1)
257         self.char_model.get_characters()
258         progress_bar.destroy()
259
260 class CharacterSheetUI(BaseUI):
261     
262     def __init__(self, controller):
263         self.controller = controller
264         self.sheet = None
265         self.char_id = None
266         self.skills_model = None
267
268     def build_window(self, treeview, path, view_column):
269         # TODO: this is a really long and ugly function, split it up somehow
270
271         win = hildon.Window()
272         win.show_all() 
273
274         progress_bar = hildon.hildon_banner_show_progress(win, None, "Loading character sheet...")
275         progress_bar.set_fraction(0.4)
276
277         # Create menu
278         # NOTE: we probably want a window-specific menu for this page, but the
279         # main appmenu works for now
280         menu = self.create_menu(win)
281         # Attach menu to the window
282         win.set_menu(menu)
283
284         model = treeview.get_model()
285         miter = model.get_iter(path)
286         
287         # column 0 is the portrait, column 1 is name
288         char_name = model.get_value(miter, 1)
289         uid = model.get_value(miter, 2)
290         self.char_id = self.controller.char_name2id(char_name)
291         
292         self.sheet = self.controller.get_char_sheet(uid, self.char_id)
293
294         win.set_title(char_name)
295         
296         hbox = gtk.HBox(False, 0)
297         info_vbox = gtk.VBox(False, 1)
298
299         portrait = gtk.Image()
300         portrait.set_from_file(self.controller.get_portrait(char_name, 256))
301         portrait.show()
302
303         hbox.pack_start(portrait, False, False, 10)
304         hbox.pack_start(info_vbox, False, False, 5)
305         
306         vbox = gtk.VBox(False, 0)
307
308         vbox.pack_start(hbox, False, False, 0)
309
310
311         self.fill_info(info_vbox)
312         self.fill_stats(info_vbox)
313
314         separator = gtk.HSeparator()
315         vbox.pack_start(separator, False, False, 5)
316         separator.show()
317  
318         
319         self.add_label("<big>Skill in Training:</big>", vbox, align="normal")
320         skill = self.controller.get_skill_in_training(uid, self.char_id)
321         
322         if skill.skillInTraining:
323
324             skilltree = self.controller.get_skill_tree()
325             
326             # I'm assuming that we will always find a skill with the skill ID
327             for group in skilltree.skillGroups:
328                 found_skill = group.skills.Get(skill.trainingTypeID, False)
329                 if found_skill:
330                     skill_name = found_skill.typeName
331                     break
332                 
333             self.add_label("%s <small>(Level %d)</small>" % (skill_name, skill.trainingToLevel),
334                     vbox, align="normal")
335             self.add_label("<small>start time: %s\t\tend time: %s</small>" %(time.ctime(skill.trainingStartTime),
336                 time.ctime(skill.trainingEndTime)), vbox, align="normal")
337             
338             progressbar = gtk.ProgressBar()
339             fraction_completed = (time.time() - skill.trainingStartTime) / \
340                     (skill.trainingEndTime - skill.trainingStartTime)
341             
342             progressbar.set_fraction(fraction_completed)
343             align = gtk.Alignment(0.5, 0.5, 0.5, 0)
344             vbox.pack_start(align, False, False, 5)
345             align.show()
346             align.add(progressbar)
347             progressbar.show()
348
349         else:
350             self.add_label("<small>No skills are currently being trained</small>", vbox, align="normal")
351
352         separator = gtk.HSeparator()
353         vbox.pack_start(separator, False, False, 0)
354         separator.show()
355         
356         self.add_label("<big>Skills:</big>", vbox, align="normal")
357
358
359         skills_model = models.CharacterSkillsModel(self.controller, self.char_id)
360         skills_treeview = gtk.TreeView(model = skills_model)
361         skills_treeview.set_model(skills_model)
362         self.add_columns_to_skills_view(skills_treeview)
363
364         vbox.pack_start(skills_treeview, False, False, 0)
365
366         win.add_with_scrollbar(vbox)
367         win.show_all()
368
369         progress_bar.set_fraction(1)
370         progress_bar.destroy()
371
372     def fill_info(self, box):
373         self.add_label("<big><big>%s</big></big>" % self.sheet.name, box)
374         self.add_label("<small>%s %s %s</small>" % (self.sheet.gender, self.sheet.race, self.sheet.bloodLine), box)
375         self.add_label("", box, markup=False)
376         self.add_label("<small><b>Corp:</b> %s</small>" % self.sheet.corporationName, box)
377         self.add_label("<small><b>Balance:</b> %s ISK</small>" % self.sheet.balance, box)
378
379     def fill_stats(self, box):
380
381         atr = self.sheet.attributes
382
383         self.add_label("<small><b>I: </b>%d  <b>M: </b>%d  <b>C: </b>%d  " \
384                 "<b>P: </b>%d  <b>W: </b>%d</small>" % (atr.intelligence,
385                     atr.memory, atr.charisma, atr.perception, atr.willpower), box)
386
387
388     def add_columns_to_skills_view(self, treeview):
389         #Column 0 for the treeview
390         renderer = gtk.CellRendererText()
391         column = gtk.TreeViewColumn('Skill Name', renderer, markup=0)
392         column.set_property("expand", True)
393         treeview.append_column(column)
394         
395         #Column 1 for the treeview
396         column = gtk.TreeViewColumn('Rank', renderer, markup=1)
397         column.set_property("expand", True)
398         treeview.append_column(column)
399         
400         #Column 2
401         renderer = gtk.CellRendererText()
402         renderer = gtk.CellRendererText()
403         column = gtk.TreeViewColumn('Points', renderer, markup=2)
404         column.set_property("expand", True)
405         treeview.append_column(column)
406         
407         #Column 3
408         column = gtk.TreeViewColumn('Level', renderer, markup=3)
409         column.set_property("expand", True)
410         treeview.append_column(column)
411
412     def refresh_clicked(self, button, window):
413         progress_bar = hildon.hildon_banner_show_progress(win, None, "Loading overview...")
414         progress_bar.set_fraction(1)
415         self.skills_model.get_skills()
416         progress_bar.destroy()
417
418 if __name__ == "__main__":
419     main()
420