#!/usr/bin/python

#run the following command
# python Testtopology  -t 100
# -t  simulation time


import sys
from subprocess import Popen, PIPE
from time import sleep
#import termcolor as T
import argparse
import os
import itertools
import random
import getopt

from mininet.util import dumpNodeConnections
from mininet.net import Mininet
from mininet.log import lg
from mininet.node import OVSKernelSwitch as Switch
from mininet.link import Link, TCLink
from mininet.util import *
from mininet.node import *
from mininet.topo import Topo
from mininet.cli  import *
from mininet.util import makeNumeric, custom

#topology
# C1-------------------------C2-------------------------------C3

class MyTopo( Topo ):
    "Simple topology example."

    def __init__( self ):
        "Create custom topo."

        # Initialize topology
        Topo.__init__( self )

        # Add hosts 
        A = self.addHost('A')
        B = self.addHost('B')
        C = self.addHost('C')
       
        
                
        # Add links
        self.addLink (A, B)
        self.addLink (B, C)
       

topos = { 'mytopo': ( lambda: MyTopo() ) }

def parse_args():
    parser = argparse.ArgumentParser(description="test")
    parser.add_argument('-t',
                        action="store",
                        help="Seconds to run the experiment",
                        default=2)
    args = parser.parse_args()                    
    args.t = float(args.t)
    return args
    
def progress(t):
    while t > 0:
        t -= 1
        sys.stdout.flush()
        sleep(1)
    print '\r\n'
    
def Config(net, args):
    A = net.getNodeByName('A')
    B = net.getNodeByName('B')
    C = net.getNodeByName('C')
   
    
    AB = A.intf(A.name+'-eth0')
    BA = B.intf(B.name+'-eth0')
    
    AB.config(bw=10, delay='10ms', loss=0.0, use_tbf=False)
    BA.config(bw=10, delay='10ms', loss=0.0, use_tbf=False)
    
    BC = B.intf(B.name+'-eth1')
    CB = C.intf(C.name+'-eth0')
    
    BC.config(bw=10, delay='10ms', loss=0.0, use_tbf=False)
    CB.config(bw=10, delay='10ms', loss=0.0, use_tbf=False)
    
    
    
    #Enter commands here 
    
    A.cmd('xterm&')
    A.cmd('xterm&')
    A.cmd('xterm&')
    B.cmd('xterm&')
    B.cmd('xterm&')
    C.cmd('xterm&')
    C.cmd('xterm&')
    C.cmd('xterm&')
   
    progress(int(args.t))
   
    
    #C3.sendInt()
    #A.sendInt()
    #B.sendInt()
    #C.sendInt()
    
    
 
def main(argv):
    args = parse_args()
    lg.setLogLevel('info')
    "Create and run experiment"
    topo = MyTopo()

    net = Mininet(topo=topo, link=TCLink, controller = OVSController)
    
    net.start()
    Config(net,args)
    #dumpNodeConnections(net.hosts)

    net.stop()

if __name__ == '__main__':
    main(sys.argv)