Health.h

Go to the documentation of this file.

/*
 *  This is free software: you can redistribute it and/or modify it
 *  under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  any later version.
 *  The software is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *  You should have received a copy of the GNU General Public License
 *  along with the software. If not, see <http://www.gnu.org/licenses/>.
 *
 *  Copyright 2017, Kuylen E, Willem L, Broeckhove J
 */

#pragma once

namespace stride {

enum class HealthStatus : unsigned short int
{
        Susceptible              = 0U,
        Exposed                  = 1U,
        Infectious               = 2U,
        Symptomatic              = 3U,
        InfectiousAndSymptomatic = 4U,
        Recovered                = 5U,
        Immune                   = 6U
};

class Health
{
public:
        explicit Health(unsigned short int start_infectiousness = 0U, unsigned int short start_symptomatic = 0U,
                        unsigned short int time_infectious = 0U, unsigned short int time_symptomatic = 0U);

        unsigned short int GetEndInfectiousness() const { return m_end_infectiousness; }

        unsigned short int GetEndSymptomatic() const { return m_end_symptomatic; }

        unsigned short int GetStartInfectiousness() const { return m_start_infectiousness; }

        unsigned short int GetStartSymptomatic() const { return m_start_symptomatic; }

        bool IsImmune() const { return m_status == HealthStatus::Immune; }

        bool IsInfected() const
        {
                return m_status == HealthStatus::Exposed || m_status == HealthStatus::Infectious ||
                       m_status == HealthStatus::InfectiousAndSymptomatic || m_status == HealthStatus::Symptomatic;
        }

        bool IsInfectious() const
        {
                return m_status == HealthStatus::Infectious || m_status == HealthStatus::InfectiousAndSymptomatic;
        }

        bool IsRecovered() const { return m_status == HealthStatus::Recovered; }

        bool IsSusceptible() const { return m_status == HealthStatus::Susceptible; }

        bool IsSymptomatic() const
        {
                return m_status == HealthStatus::Symptomatic || m_status == HealthStatus::InfectiousAndSymptomatic;
        }

        bool SymptomsStartedToday() const { return GetDiseaseCounter() == m_start_symptomatic; }

        void SetImmune() { m_status = HealthStatus::Immune; }

        void SetSusceptible() { m_status = HealthStatus::Susceptible; }

        void StartInfection();

        void StopInfection();

        void Update();

private:
        unsigned short int GetDiseaseCounter() const { return m_disease_counter; }

        void IncrementDiseaseCounter() { m_disease_counter++; }

        void ResetDiseaseCounter() { m_disease_counter = 0U; }

private:
        unsigned short int m_disease_counter; 
        HealthStatus       m_status;          

        unsigned short int m_start_infectiousness; 
        unsigned short int m_start_symptomatic;    
        unsigned short int m_end_infectiousness;   
        unsigned short int m_end_symptomatic;      
};

} // namespace stride