#
# Copyright (c) @ Regents of the University of California.
# All rights reserved.
# 
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
# 1. Redistributions of source code must retain the above copyright
#    notice, this list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright
#    notice, this list of conditions and the following disclaimer in the
#    documentation and/or other materials provided with the distribution.
# 3. All advertising materials mentioning features or use of this software
#    must display the following acknowledgement:
# 	This product includes software developed by the MASH Research
# 	Group at the University of California Berkeley.
# 4. Neither the name of the University nor of the Research Group may be
#    used to endorse or promote products derived from this software without
#    specific prior written permission.
# 
# THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
# SUCH DAMAGE.
#
# @(#) $Header: /usr/src/mash/repository/vint/ns-2/tcl/ex/scuba/complete/demo.tcl,v 1.1 1997/06/22 19:04:24 elan Exp $
#

set tcldir ../../../

source $tcldir/rtp/session-scuba.tcl
source $tcldir/rtp/session-rtp.tcl

proc flash_annotate { start duration msg } {
	global ns
	$ns at $start "trace_annotate {$msg}"
	$ns at [expr $start+$duration] "trace_annotate {}"
}

set ns [new MultiSim]

for { set i 0 } { $i < 8 } { incr i } {
	set node($i) [$ns node]
}

set f [open out.tr w]
$ns trace-all $f

Queue set limit_ 8

proc makelinks { bw delay pairs } {
	global ns node
	foreach p $pairs {
		set src $node([lindex $p 0])
		set dst $node([lindex $p 1])
		$ns duplex-link $src $dst $bw $delay DropTail
	}
}

makelinks 1.5Mb 10ms {
	{ 0 3 }
	{ 1 3 }
	{ 2 3 }
	{ 4 7 }
	{ 5 7 }
	{ 6 7 }
}

makelinks 400kb 50ms {
	{ 3 7 }
}

for { set i 0 } { $i < 8 } { incr i } {
	set dm($i) [new DM $ns $node($i)]
}

$ns at 0.0 "$ns run-mcast"

set sessbw 400kb/s

foreach n { 0 1 2 4 5 6 } {
	set sess($n) [new Session/RTP]
	$sess($n) session_bw $sessbw
	$sess($n) attach-node $node($n)
}

$ns at 0.1 {
	global sess
	foreach n { 0 1 2 4 5 6 } {
		$sess($n) join-group 0x8000
	}
}

foreach n { 0 1 2 } {
	set d [$sess($n) set dchan_]
	$d set class_ [expr $n]
}

flash_annotate 0 0.20 "Equal bandwidth - Saturate"
flash_annotate 0.25 0.1 "Starting Receivers..."

# start receivers
$ns at 0.25 {
	global sess
	$sess(4) start
	$sess(5) start
	$sess(6) start
}

# start senders
flash_annotate 0.5 0.2 "Starting sender 1 at 200kb/s..."
$ns at 0.5 "$sess(0) start"
$ns at 0.5 "$sess(0) transmit 200kb/s"

flash_annotate 1.0 0.2 "Starting sender 2 at 200kb/s..."
$ns at 1.0 "$sess(1) start"
$ns at 1.0 "$sess(1) transmit 200kb/s"

flash_annotate 1.5 0.2 "Starting sender 3 at 200kb/s..."
$ns at 1.5 "$sess(2) start"
$ns at 1.5 "$sess(2) transmit 200kb/s"

$ns at 1.75 {
	global sess
	foreach n { 0 1 2 } {
		$sess($n) stop
	}
}

$ns at 2.0 "trace_annotate {End}"


$ns at 2.5 "trace_annotate {Equal Bandwidth - No Saturation}"

# start senders
flash_annotate 3.0 0.2 "Starting sender 1 at 130kb/s..."
$ns at 3.0 "$sess(0) start"
$ns at 3.0 "$sess(0) transmit 130kb/s"

