import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(0, 100, 1000)

def tri(x, a, b, c):    
    return np.maximum(0, np.minimum((x-a)/(b-a), (c-x)/(c-b)))

def trap(x, a, b, c, d):
    return np.maximum(0, np.minimum(np.minimum((x-a)/(b-a), 1), (d-x)/(d-c)))

def gauss(x, c, sigma):
    return np.exp(-0.5 * ((x - c)/sigma)**2)

def sigmoid(x, c, k):
    return 1 / (1 + np.exp(-k*(x - c)))

# Plot all
plt.figure(figsize=(14, 8))
plt.plot(x, tri(x, 20, 50, 80), label='Triangular (Medium)', linewidth=3)
plt.plot(x, trap(x, 10, 30, 70, 90), label='Trapezoidal (Warm)', linewidth=3)
plt.plot(x, gauss(x, 50, 15), label='Gaussian (Around 50)', linewidth=3)
plt.plot(x, sigmoid(x, 60, 0.2), label='Sigmoid (Increasing)', linewidth=3)
plt.legend(fontsize=12)
plt.title('Membership Functions – Choose Wisely!', fontsize=16)
plt.xlabel('Universe (e.g., Temperature, Speed, Error)')
plt.ylabel('Degree of Membership μ(x)')
plt.grid(alpha=0.3)
plt.show()

Rule 1: IF slip_ratio is HIGH and speed is HIGH THEN brake_pressure = LOW
Rule 2: IF slip_ratio is MEDIUM THEN brake_pressure = MEDIUM
Rule 3: IF slip_ratio is LOW THEN brake_pressure = HIGH

import numpy as np
import matplotlib.pyplot as plt
import skfuzzy as fuzz
from skfuzzy import control as ctrl

# 1. Define Universe
temp = ctrl.Antecedent(np.arange(0, 41, 1), 'temperature')
fan = ctrl.Consequent(np.arange(0, 101, 1), 'fan_speed')

# 2. Membership Functions (Auto triangular/trapezoidal)
temp['cold'] = fuzz.trimf(temp.universe, [0, 0, 20])
temp['warm'] = fuzz.trimf(temp.universe, [10, 20, 30])
temp['hot'] = fuzz.trimf(temp.universe, [25, 35, 40])

fan['slow'] = fuzz.trimf(fan.universe, [0, 0, 50])
fan['medium'] = fuzz.trimf(fan.universe, [20, 50, 80])
fan['fast'] = fuzz.trimf(fan.universe, [60, 100, 100])

# 3. Fuzzy Rules
rule1 = ctrl.Rule(temp['cold'], fan['slow'])
rule2 = ctrl.Rule(temp['warm'], fan['medium'])
rule3 = ctrl.Rule(temp['hot'], fan['fast'])

# 4. Control System
fan_ctrl = ctrl.ControlSystem([rule1, rule2, rule3])
fan_sim = ctrl.ControlSystemSimulation(fan_ctrl)

#  # 5. Test!
temps = np.linspace(0, 40, 100)
speeds = []
for t in temps:
    fan_sim.input['temperature'] = t
    fan_sim.compute()
    speeds.append(fan_sim.output['fan_speed'])

# Plot
plt.figure(figsize=(12, 6))
temp.view()
fan.view()
plt.figure(figsize=(12, 6))
plt.plot(temps, speeds, 'b-', linewidth=4, label='Fuzzy Controller Output')
plt.fill_between(temps, 0, speeds, alpha=0.3)
plt.title('Fuzzy Temperature → Fan Speed Controller', fontsize=16)
plt.xlabel('Temperature (°C)')
plt.ylabel('Fan Speed (%)')
plt.grid(True)
plt.legend()
plt.show()

# Test single value
fan_sim.input['temperature'] = 28
fan_sim.compute()
print(f"At 28°C → Fan Speed = {fan_sim.output['fan_speed']:.1f}%")

# Using skfuzzy built-in
fan['slow'].view()
aggregated = np.fmax(
    np.fmin(rule1.antecedent.membership_value, fan['slow'].mf),
    np.fmax(
        np.fmin(rule2.antecedent.membership_value, fan['medium'].mf),
        np.fmin(rule3.antecedent.membership_value, fan['fast'].mf)
    )
)

# All methods
print("Defuzzification Results:")
print(f"Centroid     : {fuzz.defuzz(fan.universe, aggregated, 'centroid'):.1f}")
print(f"Bisector     : {fuzz.defuzz(fan.universe, aggregated, 'bisector'):.1f}")
print(f"Mean of Max  : {fuzz.defuzz(fan.universe, aggregated, 'mom'):.1f}")
print(f"Smallest Max : {fuzz.defuzz(fan.universe, aggregated, 'som'):.1f}")
print(f"Largest Max     : {fuzz.defuzz(fan.universe, aggregated, 'lom'):.1f}")

# Save as fuzzy_controller.py
import skfuzzy as fuzz
from skfuzzy import control as ctrl
import numpy as np

def create_ac_controller():
    temp = ctrl.Antecedent(np.arange(10, 41, 1), 'temp')
    humidity = ctrl.Antecedent(np.arange(0, 101, 1), 'humidity')
    ac = ctrl.Consequent(np.arange(0, 101, 1), 'ac_power')

    # Auto MFs
    temp.automf(names=['cool', 'good', 'hot'])
    humidity.automf(names=['dry', 'comfort', 'humid'])
    ac.automf(names=['low', 'medium', 'high'])

    # Rules
    rules = [
        ctrl.Rule(temp['hot'] | humidity['humid'], ac['high']),
        ctrl.Rule(temp['good'] & humidity['comfort'], ac['low']),
        ctrl.Rule(temp['cool'], ac['low']),
    ]

    ac_ctrl = ctrl.ControlSystem(rules)
    return ctrl.ControlSystemSimulation(ac_ctrl)

# Use it
ac = create_ac_controller()
ac.input['temp'] = 35
ac.input['humidity'] = 80
ac.compute()
print(f"AC Power: {ac.output['ac_power']:.1f}%")