diff --git a/timeseriesviewer/spectrallibraries.py b/timeseriesviewer/spectrallibraries.py
index a37843065d522d1d0825c3a4d3328a14dc895b63..9e125d83922b3656b218b90758537261b6bd0a98 100644
--- a/timeseriesviewer/spectrallibraries.py
+++ b/timeseriesviewer/spectrallibraries.py
@@ -32,9 +32,10 @@ from PyQt4.QtGui import *
import numpy as np
from osgeo import gdal, gdal_array
-from timeseriesviewer.utils import geo2px, px2geo, SpatialExtent, SpatialPoint, loadUI
-
+from timeseriesviewer.utils import geo2px, px2geo, SpatialExtent, SpatialPoint, loadUI, settings
+from timeseriesviewer.virtualrasters import describeRawFile
+FILTERS = 'ENVI Spectral Library (*.esl *.sli);;CSV Table (*.csv)'
def gdalDataset(pathOrDataset, eAccess=gdal.GA_ReadOnly):
"""
@@ -860,7 +861,7 @@ class EnviSpectralLibraryIO(SpectralLibraryIO):
if pathVrt is None:
pathVrt = tempfile.mktemp(prefix='tmpESLVrt', suffix='.esl.vrt')
- from enmapbox.gui.virtualrasters import describeRawFile
+
ds = describeRawFile(pathESL, pathVrt, xSize, ySize, bands=bands, eType=eType, byteOrder=byteOrder)
for key, value in hdr.items():
if isinstance(value, list):
@@ -1052,25 +1053,23 @@ class SpectralLibrary(QObject):
:param parent:
:return:
"""
- from enmapbox.gui.utils import settings, DIR_TESTDATA
- SETTINGS = settings()
- lastDataSourceDir = SETTINGS.value('_lastSpecLibSourceDir', None)
- if lastDataSourceDir is None:
- lastDataSourceDir = DIR_TESTDATA
+ SETTINGS = settings()
+ lastDataSourceDir = SETTINGS.value('_lastSpecLibSourceDir', '')
- if not os.path.exists(lastDataSourceDir):
+ if not QFileInfo(lastDataSourceDir).isDir():
lastDataSourceDir = None
- uris = QFileDialog.getOpenFileNames(parent, "Open spectral library", lastDataSourceDir)
+ uris = QFileDialog.getOpenFileNames(parent, "Open spectral library", lastDataSourceDir, filter=FILTERS + ';;All files (*.*)', )
if len(uris) > 0:
SETTINGS.setValue('_lastSpecLibSourceDir', os.path.dirname(uris[-1]))
- uris = [u for u in uris if os.path.isfile(u)]
+ uris = [u for u in uris if QFileInfo(u).isFile()]
speclib = SpectralLibrary()
for u in uris:
sl = SpectralLibrary.readFrom(unicode(u))
- speclib.addSpeclib(sl)
+ if isinstance(sl, SpectralLibrary):
+ speclib.addSpeclib(sl)
return speclib
@staticmethod
@@ -1177,8 +1176,8 @@ class SpectralLibrary(QObject):
def exportProfiles(self, path=None):
if path is None:
- filters = 'ENVI Spectral Library (*.esl *.sli);;CSV Table (*.csv)'
- path = QFileDialog.getSaveFileName(parent=None, caption="Save Spectral Library", filter=filters)
+
+ path = QFileDialog.getSaveFileName(parent=None, caption="Save Spectral Library", filter=FILTERS)
if len(path) > 0:
ext = os.path.splitext(path)[-1].lower()
diff --git a/timeseriesviewer/virtualrasters.py b/timeseriesviewer/virtualrasters.py
index 2e0c27a3387f088687b390d4fac1411ed46c9bd5..bf6a8dc4c8366681ee4f701df326788e0e16a9f5 100644
--- a/timeseriesviewer/virtualrasters.py
+++ b/timeseriesviewer/virtualrasters.py
@@ -1,4 +1,5 @@
# -*- coding: utf-8 -*-
+# noinspection PyPep8Naming
"""
/***************************************************************************
HUB TimeSeriesViewer
@@ -13,22 +14,53 @@
* *
* 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 *
+ * the Free Software Foundation; either version 3 of the License, or *
* (at your option) any later version. *
* *
***************************************************************************/
"""
-# noinspection PyPep8Naming
-from __future__ import absolute_import
-import os, sys, re, pickle, tempfile
+from __future__ import absolute_import, unicode_literals
+import os, sys, re, pickle, tempfile, unicodedata
from collections import OrderedDict
import tempfile
-from osgeo import gdal, osr, ogr
+from osgeo import gdal, osr, ogr, gdalconst as gc
from qgis.core import *
from qgis.gui import *
from PyQt4.QtCore import *
from PyQt4.QtGui import *
+#lookup GDAL Data Type and its size in bytes
+LUT_GDT_SIZE = {gdal.GDT_Byte:1,
+ gdal.GDT_UInt16:2,
+ gdal.GDT_Int16:2,
+ gdal.GDT_UInt32:4,
+ gdal.GDT_Int32:4,
+ gdal.GDT_Float32:4,
+ gdal.GDT_Float64:8,
+ gdal.GDT_CInt16:2,
+ gdal.GDT_CInt32:4,
+ gdal.GDT_CFloat32:4,
+ gdal.GDT_CFloat64:8}
+
+LUT_GDT_NAME = {gdal.GDT_Byte:'Byte',
+ gdal.GDT_UInt16:'UInt16',
+ gdal.GDT_Int16:'Int16',
+ gdal.GDT_UInt32:'UInt32',
+ gdal.GDT_Int32:'Int32',
+ gdal.GDT_Float32:'Float32',
+ gdal.GDT_Float64:'Float64',
+ gdal.GDT_CInt16:'Int16',
+ gdal.GDT_CInt32:'Int32',
+ gdal.GDT_CFloat32:'Float32',
+ gdal.GDT_CFloat64:'Float64'}
+
+def u2s(s):
+ if isinstance(s, unicode):
+ #s = s.encode(s, 'utf-8')
+ s = unicodedata.normalize('NFKD', s).encode('ascii', 'ignore')
+ #s = s.encode('utf-8', 'ignore')
+ return str(s)
+
def px2geo(px, gt):
#see http://www.gdal.org/gdal_datamodel.html
gx = gt[0] + px.x()*gt[1]+px.y()*gt[2]
@@ -36,7 +68,95 @@ def px2geo(px, gt):
return QgsPoint(gx,gy)
+def describeRawFile(pathRaw, pathVrt, xsize, ysize,
+ bands=1,
+ eType = gdal.GDT_Byte,
+ interleave='bsq',
+ byteOrder='LSB',
+ headerOffset=0):
+ """
+ Creates a VRT to describe a raw binary file
+ :param pathRaw: path of raw image
+ :param pathVrt: path of destination VRT
+ :param xsize: number of image samples / columns
+ :param ysize: number of image lines
+ :param bands: number of image bands
+ :param eType: the GDAL data type
+ :param interleave: can be 'bsq' (default),'bil' or 'bip'
+ :param byteOrder: 'LSB' (default) or 'MSB'
+ :param headerOffset: header offset in bytes, default = 0
+ :return: gdal.Dataset of created VRT
+ """
+ assert xsize > 0
+ assert ysize > 0
+ assert bands > 0
+ assert eType > 0
+
+ assert eType in LUT_GDT_SIZE.keys(), 'dataType "{}" is not a valid gdal datatype'.format(eType)
+ interleave = interleave.lower()
+
+ assert interleave in ['bsq','bil','bip']
+ assert byteOrder in ['LSB', 'MSB']
+ vrt = ['<VRTDataset rasterXSize="{xsize}" rasterYSize="{ysize}">'.format(xsize=xsize,ysize=ysize)]
+
+ vrtDir = os.path.dirname(pathVrt)
+ if pathRaw.startswith(vrtDir):
+ relativeToVRT = 1
+ srcFilename = os.path.relpath(pathRaw, vrtDir)
+ else:
+ relativeToVRT = 0
+ srcFilename = pathRaw
+
+ for b in range(bands):
+ vrt.append(' <VRTRasterBand dataType="{dataType}" band="{band}" subClass="VRTRawRasterBand">'.format(
+ dataType=LUT_GDT_NAME[eType], band=b+1))
+ if interleave == 'bsq':
+ imageOffset = headerOffset
+ pixelOffset = LUT_GDT_SIZE[eType]
+ lineOffset = pixelOffset * xsize
+ elif interleave == 'bip':
+ imageOffset = headerOffset + b * LUT_GDT_SIZE[eType]
+ pixelOffset = bands * LUT_GDT_SIZE[eType]
+ lineOffset = xsize * bands
+ else:
+ raise Exception('Interleave {} is not supported'.format(interleave))
+ vrt.append(""" <SourceFilename relativetoVRT="{relativeToVRT}">{srcFilename}</SourceFilename>
+ <ImageOffset>{imageOffset}</ImageOffset>
+ <PixelOffset>{pixelOffset}</PixelOffset>
+ <LineOffset>{lineOffset}</LineOffset>
+ <ByteOrder>{byteOrder}</ByteOrder>""".format(relativeToVRT=relativeToVRT,
+ srcFilename=srcFilename,
+ imageOffset=imageOffset,
+ pixelOffset=pixelOffset,
+ lineOffset=lineOffset,
+ byteOrder=byteOrder))
+
+ vrt.append(' </VRTRasterBand>')
+ vrt.append('</VRTDataset>')
+ vrt = '\n'.join(vrt)
+ open(pathVrt, 'w').write(vrt)
+
+ ds = gdal.Open(pathVrt)
+ return ds
+
+
class VRTRasterInputSourceBand(object):
+ @staticmethod
+ def fromGDALDataSet(pathOrDataSet):
+
+ srcBands = []
+
+ if isinstance(pathOrDataSet, str):
+ pathOrDataSet = gdal.Open(pathOrDataSet)
+
+ if isinstance(pathOrDataSet, gdal.Dataset):
+ path = pathOrDataSet.GetFileList()[0]
+ for b in range(pathOrDataSet.RasterCount):
+ srcBands.append(VRTRasterInputSourceBand(path, b))
+ return srcBands
+
+
+
def __init__(self, path, bandIndex, bandName=''):
self.mPath = os.path.normpath(path)
self.mBandIndex = bandIndex
@@ -75,11 +195,18 @@ class VRTRasterBand(QObject):
def __init__(self, name='', parent=None):
super(VRTRasterBand, self).__init__(parent)
self.sources = []
- self.mName = name
+ self.mName = ''
+ self.setName(name)
self.mVRT = None
def setName(self, name):
+
+ #if isinstance(name, unicode):
+ # name = name.encode('utf-8')
+ #name = u2s(name)
+ if isinstance(name, str):
+ name = unicode(name)
oldName = self.mName
self.mName = name
if oldName != self.mName:
@@ -97,10 +224,11 @@ class VRTRasterBand(QObject):
def insertSource(self, index, virtualBandInputSource):
assert isinstance(virtualBandInputSource, VRTRasterInputSourceBand)
virtualBandInputSource.mVirtualBand = self
- assert index <= len(self.sources)
- self.sources.insert(index, virtualBandInputSource)
- self.sigSourceInserted.emit(index, virtualBandInputSource)
-
+ if index <= len(self.sources):
+ self.sources.insert(index, virtualBandInputSource)
+ self.sigSourceInserted.emit(index, virtualBandInputSource)
+ else:
+ print('DEBUG: index <= len(self.sources)')
def bandIndex(self):
if isinstance(self.mVRT, VRTRaster):
return self.mVRT.mBands.index(self)
@@ -136,51 +264,14 @@ class VRTRasterBand(QObject):
infos.append('\t{} SourceFileName {} SourceBand {}'.format(i + 1, info.mPath, info.mBandIndex))
return '\n'.join(infos)
-LUT_ReampleAlg = {'nearest': gdal.GRA_NearestNeighbour,
- 'bilinear': gdal.GRA_Bilinear,
- 'mode':gdal.GRA_Mode,
- 'lanczos':gdal.GRA_Lanczos,
- 'average':gdal.GRA_Average,
- 'cubic':gdal.GRA_Cubic,
- 'cubic_splie':gdal.GRA_CubicSpline}
-
-class VRTRasterPreviewMapCanvas(QgsMapCanvas):
-
- def __init__(self, parent=None, *args, **kwds):
- super(VRTRasterPreviewMapCanvas, self).__init__(parent, *args, **kwds)
- self.setCrsTransformEnabled(True)
-
- def contextMenuEvent(self, event):
- menu = QMenu()
- action = menu.addAction('Refresh')
- action.triggered.connect(self.refresh)
-
- action = menu.addAction('Reset')
- action.triggered.connect(self.reset)
-
- menu.exec_(event.globalPos())
-
- def setLayerSet(self, layers):
- raise DeprecationWarning()
-
- def setLayers(self, layers):
- assert isinstance(layers, list)
- def area(layer):
- extent = layer.extent()
- return extent.width() * extent.height()
- layers = list(sorted(layers, key = lambda lyr: area(lyr), reverse=True))
- QgsMapLayerRegistry.instance().addMapLayers(layers)
-
- super(VRTRasterPreviewMapCanvas, self).setLayerSet([QgsMapCanvasLayer(l) for l in layers])
-
-
-
-
- def reset(self):
- extent = self.fullExtent()
- extent.scale(1.05)
- self.setExtent(extent)
- self.refresh()
+LUT_ResampleAlgs = OrderedDict()
+LUT_ResampleAlgs['nearest'] = gdal.GRA_NearestNeighbour
+LUT_ResampleAlgs['bilinear'] = gdal.GRA_Bilinear
+LUT_ResampleAlgs['mode'] = gdal.GRA_Mode
+LUT_ResampleAlgs['lanczos'] = gdal.GRA_Lanczos
+LUT_ResampleAlgs['average'] = gdal.GRA_Average
+LUT_ResampleAlgs['cubic'] = gdal.GRA_Cubic
+LUT_ResampleAlgs['cubic_spline'] = gdal.GRA_CubicSpline
class VRTRaster(QObject):
@@ -192,27 +283,125 @@ class VRTRaster(QObject):
sigBandInserted = pyqtSignal(int, VRTRasterBand)
sigBandRemoved = pyqtSignal(int, VRTRasterBand)
sigCrsChanged = pyqtSignal(QgsCoordinateReferenceSystem)
-
+ sigResolutionChanged = pyqtSignal()
+ sigResamplingAlgChanged = pyqtSignal(str)
+ sigExtentChanged = pyqtSignal()
def __init__(self, parent=None):
super(VRTRaster, self).__init__(parent)
self.mBands = []
self.mCrs = None
- self.mResampleAlg = gdal.GRA_NearestNeighbour
+ self.mResamplingAlg = gdal.GRA_NearestNeighbour
self.mMetadata = dict()
self.mSourceRasterBounds = dict()
- self.mOutputBounds = None
+ self.mExtent = None
+ self.mResolution = None
self.sigSourceBandRemoved.connect(self.updateSourceRasterBounds)
self.sigSourceBandInserted.connect(self.updateSourceRasterBounds)
self.sigBandRemoved.connect(self.updateSourceRasterBounds)
self.sigBandInserted.connect(self.updateSourceRasterBounds)
+
+ def setResamplingAlg(self, value):
+ """
+ Sets the resampling algorithm
+ :param value:
+ - Any gdal.GRA_* constant, like gdal.GRA_NearestNeighbor
+ - nearest,bilinear,cubic,cubicspline,lanczos,average,mode
+ - None (will set the default value to 'nearest'
+ """
+ last = self.mResamplingAlg
+ if value is None:
+ self.mResamplingAlg = gdal.GRA_NearestNeighbour
+ elif value in LUT_ResampleAlgs.keys():
+ self.mResamplingAlg = LUT_ResampleAlgs[value]
+ else:
+ assert value in LUT_ResampleAlgs.values()
+ self.mResamplingAlg = value
+ if last != self.mResamplingAlg:
+ self.sigResamplingAlgChanged.emit(self.resamplingAlg(asString=True))
+
+
+ def resamplingAlg(self, asString=False):
+ """
+ "Returns the resampling algorithms.
+ :param asString: Set True to return the resampling algorithm as string.
+ :return: gdal.GRA* constant or descriptive string.
+ """
+ if asString:
+ return LUT_ResampleAlgs.keys()[LUT_ResampleAlgs.values().index(self.mResamplingAlg)]
+ else:
+ self.mResamplingAlg
+
+ def setExtent(self, rectangle, crs=None):
+ last = self.mExtent
+ if rectangle is None:
+ #use implicit/automatic values
+ self.mExtent = None
+ else:
+ if isinstance(crs, QgsCoordinateReferenceSystem) and isinstance(self.mCrs, QgsCoordinateReferenceSystem):
+ trans = QgsCoordinateTransform(crs, self.mCrs)
+ rectangle = trans.transform(rectangle)
+
+ assert isinstance(rectangle, QgsRectangle)
+ assert rectangle.width() > 0
+ assert rectangle.height() > 0
+ self.mExtent = rectangle
+
+ if last != self.mExtent:
+ self.sigExtentChanged.emit()
+ pass
+
+ def extent(self):
+ return self.mExtent
+
+ def setResolution(self, xy):
+ """
+ Set the VRT resolution.
+ :param xy: explicit value given as QSizeF(x,y) object or
+ implicit as 'highest','lowest','average'
+ """
+ last = self.mResolution
+ if xy is None:
+ self.mResolution = 'average'
+ else:
+ if isinstance(xy, QSizeF):
+ assert xy.width() > 0
+ assert xy.height() > 0
+ self.mResolution = QSizeF(xy)
+ else:
+ assert type(xy) in [str, unicode]
+ xy = str(xy)
+ assert xy in ['average','highest','lowest']
+ self.mResolution = xy
+
+ if last != self.mResolution:
+ self.sigResolutionChanged.emit()
+
+ def resolution(self):
+ """
+ Returns the internal resolution descriptor, which can be
+ an explicit QSizeF(x,y) or one of following strings: 'average','highest','lowest'
+ """
+ return self.mResolution
+
+
def setCrs(self, crs):
+ """
+ Sets the output Coordinate Reference System (CRS)
+ :param crs: osr.SpatialReference or QgsCoordinateReferenceSystem
+ :return:
+ """
if isinstance(crs, osr.SpatialReference):
auth = '{}:{}'.format(crs.GetAttrValue('AUTHORITY',0), crs.GetAttrValue('AUTHORITY',1))
crs = QgsCoordinateReferenceSystem(auth)
if isinstance(crs, QgsCoordinateReferenceSystem):
if crs != self.mCrs:
+ extent = self.extent()
+ if isinstance(extent, QgsRectangle):
+ trans = QgsCoordinateTransform(self.mCrs, crs)
+ extent = trans.transform(extent)
+ self.setExtent(extent)
self.mCrs = crs
self.sigCrsChanged.emit(self.mCrs)
@@ -344,15 +533,6 @@ class VRTRaster(QObject):
def sourceRasterBounds(self):
return self.mSourceRasterBounds
- def outputBounds(self):
- if isinstance(self.mOutputBounds, RasterBounds):
- return
- #calculate from source rasters
-
- def setOutputBounds(self, bounds):
- assert isinstance(self, RasterBounds)
- self.mOutputBounds = bounds
-
def updateSourceRasterBounds(self):
@@ -377,25 +557,38 @@ class VRTRaster(QObject):
if len(toAdd) > 0:
self.sigSourceRasterAdded.emit(toAdd)
-
- def saveVRT(self, pathVRT, resampleAlg=gdal.GRA_NearestNeighbour, **kwds):
+ def loadVRT(self, pathVRT, bandIndex = None):
"""
- :param pathVRT: path to VRT that is created
- :param options --- can be be an array of strings, a string or let empty and filled from other keywords..
- :param resolution --- 'highest', 'lowest', 'average', 'user'.
- :param outputBounds --- output bounds as (minX, minY, maxX, maxY) in target SRS.
- :param xRes, yRes --- output resolution in target SRS.
- :param targetAlignedPixels --- whether to force output bounds to be multiple of output resolution.
- :param bandList --- array of band numbers (index start at 1).
- :param addAlpha --- whether to add an alpha mask band to the VRT when the source raster have none.
- :param resampleAlg --- resampling mode.
- :param outputSRS --- assigned output SRS.
- :param allowProjectionDifference --- whether to accept input datasets have not the same projection. Note: they will *not* be reprojected.
- :param srcNodata --- source nodata value(s).
- :param callback --- callback method.
- :param callback_data --- user data for callback.
- :return: gdal.DataSet(pathVRT)
+ Load the VRT definition in pathVRT and appends it to this VRT
+ :param pathVRT:
"""
+ if pathVRT in [None,'']:
+ return
+
+ if bandIndex is None:
+ bandIndex = len(self.mBands)
+
+ ds = gdal.Open(pathVRT)
+ assert isinstance(ds, gdal.Dataset)
+ assert ds.GetDriver().GetDescription() == 'VRT'
+ from xml.etree import ElementTree
+ for b in range(ds.RasterCount):
+ srcBand = ds.GetRasterBand(b+1)
+ vrtBand = VRTRasterBand(name=srcBand.GetDescription().decode('utf-8'))
+ for key, xml in srcBand.GetMetadata(str('vrt_sources')).items():
+
+ tree = ElementTree.fromstring(xml)
+ srcPath = os.path.normpath(tree.find('SourceFilename').text)
+ srcBandIndex = int(tree.find('SourceBand').text)
+ vrtBand.addSource(VRTRasterInputSourceBand(srcPath, srcBandIndex))
+
+ self.insertVirtualBand(bandIndex, vrtBand)
+ bandIndex += 1
+
+
+
+
+ def saveVRT(self, pathVRT):
if len(self.mBands) == 0:
print('No VRT Inputs defined.')
@@ -403,21 +596,6 @@ class VRTRaster(QObject):
assert os.path.splitext(pathVRT)[-1].lower() == '.vrt'
- _kwds = dict()
- supported = ['options','resolution','outputBounds','xRes','yRes','targetAlignedPixels','addAlpha','resampleAlg',
- 'outputSRS','allowProjectionDifference','srcNodata','VRTNodata','hideNodata','callback', 'callback_data']
- for k in kwds.keys():
- if k in supported:
- _kwds[k] = kwds[k]
-
- if 'resampleAlg' not in _kwds:
- _kwds['resampleAlg'] = resampleAlg
-
- if isinstance(self.mOutputBounds, RasterBounds):
- bounds = self.mOutputBounds.polygon
- xmin, ymin,xmax, ymax = bounds
- _kwds['outputBounds'] = (xmin, ymin,xmax, ymax)
-
dirVrt = os.path.dirname(pathVRT)
dirWarpedVRT = os.path.join(dirVrt, 'WarpedVRTs')
if not os.path.isdir(dirVrt):
@@ -425,6 +603,7 @@ class VRTRaster(QObject):
srcLookup = dict()
srcNodata = None
+
for i, pathSrc in enumerate(self.sourceRaster()):
dsSrc = gdal.Open(pathSrc)
assert isinstance(dsSrc, gdal.Dataset)
@@ -449,13 +628,30 @@ class VRTRaster(QObject):
tmp = None
srcLookup[pathSrc] = pathVRT2
+ srcFiles = [srcLookup[src] for src in self.sourceRaster()]
+ #1. build a temporary VRT that describes the spatial shifts of all input sources
- srcFiles = [srcLookup[src] for src in self.sourceRaster()]
+ kwds = {}
+ res = self.resolution()
+
+ if res is None:
+ res = 'average'
+ if isinstance(res, QSizeF):
+ kwds['resolution'] = 'user'
+ kwds['xRes'] = res.width()
+ kwds['yRes'] = res.height()
+ else:
+ assert res in ['highest','lowest','average']
+ kwds['resolution'] = res
+
+ extent = self.extent()
+ if isinstance(extent, QgsRectangle):
+ kwds['outputBounds'] = (extent.xMinimum(), extent.yMinimum(), extent.xMaximum(), extent.yMaximum())
+
+ vro = gdal.BuildVRTOptions(separate=True, **kwds)
- vro = gdal.BuildVRTOptions(separate=True, **_kwds)
- #1. build a temporary VRT that described the spatial shifts of all input sources
gdal.BuildVRT(pathVRT, srcFiles, options=vro)
dsVRTDst = gdal.Open(pathVRT)
assert isinstance(dsVRTDst, gdal.Dataset)
@@ -466,7 +662,7 @@ class VRTRaster(QObject):
eType = dsVRTDst.GetRasterBand(1).DataType
SOURCE_TEMPLATES = dict()
for i, srcFile in enumerate(srcFiles):
- vrt_sources = dsVRTDst.GetRasterBand(i+1).GetMetadata('vrt_sources')
+ vrt_sources = dsVRTDst.GetRasterBand(i+1).GetMetadata(str('vrt_sources'))
assert len(vrt_sources) == 1
srcXML = vrt_sources.values()[0]
assert os.path.basename(srcFile)+'</SourceFilename>' in srcXML
@@ -477,7 +673,7 @@ class VRTRaster(QObject):
os.remove(pathVRT)
#2. build final VRT from scratch
- drvVRT = gdal.GetDriverByName('VRT')
+ drvVRT = gdal.GetDriverByName(str('VRT'))
assert isinstance(drvVRT, gdal.Driver)
dsVRTDst = drvVRT.Create(pathVRT, ns, nl,0, eType=eType)
#2.1. set general properties
@@ -491,7 +687,7 @@ class VRTRaster(QObject):
assert dsVRTDst.AddBand(eType, options=['subClass=VRTSourcedRasterBand']) == 0
vrtBandDst = dsVRTDst.GetRasterBand(i+1)
assert isinstance(vrtBandDst, gdal.Band)
- vrtBandDst.SetDescription(str(vBand.name()))
+ vrtBandDst.SetDescription(vBand.name().encode('utf-8'))
md = {}
#add all input sources for this virtual band
for iSrc, sourceInfo in enumerate(vBand.sources):
@@ -500,7 +696,7 @@ class VRTRaster(QObject):
xml = SOURCE_TEMPLATES[srcLookup[sourceInfo.mPath]]
xml = re.sub('<SourceBand>1</SourceBand>','<SourceBand>{}</SourceBand>'.format(bandIndex+1), xml)
md['source_{}'.format(iSrc)] = xml
- vrtBandDst.SetMetadata(md,'vrt_sources')
+ vrtBandDst.SetMetadata(md,str('vrt_sources'))
if False:
vrtBandDst.ComputeBandStats(1)
@@ -540,111 +736,6 @@ class VRTRaster(QObject):
-class VRTRasterVectorLayer(QgsVectorLayer):
-
- def __init__(self, vrtRaster, crs=None):
- assert isinstance(vrtRaster, VRTRaster)
- if crs is None:
- crs = QgsCoordinateReferenceSystem('EPSG:4326')
-
- uri = 'polygon?crs={}'.format(crs.authid())
- super(VRTRasterVectorLayer, self).__init__(uri, 'VRTRaster', 'memory', False)
- self.mCrs = crs
- self.mVRTRaster = vrtRaster
-
- #initialize fields
- assert self.startEditing()
- # standard field names, types, etc.
- fieldDefs = [('oid', QVariant.Int, 'integer'),
- ('type', QVariant.String, 'string'),
- ('name', QVariant.String, 'string'),
- ('path', QVariant.String, 'string'),
- ]
- # initialize fields
- for fieldDef in fieldDefs:
- field = QgsField(fieldDef[0], fieldDef[1], fieldDef[2])
- self.addAttribute(field)
- self.commitChanges()
-
- symbol = QgsFillSymbolV2.createSimple({'style': 'no', 'color': 'red', 'outline_color':'black'})
- self.rendererV2().setSymbol(symbol)
- self.label().setFields(self.fields())
- self.label().setLabelField(3,3)
- self.mVRTRaster.sigSourceRasterAdded.connect(self.onRasterInserted)
- self.mVRTRaster.sigSourceRasterRemoved.connect(self.onRasterRemoved)
- self.onRasterInserted(self.mVRTRaster.sourceRaster())
-
- def path2feature(self, path):
- for f in self.dataProvider().getFeatures():
- if str(f.attribute('path')) == str(path):
- return f
- return None
-
- def path2fid(self, path):
- for f in self.dataProvider().getFeatures():
- if str(f.attribute('path')) == str(path):
- return f.id()
-
-
- return None
-
- def fid2path(self, fid):
- for f in self.dataProvider().getFeatures():
- if f.fid() == fid:
- return f
-
- return None
-
- def onRasterInserted(self, listOfNewFiles):
- assert isinstance(listOfNewFiles, list)
- if len(listOfNewFiles) == 0:
- return
- self.startEditing()
- for f in listOfNewFiles:
- bounds = self.mVRTRaster.sourceRasterBounds()[f]
- assert isinstance(bounds, RasterBounds)
- oid = str(id(bounds))
- geometry =QgsPolygonV2(bounds.polygon)
- #geometry = QgsCircularStringV2(bounds.curve)
- trans = QgsCoordinateTransform(bounds.crs, self.crs())
- geometry.transform(trans)
-
-
-
-
- feature = QgsFeature(self.pendingFields())
- #feature.setGeometry(QgsGeometry(geometry))
- feature.setGeometry(QgsGeometry.fromWkt(geometry.asWkt()))
- #feature.setFeatureId(int(oid))
- feature.setAttribute('oid', oid)
- feature.setAttribute('type', 'source file')
- feature.setAttribute('name', str(os.path.basename(f)))
- feature.setAttribute('path', str(f))
- #feature.setValid(True)
-
- assert self.dataProvider().addFeatures([feature])
- self.featureAdded.emit(feature.id())
-
-
- self.updateExtents()
- assert self.commitChanges()
- self.dataChanged.emit()
-
- def onRasterRemoved(self, files):
- self.startEditing()
- self.selectAll()
- toRemove = []
- for f in self.selectedFeatures():
- if f.attribute('path') in files:
- toRemove.append(f.id())
- self.setSelectedFeatures(toRemove)
- self.deleteSelectedFeatures()
- self.commitChanges()
- self.dataChanged.emit()
-
-
-
-
def createVirtualBandMosaic(bandFiles, pathVRT):
drv = gdal.GetDriverByName('VRT')
@@ -697,465 +788,6 @@ def createVirtualBandStack(bandFiles, pathVRT):
return vrtDS
-from timeseriesviewer.utils import loadUI
-
-class TreeNode(QObject):
- sigWillAddChildren = pyqtSignal(QObject, int, int)
- sigAddedChildren = pyqtSignal(QObject, int, int)
- sigWillRemoveChildren = pyqtSignal(QObject, int, int)
- sigRemovedChildren = pyqtSignal(QObject, int, int)
- sigUpdated = pyqtSignal(QObject)
-
- def __init__(self, parentNode, name=None):
- super(TreeNode, self).__init__()
- self.mParent = parentNode
-
- self.mChildren = []
- self.mName = name
- self.mValues = []
- self.mIcon = None
- self.mToolTip = None
-
-
- if isinstance(parentNode, TreeNode):
- parentNode.appendChildNodes(self)
-
- def nodeIndex(self):
- return self.mParent.mChildren.index(self)
-
- def next(self):
- i = self.nodeIndex()
- if i < len(self.mChildren.mChildren):
- return self.mParent.mChildren[i+1]
- else:
- return None
-
- def previous(self):
- i = self.nodeIndex()
- if i > 0:
- return self.mParent.mChildren[i - 1]
- else:
- return None
-
- def detach(self):
- """
- Detaches this TreeNode from its parent TreeNode
- :return:
- """
- if isinstance(self.mParent, TreeNode):
- self.mParent.mChildren.remove(self)
- self.setParentNode(None)
-
- def appendChildNodes(self, listOfChildNodes):
- self.insertChildNodes(len(self.mChildren), listOfChildNodes)
-
- def insertChildNodes(self, index, listOfChildNodes):
- assert index <= len(self.mChildren)
- if isinstance(listOfChildNodes, TreeNode):
- listOfChildNodes = [listOfChildNodes]
- assert isinstance(listOfChildNodes, list)
- l = len(listOfChildNodes)
- idxLast = index+l-1
- self.sigWillAddChildren.emit(self, index, idxLast)
- for i, node in enumerate(listOfChildNodes):
- assert isinstance(node, TreeNode)
- node.mParent = self
- # connect node signals
- node.sigWillAddChildren.connect(self.sigWillAddChildren)
- node.sigAddedChildren.connect(self.sigAddedChildren)
- node.sigWillRemoveChildren.connect(self.sigWillRemoveChildren)
- node.sigRemovedChildren.connect(self.sigRemovedChildren)
- node.sigUpdated.connect(self.sigUpdated)
-
- self.mChildren.insert(index+i, node)
-
- self.sigAddedChildren.emit(self, index, idxLast)
-
- def removeChildNode(self, node):
- assert node in self.mChildren
- i = self.mChildren.index(node)
- self.removeChildNodes(i, 1)
-
- def removeChildNodes(self, row, count):
-
- if row < 0 or count <= 0:
- return False
-
- rowLast = row + count - 1
-
- if rowLast >= self.childCount():
- return False
-
- self.sigWillRemoveChildren.emit(self, row, rowLast)
- to_remove = self.childNodes()[row:rowLast+1]
- for n in to_remove:
- self.mChildren.remove(n)
- #n.mParent = None
-
- self.sigRemovedChildren.emit(self, row, rowLast)
-
-
-
- def setToolTip(self, toolTip):
- self.mToolTip = toolTip
- def toolTip(self):
- return self.mToolTip
-
- def parentNode(self):
- return self.mParent
-
- def setParentNode(self, treeNode):
- assert isinstance(treeNode, TreeNode)
- self.mParent = treeNode
-
- def setIcon(self, icon):
- self.mIcon = icon
-
- def icon(self):
- return self.mIcon
-
- def setName(self, name):
- self.mName = name
-
- def name(self):
- return self.mName
-
- def contextMenu(self):
- return None
-
-
- def setValues(self, listOfValues):
- if not isinstance(listOfValues, list):
- listOfValues = [listOfValues]
- self.mValues = listOfValues[:]
- def values(self):
- return self.mValues[:]
-
- def childCount(self):
- return len(self.mChildren)
-
- def childNodes(self):
- return self.mChildren[:]
-
- def findChildNodes(self, type, recursive=True):
- results = []
- for node in self.mChildren:
- if isinstance(node, type):
- results.append(node)
- if recursive:
- results.extend(node.findChildNodes(type, recursive=True))
- return results
-
-class SourceRasterFileNode(TreeNode):
-
- def __init__(self, parentNode, path):
- super(SourceRasterFileNode, self).__init__(parentNode)
-
- self.mPath = path
- self.setName(os.path.basename(path))
- srcNode = TreeNode(self, name='Path')
- srcNode.setValues(path)
-
-
- #populate metainfo
- ds = gdal.Open(path)
- assert isinstance(ds, gdal.Dataset)
-
-
- crsNode = TreeNode(self, name='CRS')
- crsNode.setIcon(QIcon(':/timeseriesviewer/icons/CRS.png'))
- crs = osr.SpatialReference()
- crs.ImportFromWkt(ds.GetProjection())
-
- authInfo = '{}:{}'.format(crs.GetAttrValue('AUTHORITY',0), crs.GetAttrValue('AUTHORITY',1))
- crsNode.setValues([authInfo,crs.ExportToWkt()])
- self.bandNode = TreeNode(None, name='Bands')
- for b in range(ds.RasterCount):
- band = ds.GetRasterBand(b+1)
-
- inputSource = VRTRasterInputSourceBand(path, b)
- inputSource.mBandName = band.GetDescription()
- if inputSource.mBandName in [None,'']:
- inputSource.mBandName = '{}'.format(b + 1)
- inputSource.mNoData = band.GetNoDataValue()
-
- SourceRasterBandNode(self.bandNode, inputSource)
- self.bandNode.setParentNode(self)
- self.appendChildNodes(self.bandNode)
-
- def sourceBands(self):
- return [n.mSrcBand for n in self.bandNode.mChildren if isinstance(n, SourceRasterBandNode)]
-
-class SourceRasterBandNode(TreeNode):
- def __init__(self, parentNode, vrtRasterInputSourceBand):
- assert isinstance(vrtRasterInputSourceBand, VRTRasterInputSourceBand)
- super(SourceRasterBandNode, self).__init__(parentNode)
- self.setIcon(QIcon(":/timeseriesviewer/icons/mIconRaster.png"))
- self.mSrcBand = vrtRasterInputSourceBand
- self.setName(self.mSrcBand.mBandName)
- #self.setValues([self.mSrcBand.mPath])
- self.setToolTip('band {}:{}'.format(self.mSrcBand.mBandIndex+1, self.mSrcBand.mPath))
-
-class VRTRasterNode(TreeNode):
- def __init__(self, parentNode, vrtRaster):
- assert isinstance(vrtRaster, VRTRaster)
-
- super(VRTRasterNode, self).__init__(parentNode)
- self.mVRTRaster = vrtRaster
- self.mVRTRaster.sigBandInserted.connect(self.onBandInserted)
- self.mVRTRaster.sigBandRemoved.connect(self.onBandRemoved)
-
- def onBandInserted(self, index, vrtRasterBand):
- assert isinstance(vrtRasterBand, VRTRasterBand)
- i = vrtRasterBand.bandIndex()
- assert i == index
- node = VRTRasterBandNode(None, vrtRasterBand)
- self.insertChildNodes(i, [node])
-
- def onBandRemoved(self, removedIdx):
- self.removeChildNodes(removedIdx, 1)
-
-
-class VRTRasterBandNode(TreeNode):
- def __init__(self, parentNode, virtualBand):
- assert isinstance(virtualBand, VRTRasterBand)
-
- super(VRTRasterBandNode, self).__init__(parentNode)
- self.mVirtualBand = virtualBand
-
- self.setName(virtualBand.name())
- self.setIcon(QIcon(":/timeseriesviewer/icons/mIconVirtualRaster.png"))
- #self.nodeBands = TreeNode(self, name='Input Bands')
- #self.nodeBands.setToolTip('Source bands contributing to this virtual raster band')
- self.nodeBands = self
- virtualBand.sigNameChanged.connect(self.setName)
- virtualBand.sigSourceInserted.connect(lambda _, src: self.onSourceInserted(src))
- virtualBand.sigSourceRemoved.connect(self.onSourceRemoved)
- for src in self.mVirtualBand.sources:
- self.onSourceInserted(src)
-
-
- def onSourceInserted(self, inputSource):
- assert isinstance(inputSource, VRTRasterInputSourceBand)
- assert inputSource.virtualBand() == self.mVirtualBand
- i = self.mVirtualBand.sources.index(inputSource)
-
- node = VRTRasterInputSourceBandNode(None, inputSource)
- self.nodeBands.insertChildNodes(i, node)
-
- def onSourceRemoved(self, row, inputSource):
- assert isinstance(inputSource, VRTRasterInputSourceBand)
-
- node = self.nodeBands.childNodes()[row]
- if node.mSrc != inputSource:
- s = ""
- self.nodeBands.removeChildNode(node)
-
-
-
-
-class VRTRasterInputSourceBandNode(TreeNode):
- def __init__(self, parentNode, vrtRasterInputSourceBand):
- assert isinstance(vrtRasterInputSourceBand, VRTRasterInputSourceBand)
- super(VRTRasterInputSourceBandNode, self).__init__(parentNode)
- self.setIcon(QIcon(":/timeseriesviewer/icons/mIconRaster.png"))
- self.mSrc = vrtRasterInputSourceBand
- name = '{}:{}'.format(self.mSrc.mBandIndex+1, os.path.basename(self.mSrc.mPath))
- self.setName(name)
- #self.setValues([self.mSrc.mPath, self.mSrc.mBandIndex])
-
- def sourceBand(self):
- return self.mSrc
-
-class TreeView(QTreeView):
-
- def __init__(self, *args, **kwds):
- super(TreeView, self).__init__(*args, **kwds)
-
-class TreeModel(QAbstractItemModel):
- def __init__(self, parent=None, rootNode = None):
- super(TreeModel, self).__init__(parent)
-
- self.mColumnNames = ['Node','Value']
- self.mRootNode = rootNode if isinstance(rootNode, TreeNode) else TreeNode(None)
- self.mRootNode.sigWillAddChildren.connect(self.nodeWillAddChildren)
- self.mRootNode.sigAddedChildren.connect(self.nodeAddedChildren)
- self.mRootNode.sigWillRemoveChildren.connect(self.nodeWillRemoveChildren)
- self.mRootNode.sigRemovedChildren.connect(self.nodeRemovedChildren)
- self.mRootNode.sigUpdated.connect(self.nodeUpdated)
-
- self.mTreeView = None
- if isinstance(parent, QTreeView):
- self.connectTreeView(parent)
-
- def nodeWillAddChildren(self, node, idx1, idxL):
- idxNode = self.node2idx(node)
- self.beginInsertRows(idxNode, idx1, idxL)
-
-
- def nodeAddedChildren(self, node, idx1, idxL):
- self.endInsertRows()
- #for i in range(idx1, idxL+1):
- for n in node.childNodes():
- self.setColumnSpan(node)
-
- def nodeWillRemoveChildren(self, node, idx1, idxL):
- idxNode = self.node2idx(node)
- self.beginRemoveRows(idxNode, idx1, idxL)
-
- def nodeRemovedChildren(self, node, idx1, idxL):
- self.endRemoveRows()
-
-
- def nodeUpdated(self, node):
- idxNode = self.node2idx(node)
- self.dataChanged.emit(idxNode, idxNode)
- self.setColumnSpan(node)
-
- def headerData(self, section, orientation, role):
- if orientation == Qt.Horizontal and role == Qt.DisplayRole:
-
- if len(self.mColumnNames) > section:
- return self.mColumnNames[section]
- else:
- return ''
-
- else:
- return None
-
- def parent(self, index):
- if not index.isValid():
- return QModelIndex()
- node = self.idx2node(index)
- if not isinstance(node, TreeNode):
- return QModelIndex()
-
- parentNode = node.parentNode()
- if not isinstance(parentNode, TreeNode):
- return QModelIndex()
-
- return self.node2idx(parentNode)
-
- if node not in parentNode.mChildren:
- return QModelIndex
- row = parentNode.mChildren.index(node)
- return self.createIndex(row, 0, parentNode)
-
- def rowCount(self, index):
-
- node = self.idx2node(index)
- return len(node.mChildren) if isinstance(node, TreeNode) else 0
-
- def hasChildren(self, index):
- node = self.idx2node(index)
- return isinstance(node, TreeNode) and len(node.mChildren) > 0
-
- def columnNames(self):
- return self.mColumnNames
-
- def columnCount(self, index):
-
- return len(self.mColumnNames)
-
- def connectTreeView(self, treeView):
- self.mTreeView = treeView
-
- def setColumnSpan(self, node):
- if isinstance(self.mTreeView, QTreeView) \
- and isinstance(node, TreeNode) \
- and isinstance(node.parentNode(), TreeNode) :
- idxNode = self.node2idx(node)
- idxParent = self.node2idx(node.parentNode())
- span = len(node.values()) == 0
- self.mTreeView.setFirstColumnSpanned(idxNode.row(), idxParent, span)
- for n in node.childNodes():
- self.setColumnSpan(n)
-
-
- def index(self, row, column, parentIndex=None):
-
-
-
- if parentIndex is None:
- parentNode = self.mRootNode
- else:
- parentNode = self.idx2node(parentIndex)
-
- if row < 0 or row >= parentNode.childCount():
- return QModelIndex()
- if column < 0 or column >= len(self.mColumnNames):
- return QModelIndex()
-
- if isinstance(parentNode, TreeNode) and row < len(parentNode.mChildren):
- return self.createIndex(row,column,parentNode.mChildren[row])
- else:
- return QModelIndex()
-
- def findParentNode(self, node, parentNodeType):
- assert isinstance(node, TreeNode)
- while True:
- if isinstance(node, parentNodeType):
- return node
- if not isinstance(node.parentNode(), TreeNode):
- return None
- node = node.parentNode()
-
- def indexes2nodes(self, indexes):
- assert isinstance(indexes, list)
- nodes = []
- for idx in indexes:
- n = self.idx2node(idx)
- if n not in nodes:
- nodes.append(n)
- return nodes
-
- def nodes2indexes(self, nodes):
- return [self.node2idx(n) for n in nodes]
-
- def idx2node(self, index):
- if not index.isValid():
- return self.mRootNode
- else:
- return index.internalPointer()
-
- def node2idx(self, node):
- assert isinstance(node, TreeNode)
- if node == self.mRootNode:
- return QModelIndex()
- else:
- parentNode = node.parentNode()
- assert isinstance(parentNode, TreeNode)
- if node not in parentNode.mChildren:
- return QModelIndex()
- r = parentNode.mChildren.index(node)
- return self.createIndex(r,0,node)
-
-
- def data(self, index, role):
- node = self.idx2node(index)
- col = index.column()
- if role == Qt.UserRole:
- return node
-
- if col == 0:
- if role in [Qt.DisplayRole, Qt.EditRole]:
- return node.name()
- if role == Qt.DecorationRole:
- return node.icon()
- if role == Qt.ToolTipRole:
- return node.toolTip()
- if col > 0:
- i = col-1
-
- if role in [Qt.DisplayRole, Qt.EditRole] and len(node.values())>i:
- return node.values()[i]
-
- def flags(self, index):
- if not index.isValid():
- return Qt.NoItemFlags
- node = self.idx2node(index)
- return Qt.ItemIsEnabled | Qt.ItemIsSelectable
-
class RasterBounds(object):
def __init__(self, path):
@@ -1167,13 +799,6 @@ class RasterBounds(object):
if path is not None:
self.fromImage(path)
- def fromRectangle(self, crs, rectangle):
- assert isinstance(rectangle, QgsRectangle)
- assert isinstance(crs, QgsCoordinateReferenceSystem)
- self.crs = crs
- self.path = ''
- s = ""
-
def fromImage(self, path):
self.path = path
@@ -1204,784 +829,8 @@ class RasterBounds(object):
self.crs = crs
- def __repr__(self):
- return self.polygon.ExportToWkt()
-
-class SourceRasterModel(TreeModel):
- def __init__(self, parent=None):
- super(SourceRasterModel, self).__init__(parent)
-
- self.mColumnNames = ['File', 'Value']
-
-
- def files(self):
- return [n.mPath for n in self.mRootNode.childNodes() if isinstance(n, SourceRasterFileNode)]
-
- def addFile(self, file):
- self.addFiles([file])
-
-
- def addFiles(self, newFiles):
- assert isinstance(newFiles, list)
- existingFiles = self.files()
- newFiles = [os.path.normpath(f) for f in newFiles]
- newFiles = [f for f in newFiles if f not in existingFiles and isinstance(gdal.Open(f), gdal.Dataset)]
- if len(newFiles) > 0:
- for f in newFiles:
- SourceRasterFileNode(self.mRootNode, f)
-
-
- def file2node(self, file):
- for node in self.mRootNode.childNodes():
- if isinstance(node, SourceRasterFileNode) and node.mPath == file:
- return node
- return None
-
- def removeFiles(self, listOfFiles):
- assert isinstance(listOfFiles, list)
-
- toRemove = [n for n in self.mRootNode.childNodes() \
- if isinstance(n, SourceRasterFileNode) and n.mPath in listOfFiles]
-
- for n in toRemove:
- n.parentNode().removeChildNode(n)
-
-
- def flags(self, index):
- if not index.isValid():
- return Qt.NoItemFlags
-
- node = self.idx2node(index)
-
- flags = super(SourceRasterModel, self).flags(index)
- #return flags
- if isinstance(node, SourceRasterFileNode) or \
- isinstance(node, SourceRasterBandNode):
- flags |= Qt.ItemIsDragEnabled
- return flags
-
- def contextMenu(self):
-
- return None
-
- def mimeTypes(self):
- # specifies the mime types handled by this model
- types = []
- types.append('text/uri-list')
- return types
-
-
- def mimeData(self, indexes):
- indexes = sorted(indexes)
- if len(indexes) == 0:
- return None
- nodes = []
- for i in indexes:
- n = self.idx2node(i)
- if n not in nodes:
- nodes.append(n)
-
-
- sourceBands = []
-
- for node in nodes:
- if isinstance(node, SourceRasterFileNode):
- sourceBands.extend(node.sourceBands())
- if isinstance(node, SourceRasterBandNode):
- sourceBands.append(node.mSrcBand)
-
- sourceBands = list(OrderedDict.fromkeys(sourceBands))
- uriList = [sourceBand.mPath for sourceBand in sourceBands]
- uriList = list(OrderedDict.fromkeys(uriList))
-
- mimeData = QMimeData()
-
- if len(sourceBands) > 0:
- mimeData.setData('hub.vrtbuilder/bandlist', pickle.dumps(sourceBands))
-
- # set text/uri-list
- if len(uriList) > 0:
- mimeData.setUrls([QUrl(p) for p in uriList])
- #mimeData.setText('\n'.join(uriList))
- return mimeData
-
-class VRTSelectionModel(QItemSelectionModel):
-
- def __init__(self, model, mapCanvas, vectorLayer, parent=None):
- assert isinstance(model, VRTRasterTreeModel)
- assert isinstance(vectorLayer, VRTRasterVectorLayer)
- #assert isinstance(mapCanvas, VRTRasterPreviewMapCanvas)
- super(VRTSelectionModel, self).__init__(model, parent)
- self.mLyr = vectorLayer
- self.mPreviewMapHighlights = {}
-
- self.mLyr.featureDeleted.connect(lambda : self.setMapHighlights(None))
- self.mLyr.featureAdded.connect(lambda : self.setMapHighlights(None))
- self.mModel = model
- self.mCanvas = mapCanvas
- self.selectionChanged.connect(self.onTreeSelectionChanged)
-
- self.previewMapTool = QgsMapToolEmitPoint(self.mCanvas)
- self.previewMapTool.setCursor(Qt.ArrowCursor)
- self.previewMapTool.canvasClicked.connect(self.onMapFeatureIdentified)
- self.mCanvas.setMapTool(self.previewMapTool)
-
-
-
-
- @pyqtSlot(QgsFeature)
- def onMapFeatureIdentified(self, point, button):
- assert isinstance(point, QgsPoint)
-
- oldSelection = self.selectedSourceFiles()
-
- if self.sender() == self.previewMapTool:
- searchRadius = QgsTolerance.toleranceInMapUnits( \
- 1, self.mLyr,self.mCanvas.mapRenderer(), QgsTolerance.Pixels)
- searchRect = QgsRectangle()
- searchRect.setXMinimum(point.x() - searchRadius);
- searchRect.setXMaximum(point.x() + searchRadius);
- searchRect.setYMinimum(point.y() - searchRadius);
- searchRect.setYMaximum(point.y() + searchRadius);
-
- if button == Qt.LeftButton:
- """
-
- lastSelection = set([f.id() for f in lyr.selectedFeatures()])
- lyr.setSelectedFeatures([])
- lyr.select(rect, True)
- """
- # select the feature closet to the point
- selectedId = None
- if True:
- geoms = {}
- flags = QgsFeatureRequest.ExactIntersect
- features = self.mLyr.getFeatures(QgsFeatureRequest() \
- .setFilterRect(searchRect) \
- .setFlags(flags))
- feature = QgsFeature()
- while features.nextFeature(feature):
- geoms[feature.geometry().area()] = feature.id()
-
- if len(geoms) > 0:
- selectedId = geoms[min(geoms.keys())]
-
- modifiers = QApplication.keyboardModifiers()
-
- newSelection = set([selectedId])
-
- # todo: allow select modifiers to select more than one
- if modifiers & Qt.ControlModifier:
- newSelection = oldSelection.difference(newSelection)
- elif modifiers & Qt.ShiftModifier:
- newSelection = oldSelection.union(newSelection)
-
- newSelection = list(newSelection)
- self.setSelectedSourceFiles(newSelection)
-
- def onTreeSelectionChanged(self, selected, deselected):
- sourceFiles = self.selectedSourceFiles()
- features = set([self.mLyr.path2feature(path) for path in sourceFiles])
- self.setMapHighlights(features)
-
-
- def selectedSourceFileNodes(self):
- indexes = self.selectedIndexes()
- selectedFileNodes = self.mModel.indexes2nodes(indexes)
- return [n for n in selectedFileNodes if isinstance(n, VRTRasterInputSourceBandNode)]
-
- def selectedSourceFiles(self):
- return set(n.sourceBand().mPath for n in self.selectedSourceFileNodes())
-
-
- def setMapHighlights(self, features):
- if features is None:
- features = []
- for f in self.mPreviewMapHighlights.keys():
- if f not in features:
- del self.mPreviewMapHighlights[f]
-
- for f in features:
- if f not in self.mPreviewMapHighlights.keys():
- h = QgsHighlight(self.mCanvas, f.geometry(), self.mLyr)
- h.setColor(QColor(0, 255, 0, 255))
- h.setWidth(3)
- h.setFillColor(QColor(255, 0, 0, 0))
- self.mPreviewMapHighlights[f] = h
-
- def setSelectedSourceFiles(self, newSelection):
- ids = []
- paths = []
- features = []
- for f in self.mLyr.dataProvider().getFeatures(QgsFeatureRequest()):
- id = f.id()
- path = str(f.attribute('path'))
-
- if id in newSelection or path in newSelection:
- ids.append(id)
- paths.append(path)
- features.append(f)
-
- # set map overlay
- self.setMapHighlights(features)
-
- srcNodesAll = self.model().mRootNode.findChildNodes(
- VRTRasterInputSourceBandNode, recursive=True)
-
- nodeSelection = QItemSelection()
- #1. select the nodes pointing to one of the source files
- for n in srcNodesAll:
- if n.sourceBand().mPath in paths:
- idx = self.model().node2idx(n)
- nodeSelection.select(idx, idx)
- #v = self.blockSignals(True)
- self.select(nodeSelection, QItemSelectionModel.SelectCurrent)
- #self.blockSignals(v)
- #self.model().select(self.model.node2idx(n), QItemSelectionModel.Select)
-
-
-
-class VRTRasterTreeModel(TreeModel):
- def __init__(self, parent=None, vrtRaster=None):
-
- vrtRaster = vrtRaster if isinstance(vrtRaster, VRTRaster) else VRTRaster()
- rootNode = VRTRasterNode(None, vrtRaster)
- super(VRTRasterTreeModel, self).__init__(parent, rootNode=rootNode)
- self.mVRTRaster = vrtRaster
- self.mColumnNames = ['Virtual Raster']
-
- def setData(self, index, value, role):
- node = self.idx2node(index)
- col = index.column()
-
- if role == Qt.EditRole:
- if isinstance(node, VRTRasterBandNode) and col == 0:
- if len(value) > 0:
- node.setName(value)
- node.mVirtualBand.setName(value)
- return True
-
-
- return False
-
-
-
- def srcFileIndices(self, srcFile):
- srcFileNodes = self.mRootNode.findChildNodes(VRTRasterInputSourceBandNode, recursive=True)
- return self.nodes2indexes(srcFileNodes)
-
- def removeSources(self, sources):
- assert isinstance(sources, list)
- for source in sources:
- self.mVRTRaster.removeInputSource(source)
-
- def removeNodes(self, nodes):
-
- for vBandNode in [n for n in nodes if isinstance(n, VRTRasterBandNode)]:
- self.mVRTRaster.removeVirtualBand(vBandNode.mVirtualBand)
-
- for vBandSrcNode in [n for n in nodes if isinstance(n, VRTRasterInputSourceBandNode)]:
- assert isinstance(vBandSrcNode, VRTRasterInputSourceBandNode)
- srcBand = vBandSrcNode.mSrc
-
- srcBand.virtualBand().removeSource(srcBand)
-
-
- def removeRows(self, row, count, parent):
- parentNode = self.idx2node(parent)
-
-
- if isinstance(parentNode, VRTRasterBandNode):
- #self.beginRemoveRows(parent, row, row+count-1)
- vBand = parentNode.mVirtualBand
- for n in parentNode.childNodes()[row:row+count]:
- vBand.removeSource(n.mSrc)
- #self.endRemoveRows()
- return True
- else:
- return False
-
-
- def flags(self, index):
- if not index.isValid():
- return Qt.ItemIsDropEnabled
-
- node = self.idx2node(index)
- flags = super(VRTRasterTreeModel, self).flags(index)
-
- if isinstance(node, VRTRasterBandNode):
- flags |= Qt.ItemIsDropEnabled
- flags |= Qt.ItemIsEditable
- if isinstance(node, VRTRasterInputSourceBandNode):
- flags |= Qt.ItemIsDropEnabled
- flags |= Qt.ItemIsDragEnabled
- return flags
-
- def dragEnterEvent(self, event):
- assert isinstance(event, QDragEnterEvent)
- if event.mimeData().hasFormat(u'hub.vrtbuilder/bandlist'):
- event.accept()
-
- def dragMoveEvent(self, event):
- assert isinstance(event, QDragMoveEvent)
- if event.mimeData().hasFormat(u'hub.vrtbuilder/bandlist'):
- event.accept()
-
-
- def dropEvent(self, event):
- assert isinstance(event, QDropEvent)
-
- if event.mimeData().hasFormat(u'hub.vrtbuilder/bandlist'):
- parent = self.mRootNode
- p = self.node2idx(parent)
- self.dropMimeData(event.mimeData(), event.dropAction(),0,0,p )
-
-
- event.accept()
-
- s = ""
-
-
- def mimeTypes(self):
- # specifies the mime types handled by this model
- types = []
- types.append('text/uri-list')
- types.append('hub.vrtbuilder/bandlist')
- return types
-
-
- def mimeData(self, indexes):
- indexes = sorted(indexes)
- nodes = [self.idx2node(i) for i in indexes]
-
- sourceBands = []
-
- for node in nodes:
- if isinstance(node, VRTRasterInputSourceBandNode):
- sourceBand = node.sourceBand()
- assert isinstance(sourceBand, VRTRasterInputSourceBand)
- sourceBands.append(sourceBand)
-
- sourceBands = list(OrderedDict.fromkeys(sourceBands))
- uriList = [sourceBand.mPath for sourceBand in sourceBands]
- uriList = list(OrderedDict.fromkeys(uriList))
-
- mimeData = QMimeData()
-
- if len(sourceBands) > 0:
- mimeData.setData('hub.vrtbuilder/bandlist', pickle.dumps(sourceBands))
-
- # set text/uri-list
- if len(uriList) > 0:
- mimeData.setUrls([QUrl(p) for p in uriList])
- mimeData.setText('\n'.join(uriList))
-
- return mimeData
-
-
-
- def dropMimeData(self, mimeData, action, row, column, parentIndex):
- if action == Qt.IgnoreAction:
- return True
-
- assert isinstance(mimeData, QMimeData)
- #assert isinstance(action, QDropEvent)
- sourceBands = []
-
- if u'hub.vrtbuilder/bandlist' in mimeData.formats():
- dump = mimeData.data(u'hub.vrtbuilder/bandlist')
- sourceBands = pickle.loads(dump)
-
- if u'hub.vrtbuilder/vrt.indices' in mimeData.formats():
- dump = mimeData.data(u'hub.vrtbuilder/vrt.indices')
- indices = pickle.loads(dump)
- s = ""
-
- if action == Qt.MoveAction:
- s = ""
-
- if len(sourceBands) == 0:
- return False
-
- #re-order source bands by
- #1. source file band index
- #2. source file
- #create a list like [[file 1 band1, file 2 band1, file 3 band 92],
- # [file 1 band2, file 2 band2, file 3 band 93]
- # . . .
- # ]
- sourceImages = {}
- for b in sourceBands:
- assert isinstance(b, VRTRasterInputSourceBand)
- if not b.mPath in sourceImages.keys():
- sourceImages[b.mPath] = []
- sourceImages[b.mPath].append(b)
- for p in sourceImages.keys():
- sourceImages[p] = sorted(sourceImages[p], key=lambda b: b.mBandIndex)
-
- if len(sourceImages) == 0:
- return True
- sourceBands = []
- while len(sourceImages) > 0:
- sourceBands.append([])
- for k in sourceImages.keys():
- sourceBands[-1].append(sourceImages[k].pop(0))
- if len(sourceImages[k]) == 0:
- del sourceImages[k]
-
-
- #ensure that we start with a VRTRasterBandNode
- parentNode = self.idx2node(parentIndex)
- if isinstance(parentNode, VRTRasterInputSourceBandNode):
- parentNode = parentNode.parentNode()
- elif isinstance(parentNode,VRTRasterNode):
- #1. set first VirtualBand as first input node
- vBand = VRTRasterBand()
- self.mVRTRaster.addVirtualBand(vBand)
- parentNode = self.mRootNode.findChildNodes(VRTRasterBandNode, recursive=False)[0]
-
- assert isinstance(parentNode, VRTRasterBandNode)
-
- #this is the first virtual band to insert sources in
- vBand = parentNode.mVirtualBand
- assert isinstance(vBand, VRTRasterBand)
- if row < 0:
- row = 0
-
- for bands in sourceBands:
- iSrc = row
- for src in bands:
- vBand.insertSource(iSrc, src)
- iSrc += 1
-
-
-
- if bands != sourceBands[-1]:
- # switch add a new virtual band if the recent vBand is the last one
- if vBand == self.mVRTRaster.mBands[-1]:
- self.mVRTRaster.addVirtualBand(VRTRasterBand())
-
- # switch to next virtual band
- vBand = self.mVRTRaster.mBands[self.mVRTRaster.mBands.index(vBand)+1]
-
- return True
-
-
- s = ""
- return False
-
- def supportedDragActions(self):
- return Qt.CopyAction | Qt.MoveAction
-
- def supportedDropActions(self):
- return Qt.CopyAction | Qt.MoveAction
-
-class VRTBuilderWidget(QFrame, loadUI('vrtbuilder.ui')):
-
- def __init__(self, parent=None):
- super(VRTBuilderWidget, self).__init__(parent)
- self.setupUi(self)
- self.sourceFileModel = SourceRasterModel(parent=self.treeViewSourceFiles)
-
- self.treeViewSourceFiles.setModel(self.sourceFileModel)
-
-
- self.mCrsManuallySet = False
- self.mBoundsManuallySet = False
-
- self.tbNoData.setValidator(QDoubleValidator())
-
- self.tbOutputPath.textChanged.connect(self.onOutputPathChanged)
-
- filter='GDAL Virtual Raster (*.vrt);;GeoTIFF (*.tiff *.tif);;ENVI (*.bsq *.bil *.bip)'
- self.btnSelectVRTPath.clicked.connect(lambda :
- self.tbOutputPath.setText(
- QFileDialog.getSaveFileName(self,
- directory=self.tbOutputPath.text(),
- caption='Select output image',
- filter=filter)
- ))
- self.buttonBox.button(QDialogButtonBox.Ok).clicked.connect(self.saveFile)
- self.vrtRaster = VRTRaster()
- self.vrtRasterLayer = VRTRasterVectorLayer(self.vrtRaster)
- self.vrtRasterLayer.dataChanged.connect(self.resetMap)
- self.mBackgroundLayer = None
- #self.vrtRasterLayer.editingStopped.connect(self.resetMap)
-
-
- assert isinstance(self.previewMap, QgsMapCanvas)
-
- self.previewMap.setLayers([self.vrtRasterLayer])
- self.resetMap()
-
-
- self.vrtRaster.sigCrsChanged.connect(self.updateSummary)
- self.vrtRaster.sigSourceBandInserted.connect(self.updateSummary)
- self.vrtRaster.sigSourceBandRemoved.connect(self.updateSummary)
-
- self.vrtRaster.sigBandInserted.connect(self.updateSummary)
- self.vrtRaster.sigBandRemoved.connect(self.updateSummary)
-
- self.vrtBuilderModel = VRTRasterTreeModel(parent=self.treeViewVRT, vrtRaster=self.vrtRaster)
- self.treeViewVRT.setModel(self.vrtBuilderModel)
-
- self.vrtTreeSelectionModel = VRTSelectionModel(
- self.treeViewVRT.model(),
- self.previewMap,
- self.vrtRasterLayer)
-
- self.vrtTreeSelectionModel.selectionChanged.connect(self.onVRTSelectionChanged)
-
- self.treeViewVRT.setSelectionModel(self.vrtTreeSelectionModel)
-
- # 2. expand the parent nodes
- #vBandNodes = set([n.parentNode() for n in srcNodes
- # if isinstance(n.parentNode(), VRTRasterBandNode)])
- #for n in vBandNodes:
- # self.treeViewVRT.expand(self.vrtBuilderModel.node2idx(n))
-
-
- #self.vrtBuilderModel.redirectDropEvent(self.treeViewVRT)
- #self.treeViewVRT.dragEnterEvent = self.vrtBuilderModel.dragEnterEvent
- #self.treeViewVRT.dragMoveEvent = self.vrtBuilderModel.dragMoveEvent
-
- self.treeViewVRT.setAutoExpandDelay(50)
- self.treeViewVRT.setDragEnabled(True)
- self.treeViewVRT.contextMenuEvent = self.vrtTreeViewContextMenuEvent
-
-
- self.btnExpandAllVRT.clicked.connect(lambda :self.expandSelectedNodes(self.treeViewVRT, True))
- self.btnCollapseAllVRT.clicked.connect(lambda: self.expandSelectedNodes(self.treeViewVRT, False))
-
- self.btnExpandAllSrc.clicked.connect(lambda :self.expandSelectedNodes(self.treeViewSourceFiles, True))
- self.btnCollapseAllSrc.clicked.connect(lambda: self.expandSelectedNodes(self.treeViewSourceFiles, False))
-
- self.btnAddVirtualBand.clicked.connect(lambda : self.vrtRaster.addVirtualBand(VRTRasterBand(name='Band {}'.format(len(self.vrtRaster)+1))))
- self.btnRemoveVirtualBands.clicked.connect(lambda : self.vrtBuilderModel.removeNodes(
- self.vrtBuilderModel.indexes2nodes(self.treeViewVRT.selectedIndexes())
- )
- )
- self.btnAddFromRegistry.clicked.connect(self.loadSrcFromMapLayerRegistry)
- self.btnAddSrcFiles.clicked.connect(lambda :
- self.sourceFileModel.addFiles(
- QFileDialog.getOpenFileNames(self, "Open raster images",
- directory='')
- ))
-
- self.btnRemoveSrcFiles.clicked.connect(lambda : self.sourceFileModel.removeFiles(
- [n.mPath for n in self.selectedSourceFileNodes()]
- ))
-
- self.mQgsProjectionSelectionWidget.dialog().setMessage('Set VRT CRS')
- self.mQgsProjectionSelectionWidget.crsChanged.connect(self.vrtRaster.setCrs)
-
- def resetMap(self, *args):
-
- lyrs = [self.vrtRasterLayer]
- if isinstance(self.mBackgroundLayer, QgsMapLayer):
- lyrs.insert(0, self.mBackgroundLayer)
-
- if lyrs != self.previewMap.layers():
- self.previewMap.setLayers(lyrs)
- self.previewMap.reset()
-
- def onVRTSelectionChanged(self, selected, deselected):
- self.btnRemoveVirtualBands.setEnabled(selected.count() > 0)
- # 2. expand the parent nodes
- model = self.vrtBuilderModel
- nodes = [model.idx2node(idx) for idx in selected.indexes()]
- selected = set([model.node2idx(n.parentNode()) for n in nodes if isinstance(n, VRTRasterInputSourceBandNode)])
- for idx in selected:
- self.treeViewVRT.expand(idx)
-
-
-
- def loadSrcFromMapLayerRegistry(self):
-
- reg = QgsMapLayerRegistry.instance()
- for lyr in reg.mapLayers().values():
- if isinstance(lyr, QgsRasterLayer):
- self.sourceFileModel.addFile(lyr.source())
-
- def expandNodes(self, treeView, nodes, expand):
- assert isinstance(treeView, QTreeView)
- model = treeView.model()
- assert isinstance(model, TreeModel)
- for node in nodes:
- treeView.setExpanded(model.node2idx(node))
-
- def expandSelectedNodes(self, treeView, expand):
- assert isinstance(treeView, QTreeView)
-
- indices = treeView.selectedIndexes()
- if len(indices) == 0:
- treeView.selectAll()
- indices += treeView.selectedIndexes()
- treeView.clearSelection()
- for idx in indices:
- treeView.setExpanded(idx, expand)
-
-
-
- def setBackgroundLayer(self, mapLayer):
- self.mBackgroundLayer = mapLayer
- self.resetMap()
-
- def saveFile(self):
- path = self.tbOutputPath.text()
- ext = os.path.splitext(path)[-1]
-
- saveBinary = ext != '.vrt'
- if saveBinary:
- pathVrt = path+'.vrt'
- else:
- pathVrt = path
-
-
- self.vrtRaster.saveVRT(pathVrt)
-
-
- def onOutputPathChanged(self, path):
- assert isinstance(self.buttonBox, QDialogButtonBox)
- isEnabled = False
- if len(path) > 0:
- ext = os.path.splitext(path)[-1].lower()
- isEnabled = ext in ['.vrt', '.bsq', '.tif', '.tiff']
-
- self.buttonBox.button(QDialogButtonBox.Ok).setEnabled(isEnabled)
-
-
- def onSrcModelSelectionChanged(self, selected, deselected):
-
- self.btnRemoveSrcFiles.setEnabled(len(self.selectedSourceFileNodes()) > 0)
- s = ""
-
-
- def addSourceFiles(self, files):
- """
- Adds a list of source files to the source file list.
- :param files: list-of-file-paths
- """
- self.sourceFileModel.addFiles(files)
-
- def updateSummary(self):
-
-
- self.tbSourceFileCount.setText('{}'.format(len(self.vrtRaster.sourceRaster())))
- self.tbVRTBandCount.setText('{}'.format(len(self.vrtRaster)))
-
- crs = self.vrtRaster.crs()
- if isinstance(crs, QgsCoordinateReferenceSystem):
- self.previewMap.setDestinationCrs(crs)
- if crs != self.mQgsProjectionSelectionWidget.crs():
- self.mQgsProjectionSelectionWidget.setCrs(crs)
- self.previewMap.refresh()
-
-
- def vrtTreeViewContextMenuEvent(self, event):
-
- idx = self.treeViewVRT.indexAt(event.pos())
- if not idx.isValid():
- pass
-
- selectedNodes = self.vrtBuilderModel.indexes2nodes(self.treeViewVRT.selectedIndexes())
- menu = QMenu(self.treeViewVRT)
- a = menu.addAction('Remove bands')
- a.setToolTip('Remove selected nodes')
- a.triggered.connect(lambda: self.vrtBuilderModel.removeNodes(selectedNodes))
-
- srcFiles = set()
- for n in selectedNodes:
- if isinstance(n, VRTRasterInputSourceBandNode):
- srcFiles.add(n.sourceBand().mPath)
-
- if len(srcFiles) > 0:
- a = menu.addAction('Remove sources')
- a.setToolTip('Remove all bands from selected source files.')
- a.triggered.connect(lambda : self.vrtBuilderModel.removeSources(srcFiles))
-
- menu.exec_(self.treeViewVRT.viewport().mapToGlobal(event.pos()))
- """
- if (menu & & menu->actions().count() != 0 )
- menu->exec (mapToGlobal(event->pos() ) );
- delete
- menu;
- """
-
- def mapReset(self):
-
- self.previewMap.refresh()
- self.vrtRasterLayer.setSelectedFeatures([])
-
-
-
- s = ""
-if __name__ == '__main__':
- import site, sys
- #add site-packages to sys.path as done by enmapboxplugin.py
-
- from timeseriesviewer import utils, DIR_EXAMPLES
- qgsApp = utils.initQgisApplication()
-
- from example.Images import Img_2014_03_20_LC82270652014079LGN00_BOA, re_2014_08_17
-
- #r = VRTRaster()
- #r.addFilesAsStack([Img_2014_03_20_LC82270652014079LGN00_BOA, Img_2014_04_29_LE72270652014119CUB00_BOA])
- #print(r.sourceRasterBounds())
- if False:
- drv = gdal.GetDriverByName('MEM')
- ds = drv.Create('', 50, 100, 0, gdal.GDT_Byte)
- from osgeo import gdal_array
- assert isinstance(ds, gdal.Dataset)
-
- import numpy as np
- data = np.ones((100,50), dtype=np.byte)
- dPt = data.__array_interface__['data']
-
- ns, nl, nb = ds.RasterXSize, ds.RasterYSize, ds.RasterCount
- #path = 'MEM:::DATAPOINTER={dPt},PIXELS={ns},LINES={nl},BANDS={nb},DATATYPE={dt},PIXELOFFSET=1,LINEOFFSET=300,BANDOFFSET=1'.format(
- # dPt=dPt, ns=ns, nl=nl, nb=nb, dt=dt)
- ds.AddBand(gdal.GDT_Byte)
- band = ds.GetRasterBand(1)
- band.WriteArray(data)
-
- arr2 = band.ReadAsArray()
- dPt2 = arr2.__array_interface__['data']
-
- s = ""
- #ds2= gdal.Open(path, gdal.GA_ReadOnly)
- band2 = ds2.GetRasterBand(1)
- data = band2.ReadAsArray()
- s = ""
-
- w = VRTBuilderWidget()
-
- import sys
+ return self
- if sys.platform == 'darwin':
- files = [
- r'/Users/Shared/Multitemp2017/01_Data/RapidEye/re_2012-07-25.vrt'
- #p2 = r'/Users/Shared/Multitemp2017/01_Data/Landsat/LC82270652014207LGN00.vrt'
- ,r'/Users/Shared/Multitemp2017/01_Data/CBERS/CBERS_4_MUX_20150820.vrt'
- ]
+ def __repr__(self):
+ return self.polygon.asWkt()
- else:
- files = [
- r'S:/temp/temp_ar/4benjamin/05_CBERS/CBERS_4_MUX_20150603_167_107_L4_BAND5_GRID_SURFACE.tif'
- #,r'D:/Repositories/QGIS_Plugins/hub-timeseriesviewer/example/Images/re_2014-06-25.tif'
- ,r'D:/Repositories/QGIS_Plugins/hub-timeseriesviewer/example/Images/2014-08-27_LC82270652014239LGN00_BOA.tif'
- ]
-
- w.addSourceFiles(files)
- #w.vrtRaster.addFilesAsStack([p1, p2, p3])
- #p = r'S:/temp/temp_ar/4benjamin/05_CBERS/CBERS_4_MUX_20150603_167_107_L4_BAND5_GRID_SURFACE.tif'
- #bLyr = QgsRasterLayer(p, 'backgroud', 'gdal', True)
- #QgsMapLayerRegistry.instance().addMapLayer(bLyr)
- #w.setBackgroundLayer(bLyr)
- w.show()
- pathTmp = os.path.join(DIR_EXAMPLES, 'test.vrt')
- #w.vrtRaster.saveVRT(pathTmp)
- # w.vrtBuilder.addVirtualBand(VRTRasterBand(name='Band 1'))
-
- qgsApp.exec_()
- qgsApp.exitQgis()