rainynite-studio/studio/models/node_model.h

/*
 *  node_model.h - node tree model wrapper
 *  Copyright (C) 2017 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__NODE_MODEL_H_E924C400
#define __STUDIO__NODE_MODEL_H_E924C400

#include <QAbstractItemModel>

#include <core/node_info.h>

namespace studio {

class NodeModel : public QAbstractItemModel
{
    Q_OBJECT

public:
    explicit NodeModel(core::AbstractReference root, QObject* parent = 0);
    virtual ~NodeModel();

    QVariant data(QModelIndex const& index, int role) const override;
    Qt::ItemFlags flags(QModelIndex const& index) const override;
    QVariant headerData(int section, Qt::Orientation orientation,
                        int role = Qt::DisplayRole) const override;
    QModelIndex index(int row, int column,
                      QModelIndex const& parent = QModelIndex()) const override;
    QModelIndex parent(QModelIndex const& index) const override;
    int rowCount(QModelIndex const& parent = QModelIndex()) const override;
    int columnCount(QModelIndex const& parent = QModelIndex()) const override;

public:
    virtual bool removeRows(int row, int count, QModelIndex const& parent = QModelIndex()) override;

public:
    bool can_add_element(QModelIndex const& parent) const;
    void add_empty_element(QModelIndex const& parent);
    void convert_node(QModelIndex const& index, core::NodeInfo const* node_info, core::Time time);
    bool node_is_connected(QModelIndex const& index) const;
    void disconnect_node(QModelIndex const& index);
    void connect_nodes(QList<QModelIndex> const& selection, QModelIndex const& source);
    void replace_node(QModelIndex const& index, core::AbstractReference node);

public:
    template <typename T, typename F>
    T find_nodes(core::AbstractReference node_to_find, F f) const {
        // TODO: perfect forward f
        return core::traverse_once<T>(
            root,
            [this, &f, &node_to_find](auto const& node) -> boost::optional<T> {
                if (node == node_to_find)
                    return f();
                return boost::none;
            },
            core::TraverseDepth::Deeper
        );
    }

public:
    std::shared_ptr<core::AbstractValue> get_node(QModelIndex const& index) const;
    template <class T>
    std::shared_ptr<T> get_node_as(QModelIndex const& index) const {
        return std::dynamic_pointer_cast<T>(get_node(index));
    }
    inline std::shared_ptr<core::AbstractListLinked> get_list_node(QModelIndex const& index) const {
        return get_node_as<core::AbstractListLinked>(index);
    }

    size_t get_node_index(QModelIndex const& index) const;

private:
    quintptr get_id(QModelIndex const& parent, size_t i) const;

private:
    core::AbstractReference root;
private:
    mutable std::map<std::pair<QModelIndex, size_t>, quintptr> indexes;
    mutable std::map<quintptr, QModelIndex> parents;
    mutable quintptr index_count = 0;
};

} // namespace studio

#endif