def setTimeRestriction(self, timePosition, timeFrame):
TimeVectorLayer.setTimeRestriction(self, timePosition, timeFrame)
start_epoch = time_util.datetime_to_epoch(
self.getStartTime(timePosition, timeFrame))
end_epoch = time_util.datetime_to_epoch(
self.getEndTime(timePosition, timeFrame))
# info("setTimeRestriction Called {} times".format(self.n))
# info("size of layer at {}:{}".format(start_epoch,self.memLayer.featureCount(),))
geoms = self.getInterpolatedGeometries(start_epoch, end_epoch)
# Add the geometries as features
self._clearMemoryLayer()
features = []
for i, geom in enumerate(geoms):
feature = QgsFeature(id=start_epoch + i)
feature.setGeometry(QgsGeometry.fromQPointF(geom.toQPointF()))
# feature.setAttributes([start_epoch+i])
features.append(feature) # if no attributes, it will fail
self.n = self.n + 1
# info("add {}features:".format(len(features)))
res = self.memLayer.dataProvider().addFeatures(features)
assert (res)
self.memLayer.triggerRepaint()
def refreshGuiWithCurrentTime(self, currentTimePosition, sender=None):
"""
Update the gui when time has changed by refreshing/repainting the layers
and changing the time showing in dateTimeEditCurrentTime and horizontalTimeSlider
Setting the gui elements should not fire the event for
timeChanged, since they were changed to be in sync with the rest of the system on
purpose, no need to sync the system again
"""
self.setPropagateGuiChanges(False)
timeExtent = self.getTimeLayerManager().getProjectTimeExtents()
if currentTimePosition is None or timeExtent[0] is None or timeExtent[1] is None:
self.setPropagateGuiChanges(True)
return
time_util.updateUi(self.guiControl.getTimeWidget(), currentTimePosition)
timeval = time_util.datetime_to_epoch(currentTimePosition)
timeExtents = self.getTimeLayerManager().getProjectTimeExtents()
try:
pct = (timeval - time_util.datetime_to_epoch(timeExtents[0])) * 1.0 / (time_util.datetime_to_epoch(
timeExtents[1]) - time_util.datetime_to_epoch(timeExtents[0]))
sliderVal = self.guiControl.dock.horizontalTimeSlider.minimum() + int(pct * (
self.guiControl.dock.horizontalTimeSlider.maximum() -
self.guiControl.dock.horizontalTimeSlider.minimum()))
self.guiControl.dock.horizontalTimeSlider.setValue(sliderVal)
self.guiControl.repaintRasters()
self.guiControl.repaintJoined()
self.guiControl.repaintVectors()
self.guiControl.refreshMapCanvas()
self.updateLegendCount()
except Exception as e:
error(e)
finally:
self.setPropagateGuiChanges(True)
def refreshGuiTimeExtents(self, timeExtents):
"""Update time extents showing in labels and represented by horizontalTimeSlider
:param timeExtents: a tuple of start and end datetimes
"""
self.setPropagateGuiChanges(False)
if timeExtents[1] is not None: # timeExtents[0] is set in different places, so only check timeExtents[1]
startText = time_util.datetime_to_str(timeExtents[0], time_util.DEFAULT_FORMAT)
endText = time_util.datetime_to_str(timeExtents[1], time_util.DEFAULT_FORMAT)
self.guiControl.dock.labelStartTime.setText(startText)
self.guiControl.dock.labelEndTime.setText(endText)
timeLength = time_util.datetime_to_epoch(timeExtents[1]) - time_util.datetime_to_epoch(timeExtents[0])
if timeLength > MAX_TIME_LENGTH_SECONDS_SLIDER:
new_granularity = int(math.ceil(1.0 * timeLength / MAX_TIME_LENGTH_SECONDS_SLIDER))
self.setGranularitySeconds(new_granularity)
# trick because timeLength must fit in an integer
# since it interfaces with a C++ class
newTimeLength = int(math.ceil(1.0 * timeLength / new_granularity))
timeLength = newTimeLength
else:
self.setGranularitySeconds(conf.DEFAULT_GRANULARITY_IN_SECONDS)
self.guiControl.dock.horizontalTimeSlider.setMinimum(0)
self.guiControl.dock.horizontalTimeSlider.setMaximum(timeLength)
else: # set to default values
self.setGranularitySeconds(conf.DEFAULT_GRANULARITY_IN_SECONDS)
self.guiControl.dock.labelStartTime.setText('not set')
self.guiControl.dock.labelEndTime.setText('not set')
self.guiControl.dock.horizontalTimeSlider.setMinimum(conf.MIN_TIMESLIDER_DEFAULT)
self.guiControl.dock.horizontalTimeSlider.setMaximum(conf.MAX_TIMESLIDER_DEFAULT)
self.setPropagateGuiChanges(True)
def updateTimePositionFromSliderPct(self, pct):
"""See the percentage the slider is at and determine the datetime"""
if not self.propagateGuiChanges:
return
timeExtents = self.getTimeLayerManager().getProjectTimeExtents()
try:
realEpochTime = int(pct * (time_util.datetime_to_epoch(timeExtents[1]) - time_util.datetime_to_epoch(
timeExtents[0])) + time_util.datetime_to_epoch(timeExtents[0]))
except Exception:
# extents are not set yet?
return
self.getTimeLayerManager().setCurrentTimePosition(time_util.epoch_to_datetime(realEpochTime))
def updateTimePositionFromSliderPct(self, pct):
"""See the percentage the slider is at and determine the datetime"""
if not self.propagateGuiChanges:
return
timeExtents = self.getTimeLayerManager().getProjectTimeExtents()
try:
realEpochTime = int(pct *
(time_util.datetime_to_epoch(timeExtents[1]) -
time_util.datetime_to_epoch(timeExtents[0])) +
time_util.datetime_to_epoch(timeExtents[0]))
except Exception:
# extents are not set yet?
return
self.getTimeLayerManager().setCurrentTimePosition(
time_util.epoch_to_datetime(realEpochTime))
def refreshGuiTimeExtents(self, timeExtents):
"""Update time extents showing in labels and represented by horizontalTimeSlider
:param timeExtents: a tuple of start and end datetimes
"""
self.setPropagateGuiChanges(False)
if timeExtents[
1] is not None: # timeExtents[0] is set in different places, so only check timeExtents[1]
startText = time_util.datetime_to_str(timeExtents[0],
time_util.DEFAULT_FORMAT)
endText = time_util.datetime_to_str(timeExtents[1],
time_util.DEFAULT_FORMAT)
self.guiControl.dock.labelStartTime.setText(startText)
self.guiControl.dock.labelEndTime.setText(endText)
timeLength = time_util.datetime_to_epoch(
timeExtents[1]) - time_util.datetime_to_epoch(timeExtents[0])
if timeLength > MAX_TIME_LENGTH_SECONDS_SLIDER:
new_granularity = int(
math.ceil(1.0 * timeLength /
MAX_TIME_LENGTH_SECONDS_SLIDER))
self.setGranularitySeconds(new_granularity)
# trick because timeLength must fit in an integer
# since it interfaces with a C++ class
newTimeLength = int(
math.ceil(1.0 * timeLength / new_granularity))
timeLength = newTimeLength
else:
self.setGranularitySeconds(conf.DEFAULT_GRANULARITY_IN_SECONDS)
self.guiControl.dock.horizontalTimeSlider.setMinimum(0)
self.guiControl.dock.horizontalTimeSlider.setMaximum(timeLength)
else: # set to default values
self.setGranularitySeconds(conf.DEFAULT_GRANULARITY_IN_SECONDS)
self.guiControl.dock.labelStartTime.setText('not set')
self.guiControl.dock.labelEndTime.setText('not set')
self.guiControl.dock.horizontalTimeSlider.setMinimum(
conf.MIN_TIMESLIDER_DEFAULT)
self.guiControl.dock.horizontalTimeSlider.setMaximum(
conf.MAX_TIMESLIDER_DEFAULT)
self.setPropagateGuiChanges(True)
def build_query(start_dt, end_dt, from_attr, to_attr, date_type, date_format, query_idiom, acc):
"""Build subset query"""
if acc: # features never die
start_dt = time_util.get_min_dt()
comparison = "<" # simplified because of: https://github.com/anitagraser/TimeManager/issues/235
# (original: # comparison = "<" if to_attr == from_attr else "<=")
if date_type == time_util.DateTypes.IntegerTimestamps:
start_epoch = time_util.datetime_to_epoch(start_dt)
end_epoch = time_util.datetime_to_epoch(end_dt)
return INT_FORMAT.format(from_attr, comparison, end_epoch, to_attr, start_epoch)
start_str = time_util.datetime_to_str(start_dt, date_format)
end_str = time_util.datetime_to_str(end_dt, date_format)
if date_type == time_util.DateTypes.DatesAsStringsArchaelogical:
# kept <= option here since I'm not sure about implications in archaelogical mode
comparison = "<" if to_attr == from_attr else "<="
return build_query_archaelogical(start_str, end_str, from_attr, to_attr, comparison, query_idiom)
if can_compare_lexicographically(date_format):
if query_idiom == QueryIdioms.OGR:
return STRINGCAST_FORMAT.format(from_attr, comparison, end_str, to_attr, start_str)
else:
return STRING_FORMAT.format(from_attr, comparison, end_str, to_attr, start_str)
else:
# thankfully, SQL & OGR syntax agree on substr and concat
if date_type != time_util.DateTypes.DatesAsStrings:
raise QueryBuildingException()
ioy = date_format.find("%Y")
iom = date_format.find("%m")
iod = date_format.find("%d")
ioh = date_format.find("%H")
sub1 = create_ymd_substring(ioy, iom, iod, ioh, from_attr, quote_type='"') # quote type for column names
sub2 = create_ymd_substring(ioy, iom, iod, ioh, end_str, quote_type='\'') # quote type for values
sub3 = create_ymd_substring(ioy, iom, iod, ioh, to_attr, quote_type='"')
sub4 = create_ymd_substring(ioy, iom, iod, ioh, start_str, quote_type='\'')
query = "CONCAT({}) {} CONCAT({}) AND CONCAT({})>=CONCAT({})".format(sub1, comparison, sub2, sub3, sub4)
return query
def refreshGuiWithCurrentTime(self, currentTimePosition, sender=None):
"""
Update the gui when time has changed by refreshing/repainting the layers
and changing the time showing in dateTimeEditCurrentTime and horizontalTimeSlider
Setting the gui elements should not fire the event for
timeChanged, since they were changed to be in sync with the rest of the system on
purpose, no need to sync the system again
"""
self.setPropagateGuiChanges(False)
timeExtent = self.getTimeLayerManager().getProjectTimeExtents()
if currentTimePosition is None or timeExtent[0] is None or timeExtent[
1] is None:
self.setPropagateGuiChanges(True)
return
time_util.updateUi(self.guiControl.getTimeWidget(),
currentTimePosition)
timeval = time_util.datetime_to_epoch(currentTimePosition)
timeExtents = self.getTimeLayerManager().getProjectTimeExtents()
try:
pct = (timeval - time_util.datetime_to_epoch(timeExtents[0])
) * 1.0 / (time_util.datetime_to_epoch(timeExtents[1]) -
time_util.datetime_to_epoch(timeExtents[0]))
sliderVal = self.guiControl.dock.horizontalTimeSlider.minimum(
) + int(pct *
(self.guiControl.dock.horizontalTimeSlider.maximum() -
self.guiControl.dock.horizontalTimeSlider.minimum()))
self.guiControl.dock.horizontalTimeSlider.setValue(sliderVal)
self.guiControl.repaintRasters()
self.guiControl.repaintJoined()
self.guiControl.repaintVectors()
self.guiControl.refreshMapCanvas()
self.updateLegendCount()
except Exception as e:
error(e)
finally:
self.setPropagateGuiChanges(True)
def build_query(start_dt, end_dt, from_attr, to_attr, date_type, date_format,
query_idiom, acc):
"""Build subset query"""
if acc: # features never die
start_dt = time_util.get_min_dt()
comparison = "<" # simplified because of: https://github.com/anitagraser/TimeManager/issues/235
# (original: # comparison = "<" if to_attr == from_attr else "<=")
if date_type == time_util.DateTypes.IntegerTimestamps:
start_epoch = time_util.datetime_to_epoch(start_dt)
end_epoch = time_util.datetime_to_epoch(end_dt)
return INT_FORMAT.format(from_attr, comparison, end_epoch, to_attr,
start_epoch)
start_str = time_util.datetime_to_str(start_dt, date_format)
end_str = time_util.datetime_to_str(end_dt, date_format)
if date_type == time_util.DateTypes.DatesAsStringsArchaelogical:
# kept <= option here since I'm not sure about implications in archaelogical mode
comparison = "<" if to_attr == from_attr else "<="
return build_query_archaelogical(start_str, end_str, from_attr,
to_attr, comparison, query_idiom)
if can_compare_lexicographically(date_format):
if query_idiom == QueryIdioms.OGR:
return STRINGCAST_FORMAT.format(from_attr, comparison, end_str,
to_attr, start_str)
else:
return STRING_FORMAT.format(from_attr, comparison, end_str,
to_attr, start_str)
else:
# thankfully, SQL & OGR syntax agree on substr and concat
if date_type != time_util.DateTypes.DatesAsStrings:
raise QueryBuildingException()
ioy = date_format.find("%Y")
iom = date_format.find("%m")
iod = date_format.find("%d")
ioh = date_format.find("%H")
sub1 = create_ymd_substring(
ioy, iom, iod, ioh, from_attr,
quote_type='"') # quote type for column names
sub2 = create_ymd_substring(ioy,
iom,
iod,
ioh,
end_str,
quote_type='\'') # quote type for values
sub3 = create_ymd_substring(ioy,
iom,
iod,
ioh,
to_attr,
quote_type='"')
sub4 = create_ymd_substring(ioy,
iom,
iod,
ioh,
start_str,
quote_type='\'')
query = "CONCAT({}) {} CONCAT({}) AND CONCAT({})>=CONCAT({})".format(
sub1, comparison, sub2, sub3, sub4)
return query