flash_annotate 3.5 0.2 "Starting sender 2 at 130kb/s..."
$ns at 3.5 "$sess(1) start"
$ns at 3.5 "$sess(1) transmit 130kb/s"

flash_annotate 4.0 0.2 "Starting sender 3 at 130kb/s..."
$ns at 4.0 "$sess(2) start"
$ns at 4.0 "$sess(2) transmit 130kb/s"

$ns at 4.5 {
	global sess
	foreach n { 0 1 2 } {
		$sess($n) stop
	}
}

$ns at 5.0 "trace_annotate {End}"


set scuba_sim_time 5.5
$ns at $scuba_sim_time "scuba_sim $scuba_sim_time"
$ns at [expr $scuba_sim_time + 4.2] "finish"

proc scuba_sim { start } {
	global ns sess node sessbw

	flash_annotate $start 0.2 {SCUBA}
	foreach n { 0 1 2 4 5 6 } {
		set sess($n) [new Session/RTP/Scuba]
		$sess($n) session_bw $sessbw
		$sess($n) attach-node $node($n)
	}

	set t [expr $start+0.5]
	$ns at $t {
		global sess
		foreach n { 0 1 2 4 5 6 } {
			$sess($n) join-group 0x8000
		}
	}

	foreach n { 0 1 2 } {
		set d [$sess($n) set dchan_]
		$d set class_ [expr $n]
	}

	flash_annotate $t 0.2 {Starting receivers...}
	# start receivers
	$ns at $t {
		global sess
		$sess(4) start 0
		$sess(5) start 0
		$sess(6) start 0
	}

	# start senders
	flash_annotate [expr $start+1.1] 0.05 "Starting sender 1..."
	$ns at [expr $start+1.1] "$sess(0) start 1"

	flash_annotate [expr $start+1.2] 0.05 "Starting sender 2..."
	$ns at [expr $start+1.2] "$sess(1) start 1"

	flash_annotate [expr $start+1.3] 0.05 "Starting sender 3..."
	$ns at [expr $start+1.3] "$sess(2) start 1"

	# 4 focus on 0
	flash_annotate [expr $start+1.5] 0.1 "4 focussing on 0..."
	$ns at [expr $start+1.5] "[$sess(4) set repAgent_] set class_ 4"
	$ns at [expr $start+1.5] "$sess(4) scuba_focus $sess(0)"
	$ns at [expr $start+1.7] "[$sess(4) set repAgent_] set class_ 33"

	# 5 focus on 1
	flash_annotate [expr $start+2.0] 0.1 "5 focussing on 1..."
	$ns at [expr $start+2.0] "[$sess(5) set repAgent_] set class_ 4"
	$ns at [expr $start+2.0] "$sess(5) scuba_focus $sess(1)"
	$ns at [expr $start+2.2] "[$sess(5) set repAgent_] set class_ 33"

	# 5 focus on 0
	flash_annotate [expr $start+2.5] 0.1 "5 focussing on 0..."
	$ns at [expr $start+2.5] "[$sess(5) set repAgent_] set class_ 4"
	$ns at [expr $start+2.5] "$sess(5) scuba_focus $sess(0)"
	$ns at [expr $start+2.7] "[$sess(5) set repAgent_] set class_ 33"

	# 5 unfocus on 0
	flash_annotate [expr $start+3.0] 0.1 "5 unfocussing off 0..."
	$ns at [expr $start+3.0] "[$sess(5) set repAgent_] set class_ 4"
	$ns at [expr $start+3.0] "$sess(5) scuba_unfocus $sess(0)"
	$ns at [expr $start+3.2] "[$sess(5) set repAgent_] set class_ 33"
}

proc finish {} {
	trace_annotate "End Simulation"
	global tcldir
	puts "converting output to nam format..."
        global ns
        $ns flush-trace
	exec awk -f $tcldir/nam-demo/nstonam.awk out.tr > demo-nam.tr
	exec rm -f out
        #XXX
	puts "running nam..."
	exec /usr/local/src/nam/nam demo-nam &
        exit 0
}

$ns run