1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161
|
"""
***************************************************************************
gdalinfo.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
from qgis.PyQt.QtGui import QIcon
from qgis.core import (
QgsProcessingException,
QgsProcessingParameterDefinition,
QgsProcessingParameterString,
QgsProcessingParameterRasterLayer,
QgsProcessingParameterBoolean,
QgsProcessingParameterFileDestination,
)
from processing.algs.gdal.GdalAlgorithm import GdalAlgorithm
from processing.algs.gdal.GdalUtils import GdalUtils
pluginPath = os.path.split(os.path.split(os.path.dirname(__file__))[0])[0]
class gdalinfo(GdalAlgorithm):
INPUT = "INPUT"
MIN_MAX = "MIN_MAX"
STATS = "STATS"
NO_GCP = "NOGCP"
NO_METADATA = "NO_METADATA"
EXTRA = "EXTRA"
OUTPUT = "OUTPUT"
def __init__(self):
super().__init__()
def initAlgorithm(self, config=None):
self.addParameter(
QgsProcessingParameterRasterLayer(self.INPUT, self.tr("Input layer"))
)
self.addParameter(
QgsProcessingParameterBoolean(
self.MIN_MAX,
self.tr("Force computation of the actual min/max values for each band"),
defaultValue=False,
)
)
self.addParameter(
QgsProcessingParameterBoolean(
self.STATS,
self.tr(
"Read and display image statistics (force computation if necessary)"
),
defaultValue=False,
)
)
self.addParameter(
QgsProcessingParameterBoolean(
self.NO_GCP, self.tr("Suppress GCP info"), defaultValue=False
)
)
self.addParameter(
QgsProcessingParameterBoolean(
self.NO_METADATA, self.tr("Suppress metadata info"), defaultValue=False
)
)
extra_param = QgsProcessingParameterString(
self.EXTRA,
self.tr("Additional command-line parameters"),
defaultValue=None,
optional=True,
)
extra_param.setFlags(
extra_param.flags() | QgsProcessingParameterDefinition.Flag.FlagAdvanced
)
self.addParameter(extra_param)
self.addParameter(
QgsProcessingParameterFileDestination(
self.OUTPUT,
self.tr("Layer information"),
self.tr("HTML files (*.html)"),
)
)
def name(self):
return "gdalinfo"
def displayName(self):
return self.tr("Raster information")
def group(self):
return self.tr("Raster miscellaneous")
def groupId(self):
return "rastermiscellaneous"
def icon(self):
return QIcon(os.path.join(pluginPath, "images", "gdaltools", "raster-info.png"))
def commandName(self):
return "gdalinfo"
def getConsoleCommands(self, parameters, context, feedback, executing=True):
arguments = []
if self.parameterAsBoolean(parameters, self.MIN_MAX, context):
arguments.append("-mm")
if self.parameterAsBoolean(parameters, self.STATS, context):
arguments.append("-stats")
if self.parameterAsBoolean(parameters, self.NO_GCP, context):
arguments.append("-nogcp")
if self.parameterAsBoolean(parameters, self.NO_METADATA, context):
arguments.append("-nomd")
if self.EXTRA in parameters and parameters[self.EXTRA] not in (None, ""):
extra = self.parameterAsString(parameters, self.EXTRA, context)
arguments.append(extra)
raster = self.parameterAsRasterLayer(parameters, self.INPUT, context)
if raster is None:
raise QgsProcessingException(
self.invalidRasterError(parameters, self.INPUT)
)
input_details = GdalUtils.gdal_connection_details_from_layer(raster)
arguments.append(input_details.connection_string)
if input_details.open_options:
arguments.extend(input_details.open_options_as_arguments())
if input_details.credential_options:
arguments.extend(input_details.credential_options_as_arguments())
return [self.commandName(), GdalUtils.escapeAndJoin(arguments)]
def processAlgorithm(self, parameters, context, feedback):
console_output = GdalUtils.runGdal(
self.getConsoleCommands(parameters, context, feedback), feedback
)
output = self.parameterAsFileOutput(parameters, self.OUTPUT, context)
with open(output, "w") as f:
f.write("<pre>")
for s in console_output[1:]:
f.write(str(s))
f.write("</pre>")
return {self.OUTPUT: output}
|