Source code for bw2data.method

# -*- coding: utf-8 -*-
from __future__ import print_function, unicode_literals
from eight import *

from . import mapping, methods, geomapping, config
from .utils import MAX_INT_32, numpy_string
from .validate import ia_validator
from .ia_data_store import ImpactAssessmentDataStore
import numpy as np




[docs]
class Method(ImpactAssessmentDataStore):
    """A manager for an impact assessment method. This class can register or deregister methods, write intermediate data, process data to parameter arrays, validate, and copy methods.

    The Method class never holds intermediate data, but it can load or write intermediate data. The only attribute is *name*, which is the name of the method being managed.

    Instantiation does not load any data. If this method is not yet registered in the metadata store, a warning is written to ``stdout``.

    Methods are hierarchally structured, and this structure is preserved in the method name. It is a tuple of strings, like ``('ecological scarcity 2006', 'total', 'natural resources')``.

    The data schema for IA methods is:

    .. code-block:: python

            Schema([Any(
                [valid_tuple, maybe_uncertainty],         # site-generic
                [valid_tuple, maybe_uncertainty, object]  # regionalized
            )])

    where:
        * *valid_tuple* (tuple): A dataset identifier, like ``("biosphere", "CO2")``.
        * *maybe_uncertainty* (uncertainty dict or number): Either a number or an uncertainty dictionary.
        * *object* (object, optional) is a location identifier, used only for regionalized LCIA.

    Args:
        * *name* (tuple): Name of impact assessment method to manage.

    """


[docs]
    _metadata = methods




[docs]
    validator = ia_validator




[docs]
    dtype_fields = [
            (numpy_string('flow'), np.uint32),
            (numpy_string('geo'), np.uint32),
            (numpy_string('row'), np.uint32),
            (numpy_string('col'), np.uint32),
    ]





[docs]
    def add_mappings(self, data):
        mapping.add({x[0] for x in data})
        geomapping.add({x[2] for x in data if len(x) == 3})





[docs]
    def process_data(self, row):
        return (
            mapping[row[0]],
            geomapping[row[2]] if len(row) == 3 \
                else geomapping[config.global_location],
            MAX_INT_32,
            MAX_INT_32,
            ), row[1]





[docs]
    def write(self, data, process=True):
        """Serialize intermediate data to disk.

        Sets the metadata key ``num_cfs`` automatically."""
        self.metadata[u"num_cfs"] = len(data)
        self._metadata.flush()
        super(Method, self).write(data)
Support Brightway