Newer
Older
1001
1002
1003
1004
1005
1006
1007
1008
1009
1010
1011
1012
1013
1014
1015
1016
1017
1018
1019
1020
1021
1022
1023
1024
1025
1026
1027
1028
1029
1030
1031
1032
1033
1034
1035
1036
1037
1038
1039
1040
1041
1042
def displayBandNames(provider_or_dataset, bands=None):
results = None
if isinstance(provider_or_dataset, QgsRasterLayer):
return displayBandNames(provider_or_dataset.dataProvider())
elif isinstance(provider_or_dataset, QgsRasterDataProvider):
if provider_or_dataset.name() == 'gdal':
ds = gdal.Open(provider_or_dataset.dataSourceUri())
results = displayBandNames(ds, bands=bands)
else:
# same as in QgsRasterRendererWidget::displayBandName
results = []
if bands is None:
bands = range(1, provider_or_dataset.bandCount() + 1)
for band in bands:
result = provider_or_dataset.generateBandName(band)
colorInterp ='{}'.format(provider_or_dataset.colorInterpretationName(band))
if colorInterp != 'Undefined':
result += '({})'.format(colorInterp)
results.append(result)
elif isinstance(provider_or_dataset, gdal.Dataset):
results = []
if bands is None:
bands = range(1, provider_or_dataset.RasterCount+1)
for band in bands:
b = provider_or_dataset.GetRasterBand(band)
descr = b.GetDescription()
if len(descr) == 0:
descr = 'Band {}'.format(band)
results.append(descr)
return results
class RasterDataProviderMockup(QgsRasterDataProvider):
def __init__(self):
super(RasterDataProviderMockup, self).__init__('')
class MapViewRenderSettings(QgsCollapsibleGroupBox, loadUI('mapviewrendersettings.ui')):
LUT_RENDERER = {QgsMultiBandColorRenderer:QgsMultiBandColorRendererWidget,
QgsSingleBandGrayRenderer:QgsSingleBandGrayRendererWidget,
QgsSingleBandPseudoColorRenderer:QgsSingleBandPseudoColorRendererWidget,
QgsPalettedRasterRenderer:QgsPalettedRendererWidget}
LUT_RENDERER[QgsMultiBandColorRenderer]=MultiBandColorRendererWidget
LUT_RENDERER[QgsSingleBandPseudoColorRenderer]=SingleBandPseudoColorRendererWidget
LUT_RENDERER[QgsSingleBandGrayRenderer]=SingleBandGrayRendererWidget
LUT_RENDERER[QgsPalettedRasterRenderer] = PalettedRendererWidget

Benjamin Jakimow
committed
sigRendererChanged = pyqtSignal()
def __init__(self, sensor, parent=None):
"""Constructor."""
1059
1060
1061
1062
1063
1064
1065
1066
1067
1068
1069
1070
1071
1072
1073
1074
1075
1076
1077
1078
1079
1080
# Set up the user interface from Designer.
# After setupUI you can access any designer object by doing
# self.<objectname>, and you can use autoconnect slots - see
# http://qt-project.org/doc/qt-4.8/designer-using-a-ui-file.html
# #widgets-and-dialogs-with-auto-connect
self.setupUi(self)
QApplication.clipboard().dataChanged.connect(self.onClipboardChanged)
assert isinstance(sensor, SensorInstrument)
self.mSensor = sensor
self.mSensor.sigNameChanged.connect(self.onSensorNameChanged)
self.setTitle(self.mSensor.name())
from timeseriesviewer.models import OptionListModel, Option
rasterRendererModel = OptionListModel()
#rasterRendererModel.addOption(Option(QgsMultiBandColorRendererWidget, name='multibandcolor (QGIS)', mRenderType = QgsMultiBandColorRenderer))
#rasterRendererModel.addOption(Option(QgsPalettedRendererWidget, name='paletted (QGIS)', mRenderType = QgsPalettedRasterRenderer))
#rasterRendererModel.addOption(Option(QgsSingleBandGrayRendererWidget, name='singlegray (QGIS)', mRenderType = QgsSingleBandGrayRenderer))
#rasterRendererModel.addOption(Option(QgsSingleBandPseudoColorRendererWidget, name='singlebandpseudocolor (QGIS)', mRenderType = QgsSingleBandPseudoColorRenderer))
rasterRendererModel.addOption(Option(MultiBandColorRendererWidget, name='Multibandcolor', mRenderType=QgsMultiBandColorRenderer))
rasterRendererModel.addOption(Option(SingleBandGrayRendererWidget, name='Singlegray', mRenderType=QgsSingleBandGrayRenderer))
rasterRendererModel.addOption(Option(SingleBandPseudoColorRendererWidget, name='Singleband Pseudocolor', mRenderType=QgsSingleBandPseudoColorRenderer))
rasterRendererModel.addOption(Option(PalettedRendererWidget, name='Paletted', mRenderType=QgsPalettedRasterRenderer))
self.mRasterRendererModel = rasterRendererModel
self.cbRenderType.setModel(self.mRasterRendererModel)
assert isinstance(self.stackedWidget, QStackedWidget)

