UFL Specification and User Manual 0.3 - FEniCS Project

UFL Specification and User Manual 0.3Martin S. Alnæs, Anders Loggformation rules for the types you wish to handle.Shown here is maybe the simplest transformer possible:class Printer(Transformer):def __init__(self):Transformer.__init__(self)def expr(self, o, *operands):print "Visiting", str(o), "with operands:"print ", ".join(map(str,operands))return oelement = FiniteElement("CG", triangle, 1)v = TestFunction(element)u = TrialFunction(element)a = u*vp = Printer()p.visit(a)The call to visit will traverse a and call Printer.expr on all expressionnodes in post–order, with the argument operands holding the return valuesfrom visits to the operands of o. The output is:TODOImplementing expr above provides a default handler for any expression nodetype. For each subclass of Expr you can define a handler function to overridethe default by using the name of the type in underscore notation, e.g.basis function for BasisFunction. The constructor of Transformer andimplementation of Transformer.visit handles the mapping from type tohandler function automatically.Here is a simple example to show how to override default behaviour:85