model_states.h

Go to the documentation of this file.

#pragma once
 
#include <libopynsim/model_state.h>
 
#include <concepts>
#include <cstddef>
#include <memory>
#include <ranges>
#include <utility>
#include <vector>
 
namespace opyn
{
    class ModelStates final {
    public:
        using value_type = ModelState;
        using size_type = size_t;
        using reference = value_type&;
        using const_reference = const value_type&;
        using shared_handle = std::shared_ptr<value_type>;
 
        ModelStates() = default;
 
        size_t size() const { return states_.size(); }
 
        reference at(size_t pos) { return *states_.at(pos); }
 
        const_reference at(size_t pos) const { return *states_.at(pos); }
 
        void reserve(size_type new_cap) { states_.reserve(new_cap); }
 
        template<typename... Args>
        requires std::constructible_from<ModelState, Args&&...>
        reference emplace_back(Args&&... args)
        {
            return *states_.emplace_back(std::make_shared<ModelState>(std::forward<Args>(args)...));
        }
 
        void handle_push_back(shared_handle handle) { states_.push_back(std::move(handle)); }
 
        shared_handle handle_at(size_t pos) { return states_.at(pos); }
 
        std::vector<shared_handle> to_handle_list() const { return states_; }
 
    private:
        auto view() const { return states_ | std::views::transform([](const auto& handle) -> const_reference { return *handle; }); }
    public:
        auto begin() const { return view().begin(); }
        auto end() const { return view().end(); }
 
    private:
        std::vector<shared_handle> states_;
    };
}