Files
unison/examples/multi-rate-first.cc
2009-08-24 19:11:41 -07:00

399 lines
12 KiB
C++

/**
*
* Instructions:
* ./waf --run multi-rate-first
* gnuplot multi-rate-first-scen*.plt
*
* Output:
* multi-rate-first-scen1.eps
* multi-rate-first-scen2.eps
* multi-rate-first-scen3.eps
* multi-rate-first-scen4.eps
*
* Side Note: It may take some time.
*/
#include "ns3/core-module.h"
#include "ns3/common-module.h"
#include "ns3/node-module.h"
#include "ns3/helper-module.h"
#include "ns3/mobility-module.h"
#include "ns3/contrib-module.h"
#include <iostream>
#include <fstream>
NS_LOG_COMPONENT_DEFINE ("Main");
using namespace ns3;
class Experiment
{
public:
Experiment ();
Experiment (std::string name);
Gnuplot2dDataset Run (const WifiHelper &wifi, const YansWifiPhyHelper &wifiPhy,
const NqosWifiMacHelper &wifiMac, const YansWifiChannelHelper &wifiChannel, const MobilityHelper &mobility, int positionStep);
private:
void ReceivePacket (Ptr<Socket> socket);
void SetPosition (Ptr<Node> node, Vector position);
Vector GetPosition (Ptr<Node> node);
void AdvancePosition (Ptr<Node> node);
void BackTrackPosition (Ptr<Node> node);
void StationaryPosition (Ptr<Node> node);
void MultiPosition (Ptr<Node> node1, Ptr<Node> node2);
Ptr<Socket> SetupPacketReceive (Ptr<Node> node);
uint32_t m_bytesTotal;
Gnuplot2dDataset m_output;
};
Experiment::Experiment ()
{}
Experiment::Experiment (std::string name)
: m_output (name)
{
m_output.SetStyle (Gnuplot2dDataset::LINES);
}
void
Experiment::SetPosition (Ptr<Node> node, Vector position)
{
Ptr<MobilityModel> mobility = node->GetObject<MobilityModel> ();
mobility->SetPosition (position);
}
Vector
Experiment::GetPosition (Ptr<Node> node)
{
Ptr<MobilityModel> mobility = node->GetObject<MobilityModel> ();
return mobility->GetPosition ();
}
void
Experiment::AdvancePosition (Ptr<Node> node)
{
Vector pos = GetPosition (node);
double mbs = ((m_bytesTotal * 8.0) / 1000000);
m_bytesTotal = 0;
m_output.Add ((Simulator::Now()).GetSeconds(), mbs);
pos.x += 1.0;
if (pos.x >= 210.0)
{
return;
}
SetPosition (node, pos);
//std::cout << "x="<<pos.x << std::endl;
Simulator::Schedule (Seconds (1.0), &Experiment::AdvancePosition, this, node);
}
void
Experiment::BackTrackPosition (Ptr<Node> node)
{
Vector pos = GetPosition (node);
double mbs = ((m_bytesTotal * 8.0) / 1000000);
m_bytesTotal = 0;
m_output.Add ((Simulator::Now()).GetSeconds(), mbs);
pos.x -= 1.0;
if (pos.x < 0)
{
return;
}
SetPosition (node, pos);
//std::cout << "x="<<pos.x << std::endl;
Simulator::Schedule (Seconds (1.0), &Experiment::BackTrackPosition, this, node);
}
void
Experiment::StationaryPosition (Ptr<Node> node)
{
double mbs = ((m_bytesTotal * 8.0) / 1000000);
m_bytesTotal = 0;
m_output.Add ((Simulator::Now()).GetSeconds(), mbs);
}
void
Experiment::MultiPosition (Ptr<Node> n1, Ptr<Node> n2)
{
Vector pos1 = GetPosition(n1);
Vector pos2 = GetPosition(n2);
double mbs = ((m_bytesTotal * 8.0) / 1000000);
m_bytesTotal = 0;
m_output.Add ((Simulator::Now()).GetSeconds(), mbs);
if( pos1.x < 230)
{
pos1.x += 1.0;
SetPosition (n1, pos1);
}
if( pos2.x > 0)
{
pos2.x -= 1.0;
SetPosition (n2, pos2);
}
}
void
Experiment::ReceivePacket (Ptr<Socket> socket)
{
Ptr<Packet> packet;
while (packet = socket->Recv ())
{
m_bytesTotal += packet->GetSize ();
}
}
Ptr<Socket>
Experiment::SetupPacketReceive (Ptr<Node> node)
{
TypeId tid = TypeId::LookupByName ("ns3::PacketSocketFactory");
Ptr<Socket> sink = Socket::CreateSocket (node, tid);
sink->Bind ();
sink->SetRecvCallback (MakeCallback (&Experiment::ReceivePacket, this));
return sink;
}
Gnuplot2dDataset
Experiment::Run (const WifiHelper &wifi, const YansWifiPhyHelper &wifiPhy,
const NqosWifiMacHelper &wifiMac, const YansWifiChannelHelper &wifiChannel, const MobilityHelper &mobility, int positionStep)
{
m_bytesTotal = 0;
NodeContainer c;
c.Create (2);
PacketSocketHelper packetSocket;
packetSocket.Install (c);
YansWifiPhyHelper phy = wifiPhy;
phy.SetChannel (wifiChannel.Create ());
NqosWifiMacHelper mac = wifiMac;
NetDeviceContainer devices = wifi.Install (phy, mac, c);
mobility.Install (c);
PacketSocketAddress socket;
socket.SetSingleDevice(devices.Get (0)->GetIfIndex ());
socket.SetPhysicalAddress (devices.Get (1)->GetAddress ());
socket.SetProtocol (1);
OnOffHelper onoff ("ns3::PacketSocketFactory", Address (socket));
onoff.SetAttribute ("OnTime", RandomVariableValue (ConstantVariable (250)));
onoff.SetAttribute ("OffTime", RandomVariableValue (ConstantVariable (0)));
onoff.SetAttribute ("DataRate", DataRateValue (DataRate (60000000)));
onoff.SetAttribute ("PacketSize", UintegerValue (2000));
ApplicationContainer apps = onoff.Install (c.Get (0));
apps.Start (Seconds (0.5));
apps.Stop (Seconds (250.0));
Ptr<Socket> recvSink = SetupPacketReceive (c.Get (1));
if (positionStep == 1)
{
Simulator::Schedule (Seconds (1.5), &Experiment::AdvancePosition, this, c.Get (1));
}
else if (positionStep == -1)
{
Simulator::Schedule (Seconds (1.5), &Experiment::BackTrackPosition, this, c.Get (1));
}
else if (positionStep == 0)
{
for(int i = 1; i <= 210; i++)
{
Simulator::Schedule (Seconds (i), &Experiment::StationaryPosition, this, c.Get (1));
}
}
else if (positionStep == 2)
{
for(int i = 1; i <= 210; i++)
{
Simulator::Schedule (Seconds (i), &Experiment::MultiPosition, this, c.Get(0), c.Get (1));
}
}
Simulator::Run ();
Simulator::Destroy ();
return m_output;
}
int main (int argc, char *argv[])
{
std::ofstream outfile ("multi-rate-first-scen1.plt");
std::ofstream outfile2 ("multi-rate-first-scen2.plt");
std::ofstream outfile3 ("multi-rate-first-scen3.plt");
std::ofstream outfile4 ("multi-rate-first-scen4.plt");
// disable fragmentation
Config::SetDefault ("ns3::WifiRemoteStationManager::FragmentationThreshold", StringValue ("2200"));
Config::SetDefault ("ns3::WifiRemoteStationManager::RtsCtsThreshold", StringValue ("2200"));
CommandLine cmd;
cmd.Parse (argc, argv);
MobilityHelper mobility;
Experiment experiment;
Gnuplot gnuplot;
int myPositionStep;
Ptr<ListPositionAllocator> positionAlloc;
Gnuplot2dDataset dataset;
WifiHelper wifi = WifiHelper::Default ();
NqosWifiMacHelper wifiMac = NqosWifiMacHelper::Default ();
YansWifiPhyHelper wifiPhy = YansWifiPhyHelper::Default ();
YansWifiChannelHelper wifiChannel = YansWifiChannelHelper::Default ();
// Scenario 1: two nodes within transmission range about 5 meters apart
// Fix a node stationary, move the second node away from it
// moving forward
myPositionStep = 1;
gnuplot = Gnuplot ("multi-rate-first-scen1.eps");
positionAlloc = CreateObject<ListPositionAllocator> ();
positionAlloc->Add (Vector (0.0, 0.0, 0.0));
positionAlloc->Add (Vector (5.0, 0.0, 0.0));
mobility.SetPositionAllocator (positionAlloc);
mobility.SetMobilityModel ("ns3::ConstantPositionMobilityModel");
wifiMac.SetType ("ns3::AdhocWifiMac");
wifi.SetStandard (WIFI_PHY_STANDARD_holland);
experiment = Experiment ("minstrel");
wifi.SetRemoteStationManager ("ns3::MinstrelWifiManager");
dataset = experiment.Run (wifi, wifiPhy, wifiMac, wifiChannel, mobility, myPositionStep);
gnuplot.AddDataset (dataset);
experiment = Experiment ("ideal");
wifi.SetRemoteStationManager ("ns3::IdealWifiManager");
dataset = experiment.Run (wifi, wifiPhy, wifiMac, wifiChannel, mobility, myPositionStep);
gnuplot.AddDataset (dataset);
gnuplot.SetTerminal ("postscript eps color enh \"Times-BoldItalic\"");
gnuplot.SetLegend ("Time (Seconds)", "Throughput(Mbps)");
gnuplot.SetExtra ("set xrange [0:250]");
gnuplot.SetTitle ("Throughput vs Time");
gnuplot.GenerateOutput (outfile);
outfile.close ();
// Scenario 2: two nodes out of transmission range about 230 meters apart
// Fix a node stationary, move the second node into transmission range
// moving backward
myPositionStep = -1;
gnuplot = Gnuplot ("multi-rate-first-scen2.eps");
positionAlloc = CreateObject<ListPositionAllocator> ();
positionAlloc->Add (Vector (0.0, 0.0, 0.0));
positionAlloc->Add (Vector (230.0, 0.0, 0.0));
mobility.SetPositionAllocator (positionAlloc);
mobility.SetMobilityModel ("ns3::ConstantPositionMobilityModel");
wifiMac.SetType ("ns3::AdhocWifiMac");
wifi.SetStandard (WIFI_PHY_STANDARD_holland);
experiment = Experiment ("minstrel");
wifi.SetRemoteStationManager ("ns3::MinstrelWifiManager");
dataset= experiment.Run (wifi, wifiPhy, wifiMac, wifiChannel, mobility, myPositionStep);
gnuplot.AddDataset (dataset);
experiment = Experiment ("ideal");
wifi.SetRemoteStationManager ("ns3::IdealWifiManager");
dataset= experiment.Run (wifi, wifiPhy, wifiMac, wifiChannel, mobility, myPositionStep);
gnuplot.AddDataset (dataset);
gnuplot.SetTerminal ("postscript eps color enh \"Times-BoldItalic\"");
gnuplot.SetLegend ("Time (Seconds)", "Throughput(Mbps)");
gnuplot.SetExtra ("set xrange [0:250]");
gnuplot.SetTitle ("Throughput vs Time");
gnuplot.GenerateOutput (outfile2);
outfile2.close ();
// Scenario 3: two nodes within transmission range 25 meters part
// Set both nodes stationary
// this is more like a sanity check
// Set position stationary
myPositionStep = 0;
gnuplot = Gnuplot ("multi-rate-first-scen3.eps");
positionAlloc = CreateObject<ListPositionAllocator> ();
positionAlloc->Add (Vector (0.0, 0.0, 0.0));
positionAlloc->Add (Vector (25.0, 0.0, 0.0));
mobility.SetPositionAllocator (positionAlloc);
mobility.SetMobilityModel ("ns3::ConstantPositionMobilityModel");
wifiMac.SetType ("ns3::AdhocWifiMac");
wifi.SetStandard (WIFI_PHY_STANDARD_holland);
experiment = Experiment ("minstrel");
wifi.SetRemoteStationManager ("ns3::MinstrelWifiManager");
dataset= experiment.Run (wifi, wifiPhy, wifiMac, wifiChannel, mobility, myPositionStep);
gnuplot.AddDataset (dataset);
experiment = Experiment ("ideal");
wifi.SetRemoteStationManager ("ns3::IdealWifiManager");
dataset= experiment.Run (wifi, wifiPhy, wifiMac, wifiChannel, mobility, myPositionStep);
gnuplot.AddDataset (dataset);
gnuplot.SetTerminal ("postscript eps color enh \"Times-BoldItalic\"");
gnuplot.SetLegend ("Time (Seconds)", "Throughput(Mbps)");
gnuplot.SetExtra ("set xrange [0:250]");
gnuplot.SetTitle ("Throughput vs Time");
gnuplot.GenerateOutput (outfile3);
outfile3.close ();
// Scenario 4: Two nodes in opposite direction about 230 meters apart
// moving into transmission range and out of transmission range
myPositionStep = 2;
gnuplot = Gnuplot ("multi-rate-first-scen4.eps");
positionAlloc = CreateObject<ListPositionAllocator> ();
// initial position of node 1
positionAlloc->Add (Vector (0.0, 25.0, 0.0));
// initial position of node 2
positionAlloc->Add (Vector (230.0, 0.0, 0.0));
mobility.SetPositionAllocator (positionAlloc);
mobility.SetMobilityModel ("ns3::ConstantPositionMobilityModel");
wifiMac.SetType ("ns3::AdhocWifiMac");
wifi.SetStandard (WIFI_PHY_STANDARD_holland);
experiment = Experiment ("minstrel");
wifi.SetRemoteStationManager ("ns3::MinstrelWifiManager");
dataset= experiment.Run (wifi, wifiPhy, wifiMac, wifiChannel, mobility, myPositionStep);
gnuplot.AddDataset (dataset);
experiment = Experiment ("ideal");
wifi.SetRemoteStationManager ("ns3::IdealWifiManager");
dataset= experiment.Run (wifi, wifiPhy, wifiMac, wifiChannel, mobility, myPositionStep);
gnuplot.AddDataset (dataset);
gnuplot.SetTerminal ("postscript eps color enh \"Times-BoldItalic\"");
gnuplot.SetLegend ("Time (Seconds)", "Throughput(Mbps)");
gnuplot.SetExtra ("set xrange [0:250]");
gnuplot.SetTitle ("Throughput vs Time");
gnuplot.GenerateOutput (outfile4);
outfile4.close ();
return 0;
}