ContactPool.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

#include "contact/ContactLogMode.h"
#include "contact/ContactType.h"

#include <tuple>
#include <vector>

namespace stride {

class Person;

class ContactPool
{
public:
        ContactPool(unsigned int poolId, ContactType::Id type);

        ~ContactPool() = default;

        void AddMember(Person* p);

        unsigned int GetId() const { return m_pool_id; }

        unsigned int GetInfectedCount() const;

        const std::vector<Person*>& GetPool() const { return m_members; }

        ContactType::Id GetType() const { return m_pool_type; }

public:
        // To iterate over the members.
        using iterator = std::vector<stride::Person*>::iterator;

        iterator begin() { return m_members.begin(); }

        iterator end() { return m_members.end(); }

        size_t size() const { return m_members.size(); }

        Person* const& operator[](size_t index) const { return m_members[index]; }

private:
        std::tuple<bool, unsigned int> SortMembers();

        template <ContactLogMode::Id LL, bool TIC, bool TO>
        friend class Infector;

private:
        unsigned int         m_index_immune; 
        unsigned int         m_pool_id;      
        ContactType::Id      m_pool_type;    
        std::vector<Person*> m_members;      
};

} // namespace stride