-
Notifications
You must be signed in to change notification settings - Fork 0
/
find_cross_line.py
61 lines (49 loc) · 2.74 KB
/
find_cross_line.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
import sympy.geometry as symg
import random
def findcross(one_line, second_line):
"""Линии состоят из [[x1,y1],[x2,y2]] для поиска пересечений,
исключая крайние точки. Если пересечения нет, на выходе """
x1, y1, x2, y2 = one_line[0][0], one_line[0][1], one_line[1][0], one_line[1][1]
x3, y3, x4, y4 = second_line[0][0], second_line[0][1], second_line[1][0], second_line[1][1]
p1, p2, p3, p4 = symg.Point(x1, y1), symg.Point(x2, y2), symg.Point(x3, y3), symg.Point(x4, y4)
l1, l2 = symg.Segment(p1, p2), symg.Segment(p3, p4)
pointxy = symg.intersection(l1, l2)
if pointxy:
pointxy = [pointxy[0].x, pointxy[0].y]
""" кажется, что if работает быстрее
try:
pointxy = [pointxy[0].x, pointxy[0].y]
except IndexError:
return pointxy
"""
return pointxy
def chose_poit(point_index_f, pointsall_f, lines_all_f):
"""point_index_f - индекс последней точки
pointsall_f - список всех точек
выбор следующей точки по критерию уникальности новой линии, возвращает точку [x,y]"""
one_more = 1
next_point1_f = []
rand_range = list(range(len(pointsall_f) - 1)) #.remove(point_index_f)
print(rand_range)
while one_more == 1:
chose1_f = random.choice(rand_range)
while chose1_f == point_index_f:
# чтобы не было совпадений с точкой начала
rand_range.remove(chose1_f)
chose1_f = random.choice(rand_range)
# выбор точки, куда потянеться линя из первой точки
next_point1_f = pointsall_f[chose1_f]
# создание линии для проверки на совпадение
p01 = symg.Point(pointsall_f[point_index_f][0], pointsall_f[point_index_f][1])
p02 = symg.Point(next_point1_f[0], next_point1_f[1])
one_more = 0
for j in range(len(lines_all_f) - 1):
p1 = symg.Point(lines_all_f[j][0], lines_all_f[j][1])
p2 = symg.Point(lines_all_f[j + 1][0], lines_all_f[j + 1][1])
# проверка на коллинеарность из-за возможных наложений линий при несовпадении их длин, начала и конца
if symg.Point.is_collinear(p1, p2, p01, p02):
one_more = 1
# удалить из списка неудачную точку
rand_range.remove(chose1_f)
break
return next_point1_f