#!/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=r"xyz.qgz", required=False)
    parser.add_argument("-file", type=str, default=r"D:\LF\data\xyz\tif.txt", required=False)
    parser.add_argument("-out", type=str, default=r"D:\LF\data\xyz", required=False)
    parser.add_argument("-min", type=int, default=12, required=False)
    parser.add_argument("-max", type=int, default=15, required=False)
    parser.add_argument("-noData", type=int, default=255, required=False)

    return parser.parse_args()


# 读取文本文件
def read_txt(path):
    f = open(path, encoding="utf-8")
    lines = f.readlines()
    f.close()

    for i in range(0, len(lines)):
        lines[i] = lines[i].replace('\n', '')

    return lines


# 加载图层
def load_layers(prj, args):
    for layer in prj.mapLayers().values():
        prj.removeMapLayer(layer)

    layer = None
    args.crs = None
    lines = read_txt(args.file)
    for i in range(0, len(lines)):
        if len(lines[i]) == 0:
            continue

        strs = lines[i].split(",")
        prefix = "__" if "1" == strs[1] else "_"
        print("layer" + prefix + str(i) + ": " + strs[0])
        layer = QgsRasterLayer(strs[0], "layer" + prefix + str(i))

        if not layer.isValid() or layer.crs() is None:
            print("layer" + prefix + str(i) + ": failed to load!")
            continue

        if args.crs is None and len(layer.crs().authid()) > 0:
            args.crs = layer.crs()
            prj.setCrs(layer.crs())
            print("prj.authid: " + args.crs.authid())

        for j in range(1, layer.bandCount() + 1):
            try:
                layer.dataProvider().setNoDataValue(j, args.noData)
            except Exception as e:
                print(e)

        prj.addMapLayer(layer)

    if args.crs is None and layer is not None:
        args.crs = layer.crs()
        prj.setCrs(layer.crs())


# 边界类
class Rectangle:
    def __init__(self, xmin, xmax, ymin, ymax):
        self.xmin = xmin
        self.xmax = xmax
        self.ymin = ymin
        self.ymax = ymax


# 获取XYZ参数
def get_xyz_ops(args):
    ops = {
        'BACKGROUND_COLOR': QColor(0, 0, 0, 0),
        'DPI': 96,
        'EXTENT': args.ext,
        'METATILESIZE': 4,
        'OUTPUT_DIRECTORY': args.out,
        # 'OUTPUT_HTML': 'TEMPORARY_OUTPUT',
        'OUTPUT_HTML': args.out + "\\view.html",
        'QUALITY': 100,
        'TILE_FORMAT': 0,
        'TILE_HEIGHT': 256,
        'TILE_WIDTH': 256,
        'TMS_CONVENTION': False,
        'ZOOM_MAX': args.max,
        'ZOOM_MIN': args.min
    }
    print(ops)

    return ops


# 创建XYZ瓦片
def create_xyz(prj, args):
    import processing

    authid = args.crs.authid()
    for layer in prj.mapLayers().values():
        if "layer__" in layer.id():
            continue

        rect = layer.extent()
        if authid != layer.crs().authid():
            transform = QgsCoordinateTransform(layer.crs(), args.crs, prj)
            min = QgsPoint(rect.xMinimum(), rect.yMinimum())
            max = QgsPoint(rect.xMaximum(), rect.yMaximum())
            min.transform(transform)
            max.transform(transform)

            args.ext = str(min.x()) + "," + str(max.x()) + "," + \
                       str(min.y()) + "," + str(max.y()) + ' [' + authid + ']'
        else:
            args.ext = str(rect.xMinimum()) + "," + str(rect.xMaximum()) + "," + \
                       str(rect.yMinimum()) + "," + str(rect.yMaximum()) + ' [' + authid + ']'

        ops = get_xyz_ops(args)
        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))
    # prj.read(args.qgz)
    print("FileName: " + prj.fileName())

    load_layers(prj, args)
    # prj.write(args.file.replace(".txt", ".qgz"))

    # args.ext = getExtent(prj, args)
    create_xyz(prj, args)

    qgs.exitQgis()


# main
if __name__ == '__main__':
    timer = time.time()
    init()
    print(f'耗时:{time.time() - timer:.2f}s')