Benjamin Jakimow
committed
self.mMockupCanvas = QgsMapCanvas(parent)
self.mMockupCanvas.setVisible(False)
self.mMockupRasterLayer = self.mSensor.mockupLayer()
self.mMockupCanvas.setLayers([self.mMockupRasterLayer])
for func in rasterRendererModel.optionValues():
#extent = self.canvas.extent()
#w = func.create(self.mMockupRasterLayer, self.mMockupRasterLayer.extent())
w = func(self.mMockupRasterLayer, self.mMockupRasterLayer.extent())

Benjamin Jakimow
committed
assert isinstance(w, QgsRasterRendererWidget)
w.setParent(self.stackedWidget)
w.setSizePolicy(QSizePolicy(QSizePolicy.Preferred, QSizePolicy.Preferred))
w.setMapCanvas(self.mMockupCanvas)
self.stackedWidget.addWidget(w)

Benjamin Jakimow
committed
s = ""
self.initActions()
def initActions(self):
self.btnPasteStyle.setDefaultAction(self.actionPasteStyle)
self.btnCopyStyle.setDefaultAction(self.actionCopyStyle)
self.btnApplyStyle.setDefaultAction(self.actionApplyStyle)
clipboardRenderer = rendererFromXml(QApplication.clipboard().mimeData())
self.actionPasteStyle.setEnabled(isinstance(clipboardRenderer, QgsRasterRenderer))
self.actionPasteStyle.triggered.connect(self.pasteStyleFromClipboard)
self.actionCopyStyle.triggered.connect(self.pasteStyleToClipboard)
self.actionApplyStyle.triggered.connect(self.applyStyle)
def mapCanvases(self):
return self.mMapCanvases[:]
def registerMapCanvas(self, mapCanvas):
assert isinstance(mapCanvas, MapCanvas)
self.mMapCanvases.append(mapCanvas)
mapCanvas.sigChangeSVRequest.connect(self.onMapCanvasRendererChangeRequest)
def onMapCanvasRendererChangeRequest(self, mapCanvas, renderer):
self.setRasterRenderer(renderer)
self.applyStyle()
s = ""
def onSensorNameChanged(self, newName):
self.setTitle(self.mSensor.name())
self.actionApplyStyle.setToolTip('Apply style to all map view images from "{}"'.format(self.mSensor.name()))
1143
1144
1145
1146
1147
1148
1149
1150
1151
1152
1153
1154
1155
1156
1157
1158
1159
1160
1161
1162
1163
1164
1165
1166
1167
1168
1169
1170
1171
1172
1173
1174
1175
1176
1177
def currentRenderWidget(self):
"""
Returns the current QgsRasterRendererWidget
:return: QgsRasterRendererWidget
"""
return self.stackedWidget.currentWidget()
def setRasterRenderer(self, renderer):
assert isinstance(renderer, QgsRasterRenderer)
assert isinstance(self.stackedWidget, QStackedWidget)
self.mMockupRasterLayer.setRenderer(renderer)
#find the widget class that fits
cls = None
for option in self.mRasterRendererModel:
if type(renderer) == option.mRenderType:
cls = option.value()
break
if cls == None:
return
widgets = []
for i in range(self.stackedWidget.count()):
w = self.stackedWidget.widget(i)
if isinstance(w, cls):
widgets.append(w)
if len(widgets) > 0:
for w in widgets:
assert isinstance(w, QgsRasterRendererWidget)
#w.setRasterLayer(self.mMockupRasterLayer)
#w.setFromRenderer(cloneRenderer(renderer))
w.setFromRenderer(renderer)
#w.doComputations()
w = widgets[0]
self.stackedWidget.setCurrentWidget(w)
self.cbRenderType.setCurrentIndex(self.stackedWidget.currentIndex())
def rasterRenderer(self):
return self.stackedWidget.currentWidget().renderer()
def apply(self):
mRendererWidget = self.currentRenderWidget()
mRendererWidget.doComputations()

benjamin.jakimow@geo.hu-berlin.de
committed
def onClipboardChanged(self):
mimeData = QApplication.clipboard().mimeData()
renderer = rendererFromXml(mimeData)
b = isinstance(renderer, QgsRasterRenderer)
#if b == False:
# print(mimeData.formats())
# s = ""
self.actionPasteStyle.setEnabled(b)

