annotated-topology-reader.hpp

Go to the documentation of this file.

/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
// Based on the code by Hajime Tazaki <tazaki@sfc.wide.ad.jp>

#ifndef __ANNOTATED_TOPOLOGY_READER_H__
#define __ANNOTATED_TOPOLOGY_READER_H__

#include "ns3/topology-reader.h"
#include "ns3/random-variable-stream.h"
#include "ns3/object-factory.h"
#include "ns3/node-container.h"

namespace ns3 {

class AnnotatedTopologyReader : public TopologyReader {
public:
  AnnotatedTopologyReader(const std::string& path = "", double scale = 1.0);
  virtual ~AnnotatedTopologyReader();

  virtual NodeContainer
  Read();

  virtual NodeContainer
  GetNodes() const;

  virtual const std::list<Link>&
  GetLinks() const;

  virtual void
  AssignIpv4Addresses(Ipv4Address base);

  virtual void
  SetBoundingBox(double ulx, double uly, double lrx, double lry);

  virtual void
  SetMobilityModel(const std::string& model);

  virtual void
  ApplyOspfMetric();

  virtual void
  SaveTopology(const std::string& file);

  virtual void
  SaveGraphviz(const std::string& file);

protected:
  Ptr<Node>
  CreateNode(const std::string name, uint32_t systemId);

  Ptr<Node>
  CreateNode(const std::string name, double posX, double posY, uint32_t systemId);

protected:
  void
  ApplySettings();

protected:
  std::string m_path;
  NodeContainer m_nodes;

private:
  AnnotatedTopologyReader(const AnnotatedTopologyReader&);
  AnnotatedTopologyReader&
  operator=(const AnnotatedTopologyReader&);

  Ptr<UniformRandomVariable> m_randX;
  Ptr<UniformRandomVariable> m_randY;

  ObjectFactory m_mobilityFactory;
  double m_scale;

  uint32_t m_requiredPartitions;
};
}

#endif