import csv
import geopandas as gpd
def get_map_extent(shp_file_path):
    """
    讀取.shp文件并獲取其邊界范圍。
    :param shp_file_path: .shp文件路徑
    :return: 地圖邊界的范圍 (minx, miny, maxx, maxy)
    """
    # 讀取.shp文件
    gdf = gpd.read_file(shp_file_path)
    # 獲取地圖邊界
    boundary = gdf.geometry.total_bounds
    # 返回地圖范圍
    return boundary
def calculate_sheet_ranges(map_extent):
    """
    根據(jù)地圖邊界計(jì)算圖幅范圍，并按照南北緯和東西經(jīng)的5度分隔規(guī)則返回圖幅范圍的格式。
    :param map_extent: 地圖邊界的范圍 (minx, miny, maxx, maxy)
    :return: 圖幅范圍列表，每個(gè)元素包含(lon_start, lat_start, lon_end, lat_end, sheet_range)
    """
    minx, miny, maxx, maxy = map_extent
    # 初始化圖幅范圍列表
    sheet_ranges = []
    # 計(jì)算圖幅編號(hào)
    # for lon_start in range(int(minx // 5) * 5, int(maxx // 5 + 1) * 5, 5):
    #     for lat_start in range(int(miny // 5) * 5, int(maxy // 5 + 1) * 5, 5):
    for lon_start in range(int(minx // 5) * 5, int(maxx // 5 + 1) * 5, 5):
        for lat_start in range(int(miny // 5 + 1) * 5, int(maxy // 5 + 2) * 5, 5):  # GLC_FCS30D 緯度需要上調(diào)一格5度
            lon_end = lon_start + 5
            lat_end = lat_start + 5
            sheet_range = f"{'W' if lon_start < 0 else 'E'}{abs(lon_start)}" \
                          f"{'S' if lat_start < 0 else 'N'}{abs(lat_start)}"
            sheet_ranges.append(sheet_range)
    return sheet_ranges
def main():
    iso_data = []
    with open('../../ISO.csv', newline='') as csvfile:
        # 創(chuàng)建 CSV 讀取器
        csv_reader = csv.reader(csvfile)
        data = list(csv_reader)  # 將文件內(nèi)容讀取到內(nèi)存中
        # 初始化行數(shù)計(jì)數(shù)器
        iso_count = 0
        for country in data:
            iso_count += 1
            sids_country = country[0]
            iso_map_extent = [iso_count, sids_country]  # [1, 'BMU']
            # 示例文件路徑
            shp_file_path = fr'path_to_admin_division\{sids_country}.shp'
            # 獲取地圖范圍
            map_extent = get_map_extent(shp_file_path=shp_file_path)
            # 計(jì)算圖幅范圍
            sheet_ranges = calculate_sheet_ranges(map_extent=map_extent)
            # 打印結(jié)果
            print(iso_count, sids_country, map_extent)
            iso_map_extent.extend(sheet_ranges)
            iso_data.append(iso_map_extent)
            print('- Map Extent Num: ', len(sheet_ranges), iso_map_extent[2:], '\n')
if __name__ == '__main__':
    main()