Pt2Pt.py revision 8322
1# Copyright (c) 2011 Advanced Micro Devices, Inc. 2# 2011 Massachusetts Institute of Technology 3# All rights reserved. 4# 5# Redistribution and use in source and binary forms, with or without 6# modification, are permitted provided that the following conditions are 7# met: redistributions of source code must retain the above copyright 8# notice, this list of conditions and the following disclaimer; 9# redistributions in binary form must reproduce the above copyright 10# notice, this list of conditions and the following disclaimer in the 11# documentation and/or other materials provided with the distribution; 12# neither the name of the copyright holders nor the names of its 13# contributors may be used to endorse or promote products derived from 14# this software without specific prior written permission. 15# 16# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 17# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 18# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 19# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 20# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 21# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 22# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 23# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 24# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 25# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 26# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 27# 28# Authors: Brad Beckmann 29# Tushar Krishna 30 31from m5.params import * 32from m5.objects import * 33 34class Pt2Pt(Topology): 35 description='Pt2Pt' 36 37def makeTopology(nodes, options, IntLink, ExtLink, Router): 38 # Create an individual router for each controller, and connect all to all. 39 pt2pt = Pt2Pt() 40 pt2pt.routers = [Router(router_id=i) for i in range(len(nodes))] 41 ext_links = [ExtLink(link_id=i, ext_node=n, int_node=pt2pt.routers[i]) 42 for (i, n) in enumerate(nodes)] 43 link_count = len(nodes) 44 45 int_links = [] 46 for i in xrange(len(nodes)): 47 for j in xrange(len(nodes)): 48 if (i != j): 49 link_count += 1 50 int_links.append(IntLink(link_id=link_count, 51 node_a=pt2pt.routers[i], 52 node_b=pt2pt.routers[j])) 53 54 pt2pt.ext_links = ext_links 55 pt2pt.int_links = int_links 56 57 return pt2pt 58