Add "create non-existing cells" flag to TensorModifyUpdate.

When this is true, non-existing cells in the input tensor is created
before applying the modify update.
The default cell value is 0.0 for REPLACE and ADD operations,
and 1.0 for MULTIPLY operation.

1 files changed, 17 insertions, 2 deletions

diff --git a/document/src/main/java/com/yahoo/document/serialization/VespaDocumentDeserializerHead.java b/document/src/main/java/com/yahoo/document/serialization/VespaDocumentDeserializerHead.java
index c6fdc915401..765d999dbc9 100644
--- a/document/src/main/java/com/yahoo/document/serialization/VespaDocumentDeserializerHead.java
+++ b/document/src/main/java/com/yahoo/document/serialization/VespaDocumentDeserializerHead.java
@@ -26,20 +26,35 @@ public class VespaDocumentDeserializerHead extends VespaDocumentDeserializer6 {
     @Override
     protected ValueUpdate readTensorModifyUpdate(DataType type) {
         byte operationId = getByte(null);
-        TensorModifyUpdate.Operation operation = TensorModifyUpdate.Operation.getOperation(operationId);
+        var operation = decodeOperation(operationId);
         if (operation == null) {
             throw new DeserializationException("Unknown operation id " + operationId + " for tensor modify update");
         }
         if (!(type instanceof TensorDataType)) {
             throw new DeserializationException("Expected tensor data type, got " + type);
         }
+        var createNonExistingCells = decodeCreateNonExistingCells(operationId);
+        if (createNonExistingCells) {
+            // Read the default cell value (but it is not used by TensorModifyUpdate).
+            getDouble(null);
+        }
         TensorDataType tensorDataType = (TensorDataType)type;
         TensorType tensorType = tensorDataType.getTensorType();
         TensorType convertedType = TensorModifyUpdate.convertDimensionsToMapped(tensorType);
 
         TensorFieldValue tensor = new TensorFieldValue(convertedType);
         tensor.deserialize(this);
-        return new TensorModifyUpdate(operation, tensor);
+        return new TensorModifyUpdate(operation, tensor, createNonExistingCells);
+    }
+
+    private TensorModifyUpdate.Operation decodeOperation(byte operationId) {
+        byte OP_MASK = 0b01111111;
+        return TensorModifyUpdate.Operation.getOperation(operationId & OP_MASK);
+    }
+
+    private boolean decodeCreateNonExistingCells(byte operationId) {
+        byte CREATE_FLAG = -0b10000000;
+        return (operationId & CREATE_FLAG) != 0;
     }
 
     @Override