1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24 package org.dishevelled.venn.model;
25
26 import java.util.HashSet;
27 import java.util.Set;
28
29 import com.google.common.base.Joiner;
30
31 import org.dishevelled.observable.ObservableSet;
32
33 import org.dishevelled.observable.event.SetChangeEvent;
34 import org.dishevelled.observable.event.SetChangeListener;
35
36 import org.dishevelled.observable.impl.ObservableSetImpl;
37
38
39
40
41
42
43
44
45 final class SelectionView<E>
46 extends ObservableSetImpl<E>
47 {
48
49 private static final int DEFAULT_SIZE = 16;
50
51
52 private final Set<E> union;
53
54
55 private final SetChangeListener<E> purgeSelection = new SetChangeListener<E>()
56 {
57
58 public void setChanged(final SetChangeEvent<E> event)
59 {
60 purgeSelection();
61 }
62 };
63
64
65
66
67
68
69
70
71 SelectionView(final Set<E> union, final ObservableSet<E>... views)
72 {
73 super(new HashSet<E>(Math.max(DEFAULT_SIZE, union == null ? DEFAULT_SIZE : union.size())));
74 this.union = union;
75 for (ObservableSet<E> view : views)
76 {
77 view.addSetChangeListener(purgeSelection);
78 }
79 }
80
81
82
83
84
85 private void purgeSelection()
86 {
87 Set<E> toRemove = new HashSet<E>(Math.max(DEFAULT_SIZE, (int) (size() / 10)));
88 for (E e : this)
89 {
90 if (!union.contains(e))
91 {
92 toRemove.add(e);
93 }
94 }
95 removeAll(toRemove);
96 }
97
98
99 protected boolean preAdd(final E e)
100 {
101 if (!union.contains(e))
102 {
103 throw new IllegalArgumentException("can not select an element not in union");
104 }
105 return super.preAdd(e);
106 }
107
108
109 public String toString()
110 {
111 StringBuilder sb = new StringBuilder();
112 sb.append("[");
113 Joiner.on(", ").appendTo(sb, this);
114 sb.append("]");
115 return sb.toString();
116 }
117 }