1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103
|
#pragma once
#include <QtCore/QObject>
#include <memory>
#include <nodes/NodeData>
#include <nodes/NodeDataModel>
using QtNodes::NodeData;
using QtNodes::NodeDataModel;
using QtNodes::NodeDataType;
using QtNodes::PortIndex;
using QtNodes::PortType;
/// The class can potentially incapsulate any user data which
/// need to be transferred within the Node Editor graph
class MyNodeData : public NodeData
{
public:
std::shared_ptr<NodeDataType> type() const override
{
return std::make_shared<NodeDataType>("MyNodeData", "My Node Data");
}
};
class SimpleNodeData : public NodeData
{
public:
std::shared_ptr<NodeDataType> type() const override
{
return std::make_shared<NodeDataType>("SimpleData", "Simple Data");
}
};
//------------------------------------------------------------------------------
/// The model dictates the number of inputs and outputs for the Node.
/// In this example it has no logic.
class NaiveDataModel : public NodeDataModel
{
Q_OBJECT
public:
virtual ~NaiveDataModel()
{
}
public:
QString caption() const override
{
return QString("Naive Data Model");
}
QString name() const override
{
return QString("NaiveDataModel");
}
public:
unsigned int nPorts(PortType portType) const override
{
unsigned int result = 1;
switch (portType)
{
case PortType::In: result = 2; break;
case PortType::Out: result = 2; break;
case PortType::None: break;
}
return result;
}
std::shared_ptr<NodeDataType> dataType(PortType portType, PortIndex portIndex) const override
{
switch (portType)
{
case PortType::In:
switch (portIndex)
{
case 0: return MyNodeData().type();
case 1: return SimpleNodeData().type();
}
break;
case PortType::Out:
switch (portIndex)
{
case 0: return MyNodeData().type();
case 1: return SimpleNodeData().type();
}
break;
case PortType::None: break;
}
// FIXME: control may reach end of non-void function [-Wreturn-type]
return std::make_shared<NodeDataType>();
}
std::shared_ptr<NodeData> outData(PortIndex port) override
{
if (port < 1)
return std::make_shared<MyNodeData>();
return std::make_shared<SimpleNodeData>();
}
void setInData(std::shared_ptr<NodeData>, int) override
{
//
}
std::unique_ptr<NodeDataModel> clone() const override
{
return std::make_unique<NaiveDataModel>();
}
QWidget *embeddedWidget() override
{
return nullptr;
}
};
|