MemoryBase.h

Go to the documentation of this file.

#pragma once
 
#include <functional>
 
 
// BaseClass
#include "MemoryItem.h"
 
// ChildType
 
// ArmarX
#include <ArmarXCore/core/Component.h>
#include <ArmarXCore/core/logging/LoggingUtil.h>
#include <ArmarXCore/core/time.h>
 
#include <RobotAPI/libraries/armem/core/MemoryID.h>
#include <RobotAPI/libraries/armem/core/operations.h>
#include <RobotAPI/libraries/armem/core/wm/aron_conversions.h>
#include <RobotAPI/libraries/armem/core/wm/memory_definitions.h>
#include <RobotAPI/libraries/armem/server/test/ForgettingExperiments.h>
#include <RobotAPI/libraries/armem/server/wm/memory_definitions.h>
 
namespace armarx::armem::server::ltm::detail
{
    /// @brief Interface functions for the longterm memory classes
    template <class _CoreSegmentT>
    class MemoryBase : public MemoryItem
    {
    public:
        struct Statistics
        {
            armarx::core::time::DateTime lastEnabled = armarx::core::time::DateTime::Invalid();
            long recordedCoreSegments = 0;
            bool firstStart = true;
            bool firstStop = true;
            armarx::core::time::DateTime firstStarted = armarx::core::time::DateTime::Invalid();
            armarx::core::time::DateTime firstStopped = armarx::core::time::DateTime::Invalid();
            armarx::core::time::DateTime lastStarted = armarx::core::time::DateTime::Invalid();
            armarx::core::time::DateTime lastStopped = armarx::core::time::DateTime::Invalid();
        };
 
    public:
        using CoreSegmentT = _CoreSegmentT;
 
        MemoryBase(const std::string& exportName, const MemoryID& id) :
            MemoryItem(exportName, id), enabled(false)
        {
        }
 
        /// initialize config
        void
        configure()
        {
            bool en = p.enabled_on_startup;
            ARMARX_INFO << VAROUT(p.configuration_on_startup);
            ARMARX_INFO << VAROUT(p.export_name);
            this->setExportName(p.export_name);
 
            try
            {
                const auto j = nlohmann::json::parse(p.configuration_on_startup);
 
                // Processors are shared. So we only need to configure the root
                processors->configure(j);
 
                this->_configure(j);
            }
            catch (...)
            {
                ARMARX_WARNING << "Failed to parse `" << p.configuration_on_startup << "`";
                en = false;
            }
 
            if (en)
            {
                this->startRecording();
            }
        }
 
        /// enable this LTM
        void
        enable()
        {
            processors->resetFilterStatisticsForNewEpisode(); //make sure the statistics are
            //resetted before the recording starts
            auto now = armarx::core::time::DateTime::Now();
            ARMARX_INFO << "Enabling LTM " << id().str() << " at " << now.toDateTimeString();
            ARMARX_INFO << "Storing information at " << this->getExportName();
            enabled = true;
            if (statistics.firstStart)
            {
                statistics.firstStarted = now;
                statistics.firstStart = false;
            }
            statistics.lastStarted = now;
            this->_enable();
        }
 
        /// disable this LTM
        void
        disable()
        {
            auto now = armarx::core::time::DateTime::Now();
            ARMARX_INFO << "Disabling LTM " << id().str() << " at " << now.toDateTimeString();
            enabled = false;
            if (statistics.firstStop)
            {
                statistics.firstStopped = now;
                statistics.firstStop = false;
            }
            statistics.lastStopped = now;
            this->_disable();
            ARMARX_INFO << "Disabling of LTM finished";
        }
 
        /// return the full ltm as a wm::Memory with only references
        /// the ltm may be huge, use with caution
        armem::wm::Memory
        loadAllReferences()
        {
            armem::wm::Memory ret;
            loadAllReferences(ret);
            return ret;
        }
 
        void
        loadAllReferences(armem::wm::Memory& memory)
        {
            TIMING_START(LTM_Memory_LoadAll);
            _loadAllReferences(memory);
            TIMING_END_STREAM(LTM_Memory_LoadAll, ARMARX_DEBUG);
        }
 
        /// return the full ltm as a wm::Memory and resolves the references
        /// the ltm may be huge, use with caution
        armem::wm::Memory
        loadAllAndResolve()
        {
            armem::wm::Memory ret;
            loadAllAndResolve(ret);
            return ret;
        }
 
