Skip to content

Commit

Permalink
remove tdims
Browse files Browse the repository at this point in the history
  • Loading branch information
mscroggs committed Aug 18, 2023
1 parent 1206529 commit d758e8e
Showing 1 changed file with 5 additions and 7 deletions.
12 changes: 5 additions & 7 deletions symfem/functionals.py
Original file line number Diff line number Diff line change
Expand Up @@ -880,15 +880,13 @@ def __init__(self, reference: Reference, f_in: FunctionInput,
assert len(f) == self.integral_domain.tdim
self.f = mappings.contravariant(
f, self.integral_domain.get_map_to_self(),
self.integral_domain.get_inverse_map_to_self(),
self.integral_domain.tdim)
self.integral_domain.get_inverse_map_to_self())
elif f.is_matrix:
assert f.shape[0] == self.integral_domain.tdim
assert f.shape[1] == self.integral_domain.tdim
self.f = mappings.double_contravariant(
f, self.integral_domain.get_map_to_self(),
self.integral_domain.get_inverse_map_to_self(),
self.integral_domain.tdim)
self.integral_domain.get_inverse_map_to_self())
else:
self.f = f

Expand Down Expand Up @@ -1152,13 +1150,13 @@ def __init__(self, reference: Reference,
if self.f.is_vector and len(self.f) != reference.gdim:
self.f = mappings.contravariant(
self.f, id_def.get_map_to_self(),
id_def.get_inverse_map_to_self(), id_def.tdim)
id_def.get_inverse_map_to_self())
elif self.f.is_matrix:
assert self.f.shape[0] == id_def.tdim
assert self.f.shape[1] == id_def.tdim
self.f = mappings.double_contravariant(
self.f, id_def.get_map_to_self(),
id_def.get_inverse_map_to_self(), id_def.tdim)
id_def.get_inverse_map_to_self())

# Map from default reference to reference
if map_function and reference != reference.default_reference():
Expand All @@ -1167,7 +1165,7 @@ def __init__(self, reference: Reference,
mf = getattr(mappings, f"{mapping}_inverse_transpose")
self.f = mf(self.f, reference.get_map_to_self(),
reference.get_inverse_map_to_self(),
reference.tdim, substitute=False)
substitute=False)
self.f *= id_def.volume() / self.integral_domain.volume()

def _eval_symbolic(self, function: AnyFunction) -> AnyFunction:
Expand Down

0 comments on commit d758e8e

Please sign in to comment.