From: Sascha Bischoff <sascha.bischoff@arm.com>
Date: Wed, 23 Apr 2014 09:18:25 +0000 (-0400)
Subject: misc: Proper type check and import for PortRef
X-Git-Url: https://git.libre-soc.org/?a=commitdiff_plain;h=2031c03c09ad3753e53fa29a6555fdf288c821e2;p=gem5.git

misc: Proper type check and import for PortRef

Rewriting the type checking around PortRef, which was interacting strangely
with other Python scripts.

Tested-by: stephan.diestelhorst@arm.com
---

diff --git a/src/python/m5/util/dot_writer.py b/src/python/m5/util/dot_writer.py
index a6f721a0c..ab4687f7f 100644
--- a/src/python/m5/util/dot_writer.py
+++ b/src/python/m5/util/dot_writer.py
@@ -58,6 +58,7 @@
 
 import m5, os, re
 from m5.SimObject import isRoot, isSimObjectVector
+from m5.params import PortRef
 from m5.util import warn
 try:
     import pydot
@@ -106,7 +107,7 @@ def dot_create_edges(simNode, callgraph):
             full_port_name = full_path + "_" + port_name
             port_node = dot_create_node(simNode, full_port_name, port_name)
             # create edges
-            if type(port) is m5.params.PortRef:
+            if isinstance(port, PortRef):
                 dot_add_edge(simNode, callgraph, full_port_name, port)
             else:
                 for p in port.elements: