def _create_foo_integral(prefix, form_id, integral_type, form_data):
"""Compute intermediate representation of create_foo_integral."""
subdomain_ids = []
classnames = []
itg_data = [
itg_data for itg_data in form_data.integral_data
if (itg_data.integral_type == integral_type
and itg_data.subdomain_id == "otherwise")
]
if len(itg_data) > 1:
raise RuntimeError(
"Expecting at most one default integral of each type.")
elif len(itg_data) == 1:
subdomain_ids += [-1]
classnames += [
naming.integral_name(integral_type, form_data.original_form,
form_id, "otherwise")
]
for itg_data in form_data.integral_data:
if isinstance(itg_data.subdomain_id, int):
if itg_data.subdomain_id < 0:
raise ValueError(
"Integral subdomain ID must be non-negative, not {}".
format(itg_data.subdomain_id))
if (itg_data.integral_type == integral_type):
subdomain_ids += [itg_data.subdomain_id]
classnames += [
naming.integral_name(integral_type,
form_data.original_form, form_id,
itg_data.subdomain_id)
]
return subdomain_ids, classnames
def compute_ir(analysis: namedtuple, object_names, prefix, parameters, visualise):
"""Compute intermediate representation."""
logger.info(79 * "*")
logger.info("Compiler stage 2: Computing intermediate representation of objects")
logger.info(79 * "*")
# Compute object names
# NOTE: This is done here for performance reasons, because repeated calls
# within each IR computation would be expensive due to UFL signature computations
finite_element_names = {e: naming.finite_element_name(e, prefix) for e in analysis.unique_elements}
dofmap_names = {e: naming.dofmap_name(e, prefix) for e in analysis.unique_elements}
integral_names = {}
form_names = {}
for fd_index, fd in enumerate(analysis.form_data):
form_names[fd_index] = naming.form_name(fd.original_form, fd_index, prefix)
for itg_index, itg_data in enumerate(fd.integral_data):
integral_names[(fd_index, itg_index)] = naming.integral_name(fd.original_form, itg_data.integral_type,
fd_index, itg_data.subdomain_id, prefix)
ir_elements = [
_compute_element_ir(e, analysis.element_numbers, finite_element_names)
for e in analysis.unique_elements
]
ir_dofmaps = [
_compute_dofmap_ir(e, analysis.element_numbers, dofmap_names)
for e in analysis.unique_elements
]
irs = [
_compute_integral_ir(fd, i, analysis.element_numbers, integral_names, parameters, visualise)
for (i, fd) in enumerate(analysis.form_data)
]
ir_integrals = list(itertools.chain(*irs))
ir_forms = [
_compute_form_ir(fd, i, prefix, form_names, integral_names, analysis.element_numbers, finite_element_names,
dofmap_names, object_names)
for (i, fd) in enumerate(analysis.form_data)
]
ir_expressions = [_compute_expression_ir(expr, i, prefix, analysis, parameters, visualise)
for i, expr in enumerate(analysis.expressions)]
return ir_data(elements=ir_elements, dofmaps=ir_dofmaps,
integrals=ir_integrals, forms=ir_forms,
expressions=ir_expressions)