#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Tue Aug  2 12:38:31 2022

@author: syed
"""
import regex_spatial
from utils import geoutil
import geopandas as gpd
import pandas as pd
import re
from shapely.geometry import Polygon,mapping
import numpy as np
from shapely.geometry import Polygon, MultiPoint, LineString, Point
from shapely.geometry.base import geom_factory
from shapely.geos import lgeos
from geocoder import geo_level1



def get_level3(level3):
    digits = re.findall('[0-9]+', level3)[0]
    unit = re.findall('[A-Za-z]+', level3)[0]
    return digits, unit


def get_directional_coordinates_by_angle(coordinates, centroid, direction, minimum, maximum):
    direction_coordinates = []
    for p in coordinates:
        angle = geoutil.calculate_bearing(centroid, p)
        p2= (p[0],p[1],angle)
        if direction in geo_level1.east:
            if angle >= minimum or angle <= maximum:
                direction_coordinates.append(p2)
                
        else:
             if angle >= minimum and angle <= maximum:
                direction_coordinates.append(p2)
    #print(type(direction_coordinates[0])) 
    #if(direction in geo_level1.west):
    #    direction_coordinates.sort(key=lambda k: k[2], reverse=True)
   
    return direction_coordinates 

def sort_west(poly1, poly2, centroid):
    coords1 = mapping(poly1)["features"][0]["geometry"]["coordinates"]
    coords2 = mapping(poly2)["features"][0]["geometry"]["coordinates"]
    coord1 = []
    coord2 = []
    coord = []
    for c in coords1:
        pol = list(c[::-1])
        coord1.extend(pol)
    for c in coords2:
        pol = list(c[::-1])
        coord2.extend(pol)
    coo1 = []
    coo2 = []
    for p in coord1:
        angle = geoutil.calculate_bearing(centroid, p)
        if angle >= 157 and angle <= 202:
            coo1.append((p[0], p[1], angle))
    for p in coord2:
        angle = geoutil.calculate_bearing(centroid, p)
        if angle >= 157 and angle <= 202:
            coo2.append((p[0], p[1], angle))
    coo1.extend(coo2)
    return coo1
    
def get_direction_coordinates(coordinates, centroid, level1):
    min_max = geo_level1.get_min_max(level1)
    if min_max is not None:
        coord = get_directional_coordinates_by_angle(coordinates, centroid, level1, min_max[0], min_max[1])
        return coord
    return coordinates
    
def get_level3_coordinates(coordinates, centroid, level_3, level1):
    distance, unit = get_level3(level_3)
    
    kms = geoutil.get_kilometers(distance, unit)
    print(distance, unit, kms, 'duk')
    
    coord = []
    
    poly1 = Polygon(coordinates)
    polygon1 = gpd.GeoSeries(poly1)
    poly2 = polygon1.buffer(0.0095*kms, join_style=2)                       # 扩大小面积
    poly3 = polygon1.buffer(0.013*kms, join_style=2)                        # 扩大大面积
    poly = poly3.difference(poly2)                                          # 合二为一成为环状
    coords = mapping(poly)["features"][0]["geometry"]["coordinates"]        # 改为坐标格式
    # print(mapping(poly2)["features"][0]["geometry"]["coordinates"], 'cmcm')
    # print(mapping(poly3)["features"][0]["geometry"]["coordinates"], 'cmcm')
    # print(coords, 'cmcm')

    
    for c in coords:
        pol = list(c[::-1])
        coord.extend(pol)
    if level1 is not None:
        coord = get_direction_coordinates(coord, centroid, level1)
        if level1 in geo_level1.west:
            coord = sort_west(poly3, poly2, centroid)
    print("Level 3 Coordinates")
    for idx, p in enumerate(coord):
        print(idx, p)
    return coord, centroid


def get_between_coordinates(coordinates1, coordinates2, centroid1, centroid2):
    # 创建多边形对象
    poly1 = Polygon(coordinates1)
    poly2 = Polygon(coordinates2)

    # 计算中心点连线
    center_line = LineString([centroid1, centroid2])

    def max_perpendicular_distance(poly, line):
        max_dist = 0
        farthest_points = None

        for point in poly.exterior.coords:
            p = Point(point)
            # 计算点到中心线的垂直距离
            dist = p.distance(line)
            if dist > max_dist:
                max_dist = dist
                farthest_points = p

        return max_dist * 2  # 计算直径（双倍最大垂直距离）

    # 计算两个区域的最大垂线距离
    diameter1 = max_perpendicular_distance(poly1, center_line)
    diameter2 = max_perpendicular_distance(poly2, center_line)

    # 计算平均直径 R
    R = (diameter1 + diameter2) / 2

    # 计算圆心（两个中心点的中点）
    midpoint = ((centroid1[0] + centroid2[0]) / 2, (centroid1[1] + centroid2[1]) / 2)

    # 生成圆形区域
    circle = Point(midpoint).buffer(R / 2, resolution=100)

    # 获取圆的坐标
    circle_coords = list(circle.exterior.coords)

    return [circle_coords], midpoint