def main():
system = Reader().load_from_file("napiwek.fcl")
d = Doc(".")
d.createDoc(system)
input_variables = {
"obsluga": 3.,
"jedzenie": 8.,
}
output_variables = {
"napiwek": None
}
system.calculate(input_variables, output_variables)
print(output_variables)
d.createDocSets(
{"accumulated": system.variables["napiwek"].defuzzify.accumulated_set},
"accumulated set"
)
d.createDocSets(system.variables["napiwek"].defuzzify.activated_sets, "activated sets")
#other possibilitis:
import matplotlib.pyplot as plt
plt.figure(6)
d.build2DPlot(plt.gca(), system, "obsluga", "napiwek")
plt.figure(7)
ax = plt.gca(projection="3d")
d.build3DPlot(ax, system, 'obsluga', 'jedzenie', 'napiwek')
ax.set_zlim(*d.getValues(system.variables['napiwek']))
plt.show()
class FuzzyController:
def __init__(self, fcl_path):
self.system = Reader().load_from_file(fcl_path)
def _make_input(self, world):
return dict(cp=world.x,
cv=world.v,
pa=degrees(world.theta),
pv=degrees(world.omega))
def _make_output(self):
return dict(force=0.)
def decide(self, world):
output = self._make_output()
self.system.calculate(self._make_input(world), output)
return output['force']
def get_fuzzy_size(image):
current_dir = os.path.dirname(os.path.realpath(__file__))
controller = Reader().load_from_file(
"{}/size.fcl".format(current_dir)
)
width, height = image.size
input_variables = {
"width": width,
"height": height
}
output_variables = {
"size": None
}
controller.calculate(input_variables, output_variables)
return output_variables
class Cart:
def __init__(self, canvas):
self.history = defaultdict(list)
# view
self.canvas = canvas
self.last_x = 400
self.rod_len = 80
self.rod_y_start = 300-20-30
rod_y_end = self.rod_y_start - self.rod_len
self.rect = canvas.create_rectangle(self.last_x-35, 300-20-30, self.last_x+35, 300-20)
self.w1 = canvas.create_oval(self.last_x-30, 300-20, self.last_x-10, 300)
self.w2 = canvas.create_oval(self.last_x+10, 300-20, self.last_x+30, 300)
self.ball = canvas.create_oval(self.last_x - 10, rod_y_end - 10,
self.last_x + 10, rod_y_end + 10
, fill="gray")
self.rod = canvas.create_line(self.last_x, self.rod_y_start,
self.last_x, rod_y_end
, width=3, fill="gray")
self.x_offset = 400
# simulation
self.x = 0.0
self.x_dot = 0.0
self.x_dot_dot = 0.0
self.theta = 0.0
self.theta_dot = 0.0
self.theta_dot_dot = 0.0
self.time_scale = 0.01
self.f = 0
# controller
self.input = {}
self.output = {"F":0}
self.controller = Reader().load_from_file("inverted_pendulum.fcl")
def initialize(self):
self.x = 0.0
self.x_dot = 0.0
self.x_dot_dot = 0.0
self.initial_theta = randint(0, 360)
self.theta = radians(self.initial_theta)
self.theta_dot = 0.0
self.theta_dot_dot = 0.0
self.f = 0
self.history = defaultdict(list)
def simulate(self):
print("x:{} x_dot:{} x_dot_dot:{} theta:{} theta_dot:{} theta_dot_dot:{}"
.format(self.x, self.x_dot, self.x_dot_dot, self.theta, self.theta_dot, self.theta_dot_dot))
N = m*(self.x_dot_dot-l*self.theta_dot*self.theta_dot*sin(self.theta)+l*self.theta_dot_dot*cos(self.theta))
P = g + m*(l*self.theta_dot*self.theta_dot*cos(self.theta)+l*self.theta_dot_dot*sin(self.theta))
self.x_dot_dot = 1/M*(self.f-N-b*self.x_dot)
self.theta_dot_dot = 1/I*(-N*l*cos(self.theta)-P*l*sin(self.theta))
self.x_dot += self.x_dot_dot*self.time_scale
self.theta_dot += self.theta_dot_dot * self.time_scale
self.theta = (self.theta + self.theta_dot * self.time_scale) % (2 * pi)
self.x += self.x_dot*self.time_scale
def control(self):
self.input["theta"] = degrees(self.theta)
self.input["theta_dot"] = self.theta_dot
self.input["theta_dot_dot"] = self.theta_dot_dot
self.input["x"] = self.x
self.input["x_dot"] = self.x_dot
self.input["x_dot_dot"] = self.x_dot_dot
self.controller.calculate(self.input, self.output)
self.f = self.output["F"]
print(str(self.f))
def update_view(self):
delta_x = (self.x + self.x_offset) % 800 - self.last_x
self.last_x = (self.x + self.x_offset) % 800
theta = self.theta - radians(180)
self.canvas.move(self.w1, delta_x, 0)
self.canvas.move(self.w2, delta_x, 0)
self.canvas.move(self.rect, delta_x, 0)
mass_x, mass_y = int(self.last_x-sin(theta)*self.rod_len), int(self.rod_y_start-cos(theta)*self.rod_len)
rod_x_1, rod_y_1 = int(self.last_x), int(self.rod_y_start)
rod_x_2, rod_y_2 = int(self.last_x-sin(theta)*self.rod_len), int(self.rod_y_start-cos(theta)*self.rod_len)
self.canvas.coords(self.ball, mass_x - 10, mass_y - 10,
mass_x + 10, mass_y + 10)
self.canvas.coords(self.rod, rod_x_1, rod_y_1,
rod_x_2, rod_y_2)
def update(self):
self.simulate()
self.update_view()
self.control()
self.add_stats()
def finish(self):
self.show_stats()
def add_stats(self):
self.history['x'].append(self.x)
self.history['x_dot'].append(self.x_dot)
self.history['x_dot_dot'].append(self.x_dot_dot)
self.history['theta - 180'].append(degrees(self.theta) - 180)
self.history['theta_dot'].append(self.theta_dot)
self.history['theta_dot_dot'].append(self.theta_dot_dot)
self.history['f'].append(self.f)
def show_stats(self):
plots = ['x', 'x_dot', 'x_dot_dot', 'theta - 180', 'theta_dot', 'theta_dot_dot', 'f']
time_lable = [i for i in range(len(self.history['x']))]
plt.suptitle('Report for initial_theta: {}'.format(self.initial_theta - 180))
for i in range(len(plots)):
plt.subplot(3, 3, i+1)
plt.plot(time_lable, self.history[plots[i]], '-')
plt.title(plots[i])
plt.axhline(y=0, color='gray', linestyle='-', alpha=0.7)
plt.tight_layout(rect=[0, 0.03, 1, 0.95])
plt.show()
class Cart:
def __init__(self, canvas):
# view
self.canvas = canvas
self.last_x = 400
self.rod_len = 80
self.rod_y_start = 300 - 20 - 30
rod_y_end = self.rod_y_start - self.rod_len
self.rect = canvas.create_rectangle(self.last_x - 35, 300 - 20 - 30,
self.last_x + 35, 300 - 20)
self.w1 = canvas.create_oval(self.last_x - 30, 300 - 20,
self.last_x - 10, 300)
self.w2 = canvas.create_oval(self.last_x + 10, 300 - 20,
self.last_x + 30, 300)
self.ball = canvas.create_oval(self.last_x - 10,
rod_y_end - 10,
self.last_x + 10,
rod_y_end + 10,
fill="gray")
self.rod = canvas.create_line(self.last_x,
self.rod_y_start,
self.last_x,
rod_y_end,
width=3,
fill="gray")
# simulation
self.x = 400.0
self.x_dot = 0.0
self.x_dot_dot = 0.0
self.theta = radians(10)
self.theta_dot = 0.0
self.theta_dot_dot = 0.0
self.time_scale = 0.01
self.F = 0
# controller
self.input = {}
self.output = {"F": 0}
self.controller = Reader().load_from_file("inverted_pendulum.fcl")
def initialize(self):
self.x = 400.0
self.x_dot = 0.0
self.x_dot_dot = 0.0
self.theta = radians(randint(0, 360))
self.theta_dot = 0.0
self.theta_dot_dot = 0.0
self.time_scale = 0.01
self.F = 0
def simulate(self):
# print("x:{} x_dot:{} x_dot_dot:{} theta:{} theta_dot:{} theta_dot_dot:{}"
# .format(self.x, self.x_dot, self.x_dot_dot, self.theta, self.theta_dot, self.theta_dot_dot))
N = m * (self.x_dot_dot - l * self.theta_dot * self.theta_dot *
sin(self.theta) + l * self.theta_dot_dot * cos(self.theta))
P = g + m * (l * self.theta_dot * self.theta_dot * cos(self.theta) +
l * self.theta_dot_dot * sin(self.theta))
self.x_dot_dot = 1 / M * (self.F - N - b * self.x_dot)
self.theta_dot_dot = 1 / I * (-N * l * cos(self.theta) -
P * l * sin(self.theta))
self.x_dot += self.x_dot_dot * self.time_scale
self.theta_dot += self.theta_dot_dot * self.time_scale
self.theta = (self.theta + self.theta_dot * self.time_scale) % (2 * pi)
self.x += self.x_dot * self.time_scale
def control(self):
self.input["theta"] = degrees(self.theta)
self.input["theta_dot"] = self.theta_dot
self.input["x_dot"] = self.x_dot
self.controller.calculate(self.input, self.output)
self.F = self.output["F"]
# print(self.F)
def update_view(self):
delta_x = self.x % 800 - self.last_x
self.last_x = self.x % 800
theta = self.theta - radians(180)
self.canvas.move(self.w1, delta_x, 0)
self.canvas.move(self.w2, delta_x, 0)
self.canvas.move(self.rect, delta_x, 0)
mass_x, mass_y = int(self.last_x - sin(theta) *
self.rod_len), int(self.rod_y_start -
cos(theta) * self.rod_len)
rod_x_1, rod_y_1 = int(self.last_x), int(self.rod_y_start)
rod_x_2, rod_y_2 = int(self.last_x - sin(theta) *
self.rod_len), int(self.rod_y_start -
cos(theta) * self.rod_len)
self.canvas.coords(self.ball, mass_x - 10, mass_y - 10, mass_x + 10,
mass_y + 10)
self.canvas.coords(self.rod, rod_x_1, rod_y_1, rod_x_2, rod_y_2)