221 lines
7.9 KiB
Python
221 lines
7.9 KiB
Python
from mesa import Agent
|
|
import random
|
|
|
|
def dice(number):
|
|
return random.randint(1,number)
|
|
|
|
class Wall(Agent):
|
|
def __init__(self, unique_id, model):
|
|
super().__init__(unique_id, model)
|
|
|
|
def step(self):
|
|
pass
|
|
|
|
class Box(Agent):
|
|
def __init__(self, unique_id, model):
|
|
super().__init__(unique_id, model)
|
|
self.gold=3*dice(6)
|
|
self.isBox=True
|
|
self.isCreature=False
|
|
|
|
def step(self):
|
|
pass
|
|
|
|
class Weapon():
|
|
def __init__(self,name,type,damage):
|
|
self.name = name
|
|
self.type = type
|
|
self.damage = damage
|
|
|
|
class Armor():
|
|
def __init__(self,name,defence,mp):
|
|
self.name = name
|
|
self.defence = defence
|
|
self.mag_protection = mp
|
|
|
|
class Creature(Agent):
|
|
def __init__(self, unique_id, model, n, s, a, w, maxhp, hp, weap, arm, g):
|
|
super().__init__(unique_id, model)
|
|
self.name = n
|
|
self.strength = s
|
|
self.agility = a
|
|
self.wisdom = w
|
|
self.maxHealth = maxhp
|
|
self.health = hp
|
|
self.gold = g
|
|
self.weapon1 = weap
|
|
self.armor = arm
|
|
self.isBox = False
|
|
self.isCreature=True
|
|
|
|
|
|
def meleeAttack(self,opponent):
|
|
attackValue = self.strength + dice(6)
|
|
defenseValue = opponent.strength + opponent.armor.defence
|
|
damage = attackValue - defenseValue
|
|
if damage > 0:
|
|
opponent.health = opponent.health - (damage + self.weapon1.damage)
|
|
|
|
def rangeAttack(self,opponent):
|
|
attackValue = self.agility + dice(6)
|
|
defenseValue = opponent.agility
|
|
damage = attackValue - defenseValue
|
|
if (damage > 0) and (damage + self.weapon1.damage - opponent.armor.defence > 0):
|
|
opponent.health = opponent.health - (damage + self.weapon1.damage - opponent.armor.defence)
|
|
|
|
def magicAttack(self,opponent):
|
|
attackValue = self.wisdom + dice(6)
|
|
defenseValue = opponent.wisdom
|
|
damage = attackValue - defenseValue
|
|
if (damage > 0) and (damage + self.weapon1.damage - opponent.armor.mag_protection > 0):
|
|
opponent.health = opponent.health - (damage + self.weapon1.damage - opponent.armor.mag_protection)
|
|
|
|
def defaultAttack(self,opponent):
|
|
if self.weapon1.type == "Meele":
|
|
self.meleeAttack(opponent)
|
|
elif self.weapon1.type == "Range":
|
|
self.rangeAttack(opponent)
|
|
else:
|
|
self.magicAttack(opponent)
|
|
|
|
class Player(Creature):
|
|
def __init__(self, unique_id, model, n, s, a, w, maxhp, hp, weap, arm, g, w2, w3):
|
|
super().__init__(unique_id, model, n, s, a, w, maxhp, hp, weap, arm, g)
|
|
self.name = n
|
|
self.strength = s
|
|
self.agility = a
|
|
self.wisdom = w
|
|
self.maxHealth = maxhp
|
|
self.health = hp
|
|
self.gold = g
|
|
self.weapon1 = weap
|
|
self.weapon2 = w2
|
|
self.weapon3 = w3
|
|
self.armor = arm
|
|
self.isBox = False
|
|
self.isCreature = False
|
|
|
|
def move(self):
|
|
possible_steps = self.model.grid.get_neighborhood(
|
|
self.pos,
|
|
moore=True,
|
|
include_center=False)
|
|
new_position = self.random.choice(possible_steps)
|
|
self.model.grid.move_agent(self, new_position)
|
|
|
|
def meleeAttack(self,opponent):
|
|
attackValue = self.strength + dice(6)
|
|
defenseValue = opponent.strength + opponent.armor.defence
|
|
damage = attackValue - defenseValue
|
|
if damage > 0:
|
|
opponent.health = opponent.health - (damage + self.weapon1.damage)
|
|
|
|
def rangeAttack(self,opponent):
|
|
attackValue = self.agility + dice(6)
|
|
defenseValue = opponent.agility
|
|
damage = attackValue - defenseValue
|
|
if (damage > 0) and (damage + self.weapon2.damage - opponent.armor.defence > 0):
|
|
opponent.health = opponent.health - (damage + self.weapon2.damage - opponent.armor.defence)
|
|
|
|
def magicAttack(self,opponent):
|
|
attackValue = self.wisdom + dice(6)
|
|
defenseValue = opponent.wisdom
|
|
damage = attackValue - defenseValue
|
|
if (damage > 0) and (damage + self.weapon3.damage - opponent.armor.mag_protection > 0):
|
|
opponent.health = opponent.health - (damage + self.weapon3.damage - opponent.armor.mag_protection)
|
|
|
|
def fightOrFlight(self, opponent):
|
|
combat = True
|
|
while combat:
|
|
choice = dice(4)
|
|
print("dice rolled:", choice)
|
|
if choice == 1:
|
|
running_speed = self.agility + dice(6)
|
|
opponent_speed = opponent.agility + dice(6)
|
|
if running_speed > opponent_speed:
|
|
combat = False
|
|
print("Player ran away")
|
|
self.step()
|
|
else:
|
|
opponent.defaultAttack(self)
|
|
if self.health <= 0:
|
|
combat = False
|
|
print("Player died :/")
|
|
elif choice == 2:
|
|
self.meleeAttack(opponent)
|
|
if opponent.health > 0:
|
|
opponent.defaultAttack(self)
|
|
if self.health <= 0:
|
|
combat = False
|
|
print("Player died :/")
|
|
else:
|
|
combat = False
|
|
self.gold = self.gold + opponent.gold
|
|
opponent.gold = 0
|
|
opponent.model.grid.remove_agent(opponent)
|
|
print("Fight won")
|
|
elif choice == 3:
|
|
self.rangeAttack(opponent)
|
|
if opponent.health > 0:
|
|
opponent.defaultAttack(self)
|
|
if self.health <= 0:
|
|
combat = False
|
|
print("Player died :/")
|
|
else:
|
|
combat = False
|
|
self.gold = self.gold + opponent.gold
|
|
opponent.gold = 0
|
|
opponent.model.grid.remove_agent(opponent)
|
|
print("Fight won")
|
|
else:
|
|
self.magicAttack(opponent)
|
|
if opponent.health > 0:
|
|
opponent.defaultAttack(self)
|
|
if self.health <= 0:
|
|
combat = False
|
|
print("Player died :/")
|
|
else:
|
|
combat = False
|
|
self.gold = self.gold + opponent.gold
|
|
opponent.gold = 0
|
|
opponent.model.grid.remove_agent(opponent)
|
|
print("Fight won")
|
|
|
|
def openChest(self,chest):
|
|
self.gold = self.gold + chest.gold
|
|
print("Chest opened. Gold inside:", chest.gold)
|
|
chest.gold = 0
|
|
# if isinstance(chest.loot,Armor):
|
|
# buffer = self.armor
|
|
# self.armor = chest.loot
|
|
# chest.loot = buffer
|
|
# if isinstance(chest.loot,Weapon):
|
|
# if chest.loot.type == "Melee":
|
|
# buffer = self.weapon1
|
|
# self.weapon1 = chest.loot
|
|
# chest.loot = buffer
|
|
# elif chest.loot.type == "Range":
|
|
# buffer = self.weapon2
|
|
# self.weapon2 = chest.loot
|
|
# chest.loot = buffer
|
|
# elif chest.loot.type == "Magic":
|
|
# buffer = self.weapon3
|
|
# self.weapon3 = chest.loot
|
|
# chest.loot = buffer
|
|
|
|
def step(self):
|
|
if self.health > 0:
|
|
self.move()
|
|
cellmates = self.model.grid.get_cell_list_contents([self.pos])
|
|
if len(cellmates) > 1:
|
|
if isinstance(cellmates[0],Box):
|
|
self.openChest(cellmates[0])
|
|
else:
|
|
opponent = cellmates[0]
|
|
print("Fighting")
|
|
self.fightOrFlight(opponent)
|
|
print("HP: "+str(self.health)+" / "+str(self.maxHealth))
|
|
print("Gold: "+str(self.gold))
|
|
else:
|
|
print("HP: 0 / "+str(self.maxHealth))
|
|
|