Added new transition pattern

Author: willgebhardt

ngcsimlib/compilers/component_compiler.py

@@ -52,8 +52,13 @@ def parse(component, compile_key):
         args = []
         parameters = []
         compartments = []
-        varnames = pure_fn.__func__.__code__.co_varnames[
-                   :pure_fn.__func__.__code__.co_argcount]
+        try:
+            func = pure_fn.__func__
+        except:
+            func = pure_fn
+
+        varnames = func.__code__.co_varnames[
+                   :func.__code__.co_argcount]
 
         for name in varnames:
             if name not in component.__dict__.keys():
@@ -104,11 +109,16 @@ def compile(component, resolver):
 
     comp_key_key = [(narg.split('/')[-1], narg) for narg in comp_ids]
 
+    try:
+        func = pure_fn.__func__
+    except:
+        func = pure_fn
+
     def compiled(**kwargs):
         funArgs = {narg: kwargs.get(narg) for narg in _args}
         funComps = {knarg: kwargs.get(narg) for knarg, narg in comp_key_key}
 
-        return pure_fn.__func__(**funParams, **funArgs, **funComps)
+        return func(**funParams, **funArgs, **funComps)
 
     exc_order.append((compiled, out_ids, component.name, comp_ids))
     return exc_order

ngcsimlib/transition.py

@@ -0,0 +1,21 @@
+from ngcsimlib.logger import warn
+from ngcsimlib.utils import add_component_resolver, add_resolver_meta
+
+def add_transition(pure_fn, cls, output_compartments=None, transition_key=None):
+    if output_compartments is None:
+        warn(f"Transition {pure_fn.__qualname__.split('.')[-1]} has no output_compartments and thus does not do anything")
+
+    key = pure_fn.__qualname__.split('.')[-1] if transition_key is None else transition_key
+
+    class_name = cls.__qualname__
+    add_component_resolver(class_name, key,
+                           (pure_fn, output_compartments))
+    add_resolver_meta(class_name, key,
+                      (None, None, None, True))
+
+
+def transition(pure_fn, output_compartments=None, transition_key=None):
+    def _transition(cls):
+        add_transition(pure_fn, cls, output_compartments, transition_key)
+        return cls
+    return _transition