ScriptNode.cpp

Go to the documentation of this file.

#include "run/ScriptNode.h"
#include "run/ScriptUtils.h"
 
NAMESPACE_SPH_BEGIN
 
#ifdef SPH_USE_CHAISCRIPT
 
namespace Chai {
 
class Node {
private:
    SharedPtr<JobNode> node;
    VirtualSettings settings;
 
    const RunSettings& global;
    IJobCallbacks& callbacks;
 
public:
    Node(SharedPtr<JobNode> node, const RunSettings& global, IJobCallbacks& callbacks)
        : node(node)
        , global(global)
        , callbacks(callbacks) {
        settings = node->getSettings();
    }
 
    template <typename T>
    void setParam(const std::string& key, const T value) {
        IVirtualEntry::Value current = settings.get(key);
        if (!current.has<T>()) {
            throw InvalidSetup("Type mismatch when setting parameter '" + key + "'");
        }
        settings.set(key, value);
    }
 
    void run() {
        if (!callbacks.shouldAbortRun()) {
            node->run(global, callbacks);
        }
    }
};
 
 
// either string, Path, or EnumWrapper
template <>
void Node::setParam(const std::string& key, const std::string value) {
    IVirtualEntry::Value current = settings.get(key);
    if (current.has<std::string>()) {
        settings.set(key, value);
    } else if (current.has<Path>()) {
        settings.set(key, Path(value));
    } else if (current.has<EnumWrapper>()) {
        EnumWrapper ew = current.get<EnumWrapper>();
        Optional<int> enumValue = EnumMap::fromString(value, ew.index);
        if (!enumValue) {
            throw InvalidSetup("Unknown value of parameter '" + key + "': " + value);
        }
        ew.value = enumValue.value();
        settings.set(key, ew);
    } else {
        throw InvalidSetup("Type mismatch when setting parameter '" + key + "'");
    }
}
 
} // namespace Chai
 
void ScriptNode::run(const RunSettings& global, IJobCallbacks& callbacks) {
    chaiscript::ChaiScript chai;
    Chai::registerBindings(chai);
 
    chai.add(chaiscript::user_type<Chai::Node>(), "Node");
    chai.add(chaiscript::fun(&Chai::Node::setParam<Float>), "setParam");
    chai.add(chaiscript::fun(&Chai::Node::setParam<int>), "setParam");
    chai.add(chaiscript::fun(&Chai::Node::setParam<std::string>), "setParam");
    chai.add(chaiscript::fun(&Chai::Node::run), "run");
 
    chai.add(chaiscript::fun<std::function<Chai::Node(std::string)>>([&](std::string name) {
        for (const auto& node : nodes) {
            if (node->instanceName() == name) {
                return Chai::Node(node, global, callbacks);
            }
        }
        throw InvalidSetup("Unknown node '" + name + "'");
    }),
        "getNode");
 
    chai.eval_file(file.native());
}
 
#endif
 
NAMESPACE_SPH_END