class GuiUnitPanel:
def __init__(self, aspect, unit_id, panel_pos, unit_type, default_hp, hp, default_ap, ap):
self.unit_id = unit_id
self.frameWidth = 0.30
self.frameHeight = 0.06
self.frame = DirectFrame( # relief = DGG.FLAT
scale = 1
, frameSize = (0, self.frameWidth, 0, -self.frameHeight)
, frameColor = (0.2, 0.2, 0.2, 0.8)
, parent = base.a2dTopLeft )#@UndefinedVariable
self.pos = Point3(0, 0, -GUI_TOP_OFFSET-0.05 - (0.069*panel_pos))
self.frame.setPos(self.pos)
self.ap_bar = DirectWaitBar(parent = self.frame
, text = ""
, range = default_ap
, value = ap
, pos = (0.2,0,-0.045)
, barColor = (0,0,1,1)
, frameColor = (0,0,0.5,0.2)
, scale = (0.1,0.5,0.1))
self.hp_bar = DirectWaitBar(parent = self.frame
, text = ""
, range = default_hp
, value = hp
, pos = (0.2,0,-0.015)
, barColor = (0,1,0,1)
, frameColor = (1,0,0,0.9)
, scale = (0.1,0.5,0.08))
self.insignia = OnscreenImage(parent = self.frame
,image = "unit_" + unit_type + "_big_transparent_32.png"
,pos = (0.035,0,-0.03)
,scale = 0.025)
self.insignia.setTransparency(TransparencyAttrib.MAlpha)
self.all_icons = {}
self.visible_icons = {}
self.addIcon("overwatch")
self.addIcon("set_up")
self.getBounds(aspect)
def addIcon(self, name):
self.all_icons[name] = OnscreenImage(parent = self.frame
,image = name + "_icon.png"
#,pos = offset + (0,0,-0.1)
,scale = 0.035)
self.all_icons[name].setTransparency(TransparencyAttrib.MAlpha)
self.all_icons[name].hide()
def refreshBars(self, hp, ap):
self.ap_bar['value'] = ap
self.hp_bar['value'] = hp
self.ap_bar.setValue()
self.hp_bar.setValue()
def refreshIcons(self):
count = len(self.visible_icons)
start_pos = 0.34
for icon in self.all_icons:
if icon in self.visible_icons:
self.visible_icons[icon].setPos((start_pos, 0, -0.032))
self.visible_icons[icon].show()
start_pos += 0.07
else:
self.all_icons[icon].hide()
def hideOverwatch(self):
if "overwatch" in self.visible_icons:
self.visible_icons.pop("overwatch")
self.refreshIcons()
def showOverwatch(self):
self.visible_icons["overwatch"] = self.all_icons["overwatch"]
self.refreshIcons()
def hideSetUp(self):
if "set_up" in self.visible_icons:
self.visible_icons.pop("set_up")
self.refreshIcons()
def showSetUp(self):
self.visible_icons["set_up"] = self.all_icons["set_up"]
self.refreshIcons()
def getBounds(self, aspect):
posx, posy = self.frame.getTightBounds()
self.pos_min_x = -1
self.pos_min_y = self.pos.getZ() + 1 - self.frameHeight
self.pos_max_x = -1 + self.frameWidth/aspect
self.pos_max_y = self.pos.getZ() + 1
class GuiButton2:
def __init__(self, hugpos, offset, aspect, plane, name):
self.enabled = True
self.name = name
self.frame = DirectFrame( relief = DGG.FLAT
, frameColor = (0, 0, 0, 0)
, scale = 1
, frameSize = (-0.3, 0.3, -0.3, 0.3) ) #32/600=0.05333
self.frame.setScale(0.0533)
self.imageObject = OnscreenImage(image = name+".png", pos = (0, 0, 0))
self.imageObject.setTransparency(TransparencyAttrib.MAlpha)
self.imageObject.setAlphaScale(1)
self.imageObject.reparentTo(self.frame)
self.hugpos = hugpos
self.offset = offset
self.redraw(aspect)
def redraw(self, aspect, flag="wide"):
if self.hugpos == "top":
p = base.a2dTopLeft.getPos()#@UndefinedVariable
p.setX(p.getX() + self.offset.getX() + 0.05)
p.setZ(p.getZ() + self.offset.getZ() - GUI_TOP_OFFSET - 0.05)
elif self.hugpos == "bottom":
p = base.a2dBottomLeft.getPos()#@UndefinedVariable
p.setX(p.getX() + self.offset.getX() + 0.05)
p.setZ(p.getZ() + self.offset.getZ() + GUI_BOTTOM_OFFSET - 0.05)
elif self.hugpos == "right":
p = base.a2dBottomRight.getPos()#@UndefinedVariable
p.setX(p.getX() + self.offset.getX() + 0.05)
p.setZ(p.getZ() + self.offset.getZ() + GUI_BOTTOM_OFFSET - 0.05)
self.frame.setPos(p)
if flag == "wide":
posx, posy = self.frame.getTightBounds()
self.pos_min_x = posx.getX() / aspect
self.pos_min_y = posx.getZ()
self.pos_max_x = posy.getX() / aspect
self.pos_max_y = posy.getZ()
elif flag == "tall":
posx, posy = self.frame.getTightBounds()
self.pos_min_x = posx.getX()
self.pos_min_y = posx.getZ() / aspect
self.pos_max_x = posy.getX()
self.pos_max_y = posy.getZ() / aspect
def turnOn(self):
self.imageObject.setImage(self.name+"_on.png")#@UndefinedVariable
self.imageObject.setTransparency(TransparencyAttrib.MAlpha)
def turnOff(self):
self.imageObject.setImage(self.name+".png")#@UndefinedVariable
self.imageObject.setTransparency(TransparencyAttrib.MAlpha)
def enable(self):
self.imageObject.setImage(self.name+".png")#@UndefinedVariable
self.imageObject.setTransparency(TransparencyAttrib.MAlpha)
self.enabled = True
def disable(self):
self.imageObject.setImage("empty.png")#@UndefinedVariable
self.imageObject.setTransparency(TransparencyAttrib.MAlpha)
self.enabled = False
def removeNode(self):
self.frame.removeNode()
def setAbility(self, ability):
self.name = ability
self.imageObject.setImage(self.name+".png")
self.imageObject.setTransparency(TransparencyAttrib.MAlpha)
self.imageObject.setAlphaScale(1)
