LMServer/create_user_graph.py at master · hsiaoching/LMServer · 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
import math
from igraph import *

def create_user_graph(path):
    #path = [(0, 0), (3,5), (320,568)]

    g = Graph()
    g.add_vertices(len(path))

    for i in range(0, len(path)):
        v = g.vs[i]
        v["checked"] = False
        v["nid"] = str(v.index)
        v["lat"] = path[i][1]
        v["lng"] = path[i][0]

    edge_length = 0
    if path[0] == path[len(path)-1]:
        edge_length = len(path)-1
    else:
        edge_length = len(path)

    for i in range(0, edge_length-1):
        v1 = g.vs[i]
        v2 = g.vs[i+1]
        g.add_edges((v1.index,v2.index))
        (bearing1, bearing2) = bearing(v1, v2)
        g.es[i]["n"] = {v1["nid"]:bearing1, v2["nid"]:bearing2}
        length = math.hypot(v1["lng"]-v2["lng"], v1["lat"]-v2["lat"])
        g.es[i]["length"] = length
        g.es[i]["n1"] = v1["nid"]
        g.es[i]["n2"] = v2["nid"]
        g.es[i]["bearing1"] = bearing1
        g.es[i]["bearing2"] = bearing2
        g.es[i]["checked"] = False

    if path[0] == path[len(path)-1]:
        v1 = g.vs[edge_length-1]
        v2 = g.vs[0]
        g.add_edges((v1.index,v2.index))
        (bearing1, bearing2) = bearing(v1, v2)
        g.es[edge_length-1]["n"] = {v1["nid"]:bearing1, v2["nid"]:bearing2}
        length = math.hypot(v1["lng"]-v2["lng"], v1["lat"]-v2["lat"])
        g.es[edge_length-1]["length"] = length
        g.es[edge_length-1]["n1"] = v1["nid"]
        g.es[edge_length-1]["n2"] = v2["nid"]
        g.es[edge_length-1]["bearing1"] = bearing1
        g.es[edge_length-1]["bearing2"] = bearing2
        g.es[edge_length-1]["checked"] = False

    print g
    return g


def bearing(v1, v2):
    x = v2["lng"]-v1["lng"]
    y = v2["lat"]-v1["lat"]
    theta1 = rectify( math.degrees(math.atan2(y,x))+90 )
    theta2 = rectify( -180 + theta1 )
    return (theta1, theta2)


# path = [(159,326), (248,345), (296,441), (385,412), (477,424), (477,489), (270,571), (167,428), (159,326)]
# g = create_user_graph(path)
# for i, edge in enumerate(g.es):
#     print g.es[i]["bearing1"], g.es[i]["bearing2"]