StuckDetection.hpp

Go to the documentation of this file.

 
#ifndef STUCK_DETECTION_HPP
#define STUCK_DETECTION_HPP
 
// Put implicit includes in here.
#include <chrono>
#include <cmath>
 
 
 
namespace statemachine
{
 
    class TimeIntervalBasedStuckDetector
    {
        private:
            // Declare private member variables.
 
            double m_dMaximumStuckCount;
            double m_dStuckCheckIntervalSeconds;
            unsigned int m_unStuckChecksSoFar;
            std::chrono::system_clock::time_point m_tmTimeSinceLastStuckCheck;
 
        public:
            // Declare public class methods.
 
 
            TimeIntervalBasedStuckDetector(double dMaximumStuckCount = 3, double dStuckCheckIntervalSeconds = 3)
            {
                // Initialize member variables.
                m_dMaximumStuckCount         = dMaximumStuckCount;
                m_dStuckCheckIntervalSeconds = dStuckCheckIntervalSeconds;
                m_unStuckChecksSoFar         = 0;
                m_tmTimeSinceLastStuckCheck  = std::chrono::system_clock::now();
            }
 
 
            ~TimeIntervalBasedStuckDetector() {}
 
 
            bool CheckIfStuck(double dCurrentVelocity, double dCurrentAngularVelocity, double dVelocityThreshold = 0.1, double dAngularVelocityThreshold = 0.1)
            {
                // Create instance variables.
                bool bStuck = false;
 
                // Time since we last checked if the rover is stuck.
                std::chrono::system_clock::time_point tmCurrentTime = std::chrono::system_clock::now();
                double dTimeSinceLastCheck = std::chrono::duration_cast<std::chrono::milliseconds>(tmCurrentTime - m_tmTimeSinceLastStuckCheck).count() / 1000.0;
                if (dTimeSinceLastCheck > m_dStuckCheckIntervalSeconds)
                {
                    // Update time since last check to now.
                    m_tmTimeSinceLastStuckCheck = tmCurrentTime;
 
                    // Check if the rover is rotating or moving linearly.
                    if (std::abs(dCurrentVelocity) < dVelocityThreshold && std::abs(dCurrentAngularVelocity) < dAngularVelocityThreshold)
                    {
                        ++m_unStuckChecksSoFar;
                    }
                    else
                    {
                        m_unStuckChecksSoFar = 0;
                    }
                }
 
                // Check if we met stuck criteria.
                if (m_unStuckChecksSoFar > m_dMaximumStuckCount)
                {
                    // Reset stuck checks.
                    m_unStuckChecksSoFar = 0;
                    // Set stuck toggle.
                    bStuck = true;
                }
 
                // Return if stuck or not.
                return bStuck;
            }
 
 
            void ResetStuckChecks() { m_unStuckChecksSoFar = 0; }
    };
}    // namespace statemachine
 
#endif