Add a first implementation of hooks

python/dune/perftool/generation/__init__.py

@@ -31,6 +31,10 @@ from dune.perftool.generation.cpp import (base_class,
                                           template_parameter,
                                           )
 
+from dune.perftool.generation.hooks import (register_hook,
+                                            run_hook,
+                                            )
+
 from dune.perftool.generation.loopy import (barrier,
                                             constantarg,
                                             domain,

python/dune/perftool/generation/hooks.py

+""" All the infrastructure code related to adding hooks to the code generation process """
+
+
+_hooks = {}
+
+
+def register_hook(hookname, func):
+    current = _hooks.setdefault(hookname, ())
+    current = list(current)
+    current.append(func)
+    _hooks[hookname] = tuple(current)
+
+
+def run_hook(hookname, *args, **kwargs):
+    for hook in _hooks[hookname]:
+        hook(*args, **kwargs)

python/dune/perftool/pdelab/localoperator.py

@@ -28,6 +28,7 @@ from dune.perftool.generation import (backend,
                                       post_include,
                                       retrieve_cache_functions,
                                       retrieve_cache_items,
+                                      run_hook,
                                       template_parameter,
                                       )
 from dune.perftool.cgen.clazz import (AccessModifier,
@@ -452,6 +453,8 @@ def visit_integral(integral):
     visitor = get_visitor(measure, subdomain_id)
     visitor.accumulate(integrand)
 
+    run_hook("after_visit", visitor)
+
 
 def generate_kernel(integrals):
     logger = logging.getLogger(__name__)