benjamin.jakimow@geo.hu-berlin.de
committed
def pasteStyleFromClipboard(self):
mimeData = QApplication.clipboard().mimeData()
renderer = rendererFromXml(mimeData)
if isinstance(renderer, QgsRasterRenderer):
self.setRasterRenderer(renderer)
def pasteStyleToClipboard(self):
xml = rendererToXml(self.rasterRenderer())
assert isinstance(xml, QDomDocument)
md = QMimeData()
#['application/qgis.style', 'text/plain']
md.setData('application/qgis.style', xml.toByteArray())
md.setData('text/plain', xml.toByteArray())
QApplication.clipboard().setMimeData(md)
if isinstance(r, QgsRasterRenderer):
for mapCanvas in self.mapCanvases():
assert isinstance(mapCanvas, MapCanvas)
mapCanvas.addToRefreshPipeLine(MapCanvas.Command.RefreshRenderer)
RENDER_CLASSES = {}
RENDER_CLASSES['rasterrenderer'] = {
'singlebandpseudocolor': QgsSingleBandPseudoColorRenderer,
'singlebandgray': QgsSingleBandGrayRenderer,
'paletted': QgsPalettedRasterRenderer,
'multibandcolor': QgsMultiBandColorRenderer,
'hillshade': QgsHillshadeRenderer
}
RENDER_CLASSES['renderer-v2'] = {
'categorizedSymbol':QgsCategorizedSymbolRenderer,
'singleSymbol':QgsSingleSymbolRenderer
}
"""
Returns a renderer XML representation
:param renderer: QgsRasterRender | QgsFeatureRenderer
:return: QDomDocument
"""
if isinstance(renderer, QgsRasterRenderer):
1262
1263
1264
1265
1266
1267
1268
1269
1270
1271
1272
1273
1274
1275
1276
1277
1278
1279
1280
1281
1282
1283
1284
1285
1286
1287
#create a dummy raster layer
import uuid
from timeseriesviewer.virtualrasters import write_vsimem, read_vsimem
xml = """<VRTDataset rasterXSize="1" rasterYSize="1">
<GeoTransform> 0.0000000000000000e+00, 1.0000000000000000e+00, 0.0000000000000000e+00, 0.0000000000000000e+00, 0.0000000000000000e+00, -1.0000000000000000e+00</GeoTransform>
<VRTRasterBand dataType="Float32" band="1">
<Metadata>
<MDI key="STATISTICS_MAXIMUM">0</MDI>
<MDI key="STATISTICS_MEAN">0</MDI>
<MDI key="STATISTICS_MINIMUM">0</MDI>
<MDI key="STATISTICS_STDDEV">0</MDI>
</Metadata>
<Description>Band 1</Description>
<Histograms>
<HistItem>
<HistMin>0</HistMin>
<HistMax>0</HistMax>
<BucketCount>1</BucketCount>
<IncludeOutOfRange>0</IncludeOutOfRange>
<Approximate>0</Approximate>
<HistCounts>0</HistCounts>
</HistItem>
</Histograms>
</VRTRasterBand>
</VRTDataset>
"""

Benjamin Jakimow
committed
path = '/vsimem/{}.vrt'.format(uuid.uuid4())
drv = gdal.GetDriverByName('VRT')
assert isinstance(drv, gdal.Driver)
write_vsimem(path, xml)
lyr.setRenderer(renderer.clone())
lyr.exportNamedStyle(doc)
#remove dummy raster layer
lyr = None
drv.Delete(path)
elif isinstance(renderer, QgsFeatureRenderer):
#todo: distinguish vector type from requested renderer
lyr = QgsVectorLayer('Point?crs=epsg:4326&field=id:integer', 'dummy', 'memory')
lyr.setRenderer(renderer.clone())
lyr.exportNamedStyle(doc)
else:
raise NotImplementedError()
1310
1311
1312
1313
1314
1315
1316
1317
1318
1319
1320
1321
1322
1323
1324
1325
1326
1327
1328
1329
1330
1331
1332
1333
1334
1335
1336
1337
1338
1339
1340
1341
1342
1343
1344
1345
1346
1347
1348
1349
1350
1351
1352
1353
1354
1355
return doc
def rendererFromXml(xml):
"""
Reads a string `text` and returns the first QgsRasterRenderer or QgsFeatureRenderer (if defined).
:param text:
:return:
"""
if isinstance(xml, QMimeData):
for format in ['application/qgis.style', 'text/plain']:
if format in xml.formats():
dom = QDomDocument()
dom.setContent(xml.data(format))
return rendererFromXml(dom)
return None
elif isinstance(xml, str):
dom = QDomDocument()
dom.setContent(xml)
return rendererFromXml(dom)
assert isinstance(xml, QDomDocument)
root = xml.documentElement()
for baseClass, renderClasses in RENDER_CLASSES.items():
elements = root.elementsByTagName(baseClass)
if elements.count() > 0:
elem = elements.item(0).toElement()
typeName = elem.attributes().namedItem('type').nodeValue()
if typeName in renderClasses.keys():
rClass = renderClasses[typeName]
if baseClass == 'rasterrenderer':
return rClass.create(elem, DUMMY_RASTERINTERFACE)
elif baseClass == 'renderer-v2':
context = QgsReadWriteContext()
return rClass.load(elem, context)
#return rClass.create(elem)
else:
print(typeName)
s =""
return None

