#
 # This code was developed as part of the MIT uAMPS project. (June, 2000)
 #
 
 
global opt bs
 
source $env(RCA_LIBRARY)/ns-ranode.tcl
 source $env(uAMPS_LIBRARY)/ns-bsapp.tcl
 source $env(uAMPS_LIBRARY)/extras.tcl
 #source $env(uAMPS_LIBRARY)/stats.tcl
 #Uncomment these lines to use gdb to debug the c code
 #source mit/uAMPS/ns-bsapp.tcl
 #source mit/uAMPS/extras.tcl
 source $env(uAMPS_LIBRARY)/stats.tcl
 source $env(RCA_LIBRARY)/resources/ns-resource-manager.tcl
 source $env(RCA_LIBRARY)/resources/ns-energy-resource.tcl
 source $env(RCA_LIBRARY)/resources/ns-neighbor-resource.tcl
 
# ========================================================================
 # Default Script Options
 # ========================================================================
 
set opt(bsapp) "Application/BSApp" ;# BS application type
 set opt(mtype) "" ;# No meta-data used 
set opt(nn_**) [expr $opt(nn) - 1] ;# Number of non-BS nodes
 set opt(bsID) $opt(nn_) ;# BS node number
 set opt(bsCode) 0 ;# Spreading code for BS
 set opt(quiet) 1 ;# 0=print info, 1=quiet
 
set opt(bw) 1e6 ;# 1 Mbps radio speed
 set opt(delay) 1e-12 ;# Links delay
 set opt(prop_speed) 3e8; ;# Meters per second
 set opt(ll) RCALinkLayer ;# Arpless link-layer
 set opt(mac) Mac/Sensor ;# Sensor mac protocol
 set opt(ifq) Queue/DropTail ;# DropTail Q
 set opt(ifqlen) 100 ;# Max packets in ifq
 set opt(netif) Phy/WirelessPhy ;# Wireless channel
 set opt(ant) Antenna/OmniAntenna ;# Omnidirectional antena
 
# Time required to transmit numbytes bytes of data
 proc TxTime {numbytes} {
 global opt
 return [expr $numbytes*8/$opt(bw)]
 }
 
set opt(hdr_size) 25 ;# Bytes for header
 set opt(sig_size) 500 ;# Bytes for data signal
 # Packet transmission time
 set opt(slot_time) [expr [TxTime [expr $opt(sig_size)+$opt(hdr_size)]]]
 # Spread-spectrum packet transmission time
 set opt(ss_slot_time) [expr $opt(slot_time) * $opt(spreading)]
 # Maximum TDMA frame time (if all nodes in one cluster)
 set opt(frame_time) [expr $opt(ss_slot_time) * $opt(nn_)]
 
set opt(ch_change) [expr 10 * $opt(init_energy)] ;# Time for each round
 set opt(check_energy) 10 ;# Time btwn energy traces 

set opt(freq) 914e+6 ;# Carrier frequency
 set opt(L) 1.0 ;# System (non-propogation) loss
 set opt(Gt) 1.0 ;# Tx antenna gain
 set opt(Gr) 1.0 ;# Rx antenna gain
 set opt(ht) 1.5 ;# Antenna height
 set opt(CSThresh) 1e-9 ;# Receive threshold is 1 nW
 set opt(RXThresh) 6e-9 ;# Success threshold is 6 nW
 set PI 3.1415926
 set l [expr 3e8 / $opt(freq)] ;# Wavelength of carrier
 

 #
 # Energy Models
 #

 
# Efriss_amp = RXThresh * (4pi)^2 / (Rb Gt Gr lambda^2)
 set opt(Efriss_amp) [expr [expr 1.1 * $opt(RXThresh) * 16 * $PI * $PI] / \
 [expr $opt(bw) * $opt(Gt) * $opt(Gr) * $l * $l]]
 # Etwo_ray_amp = RXThresh / (Rb Gt Gr ht^2 hr^2)
 set opt(Etwo_ray_amp) [expr 1.1 * $opt(RXThresh) / \
 [expr $opt(bw) * $opt(Gt) * $opt(Gr) * \
 $opt(ht) * $opt(ht) * $opt(ht) * $opt(ht)]]
 set opt(EXcvr) 50e-9 ;# Energy for radio circuitry
 set opt(e_bf) 5e-9 ;# Beamforming energy (J/bit)
 set opt(Esense) 0 ;# Sensing energy (J/bit)
 set opt(thresh_energy) 0.00 ;# Threshold for power adaptation
 set opt(Pidle) 0 ;# Idle power (W)
 set opt(Psleep) 0 ;# Sleep power (W)
 

