gemrb/GUIScripts/bg2/GUIPR.py

   1 # -*-python-*-
   2 # GemRB - Infinity Engine Emulator
   3 # Copyright (C) 2003-2004 The GemRB Project
   4 #
   5 # This program is free software; you can redistribute it and/or
   6 # modify it under the terms of the GNU General Public License
   7 # as published by the Free Software Foundation; either version 2
   8 # of the License, or (at your option) any later version.
   9 #
  10 # This program 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, write to the Free Software
  17 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  18 #
  19
  20
  21 # GUIPR.py - scripts to control the priest spells windows from the GUIPR winpack
  22
  23 ###################################################
  24
  25 import GemRB
  26 import GUICommonWindows
  27 from GUIDefines import *
  28 from ie_stats import *
  29 from GUICommon import CloseOtherWindow
  30 from GUICommon import GameWindow
  31 from GUICommonWindows import *
  32
  33 PriestWindow = None
  34 PriestSpellInfoWindow = None
  35 PriestSpellLevel = 0
  36 PriestSpellUnmemorizeWindow = None
  37 PortraitWindow = None
  38 OptionsWindow = None
  39 OldPortraitWindow = None
  40 OldOptionsWindow = None
  41
  42 def OpenPriestWindow ():
  43         global PriestWindow, OptionsWindow, PortraitWindow
  44         global OldPortraitWindow, OldOptionsWindow
  45
  46         if CloseOtherWindow (OpenPriestWindow):
  47                 if PriestWindow:
  48                         PriestWindow.Unload ()
  49                 if OptionsWindow:
  50                         OptionsWindow.Unload ()
  51                 if PortraitWindow:
  52                         PortraitWindow.Unload ()
  53
  54                 PriestWindow = None
  55                 GemRB.SetVar ("OtherWindow", -1)
  56                 GameWindow.SetVisible(WINDOW_VISIBLE)
  57                 GemRB.UnhideGUI ()
  58                 GUICommonWindows.PortraitWindow = OldPortraitWindow
  59                 OldPortraitWindow = None
  60                 GUICommonWindows.OptionsWindow = OldOptionsWindow
  61                 OldOptionsWindow = None
  62                 SetSelectionChangeHandler (None)
  63                 return
  64
  65         GemRB.HideGUI ()
  66         GameWindow.SetVisible(WINDOW_INVISIBLE)
  67
  68         GemRB.LoadWindowPack ("GUIPR", 640, 480)
  69         PriestWindow = Window = GemRB.LoadWindowObject (2)
  70         GemRB.SetVar ("OtherWindow", PriestWindow.ID)
  71         #saving the original portrait window
  72         OldOptionsWindow = GUICommonWindows.OptionsWindow
  73         OptionsWindow = GemRB.LoadWindowObject (0)
  74         MarkMenuButton (OptionsWindow)
  75         SetupMenuWindowControls (OptionsWindow, 0, "OpenPriestWindow")
  76         OptionsWindow.SetFrame ()
  77         OldPortraitWindow = GUICommonWindows.PortraitWindow
  78         PortraitWindow = OpenPortraitWindow (0)
  79
  80         Button = Window.GetControl (1)
  81         Button.SetEvent (IE_GUI_BUTTON_ON_PRESS, "PriestPrevLevelPress")
  82
  83         Button = Window.GetControl (2)
  84         Button.SetEvent (IE_GUI_BUTTON_ON_PRESS, "PriestNextLevelPress")
  85
  86         #setup level buttons
  87         for i in range (7):
  88                 Button = Window.GetControl (55 + i)
  89                 Button.SetEvent (IE_GUI_BUTTON_ON_PRESS, "RefreshPriestLevel")
  90                 Button.SetFlags (IE_GUI_BUTTON_RADIOBUTTON, OP_OR)
  91
  92         for i in range (7):
  93                 Button = Window.GetControl (55 + i)
  94                 Button.SetVarAssoc ("PriestSpellLevel", i)
  95
  96         # Setup memorized spells buttons
  97         for i in range (12):
  98                 Button = Window.GetControl (3 + i)
  99                 Button.SetBorder (0,0,0,0,0,0,0,0,64,0,1)
 100                 Button.SetSprites ("SPELFRAM",0,0,0,0,0)
 101                 Button.SetFlags (IE_GUI_BUTTON_PICTURE, OP_OR)
 102                 Button.SetState (IE_GUI_BUTTON_LOCKED)
 103
 104         # Setup book spells buttons
 105         for i in range (24):
 106                 Button = Window.GetControl (27 + i)
 107                 Button.SetFlags (IE_GUI_BUTTON_NO_IMAGE, OP_OR)
 108                 Button.SetState (IE_GUI_BUTTON_LOCKED)
 109
 110         SetSelectionChangeHandler (UpdatePriestWindow)
 111         UpdatePriestWindow ()
 112         OptionsWindow.SetVisible (WINDOW_VISIBLE)
 113         #bringing window front
 114         Window.SetVisible (WINDOW_FRONT)
 115         PortraitWindow.SetVisible (WINDOW_VISIBLE)
 116         return
 117
 118 def UpdatePriestWindow ():
 119         global PriestMemorizedSpellList, PriestKnownSpellList
 120
 121         PriestMemorizedSpellList = []
 122         PriestKnownSpellList = []
 123
 124         Window = PriestWindow
 125         pc = GemRB.GameGetSelectedPCSingle ()
 126         type = IE_SPELL_TYPE_PRIEST
 127         level = PriestSpellLevel
 128         max_mem_cnt = GemRB.GetMemorizableSpellsCount (pc, type, level)
 129
 130         Label = Window.GetControl (0x10000032)
 131         GemRB.SetToken("SPELLLEVEL", str(level+1) )
 132         Label.SetText (10345)
 133
 134         Name = GemRB.GetPlayerName (pc, 0)
 135         Label = Window.GetControl (0x10000035)
 136         Label.SetText (Name)
 137
 138         mem_cnt = GemRB.GetMemorizedSpellsCount (pc, type, level)
 139         for i in range (12):
 140                 Button = Window.GetControl (3 + i)
 141                 if i < mem_cnt:
 142                         ms = GemRB.GetMemorizedSpell (pc, type, level, i)
 143                         Button.SetSpellIcon (ms['SpellResRef'], 0)
 144                         Button.SetFlags (IE_GUI_BUTTON_NO_IMAGE, OP_NAND)
 145                         Button.SetFlags (IE_GUI_BUTTON_PICTURE, OP_OR)
 146                         if ms['Flags']:
 147                                 Button.SetEvent (IE_GUI_BUTTON_ON_PRESS, "OpenPriestSpellUnmemorizeWindow")
 148                         else:
 149                                 Button.SetEvent (IE_GUI_BUTTON_ON_PRESS, "OnPriestUnmemorizeSpell")
 150                         Button.SetEvent (IE_GUI_BUTTON_ON_RIGHT_PRESS, "OpenPriestSpellInfoWindow")
 151                         spell = GemRB.GetSpell (ms['SpellResRef'])
 152                         Button.SetTooltip (spell['SpellName'])
 153                         PriestMemorizedSpellList.append (ms['SpellResRef'])
 154                         Button.SetVarAssoc ("SpellButton", i)
 155                         Button.EnableBorder (0, ms['Flags'] == 0)
 156                 else:
 157                         if i < max_mem_cnt:
 158                                 Button.SetFlags (IE_GUI_BUTTON_NORMAL, OP_SET)
 159                         else:
 160                                 Button.SetFlags (IE_GUI_BUTTON_NO_IMAGE, OP_SET)
 161                         Button.SetEvent (IE_GUI_BUTTON_ON_PRESS, "")
 162                         Button.SetEvent (IE_GUI_BUTTON_ON_RIGHT_PRESS, "")
 163                         Button.SetTooltip ('')
 164                         Button.EnableBorder (0, 0)
 165
 166
 167         known_cnt = GemRB.GetKnownSpellsCount (pc, type, level)
 168         for i in range (24):
 169                 Button = Window.GetControl (27 + i)
 170                 if i < known_cnt:
 171                         ks = GemRB.GetKnownSpell (pc, type, level, i)
 172                         Button.SetSpellIcon (ks['SpellResRef'], 0)
 173                         Button.SetEvent (IE_GUI_BUTTON_ON_PRESS, "OnPriestMemorizeSpell")
 174                         Button.SetEvent (IE_GUI_BUTTON_ON_RIGHT_PRESS, "OpenPriestSpellInfoWindow")
 175                         spell = GemRB.GetSpell (ks['SpellResRef'])
 176                         Button.SetTooltip (spell['SpellName'])
 177                         PriestKnownSpellList.append (ks['SpellResRef'])
 178                         Button.SetVarAssoc ("SpellButton", 100 + i)
 179
 180                 else:
 181                         Button.SetFlags (IE_GUI_BUTTON_NO_IMAGE, OP_OR)
 182                         Button.SetFlags (IE_GUI_BUTTON_PICTURE, OP_NAND)
 183                         Button.SetEvent (IE_GUI_BUTTON_ON_PRESS, "")
 184                         Button.SetEvent (IE_GUI_BUTTON_ON_RIGHT_PRESS, "")
 185                         Button.SetTooltip ('')
 186                         Button.EnableBorder (0, 0)
 187
 188         Class = GemRB.GetPlayerStat (GemRB.GameGetSelectedPCSingle(), IE_CLASS)
 189         DivineCaster = ClassSkillsTable.GetValue (Class, 1)
 190         if DivineCaster == "*":
 191                 # also check the DRUIDSPELL column
 192                 DivineCaster = ClassSkillsTable.GetValue (Class, 0)
 193         if DivineCaster == "*":
 194                 Window.SetVisible (WINDOW_GRAYED)
 195         else:
 196                 Window.SetVisible (WINDOW_VISIBLE)
 197         return
 198
 199 def PriestPrevLevelPress ():
 200         global PriestSpellLevel
 201
 202         if PriestSpellLevel > 0:
 203                 PriestSpellLevel = PriestSpellLevel - 1
 204                 UpdatePriestWindow ()
 205         return
 206
 207 def PriestNextLevelPress ():
 208         global PriestSpellLevel
 209
 210         if PriestSpellLevel < 6:
 211                 PriestSpellLevel = PriestSpellLevel + 1
 212                 UpdatePriestWindow ()
 213         return
 214
 215 def RefreshPriestLevel ():
 216         global PriestSpellLevel
 217
 218         PriestSpellLevel = GemRB.GetVar ("PriestSpellLevel")
 219         UpdatePriestWindow ()
 220         return
 221
 222 def OpenPriestSpellInfoWindow ():
 223         global PriestSpellInfoWindow
 224
 225         if PriestSpellInfoWindow != None:
 226                 if PriestSpellInfoWindow:
 227                         PriestSpellInfoWindow.Unload ()
 228                 PriestSpellInfoWindow = None
 229                 return
 230
 231         PriestSpellInfoWindow = Window = GemRB.LoadWindowObject (3)
 232
 233         #back
 234         Button = Window.GetControl (5)
 235         Button.SetText (15416)
 236         Button.SetEvent (IE_GUI_BUTTON_ON_PRESS, "OpenPriestSpellInfoWindow")
 237
 238         index = GemRB.GetVar ("SpellButton")
 239         if index < 100:
 240                 ResRef = PriestMemorizedSpellList[index]
 241         else:
 242                 ResRef = PriestKnownSpellList[index - 100]
 243
 244         spell = GemRB.GetSpell (ResRef)
 245
 246         Label = Window.GetControl (0x0fffffff)
 247         Label.SetText (spell['SpellName'])
 248
 249         Button = Window.GetControl (2)
 250         Button.SetSpellIcon (ResRef, 1)
 251
 252         Text = Window.GetControl (3)
 253         Text.SetText (spell['SpellDesc'])
 254
 255         Window.ShowModal (MODAL_SHADOW_GRAY)
 256         return
 257
 258 def OnPriestMemorizeSpell ():
 259         pc = GemRB.GameGetSelectedPCSingle ()
 260         level = PriestSpellLevel
 261         type = IE_SPELL_TYPE_PRIEST
 262
 263         index = GemRB.GetVar ("SpellButton") - 100
 264
 265         if GemRB.MemorizeSpell (pc, type, level, index):
 266                 UpdatePriestWindow ()
 267         return
 268
 269 def OpenPriestSpellRemoveWindow ():
 270         global PriestSpellUnmemorizeWindow
 271
 272         PriestSpellUnmemorizeWindow = Window = GemRB.LoadWindowObject (5)
 273
 274         # "Are you sure you want to ....?"
 275         TextArea = Window.GetControl (3)
 276         TextArea.SetText (11824)
 277
 278         # Remove
 279         Button = Window.GetControl (0)
 280         Button.SetText (17507)
 281         Button.SetEvent (IE_GUI_BUTTON_ON_PRESS, "OnPriestRemoveSpell")
 282         Button.SetFlags (IE_GUI_BUTTON_DEFAULT, OP_OR)
 283
 284         # Cancel
 285         Button = Window.GetControl (1)
 286         Button.SetText (13727)
 287         Button.SetEvent (IE_GUI_BUTTON_ON_PRESS, "ClosePriestSpellUnmemorizeWindow")
 288         Button.SetFlags (IE_GUI_BUTTON_CANCEL, OP_OR)
 289
 290         Window.ShowModal (MODAL_SHADOW_GRAY)
 291         return
 292
 293 def ClosePriestSpellUnmemorizeWindow ():
 294         global PriestSpellUnmemorizeWindow
 295
 296         if PriestSpellUnmemorizeWindow:
 297                 PriestSpellUnmemorizeWindow.Unload ()
 298         PriestSpellUnmemorizeWindow = None
 299         return
 300
 301 def OpenPriestSpellUnmemorizeWindow ():
 302         global PriestSpellUnmemorizeWindow
 303
 304         PriestSpellUnmemorizeWindow = Window = GemRB.LoadWindowObject (5)
 305
 306         # "Are you sure you want to ....?"
 307         TextArea = Window.GetControl (3)
 308         TextArea.SetText (11824)
 309
 310         # Remove
 311         Button = Window.GetControl (0)
 312         Button.SetText (17507)
 313         Button.SetEvent (IE_GUI_BUTTON_ON_PRESS, "OnPriestUnmemorizeSpell")
 314         Button.SetFlags (IE_GUI_BUTTON_DEFAULT, OP_OR)
 315
 316         # Cancel
 317         Button = Window.GetControl (1)
 318         Button.SetText (13727)
 319         Button.SetEvent (IE_GUI_BUTTON_ON_PRESS, "ClosePriestSpellUnmemorizeWindow")
 320         Button.SetFlags (IE_GUI_BUTTON_CANCEL, OP_OR)
 321
 322         Window.ShowModal (MODAL_SHADOW_GRAY)
 323         return
 324
 325 def OnPriestUnmemorizeSpell ():
 326         if PriestSpellUnmemorizeWindow:
 327                 ClosePriestSpellUnmemorizeWindow ()
 328
 329         pc = GemRB.GameGetSelectedPCSingle ()
 330         level = PriestSpellLevel
 331         type = IE_SPELL_TYPE_PRIEST
 332
 333         index = GemRB.GetVar ("SpellButton")
 334
 335         if GemRB.UnmemorizeSpell (pc, type, level, index):
 336                 UpdatePriestWindow ()
 337         return
 338
 339 def OnPriestRemoveSpell ():
 340         ClosePriestSpellUnmemorizeWindow()
 341         OpenPriestSpellInfoWindow()
 342
 343         pc = GemRB.GameGetSelectedPCSingle ()
 344         level = PriestSpellLevel
 345         type = IE_SPELL_TYPE_PRIEST
 346
 347         index = GemRB.GetVar ("SpellButton") - 100
 348
 349         #remove spell from book
 350         GemRB.RemoveSpell (pc, type, level, index)
 351         UpdatePriestWindow ()
 352         return
 353
 354 ###################################################
 355 # End of file GUIPR.py