Newer
Older
# -*- coding: utf-8 -*-
"""
/***************************************************************************

Benjamin Jakimow
committed
EO Time Series Viewer
-------------------
begin : 2017-08-04
git sha : $Format:%H$
copyright : (C) 2017 by HU-Berlin
email : benjamin.jakimow@geo.hu-berlin.de
***************************************************************************/
/***************************************************************************
* *
* 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. *
* *
***************************************************************************/
"""
# noinspection PyPep8Naming
import os, sys, pickle, datetime, re, collections
from collections import OrderedDict
from qgis.gui import *
from qgis.core import *
from PyQt5.Qt import *
from PyQt5.QtCore import *
from PyQt5.QtXml import *
from PyQt5.QtGui import *
from PyQt5.QtWidgets import *
import numpy as np
import pyqtgraph as pg
from pyqtgraph import functions as fn
from pyqtgraph import AxisItem
from osgeo import ogr, osr, gdal
from timeseriesviewer.timeseries import TimeSeries, TimeSeriesDatum, SensorInstrument
from timeseriesviewer.plotstyling import PlotStyle
from timeseriesviewer.pixelloader import PixelLoader, PixelLoaderTask
from timeseriesviewer.utils import *
from timeseriesviewer.models import OptionListModel, Option
LABEL_EXPRESSION_2D = 'DN or Index'
LABEL_TIME = 'Date'
DEBUG = False
OPENGL_AVAILABLE = False
try:
import OpenGL
def qgsFieldFromKeyValue(fieldName, value):
t = type(value)
if t in [int, float] or np.isreal(value):
fLen = 0
fPrec = 0
fComm = ''
fType = ''
f = QgsField(fieldName, QVariant.Double, 'double', 40, 5)
else:
f = QgsField(fieldName, QVariant.String, 'text', 40, 5)
return f
def sensorExampleQgsFeature(sensor, singleBandOnly=False):
# populate with exemplary band values (generally stored as floats)
if sensor is None:
singleBandOnly = True
fieldValues = collections.OrderedDict()
if singleBandOnly:
fieldValues['b'] = 1.0
else:
for b in range(sensor.nb):
fn = bandIndex2bandKey(b)
fieldValues[fn] = 1.0
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
date = datetime.date.today()
doy = dateDOY(date)
fieldValues['doy'] = doy
fieldValues['date'] = str(date)
fields = QgsFields()
for k, v in fieldValues.items():
fields.append(qgsFieldFromKeyValue(k,v))
f = QgsFeature(fields)
for k, v in fieldValues.items():
f.setAttribute(k, v)
return f
def dateDOY(date):
if isinstance(date, np.datetime64):
date = date.astype(datetime.date)
return date.timetuple().tm_yday
def daysPerYear(year):
if isinstance(year, np.datetime64):
year = year.astype(datetime.date)
if isinstance(year, datetime.date):
year = year.timetuple().tm_year
return dateDOY(datetime.date(year=year, month=12, day=31))
def date2num(d):
#kindly taken from https://stackoverflow.com/questions/6451655/python-how-to-convert-datetime-dates-to-decimal-years
if isinstance(d, np.datetime64):
d = d.astype(datetime.datetime)
if isinstance(d, QDate):
d = datetime.date(d.year(), d.month(), d.day())
assert isinstance(d, datetime.date)
yearDuration = daysPerYear(d)
yearElapsed = d.timetuple().tm_yday
fraction = float(yearElapsed) / float(yearDuration)
if fraction == 1.0:
fraction = 0.9999999
return float(d.year) + fraction
def num2date(n, dt64=True, qDate=False):
n = float(n)
if n < 1:
n += 1
year = int(n)
fraction = n - year
yearDuration = daysPerYear(year)
yearElapsed = fraction * yearDuration
import math
doy = round(yearElapsed)
if doy < 1:
doy = 1
try:
date = datetime.date(year, 1, 1) + datetime.timedelta(days=doy-1)
except:
s = ""
if qDate:
return QDate(date.year, date.month, date.day)
if dt64:
return np.datetime64(date)
else:
return date
#return np.datetime64('{:04}-01-01'.format(year), 'D') + np.timedelta64(int(yearElapsed), 'D')
def saveTemporalProfiles(profiles, path, mode='all', sep=',', loadMissingValues=False):
if path is None or len(path) == 0:
return
assert mode in ['coordinate','all']
nbMax = 0
for sensor in profiles[0].timeSeries().sensors():
assert isinstance(sensor, SensorInstrument)
nbMax = max(nbMax, sensor.nb)
ext = os.path.splitext(path)[1].lower()
assert isinstance(ext, str)
if ext.startswith('.'):
ext = ext[1:]
if loadMissingValues:
for p in profiles:
assert isinstance(p, TemporalProfile)
p.loadMissingData()
#write a flat list of profiles
lines = ['Temporal Profiles']
lines.append(sep.join(['pid', 'name', 'date', 'img', 'location', 'band values']))
for nP, p in enumerate(profiles):
assert isinstance(p, TemporalProfile)
lines.append('Profile {} "{}": {}'.format(p.mID, p.name(), p.mCoordinate))
assert isinstance(p, TemporalProfile)
c = p.coordinate()
for i, tsd in enumerate(p.mTimeSeries):
assert isinstance(tsd, TimeSeriesDatum)
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
data = p.data(tsd)
line = [p.id(), p.name(), data['date'], tsd.pathImg, c.asWkt()]
for b in range(tsd.sensor.nb):
key = 'b{}'.format(b+1)
if key in data.keys():
line.append(data[key])
else:
line.append('')
line = sep.join([str(l) for l in line])
lines.append(line)
file = open(path, 'w', encoding='utf8')
file.writelines('\n'.join(lines))
file.flush()
file.close()
else:
drv = None
for i in range(ogr.GetDriverCount()):
d = ogr.GetDriver(i)
driverExtensions = d.GetMetadataItem('DMD_EXTENSIONS')
if driverExtensions is not None and ext in driverExtensions:
drv = d
break
if not isinstance(drv, ogr.Driver):
raise Exception('Unable to find a OGR driver to write {}'.format(path))
drvMEM = ogr.GetDriverByName('Memory')
assert isinstance(drvMEM, ogr.Driver)
dsMEM = drvMEM.CreateDataSource('')
srs = osr.SpatialReference()
srs.ImportFromEPSG(4326)
crs = QgsCoordinateReferenceSystem('EPSG:4326')
assert isinstance(dsMEM, ogr.DataSource)
#lines.append(sep.join(['pid', 'name', 'date', 'img', 'location', 'band values']))
lyr = dsMEM.CreateLayer(os.path.basename(path), srs, ogr.wkbPoint)
lyr.CreateField(ogr.FieldDefn('pid', ogr.OFTInteger64))
lyr.CreateField(ogr.FieldDefn('name', ogr.OFTString))
lyr.CreateField(ogr.FieldDefn('date', ogr.OFTDateTime))
lyr.CreateField(ogr.FieldDefn('doy', ogr.OFTInteger))
lyr.CreateField(ogr.FieldDefn('img', ogr.OFTString))
lyr.CreateField(ogr.FieldDefn('lat', ogr.OFTReal))
lyr.CreateField(ogr.FieldDefn('lon', ogr.OFTReal))
lyr.CreateField(ogr.FieldDefn('nodata', ogr.OFTBinary))
for b in range(nbMax):
lyr.CreateField(ogr.FieldDefn('b{}'.format(b+1), ogr.OFTReal))
for iP, p in enumerate(profiles):
assert isinstance(p, TemporalProfile)
c = p.coordinate().toCrs(crs)
assert isinstance(c, SpatialPoint)
for iT, tsd in enumerate(p.timeSeries()):
feature = ogr.Feature(lyr.GetLayerDefn())
assert isinstance(feature, ogr.Feature)
data = p.data(tsd)
assert isinstance(tsd, TimeSeriesDatum)
feature.SetField('pid', p.id())
feature.SetField('name', p.name())
feature.SetField('date', str(tsd.date))
feature.SetField('doy', tsd.doy)
feature.SetField('img', tsd.pathImg)
feature.SetField('lon', c.x())
feature.SetField('lat', c.y())
point = ogr.CreateGeometryFromWkt(c.asWkt())
feature.SetGeometry(point)
if 'nodata' in data.keys():
feature.SetField('nodata', data['nodata'])
for b in range(tsd.sensor.nb):
key = 'b{}'.format(b+1)
if key in data.keys():
feature.SetField(key, data[key])
lyr.CreateFeature(feature)
drv.CopyDataSource(dsMEM, path)
regBandKey = re.compile(r"(?<!\w)b\d+(?!\w)", re.IGNORECASE)
regBandKeyExact = re.compile(r'^' + regBandKey.pattern + '$', re.IGNORECASE)
def bandIndex2bandKey(i):
assert isinstance(i, int)
assert i >= 0
return 'b{}'.format(i + 1)
def bandKey2bandIndex(key):
match = regBandKeyExact.search(key)
assert match
idx = int(match.group()[1:]) - 1
return idx
class DateTimePlotWidget(pg.PlotWidget):
"""
Subclass of PlotWidget
"""
def __init__(self, parent=None):
"""
Constructor of the widget
"""
super(DateTimePlotWidget, self).__init__(parent)
self.plotItem = pg.PlotItem(
axisItems={'bottom':DateTimeAxis(orientation='bottom')}
,viewBox=DateTimeViewBox()
)
self.setCentralItem(self.plotItem)
#self.xAxisInitialized = False
pi = self.getPlotItem()
pi.getAxis('bottom').setLabel(LABEL_TIME)
pi.getAxis('left').setLabel(LABEL_EXPRESSION_2D)
self.mInfoColor = QColor('yellow')
self.mCrosshairLineV = pg.InfiniteLine(angle=90, movable=False)
self.mCrosshairLineH = pg.InfiniteLine(angle=0, movable=False)
self.mInfoLabelCursor = pg.TextItem(text='<cursor position>', anchor=(1.0, 0.0))
self.mInfoLabelCursor.setColor(QColor('yellow'))
self.scene().addItem(self.mInfoLabelCursor)
self.mInfoLabelCursor.setParentItem(self.getPlotItem())
#self.plot2DLabel.setAnchor()
#self.plot2DLabel.anchor(itemPos=(0, 0), parentPos=(0, 0), offset=(0, 0))
pi.addItem(self.mCrosshairLineV, ignoreBounds=True)
pi.addItem(self.mCrosshairLineH, ignoreBounds=True)
self.proxy2D = pg.SignalProxy(self.scene().sigMouseMoved, rateLimit=60, slot=self.onMouseMoved2D)
def onMouseMoved2D(self, evt):
pos = evt[0] ## using signal proxy turns original arguments into a tuple
plotItem = self.getPlotItem()
if plotItem.sceneBoundingRect().contains(pos):

