Python 射线法判断一个点坐标是否在一个坐标区域内

Python 射线法判断一个点坐标是否在一个坐标区域内

class Point:
    lng = ''
    lat = ''
 
    def __init__(self, lng, lat):
        self.lng = lng
        self.lat = lat
 
 
# 求外包矩形
def get_polygon_bounds(points):
    length = len(points)
    top = down = left = right = points[0]
    for i in range(1, length):
        if points[i].lng > top.lng:
            top = points[i]
        elif points[i].lng < down.lng:
            down = points[i]
        else:
            pass
        if points[i].lat > right.lat:
            right = points[i]
        elif points[i].lat < left.lat:
            left = points[i]
        else:
            pass
    top_left = Point(top.lng, left.lat)
    top_right = Point(top.lng, right.lat)
    down_right = Point(down.lng, right.lat)
    down_left = Point(down.lng, left.lat)
    return [top_left, top_right, down_right, down_left]
 
 
# 判断点是否在外包矩形外
def is_point_in_rect(point, polygon_bounds):
    top_left = polygon_bounds[0]
    top_right = polygon_bounds[1]
    down_right = polygon_bounds[2]
    down_left = polygon_bounds[3]
    return (down_left.lng <= point.lng <= top_right.lng
            and top_left.lat <= point.lat <= down_right.lat)
 
 
def is_point_in_polygon(point, points):
    polygon_bounds = get_polygon_bounds(points)
    if not is_point_in_rect(point, polygon_bounds):
        return False
    length = len(points)
    point_start = points[0]
    flag = False
    for i in range(1, length):
        point_end = points[i]
        # 点与多边形顶点重合
        if (point.lng == point_start.lng and point.lat == point_start.lat) or (
                point.lng == point_end.lng and point.lat == point_end.lat):
            return True
        # 判断线段两端点是否在射线两侧
        if (point_end.lat < point.lat <= point_start.lat) or (
                point_end.lat >= point.lat > point_start.lat):
            # 线段上与射线 Y 坐标相同的点的 X 坐标
            if point_end.lat == point_start.lat:
                x = (point_start.lng + point_end.lng) / 2
            else:
                x = point_end.lng - (point_end.lat - point.lat) * (
                        point_end.lng - point_start.lng) / (
                        point_end.lat - point_start.lat)
            # 点在多边形的边上
            if x == point.lng:
                return True
            # 射线穿过多边形的边界
            if x > point.lng:
                flag = not flag
            else:
                pass
        else:
            pass
 
        point_start = point_end
    return flag
 
 
def test(input_lat=116.732617, input_lon=39.722676):
    # polyline 是多个坐标点，形如
    polyline = ['115.732617,40.722676', '117.732617,40.722676', '115.732617,38.722676',
    '117.732617,38.722676', '118.732617,39.722676']
    # polyline = []

    points = []
    for line in polyline:
        if line:
            try:
                lat, lon = line.split(',')
                points.append(Point(float(lat), float(lon)))
            except ValueError:
                pass
    if points:
        a = is_point_in_polygon(Point(float(input_lat), float(input_lon)), points)
        print(a)

test(116.732617, 39.722676)