# ===== Get rid of the warnings in bind ================================
 Resource/Energy set energyLevel_ $opt(init_energy)
 Resource/Energy set alarmLevel_ $opt(thresh_energy)
 Resource/Energy set expended_ 0
 
Agent/RCAgent set sport_ 0
 Agent/RCAgent set dport_ 0
 Agent/RCAgent set packetMsg_ 0
 Agent/RCAgent set distEst_ 0
 Agent/RCAgent set packetSize_ 0
 Agent/BSAgent set packetMsg_ 0
 Agent/BSAgent set packetSize_ 0
 Agent/BSAgent set recv_code_ 0
 
RCALinkLayer set delay_ 25us
 RCALinkLayer set bandwidth_ 0 
RCALinkLayer set off_prune_ 0 
RCALinkLayer set off_CtrMcast_ 0 
RCALinkLayer set macDA_ 0 
RCALinkLayer set debug_ 0 
RCALinkLayer set avoidReordering_ 0 

Phy/WirelessPhy set bandwidth_ $opt(bw)
 Phy/WirelessPhy set CSThresh_ $opt(CSThresh)
 Phy/WirelessPhy set RXThresh_ $opt(RXThresh)
 Phy/WirelessPhy set Efriss_amp_ $opt(Efriss_amp)
 Phy/WirelessPhy set Etwo_ray_amp_ $opt(Etwo_ray_amp)
 Phy/WirelessPhy set EXcvr_ $opt(EXcvr)
 Phy/WirelessPhy set freq_ $opt(freq)
 Phy/WirelessPhy set L_ $opt(L)
 Phy/WirelessPhy set sleep_ 0
 Phy/WirelessPhy set alive_ 1
 Phy/WirelessPhy set ss_ $opt(spreading)
 Phy/WirelessPhy set dist_ 0
 
Antenna/OmniAntenna set Gt_ $opt(Gt)
 Antenna/OmniAntenna set Gr_ $opt(Gr)
 Antenna/OmniAntenna set Z_ $opt(ht)
 
set MacTrace OFF
 Mac set bandwidth_ $opt(bw)
 Mac/Sensor set code_ 0
 Mac/Sensor set node_num_ 0
 Mac/Sensor set ss_ $opt(spreading) 
Mac/Sensor set CHheard_ 0
 Mac/Sensor set myADVnum_ 0
 
set bs[list $opt(bs_x) $opt(bs_y)]
 set BS_NODE 1
 
set outf [open "$opt(dirname)/conditions.txt" a]
 puts $outf "Simulation will stop after $opt(stop) seconds."
 puts $outf "Base station at ($opt(bs_x), $opt(bs_y))"
 if {$opt(eq_energy) == 1} {
 puts $outf "Each node starting with $opt(init_energy) Joules of energy.\n"
 }
 puts $outf "Energy Model:"
 puts $outf "\t\tRXThresh = $opt(RXThresh)"
 puts $outf "\t\tCSThresh = $opt(CSThresh)"
 puts $outf "\t\tRb = $opt(bw)"
 puts $outf "\t\tExcvr = $opt(EXcvr)"
 puts $outf "\t\tEfriss_amp = $opt(Efriss_amp)"
 puts $outf "\t\tEtwo_ray_amp = $opt(Etwo_ray_amp)"
 puts $outf "\t\tEbf = $opt(e_bf)"
 puts $outf "\t\tPidle = $opt(Pidle)"
 puts $outf "\t\tPsleep = $opt(Psleep)\n"
 close $outf
 
set initialized 0
 set rng_ [new RNG]
 
