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