Checkpointable.hpp

// <Checkpointable> -*- C++ -*-
 
#pragma once
 
#include <cassert>
#include <cstring>
#include <list>
#include <memory>
#include <type_traits>
#include <vector>
 
#include "sparta/functional/ArchData.hpp"
#include "sparta/simulation/TreeNode.hpp"
#include "sparta/utils/MathUtils.hpp"
 
namespace sparta::serialization::checkpoint {
    class Checkpointable
    {
    public:
 
        Checkpointable(sparta::TreeNode *cp_node) : cp_node_(cp_node) {}
 
        template <class CheckpointableT, typename... CpArgs>
        CheckpointableT &allocateCheckpointable(CpArgs... cp_args)
        {
            static_assert(false == std::is_pointer_v<CheckpointableT>,
                          "Checkpointable object cannot be a pointer");
            constexpr size_t checkpointable_size =
                          utils::next_power_of_2(sizeof(CheckpointableT));
            auto &cp_component = checkpoint_components_.emplace_back(
                new CheckpointComponent(cp_node_, checkpointable_size));
 
            auto cp_mem = cp_component->getRawDataPtr();
            return *(new (cp_mem) CheckpointableT(cp_args...));
        }
 
    private:
        sparta::TreeNode *cp_node_ = nullptr;
 
        struct CheckpointComponent
        {
            CheckpointComponent(sparta::TreeNode *cp_node,
                                ArchData::offset_type line_size)
                : adata_(cp_node, line_size, ArchData::DEFAULT_INITIAL_FILL,
                         ArchData::DEFAULT_INITIAL_FILL_SIZE,
                         false), // Cannot delete lines
                  dview_(&adata_, 0, line_size,
                         ArchDataSegment::INVALID_ID, // subset of
                         0)                           // subset_offset
            {
                adata_.layout();
            }
 
            uint8_t *getRawDataPtr() { return dview_.getLine()->getRawDataPtr(0); }
 
            ArchData adata_;
 
            DataView dview_;
        };
 
        std::vector<std::unique_ptr<CheckpointComponent>> checkpoint_components_;
    };
} // namespace sparta