GeneticAlgorithmAndCom/ploting.py at main · ermakov-dmitry/GeneticAlgorithmAndCom · 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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
#!/usr/bin/env python
# -*- coding: utf-8 -*-

import matplotlib.pyplot as plt
from matplotlib.patches import Rectangle
import tikzplotlib


def draw_rectangle(x, y, length, width):
    plt.plot([x, x], [y, y + width], linewidth=1, color='k')
    plt.plot([x, x + length], [y + width, y + width], linewidth=1, color='k')
    plt.plot([x + length, x + length], [y + width, y], linewidth=1, color='k')
    plt.plot([x + length, x], [y, y], linewidth=1, color='k')


def draw_small_rectangle(x, y, length, width):
    plt.plot([x, x], [y, y + width], linewidth=1, color='k')
    plt.plot([x + length, x + length], [y + width, y], linewidth=1, color='k')
    plt.plot([x + length, x], [y, y], linewidth=1, color='k')


def plot_rectangles():
    # first rectangle
    draw_rectangle(1, 6, 5, 3)
    draw_small_rectangle(5, 5, 1, 1)
    plt.text(3.25, 7, '1', fontsize=12)

    # second rectangle
    draw_rectangle(1, 12, 5, 3)
    draw_small_rectangle(5, 11, 1, 1)
    plt.text(3.25, 13, '2', fontsize=12)

    # third rectangle
    draw_rectangle(1, 18, 5, 3)
    draw_small_rectangle(5, 17, 1, 1)
    plt.text(3.25, 19, '3', fontsize=12)

    # 4th rectangle
    draw_rectangle(9, 6, 5, 3)
    draw_small_rectangle(13, 5, 1, 1)
    plt.text(11.25, 7, '4', fontsize=12)

    # 5th rectangle
    draw_rectangle(9, 12, 5, 3)
    draw_small_rectangle(13, 11, 1, 1)
    plt.text(11.25, 13, '5', fontsize=12)

    # 6th rectangle
    draw_rectangle(9, 18, 5, 3)
    draw_small_rectangle(13, 17, 1, 1)
    plt.text(11.25, 19, '6', fontsize=12)

    # 7th rectangle
    draw_rectangle(17, 6, 5, 3)
    draw_small_rectangle(17, 5, 1, 1)
    plt.text(19.25, 7, '7', fontsize=12)

    # 8th rectangle
    draw_rectangle(17, 12, 5, 3)
    draw_small_rectangle(17, 11, 1, 1)
    plt.text(19.25, 13, '8', fontsize=12)

    # 9th rectangle
    draw_rectangle(17, 18, 5, 3)
    draw_small_rectangle(17, 17, 1, 1)
    plt.text(19.25, 19, '9', fontsize=12)

    # zero rectangle
    draw_rectangle(9, 0, 5, 2)
    plt.text(11.25, 0.35, '0', fontsize=12)


def plot_solution(route):
    coordinats = [[14.5, 2.5],
                  [6.5, 5.5],
                  [6.5, 11.5],
                  [6.5, 17.5],
                  [14.5, 5.5],
                  [14.5, 11.5],
                  [14.5, 17.5],
                  [16.5, 5.5],
                  [16.5, 11.5],
                  [16.5, 17.5],
                  [6.5, 2.5],
                  [16.5, 2.5],
                  [8.5, 4.5],
                  [14.5, 4.5],
                  [8.5, 5.5],
                  [8.5, 10.5],
                  [14.5, 10.5],
                  [8.5, 11.5],
                  [8.5, 16.5],
                  [14.5, 16.5],
                  [8.5, 17.5]]
    x_route = []
    y_route = []
    # plt.figure(figsize=(10, 8), dpi=20)
    for idx in route:
        x_route.append(coordinats[idx][0])
        y_route.append(coordinats[idx][1])
    plt.plot(x_route, y_route, '--')
    for i in range(10):
        plt.plot(coordinats[i][0], coordinats[i][1], marker='.', color='k', markersize=2)
    for i in range(10, len(coordinats)):
        plt.plot(coordinats[i][0], coordinats[i][1], marker='.', color='k', markersize=1)
    plt.xlabel('$x$, m')
    plt.ylabel('$y$, m')
    plt.xlim([0, 25])
    plt.ylim([0, 25])

    plot_rectangles()
    tikzplotlib.save('test.tex')
    plt.show()