1 files changed, 4 insertions, 1 deletions
diff --git a/src/gui/undocommands.h b/src/gui/undocommands.h
index 42fed30..94b98be 100644
--- a/src/gui/undocommands.h
+++ b/src/gui/undocommands.h
@@ -121,11 +121,14 @@ private:
 class AddEdgeCommand : public GraphUpdateCommand
 {
 public:
-    explicit AddEdgeCommand(TikzScene *scene, Edge *edge, QUndoCommand *parent = nullptr);
+    explicit AddEdgeCommand(TikzScene *scene, Edge *edge, bool selectEdge, QSet<Node *> selNodes, QSet<Edge *> selEdges, QUndoCommand *parent = nullptr);
     void undo() override;
     void redo() override;
 private:
+    bool _selectEdge;
     Edge *_edge;
+    QSet<Node*> _selNodes;
+    QSet<Edge*> _selEdges;
 };
 
 class ChangeEdgeModeCommand : public GraphUpdateCommand