proc leach-create-mobile-node { id } {
 global ns_ chan prop topo tracefd opt node_ 
global initialized BS_NODE rng_
 
if {$initialized == 0} {
 sens_init
 set initialized 1
 }
 
# Create nodes.
 if {$id != $opt(nn_)} {
 puts -nonewline "$id "
 set node_($id) [new MobileNode/ResourceAwareNode]
 } else {
 puts "($opt(nn_) == BS)"
 set node_($id) [new MobileNode/ResourceAwareNode $BS_NODE]
 }
 
set node $node_($id)
 if {$id != $opt(nn_)} {
 # Set initial node energy.
 if {$opt(eq_energy) == 1} {
 $node set-energy $opt(init_energy) $opt(thresh_energy)
 } else {
 #set E [$rng_ uniform $opt(lower_e) $opt(upper_e)]
 #set rn [$rng_ uniform 0 1]
 #if {$rn 0.1} {
 # set E 200
 #} else {
 # set E 2
 #}
 set high_e_nodes[list 97 19 12 87 8 22 83 55 34 72]
 if {[lsearch $high_e_nodes $id] == -1} {
 set E 2
 } else {
 set E 200
 }
 $node set-energy $E $opt(thresh_energy)
 set initf [open "$opt(dirname)/init.energy" a]
 puts $initf "$id\t$E"
 close $initf
 }
 } else {
 # Base station has an infinite amount of energy.
 $node set-energy 50000 $opt(thresh_energy)
 }
 
# Disable random motion.
 $node random-motion 0 
$node topography $topo
 
if ![info exist inerrProc_] {
 set inerrProc_ ""
 }
 if ![info exist outerrProc_] {
 set outerrProc_ ""
 }
 if ![info exist FECProc_] {
 set FECProc_ ""
 }
 
# Connect the node to the channel.
 $node add-interface $chan $prop $opt(ll) $opt(mac) \
 $opt(ifq) $opt(ifqlen) $opt(netif) $opt(ant) \
 $topo $inerrProc_ $outerrProc_ $FECProc_
 
# Set up the trace target.
 set T [new Trace/Generic]
 $T target [$ns_ set nullAgent_]
 $T attach $tracefd
 $T set src_ $id
 $node log-target $T
 
$ns_ at 0.0 "$node_($id) start-app"
 }
 
proc sens_init {} {
 
global ns_ opt ns
 
# The timer code has hard-coded the global variable ns to
 # be the simulator.
 set ns $ns_
 
# Remove old trace files.
 catch "eval exec rm [glob -nocomplain $opt(dirname)/TDMAschedule.*.txt]"
 catch "exec rm $opt(dirname)/$opt(filename).energy"
 catch "exec rm $opt(dirname)/$opt(filename).data"
 catch "exec rm $opt(dirname)/$opt(filename).alive"
 catch "exec rm $opt(dirname)/startup.energy"
 catch "exec rm $opt(dirname)/init.energy"
 
puts "Creating sensor nodes..."

 sens_init_stats "$opt(dirname)/$opt(filename)"
 
$ns_ at $opt(stop) "sens_finish"
 # Start logging simulation statistics.
 $ns_ at $opt(check_energy) "sens_gather_stats"
 }

luuh@ubuntu:~$ cd ns-allinone-2.34/ns-2.34/mit/
luuh@ubuntu:~/ns-allinone-2.34/ns-2.34/mit$ ns leacttcl.tcl
syntax error in expression "$i  $argc": extra tokens at end of expression
    ("for" test expression)
    while compiling
"for {set i 0} {$i  $argc} {incr i} {
		set arg [lindex $argv $i]
		if {[string range $arg 0 0] != "-"} continue

		set name [string range $arg 1 end]
..."
    (compiling body of proc "getopt", line 5)
    invoked from within
"getopt $argc $argv"
    (file "leacttcl.tcl" line 180)
luuh@ubuntu:~/ns-allinone-2.34/ns-2.34/mit$

# Copyright (c) 1997 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 Computer Systems
#      Engineering Group at Lawrence Berkeley Laboratory.
# 4. Neither the name of the University nor of the Laboratory 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/wireless.tcl,v 1.2 1999/02/24 23:27:34 haldar Exp $
#
# Ported from CMU/Monarch's code, nov'98 -Padma.