        void
        loadAllAndResolve(armem::wm::Memory& memory)
        {
            TIMING_START(LTM_Memory_LoadAllAndResolve);
            _loadAllReferences(memory);
            _resolve(memory);
            TIMING_END_STREAM(LTM_Memory_LoadAllAndResolve, ARMARX_DEBUG);
        }
 
        /// convert the references of the input into a wm::Memory
        void
        resolve(armem::wm::Memory& memory)
        {
            TIMING_START(LTM_Memory_Load);
            _resolve(memory);
            TIMING_END_STREAM(LTM_Memory_Load, ARMARX_DEBUG);
        }
 
        /// append a wm::Memory instance to the ltm
        void
        store(const armem::wm::Memory& memory)
        {
            TIMING_START(LTM_Memory_Append);
 
            for (auto& f : processors->memFilters)
            {
                if (!f->accept(memory))
                {
                    ARMARX_INFO << deactivateSpam()
                                << "Ignoring to put a Memory into the LTM because it got filtered.";
                    return;
                }
            }
            _directlyStore(memory);
            TIMING_END_STREAM(LTM_Memory_Append, ARMARX_DEBUG);
        }
 
        /// append a wm::Memory instance to the ltm
        void
        store(const armem::server::wm::Memory& serverMemory)
        {
            wm::Memory memory;
            memory.update(armem::toCommit(serverMemory));
            this->store(memory);
        }
 
        /// iterate over all core segments of this ltm
        virtual bool forEachCoreSegment(std::function<void(CoreSegmentT&)> func) const = 0;
 
        /// check if core segment exists
        virtual bool hasCoreSegment(const std::string&) const = 0;
 
        /// find core segment
        virtual std::unique_ptr<CoreSegmentT> findCoreSegment(const std::string&) const = 0;
 
        /// default parameters. Implementation should use the configuration to configure
        virtual void
        createPropertyDefinitions(PropertyDefinitionsPtr& defs, const std::string& prefix)
        {
            defs->optional(p.enabled_on_startup, prefix + "enabled");
            defs->optional(p.configuration_on_startup, prefix + "configuration");
            defs->optional(p.export_name, prefix + "exportName");
        }
 
        /// enable/disable
        void
        startRecording()
        {
            statistics.lastEnabled = armarx::core::time::DateTime::Now();
 
            enable();
        }
 
        void
        stopRecording()
        {
 
            disable();
        }
 
        bool
        isRecording() const
        {
            return enabled;
        }
 
        /// statistics
        virtual void
        resetStatistics()
        {
            // enabled stays the same
            statistics.lastEnabled = armarx::core::time::DateTime::Invalid();
            statistics.recordedCoreSegments = 0;
        }
 
        Statistics
        getStatistics() const
        {
            return statistics;
        }
 
        std::map<std::string, processor::SnapshotFilter::FilterStatistics>
        getFilterStatistics()
        {
            try
            {
                ARMARX_INFO << "Trying to save statistics";
                auto stats = processors->getSnapshotFilterStatistics();
                return stats;
            }
            catch (InvalidArgumentException& e)
            { //no data was actually recorded
                ARMARX_INFO << e.what();
            }
            catch (...)
            {
                ARMARX_WARNING << "Saving statistics did not work";
            }
            std::map<std::string, processor::SnapshotFilter::FilterStatistics> emptyStatistics;
            return emptyStatistics;
        }
 
        /// get level name1
        static std::string
        getLevelName()
        {
            return "LT-Memory";
        }
 
    protected:
        /// configuration
        virtual void
        _configure(const nlohmann::json&)
        {
        }
 
        virtual void
        _enable()
        {
        }
 
        virtual void
        _disable()
        {
        }
 
        virtual void
        _setExportName(const std::string&)
        {
        }
 
        virtual void _loadAllReferences(armem::wm::Memory& memory) = 0;
        virtual void _resolve(armem::wm::Memory& memory) = 0;
        virtual void _store(const armem::wm::Memory& memory) = 0;
        virtual void _directlyStore(const armem::wm::Memory& memory) = 0;
 
    public:
        // stuff for scenario parameters
        struct Properties
        {
            bool enabled_on_startup = false;
            std::string configuration_on_startup =
                "{ \"SnapshotFrequencyFilter\": {\"WaitingTimeInMsForFilter\" : 50}, "
                "\"PngConverter\": {}, \"ExrConverter\": {}}"; //record with 20 fps as standard
            std::string export_name = "MemoryExport";
            std::string export_path = "/tmp/ltm";
        } p;
 
    protected:
        mutable std::recursive_mutex ltm_mutex;
 
        mutable Statistics statistics;
 
        std::atomic_bool enabled = false;
    };
} // namespace armarx::armem::server::ltm::detail