1 files changed, 171 insertions, 0 deletions
diff --git a/src/common/PickSupport.m b/src/common/PickSupport.m
new file mode 100644
index 0000000..5ea13c0
--- /dev/null
+++ b/src/common/PickSupport.m
@@ -0,0 +1,171 @@
+//
+//  PickSupport.m
+//  TikZiT
+//  
+//  Copyright 2010 Aleks Kissinger. All rights reserved.
+//  
+//  
+//  This file is part of TikZiT.
+//  
+//  TikZiT is free software: you can redistribute it and/or modify
+//  it under the terms of the GNU General Public License as published by
+//  the Free Software Foundation, either version 3 of the License, or
+//  (at your option) any later version.
+//  
+//  TikZiT is distributed in the hope that it will be useful,
+//  but WITHOUT ANY WARRANTY; without even the implied warranty of
+//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+//  GNU General Public License for more details.
+//  
+//  You should have received a copy of the GNU General Public License
+//  along with TikZiT.  If not, see <http://www.gnu.org/licenses/>.
+//  
+
+#import "PickSupport.h"
+
+
+@implementation PickSupport
+
+- (void) postNodeSelectionChanged {
+	[[NSNotificationCenter defaultCenter]
+		postNotificationName:@"NodeSelectionChanged"
+		object:self];
+}
+
+- (void) postEdgeSelectionChanged {
+	[[NSNotificationCenter defaultCenter]
+		postNotificationName:@"EdgeSelectionChanged"
+		object:self];
+}
+
+- (id) init {
+	self = [super init];
+
+	if (self) {
+		selectedNodes = [[NSMutableSet set] retain];
+		selectedEdges = [[NSMutableSet set] retain];
+	}
+
+	return self;
+}
+
++ (PickSupport*)pickSupport {
+	return [[[PickSupport alloc] init] autorelease];
+}
+
+- (NSSet*)selectedNodes { return selectedNodes; }
+- (NSSet*)selectedEdges { return selectedEdges; }
+
+- (BOOL)isNodeSelected:(Node*)nd {
+	return [selectedNodes containsObject:nd];
+}
+
+- (BOOL)isEdgeSelected:(Edge*)e {
+	return [selectedEdges containsObject:e];
+}
+
+- (void)selectNode:(Node*)nd {
+	if (nd != nil && ![selectedNodes member:nd]) {
+		[selectedNodes addObject:nd];
+		[[NSNotificationCenter defaultCenter]
+			postNotificationName:@"NodeSelected"
+			object:self
+			userInfo:[NSDictionary dictionaryWithObject:nd forKey:@"node"]];
+		[self postNodeSelectionChanged];
+	}
+}
+
+- (void)deselectNode:(Node*)nd {
+	if (nd != nil && [selectedNodes member:nd]) {
+		[selectedNodes removeObject:nd];
+		[[NSNotificationCenter defaultCenter]
+			postNotificationName:@"NodeDeselected"
+			object:self
+			userInfo:[NSDictionary dictionaryWithObject:nd forKey:@"node"]];
+		[self postNodeSelectionChanged];
+	}
+}
+
+- (void)selectEdge:(Edge*)e {
+	if (e != nil && ![selectedEdges member:e]) {
+		[selectedEdges addObject:e];
+		[[NSNotificationCenter defaultCenter]
+			postNotificationName:@"EdgeSelected"
+			object:self
+			userInfo:[NSDictionary dictionaryWithObject:e forKey:@"edge"]];
+		[self postEdgeSelectionChanged];
+	}
+}
+
+- (void)deselectEdge:(Edge*)e {
+	if (e != nil && [selectedEdges member:e]) {
+		[selectedEdges removeObject:e];
+		[[NSNotificationCenter defaultCenter]
+			postNotificationName:@"EdgeDeselected"
+			object:self
+			userInfo:[NSDictionary dictionaryWithObject:e forKey:@"edge"]];
+		[self postEdgeSelectionChanged];
+	}
+}
+
+- (void)toggleNodeSelected:(Node*)nd {
+	if ([self isNodeSelected:nd])
+		[self deselectNode:nd];
+	else
+		[self selectNode:nd];
+}
+
+- (void)selectAllNodes:(NSSet*)nodes {
+	[self selectAllNodes:nodes replacingSelection:YES];
+}
+
+- (void)selectAllNodes:(NSSet*)nodes replacingSelection:(BOOL)replace {
+	if (selectedNodes == nodes) {
+		return;
+	}
+	if (!replace && [nodes count] == 0) {
+		return;
+	}
+
+	if (replace) {
+		[selectedNodes release];
+		selectedNodes = [nodes mutableCopy];
+	} else {
+		[selectedNodes unionSet:nodes];
+	}
+	[[NSNotificationCenter defaultCenter]
+		postNotificationName:@"NodeSelectionReplaced"
+		object:self];
+	[self postNodeSelectionChanged];
+}
+
+- (void)deselectAllNodes {
+	if ([selectedNodes count] > 0) {
+		[selectedNodes removeAllObjects];
+		[[NSNotificationCenter defaultCenter]
+			postNotificationName:@"NodeSelectionReplaced"
+			object:self];
+		[self postNodeSelectionChanged];
+	}
+}
+
+- (void)deselectAllEdges {
+	if ([selectedEdges count] > 0) {
+		[selectedEdges removeAllObjects];
+		[[NSNotificationCenter defaultCenter]
+			postNotificationName:@"EdgeSelectionReplaced"
+			object:self];
+		[self postEdgeSelectionChanged];
+	}
+}
+
+- (void)dealloc {
+	[selectedNodes release];
+	[selectedEdges release];
+	
+	[super dealloc];
+}
+
+@end
+
+// vi:ft=objc:ts=4:noet:sts=4:sw=4