import math

def wgs84_to_web_mercator(lon, lat):
    """
    將WGS84經(jīng)緯度轉(zhuǎn)換為Web墨卡托坐標(biāo)（米）。
    Args:
        lon: 經(jīng)度
        lat: 緯度
    Returns:
        (x, y) 單位：米
    """
    x = lon * 20037508.34 / 180.0
    y = math.log(math.tan((90.0 + lat) * math.pi / 360.0)) / (math.pi / 180.0)
    y = y * 20037508.34 / 180.0
    return x, y

def web_mercator_to_wgs84(x, y):
    """
    將Web墨卡托坐標(biāo)轉(zhuǎn)換回WGS84經(jīng)緯度。
    Args:
        x: 墨卡托x坐標(biāo)（米）
        y: 墨卡托y坐標(biāo)（米）
    Returns:
        (lon, lat)
    """
    lon = x / 20037508.34 * 180.0
    lat = y / 20037508.34 * 180.0
    lat = 180.0 / math.pi * (2 * math.atan(math.exp(lat * math.pi / 180.0)) - math.pi / 2.0)
    return lon, lat

def deg2num(lon_deg, lat_deg, zoom):
    """
    將經(jīng)緯度轉(zhuǎn)換為特定縮放級(jí)別下的瓦片索引。
    Args:
        lon_deg: 經(jīng)度 (WGS84)
        lat_deg: 緯度 (WGS84)
        zoom: 縮放級(jí)別 (0-通常為最大)
    Returns:
        (xtile, ytile)
    """
    lat_rad = math.radians(lat_deg)
    n = 2.0 ** zoom
    # 先換算為0-1范圍內(nèi)的比例
    xtile = (lon_deg + 180.0) / 360.0 * n
    # 注意這里的公式，使用了tan和log
    ytile = (1.0 - math.log(math.tan(lat_rad) + (1 / math.cos(lat_rad))) / math.pi) / 2.0 * n
    # 關(guān)鍵步驟：使用 floor 確保索引正確
    return int(math.floor(xtile)), int(math.floor(ytile))

import matplotlib.pyplot as plt
import numpy as np
from PIL import Image

def get_tile_extent(xtile, ytile, zoom):
    """
    根據(jù)瓦片索引計(jì)算其地理范圍 (Web墨卡托坐標(biāo))。
    Args:
        xtile, ytile: 瓦片索引
        zoom: 縮放級(jí)別
    Returns:
        extent list: [x_min, x_max, y_min, y_max] in Web Mercator meters.
    """
    # 每個(gè)縮放級(jí)別的瓦片總數(shù)
    n = 2.0 ** zoom
    # 整個(gè)世界的墨卡托坐標(biāo)范圍
    world_size = 20037508.34 * 2
    # 單個(gè)瓦片的寬度/高度（米）
    tile_size = world_size / n

    # 計(jì)算當(dāng)前瓦片的邊界（米）
    x_min = -20037508.34 + xtile * tile_size
    x_max = -20037508.34 + (xtile + 1) * tile_size
    # 注意：墨卡托y軸原點(diǎn)在頂部，向下為正
    y_max = 20037508.34 - ytile * tile_size
    y_min = 20037508.34 - (ytile + 1) * tile_size

    return [x_min, x_max, y_min, y_max]

# 拼接示例
fig, ax = plt.subplots(figsize=(10, 8))
ax.set_aspect('equal') # 保持縱橫比，防止地圖變形

# 假設(shè)我們有三張相鄰的瓦片
tiles_to_draw = [(x, y) for x in range(10, 13) for y in range(5, 8)]
zoom = 12

for xtile, ytile in tiles_to_draw:
    # 1. 加載瓦片圖像（這里用隨機(jī)數(shù)組模擬）
    tile_path = f'./tiles/{zoom}/{xtile}/{ytile}.png'
    try:
        img = np.array(Image.open(tile_path)) # 或使用 matplotlib.image.imread
    except FileNotFoundError:
        continue # 處理缺失瓦片

    # 2. 計(jì)算該瓦片的地理范圍
    extent = get_tile_extent(xtile, ytile, zoom)

    # 3. 使用正確的extent和origin顯示
    ax.imshow(img, extent=extent, origin='upper', zorder=0)

# 現(xiàn)在，你可以用相同的數(shù)據(jù)坐標(biāo)（墨卡托米）在ax上繪制你的業(yè)務(wù)數(shù)據(jù)了
# ax.plot([x1, x2], [y1, y2], 'r-', linewidth=2) # 這會(huì)精確疊加在地圖上

import concurrent.futures
import requests
import os
from urllib.parse import urljoin

