Gabe Black has uploaded this change for review. (
https://gem5-review.googlesource.com/4849
Change subject: util: Make dot_writer ignore NULL simobjects.
......................................................................
util: Make dot_writer ignore NULL simobjects.
Also centralize the code which iterates over the parameters of a simnode's
children, and remove a copy/paste block of code in the DVFS block of code.
Change-Id: I01305fbff59346010780ee386ba97ad9106b1f5d
---
M src/python/m5/util/dot_writer.py
1 file changed, 33 insertions(+), 57 deletions(-)
diff --git a/src/python/m5/util/dot_writer.py
b/src/python/m5/util/dot_writer.py
index 501af51..1177ae7 100644
--- a/src/python/m5/util/dot_writer.py
+++ b/src/python/m5/util/dot_writer.py
@@ -59,13 +59,24 @@
import m5, os, re
from m5.SimObject import isRoot, isSimObjectVector
-from m5.params import PortRef
+from m5.params import PortRef, isNullPointer
from m5.util import warn
try:
import pydot
except:
pydot = False
+def simnode_children(simNode):
+ for child in simNode._children.itervalues():
+ if isNullPointer(child):
+ continue
+ if isSimObjectVector(child):
+ for obj in child:
+ if not isNullPointer(obj):
+ yield obj
+ else:
+ yield child
+
# need to create all nodes (components) before creating edges (memory
channels)
def dot_create_nodes(simNode, callgraph):
if isRoot(simNode):
@@ -88,14 +99,8 @@
cluster.add_node(port_node)
# recurse to children
- if simNode._children:
- for c in simNode._children:
- child = simNode._children[c]
- if isSimObjectVector(child):
- for obj in child:
- dot_create_nodes(obj, cluster)
- else:
- dot_create_nodes(child, cluster)
+ for child in simnode_children(simNode):
+ dot_create_nodes(child, cluster)
callgraph.add_subgraph(cluster)
@@ -115,14 +120,8 @@
dot_add_edge(simNode, callgraph, full_port_name, p)
# recurse to children
- if simNode._children:
- for c in simNode._children:
- child = simNode._children[c]
- if isSimObjectVector(child):
- for obj in child:
- dot_create_edges(obj, callgraph)
- else:
- dot_create_edges(child, callgraph)
+ for child in simnode_children(simNode):
+ dot_create_edges(child, callgraph)
def dot_add_edge(simNode, callgraph, full_port_name, peerPort):
if peerPort.role == "MASTER":
@@ -299,48 +298,25 @@
dvfs_domains = {}
# recurse to children
- if simNode._children:
- for c in simNode._children:
- child = simNode._children[c]
- if isSimObjectVector(child):
- for obj in child:
- try:
- c_dom = obj.__getattr__('clk_domain')
- v_dom = c_dom.__getattr__('voltage_domain')
- except AttributeError:
- # Just re-use the domain from above
- c_dom = domain
- v_dom = c_dom.__getattr__('voltage_domain')
- pass
+ for child in simnode_children(simNode):
+ try:
+ c_dom = obj.__getattr__('clk_domain')
+ v_dom = c_dom.__getattr__('voltage_domain')
+ except AttributeError:
+ # Just re-use the domain from above
+ c_dom = domain
+ v_dom = c_dom.__getattr__('voltage_domain')
+ pass
- if c_dom == domain or c_dom == None:
- dot_create_dvfs_nodes(obj, cluster, domain)
- else:
- if c_dom not in dvfs_domains:
- dvfs_cluster = dot_add_clk_domain(c_dom, v_dom)
- dvfs_domains[c_dom] = dvfs_cluster
- else:
- dvfs_cluster = dvfs_domains[c_dom]
- dot_create_dvfs_nodes(obj, dvfs_cluster, c_dom)
+ if c_dom == domain or c_dom == None:
+ dot_create_dvfs_nodes(child, cluster, domain)
+ else:
+ if c_dom not in dvfs_domains:
+ dvfs_cluster = dot_add_clk_domain(c_dom, v_dom)
+ dvfs_domains[c_dom] = dvfs_cluster
else:
- try:
- c_dom = child.__getattr__('clk_domain')
- v_dom = c_dom.__getattr__('voltage_domain')
- except AttributeError:
- # Just re-use the domain from above
- c_dom = domain
- v_dom = c_dom.__getattr__('voltage_domain')
- pass
-
- if c_dom == domain or c_dom == None:
- dot_create_dvfs_nodes(child, cluster, domain)
- else:
- if c_dom not in dvfs_domains:
- dvfs_cluster = dot_add_clk_domain(c_dom, v_dom)
- dvfs_domains[c_dom] = dvfs_cluster
- else:
- dvfs_cluster = dvfs_domains[c_dom]
- dot_create_dvfs_nodes(child, dvfs_cluster, c_dom)
+ dvfs_cluster = dvfs_domains[c_dom]
+ dot_create_dvfs_nodes(child, dvfs_cluster, c_dom)
for key in dvfs_domains:
cluster.add_subgraph(dvfs_domains[key])
--
To view, visit https://gem5-review.googlesource.com/4849
To unsubscribe, or for help writing mail filters, visit
https://gem5-review.googlesource.com/settings
Gerrit-Project: public/gem5
Gerrit-Branch: master
Gerrit-MessageType: newchange
Gerrit-Change-Id: I01305fbff59346010780ee386ba97ad9106b1f5d
Gerrit-Change-Number: 4849
Gerrit-PatchSet: 1
Gerrit-Owner: Gabe Black <[email protected]>
_______________________________________________
gem5-dev mailing list
[email protected]
http://m5sim.org/mailman/listinfo/gem5-dev
