Fix virtual nodes

gui/GraphEditorWidget.cpp

@@ -10,6 +10,7 @@
 
 #include <QAction>
 #include <QClipboard>
+#include <QContextMenuEvent>
 #include <QDateTime>
 #include <QDir>
 #include <QGraphicsItem>
@@ -135,6 +136,9 @@ GraphEditorWidget::GraphEditorWidget(warppipe::Client *client,
       }})");
 
   m_view = new QtNodes::GraphicsView(m_scene);
+  m_view->setFocusPolicy(Qt::StrongFocus);
+  m_view->viewport()->setFocusPolicy(Qt::StrongFocus);
+  m_view->viewport()->installEventFilter(this);
 
   auto *layout = new QVBoxLayout(this);
   layout->setContentsMargins(0, 0, 0, 0);
@@ -292,10 +296,21 @@ void GraphEditorWidget::captureDebugScreenshot(const QString &event) {
   pixmap.save(m_debugScreenshotDir + "/" + filename);
 }
 
+bool GraphEditorWidget::eventFilter(QObject *obj, QEvent *event) {
+  if (obj == m_view->viewport() &&
+      event->type() == QEvent::ContextMenu) {
+    auto *cme = static_cast<QContextMenuEvent *>(event);
+    m_lastContextMenuScenePos = m_view->mapToScene(cme->pos());
+  }
+  return QWidget::eventFilter(obj, event);
+}
+
 void GraphEditorWidget::onContextMenuRequested(const QPoint &pos) {
   QPointF scenePos = m_view->mapToScene(pos);
+  m_lastContextMenuScenePos = scenePos;
 
   uint32_t hitPwNodeId = 0;
+  QtNodes::NodeId hitQtNodeId = 0;
   for (auto nodeId : m_model->allNodeIds()) {
     const WarpNodeData *data = m_model->warpNodeData(nodeId);
     if (!data) {
@@ -308,13 +323,14 @@ void GraphEditorWidget::onContextMenuRequested(const QPoint &pos) {
     QRectF nodeRect(nodePos, QSizeF(nodeSize));
     if (nodeRect.contains(scenePos)) {
       hitPwNodeId = data->info.id.value;
+      hitQtNodeId = nodeId;
       break;
     }
   }
 
   QPoint screenPos = m_view->mapToGlobal(pos);
   if (hitPwNodeId != 0) {
-    showNodeContextMenu(screenPos, hitPwNodeId);
+    showNodeContextMenu(screenPos, hitPwNodeId, hitQtNodeId);
   } else {
     showCanvasContextMenu(screenPos, scenePos);
   }
@@ -327,6 +343,14 @@ void GraphEditorWidget::showCanvasContextMenu(const QPoint &screenPos,
   QAction *createSource =
       menu.addAction(QStringLiteral("Create Virtual Source"));
   menu.addSeparator();
+  QAction *pasteAction = menu.addAction(QStringLiteral("Paste"));
+  pasteAction->setShortcut(QKeySequence::Paste);
+  pasteAction->setEnabled(!m_clipboardJson.isEmpty() ||
+                          (QGuiApplication::clipboard()->mimeData() &&
+                           QGuiApplication::clipboard()->mimeData()->hasFormat(
+                               QStringLiteral(
+                                   "application/warppipe-virtual-graph"))));
+  menu.addSeparator();
   QAction *autoArrange = menu.addAction(QStringLiteral("Auto-Arrange"));
 
   QAction *chosen = menu.exec(screenPos);
@@ -334,15 +358,17 @@ void GraphEditorWidget::showCanvasContextMenu(const QPoint &screenPos,
     createVirtualNode(true, scenePos);
   } else if (chosen == createSource) {
     createVirtualNode(false, scenePos);
+  } else if (chosen == pasteAction) {
+    pasteSelection(QPointF(0, 0));
   } else if (chosen == autoArrange) {
     m_model->autoArrange();
   }
 }
 
 void GraphEditorWidget::showNodeContextMenu(const QPoint &screenPos,
-                                            uint32_t pwNodeId) {
-  QtNodes::NodeId qtId = m_model->qtNodeIdForPw(pwNodeId);
-  const WarpNodeData *data = m_model->warpNodeData(qtId);
+                                            uint32_t pwNodeId,
+                                            QtNodes::NodeId qtNodeId) {
+  const WarpNodeData *data = m_model->warpNodeData(qtNodeId);
   if (!data) {
     return;
   }
@@ -351,17 +377,42 @@ void GraphEditorWidget::showNodeContextMenu(const QPoint &screenPos,
   bool isVirtual =
       type == WarpNodeType::kVirtualSink || type == WarpNodeType::kVirtualSource;
 
-  if (!isVirtual) {
-    return;
+  QMenu menu;
+
+  QAction *copyAction = nullptr;
+  QAction *duplicateAction = nullptr;
+  QAction *deleteAction = nullptr;
+
+  if (isVirtual) {
+    copyAction = menu.addAction(QStringLiteral("Copy"));
+    copyAction->setShortcut(QKeySequence::Copy);
+    duplicateAction = menu.addAction(QStringLiteral("Duplicate"));
+    duplicateAction->setShortcut(QKeySequence(Qt::CTRL | Qt::Key_D));
+    menu.addSeparator();
+    deleteAction = menu.addAction(QStringLiteral("Delete"));
+    deleteAction->setShortcut(QKeySequence::Delete);
   }
 
-  QMenu menu;
-  QAction *deleteAction = menu.addAction(QStringLiteral("Delete Node"));
+  QAction *pasteAction = menu.addAction(QStringLiteral("Paste"));
+  pasteAction->setShortcut(QKeySequence::Paste);
+  pasteAction->setEnabled(!m_clipboardJson.isEmpty() ||
+                          (QGuiApplication::clipboard()->mimeData() &&
+                           QGuiApplication::clipboard()->mimeData()->hasFormat(
+                               QStringLiteral(
+                                   "application/warppipe-virtual-graph"))));
 
   QAction *chosen = menu.exec(screenPos);
-  if (chosen == deleteAction && m_client) {
-    m_client->RemoveNode(warppipe::NodeId{pwNodeId});
-    m_model->refreshFromClient();
+  if (!chosen) {
+    return;
+  }
+  if (chosen == copyAction) {
+    copySelection();
+  } else if (chosen == duplicateAction) {
+    duplicateSelection();
+  } else if (chosen == deleteAction && m_client) {
+    deleteSelection();
+  } else if (chosen == pasteAction) {
+    pasteSelection(QPointF(0, 0));
   }
 }
 

gui/GraphEditorWidget.h

@@ -3,6 +3,7 @@
 #include <warppipe/warppipe.hpp>
 
 #include <QJsonObject>
+#include <QPointF>
 #include <QString>
 #include <QWidget>
 
@@ -10,6 +11,7 @@
 #include <vector>
 
 namespace QtNodes {
+using NodeId = unsigned int;
 class BasicGraphicsScene;
 class GraphicsView;
 }  // namespace QtNodes
@@ -33,6 +35,8 @@ public:
 
   void setDebugScreenshotDir(const QString &dir);
 
+  bool eventFilter(QObject *obj, QEvent *event) override;
+
 Q_SIGNALS:
   void graphReady();
 
@@ -43,7 +47,8 @@ private slots:
 
 private:
   void showCanvasContextMenu(const QPoint &screenPos, const QPointF &scenePos);
-  void showNodeContextMenu(const QPoint &screenPos, uint32_t pwNodeId);
+  void showNodeContextMenu(const QPoint &screenPos, uint32_t pwNodeId,
+                           QtNodes::NodeId qtNodeId);
   void createVirtualNode(bool isSink, const QPointF &scenePos);
   void captureDebugScreenshot(const QString &event);
 
@@ -72,4 +77,5 @@ private:
   bool m_graphReady = false;
   QJsonObject m_clipboardJson;
   std::vector<PendingPasteLink> m_pendingPasteLinks;
+  QPointF m_lastContextMenuScenePos;
 };

gui/WarpGraphModel.cpp

@@ -628,14 +628,13 @@ WarpNodeType
 WarpGraphModel::classifyNode(const warppipe::NodeInfo &info) {
   const std::string &mc = info.media_class;
 
-  bool isVirtual = (info.name.find("warppipe") != std::string::npos);
-
   if (mc == "Audio/Sink" || mc == "Audio/Duplex") {
-    return isVirtual ? WarpNodeType::kVirtualSink : WarpNodeType::kHardwareSink;
+    return info.is_virtual ? WarpNodeType::kVirtualSink
+                           : WarpNodeType::kHardwareSink;
   }
   if (mc == "Audio/Source") {
-    return isVirtual ? WarpNodeType::kVirtualSource
-                     : WarpNodeType::kHardwareSource;
+    return info.is_virtual ? WarpNodeType::kVirtualSource
+                           : WarpNodeType::kHardwareSource;
   }
   if (mc == "Stream/Output/Audio" || mc == "Stream/Input/Audio") {
     return WarpNodeType::kApplication;

gui/main.cpp

@@ -83,6 +83,10 @@ int main(int argc, char *argv[]) {
 
   warppipe::ConnectionOptions opts;
   opts.application_name = "warppipe-gui";
+  opts.config_path =
+      (QStandardPaths::writableLocation(QStandardPaths::AppConfigLocation) +
+       QStringLiteral("/config.json"))
+          .toStdString();
 
   auto result = warppipe::Client::Create(opts);
   if (!result.ok()) {