def decorator(cls): orig_ufl_type(*args, **kwargs)(cls) classes.all_ufl_classes.add(cls) if cls._ufl_is_abstract_: classes.abstract_classes.add(cls) else: classes.ufl_classes.add(cls) if cls._ufl_is_terminal_: classes.terminal_classes.add(cls) else: classes.nonterminal_classes.add(cls) return cls
def decorator(cls): orig_ufl_type(*args, **kwargs)(cls) classes.all_ufl_classes.add(cls) if cls._ufl_is_abstract_: classes.abstract_classes.add(cls) else: classes.ufl_classes.add(cls) if cls._ufl_is_terminal_: classes.terminal_classes.add(cls) else: classes.nonterminal_classes.add(cls) return cls