benjamin.jakimow@geo.hu-berlin.de
committed
class DatumViewUI(QFrame, loadUI('timeseriesdatumview.ui')):
1359
1360
1361
1362
1363
1364
1365
1366
1367
1368
1369
1370
1371
1372
1373
1374
1375
1376
1377
1378
1379
1380
1381
1382
1383
1384
1385
1386
"""
Widget to host the MapCanvases of all map views that relate to a single Datum-Sensor combinbation.
"""
def __init__(self, title='<#>', parent=None):
super(DatumViewUI, self).__init__(parent)
self.setupUi(self)
def sizeHint(self):
m = self.layout().contentsMargins()
s = QSize(0, 0)
map = None
widgets = [self.layout().itemAt(i).widget() for i in range(self.layout().count())]
widgets = [w for w in widgets if isinstance(w, QWidget)]
maps = [w for w in widgets if isinstance(w, MapCanvas)]
others = [w for w in widgets if not isinstance(w, MapCanvas)]
s = self.layout().spacing()
m = self.layout().contentsMargins()
def totalHeight(widgetList):
total = QSize(0,0)
for w in widgetList:
ws = w.size()
if ws.width() == 0:
ws = w.sizeHint()
if w.isVisible():
total.setWidth(max([total.width(), ws.width()]))
total.setHeight(total.height() + ws.height())
return total
baseSize = totalHeight(widgets)
if baseSize.width() == 0:
for o in others:
baseSize.setWidth(9999)
s = QSize(baseSize.width() + m.left() + m.right(),
baseSize.height() + m.top() + m.bottom())
1399
1400
1401
1402
1403
1404
1405
1406
1407
1408
1409
1410
1411
1412
1413
1414
1415
1416
1417
1418
1419
1420
1421
1422
1423
1424
1425
1426
1427
1428
return s
"""
def sizeHint(self):
if not self.ui.isVisible():
return QSize(0,0)
else:
#return self.ui.sizeHint()
size = self.ui.sizeHint()
s = self.ui.layout().spacing()
m = self.ui.layout().contentsMargins()
dx = m.left() + m.right() + s
dy = self.ui.layout().spacing()
n = len([m for m in self.mapCanvases.keys() if m.isVisible()])
if n > 0:
baseSize = self.mapCanvases.values()[0].size()
size = QSize(baseSize.width()+ dx, \
size.height()+ (n+1)*(dy+2*s))
else:
s = ""
return size
"""

benjamin.jakimow@geo.hu-berlin.de
committed
sigRenderProgress = pyqtSignal(int,int)
sigLoadingStarted = pyqtSignal(MapView, TimeSeriesDatum)
sigLoadingFinished = pyqtSignal(MapView, TimeSeriesDatum)
sigVisibilityChanged = pyqtSignal(bool)
def __init__(self, timeSeriesDatum:TimeSeriesDatum, stv, parent=None):
assert isinstance(timeSeriesDatum, TimeSeriesDatum)
assert isinstance(stv, SpatialTemporalVisualization)

benjamin.jakimow@geo.hu-berlin.de
committed

benjamin.jakimow@geo.hu-berlin.de
committed
self.ui.create()
self.showLoading(False)
self.wOffset = self.ui.layout().count()-1

benjamin.jakimow@geo.hu-berlin.de
committed
self.minHeight = self.ui.height()

benjamin.jakimow@geo.hu-berlin.de
committed
self.renderProgress = dict()
assert isinstance(stv, SpatialTemporalVisualization)
self.STV = stv
self.mSensor.sigNameChanged.connect(lambda :self.setColumnInfo())
self.TSD.sigVisibilityChanged.connect(self.setVisibility)

benjamin.jakimow@geo.hu-berlin.de
committed
self.setColumnInfo()
self.mRenderState = dict()

benjamin.jakimow@geo.hu-berlin.de
committed

benjamin.jakimow@geo.hu-berlin.de
committed
def setColumnInfo(self):
labelTxt = '{}\n{}'.format(str(self.TSD.mDate), self.TSD.mSensor.name())
tooltip = '\n'.join([tss.uri()for tss in self.TSD.sources()])

benjamin.jakimow@geo.hu-berlin.de
committed
self.ui.labelTitle.setText(labelTxt)
self.ui.labelTitle.setToolTip(tooltip)

benjamin.jakimow@geo.hu-berlin.de
committed
def setVisibility(self, b):
self.ui.setVisible(b)
self.sigVisibilityChanged.emit(b)

benjamin.jakimow@geo.hu-berlin.de
committed
def setHighlighted(self, b=True, timeout=1000):
styleOn = """.QFrame {
border: 4px solid red;
border-radius: 4px;
}"""
styleOff = """"""
if b is True:
self.ui.setStyleSheet(styleOn)
if timeout > 0:
QTimer.singleShot(timeout, lambda : self.setHighlighted(b=False))
else:
self.ui.setStyleSheet(styleOff)

