mirror of
https://github.com/qgis/QGIS.git
synced 2025-02-27 00:33:48 -05:00
123 lines
4.2 KiB
Python
123 lines
4.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
***************************************************************************
|
|
GdalUtils.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'
|
|
# This will get replaced with a git SHA1 when you do a git archive
|
|
__revision__ = '$Format:%H$'
|
|
|
|
import os
|
|
import subprocess
|
|
|
|
from PyQt4.QtCore import *
|
|
|
|
from sextante.core.SextanteLog import SextanteLog
|
|
|
|
try:
|
|
from osgeo import gdal
|
|
gdalAvailable = True
|
|
except:
|
|
gdalAvailable = False
|
|
|
|
class GdalUtils:
|
|
|
|
supportedRasters = None
|
|
|
|
@staticmethod
|
|
def runGdal(commands, progress):
|
|
settings = QSettings()
|
|
path = str(settings.value( "/GdalTools/gdalPath", QVariant( "" ) ).toString())
|
|
envval = str(os.getenv("PATH"))
|
|
if not path.lower() in envval.lower().split(os.pathsep):
|
|
envval += "%s%s" % (os.pathsep, str(path))
|
|
os.putenv( "PATH", envval )
|
|
loglines = []
|
|
loglines.append("GDAL execution console output")
|
|
fused_command = ''.join(['%s ' % c for c in commands])
|
|
proc = subprocess.Popen(fused_command, shell=True, stdout=subprocess.PIPE, stdin=subprocess.PIPE,stderr=subprocess.STDOUT, universal_newlines=False).stdout
|
|
for line in iter(proc.readline, ""):
|
|
loglines.append(line)
|
|
SextanteLog.addToLog(SextanteLog.LOG_INFO, loglines)
|
|
GdalUtils.consoleOutput = loglines
|
|
|
|
@staticmethod
|
|
def getConsoleOutput():
|
|
return GdalUtils.consoleOutput
|
|
|
|
@staticmethod
|
|
def getSupportedRasters():
|
|
if not gdalAvailable:
|
|
return {}
|
|
|
|
'''this has been adapted from GdalTools plugin'''
|
|
if GdalUtils.supportedRasters != None:
|
|
return GdalUtils.supportedRasters
|
|
|
|
if gdal.GetDriverCount() == 0:
|
|
gdal.AllRegister()
|
|
|
|
GdalUtils.supportedRasters = {}
|
|
GdalUtils.supportedRasters["GTiff"] = ["tif"]
|
|
for i in range(gdal.GetDriverCount()):
|
|
driver = gdal.GetDriver(i)
|
|
if driver == None:
|
|
continue
|
|
|
|
shortName = str(QString(driver.ShortName).remove( QRegExp( '\(.*$' ) ).trimmed())
|
|
metadata = driver.GetMetadata()
|
|
if not metadata.has_key(gdal.DCAP_CREATE) or metadata[gdal.DCAP_CREATE] != 'YES':
|
|
continue
|
|
if metadata.has_key(gdal.DMD_EXTENSION):
|
|
extensions = metadata[gdal.DMD_EXTENSION].split("/")
|
|
if extensions:
|
|
GdalUtils.supportedRasters[shortName] = extensions
|
|
|
|
return GdalUtils.supportedRasters
|
|
|
|
@staticmethod
|
|
def getSupportedRasterExtensions():
|
|
allexts = ["tif"]
|
|
for exts in GdalUtils.getSupportedRasters().values():
|
|
for ext in exts:
|
|
if ext not in allexts and ext != "":
|
|
allexts.append(ext)
|
|
return allexts
|
|
|
|
@staticmethod
|
|
def getFormatShortNameFromFilename(filename):
|
|
ext = filename[filename.rfind(".")+1:]
|
|
supported = GdalUtils.getSupportedRasters()
|
|
for name in supported.keys():
|
|
exts = supported[name]
|
|
if ext in exts:
|
|
return name
|
|
return "GTiff"
|
|
|
|
@staticmethod
|
|
def escapeAndJoin(strList):
|
|
joined = QString()
|
|
for s in strList:
|
|
if " " in s:
|
|
escaped = '"' + s.replace('\\', '\\\\').replace('"', '\\"') + '"'
|
|
else:
|
|
escaped = s
|
|
joined += escaped + " "
|
|
return joined.trimmed()
|