本文档旨在指导用户如何使用 PyQGIS 脚本为 QGIS 项目中的每个图层组自动创建地图主题。通过脚本控制图层的可见性，并利用 QgsMapThemeCollection 类，可以高效地管理和切换不同的地图显示状态，从而提升地图项目的组织性和可交互性。
使用 PyQGIS 脚本生成地图主题
以下代码展示了如何使用 PyQGIS 为 QGIS 项目中的每个图层组自动创建地图主题。该脚本遍历项目中的所有图层组，为每个组创建一个地图主题，该主题仅显示该组中的图层。
# 获取项目图层树的根节点
root = QgsProject.instance().layerTreeRoot()
# 获取所有图层组
groups = root.findGroups()
# 获取所有图层
AllLayers = root.findLayers()
# 获取地图主题集合
mapThemesCollection = QgsProject.instance().mapThemeCollection()
# 隐藏所有图层
for layer in AllLayers:
layer.setItemVisibilityChecked(False)
# 隐藏所有图层组
for group in groups:
group.setItemVisibilityChecked(False)
# 遍历每个图层组
for group in groups:
# 显示当前图层组
group.setItemVisibilityChecked(True)
# 获取当前图层组中的所有图层
groupLayers = group.findLayers()
# 显示当前图层组中的所有图层
for groupLayer in groupLayers:
groupLayer.setItemVisibilityChecked(True)
# 为当前图层组创建地图主题
mapThemeRecord = QgsMapThemeCollection.createThemeFromCurrentState(
QgsProject.instance().layerTreeRoot(),
iface.layerTreeView().layerTreeModel()
)
# 将地图主题添加到地图主题集合中，以图层组的名称作为主题名称
mapThemesCollection.insert(str(group), mapThemeRecord)代码解释:
-
获取项目元素:
- QgsProject.instance().layerTreeRoot(): 获取当前 QGIS 项目的图层树根节点，是后续操作的基础。
- root.findGroups(): 查找图层树根节点下的所有图层组。
- root.findLayers(): 查找图层树根节点下的所有图层。
- QgsProject.instance().mapThemeCollection(): 获取当前 QGIS 项目的地图主题集合，用于存储和管理创建的地图主题。
-
设置初始可见性:
- 循环遍历所有图层和图层组，将它们的可见性设置为 False，确保在创建每个主题之前，所有图层都处于隐藏状态。
-
遍历图层组并创建主题:
- 外层循环遍历每个图层组。
- group.setItemVisibilityChecked(True): 显示当前正在处理的图层组。
- group.findLayers(): 查找当前图层组下的所有图层。
- 内层循环遍历当前图层组中的所有图层，并将它们的可见性设置为 True，确保只有当前图层组及其包含的图层是可见的。
- QgsMapThemeCollection.createThemeFromCurrentState(...): 基于当前图层树的状态（即哪些图层可见）创建一个新的地图主题。
- mapThemesCollection.insert(str(group), mapThemeRecord): 将新创建的地图主题添加到地图主题集合中。这里使用图层组的名称作为地图主题的名称，方便后续识别和管理。
注意事项
- iface 对象: 代码中使用了 iface 对象，它代表 QGIS 的主界面。 确保在 QGIS 环境中运行此脚本，或者在独立脚本中使用 QgisInterface 类来模拟 iface 对象。 在 QGIS 插件或控制台中运行此脚本通常不需要额外处理。
- 图层和图层组的命名: 建议图层和图层组使用有意义的名称，以便更好地组织和管理地图主题。
- 性能: 对于包含大量图层和图层组的项目，此脚本可能需要一些时间才能完成。 可以考虑优化代码，例如使用更高效的查找方法或并行处理。
- 依赖: 该脚本依赖于 PyQGIS 库，确保 QGIS 环境配置正确，并且可以访问 PyQGIS 模块。
总结
通过这个脚本，可以自动化地为 QGIS 项目中的每个图层组创建地图主题，极大地简化了地图项目的管理和维护。 这种方法特别适用于需要频繁切换不同图层组合的项目，例如地质调查、城市规划等。 掌握这种方法，可以显著提高 QGIS 的使用效率，并提升地图项目的专业性。