benjamin.jakimow@geo.hu-berlin.de
committed
def removeMapView(self, mapView):
self.ui.layout().removeWidget(canvas)

benjamin.jakimow@geo.hu-berlin.de
committed
canvas.close()
def mapCanvases(self)->list:
"""
Retuns the MapCanvases of this DataView
:return: [list-of-MapCanvases]
"""
return self.mMapCanvases.values()
def refresh(self):
"""
Refreshes the MapCanvases in this DatumView, if they are not hidden behind a scroll area.
"""

benjamin.jakimow@geo.hu-berlin.de
committed
if self.ui.isVisible():
for c in self.mapCanvases():

benjamin.jakimow@geo.hu-berlin.de
committed
if c.isVisible():

benjamin.jakimow@geo.hu-berlin.de
committed
def insertMapView(self, mapView):
assert isinstance(mapView, MapView)
from timeseriesviewer.mapcanvas import MapCanvas

benjamin.jakimow@geo.hu-berlin.de
committed
mapCanvas.setObjectName('MapCanvas {} {}'.format(mapView.title(), self.TSD.mDate))

Benjamin Jakimow
committed
self.registerMapCanvas(mapView, mapCanvas)
mapCanvas.setMapView(mapView)
mapCanvas.setTSD(self.TSD)

benjamin.jakimow@geo.hu-berlin.de
committed
mapView.registerMapCanvas(self.mSensor, mapCanvas)
mapCanvas.renderComplete.connect(lambda : self.onRenderingChange(False))
mapCanvas.renderStarting.connect(lambda : self.onRenderingChange(True))
#mapCanvas.sigMapRefreshed[float, float].connect(
# lambda dt: self.STV.TSV.ui.dockSystemInfo.addTimeDelta('Map {}'.format(self.mSensor.name()), dt))
#mapCanvas.sigMapRefreshed.connect(
# lambda dt: self.STV.TSV.ui.dockSystemInfo.addTimeDelta('All Sensors', dt))

benjamin.jakimow@geo.hu-berlin.de
committed
def showLoading(self, b):
if b:
self.ui.progressBar.setRange(0, 0)
self.ui.progressBar.setValue(-1)
else:
self.ui.progressBar.setRange(0,1)
self.ui.progressBar.setValue(0)
def onRenderingChange(self, b):
mc = self.sender()
#assert isinstance(mc, QgsMapCanvas)
self.mRenderState[mc] = b
self.showLoading(any(self.mRenderState.values()))
def onRendering(self, *args):
renderFlags = [m.renderFlag() for m in self.mMapCanvases.values()]
drawFlags = [m.isDrawing() for m in self.mMapCanvases.values()]

benjamin.jakimow@geo.hu-berlin.de
committed
# print((renderFlags, drawFlags))
isLoading = any(renderFlags)
self.showLoading(isLoading)
s = ""

benjamin.jakimow@geo.hu-berlin.de
committed
def registerMapCanvas(self, mapView, mapCanvas):
from timeseriesviewer.mapcanvas import MapCanvas
assert isinstance(mapCanvas, MapCanvas)

Benjamin Jakimow
committed
mapCanvas.setVisible(mapView.isVisible())
#mapView.sigTitleChanged.connect(lambda title : mapCanvas.setSaveFileName('{}_{}'.format(self.TSD.date, title)))
mapCanvas.mapLayerModel().addMapLayerSources(self.TSD.qgsMimeDataUtilsUris())
#self.ui.layout().insertWidget(self.wOffset + len(self.mapCanvases), mapCanvas)
self.ui.layout().insertWidget(self.ui.layout().count() - 1, mapCanvas)
self.ui.update()
#register signals handled on (this) DV level
mapCanvas.renderStarting.connect(lambda: self.sigLoadingStarted.emit(mapView, self.TSD))
mapCanvas.mapCanvasRefreshed.connect(lambda: self.sigLoadingFinished.emit(mapView, self.TSD))
mapCanvas.sigShowProfiles.connect(lambda c, t : mapView.sigShowProfiles.emit(c,mapCanvas, t))
mapCanvas.sigChangeDVRequest.connect(self.onMapCanvasRequest)

benjamin.jakimow@geo.hu-berlin.de
committed
def onMapCanvasRequest(self, mapCanvas, key):
if key == 'hide_date':
self.TSD.setVisibility(False)
QApplication.clipboard().setText(self.TSD.mSensor.name())
QApplication.clipboard().setText('\n'.join(self.TSD.sourceUris()))

benjamin.jakimow@geo.hu-berlin.de
committed
def __lt__(self, other):
assert isinstance(other, DatumView)
return self.TSD < other.TSD

benjamin.jakimow@geo.hu-berlin.de
committed
def __eq__(self, other):
assert isinstance(other, DatumView)
return self.TSD == other.TSD

