from mesa import Agent, Model
import random
from agent2 import Weapon, Armor, Box, Creature, Player, dice

WM1 = Weapon("Log","Melee",1)
WM2 = Weapon("Log","Melee",1)
WM3 = Weapon("Log","Melee",1)
WM4 = Weapon("Dagger","Melee",1)
WM5 = Weapon("Dagger","Melee",1)
WM6 = Weapon("Dagger","Melee",1)
WM7 = Weapon("Sword","Melee",2)
WM8 = Weapon("Sword","Melee",2)
WM9 = Weapon("Sword","Melee",2)
WM10 = Weapon("Axe","Melee",3)
WM11 = Weapon("Axe","Melee",3)
WM12 = Weapon("Battle axe","Melee",4)

WR1 = Weapon("Sling","Range",1)
WR2 = Weapon("Sling","Range",1)
WR3 = Weapon("Sling","Range",1)
WR4 = Weapon("Bow","Range",2)
WR5 = Weapon("Bow","Range",2)
WR6 = Weapon("Bow","Range",2)
WR7 = Weapon("Bow","Range",2)
WR8 = Weapon("Longbow","Range",3)
WR9 = Weapon("Longbow","Range",3)
WR10 = Weapon("Crossbow","Range",4)

S1 = Weapon("Push","Magic",2)
S2 = Weapon("Push","Magic",2)
S3 = Weapon("Punch","Magic",4)
S4 = Weapon("Punch","Magic",4)
S5 = Weapon("Fireball","Magic",6)
S6 = Weapon("Firestorm","Magic",10)

A1 = Armor("None",0,0)
A2 = Armor("None",0,0)
A3 = Armor("None",0,0)
A4 = Armor("None",0,0)
A5 = Armor("None",0,0)
A6 = Armor("None",0,0)
A7 = Armor("None",0,0)
A8 = Armor("Leather Armor",1,0)
A9 = Armor("Leather Armor",1,0)
A10 = Armor("Leather Armor",1,0)
A11 = Armor("Chain Mail",2,0)
A12 = Armor("Chain Mail",2,0)
A13 = Armor("Plate Armor",3,0)
A14 = Armor("Robe",1,2)
A15 = Armor("Robe",1,2)
A16 = Armor("Magical Plate Armor",3,2)

# C1 = Box(WM6)
# C2 = Box(WR3)
# C3 = Box(WR6)
# C4 = Box(WR7)
# C5 = Box(WR9)
# C6 = Box(WM9)
# C7 = Box(WM10)
# C8 = Box(S4)
# C9 = Box(S5)
# C10 = Box(A8)
# C11 = Box(A12)
# C12 = Box(A14)

#Gracz = Player(1000, self, "Janusz",3,3,3,20,20,WM1,A1,0,WR1,S1)
#    def __init__(self, unique_id, model, n, s, a, w, maxhp, hp, weap, arm, g):
# M1 = Creature("Goblin",2,2,1,10,10,WM2,A2,dice(6))
# M2 = Creature("Goblin",2,2,1,10,10,WM3,A3,dice(6))
# M3 = Creature("Goblin",2,2,1,10,10,WR2,A4,dice(6))
# M4 = Creature("Skeleton Archer",2,3,4,15,15,WR4,A5,dice(6))
# M5 = Creature("Skeleton Archer",2,3,4,15,15,WR5,A6,dice(6))
# M6 = Creature("Skeleton",3,2,4,15,15,WM7,A7,dice(6))
# M7 = Creature("Skeleton",3,2,4,15,15,WM8,A7,dice(6))
# M8 = Creature("Bandit",3,3,2,15,15,WM4,A9,2*dice(6))
# M9 = Creature("Bandit",3,3,2,15,15,WM5,A10,2*dice(6))
# M10 = Creature("Chieftain",4,4,4,20,20,WR8,A11,3*dice(6))
# M11 = Creature("Sorcerer",3,3,6,25,25,S3,A15,3*dice(6))