From 19ebf81e3d32dbdda4ef06406c28566c6d771f9d Mon Sep 17 00:00:00 2001
From: Alexander Afanasyev <alexander.afanasyev@ucla.edu>
Date: Mon, 12 Dec 2011 18:01:12 -0800
Subject: topology-read: Several extensions of TopologyReader

diff --git a/src/topology-read/model/topology-reader.cc b/src/topology-read/model/topology-reader.cc
index b531a6d..82a8343 100644
--- a/src/topology-read/model/topology-reader.cc
+++ b/src/topology-read/model/topology-reader.cc
@@ -106,12 +106,23 @@ TopologyReader::Link::Link ()
 {
 }
 
+void
+TopologyReader::Link::SetNetDevices (Ptr<NetDevice> from, Ptr<NetDevice> to)
+{
+  m_fromNetDevice = from;
+  m_toNetDevice = to;
+}
 
 Ptr<Node> TopologyReader::Link::GetFromNode (void) const
 {
   return m_fromPtr;
 }
 
+Ptr<NetDevice> TopologyReader::Link::GetFromNetDevice (void) const
+{
+  return m_fromNetDevice;
+}
+
 std::string
 TopologyReader::Link::GetFromNodeName (void) const
 {
@@ -124,6 +135,12 @@ TopologyReader::Link::GetToNode (void) const
   return m_toPtr;
 }
 
+Ptr<NetDevice>
+TopologyReader::Link::GetToNetDevice (void) const
+{
+  return m_toNetDevice;
+}
+
 std::string
 TopologyReader::Link::GetToNodeName (void) const
 {
diff --git a/src/topology-read/model/topology-reader.h b/src/topology-read/model/topology-reader.h
index 865b7d3..48c0c1d 100644
--- a/src/topology-read/model/topology-reader.h
+++ b/src/topology-read/model/topology-reader.h
@@ -32,6 +32,8 @@
 
 namespace ns3 {
 
+class NetDevice;
+
 /**
  * \ingroup topology
  *
@@ -53,7 +55,7 @@ public:
    */
   class Link
   {
-public:
+  public:
   /**
    * \brief Constant iterator to scan the map of link attributes.
    */
@@ -66,14 +68,27 @@ public:
      * \param toPtr Ptr to the node the link is directed to
      * \param toName name of the node the link is directed to
      */
-    Link ( Ptr<Node> fromPtr, const std::string &fromName, Ptr<Node> toPtr, const std::string &toName );
+    Link (Ptr<Node> fromPtr, const std::string &fromName, Ptr<Node> toPtr, const std::string &toName);
 
     /**
+     * \brief Set netdevices associated with the link
+     * \param from NetDevice associated with From node
+     * \param to NetDevice associated with To node
+     */
+    void
+    SetNetDevices (Ptr<NetDevice> from, Ptr<NetDevice> to);
+    
+    /**
      * \brief Returns a Ptr<Node> to the "from" node of the link
      * \return a Ptr<Node> to the "from" node of the link
      */
     Ptr<Node> GetFromNode (void) const;
     /**
+     * \brief Returns a Ptr<NetDevice> of the "from" node of the link
+     * \return a Ptr<NetDevice> of the "from" node of the link
+     */
+    Ptr<NetDevice> GetFromNetDevice (void) const;
+    /**
      * \brief Returns the name of the "from" node of the link
      * \return the name of the "from" node of the link
      */
@@ -84,6 +99,11 @@ public:
      */
     Ptr<Node> GetToNode (void) const;
     /**
+     * \brief Returns a Ptr<NetDevice> of the "to" node of the link
+     * \return a Ptr<NetDevice> of the "to" node of the link
+     */
+    Ptr<NetDevice> GetToNetDevice (void) const;
+    /**
      * \brief Returns the name of the "to" node of the link
      * \return the name of the "to" node of the link
      */
@@ -127,7 +147,9 @@ private:
     Ptr< Node > m_fromPtr;
     std::string m_toName;
     Ptr< Node > m_toPtr;
-    std::map<std::string, std::string> m_linkAttr;
+    Ptr< NetDevice > m_fromNetDevice;
+    Ptr< NetDevice > m_toNetDevice;
+    std::map<std::string, std::string > m_linkAttr;
   };
 
   /**
@@ -195,13 +217,14 @@ private:
    */
   void AddLink (Link link);
 
+protected:
+  std::string m_fileName;
+  std::list<Link> m_linksList;
+
 private:
   TopologyReader (const TopologyReader&);
   TopologyReader& operator= (const TopologyReader&);
 
-  std::string m_fileName;
-  std::list<Link> m_linksList;
-
   // end class TopologyReader
 };
 
-- 
1.7.9.5