Source code for ufl.restriction

# -*- coding: utf-8 -*-
"""Restriction operations."""

# Copyright (C) 2008-2016 Martin Sandve Aln├Žs
#
# This file is part of UFL (https://www.fenicsproject.org)
#
# SPDX-License-Identifier:    LGPL-3.0-or-later

from ufl.core.operator import Operator
from ufl.precedence import parstr
from ufl.core.ufl_type import ufl_type


# --- Restriction operators ---

[docs]@ufl_type(is_abstract=True, num_ops=1, inherit_shape_from_operand=0, inherit_indices_from_operand=0, is_restriction=True) class Restricted(Operator): __slots__ = () # TODO: Add __new__ operator here, e.g. restricted(literal) == literal def __init__(self, f): Operator.__init__(self, (f,))
[docs] def side(self): return self._side
[docs] def evaluate(self, x, mapping, component, index_values): return self.ufl_operands[0].evaluate(x, mapping, component, index_values)
def __str__(self): return "%s('%s')" % (parstr(self.ufl_operands[0], self), self._side)
[docs]@ufl_type(is_terminal_modifier=True) class PositiveRestricted(Restricted): __slots__ = () _side = "+"
[docs]@ufl_type(is_terminal_modifier=True) class NegativeRestricted(Restricted): __slots__ = () _side = "-"