#!/usr/bin/env
|
# -*- coding: utf-8 -*-
|
|
|
import os
|
import sys
|
import time
|
|
sys.path.append(r"C:\Program Files\QGIS 3.16\apps\qgis-ltr\python\plugins")
|
import math
|
import argparse
|
from qgis.core import *
|
from qgis.gui import *
|
from qgis.PyQt.QtGui import *
|
from qgis.PyQt.QtCore import *
|
from qgis.PyQt.QtWidgets import *
|
from processing.core.Processing import Processing
|
|
|
# 获取完整路径
|
def get_full_path():
|
return os.path.split(sys.argv[0])[0]
|
|
|
# 获取参数
|
def get_args():
|
print("argv = ", sys.argv[1:])
|
parser = argparse.ArgumentParser(description='ArgUtils')
|
parser.add_argument("-src", type=str, default=get_full_path(), required=False)
|
parser.add_argument("-qgz", type=str, default="xyz.qgz", required=False)
|
parser.add_argument("-file", type=str, default=r"D:\xyz\dq.txt", required=False)
|
parser.add_argument("-out", type=str, default=r"D:\xyz\tiles\dq", required=False)
|
parser.add_argument("-min", type=int, default=4, required=False)
|
parser.add_argument("-max", type=int, default=18, required=False)
|
|
return parser.parse_args()
|
|
|
# 读取文本文件
|
def readTxt(filePath):
|
f = open(filePath, encoding="utf-8")
|
lines = f.readlines()
|
f.close()
|
|
for i in range(0, len(lines)):
|
lines[i] = lines[i].replace('\n', '')
|
|
return lines
|
|
|
# 加载图层
|
def loadLayers(prj, args):
|
for layer in prj.mapLayers().values():
|
prj.removeMapLayer(layer)
|
|
lines = readTxt(args.file)
|
for i in range(0, len(lines)):
|
if len(lines[i]) == 0:
|
continue
|
|
print("layer: " + lines[i])
|
layer = QgsRasterLayer(lines[i], "layer_" + str(i))
|
if not layer.isValid() or layer.crs() is None:
|
print("layer_" + str(i) + ": failed to load!")
|
continue
|
|
if len(prj.mapLayers()) == 0:
|
prj.setCrs(layer.crs())
|
|
for j in range(1, layer.bandCount() + 1):
|
layer.dataProvider().setNoDataValue(j, 0)
|
|
prj.addMapLayer(layer)
|
|
|
# 边界类
|
class Rectangle:
|
def __init__(self, xmin, xmax, ymin, ymax):
|
self.xmin = xmin
|
self.xmax = xmax
|
self.ymin = ymin
|
self.ymax = ymax
|
|
|
# 获取边界
|
def getExtent(prj):
|
ex = None
|
for layer in prj.mapLayers().values():
|
rect = layer.extent()
|
print(layer.name() + ", " + rect.toString())
|
|
if ex is None:
|
ex = Rectangle(rect.xMinimum(), rect.xMaximum(), rect.yMinimum(), rect.yMaximum())
|
continue
|
|
if rect.xMinimum() < ex.xmin:
|
ex.xmin = rect.xMinimum()
|
if rect.yMinimum() < ex.ymin:
|
ex.ymin = rect.yMinimum()
|
if rect.xMaximum() > ex.xmax:
|
ex.xmax = rect.xMaximum()
|
if rect.yMaximum() > ex.ymax:
|
ex.ymax = rect.yMaximum()
|
|
# '38399769,38403264,3558330,3561632'
|
return str(ex.xmin) + "," + str(ex.xmax) + "," + str(ex.ymin) + "," + str(ex.ymax)
|
|
|
# 获取边界2
|
def getExtent2(prj):
|
rt = None
|
for layer in prj.mapLayers().values():
|
if rt is None:
|
rt = layer.extent()
|
continue
|
|
rt = rt.combineExtentWith(layer.extent())
|
|
return str(rt.xMinimum()) + "," + str(rt.xMaximum()) + "," + str(rt.yMinimum()) + "," + str(rt.yMaximum())
|
|
|
# 获取XYZ参数
|
def getXYZOps(args, level):
|
return {
|
'BACKGROUND_COLOR': QColor(0, 0, 0, 0),
|
'DPI': 96,
|
'EXTENT': args.ext,
|
'METATILESIZE': 4,
|
'OUTPUT_DIRECTORY': args.out,
|
'OUTPUT_HTML': args.out + "\\view.html" if args.min == level else 'TEMPORARY_OUTPUT',
|
'QUALITY': 100,
|
'TILE_FORMAT': 0,
|
'TILE_HEIGHT': 256,
|
'TILE_WIDTH': 256,
|
'TMS_CONVENTION': True,
|
'ZOOM_MAX': level,
|
'ZOOM_MIN': level
|
}
|
|
|
# 创建XYZ瓦片
|
def createXYZ(args):
|
import processing
|
|
for i in range(args.min, args.max + 1):
|
ops = getXYZOps(args, i)
|
print(ops)
|
processing.run("qgis:tilesxyzdirectory", ops)
|
|
|
# 初始化
|
def init():
|
# QgsApplication.setPrefixPath("C:\Program Files\QGIS 3.16", True)
|
qgs = QgsApplication([], False)
|
qgs.initQgis()
|
Processing.initialize()
|
|
args = get_args()
|
prj = QgsProject.instance()
|
prj.read(os.path.join(args.src, args.qgz))
|
print("FileName: " + prj.fileName())
|
|
loadLayers(prj, args)
|
prj.write(args.file.replace(".txt", ".qgz"))
|
|
args.ext = getExtent(prj)
|
createXYZ(args)
|
|
qgs.exitQgis()
|
|
|
# main
|
if __name__ == '__main__':
|
timer = time.time()
|
init()
|
print(f'耗时:{time.time() - timer:.2f}s')
|
