forked from ranshiju/T-Nalg
-
Notifications
You must be signed in to change notification settings - Fork 0
/
SpottiesGame.py
126 lines (106 loc) · 4.56 KB
/
SpottiesGame.py
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
import numpy as np
import BasicFunctionsSJR as bfr
from Spotties import Spotty, save_intel_linear_random, spotty_copy, find_position_from_movement, decide_by_intel
from Universe import Universe
_is_debug = True
class GameBasic:
def __init__(self, info, lx, ly, max_age, split_energy, is_debug=False):
self.universe = Universe(info, lx, ly)
self.spotties = list()
self.population = dict()
self.population_tot = 0
self.max_age = max_age
self.split_energy = split_energy
self._is_debug = is_debug
def add_spotties_random(self, new_pop, info):
pop0 = self.spotties.__len__()
for n in range(0, new_pop):
self.spotties.append(Spotty(info, self.max_age, self.split_energy))
self.spotties[n + pop0].enter_map_random(self.universe.map['tribe'])
self.universe.put_universe(self.spotties[n + pop0])
def add_spotty_positions(self, info, positions):
new_pop = len(positions)
pop0 = self.spotties.__len__()
for n in range(0, new_pop):
self.spotties.append(Spotty(info, self.max_age, self.split_energy))
self.spotties[n + pop0].enter_map_position(positions[n])
self.universe.put_universe(self.spotties[n + pop0])
def spotty_split_random(self, nth):
# split the nth spotty
neighbor1 = self.universe.read_map(self.spotties[nth], 1)
movement = np.nonzero(neighbor1['tribe'] == 0)
rand_pos = np.random.randint(0, len(movement[1]), 1)
movement = movement[1][rand_pos[0]] + 1
new_pos = find_position_from_movement(self.spotties[nth].position, movement)
new = spotty_copy(self.spotties[nth])
new.enter_map_position(new_pos)
self.universe.put_universe(new)
self.spotties.append(new)
self.spotties[nth].energy = 0
self.spotties[nth].age += 1
def spotty_move(self, nth, decision):
self.universe.delete_universe(self.spotties[nth])
self.spotties[nth].move(decision)
self.universe.put_universe(self.spotties[nth])
self.spotties[nth].age += 1
def spotty_die(self, nth):
self.universe.delete_universe(self.spotties[nth])
self.spotties.__delitem__(nth)
def update_population_info(self):
self.population_tot = self.spotties.__len__()
self.population = dict()
for s in self.spotties:
for p in s.info:
if p in self.population:
self.population[p] += 1
else:
self.population[p] = 1
class SpottiesGameV0(GameBasic):
# One tribe
# empty sites > self.cond_gain_energy: gain energy
def __init__(self, info, cond_gain_energy, lx, ly, max_age, split_energy, is_debug=False):
super(SpottiesGameV0, self).__init__(info, lx, ly, max_age, split_energy, is_debug)
self.cond_gain_energy = cond_gain_energy
def update_one_spotty(self, nth, intel):
if _is_debug:
self.spotties[nth].report_yourself()
if self.spotties[nth].age > self.max_age:
self.spotty_die(nth)
else:
env = self.universe.read_map(self.spotties[nth], 2)
print(env)
if self.if_gain_energy(env, self.cond_gain_energy):
self.spotties[nth].energy += 1
if self.spotties[nth].energy == self.split_energy:
self.spotty_split_random(nth)
else:
decision = decide_by_intel(intel, env['tribe'])
print(decision)
self.spotty_move(nth, decision)
@ staticmethod
def if_gain_energy(env2, cond_energy):
return np.count_nonzero(env2 == 0) < cond_energy + 0.1
def game_v0():
# iteration time
time = 200
# map size
lx = 9
ly = 9
# properties of spotties
ini_pos = [[4, 4]]
max_age = 100
split_energy = 100
cond_gain_energy = 3
# initial game
info = {'tribe': 1}
game = SpottiesGameV0(info, cond_gain_energy, lx, ly, max_age, split_energy)
game.add_spotty_positions(info, ini_pos)
save_intel_linear_random(8, 5, 'linear_intel.pr')
intel1 = dict()
intel1['type'] = 'linear'
intel1['data'] = bfr.load_pr('.\\Intels\\linear_intel.pr', 'intel')
intel = [intel1]
for t in range(0, time):
for n in range(game.spotties.__len__() - 1, -1, -1):
game.update_one_spotty(n, intel[game.spotties[n].info['tribe'] - 1])
game.universe.plot_universe()