def __init__(self):
self.acceleration = random.randint(10, 30) / 10 # 1-3 by 0.1
self.wind = 0.2 * (random.random() - 0.5)
self.max_safe_landing_speed = 4.0
self.min_safe_x = -0.2
self.max_safe_x = 0.2
self.landed = 'in_air'
self.height = 100.0
self.x_position = 0.0
self.y_velocity = 10.0 * random.random()
self.x_velocity = 0.0
self.fuel = 100.0
self.burn = None
self.thrust = None
self.controller = FuzzyControl()
def __init__(self):
self.acceleration = random.randint(10, 30)/10 # 1-3 by 0.1
self.wind = 0.2 * (random.random()-0.5)
self.max_safe_landing_speed = 4.0
self.min_safe_x = -0.2
self.max_safe_x = 0.2
self.landed = 'in_air'
self.height = 100.0
self.x_position = 0
self.y_velocity = 10.0 * random.random()
self.x_velocity = 0
self.fuel = 100.0
self.burn = None
self.thrust = None
self.controller = FuzzyControl()
class Moonlander:
def __init__(self):
self.acceleration = random.randint(10, 30)/10 # 1-3 by 0.1
self.wind = 0.2 * (random.random()-0.5)
self.max_safe_landing_speed = 4.0
self.min_safe_x = -0.2
self.max_safe_x = 0.2
self.landed = 'in_air'
self.height = 100.0
self.x_position = 0
self.y_velocity = 10.0 * random.random()
self.x_velocity = 0
self.fuel = 100.0
self.burn = None
self.thrust = None
self.controller = FuzzyControl()
def control(self):
"""
This needs to get burn and thrust from the ANN
"""
return self.controller.control_input(self.height,
self.y_velocity,
self.x_position,
self.x_velocity,
self.wind)
def test(self):
if self.height > 0:
return 'in_air'
if (self.y_velocity > self.max_safe_landing_speed or
self.x_position < self.min_safe_x or
self.x_position > self.max_safe_x):
return 'crashed'
else:
return 'landed'
def update(self):
self.y_velocity += self.acceleration
burn, thrust = self.control()
if self.fuel < burn:
burn = self.fuel
self.fuel -= abs(burn)
self.y_velocity -= burn
if self.fuel < abs(thrust):
if thrust > 0:
thrust = self.fuel
elif thrust <= 0:
thrust = -self.fuel
self.fuel -= abs(thrust)
self.x_velocity -= thrust
self.height -= self.y_velocity
self.x_position += (self.x_velocity + self.wind)
self.landed = self.test()
self.output(burn, thrust)
def output(self, burn, thrust):
output = {'Height': self.height,
'Y-Velocity': self.y_velocity,
'Burn': burn,
'Wind': self.wind,
'X-Velocity': self.x_velocity,
'X-Position': self.x_position,
'Thrust': thrust,
'Fuel': self.fuel}
for string, variable in output.items():
print(string+': {}'.format(variable))
print('')
class Moonlander:
def __init__(self):
self.acceleration = random.randint(10, 30) / 10 # 1-3 by 0.1
self.wind = 0.2 * (random.random() - 0.5)
self.max_safe_landing_speed = 4.0
self.min_safe_x = -0.2
self.max_safe_x = 0.2
self.landed = 'in_air'
self.height = 100.0
self.x_position = 0.0
self.y_velocity = 10.0 * random.random()
self.x_velocity = 0.0
self.fuel = 100.0
self.burn = None
self.thrust = None
self.controller = FuzzyControl()
def control(self):
"""
This needs to get burn and thrust from the ANN
"""
return self.controller.control_input(self.height, self.y_velocity,
self.x_position, self.x_velocity,
self.wind)
def test(self):
if self.height > 0:
return 'in_air'
if (self.y_velocity > self.max_safe_landing_speed
or self.x_position < self.min_safe_x
or self.x_position > self.max_safe_x):
return 'crashed'
else:
return 'landed'
def update(self):
self.y_velocity += self.acceleration
burn, thrust = self.control()
if self.fuel < burn:
burn = self.fuel
self.fuel -= abs(burn)
self.y_velocity -= burn
if self.fuel < abs(thrust):
if thrust > 0:
thrust = self.fuel
elif thrust <= 0:
thrust = -self.fuel
self.fuel -= abs(thrust)
self.x_velocity -= thrust
self.height -= self.y_velocity
self.x_position += (self.x_velocity + self.wind)
self.landed = self.test()
self.output(burn, thrust)
def output(self, burn, thrust):
output = {
'Height': self.height,
'Y-Velocity': self.y_velocity,
'Burn': burn,
'Wind': self.wind,
'X-Velocity': self.x_velocity,
'X-Position': self.x_position,
'Thrust': thrust,
'Fuel': self.fuel
}
for string, variable in output.items():
print(string + ': {}'.format(variable))
print('')
from fuzz.visualization import VisManager
from fuzzy_control import FuzzyControl
controller = FuzzyControl()
sets = {
'height': controller.height_sets,
'y_velocity': controller.y_velocity_sets,
'position': controller.position_sets,
'x_velocity': controller.x_velocity_sets
}
plugins = {}
for name, num in sets.items():
for key, item in num.items():
plugins[name + '_' + key] = VisManager.create_backend(item)
for name, plugin in plugins.items():
print(name, plugin)
(vis_format, vis_data) = plugin.visualize()
with (open("{}.{}".format(name, vis_format), "wb")) as fp:
fp.write(vis_data)
