Subversion Repositories nocmodel

#!/usr/bin/env python

# -*- coding: utf-8 -*-

#

# Helper functions for NoC Objects

#   Functions that easily construct predefined NoC structures and objects

#

# Author:  Oscar Diaz

# Version: 0.2

# Date:    14-03-2011

#

# This code is free software; you can redistribute it and/or

# modify it under the terms of the GNU Lesser General Public

# License as published by the Free Software Foundation; either

# 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,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

# Lesser General Public License for more details.

#

# You should have received a copy of the GNU Lesser General Public

# License along with this library; if not, write to the

# Free Software  Foundation, Inc., 59 Temple Place, Suite 330,

# Boston, MA  02111-1307  USA

#

#

# Changelog:

#

# 14-03-2011 : (OD) initial release

#

import networkx as nx

from noc_base import *

def generate_squarenoc(orderx=3, ordery=None, with_ipcore=False):

    """

    NoC generator helper: generates a 2D grid NoC object

    Arguments

    * orderx: optional X-axis length of the grid. By default build a 3x3 square

      grid.

    * ordery: optional Y-axis length of the grid. By default build a square grid

      of order "orderx". This argument is used to build rectangular grids.

    * with_ipcore: If True, add ipcores to routers automatically.

    """

    if ordery == None:

        ordery = orderx

    # 1. generate a 2d grid

    basegrid = nx.grid_2d_graph(orderx, ordery)

    # 2. convert to a graph with ints as nodes

    convgrid = nx.Graph()

    for n in basegrid.nodes_iter():

        n2 = n[0] + n[1]*orderx

        convgrid.add_node(n2, coord_x=n[0], coord_y=n[1])

    for e in basegrid.edges_iter():

        e1 = e[0][0] + e[0][1]*orderx

        e2 = e[1][0] + e[1][1]*orderx

        convgrid.add_edge(e1, e2)

    nocbase = noc(name="NoC grid %dx%d" % (orderx, ordery), data=convgrid)

    # 2. for each node add router object

    for n in nocbase.nodes_iter():

        cx = nocbase.node[n]["coord_x"]

        cy = nocbase.node[n]["coord_y"]

        r = nocbase._add_router_from_node(n, coord_x=cx, coord_y=cy)

        if with_ipcore:

            nocbase.add_ipcore(r)

    # 3. for each edge add channel object

    for e in nocbase.edges_iter():

        nocbase._add_channel_from_edge(e)

    return nocbase