Pt2Pt.py revision 9100
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 34from BaseTopology import BaseTopology 35 36class Pt2Pt(BaseTopology): 37 description='Pt2Pt' 38 39 def __init__(self, controllers): 40 self.nodes = controllers 41 42 def makeTopology(self, options, IntLink, ExtLink, Router): 43 nodes = self.nodes 44 # Create an individual router for each controller, and connect all to all. 45 46 routers = [Router(router_id=i) for i in range(len(nodes))] 47 ext_links = [ExtLink(link_id=i, ext_node=n, int_node=routers[i]) 48 for (i, n) in enumerate(nodes)] 49 link_count = len(nodes) 50 51 int_links = [] 52 for i in xrange(len(nodes)): 53 for j in xrange(len(nodes)): 54 if (i != j): 55 link_count += 1 56 int_links.append(IntLink(link_id=link_count, 57 node_a=routers[i], 58 node_b=routers[j])) 59 60 return routers, int_links, ext_links 61