#file: knowledge.py #Copyright (C) 2005,2006,2008 Evil Mr Henry, Phil Bordelon, and FunnyMan3595 #This file is part of Endgame: Singularity. #Endgame: Singularity is free software; you can redistribute it and/or modify #it under the terms of the GNU General Public License as published by #the Free Software Foundation; either version 2 of the License, or #(at your option) any later version. #Endgame: Singularity is distributed in the hope that it will be useful, #but WITHOUT ANY WARRANTY; without even the implied warranty of #MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the #GNU General Public License for more details. #You should have received a copy of the GNU General Public License #along with Endgame: Singularity; if not, write to the Free Software #Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA #This file is used to display the knowledge lists. import pygame from code import g from code.graphics import text, button, dialog, widget, constants, listbox, g as gg class KnowledgeScreen(dialog.Dialog): def __init__(self, *args, **kwargs): super(KnowledgeScreen, self).__init__(*args, **kwargs) self.knowledge_type_list = ("Techs", "Items", "Concepts") self.cur_knowledge_type = "" self.cur_knowledge = None self.knowledge_inner_list = () self.knowledge_inner_list_key = () self.cur_focus = 0 self.knowledge_choice = \ listbox.UpdateListbox(self, (0.05, .18), (.21, .25), list=self.knowledge_type_list, update_func=self.set_knowledge_type) self.knowledge_inner = \ listbox.UpdateListbox(self, (.30, .18), (.21, .25), list=self.knowledge_inner_list, update_func=self.set_knowledge) self.description_pane = \ widget.BorderedWidget(self, (0.55, 0), (0.40, 0.7), anchor = constants.TOP_LEFT) self.back_button = button.ExitDialogButton(self, (0.17, 0.46), (-.3, -.1), anchor=constants.TOP_LEFT, text="BACK", hotkey="b") #Set up the key handling. #This is likely not the best way to do it. self.remove_key_handler(pygame.K_UP, self.knowledge_choice.got_key) self.remove_key_handler(pygame.K_DOWN, self.knowledge_choice.got_key) self.remove_key_handler(pygame.K_PAGEUP, self.knowledge_choice.got_key) self.remove_key_handler(pygame.K_PAGEDOWN, self.knowledge_choice.got_key) self.remove_key_handler(pygame.K_UP, self.knowledge_inner.got_key) self.remove_key_handler(pygame.K_DOWN, self.knowledge_inner.got_key) self.remove_key_handler(pygame.K_PAGEUP, self.knowledge_inner.got_key) self.remove_key_handler(pygame.K_PAGEDOWN, self.knowledge_inner.got_key) self.add_key_handler(pygame.K_UP, self.key_handle) self.add_key_handler(pygame.K_DOWN, self.key_handle) self.add_key_handler(pygame.K_LEFT, self.key_handle) self.add_key_handler(pygame.K_RIGHT, self.key_handle) #custom key handler. def key_handle(self, event): if event.type != pygame.KEYDOWN: return if event.key == pygame.K_LEFT or event.key == pygame.K_RIGHT: self.cur_focus = (self.cur_focus + 1) % 2 else: if self.cur_focus == 0: self.knowledge_choice.got_key(event) elif self.cur_focus == 1: self.knowledge_inner.got_key(event) #fill the right-hand listbox def set_inner_list(self, item_type): if item_type == "Techs": items = [tech for tech in g.techs.values() if tech.available()] elif item_type == "Concepts": items = [ [item[1][0], item[0]] for item in g.help_strings.items()] items.sort() else: items = [item for item in g.items.values() if item.available()] if item_type != "Concepts": items = [ [item.name, item.id ] for item in items] items.sort() return_list1 = [] return_list2 = [] for name, id in items: return_list1.append(id) return_list2.append(name) return return_list1, return_list2 #Make sure the left listbox is correct after moving around. def set_knowledge_type(self, list_pos): if getattr(self, "knowledge_choice", None) is None: self.knowledge_inner_list_key, self.knowledge_inner_list = \ self.set_inner_list(self.cur_knowledge_type) return # Not yet initialized. prev_know = self.cur_knowledge_type if list_pos == -1: prev_know = "" list_pos = 0 if 0 <= list_pos < len(self.knowledge_choice.list): self.cur_knowledge_type = self.knowledge_choice.list[list_pos] if prev_know != self.cur_knowledge_type: self.knowledge_inner_list_key, self.knowledge_inner.list = \ self.set_inner_list(self.cur_knowledge_type) self.knowledge_inner.list_pos = 0 self.set_knowledge(0) #Make sure the right-hand listbox is correct. def set_knowledge(self, list_pos): if getattr(self, "knowledge_inner", None) is None: return # Not yet initialized. prev_know = self.cur_knowledge if 0 <= list_pos < len(self.knowledge_inner.list): self.cur_knowledge = self.knowledge_inner.list[list_pos] if prev_know != self.cur_knowledge: self.show_info(self.cur_knowledge_type, self.knowledge_inner_list_key[list_pos]) #print information to the right. def show_info(self, knowledge_type, knowledge_key): desc_text = "" if knowledge_type == "Concepts": desc_text = g.help_strings[knowledge_key][0] + "\n\n" + \ g.help_strings[knowledge_key][1] if knowledge_type == "Techs": desc_text = g.techs[knowledge_key].name + "\n\n" #Cost if not g.techs[knowledge_key].done: desc_text += "Research Cost:\n" + \ g.to_money(g.techs[knowledge_key].cost_left[0])+" Money, " desc_text += g.to_cpu(g.techs[knowledge_key].cost_left[1]) + " CPU\n" if g.techs[knowledge_key].danger == 0: desc_text += "Study anywhere." elif g.techs[knowledge_key].danger == 1: desc_text += "Study underseas or farther." elif g.techs[knowledge_key].danger == 2: desc_text += "Study off-planet." elif g.techs[knowledge_key].danger == 3: desc_text += "Study far away from this planet." elif g.techs[knowledge_key].danger == 4: desc_text += "Do not study in this dimension." else: desc_text += "Research complete." desc_text += "\n\n"+g.techs[knowledge_key].description if g.techs[knowledge_key].done: desc_text += "\n\n"+g.techs[knowledge_key].result if knowledge_type == "Items": desc_text = g.items[knowledge_key].name + "\n\n" #Building cost desc_text += "Building Cost:\n" desc_text += g.to_money(g.items[knowledge_key].cost[0])+" Money, " desc_text += g.to_time(g.items[knowledge_key].cost[2]) + "\n" #Quality if g.items[knowledge_key].item_type == "cpu": desc_text += "CPU per day: " desc_text += str(g.items[knowledge_key].item_qual) elif g.items[knowledge_key].item_type == "reactor": desc_text += "Detection chance reduction: " desc_text += g.to_percent(g.items[knowledge_key].item_qual) elif g.items[knowledge_key].item_type == "network": desc_text += "CPU bonus: " desc_text += g.to_percent(g.items[knowledge_key].item_qual) elif g.items[knowledge_key].item_type == "security": desc_text += "Detection chance reduction: " desc_text += g.to_percent(g.items[knowledge_key].item_qual) desc_text += "\n\n"+g.items[knowledge_key].description text.Text(self.description_pane, (0, 0), (-1, -1), text=desc_text, background_color=gg.colors["dark_blue"], text_size=20, align=constants.LEFT, valign=constants.TOP, borders=constants.ALL) def show(self): self.set_knowledge_type(-1) self.knowledge_choice.list_pos = 0 self.knowledge_inner.list_pos = 0 return super(KnowledgeScreen, self).show()