# ======================================================================
# Default Script Options
# ======================================================================
set opt(chan)		Channel/WirelessChannel
set opt(prop)		Propagation/TwoRayGround
#set opt(netif)		NetIf/SharedMedia
set opt(netif)		Phy/WirelessPhy
#set opt(mac)		Mac/802_11
set opt(mac)		Mac/802_11
set opt(ifq)		Queue/DropTail/PriQueue
set opt(ll)		LL
set opt(ant)            Antenna/OmniAntenna

set opt(x)		0	;# X dimension of the topography
set opt(y)		0		;# Y dimension of the topography
#set opt(cp)		"../mobility/scene/cbr-50-20-4-512" ;# connection pattern file
set opt(cp)		""
set opt(sc)		"../mobility/scene/scen-670x670-50-600-20-2" ;# scenario file


set opt(ifqlen)		50		;# max packet in ifq
set opt(nn)		51		;# number of nodes
set opt(seed)		0.0
set opt(stop)		10.0		;# simulation time
set opt(tr)		out.tr		;# trace file
set opt(rp)             dsdv            ;# routing protocol script
set opt(lm)             "on"           ;# log movement

# ======================================================================

set AgentTrace			ON
set RouterTrace			ON
set MacTrace			OFF

LL set mindelay_		50us
LL set delay_			25us
LL set bandwidth_		0	;# not used
LL set off_prune_		0	;# not used
LL set off_CtrMcast_		0	;# not used

Agent/Null set sport_		0
Agent/Null set dport_		0

Agent/CBR set sport_		0
Agent/CBR set dport_		0

Agent/TCPSink set sport_	0
Agent/TCPSink set dport_	0

Agent/TCP set sport_		0
Agent/TCP set dport_		0
Agent/TCP set packetSize_	1460

Queue/DropTail/PriQueue set Prefer_Routing_Protocols    1

# unity gain, omni-directional antennas
# set up the antennas to be centered in the node and 1.5 meters above it
Antenna/OmniAntenna set X_ 0
Antenna/OmniAntenna set Y_ 0
Antenna/OmniAntenna set Z_ 1.5
Antenna/OmniAntenna set Gt_ 1.0
Antenna/OmniAntenna set Gr_ 1.0

# Initialize the SharedMedia interface with parameters to make
# it work like the 914MHz Lucent WaveLAN DSSS radio interface
Phy/WirelessPhy set CPThresh_ 10.0
Phy/WirelessPhy set CSThresh_ 1.559e-11
Phy/WirelessPhy set RXThresh_ 3.652e-10
Phy/WirelessPhy set Rb_ 2*1e6
Phy/WirelessPhy set Pt_ 0.2818
Phy/WirelessPhy set freq_ 914e+6 
Phy/WirelessPhy set L_ 1.0

# ======================================================================

proc usage { argv0 }  {
	puts "Usage: $argv0"
	puts "\tmandatory arguments:"
	puts "\t\t\[-x MAXX\] \[-y MAXY\]"
	puts "\toptional arguments:"
	puts "\t\t\[-cp conn pattern\] \[-sc scenario\] \[-nn nodes\]"
	puts "\t\t\[-seed seed\] \[-stop sec\] \[-tr tracefile\]\n"
}


proc getopt {argc argv} {
	global opt
	lappend optlist cp nn seed sc stop tr x y

	for {set i 0} {$i  $argc} {incr i} {
		set arg [lindex $argv $i]
		if {[string range $arg 0 0] != "-"} continue

		set name [string range $arg 1 end]
		set opt($name) [lindex $argv [expr $i+1]]
	}
}


proc cmu-trace { ttype atype node } {
	global ns_ tracefd

	if { $tracefd == "" } {
		return ""
	}
	set T [new CMUTrace/$ttype $atype]
	$T target [$ns_ set nullAgent_]
	$T attach $tracefd
        $T set src_ [$node id]

        $T node $node

	return $T
}


proc create-god { nodes } {
	global ns_ god_ tracefd

	set god_ [new God]
	$god_ num_nodes $nodes
}

