From 1b42a037fbd6e18bd0ae61c77f018a6afbc99bab Mon Sep 17 00:00:00 2001 From: "benjamin.jakimow@geo.hu-berlin.de" <q8DTkxUg-BB> Date: Tue, 31 Jan 2017 17:12:19 +0100 Subject: [PATCH] added sensorOfInterest keyword to TimeSeries::getTSDs() --- timeseriesviewer/timeseries.py | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) diff --git a/timeseriesviewer/timeseries.py b/timeseriesviewer/timeseries.py index 6083f15a..a1295968 100644 --- a/timeseriesviewer/timeseries.py +++ b/timeseriesviewer/timeseries.py @@ -268,6 +268,7 @@ class TimeSeriesDatum(QObject): self.pathImg = pathImg self.pathMsk = None + assert os.path.exists(pathImg) self.lyrImg = QgsRasterLayer(pathImg, os.path.basename(pathImg), False) assert self.lyrImg.isValid() self.uriImg = self.lyrImg.dataProvider().dataSourceUri() @@ -275,7 +276,7 @@ class TimeSeriesDatum(QObject): self.crs = self.lyrImg.dataProvider().crs() self.sensor = SensorInstrument(self.lyrImg) - self.date = getImageDate2(self.lyrImg) + self.date = getImageDate(self.lyrImg) assert self.date is not None, 'Unable to find acquisition date of {}'.format(pathImg) self.ns = self.lyrImg.width() @@ -413,8 +414,6 @@ class TimeSeries(QObject): with open(path, 'w') as f: f.writelines(lines) - - def getMaxSpatialExtent(self, crs=None): if len(self.data) == 0: return None @@ -429,11 +428,13 @@ class TimeSeries(QObject): def getObservationDates(self): return [tsd.getDate() for tsd in self.data] - def getTSDs(self, date_of_interest=None): - if date_of_interest: - tsds = [tsd for tsd in self.data if tsd.getDate() == date_of_interest] + def getTSDs(self, dateOfInterest=None, sensorOfInterest=None): + if dateOfInterest: + tsds = [tsd for tsd in self.data if tsd.getDate() == dateOfInterest] else: tsds = self.data + if sensorOfInterest: + tsds = [tsd for tsd in tsds if tsd.sensor == sensorOfInterest] return tsds def clear(self): @@ -532,7 +533,7 @@ class TimeSeries(QObject): regAcqDate = re.compile(r'acquisition[ ]*(time|date|day)', re.I) regLandsatSceneID = re.compile(r"L[EMCT][1234578]{1}[12]\d{12}[a-zA-Z]{3}\d{2}") -def getImageDate2(lyr): +def getImageDate(lyr): assert isinstance(lyr, QgsRasterLayer) mdLines = str(lyr.metadata()).splitlines() date = None -- GitLab