evolutionsim/environment.py at master · esmason/evolutionsim · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
from Animal import Animal
from MaleAnimal import Male
from Female import Female
import random

class Environment:
    def __init__(self):
        self.males = []
        self.females = []
        self.juveniles = []
        self.growUpAge = 2 #age animals start mating
        self.dieAge = 4 #age adult animals die
        self.maxPop = 400 #runs very slow if much above 1E3


    def addMale(self,m):
        self.males.append(m)
        m.addEnvironment(self)

    def addFemale(self,f):
        self.females.append(f)

    def addJuvenile(self,j):
        self.juveniles.append(j)

    def getMales(self):
        return self.males

    def getFemales(self):
        return self.females

    def getJuveniles(self):
        return self.juveniles

    def getNumMales(self):
        return len(self.getMales())

    def hasFemales(self):
        return len(self.females) > 0


    def getNumFemales(self):
        return len(getFemales())

    def getNumJuveniles(self):
        return len(getJuveniles())

    def removeAdult(self, a):
        if a.isFemale():
            self.females.remove(a)
        else:
            self.males.remove(a)
    def growUp(self, a):
        self.juveniles.remove(a)
        if a.isFemale():
            self.addFemale(a)
        else:
            self.addMale(a)


    def updateAnimalsAge(self):
        for a in self.males:
            a.older()
            if (a.age > self.dieAge):
                self.removeAdult(a)

        for a in self.females:
            a.older()
            if (a.age > self.dieAge):
                self.removeAdult(a)

        for a in self.juveniles:
            a.older()
            if (a.age==self.growUpAge):
                self.growUp(a)

    def removeJuvenile(self, j):
        self.juveniles.remove(j)

    def getFemalesNotTaken(self):
        notTaken=[]
        for f in self.females:
            if not f.getHasPartner():
                notTaken.append(f)
        return notTaken


    # maintains M/F/J ratio and randomly removes enough animals to keep total animals < maxPop
    def dieOff(self):
        total = len(self.getFemales()) + len(self.getMales()) + len(self.getJuveniles())
        percentFemale = len(self.getFemales())/total
        percentMale = len(self.getMales())/total
        percentJuvenile = len(self.getJuveniles())/total

        while len(self.getMales()) > round(percentMale*self.maxPop):

            self.removeAdult(random.choice(self.getMales()))
        while len(self.getFemales()) > (percentFemale*self.maxPop):
            self.removeAdult(random.choice(self.getFemales()))
        while len(self.getJuveniles()) > round(percentJuvenile*self.maxPop):
            self.removeJuvenile(random.choice(self.getJuveniles()))


    #invariants: total population less than maxPop, organisms in appropriate category for age, currently no organisms
    # have partners

    def update(self):

        self.updateAnimalsAge()
        males = self.getMales()
        for m in males:
            m.findPartner()

            if m.hasPartner():
                m.getPartner().setPartner(True) #to maintain 1-1 matingm, aka 'monogamy' each update
                m.mate()
                for c in m.getChilds():
                    self.addJuvenile(c)
                m.clearChilds()
            for m in males:
                m.setPartner(None)
        for f in self.getFemales():
            f.mutate()
            f.setPartner(False)

        if len(self.getFemales()) >100 and len(self.getMales())>100: #animals die when population gets too big
            self.dieOff()

    def getAverageRatio(self):
        sum = 0
        allAnimals =  self.getMales()+self.getFemales()+self.getJuveniles()
        for a in allAnimals:
            sum = sum + a.getRatio()
        return sum/len(allAnimals)