/*=========================================================================

  Program:   ParaView
  Plugin:    NodeEditor

  Copyright (c) Kitware, Inc.
  All rights reserved.
  See Copyright.txt or http://www.paraview.org/HTML/Copyright.html for details.

     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notice for more information.

=========================================================================*/
/*-------------------------------------------------------------------------
  ParaViewPluginsNodeEditor - BSD 3-Clause License - Copyright (C) 2021 Jonas Lukasczyk

  See the Copyright.txt file provided
  with ParaViewPluginsNodeEditor for license information.
-------------------------------------------------------------------------*/

#ifndef pqNodeEditorScene_h
#define pqNodeEditorScene_h

#include <QGraphicsScene>

#include "vtkType.h" // for vtkIdType

#include <unordered_map> // for std::unordered_map
#include <vector>        // for std::vector

class pqNodeEditorNode;
class pqNodeEditorEdge;

/**
 * This class extends QGraphicsScene to draw a grid background;
 * monitor the creation/modification/destruction of proxies to automatically
 * modify the scene accordingly; and manage the instances of nodes and edges.
 */
class pqNodeEditorScene : public QGraphicsScene
{
  Q_OBJECT

public:
  pqNodeEditorScene(QObject* parent = nullptr);
  ~pqNodeEditorScene() override = default;

Q_SIGNALS:
  /**
   * Fired when an en edge created from the drag and drop interaction of a port disc
   * is released. @c fromNode is the ID of the producer proxy and @c fromPort its port number.
   * @c toNode and @c toPort re the same informations but relative to the consumer.
   */
  void edgeDragAndDropRelease(vtkIdType fromNode, int fromPort, vtkIdType toNode, int toPort);

public Q_SLOTS:
  /**
   * Compute an optimized layout for the nodes in the scene.
   * Return 1 if success, 0 else.
   *
   * If GraphViz has not been at the compilation this function will do nothing.
   */
  int computeLayout(const std::unordered_map<vtkIdType, pqNodeEditorNode*>& nodes,
    std::unordered_map<vtkIdType, std::vector<pqNodeEditorEdge*>>& edges);

protected:
  /**
   *  Draws a grid background.
   */
  void drawBackground(QPainter* painter, const QRectF& rect) override;
};

#endif // pqNodeEditorScene_h
