ProviderSegmentBase.h

Go to the documentation of this file.

#pragma once
 
#include <functional>
 
#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 _EntityT>
    class ProviderSegmentBase : public MemoryItem
    {
    public:
        struct Statistics
        {
            long recordedEntities = 0;
        };
 
    public:
        using EntityT = _EntityT;
 
        using MemoryItem::MemoryItem;
 
        /// return the full sub-ltm as a wm::ProviderSegment with only references
        /// the ltm may be huge, use with caution
        void
        loadAllReferences(armem::wm::ProviderSegment& provSeg)
        {
            _loadAllReferences(provSeg);
        }
 
        void
        loadLatestNReferences(int n, armem::wm::ProviderSegment& provSeg)
        {
            _loadLatestNReferences(n, provSeg);
        }
 
        /// convert the references of the input into a wm::Memory
        void
        resolve(armem::wm::ProviderSegment& provSeg)
        {
            _resolve(provSeg);
        }
 
        /// encode the content of a wm::Memory and store
        void
        store(const armem::wm::ProviderSegment& provSeg, bool simulatedVersion)
        {
            _store(provSeg, simulatedVersion);
        }
 
        /// statistics
        void
        resetStatistics()
        {
            statistics.recordedEntities = 0;
        }
 
        Statistics
        getStatistics() const
        {
            return statistics;
        }
 
        /// iterate over all core segments of this ltm
        bool
        forEachEntity(std::function<void(EntityT&)> func) const
        {
            return _implForEachEntity(func);
        }
 
        /// check if entity segment exists
        bool
        hasEntity(const std::string& entityName) const
        {
            return _implHasEntity(entityName);
        }
 
        /// find entity segment
        std::shared_ptr<EntityT>
        findEntity(const std::string& entityName) const
        {
            return _implFindEntity(entityName);
        }
 
        ///get aron type
        aron::type::ObjectPtr
        aronType() const
        {
            return nullptr;
        }
 
        static std::string
        getLevelName()
        {
            return "LT-ProviderSegment";
        }
 
    protected:
        virtual void _loadAllReferences(armem::wm::ProviderSegment&) = 0;
        virtual void _loadLatestNReferences(int n, armem::wm::ProviderSegment& p) = 0;
        virtual void _resolve(armem::wm::ProviderSegment&) = 0;
        virtual void _store(const armem::wm::ProviderSegment& p, bool simulatedVersion = false) = 0;
 
        // Implementation methods - subclasses override these
        // Thread safety is handled by DiskPersistence's per-directory mutexes
        virtual bool _implForEachEntity(std::function<void(EntityT&)> func) const = 0;
        virtual bool _implHasEntity(const std::string& entityName) const = 0;
        virtual std::shared_ptr<EntityT> _implFindEntity(const std::string& entityName) const = 0;
 
    protected:
        Statistics statistics;
    };
} // namespace armarx::armem::server::ltm::detail