#!/usr/bin/python

"A FatTree network, using Brandon Heller's ripcord system."

import ripcord
from ripcord.topo import FatTreeTopo

from mininet.mininet import init, Controller, Network, Host, nameGen, Cli
from mininet.mininet import createLink, flush

class NoxController( Controller ):
   def __init__( self, name, kernel=False, **kwargs ):
      Controller.__init__( self, name, kernel=kernel,
         controller='nox_core', 
         cargs='-v --libdir=/usr/local/lib -i ptcp:', 
         cdir='/usr/local/bin', **kwargs)
   
class FatTree( Network ):
   "A customized Network that uses ripcord's FatTree."
   def __init__( self, depth, **kwargs ):
      self.depth = depth
      Network.__init__( self, **kwargs )
   def makeNet( self, controller ):
      ft = FatTreeTopo( self.depth )
      graph = ft.g
      switches = []
      hosts = []
      hostnames = nameGen( 'h' )
      switchnames = nameGen( 's' )
      dpnames = nameGen( 'nl:')
      graphToMini = {}
      miniToGraph = {}
      # Create nodes
      for graphNode in graph.nodes():
         isLeaf = len( graph.neighbors( graphNode ) ) == 1
         if isLeaf:
            mininetNode = Host( hostnames.next() )
            hosts += [ mininetNode ]
         else:
            mininetNode = self.Switch( switchnames.next(), dpnames.next() )
            switches += [ mininetNode ]
         print mininetNode.name, ; flush()
         miniToGraph[ mininetNode ] = graphNode
         graphToMini[ graphNode ] = mininetNode
      print
      print "*** Creating links"
      for switch in switches:
         currentNeighbors = [ switch.connection[ intf ][ 0 ] 
            for intf in switch.intfs ]
         for neighbor in graph.neighbors( miniToGraph[ switch ] ):
            miniNeighbor = graphToMini[ neighbor ]
            if miniNeighbor not in currentNeighbors:
               print '.', ; flush()
               createLink( switch, graphToMini[ neighbor ] )
      print
      return switches, hosts
      
if __name__ == '__main__':
   init()   
   network = FatTree( depth=4, kernel=True, Controller=NoxController)
   network.run( Cli )