예제 #1
0
 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
예제 #2
0
 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