WatchRouteProblem/samples.py at main · NeSC-IV/WatchRouteProblem · 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
import shapely
import getopt
import sys
import random_polygons_generate
from polygons_coverage import PolygonCover, SelectPointFromPolygon
from draw_pictures import *


def getLineList(lines):
    lineList = []
    if (type(lines) == shapely.LineString):
        lineList.append(lines)
    elif (type(lines) == shapely.MultiLineString):
        for line in lines.geoms:
            lineList.append(line)
    return lineList


def GetSample(polygonList, freeSpace, dSample, image):
    sampleList = []
    for polygon in polygonList:
        pointList = []
        lineString = polygon.boundary
        lineList = getLineList(lineString.difference(freeSpace.boundary))
        for line in lineList:
            start = shapely.get_point(line, 0)
            if freeSpace.buffer(-20).contains(start):
                pointList.append(start)
            end = shapely.get_point(line, -1)
            if freeSpace.buffer(-20).contains(end):
                pointList.append(end)
        path = 0
        while (path < lineString.length):
            point = shapely.line_interpolate_point(lineString, path)
            if freeSpace.buffer(-20).contains(point):
                pointList.append(point)
            path += dSample

        sampleList.append(pointList)
    return sampleList


if __name__ == '__main__':
    edgeNum = None
    iterationNum = None
    coverageRate = 0.98
    try:
        opts, args = getopt.getopt(sys.argv[1:], 'he:i:c:')
    except getopt.GetoptError:
        print("Usage:")
        print(
            "python/python3 polygons_coverage.py -e <num_of_edge> -i <num_of_iteration> {-c <coverage_rate>}")
        print("For example:")
        print("python3 polygons_coverage.py -e 20 -i 10 {-c 0.98}")
        sys.exit(2)
    for opt, arg in opts:
        if opt == '-h':
            print("Usage:")
            print(
                "python/python3 polygons_coverage.py -e <num_of_edge> -i <num_of_iteration>")
            print("For example:")
            print("python3 polygons_coverage.py -e 20 -i 10 {-c 0.98}")
            sys.exit(0)
        if opt == '-e':
            edgeNum = int(arg)
        elif opt == '-i':
            iterationNum = int(arg)
        elif opt == '-c':
            coverageRate = float(arg)
    polygon = random_polygons_generate.GetPolygon(edgeNum)
    # polygon = shapely.Polygon([(0, 0), (10000, 0), (10000, 3000), (8000, 3000),
    #                            (8000, 6000), (10000, 6000), (10000, 10000), (0, 10000), (0, 6000), (2000, 6000), (2000, 3000), (0, 3000)])
    watcher = SelectPointFromPolygon(polygon)

    image = np.zeros((pic_size, pic_size, 3), dtype=np.uint8)
    DrawPolygon((pic_size, pic_size, 3), list(
        polygon.exterior.coords), (255, 255, 255), image)
    image = DrawPoints(image, watcher.x, watcher.y)
    cv2.imshow('polygons', image)
    print("Press any key to continue!")
    cv2.waitKey(0)
    cv2.destroyAllWindows()

    polygonCoverList = PolygonCover(
        polygon, 3000, 1-coverageRate, iterationNum)
    print("The number of convex polygonlen is " + str(len(polygonCoverList)))
    n = 0
    m = 255
    o = 255
    image = np.zeros((pic_size, pic_size, 3), dtype=np.uint8)

    freeSpace = shapely.Polygon()
    for p in polygonCoverList:
        p = p.simplify(0.05, preserve_topology=False)
        image = DrawPolygon((pic_size, pic_size, 3), list(
            p.exterior.coords), (o, n, m), image)
        n += 55
        if (n >= 255):
            m -= 55
        if (m <= 0):
            o -= 55
        freeSpace = freeSpace.union(p)
    sampleList = GetSample(
        polygonCoverList, freeSpace, 2000, image)
    print(sampleList)
    for sample in sampleList:
        for point in sample:
            DrawPoints(image, point.x, point.y)
    cv2.imshow('polygons', image)
    cv2.waitKey(0)
    cv2.destroyAllWindows()