/usr/lib/python2.7/dist-packages/ufl/indexsum.py is in python-ufl 1.6.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 | """This module defines the IndexSum 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 six.moves import xrange as range
from ufl.log import error
from ufl.assertions import ufl_assert
from ufl.core.expr import Expr
from ufl.core.operator import Operator
from ufl.core.multiindex import Index, MultiIndex, as_multi_index
from ufl.precedence import parstr
from ufl.common import EmptyDict
from ufl.core.ufl_type import ufl_type
from ufl.constantvalue import Zero
#--- Sum over an index ---
@ufl_type(num_ops=2)
class IndexSum(Operator):
__slots__ = ("_dimension", "ufl_free_indices", "ufl_index_dimensions")
def __new__(cls, summand, index):
# Error checks
if not isinstance(summand, Expr):
error("Expecting Expr instance, not %s." % repr(summand))
if not isinstance(index, MultiIndex):
error("Expecting MultiIndex instance, not %s." % repr(index))
if len(index) != 1:
error("Expecting a single Index only.")
# Simplification to zero
if isinstance(summand, Zero):
sh = summand.ufl_shape
j, = index
fi = summand.ufl_free_indices
fid = summand.ufl_index_dimensions
pos = fi.index(j.count())
fi = fi[:pos] + fi[pos+1:]
fid = fid[:pos] + fid[pos+1:]
return Zero(sh, fi, fid)
return Operator.__new__(cls)
def __init__(self, summand, index):
j, = index
fi = summand.ufl_free_indices
fid = summand.ufl_index_dimensions
pos = fi.index(j.count())
self._dimension = fid[pos]
self.ufl_free_indices = fi[:pos] + fi[pos+1:]
self.ufl_index_dimensions = fid[:pos] + fid[pos+1:]
Operator.__init__(self, (summand, index))
def index(self):
return self.ufl_operands[1][0]
def dimension(self):
return self._dimension
@property
def ufl_shape(self):
return self.ufl_operands[0].ufl_shape
def is_cellwise_constant(self):
"Return whether this expression is spatially constant over each cell."
return self.ufl_operands[0].is_cellwise_constant()
def evaluate(self, x, mapping, component, index_values):
i, = self.ufl_operands[1]
tmp = 0
for k in range(self._dimension):
index_values.push(i, k)
tmp += self.ufl_operands[0].evaluate(x, mapping, component, index_values)
index_values.pop()
return tmp
def __str__(self):
return "sum_{%s} %s " % (str(self.ufl_operands[1]), parstr(self.ufl_operands[0], self))
def __repr__(self):
return "IndexSum(%r, %r)" % (self.ufl_operands[0], self.ufl_operands[1])
|