modelBuilderBase.hpp

Go to the documentation of this file.

 
#ifndef RTCTK_COMPONENTFRAMEWORK_MODELBUILDERBASE_HPP
#define RTCTK_COMPONENTFRAMEWORK_MODELBUILDERBASE_HPP
 
#include <rtctk/componentFramework/logger.hpp>
#include <rtctk/componentFramework/model.hpp>
#include <rtctk/componentFramework/modelExport.hpp>
#include <rtctk/componentFramework/modelManipulator.hpp>
#include <rtctk/componentFramework/stateMachineEngine.hpp>
 
#include <fstream>
#include <iostream>
#include <map>
#include <tuple>
#include <vector>
 
namespace rtctk::componentFramework {
 
class ModelBuilderBase {
public:
    ModelBuilderBase(StateMachineEngine& engine) : sm("StateMachine"), mm(sm), m_engine(engine) {
    }
 
    virtual ~ModelBuilderBase() = default;
 
    void RegisterModel() {
        LOG4CPLUS_DEBUG(GetLogger(), "StateMachine.DEBUG\n" + Export(sm, "debug").str());
        LOG4CPLUS_DEBUG(GetLogger(), "StateMachine.SCXML\n" + Export(sm, "scxml").str());
 
        // TODO: validate model
 
        m_engine.RegisterModel("model", Export(sm, "scxml").str());
    }
 
    void ExportModel(std::string const& file_name) {
        auto ends_with = [](std::string const& value, std::string const& ending) {
            if (ending.size() > value.size())
                return false;
            return std::equal(ending.rbegin(), ending.rend(), value.rbegin());
        };
 
        if (ends_with(file_name, "scxml")) {
            std::ofstream out(file_name);
            out << Export(sm, "scxml").str();
        } else if (ends_with(file_name, "smcat")) {
            std::ofstream out(file_name);
            out << Export(sm, "smcat").str();
        } else if (ends_with(file_name, "debug")) {
            std::ofstream out(file_name);
            out << Export(sm, "debug").str();
        } else {
            CII_THROW(RtctkException, "Unsupported export format: '" + file_name + "'");
        }
    }
 
protected:
    using LayerOptions = std::map<std::string, bool>;
    using Layer = std::tuple<std::string, LayerOptions>;
    using Layers = std::vector<Layer>;
 
    void RegisterLayer(Layer layer) {
        std::ostringstream msg;
        msg << "Applying Layer '" << std::get<0>(layer) << "' with Options:";
        for (const auto& [key, value] : std::get<1>(layer)) {
            msg << " " << key << ":" << value;
        }
        LOG4CPLUS_DEBUG(GetLogger(), msg.str());
 
        layers.push_back(layer);
    }
 
    StateMachine sm;
    ModelManipulator mm;
    StateMachineEngine& m_engine;
 
private:
    Layers layers;
};
 
}  // namespace rtctk::componentFramework
 
#endif