ltbams 0.9.9__py3-none-any.whl → 1.0.2__py3-none-any.whl

ams/core/model.py

@@ -0,0 +1,330 @@
+"""
+Module for Model class.
+"""
+
+import logging
+from collections import OrderedDict
+from typing import Iterable
+
+import numpy as np
+from andes.core.common import Config
+from andes.core.param import ExtParam
+from andes.core.service import BaseService, BackRef
+from andes.utils.func import list_flatten
+
+from ams.core.documenter import Documenter
+from ams.core.var import Algeb
+
+from ams.utils.misc import deprec_get_idx
+
+logger = logging.getLogger(__name__)
+
+
+class Model:
+    """
+    Base class for power system scheduling models.
+
+    This class is revised from ``andes.core.model.Model``.
+    """
+
+    def __init__(self, system=None, config=None):
+
+        # --- Model ---
+        self.system = system
+        self.group = 'Undefined'
+
+        self.algebs = OrderedDict()  # internal algebraic variables
+        self.vars_decl_order = OrderedDict()  # variable in the order of declaration
+
+        self.params_ext = OrderedDict()  # external parameters
+
+        self.services = OrderedDict()  # service/temporary variables
+        self.services_ref = OrderedDict()  # BackRefs
+        self.services_sum = OrderedDict()  # VarSums
+
+        self.config = Config(name=self.class_name)  # `config` that can be exported
+        if config is not None:
+            self.config.load(config)
+
+        # basic configs
+        self.config.add(OrderedDict((('allow_adjust', 1),
+                                    ('adjust_lower', 0),
+                                    ('adjust_upper', 1),
+                                     )))
+        self.docum = Documenter(self)
+
+    def _all_vars(self):
+        """
+        An OrderedDict of States, ExtStates, Algebs, ExtAlgebs
+        """
+        return OrderedDict(list(self.algebs.items()))
+
+    def _check_attribute(self, key, value):
+        """
+        Check the attribute pair for valid names while instantiating the class.
+
+        This function assigns `owner` to the model itself, assigns the name and tex_name.
+        """
+        if isinstance(value, (Algeb, BaseService)):
+            if not value.owner:
+                value.owner = self
+            if not value.name:
+                value.name = key
+            if not value.tex_name:
+                value.tex_name = key
+            if key in self.__dict__:
+                logger.warning(f"{self.class_name}: redefinition of member <{key}>. Likely a modeling error.")
+
+    def __setattr__(self, key, value):
+        """
+        Overload the setattr function to register attributes.
+
+        Parameters
+        ----------
+        key : str
+            name of the attribute
+        value : [Algeb]
+            value of the attribute
+        """
+        self._check_attribute(key, value)
+        self._register_attribute(key, value)
+
+        super(Model, self).__setattr__(key, value)
+
+    def _register_attribute(self, key, value):
+        """
+        Register a pair of attributes to the model instance.
+
+        Called within ``__setattr__``, this is where the magic happens.
+        Subclass attributes are automatically registered based on the variable type.
+        Block attributes will be exported and registered recursively.
+        """
+        if isinstance(value, Algeb):
+            self.algebs[key] = value
+        elif isinstance(value, ExtParam):
+            self.params_ext[key] = value
+        elif isinstance(value, BackRef):
+            self.services_ref[key] = value
+            self.services[key] = value
+
+    @property
+    def class_name(self):
+        """
+        Return the class name
+        """
+        return self.__class__.__name__
+
+    def list2array(self):
+        """
+        Convert all the value attributes ``v`` to NumPy arrays.
+
+        Value attribute arrays should remain in the same address afterwards.
+        Namely, all assignments to value array should be operated in place (e.g., with [:]).
+        """
+
+        for instance in self.num_params.values():
+            instance.to_array()
+
+    def set_backref(self, name, from_idx, to_idx):
+        """
+        Helper function for setting idx-es to ``BackRef``.
+        """
+
+        if name not in self.services_ref:
+            return
+
+        uid = self.idx2uid(to_idx)
+        self.services_ref[name].v[uid].append(from_idx)
+
+    def get(self, src: str, idx, attr: str = 'v', allow_none=False, default=0.0):
+        """
+        Get the value of an attribute of a model property.
+
+        The return value is ``self.<src>.<attr>[idx]``
+
+        Parameters
+        ----------
+        src : str
+            Name of the model property
+        idx : str, int, float, array-like
+            Indices of the devices
+        attr : str, optional, default='v'
+            The attribute of the property to get.
+            ``v`` for values, ``a`` for address, and ``e`` for equation value.
+        allow_none : bool
+            True to allow None values in the indexer
+        default : float
+            If `allow_none` is true, the default value to use for None indexer.
+
+        Returns
+        -------
+        array-like
+            ``self.<src>.<attr>[idx]``
+
+        """
+        uid = self.idx2uid(idx)
+        if isinstance(self.__dict__[src].__dict__[attr], list):
+            if isinstance(uid, Iterable):
+                if not allow_none and (uid is None or None in uid):
+                    raise KeyError('None not allowed in uid/idx. Enable through '
+                                   '`allow_none` and provide a `default` if needed.')
+                return [self.__dict__[src].__dict__[attr][i] if i is not None else default
+                        for i in uid]
+        # FIXME: this seems to be an unexpected case originted from ANDES
+        if isinstance(uid, Iterable):
+            if None in uid:
+                return [self.__dict__[src].__dict__[attr][i] if i is not None else default
+                        for i in uid]
+        return self.__dict__[src].__dict__[attr][uid]
+
+    def set(self, src, idx, attr, value):
+        """
+        Set the value of an attribute of a model property.
+
+        Performs ``self.<src>.<attr>[idx] = value``. This method will not modify
+        the input values from the case file that have not been converted to the
+        system base. As a result, changes applied by this method will not affect
+        the dumped case file.
+
+        To alter parameters and reflect it in the case file, use :meth:`alter`
+        instead.
+
+        Parameters
+        ----------
+        src : str
+            Name of the model property
+        idx : str, int, float, array-like
+            Indices of the devices
+        attr : str, optional, default='v'
+            The internal attribute of the property to get.
+            ``v`` for values, ``a`` for address, and ``e`` for equation value.
+        value : array-like
+            New values to be set
+
+        Returns
+        -------
+        bool
+            True when successful.
+        """
+        uid = self.idx2uid(idx)
+        self.__dict__[src].__dict__[attr][uid] = value
+        return True
+
+    def alter(self, src, idx, value, attr='v'):
+        """
+        Alter values of input parameters or constant service.
+
+        If the method operates on an input parameter, the new data should be in
+        the same base as that in the input file. This function will convert
+        ``value`` to per unit in the system base whenever necessary.
+
+        The values for storing the input data, i.e., the parameter's ``vin``
+        field, will be overwritten. As a result, altered values will be
+        reflected in the dumped case file.
+
+        Parameters
+        ----------
+        src : str
+            The parameter name to alter
+        idx : str, float, int
+            The device to alter
+        value : float
+            The desired value
+        attr : str
+            The attribute to alter, default is 'v'.
+
+        Notes
+        -----
+        New in version 0.9.14: Added the signature `attr` to alter specific attributes.
+        This feature is useful when you need to manipulate parameter values in the system
+        base and ensure that these changes are reflected in the dumped case file.
+        """
+
+        instance = self.__dict__[src]
+
+        if hasattr(instance, 'vin') and (instance.vin is not None):
+            uid = self.idx2uid(idx)
+            if attr == 'vin':
+                self.set(src, idx, 'vin', value / instance.pu_coeff[uid])
+                self.set(src, idx, 'v', value=value)
+            else:
+                self.set(src, idx, 'vin', value)
+                self.set(src, idx, 'v', value * instance.pu_coeff[uid])
+        elif not hasattr(instance, 'vin') and attr == 'vin':
+            logger.warning(f"{self.class_name}.{src} has no `vin` attribute, changing `v`.")
+            self.set(src, idx, 'v', value)
+        else:
+            self.set(src, idx, attr=attr, value=value)
+
+    def idx2uid(self, idx):
+        """
+        Convert idx to the 0-indexed unique index.
+
+        Parameters
+        ----------
+        idx : array-like, numbers, or str
+            idx of devices
+
+        Returns
+        -------
+        list
+            A list containing the unique indices of the devices
+        """
+        if idx is None:
+            logger.debug("idx2uid returned None for idx None")
+            return None
+        if isinstance(idx, (float, int, str, np.integer, np.floating)):
+            return self._one_idx2uid(idx)
+        elif isinstance(idx, Iterable):
+            if len(idx) > 0 and isinstance(idx[0], (list, np.ndarray)):
+                idx = list_flatten(idx)
+            return [self._one_idx2uid(i) if i is not None else None
+                    for i in idx]
+        else:
+            raise NotImplementedError(f'Unknown idx type {type(idx)}')
+
+    def _one_idx2uid(self, idx):
+        """
+        Helper function for checking if an idx exists and
+        converting it to uid.
+        """
+
+        if idx not in self.uid:
+            raise KeyError("<%s>: device not exist with idx=%s." %
+                           (self.class_name, idx))
+
+        return self.uid[idx]
+
+    def doc(self, max_width=78, export='plain'):
+        """
+        Retrieve model documentation as a string.
+        """
+        return self.docum.get(max_width=max_width, export=export)
+
+    @deprec_get_idx
+    def get_idx(self):
+        """
+        Return the index of the model instance.
+        Equivalent to ``self.idx.v``, develoepd for consistency with group method
+        ``get_idx``.
+        """
+        return self.idx.v
+
+    def get_all_idxes(self):
+        """
+        Return all the indexes of this model.
+
+        .. note::
+            New in version 1.0.0. Add to follow the group method ``get_all_idxes``.
+
+        Returns
+        -------
+        list
+            A list of indexes.
+        """
+        return self.idx.v
+
+    def __repr__(self):
+        dev_text = 'device' if self.n == 1 else 'devices'
+
+        return f'{self.class_name} ({self.n} {dev_text}) at {hex(id(self))}'

