gemrb/GUIScripts/bg2/GUIWORLD.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 # GUIW.py - scripts to control some windows from the GUIWORLD winpack
  22 # except of Actions, Portrait, Options and Dialog windows
  23 #################################################################
  24
  25 import GemRB
  26 from GUIDefines import *
  27 import GUICommonWindows
  28 import GUICommon
  29 from GUIClasses import GWindow
  30
  31 FRAME_PC_SELECTED = 0
  32 FRAME_PC_TARGET   = 1
  33
  34 ContainerWindow = None
  35 ContinueWindow = None
  36 ReformPartyWindow = None
  37 OldActionsWindow = None
  38 OldMessageWindow = None
  39 Container = None
  40
  41 def CloseContinueWindow ():
  42         if ContinueWindow:
  43                 # don't close the actual window now to avoid flickering: we might still want it open
  44                 GemRB.SetVar ("DialogChoose", GemRB.GetVar ("DialogOption"))
  45
  46 def NextDialogState ():
  47         global ContinueWindow, OldActionsWindow
  48
  49         if ContinueWindow == None:
  50                 return
  51
  52         hideflag = GemRB.HideGUI ()
  53
  54         if ContinueWindow:
  55                 ContinueWindow.Unload ()
  56         GemRB.SetVar ("ActionsWindow", OldActionsWindow.ID)
  57         ContinueWindow = None
  58         OldActionsWindow = None
  59         if hideflag:
  60                 GemRB.UnhideGUI ()
  61
  62
  63 def OpenEndMessageWindow ():
  64         global ContinueWindow, OldActionsWindow
  65
  66         hideflag = GemRB.HideGUI ()
  67
  68         if not ContinueWindow:
  69                 GemRB.LoadWindowPack (GUICommon.GetWindowPack())
  70                 ContinueWindow = Window = GemRB.LoadWindow (9)
  71                 OldActionsWindow = GWindow( GemRB.GetVar ("ActionsWindow") )
  72                 GemRB.SetVar ("ActionsWindow", Window.ID)
  73
  74         #end dialog
  75         Button = ContinueWindow.GetControl (0)
  76         Button.SetText (9371)
  77         Button.SetEventByName (IE_GUI_BUTTON_ON_PRESS, "CloseContinueWindow")
  78         if GUICommonWindows.PortraitWindow:
  79                 GUICommonWindows.UpdatePortraitWindow ()
  80         if hideflag:
  81                 GemRB.UnhideGUI ()
  82
  83
  84 def OpenContinueMessageWindow ():
  85         global ContinueWindow, OldActionsWindow
  86
  87         hideflag = GemRB.HideGUI ()
  88
  89         if not ContinueWindow:
  90                 GemRB.LoadWindowPack (GUICommon.GetWindowPack())
  91                 ContinueWindow = Window = GemRB.LoadWindow (9)
  92                 OldActionsWindow = GWindow( GemRB.GetVar ("ActionsWindow") )
  93                 GemRB.SetVar ("ActionsWindow", Window.ID)
  94
  95         #continue
  96         Button = ContinueWindow.GetControl (0)
  97         Button.SetText (9372)
  98         Button.SetEventByName (IE_GUI_BUTTON_ON_PRESS, "CloseContinueWindow")
  99         if hideflag:
 100                 GemRB.UnhideGUI ()
 101
 102
 103 def CloseContainerWindow ():
 104         global OldActionsWindow, OldMessageWindow, ContainerWindow
 105
 106         if ContainerWindow == None:
 107                 return
 108
 109         hideflag = GemRB.HideGUI ()
 110
 111         if ContainerWindow:
 112                 ContainerWindow.Unload ()
 113         ContainerWindow = None
 114         GemRB.SetVar ("ActionsWindow", OldActionsWindow.ID)
 115         GemRB.SetVar ("MessageWindow", OldMessageWindow.ID)
 116         Table = GemRB.LoadTable ("containr")
 117         row = Container['Type']
 118         tmp = Table.GetValue (row, 2)
 119         #play closing sound if applicable
 120         if tmp!='*':
 121                 GemRB.PlaySound (tmp)
 122
 123         #it is enough to close here
 124
 125         if hideflag:
 126                 GemRB.UnhideGUI ()
 127
 128
 129 def UpdateContainerWindow ():
 130         global Container
 131
 132         Window = ContainerWindow
 133
 134         pc = GemRB.GameGetFirstSelectedPC ()
 135         SetEncumbranceLabels( Window, 0x10000043, 0x10000044, pc)
 136
 137         party_gold = GemRB.GameGetPartyGold ()
 138         Text = Window.GetControl (0x10000036)
 139         Text.SetText (str (party_gold))
 140
 141         Container = GemRB.GetContainer(0) #will use first selected pc anyway
 142         LeftCount = Container['ItemCount']
 143         ScrollBar = Window.GetControl (52)
 144         Count = LeftCount/3
 145         if Count<1:
 146                 Count=1
 147         ScrollBar.SetVarAssoc ("LeftTopIndex", Count)
 148
 149         inventory_slots = GemRB.GetSlots (pc, 0x8000)
 150         RightCount = len(inventory_slots)
 151         ScrollBar = Window.GetControl (53)
 152         Count = RightCount/2
 153         if Count<1:
 154                 Count=1
 155         ScrollBar.SetVarAssoc ("RightTopIndex", Count)
 156
 157         RedrawContainerWindow ()
 158
 159
 160 def RedrawContainerWindow ():
 161         Window = ContainerWindow
 162
 163         LeftTopIndex = GemRB.GetVar ("LeftTopIndex") * 3
 164         LeftIndex = GemRB.GetVar ("LeftIndex")
 165         RightTopIndex = GemRB.GetVar ("RightTopIndex") * 2
 166         RightIndex = GemRB.GetVar ("RightIndex")
 167         LeftCount = Container['ItemCount']
 168         pc = GemRB.GameGetFirstSelectedPC ()
 169         inventory_slots = GemRB.GetSlots (pc, 0x8000)
 170         RightCount = len(inventory_slots)
 171
 172         for i in range (6):
 173                 #this is an autoselected container, but we could use PC too
 174                 Slot = GemRB.GetContainerItem (0, i+LeftTopIndex)
 175                 Button = Window.GetControl (i)
 176                 if Slot != None:
 177                         Button.SetVarAssoc ("LeftIndex", LeftTopIndex+i)
 178                 else:
 179                         Button.SetVarAssoc ("LeftIndex", -1)
 180                 GUICommon.UpdateInventorySlot (pc, Button, Slot, "container")
 181
 182         for i in range (4):
 183                 if i+RightTopIndex<RightCount:
 184                         Slot = GemRB.GetSlotItem (pc, inventory_slots[i+RightTopIndex])
 185                 else:
 186                         Slot = None
 187                 Button = Window.GetControl (i+10)
 188                 if Slot!=None:
 189                         Button.SetVarAssoc ("RightIndex", RightTopIndex+i)
 190                 else:
 191                         Button.SetVarAssoc ("RightIndex", -1)
 192                 GUICommon.UpdateInventorySlot (pc, Button, Slot, "inventory")
 193
 194
 195 def OpenContainerWindow ():
 196         global OldActionsWindow, OldMessageWindow
 197         global ContainerWindow, Container
 198
 199         if ContainerWindow:
 200                 return
 201
 202         hideflag = GemRB.HideGUI ()
 203
 204         GemRB.LoadWindowPack (GUICommon.GetWindowPack())
 205         ContainerWindow = Window = GemRB.LoadWindow (8)
 206         OldActionsWindow = GWindow( GemRB.GetVar ("ActionsWindow") )
 207         OldMessageWindow = GWindow( GemRB.GetVar ("MessageWindow") )
 208         GemRB.SetVar ("ActionsWindow", Window.ID)
 209         GemRB.SetVar ("MessageWindow", -1)
 210
 211         Container = GemRB.GetContainer(0)
 212
 213         # Gears (time) when options pane is down
 214         Button = Window.GetControl (62)
 215         Label = Button.CreateLabelOnButton (0x1000003e, "NORMAL", 0)
 216
 217         Label.SetAnimation ("CPEN")
 218         Button.SetAnimation ("CGEAR")
 219         Button.SetBAM ("CDIAL", 0, 0)
 220         Button.SetState (IE_GUI_BUTTON_ENABLED)
 221         Button.SetFlags (IE_GUI_BUTTON_PICTURE|IE_GUI_BUTTON_ANIMATED|IE_GUI_BUTTON_NORMAL, OP_SET)
 222         Button.SetEventByName(IE_GUI_BUTTON_ON_PRESS, "GearsClicked")
 223         GUICommon.SetGamedaysAndHourToken()
 224         Button.SetTooltip(16041)
 225
 226         # 0 - 5 - Ground Item
 227         # 10 - 13 - Personal Item
 228         # 50 hand
 229         # 52, 53 scroller ground, scroller personal
 230         # 54 - encumbrance
 231
 232         for i in range (6):
 233                 Button = Window.GetControl (i)
 234                 Button.SetVarAssoc ("LeftIndex", i)
 235                 #Button.SetFlags (IE_GUI_BUTTON_CHECKBOX, OP_OR)
 236                 Button.SetEventByName (IE_GUI_BUTTON_ON_PRESS, "TakeItemContainer")
 237
 238         for i in range (4):
 239                 Button = Window.GetControl (i+10)
 240                 Button.SetVarAssoc ("RightIndex", i)
 241                 #Button.SetFlags (IE_GUI_BUTTON_CHECKBOX, OP_OR)
 242                 Button.SetEventByName (IE_GUI_BUTTON_ON_PRESS, "DropItemContainer")
 243
 244         # left scrollbar
 245         ScrollBar = Window.GetControl (52)
 246         ScrollBar.SetEventByName (IE_GUI_SCROLLBAR_ON_CHANGE, "RedrawContainerWindow")
 247
 248         # right scrollbar
 249         ScrollBar = Window.GetControl (53)
 250         ScrollBar.SetEventByName (IE_GUI_SCROLLBAR_ON_CHANGE, "RedrawContainerWindow")
 251
 252         Label = Window.CreateLabel (0x10000043, 323,14,60,15,"NUMBER","0:",IE_FONT_ALIGN_LEFT|IE_FONT_ALIGN_TOP)
 253         Label = Window.CreateLabel (0x10000044, 323,20,80,15,"NUMBER","0:",IE_FONT_ALIGN_RIGHT|IE_FONT_ALIGN_TOP)
 254
 255         Button = Window.GetControl (50)
 256         Button.SetState (IE_GUI_BUTTON_LOCKED)
 257         Table = GemRB.LoadTable ("containr")
 258         row = Container['Type']
 259         tmp = Table.GetValue (row, 0)
 260         if tmp!='*':
 261                 GemRB.PlaySound (tmp)
 262         tmp = Table.GetValue (row, 1)
 263         if tmp!='*':
 264                 Button.SetSprites (tmp, 0, 0, 0, 0, 0 )
 265
 266         # Done
 267         Button = Window.GetControl (51)
 268         Button.SetText (1403)
 269         Button.SetEventByName (IE_GUI_BUTTON_ON_PRESS, "LeaveContainer")
 270
 271         GemRB.SetVar ("LeftTopIndex", 0)
 272         GemRB.SetVar ("RightTopIndex", 0)
 273         UpdateContainerWindow ()
 274         if hideflag:
 275                 GemRB.UnhideGUI ()
 276
 277
 278 #doing this way it will inform the core system too, which in turn will call
 279 #CloseContainerWindow ()
 280 def LeaveContainer ():
 281         GemRB.LeaveContainer()
 282
 283 def DropItemContainer ():
 284         RightIndex = GemRB.GetVar ("RightIndex")
 285         if RightIndex<0:
 286                 return
 287
 288         #we need to get the right slot number
 289         pc = GemRB.GameGetFirstSelectedPC ()
 290         inventory_slots = GemRB.GetSlots (pc, 0x8000)
 291         if RightIndex >= len(inventory_slots):
 292                 return
 293         GemRB.ChangeContainerItem (0, inventory_slots[RightIndex], 0)
 294         UpdateContainerWindow ()
 295
 296
 297 def TakeItemContainer ():
 298         LeftIndex = GemRB.GetVar ("LeftIndex")
 299         if LeftIndex<0:
 300                 return
 301
 302         if LeftIndex >= Container['ItemCount']:
 303                 return
 304         GemRB.ChangeContainerItem (0, LeftIndex, 1)
 305         UpdateContainerWindow ()
 306
 307
 308 def UpdateReformWindow ():
 309         Window = ReformPartyWindow
 310
 311         select = GemRB.GetVar ("Selected")
 312
 313         need_to_drop = GemRB.GetPartySize ()-PARTY_SIZE
 314         if need_to_drop<0:
 315                 need_to_drop = 0
 316
 317         #excess player number
 318         Label = Window.GetControl (0x1000000f)
 319         Label.SetText (str(need_to_drop) )
 320
 321         #done
 322         Button = Window.GetControl (8)
 323         if need_to_drop:
 324                 Button.SetState (IE_GUI_BUTTON_DISABLED)
 325         else:
 326                 Button.SetState (IE_GUI_BUTTON_ENABLED)
 327
 328         #remove
 329         Button = Window.GetControl (15)
 330         if select:
 331                 Button.SetState (IE_GUI_BUTTON_ENABLED)
 332         else:
 333                 Button.SetState (IE_GUI_BUTTON_DISABLED)
 334
 335         for i in range (PARTY_SIZE+1):
 336                 Button = Window.GetControl (i)
 337                 Button.EnableBorder (FRAME_PC_SELECTED, select == i+2 )
 338                 #+2 because protagonist is skipped
 339                 pic = GemRB.GetPlayerPortrait (i+2,1)
 340                 if not pic:
 341                         Button.SetFlags (IE_GUI_BUTTON_NO_IMAGE, OP_SET)
 342                         Button.SetState (IE_GUI_BUTTON_LOCKED)
 343                         continue
 344
 345                 Button.SetState (IE_GUI_BUTTON_ENABLED)
 346                 Button.SetFlags (IE_GUI_BUTTON_PICTURE|IE_GUI_BUTTON_ALIGN_BOTTOM|IE_GUI_BUTTON_ALIGN_LEFT, OP_SET)
 347                 Button.SetPicture (pic, "NOPORTSM")
 348         GUICommonWindows.UpdatePortraitWindow ()
 349         return
 350
 351 def RemovePlayer ():
 352         global ReformPartyWindow
 353
 354         hideflag = GemRB.HideGUI ()
 355
 356         GemRB.LoadWindowPack (GUICommon.GetWindowPack())
 357         if ReformPartyWindow:
 358                 ReformPartyWindow.Unload ()
 359         ReformPartyWindow = Window = GemRB.LoadWindow (25)
 360         GemRB.SetVar ("OtherWindow", Window.ID)
 361
 362         #are you sure
 363         Label = Window.GetControl (0x0fffffff)
 364         Label.SetText (17518)
 365
 366         #confirm
 367         Button = Window.GetControl (1)
 368         Button.SetText (17507)
 369         Button.SetEventByName (IE_GUI_BUTTON_ON_PRESS, "RemovePlayerConfirm")
 370         Button.SetFlags (IE_GUI_BUTTON_DEFAULT, OP_OR)
 371
 372         #cancel
 373         Button = Window.GetControl (2)
 374         Button.SetText (13727)
 375         Button.SetEventByName (IE_GUI_BUTTON_ON_PRESS, "RemovePlayerCancel")
 376         Button.SetFlags (IE_GUI_BUTTON_CANCEL, OP_OR)
 377
 378         GemRB.SetVar ("OtherWindow", Window.ID)
 379         GemRB.SetVar ("ActionsWindow", -1)
 380         if hideflag:
 381                 GemRB.UnhideGUI ()
 382         Window.ShowModal (MODAL_SHADOW_GRAY)
 383         return
 384
 385 def RemovePlayerConfirm ():
 386         slot = GemRB.GetVar ("Selected")
 387         GemRB.LeaveParty (slot, 2)
 388         OpenReformPartyWindow ()
 389         return
 390
 391 def RemovePlayerCancel ():
 392         #Once for getting rid of the confirmation window
 393         OpenReformPartyWindow ()
 394         #and once for reopening the reform party window
 395         OpenReformPartyWindow ()
 396         return
 397
 398 def OpenReformPartyWindow ():
 399         global ReformPartyWindow, OldActionsWindow, OldMessageWindow
 400
 401         GemRB.SetVar ("Selected", 0)
 402         hideflag = GemRB.HideGUI ()
 403
 404         if ReformPartyWindow:
 405                 if ReformPartyWindow:
 406                         ReformPartyWindow.Unload ()
 407
 408                 GemRB.SetVar ("ActionsWindow", OldActionsWindow.ID)
 409                 GemRB.SetVar ("MessageWindow", OldMessageWindow.ID)
 410                 GemRB.SetVar ("OtherWindow", -1)
 411
 412                 OldActionsWindow = None
 413                 OldMessageWindow = None
 414                 ReformPartyWindow = None
 415                 if hideflag:
 416                         GemRB.UnhideGUI ()
 417                 #re-enabling party size control
 418                 GemRB.GameSetPartySize (PARTY_SIZE)
 419                 GUICommonWindows.UpdatePortraitWindow()
 420                 return
 421
 422         GemRB.LoadWindowPack (GUICommon.GetWindowPack())
 423         ReformPartyWindow = Window = GemRB.LoadWindow (24)
 424         GemRB.SetVar ("OtherWindow", Window.ID)
 425
 426         #PC portraits
 427         for j in range (PARTY_SIZE+1):
 428                 Button = Window.GetControl (j)
 429                 Button.SetState (IE_GUI_BUTTON_LOCKED)
 430                 Button.SetFlags (IE_GUI_BUTTON_RADIOBUTTON|IE_GUI_BUTTON_NO_IMAGE|IE_GUI_BUTTON_PICTURE,OP_SET)
 431                 Button.SetBorder (FRAME_PC_SELECTED, 1, 1, 2, 2, 0, 255, 0, 255)
 432                 #protagonist is skipped
 433                 index = j + 2
 434                 Button.SetVarAssoc ("Selected", index)
 435                 Button.SetEventByName (IE_GUI_BUTTON_ON_PRESS, "UpdateReformWindow")
 436
 437         # Remove
 438         Button = Window.GetControl (15)
 439         Button.SetText (17507)
 440         Button.SetEventByName (IE_GUI_BUTTON_ON_PRESS, "RemovePlayer")
 441
 442         # Done
 443         Button = Window.GetControl (8)
 444         Button.SetText (11973)
 445         Button.SetEventByName (IE_GUI_BUTTON_ON_PRESS, "OpenReformPartyWindow")
 446
 447         OldActionsWindow = GWindow( GemRB.GetVar ("ActionsWindow") )
 448         OldMessageWindow = GWindow( GemRB.GetVar ("MessageWindow") )
 449         GemRB.SetVar ("ActionsWindow", -1)
 450         GemRB.SetVar ("MessageWindow", -1)
 451         UpdateReformWindow ()
 452         if hideflag:
 453                 GemRB.UnhideGUI ()
 454         Window.ShowModal (MODAL_SHADOW_GRAY)
 455         return
 456
 457 def DeathWindow ():
 458         GemRB.HideGUI ()
 459         GemRB.SetTimedEventByName ("DeathWindowEnd", 10)
 460         return
 461
 462 def DeathWindowEnd ():
 463         #playing death movie before continuing
 464         GemRB.PlayMovie ("deathand",1)
 465         GemRB.GamePause (1,1)
 466
 467         GemRB.LoadWindowPack (GUICommon.GetWindowPack())
 468         Window = GemRB.LoadWindow (17)
 469
 470         #reason for death
 471         Label = Window.GetControl (0x0fffffff)
 472         Label.SetText (16498)
 473
 474         #load
 475         Button = Window.GetControl (1)
 476         Button.SetText (15590)
 477         Button.SetEventByName (IE_GUI_BUTTON_ON_PRESS, "LoadPress")
 478
 479         #quit
 480         Button = Window.GetControl (2)
 481         Button.SetText (15417)
 482         Button.SetEventByName (IE_GUI_BUTTON_ON_PRESS, "QuitPress")
 483
 484         GemRB.HideGUI ()
 485         GemRB.SetVar ("MessageWindow", -1)
 486         GemRB.UnhideGUI ()
 487         Window.ShowModal (MODAL_SHADOW_GRAY)
 488         return
 489
 490 def QuitPress():
 491         GemRB.QuitGame ()
 492         GemRB.SetNextScript ("Start")
 493         return
 494
 495 def LoadPress():
 496         GemRB.QuitGame ()
 497         GemRB.SetNextScript ("GUILOAD")
 498         return
 499