benjamin.jakimow@geo.hu-berlin.de
committed
vb = plotItem.vb
assert isinstance(vb, DateTimeViewBox)
mousePoint = vb.mapSceneToView(pos)
x = mousePoint.x()
if x >= 0:
y = mousePoint.y()
date = num2date(x)
doy = dateDOY(date)
plotItem.vb.updateCurrentDate(num2date(x, dt64=True))
self.mInfoLabelCursor.setText('DN {:0.2f}\nDate {}\nDOY {}'.format(
mousePoint.y(), date, doy),
color=self.mInfoColor)
s = self.size()
pos = QPointF(s.width(), 0)

benjamin.jakimow@geo.hu-berlin.de
committed
self.mInfoLabelCursor.setVisible(vb.mActionShowCursorValues.isChecked())
self.mInfoLabelCursor.setPos(pos)

benjamin.jakimow@geo.hu-berlin.de
committed
b = vb.mActionShowCrosshair.isChecked()
self.mCrosshairLineH.setVisible(b)
self.mCrosshairLineV.setVisible(b)
self.mCrosshairLineH.pen.setColor(self.mInfoColor)
self.mCrosshairLineV.pen.setColor(self.mInfoColor)
self.mCrosshairLineV.setPos(mousePoint.x())
self.mCrosshairLineH.setPos(mousePoint.y())
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
class DateTimeAxis(pg.AxisItem):
def __init__(self, *args, **kwds):
super(DateTimeAxis, self).__init__(*args, **kwds)
self.setRange(1,3000)
self.enableAutoSIPrefix(False)
self.labelAngle = 0
def logTickStrings(self, values, scale, spacing):
s = ""
def tickStrings(self, values, scale, spacing):
strns = []
if len(values) == 0:
return []
#assert isinstance(values[0],
values = [num2date(v) if v > 0 else num2date(1) for v in values]
rng = max(values)-min(values)
ndays = rng.astype(int)
strns = []
for v in values:
if ndays == 0:
strns.append(v.astype(str))
else:
strns.append(v.astype(str))
return strns
def tickValues(self, minVal, maxVal, size):
d = super(DateTimeAxis, self).tickValues(minVal, maxVal, size)
return d
def drawPicture(self, p, axisSpec, tickSpecs, textSpecs):
p.setRenderHint(p.Antialiasing, False)
p.setRenderHint(p.TextAntialiasing, True)
## draw long line along axis
pen, p1, p2 = axisSpec
p.setPen(pen)
p.drawLine(p1, p2)
p.translate(0.5, 0) ## resolves some damn pixel ambiguity
## draw ticks
for pen, p1, p2 in tickSpecs:
p.setPen(pen)
p.drawLine(p1, p2)
## Draw all text
if self.tickFont is not None:
p.setFont(self.tickFont)
p.setPen(self.pen())
#for rect, flags, text in textSpecs:
# p.drawText(rect, flags, text)
# # p.drawRect(rect)
#see https://github.com/pyqtgraph/pyqtgraph/issues/322
for rect, flags, text in textSpecs:
p.save() # save the painter state
p.translate(rect.center()) # move coordinate system to center of text rect
p.rotate(self.labelAngle) # rotate text
p.translate(-rect.center()) # revert coordinate system
p.drawText(rect, flags, text)
p.restore() # restore the painter state
class DateTimeViewBox(pg.ViewBox):
"""
Subclass of ViewBox
"""
sigMoveToDate = pyqtSignal(np.datetime64)
def __init__(self, parent=None):
"""
Constructor of the CustomViewBox
"""
super(DateTimeViewBox, self).__init__(parent)
#self.menu = None # Override pyqtgraph ViewBoxMenu
#self.menu = self.getMenu() # Create the menu
#self.menu = None
self.mCurrentDate = np.datetime64('today')
self.mXAxisUnit = 'date'
xAction = [a for a in self.menu.actions() if a.text() == 'X Axis'][0]
yAction = [a for a in self.menu.actions() if a.text() == 'Y Axis'][0]
menuXAxis = self.menu.addMenu('X Axis')
#define the widget to set X-Axis options
frame = QFrame()
l = QGridLayout()
frame.setLayout(l)
#l.addWidget(self, QWidget, int, int, alignment: Qt.Alignment = 0): not enough arguments
self.rbXManualRange = QRadioButton('Manual')
self.dateEditX0 = QDateEdit()
self.dateEditX0.setDisplayFormat('yyyy-MM-dd')
self.dateEditX0.setToolTip('Start time')
self.dateEditX0.setCalendarPopup(True)
self.dateEditX0.dateChanged.connect(self.updateXRange)
self.dateEditX1 = QDateEdit()
self.dateEditX1.setDisplayFormat('yyyy-MM-dd')
self.dateEditX0.setToolTip('End time')
self.dateEditX1.setCalendarPopup(True)
self.dateEditX1.dateChanged.connect(self.updateXRange)
self.rbXAutoRange = QRadioButton('Auto')
self.rbXAutoRange.setChecked(True)
self.rbXAutoRange.toggled.connect(self.updateXRange)
l.addWidget(self.rbXManualRange, 0,0)
l.addWidget(self.dateEditX0, 0,1)
l.addWidget(self.dateEditX1, 0,2)
l.addWidget(self.rbXAutoRange, 1, 0)
l.setMargin(1)
l.setSpacing(1)
frame.setMinimumSize(l.sizeHint())
wa = QWidgetAction(menuXAxis)
wa.setDefaultWidget(frame)
menuXAxis.addAction(wa)
self.menu.insertMenu(xAction, menuXAxis)
self.menu.removeAction(xAction)
self.mActionMoveToDate = self.menu.addAction('Move to {}'.format(self.mCurrentDate))
self.mActionMoveToDate.triggered.connect(lambda : self.sigMoveToDate.emit(self.mCurrentDate))