def download_single_tile(base_url, xtile, ytile, zoom, save_dir, max_retries=3):
    """下載單張瓦片，包含重試機(jī)制"""
    url = urljoin(base_url, f'{zoom}/{xtile}/{ytile}.png')
    save_path = os.path.join(save_dir, f'{zoom}/{xtile}/{ytile}.png')
    os.makedirs(os.path.dirname(save_path), exist_ok=True)

    for attempt in range(max_retries):
        try:
            # 設(shè)置超時(shí)，避免無限等待
            resp = requests.get(url, timeout=10)
            resp.raise_for_status() # 檢查HTTP狀態(tài)碼
            with open(save_path, 'wb') as f:
                f.write(resp.content)
            print(f'成功下載: {save_path}')
            return True
        except (requests.exceptions.RequestException, IOError) as e:
            print(f'下載失敗 {url}, 嘗試 {attempt+1}/{max_retries}: {e}')
            if attempt == max_retries - 1:
                return False
    return False

def download_tile_batch(tile_list, base_url, save_dir, max_workers=20):
    """并發(fā)下載一批瓦片"""
    with concurrent.futures.ThreadPoolExecutor(max_workers=max_workers) as executor:
        future_to_tile = {}
        for xtile, ytile, zoom in tile_list:
            future = executor.submit(download_single_tile, base_url, xtile, ytile, zoom, save_dir)
            future_to_tile[future] = (xtile, ytile, zoom)

        for future in concurrent.futures.as_completed(future_to_tile):
            xtile, ytile, zoom = future_to_tile[future]
            try:
                success = future.result()
                if not success:
                    # 記錄失敗的任務(wù)，后續(xù)可能用備用源或留空
                    print(f'瓦片 {zoom}/{xtile}/{ytile} 最終下載失敗')
            except Exception as exc:
                print(f'瓦片 {zoom}/{xtile}/{ytile} 生成異常: {exc}')

from PIL import Image

def merge_tiles_vertically(tile_dir, x_range, y_start, y_end, zoom):
    """垂直拼接一列瓦片，減少內(nèi)存占用"""
    column_images = []
    for y in range(y_start, y_end + 1):
        row_images = []
        for x in range(x_range[0], x_range[1] + 1):
            path = os.path.join(tile_dir, f'{zoom}/{x}/{y}.png')
            if os.path.exists(path):
                row_images.append(Image.open(path))
            else:
                # 用空白瓦片占位
                row_images.append(Image.new('RGB', (256, 256), (200, 200, 200)))
        # 水平拼接一行
        row_merged = Image.new('RGB', (256 * len(row_images), 256))
        x_offset = 0
        for img in row_images:
            row_merged.paste(img, (x_offset, 0))
            x_offset += img.width
        column_images.append(row_merged)

    # 垂直拼接所有行
    total_height = sum(img.height for img in column_images)
    result = Image.new('RGB', (column_images[0].width, total_height))
    y_offset = 0
    for img in column_images:
        result.paste(img, (0, y_offset))
        y_offset += img.height
        img.close() # 及時(shí)關(guān)閉，釋放內(nèi)存
    return result

class TileSource:
    """封裝不同瓦片源的配置"""
    SOURCES = {
        'osm': {
            'url_template': 'https://tile.openstreetmap.org/{z}/{x}/{y}.png',
            'projection': 'epsg:3857',
            'tms': False, # XYZ 方案
            'max_zoom': 19,
            'attribution': '? OpenStreetMap contributors'
        },
        # 示例：假設(shè)一個(gè)TMS源
        'custom_tms': {
            'url_template': 'http://example.com/{z}/{x}/{y}.png',
            'projection': 'epsg:3857',
            'tms': True, # TMS 方案，y軸原點(diǎn)在底部
            'max_zoom': 18,
        }
    }

    def __init__(self, source_key='osm'):
        self.config = self.SOURCES.get(source_key)
        if not self.config:
            raise ValueError(f'未知的瓦片源: {source_key}')

    def get_tile_url(self, x, y, z):
        """根據(jù)配置生成瓦片URL"""
        if self.config['tms']:
            # TMS 轉(zhuǎn) XYZ (y軸翻轉(zhuǎn))
            y = (2 ** z - 1) - y
        url = self.config['url_template'].format(x=x, y=y, z=z)
        return url

    def download_tile(self, x, y, z, save_path):
        """下載瓦片，處理可能的404"""
        url = self.get_tile_url(x, y, z)
        try:
            resp = requests.get(url, timeout=5, headers={'User-Agent': 'Your-App/1.0'})
            if resp.status_code == 200:
                with open(save_path, 'wb') as f:
                    f.write(resp.content)
                return True
            elif resp.status_code == 404:
                # 該位置可能無瓦片（如海洋），記錄并跳過
                print(f'瓦片不存在 (404): {url}')
                return False
            else:
                print(f'下載失敗，狀態(tài)碼 {resp.status_code}: {url}')
                return False
        except requests.exceptions.RequestException as e:
            print(f'網(wǎng)絡(luò)錯(cuò)誤: {e}')
            return False