# Python Programm zum Plotten der Daten des periodisch angetriebenen Pendels
# Attraktordiagramm vieler Pendel mit unterschiedlichen Amplitudenwerten (Feigenbaumdiagramm)

import matplotlib.pyplot as plt             # Python Bibliothek zum Plotten (siehe https://matplotlib.org/ )
from matplotlib import rcParams
import numpy as np                          # Python Bibliothek fuer Mathematisches (siehe https://numpy.org/ )

data_attr = np.genfromtxt("./GetriebenesPendel_attr_dia_a.dat")  # Einlesen der berechneten Daten

plt.rcParams["figure.figsize"] = (13,7)
plt.xlabel(r"$\rm A$")
plt.ylabel(r"$\rm \dot{\theta}(t_n)$")

num_attr_points = int((len(data_attr[0,:])-1)/2)
for i in range(num_attr_points):
    plt.scatter(data_attr[:,0],data_attr[:,i+1], c=data_attr[:,i+num_attr_points+1],s=1, vmin=-np.pi, vmax=np.pi ,cmap=plt.cm.gist_ncar)
cbar = plt.colorbar()
cbar.set_label(r"Winkel $\theta$", rotation=270)

# Bild-Ausgabe
plt.savefig("Attr_diag_a.png", dpi=400,bbox_inches="tight",pad_inches=0.05,format="png")
plt.close()