| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566 |
- import random
- from qgis.PyQt.QtWidgets import QToolButton, QMenu, QMessageBox
- from .extra_map import add_tianditu_province_menu, add_extra_map_menu
- from .utils import add_raster_layer
- from .utils import get_map_uri
- from ..icons import icons
- from ...utils import TIANDITU_HOME_URL, PluginConfig, tianditu_map_url
- tianditu_map_info = {
- "vec": "天地图-矢量地图",
- "cva": "天地图-矢量注记",
- "img": "天地图-影像地图",
- "cia": "天地图-影像注记",
- "ter": "天地图-地形晕染",
- "cta": "天地图-地形注记",
- "ibo": "天地图-全球境界",
- }
- conf = PluginConfig()
- class AddMapBtn(QToolButton):
- def __init__(self, parent=None):
- super().__init__(parent)
- self.icons = icons
- self.setToolTip("添加地图")
- self.setup_action()
- def setup_action(self):
- menu = QMenu(self)
- menu.setObjectName("TianDiTuAddMap")
- for map_type, map_name in tianditu_map_info.items():
- menu.addAction(
- self.icons["map"],
- map_name,
- lambda maptype_=map_type: self.add_tianditu_basemap(maptype_),
- )
- menu.addSeparator()
- # 天地图省级节点
- add_tianditu_province_menu(menu)
- # 其他图源
- add_extra_map_menu(menu)
- self.setMenu(menu)
- self.setPopupMode(QToolButton.MenuButtonPopup)
- self.setIcon(self.icons["add"])
- def add_tianditu_basemap(self, maptype):
- key = conf.get_key()
- if key == "":
- QMessageBox.warning(
- self, "错误", "天地图Key未设置或Key无效", QMessageBox.Yes, QMessageBox.Yes
- )
- else:
- random_enabled = conf.get_bool_value("Tianditu/random")
- key_random_enabled = conf.get_bool_value("Tianditu/random_key")
- if random_enabled:
- subdomain = f"t{random.randint(0, 7)}"
- else:
- subdomain = conf.get_value("Tianditu/subdomain")
- if key_random_enabled:
- key = conf.get_random_key()
- map_url = tianditu_map_url(maptype, key, subdomain)
- uri = get_map_uri(map_url, 1, 18, TIANDITU_HOME_URL)
- add_raster_layer(uri, tianditu_map_info[maptype])
|
