def __init__(self, mapping):
MultiFunction.__init__(self)
self._mapping = mapping
if not all(k._ufl_is_terminal_ for k in mapping.keys()):
error("This implementation can only replace Terminal objects.")
if not all(k.ufl_shape == v.ufl_shape for k, v in mapping.items()):
error("Replacement expressions must have the same shape as what they replace.")
def __init__(self, side=None):
MultiFunction.__init__(self)
self.current_restriction = side
self.default_restriction = "+"
if self.current_restriction is None:
self._rp = {"+": RestrictionPropagator("+"),
"-": RestrictionPropagator("-")}
def __init__(self, argument_deps=None, coefficient_deps=None):
MultiFunction.__init__(self)
if argument_deps is None:
argument_deps = {}
if coefficient_deps is None:
coefficient_deps = {}
self.argument_deps = argument_deps
self.coefficient_deps = coefficient_deps
def __init__(self, context):
# MultiFunction.__init__ does not call further __init__
# methods, but ufl2gem.Mixin must be initialised.
# (ModifiedTerminalMixin requires no initialisation.)
MultiFunction.__init__(self)
ufl2gem.Mixin.__init__(self)
# Need context during translation!
self.context = context
def __init__(self, ir, language, symbols, parameters):
MultiFunction.__init__(self)
# Store ir and parameters
self.entitytype = ir["entitytype"]
self.integral_type = ir["integral_type"]
self.language = language
self.symbols = symbols
self.parameters = parameters
def __init__(self, return_list):
MultiFunction.__init__(self)
self.return_list = return_list
def __init__(self):
self.varmapping = OrderedDict()
MultiFunction.__init__(self)
ufl2gem.Mixin.__init__(self)
def __init__(self):
MultiFunction.__init__(self)
def __init__(self, language):
MultiFunction.__init__(self)
UFL2CNodesMixin.__init__(self, language)
def __init__(self):
MultiFunction.__init__(self)
self._one = as_ufl(1.0)
self._arg = None
def __init__(self):
MultiFunction.__init__(self)
# caches for reuse in the dispatched transformers
self.vcaches = defaultdict(dict)
self.rcaches = defaultdict(dict)
def __init__(self):
MultiFunction.__init__(self)
self.vcache = defaultdict(dict)
self.rcache = defaultdict(dict)
def __init__(self, test=None, trial=None):
self.restrictions = {0: test, 1: trial}
MultiFunction.__init__(self)
def __init__(self, language):
MultiFunction.__init__(self)
UFL2CNodesMixin.__init__(self, language)
def __init__(self, require_restriction):
MultiFunction.__init__(self)
self.current_restriction = None
self.require_restriction = require_restriction
def __init__(self):
MultiFunction.__init__(self)
def __init__(self):
MultiFunction.__init__(self)
self._ngrads = 0
self._restricted = ''
self._avg = ''
def __init__(self, test=None, trial=None):
self.restrictions = {0: test, 1: trial}
MultiFunction.__init__(self)
def __init__(self, var_shape):
MultiFunction.__init__(self)
self._var_shape = var_shape
def __init__(self):
MultiFunction.__init__(self)
self._ngrads = 0
self._restricted = ''
self._avg = ''
def __init__(self, tabulation_manager, context):
MultiFunction.__init__(self)
ufl2gem.Mixin.__init__(self)
context.tabulation_manager = tabulation_manager
self.context = context
def __init__(self, arguments):
MultiFunction.__init__(self)
self.arguments = arguments
self._et = ()
def __init__(self, argument_names=None, coefficient_names=None, colorama_bold=False):
MultiFunction.__init__(self)
self.argument_names = argument_names
self.coefficient_names = coefficient_names
self.colorama_bold = colorama_bold and has_colorama
def __init__(self):
MultiFunction.__init__(self)
self.nodetype = {}
def __init__(self, split):
MultiFunction.__init__(self)
self._split = split
def __init__(self, preserve_types=()):
MultiFunction.__init__(self)
# Store preserve_types as boolean lookup table
self._preserve_types = [False]*Expr._ufl_num_typecodes_
for cls in preserve_types:
self._preserve_types[cls._ufl_typecode_] = True
def __init__(self, coordinates):
self.coordinates = coordinates
MultiFunction.__init__(self)
def __init__(self, var_shape):
MultiFunction.__init__(self)
self._var_shape = var_shape
def __init__(self, split):
MultiFunction.__init__(self)
self._split = split
def __init__(self, argument_names=None, coefficient_names=None):
MultiFunction.__init__(self)
self.argument_names = argument_names
self.coefficient_names = coefficient_names
def __init__(self, e2i, V_sizes, V_symbols):
MultiFunction.__init__(self)
self.symbol_count = 0
self.e2i = e2i
self.V_sizes = V_sizes
self.V_symbols = V_symbols
