dodox/Columbus
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Minor bug fixes

Browse Source
This commit is contained in:
Dominik Moritz Roth 2022-06-19 15:32:12 +02:00
parent 89aeaac1eb
commit 12cac4db99
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
env.py
View File

@ -12,7 +12,7 @@ class ColumbusEnv(gym.Env):
metadata = {'render.modes': ['human']} metadata = {'render.modes': ['human']}
def __init__(self, observable=observables.Observable(), fps=60, env_seed=3.1): def __init__(self, observable=observables.Observable(), fps=60, env_seed=3.1):
super(Base2DExpEnv, self).__init__() super(ColumbusEnv, self).__init__()
self.action_space = spaces.Box( self.action_space = spaces.Box(
low=0, high=1, shape=(2,), dtype=np.float32) low=0, high=1, shape=(2,), dtype=np.float32)
observable._set_env(self) observable._set_env(self)
@ -121,7 +121,7 @@ class ColumbusEnv(gym.Env):
shapes = [e1.shape, e2.shape] shapes = [e1.shape, e2.shape]
shapes.sort() shapes.sort()
if shapes == ['circle', 'circle']: if shapes == ['circle', 'circle']:
sq_dist = ((e1.pos[0]-e2[0])*self.width) ** 2 \ sq_dist = ((e1.pos[0]-e2.pos[0])*self.width) ** 2 \
+ ((e1.pos[1]-e2.pos[1])*self.height)**2 + ((e1.pos[1]-e2.pos[1])*self.height)**2
return sq_dist < (e1.radius + e2.radius)**2 return sq_dist < (e1.radius + e2.radius)**2
else: else:
@ -139,7 +139,7 @@ class ColumbusEnv(gym.Env):
self.entities = newEntities self.entities = newEntities
def setup(self): def setup(self):
for i in range(16): for i in range(18):
enemy = entities.CircleBarrier(self) enemy = entities.CircleBarrier(self)
enemy.radius = self.random()*40+50 enemy.radius = self.random()*40+50
self.entities.append(enemy) self.entities.append(enemy)