benjamin.jakimow@geo.hu-berlin.de
committed
self.mActionShowCrosshair = self.menu.addAction('Show Crosshair')
self.mActionShowCrosshair.setCheckable(True)
self.mActionShowCrosshair.setChecked(True)
self.mActionShowCursorValues = self.menu.addAction('Show Mouse values')
self.mActionShowCursorValues.setCheckable(True)
self.mActionShowCursorValues.setChecked(True)
sigXAxisUnitChanged = pyqtSignal(str)
def setXAxisUnit(self, unit):
assert unit in ['date', 'doy']
old = self.mXAxisUnit
self.mXAxisUnit = unit
if old != self.mXAxisUnit:
self.sigXAxisUnitChanged.emit(self.mXAxisUnit)
def xAxisUnit(self):
return self.mXAxisUnit
def updateXRange(self, *args):
isAutoRange = self.rbXAutoRange.isChecked()
self.enableAutoRange('x', isAutoRange)
self.dateEditX0.setEnabled(not isAutoRange)
self.dateEditX1.setEnabled(not isAutoRange)
if not isAutoRange:
t0 = date2num(self.dateEditX0.date())
t1 = date2num(self.dateEditX1.date())
t0 = min(t0, t1)
t1 = max(t0, t1)
self.setXRange(t0, t1)
def updateCurrentDate(self, date):
if isinstance(date, np.datetime64):
self.mCurrentDate = date
self.mActionMoveToDate.setData(date)
self.mActionMoveToDate.setText('Move maps to {}'.format(date))
def raiseContextMenu(self, ev):
pt = self.mapDeviceToView(ev.pos())
self.updateCurrentDate(num2date(pt.x(), dt64=True))
xRange, yRange = self.viewRange()
t0 = num2date(xRange[0], qDate=True)
t1 = num2date(xRange[1], qDate=True)
self.dateEditX0.setDate(t0)
self.dateEditX1.setDate(t1)
menu = self.getMenu(ev)
self.scene().addParentContextMenus(self, menu, ev)
menu.exec_(ev.screenPos().toPoint())
class TemporalProfilePlotStyleBase(PlotStyle):
sigExpressionUpdated = pyqtSignal()
sigSensorChanged = pyqtSignal(SensorInstrument)
def __init__(self, parent=None, temporalProfile=None):
super(TemporalProfilePlotStyleBase, self).__init__()
self.mSensor = None
self.mTP = None
self.mExpression = 'b1'
self.mPlotItems = []
if isinstance(temporalProfile, TemporalProfile):
self.setTemporalProfile(temporalProfile)
def isPlotable(self):
return self.isVisible() and isinstance(self.temporalProfile(), TemporalProfile) and isinstance(self.sensor(), SensorInstrument)
def createPlotItem(self):
raise NotImplementedError()
def temporalProfile(self):
return self.mTP
def setTemporalProfile(self, temporalPofile):
b = temporalPofile != self.mTP
self.mTP = temporalPofile
if temporalPofile is None:
s =""
else:
assert isinstance(temporalPofile, TemporalProfile)
if b:

Benjamin Jakimow
committed
self.updateDataProperties()
def setSensor(self, sensor):
assert sensor is None or isinstance(sensor, SensorInstrument)
b = sensor != self.mSensor
self.mSensor = sensor
if b:
self.update()
self.sigSensorChanged.emit(sensor)
def sensor(self):
return self.mSensor

Benjamin Jakimow
committed
def updateStyleProperties(self):
raise NotImplementedError()

Benjamin Jakimow
committed
def updateDataProperties(self):
raise NotImplementedError()

Benjamin Jakimow
committed
def update(self):
self.updateDataProperties()
def setExpression(self, exp):
b = self.mExpression != exp
self.mExpression = exp

Benjamin Jakimow
committed
self.updateDataProperties()
if b:

Benjamin Jakimow
committed
self
self.sigExpressionUpdated.emit()
def expression(self):
return self.mExpression
def __reduce_ex__(self, protocol):
return self.__class__, (), self.__getstate__()
def __getstate__(self):
result = super(TemporalProfile2DPlotStyle, self).__getstate__()
#remove
del result['mTP']
del result['mSensor']
return result
def isVisible(self):
return self.mIsVisible
def setVisibility(self, b):
assert isinstance(b, bool)
old = self.isVisible()
self.mIsVisible = b
if b != old:

Benjamin Jakimow
committed
self.updateStyleProperties()
#self.update()
def copyFrom(self, plotStyle):

Benjamin Jakimow
committed
if isinstance(plotStyle, PlotStyle):
super(TemporalProfilePlotStyleBase, self).copyFrom(plotStyle)

Benjamin Jakimow
committed
self.updateStyleProperties()
if isinstance(plotStyle, TemporalProfilePlotStyleBase):
self.setExpression(plotStyle.expression())
self.setSensor(plotStyle.sensor())
self.setTemporalProfile(plotStyle.temporalProfile())

Benjamin Jakimow
committed
self.updateDataProperties()
class TemporalProfile2DPlotStyle(TemporalProfilePlotStyleBase):
def __init__(self, temporalProfile=None):
super(TemporalProfile2DPlotStyle, self).__init__(temporalProfile=temporalProfile)
#PlotStyle.__init__(self)
#TemporalProfilePlotStyleBase.__init__(self, temporalProfile=temporalProfile)
def createPlotItem(self, plotWidget):
pdi = TemporalProfilePlotDataItem(self)
self.mPlotItems.append(pdi)
return pdi

Benjamin Jakimow
committed
def updateStyleProperties(self):
for pdi in self.mPlotItems:
assert isinstance(pdi, TemporalProfilePlotDataItem)
pdi.updateStyle()

