#!/usr/bin/env python3
# Time-stamp: <2023-08-15 12:35:22 ycopin>

"""
Exemple un peu plus complexe de figure, incluant 2 axes, légendes, axes, etc.
"""

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-np.pi, 3*np.pi, 2*360)

# Signal carré
y = np.sign(np.sin(x))            # = ± 1

# 3 premiers termes de la décomposition en série de Fourier
y1 = 4/np.pi * np.sin(x)          # Fondamentale
y2 = 4/np.pi * np.sin(3*x) / 3    # 1re harmonique
y3 = 4/np.pi * np.sin(5*x) / 5    # 2de harmonique
# Somme des 3 premières composantes
ytot = y1 + y2 + y3

# Figure
fig = plt.figure()                # Création de la Figure

# 1er axe: composantes
ax1 = fig.add_subplot(2, 1, 1,  # 1er axe d'une série de 2 × 1
                      ylabel="Composantes",
                      title="Décomposition en série de Fourier")
ax1.plot(x, y1, label="Fondamental")
ax1.plot(x, y2, label="1re harmonique")
ax1.plot(x, y3, label="2de harmonique")
ax1.legend(loc="upper left", fontsize="x-small")

# 2nd axe: décomposition
ax2 = fig.add_subplot(2, 1, 2,  # 2d axe d'une série de 2 × 1
                      ylabel="Décomposition",
                      xlabel="x [rad]")
ax2.plot(x, y, lw=2, color='k', label="Signal carré")
ax2.plot(x, ytot, lw=2, ls=':', color='k', label="Somme des composantes")
ax2.legend(loc="upper left", fontsize="x-small")

# Sauvegarde de la figure (pas d'affichage intéractif)
fig.savefig("figure.png")