gemrb/GUIScripts/bg1/GUIMG.py

   1 # -*-python-*-
   2 # GemRB - Infinity Engine Emulator
   3 # Copyright (C) 2003 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 # GUIMG.py - scripts to control mage spells windows from GUIMG winpack
  22
  23 ###################################################
  24
  25 import GemRB
  26 import GUICommon
  27 import CommonTables
  28 import GUICommonWindows
  29 from GUIDefines import *
  30 from ie_stats import *
  31
  32 MageWindow = None
  33 MageSpellInfoWindow = None
  34 MageSpellLevel = 0
  35 MageSpellUnmemorizeWindow = None
  36 OldOptionsWindow = None
  37
  38
  39 def OpenMageWindow ():
  40         global MageWindow, OptionsWindow
  41         global OldOptionsWindow
  42
  43         if GUICommon.CloseOtherWindow (OpenMageWindow):
  44                 if MageWindow:
  45                         MageWindow.Unload ()
  46                 if OptionsWindow:
  47                         OptionsWindow.Unload ()
  48                 MageWindow = None
  49                 GemRB.SetVar ("OtherWindow", -1)
  50                 GUICommon.GameWindow.SetVisible(WINDOW_VISIBLE)
  51                 GemRB.UnhideGUI ()
  52                 OptionsWindow = OldOptionsWindow
  53                 OldOptionsWindow = None
  54                 GUICommonWindows.SetSelectionChangeHandler(None)
  55                 return
  56
  57         GemRB.HideGUI ()
  58         GUICommon.GameWindow.SetVisible(WINDOW_INVISIBLE)
  59         GemRB.LoadWindowPack ("GUIMG")
  60         MageWindow = Window = GemRB.LoadWindow (2)
  61         GemRB.SetVar ("OtherWindow", MageWindow.ID)
  62         OldOptionsWindow = GUICommonWindows.OptionsWindow
  63         OptionsWindow = GemRB.LoadWindow (0)
  64         GUICommonWindows.SetupMenuWindowControls (OptionsWindow, 0, OpenMageWindow)
  65         OptionsWindow.SetFrame ()
  66
  67         Button = Window.GetControl (1)
  68         Button.SetEvent (IE_GUI_BUTTON_ON_PRESS, MagePrevLevelPress)
  69
  70         Button = Window.GetControl (2)
  71         Button.SetEvent (IE_GUI_BUTTON_ON_PRESS, MageNextLevelPress)
  72
  73 ##      #unknown usage
  74 ##      Button = Window.GetControl (55)
  75 ##      Button.SetState (IE_GUI_BUTTON_LOCKED)
  76 ##      #Button.SetText (123)
  77 ##      #Button.SetEvent (IE_GUI_BUTTON_ON_PRESS, xxPress)
  78
  79 ##      #setup level buttons
  80 ##      for i in range (9):
  81 ##              Button = Window.GetControl (56 + i)
  82 ##              Button.SetEvent (IE_GUI_BUTTON_ON_PRESS, RefreshMageLevel)
  83 ##              Button.SetFlags (IE_GUI_BUTTON_RADIOBUTTON, OP_OR)
  84
  85 ##      for i in range (9):
  86 ##              Button = Window.GetControl (56 + i)
  87 ##              Button.SetVarAssoc ("MageSpellLevel", i)
  88
  89         # Setup memorized spells buttons
  90         for i in range (12):
  91                 Button = Window.GetControl (3 + i)
  92                 Button.SetBorder (0,0,0,0,0,0,0,0,64,0,1)
  93                 Button.SetSprites ("SPELFRAM",0,0,0,0,0)
  94                 Button.SetFlags (IE_GUI_BUTTON_PICTURE, OP_OR)
  95                 Button.SetState (IE_GUI_BUTTON_LOCKED)
  96
  97         # Setup book spells buttons
  98         for i in range (20):
  99                 Button = Window.GetControl (27 + i)
 100                 Button.SetFlags (IE_GUI_BUTTON_NO_IMAGE, OP_OR)
 101                 Button.SetState (IE_GUI_BUTTON_LOCKED)
 102
 103         GUICommonWindows.SetSelectionChangeHandler (UpdateMageWindow)
 104         UpdateMageWindow ()
 105
 106         OptionsWindow.SetVisible (WINDOW_VISIBLE)
 107         Window.SetVisible (WINDOW_FRONT)
 108         GUICommonWindows.PortraitWindow.SetVisible (WINDOW_VISIBLE)
 109         return
 110
 111
 112 def UpdateMageWindow ():
 113         global MageMemorizedSpellList, MageKnownSpellList
 114
 115         MageMemorizedSpellList = []
 116         MageKnownSpellList = []
 117
 118         Window = MageWindow
 119         pc = GemRB.GameGetSelectedPCSingle ()
 120         type = IE_SPELL_TYPE_WIZARD
 121         level = MageSpellLevel
 122         max_mem_cnt = GemRB.GetMemorizableSpellsCount (pc, type, level)
 123
 124         Label = Window.GetControl (0x10000032)
 125         GemRB.SetToken ('LEVEL', str (level + 1))
 126         Label.SetText (12137 )
 127
 128         Name = GemRB.GetPlayerName (pc, 0)
 129         Label = Window.GetControl (0x10000035)
 130         Label.SetText (Name)
 131
 132         mem_cnt = GemRB.GetMemorizedSpellsCount (pc, type, level)
 133         for i in range (12):
 134                 Button = Window.GetControl (3 + i)
 135                 if i < mem_cnt:
 136                         ms = GemRB.GetMemorizedSpell (pc, type, level, i)
 137                         Button.SetSpellIcon (ms['SpellResRef'], 0)
 138                         Button.SetFlags (IE_GUI_BUTTON_NO_IMAGE, OP_NAND)
 139                         Button.SetFlags (IE_GUI_BUTTON_PICTURE, OP_OR)
 140                         if ms['Flags']:
 141                                 Button.SetEvent (IE_GUI_BUTTON_ON_PRESS, OpenMageSpellUnmemorizeWindow)
 142                         else:
 143                                 Button.SetEvent (IE_GUI_BUTTON_ON_PRESS, OnMageUnmemorizeSpell)
 144                         Button.SetEvent (IE_GUI_BUTTON_ON_RIGHT_PRESS, OpenMageSpellInfoWindow)
 145                         spell = GemRB.GetSpell (ms['SpellResRef'])
 146                         Button.SetTooltip (spell['SpellName'])
 147                         MageMemorizedSpellList.append (ms['SpellResRef'])
 148                         Button.SetVarAssoc ("SpellButton", i)
 149                         Button.EnableBorder (0, ms['Flags'] == 0)
 150                 else:
 151                         if i < max_mem_cnt:
 152                                 Button.SetFlags (IE_GUI_BUTTON_NORMAL, OP_SET)
 153                         else:
 154                                 Button.SetFlags (IE_GUI_BUTTON_NO_IMAGE, OP_SET)
 155                         Button.SetEvent (IE_GUI_BUTTON_ON_PRESS, None)
 156                         Button.SetEvent (IE_GUI_BUTTON_ON_RIGHT_PRESS, None)
 157                         Button.SetTooltip ('')
 158                         Button.EnableBorder (0, 0)
 159
 160
 161         known_cnt = GemRB.GetKnownSpellsCount (pc, type, level)
 162         for i in range (20):
 163                 Button = Window.GetControl (27 + i)
 164                 if i < known_cnt:
 165                         ks = GemRB.GetKnownSpell (pc, type, level, i)
 166                         Button.SetSpellIcon (ks['SpellResRef'], 0)
 167                         Button.SetEvent (IE_GUI_BUTTON_ON_PRESS, OnMageMemorizeSpell)
 168                         Button.SetEvent (IE_GUI_BUTTON_ON_RIGHT_PRESS, OpenMageSpellInfoWindow)
 169                         spell = GemRB.GetSpell (ks['SpellResRef'])
 170                         Button.SetTooltip (spell['SpellName'])
 171                         MageKnownSpellList.append (ks['SpellResRef'])
 172                         Button.SetVarAssoc ("SpellButton", 100 + i)
 173
 174                 else:
 175                         Button.SetFlags (IE_GUI_BUTTON_PICTURE, OP_NAND)
 176                         Button.SetEvent (IE_GUI_BUTTON_ON_PRESS, None)
 177                         Button.SetEvent (IE_GUI_BUTTON_ON_RIGHT_PRESS, None)
 178                         Button.SetTooltip ('')
 179                         Button.EnableBorder (0, 0)
 180
 181         if (CommonTables.ClassSkills.GetValue (GemRB.GetPlayerStat (pc, IE_CLASS), 2)=="*") or \
 182         GemRB.GetPlayerStat (pc, IE_STATE_ID) & STATE_DEAD:
 183                 Window.SetVisible (WINDOW_GRAYED)
 184         else:
 185                 Window.SetVisible (WINDOW_VISIBLE)
 186         return
 187
 188 def MagePrevLevelPress ():
 189         global MageSpellLevel
 190
 191         if MageSpellLevel > 0:
 192                 MageSpellLevel = MageSpellLevel - 1
 193                 UpdateMageWindow ()
 194         return
 195
 196 def MageNextLevelPress ():
 197         global MageSpellLevel
 198
 199         if MageSpellLevel < 8:
 200                 MageSpellLevel = MageSpellLevel + 1
 201                 UpdateMageWindow ()
 202         return
 203
 204 def RefreshMageLevel ():
 205         global MageSpellLevel
 206
 207         MageSpellLevel = GemRB.GetVar ("MageSpellLevel")
 208         UpdateMageWindow ()
 209         return
 210
 211 def OpenMageSpellInfoWindow ():
 212         global MageSpellInfoWindow
 213
 214         if MageSpellInfoWindow != None:
 215                 if MageSpellInfoWindow:
 216                         MageSpellInfoWindow.Unload ()
 217                 MageSpellInfoWindow = None
 218                 return
 219
 220         MageSpellInfoWindow = Window = GemRB.LoadWindow (3)
 221
 222         #back
 223         Button = Window.GetControl (5)
 224         Button.SetText (15416)
 225         Button.SetEvent (IE_GUI_BUTTON_ON_PRESS, OpenMageSpellInfoWindow)
 226
 227         #erase
 228         #Button = Window.GetControl (6)
 229         index = GemRB.GetVar ("SpellButton")
 230         if index < 100:
 231                 ResRef = MageMemorizedSpellList[index]
 232         else:
 233                 ResRef = MageKnownSpellList[index - 100]
 234
 235         spell = GemRB.GetSpell (ResRef)
 236
 237         Label = Window.GetControl (0x0fffffff)
 238         Label.SetText (spell['SpellName'])
 239
 240         Button = Window.GetControl (2)
 241         Button.SetSpellIcon (ResRef, 1)
 242
 243         Text = Window.GetControl (3)
 244         Text.SetText (spell['SpellDesc'])
 245
 246         Window.ShowModal (MODAL_SHADOW_GRAY)
 247         return
 248
 249 def OnMageMemorizeSpell ():
 250         pc = GemRB.GameGetSelectedPCSingle ()
 251         level = MageSpellLevel
 252         type = IE_SPELL_TYPE_WIZARD
 253
 254         index = GemRB.GetVar ("SpellButton") - 100
 255
 256         if GemRB.MemorizeSpell (pc, type, level, index):
 257                 UpdateMageWindow ()
 258         return
 259
 260 def CloseMageSpellUnmemorizeWindow ():
 261         global MageSpellUnmemorizeWindow
 262
 263         if MageSpellUnmemorizeWindow:
 264                 MageSpellUnmemorizeWindow.Unload ()
 265         MageSpellUnmemorizeWindow = None
 266         return
 267
 268 #def OpenMageSpellRemoveWindow ():
 269 #       global MageSpellUnmemorizeWindow
 270 #
 271 #       MageSpellUnmemorizeWindow = Window = GemRB.LoadWindow (5)
 272 #
 273 #       # "Are you sure you want to ....?"
 274 #       TextArea = Window.GetControl (3)
 275 #       TextArea.SetText (63745)
 276 #
 277 #       # Remove
 278 #       Button = Window.GetControl (0)
 279 #       Button.SetText (17507)
 280 #       Button.SetEvent (IE_GUI_BUTTON_ON_PRESS, OnMageRemoveSpell)
 281 #       Button.SetFlags (IE_GUI_BUTTON_DEFAULT, OP_OR)
 282 #
 283 #       # Cancel
 284 #       Button = Window.GetControl (1)
 285 #       Button.SetText (13727)
 286 #       Button.SetEvent (IE_GUI_BUTTON_ON_PRESS, CloseMageSpellUnmemorizeWindow)
 287 #       Button.SetFlags (IE_GUI_BUTTON_CANCEL, OP_OR)
 288 #
 289 #       Window.ShowModal (MODAL_SHADOW_GRAY)
 290 #       return
 291
 292 def OpenMageSpellUnmemorizeWindow ():
 293         global MageSpellUnmemorizeWindow
 294
 295         MageSpellUnmemorizeWindow = Window = GemRB.LoadWindow (5)
 296
 297         # "Are you sure you want to ....?"
 298         TextArea = Window.GetControl (3)
 299         TextArea.SetText (11824)
 300
 301         # Remove
 302         Button = Window.GetControl (0)
 303         Button.SetText (17507)
 304         Button.SetEvent (IE_GUI_BUTTON_ON_PRESS, OnMageUnmemorizeSpell)
 305         Button.SetFlags (IE_GUI_BUTTON_DEFAULT, OP_OR)
 306
 307         # Cancel
 308         Button = Window.GetControl (1)
 309         Button.SetText (13727)
 310         Button.SetEvent (IE_GUI_BUTTON_ON_PRESS, CloseMageSpellUnmemorizeWindow)
 311         Button.SetFlags (IE_GUI_BUTTON_CANCEL, OP_OR)
 312
 313         Window.ShowModal (MODAL_SHADOW_GRAY)
 314         return
 315
 316 def OnMageUnmemorizeSpell ():
 317         if MageSpellUnmemorizeWindow:
 318                 CloseMageSpellUnmemorizeWindow()
 319
 320         pc = GemRB.GameGetSelectedPCSingle ()
 321         level = MageSpellLevel
 322         type = IE_SPELL_TYPE_WIZARD
 323
 324         index = GemRB.GetVar ("SpellButton")
 325
 326         if GemRB.UnmemorizeSpell (pc, type, level, index):
 327                 UpdateMageWindow ()
 328         return
 329
 330 #def OnMageRemoveSpell ():
 331 #       CloseMageSpellUnmemorizeWindow()
 332 #       OpenMageSpellInfoWindow()
 333 #
 334 #       pc = GemRB.GameGetSelectedPCSingle ()
 335 #       level = MageSpellLevel
 336 #       type = IE_SPELL_TYPE_WIZARD
 337 #
 338 #       index = GemRB.GetVar ("SpellButton")-100
 339 #
 340 #       #remove spell from book
 341 #       GemRB.RemoveSpell (pc, type, level, index)
 342 #       UpdateMageWindow ()
 343 #       return
 344
 345 ###################################################
 346 # End of file GUIMG.py