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:
    explicit ModelBuilderBase(StateMachineEngine& engine)
        : sm("StateMachine"), mm(sm), m_engine(engine) {
    }
 
    virtual ~ModelBuilderBase() = default;
 
    void RegisterModel() {
        auto& logger = GetLogger("rtctk");
        LOG4CPLUS_DEBUG(logger, "StateMachine.DEBUG\n" + Export(sm, "debug").str());
        LOG4CPLUS_DEBUG(logger, "StateMachine.SCXML\n" + Export(sm, "scxml").str());
 
        // TODO: validate model
 
        m_engine.RegisterModel("model", Export(sm, "scxml").str());
    }
 
    void ExportModel(const std::string& file_name) {
        auto ends_with = [](const std::string& value, const std::string& 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, "puml")) {
            std::ofstream out(file_name);
            out << Export(sm, "puml").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 + "'");
        }
    }
 
    StateMachine sm;
    ModelManipulator mm;
    StateMachineEngine& m_engine;
};
 
}  // namespace rtctk::componentFramework
 
#endif