FEniCS / ffcx / 19801120260

import logging

import basix

from ffcx.codegeneration.backend import FFCXBackend

from ffcx.codegeneration.cpp import integrals_template as ufcx_integrals

from ffcx.codegeneration.cpp.cpp_implementation import CppFormatter

from ffcx.codegeneration.integral_generator import IntegralGenerator

from ffcx.ir.representation import IntegralIR

logger = logging.getLogger("ffcx")

def generator(ir: IntegralIR, domain: basix.CellType, options):

    logger.info("Generating code for integral:")

    logger.info(f"--- type: {ir.expression.integral_type}")

    logger.info(f"--- name: {ir.expression.name}")

    factory_name = ir.expression.name

    declaration = ufcx_integrals.declaration.format(factory_name=factory_name)

    backend = FFCXBackend(ir, options)

    ig = IntegralGenerator(ir, backend)

    parts = ig.generate(domain)

    CF = CppFormatter(options["scalar_type"])

    body = CF.c_format(parts)

    code = {}

    code["class_type"] = ir.expression.integral_type + "_integral"

    code["name"] = ir.expression.name

    vals = ", ".join("true" if i else "false" for i in ir.enabled_coefficients)

    code["enabled_coefficients"] = f"{{{vals}}}"

    code["additional_includes_set"] = set()  # FIXME: Get this out of code[]

    code["tabulate_tensor"] = body

    implementation = ufcx_integrals.factory.format(

    return declaration + implementation, ""