#!/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)

   
    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)

    
    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"]        # 改为坐标格式

    
    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 = (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