benjamin.jakimow@geo.hu-berlin.de
committed
class SpatialTemporalVisualization(QObject):
"""
"""
sigLoadingStarted = pyqtSignal(DatumView, MapView)
sigLoadingFinished = pyqtSignal(DatumView, MapView)
sigShowProfiles = pyqtSignal(SpatialPoint, MapCanvas, str)

benjamin.jakimow@geo.hu-berlin.de
committed
sigShowMapLayerInfo = pyqtSignal(dict)
sigSpatialExtentChanged = pyqtSignal(SpatialExtent)
sigCRSChanged = pyqtSignal(QgsCoordinateReferenceSystem)

benjamin.jakimow@geo.hu-berlin.de
committed
def __init__(self, timeSeriesViewer):
super(SpatialTemporalVisualization, self).__init__()

benjamin.jakimow@geo.hu-berlin.de
committed
#assert isinstance(timeSeriesViewer, TimeSeriesViewer), timeSeriesViewer
#default map settings
self.mSpatialExtent = SpatialExtent.world()
self.mCRS = self.mSpatialExtent.crs()

Benjamin Jakimow
committed
self.mSize = QSize(200, 200)
self.mColor = Qt.black

benjamin.jakimow@geo.hu-berlin.de
committed
self.ui = timeSeriesViewer.ui
#map-tool handling
self.mMapToolActivator = None
self.mMapTools = []

benjamin.jakimow@geo.hu-berlin.de
committed
self.scrollArea = self.ui.scrollAreaSubsets
assert isinstance(self.scrollArea, MapViewScrollArea)
#self.scrollArea.sigResized.connect(self.refresh())
#self.scrollArea.horizontalScrollBar().valueChanged.connect(self.mRefreshTimer.start)

benjamin.jakimow@geo.hu-berlin.de
committed
self.TSV = timeSeriesViewer

benjamin.jakimow@geo.hu-berlin.de
committed
self.targetLayout = self.ui.scrollAreaSubsetContent.layout()
#self.MVC = MapViewCollection(self)
#self.MVC.sigShowProfiles.connect(self.sigShowProfiles.emit)
self.MVC = self.ui.dockMapViews
assert isinstance(self.MVC, MapViewCollectionDock)

benjamin.jakimow@geo.hu-berlin.de
committed
self.MVC.sigShowProfiles.connect(self.sigShowProfiles.emit)
self.MVC.sigMapViewAdded.connect(self.onMapViewAdded)

benjamin.jakimow@geo.hu-berlin.de
committed
self.vectorOverlay = None
self.DVC.sigResizeRequired.connect(self.adjustScrollArea)

Benjamin Jakimow
committed
#self.DVC.sigLoadingStarted.connect(self.ui.dockRendering.addStartedWork)
#self.DVC.sigLoadingFinished.connect(self.ui.dockRendering.addFinishedWork)
#self.timeSeriesDateViewCollection.sigSpatialExtentChanged.connect(self.setSpatialExtent)
self.TS.sigTimeSeriesDatesAdded.connect(self.DVC.addDates)
self.TS.sigTimeSeriesDatesRemoved.connect(self.DVC.removeDates)

benjamin.jakimow@geo.hu-berlin.de
committed
#add dates, if already existing

benjamin.jakimow@geo.hu-berlin.de
committed
if len(self.TS) > 0:

