-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathmain.py
138 lines (107 loc) · 4.58 KB
/
main.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
# -*- coding: utf-8 -*-：
from grid import Hexagon
from drawing import *
from LL2UTM import LL2UTM_USGS
from style import readDrawingSetting
from func import readGids
# Load data
def readData(filename, dgids, dnum, minSpeed = 2, maxSpeed = 150):
flows = {}
grids = {}
for gid in dgids:
grids[gid] = Hexagon(gid, dnum)
with open(filename, 'r') as f:
f.readline()
while True:
line = f.readline().strip()
if line:
sl = line.split(',')
if float(sl[-2]) < minSpeed or float(sl[-2]) > maxSpeed:
continue
fid = int(sl[-4])
gid = int(sl[-1])
if fid not in flows:
flows[fid] = [(), ()]
x, y = LL2UTM_USGS(float(sl[-5]), float(sl[-6]))
if sl[-3] == '1':
flows[fid][0] = (x, y)
if gid in dgids:
grids[gid].addOutFlow(fid)
elif sl[-3] == '0':
flows[fid][1] = (x, y)
if gid in dgids:
grids[gid].addInFlow(fid)
else:
break
return grids, flows
# Load data within the 5th ring road
def readData_Inside(filename, dgids, dnum, minSpeed = 2, maxSpeed = 150):
flows = {}
grids = {}
for gid in dgids:
grids[gid] = Hexagon(gid, dnum)
with open(filename, 'r') as f:
f.readline()
while True:
line1 = f.readline().strip()
if line1:
line2 = f.readline().strip()
sl1 = line1.split(',')
sl2 = line2.split(',')
if int(sl1[1]) == 0 or int(sl2[1]) == 0:
continue
if float(sl1[-2]) < minSpeed or float(sl1[-2]) > maxSpeed:
continue
fid = int(sl1[-4])
ogid = int(sl1[-1])
dgid = int(sl2[-1])
ox, oy = LL2UTM_USGS(float(sl1[-5]), float(sl1[-6]))
dx, dy = LL2UTM_USGS(float(sl2[-5]), float(sl2[-6]))
flows[fid] = [(ox, oy), (dx, dy)]
assert ogid in grids and dgid in grids
grids[ogid].addOutFlow(fid)
grids[dgid].addInFlow(fid)
else:
break
return grids, flows
# Visualize
def SIPatterns(dataFileName, dgids, ia, scale, mode, inside=False):
if inside:
grids, flows = readData_Inside(dataFileName+'.csv', dgids, ia['dnum'])
saveFileName = './figure/p_' + dataFileName[10:] + '_' + mode + '_in.jpg'
else:
grids, flows = readData(dataFileName + '.csv', dgids, ia['dnum'])
saveFileName = './figure/p_' + dataFileName[10:] + '_' + mode + '.jpg'
if mode == 'pm_bs':
drawPattern_bs(grids, flows, ia, scale, saveFileName)
elif mode == 'pm_bc':
drawPattern_bc(grids, flows, ia, saveFileName)
def SIPatterns_highlight(fileName, dgids, ia, mode='bs'):
grids, flows = readData(fileName + '.csv', dgids, ia['dnum'])
saveFileName = './figure/p_' + fileName[-15:] + '_' + mode + '.jpg'
if mode == 'pm_bs':
drawPattern_bs_highlight(grids, flows, ia, saveFileName)
def singlePattern(gid, fileName, dgids, ia, mode='bs'):
grids, flows = readData(fileName + '.csv', dgids, ia['dnum'])
saveFileName = './figure/p_' + fileName[-15:] + '_' + str(gid) + '.jpg'
if mode == 'pm_bs':
drawSinglePattern_bs(gid, grids, flows, ia, saveFileName)
if __name__ == '__main__':
# -----------------------------data----------------------------------
fileNames = ['./data/sj_051316_0105','./data/sj_051316_0509', './data/sj_051316_0913',
'./data/sj_051316_1317', './data/sj_051316_1721', './data/sj_051316_2101']
scale = '1km' # '1km' or '500m'
mode = 'pm_bs'
dgids = readGids('./data/5th_rr_hexagon_'+scale+'.csv')
ia = readDrawingSetting(mode, scale)
# -----------------------------drawing--------------------------------
#drawGlyph_bs(ia)
SIPatterns(fileNames[4]+'_'+scale, dgids, ia, scale, mode, True) # True means visualizing data with the 5th ring road.
#SIPatterns_sp(fileNames[0]+'_'+scale, dgids, ia, mode)
#singlePattern(392, fileNames[0]+'_'+scale, dgids, ia, mode)
# if False:
# labels = ['T', 'S', 'C', 'R']
# #gids = [124, 150, 437, 356] #scale = 1km
# gids = [124, 150, 437, 165] # scale = 1km
# colors = ['#eaff56', '#44cef6', '#ff461f', '#bddd22']
# difVar([fileNames[i]+'_'+scale for i in [1,0,2,3,4,5]], dgids, gids, labels, colors, 0.7, 6)