system_menu.h

#pragma once
 
#include <QList>
#include <QTreeWidget>
 
#include "result.h"
#include "test_and_set.h"
#include "types.h"
 
class NodeItem;
class Flow;
class QTreeWidgetItem;
class FlowSaveInfo;

class SystemMenu : public QTreeWidget
{
  Q_OBJECT
public:
  SystemMenu(QWidget* parent);

  VoidResult onNodeAdded(const QString& flowId, NodeItem* node, const Types::LibraryTypes canvasType);

  VoidResult onNodeRemoved(const QString& flowId, const QString& nodeId, const QString& parentId);

  VoidResult onNodeModified(const QString& flowId, NodeItem* node);

  VoidResult onNodeSelected(const QString& flowId, NodeItem* node, bool selected);

  VoidResult onFlowAdded(Flow* flow, NodeItem* node);

  VoidResult onFlowRemoved(const QString& flowId, const QString& nodeId);
 
  VoidResult onPipelineAdded(std::shared_ptr<const FlowSaveInfo> info);
  VoidResult onPipelineRemoved(const QString& id);
 
signals:
  void nodeFocused(const QString& flowId, const QString& nodeId);

  void nodeRemoved(const QString& flowId, const QString& nodeId);

  void flowSelected(const QString& flowId, const QString& nodeId);

  void flowRemoved(const QString& flowId, const QString& nodeId);

  void flowRenamed(const QString& flowId, const QString& nodeId);
 
  void editPipeline(const QString& pipelineId);
  void removePipeline(const QString& pipelineId);
 
private slots:
  void showContextMenu(const QPoint& pos);

  void onItemClicked(QTreeWidgetItem* item, int /* column */);
 
protected:
  void resizeEvent(QResizeEvent* event) override;
 
private:
  enum Roles
  {
    ToplevelRole = 0,  
    Capabilities,
    SubTasks,
    Flows,
    PipelineRoot,
    Pipeline,
    ComponentRole,  
    FlowRole,       
    NodeRole,       
    PipelineRole    
  };
 
  qreal mNameColumnRatio = 0.7;
  TestAndSet<bool> mAdjustingColumns = false;
 
  void resizeColumns(int changedColumn = -1, int requestedWidth = -1);

  VoidResult addRootNode(NodeItem* node);

  VoidResult addLeafNode(NodeItem* node);

  void populateItem(QTreeWidgetItem* item, const QIcon& icon, const QString& name, const QString& type, const QString& data, const Roles role,
                    const QString& canvas = "");
 
  void populateTaskItem(QTreeWidgetItem* item, NodeItem* node);

  QTreeWidgetItem* getItemById(const QString& id) const;
 
  QTreeWidgetItem* findParentItemByRole(const QString& id, Roles role) const;
  QTreeWidgetItem* getOrCreateChildGroup(const QString& parentId, Roles role);
  QTreeWidgetItem* getOrCreateGroup(const QString& id, Roles role, QTreeWidgetItem* parent);
 
  void removeGroupIfEmpty(QTreeWidgetItem* group);

  QTreeWidgetItem* systemFlows();

  QTreeWidgetItem* componentFlows();

  void editFlow(QTreeWidgetItem* item);

  void removeFlow(QTreeWidgetItem* item);
};