rainynite-studio/studio/generic/process_node.h

/*  process_node.h - node processor
 *  Copyright (C) 2017-2018 caryoscelus
 *
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef STUDIO_ACTIONS_PROCESS_NODE_H_D9E2969B_5C37_5085_828C_4708ED7EC469
#define STUDIO_ACTIONS_PROCESS_NODE_H_D9E2969B_5C37_5085_828C_4708ED7EC469

#include <core/util/class_init.h>
#include <core/abstract_factory.h>

#include <generic/context_listener.h>

namespace rainynite::studio {

template <typename T, typename Tag>
class ProcessNode : public ContextListener {
public:
    virtual bool accept(core::NodeTreeIndex index) const = 0;
    virtual void feed(T const& argument) = 0;
    virtual void start_list() {}
    virtual void end_list() {}

    bool set_node(core::NodeTreeIndex index_) {
        if (accept(index_)) {
            index = index_;
            return true;
        }
        return false;
    }

    core::NodeTreeIndex get_node_index() const {
        return index;
    }

    shared_ptr<core::AbstractValue> get_node() const {
        return get_context()->tree()->get_node(index);
    }

private:
    core::NodeTreeIndex index;
};

template <class P, typename Base>
struct ProcessNodeFactoryImpl :
    public AbstractFactoryImpl<Base, P>,
    private class_init::ListRegistered<
        ProcessNodeFactoryImpl<P, Base>,
        AbstractFactory<Base>
    >
{};

#define REGISTER_PROCESS_NODE(Processor, Base) \
template struct ProcessNodeFactoryImpl<Processor, Base>;

template <typename Processor>
class FindTargetNode {
public:
    unique_ptr<Processor> find_appropriate_target(core::NodeTree const& tree, shared_ptr<EditorContext> ctx) {
        if (auto node = ctx->get_active_node_index()) {
            if (auto r = check_target(node, ctx))
                return r;
            for (size_t i = 0; i < tree.children_count(node); ++i) {
                if (auto r = check_target(tree.index(node, i), ctx))
                    return r;
            }
        }
        auto doc_root = tree.get_root_index();
        for (size_t i = 0; i < tree.children_count(doc_root); ++i) {
            if (auto r = check_target(tree.index(doc_root, i), ctx))
                return r;
        }

        core::NodeTreeIndex root;
        try {
            root = index_of_property(tree, doc_root, "root");
        } catch (...) {
            return nullptr;
        }
        for (size_t i = 0; i < tree.children_count(root); ++i) {
            if (auto r = check_target(tree.index(root, i), ctx))
                return r;
        }
        return {};
    }

    unique_ptr<Processor> check_target(core::NodeTreeIndex node, shared_ptr<EditorContext> ctx) const {
        using namespace class_init;
        for (auto factory : class_list_registry<AbstractFactory<Processor>>()) {
            auto r = (*factory)();
            r->set_context(ctx);
            if (r->set_node(node))
                return r;
        }
        return nullptr;
    }
};

} // namespace rainynite::studio

#endif