from functools import cmp_to_key

from shapely import from_wkt
from shapely.geometry import LineString

# 定义车道线（假设为 LineString 对象）
lines_dict = {
    "4": "LINESTRING Z(116.222517 40.147647 0, 116.222515 40.147649 0, 116.222455 40.147705 0, 116.222418 40.147741 0, 116.222252 40.1479 0)",
    "3": "LINESTRING Z(116.222472 40.14762 0, 116.222375 40.147716 0, 116.22227 40.14782 0, 116.222212 40.147877 0)",
    "-1": "LINESTRING Z(116.22238 40.147565 0, 116.222283 40.147661 0, 116.222121 40.147823 0)",
    "2": "LINESTRING Z(116.222436 40.147599 0, 116.222339 40.147694 0, 116.222177 40.147856 0)",
    "1": "LINESTRING Z(116.222401 40.147577 0, 116.222304 40.147674 0, 116.222141 40.147835 0)",
    "5": "LINESTRING Z(116.222558 40.147672 0, 116.222398 40.147818 0, 116.222287 40.147922 0)"
}


def cross_product(vec1, vec2):
    return vec1[0] * vec2[1] - vec1[1] * vec2[0]


def cross_sort(item1, item2):
    line1, line2 = item1[1], item2[1]
    line1_coords = line1.coords
    line2_coords = line2.coords
    start1, end1 = line1_coords[0], line1_coords[-1]
    start2, end2 = line2_coords[0], line2_coords[-1]

    vector1 = (end1[0] - start1[0], end1[1] - start1[1])
    vector2_start = (start2[0] - start1[0], start2[1] - start1[1])
    vector2_end = (end2[0] - start1[0], end2[1] - start1[1])

    cross_start = vector1[0] * vector2_start[1] - vector1[1] * vector2_start[0]
    cross_end = vector1[0] * vector2_end[1] - vector1[1] * vector2_end[0]

    # # 判断位置关系
    # if cross_start > 0 and cross_end > 0:
    #     return "left"
    # elif cross_start < 0 and cross_end < 0:
    #     return "right"
    # elif cross_start == 0 and cross_end == 0:
    #     return "collinear"
    # else:
    #     return "intersecting or mixed"

    if cross_start > 0 and cross_end > 0:
        return 1
    elif cross_start < 0 and cross_end < 0:
        return -1


def lines_sorted(lines_dict):
    if lines_dict and len(lines_dict) > 1:
        sorted_items = sorted(lines_dict.items(), key=cmp_to_key(cross_sort))
        for sorted_item in sorted_items:
            print(sorted_item)


if __name__ == '__main__':
    d = {}
    for k, v in lines_dict.items():
        d[k] = from_wkt(v)
    print(d)

    lines_sorted(d)