Bicycle_Relocation/Drawing.py at main · Sigma4749/Bicycle_Relocation · 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
114
115
116
117
118
119
120
121
122
123
124
125
import math
import random
import numpy as np
import matplotlib.mlab as mlab
import matplotlib.pyplot as plt

def distance(point1, point2):
    return (   (point1[0]-point2[0])**2  +  (point1[1]-point2[1])**2   )**0.5

def label_correction(current_point, labels_positions):
    increments = [[0,-1.75],[0,1.75]]
    new_current_point = current_point
    tries = -1
    while(tries < len(increments)):
        nearest_point = labels_positions[0]
        for dummy_index in range(1,len(labels_positions)):
            if (distance(new_current_point, labels_positions[dummy_index]) < distance(new_current_point, nearest_point) ):
                nearest_point = labels_positions[dummy_index]
        if (distance(new_current_point, nearest_point) > 0.25 ):
            return new_current_point
        else:
            tries += 1
            new_current_point = [current_point[0]+increments[tries][0], current_point[1]+increments[tries][1]]

    return new_current_point


def draw_graph(vertices, aristas, MIN_WIDTH, MIN_HEIGHT, MAX_WIDTH, MAX_HEIGHT):
    labels_positions = [] #avoid superposing labels
    for dummy_index in range(len(vertices)):
        plt.plot(vertices[dummy_index][0],vertices[dummy_index][1], ".", color = "black", markersize = 4)
        plt.text(vertices[dummy_index][0]+0.1, vertices[dummy_index][1]+0.1, str(dummy_index), fontsize=8)
        labels_positions.append([vertices[dummy_index][0]+0.1, vertices[dummy_index][1]+0.1])

        # Subindices vertices
        if (vertices[dummy_index][2] != 0 and dummy_index < 10):
        	plt.text(vertices[dummy_index][0], vertices[dummy_index][1], "    "+str(vertices[dummy_index][2]), fontsize=4)
        elif (vertices[dummy_index][2] != 0 and dummy_index >= 10):
        	plt.text(vertices[dummy_index][0], vertices[dummy_index][1], "    "+str(vertices[dummy_index][2]), fontsize=4)

    # edge labels
    for dummy_index in range(len(edges)):
        point1 = (vertices[edges[dummy_index][0]][0], vertices[edges[dummy_index][0]][1])
        point2 = (vertices[edges[dummy_index][1]][0], vertices[edges[dummy_index][1]][1])
        mid_point = [0.5*(point1[0]+point2[0]), 0.5*(point1[1]+point2[1])]

        plt.annotate(text='', xy=point1, xytext=point2, arrowprops=dict(arrowstyle='<-', color="blue",  alpha=0.95, linewidth=0.25*random.random()+0.35))

        if(len("["+str(edges[dummy_index][0])+","+str(edges[dummy_index][1])+"]") >= 6):
            correct_position = label_correction([mid_point[0]-0.5,mid_point[1]], labels_positions)
            plt.text(correct_position[0], correct_position[1], "     "+str(edges[dummy_index][2]), color= "brown", fontsize=7)
            plt.text(correct_position[0], correct_position[1]-0.1, "["+str(edges[dummy_index][0])+","+str(edges[dummy_index][1])+"]" , color= "brown", fontsize=4)
            labels_positions.append([correct_position[0], correct_position[1]])
        else:
            correct_position = label_correction([mid_point[0]-0.25,mid_point[1]], labels_positions)
            plt.text(correct_position[0], correct_position[1], "    "+str(edges[dummy_index][2]), color= "brown", fontsize=7)
            plt.text(correct_position[0], correct_position[1]-0.1, "["+str(edges[dummy_index][0])+","+str(edges[dummy_index][1])+"]" , color= "brown", fontsize=4)
            labels_positions.append([correct_position[0], correct_position[1]])

        if(edges[dummy_index][3] >= 10):
            correct_position = label_correction([mid_point[0]+0.15,mid_point[1]], labels_positions)
            plt.text(correct_position[0], correct_position[1], "        "+str(edges[dummy_index][3]), color= "red", fontsize=7)
            plt.text(correct_position[0], correct_position[1], "                       ["+str(edges[dummy_index][0])+","+str(edges[dummy_index][1])+"]" , color= "red", fontsize=4)
            labels_positions.append([correct_position[0], correct_position[1]])
        else:
            correct_position = label_correction([mid_point[0]+0.25,mid_point[1]], labels_positions)
            plt.text(correct_position[0], correct_position[1], "       "+str(edges[dummy_index][3]), color= "red", fontsize=7)
            plt.text(correct_position[0], correct_position[1]-0.1, "                     ["+str(edges[dummy_index][0])+","+str(edges[dummy_index][1])+"]" , color= "red", fontsize=4)
            labels_positions.append([correct_position[0], correct_position[1]])

    for dummy_index in range(len(vertices)):
        if (vertices[dummy_index][2] != 0 and dummy_index < 10):
            plt.text(vertices[dummy_index][0], vertices[dummy_index][1], "    "+str(vertices[dummy_index][2]), fontsize=4)
        elif (vertices[dummy_index][2] != 0 and dummy_index >= 10):
            plt.text(vertices[dummy_index][0], vertices[dummy_index][1], "    "+str(vertices[dummy_index][2]), fontsize=4)

    plt.savefig('Instance.pdf', format='pdf')


    plt.pause(0.00000005)   # Mini-pause before closing plot
    plt.show()


print("\nDrawing instance...")
""" Read vertices file """
vertices = []

with open("vertices_data.txt") as f:
    line = f.readline()
    MAX_WIDTH = int(line)
    line = f.readline()
    MAX_HEIGHT = int(line)
    line = f.readline()
    MIN_WIDTH  = MAX_WIDTH
    MIN_HEIGHT = MAX_HEIGHT
    while True:
        line = f.readline()
        if line == '':
            break
        row = line.split(' ')
        if ([int(row[0]), int(row[1]) ]  not in vertices):
                vertices.append( [int(row[0]), int(row[1]), int(row[2]) ] )
                if (MIN_WIDTH > int(row[0])):
                    MIN_WIDTH = int(row[0])
                if (MIN_HEIGHT > int(row[1])):
                    MIN_HEIGHT = int(row[1])
f.close()


""" Read edges file """
edges = []
with open("arcs_data.txt") as h:
    line = h.readline()
    while True:
        line = h.readline()
        if line == '':
            break
        row = line.split(' ')
        if ([int(row[0]), int(row[1]) ]  not in vertices):
                edges.append( [int(row[0]), int(row[1]), int(row[2]), int(row[3]) ] )
h.close()


draw_graph(vertices, edges, MIN_WIDTH, MIN_HEIGHT, MAX_WIDTH, MAX_HEIGHT)