class CircleLabel:
def __init__(self,
text='',
x=0,
y=0,
rad=5,
fc=(255, 255, 255),
bc=(0, 0, 0),
ai=False):
self.x = x
self.y = y
self.label = Label(text, x, y, size=(rad - 1))
self.fill_color = fc
self.border_color = bc
self.radius = rad if rad > 5 else 5 # minimum radius is 5
self.ai = ai
def render(self, surface):
pygame.draw.circle(surface, self.border_color, (self.x, self.y),
self.radius)
pygame.draw.circle(surface, self.fill_color, (self.x, self.y),
self.radius - 3)
if not self.ai:
self.label.render(surface)
def set_text(self, text):
self.label.set_text(text)
def set_pos(self, pos):
self.x, self.y = pos
def set_radius(self, radius):
self.radius = radius
def set_fill_color(self, fc):
self.fill_color = fc
def set_border_color(self, bc):
self.border_color = bc
class Troop:
def __init__(self, x=0, y=0, ai=False, t=0):
self.x, self.y = x, y # position to draw troop
self.health = 100 # troop's health
self.ai = ai # flag for specifying if troop is ai
self.type = t # 0 = default, 1 = medic, 2 = sniper
if self.type == 2:
self.health = 75
self.default_stats() # default all troops state
# read in sprite image for troop
# if ai, flip the image
if self.ai and self.type == 0:
self.image = pygame.transform.flip(pygame.image.load(os.path.join('assets', 'ai_commando.png')), True, False)
elif self.ai and self.type == 1:
self.image = pygame.transform.flip(pygame.image.load(os.path.join('assets', 'ai_medic.png')), True, False)
elif self.ai and self.type == 2:
self.image = pygame.transform.flip(pygame.image.load(os.path.join('assets', 'ai_soldier.png')), True, False)
elif not self.ai and self.type == 0:
self.image = pygame.image.load(os.path.join('assets', 'player_commando.png'))
elif not self.ai and self.type == 1:
self.image = pygame.image.load(os.path.join('assets', 'player_medic.png'))
else:
self.image = pygame.image.load(os.path.join('assets', 'player_soldier.png'))
# rectangle surrounding troop (useful for detecting clicks)
self.rect = self.image.get_rect()
self.rect.center = int(self.x), int(self.y)
# label for drawing troops health
self.health_label = Label(str(int(self.health)), size=15)
# detect if troop was clicked on
def get_event(self, event):
if event.type == pygame.MOUSEBUTTONUP and self.rect.collidepoint(event.pos):
return True
return False
# update health label to display current health
def update(self):
self.health_label.set_text(str(int(self.health)))
# draw troop to screen
def render(self, surface):
surface.blit(self.image, self.rect)
self.health_label.render(surface)
# add attack boost to troop
def attack_boost(self):
self.attack += 10
if self.type == 2:
self.attack += 10
self.defense = 0
if self.type == 0 and self.attack > 40:
self.attack = 40
elif self.type == 1 and self.attack > 30:
self.attack = 30
elif self.type == 2 and self.attack > 70:
self.attack = 70
# add defense boost to troop
def defense_boost(self):
self.defense += 10
if self.type == 2:
self.attack += 10
self.defense -= 5
if self.type == 0 and self.defense > 40:
self.defense = 40
elif self.type == 1 and self.defense > 20:
self.defense = 30
elif self.type == 2 and self.defense > 10:
self.defense = 20
# add healing property
def heal(self,num):
self.health += num
if self.health > 100:
self.health = 100
if self.type == 2 and self.health > 75:
self.health = 75
# set stats to default values
def default_stats(self):
self.attack = 20
if self.type == 2:
self.attack = 10
self.defense = 10
# set pos to draw troop
def set_pos(self, pos):
self.x, self.y = pos
self.rect.center = int(self.x), int(self.y)
if not self.ai:
self.health_label.set_pos((int(self.x - self.rect.w/4), int(self.y - self.rect.h/2 - 15)))
else:
self.health_label.set_pos((int(self.x + self.rect.w / 4), int(self.y - self.rect.h / 2 - 15)))