proc log-movement {} {
    global logtimer ns_ ns

    set ns $ns_
    source tcl/mobility/timer.tcl
    Class LogTimer -superclass Timer
    LogTimer instproc timeout {} {
	global opt node_;
	for {set i 0} {$i  $opt(nn)} {incr i} {
	    $node_($i) log-movement
	}
	$self sched 0.1
    }

    set logtimer [new LogTimer]
    $logtimer sched 0.1
}

# ======================================================================
# Main Program
# ======================================================================
getopt $argc $argv

#
# Source External TCL Scripts
#
source tcl/lib/ns-mobilenode.tcl

#if { $opt(rp) != "" } {
	source tcl/mobility/$opt(rp).tcl
	#} elseif { [catch { set env(NS_PROTO_SCRIPT) } ] == 1 } {
	#puts "\nenvironment variable NS_PROTO_SCRIPT not set!\n"
	#exit
#} else {
	#puts "\n*** using script $env(NS_PROTO_SCRIPT)\n\n";
        #source $env(NS_PROTO_SCRIPT)
#}
source tcl/lib/ns-cmutrace.tcl

# do the get opt again incase the routing protocol file added some more
# options to look for
getopt $argc $argv

if { $opt(x) == 0 || $opt(y) == 0 } {
	usage $argv0
	exit 1
}

if {$opt(seed) > 0} {
	puts "Seeding Random number generator with $opt(seed)\n"
	ns-random $opt(seed)
}

#
# Initialize Global Variables
#
set ns_		[new Simulator]
set chan	[new $opt(chan)]
set prop	[new $opt(prop)]
set topo	[new Topography]
set tracefd	[open $opt(tr) w]

$topo load_flatgrid $opt(x) $opt(y)

$prop topography $topo

#
# Create God
#
create-god $opt(nn)


#
# log the mobile nodes movements if desired
#
if { $opt(lm) == "on" } {
    log-movement
}

#
#  Create the specified number of nodes $opt(nn) and "attach" them
#  the channel.
#  Each routing protocol script is expected to have defined a proc
#  create-mobile-node that builds a mobile node and inserts it into the
#  array global $node_($i)
#

if { [string compare $opt(rp) "dsr"] == 0} { 
	for {set i 0} {$i  $opt(nn) } {incr i} {
		dsr-create-mobile-node $i
	}
} elseif { [string compare $opt(rp) "dsdv"] == 0} { 
	for {set i 0} {$i  $opt(nn) } {incr i} {
		dsdv-create-mobile-node $i
	}
} elseif { [string compare $opt(rp) "leach"] == 0} { 
	for {set i 0} {$i  $opt(nn) } {incr i} {
		leach-create-mobile-node $i
	}
} elseif { [string compare $opt(rp) "leach-c"] == 0} { 
	for {set i 0} {$i  $opt(nn) } {incr i} {
		leach-create-mobile-node $i
	}
} elseif { [string compare $opt(rp) "stat-clus"] == 0} { 
	for {set i 0} {$i  $opt(nn) } {incr i} {
		leach-create-mobile-node $i
	}
} elseif { [string compare $opt(rp) "mte"] == 0} { 
	for {set i 0} {$i  $opt(nn) } {incr i} {
		leach-create-mobile-node $i
	}
}


#
# Source the Connection and Movement scripts
#
if { $opt(cp) == "" } {
	puts "*** NOTE: no connection pattern specified."
        set opt(cp) "none"
} else {
	puts "Loading connection pattern..."
	source $opt(cp)
}


#
# Tell all the nodes when the simulation ends
#
for {set i 0} {$i  $opt(nn) } {incr i} {
    $ns_ at $opt(stop).000000001 "$node_($i) reset";
}
$ns_ at $opt(stop).00000001 "puts \"NS EXITING...\" ; $ns_ halt"


if { $opt(sc) == "" } {
	puts "*** NOTE: no scenario file specified."
        set opt(sc) "none"
} else {
	puts "Loading scenario file..."
	source $opt(sc)
	puts "Load complete..."
}

puts $tracefd "M 0.0 nn $opt(nn) x $opt(x) y $opt(y) rp $opt(rp)"
puts $tracefd "M 0.0 sc $opt(sc) cp $opt(cp) seed $opt(seed)"
puts $tracefd "M 0.0 prop $opt(prop) ant $opt(ant)"

puts "Starting Simulation..."
$ns_ run