Support arithmetic operations for dense_vectors: scalar version

Adds support for arithmetic operations - '+', '-', '*' and '/' when one operands is a dense_vector and other is scalar. Performs the arithmetic operation
on each element of the dense_vector against the scalar operand.

Closes #140538

Author: kkharbas

x-pack/plugin/esql-core/src/main/java/org/elasticsearch/xpack/esql/core/util/NumericUtils.java

@@ -166,7 +166,7 @@ public static double asFiniteNumber(double dbl) {
      */
     public static float asFiniteNumber(float flt) {
         if (Double.isNaN(flt) || Double.isInfinite(flt)) {
-            throw new ArithmeticException("not a finite double number: " + flt);
+            throw new ArithmeticException("not a finite float number: " + flt);
         }
         return flt;
     }

x-pack/plugin/esql/compute/ann/src/main/java/org/elasticsearch/compute/ann/DenseVectorEvaluator.java

@@ -0,0 +1,29 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License
+ * 2.0; you may not use this file except in compliance with the Elastic License
+ * 2.0.
+ */
+
+package org.elasticsearch.compute.ann;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+@Target(ElementType.METHOD)
+@Retention(RetentionPolicy.SOURCE)
+public @interface DenseVectorEvaluator {
+    /**
+     * Extra part of the name of the evaluator. Use for disambiguating
+     * when there are multiple ways to evaluate a function.
+     */
+    String extraName() default "";
+
+    /**
+     * Exceptions thrown by the process method to catch and convert
+     * into a warning and turn into a null value.
+     */
+    Class<? extends Exception>[] warnExceptions() default {};
+}

x-pack/plugin/esql/compute/gen/src/main/java/org/elasticsearch/compute/gen/DenseVectorEvaluatorImplementer.java

@@ -8,6 +8,7 @@
 package org.elasticsearch.compute.gen;
 
 import org.elasticsearch.compute.ann.ConvertEvaluator;
+import org.elasticsearch.compute.ann.DenseVectorEvaluator;
 import org.elasticsearch.compute.ann.Evaluator;
 import org.elasticsearch.compute.ann.MvEvaluator;
 
@@ -39,7 +40,8 @@ public Set<String> getSupportedOptions() {
 
     @Override
     public Set<String> getSupportedAnnotationTypes() {
-        return Set.of(Evaluator.class.getName(), MvEvaluator.class.getName(), ConvertEvaluator.class.getName());
+        return Set.of(Evaluator.class.getName(), MvEvaluator.class.getName(), ConvertEvaluator.class.getName(),
+            DenseVectorEvaluator.class.getName());
     }
 
     @Override
@@ -68,7 +70,7 @@ public boolean process(Set<? extends TypeElement> set, RoundEnvironment roundEnv
             for (Element evaluatorMethod : roundEnvironment.getElementsAnnotatedWith(ann)) {
                 var warnExceptionsTypes = Annotations.listAttributeValues(
                     evaluatorMethod,
-                    Set.of(Evaluator.class, MvEvaluator.class, ConvertEvaluator.class),
+                    Set.of(Evaluator.class, MvEvaluator.class, ConvertEvaluator.class, DenseVectorEvaluator.class),
                     "warnExceptions"
                 );
                 Evaluator evaluatorAnn = evaluatorMethod.getAnnotation(Evaluator.class);
@@ -134,6 +136,26 @@ public boolean process(Set<? extends TypeElement> set, RoundEnvironment roundEnv
                         throw e;
                     }
                 }
+                DenseVectorEvaluator denseVectorEvaluatorAnn = evaluatorMethod.getAnnotation(DenseVectorEvaluator.class);
+                if (denseVectorEvaluatorAnn != null) {
+                    try {
+                        AggregatorProcessor.write(
+                            evaluatorMethod,
+                            "evaluator",
+                            new DenseVectorEvaluatorImplementer(
+                                env.getElementUtils(),
+                                env.getTypeUtils(),
+                                (ExecutableElement) evaluatorMethod,
+                                denseVectorEvaluatorAnn.extraName(),
+                                warnExceptionsTypes
+                            ).sourceFile(),
+                            env
+                        );
+                    } catch (Exception e) {
+                        env.getMessager().printMessage(Diagnostic.Kind.ERROR, "failed to build " + evaluatorMethod.getEnclosingElement());
+                        throw e;
+                    }
+                }
             }
         }
         return true;

x-pack/plugin/esql/compute/gen/src/main/java/org/elasticsearch/compute/gen/EvaluatorProcessor.java

@@ -1,5 +1,5 @@
-id:l, vector_field_1:dense_vector, vector_field_2:dense_vector
-0,    [1.0, 2.0, 3.0],             [0.1, 0.2, 0.3]
-1,    [4.0, 5.0, 6.0],             [0.4, 0.5, 0.6]
-2,    [9.0, 8.0, 7.0],             [0.9, 0.8, 0.7]
-3,    [0.054, 0.032, 0.012],       [1.0, 2.0, 3.0]
+id:l, vector_field_1:dense_vector, vector_field_2:dense_vector, int_field:integer
+0,    [1.0, 2.0, 3.0],             [0.1, 0.2, 0.3],             4
+1,    [4.0, 5.0, 6.0],             [0.4, 0.5, 0.6],             5
+2,    [9.0, 8.0, 7.0],             [0.9, 0.8, 0.7],             6
+3,    [0.054, 0.032, 0.012],       [1.0, 2.0, 3.0],             7

x-pack/plugin/esql/qa/testFixtures/src/main/resources/data/dense_vector_arithmetic.csv

@@ -86,6 +86,20 @@ null
 null
 ;
 
+addDenseVectorInt
+required_capability: dense_vector_arithmetic
+FROM dense_vector_arithmetic
+| eval result_vector = vector_field_1 + int_field
+| SORT id
+| KEEP result_vector;
+
+result_vector:dense_vector
+[5.0, 6.0, 7.0]
+[9.0, 10.0, 11.0]
+[15.0, 14.0, 13.0]
+[7.054, 7.032, 7.012]
+;
+
 // tests for sub operation
 
 subDenseVectors
@@ -174,6 +188,20 @@ null
 null
 ;
 
+subDenseVectorInt
+required_capability: dense_vector_arithmetic
+FROM dense_vector_arithmetic
+| eval result_vector = vector_field_1 - int_field
+| SORT id
+| KEEP result_vector;
+
+result_vector:dense_vector
+[-3.0, -2.0, -1.0]
+[-1.0, 0.0, 1.0]
+[3.0, 2.0, 1.0]
+[-6.946, -6.968, -6.988]
+;
+
 // tests for mul operation
 
 mulDenseVectors
@@ -262,6 +290,20 @@ null
 null
 ;
 
+mulDenseVectorInt
+required_capability: dense_vector_arithmetic
+FROM dense_vector_arithmetic
+| eval result_vector = vector_field_1 * int_field
+| SORT id
+| KEEP result_vector;
+
+result_vector:dense_vector
+[4.0, 8.0, 12.0]
+[20.0, 25.0, 30.0]
+[54.0, 48.0, 42.0]
+[0.37800002, 0.224, 0.084]
+;
+
 // tests for div operation
 
 divDenseVectors
@@ -349,3 +391,17 @@ null
 null
 null
 ;
+
+divDenseVectorInt
+required_capability: dense_vector_arithmetic
+FROM dense_vector_arithmetic
+| eval result_vector = vector_field_1 / int_field
+| SORT id
+| KEEP result_vector;
+
+result_vector:dense_vector
+[0.25, 0.5, 0.75]
+[0.8, 1.0, 1.2]
+[1.5, 1.3333334, 1.1666666]
+[0.007714286, 0.004571429, 0.0017142857]
+;

x-pack/plugin/esql/qa/testFixtures/src/main/resources/dense_vector-arithmetic.csv-spec

@@ -20,6 +20,9 @@
         "m": 16,
         "ef_construction": 100
       }
+    },
+    "int_field": {
+      "type": "integer"
     }
   }
 }