testing_strategy.h

Go to the documentation of this file.

/*
* Copyright (C) 2020-2026 MEmilio
*
* Authors: Elisabeth Kluth, David Kerkmann, Sascha Korf, Martin J. Kuehn, Khoa Nguyen
*
* Contact: Martin J. Kuehn <Martin.Kuehn@DLR.de>
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
*     http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef MIO_ABM_TESTING_SCHEME_H
#define MIO_ABM_TESTING_SCHEME_H
 
#include "abm/config.h"
#include "abm/location_id.h"
#include "abm/location_type.h"
#include "abm/parameters.h"
#include "abm/person.h"
#include "abm/location.h"
#include "abm/time.h"
#include "memilio/io/default_serialize.h"
 
#include <bitset>
#include <vector>
 
namespace mio
{
namespace abm
{
 
class TestingCriteria
{
public:
    TestingCriteria() = default;
 
    TestingCriteria(const std::vector<AgeGroup>& ages, const std::vector<InfectionState>& infection_states);
 
    bool operator==(const TestingCriteria& other) const;
 
    bool evaluate(const Person& p, TimePoint t) const;
 
    auto default_serialize()
    {
        return Members("TestingCriteria").add("ages", m_ages).add("infection_states", m_infection_states);
    }
 
private:
    std::bitset<MAX_NUM_AGE_GROUPS> m_ages; 
    std::bitset<(size_t)InfectionState::Count>
        m_infection_states; 
};
 
class TestingScheme
{
public:
    TestingScheme(const TestingCriteria& testing_criteria, TimeSpan validity_period, TimePoint start_date,
                  TimePoint end_date, TestParameters test_parameters, ScalarType probability);
 
    bool operator==(const TestingScheme& other) const;
 
    bool is_active(TimePoint t) const;
 
    bool run_and_test(PersonalRandomNumberGenerator& rng, Person& person, TimePoint t) const;
 
    auto default_serialize()
    {
        return Members("TestingScheme")
            .add("criteria", m_testing_criteria)
            .add("validity_period", m_validity_period)
            .add("start_date", m_start_date)
            .add("end_date", m_end_date)
            .add("test_params", m_test_parameters)
            .add("probability", m_probability);
    }
 
private:
    friend DefaultFactory<TestingScheme>;
    TestingScheme() = default;
 
    mio::abm::TimePoint get_start_date() const
    {
        return m_start_date;
    }
 
    mio::abm::TimePoint get_end_date() const
    {
        return m_end_date;
    }
 
    TestingCriteria m_testing_criteria; 
    TimeSpan m_validity_period; 
    TimePoint m_start_date; 
    TimePoint m_end_date; 
    TestParameters m_test_parameters; 
    ScalarType m_probability; 
};
 
class TestingStrategy
{
public:
    struct LocalStrategy {
        std::vector<TestingScheme> schemes;
 
        auto default_serialize()
        {
            return Members("LocalStrategy").add("schemes", schemes);
        }
    };
 
    TestingStrategy() = default;
    explicit TestingStrategy(const std::vector<LocalStrategy>& location_to_schemes_id,
                             const std::vector<LocalStrategy>& location_to_schemes_type);
 
    void add_scheme(const LocationId& loc_id, const TestingScheme& scheme);
    void add_scheme(const LocationType& loc_type, const TestingScheme& scheme);
 
    void add_scheme(const std::vector<LocationType>& loc_type, const TestingScheme& scheme)
    {
        for (auto& type : loc_type) {
            add_scheme(type, scheme);
        }
    }
 
    bool run_and_check(PersonalRandomNumberGenerator& rng, Person& person, const Location& location, TimePoint t);
 
    auto default_serialize()
    {
        return Members("TestingStrategy")
            .add("schemes_id", m_testing_schemes_at_location_id)
            .add("schemes_type", m_testing_schemes_at_location_type);
    }
 
private:
    std::vector<LocalStrategy>
        m_testing_schemes_at_location_id; 
    std::vector<LocalStrategy>
        m_testing_schemes_at_location_type; 
};
 
} // namespace abm
} // namespace mio
 
#endif