class UnitBox():
"""
One of the tiny boxes displayed when a unit is selected
@param pos: position of UnitBox on screen
@type pos:tuple(int, int)
@param baselayer: background image
@type baselayer: DrawableObjectGroup
@param entity: entity displayed
@type entity: Entity
@param thumbnailOffset: position of thumbnail image relative to UnitBox
@type thumbnailOffset: tuple(int, int)
@param thumbnail: image of entity displayed
@type thumbnail: pygame.Surface
@param healthBar: shows health of entity
@type healthBar: Bar
"""
def __init__(self,entity,pos=(0,0),endCap=False):
import pygame #FIXME Remove this dependency - replace image with thumbnail
from Overlay import Bar
#Format for each tuple is (imagePath, [colorKey, [offset]])
images = ["UnitInfoBoxMain.png"]
if endCap:
images.append(("UnitInfoBoxRight.png",'alpha',(46,0)))
self.pos = pos
self.baseLayer = DrawableObjectGroup(images,pos=self.pos)
self.entity = entity
self.thumbnailOffset = (6,10)
self.thumbnail = pygame.transform.scale(self.entity.image, (36, 36))
self.healthBar = Bar(self.entity.maxHealth,38,5,fullColor=(0,255,0),emptyColor=(30,30,30))
def draw(self,screen):
"""updates and draws unitBox to screen"""
self.healthBar.updateBarWithValue(self.entity.curHealth)
self.baseLayer.draw(screen)
screen.blit(self.thumbnail, (self.pos[0]+self.thumbnailOffset[0],self.pos[1]+self.thumbnailOffset[1]))
self.healthBar.draw(screen,(self.pos[0]+4,self.pos[1]+50))
def updateDisplayedEntity(self,entity):
"""
Changes entity displayed
"""
import pygame
from Overlay import Bar
self.entity = entity
self.thumbnail = self.entity.getDefaultImage()
self.thumbnail = pygame.transform.scale(self.thumbnail, (44, 44))
self.description.clear()
self.description.addtext(entity.description)
self.healthBar = Bar(self.entity.maxHealth,118,6,fullColor=(0,255,0),emptyColor=(30,30,30))
def __init__(self,entity,pos=(0,0),endCap=False):
import pygame #FIXME Remove this dependency - replace image with thumbnail
from Overlay import Bar
#Format for each tuple is (imagePath, [colorKey, [offset]])
images = ["UnitInfoBoxMain.png"]
if endCap:
images.append(("UnitInfoBoxRight.png",'alpha',(46,0)))
self.pos = pos
self.baseLayer = DrawableObjectGroup(images,pos=self.pos)
self.entity = entity
self.thumbnailOffset = (6,10)
self.thumbnail = pygame.transform.scale(self.entity.image, (36, 36))
self.healthBar = Bar(self.entity.maxHealth,38,5,fullColor=(0,255,0),emptyColor=(30,30,30))
def draw(self,screen):
self.baseLayer.draw(screen)
if self.entity:
self.healthBar.updateBarWithValue(self.entity.curHealth)
#if entity is building something, displays
#building status
if hasattr(self.entity,'currentTask'):
if self.entity.currentTask:
if hasattr(self.entity.currentTask,'timeSpentBuilding'):
from Overlay import Bar
self.buildBar = Bar(self.entity.currentTask.timeToBuild,118,6,fullColor=(0,180,255),emptyColor=(30,30,30))
self.buildBar.updateBarWithValue(self.entity.currentTask.timeSpentBuilding)
self.buildBar.draw(screen,(self.pos[0]+78,self.pos[1]+69))
screen.blit(self.thumbnail, (self.pos[0]+self.thumbnailOffset[0],self.pos[1]+self.thumbnailOffset[1]))
self.healthBar.draw(screen,(self.pos[0]+78,self.pos[1]+56))
self.description.render(screen)
class DescriptionBox():
"""
The largest HUD element which describes the hovered unit
and its properties (those which are player-relevant)
@param baseLayer: background image
@type baseLayer: DrawableObjectGroup(list(tuple(str, str)), tuple(int, int))
@param entity: Entity whose information is being displayed
@type entity: Entity
@param descriptionOffset: position of description of entity relative to DescriptionBox
@type descriptionOffset: tuple(int, int)
@param description: Text description of entity
@type descritpion: Sign
@param pos: position of DescriptionBox on screen
@type pos: tuple(int, int)
@param thumbnailOffset: position of thumbnail relative to DescriptionBox
@type thumbnailOffset: tuple(int, int)
"""
def __init__(self,pos = (0,768-36-186)):
#FIXME Remove this dependency - replace image with thumbnail
from Overlay import Bar
#Format for each tuple is (imagePath, [colorKey, [offset]])
images = [("BarTop.png",'alpha')]
images.append(("BarRight.png",'alpha',(325,36)))
images.append(("DescBoxCentral.png",None,(0,36)))
self.baseLayer = DrawableObjectGroup(images,pos=pos)
self.entity = None
self.descriptionOffset = (35,73+36)
self.description = Sign(260, (pos[0]+self.descriptionOffset[0],pos[1]+self.descriptionOffset[1]),fsize = 18)
self.description.tcolor=(180,180,0)
self.pos = pos
self.thumbnailOffset = (12,14+36)
def updateDisplayedEntity(self,entity):
"""
Changes entity displayed
"""
import pygame
from Overlay import Bar
self.entity = entity
self.thumbnail = self.entity.getDefaultImage()
self.thumbnail = pygame.transform.scale(self.thumbnail, (44, 44))
self.description.clear()
self.description.addtext(entity.description)
self.healthBar = Bar(self.entity.maxHealth,118,6,fullColor=(0,255,0),emptyColor=(30,30,30))
def draw(self,screen):
self.baseLayer.draw(screen)
if self.entity:
self.healthBar.updateBarWithValue(self.entity.curHealth)
#if entity is building something, displays
#building status
if hasattr(self.entity,'currentTask'):
if self.entity.currentTask:
if hasattr(self.entity.currentTask,'timeSpentBuilding'):
from Overlay import Bar
self.buildBar = Bar(self.entity.currentTask.timeToBuild,118,6,fullColor=(0,180,255),emptyColor=(30,30,30))
self.buildBar.updateBarWithValue(self.entity.currentTask.timeSpentBuilding)
self.buildBar.draw(screen,(self.pos[0]+78,self.pos[1]+69))
screen.blit(self.thumbnail, (self.pos[0]+self.thumbnailOffset[0],self.pos[1]+self.thumbnailOffset[1]))
self.healthBar.draw(screen,(self.pos[0]+78,self.pos[1]+56))
self.description.render(screen)