ams/core/param.py

@@ -0,0 +1,322 @@
+"""
+Base class for parameters.
+"""
+
+
+import logging
+
+from typing import Optional, Iterable
+
+import numpy as np
+from scipy.sparse import issparse
+
+from ams.opt import Param
+
+from ams.utils.misc import deprec_get_idx
+
+logger = logging.getLogger(__name__)
+
+
+class RParam(Param):
+    """
+    Class for parameters used in a routine.
+    This class is developed to simplify the routine definition.
+
+    `RParm` is further used to define `Parameter` in the optimization model.
+
+    `no_parse` is used to skip parsing the `RParam` in optimization
+    model.
+    It means that the `RParam` will not be added to the optimization model.
+    This is useful when the RParam contains non-numeric values,
+    or it is not necessary to be added to the optimization model.
+
+    Parameters
+    ----------
+    name : str, optional
+        Name of this parameter. If not provided, `name` will be set
+        to the attribute name.
+    tex_name : str, optional
+        LaTeX-formatted parameter name. If not provided, `tex_name`
+        will be assigned the same as `name`.
+    info : str, optional
+        A description of this parameter
+    src : str, optional
+        Source name of the parameter.
+    unit : str, optional
+        Unit of the parameter.
+    model : str, optional
+        Name of the owner model or group.
+    v : np.ndarray, optional
+        External value of the parameter.
+    indexer : str, optional
+        Indexer of the parameter.
+    imodel : str, optional
+        Name of the owner model or group of the indexer.
+    no_parse: bool, optional
+        True to skip parsing the parameter.
+    nonneg: bool, optional
+        True to set the parameter as non-negative.
+    nonpos: bool, optional
+        True to set the parameter as non-positive.
+    cplx: bool, optional
+        True to set the parameter as complex.
+    imag: bool, optional
+        True to set the parameter as imaginary.
+    symmetric: bool, optional
+        True to set the parameter as symmetric.
+    diag: bool, optional
+        True to set the parameter as diagonal.
+    hermitian: bool, optional
+        True to set the parameter as hermitian.
+    boolean: bool, optional
+        True to set the parameter as boolean.
+    integer: bool, optional
+        True to set the parameter as integer.
+    pos: bool, optional
+        True to set the parameter as positive.
+    neg: bool, optional
+        True to set the parameter as negative.
+    sparse: bool, optional
+        True to set the parameter as sparse.
+
+    Examples
+    --------
+    Example 1: Define a routine parameter from a source model or group.
+
+    In this example, we define the parameter `cru` from the source model
+    `SFRCost` with the parameter `cru`.
+
+    >>> self.cru = RParam(info='RegUp reserve coefficient',
+    >>>                   tex_name=r'c_{r,u}',
+    >>>                   unit=r'$/(p.u.)',
+    >>>                   name='cru',
+    >>>                   src='cru',
+    >>>                   model='SFRCost'
+    >>>                   )
+
+    Example 2: Define a routine parameter with a user-defined value.
+
+    In this example, we define the parameter with a user-defined value.
+    TODO: Add example
+    """
+
+    def __init__(self,
+                 name: Optional[str] = None,
+                 tex_name: Optional[str] = None,
+                 info: Optional[str] = None,
+                 src: Optional[str] = None,
+                 unit: Optional[str] = None,
+                 model: Optional[str] = None,
+                 v: Optional[np.ndarray] = None,
+                 indexer: Optional[str] = None,
+                 imodel: Optional[str] = None,
+                 expand_dims: Optional[int] = None,
+                 no_parse: Optional[bool] = False,
+                 nonneg: Optional[bool] = False,
+                 nonpos: Optional[bool] = False,
+                 cplx: Optional[bool] = False,
+                 imag: Optional[bool] = False,
+                 symmetric: Optional[bool] = False,
+                 diag: Optional[bool] = False,
+                 hermitian: Optional[bool] = False,
+                 boolean: Optional[bool] = False,
+                 integer: Optional[bool] = False,
+                 pos: Optional[bool] = False,
+                 neg: Optional[bool] = False,
+                 sparse: Optional[list] = None,
+                 ):
+        Param.__init__(self, nonneg=nonneg, nonpos=nonpos,
+                       cplx=cplx, imag=imag, symmetric=symmetric,
+                       diag=diag, hermitian=hermitian, boolean=boolean,
+                       integer=integer, pos=pos, neg=neg, sparse=sparse)
+        self.name = name
+        self.tex_name = tex_name if (tex_name is not None) else name
+        self.info = info
+        self.src = name if (src is None) else src
+        self.unit = unit
+        self.is_group = False
+        self.model = model  # name of a group or model
+        self.indexer = indexer  # name of the indexer
+        self.imodel = imodel  # name of a group or model of the indexer
+        self.expand_dims = expand_dims
+        self.no_parse = no_parse
+        self.owner = None  # instance of the owner model or group
+        self.is_ext = False  # indicate if the value is set externally
+        self._v = None  # external value
+        if v is not None:
+            self._v = v
+            self.is_ext = True
+
+    # FIXME: might need a better organization
+    @property
+    def v(self):
+        """
+        The value of the parameter.
+
+        Notes
+        -----
+        - This property is a wrapper for the ``get`` method of the owner class.
+        - The value will sort by the indexer if indexed, used for optmization modeling.
+        """
+        out = None
+        if self.sparse and self.expand_dims is not None:
+            msg = 'Sparse matrix does not support expand_dims.'
+            raise NotImplementedError(msg)
+        if self.indexer is None:
+            if self.is_ext:
+                if issparse(self._v):
+                    out = self._v.toarray()
+                else:
+                    out = self._v
+            elif self.is_group:
+                out = self.owner.get(src=self.src, attr='v',
+                                     idx=self.owner.get_all_idxes())
+            else:
+                src_param = getattr(self.owner, self.src)
+                out = getattr(src_param, 'v')
+        else:
+            try:
+                imodel = getattr(self.rtn.system, self.imodel)
+            except AttributeError:
+                msg = f'Indexer source model <{self.imodel}> not found, '
+                msg += 'likely a modeling error.'
+                raise AttributeError(msg)
+            try:
+                sorted_idx = self.owner.find_idx(keys=self.indexer, values=imodel.get_all_idxes())
+            except AttributeError:
+                sorted_idx = self.owner.idx.v
+            except Exception as e:
+                raise e
+            model = getattr(self.rtn.system, self.model)
+            out = model.get(src=self.src, attr='v', idx=sorted_idx)
+        if self.expand_dims is not None:
+            out = np.expand_dims(out, axis=self.expand_dims)
+        return out
+
+    @property
+    def shape(self):
+        """
+        Return the shape of the parameter.
+        """
+        if self.is_ext:
+            return np.shape(self._v)
+        elif self.no_parse:
+            return None
+        else:
+            return np.shape(self.v)
+
+    @property
+    def dtype(self):
+        """
+        Return the data type of the parameter value.
+        """
+        if isinstance(self.v, (str, bytes)):
+            return str
+        elif isinstance(self.v, Iterable):
+            return type(self.v[0])
+        else:
+            return type(self.v)
+
+    @property
+    def n(self):
+        """
+        Return the szie of the parameter.
+        """
+        if self.is_ext:
+            return self._v.shape[0]
+        else:
+            return self.owner.n
+
+    @property
+    def class_name(self):
+        """
+        Return the class name
+        """
+        return self.__class__.__name__
+
+    def __repr__(self):
+        owner = self.owner.__class__.__name__ if self.owner is not None else self.rtn.__class__.__name__
+        postfix = '' if self.src is None else f'.{self.src}'
+        return f'{self.__class__.__name__}: {owner}' + postfix
+
+    @deprec_get_idx
+    def get_idx(self):
+        """
+        Get the index of the parameter.
+
+        Returns
+        -------
+        idx : list
+            Index of the parameter.
+
+        Notes
+        -----
+        - The value will sort by the indexer if indexed.
+        """
+        if self.indexer is None:
+            if self.is_group:
+                return self.owner.get_all_idxes()
+            elif self.owner is None:
+                logger.info(f'Param <{self.name}> has no owner.')
+                return None
+            elif hasattr(self.owner, 'idx'):
+                return self.owner.idx.v
+            else:
+                logger.info(f'Param <{self.name}> owner <{self.owner.class_name}> has no idx.')
+                return None
+        else:
+            try:
+                imodel = getattr(self.rtn.system, self.imodel)
+            except AttributeError:
+                msg = f'Indexer source model <{self.imodel}> not found, '
+                msg += 'likely a modeling error.'
+                raise AttributeError(msg)
+            try:
+                sorted_idx = self.owner.find_idx(keys=self.indexer, values=imodel.get_all_idxes())
+            except AttributeError:
+                msg = f'Indexer <{self.indexer}> not found in <{self.imodel}>, '
+                msg += 'likely a modeling error.'
+                raise AttributeError(msg)
+            return sorted_idx
+
+    def get_all_idxes(self):
+        """
+        Get all the indexes of the parameter.
+
+        .. note::
+            New in version 1.0.0.
+
+        Returns
+        -------
+        idx : list
+            Index of the parameter.
+
+        Notes
+        -----
+        - The value will sort by the indexer if indexed.
+        """
+        if self.indexer is None:
+            if self.is_group:
+                return self.owner.get_all_idxes()
+            elif self.owner is None:
+                logger.info(f'Param <{self.name}> has no owner.')
+                return None
+            elif hasattr(self.owner, 'idx'):
+                return self.owner.idx.v
+            else:
+                logger.info(f'Param <{self.name}> owner <{self.owner.class_name}> has no idx.')
+                return None
+        else:
+            try:
+                imodel = getattr(self.rtn.system, self.imodel)
+            except AttributeError:
+                msg = f'Indexer source model <{self.imodel}> not found, '
+                msg += 'likely a modeling error.'
+                raise AttributeError(msg)
+            try:
+                sorted_idx = self.owner.find_idx(keys=self.indexer, values=imodel.get_all_idxes())
+            except AttributeError:
+                msg = f'Indexer <{self.indexer}> not found in <{self.imodel}>, '
+                msg += 'likely a modeling error.'
+                raise AttributeError(msg)
+            return sorted_idx