#include "DataFlowGraphicsScene.hpp"

#include "ConnectionGraphicsObject.hpp"

#include "GraphicsView.hpp"

#include "NodeDelegateModelRegistry.hpp"

#include "NodeGraphicsObject.hpp"

#include "UndoCommands.hpp"

#include <QtWidgets/QFileDialog>

#include <QtWidgets/QGraphicsSceneMoveEvent>

#include <QtWidgets/QHeaderView>

#include <QtWidgets/QLineEdit>

#include <QtWidgets/QTreeWidget>

#include <QtWidgets/QWidgetAction>

#include <QtCore/QBuffer>

#include <QtCore/QByteArray>

#include <QtCore/QDataStream>

#include <QtCore/QDebug>

#include <QtCore/QFile>

#include <QtCore/QJsonArray>

#include <QtCore/QJsonDocument>

#include <QtCore/QJsonObject>

#include <QtCore/QtGlobal>

#include <stdexcept>

#include <utility>

namespace QtNodes {

DataFlowGraphicsScene::DataFlowGraphicsScene(DataFlowGraphModel &graphModel, QObject *parent)

}

std::vector<NodeId> DataFlowGraphicsScene::selectedNodes() const

}

QMenu *DataFlowGraphicsScene::createSceneMenu(QPointF const scenePos)

{

if (_graphModel.is(DataFlowGraphModel::CtlOption::NoSceneMenu)) {

return nullptr;

}

QMenu *modelMenu = new QMenu();

// Add filterbox to the context menu

auto *txtBox = new QLineEdit(modelMenu);

txtBox->setPlaceholderText(QStringLiteral("Filter"));

txtBox->setClearButtonEnabled(true);

auto *txtBoxAction = new QWidgetAction(modelMenu);

txtBoxAction->setDefaultWidget(txtBox);

// 1.

modelMenu->addAction(txtBoxAction);

// Add result treeview to the context menu

QTreeWidget *treeView = new QTreeWidget(modelMenu);

treeView->header()->close();

auto *treeViewAction = new QWidgetAction(modelMenu);

treeViewAction->setDefaultWidget(treeView);

// 2.

modelMenu->addAction(treeViewAction);

auto registry = _graphModel.dataModelRegistry();

for (auto const &cat : registry->categories()) {

auto item = new QTreeWidgetItem(treeView);

item->setText(0, cat);

item->setFlags(item->flags() & ~Qt::ItemIsSelectable);

}

for (auto const &assoc : registry->registeredModelsCategoryAssociation()) {

QList<QTreeWidgetItem *> parent = treeView->findItems(assoc.second, Qt::MatchExactly);

if (parent.count() <= 0)

continue;

auto item = new QTreeWidgetItem(parent.first());

item->setText(0, assoc.first);

}

treeView->expandAll();

connect(treeView,

&QTreeWidget::itemClicked,

[this, modelMenu, scenePos](QTreeWidgetItem *item, int) {

if (!(item->flags() & (Qt::ItemIsSelectable))) {

return;

}

this->undoStack().push(new CreateCommand(this, item->text(0), scenePos));

modelMenu->close();

});

//Setup filtering

connect(txtBox, &QLineEdit::textChanged, [treeView](const QString &text) {

QTreeWidgetItemIterator categoryIt(treeView, QTreeWidgetItemIterator::HasChildren);

while (*categoryIt)

(*categoryIt++)->setHidden(true);

QTreeWidgetItemIterator it(treeView, QTreeWidgetItemIterator::NoChildren);

while (*it) {

auto modelName = (*it)->text(0);

const bool match = (modelName.contains(text, Qt::CaseInsensitive));

(*it)->setHidden(!match);

if (match) {

QTreeWidgetItem *parent = (*it)->parent();

while (parent) {

parent->setHidden(false);

parent = parent->parent();

}

}

++it;

}

});

// make sure the text box gets focus so the user doesn't have to click on it

txtBox->setFocus();

// QMenu's instance auto-destruction

modelMenu->setAttribute(Qt::WA_DeleteOnClose);

return modelMenu;

}

bool DataFlowGraphicsScene::save() const

}

bool DataFlowGraphicsScene::load()

}

} // namespace QtNodes