1 files changed, 9 insertions, 0 deletions
diff --git a/tikzit/src/gtk/SelectTool.h b/tikzit/src/gtk/SelectTool.h
index 172a7ae..65f511a 100644
--- a/tikzit/src/gtk/SelectTool.h
+++ b/tikzit/src/gtk/SelectTool.h
@@ -31,10 +31,17 @@ typedef enum {
     DragEdgeControlPoint2
 } SelectToolState;
 
+typedef enum {
+    DragSelectsNodes = 1,
+    DragSelectsEdges = 2,
+    DragSelectsBoth = DragSelectsNodes | DragSelectsEdges
+} DragSelectMode;
+
 @interface SelectTool : NSObject <Tool> {
     GraphRenderer      *renderer;
     SelectToolState     state;
     float               edgeFuzz;
+    DragSelectMode      dragSelectMode;
     NSPoint             dragOrigin;
     Node               *leaderNode;
     NSPoint             oldLeaderPos;
@@ -43,9 +50,11 @@ typedef enum {
     NSMutableSet       *selectionBoxContents;
 
     GtkWidget          *configWidget;
+    GSList             *dragSelectModeButtons;
 }
 
 @property (assign) float edgeFuzz;
+@property (assign) DragSelectMode dragSelectMode;
 
 - (id) init;
 + (id) tool;