ringedseal.py

from animal import Animal
from random import random, uniform, choice, shuffle
from numpy.random import exponential


class RingedSeal(Animal):

    count = 0
    initial_population = 400
    capacity = 600

    def __init__(self, gender, parents, age=2555):
        super().__init__(gender=gender,
                         age=age,
                         probability_death=0.15,
                         probability_birth=0.3,
                         movement_speed=0.5,
                         hunger=0,
                         radius=30,
                         weaning=42,
                         mating={'m': 1825, 'f': 1095},
                         parents=parents,
                         pregnancy=[269, 271])
        self.x = uniform(0, 100)
        self.y = self.restrict(100 - exponential(1.65) * 15, 0, 100)
        self.uid = RingedSeal.count
        RingedSeal.count += 1

    def check_death(self, agents, neighbours):
        if len(neighbours) > 0 and random() < self.probability_death and self.age > self.weaning:
            temp_neighbours = [
                i for i in neighbours if i.age > i.weaning]
            shuffle(temp_neighbours)
            temp_uid = 0
            temp = 0
            for i in temp_neighbours:
                if i.hunger > 0.7:
                    i.hunger = 0.1
                    i.probability_death = (i.hunger) + (i.age)/1000
                    deaths = []
                    deaths.append(self)
                    RingedSeal.count -= 1
                    for child in self.children:
                        if child.age < child.weaning:
                            deaths.append(child)
                            RingedSeal.count -= 1
                    return deaths
        return False

# 	def give_birth(self, chosen):
# 		if self.gender == 'f':
# 			female = self
# 			male = chosen
# 		else:
# 			female = chosen
# 			male = self
# =======
# 			temp_neighbours = neighbours.copy()
# 			shuffle(temp_neighbours)
# 			temp_uid = 0
# 			temp = 0
# 			for i in temp_neighbours:
# 				if i.hunger > 0.5:
# 					temp += 1
# 					i.hunger = 0.1
# 					i.probability_death = i.probability_death * i.hunger
# 					temp_uid = i.uid
# 					deaths = []
# 					deaths.append(self)
# 					RingedSeal.count -= 1
# 					for child in self.children:
# 						if child.age < child.weaning:
# 							deaths.append(child)
# 							RingedSeal.count -= 1
# 					return deaths
# 					break

# 			# for i in temp_neighbours:
# 			# 	if i.uid != temp_uid:
# 			# 		i.hunger += 0.1
# 		return False

    def give_birth(self, chosen):
        if self.gender == 'f':
            female = self
            male = chosen
        else:
            female = chosen
            male = self
        parents = {
            'm': male,
            'f': female
        }
        child = RingedSeal(choice(['f', 'm']), parents, 0)
        self.children.append(child)
        self.isPregnant = False
        return [child]

    def check_birth(self, agents, same_neighbours, day):
        if day < 130 or day > self.seasons['autumn']:
            return False
        if self.age > self.mating:
            if len(same_neighbours) > 0 and random() < self.probability_birth * (1 - RingedSeal.count / RingedSeal.capacity):
                opp_gender = [ag for ag in same_neighbours if ag.gender !=
                              self.gender and not ag.isPregnant and ag.age > ag.weaning]
                for i in opp_gender:
                    hasWeaningChildren = False
                    if len(i.children) != 0:
                        for j in i.children:
                            if j.age < j.weaning:
                                hasWeaningChildren = True
                                break
                    if hasWeaningChildren:
                        opp_gender.remove(i)
                if len(opp_gender) == 0:
                    return False
                chosen = choice(opp_gender)
                if self.gender == 'f':
                    self.isPregnant = True
                    self.daysSpentInPregnancy = 0
                else:
                    chosen.isPregnant = True
                    chosen.daysSpentInPregnancy = 0
                self.partner = chosen
                return True
        return False

    def move(self, agents, day):
        if self.age > self.weaning:
            self.x = self.restrict(
                self.x + uniform(-self.movement_speed, self.movement_speed), 0, 100)
            self.y = self.restrict(
                self.y + uniform(-self.movement_speed, self.movement_speed), 0, 100)
        else:
            self.x = self.parents['f'].x + uniform(-2, 2)
            self.y = self.parents['f'].y + uniform(-2, 2)