Automatic-Control-Theory/lab_3.py at master · shinbatsu/Automatic-Control-Theory · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
import numpy as np
import matplotlib.pyplot as plt
from scipy import signal


def define_systems(T, T2, K, eps):
    """Определение трёх различных передаточных функций."""
    return {
        'K/(Ts+1)': signal.TransferFunction([K], [T, 1]),
        'K/((Ts+1)(T2s+1))': signal.TransferFunction([K], [T*T2, T+T2, 1]),
        'K/(T^2s^2 + 2T2(eps)s + 1)': signal.TransferFunction([K], [T**2, 2*T2*eps, 1])
    }


def plot_bode_magnitude(systems, w):
    """Построение АЧХ."""
    plt.figure()
    for name, sys in systems.items():
        _, mag, _ = signal.bode(sys, w=w)
        plt.semilogx(w, mag, label=name)
    plt.title('АЧХ (Amplitude-Frequency Response)')
    plt.xlabel('Frequency (rad/s)')
    plt.ylabel('Magnitude (dB)')
    plt.legend()
    plt.grid(True)


def plot_bode_phase(systems, w):
    """Построение ФЧХ."""
    plt.figure()
    for name, sys in systems.items():
        _, _, phase = signal.bode(sys, w=w)
        plt.semilogx(w, phase, label=name)
    plt.title('ФЧХ (Phase-Frequency Response)')
    plt.xlabel('Frequency (rad/s)')
    plt.ylabel('Phase (degrees)')
    plt.legend()
    plt.grid(True)


def plot_log_bode(systems, w):
    """Построение логарифмической АЧХ и ФЧХ."""
    plt.figure()
    for name, sys in systems.items():
        _, mag, _ = signal.bode(sys, w=w)
        plt.semilogx(w, mag, label=name)
    plt.title('ЛАЧХ (Logarithmic Amplitude-Frequency Response)')
    plt.xlabel('Frequency (rad/s)')
    plt.ylabel('Log Magnitude (dB)')
    plt.legend()
    plt.grid(True)

    plt.figure()
    for name, sys in systems.items():
        _, _, phase = signal.bode(sys, w=w)
        plt.semilogx(w, phase, label=name)
    plt.title('ЛФЧХ (Logarithmic Phase-Frequency Response)')
    plt.xlabel('Frequency (rad/s)')
    plt.ylabel('Log Phase (degrees)')
    plt.legend()
    plt.grid(True)


def plot_amplitude_phase(systems, w):
    """Построение АФЧХ (Амплитуда-Фаза)."""
    plt.figure()
    for name, sys in systems.items():
        _, mag, phase = signal.bode(sys, w=w)
        plt.plot(mag, phase, label=name)
    plt.title('АФЧХ (Amplitude-Phase Response)')
    plt.xlabel('Magnitude (dB)')
    plt.ylabel('Phase (degrees)')
    plt.legend()
    plt.grid(True)


def main():
    # Задание параметров
    T = 0.01 * 4
    T2 = T * 2
    K = 4.0
    eps = 0.4
    w = np.logspace(-2, 2, 500)

    systems = define_systems(T, T2, K, eps)

    # Построение графиков
    plot_bode_magnitude(systems, w)
    plot_bode_phase(systems, w)
    plot_log_bode(systems, w)
    plot_amplitude_phase(systems, w)

    plt.show()


if __name__ == "__main__":
    main()