summaryrefslogtreecommitdiffstats
path: root/src/jrummikub/model
diff options
context:
space:
mode:
Diffstat (limited to 'src/jrummikub/model')
-rw-r--r--src/jrummikub/model/IStoneTray.java2
-rw-r--r--src/jrummikub/model/StoneTray.java11
2 files changed, 12 insertions, 1 deletions
diff --git a/src/jrummikub/model/IStoneTray.java b/src/jrummikub/model/IStoneTray.java
index d411c9e..437ecda 100644
--- a/src/jrummikub/model/IStoneTray.java
+++ b/src/jrummikub/model/IStoneTray.java
@@ -55,4 +55,6 @@ public interface IStoneTray<E extends Sizeable> extends
*/
public int getSize();
+ public boolean contains(E object);
+
} \ No newline at end of file
diff --git a/src/jrummikub/model/StoneTray.java b/src/jrummikub/model/StoneTray.java
index aaedaad..9148934 100644
--- a/src/jrummikub/model/StoneTray.java
+++ b/src/jrummikub/model/StoneTray.java
@@ -189,7 +189,16 @@ public class StoneTray<E extends Sizeable> implements IStoneTray<E> {
*/
@Override
public Position getPosition(E object) {
- return objects.get(object).getSecond();
+ Pair<E, Position> entry = objects.get(object);
+ if (entry == null) {
+ return null;
+ }
+ return entry.getSecond();
+ }
+
+ @Override
+ public boolean contains(E object) {
+ return objects.containsKey(objects);
}
@Override