forked from andikleen/pmu-tools
-
Notifications
You must be signed in to change notification settings - Fork 0
/
node.py
62 lines (49 loc) · 1.88 KB
/
node.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
# Helper classes and functions for nodes
# Decorator class to declare reference dependecies between classes
class requires(object):
"""Decorator to mark required references. These references will
be added to the object as instance attributes. Example:
@requires("ref1", "ref2")
class SomeClass(object):
def some_method(self):
return self.ref1 self.ref2
"""
def __init__(self, *required_refs):
self.required_refs = required_refs
def __call__(self, cls):
setattr(cls, "required_refs", self.required_refs)
return cls
def set_parent(parent, nodes):
for node in nodes:
node.parent = parent
# Check that all required references are set
def check_refs(fn):
"""Decorator to check if required references for an object
are set. If it finds missing references, it will raise an
exception. Example:
@requires("retiring", "bad_speculation", "frontend_bound")
class BackendBound(object):
@check_refs
def _compute(self, ev):
# checks if required refs are set before executing
"""
def wrapped(self, *args, **kwargs):
if not hasattr(self, "required_refs"):
raise Exception("Missing required_refs object")
missing_refs = [ref for ref in self.required_refs
if not hasattr(self, ref)]
if missing_refs:
raise Exception("Missing references: {0}".format(missing_refs))
return fn(self, *args, **kwargs)
wrapped.__name__ = fn.__name__
return wrapped
def add_references(node, **refs):
"""Adds an attribute to node, as specified in the **refs argument.
Example:
...
backend = BackendBound()
add_references(backend, retiring=retiring, frontend_bound=frontend,
bad_speculation=bad_speculation)
"""
for name, obj in refs.items():
setattr(node, name, obj)