def test_copy_vector_layer(self):
"""Test we can copy a vector layer."""
layer = load_test_vector_layer('exposure', 'buildings.shp')
new_layer = create_memory_layer('New layer', layer.geometryType(),
layer.crs(), layer.fields())
new_layer.keywords = layer.keywords
copy_layer(layer, new_layer)
self.assertEqual(layer.featureCount(), new_layer.featureCount())
self.assertEqual(len(layer.fields().toList()),
len(new_layer.fields().toList()))
expected = len(new_layer.fields().toList()) + 1
new_fields = {'STRUCTURE': 'my_new_field'}
copy_fields(new_layer, new_fields)
self.assertEqual(len(new_layer.fields().toList()), expected)
self.assertGreater(new_layer.fieldNameIndex('my_new_field'), -1)
remove_fields(new_layer, ['STRUCTURE', 'OSM_TYPE'])
self.assertEqual(len(new_layer.fields().toList()), expected - 2)
self.assertEqual(new_layer.fieldNameIndex('STRUCTURE'), -1)
self.assertEqual(new_layer.fieldNameIndex('OSM_TYPE'), -1)
_add_id_column(new_layer)
field_name = exposure_id_field['field_name']
self.assertGreater(new_layer.fieldNameIndex(field_name), -1)
def test_copy_vector_layer(self):
"""Test we can copy a vector layer."""
layer = load_test_vector_layer('exposure', 'buildings.shp')
new_layer = create_memory_layer(
'New layer', layer.geometryType(), layer.crs(), layer.fields())
new_layer.keywords = layer.keywords
copy_layer(layer, new_layer)
self.assertEqual(layer.featureCount(), new_layer.featureCount())
self.assertEqual(
len(layer.fields().toList()), len(new_layer.fields().toList()))
expected = len(new_layer.fields().toList()) + 1
new_fields = {'STRUCTURE': 'my_new_field'}
copy_fields(new_layer, new_fields)
self.assertEqual(
len(new_layer.fields().toList()), expected)
self.assertGreater(new_layer.fieldNameIndex('my_new_field'), -1)
remove_fields(new_layer, ['STRUCTURE', 'OSM_TYPE'])
self.assertEqual(
len(new_layer.fields().toList()), expected - 2)
self.assertEqual(new_layer.fieldNameIndex('STRUCTURE'), -1)
self.assertEqual(new_layer.fieldNameIndex('OSM_TYPE'), -1)
_add_id_column(new_layer)
field_name = exposure_id_field['field_name']
self.assertGreater(new_layer.fieldNameIndex(field_name), -1)
def zonal_stats(raster, vector, callback=None):
"""Reclassify a continuous raster layer.
Issue https://github.com/inasafe/inasafe/issues/3190
:param raster: The raster layer.
:type raster: QgsRasterLayer
:param vector: The vector layer.
:type vector: QgsVectorLayer
:param callback: A function to all to indicate progress. The function
should accept params 'current' (int), 'maximum' (int) and 'step' (str).
Defaults to None.
:type callback: function
:return: The output of the zonal stats.
:rtype: QgsVectorLayer
.. versionadded:: 4.0
"""
output_layer_name = zonal_stats_steps['output_layer_name']
processing_step = zonal_stats_steps['step_name']
layer = create_memory_layer(
output_layer_name,
vector.geometryType(),
vector.crs(),
vector.fields()
)
copy_layer(vector, layer)
input_band = layer.keywords.get('active_band', 1)
analysis = QgsZonalStatistics(
layer,
raster.source(),
'exposure_',
input_band,
QgsZonalStatistics.Sum)
result = analysis.calculateStatistics(None)
LOGGER.debug(tr('Zonal stats on %s : %s' % (raster.source(), result)))
layer.startEditing()
exposure = raster.keywords['exposure']
output_field = exposure_count_field['field_name'] % exposure
fields_to_rename = {
'exposure_sum': output_field
}
copy_fields(layer, fields_to_rename)
remove_fields(layer, fields_to_rename.keys())
layer.commitChanges()
# The zonal stats is producing some None values. We need to fill these
# with 0. See issue : #3778
# We should start a new editing session as previous fields need to be
# commited first.
layer.startEditing()
request = QgsFeatureRequest()
expression = '\"%s\" is None' % output_field
request.setFilterExpression(expression)
request.setFlags(QgsFeatureRequest.NoGeometry)
index = layer.fieldNameIndex(output_field)
for feature in layer.getFeatures():
if feature[output_field] is None:
layer.changeAttributeValue(feature.id(), index, 0)
layer.commitChanges()
layer.keywords = raster.keywords.copy()
layer.keywords['inasafe_fields'] = vector.keywords['inasafe_fields'].copy()
layer.keywords['inasafe_default_values'] = (
raster.keywords['inasafe_default_values'].copy())
key = exposure_count_field['key'] % raster.keywords['exposure']
# Special case here, one field is the exposure count and the total.
layer.keywords['inasafe_fields'][key] = output_field
layer.keywords['inasafe_fields'][total_field['key']] = output_field
layer.keywords['exposure_keywords'] = raster.keywords.copy()
layer.keywords['hazard_keywords'] = vector.keywords[
'hazard_keywords'].copy()
layer.keywords['aggregation_keywords'] = (
vector.keywords['aggregation_keywords'])
layer.keywords['layer_purpose'] = (
layer_purpose_aggregate_hazard_impacted['key'])
layer.keywords['title'] = output_layer_name
check_layer(layer)
return layer
def zonal_stats(raster, vector, callback=None):
"""Reclassify a continuous raster layer.
Issue https://github.com/inasafe/inasafe/issues/3190
:param raster: The raster layer.
:type raster: QgsRasterLayer
:param vector: The vector layer.
:type vector: QgsVectorLayer
:param callback: A function to all to indicate progress. The function
should accept params 'current' (int), 'maximum' (int) and 'step' (str).
Defaults to None.
:type callback: function
:return: The output of the zonal stats.
:rtype: QgsVectorLayer
.. versionadded:: 4.0
"""
output_layer_name = zonal_stats_steps['output_layer_name']
processing_step = zonal_stats_steps['step_name']
layer = create_memory_layer(
output_layer_name,
vector.geometryType(),
vector.crs(),
vector.fields()
)
copy_layer(vector, layer)
analysis = QgsZonalStatistics(
layer, raster.source(), 'exposure_', 1, QgsZonalStatistics.Sum)
result = analysis.calculateStatistics(None)
LOGGER.debug(tr('Zonal stats on %s : %s' % (raster.source(), result)))
layer.startEditing()
exposure = raster.keywords['exposure']
output_field = exposure_count_field['field_name'] % exposure
fields_to_rename = {
'exposure_sum': output_field
}
copy_fields(layer, fields_to_rename)
remove_fields(layer, fields_to_rename.keys())
layer.commitChanges()
# The zonal stats is producing some None values. We need to fill these
# with 0. See issue : #3778
# We should start a new editing session as previous fields need to be
# commited first.
layer.startEditing()
request = QgsFeatureRequest()
expression = '\"%s\" is None' % output_field
request.setFilterExpression(expression)
request.setFlags(QgsFeatureRequest.NoGeometry)
index = layer.fieldNameIndex(output_field)
for feature in layer.getFeatures():
if feature[output_field] is None:
layer.changeAttributeValue(feature.id(), index, 0)
layer.commitChanges()
layer.keywords = raster.keywords.copy()
layer.keywords['inasafe_fields'] = vector.keywords['inasafe_fields'].copy()
layer.keywords['inasafe_default_values'] = (
raster.keywords['inasafe_default_values'].copy())
key = exposure_count_field['key'] % raster.keywords['exposure']
# Special case here, one field is the exposure count and the total.
layer.keywords['inasafe_fields'][key] = output_field
layer.keywords['inasafe_fields'][total_field['key']] = output_field
layer.keywords['exposure_keywords'] = raster.keywords.copy()
layer.keywords['hazard_keywords'] = vector.keywords[
'hazard_keywords'].copy()
layer.keywords['aggregation_keywords'] = (
vector.keywords['aggregation_keywords'])
layer.keywords['layer_purpose'] = (
layer_purpose_aggregate_hazard_impacted['key'])
layer.keywords['title'] = output_layer_name
check_layer(layer)
return layer