From 35ed9727d3528b84eb551cbf0784c92e0e9abac0 Mon Sep 17 00:00:00 2001
From: "Gustavo J. A. M. Carneiro" <gjc@inescporto.pt>
Date: Fri, 28 Sep 2007 18:17:25 +0100
Subject: [PATCH] Add NotifyCourseChange support to
 RandomWaypointMobilityModel, and fix a bug in GetSpeed.

---
 .../random-waypoint-mobility-model.cc         | 24 +++++++++++++------
 src/mobility/random-waypoint-mobility-model.h |  2 ++
 2 files changed, 19 insertions(+), 7 deletions(-)

diff --git a/src/mobility/random-waypoint-mobility-model.cc b/src/mobility/random-waypoint-mobility-model.cc
index 678273178..4db4f2e11 100644
--- a/src/mobility/random-waypoint-mobility-model.cc
+++ b/src/mobility/random-waypoint-mobility-model.cc
@@ -1,3 +1,4 @@
+#include <cmath>
 #include "ns3/simulator.h"
 #include "ns3/random-variable.h"
 #include "ns3/random-variable-default-value.h"
@@ -91,23 +92,32 @@ RandomWaypointMobilityModel::RandomWaypointMobilityModel (Ptr<RandomWaypointMobi
   : m_parameters (parameters)
 {
   Simulator::ScheduleNow (&RandomWaypointMobilityModel::Start, this);
+  NotifyCourseChange ();
+}
+
+void
+RandomWaypointMobilityModel::BeginPause (void)
+{
+  m_event = Simulator::Schedule (m_pause, &RandomWaypointMobilityModel::Start, this);
 }
 
 void
 RandomWaypointMobilityModel::Start (void)
 {
+  m_pause = Seconds (m_parameters->m_pause->GetValue ());
   Position m_current = m_helper.GetCurrentPosition ();
   Position destination = m_parameters->m_position->Get ();
   double speed = m_parameters->m_speed->GetValue ();
-  Time pause = Seconds (m_parameters->m_pause->GetValue ());
-  double dx = (destination.x - m_current.x) * speed;
-  double dy = (destination.y - m_current.y) * speed;
-  double dz = (destination.z - m_current.z) * speed;
+  double dx = (destination.x - m_current.x);
+  double dy = (destination.y - m_current.y);
+  double dz = (destination.z - m_current.z);
+  double k = speed / std::sqrt (dx*dx + dy*dy + dz*dz);
 
-  m_helper.Reset (Speed (dx,dy,dz), pause);
+  m_helper.Reset (Speed (k*dx, k*dy, k*dz), m_pause);
   Time travelDelay = Seconds (CalculateDistance (destination, m_current) / speed);
-  m_event = Simulator::Schedule (travelDelay + pause,
-				 &RandomWaypointMobilityModel::Start, this);
+  m_event = Simulator::Schedule (travelDelay,
+				 &RandomWaypointMobilityModel::BeginPause, this);
+  NotifyCourseChange ();
 }
 
 Position 
diff --git a/src/mobility/random-waypoint-mobility-model.h b/src/mobility/random-waypoint-mobility-model.h
index aef1c3d86..147302383 100644
--- a/src/mobility/random-waypoint-mobility-model.h
+++ b/src/mobility/random-waypoint-mobility-model.h
@@ -97,6 +97,7 @@ public:
   RandomWaypointMobilityModel (Ptr<RandomWaypointMobilityModelParameters> parameters);
 private:
   void Start (void);
+  void BeginPause (void);
   virtual Position DoGet (void) const;
   virtual void DoSet (const Position &position);
   virtual Speed DoGetSpeed (void) const;
@@ -104,6 +105,7 @@ private:
   StaticSpeedHelper m_helper;
   Ptr<RandomWaypointMobilityModelParameters> m_parameters;
   EventId m_event;
+  Time m_pause;
 };
 
 } // namespace ns3