Add sat and gnd node helpers

helper/ground-node-helper.cc

@@ -0,0 +1,55 @@
+#include <fstream>
+
+#include "ns3/log.h"
+#include "ns3/config.h"
+#include "ns3/waypoint.h"
+
+#include "ground-node-helper.h"
+
+using namespace std;
+
+namespace ns3
+{
+NS_LOG_COMPONENT_DEFINE ("LeoGndNodeHelper");
+
+LeoGndNodeHelper::LeoGndNodeHelper ()
+{
+  m_gndNodeFactory.SetTypeId ("ns3::Node");
+}
+
+LeoGndNodeHelper::~LeoGndNodeHelper ()
+{
+}
+
+void
+LeoGndNodeHelper::SetAttribute (string name, const AttributeValue &value)
+{
+  m_gndNodeFactory.Set (name, value);
+}
+
+NodeContainer
+LeoGndNodeHelper::Install (const std::string &wpFile)
+{
+  NS_LOG_FUNCTION (wpFile);
+
+  NodeContainer nodes;
+  ifstream waypoints;
+  waypoints.open (wpFile, ifstream::in);
+  Vector pos;
+  while ((waypoints >> pos))
+    {
+      Ptr<ConstantPositionMobilityModel> mob = CreateObject<ConstantPositionMobilityModel> ();
+      mob->SetPosition (pos);
+      Ptr<Node> node = m_gndNodeFactory.Create<Node> ();
+      node->AggregateObject (mob);
+      nodes.Add (node);
+      NS_LOG_INFO ("Added ground node at " << pos);
+    }
+  waypoints.close ();
+
+  NS_LOG_INFO ("Added " << nodes.GetN () << " ground nodes");
+
+  return nodes;
+}
+
+}; // namespace ns3

helper/ground-node-helper.h

@@ -0,0 +1,48 @@
+/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
+
+#ifndef LEO_GND_NODE_HELPER_H
+#define LEO_GND_NODE_HELPER_H
+
+#include <string>
+
+#include "ns3/object-factory.h"
+#include "ns3/node-container.h"
+#include "ns3/constant-position-mobility-model.h"
+
+/**
+ * \brief Builds a node container of nodes with constant positions
+ *
+ * Adds waypoints from file for each node.
+ */
+
+namespace ns3
+{
+
+class LeoGndNodeHelper
+{
+public:
+  LeoGndNodeHelper ();
+  virtual ~LeoGndNodeHelper ();
+
+  /**
+   *
+   * \param wpFile path to waypoint file
+   * \returns a node container containing nodes using the specified attributes
+   */
+  NodeContainer Install (const std::string &wpFile);
+
+  /**
+   * Set an attribute for each node
+   *
+   * \param name name of the attribute
+   * \param value value of the attribute
+   */
+  void SetAttribute (std::string name, const AttributeValue &value);
+
+private:
+  ObjectFactory m_gndNodeFactory;
+};
+
+}; // namespace ns3
+
+#endif

helper/satellite-node-helper.cc

@@ -0,0 +1,55 @@
+#include "ns3/log.h"
+#include "ns3/config.h"
+#include "ns3/waypoint-mobility-model.h"
+
+#include "satellite-node-helper.h"
+
+using namespace std;
+
+namespace ns3
+{
+NS_LOG_COMPONENT_DEFINE ("LeoSatNodeHelper");
+
+LeoSatNodeHelper::LeoSatNodeHelper ()
+{
+  m_satNodeFactory.SetTypeId ("ns3::Node");
+}
+
+LeoSatNodeHelper::~LeoSatNodeHelper ()
+{
+}
+
+void
+LeoSatNodeHelper::SetAttribute (string name, const AttributeValue &value)
+{
+  m_satNodeFactory.Set (name, value);
+}
+
+NodeContainer
+LeoSatNodeHelper::Install (vector<string> &wpFiles)
+{
+  NS_LOG_FUNCTION (wpFiles);
+
+  NodeContainer nodes;
+  for (size_t i = 0; i < wpFiles.size (); i ++)
+    {
+      Ptr<WaypointMobilityModel> mob = CreateObject<WaypointMobilityModel> ();
+      string fileName = wpFiles[i];
+      m_fileStreamContainer.SetFile (fileName);
+      Waypoint wp;
+      while (m_fileStreamContainer.GetNextSample (wp))
+      	{
+      	  mob->AddWaypoint (wp);
+          NS_LOG_DEBUG ("Added waypoint " << wp);
+      	}
+      Ptr<Node> node = m_satNodeFactory.Create<Node> ();
+      node->AggregateObject (mob);
+
+      nodes.Add (node);
+      NS_LOG_INFO ("Added satellite node " << node->GetId ());
+    }
+
+  return nodes;
+}
+
+}; // namespace ns3

helper/satellite-node-helper.h

@@ -0,0 +1,51 @@
+/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
+
+#ifndef SAT_NODE_HELPER_H
+#define SAT_NODE_HELPER_H
+
+#include <string>
+
+#include "ns3/object-factory.h"
+#include "ns3/node-container.h"
+
+#include "ns3/leo-input-fstream-container.h"
+
+/**
+ * \brief Builds a node container with a waypoint mobility model
+ *
+ * Adds waypoints from file for each node.
+ * The node satId must must correspond to the NORAD id from Celestrack.
+ */
+
+namespace ns3
+{
+
+class LeoSatNodeHelper
+{
+public:
+  LeoSatNodeHelper ();
+  virtual ~LeoSatNodeHelper ();
+
+  /**
+   *
+   * \param nodeIds paths to satellite to waypoint files
+   * \returns a node container containing nodes using the specified attributes
+   */
+  NodeContainer Install (std::vector<std::string> &wpFiles);
+
+  /**
+   * Set an attribute for each node
+   *
+   * \param name name of the attribute
+   * \param value value of the attribute
+   */
+  void SetAttribute (std::string name, const AttributeValue &value);
+
+private:
+  ObjectFactory m_satNodeFactory;
+  LeoWaypointInputFileStreamContainer m_fileStreamContainer;
+};
+
+}; // namespace ns3
+
+#endif