-
Notifications
You must be signed in to change notification settings - Fork 0
/
route.py
103 lines (78 loc) · 2.64 KB
/
route.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
# -*- coding: utf-8 -*-
"""
Created on Sat Jun 07 16:30:50 2014
@author: Yunsheng Wei
"""
import matplotlib.pyplot as plt
from helper import dist
class Route:
def __init__(self, route_no, sites):
"""
route_no should be int
"""
self.route_no = route_no
self.sites = sites
self.site_backup = self.sites
self.total_len = None
def get_no(self):
return self.route_no
def get_location(self, i):
"""
return as (longitude, latitude)
"""
return (self.get_longitude(i), self.get_latitude(i))
def site_location(self, i):
return (self.site_backup[i][0], self.site_backup[i][1])
def site_count(self):
return len(self.site_backup)
def get_latitude(self, i):
return self.sites[i][1]
def get_longitude(self, i):
return self.sites[i][0]
def get_dest_loc(self):
return self.get_location(-1)
def get_start_loc(self):
return self.get_location(0)
def __len__(self):
"""
count of sites
"""
return len(self.sites)
def __iter__(self):
return iter(self.sites)
def plot(self, start, end, *args1, **args2):
"""
plot route data on a 2d plane on current figure.
[start, end)
"""
x, y = zip(*[self.get_location(i) for i in xrange(start, end)])
plt.plot(x, y, *args1, **args2)
x, y = zip(*[self.site_location(i) for i in xrange(len(self.site_backup))])
plt.plot(x, y, 'ro', markersize = 7)
def length(self):
"""
total distance between adjacent sites,
use lines to link adjacent sites.
"""
if self.total_len == None:
self.total_len = sum(dist(self.get_location(i), self.get_location(i - 1))
for i in xrange(1, len(self)))
return self.total_len
def add_supplement_info(self, sites):
self.site_backup = self.sites
self.sites = sites
class Routes:
def __init__(self):
self.routes = {}
def get_route_len(self, i):
return len(self.routes[i])
def add(self, route):
self.routes[route.get_no()] = route
def get_route(self, no):
return self.routes[no]
def all_routes_no(self):
return self.routes.keys()
def __len__(self):
return len(self.routes)
def __iter__(self):
return self.routes.itervalues()