-
Notifications
You must be signed in to change notification settings - Fork 3
/
04distantmatrix_milan.py
151 lines (126 loc) · 6.93 KB
/
04distantmatrix_milan.py
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
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
import numpy as np
import sys
import re
import time
import os
dict_sensor2id = {"D001": 0, "D002": 1, "D003": 2, "M001": 3, "M002": 4, "M003": 5, "M004": 6, "M005": 7, "M006": 8, "M007": 9, "M008": 10, "M009": 11, "M010": 12, "M011": 13, "M012": 14, "M013": 15, "M014": 16, "M015": 17, "M016": 18, "M017": 19, "M018": 20, "M019": 21, "M020": 22, "M021": 23, "M022": 24, "M023": 25, "M024": 26, "M025": 27, "M026": 28, "M027": 29, "M028": 30, "T001": 31, "T002": 32}
def dict_key2value(dict, key):
try:
return dict[key]
except:
print('Key input error, not found!')
def dict_value2key(dict, value):
try:
return list(dict.keys())[list(dict.values()).index(value)]
except:
print('Value input error, not found!')
def init_matrix(shape=(94, 94)):
distance_matrix = np.zeros(shape)
return distance_matrix
def save_matrix(matrix_name, matrix_value, path='./', delimiter='\t', encoding=None):
np.savetxt(matrix_name, matrix_value, fmt='%.0f', delimiter='\t')
return 'OK'
if __name__ == '__main__':
dataset_name = 'milan'
matrix_value = ''
opdir = os.path.join(os.getcwd(), 'datasets', 'distant')
matrix_name = os.path.join(opdir, dataset_name + '-distance_matrix.txt')
assert (os.path.exists(matrix_name),
'Please confirm that the file distance matrix file exists first! Otherwise, please create it first and initialize an empty one.')
operate_select = '' # 0: initialize 1: view 2: repair 3: save
while operate_select != 10: # Setting 10 is because it requires pressing two keys at a long distance to avoid wrong pressing
print(operate_select)
operate_select = int(input('\n''\nPlease enter the action you want to perform: \n%s\n%s\n%s\n%s\n%s\n' %
('0\tinitialization',
'1\tView data',
'2\tRepair and improve data',
'3\tmodel save',
'10\texit')
))
if operate_select == 0:
print('Initialization operation, all zeroing processing...')
matrix_value = init_matrix(shape=(len(dict_sensor2id), len(dict_sensor2id)))
print('shape: %s' % str(matrix_value.shape))
print(save_matrix(matrix_name, matrix_value)) # The default is to save as an integer
if operate_select == 1:
matrix_value = np.loadtxt(matrix_name)
print('shape: %s' % str((matrix_value.shape)))
print(dict_sensor2id.keys())
sensor_one = input('Please enter the name of the sensor to view 1：')
sensor_two = input('Please enter the name of the sensor to view 2：')
sensor_one_value = int(dict_key2value(dict_sensor2id, sensor_one))
sensor_two_value = int(dict_key2value(dict_sensor2id, sensor_two))
print('(%s,%s)=>(%d,%d) distance: %d' %
(sensor_one, sensor_two,
sensor_one_value, sensor_two_value,
matrix_value[sensor_one_value, sensor_two_value]))
print('location: \n%s' % (matrix_name))
if operate_select == 2:
operate_sensors = ['AD1-A', 'AD1-B', 'AD1-C', 'M', 'D']
matrix_value = np.loadtxt(matrix_name)
copy_matrix_value = matrix_value.copy()
s = matrix_value.shape[0]
row_number = -1
for i in range(1, int(s)):
if float(copy_matrix_value[i][0]) == 0:
row_number = i
break
if row_number == -1:
print('You have completed all the data')
else:
print('Suggestion: last operation to% d lines' % (row_number))
print('Generally, it is the lower triangular matrix of operation...\n')
row_number = int(input('Please enter the line you want to modify from? Cannot be greater than %s' % (s)))
column_number = int(input('Please enter the column you want to modify from?'))
if row_number > int(s) - 1 or column_number >= row_number:
print('There is a problem with your input.')
sys.exit(999)
keypress = ''
while keypress != 'Q':
try:
if int(s) > int(row_number) > int(column_number):
row_sensor = dict_value2key(dict_sensor2id, int(row_number))
column_sensor = dict_value2key(dict_sensor2id, int(column_number))
print('The row sensor to operate is: %s' % row_sensor)
print('The column sensor to operate is: %s' % column_sensor)
if re.sub('[\d]{3}', '', row_sensor) not in operate_sensors or \
re.sub('[\d]{3}', '', column_sensor) not in operate_sensors:
print(
'The current distance is not within the calculation range，(%s, %s) is initialized as inf' % (
row_sensor, column_sensor))
input_value = np.PINF # np.PINF np.NINF
time.sleep(0.2)
else:
input_value = input('current value：%d\t row:%d column:%d\t\t[(%s, %s)]\t：'
% (copy_matrix_value[row_number][column_number],
row_number, column_number,
row_sensor,
column_sensor))
copy_matrix_value[row_number][column_number] = input_value
column_number += 1
if column_number == row_number:
column_number = 0
row_number += 1
print('-' * 30)
print('\nThis line is over and will go to the next new line, sensor is：%s ' %
(dict_value2key(dict_sensor2id, int(row_number))))
if row_number > s - 1:
break
except ValueError:
keypress = input_value
print('exit \'Q\'')
print('What you entered is %s' % (keypress))
finally:
pass
confirm_edit = input('Are you sure you want to save the new matrix, Y/ ')
if confirm_edit:
print('Saving...')
from tools.uppertriangular2symmetry import low2sym
matrix_value = low2sym(copy_matrix_value)
print(save_matrix(matrix_name, matrix_value))
else:
print('You didnot save, that is, you didnot do anything...')
elif operate_select == 3:
save_matrix(matrix_name, matrix_value, path='./')
pass
print('exit')