Merge pull request #17693 from vespa-engine/arnej/tensor-has-cell-with-address

add api for detecting cell existence

1 files changed, 11 insertions, 0 deletions

diff --git a/vespajlib/src/main/java/com/yahoo/tensor/MixedTensor.java b/vespajlib/src/main/java/com/yahoo/tensor/MixedTensor.java
index 7631a2e4eab..e686a42d530 100644
--- a/vespajlib/src/main/java/com/yahoo/tensor/MixedTensor.java
+++ b/vespajlib/src/main/java/com/yahoo/tensor/MixedTensor.java
@@ -61,6 +61,17 @@ public class MixedTensor implements Tensor {
         return cell.getValue();
     }
 
+    @Override
+    public boolean has(TensorAddress address) {
+        long cellIndex = index.indexOf(address);
+        if (cellIndex < 0 || cellIndex >= cells.size())
+            return false;
+        Cell cell = cells.get((int)cellIndex);
+        if ( ! address.equals(cell.getKey()))
+            return false;
+        return true;
+    }
+
     /**
      * Returns an iterator over the cells of this tensor.
      * Cells are returned in order of increasing indexes in the