Benjamin Jakimow
committed
def updateDataProperties(self):
for pdi in self.mPlotItems:
assert isinstance(pdi, TemporalProfilePlotDataItem)
pdi.updateDataAndStyle()
class TemporalProfile(QObject):
_mNextID = 0
@staticmethod
def nextID():
n = TemporalProfile._mNextID
TemporalProfile._mNextID += 1
return n
sigNameChanged = pyqtSignal(str)
sigDataChanged = pyqtSignal()
def __init__(self, timeSeries, spatialPoint):
super(TemporalProfile, self).__init__()
assert isinstance(timeSeries, TimeSeries)
assert isinstance(spatialPoint, SpatialPoint)
self.mTimeSeries = timeSeries
self.mCoordinate = spatialPoint
self.mID = TemporalProfile.nextID()
self.mData = {}
self.mUpdated = False
self.mLoaded = self.mLoadedMax = self.mNoData = 0
self.initMetadata()
self.updateLoadingStatus()
def __eq__(self, other):
"""
Two temporal profiles are equal if they point to the same geometry
:param other:
:return:
"""
if not isinstance(other, TemporalProfile):
return False
return other.mCoordinate == self.mCoordinate
def coordinate(self):
return self.mCoordinate
def id(self):
return self.mID
def data(self):
return self.mData
def timeSeries(self):
return self.mTimeSeries
def initMetadata(self):
for tsd in self.mTimeSeries:
assert isinstance(tsd, TimeSeriesDatum)
meta = {'doy':tsd.doy,
'date':str(tsd.date),
'nodata':False}
self.updateData(tsd, meta)
def pullDataUpdate(self, d):
assert isinstance(d, PixelLoaderTask)
if d.success() and self.mID in d.temporalProfileIDs:
i = d.temporalProfileIDs.index(self.mID)
tsd = self.mTimeSeries.getTSD(d.sourcePath)
assert isinstance(tsd, TimeSeriesDatum)
if d.validPixelValues(i):
profileData = d.resProfiles[i]
vMean, vStd = profileData
validValues = not isinstance(vMean, str)
# 1. add the pixel values per returned band
for iBand, bandIndex in enumerate(d.bandIndices):
key = 'b{}'.format(bandIndex + 1)
values[key] = vMean[iBand] if validValues else None
key = 'std{}'.format(bandIndex + 1)
values[key] = vStd[iBand] if validValues else None
else:
values['nodata'] = True
self.updateData(tsd, values)
def loadMissingData(self, showGUI=False):
"""
Loads the missing data
:return:
"""
from timeseriesviewer.pixelloader import PixelLoaderTask, doLoaderTask
tasks = []
for tsd in self.mTimeSeries:
missingIndices = self.missingBandIndices(tsd)
if len(missingIndices) > 0:
task = PixelLoaderTask(tsd.pathImg, [self.mCoordinate],
bandIndices=missingIndices,
temporalProfileIDs=[self.mID])
tasks.append(task)
for task in tasks:
result = doLoaderTask(task)
assert isinstance(result, PixelLoaderTask)
self.pullDataUpdate(result)
def missingBandIndices(self, tsd, requiredIndices=None):
"""
Returns the band indices [0, sensor.nb) that have not been loaded yet.
:param tsd: TimeSeriesDatum of interest
:param requiredIndices: optional subset of possible band-indices to return the missing ones from.
:return: [list-of-indices]
"""
assert isinstance(tsd, TimeSeriesDatum)
if requiredIndices is None:
requiredIndices = list(range(tsd.sensor.nb))
requiredIndices = [i for i in requiredIndices if i >= 0 and i < tsd.sensor.nb]
existingBandIndices = [bandKey2bandIndex(k) for k in self.data(tsd).keys() if regBandKeyExact.search(k)]
return [i for i in requiredIndices if i not in existingBandIndices]
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
def setName(self, name):
if name != self.mName:
self.mName = name
self.sigNameChanged.emit(self.mName)
def name(self):
return self.mName
def plot(self):
import pyqtgraph as pg
for sensor in self.mTimeSeries.sensors():
assert isinstance(sensor, SensorInstrument)
plotStyle = TemporalProfile2DPlotStyle(self)
plotStyle.setSensor(sensor)
pi = TemporalProfilePlotDataItem(plotStyle)
pi.setClickable(True)
pw = pg.plot(title=self.name())
pw.getPlotItem().addItem(pi)
pi.setColor('green')
pg.QAPP.exec_()
def updateData(self, tsd, values):
assert isinstance(tsd, TimeSeriesDatum)
assert isinstance(values, dict)
if tsd not in self.mData.keys():
self.mData[tsd] = {}
self.mData[tsd].update(values)
self.updateLoadingStatus()
self.mUpdated = True
def resetUpdatedFlag(self):
self.mUpdated = False
def updated(self):
return self.mUpdated
def dataFromExpression(self, sensor, expression, dateType='date'):
assert dateType in ['date','doy']
x = []
y = []
if not isinstance(expression, QgsExpression):
expression = QgsExpression(expression)
assert isinstance(expression, QgsExpression)
expression = QgsExpression(expression)
fields = QgsFields()
sensorTSDs = sorted([tsd for tsd in self.mData.keys() if tsd.sensor == sensor])
for tsd in sensorTSDs:
data = self.mData[tsd]
for k, v in data.items():
if v is not None and fields.indexFromName(k) == -1:
fields.append(qgsFieldFromKeyValue(k, v))
for i, tsd in enumerate(sensorTSDs):
assert isinstance(tsd, TimeSeriesDatum)
data = self.mData[tsd]
context = QgsExpressionContext()
scope = QgsExpressionContextScope()
f = QgsFeature()
f.setFields(fields)
f.setValid(True)
for k, v in data.items():
if v is None:
continue
idx = f.fields().indexFromName(k)
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
field = f.fields().field(idx)
if field.typeName() == 'text':
v = str(v)
else:
v = float(v)
f.setAttribute(k,v)
scope.setFeature(f)
context.appendScope(scope)
#value = expression.evaluatePrepared(f)
value = expression.evaluate(context)
if value in [None]:
s = ""
else:
if dateType == 'date':
x.append(date2num(tsd.date))
elif dateType == 'doy':
x.append(tsd.doy)
y.append(value)
#return np.asarray(x), np.asarray(y)
assert len(x) == len(y)
return x, y
def data(self, tsd):
assert isinstance(tsd, TimeSeriesDatum)
if self.hasData(tsd):
return self.mData[tsd]
else:
return {}
def loadingStatus(self):
"""
Returns the loading status in terms of single pixel values.
nLoaded = sum of single band values
nLoadedMax = potential maximum of band values that might be loaded
:return: (nLoaded, nLoadedMax)
"""
return self.mLoaded, self.mNoData, self.mLoadedMax
def updateLoadingStatus(self):
"""
Calculates the loading status in terms of single pixel values.
nMax is the sum of all bands over each TimeSeriesDatum and Sensors
"""
for tsd in self.mTimeSeries:
assert isinstance(tsd, TimeSeriesDatum)
nb = tsd.sensor.nb
self.mLoadedMax += nb
if self.hasData(tsd):
if self.isNoData(tsd):
self.mNoData += nb
else:
self.mLoaded += len([k for k in self.mData[tsd].keys() if regBandKey.search(k)])
def isNoData(self, tsd):
assert isinstance(tsd, TimeSeriesDatum)
return self.mData[tsd]['nodata']
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
assert isinstance(tsd, TimeSeriesDatum)
return tsd in self.mData.keys()
def __repr__(self):
return 'TemporalProfile {}'.format(self.mCoordinate)
class TemporalProfilePlotDataItem(pg.PlotDataItem):
def __init__(self, plotStyle, parent=None):
assert isinstance(plotStyle, TemporalProfile2DPlotStyle)
super(TemporalProfilePlotDataItem, self).__init__([], [], parent=parent)
self.menu = None
#self.setFlags(QGraphicsItem.ItemIsSelectable)
self.mPlotStyle = plotStyle
self.setAcceptedMouseButtons(Qt.LeftButton | Qt.RightButton)
self.mPlotStyle.sigUpdated.connect(self.updateDataAndStyle)
self.updateStyle()
# On right-click, raise the context menu
def mouseClickEvent(self, ev):
if ev.button() == QtCore.Qt.RightButton:
if self.raiseContextMenu(ev):
ev.accept()
def raiseContextMenu(self, ev):
menu = self.getContextMenus()
# Let the scene add on to the end of our context menu
# (this is optional)