import numpy as np
import matplotlib.pyplot as plt
import skfuzzy as fuzz

# Zakres zmiennej wyjściowej – poziom ataku
x_attack = np.linspace(0, 100, 1000)

# Funkcje przynależności – atak
attack_none = fuzz.trimf(x_attack, [0, 0, 30])
attack_careful = fuzz.trimf(x_attack, [50, 70, 90])
attack_strong = fuzz.trimf(x_attack, [80, 100, 100])

# Wykres
plt.figure(figsize=(10, 4))
plt.plot(x_attack, attack_none, label='Brak', color='blue')
plt.plot(x_attack, attack_careful, label='Ostrożny', color='orange')
plt.plot(x_attack, attack_strong, label='Zdecydowany', color='red')
plt.title('Funkcje przynależności – Poziom ataku')
plt.xlabel('Atak [%]')
plt.ylabel('Przynależność μ(x)')
plt.legend()
plt.grid(True)
plt.ylim(0, 1.1)
plt.show()