sanya-dbms/processing/algs/gdal/GdalAlgorithmProvider.py

"""
***************************************************************************
    GdalAlgorithmProvider.py
    ---------------------
    Date                 : August 2012
    Copyright            : (C) 2012 by Victor Olaya
    Email                : volayaf at gmail dot com
***************************************************************************
*                                                                         *
*   This program is free software; you can redistribute it and/or modify  *
*   it under the terms of the GNU General Public License as published by  *
*   the Free Software Foundation; either version 2 of the License, or     *
*   (at your option) any later version.                                   *
*                                                                         *
***************************************************************************
"""

__author__ = 'Victor Olaya'
__date__ = 'August 2012'
__copyright__ = '(C) 2012, Victor Olaya'

import os
import importlib

from PyQt5.QtGui import QIcon
from processing.core.ProcessingConfig import ProcessingConfig, Setting
from qgis.PyQt.QtCore import QCoreApplication
from qgis.core import (QgsApplication,
                       QgsProcessingProvider,
                       QgsRuntimeProfiler)

pluginPath = os.path.normpath(os.path.join(
    os.path.split(os.path.dirname(__file__))[0], os.pardir))


class GdalAlgorithmProvider(QgsProcessingProvider):

    def __init__(self):
        super().__init__()
        self.algs = []
        self._algorithm_definitions = None
        QgsApplication.processingRegistry().addAlgorithmAlias('qgis:buildvirtualvector', 'gdal:buildvirtualvector')

    def load(self):
        with QgsRuntimeProfiler.profile('GDAL Provider'):
            ProcessingConfig.settingIcons[self.name()] = self.icon()
            ProcessingConfig.addSetting(Setting(self.name(), 'ACTIVATE_GDAL',
                                                self.tr('Activate'), True))
            ProcessingConfig.readSettings()
            self.refreshAlgorithms()
        return True

    def unload(self):
        ProcessingConfig.removeSetting('ACTIVATE_GDAL')

    def isActive(self):
        return ProcessingConfig.getSetting('ACTIVATE_GDAL')

    def setActive(self, active):
        ProcessingConfig.setSettingValue('ACTIVATE_GDAL', active)

    def name(self):
        return '数管工具集'

    def longName(self):
        # 延迟导入GdalUtils，只在需要时才加载
        try:
            from .GdalUtils import GdalUtils
            version = GdalUtils.readableVersion()
        except ImportError:
            version = 'Unknown'
        return f'数管工具集 ({version})'

    def id(self):
        return 'gdal'

    def helpId(self):
        return 'gdal'

    def icon(self):
        return QIcon(os.path.join(pluginPath, 'images', 'dbms', 'tools.png'))

    def svgIconPath(self):
        return os.path.join(pluginPath, 'images', 'dbms', 'tools.png')

    def _get_algorithm_definitions(self):
        """获取算法定义列表，使用延迟加载"""
        if self._algorithm_definitions is None:
            self._algorithm_definitions = [
                # GDAL Raster Tools
                ('AssignProjection', '.AssignProjection'),
                ('aspect', '.aspect'),
                ('buildvrt', '.buildvrt'),
                ('ClipRasterByExtent', '.ClipRasterByExtent'),
                ('ClipRasterByMask', '.ClipRasterByMask'),
                ('ColorRelief', '.ColorRelief'),
                ('contour', '.contour'),
                ('contour_polygon', '.contour'),
                ('Datasources2Vrt', '.Datasources2Vrt'),
                ('fillnodata', '.fillnodata'),
                ('gdalinfo', '.gdalinfo'),
                ('gdal2tiles', '.gdal2tiles'),
                ('gdal2xyz', '.gdal2xyz'),
                ('gdaladdo', '.gdaladdo'),
                ('gdalcalc', '.gdalcalc'),
                ('gdaltindex', '.gdaltindex'),
                ('GridAverage', '.GridAverage'),
                ('GridDataMetrics', '.GridDataMetrics'),
                ('GridInverseDistance', '.GridInverseDistance'),
                ('GridInverseDistanceNearestNeighbor', '.GridInverseDistanceNearestNeighbor'),
                ('GridLinear', '.GridLinear'),
                ('GridNearestNeighbor', '.GridNearestNeighbor'),
                ('hillshade', '.hillshade'),
                ('merge', '.merge'),
                ('nearblack', '.nearblack'),
                ('pct2rgb', '.pct2rgb'),
                ('polygonize', '.polygonize'),
                ('proximity', '.proximity'),
                ('rasterize', '.rasterize'),
                ('rearrange_bands', '.rearrange_bands'),
                ('retile', '.retile'),
                ('rgb2pct', '.rgb2pct'),
                ('roughness', '.roughness'),
                ('sieve', '.sieve'),
                ('slope', '.slope'),
                ('translate', '.translate'),
                ('tpi', '.tpi'),
                ('tri', '.tri'),
                ('warp', '.warp'),
                ('pansharp', '.pansharp'),
                ('ExtractProjection', '.extractprojection'),
                ('rasterize_over', '.rasterize_over'),
                ('rasterize_over_fixed_value', '.rasterize_over_fixed_value'),

                # OGR Vector Tools
                ('Buffer', '.Buffer'),
                ('ClipVectorByExtent', '.ClipVectorByExtent'),
                ('ClipVectorByMask', '.ClipVectorByMask'),
                ('Dissolve', '.Dissolve'),
                ('ExecuteSql', '.ExecuteSql'),
                ('OffsetCurve', '.OffsetCurve'),
                ('ogr2ogr', '.ogr2ogr'),
                ('ogrinfo', '.ogrinfo'),
                ('OneSideBuffer', '.OneSideBuffer'),
                ('PointsAlongLines', '.PointsAlongLines'),
                ('VectorProject', '.VectorProject'),

                # PostGIS Tools
                ('OgrToPostGis', '.OgrToPostGis'),
                ('Ogr2OgrToPostGisList', '.ogr2ogrtopostgislist'),
                ('Ogr2PostGisList', '.rastertopostgislist'),
                ('Postgisupdate', '.postgisupdate'),
                ('Postgisrestore', '.postgisrestore'),
                ('Postgistogeoserver', '.postgistogeoserver'),
                ('ImportOSGBToPostGIS', '.ImportOSGBToPostGIS'),
                ('ImportTableToPostGIS', '.ImportTableToPostGIS'),
                ('ImportSingleOSGBToPostGIS', '.ImportSingleOSGBToPostGIS'),
                ('DataStorageStatistics', '.DataStorageStatistics'),
                ('GdbToPostGisList', '.gdbtopostgislist'),

                # Export Tools
                ('ExportVectorByMask', '.ExportVectorByMask'),
                ('ExportRasterByMask', '.ExportRasterByMask'),

                # Other Tools
                ('LicenseMake', '.LicenseMake'),

                # Topology Tools
                ('A00', '.Topology.A00'),
                ('A01', '.Topology.A01'),
                ('A02', '.Topology.A02'),
                ('A03', '.Topology.A03'),
                ('A04', '.Topology.A04'),
                ('A05', '.Topology.A05'),
                ('P00', '.Topology.P00'),
                ('P01', '.Topology.P01'),
                ('L00', '.Topology.L00'),
                ('L01', '.Topology.L01'),
                ('L02', '.Topology.L02'),
                ('L03', '.Topology.L03'),
                ('L04', '.Topology.L04'),
                ('L05', '.Topology.L05'),
                ('L06', '.Topology.L06'),
                ('L07', '.Topology.L07'),
                ('L08', '.Topology.L08'),
                ('L09', '.Topology.L09'),
                ('L10', '.Topology.L10'),
                ('PP01', '.Topology.PP01'),
                ('PP02', '.Topology.PP02'),
                ('PL01', '.Topology.PL01'),
                ('PL02', '.Topology.PL02'),
                ('PA01', '.Topology.PA01'),
                ('PA02', '.Topology.PA02'),
                ('PA03', '.Topology.PA03'),
                ('LP01', '.Topology.LP01'),
                ('LL01', '.Topology.LL01'),
                ('LL02', '.Topology.LL02'),
                ('LL03', '.Topology.LL03'),
                ('LL04', '.Topology.LL04'),
                ('LA01', '.Topology.LA01'),
                ('LA02', '.Topology.LA02'),
                ('LA03', '.Topology.LA03'),
                ('AP01', '.Topology.AP01'),
                ('AP02', '.Topology.AP02'),
                ('AL01', '.Topology.AL01'),
                ('AA01', '.Topology.AA01'),
                ('AA02', '.Topology.AA02'),
                ('AA03', '.Topology.AA03'),
                ('AA04', '.Topology.AA04'),
                ('AA05', '.Topology.AA05'),
            ]

        return self._algorithm_definitions

    def _load_algorithm_class(self, class_name, module_path):
        """延迟加载算法类"""
        try:
            module = importlib.import_module(module_path, package=__package__)
            return getattr(module, class_name)
        except (ImportError, AttributeError) as e:
            print(f"Warning: Could not load algorithm {class_name} from {module_path}: {e}")
            return None

    def _load_viewshed_if_supported(self):
        """检查GDAL版本并加载viewshed算法"""
        try:
            from osgeo import gdal
            if int(gdal.VersionInfo()) > 3010000:
                viewshed_class = self._load_algorithm_class('viewshed', '.viewshed')
                if viewshed_class:
                    return viewshed_class()
        except Exception as e:
            print(f"Warning: Could not check GDAL version or load viewshed: {e}")
        return None

    def loadAlgorithms(self):
        """使用延迟加载方式加载算法"""
        self.algs = []

        # 获取算法定义
        algorithm_definitions = self._get_algorithm_definitions()

        # 延迟加载每个算法
        for class_name, module_path in algorithm_definitions:
            algorithm_class = self._load_algorithm_class(class_name, module_path)
            if algorithm_class:
                try:
                    algorithm_instance = algorithm_class()
                    self.algs.append(algorithm_instance)
                except Exception as e:
                    print(f"Warning: Could not instantiate algorithm {class_name}: {e}")

        # 检查并加载viewshed算法（需要GDAL版本检查）
        viewshed_instance = self._load_viewshed_if_supported()
        if viewshed_instance:
            self.algs.append(viewshed_instance)

        # 将所有算法添加到提供者
        for algorithm in self.algs:
            self.addAlgorithm(algorithm)

    def supportedOutputRasterLayerExtensions(self):
        """延迟加载GdalUtils"""
        try:
            from .GdalUtils import GdalUtils
            return GdalUtils.getSupportedOutputRasterExtensions()
        except ImportError:
            return []

    def supportsNonFileBasedOutput(self):
        """
        GDAL Provider doesn't support non file based outputs
        """
        return False

    def tr(self, string, context=''):
        if context == '':
            context = 'GdalAlgorithmProvider'
        return QCoreApplication.translate(context, string)