A Discrete-Event Network Simulator
API
inet-topology-reader.cc
Go to the documentation of this file.
1 /* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
2 /*
3  * Copyright (c) 2010 Universita' di Firenze, Italy
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License version 2 as
7  * published by the Free Software Foundation;
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software
16  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17  *
18  * Author: Tommaso Pecorella (tommaso.pecorella@unifi.it)
19  * Author: Valerio Sartini (Valesar@gmail.com)
20  */
21 
22 #include <fstream>
23 #include <cstdlib>
24 #include <sstream>
25 
26 #include "ns3/log.h"
27 
28 #include "inet-topology-reader.h"
29 
30 
31 namespace ns3 {
32 
33 NS_LOG_COMPONENT_DEFINE ("InetTopologyReader");
34 
35 NS_OBJECT_ENSURE_REGISTERED (InetTopologyReader);
36 
38 {
39  static TypeId tid = TypeId ("ns3::InetTopologyReader")
41  .SetGroupName ("TopologyReader")
42  .AddConstructor<InetTopologyReader> ()
43  ;
44  return tid;
45 }
46 
48 {
49  NS_LOG_FUNCTION (this);
50 }
51 
53 {
54  NS_LOG_FUNCTION (this);
55 }
56 
59 {
60  std::ifstream topgen;
61  topgen.open (GetFileName ().c_str ());
62  std::map<std::string, Ptr<Node> > nodeMap;
64 
65  if ( !topgen.is_open () )
66  {
67  NS_LOG_WARN ("Inet topology file object is not open, check file name and permissions");
68  return nodes;
69  }
70 
71  std::string from;
72  std::string to;
73  std::string linkAttr;
74 
75  int linksNumber = 0;
76  int nodesNumber = 0;
77 
78  int totnode = 0;
79  int totlink = 0;
80 
81  std::istringstream lineBuffer;
82  std::string line;
83 
84  getline (topgen,line);
85  lineBuffer.str (line);
86 
87  lineBuffer >> totnode;
88  lineBuffer >> totlink;
89  NS_LOG_INFO ("Inet topology should have " << totnode << " nodes and " << totlink << " links");
90 
91  for (int i = 0; i < totnode && !topgen.eof (); i++)
92  {
93  getline (topgen,line);
94  }
95 
96  for (int i = 0; i < totlink && !topgen.eof (); i++)
97  {
98  getline (topgen,line);
99  lineBuffer.clear ();
100  lineBuffer.str (line);
101 
102  lineBuffer >> from;
103  lineBuffer >> to;
104  lineBuffer >> linkAttr;
105 
106  if ( (!from.empty ()) && (!to.empty ()) )
107  {
108  NS_LOG_INFO ( "Link " << linksNumber << " from: " << from << " to: " << to);
109 
110  if ( nodeMap[from] == 0 )
111  {
112  NS_LOG_INFO ( "Node " << nodesNumber << " name: " << from);
113  Ptr<Node> tmpNode = CreateObject<Node> ();
114  nodeMap[from] = tmpNode;
115  nodes.Add (tmpNode);
116  nodesNumber++;
117  }
118 
119  if (nodeMap[to] == 0)
120  {
121  NS_LOG_INFO ( "Node " << nodesNumber << " name: " << to);
122  Ptr<Node> tmpNode = CreateObject<Node> ();
123  nodeMap[to] = tmpNode;
124  nodes.Add (tmpNode);
125  nodesNumber++;
126  }
127 
128  Link link ( nodeMap[from], from, nodeMap[to], to );
129  if ( !linkAttr.empty () )
130  {
131  NS_LOG_INFO ( "Link " << linksNumber << " weight: " << linkAttr);
132  link.SetAttribute ("Weight", linkAttr);
133  }
134  AddLink (link);
135 
136  linksNumber++;
137  }
138  }
139 
140  NS_LOG_INFO ("Inet topology created with " << nodesNumber << " nodes and " << linksNumber << " links");
141  topgen.close ();
142 
143  return nodes;
144 }
145 
146 } /* namespace ns3 */
#define NS_LOG_FUNCTION(parameters)
If log level LOG_FUNCTION is enabled, this macro will output all input parameters separated by "...
#define NS_OBJECT_ENSURE_REGISTERED(type)
Register an Object subclass with the TypeId system.
Definition: object-base.h:45
static TypeId GetTypeId(void)
Get the type ID.
#define NS_LOG_COMPONENT_DEFINE(name)
Define a Log component with a specific name.
Definition: log.h:202
#define NS_LOG_INFO(msg)
Use NS_LOG to output a message of level LOG_INFO.
Definition: log.h:278
void AddLink(Link link)
Adds a link to the topology.
Interface for input file readers management.
nodes
Definition: first.py:25
Every class exported by the ns3 library is enclosed in the ns3 namespace.
virtual NodeContainer Read(void)
Main topology reading function.
keep track of a set of node pointers.
std::string GetFileName(void) const
Returns the input file name.
#define NS_LOG_WARN(msg)
Use NS_LOG to output a message of level LOG_WARN.
Definition: log.h:262
Topology file reader (Inet-format type).
a unique identifier for an interface.
Definition: type-id.h:58
TypeId SetParent(TypeId tid)
Set the parent TypeId.
Definition: type-id.cc:915