#!/usr/bin/env python
|
#!/usr/bin/env python
|
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
|
|
#
|
#
|
# Support for graphical representation of NoC objects
|
# Support for graphical representation of NoC objects
|
#
|
#
|
# Author: Oscar Diaz
|
# Author: Oscar Diaz
|
# Version: 0.1
|
# Version: 0.1
|
# Date: 03-03-2011
|
# Date: 03-03-2011
|
|
|
#
|
#
|
# This code is free software; you can redistribute it and/or
|
# This code is free software; you can redistribute it and/or
|
# modify it under the terms of the GNU Lesser General Public
|
# modify it under the terms of the GNU Lesser General Public
|
# License as published by the Free Software Foundation; either
|
# License as published by the Free Software Foundation; either
|
# version 2.1 of the License, or (at your option) any later version.
|
# version 2.1 of the License, or (at your option) any later version.
|
#
|
#
|
# This code is distributed in the hope that it will be useful,
|
# This code is distributed in the hope that it will be useful,
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
# Lesser General Public License for more details.
|
# Lesser General Public License for more details.
|
#
|
#
|
# You should have received a copy of the GNU Lesser General Public
|
# You should have received a copy of the GNU Lesser General Public
|
# License along with this library; if not, write to the
|
# License along with this library; if not, write to the
|
# Free Software Foundation, Inc., 59 Temple Place, Suite 330,
|
# Free Software Foundation, Inc., 59 Temple Place, Suite 330,
|
# Boston, MA 02111-1307 USA
|
# Boston, MA 02111-1307 USA
|
#
|
#
|
|
|
#
|
#
|
# Changelog:
|
# Changelog:
|
#
|
#
|
# 03-03-2011 : (OD) initial release
|
# 03-03-2011 : (OD) initial release
|
#
|
#
|
|
|
"""
|
"""
|
================
|
================
|
NoCmodel Graphic utilities
|
NoCmodel Graphic utilities
|
================
|
================
|
|
|
This module declares functions to draw graphical representations of
|
This module declares functions to draw graphical representations of
|
NoCmodel objects.
|
NoCmodel objects.
|
|
|
"""
|
"""
|
try:
|
try:
|
import matplotlib.pyplot as plt
|
import matplotlib.pyplot as plt
|
has_matplotlib = True
|
has_matplotlib = True
|
except:
|
except:
|
print("Matplotlib package not found. Drawing functions will not work.")
|
print("Matplotlib package not found. Drawing functions will not work.")
|
has_matplotlib = False
|
has_matplotlib = False
|
|
|
import networkx as nx
|
import networkx as nx
|
|
import warnings
|
|
|
from noc_base import *
|
from noc_base import *
|
|
|
def draw_noc(noc, rectangular=True, nodepos=None):
|
def draw_noc(noc, rectangular=True, nodepos=None):
|
"""
|
"""
|
Draw a representation of a NoC
|
Draw a representation of a NoC
|
|
|
Arguments:
|
Arguments:
|
* noc: Model to draw
|
* noc: Model to draw
|
* rectangular: If True assumes rectangular layout, with routers
|
* rectangular: If True assumes rectangular layout, with routers
|
having coord_x and coord_y attributes. If false, expect router's
|
having coord_x and coord_y attributes. If false, expect router's
|
positions in nodepos argument
|
positions in nodepos argument
|
* nodepos: Optional dictionary where keys are router's indexes and values
|
* nodepos: Optional dictionary where keys are router's indexes and values
|
are tuples with x and y positions.
|
are tuples with x and y positions.
|
"""
|
"""
|
if not has_matplotlib:
|
if not has_matplotlib:
|
print("Function not available")
|
warnings.warn("Function not available")
|
return None
|
return None
|
|
|
# node positions
|
# node positions
|
if rectangular:
|
if rectangular:
|
if nodepos == None:
|
if nodepos == None:
|
nodepos = {}
|
nodepos = {}
|
for i in noc.router_list():
|
for i in noc.router_list():
|
nodepos[i.index] = (i.coord_x, i.coord_y)
|
nodepos[i.index] = (i.coord_x, i.coord_y)
|
else:
|
else:
|
if nodepos == None:
|
if nodepos == None:
|
raise ValueError("For non-rectangular layouts this function needs argument 'nodepos'")
|
raise ValueError("For non-rectangular layouts this function needs argument 'nodepos'")
|
|
|
# some parameters
|
# some parameters
|
ip_relpos = (-0.3, 0.3) # relative to router
|
ip_relpos = (-0.3, 0.3) # relative to router
|
# node labels
|
# node labels
|
nodelabels = {}
|
nodelabels = {}
|
for i in nodepos.iterkeys():
|
for i in nodepos.iterkeys():
|
nodelabels[i] = noc.node[i]["router_ref"].name
|
nodelabels[i] = noc.node[i]["router_ref"].name
|
|
|
# channel positions
|
# channel positions
|
chpos = {}
|
chpos = {}
|
for i in noc.channel_list():
|
for i in noc.channel_list():
|
ep = i.endpoints
|
ep = i.endpoints
|
ep_1x = nodepos[ep[0].index][0]
|
ep_1x = nodepos[ep[0].index][0]
|
ep_1y = nodepos[ep[0].index][1]
|
ep_1y = nodepos[ep[0].index][1]
|
ep_2x = nodepos[ep[1].index][0]
|
ep_2x = nodepos[ep[1].index][0]
|
ep_2y = nodepos[ep[1].index][1]
|
ep_2y = nodepos[ep[1].index][1]
|
thepos = (ep_2x + ((ep_1x - ep_2x)/2.0), ep_2y + ((ep_1y - ep_2y)/2.0))
|
thepos = (ep_2x + ((ep_1x - ep_2x)/2.0), ep_2y + ((ep_1y - ep_2y)/2.0))
|
chpos[i.index] = {"pos": thepos, "text": i.name}
|
chpos[i.index] = {"pos": thepos, "text": i.name}
|
|
|
# start drawing
|
# start drawing
|
nx.draw_networkx_nodes(noc, pos=nodepos, node_size=1000, node_color="blue", alpha=0.5)
|
nx.draw_networkx_nodes(noc, pos=nodepos, node_size=1000, node_color="blue", alpha=0.5)
|
nx.draw_networkx_edges(noc, pos=nodepos, edge_color="black", alpha=1.0, width=3.0)
|
nx.draw_networkx_edges(noc, pos=nodepos, edge_color="black", alpha=1.0, width=3.0)
|
nx.draw_networkx_labels(noc, pos=nodepos, labels=nodelabels, font_color="red")
|
nx.draw_networkx_labels(noc, pos=nodepos, labels=nodelabels, font_color="red")
|
|
|
ax=plt.gca()
|
ax=plt.gca()
|
# channel labels
|
# channel labels
|
for i in chpos.itervalues():
|
for i in chpos.itervalues():
|
ax.text(i["pos"][0], i["pos"][1], i["text"], horizontalalignment="center", verticalalignment="top", bbox=dict(facecolor='red', alpha=0.2))
|
ax.text(i["pos"][0], i["pos"][1], i["text"], horizontalalignment="center", verticalalignment="top", bbox=dict(facecolor='red', alpha=0.2))
|
|
|
# ipcore with channels and labels
|
# ipcore with channels and labels
|
for i in noc.ipcore_list():
|
for i in noc.ipcore_list():
|
thepos = nodepos[i.router_ref.index]
|
thepos = nodepos[i.router_ref.index]
|
# channel
|
# channel
|
ax.arrow(thepos[0], thepos[1], ip_relpos[0], ip_relpos[1])
|
ax.arrow(thepos[0], thepos[1], ip_relpos[0], ip_relpos[1])
|
# ip channel label
|
# ip channel label
|
ax.text(thepos[0]+(ip_relpos[0]/2), thepos[1]+(ip_relpos[1]/2), i.channel_ref.name, horizontalalignment="center", bbox=dict(facecolor='red', alpha=0.2))
|
ax.text(thepos[0]+(ip_relpos[0]/2), thepos[1]+(ip_relpos[1]/2), i.channel_ref.name, horizontalalignment="center", bbox=dict(facecolor='red', alpha=0.2))
|
# box with ipcore labels
|
# box with ipcore labels
|
ax.text(thepos[0]+ip_relpos[0], thepos[1]+ip_relpos[1], i.name, horizontalalignment="center", bbox=dict(facecolor='green', alpha=0.2))
|
ax.text(thepos[0]+ip_relpos[0], thepos[1]+ip_relpos[1], i.name, horizontalalignment="center", bbox=dict(facecolor='green', alpha=0.2))
|
|
|
|
# adjust axis TODO!
|
plt.show()
|
plt.show()
|
|
|
