qtDiagramScene.h

//=========================================================================
//  Copyright (c) Kitware, Inc.
//  All rights reserved.
//  See LICENSE.txt 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.
//=========================================================================
#ifndef smtk_extension_qtDiagramScene_h
#define smtk_extension_qtDiagramScene_h
 
#include "smtk/extension/qt/Exports.h"
#include "smtk/extension/qt/qtBaseView.h"
 
#include "smtk/string/Token.h"
 
#include "smtk/common/TypeContainer.h"
 
#include "smtk/PublicPointerDefs.h"
 
#include <QGraphicsScene>
 
#include <unordered_set>
 
class QAbstractItemModel;
class QItemSelection;
class QTreeView;
 
namespace smtk
{
namespace extension
{
 
class qtBaseArc;
class qtDiagram;
class qtBaseNode;
class qtDiagramViewConfiguration;
 
class SMTKQTEXT_EXPORT qtDiagramScene : public QGraphicsScene
{
  Q_OBJECT
 
public:
  using Superclass = QGraphicsScene;
 
  qtDiagramScene(qtDiagram*);
  ~qtDiagramScene() override;
 
  void setConfiguration(qtDiagramViewConfiguration* config) { m_config = config; }
  qtDiagramViewConfiguration* configuration() const { return m_config; }
 
  qtDiagram* diagram() { return m_diagram; }
 
  std::unordered_set<qtBaseNode*> nodesOfSelection() const;
 
public Q_SLOTS:
  bool computeLayout(
    const std::unordered_set<qtBaseNode*>& nodes,
    const std::unordered_set<qtBaseArc*>& arcs);
 
  virtual void alignHorizontal(const std::unordered_set<qtBaseNode*>& nodeSet, int alignment);
  virtual void alignVertical(const std::unordered_set<qtBaseNode*>& nodeSet, int alignment);
 
  virtual void distributeHorizontal(
    const std::unordered_set<qtBaseNode*>& nodeSet,
    smtk::string::Token distribution);
  virtual void distributeVertical(
    const std::unordered_set<qtBaseNode*>& nodeSet,
    smtk::string::Token distribution);
 
protected:
  static QPointF snapToGrid(const qreal& x, const qreal& y, const qreal& resolution = 1.0);
 
  void drawBackground(QPainter* painter, const QRectF& rect) override;
 
  qtDiagramViewConfiguration* m_config{ nullptr };
  qtDiagram* m_diagram{ nullptr };
};
 
} // namespace extension
} // namespace smtk
#endif // smtk_extension_qtDiagramScene_h