EntitySnapshotBase.h

Go to the documentation of this file.

#pragma once
 
#include <functional>
 
// BaseClass
#include <RobotAPI/libraries/armem/core/MemoryID.h>
#include <RobotAPI/libraries/armem/core/wm/aron_conversions.h>
#include <RobotAPI/libraries/armem/core/wm/memory_definitions.h>
 
#include "EntityBase.h"
#include "MemoryItem.h"
 
namespace armarx::armem::server::ltm::detail
{
    /// @brief Interface functions for the longterm memory classes
    template <class InstanceT>
    class EntitySnapshotBase : public MemoryItem
    {
    public:
        struct Statistics
        {
            long recordedInstances = 0;
        };
 
    public:
        using MemoryItem::MemoryItem;
 
        /// return the full sub-ltm as a wm::EntitySnapshot with only references
        /// the ltm may be huge, use with caution
        void
        loadAllReferences(armem::wm::EntitySnapshot& e) const
        {
            _loadAllReferences(e);
        }
 
        /// convert the references of the input into a wm::Memory
        void
        resolve(armem::wm::EntitySnapshot& e) const
        {
            _resolve(e);
        }
 
        /// encode the content of a wm::Memory and store
        void
        store(const armem::wm::EntitySnapshot& e)
        {
            _store(e);
        }
 
        /// statistics
        void
        resetStatistics()
        {
            statistics.recordedInstances = 0;
        }
 
        Statistics
        getStatistics() const
        {
            return statistics;
        }
 
        static std::string
        getLevelName()
        {
            return "LT-EntitySnapshot";
        }
 
        /// iterate over all Instance segments of this ltm
        bool
        forEachInstance(std::function<void(InstanceT&)> func) const
        {
            return _implForEachInstance(func);
        }
 
        /// check if Instance segment exists
        bool
        hasInstance(const int instanceId) const
        {
            return _implHasInstance(instanceId);
        }
 
        /// find Instance segment
        std::shared_ptr<InstanceT>
        findInstance(const int instanceId) const
        {
            return _implFindInstance(instanceId);
        }
 
    protected:
        virtual void _loadAllReferences(armem::wm::EntitySnapshot&) const = 0;
        virtual void _resolve(armem::wm::EntitySnapshot&) const = 0;
        virtual void _store(const armem::wm::EntitySnapshot& s) = 0;
 
        // Implementation methods - subclasses override these
        // Thread safety is handled by DiskPersistence's per-directory mutexes
        virtual bool _implForEachInstance(std::function<void(InstanceT&)> func) const = 0;
        virtual bool _implHasInstance(const int instanceId) const = 0;
        virtual std::shared_ptr<InstanceT> _implFindInstance(const int instanceId) const = 0;
 
    protected:
        Statistics statistics;
    };
} // namespace armarx::armem::server::ltm::detail