Key validation

3 files changed, 11 insertions, 0 deletions

diff --git a/model-data/src/main/java/org/eclipse/viatra/solver/data/model/representation/AuxilaryDataRepresentation.java b/model-data/src/main/java/org/eclipse/viatra/solver/data/model/representation/AuxilaryDataRepresentation.java
index c524ba37..15116afd 100644
--- a/model-data/src/main/java/org/eclipse/viatra/solver/data/model/representation/AuxilaryDataRepresentation.java
+++ b/model-data/src/main/java/org/eclipse/viatra/solver/data/model/representation/AuxilaryDataRepresentation.java
@@ -8,4 +8,9 @@ public class AuxilaryDataRepresentation<K,V> extends DataRepresentation<K, V> {
                         V defaultValue) {
                 super(keyClass, hashProvider, valueClass, defaultValue);
         }
+        
+        @Override
+        public boolean isValidKey(K key) {
+                return true;
+        }
 }
diff --git a/model-data/src/main/java/org/eclipse/viatra/solver/data/model/representation/DataRepresentation.java b/model-data/src/main/java/org/eclipse/viatra/solver/data/model/representation/DataRepresentation.java
index 41205c83..e04606f5 100644
--- a/model-data/src/main/java/org/eclipse/viatra/solver/data/model/representation/DataRepresentation.java
+++ b/model-data/src/main/java/org/eclipse/viatra/solver/data/model/representation/DataRepresentation.java
@@ -27,6 +27,7 @@ abstract public class DataRepresentation<K, V> {
         public ContinousHashProvider<K> getHashProvider() {
                 return hashProvider;
         }
+        public abstract boolean isValidKey(K key);
 
         public V getDefaultValue() {
                 return defaultValue;
diff --git a/model-data/src/main/java/org/eclipse/viatra/solver/data/model/representation/SymbolRepresentation.java b/model-data/src/main/java/org/eclipse/viatra/solver/data/model/representation/SymbolRepresentation.java
index 442f9dc6..c8b7473f 100644
--- a/model-data/src/main/java/org/eclipse/viatra/solver/data/model/representation/SymbolRepresentation.java
+++ b/model-data/src/main/java/org/eclipse/viatra/solver/data/model/representation/SymbolRepresentation.java
@@ -15,4 +15,9 @@ public class SymbolRepresentation <D> extends DataRepresentation<Tuple,D> {
         public Symbol getSymbol() {
                 return symbol;
         }
+        
+        @Override
+        public boolean isValidKey(Tuple key) {
+                return key.getSize() == symbol.getArity();
+        }
 }