Source code for safe.impact_functions.bases.layer_types.continuous_vector_exposure

# coding=utf-8
import sys

from safe.common.exceptions import NoAttributeInLayerError
from safe.impact_functions.bases.utilities import check_attribute_exist

__author__ = 'Rizky Maulana Nugraha "lucernae" <[email protected]>'
__date__ = '08/05/15'


[docs]class ContinuousVectorExposureMixin(object):

    def __init__(self):
        self._exposure_value_attribute = None
        self._exposure_min_value = None
        self._exposure_max_value = None

    @property
    def exposure_value_attribute(self):
        return self._exposure_value_attribute

    @exposure_value_attribute.setter
[docs]    def exposure_value_attribute(self, value):
        # self.exposure is from IF base class.
        exposure_layer = self.exposure.qgis_vector_layer()
        if (exposure_layer and
                check_attribute_exist(exposure_layer, value)):
            self._exposure_value_attribute = value
        else:
            message = ('The attribute "%s" does not exist in the hazard '
                       'layer.') % value
            raise NoAttributeInLayerError(message)

        # finding minima and maxima in a layer
        if exposure_layer:
            attr_index = exposure_layer.dataProvider().\
                fieldNameIndex(value)
            min_val = sys.maxint
            max_val = -sys.maxint - 1
            for feature in exposure_layer.getFeatures():
                try:
                    feature_value = feature.attributes()[attr_index]
                    feature_value = float(feature_value)
                    if feature_value < min_val:
                        min_val = feature_value
                    if feature_value > max_val:
                        max_val = feature_value
                except (ValueError, TypeError):
                    pass

            self.exposure_min_value = min_val
            self.exposure_max_value = max_val

    @property
    def exposure_min_value(self):
        return self._exposure_min_value

    @exposure_min_value.setter
[docs]    def exposure_min_value(self, value):
        self._exposure_min_value = value

    @property
    def exposure_max_value(self):
        return self._exposure_max_value

    @exposure_max_value.setter
[docs]    def exposure_max_value(self, value):
        self._exposure_max_value = value