nodeeditor/include/nodes/internal/FlowScene.hpp

#pragma once

#include <QtCore/QUuid>
#include <QtWidgets/QGraphicsScene>

#include <unordered_map>
#include <tuple>
#include <functional>

#include "QUuidStdHash.hpp"
#include "Export.hpp"
#include "ConnectionID.hpp"
#include "DataModelRegistry.hpp"
#include "TypeConverter.hpp"
#include "memory.hpp"

namespace QtNodes
{

class FlowSceneModel;
class ConnectionGraphicsObject;
class NodeGraphicsObject;

/// The FlowScene is responsible for rendering a FlowSceneModel
/// If you're looking for a basic dataflow model, see DataFlowScene
class NODE_EDITOR_PUBLIC FlowScene
  : public QGraphicsScene
{
  Q_OBJECT

  friend NodeGraphicsObject;
  friend ConnectionGraphicsObject;
public:

  FlowScene(FlowSceneModel* model, QObject* parent = Q_NULLPTR);

  ~FlowScene();

public:

  FlowSceneModel*
  model() const { return _model; }

  NodeGraphicsObject*
  nodeGraphicsObject(const NodeIndex& index);

  std::vector<NodeIndex>
  selectedNodes() const;

private slots:

  void
  nodeRemoved(const QUuid& id);

  void
  nodeAdded(const QUuid& newID);

  void
  nodePortUpdated(NodeIndex const& id);

  void
  nodeValidationUpdated(NodeIndex const& id);

  void
  connectionRemoved(NodeIndex const& leftNode, PortIndex leftPortID,
                    NodeIndex const& rightNode, PortIndex rightPortID);

  void
  connectionAdded(NodeIndex const& leftNode, PortIndex leftPortID,
                  NodeIndex const& rightNode, PortIndex rightPortID);

  void
  nodeMoved(NodeIndex const& index);

private:

  FlowSceneModel* _model;

  std::unordered_map<QUuid, NodeGraphicsObject*> _nodeGraphicsObjects;

  std::unordered_map<ConnectionID, ConnectionGraphicsObject*> _connGraphicsObjects;

  // This is for when you're creating a connection
  ConnectionGraphicsObject* _temporaryConn = nullptr;

};

NodeGraphicsObject*
locateNodeAt(QPointF scenePoint, FlowScene &scene,
             QTransform const & viewTransform);
}