model/annotated-topology-reader.cc - ndnSIM - Gitiles

 /* -*-  Mode: C++; c-file-style: "gnu"; indent-tabs-mode:nil; -*- */
 /*
  * Copyright (c) 2011 University of California, Los Angeles
  *
  * This program is free software; you can redistribute it and/or modify
  * it under the terms of the GNU General Public License version 2 as
  * published by the Free Software Foundation;
  *
  * This program is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  * GNU General Public License for more details.
  *
  * You should have received a copy of the GNU General Public License
  * along with this program; if not, write to the Free Software
  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  *
  * Author: Ilya Moiseenko <iliamo@cs.ucla.edu>
  */

 #include "annotated-topology-reader.h"

 #include "ns3/nstime.h"
 #include "ns3/log.h"
 #include "ns3/assert.h"
 #include "ns3/names.h"
 #include "ns3/net-device-container.h"
 #include "ns3/point-to-point-helper.h"
 #include "ns3/point-to-point-net-device.h"
 #include "ns3/internet-stack-helper.h"
 #include "ns3/ipv4-address-helper.h"
 #include "ns3/ipv4-global-routing-helper.h"
 #include "ns3/drop-tail-queue.h"
 #include "ns3/ipv4-interface.h"
 #include "ns3/ipv4.h"
 #include "ns3/string.h"
 #include "ns3/pointer.h"
 #include "ns3/uinteger.h"
 #include "ns3/ipv4-address.h"

 #include "ns3/constant-position-mobility-model.h"
 #include "ns3/random-variable.h"

 #include <boost/foreach.hpp>
 #include <boost/lexical_cast.hpp>

 #include <set>

 using namespace std;

 namespace ns3
 {

 NS_LOG_COMPONENT_DEFINE ("AnnotatedTopologyReader");

 NS_OBJECT_ENSURE_REGISTERED (AnnotatedTopologyReader);

 TypeId AnnotatedTopologyReader::GetTypeId (void)
 {
   static TypeId tid = TypeId ("ns3::AnnotatedTopologyReader")
     .SetParent<Object> ()
     ;
   return tid;
 }

 AnnotatedTopologyReader::AnnotatedTopologyReader (const std::string &path)
   : m_path (path)
   , m_ulx (0)
   , m_uly (0)
   , m_lrx (100.0)
   , m_lry (100.0)
 {
   NS_LOG_FUNCTION (this);
 }

 void
 AnnotatedTopologyReader::SetBoundingBox (double ulx, double uly, double lrx, double lry)
 {
   NS_LOG_FUNCTION (this << ulx << uly << lrx << lry);

   m_ulx = ulx;
   m_uly = uly;
   m_lrx = lrx;
   m_lry = lry;
 }

 AnnotatedTopologyReader::~AnnotatedTopologyReader ()
 {
   NS_LOG_FUNCTION (this);
 }

 NodeContainer
 AnnotatedTopologyReader::Read (void)
 {
   ifstream topgen;
   topgen.open (GetFileName ().c_str ());
   NodeContainer nodes;

   if ( !topgen.is_open () )
     {
       NS_LOG_ERROR ("Cannot open file " << GetFileName () << " for reading");
       return nodes;
     }

   uint32_t linksNumber = 0;
   uint32_t nodesNumber = 0;

   while (!topgen.eof ())
     {
       string line;
       getline (topgen, line);

       if (line == "router") break;
     }

   while (!topgen.eof ())
     {
       string line;
       getline (topgen,line);
       if (line[0] == '#') continue; // comments
       if (line=="link") break; // stop reading nodes

       istringstream lineBuffer (line);
       string name, city;
       double latitude, longitude;

       lineBuffer >> name >> city >> latitude >> longitude;
       Ptr<Node> node = CreateObject<Node> ();
       Ptr<ConstantPositionMobilityModel> loc = CreateObject<ConstantPositionMobilityModel> ();
       node->AggregateObject (loc);

       loc->SetPosition (Vector (latitude, longitude, 0));

       Names::Add (m_path, name, node);
       nodes.Add (node);
       nodesNumber++;
     }

   map<string, set<string> > processedLinks; // to eliminate duplications

   // SeekToSection ("link");
   while (!topgen.eof ())
     {
       string line;
       getline (topgen,line);
       if (line == "") continue;
       if (line[0] == '#') continue; // comments

       // NS_LOG_DEBUG ("Input: [" << line << "]");

       istringstream lineBuffer (line);
       string from, to, capacity, metric;

       lineBuffer >> from >> to >> capacity >> metric;

       if (processedLinks[to].size () != 0 &&
           processedLinks[to].find (from) != processedLinks[to].end ())
         {
           continue; // duplicated link
         }
       processedLinks[from].insert (to);

       Ptr<Node> fromNode = Names::Find<Node> (m_path, from);
       NS_ASSERT (fromNode != 0);
       Ptr<Node> toNode   = Names::Find<Node> (m_path, to);
       NS_ASSERT (fromNode != 0);

       Link link (fromNode, from, toNode, to);

       link.SetAttribute ("DataRate", capacity);
       link.SetAttribute ("OSPF", metric);
       // link.SetAttribute ("Delay", delay);
       // link.SetAttribute ("QueueSizeNode1", queueSizeNode1);
       // link.SetAttribute ("QueueSizeNode2", queueSizeNode2);

       AddLink (link);
       NS_LOG_DEBUG ("New link " << from << " <==> " << to << " / " << capacity << "Kbps with " << metric << " metric");

       linksNumber++;
     }

   NS_LOG_INFO ("Annotated topology created with " << nodesNumber << " nodes and " << linksNumber << " links");
   topgen.close ();

   ApplySettings ();

   return nodes;
 }

 void
 AnnotatedTopologyReader::AssignIpv4Addresses (Ipv4Address base)
 {
   Ipv4AddressHelper address (base, Ipv4Mask ("/24"));

   BOOST_FOREACH (const Link &link, m_linksList)
     {
       address.Assign (NetDeviceContainer (link.GetFromNetDevice (),
                                           link.GetToNetDevice ()));

       base = Ipv4Address (base.Get () + 256);
       address.SetBase (base, Ipv4Mask ("/24"));
     }

   ApplyOspfMetric ();
 }

 void
 AnnotatedTopologyReader::ApplyOspfMetric ()
 {
   BOOST_FOREACH (const Link &link, m_linksList)
     {
       uint16_t metric = boost::lexical_cast<uint16_t> (link.GetAttribute ("OSPF"));

       {
         Ptr<Ipv4> ipv4 = link.GetFromNode ()->GetObject<Ipv4> ();
         NS_ASSERT (ipv4 != 0);

         int32_t interfaceId = ipv4->GetInterfaceForDevice (link.GetFromNetDevice ());
         NS_ASSERT (interfaceId >= 0);

         ipv4->SetMetric (interfaceId,metric);
       }

       {
         Ptr<Ipv4> ipv4 = link.GetToNode ()->GetObject<Ipv4> ();
         NS_ASSERT (ipv4 != 0);

         int32_t interfaceId = ipv4->GetInterfaceForDevice (link.GetToNetDevice ());
         NS_ASSERT (interfaceId >= 0);

         ipv4->SetMetric (interfaceId,metric);
       }
     }
 }

 void
 AnnotatedTopologyReader::ApplySettings ()
 {
   PointToPointHelper p2p;

   // temporary queue, will be changed later
   p2p.SetQueue ("ns3::DropTailQueue",
                 "MaxPackets", StringValue("100"));

   BOOST_FOREACH (Link &link, m_linksList)
     {
       string tmp;

       NS_LOG_INFO ("DataRate = " + link.GetAttribute("DataRate")+"Kbps");
       p2p.SetDeviceAttribute ("DataRate", StringValue(link.GetAttribute("DataRate")+"Kbps"));

       if (link.GetAttributeFailSafe("Delay", tmp))
         {
           NS_LOG_INFO ("Delay = " + link.GetAttribute("Delay")+"ms");
           p2p.SetChannelAttribute ("Delay", StringValue(link.GetAttribute("Delay")+"ms"));
         }
       else
         {
           NS_LOG_INFO ("Default delay 1ms");
           p2p.SetChannelAttribute ("Delay", StringValue("1ms"));
         }

       NetDeviceContainer nd = p2p.Install(link.GetFromNode (), link.GetToNode ());
       link.SetNetDevices (nd.Get (0), nd.Get (1));

       // NS_LOG_INFO ("Queue: " << link.GetAttribute("QueueSizeNode1") << " <==> " << link.GetAttribute("QueueSizeNode2"));

       if (link.GetAttributeFailSafe("QueueSizeNode1", tmp))
         {
           PointerValue txQueueFrom;
           link.GetFromNetDevice ()->GetAttribute ("TxQueue", txQueueFrom);
           NS_ASSERT (txQueueFrom.Get<DropTailQueue> () != 0);

           txQueueFrom.Get<DropTailQueue> ()->SetAttribute ("MaxPackets", StringValue (link.GetAttribute("QueueSizeNode1")));
         }

       if (link.GetAttributeFailSafe("QueueSizeNode2", tmp))
         {
           PointerValue txQueueTo;
           link.GetToNetDevice ()->GetAttribute ("TxQueue", txQueueTo);
           NS_ASSERT (txQueueTo.Get<DropTailQueue> () != 0);

           txQueueTo.  Get<DropTailQueue> ()->SetAttribute ("MaxPackets", StringValue (link.GetAttribute("QueueSizeNode2")));
         }
     }
 }

 }
	/* -- Mode: C++; c-file-style: "gnu"; indent-tabs-mode:nil; -- */
	/*
	* Copyright (c) 2011 University of California, Los Angeles
	*
	* This program is free software; you can redistribute it and/or modify
	* it under the terms of the GNU General Public License version 2 as
	* published by the Free Software Foundation;
	*
	* This program is distributed in the hope that it will be useful,
	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	* GNU General Public License for more details.
	*
	* You should have received a copy of the GNU General Public License
	* along with this program; if not, write to the Free Software
	* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
	*
	* Author: Ilya Moiseenko <iliamo@cs.ucla.edu>
	*/

	#include "annotated-topology-reader.h"

	#include "ns3/nstime.h"
	#include "ns3/log.h"
	#include "ns3/assert.h"
	#include "ns3/names.h"
	#include "ns3/net-device-container.h"
	#include "ns3/point-to-point-helper.h"
	#include "ns3/point-to-point-net-device.h"
	#include "ns3/internet-stack-helper.h"
	#include "ns3/ipv4-address-helper.h"
	#include "ns3/ipv4-global-routing-helper.h"
	#include "ns3/drop-tail-queue.h"
	#include "ns3/ipv4-interface.h"
	#include "ns3/ipv4.h"
	#include "ns3/string.h"
	#include "ns3/pointer.h"
	#include "ns3/uinteger.h"
	#include "ns3/ipv4-address.h"

	#include "ns3/constant-position-mobility-model.h"
	#include "ns3/random-variable.h"

	#include <boost/foreach.hpp>
	#include <boost/lexical_cast.hpp>

	#include <set>

	using namespace std;

	namespace ns3
	{

	NS_LOG_COMPONENT_DEFINE ("AnnotatedTopologyReader");

	NS_OBJECT_ENSURE_REGISTERED (AnnotatedTopologyReader);

	TypeId AnnotatedTopologyReader::GetTypeId (void)
	{
	static TypeId tid = TypeId ("ns3::AnnotatedTopologyReader")
	.SetParent<Object> ()
	;
	return tid;
	}

	AnnotatedTopologyReader::AnnotatedTopologyReader (const std::string &path)
	: m_path (path)
	, m_ulx (0)
	, m_uly (0)
	, m_lrx (100.0)
	, m_lry (100.0)
	{
	NS_LOG_FUNCTION (this);
	}

	void
	AnnotatedTopologyReader::SetBoundingBox (double ulx, double uly, double lrx, double lry)
	{
	NS_LOG_FUNCTION (this << ulx << uly << lrx << lry);

	m_ulx = ulx;
	m_uly = uly;
	m_lrx = lrx;
	m_lry = lry;
	}

	AnnotatedTopologyReader::~AnnotatedTopologyReader ()
	{
	NS_LOG_FUNCTION (this);
	}

	NodeContainer
	AnnotatedTopologyReader::Read (void)
	{
	ifstream topgen;
	topgen.open (GetFileName ().c_str ());
	NodeContainer nodes;

	if ( !topgen.is_open () )
	{
	NS_LOG_ERROR ("Cannot open file " << GetFileName () << " for reading");
	return nodes;
	}

	uint32_t linksNumber = 0;
	uint32_t nodesNumber = 0;

	while (!topgen.eof ())
	{
	string line;
	getline (topgen, line);

	if (line == "router") break;
	}

	while (!topgen.eof ())
	{
	string line;
	getline (topgen,line);
	if (line[0] == '#') continue; // comments
	if (line=="link") break; // stop reading nodes

	istringstream lineBuffer (line);
	string name, city;
	double latitude, longitude;

	lineBuffer >> name >> city >> latitude >> longitude;
	Ptr<Node> node = CreateObject<Node> ();
	Ptr<ConstantPositionMobilityModel> loc = CreateObject<ConstantPositionMobilityModel> ();
	node->AggregateObject (loc);

	loc->SetPosition (Vector (latitude, longitude, 0));

	Names::Add (m_path, name, node);
	nodes.Add (node);
	nodesNumber++;
	}

	map<string, set<string> > processedLinks; // to eliminate duplications

	// SeekToSection ("link");
	while (!topgen.eof ())
	{
	string line;
	getline (topgen,line);
	if (line == "") continue;
	if (line[0] == '#') continue; // comments

	// NS_LOG_DEBUG ("Input: [" << line << "]");

	istringstream lineBuffer (line);
	string from, to, capacity, metric;

	lineBuffer >> from >> to >> capacity >> metric;

	if (processedLinks[to].size () != 0 &&
	processedLinks[to].find (from) != processedLinks[to].end ())
	{
	continue; // duplicated link
	}
	processedLinks[from].insert (to);

	Ptr<Node> fromNode = Names::Find<Node> (m_path, from);
	NS_ASSERT (fromNode != 0);
	Ptr<Node> toNode = Names::Find<Node> (m_path, to);
	NS_ASSERT (fromNode != 0);

	Link link (fromNode, from, toNode, to);

	link.SetAttribute ("DataRate", capacity);
	link.SetAttribute ("OSPF", metric);
	// link.SetAttribute ("Delay", delay);
	// link.SetAttribute ("QueueSizeNode1", queueSizeNode1);
	// link.SetAttribute ("QueueSizeNode2", queueSizeNode2);

	AddLink (link);
	NS_LOG_DEBUG ("New link " << from << " <==> " << to << " / " << capacity << "Kbps with " << metric << " metric");

	linksNumber++;
	}

	NS_LOG_INFO ("Annotated topology created with " << nodesNumber << " nodes and " << linksNumber << " links");
	topgen.close ();

	ApplySettings ();

	return nodes;
	}

	void
	AnnotatedTopologyReader::AssignIpv4Addresses (Ipv4Address base)
	{
	Ipv4AddressHelper address (base, Ipv4Mask ("/24"));

	BOOST_FOREACH (const Link &link, m_linksList)
	{
	address.Assign (NetDeviceContainer (link.GetFromNetDevice (),
	link.GetToNetDevice ()));

	base = Ipv4Address (base.Get () + 256);
	address.SetBase (base, Ipv4Mask ("/24"));
	}

	ApplyOspfMetric ();
	}

	void
	AnnotatedTopologyReader::ApplyOspfMetric ()
	{
	BOOST_FOREACH (const Link &link, m_linksList)
	{
	uint16_t metric = boost::lexical_cast<uint16_t> (link.GetAttribute ("OSPF"));

	{
	Ptr<Ipv4> ipv4 = link.GetFromNode ()->GetObject<Ipv4> ();
	NS_ASSERT (ipv4 != 0);

	int32_t interfaceId = ipv4->GetInterfaceForDevice (link.GetFromNetDevice ());
	NS_ASSERT (interfaceId >= 0);

	ipv4->SetMetric (interfaceId,metric);
	}

	{
	Ptr<Ipv4> ipv4 = link.GetToNode ()->GetObject<Ipv4> ();
	NS_ASSERT (ipv4 != 0);

	int32_t interfaceId = ipv4->GetInterfaceForDevice (link.GetToNetDevice ());
	NS_ASSERT (interfaceId >= 0);

	ipv4->SetMetric (interfaceId,metric);
	}
	}
	}

	void
	AnnotatedTopologyReader::ApplySettings ()
	{
	PointToPointHelper p2p;

	// temporary queue, will be changed later
	p2p.SetQueue ("ns3::DropTailQueue",
	"MaxPackets", StringValue("100"));

	BOOST_FOREACH (Link &link, m_linksList)
	{
	string tmp;

	NS_LOG_INFO ("DataRate = " + link.GetAttribute("DataRate")+"Kbps");
	p2p.SetDeviceAttribute ("DataRate", StringValue(link.GetAttribute("DataRate")+"Kbps"));

	if (link.GetAttributeFailSafe("Delay", tmp))
	{
	NS_LOG_INFO ("Delay = " + link.GetAttribute("Delay")+"ms");
	p2p.SetChannelAttribute ("Delay", StringValue(link.GetAttribute("Delay")+"ms"));
	}
	else
	{
	NS_LOG_INFO ("Default delay 1ms");
	p2p.SetChannelAttribute ("Delay", StringValue("1ms"));
	}

	NetDeviceContainer nd = p2p.Install(link.GetFromNode (), link.GetToNode ());
	link.SetNetDevices (nd.Get (0), nd.Get (1));

	// NS_LOG_INFO ("Queue: " << link.GetAttribute("QueueSizeNode1") << " <==> " << link.GetAttribute("QueueSizeNode2"));

	if (link.GetAttributeFailSafe("QueueSizeNode1", tmp))
	{
	PointerValue txQueueFrom;
	link.GetFromNetDevice ()->GetAttribute ("TxQueue", txQueueFrom);
	NS_ASSERT (txQueueFrom.Get<DropTailQueue> () != 0);

	txQueueFrom.Get<DropTailQueue> ()->SetAttribute ("MaxPackets", StringValue (link.GetAttribute("QueueSizeNode1")));
	}

	if (link.GetAttributeFailSafe("QueueSizeNode2", tmp))
	{
	PointerValue txQueueTo;
	link.GetToNetDevice ()->GetAttribute ("TxQueue", txQueueTo);
	NS_ASSERT (txQueueTo.Get<DropTailQueue> () != 0);

	txQueueTo. Get<DropTailQueue> ()->SetAttribute ("MaxPackets", StringValue (link.GetAttribute("QueueSizeNode2")));
	}
	}
	}

	}