/usr/lib/python2.7/dist-packages/ufl/indexed.py is in python-ufl 1.4.0-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 | """This module defines the Indexed class."""
# Copyright (C) 2008-2014 Martin Sandve Alnes
#
# This file is part of UFL.
#
# UFL is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# UFL is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with UFL. If not, see <http://www.gnu.org/licenses/>.
from itertools import izip
from ufl.log import error
from ufl.expr import Expr
from ufl.operatorbase import WrapperType
from ufl.indexing import Index, FixedIndex, as_multi_index
from ufl.indexutils import unique_indices
from ufl.precedence import parstr
from ufl.common import EmptyDict
#--- Indexed expression ---
class Indexed(WrapperType):
__slots__ = ("_ops",
"_free_indices", "_index_dimensions",)
def __init__(self, expression, indices):
WrapperType.__init__(self)
# Error checking
if not isinstance(expression, Expr):
error("Expecting Expr instance, not %s." % repr(expression))
shape = expression.shape()
# Error checking
if len(shape) != len(indices):
error("Invalid number of indices (%d) for tensor "\
"expression of rank %d:\n\t%r\n"\
% (len(indices), expression.rank(), expression))
# Store operands
indices = as_multi_index(indices, shape)
self._ops = (expression, indices)
# Error checking
for si, di in izip(shape, indices):
if isinstance(di, FixedIndex) and int(di) >= int(si):
error("Fixed index out of range!")
# Build free index tuple and dimensions
idims = dict((i, s) for (i, s) in izip(indices._indices, shape)
if isinstance(i, Index))
idims.update(expression.index_dimensions())
fi = unique_indices(expression.free_indices() + indices._indices)
# Cache free index and dimensions
self._free_indices = fi
self._index_dimensions = idims or EmptyDict
def operands(self):
return self._ops
def free_indices(self):
return self._free_indices
def index_dimensions(self):
return self._index_dimensions
def shape(self):
return ()
def is_cellwise_constant(self):
"Return whether this expression is spatially constant over each cell."
return self._ops[0].is_cellwise_constant()
def evaluate(self, x, mapping, component, index_values, derivatives=()):
A, ii = self.operands()
component = ii.evaluate(x, mapping, None, index_values)
if derivatives:
return A.evaluate(x, mapping, component, index_values, derivatives)
else:
return A.evaluate(x, mapping, component, index_values)
def __str__(self):
return "%s[%s]" % (parstr(self._ops[0], self), self._ops[1])
def __repr__(self):
return "Indexed(%r, %r)" % self._ops
def __getitem__(self, key):
error("Attempting to index with %r, but object is already indexed: %r" % (key, self))
|