benjamin.jakimow@geo.hu-berlin.de
committed
self.mMapRefreshTimer = QTimer(self)
self.mMapRefreshTimer.timeout.connect(self.timedCanvasRefresh)
self.mMapRefreshTimer.setInterval(500)
self.mNumberOfHiddenMapsToRefresh = 2
def timedCanvasRefresh(self, *args, force:bool=False):
#do refresh maps
assert isinstance(self.scrollArea, MapViewScrollArea)
visibleMaps = [m for m in self.mapCanvases() if m.isVisibleToViewport()]
hiddenMaps = sorted([m for m in self.mapCanvases() if not m.isVisibleToViewport()],
n = 0
#redraw all visible maps
for c in visibleMaps:
assert isinstance(c, MapCanvas)
c.timedRefresh()
n += 1
if n < 10:
#refresh up to mNumberOfHiddenMapsToRefresh maps which are not visible to the user
i = 0
for c in hiddenMaps:
assert isinstance(c, MapCanvas)
c.timedRefresh()
i += 1
if i >= self.mNumberOfHiddenMapsToRefresh and not force:
break

Benjamin Jakimow
committed
def mapViewFromCanvas(self, mapCanvas:MapCanvas)->MapView:
"""
Returns the MapView a mapCanvas belongs to
:param mapCanvas: MapCanvas
:return: MapView
"""
for mapView in self.MVC:
assert isinstance(mapView, MapView)
if mapCanvas in mapView.mapCanvases():
return mapView
return None
def onMapViewAdded(self, *args):
self.adjustScrollArea()
s = ""

benjamin.jakimow@geo.hu-berlin.de
committed
def createMapView(self):
self.MVC.createMapView()
def registerMapCanvas(self, mapCanvas:MapCanvas):
"""
Connects a MapCanvas and its signals
:param mapCanvas: MapCanvas
"""
from timeseriesviewer.mapcanvas import MapCanvas
assert isinstance(mapCanvas, MapCanvas)
mapCanvas.setMapLayerStore(self.TSV.mMapLayerStore)
self.mMapCanvases.append(mapCanvas)
#set general canvas properties
mapCanvas.setFixedSize(self.mSize)
mapCanvas.setSpatialExtent(self.mSpatialExtent)

Benjamin Jakimow
committed
mapCanvas.setLabelingModel(self.TSV.ui.dockLabeling.mLabelAttributeModel)
#register on map canvas signals
def onChanged(e, mapCanvas0=None):
self.setSpatialExtent(e, mapCanvas0=mapCanvas0)
#mapCanvas.sigSpatialExtentChanged.connect(lambda e: self.setSpatialExtent(e, mapCanvas0=mapCanvas))
mapCanvas.sigSpatialExtentChanged.connect(lambda e: onChanged(e, mapCanvas0=mapCanvas))

Benjamin Jakimow
committed
mapCanvas.sigCrosshairPositionChanged.connect(self.onCrosshairChanged)
def onCrosshairChanged(self, spatialPoint:SpatialPoint):

Benjamin Jakimow
committed
"""
Synchronizes all crosshair positions. Takes care of CRS differences.
:param spatialPoint: SpatialPoint of new Crosshair position

Benjamin Jakimow
committed
"""
from timeseriesviewer.crosshair import CrosshairStyle

Benjamin Jakimow
committed
srcCanvas = self.sender()
if isinstance(srcCanvas, MapCanvas):
dstCanvases = [c for c in self.mapCanvases() if c != srcCanvas]
else:
dstCanvases = [c for c in self.mapCanvases()]
if isinstance(spatialPoint, SpatialPoint):

Benjamin Jakimow
committed
for mapCanvas in dstCanvases:
mapCanvas.setCrosshairPosition(spatialPoint, emitSignal=False)

Benjamin Jakimow
committed
def setCrosshairStyle(self, crosshairStyle:CrosshairStyle):
"""
Sets a crosshair style to all map canvas
:param crosshairStyle: CrosshairStyle
"""
for mapView in self.mapViews():
assert isinstance(mapView, MapView)
mapView.setCrosshairStyle(crosshairStyle)

benjamin.jakimow@geo.hu-berlin.de
committed
def setCrosshairVisibility(self, b:bool):
assert isinstance(b, bool)
self.onCrosshairChanged(b)

benjamin.jakimow@geo.hu-berlin.de
committed
def setVectorLayer(self, lyr):
self.MVC.setVectorLayer(lyr)

benjamin.jakimow@geo.hu-berlin.de
committed
assert isinstance(size, QSize)
from timeseriesviewer.mapcanvas import MapCanvas
for mapCanvas in self.mMapCanvases:
assert isinstance(mapCanvas, MapCanvas)
mapCanvas.setFixedSize(size)
self.sigMapSizeChanged.emit(self.mSize)

benjamin.jakimow@geo.hu-berlin.de
committed
self.adjustScrollArea()

benjamin.jakimow@geo.hu-berlin.de
committed
def refresh(self):
"""
Refreshes all visible MapCanvases
"""
for c in self.mapCanvases():
assert isinstance(c, MapCanvas)
c.refresh()
#self.mMapRefreshTimer.stop()

benjamin.jakimow@geo.hu-berlin.de
committed
def adjustScrollArea(self):
#adjust scroll area widget to fit all visible widgets
m = self.targetLayout.contentsMargins()

benjamin.jakimow@geo.hu-berlin.de
committed
w = h = 0
s = QSize()
r = None
tsdViews = [v for v in self.DVC if v.ui.isVisible()]
mapViews = [v for v in self.MVC if v.isVisible()]
nX = len(tsdViews)
nY = len(mapViews)
spacing = self.targetLayout.spacing()
margins = self.targetLayout.contentsMargins()

benjamin.jakimow@geo.hu-berlin.de
committed
sizeX = 1
sizeY = 50
if nX > 0:
s = tsdViews[0].ui.sizeHint().width()
s = nX * (s + spacing) + margins.left() + margins.right()
sizeX = s
if nY > 0 and nX > 0:
s = tsdViews[0].ui.sizeHint().height()
s = s + margins.top() + margins.bottom()

benjamin.jakimow@geo.hu-berlin.de
committed
#s = tsdViews[0].ui.sizeHint()
#s = QSize(nX * (s.width() + spacing) + margins.left() + margins.right(),
# s.height() + margins.top() + margins.bottom())

benjamin.jakimow@geo.hu-berlin.de
committed
#print(sizeX, sizeY)
self.targetLayout.parentWidget().resize(QSize(sizeX, sizeY))

benjamin.jakimow@geo.hu-berlin.de
committed
def setMapTool(self, mapToolKey, *args, **kwds):
# filter map tools
self.mMapToolActivator = self.sender()
del self.mMapTools[:]

Benjamin Jakimow
committed
from timeseriesviewer.maptools import MapTools, CursorLocationMapTool, SpectralProfileMapTool, TemporalProfileMapTool
for canvas in self.mMapCanvases:
mt = None
if mapToolKey in MapTools.mapToolKeys():
mt = MapTools.create(mapToolKey, canvas, *args, **kwds)
if isinstance(mapToolKey, QgsMapTool):
mt = MapTools.copy(mapToolKey, canvas, *args, **kwds)
if isinstance(mt, QgsMapTool):
canvas.setMapTool(mt)
self.mMapTools.append(mt)
#if required, link map-tool with specific slots
if isinstance(mt, CursorLocationMapTool):
mt.sigLocationRequest[SpatialPoint, QgsMapCanvas].connect(lambda c, m : self.sigShowProfiles.emit(c,m, mapToolKey))
return self.mMapTools

benjamin.jakimow@geo.hu-berlin.de
committed
def setMaxTSDViews(self, n=-1):
self.nMaxTSDViews = n
#todo: remove views
def setSpatialCenter(self, center, mapCanvas0=None):
assert isinstance(center, SpatialPoint)
center = center.toCrs(self.mCRS)
if not isinstance(center, SpatialPoint):
return
self.mSpatialExtent.setCenter(center)
for mapCanvas in self.mMapCanvases:
if mapCanvas != mapCanvas0:
oldState = mapCanvas.blockSignals(True)
mapCanvas.setCenter(center)
mapCanvas.blockSignals(oldState)
self.sigSpatialExtentChanged.emit(self.mSpatialExtent)
def setSpatialCenter(self, center:SpatialPoint, mapCanvas0=None):
"""
Sets the MapCanvas center.
:param center: SpatialPoint
:param mapCanvas0: MapCanvas0 optional
"""
assert isinstance(center, SpatialPoint)
center = center.toCrs(self.mCRS)
if not isinstance(center, SpatialPoint):
return None
for mapCanvas in self.mapCanvases():
assert isinstance(mapCanvas, MapCanvas)
if mapCanvas != mapCanvas0:
center0 = mapCanvas.spatialCenter()
if center0 != center:
oldState = mapCanvas.blockSignals(True)
mapCanvas.setCenter(center)
mapCanvas.blockSignals(oldState)
self.mMapRefreshTimer.start()
def setSpatialExtent(self, extent, mapCanvas0=None):
"""
Sets the spatial extent of all MapCanvases
:param extent: SpatialExtent
:param mapCanvas0:
:return:
"""
assert isinstance(extent, SpatialExtent)
extent = extent.toCrs(self.mCRS)
if not isinstance(extent, SpatialExtent) \
or extent.isEmpty() or not extent.isFinite() \
or extent.width() <= 0 \
or extent.height() <= 0 \
or extent == self.mSpatialExtent:
if self.mSpatialExtent == extent:
return
for mapCanvas in self.mapCanvases():
assert isinstance(mapCanvas, MapCanvas)
extent0 = mapCanvas.spatialExtent()
if mapCanvas != mapCanvas0 and extent0 != extent:
self.sigSpatialExtentChanged.emit(extent)

benjamin.jakimow@geo.hu-berlin.de
committed
def setBackgroundColor(self, color:QColor):
"""
Sets the MapCanvas background color
:param color: QColor
"""
assert isinstance(color, QColor)
self.mColor = color
for mapCanvas in self.mMapCanvases:
assert isinstance(mapCanvas, MapCanvas)
mapCanvas.setCanvasColor(color)
def backgroundColor(self)->QColor:
"""
Returns the MapCanvas background color
:return: QColor
"""
return self.mColor

Benjamin Jakimow
committed
def mapCanvases(self, mapView=None)->list:
"""
Returns MapCanvases
:param mapView: a MapView to return MapCanvases from only, defaults to None
:return: [list-of-MapCanvas]
"""
if isinstance(mapView, MapView):
s = ""

Benjamin Jakimow
committed
def mapViews(self)->list:
"""
Returns a list of all mapviews
:return [list-of-MapViews]:
"""
return self.MVC[:]
def setCrs(self, crs):
assert isinstance(crs, QgsCoordinateReferenceSystem)
transform = QgsCoordinateTransform()
transform.setSourceCrs(self.mCRS)
transform.setDestinationCrs(crs)
if transform.isValid() and not transform.isShortCircuited():

Benjamin Jakimow
committed
for mapCanvas in self.mapCanvases():
# print(('STV set CRS {} {}', str(mapCanvas), self.mCRS.description()))
mapCanvas.setDestinationCrs(QgsCoordinateReferenceSystem(crs))
"""
from timeseriesviewer.utils import saveTransform
if saveTransform(self.mSpatialExtent, self.mCRS, crs):
self.mCRS = crs