diff --git a/.gitignore b/.gitignore index 03e2446ec4f13..acb5a9e3054e8 100644 --- a/.gitignore +++ b/.gitignore @@ -69,3 +69,5 @@ testfixtures_shared/ # Generated checkstyle_ide.xml x-pack/plugin/esql/gen/ +/x-pack/plugin/esql/compute/src/main/generated +/x-pack/plugin/esql/src/main/java/generated diff --git a/docs/changelog/107658.yaml b/docs/changelog/107658.yaml new file mode 100644 index 0000000000000..38c955d786a8b --- /dev/null +++ b/docs/changelog/107658.yaml @@ -0,0 +1,5 @@ +pr: 105660 +summary: "Ignore generated code in git" +area: Delivery/Build +type: enhancement +issues: [] diff --git a/x-pack/plugin/esql/build.gradle b/x-pack/plugin/esql/build.gradle index 86245e1c93e97..5eba175233521 100644 --- a/x-pack/plugin/esql/build.gradle +++ b/x-pack/plugin/esql/build.gradle @@ -48,7 +48,7 @@ dependencies { * them and *then* compile them. */ tasks.named("compileJava").configure { - options.compilerArgs.addAll(["-s", "${projectDir}/src/main/java/generated"]) + options.generatedSourceOutputDirectory = file("src/main/java/generated") } tasks.named("javadoc").configure { diff --git a/x-pack/plugin/esql/compute/build.gradle b/x-pack/plugin/esql/compute/build.gradle index d04daf6631447..9f8b71856b8e8 100644 --- a/x-pack/plugin/esql/compute/build.gradle +++ b/x-pack/plugin/esql/compute/build.gradle @@ -10,8 +10,9 @@ dependencies { testImplementation(project(xpackModule('ql'))) } + tasks.named("compileJava").configure { - options.compilerArgs.addAll(["-s", "${projectDir}/src/main/generated"]) + options.generatedSourceOutputDirectory = file("src/main/generated") } tasks.named('checkstyleMain').configure { diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctBooleanAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctBooleanAggregatorFunction.java deleted file mode 100644 index e9b4498d50265..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctBooleanAggregatorFunction.java +++ /dev/null @@ -1,127 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunction} implementation for {@link CountDistinctBooleanAggregator}. - * This class is generated. Do not edit it. - */ -public final class CountDistinctBooleanAggregatorFunction implements AggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("fbit", ElementType.BOOLEAN), - new IntermediateStateDesc("tbit", ElementType.BOOLEAN) ); - - private final DriverContext driverContext; - - private final CountDistinctBooleanAggregator.SingleState state; - - private final List channels; - - public CountDistinctBooleanAggregatorFunction(DriverContext driverContext, List channels, - CountDistinctBooleanAggregator.SingleState state) { - this.driverContext = driverContext; - this.channels = channels; - this.state = state; - } - - public static CountDistinctBooleanAggregatorFunction create(DriverContext driverContext, - List channels) { - return new CountDistinctBooleanAggregatorFunction(driverContext, channels, CountDistinctBooleanAggregator.initSingle()); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public void addRawInput(Page page) { - BooleanBlock block = page.getBlock(channels.get(0)); - BooleanVector vector = block.asVector(); - if (vector != null) { - addRawVector(vector); - } else { - addRawBlock(block); - } - } - - private void addRawVector(BooleanVector vector) { - for (int i = 0; i < vector.getPositionCount(); i++) { - CountDistinctBooleanAggregator.combine(state, vector.getBoolean(i)); - } - } - - private void addRawBlock(BooleanBlock block) { - for (int p = 0; p < block.getPositionCount(); p++) { - if (block.isNull(p)) { - continue; - } - int start = block.getFirstValueIndex(p); - int end = start + block.getValueCount(p); - for (int i = start; i < end; i++) { - CountDistinctBooleanAggregator.combine(state, block.getBoolean(i)); - } - } - } - - @Override - public void addIntermediateInput(Page page) { - assert channels.size() == intermediateBlockCount(); - assert page.getBlockCount() >= channels.get(0) + intermediateStateDesc().size(); - Block fbitUncast = page.getBlock(channels.get(0)); - if (fbitUncast.areAllValuesNull()) { - return; - } - BooleanVector fbit = ((BooleanBlock) fbitUncast).asVector(); - assert fbit.getPositionCount() == 1; - Block tbitUncast = page.getBlock(channels.get(1)); - if (tbitUncast.areAllValuesNull()) { - return; - } - BooleanVector tbit = ((BooleanBlock) tbitUncast).asVector(); - assert tbit.getPositionCount() == 1; - CountDistinctBooleanAggregator.combineIntermediate(state, fbit.getBoolean(0), tbit.getBoolean(0)); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, DriverContext driverContext) { - state.toIntermediate(blocks, offset, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, DriverContext driverContext) { - blocks[offset] = CountDistinctBooleanAggregator.evaluateFinal(state, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctBooleanAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctBooleanAggregatorFunctionSupplier.java deleted file mode 100644 index 9512f4e76c49c..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctBooleanAggregatorFunctionSupplier.java +++ /dev/null @@ -1,39 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link CountDistinctBooleanAggregator}. - * This class is generated. Do not edit it. - */ -public final class CountDistinctBooleanAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - public CountDistinctBooleanAggregatorFunctionSupplier(List channels) { - this.channels = channels; - } - - @Override - public CountDistinctBooleanAggregatorFunction aggregator(DriverContext driverContext) { - return CountDistinctBooleanAggregatorFunction.create(driverContext, channels); - } - - @Override - public CountDistinctBooleanGroupingAggregatorFunction groupingAggregator( - DriverContext driverContext) { - return CountDistinctBooleanGroupingAggregatorFunction.create(channels, driverContext); - } - - @Override - public String describe() { - return "count_distinct of booleans"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctBooleanGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctBooleanGroupingAggregatorFunction.java deleted file mode 100644 index a12677e70e8a9..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctBooleanGroupingAggregatorFunction.java +++ /dev/null @@ -1,202 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link CountDistinctBooleanAggregator}. - * This class is generated. Do not edit it. - */ -public final class CountDistinctBooleanGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("fbit", ElementType.BOOLEAN), - new IntermediateStateDesc("tbit", ElementType.BOOLEAN) ); - - private final CountDistinctBooleanAggregator.GroupingState state; - - private final List channels; - - private final DriverContext driverContext; - - public CountDistinctBooleanGroupingAggregatorFunction(List channels, - CountDistinctBooleanAggregator.GroupingState state, DriverContext driverContext) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - } - - public static CountDistinctBooleanGroupingAggregatorFunction create(List channels, - DriverContext driverContext) { - return new CountDistinctBooleanGroupingAggregatorFunction(channels, CountDistinctBooleanAggregator.initGrouping(driverContext.bigArrays()), driverContext); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - BooleanBlock valuesBlock = page.getBlock(channels.get(0)); - BooleanVector valuesVector = valuesBlock.asVector(); - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, BooleanBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - CountDistinctBooleanAggregator.combine(state, groupId, values.getBoolean(v)); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, BooleanVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - CountDistinctBooleanAggregator.combine(state, groupId, values.getBoolean(groupPosition + positionOffset)); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, BooleanBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - CountDistinctBooleanAggregator.combine(state, groupId, values.getBoolean(v)); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, BooleanVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - CountDistinctBooleanAggregator.combine(state, groupId, values.getBoolean(groupPosition + positionOffset)); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block fbitUncast = page.getBlock(channels.get(0)); - if (fbitUncast.areAllValuesNull()) { - return; - } - BooleanVector fbit = ((BooleanBlock) fbitUncast).asVector(); - Block tbitUncast = page.getBlock(channels.get(1)); - if (tbitUncast.areAllValuesNull()) { - return; - } - BooleanVector tbit = ((BooleanBlock) tbitUncast).asVector(); - assert fbit.getPositionCount() == tbit.getPositionCount(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - CountDistinctBooleanAggregator.combineIntermediate(state, groupId, fbit.getBoolean(groupPosition + positionOffset), tbit.getBoolean(groupPosition + positionOffset)); - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - CountDistinctBooleanAggregator.GroupingState inState = ((CountDistinctBooleanGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - CountDistinctBooleanAggregator.combineStates(state, groupId, inState, position); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = CountDistinctBooleanAggregator.evaluateFinal(state, selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctBytesRefAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctBytesRefAggregatorFunction.java deleted file mode 100644 index 83917ba218285..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctBytesRefAggregatorFunction.java +++ /dev/null @@ -1,127 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunction} implementation for {@link CountDistinctBytesRefAggregator}. - * This class is generated. Do not edit it. - */ -public final class CountDistinctBytesRefAggregatorFunction implements AggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("hll", ElementType.BYTES_REF) ); - - private final DriverContext driverContext; - - private final HllStates.SingleState state; - - private final List channels; - - private final int precision; - - public CountDistinctBytesRefAggregatorFunction(DriverContext driverContext, - List channels, HllStates.SingleState state, int precision) { - this.driverContext = driverContext; - this.channels = channels; - this.state = state; - this.precision = precision; - } - - public static CountDistinctBytesRefAggregatorFunction create(DriverContext driverContext, - List channels, int precision) { - return new CountDistinctBytesRefAggregatorFunction(driverContext, channels, CountDistinctBytesRefAggregator.initSingle(driverContext.bigArrays(), precision), precision); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public void addRawInput(Page page) { - BytesRefBlock block = page.getBlock(channels.get(0)); - BytesRefVector vector = block.asVector(); - if (vector != null) { - addRawVector(vector); - } else { - addRawBlock(block); - } - } - - private void addRawVector(BytesRefVector vector) { - BytesRef scratch = new BytesRef(); - for (int i = 0; i < vector.getPositionCount(); i++) { - CountDistinctBytesRefAggregator.combine(state, vector.getBytesRef(i, scratch)); - } - } - - private void addRawBlock(BytesRefBlock block) { - BytesRef scratch = new BytesRef(); - for (int p = 0; p < block.getPositionCount(); p++) { - if (block.isNull(p)) { - continue; - } - int start = block.getFirstValueIndex(p); - int end = start + block.getValueCount(p); - for (int i = start; i < end; i++) { - CountDistinctBytesRefAggregator.combine(state, block.getBytesRef(i, scratch)); - } - } - } - - @Override - public void addIntermediateInput(Page page) { - assert channels.size() == intermediateBlockCount(); - assert page.getBlockCount() >= channels.get(0) + intermediateStateDesc().size(); - Block hllUncast = page.getBlock(channels.get(0)); - if (hllUncast.areAllValuesNull()) { - return; - } - BytesRefVector hll = ((BytesRefBlock) hllUncast).asVector(); - assert hll.getPositionCount() == 1; - BytesRef scratch = new BytesRef(); - CountDistinctBytesRefAggregator.combineIntermediate(state, hll.getBytesRef(0, scratch)); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, DriverContext driverContext) { - state.toIntermediate(blocks, offset, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, DriverContext driverContext) { - blocks[offset] = CountDistinctBytesRefAggregator.evaluateFinal(state, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctBytesRefAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctBytesRefAggregatorFunctionSupplier.java deleted file mode 100644 index b05c529c2ce9b..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctBytesRefAggregatorFunctionSupplier.java +++ /dev/null @@ -1,42 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link CountDistinctBytesRefAggregator}. - * This class is generated. Do not edit it. - */ -public final class CountDistinctBytesRefAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - private final int precision; - - public CountDistinctBytesRefAggregatorFunctionSupplier(List channels, int precision) { - this.channels = channels; - this.precision = precision; - } - - @Override - public CountDistinctBytesRefAggregatorFunction aggregator(DriverContext driverContext) { - return CountDistinctBytesRefAggregatorFunction.create(driverContext, channels, precision); - } - - @Override - public CountDistinctBytesRefGroupingAggregatorFunction groupingAggregator( - DriverContext driverContext) { - return CountDistinctBytesRefGroupingAggregatorFunction.create(channels, driverContext, precision); - } - - @Override - public String describe() { - return "count_distinct of bytes"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctBytesRefGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctBytesRefGroupingAggregatorFunction.java deleted file mode 100644 index 4879df5cf1c2c..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctBytesRefGroupingAggregatorFunction.java +++ /dev/null @@ -1,204 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link CountDistinctBytesRefAggregator}. - * This class is generated. Do not edit it. - */ -public final class CountDistinctBytesRefGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("hll", ElementType.BYTES_REF) ); - - private final HllStates.GroupingState state; - - private final List channels; - - private final DriverContext driverContext; - - private final int precision; - - public CountDistinctBytesRefGroupingAggregatorFunction(List channels, - HllStates.GroupingState state, DriverContext driverContext, int precision) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - this.precision = precision; - } - - public static CountDistinctBytesRefGroupingAggregatorFunction create(List channels, - DriverContext driverContext, int precision) { - return new CountDistinctBytesRefGroupingAggregatorFunction(channels, CountDistinctBytesRefAggregator.initGrouping(driverContext.bigArrays(), precision), driverContext, precision); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - BytesRefBlock valuesBlock = page.getBlock(channels.get(0)); - BytesRefVector valuesVector = valuesBlock.asVector(); - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, BytesRefBlock values) { - BytesRef scratch = new BytesRef(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - CountDistinctBytesRefAggregator.combine(state, groupId, values.getBytesRef(v, scratch)); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, BytesRefVector values) { - BytesRef scratch = new BytesRef(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - CountDistinctBytesRefAggregator.combine(state, groupId, values.getBytesRef(groupPosition + positionOffset, scratch)); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, BytesRefBlock values) { - BytesRef scratch = new BytesRef(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - CountDistinctBytesRefAggregator.combine(state, groupId, values.getBytesRef(v, scratch)); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, BytesRefVector values) { - BytesRef scratch = new BytesRef(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - CountDistinctBytesRefAggregator.combine(state, groupId, values.getBytesRef(groupPosition + positionOffset, scratch)); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block hllUncast = page.getBlock(channels.get(0)); - if (hllUncast.areAllValuesNull()) { - return; - } - BytesRefVector hll = ((BytesRefBlock) hllUncast).asVector(); - BytesRef scratch = new BytesRef(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - CountDistinctBytesRefAggregator.combineIntermediate(state, groupId, hll.getBytesRef(groupPosition + positionOffset, scratch)); - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - HllStates.GroupingState inState = ((CountDistinctBytesRefGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - CountDistinctBytesRefAggregator.combineStates(state, groupId, inState, position); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = CountDistinctBytesRefAggregator.evaluateFinal(state, selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctDoubleAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctDoubleAggregatorFunction.java deleted file mode 100644 index c720df313bd99..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctDoubleAggregatorFunction.java +++ /dev/null @@ -1,127 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunction} implementation for {@link CountDistinctDoubleAggregator}. - * This class is generated. Do not edit it. - */ -public final class CountDistinctDoubleAggregatorFunction implements AggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("hll", ElementType.BYTES_REF) ); - - private final DriverContext driverContext; - - private final HllStates.SingleState state; - - private final List channels; - - private final int precision; - - public CountDistinctDoubleAggregatorFunction(DriverContext driverContext, List channels, - HllStates.SingleState state, int precision) { - this.driverContext = driverContext; - this.channels = channels; - this.state = state; - this.precision = precision; - } - - public static CountDistinctDoubleAggregatorFunction create(DriverContext driverContext, - List channels, int precision) { - return new CountDistinctDoubleAggregatorFunction(driverContext, channels, CountDistinctDoubleAggregator.initSingle(driverContext.bigArrays(), precision), precision); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public void addRawInput(Page page) { - DoubleBlock block = page.getBlock(channels.get(0)); - DoubleVector vector = block.asVector(); - if (vector != null) { - addRawVector(vector); - } else { - addRawBlock(block); - } - } - - private void addRawVector(DoubleVector vector) { - for (int i = 0; i < vector.getPositionCount(); i++) { - CountDistinctDoubleAggregator.combine(state, vector.getDouble(i)); - } - } - - private void addRawBlock(DoubleBlock block) { - for (int p = 0; p < block.getPositionCount(); p++) { - if (block.isNull(p)) { - continue; - } - int start = block.getFirstValueIndex(p); - int end = start + block.getValueCount(p); - for (int i = start; i < end; i++) { - CountDistinctDoubleAggregator.combine(state, block.getDouble(i)); - } - } - } - - @Override - public void addIntermediateInput(Page page) { - assert channels.size() == intermediateBlockCount(); - assert page.getBlockCount() >= channels.get(0) + intermediateStateDesc().size(); - Block hllUncast = page.getBlock(channels.get(0)); - if (hllUncast.areAllValuesNull()) { - return; - } - BytesRefVector hll = ((BytesRefBlock) hllUncast).asVector(); - assert hll.getPositionCount() == 1; - BytesRef scratch = new BytesRef(); - CountDistinctDoubleAggregator.combineIntermediate(state, hll.getBytesRef(0, scratch)); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, DriverContext driverContext) { - state.toIntermediate(blocks, offset, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, DriverContext driverContext) { - blocks[offset] = CountDistinctDoubleAggregator.evaluateFinal(state, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctDoubleAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctDoubleAggregatorFunctionSupplier.java deleted file mode 100644 index 0a15ebb07ecf4..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctDoubleAggregatorFunctionSupplier.java +++ /dev/null @@ -1,42 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link CountDistinctDoubleAggregator}. - * This class is generated. Do not edit it. - */ -public final class CountDistinctDoubleAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - private final int precision; - - public CountDistinctDoubleAggregatorFunctionSupplier(List channels, int precision) { - this.channels = channels; - this.precision = precision; - } - - @Override - public CountDistinctDoubleAggregatorFunction aggregator(DriverContext driverContext) { - return CountDistinctDoubleAggregatorFunction.create(driverContext, channels, precision); - } - - @Override - public CountDistinctDoubleGroupingAggregatorFunction groupingAggregator( - DriverContext driverContext) { - return CountDistinctDoubleGroupingAggregatorFunction.create(channels, driverContext, precision); - } - - @Override - public String describe() { - return "count_distinct of doubles"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctDoubleGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctDoubleGroupingAggregatorFunction.java deleted file mode 100644 index 1e0ce58377f9e..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctDoubleGroupingAggregatorFunction.java +++ /dev/null @@ -1,202 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link CountDistinctDoubleAggregator}. - * This class is generated. Do not edit it. - */ -public final class CountDistinctDoubleGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("hll", ElementType.BYTES_REF) ); - - private final HllStates.GroupingState state; - - private final List channels; - - private final DriverContext driverContext; - - private final int precision; - - public CountDistinctDoubleGroupingAggregatorFunction(List channels, - HllStates.GroupingState state, DriverContext driverContext, int precision) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - this.precision = precision; - } - - public static CountDistinctDoubleGroupingAggregatorFunction create(List channels, - DriverContext driverContext, int precision) { - return new CountDistinctDoubleGroupingAggregatorFunction(channels, CountDistinctDoubleAggregator.initGrouping(driverContext.bigArrays(), precision), driverContext, precision); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - DoubleBlock valuesBlock = page.getBlock(channels.get(0)); - DoubleVector valuesVector = valuesBlock.asVector(); - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, DoubleBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - CountDistinctDoubleAggregator.combine(state, groupId, values.getDouble(v)); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, DoubleVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - CountDistinctDoubleAggregator.combine(state, groupId, values.getDouble(groupPosition + positionOffset)); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, DoubleBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - CountDistinctDoubleAggregator.combine(state, groupId, values.getDouble(v)); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, DoubleVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - CountDistinctDoubleAggregator.combine(state, groupId, values.getDouble(groupPosition + positionOffset)); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block hllUncast = page.getBlock(channels.get(0)); - if (hllUncast.areAllValuesNull()) { - return; - } - BytesRefVector hll = ((BytesRefBlock) hllUncast).asVector(); - BytesRef scratch = new BytesRef(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - CountDistinctDoubleAggregator.combineIntermediate(state, groupId, hll.getBytesRef(groupPosition + positionOffset, scratch)); - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - HllStates.GroupingState inState = ((CountDistinctDoubleGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - CountDistinctDoubleAggregator.combineStates(state, groupId, inState, position); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = CountDistinctDoubleAggregator.evaluateFinal(state, selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctIntAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctIntAggregatorFunction.java deleted file mode 100644 index 083c483a9c9bc..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctIntAggregatorFunction.java +++ /dev/null @@ -1,127 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunction} implementation for {@link CountDistinctIntAggregator}. - * This class is generated. Do not edit it. - */ -public final class CountDistinctIntAggregatorFunction implements AggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("hll", ElementType.BYTES_REF) ); - - private final DriverContext driverContext; - - private final HllStates.SingleState state; - - private final List channels; - - private final int precision; - - public CountDistinctIntAggregatorFunction(DriverContext driverContext, List channels, - HllStates.SingleState state, int precision) { - this.driverContext = driverContext; - this.channels = channels; - this.state = state; - this.precision = precision; - } - - public static CountDistinctIntAggregatorFunction create(DriverContext driverContext, - List channels, int precision) { - return new CountDistinctIntAggregatorFunction(driverContext, channels, CountDistinctIntAggregator.initSingle(driverContext.bigArrays(), precision), precision); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public void addRawInput(Page page) { - IntBlock block = page.getBlock(channels.get(0)); - IntVector vector = block.asVector(); - if (vector != null) { - addRawVector(vector); - } else { - addRawBlock(block); - } - } - - private void addRawVector(IntVector vector) { - for (int i = 0; i < vector.getPositionCount(); i++) { - CountDistinctIntAggregator.combine(state, vector.getInt(i)); - } - } - - private void addRawBlock(IntBlock block) { - for (int p = 0; p < block.getPositionCount(); p++) { - if (block.isNull(p)) { - continue; - } - int start = block.getFirstValueIndex(p); - int end = start + block.getValueCount(p); - for (int i = start; i < end; i++) { - CountDistinctIntAggregator.combine(state, block.getInt(i)); - } - } - } - - @Override - public void addIntermediateInput(Page page) { - assert channels.size() == intermediateBlockCount(); - assert page.getBlockCount() >= channels.get(0) + intermediateStateDesc().size(); - Block hllUncast = page.getBlock(channels.get(0)); - if (hllUncast.areAllValuesNull()) { - return; - } - BytesRefVector hll = ((BytesRefBlock) hllUncast).asVector(); - assert hll.getPositionCount() == 1; - BytesRef scratch = new BytesRef(); - CountDistinctIntAggregator.combineIntermediate(state, hll.getBytesRef(0, scratch)); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, DriverContext driverContext) { - state.toIntermediate(blocks, offset, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, DriverContext driverContext) { - blocks[offset] = CountDistinctIntAggregator.evaluateFinal(state, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctIntAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctIntAggregatorFunctionSupplier.java deleted file mode 100644 index fec5b7df48a21..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctIntAggregatorFunctionSupplier.java +++ /dev/null @@ -1,42 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link CountDistinctIntAggregator}. - * This class is generated. Do not edit it. - */ -public final class CountDistinctIntAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - private final int precision; - - public CountDistinctIntAggregatorFunctionSupplier(List channels, int precision) { - this.channels = channels; - this.precision = precision; - } - - @Override - public CountDistinctIntAggregatorFunction aggregator(DriverContext driverContext) { - return CountDistinctIntAggregatorFunction.create(driverContext, channels, precision); - } - - @Override - public CountDistinctIntGroupingAggregatorFunction groupingAggregator( - DriverContext driverContext) { - return CountDistinctIntGroupingAggregatorFunction.create(channels, driverContext, precision); - } - - @Override - public String describe() { - return "count_distinct of ints"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctIntGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctIntGroupingAggregatorFunction.java deleted file mode 100644 index 99e6ace52b256..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctIntGroupingAggregatorFunction.java +++ /dev/null @@ -1,200 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link CountDistinctIntAggregator}. - * This class is generated. Do not edit it. - */ -public final class CountDistinctIntGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("hll", ElementType.BYTES_REF) ); - - private final HllStates.GroupingState state; - - private final List channels; - - private final DriverContext driverContext; - - private final int precision; - - public CountDistinctIntGroupingAggregatorFunction(List channels, - HllStates.GroupingState state, DriverContext driverContext, int precision) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - this.precision = precision; - } - - public static CountDistinctIntGroupingAggregatorFunction create(List channels, - DriverContext driverContext, int precision) { - return new CountDistinctIntGroupingAggregatorFunction(channels, CountDistinctIntAggregator.initGrouping(driverContext.bigArrays(), precision), driverContext, precision); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - IntBlock valuesBlock = page.getBlock(channels.get(0)); - IntVector valuesVector = valuesBlock.asVector(); - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, IntBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - CountDistinctIntAggregator.combine(state, groupId, values.getInt(v)); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, IntVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - CountDistinctIntAggregator.combine(state, groupId, values.getInt(groupPosition + positionOffset)); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, IntBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - CountDistinctIntAggregator.combine(state, groupId, values.getInt(v)); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, IntVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - CountDistinctIntAggregator.combine(state, groupId, values.getInt(groupPosition + positionOffset)); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block hllUncast = page.getBlock(channels.get(0)); - if (hllUncast.areAllValuesNull()) { - return; - } - BytesRefVector hll = ((BytesRefBlock) hllUncast).asVector(); - BytesRef scratch = new BytesRef(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - CountDistinctIntAggregator.combineIntermediate(state, groupId, hll.getBytesRef(groupPosition + positionOffset, scratch)); - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - HllStates.GroupingState inState = ((CountDistinctIntGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - CountDistinctIntAggregator.combineStates(state, groupId, inState, position); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = CountDistinctIntAggregator.evaluateFinal(state, selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctLongAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctLongAggregatorFunction.java deleted file mode 100644 index ee6fb5b470442..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctLongAggregatorFunction.java +++ /dev/null @@ -1,127 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunction} implementation for {@link CountDistinctLongAggregator}. - * This class is generated. Do not edit it. - */ -public final class CountDistinctLongAggregatorFunction implements AggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("hll", ElementType.BYTES_REF) ); - - private final DriverContext driverContext; - - private final HllStates.SingleState state; - - private final List channels; - - private final int precision; - - public CountDistinctLongAggregatorFunction(DriverContext driverContext, List channels, - HllStates.SingleState state, int precision) { - this.driverContext = driverContext; - this.channels = channels; - this.state = state; - this.precision = precision; - } - - public static CountDistinctLongAggregatorFunction create(DriverContext driverContext, - List channels, int precision) { - return new CountDistinctLongAggregatorFunction(driverContext, channels, CountDistinctLongAggregator.initSingle(driverContext.bigArrays(), precision), precision); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public void addRawInput(Page page) { - LongBlock block = page.getBlock(channels.get(0)); - LongVector vector = block.asVector(); - if (vector != null) { - addRawVector(vector); - } else { - addRawBlock(block); - } - } - - private void addRawVector(LongVector vector) { - for (int i = 0; i < vector.getPositionCount(); i++) { - CountDistinctLongAggregator.combine(state, vector.getLong(i)); - } - } - - private void addRawBlock(LongBlock block) { - for (int p = 0; p < block.getPositionCount(); p++) { - if (block.isNull(p)) { - continue; - } - int start = block.getFirstValueIndex(p); - int end = start + block.getValueCount(p); - for (int i = start; i < end; i++) { - CountDistinctLongAggregator.combine(state, block.getLong(i)); - } - } - } - - @Override - public void addIntermediateInput(Page page) { - assert channels.size() == intermediateBlockCount(); - assert page.getBlockCount() >= channels.get(0) + intermediateStateDesc().size(); - Block hllUncast = page.getBlock(channels.get(0)); - if (hllUncast.areAllValuesNull()) { - return; - } - BytesRefVector hll = ((BytesRefBlock) hllUncast).asVector(); - assert hll.getPositionCount() == 1; - BytesRef scratch = new BytesRef(); - CountDistinctLongAggregator.combineIntermediate(state, hll.getBytesRef(0, scratch)); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, DriverContext driverContext) { - state.toIntermediate(blocks, offset, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, DriverContext driverContext) { - blocks[offset] = CountDistinctLongAggregator.evaluateFinal(state, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctLongAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctLongAggregatorFunctionSupplier.java deleted file mode 100644 index e3cc788215d39..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctLongAggregatorFunctionSupplier.java +++ /dev/null @@ -1,42 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link CountDistinctLongAggregator}. - * This class is generated. Do not edit it. - */ -public final class CountDistinctLongAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - private final int precision; - - public CountDistinctLongAggregatorFunctionSupplier(List channels, int precision) { - this.channels = channels; - this.precision = precision; - } - - @Override - public CountDistinctLongAggregatorFunction aggregator(DriverContext driverContext) { - return CountDistinctLongAggregatorFunction.create(driverContext, channels, precision); - } - - @Override - public CountDistinctLongGroupingAggregatorFunction groupingAggregator( - DriverContext driverContext) { - return CountDistinctLongGroupingAggregatorFunction.create(channels, driverContext, precision); - } - - @Override - public String describe() { - return "count_distinct of longs"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctLongGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctLongGroupingAggregatorFunction.java deleted file mode 100644 index 85f823296c886..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/CountDistinctLongGroupingAggregatorFunction.java +++ /dev/null @@ -1,202 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link CountDistinctLongAggregator}. - * This class is generated. Do not edit it. - */ -public final class CountDistinctLongGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("hll", ElementType.BYTES_REF) ); - - private final HllStates.GroupingState state; - - private final List channels; - - private final DriverContext driverContext; - - private final int precision; - - public CountDistinctLongGroupingAggregatorFunction(List channels, - HllStates.GroupingState state, DriverContext driverContext, int precision) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - this.precision = precision; - } - - public static CountDistinctLongGroupingAggregatorFunction create(List channels, - DriverContext driverContext, int precision) { - return new CountDistinctLongGroupingAggregatorFunction(channels, CountDistinctLongAggregator.initGrouping(driverContext.bigArrays(), precision), driverContext, precision); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - LongBlock valuesBlock = page.getBlock(channels.get(0)); - LongVector valuesVector = valuesBlock.asVector(); - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, LongBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - CountDistinctLongAggregator.combine(state, groupId, values.getLong(v)); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, LongVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - CountDistinctLongAggregator.combine(state, groupId, values.getLong(groupPosition + positionOffset)); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, LongBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - CountDistinctLongAggregator.combine(state, groupId, values.getLong(v)); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, LongVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - CountDistinctLongAggregator.combine(state, groupId, values.getLong(groupPosition + positionOffset)); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block hllUncast = page.getBlock(channels.get(0)); - if (hllUncast.areAllValuesNull()) { - return; - } - BytesRefVector hll = ((BytesRefBlock) hllUncast).asVector(); - BytesRef scratch = new BytesRef(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - CountDistinctLongAggregator.combineIntermediate(state, groupId, hll.getBytesRef(groupPosition + positionOffset, scratch)); - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - HllStates.GroupingState inState = ((CountDistinctLongGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - CountDistinctLongAggregator.combineStates(state, groupId, inState, position); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = CountDistinctLongAggregator.evaluateFinal(state, selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MaxDoubleAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MaxDoubleAggregatorFunction.java deleted file mode 100644 index f78a8773ccfcd..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MaxDoubleAggregatorFunction.java +++ /dev/null @@ -1,138 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunction} implementation for {@link MaxDoubleAggregator}. - * This class is generated. Do not edit it. - */ -public final class MaxDoubleAggregatorFunction implements AggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("max", ElementType.DOUBLE), - new IntermediateStateDesc("seen", ElementType.BOOLEAN) ); - - private final DriverContext driverContext; - - private final DoubleState state; - - private final List channels; - - public MaxDoubleAggregatorFunction(DriverContext driverContext, List channels, - DoubleState state) { - this.driverContext = driverContext; - this.channels = channels; - this.state = state; - } - - public static MaxDoubleAggregatorFunction create(DriverContext driverContext, - List channels) { - return new MaxDoubleAggregatorFunction(driverContext, channels, new DoubleState(MaxDoubleAggregator.init())); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public void addRawInput(Page page) { - DoubleBlock block = page.getBlock(channels.get(0)); - DoubleVector vector = block.asVector(); - if (vector != null) { - addRawVector(vector); - } else { - addRawBlock(block); - } - } - - private void addRawVector(DoubleVector vector) { - state.seen(true); - for (int i = 0; i < vector.getPositionCount(); i++) { - state.doubleValue(MaxDoubleAggregator.combine(state.doubleValue(), vector.getDouble(i))); - } - } - - private void addRawBlock(DoubleBlock block) { - for (int p = 0; p < block.getPositionCount(); p++) { - if (block.isNull(p)) { - continue; - } - state.seen(true); - int start = block.getFirstValueIndex(p); - int end = start + block.getValueCount(p); - for (int i = start; i < end; i++) { - state.doubleValue(MaxDoubleAggregator.combine(state.doubleValue(), block.getDouble(i))); - } - } - } - - @Override - public void addIntermediateInput(Page page) { - assert channels.size() == intermediateBlockCount(); - assert page.getBlockCount() >= channels.get(0) + intermediateStateDesc().size(); - Block maxUncast = page.getBlock(channels.get(0)); - if (maxUncast.areAllValuesNull()) { - return; - } - DoubleVector max = ((DoubleBlock) maxUncast).asVector(); - assert max.getPositionCount() == 1; - Block seenUncast = page.getBlock(channels.get(1)); - if (seenUncast.areAllValuesNull()) { - return; - } - BooleanVector seen = ((BooleanBlock) seenUncast).asVector(); - assert seen.getPositionCount() == 1; - if (seen.getBoolean(0)) { - state.doubleValue(MaxDoubleAggregator.combine(state.doubleValue(), max.getDouble(0))); - state.seen(true); - } - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, DriverContext driverContext) { - state.toIntermediate(blocks, offset, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, DriverContext driverContext) { - if (state.seen() == false) { - blocks[offset] = driverContext.blockFactory().newConstantNullBlock(1); - return; - } - blocks[offset] = driverContext.blockFactory().newConstantDoubleBlockWith(state.doubleValue(), 1); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MaxDoubleAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MaxDoubleAggregatorFunctionSupplier.java deleted file mode 100644 index af878fc778985..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MaxDoubleAggregatorFunctionSupplier.java +++ /dev/null @@ -1,38 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link MaxDoubleAggregator}. - * This class is generated. Do not edit it. - */ -public final class MaxDoubleAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - public MaxDoubleAggregatorFunctionSupplier(List channels) { - this.channels = channels; - } - - @Override - public MaxDoubleAggregatorFunction aggregator(DriverContext driverContext) { - return MaxDoubleAggregatorFunction.create(driverContext, channels); - } - - @Override - public MaxDoubleGroupingAggregatorFunction groupingAggregator(DriverContext driverContext) { - return MaxDoubleGroupingAggregatorFunction.create(channels, driverContext); - } - - @Override - public String describe() { - return "max of doubles"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MaxDoubleGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MaxDoubleGroupingAggregatorFunction.java deleted file mode 100644 index da93320eaf96e..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MaxDoubleGroupingAggregatorFunction.java +++ /dev/null @@ -1,208 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link MaxDoubleAggregator}. - * This class is generated. Do not edit it. - */ -public final class MaxDoubleGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("max", ElementType.DOUBLE), - new IntermediateStateDesc("seen", ElementType.BOOLEAN) ); - - private final DoubleArrayState state; - - private final List channels; - - private final DriverContext driverContext; - - public MaxDoubleGroupingAggregatorFunction(List channels, DoubleArrayState state, - DriverContext driverContext) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - } - - public static MaxDoubleGroupingAggregatorFunction create(List channels, - DriverContext driverContext) { - return new MaxDoubleGroupingAggregatorFunction(channels, new DoubleArrayState(driverContext.bigArrays(), MaxDoubleAggregator.init()), driverContext); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - DoubleBlock valuesBlock = page.getBlock(channels.get(0)); - DoubleVector valuesVector = valuesBlock.asVector(); - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, DoubleBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - state.set(groupId, MaxDoubleAggregator.combine(state.getOrDefault(groupId), values.getDouble(v))); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, DoubleVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - state.set(groupId, MaxDoubleAggregator.combine(state.getOrDefault(groupId), values.getDouble(groupPosition + positionOffset))); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, DoubleBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - state.set(groupId, MaxDoubleAggregator.combine(state.getOrDefault(groupId), values.getDouble(v))); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, DoubleVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - state.set(groupId, MaxDoubleAggregator.combine(state.getOrDefault(groupId), values.getDouble(groupPosition + positionOffset))); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block maxUncast = page.getBlock(channels.get(0)); - if (maxUncast.areAllValuesNull()) { - return; - } - DoubleVector max = ((DoubleBlock) maxUncast).asVector(); - Block seenUncast = page.getBlock(channels.get(1)); - if (seenUncast.areAllValuesNull()) { - return; - } - BooleanVector seen = ((BooleanBlock) seenUncast).asVector(); - assert max.getPositionCount() == seen.getPositionCount(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (seen.getBoolean(groupPosition + positionOffset)) { - state.set(groupId, MaxDoubleAggregator.combine(state.getOrDefault(groupId), max.getDouble(groupPosition + positionOffset))); - } - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - DoubleArrayState inState = ((MaxDoubleGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - if (inState.hasValue(position)) { - state.set(groupId, MaxDoubleAggregator.combine(state.getOrDefault(groupId), inState.get(position))); - } - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = state.toValuesBlock(selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MaxIntAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MaxIntAggregatorFunction.java deleted file mode 100644 index 6f83ee7224879..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MaxIntAggregatorFunction.java +++ /dev/null @@ -1,138 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunction} implementation for {@link MaxIntAggregator}. - * This class is generated. Do not edit it. - */ -public final class MaxIntAggregatorFunction implements AggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("max", ElementType.INT), - new IntermediateStateDesc("seen", ElementType.BOOLEAN) ); - - private final DriverContext driverContext; - - private final IntState state; - - private final List channels; - - public MaxIntAggregatorFunction(DriverContext driverContext, List channels, - IntState state) { - this.driverContext = driverContext; - this.channels = channels; - this.state = state; - } - - public static MaxIntAggregatorFunction create(DriverContext driverContext, - List channels) { - return new MaxIntAggregatorFunction(driverContext, channels, new IntState(MaxIntAggregator.init())); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public void addRawInput(Page page) { - IntBlock block = page.getBlock(channels.get(0)); - IntVector vector = block.asVector(); - if (vector != null) { - addRawVector(vector); - } else { - addRawBlock(block); - } - } - - private void addRawVector(IntVector vector) { - state.seen(true); - for (int i = 0; i < vector.getPositionCount(); i++) { - state.intValue(MaxIntAggregator.combine(state.intValue(), vector.getInt(i))); - } - } - - private void addRawBlock(IntBlock block) { - for (int p = 0; p < block.getPositionCount(); p++) { - if (block.isNull(p)) { - continue; - } - state.seen(true); - int start = block.getFirstValueIndex(p); - int end = start + block.getValueCount(p); - for (int i = start; i < end; i++) { - state.intValue(MaxIntAggregator.combine(state.intValue(), block.getInt(i))); - } - } - } - - @Override - public void addIntermediateInput(Page page) { - assert channels.size() == intermediateBlockCount(); - assert page.getBlockCount() >= channels.get(0) + intermediateStateDesc().size(); - Block maxUncast = page.getBlock(channels.get(0)); - if (maxUncast.areAllValuesNull()) { - return; - } - IntVector max = ((IntBlock) maxUncast).asVector(); - assert max.getPositionCount() == 1; - Block seenUncast = page.getBlock(channels.get(1)); - if (seenUncast.areAllValuesNull()) { - return; - } - BooleanVector seen = ((BooleanBlock) seenUncast).asVector(); - assert seen.getPositionCount() == 1; - if (seen.getBoolean(0)) { - state.intValue(MaxIntAggregator.combine(state.intValue(), max.getInt(0))); - state.seen(true); - } - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, DriverContext driverContext) { - state.toIntermediate(blocks, offset, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, DriverContext driverContext) { - if (state.seen() == false) { - blocks[offset] = driverContext.blockFactory().newConstantNullBlock(1); - return; - } - blocks[offset] = driverContext.blockFactory().newConstantIntBlockWith(state.intValue(), 1); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MaxIntAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MaxIntAggregatorFunctionSupplier.java deleted file mode 100644 index 5e0a4e2172696..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MaxIntAggregatorFunctionSupplier.java +++ /dev/null @@ -1,38 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link MaxIntAggregator}. - * This class is generated. Do not edit it. - */ -public final class MaxIntAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - public MaxIntAggregatorFunctionSupplier(List channels) { - this.channels = channels; - } - - @Override - public MaxIntAggregatorFunction aggregator(DriverContext driverContext) { - return MaxIntAggregatorFunction.create(driverContext, channels); - } - - @Override - public MaxIntGroupingAggregatorFunction groupingAggregator(DriverContext driverContext) { - return MaxIntGroupingAggregatorFunction.create(channels, driverContext); - } - - @Override - public String describe() { - return "max of ints"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MaxIntGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MaxIntGroupingAggregatorFunction.java deleted file mode 100644 index c8b1b6910c0aa..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MaxIntGroupingAggregatorFunction.java +++ /dev/null @@ -1,206 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link MaxIntAggregator}. - * This class is generated. Do not edit it. - */ -public final class MaxIntGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("max", ElementType.INT), - new IntermediateStateDesc("seen", ElementType.BOOLEAN) ); - - private final IntArrayState state; - - private final List channels; - - private final DriverContext driverContext; - - public MaxIntGroupingAggregatorFunction(List channels, IntArrayState state, - DriverContext driverContext) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - } - - public static MaxIntGroupingAggregatorFunction create(List channels, - DriverContext driverContext) { - return new MaxIntGroupingAggregatorFunction(channels, new IntArrayState(driverContext.bigArrays(), MaxIntAggregator.init()), driverContext); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - IntBlock valuesBlock = page.getBlock(channels.get(0)); - IntVector valuesVector = valuesBlock.asVector(); - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, IntBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - state.set(groupId, MaxIntAggregator.combine(state.getOrDefault(groupId), values.getInt(v))); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, IntVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - state.set(groupId, MaxIntAggregator.combine(state.getOrDefault(groupId), values.getInt(groupPosition + positionOffset))); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, IntBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - state.set(groupId, MaxIntAggregator.combine(state.getOrDefault(groupId), values.getInt(v))); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, IntVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - state.set(groupId, MaxIntAggregator.combine(state.getOrDefault(groupId), values.getInt(groupPosition + positionOffset))); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block maxUncast = page.getBlock(channels.get(0)); - if (maxUncast.areAllValuesNull()) { - return; - } - IntVector max = ((IntBlock) maxUncast).asVector(); - Block seenUncast = page.getBlock(channels.get(1)); - if (seenUncast.areAllValuesNull()) { - return; - } - BooleanVector seen = ((BooleanBlock) seenUncast).asVector(); - assert max.getPositionCount() == seen.getPositionCount(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (seen.getBoolean(groupPosition + positionOffset)) { - state.set(groupId, MaxIntAggregator.combine(state.getOrDefault(groupId), max.getInt(groupPosition + positionOffset))); - } - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - IntArrayState inState = ((MaxIntGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - if (inState.hasValue(position)) { - state.set(groupId, MaxIntAggregator.combine(state.getOrDefault(groupId), inState.get(position))); - } - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = state.toValuesBlock(selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MaxLongAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MaxLongAggregatorFunction.java deleted file mode 100644 index 8826128a68837..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MaxLongAggregatorFunction.java +++ /dev/null @@ -1,138 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunction} implementation for {@link MaxLongAggregator}. - * This class is generated. Do not edit it. - */ -public final class MaxLongAggregatorFunction implements AggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("max", ElementType.LONG), - new IntermediateStateDesc("seen", ElementType.BOOLEAN) ); - - private final DriverContext driverContext; - - private final LongState state; - - private final List channels; - - public MaxLongAggregatorFunction(DriverContext driverContext, List channels, - LongState state) { - this.driverContext = driverContext; - this.channels = channels; - this.state = state; - } - - public static MaxLongAggregatorFunction create(DriverContext driverContext, - List channels) { - return new MaxLongAggregatorFunction(driverContext, channels, new LongState(MaxLongAggregator.init())); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public void addRawInput(Page page) { - LongBlock block = page.getBlock(channels.get(0)); - LongVector vector = block.asVector(); - if (vector != null) { - addRawVector(vector); - } else { - addRawBlock(block); - } - } - - private void addRawVector(LongVector vector) { - state.seen(true); - for (int i = 0; i < vector.getPositionCount(); i++) { - state.longValue(MaxLongAggregator.combine(state.longValue(), vector.getLong(i))); - } - } - - private void addRawBlock(LongBlock block) { - for (int p = 0; p < block.getPositionCount(); p++) { - if (block.isNull(p)) { - continue; - } - state.seen(true); - int start = block.getFirstValueIndex(p); - int end = start + block.getValueCount(p); - for (int i = start; i < end; i++) { - state.longValue(MaxLongAggregator.combine(state.longValue(), block.getLong(i))); - } - } - } - - @Override - public void addIntermediateInput(Page page) { - assert channels.size() == intermediateBlockCount(); - assert page.getBlockCount() >= channels.get(0) + intermediateStateDesc().size(); - Block maxUncast = page.getBlock(channels.get(0)); - if (maxUncast.areAllValuesNull()) { - return; - } - LongVector max = ((LongBlock) maxUncast).asVector(); - assert max.getPositionCount() == 1; - Block seenUncast = page.getBlock(channels.get(1)); - if (seenUncast.areAllValuesNull()) { - return; - } - BooleanVector seen = ((BooleanBlock) seenUncast).asVector(); - assert seen.getPositionCount() == 1; - if (seen.getBoolean(0)) { - state.longValue(MaxLongAggregator.combine(state.longValue(), max.getLong(0))); - state.seen(true); - } - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, DriverContext driverContext) { - state.toIntermediate(blocks, offset, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, DriverContext driverContext) { - if (state.seen() == false) { - blocks[offset] = driverContext.blockFactory().newConstantNullBlock(1); - return; - } - blocks[offset] = driverContext.blockFactory().newConstantLongBlockWith(state.longValue(), 1); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MaxLongAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MaxLongAggregatorFunctionSupplier.java deleted file mode 100644 index f4d17da186d58..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MaxLongAggregatorFunctionSupplier.java +++ /dev/null @@ -1,38 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link MaxLongAggregator}. - * This class is generated. Do not edit it. - */ -public final class MaxLongAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - public MaxLongAggregatorFunctionSupplier(List channels) { - this.channels = channels; - } - - @Override - public MaxLongAggregatorFunction aggregator(DriverContext driverContext) { - return MaxLongAggregatorFunction.create(driverContext, channels); - } - - @Override - public MaxLongGroupingAggregatorFunction groupingAggregator(DriverContext driverContext) { - return MaxLongGroupingAggregatorFunction.create(channels, driverContext); - } - - @Override - public String describe() { - return "max of longs"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MaxLongGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MaxLongGroupingAggregatorFunction.java deleted file mode 100644 index 41d893f9bbf0c..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MaxLongGroupingAggregatorFunction.java +++ /dev/null @@ -1,208 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link MaxLongAggregator}. - * This class is generated. Do not edit it. - */ -public final class MaxLongGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("max", ElementType.LONG), - new IntermediateStateDesc("seen", ElementType.BOOLEAN) ); - - private final LongArrayState state; - - private final List channels; - - private final DriverContext driverContext; - - public MaxLongGroupingAggregatorFunction(List channels, LongArrayState state, - DriverContext driverContext) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - } - - public static MaxLongGroupingAggregatorFunction create(List channels, - DriverContext driverContext) { - return new MaxLongGroupingAggregatorFunction(channels, new LongArrayState(driverContext.bigArrays(), MaxLongAggregator.init()), driverContext); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - LongBlock valuesBlock = page.getBlock(channels.get(0)); - LongVector valuesVector = valuesBlock.asVector(); - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, LongBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - state.set(groupId, MaxLongAggregator.combine(state.getOrDefault(groupId), values.getLong(v))); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, LongVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - state.set(groupId, MaxLongAggregator.combine(state.getOrDefault(groupId), values.getLong(groupPosition + positionOffset))); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, LongBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - state.set(groupId, MaxLongAggregator.combine(state.getOrDefault(groupId), values.getLong(v))); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, LongVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - state.set(groupId, MaxLongAggregator.combine(state.getOrDefault(groupId), values.getLong(groupPosition + positionOffset))); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block maxUncast = page.getBlock(channels.get(0)); - if (maxUncast.areAllValuesNull()) { - return; - } - LongVector max = ((LongBlock) maxUncast).asVector(); - Block seenUncast = page.getBlock(channels.get(1)); - if (seenUncast.areAllValuesNull()) { - return; - } - BooleanVector seen = ((BooleanBlock) seenUncast).asVector(); - assert max.getPositionCount() == seen.getPositionCount(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (seen.getBoolean(groupPosition + positionOffset)) { - state.set(groupId, MaxLongAggregator.combine(state.getOrDefault(groupId), max.getLong(groupPosition + positionOffset))); - } - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - LongArrayState inState = ((MaxLongGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - if (inState.hasValue(position)) { - state.set(groupId, MaxLongAggregator.combine(state.getOrDefault(groupId), inState.get(position))); - } - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = state.toValuesBlock(selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MedianAbsoluteDeviationDoubleAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MedianAbsoluteDeviationDoubleAggregatorFunction.java deleted file mode 100644 index 4bcf08ce0fa35..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MedianAbsoluteDeviationDoubleAggregatorFunction.java +++ /dev/null @@ -1,124 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunction} implementation for {@link MedianAbsoluteDeviationDoubleAggregator}. - * This class is generated. Do not edit it. - */ -public final class MedianAbsoluteDeviationDoubleAggregatorFunction implements AggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("quart", ElementType.BYTES_REF) ); - - private final DriverContext driverContext; - - private final QuantileStates.SingleState state; - - private final List channels; - - public MedianAbsoluteDeviationDoubleAggregatorFunction(DriverContext driverContext, - List channels, QuantileStates.SingleState state) { - this.driverContext = driverContext; - this.channels = channels; - this.state = state; - } - - public static MedianAbsoluteDeviationDoubleAggregatorFunction create(DriverContext driverContext, - List channels) { - return new MedianAbsoluteDeviationDoubleAggregatorFunction(driverContext, channels, MedianAbsoluteDeviationDoubleAggregator.initSingle()); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public void addRawInput(Page page) { - DoubleBlock block = page.getBlock(channels.get(0)); - DoubleVector vector = block.asVector(); - if (vector != null) { - addRawVector(vector); - } else { - addRawBlock(block); - } - } - - private void addRawVector(DoubleVector vector) { - for (int i = 0; i < vector.getPositionCount(); i++) { - MedianAbsoluteDeviationDoubleAggregator.combine(state, vector.getDouble(i)); - } - } - - private void addRawBlock(DoubleBlock block) { - for (int p = 0; p < block.getPositionCount(); p++) { - if (block.isNull(p)) { - continue; - } - int start = block.getFirstValueIndex(p); - int end = start + block.getValueCount(p); - for (int i = start; i < end; i++) { - MedianAbsoluteDeviationDoubleAggregator.combine(state, block.getDouble(i)); - } - } - } - - @Override - public void addIntermediateInput(Page page) { - assert channels.size() == intermediateBlockCount(); - assert page.getBlockCount() >= channels.get(0) + intermediateStateDesc().size(); - Block quartUncast = page.getBlock(channels.get(0)); - if (quartUncast.areAllValuesNull()) { - return; - } - BytesRefVector quart = ((BytesRefBlock) quartUncast).asVector(); - assert quart.getPositionCount() == 1; - BytesRef scratch = new BytesRef(); - MedianAbsoluteDeviationDoubleAggregator.combineIntermediate(state, quart.getBytesRef(0, scratch)); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, DriverContext driverContext) { - state.toIntermediate(blocks, offset, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, DriverContext driverContext) { - blocks[offset] = MedianAbsoluteDeviationDoubleAggregator.evaluateFinal(state, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MedianAbsoluteDeviationDoubleAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MedianAbsoluteDeviationDoubleAggregatorFunctionSupplier.java deleted file mode 100644 index 4720ce08fa282..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MedianAbsoluteDeviationDoubleAggregatorFunctionSupplier.java +++ /dev/null @@ -1,39 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link MedianAbsoluteDeviationDoubleAggregator}. - * This class is generated. Do not edit it. - */ -public final class MedianAbsoluteDeviationDoubleAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - public MedianAbsoluteDeviationDoubleAggregatorFunctionSupplier(List channels) { - this.channels = channels; - } - - @Override - public MedianAbsoluteDeviationDoubleAggregatorFunction aggregator(DriverContext driverContext) { - return MedianAbsoluteDeviationDoubleAggregatorFunction.create(driverContext, channels); - } - - @Override - public MedianAbsoluteDeviationDoubleGroupingAggregatorFunction groupingAggregator( - DriverContext driverContext) { - return MedianAbsoluteDeviationDoubleGroupingAggregatorFunction.create(channels, driverContext); - } - - @Override - public String describe() { - return "median_absolute_deviation of doubles"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MedianAbsoluteDeviationDoubleGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MedianAbsoluteDeviationDoubleGroupingAggregatorFunction.java deleted file mode 100644 index e08488685d2cb..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MedianAbsoluteDeviationDoubleGroupingAggregatorFunction.java +++ /dev/null @@ -1,199 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link MedianAbsoluteDeviationDoubleAggregator}. - * This class is generated. Do not edit it. - */ -public final class MedianAbsoluteDeviationDoubleGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("quart", ElementType.BYTES_REF) ); - - private final QuantileStates.GroupingState state; - - private final List channels; - - private final DriverContext driverContext; - - public MedianAbsoluteDeviationDoubleGroupingAggregatorFunction(List channels, - QuantileStates.GroupingState state, DriverContext driverContext) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - } - - public static MedianAbsoluteDeviationDoubleGroupingAggregatorFunction create( - List channels, DriverContext driverContext) { - return new MedianAbsoluteDeviationDoubleGroupingAggregatorFunction(channels, MedianAbsoluteDeviationDoubleAggregator.initGrouping(driverContext.bigArrays()), driverContext); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - DoubleBlock valuesBlock = page.getBlock(channels.get(0)); - DoubleVector valuesVector = valuesBlock.asVector(); - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, DoubleBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - MedianAbsoluteDeviationDoubleAggregator.combine(state, groupId, values.getDouble(v)); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, DoubleVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - MedianAbsoluteDeviationDoubleAggregator.combine(state, groupId, values.getDouble(groupPosition + positionOffset)); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, DoubleBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - MedianAbsoluteDeviationDoubleAggregator.combine(state, groupId, values.getDouble(v)); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, DoubleVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - MedianAbsoluteDeviationDoubleAggregator.combine(state, groupId, values.getDouble(groupPosition + positionOffset)); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block quartUncast = page.getBlock(channels.get(0)); - if (quartUncast.areAllValuesNull()) { - return; - } - BytesRefVector quart = ((BytesRefBlock) quartUncast).asVector(); - BytesRef scratch = new BytesRef(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - MedianAbsoluteDeviationDoubleAggregator.combineIntermediate(state, groupId, quart.getBytesRef(groupPosition + positionOffset, scratch)); - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - QuantileStates.GroupingState inState = ((MedianAbsoluteDeviationDoubleGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - MedianAbsoluteDeviationDoubleAggregator.combineStates(state, groupId, inState, position); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = MedianAbsoluteDeviationDoubleAggregator.evaluateFinal(state, selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MedianAbsoluteDeviationIntAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MedianAbsoluteDeviationIntAggregatorFunction.java deleted file mode 100644 index db9dbdab52244..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MedianAbsoluteDeviationIntAggregatorFunction.java +++ /dev/null @@ -1,124 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunction} implementation for {@link MedianAbsoluteDeviationIntAggregator}. - * This class is generated. Do not edit it. - */ -public final class MedianAbsoluteDeviationIntAggregatorFunction implements AggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("quart", ElementType.BYTES_REF) ); - - private final DriverContext driverContext; - - private final QuantileStates.SingleState state; - - private final List channels; - - public MedianAbsoluteDeviationIntAggregatorFunction(DriverContext driverContext, - List channels, QuantileStates.SingleState state) { - this.driverContext = driverContext; - this.channels = channels; - this.state = state; - } - - public static MedianAbsoluteDeviationIntAggregatorFunction create(DriverContext driverContext, - List channels) { - return new MedianAbsoluteDeviationIntAggregatorFunction(driverContext, channels, MedianAbsoluteDeviationIntAggregator.initSingle()); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public void addRawInput(Page page) { - IntBlock block = page.getBlock(channels.get(0)); - IntVector vector = block.asVector(); - if (vector != null) { - addRawVector(vector); - } else { - addRawBlock(block); - } - } - - private void addRawVector(IntVector vector) { - for (int i = 0; i < vector.getPositionCount(); i++) { - MedianAbsoluteDeviationIntAggregator.combine(state, vector.getInt(i)); - } - } - - private void addRawBlock(IntBlock block) { - for (int p = 0; p < block.getPositionCount(); p++) { - if (block.isNull(p)) { - continue; - } - int start = block.getFirstValueIndex(p); - int end = start + block.getValueCount(p); - for (int i = start; i < end; i++) { - MedianAbsoluteDeviationIntAggregator.combine(state, block.getInt(i)); - } - } - } - - @Override - public void addIntermediateInput(Page page) { - assert channels.size() == intermediateBlockCount(); - assert page.getBlockCount() >= channels.get(0) + intermediateStateDesc().size(); - Block quartUncast = page.getBlock(channels.get(0)); - if (quartUncast.areAllValuesNull()) { - return; - } - BytesRefVector quart = ((BytesRefBlock) quartUncast).asVector(); - assert quart.getPositionCount() == 1; - BytesRef scratch = new BytesRef(); - MedianAbsoluteDeviationIntAggregator.combineIntermediate(state, quart.getBytesRef(0, scratch)); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, DriverContext driverContext) { - state.toIntermediate(blocks, offset, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, DriverContext driverContext) { - blocks[offset] = MedianAbsoluteDeviationIntAggregator.evaluateFinal(state, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MedianAbsoluteDeviationIntAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MedianAbsoluteDeviationIntAggregatorFunctionSupplier.java deleted file mode 100644 index e72918359b2f6..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MedianAbsoluteDeviationIntAggregatorFunctionSupplier.java +++ /dev/null @@ -1,39 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link MedianAbsoluteDeviationIntAggregator}. - * This class is generated. Do not edit it. - */ -public final class MedianAbsoluteDeviationIntAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - public MedianAbsoluteDeviationIntAggregatorFunctionSupplier(List channels) { - this.channels = channels; - } - - @Override - public MedianAbsoluteDeviationIntAggregatorFunction aggregator(DriverContext driverContext) { - return MedianAbsoluteDeviationIntAggregatorFunction.create(driverContext, channels); - } - - @Override - public MedianAbsoluteDeviationIntGroupingAggregatorFunction groupingAggregator( - DriverContext driverContext) { - return MedianAbsoluteDeviationIntGroupingAggregatorFunction.create(channels, driverContext); - } - - @Override - public String describe() { - return "median_absolute_deviation of ints"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MedianAbsoluteDeviationIntGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MedianAbsoluteDeviationIntGroupingAggregatorFunction.java deleted file mode 100644 index 02866ee15b961..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MedianAbsoluteDeviationIntGroupingAggregatorFunction.java +++ /dev/null @@ -1,197 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link MedianAbsoluteDeviationIntAggregator}. - * This class is generated. Do not edit it. - */ -public final class MedianAbsoluteDeviationIntGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("quart", ElementType.BYTES_REF) ); - - private final QuantileStates.GroupingState state; - - private final List channels; - - private final DriverContext driverContext; - - public MedianAbsoluteDeviationIntGroupingAggregatorFunction(List channels, - QuantileStates.GroupingState state, DriverContext driverContext) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - } - - public static MedianAbsoluteDeviationIntGroupingAggregatorFunction create(List channels, - DriverContext driverContext) { - return new MedianAbsoluteDeviationIntGroupingAggregatorFunction(channels, MedianAbsoluteDeviationIntAggregator.initGrouping(driverContext.bigArrays()), driverContext); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - IntBlock valuesBlock = page.getBlock(channels.get(0)); - IntVector valuesVector = valuesBlock.asVector(); - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, IntBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - MedianAbsoluteDeviationIntAggregator.combine(state, groupId, values.getInt(v)); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, IntVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - MedianAbsoluteDeviationIntAggregator.combine(state, groupId, values.getInt(groupPosition + positionOffset)); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, IntBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - MedianAbsoluteDeviationIntAggregator.combine(state, groupId, values.getInt(v)); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, IntVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - MedianAbsoluteDeviationIntAggregator.combine(state, groupId, values.getInt(groupPosition + positionOffset)); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block quartUncast = page.getBlock(channels.get(0)); - if (quartUncast.areAllValuesNull()) { - return; - } - BytesRefVector quart = ((BytesRefBlock) quartUncast).asVector(); - BytesRef scratch = new BytesRef(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - MedianAbsoluteDeviationIntAggregator.combineIntermediate(state, groupId, quart.getBytesRef(groupPosition + positionOffset, scratch)); - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - QuantileStates.GroupingState inState = ((MedianAbsoluteDeviationIntGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - MedianAbsoluteDeviationIntAggregator.combineStates(state, groupId, inState, position); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = MedianAbsoluteDeviationIntAggregator.evaluateFinal(state, selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MedianAbsoluteDeviationLongAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MedianAbsoluteDeviationLongAggregatorFunction.java deleted file mode 100644 index bf5fd51d7ed17..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MedianAbsoluteDeviationLongAggregatorFunction.java +++ /dev/null @@ -1,124 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunction} implementation for {@link MedianAbsoluteDeviationLongAggregator}. - * This class is generated. Do not edit it. - */ -public final class MedianAbsoluteDeviationLongAggregatorFunction implements AggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("quart", ElementType.BYTES_REF) ); - - private final DriverContext driverContext; - - private final QuantileStates.SingleState state; - - private final List channels; - - public MedianAbsoluteDeviationLongAggregatorFunction(DriverContext driverContext, - List channels, QuantileStates.SingleState state) { - this.driverContext = driverContext; - this.channels = channels; - this.state = state; - } - - public static MedianAbsoluteDeviationLongAggregatorFunction create(DriverContext driverContext, - List channels) { - return new MedianAbsoluteDeviationLongAggregatorFunction(driverContext, channels, MedianAbsoluteDeviationLongAggregator.initSingle()); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public void addRawInput(Page page) { - LongBlock block = page.getBlock(channels.get(0)); - LongVector vector = block.asVector(); - if (vector != null) { - addRawVector(vector); - } else { - addRawBlock(block); - } - } - - private void addRawVector(LongVector vector) { - for (int i = 0; i < vector.getPositionCount(); i++) { - MedianAbsoluteDeviationLongAggregator.combine(state, vector.getLong(i)); - } - } - - private void addRawBlock(LongBlock block) { - for (int p = 0; p < block.getPositionCount(); p++) { - if (block.isNull(p)) { - continue; - } - int start = block.getFirstValueIndex(p); - int end = start + block.getValueCount(p); - for (int i = start; i < end; i++) { - MedianAbsoluteDeviationLongAggregator.combine(state, block.getLong(i)); - } - } - } - - @Override - public void addIntermediateInput(Page page) { - assert channels.size() == intermediateBlockCount(); - assert page.getBlockCount() >= channels.get(0) + intermediateStateDesc().size(); - Block quartUncast = page.getBlock(channels.get(0)); - if (quartUncast.areAllValuesNull()) { - return; - } - BytesRefVector quart = ((BytesRefBlock) quartUncast).asVector(); - assert quart.getPositionCount() == 1; - BytesRef scratch = new BytesRef(); - MedianAbsoluteDeviationLongAggregator.combineIntermediate(state, quart.getBytesRef(0, scratch)); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, DriverContext driverContext) { - state.toIntermediate(blocks, offset, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, DriverContext driverContext) { - blocks[offset] = MedianAbsoluteDeviationLongAggregator.evaluateFinal(state, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MedianAbsoluteDeviationLongAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MedianAbsoluteDeviationLongAggregatorFunctionSupplier.java deleted file mode 100644 index aa79691ba220e..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MedianAbsoluteDeviationLongAggregatorFunctionSupplier.java +++ /dev/null @@ -1,39 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link MedianAbsoluteDeviationLongAggregator}. - * This class is generated. Do not edit it. - */ -public final class MedianAbsoluteDeviationLongAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - public MedianAbsoluteDeviationLongAggregatorFunctionSupplier(List channels) { - this.channels = channels; - } - - @Override - public MedianAbsoluteDeviationLongAggregatorFunction aggregator(DriverContext driverContext) { - return MedianAbsoluteDeviationLongAggregatorFunction.create(driverContext, channels); - } - - @Override - public MedianAbsoluteDeviationLongGroupingAggregatorFunction groupingAggregator( - DriverContext driverContext) { - return MedianAbsoluteDeviationLongGroupingAggregatorFunction.create(channels, driverContext); - } - - @Override - public String describe() { - return "median_absolute_deviation of longs"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MedianAbsoluteDeviationLongGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MedianAbsoluteDeviationLongGroupingAggregatorFunction.java deleted file mode 100644 index 36c40e10e54d5..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MedianAbsoluteDeviationLongGroupingAggregatorFunction.java +++ /dev/null @@ -1,199 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link MedianAbsoluteDeviationLongAggregator}. - * This class is generated. Do not edit it. - */ -public final class MedianAbsoluteDeviationLongGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("quart", ElementType.BYTES_REF) ); - - private final QuantileStates.GroupingState state; - - private final List channels; - - private final DriverContext driverContext; - - public MedianAbsoluteDeviationLongGroupingAggregatorFunction(List channels, - QuantileStates.GroupingState state, DriverContext driverContext) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - } - - public static MedianAbsoluteDeviationLongGroupingAggregatorFunction create(List channels, - DriverContext driverContext) { - return new MedianAbsoluteDeviationLongGroupingAggregatorFunction(channels, MedianAbsoluteDeviationLongAggregator.initGrouping(driverContext.bigArrays()), driverContext); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - LongBlock valuesBlock = page.getBlock(channels.get(0)); - LongVector valuesVector = valuesBlock.asVector(); - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, LongBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - MedianAbsoluteDeviationLongAggregator.combine(state, groupId, values.getLong(v)); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, LongVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - MedianAbsoluteDeviationLongAggregator.combine(state, groupId, values.getLong(groupPosition + positionOffset)); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, LongBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - MedianAbsoluteDeviationLongAggregator.combine(state, groupId, values.getLong(v)); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, LongVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - MedianAbsoluteDeviationLongAggregator.combine(state, groupId, values.getLong(groupPosition + positionOffset)); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block quartUncast = page.getBlock(channels.get(0)); - if (quartUncast.areAllValuesNull()) { - return; - } - BytesRefVector quart = ((BytesRefBlock) quartUncast).asVector(); - BytesRef scratch = new BytesRef(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - MedianAbsoluteDeviationLongAggregator.combineIntermediate(state, groupId, quart.getBytesRef(groupPosition + positionOffset, scratch)); - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - QuantileStates.GroupingState inState = ((MedianAbsoluteDeviationLongGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - MedianAbsoluteDeviationLongAggregator.combineStates(state, groupId, inState, position); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = MedianAbsoluteDeviationLongAggregator.evaluateFinal(state, selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MinDoubleAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MinDoubleAggregatorFunction.java deleted file mode 100644 index 7d7544e5d8470..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MinDoubleAggregatorFunction.java +++ /dev/null @@ -1,138 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunction} implementation for {@link MinDoubleAggregator}. - * This class is generated. Do not edit it. - */ -public final class MinDoubleAggregatorFunction implements AggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("min", ElementType.DOUBLE), - new IntermediateStateDesc("seen", ElementType.BOOLEAN) ); - - private final DriverContext driverContext; - - private final DoubleState state; - - private final List channels; - - public MinDoubleAggregatorFunction(DriverContext driverContext, List channels, - DoubleState state) { - this.driverContext = driverContext; - this.channels = channels; - this.state = state; - } - - public static MinDoubleAggregatorFunction create(DriverContext driverContext, - List channels) { - return new MinDoubleAggregatorFunction(driverContext, channels, new DoubleState(MinDoubleAggregator.init())); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public void addRawInput(Page page) { - DoubleBlock block = page.getBlock(channels.get(0)); - DoubleVector vector = block.asVector(); - if (vector != null) { - addRawVector(vector); - } else { - addRawBlock(block); - } - } - - private void addRawVector(DoubleVector vector) { - state.seen(true); - for (int i = 0; i < vector.getPositionCount(); i++) { - state.doubleValue(MinDoubleAggregator.combine(state.doubleValue(), vector.getDouble(i))); - } - } - - private void addRawBlock(DoubleBlock block) { - for (int p = 0; p < block.getPositionCount(); p++) { - if (block.isNull(p)) { - continue; - } - state.seen(true); - int start = block.getFirstValueIndex(p); - int end = start + block.getValueCount(p); - for (int i = start; i < end; i++) { - state.doubleValue(MinDoubleAggregator.combine(state.doubleValue(), block.getDouble(i))); - } - } - } - - @Override - public void addIntermediateInput(Page page) { - assert channels.size() == intermediateBlockCount(); - assert page.getBlockCount() >= channels.get(0) + intermediateStateDesc().size(); - Block minUncast = page.getBlock(channels.get(0)); - if (minUncast.areAllValuesNull()) { - return; - } - DoubleVector min = ((DoubleBlock) minUncast).asVector(); - assert min.getPositionCount() == 1; - Block seenUncast = page.getBlock(channels.get(1)); - if (seenUncast.areAllValuesNull()) { - return; - } - BooleanVector seen = ((BooleanBlock) seenUncast).asVector(); - assert seen.getPositionCount() == 1; - if (seen.getBoolean(0)) { - state.doubleValue(MinDoubleAggregator.combine(state.doubleValue(), min.getDouble(0))); - state.seen(true); - } - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, DriverContext driverContext) { - state.toIntermediate(blocks, offset, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, DriverContext driverContext) { - if (state.seen() == false) { - blocks[offset] = driverContext.blockFactory().newConstantNullBlock(1); - return; - } - blocks[offset] = driverContext.blockFactory().newConstantDoubleBlockWith(state.doubleValue(), 1); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MinDoubleAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MinDoubleAggregatorFunctionSupplier.java deleted file mode 100644 index f91cdfe54c89e..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MinDoubleAggregatorFunctionSupplier.java +++ /dev/null @@ -1,38 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link MinDoubleAggregator}. - * This class is generated. Do not edit it. - */ -public final class MinDoubleAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - public MinDoubleAggregatorFunctionSupplier(List channels) { - this.channels = channels; - } - - @Override - public MinDoubleAggregatorFunction aggregator(DriverContext driverContext) { - return MinDoubleAggregatorFunction.create(driverContext, channels); - } - - @Override - public MinDoubleGroupingAggregatorFunction groupingAggregator(DriverContext driverContext) { - return MinDoubleGroupingAggregatorFunction.create(channels, driverContext); - } - - @Override - public String describe() { - return "min of doubles"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MinDoubleGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MinDoubleGroupingAggregatorFunction.java deleted file mode 100644 index 7d0374b3d21f7..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MinDoubleGroupingAggregatorFunction.java +++ /dev/null @@ -1,208 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link MinDoubleAggregator}. - * This class is generated. Do not edit it. - */ -public final class MinDoubleGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("min", ElementType.DOUBLE), - new IntermediateStateDesc("seen", ElementType.BOOLEAN) ); - - private final DoubleArrayState state; - - private final List channels; - - private final DriverContext driverContext; - - public MinDoubleGroupingAggregatorFunction(List channels, DoubleArrayState state, - DriverContext driverContext) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - } - - public static MinDoubleGroupingAggregatorFunction create(List channels, - DriverContext driverContext) { - return new MinDoubleGroupingAggregatorFunction(channels, new DoubleArrayState(driverContext.bigArrays(), MinDoubleAggregator.init()), driverContext); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - DoubleBlock valuesBlock = page.getBlock(channels.get(0)); - DoubleVector valuesVector = valuesBlock.asVector(); - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, DoubleBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - state.set(groupId, MinDoubleAggregator.combine(state.getOrDefault(groupId), values.getDouble(v))); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, DoubleVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - state.set(groupId, MinDoubleAggregator.combine(state.getOrDefault(groupId), values.getDouble(groupPosition + positionOffset))); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, DoubleBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - state.set(groupId, MinDoubleAggregator.combine(state.getOrDefault(groupId), values.getDouble(v))); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, DoubleVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - state.set(groupId, MinDoubleAggregator.combine(state.getOrDefault(groupId), values.getDouble(groupPosition + positionOffset))); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block minUncast = page.getBlock(channels.get(0)); - if (minUncast.areAllValuesNull()) { - return; - } - DoubleVector min = ((DoubleBlock) minUncast).asVector(); - Block seenUncast = page.getBlock(channels.get(1)); - if (seenUncast.areAllValuesNull()) { - return; - } - BooleanVector seen = ((BooleanBlock) seenUncast).asVector(); - assert min.getPositionCount() == seen.getPositionCount(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (seen.getBoolean(groupPosition + positionOffset)) { - state.set(groupId, MinDoubleAggregator.combine(state.getOrDefault(groupId), min.getDouble(groupPosition + positionOffset))); - } - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - DoubleArrayState inState = ((MinDoubleGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - if (inState.hasValue(position)) { - state.set(groupId, MinDoubleAggregator.combine(state.getOrDefault(groupId), inState.get(position))); - } - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = state.toValuesBlock(selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MinIntAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MinIntAggregatorFunction.java deleted file mode 100644 index 0f2385cc120f9..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MinIntAggregatorFunction.java +++ /dev/null @@ -1,138 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunction} implementation for {@link MinIntAggregator}. - * This class is generated. Do not edit it. - */ -public final class MinIntAggregatorFunction implements AggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("min", ElementType.INT), - new IntermediateStateDesc("seen", ElementType.BOOLEAN) ); - - private final DriverContext driverContext; - - private final IntState state; - - private final List channels; - - public MinIntAggregatorFunction(DriverContext driverContext, List channels, - IntState state) { - this.driverContext = driverContext; - this.channels = channels; - this.state = state; - } - - public static MinIntAggregatorFunction create(DriverContext driverContext, - List channels) { - return new MinIntAggregatorFunction(driverContext, channels, new IntState(MinIntAggregator.init())); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public void addRawInput(Page page) { - IntBlock block = page.getBlock(channels.get(0)); - IntVector vector = block.asVector(); - if (vector != null) { - addRawVector(vector); - } else { - addRawBlock(block); - } - } - - private void addRawVector(IntVector vector) { - state.seen(true); - for (int i = 0; i < vector.getPositionCount(); i++) { - state.intValue(MinIntAggregator.combine(state.intValue(), vector.getInt(i))); - } - } - - private void addRawBlock(IntBlock block) { - for (int p = 0; p < block.getPositionCount(); p++) { - if (block.isNull(p)) { - continue; - } - state.seen(true); - int start = block.getFirstValueIndex(p); - int end = start + block.getValueCount(p); - for (int i = start; i < end; i++) { - state.intValue(MinIntAggregator.combine(state.intValue(), block.getInt(i))); - } - } - } - - @Override - public void addIntermediateInput(Page page) { - assert channels.size() == intermediateBlockCount(); - assert page.getBlockCount() >= channels.get(0) + intermediateStateDesc().size(); - Block minUncast = page.getBlock(channels.get(0)); - if (minUncast.areAllValuesNull()) { - return; - } - IntVector min = ((IntBlock) minUncast).asVector(); - assert min.getPositionCount() == 1; - Block seenUncast = page.getBlock(channels.get(1)); - if (seenUncast.areAllValuesNull()) { - return; - } - BooleanVector seen = ((BooleanBlock) seenUncast).asVector(); - assert seen.getPositionCount() == 1; - if (seen.getBoolean(0)) { - state.intValue(MinIntAggregator.combine(state.intValue(), min.getInt(0))); - state.seen(true); - } - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, DriverContext driverContext) { - state.toIntermediate(blocks, offset, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, DriverContext driverContext) { - if (state.seen() == false) { - blocks[offset] = driverContext.blockFactory().newConstantNullBlock(1); - return; - } - blocks[offset] = driverContext.blockFactory().newConstantIntBlockWith(state.intValue(), 1); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MinIntAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MinIntAggregatorFunctionSupplier.java deleted file mode 100644 index e09102a1a88ec..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MinIntAggregatorFunctionSupplier.java +++ /dev/null @@ -1,38 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link MinIntAggregator}. - * This class is generated. Do not edit it. - */ -public final class MinIntAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - public MinIntAggregatorFunctionSupplier(List channels) { - this.channels = channels; - } - - @Override - public MinIntAggregatorFunction aggregator(DriverContext driverContext) { - return MinIntAggregatorFunction.create(driverContext, channels); - } - - @Override - public MinIntGroupingAggregatorFunction groupingAggregator(DriverContext driverContext) { - return MinIntGroupingAggregatorFunction.create(channels, driverContext); - } - - @Override - public String describe() { - return "min of ints"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MinIntGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MinIntGroupingAggregatorFunction.java deleted file mode 100644 index 6625fd327237b..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MinIntGroupingAggregatorFunction.java +++ /dev/null @@ -1,206 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link MinIntAggregator}. - * This class is generated. Do not edit it. - */ -public final class MinIntGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("min", ElementType.INT), - new IntermediateStateDesc("seen", ElementType.BOOLEAN) ); - - private final IntArrayState state; - - private final List channels; - - private final DriverContext driverContext; - - public MinIntGroupingAggregatorFunction(List channels, IntArrayState state, - DriverContext driverContext) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - } - - public static MinIntGroupingAggregatorFunction create(List channels, - DriverContext driverContext) { - return new MinIntGroupingAggregatorFunction(channels, new IntArrayState(driverContext.bigArrays(), MinIntAggregator.init()), driverContext); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - IntBlock valuesBlock = page.getBlock(channels.get(0)); - IntVector valuesVector = valuesBlock.asVector(); - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, IntBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - state.set(groupId, MinIntAggregator.combine(state.getOrDefault(groupId), values.getInt(v))); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, IntVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - state.set(groupId, MinIntAggregator.combine(state.getOrDefault(groupId), values.getInt(groupPosition + positionOffset))); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, IntBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - state.set(groupId, MinIntAggregator.combine(state.getOrDefault(groupId), values.getInt(v))); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, IntVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - state.set(groupId, MinIntAggregator.combine(state.getOrDefault(groupId), values.getInt(groupPosition + positionOffset))); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block minUncast = page.getBlock(channels.get(0)); - if (minUncast.areAllValuesNull()) { - return; - } - IntVector min = ((IntBlock) minUncast).asVector(); - Block seenUncast = page.getBlock(channels.get(1)); - if (seenUncast.areAllValuesNull()) { - return; - } - BooleanVector seen = ((BooleanBlock) seenUncast).asVector(); - assert min.getPositionCount() == seen.getPositionCount(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (seen.getBoolean(groupPosition + positionOffset)) { - state.set(groupId, MinIntAggregator.combine(state.getOrDefault(groupId), min.getInt(groupPosition + positionOffset))); - } - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - IntArrayState inState = ((MinIntGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - if (inState.hasValue(position)) { - state.set(groupId, MinIntAggregator.combine(state.getOrDefault(groupId), inState.get(position))); - } - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = state.toValuesBlock(selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MinLongAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MinLongAggregatorFunction.java deleted file mode 100644 index 805729588158e..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MinLongAggregatorFunction.java +++ /dev/null @@ -1,138 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunction} implementation for {@link MinLongAggregator}. - * This class is generated. Do not edit it. - */ -public final class MinLongAggregatorFunction implements AggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("min", ElementType.LONG), - new IntermediateStateDesc("seen", ElementType.BOOLEAN) ); - - private final DriverContext driverContext; - - private final LongState state; - - private final List channels; - - public MinLongAggregatorFunction(DriverContext driverContext, List channels, - LongState state) { - this.driverContext = driverContext; - this.channels = channels; - this.state = state; - } - - public static MinLongAggregatorFunction create(DriverContext driverContext, - List channels) { - return new MinLongAggregatorFunction(driverContext, channels, new LongState(MinLongAggregator.init())); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public void addRawInput(Page page) { - LongBlock block = page.getBlock(channels.get(0)); - LongVector vector = block.asVector(); - if (vector != null) { - addRawVector(vector); - } else { - addRawBlock(block); - } - } - - private void addRawVector(LongVector vector) { - state.seen(true); - for (int i = 0; i < vector.getPositionCount(); i++) { - state.longValue(MinLongAggregator.combine(state.longValue(), vector.getLong(i))); - } - } - - private void addRawBlock(LongBlock block) { - for (int p = 0; p < block.getPositionCount(); p++) { - if (block.isNull(p)) { - continue; - } - state.seen(true); - int start = block.getFirstValueIndex(p); - int end = start + block.getValueCount(p); - for (int i = start; i < end; i++) { - state.longValue(MinLongAggregator.combine(state.longValue(), block.getLong(i))); - } - } - } - - @Override - public void addIntermediateInput(Page page) { - assert channels.size() == intermediateBlockCount(); - assert page.getBlockCount() >= channels.get(0) + intermediateStateDesc().size(); - Block minUncast = page.getBlock(channels.get(0)); - if (minUncast.areAllValuesNull()) { - return; - } - LongVector min = ((LongBlock) minUncast).asVector(); - assert min.getPositionCount() == 1; - Block seenUncast = page.getBlock(channels.get(1)); - if (seenUncast.areAllValuesNull()) { - return; - } - BooleanVector seen = ((BooleanBlock) seenUncast).asVector(); - assert seen.getPositionCount() == 1; - if (seen.getBoolean(0)) { - state.longValue(MinLongAggregator.combine(state.longValue(), min.getLong(0))); - state.seen(true); - } - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, DriverContext driverContext) { - state.toIntermediate(blocks, offset, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, DriverContext driverContext) { - if (state.seen() == false) { - blocks[offset] = driverContext.blockFactory().newConstantNullBlock(1); - return; - } - blocks[offset] = driverContext.blockFactory().newConstantLongBlockWith(state.longValue(), 1); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MinLongAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MinLongAggregatorFunctionSupplier.java deleted file mode 100644 index 85805767c9168..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MinLongAggregatorFunctionSupplier.java +++ /dev/null @@ -1,38 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link MinLongAggregator}. - * This class is generated. Do not edit it. - */ -public final class MinLongAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - public MinLongAggregatorFunctionSupplier(List channels) { - this.channels = channels; - } - - @Override - public MinLongAggregatorFunction aggregator(DriverContext driverContext) { - return MinLongAggregatorFunction.create(driverContext, channels); - } - - @Override - public MinLongGroupingAggregatorFunction groupingAggregator(DriverContext driverContext) { - return MinLongGroupingAggregatorFunction.create(channels, driverContext); - } - - @Override - public String describe() { - return "min of longs"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MinLongGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MinLongGroupingAggregatorFunction.java deleted file mode 100644 index f0c3727d7db0b..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/MinLongGroupingAggregatorFunction.java +++ /dev/null @@ -1,208 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link MinLongAggregator}. - * This class is generated. Do not edit it. - */ -public final class MinLongGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("min", ElementType.LONG), - new IntermediateStateDesc("seen", ElementType.BOOLEAN) ); - - private final LongArrayState state; - - private final List channels; - - private final DriverContext driverContext; - - public MinLongGroupingAggregatorFunction(List channels, LongArrayState state, - DriverContext driverContext) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - } - - public static MinLongGroupingAggregatorFunction create(List channels, - DriverContext driverContext) { - return new MinLongGroupingAggregatorFunction(channels, new LongArrayState(driverContext.bigArrays(), MinLongAggregator.init()), driverContext); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - LongBlock valuesBlock = page.getBlock(channels.get(0)); - LongVector valuesVector = valuesBlock.asVector(); - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, LongBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - state.set(groupId, MinLongAggregator.combine(state.getOrDefault(groupId), values.getLong(v))); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, LongVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - state.set(groupId, MinLongAggregator.combine(state.getOrDefault(groupId), values.getLong(groupPosition + positionOffset))); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, LongBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - state.set(groupId, MinLongAggregator.combine(state.getOrDefault(groupId), values.getLong(v))); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, LongVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - state.set(groupId, MinLongAggregator.combine(state.getOrDefault(groupId), values.getLong(groupPosition + positionOffset))); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block minUncast = page.getBlock(channels.get(0)); - if (minUncast.areAllValuesNull()) { - return; - } - LongVector min = ((LongBlock) minUncast).asVector(); - Block seenUncast = page.getBlock(channels.get(1)); - if (seenUncast.areAllValuesNull()) { - return; - } - BooleanVector seen = ((BooleanBlock) seenUncast).asVector(); - assert min.getPositionCount() == seen.getPositionCount(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (seen.getBoolean(groupPosition + positionOffset)) { - state.set(groupId, MinLongAggregator.combine(state.getOrDefault(groupId), min.getLong(groupPosition + positionOffset))); - } - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - LongArrayState inState = ((MinLongGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - if (inState.hasValue(position)) { - state.set(groupId, MinLongAggregator.combine(state.getOrDefault(groupId), inState.get(position))); - } - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = state.toValuesBlock(selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/PercentileDoubleAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/PercentileDoubleAggregatorFunction.java deleted file mode 100644 index cd7a5b5974442..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/PercentileDoubleAggregatorFunction.java +++ /dev/null @@ -1,127 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunction} implementation for {@link PercentileDoubleAggregator}. - * This class is generated. Do not edit it. - */ -public final class PercentileDoubleAggregatorFunction implements AggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("quart", ElementType.BYTES_REF) ); - - private final DriverContext driverContext; - - private final QuantileStates.SingleState state; - - private final List channels; - - private final double percentile; - - public PercentileDoubleAggregatorFunction(DriverContext driverContext, List channels, - QuantileStates.SingleState state, double percentile) { - this.driverContext = driverContext; - this.channels = channels; - this.state = state; - this.percentile = percentile; - } - - public static PercentileDoubleAggregatorFunction create(DriverContext driverContext, - List channels, double percentile) { - return new PercentileDoubleAggregatorFunction(driverContext, channels, PercentileDoubleAggregator.initSingle(percentile), percentile); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public void addRawInput(Page page) { - DoubleBlock block = page.getBlock(channels.get(0)); - DoubleVector vector = block.asVector(); - if (vector != null) { - addRawVector(vector); - } else { - addRawBlock(block); - } - } - - private void addRawVector(DoubleVector vector) { - for (int i = 0; i < vector.getPositionCount(); i++) { - PercentileDoubleAggregator.combine(state, vector.getDouble(i)); - } - } - - private void addRawBlock(DoubleBlock block) { - for (int p = 0; p < block.getPositionCount(); p++) { - if (block.isNull(p)) { - continue; - } - int start = block.getFirstValueIndex(p); - int end = start + block.getValueCount(p); - for (int i = start; i < end; i++) { - PercentileDoubleAggregator.combine(state, block.getDouble(i)); - } - } - } - - @Override - public void addIntermediateInput(Page page) { - assert channels.size() == intermediateBlockCount(); - assert page.getBlockCount() >= channels.get(0) + intermediateStateDesc().size(); - Block quartUncast = page.getBlock(channels.get(0)); - if (quartUncast.areAllValuesNull()) { - return; - } - BytesRefVector quart = ((BytesRefBlock) quartUncast).asVector(); - assert quart.getPositionCount() == 1; - BytesRef scratch = new BytesRef(); - PercentileDoubleAggregator.combineIntermediate(state, quart.getBytesRef(0, scratch)); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, DriverContext driverContext) { - state.toIntermediate(blocks, offset, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, DriverContext driverContext) { - blocks[offset] = PercentileDoubleAggregator.evaluateFinal(state, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/PercentileDoubleAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/PercentileDoubleAggregatorFunctionSupplier.java deleted file mode 100644 index 996cebd805aa8..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/PercentileDoubleAggregatorFunctionSupplier.java +++ /dev/null @@ -1,42 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link PercentileDoubleAggregator}. - * This class is generated. Do not edit it. - */ -public final class PercentileDoubleAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - private final double percentile; - - public PercentileDoubleAggregatorFunctionSupplier(List channels, double percentile) { - this.channels = channels; - this.percentile = percentile; - } - - @Override - public PercentileDoubleAggregatorFunction aggregator(DriverContext driverContext) { - return PercentileDoubleAggregatorFunction.create(driverContext, channels, percentile); - } - - @Override - public PercentileDoubleGroupingAggregatorFunction groupingAggregator( - DriverContext driverContext) { - return PercentileDoubleGroupingAggregatorFunction.create(channels, driverContext, percentile); - } - - @Override - public String describe() { - return "percentile of doubles"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/PercentileDoubleGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/PercentileDoubleGroupingAggregatorFunction.java deleted file mode 100644 index 9d486b9614dab..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/PercentileDoubleGroupingAggregatorFunction.java +++ /dev/null @@ -1,202 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link PercentileDoubleAggregator}. - * This class is generated. Do not edit it. - */ -public final class PercentileDoubleGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("quart", ElementType.BYTES_REF) ); - - private final QuantileStates.GroupingState state; - - private final List channels; - - private final DriverContext driverContext; - - private final double percentile; - - public PercentileDoubleGroupingAggregatorFunction(List channels, - QuantileStates.GroupingState state, DriverContext driverContext, double percentile) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - this.percentile = percentile; - } - - public static PercentileDoubleGroupingAggregatorFunction create(List channels, - DriverContext driverContext, double percentile) { - return new PercentileDoubleGroupingAggregatorFunction(channels, PercentileDoubleAggregator.initGrouping(driverContext.bigArrays(), percentile), driverContext, percentile); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - DoubleBlock valuesBlock = page.getBlock(channels.get(0)); - DoubleVector valuesVector = valuesBlock.asVector(); - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, DoubleBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - PercentileDoubleAggregator.combine(state, groupId, values.getDouble(v)); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, DoubleVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - PercentileDoubleAggregator.combine(state, groupId, values.getDouble(groupPosition + positionOffset)); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, DoubleBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - PercentileDoubleAggregator.combine(state, groupId, values.getDouble(v)); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, DoubleVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - PercentileDoubleAggregator.combine(state, groupId, values.getDouble(groupPosition + positionOffset)); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block quartUncast = page.getBlock(channels.get(0)); - if (quartUncast.areAllValuesNull()) { - return; - } - BytesRefVector quart = ((BytesRefBlock) quartUncast).asVector(); - BytesRef scratch = new BytesRef(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - PercentileDoubleAggregator.combineIntermediate(state, groupId, quart.getBytesRef(groupPosition + positionOffset, scratch)); - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - QuantileStates.GroupingState inState = ((PercentileDoubleGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - PercentileDoubleAggregator.combineStates(state, groupId, inState, position); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = PercentileDoubleAggregator.evaluateFinal(state, selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/PercentileIntAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/PercentileIntAggregatorFunction.java deleted file mode 100644 index b9b1c2e90b768..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/PercentileIntAggregatorFunction.java +++ /dev/null @@ -1,127 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunction} implementation for {@link PercentileIntAggregator}. - * This class is generated. Do not edit it. - */ -public final class PercentileIntAggregatorFunction implements AggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("quart", ElementType.BYTES_REF) ); - - private final DriverContext driverContext; - - private final QuantileStates.SingleState state; - - private final List channels; - - private final double percentile; - - public PercentileIntAggregatorFunction(DriverContext driverContext, List channels, - QuantileStates.SingleState state, double percentile) { - this.driverContext = driverContext; - this.channels = channels; - this.state = state; - this.percentile = percentile; - } - - public static PercentileIntAggregatorFunction create(DriverContext driverContext, - List channels, double percentile) { - return new PercentileIntAggregatorFunction(driverContext, channels, PercentileIntAggregator.initSingle(percentile), percentile); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public void addRawInput(Page page) { - IntBlock block = page.getBlock(channels.get(0)); - IntVector vector = block.asVector(); - if (vector != null) { - addRawVector(vector); - } else { - addRawBlock(block); - } - } - - private void addRawVector(IntVector vector) { - for (int i = 0; i < vector.getPositionCount(); i++) { - PercentileIntAggregator.combine(state, vector.getInt(i)); - } - } - - private void addRawBlock(IntBlock block) { - for (int p = 0; p < block.getPositionCount(); p++) { - if (block.isNull(p)) { - continue; - } - int start = block.getFirstValueIndex(p); - int end = start + block.getValueCount(p); - for (int i = start; i < end; i++) { - PercentileIntAggregator.combine(state, block.getInt(i)); - } - } - } - - @Override - public void addIntermediateInput(Page page) { - assert channels.size() == intermediateBlockCount(); - assert page.getBlockCount() >= channels.get(0) + intermediateStateDesc().size(); - Block quartUncast = page.getBlock(channels.get(0)); - if (quartUncast.areAllValuesNull()) { - return; - } - BytesRefVector quart = ((BytesRefBlock) quartUncast).asVector(); - assert quart.getPositionCount() == 1; - BytesRef scratch = new BytesRef(); - PercentileIntAggregator.combineIntermediate(state, quart.getBytesRef(0, scratch)); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, DriverContext driverContext) { - state.toIntermediate(blocks, offset, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, DriverContext driverContext) { - blocks[offset] = PercentileIntAggregator.evaluateFinal(state, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/PercentileIntAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/PercentileIntAggregatorFunctionSupplier.java deleted file mode 100644 index 7ce62dd7b600b..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/PercentileIntAggregatorFunctionSupplier.java +++ /dev/null @@ -1,41 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link PercentileIntAggregator}. - * This class is generated. Do not edit it. - */ -public final class PercentileIntAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - private final double percentile; - - public PercentileIntAggregatorFunctionSupplier(List channels, double percentile) { - this.channels = channels; - this.percentile = percentile; - } - - @Override - public PercentileIntAggregatorFunction aggregator(DriverContext driverContext) { - return PercentileIntAggregatorFunction.create(driverContext, channels, percentile); - } - - @Override - public PercentileIntGroupingAggregatorFunction groupingAggregator(DriverContext driverContext) { - return PercentileIntGroupingAggregatorFunction.create(channels, driverContext, percentile); - } - - @Override - public String describe() { - return "percentile of ints"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/PercentileIntGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/PercentileIntGroupingAggregatorFunction.java deleted file mode 100644 index 8c2bd7091143f..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/PercentileIntGroupingAggregatorFunction.java +++ /dev/null @@ -1,200 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link PercentileIntAggregator}. - * This class is generated. Do not edit it. - */ -public final class PercentileIntGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("quart", ElementType.BYTES_REF) ); - - private final QuantileStates.GroupingState state; - - private final List channels; - - private final DriverContext driverContext; - - private final double percentile; - - public PercentileIntGroupingAggregatorFunction(List channels, - QuantileStates.GroupingState state, DriverContext driverContext, double percentile) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - this.percentile = percentile; - } - - public static PercentileIntGroupingAggregatorFunction create(List channels, - DriverContext driverContext, double percentile) { - return new PercentileIntGroupingAggregatorFunction(channels, PercentileIntAggregator.initGrouping(driverContext.bigArrays(), percentile), driverContext, percentile); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - IntBlock valuesBlock = page.getBlock(channels.get(0)); - IntVector valuesVector = valuesBlock.asVector(); - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, IntBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - PercentileIntAggregator.combine(state, groupId, values.getInt(v)); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, IntVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - PercentileIntAggregator.combine(state, groupId, values.getInt(groupPosition + positionOffset)); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, IntBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - PercentileIntAggregator.combine(state, groupId, values.getInt(v)); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, IntVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - PercentileIntAggregator.combine(state, groupId, values.getInt(groupPosition + positionOffset)); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block quartUncast = page.getBlock(channels.get(0)); - if (quartUncast.areAllValuesNull()) { - return; - } - BytesRefVector quart = ((BytesRefBlock) quartUncast).asVector(); - BytesRef scratch = new BytesRef(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - PercentileIntAggregator.combineIntermediate(state, groupId, quart.getBytesRef(groupPosition + positionOffset, scratch)); - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - QuantileStates.GroupingState inState = ((PercentileIntGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - PercentileIntAggregator.combineStates(state, groupId, inState, position); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = PercentileIntAggregator.evaluateFinal(state, selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/PercentileLongAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/PercentileLongAggregatorFunction.java deleted file mode 100644 index cc785ce55bb55..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/PercentileLongAggregatorFunction.java +++ /dev/null @@ -1,127 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunction} implementation for {@link PercentileLongAggregator}. - * This class is generated. Do not edit it. - */ -public final class PercentileLongAggregatorFunction implements AggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("quart", ElementType.BYTES_REF) ); - - private final DriverContext driverContext; - - private final QuantileStates.SingleState state; - - private final List channels; - - private final double percentile; - - public PercentileLongAggregatorFunction(DriverContext driverContext, List channels, - QuantileStates.SingleState state, double percentile) { - this.driverContext = driverContext; - this.channels = channels; - this.state = state; - this.percentile = percentile; - } - - public static PercentileLongAggregatorFunction create(DriverContext driverContext, - List channels, double percentile) { - return new PercentileLongAggregatorFunction(driverContext, channels, PercentileLongAggregator.initSingle(percentile), percentile); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public void addRawInput(Page page) { - LongBlock block = page.getBlock(channels.get(0)); - LongVector vector = block.asVector(); - if (vector != null) { - addRawVector(vector); - } else { - addRawBlock(block); - } - } - - private void addRawVector(LongVector vector) { - for (int i = 0; i < vector.getPositionCount(); i++) { - PercentileLongAggregator.combine(state, vector.getLong(i)); - } - } - - private void addRawBlock(LongBlock block) { - for (int p = 0; p < block.getPositionCount(); p++) { - if (block.isNull(p)) { - continue; - } - int start = block.getFirstValueIndex(p); - int end = start + block.getValueCount(p); - for (int i = start; i < end; i++) { - PercentileLongAggregator.combine(state, block.getLong(i)); - } - } - } - - @Override - public void addIntermediateInput(Page page) { - assert channels.size() == intermediateBlockCount(); - assert page.getBlockCount() >= channels.get(0) + intermediateStateDesc().size(); - Block quartUncast = page.getBlock(channels.get(0)); - if (quartUncast.areAllValuesNull()) { - return; - } - BytesRefVector quart = ((BytesRefBlock) quartUncast).asVector(); - assert quart.getPositionCount() == 1; - BytesRef scratch = new BytesRef(); - PercentileLongAggregator.combineIntermediate(state, quart.getBytesRef(0, scratch)); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, DriverContext driverContext) { - state.toIntermediate(blocks, offset, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, DriverContext driverContext) { - blocks[offset] = PercentileLongAggregator.evaluateFinal(state, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/PercentileLongAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/PercentileLongAggregatorFunctionSupplier.java deleted file mode 100644 index 7e32bfc9d9937..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/PercentileLongAggregatorFunctionSupplier.java +++ /dev/null @@ -1,41 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link PercentileLongAggregator}. - * This class is generated. Do not edit it. - */ -public final class PercentileLongAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - private final double percentile; - - public PercentileLongAggregatorFunctionSupplier(List channels, double percentile) { - this.channels = channels; - this.percentile = percentile; - } - - @Override - public PercentileLongAggregatorFunction aggregator(DriverContext driverContext) { - return PercentileLongAggregatorFunction.create(driverContext, channels, percentile); - } - - @Override - public PercentileLongGroupingAggregatorFunction groupingAggregator(DriverContext driverContext) { - return PercentileLongGroupingAggregatorFunction.create(channels, driverContext, percentile); - } - - @Override - public String describe() { - return "percentile of longs"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/PercentileLongGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/PercentileLongGroupingAggregatorFunction.java deleted file mode 100644 index c1c332ba0094d..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/PercentileLongGroupingAggregatorFunction.java +++ /dev/null @@ -1,202 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link PercentileLongAggregator}. - * This class is generated. Do not edit it. - */ -public final class PercentileLongGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("quart", ElementType.BYTES_REF) ); - - private final QuantileStates.GroupingState state; - - private final List channels; - - private final DriverContext driverContext; - - private final double percentile; - - public PercentileLongGroupingAggregatorFunction(List channels, - QuantileStates.GroupingState state, DriverContext driverContext, double percentile) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - this.percentile = percentile; - } - - public static PercentileLongGroupingAggregatorFunction create(List channels, - DriverContext driverContext, double percentile) { - return new PercentileLongGroupingAggregatorFunction(channels, PercentileLongAggregator.initGrouping(driverContext.bigArrays(), percentile), driverContext, percentile); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - LongBlock valuesBlock = page.getBlock(channels.get(0)); - LongVector valuesVector = valuesBlock.asVector(); - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, LongBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - PercentileLongAggregator.combine(state, groupId, values.getLong(v)); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, LongVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - PercentileLongAggregator.combine(state, groupId, values.getLong(groupPosition + positionOffset)); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, LongBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - PercentileLongAggregator.combine(state, groupId, values.getLong(v)); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, LongVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - PercentileLongAggregator.combine(state, groupId, values.getLong(groupPosition + positionOffset)); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block quartUncast = page.getBlock(channels.get(0)); - if (quartUncast.areAllValuesNull()) { - return; - } - BytesRefVector quart = ((BytesRefBlock) quartUncast).asVector(); - BytesRef scratch = new BytesRef(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - PercentileLongAggregator.combineIntermediate(state, groupId, quart.getBytesRef(groupPosition + positionOffset, scratch)); - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - QuantileStates.GroupingState inState = ((PercentileLongGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - PercentileLongAggregator.combineStates(state, groupId, inState, position); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = PercentileLongAggregator.evaluateFinal(state, selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/RateDoubleAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/RateDoubleAggregatorFunctionSupplier.java deleted file mode 100644 index 8806e1ed865c2..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/RateDoubleAggregatorFunctionSupplier.java +++ /dev/null @@ -1,41 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link RateDoubleAggregator}. - * This class is generated. Do not edit it. - */ -public final class RateDoubleAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - private final long unitInMillis; - - public RateDoubleAggregatorFunctionSupplier(List channels, long unitInMillis) { - this.channels = channels; - this.unitInMillis = unitInMillis; - } - - @Override - public AggregatorFunction aggregator(DriverContext driverContext) { - throw new UnsupportedOperationException("non-grouping aggregator is not supported"); - } - - @Override - public RateDoubleGroupingAggregatorFunction groupingAggregator(DriverContext driverContext) { - return RateDoubleGroupingAggregatorFunction.create(channels, driverContext, unitInMillis); - } - - @Override - public String describe() { - return "rate of doubles"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/RateDoubleGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/RateDoubleGroupingAggregatorFunction.java deleted file mode 100644 index 8d9e011891e95..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/RateDoubleGroupingAggregatorFunction.java +++ /dev/null @@ -1,225 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link RateDoubleAggregator}. - * This class is generated. Do not edit it. - */ -public final class RateDoubleGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("timestamps", ElementType.LONG), - new IntermediateStateDesc("values", ElementType.DOUBLE), - new IntermediateStateDesc("resets", ElementType.DOUBLE) ); - - private final RateDoubleAggregator.DoubleRateGroupingState state; - - private final List channels; - - private final DriverContext driverContext; - - private final long unitInMillis; - - public RateDoubleGroupingAggregatorFunction(List channels, - RateDoubleAggregator.DoubleRateGroupingState state, DriverContext driverContext, - long unitInMillis) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - this.unitInMillis = unitInMillis; - } - - public static RateDoubleGroupingAggregatorFunction create(List channels, - DriverContext driverContext, long unitInMillis) { - return new RateDoubleGroupingAggregatorFunction(channels, RateDoubleAggregator.initGrouping(driverContext, unitInMillis), driverContext, unitInMillis); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - DoubleBlock valuesBlock = page.getBlock(channels.get(0)); - DoubleVector valuesVector = valuesBlock.asVector(); - LongBlock timestampsBlock = page.getBlock(channels.get(1)); - LongVector timestampsVector = timestampsBlock.asVector(); - if (timestampsVector == null) { - throw new IllegalStateException("expected @timestamp vector; but got a block"); - } - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock, timestampsVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock, timestampsVector); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector, timestampsVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector, timestampsVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, DoubleBlock values, - LongVector timestamps) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - RateDoubleAggregator.combine(state, groupId, timestamps.getLong(v), values.getDouble(v)); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, DoubleVector values, - LongVector timestamps) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - var valuePosition = groupPosition + positionOffset; - RateDoubleAggregator.combine(state, groupId, timestamps.getLong(valuePosition), values.getDouble(valuePosition)); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, DoubleBlock values, - LongVector timestamps) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - RateDoubleAggregator.combine(state, groupId, timestamps.getLong(v), values.getDouble(v)); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, DoubleVector values, - LongVector timestamps) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - var valuePosition = groupPosition + positionOffset; - RateDoubleAggregator.combine(state, groupId, timestamps.getLong(valuePosition), values.getDouble(valuePosition)); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block timestampsUncast = page.getBlock(channels.get(0)); - if (timestampsUncast.areAllValuesNull()) { - return; - } - LongBlock timestamps = (LongBlock) timestampsUncast; - Block valuesUncast = page.getBlock(channels.get(1)); - if (valuesUncast.areAllValuesNull()) { - return; - } - DoubleBlock values = (DoubleBlock) valuesUncast; - Block resetsUncast = page.getBlock(channels.get(2)); - if (resetsUncast.areAllValuesNull()) { - return; - } - DoubleVector resets = ((DoubleBlock) resetsUncast).asVector(); - assert timestamps.getPositionCount() == values.getPositionCount() && timestamps.getPositionCount() == resets.getPositionCount(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - RateDoubleAggregator.combineIntermediate(state, groupId, timestamps, values, resets.getDouble(groupPosition + positionOffset), groupPosition + positionOffset); - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - RateDoubleAggregator.DoubleRateGroupingState inState = ((RateDoubleGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - RateDoubleAggregator.combineStates(state, groupId, inState, position); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = RateDoubleAggregator.evaluateFinal(state, selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/RateIntAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/RateIntAggregatorFunctionSupplier.java deleted file mode 100644 index a98f0217ef90e..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/RateIntAggregatorFunctionSupplier.java +++ /dev/null @@ -1,41 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link RateIntAggregator}. - * This class is generated. Do not edit it. - */ -public final class RateIntAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - private final long unitInMillis; - - public RateIntAggregatorFunctionSupplier(List channels, long unitInMillis) { - this.channels = channels; - this.unitInMillis = unitInMillis; - } - - @Override - public AggregatorFunction aggregator(DriverContext driverContext) { - throw new UnsupportedOperationException("non-grouping aggregator is not supported"); - } - - @Override - public RateIntGroupingAggregatorFunction groupingAggregator(DriverContext driverContext) { - return RateIntGroupingAggregatorFunction.create(channels, driverContext, unitInMillis); - } - - @Override - public String describe() { - return "rate of ints"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/RateIntGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/RateIntGroupingAggregatorFunction.java deleted file mode 100644 index 6bd4b833dc9e6..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/RateIntGroupingAggregatorFunction.java +++ /dev/null @@ -1,225 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link RateIntAggregator}. - * This class is generated. Do not edit it. - */ -public final class RateIntGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("timestamps", ElementType.LONG), - new IntermediateStateDesc("values", ElementType.INT), - new IntermediateStateDesc("resets", ElementType.DOUBLE) ); - - private final RateIntAggregator.IntRateGroupingState state; - - private final List channels; - - private final DriverContext driverContext; - - private final long unitInMillis; - - public RateIntGroupingAggregatorFunction(List channels, - RateIntAggregator.IntRateGroupingState state, DriverContext driverContext, - long unitInMillis) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - this.unitInMillis = unitInMillis; - } - - public static RateIntGroupingAggregatorFunction create(List channels, - DriverContext driverContext, long unitInMillis) { - return new RateIntGroupingAggregatorFunction(channels, RateIntAggregator.initGrouping(driverContext, unitInMillis), driverContext, unitInMillis); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - IntBlock valuesBlock = page.getBlock(channels.get(0)); - IntVector valuesVector = valuesBlock.asVector(); - LongBlock timestampsBlock = page.getBlock(channels.get(1)); - LongVector timestampsVector = timestampsBlock.asVector(); - if (timestampsVector == null) { - throw new IllegalStateException("expected @timestamp vector; but got a block"); - } - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock, timestampsVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock, timestampsVector); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector, timestampsVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector, timestampsVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, IntBlock values, - LongVector timestamps) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - RateIntAggregator.combine(state, groupId, timestamps.getLong(v), values.getInt(v)); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, IntVector values, - LongVector timestamps) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - var valuePosition = groupPosition + positionOffset; - RateIntAggregator.combine(state, groupId, timestamps.getLong(valuePosition), values.getInt(valuePosition)); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, IntBlock values, - LongVector timestamps) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - RateIntAggregator.combine(state, groupId, timestamps.getLong(v), values.getInt(v)); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, IntVector values, - LongVector timestamps) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - var valuePosition = groupPosition + positionOffset; - RateIntAggregator.combine(state, groupId, timestamps.getLong(valuePosition), values.getInt(valuePosition)); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block timestampsUncast = page.getBlock(channels.get(0)); - if (timestampsUncast.areAllValuesNull()) { - return; - } - LongBlock timestamps = (LongBlock) timestampsUncast; - Block valuesUncast = page.getBlock(channels.get(1)); - if (valuesUncast.areAllValuesNull()) { - return; - } - IntBlock values = (IntBlock) valuesUncast; - Block resetsUncast = page.getBlock(channels.get(2)); - if (resetsUncast.areAllValuesNull()) { - return; - } - DoubleVector resets = ((DoubleBlock) resetsUncast).asVector(); - assert timestamps.getPositionCount() == values.getPositionCount() && timestamps.getPositionCount() == resets.getPositionCount(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - RateIntAggregator.combineIntermediate(state, groupId, timestamps, values, resets.getDouble(groupPosition + positionOffset), groupPosition + positionOffset); - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - RateIntAggregator.IntRateGroupingState inState = ((RateIntGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - RateIntAggregator.combineStates(state, groupId, inState, position); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = RateIntAggregator.evaluateFinal(state, selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/RateLongAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/RateLongAggregatorFunctionSupplier.java deleted file mode 100644 index b8100dbbe4455..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/RateLongAggregatorFunctionSupplier.java +++ /dev/null @@ -1,41 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link RateLongAggregator}. - * This class is generated. Do not edit it. - */ -public final class RateLongAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - private final long unitInMillis; - - public RateLongAggregatorFunctionSupplier(List channels, long unitInMillis) { - this.channels = channels; - this.unitInMillis = unitInMillis; - } - - @Override - public AggregatorFunction aggregator(DriverContext driverContext) { - throw new UnsupportedOperationException("non-grouping aggregator is not supported"); - } - - @Override - public RateLongGroupingAggregatorFunction groupingAggregator(DriverContext driverContext) { - return RateLongGroupingAggregatorFunction.create(channels, driverContext, unitInMillis); - } - - @Override - public String describe() { - return "rate of longs"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/RateLongGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/RateLongGroupingAggregatorFunction.java deleted file mode 100644 index 27318d6496737..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/RateLongGroupingAggregatorFunction.java +++ /dev/null @@ -1,225 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link RateLongAggregator}. - * This class is generated. Do not edit it. - */ -public final class RateLongGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("timestamps", ElementType.LONG), - new IntermediateStateDesc("values", ElementType.LONG), - new IntermediateStateDesc("resets", ElementType.DOUBLE) ); - - private final RateLongAggregator.LongRateGroupingState state; - - private final List channels; - - private final DriverContext driverContext; - - private final long unitInMillis; - - public RateLongGroupingAggregatorFunction(List channels, - RateLongAggregator.LongRateGroupingState state, DriverContext driverContext, - long unitInMillis) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - this.unitInMillis = unitInMillis; - } - - public static RateLongGroupingAggregatorFunction create(List channels, - DriverContext driverContext, long unitInMillis) { - return new RateLongGroupingAggregatorFunction(channels, RateLongAggregator.initGrouping(driverContext, unitInMillis), driverContext, unitInMillis); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - LongBlock valuesBlock = page.getBlock(channels.get(0)); - LongVector valuesVector = valuesBlock.asVector(); - LongBlock timestampsBlock = page.getBlock(channels.get(1)); - LongVector timestampsVector = timestampsBlock.asVector(); - if (timestampsVector == null) { - throw new IllegalStateException("expected @timestamp vector; but got a block"); - } - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock, timestampsVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock, timestampsVector); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector, timestampsVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector, timestampsVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, LongBlock values, - LongVector timestamps) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - RateLongAggregator.combine(state, groupId, timestamps.getLong(v), values.getLong(v)); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, LongVector values, - LongVector timestamps) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - var valuePosition = groupPosition + positionOffset; - RateLongAggregator.combine(state, groupId, timestamps.getLong(valuePosition), values.getLong(valuePosition)); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, LongBlock values, - LongVector timestamps) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - RateLongAggregator.combine(state, groupId, timestamps.getLong(v), values.getLong(v)); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, LongVector values, - LongVector timestamps) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - var valuePosition = groupPosition + positionOffset; - RateLongAggregator.combine(state, groupId, timestamps.getLong(valuePosition), values.getLong(valuePosition)); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block timestampsUncast = page.getBlock(channels.get(0)); - if (timestampsUncast.areAllValuesNull()) { - return; - } - LongBlock timestamps = (LongBlock) timestampsUncast; - Block valuesUncast = page.getBlock(channels.get(1)); - if (valuesUncast.areAllValuesNull()) { - return; - } - LongBlock values = (LongBlock) valuesUncast; - Block resetsUncast = page.getBlock(channels.get(2)); - if (resetsUncast.areAllValuesNull()) { - return; - } - DoubleVector resets = ((DoubleBlock) resetsUncast).asVector(); - assert timestamps.getPositionCount() == values.getPositionCount() && timestamps.getPositionCount() == resets.getPositionCount(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - RateLongAggregator.combineIntermediate(state, groupId, timestamps, values, resets.getDouble(groupPosition + positionOffset), groupPosition + positionOffset); - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - RateLongAggregator.LongRateGroupingState inState = ((RateLongGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - RateLongAggregator.combineStates(state, groupId, inState, position); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = RateLongAggregator.evaluateFinal(state, selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/SumDoubleAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/SumDoubleAggregatorFunction.java deleted file mode 100644 index 354726f82b8f3..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/SumDoubleAggregatorFunction.java +++ /dev/null @@ -1,142 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunction} implementation for {@link SumDoubleAggregator}. - * This class is generated. Do not edit it. - */ -public final class SumDoubleAggregatorFunction implements AggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("value", ElementType.DOUBLE), - new IntermediateStateDesc("delta", ElementType.DOUBLE), - new IntermediateStateDesc("seen", ElementType.BOOLEAN) ); - - private final DriverContext driverContext; - - private final SumDoubleAggregator.SumState state; - - private final List channels; - - public SumDoubleAggregatorFunction(DriverContext driverContext, List channels, - SumDoubleAggregator.SumState state) { - this.driverContext = driverContext; - this.channels = channels; - this.state = state; - } - - public static SumDoubleAggregatorFunction create(DriverContext driverContext, - List channels) { - return new SumDoubleAggregatorFunction(driverContext, channels, SumDoubleAggregator.initSingle()); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public void addRawInput(Page page) { - DoubleBlock block = page.getBlock(channels.get(0)); - DoubleVector vector = block.asVector(); - if (vector != null) { - addRawVector(vector); - } else { - addRawBlock(block); - } - } - - private void addRawVector(DoubleVector vector) { - state.seen(true); - for (int i = 0; i < vector.getPositionCount(); i++) { - SumDoubleAggregator.combine(state, vector.getDouble(i)); - } - } - - private void addRawBlock(DoubleBlock block) { - for (int p = 0; p < block.getPositionCount(); p++) { - if (block.isNull(p)) { - continue; - } - state.seen(true); - int start = block.getFirstValueIndex(p); - int end = start + block.getValueCount(p); - for (int i = start; i < end; i++) { - SumDoubleAggregator.combine(state, block.getDouble(i)); - } - } - } - - @Override - public void addIntermediateInput(Page page) { - assert channels.size() == intermediateBlockCount(); - assert page.getBlockCount() >= channels.get(0) + intermediateStateDesc().size(); - Block valueUncast = page.getBlock(channels.get(0)); - if (valueUncast.areAllValuesNull()) { - return; - } - DoubleVector value = ((DoubleBlock) valueUncast).asVector(); - assert value.getPositionCount() == 1; - Block deltaUncast = page.getBlock(channels.get(1)); - if (deltaUncast.areAllValuesNull()) { - return; - } - DoubleVector delta = ((DoubleBlock) deltaUncast).asVector(); - assert delta.getPositionCount() == 1; - Block seenUncast = page.getBlock(channels.get(2)); - if (seenUncast.areAllValuesNull()) { - return; - } - BooleanVector seen = ((BooleanBlock) seenUncast).asVector(); - assert seen.getPositionCount() == 1; - SumDoubleAggregator.combineIntermediate(state, value.getDouble(0), delta.getDouble(0), seen.getBoolean(0)); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, DriverContext driverContext) { - state.toIntermediate(blocks, offset, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, DriverContext driverContext) { - if (state.seen() == false) { - blocks[offset] = driverContext.blockFactory().newConstantNullBlock(1); - return; - } - blocks[offset] = SumDoubleAggregator.evaluateFinal(state, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/SumDoubleAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/SumDoubleAggregatorFunctionSupplier.java deleted file mode 100644 index b68bed30013c6..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/SumDoubleAggregatorFunctionSupplier.java +++ /dev/null @@ -1,38 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link SumDoubleAggregator}. - * This class is generated. Do not edit it. - */ -public final class SumDoubleAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - public SumDoubleAggregatorFunctionSupplier(List channels) { - this.channels = channels; - } - - @Override - public SumDoubleAggregatorFunction aggregator(DriverContext driverContext) { - return SumDoubleAggregatorFunction.create(driverContext, channels); - } - - @Override - public SumDoubleGroupingAggregatorFunction groupingAggregator(DriverContext driverContext) { - return SumDoubleGroupingAggregatorFunction.create(channels, driverContext); - } - - @Override - public String describe() { - return "sum of doubles"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/SumDoubleGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/SumDoubleGroupingAggregatorFunction.java deleted file mode 100644 index 5085cfc3bebcf..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/SumDoubleGroupingAggregatorFunction.java +++ /dev/null @@ -1,210 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link SumDoubleAggregator}. - * This class is generated. Do not edit it. - */ -public final class SumDoubleGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("value", ElementType.DOUBLE), - new IntermediateStateDesc("delta", ElementType.DOUBLE), - new IntermediateStateDesc("seen", ElementType.BOOLEAN) ); - - private final SumDoubleAggregator.GroupingSumState state; - - private final List channels; - - private final DriverContext driverContext; - - public SumDoubleGroupingAggregatorFunction(List channels, - SumDoubleAggregator.GroupingSumState state, DriverContext driverContext) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - } - - public static SumDoubleGroupingAggregatorFunction create(List channels, - DriverContext driverContext) { - return new SumDoubleGroupingAggregatorFunction(channels, SumDoubleAggregator.initGrouping(driverContext.bigArrays()), driverContext); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - DoubleBlock valuesBlock = page.getBlock(channels.get(0)); - DoubleVector valuesVector = valuesBlock.asVector(); - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, DoubleBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - SumDoubleAggregator.combine(state, groupId, values.getDouble(v)); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, DoubleVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - SumDoubleAggregator.combine(state, groupId, values.getDouble(groupPosition + positionOffset)); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, DoubleBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - SumDoubleAggregator.combine(state, groupId, values.getDouble(v)); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, DoubleVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - SumDoubleAggregator.combine(state, groupId, values.getDouble(groupPosition + positionOffset)); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block valueUncast = page.getBlock(channels.get(0)); - if (valueUncast.areAllValuesNull()) { - return; - } - DoubleVector value = ((DoubleBlock) valueUncast).asVector(); - Block deltaUncast = page.getBlock(channels.get(1)); - if (deltaUncast.areAllValuesNull()) { - return; - } - DoubleVector delta = ((DoubleBlock) deltaUncast).asVector(); - Block seenUncast = page.getBlock(channels.get(2)); - if (seenUncast.areAllValuesNull()) { - return; - } - BooleanVector seen = ((BooleanBlock) seenUncast).asVector(); - assert value.getPositionCount() == delta.getPositionCount() && value.getPositionCount() == seen.getPositionCount(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - SumDoubleAggregator.combineIntermediate(state, groupId, value.getDouble(groupPosition + positionOffset), delta.getDouble(groupPosition + positionOffset), seen.getBoolean(groupPosition + positionOffset)); - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - SumDoubleAggregator.GroupingSumState inState = ((SumDoubleGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - SumDoubleAggregator.combineStates(state, groupId, inState, position); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = SumDoubleAggregator.evaluateFinal(state, selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/SumIntAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/SumIntAggregatorFunction.java deleted file mode 100644 index e210429991aa6..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/SumIntAggregatorFunction.java +++ /dev/null @@ -1,140 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunction} implementation for {@link SumIntAggregator}. - * This class is generated. Do not edit it. - */ -public final class SumIntAggregatorFunction implements AggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("sum", ElementType.LONG), - new IntermediateStateDesc("seen", ElementType.BOOLEAN) ); - - private final DriverContext driverContext; - - private final LongState state; - - private final List channels; - - public SumIntAggregatorFunction(DriverContext driverContext, List channels, - LongState state) { - this.driverContext = driverContext; - this.channels = channels; - this.state = state; - } - - public static SumIntAggregatorFunction create(DriverContext driverContext, - List channels) { - return new SumIntAggregatorFunction(driverContext, channels, new LongState(SumIntAggregator.init())); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public void addRawInput(Page page) { - IntBlock block = page.getBlock(channels.get(0)); - IntVector vector = block.asVector(); - if (vector != null) { - addRawVector(vector); - } else { - addRawBlock(block); - } - } - - private void addRawVector(IntVector vector) { - state.seen(true); - for (int i = 0; i < vector.getPositionCount(); i++) { - state.longValue(SumIntAggregator.combine(state.longValue(), vector.getInt(i))); - } - } - - private void addRawBlock(IntBlock block) { - for (int p = 0; p < block.getPositionCount(); p++) { - if (block.isNull(p)) { - continue; - } - state.seen(true); - int start = block.getFirstValueIndex(p); - int end = start + block.getValueCount(p); - for (int i = start; i < end; i++) { - state.longValue(SumIntAggregator.combine(state.longValue(), block.getInt(i))); - } - } - } - - @Override - public void addIntermediateInput(Page page) { - assert channels.size() == intermediateBlockCount(); - assert page.getBlockCount() >= channels.get(0) + intermediateStateDesc().size(); - Block sumUncast = page.getBlock(channels.get(0)); - if (sumUncast.areAllValuesNull()) { - return; - } - LongVector sum = ((LongBlock) sumUncast).asVector(); - assert sum.getPositionCount() == 1; - Block seenUncast = page.getBlock(channels.get(1)); - if (seenUncast.areAllValuesNull()) { - return; - } - BooleanVector seen = ((BooleanBlock) seenUncast).asVector(); - assert seen.getPositionCount() == 1; - if (seen.getBoolean(0)) { - state.longValue(SumIntAggregator.combine(state.longValue(), sum.getLong(0))); - state.seen(true); - } - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, DriverContext driverContext) { - state.toIntermediate(blocks, offset, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, DriverContext driverContext) { - if (state.seen() == false) { - blocks[offset] = driverContext.blockFactory().newConstantNullBlock(1); - return; - } - blocks[offset] = driverContext.blockFactory().newConstantLongBlockWith(state.longValue(), 1); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/SumIntAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/SumIntAggregatorFunctionSupplier.java deleted file mode 100644 index dcb48944dc557..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/SumIntAggregatorFunctionSupplier.java +++ /dev/null @@ -1,38 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link SumIntAggregator}. - * This class is generated. Do not edit it. - */ -public final class SumIntAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - public SumIntAggregatorFunctionSupplier(List channels) { - this.channels = channels; - } - - @Override - public SumIntAggregatorFunction aggregator(DriverContext driverContext) { - return SumIntAggregatorFunction.create(driverContext, channels); - } - - @Override - public SumIntGroupingAggregatorFunction groupingAggregator(DriverContext driverContext) { - return SumIntGroupingAggregatorFunction.create(channels, driverContext); - } - - @Override - public String describe() { - return "sum of ints"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/SumIntGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/SumIntGroupingAggregatorFunction.java deleted file mode 100644 index 6891fe548908f..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/SumIntGroupingAggregatorFunction.java +++ /dev/null @@ -1,208 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link SumIntAggregator}. - * This class is generated. Do not edit it. - */ -public final class SumIntGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("sum", ElementType.LONG), - new IntermediateStateDesc("seen", ElementType.BOOLEAN) ); - - private final LongArrayState state; - - private final List channels; - - private final DriverContext driverContext; - - public SumIntGroupingAggregatorFunction(List channels, LongArrayState state, - DriverContext driverContext) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - } - - public static SumIntGroupingAggregatorFunction create(List channels, - DriverContext driverContext) { - return new SumIntGroupingAggregatorFunction(channels, new LongArrayState(driverContext.bigArrays(), SumIntAggregator.init()), driverContext); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - IntBlock valuesBlock = page.getBlock(channels.get(0)); - IntVector valuesVector = valuesBlock.asVector(); - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, IntBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - state.set(groupId, SumIntAggregator.combine(state.getOrDefault(groupId), values.getInt(v))); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, IntVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - state.set(groupId, SumIntAggregator.combine(state.getOrDefault(groupId), values.getInt(groupPosition + positionOffset))); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, IntBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - state.set(groupId, SumIntAggregator.combine(state.getOrDefault(groupId), values.getInt(v))); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, IntVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - state.set(groupId, SumIntAggregator.combine(state.getOrDefault(groupId), values.getInt(groupPosition + positionOffset))); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block sumUncast = page.getBlock(channels.get(0)); - if (sumUncast.areAllValuesNull()) { - return; - } - LongVector sum = ((LongBlock) sumUncast).asVector(); - Block seenUncast = page.getBlock(channels.get(1)); - if (seenUncast.areAllValuesNull()) { - return; - } - BooleanVector seen = ((BooleanBlock) seenUncast).asVector(); - assert sum.getPositionCount() == seen.getPositionCount(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (seen.getBoolean(groupPosition + positionOffset)) { - state.set(groupId, SumIntAggregator.combine(state.getOrDefault(groupId), sum.getLong(groupPosition + positionOffset))); - } - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - LongArrayState inState = ((SumIntGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - if (inState.hasValue(position)) { - state.set(groupId, SumIntAggregator.combine(state.getOrDefault(groupId), inState.get(position))); - } - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = state.toValuesBlock(selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/SumLongAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/SumLongAggregatorFunction.java deleted file mode 100644 index 38d1b3de78265..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/SumLongAggregatorFunction.java +++ /dev/null @@ -1,138 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunction} implementation for {@link SumLongAggregator}. - * This class is generated. Do not edit it. - */ -public final class SumLongAggregatorFunction implements AggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("sum", ElementType.LONG), - new IntermediateStateDesc("seen", ElementType.BOOLEAN) ); - - private final DriverContext driverContext; - - private final LongState state; - - private final List channels; - - public SumLongAggregatorFunction(DriverContext driverContext, List channels, - LongState state) { - this.driverContext = driverContext; - this.channels = channels; - this.state = state; - } - - public static SumLongAggregatorFunction create(DriverContext driverContext, - List channels) { - return new SumLongAggregatorFunction(driverContext, channels, new LongState(SumLongAggregator.init())); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public void addRawInput(Page page) { - LongBlock block = page.getBlock(channels.get(0)); - LongVector vector = block.asVector(); - if (vector != null) { - addRawVector(vector); - } else { - addRawBlock(block); - } - } - - private void addRawVector(LongVector vector) { - state.seen(true); - for (int i = 0; i < vector.getPositionCount(); i++) { - state.longValue(SumLongAggregator.combine(state.longValue(), vector.getLong(i))); - } - } - - private void addRawBlock(LongBlock block) { - for (int p = 0; p < block.getPositionCount(); p++) { - if (block.isNull(p)) { - continue; - } - state.seen(true); - int start = block.getFirstValueIndex(p); - int end = start + block.getValueCount(p); - for (int i = start; i < end; i++) { - state.longValue(SumLongAggregator.combine(state.longValue(), block.getLong(i))); - } - } - } - - @Override - public void addIntermediateInput(Page page) { - assert channels.size() == intermediateBlockCount(); - assert page.getBlockCount() >= channels.get(0) + intermediateStateDesc().size(); - Block sumUncast = page.getBlock(channels.get(0)); - if (sumUncast.areAllValuesNull()) { - return; - } - LongVector sum = ((LongBlock) sumUncast).asVector(); - assert sum.getPositionCount() == 1; - Block seenUncast = page.getBlock(channels.get(1)); - if (seenUncast.areAllValuesNull()) { - return; - } - BooleanVector seen = ((BooleanBlock) seenUncast).asVector(); - assert seen.getPositionCount() == 1; - if (seen.getBoolean(0)) { - state.longValue(SumLongAggregator.combine(state.longValue(), sum.getLong(0))); - state.seen(true); - } - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, DriverContext driverContext) { - state.toIntermediate(blocks, offset, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, DriverContext driverContext) { - if (state.seen() == false) { - blocks[offset] = driverContext.blockFactory().newConstantNullBlock(1); - return; - } - blocks[offset] = driverContext.blockFactory().newConstantLongBlockWith(state.longValue(), 1); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/SumLongAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/SumLongAggregatorFunctionSupplier.java deleted file mode 100644 index b4d36aa526075..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/SumLongAggregatorFunctionSupplier.java +++ /dev/null @@ -1,38 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link SumLongAggregator}. - * This class is generated. Do not edit it. - */ -public final class SumLongAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - public SumLongAggregatorFunctionSupplier(List channels) { - this.channels = channels; - } - - @Override - public SumLongAggregatorFunction aggregator(DriverContext driverContext) { - return SumLongAggregatorFunction.create(driverContext, channels); - } - - @Override - public SumLongGroupingAggregatorFunction groupingAggregator(DriverContext driverContext) { - return SumLongGroupingAggregatorFunction.create(channels, driverContext); - } - - @Override - public String describe() { - return "sum of longs"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/SumLongGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/SumLongGroupingAggregatorFunction.java deleted file mode 100644 index 507aa343aa74e..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/SumLongGroupingAggregatorFunction.java +++ /dev/null @@ -1,208 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link SumLongAggregator}. - * This class is generated. Do not edit it. - */ -public final class SumLongGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("sum", ElementType.LONG), - new IntermediateStateDesc("seen", ElementType.BOOLEAN) ); - - private final LongArrayState state; - - private final List channels; - - private final DriverContext driverContext; - - public SumLongGroupingAggregatorFunction(List channels, LongArrayState state, - DriverContext driverContext) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - } - - public static SumLongGroupingAggregatorFunction create(List channels, - DriverContext driverContext) { - return new SumLongGroupingAggregatorFunction(channels, new LongArrayState(driverContext.bigArrays(), SumLongAggregator.init()), driverContext); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - LongBlock valuesBlock = page.getBlock(channels.get(0)); - LongVector valuesVector = valuesBlock.asVector(); - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, LongBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - state.set(groupId, SumLongAggregator.combine(state.getOrDefault(groupId), values.getLong(v))); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, LongVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - state.set(groupId, SumLongAggregator.combine(state.getOrDefault(groupId), values.getLong(groupPosition + positionOffset))); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, LongBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - state.set(groupId, SumLongAggregator.combine(state.getOrDefault(groupId), values.getLong(v))); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, LongVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - state.set(groupId, SumLongAggregator.combine(state.getOrDefault(groupId), values.getLong(groupPosition + positionOffset))); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block sumUncast = page.getBlock(channels.get(0)); - if (sumUncast.areAllValuesNull()) { - return; - } - LongVector sum = ((LongBlock) sumUncast).asVector(); - Block seenUncast = page.getBlock(channels.get(1)); - if (seenUncast.areAllValuesNull()) { - return; - } - BooleanVector seen = ((BooleanBlock) seenUncast).asVector(); - assert sum.getPositionCount() == seen.getPositionCount(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (seen.getBoolean(groupPosition + positionOffset)) { - state.set(groupId, SumLongAggregator.combine(state.getOrDefault(groupId), sum.getLong(groupPosition + positionOffset))); - } - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - LongArrayState inState = ((SumLongGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - if (inState.hasValue(position)) { - state.set(groupId, SumLongAggregator.combine(state.getOrDefault(groupId), inState.get(position))); - } - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = state.toValuesBlock(selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesBooleanAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesBooleanAggregatorFunction.java deleted file mode 100644 index 3e9bc91e0039a..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesBooleanAggregatorFunction.java +++ /dev/null @@ -1,120 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunction} implementation for {@link ValuesBooleanAggregator}. - * This class is generated. Do not edit it. - */ -public final class ValuesBooleanAggregatorFunction implements AggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("values", ElementType.BOOLEAN) ); - - private final DriverContext driverContext; - - private final ValuesBooleanAggregator.SingleState state; - - private final List channels; - - public ValuesBooleanAggregatorFunction(DriverContext driverContext, List channels, - ValuesBooleanAggregator.SingleState state) { - this.driverContext = driverContext; - this.channels = channels; - this.state = state; - } - - public static ValuesBooleanAggregatorFunction create(DriverContext driverContext, - List channels) { - return new ValuesBooleanAggregatorFunction(driverContext, channels, ValuesBooleanAggregator.initSingle()); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public void addRawInput(Page page) { - BooleanBlock block = page.getBlock(channels.get(0)); - BooleanVector vector = block.asVector(); - if (vector != null) { - addRawVector(vector); - } else { - addRawBlock(block); - } - } - - private void addRawVector(BooleanVector vector) { - for (int i = 0; i < vector.getPositionCount(); i++) { - ValuesBooleanAggregator.combine(state, vector.getBoolean(i)); - } - } - - private void addRawBlock(BooleanBlock block) { - for (int p = 0; p < block.getPositionCount(); p++) { - if (block.isNull(p)) { - continue; - } - int start = block.getFirstValueIndex(p); - int end = start + block.getValueCount(p); - for (int i = start; i < end; i++) { - ValuesBooleanAggregator.combine(state, block.getBoolean(i)); - } - } - } - - @Override - public void addIntermediateInput(Page page) { - assert channels.size() == intermediateBlockCount(); - assert page.getBlockCount() >= channels.get(0) + intermediateStateDesc().size(); - Block valuesUncast = page.getBlock(channels.get(0)); - if (valuesUncast.areAllValuesNull()) { - return; - } - BooleanBlock values = (BooleanBlock) valuesUncast; - assert values.getPositionCount() == 1; - ValuesBooleanAggregator.combineIntermediate(state, values); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, DriverContext driverContext) { - state.toIntermediate(blocks, offset, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, DriverContext driverContext) { - blocks[offset] = ValuesBooleanAggregator.evaluateFinal(state, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesBooleanAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesBooleanAggregatorFunctionSupplier.java deleted file mode 100644 index 20d5a5fda7726..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesBooleanAggregatorFunctionSupplier.java +++ /dev/null @@ -1,38 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link ValuesBooleanAggregator}. - * This class is generated. Do not edit it. - */ -public final class ValuesBooleanAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - public ValuesBooleanAggregatorFunctionSupplier(List channels) { - this.channels = channels; - } - - @Override - public ValuesBooleanAggregatorFunction aggregator(DriverContext driverContext) { - return ValuesBooleanAggregatorFunction.create(driverContext, channels); - } - - @Override - public ValuesBooleanGroupingAggregatorFunction groupingAggregator(DriverContext driverContext) { - return ValuesBooleanGroupingAggregatorFunction.create(channels, driverContext); - } - - @Override - public String describe() { - return "values of booleans"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesBooleanGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesBooleanGroupingAggregatorFunction.java deleted file mode 100644 index 16e92a7c69ca8..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesBooleanGroupingAggregatorFunction.java +++ /dev/null @@ -1,195 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link ValuesBooleanAggregator}. - * This class is generated. Do not edit it. - */ -public final class ValuesBooleanGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("values", ElementType.BOOLEAN) ); - - private final ValuesBooleanAggregator.GroupingState state; - - private final List channels; - - private final DriverContext driverContext; - - public ValuesBooleanGroupingAggregatorFunction(List channels, - ValuesBooleanAggregator.GroupingState state, DriverContext driverContext) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - } - - public static ValuesBooleanGroupingAggregatorFunction create(List channels, - DriverContext driverContext) { - return new ValuesBooleanGroupingAggregatorFunction(channels, ValuesBooleanAggregator.initGrouping(driverContext.bigArrays()), driverContext); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - BooleanBlock valuesBlock = page.getBlock(channels.get(0)); - BooleanVector valuesVector = valuesBlock.asVector(); - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, BooleanBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - ValuesBooleanAggregator.combine(state, groupId, values.getBoolean(v)); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, BooleanVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - ValuesBooleanAggregator.combine(state, groupId, values.getBoolean(groupPosition + positionOffset)); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, BooleanBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - ValuesBooleanAggregator.combine(state, groupId, values.getBoolean(v)); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, BooleanVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - ValuesBooleanAggregator.combine(state, groupId, values.getBoolean(groupPosition + positionOffset)); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block valuesUncast = page.getBlock(channels.get(0)); - if (valuesUncast.areAllValuesNull()) { - return; - } - BooleanBlock values = (BooleanBlock) valuesUncast; - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - ValuesBooleanAggregator.combineIntermediate(state, groupId, values, groupPosition + positionOffset); - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - ValuesBooleanAggregator.GroupingState inState = ((ValuesBooleanGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - ValuesBooleanAggregator.combineStates(state, groupId, inState, position); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = ValuesBooleanAggregator.evaluateFinal(state, selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesBytesRefAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesBytesRefAggregatorFunction.java deleted file mode 100644 index 24b88f59e38f4..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesBytesRefAggregatorFunction.java +++ /dev/null @@ -1,124 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunction} implementation for {@link ValuesBytesRefAggregator}. - * This class is generated. Do not edit it. - */ -public final class ValuesBytesRefAggregatorFunction implements AggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("values", ElementType.BYTES_REF) ); - - private final DriverContext driverContext; - - private final ValuesBytesRefAggregator.SingleState state; - - private final List channels; - - public ValuesBytesRefAggregatorFunction(DriverContext driverContext, List channels, - ValuesBytesRefAggregator.SingleState state) { - this.driverContext = driverContext; - this.channels = channels; - this.state = state; - } - - public static ValuesBytesRefAggregatorFunction create(DriverContext driverContext, - List channels) { - return new ValuesBytesRefAggregatorFunction(driverContext, channels, ValuesBytesRefAggregator.initSingle(driverContext.bigArrays())); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public void addRawInput(Page page) { - BytesRefBlock block = page.getBlock(channels.get(0)); - BytesRefVector vector = block.asVector(); - if (vector != null) { - addRawVector(vector); - } else { - addRawBlock(block); - } - } - - private void addRawVector(BytesRefVector vector) { - BytesRef scratch = new BytesRef(); - for (int i = 0; i < vector.getPositionCount(); i++) { - ValuesBytesRefAggregator.combine(state, vector.getBytesRef(i, scratch)); - } - } - - private void addRawBlock(BytesRefBlock block) { - BytesRef scratch = new BytesRef(); - for (int p = 0; p < block.getPositionCount(); p++) { - if (block.isNull(p)) { - continue; - } - int start = block.getFirstValueIndex(p); - int end = start + block.getValueCount(p); - for (int i = start; i < end; i++) { - ValuesBytesRefAggregator.combine(state, block.getBytesRef(i, scratch)); - } - } - } - - @Override - public void addIntermediateInput(Page page) { - assert channels.size() == intermediateBlockCount(); - assert page.getBlockCount() >= channels.get(0) + intermediateStateDesc().size(); - Block valuesUncast = page.getBlock(channels.get(0)); - if (valuesUncast.areAllValuesNull()) { - return; - } - BytesRefBlock values = (BytesRefBlock) valuesUncast; - assert values.getPositionCount() == 1; - BytesRef scratch = new BytesRef(); - ValuesBytesRefAggregator.combineIntermediate(state, values); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, DriverContext driverContext) { - state.toIntermediate(blocks, offset, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, DriverContext driverContext) { - blocks[offset] = ValuesBytesRefAggregator.evaluateFinal(state, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesBytesRefAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesBytesRefAggregatorFunctionSupplier.java deleted file mode 100644 index 31de817edf868..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesBytesRefAggregatorFunctionSupplier.java +++ /dev/null @@ -1,38 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link ValuesBytesRefAggregator}. - * This class is generated. Do not edit it. - */ -public final class ValuesBytesRefAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - public ValuesBytesRefAggregatorFunctionSupplier(List channels) { - this.channels = channels; - } - - @Override - public ValuesBytesRefAggregatorFunction aggregator(DriverContext driverContext) { - return ValuesBytesRefAggregatorFunction.create(driverContext, channels); - } - - @Override - public ValuesBytesRefGroupingAggregatorFunction groupingAggregator(DriverContext driverContext) { - return ValuesBytesRefGroupingAggregatorFunction.create(channels, driverContext); - } - - @Override - public String describe() { - return "values of bytes"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesBytesRefGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesBytesRefGroupingAggregatorFunction.java deleted file mode 100644 index f9a51fcc52221..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesBytesRefGroupingAggregatorFunction.java +++ /dev/null @@ -1,201 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link ValuesBytesRefAggregator}. - * This class is generated. Do not edit it. - */ -public final class ValuesBytesRefGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("values", ElementType.BYTES_REF) ); - - private final ValuesBytesRefAggregator.GroupingState state; - - private final List channels; - - private final DriverContext driverContext; - - public ValuesBytesRefGroupingAggregatorFunction(List channels, - ValuesBytesRefAggregator.GroupingState state, DriverContext driverContext) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - } - - public static ValuesBytesRefGroupingAggregatorFunction create(List channels, - DriverContext driverContext) { - return new ValuesBytesRefGroupingAggregatorFunction(channels, ValuesBytesRefAggregator.initGrouping(driverContext.bigArrays()), driverContext); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - BytesRefBlock valuesBlock = page.getBlock(channels.get(0)); - BytesRefVector valuesVector = valuesBlock.asVector(); - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, BytesRefBlock values) { - BytesRef scratch = new BytesRef(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - ValuesBytesRefAggregator.combine(state, groupId, values.getBytesRef(v, scratch)); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, BytesRefVector values) { - BytesRef scratch = new BytesRef(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - ValuesBytesRefAggregator.combine(state, groupId, values.getBytesRef(groupPosition + positionOffset, scratch)); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, BytesRefBlock values) { - BytesRef scratch = new BytesRef(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - ValuesBytesRefAggregator.combine(state, groupId, values.getBytesRef(v, scratch)); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, BytesRefVector values) { - BytesRef scratch = new BytesRef(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - ValuesBytesRefAggregator.combine(state, groupId, values.getBytesRef(groupPosition + positionOffset, scratch)); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block valuesUncast = page.getBlock(channels.get(0)); - if (valuesUncast.areAllValuesNull()) { - return; - } - BytesRefBlock values = (BytesRefBlock) valuesUncast; - BytesRef scratch = new BytesRef(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - ValuesBytesRefAggregator.combineIntermediate(state, groupId, values, groupPosition + positionOffset); - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - ValuesBytesRefAggregator.GroupingState inState = ((ValuesBytesRefGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - ValuesBytesRefAggregator.combineStates(state, groupId, inState, position); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = ValuesBytesRefAggregator.evaluateFinal(state, selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesDoubleAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesDoubleAggregatorFunction.java deleted file mode 100644 index a6295038dbd7a..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesDoubleAggregatorFunction.java +++ /dev/null @@ -1,120 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunction} implementation for {@link ValuesDoubleAggregator}. - * This class is generated. Do not edit it. - */ -public final class ValuesDoubleAggregatorFunction implements AggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("values", ElementType.DOUBLE) ); - - private final DriverContext driverContext; - - private final ValuesDoubleAggregator.SingleState state; - - private final List channels; - - public ValuesDoubleAggregatorFunction(DriverContext driverContext, List channels, - ValuesDoubleAggregator.SingleState state) { - this.driverContext = driverContext; - this.channels = channels; - this.state = state; - } - - public static ValuesDoubleAggregatorFunction create(DriverContext driverContext, - List channels) { - return new ValuesDoubleAggregatorFunction(driverContext, channels, ValuesDoubleAggregator.initSingle(driverContext.bigArrays())); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public void addRawInput(Page page) { - DoubleBlock block = page.getBlock(channels.get(0)); - DoubleVector vector = block.asVector(); - if (vector != null) { - addRawVector(vector); - } else { - addRawBlock(block); - } - } - - private void addRawVector(DoubleVector vector) { - for (int i = 0; i < vector.getPositionCount(); i++) { - ValuesDoubleAggregator.combine(state, vector.getDouble(i)); - } - } - - private void addRawBlock(DoubleBlock block) { - for (int p = 0; p < block.getPositionCount(); p++) { - if (block.isNull(p)) { - continue; - } - int start = block.getFirstValueIndex(p); - int end = start + block.getValueCount(p); - for (int i = start; i < end; i++) { - ValuesDoubleAggregator.combine(state, block.getDouble(i)); - } - } - } - - @Override - public void addIntermediateInput(Page page) { - assert channels.size() == intermediateBlockCount(); - assert page.getBlockCount() >= channels.get(0) + intermediateStateDesc().size(); - Block valuesUncast = page.getBlock(channels.get(0)); - if (valuesUncast.areAllValuesNull()) { - return; - } - DoubleBlock values = (DoubleBlock) valuesUncast; - assert values.getPositionCount() == 1; - ValuesDoubleAggregator.combineIntermediate(state, values); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, DriverContext driverContext) { - state.toIntermediate(blocks, offset, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, DriverContext driverContext) { - blocks[offset] = ValuesDoubleAggregator.evaluateFinal(state, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesDoubleAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesDoubleAggregatorFunctionSupplier.java deleted file mode 100644 index 049deda37c460..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesDoubleAggregatorFunctionSupplier.java +++ /dev/null @@ -1,38 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link ValuesDoubleAggregator}. - * This class is generated. Do not edit it. - */ -public final class ValuesDoubleAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - public ValuesDoubleAggregatorFunctionSupplier(List channels) { - this.channels = channels; - } - - @Override - public ValuesDoubleAggregatorFunction aggregator(DriverContext driverContext) { - return ValuesDoubleAggregatorFunction.create(driverContext, channels); - } - - @Override - public ValuesDoubleGroupingAggregatorFunction groupingAggregator(DriverContext driverContext) { - return ValuesDoubleGroupingAggregatorFunction.create(channels, driverContext); - } - - @Override - public String describe() { - return "values of doubles"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesDoubleGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesDoubleGroupingAggregatorFunction.java deleted file mode 100644 index 11a0eb96c6a8e..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesDoubleGroupingAggregatorFunction.java +++ /dev/null @@ -1,195 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link ValuesDoubleAggregator}. - * This class is generated. Do not edit it. - */ -public final class ValuesDoubleGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("values", ElementType.DOUBLE) ); - - private final ValuesDoubleAggregator.GroupingState state; - - private final List channels; - - private final DriverContext driverContext; - - public ValuesDoubleGroupingAggregatorFunction(List channels, - ValuesDoubleAggregator.GroupingState state, DriverContext driverContext) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - } - - public static ValuesDoubleGroupingAggregatorFunction create(List channels, - DriverContext driverContext) { - return new ValuesDoubleGroupingAggregatorFunction(channels, ValuesDoubleAggregator.initGrouping(driverContext.bigArrays()), driverContext); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - DoubleBlock valuesBlock = page.getBlock(channels.get(0)); - DoubleVector valuesVector = valuesBlock.asVector(); - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, DoubleBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - ValuesDoubleAggregator.combine(state, groupId, values.getDouble(v)); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, DoubleVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - ValuesDoubleAggregator.combine(state, groupId, values.getDouble(groupPosition + positionOffset)); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, DoubleBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - ValuesDoubleAggregator.combine(state, groupId, values.getDouble(v)); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, DoubleVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - ValuesDoubleAggregator.combine(state, groupId, values.getDouble(groupPosition + positionOffset)); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block valuesUncast = page.getBlock(channels.get(0)); - if (valuesUncast.areAllValuesNull()) { - return; - } - DoubleBlock values = (DoubleBlock) valuesUncast; - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - ValuesDoubleAggregator.combineIntermediate(state, groupId, values, groupPosition + positionOffset); - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - ValuesDoubleAggregator.GroupingState inState = ((ValuesDoubleGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - ValuesDoubleAggregator.combineStates(state, groupId, inState, position); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = ValuesDoubleAggregator.evaluateFinal(state, selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesIntAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesIntAggregatorFunction.java deleted file mode 100644 index 19e578936cd14..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesIntAggregatorFunction.java +++ /dev/null @@ -1,120 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunction} implementation for {@link ValuesIntAggregator}. - * This class is generated. Do not edit it. - */ -public final class ValuesIntAggregatorFunction implements AggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("values", ElementType.INT) ); - - private final DriverContext driverContext; - - private final ValuesIntAggregator.SingleState state; - - private final List channels; - - public ValuesIntAggregatorFunction(DriverContext driverContext, List channels, - ValuesIntAggregator.SingleState state) { - this.driverContext = driverContext; - this.channels = channels; - this.state = state; - } - - public static ValuesIntAggregatorFunction create(DriverContext driverContext, - List channels) { - return new ValuesIntAggregatorFunction(driverContext, channels, ValuesIntAggregator.initSingle(driverContext.bigArrays())); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public void addRawInput(Page page) { - IntBlock block = page.getBlock(channels.get(0)); - IntVector vector = block.asVector(); - if (vector != null) { - addRawVector(vector); - } else { - addRawBlock(block); - } - } - - private void addRawVector(IntVector vector) { - for (int i = 0; i < vector.getPositionCount(); i++) { - ValuesIntAggregator.combine(state, vector.getInt(i)); - } - } - - private void addRawBlock(IntBlock block) { - for (int p = 0; p < block.getPositionCount(); p++) { - if (block.isNull(p)) { - continue; - } - int start = block.getFirstValueIndex(p); - int end = start + block.getValueCount(p); - for (int i = start; i < end; i++) { - ValuesIntAggregator.combine(state, block.getInt(i)); - } - } - } - - @Override - public void addIntermediateInput(Page page) { - assert channels.size() == intermediateBlockCount(); - assert page.getBlockCount() >= channels.get(0) + intermediateStateDesc().size(); - Block valuesUncast = page.getBlock(channels.get(0)); - if (valuesUncast.areAllValuesNull()) { - return; - } - IntBlock values = (IntBlock) valuesUncast; - assert values.getPositionCount() == 1; - ValuesIntAggregator.combineIntermediate(state, values); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, DriverContext driverContext) { - state.toIntermediate(blocks, offset, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, DriverContext driverContext) { - blocks[offset] = ValuesIntAggregator.evaluateFinal(state, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesIntAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesIntAggregatorFunctionSupplier.java deleted file mode 100644 index 21402b5913813..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesIntAggregatorFunctionSupplier.java +++ /dev/null @@ -1,38 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link ValuesIntAggregator}. - * This class is generated. Do not edit it. - */ -public final class ValuesIntAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - public ValuesIntAggregatorFunctionSupplier(List channels) { - this.channels = channels; - } - - @Override - public ValuesIntAggregatorFunction aggregator(DriverContext driverContext) { - return ValuesIntAggregatorFunction.create(driverContext, channels); - } - - @Override - public ValuesIntGroupingAggregatorFunction groupingAggregator(DriverContext driverContext) { - return ValuesIntGroupingAggregatorFunction.create(channels, driverContext); - } - - @Override - public String describe() { - return "values of ints"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesIntGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesIntGroupingAggregatorFunction.java deleted file mode 100644 index 67722cd1318c0..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesIntGroupingAggregatorFunction.java +++ /dev/null @@ -1,193 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link ValuesIntAggregator}. - * This class is generated. Do not edit it. - */ -public final class ValuesIntGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("values", ElementType.INT) ); - - private final ValuesIntAggregator.GroupingState state; - - private final List channels; - - private final DriverContext driverContext; - - public ValuesIntGroupingAggregatorFunction(List channels, - ValuesIntAggregator.GroupingState state, DriverContext driverContext) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - } - - public static ValuesIntGroupingAggregatorFunction create(List channels, - DriverContext driverContext) { - return new ValuesIntGroupingAggregatorFunction(channels, ValuesIntAggregator.initGrouping(driverContext.bigArrays()), driverContext); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - IntBlock valuesBlock = page.getBlock(channels.get(0)); - IntVector valuesVector = valuesBlock.asVector(); - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, IntBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - ValuesIntAggregator.combine(state, groupId, values.getInt(v)); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, IntVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - ValuesIntAggregator.combine(state, groupId, values.getInt(groupPosition + positionOffset)); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, IntBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - ValuesIntAggregator.combine(state, groupId, values.getInt(v)); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, IntVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - ValuesIntAggregator.combine(state, groupId, values.getInt(groupPosition + positionOffset)); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block valuesUncast = page.getBlock(channels.get(0)); - if (valuesUncast.areAllValuesNull()) { - return; - } - IntBlock values = (IntBlock) valuesUncast; - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - ValuesIntAggregator.combineIntermediate(state, groupId, values, groupPosition + positionOffset); - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - ValuesIntAggregator.GroupingState inState = ((ValuesIntGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - ValuesIntAggregator.combineStates(state, groupId, inState, position); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = ValuesIntAggregator.evaluateFinal(state, selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesLongAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesLongAggregatorFunction.java deleted file mode 100644 index 420da87076a37..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesLongAggregatorFunction.java +++ /dev/null @@ -1,120 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunction} implementation for {@link ValuesLongAggregator}. - * This class is generated. Do not edit it. - */ -public final class ValuesLongAggregatorFunction implements AggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("values", ElementType.LONG) ); - - private final DriverContext driverContext; - - private final ValuesLongAggregator.SingleState state; - - private final List channels; - - public ValuesLongAggregatorFunction(DriverContext driverContext, List channels, - ValuesLongAggregator.SingleState state) { - this.driverContext = driverContext; - this.channels = channels; - this.state = state; - } - - public static ValuesLongAggregatorFunction create(DriverContext driverContext, - List channels) { - return new ValuesLongAggregatorFunction(driverContext, channels, ValuesLongAggregator.initSingle(driverContext.bigArrays())); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public void addRawInput(Page page) { - LongBlock block = page.getBlock(channels.get(0)); - LongVector vector = block.asVector(); - if (vector != null) { - addRawVector(vector); - } else { - addRawBlock(block); - } - } - - private void addRawVector(LongVector vector) { - for (int i = 0; i < vector.getPositionCount(); i++) { - ValuesLongAggregator.combine(state, vector.getLong(i)); - } - } - - private void addRawBlock(LongBlock block) { - for (int p = 0; p < block.getPositionCount(); p++) { - if (block.isNull(p)) { - continue; - } - int start = block.getFirstValueIndex(p); - int end = start + block.getValueCount(p); - for (int i = start; i < end; i++) { - ValuesLongAggregator.combine(state, block.getLong(i)); - } - } - } - - @Override - public void addIntermediateInput(Page page) { - assert channels.size() == intermediateBlockCount(); - assert page.getBlockCount() >= channels.get(0) + intermediateStateDesc().size(); - Block valuesUncast = page.getBlock(channels.get(0)); - if (valuesUncast.areAllValuesNull()) { - return; - } - LongBlock values = (LongBlock) valuesUncast; - assert values.getPositionCount() == 1; - ValuesLongAggregator.combineIntermediate(state, values); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, DriverContext driverContext) { - state.toIntermediate(blocks, offset, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, DriverContext driverContext) { - blocks[offset] = ValuesLongAggregator.evaluateFinal(state, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesLongAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesLongAggregatorFunctionSupplier.java deleted file mode 100644 index a025bd0ade17a..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesLongAggregatorFunctionSupplier.java +++ /dev/null @@ -1,38 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link ValuesLongAggregator}. - * This class is generated. Do not edit it. - */ -public final class ValuesLongAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - public ValuesLongAggregatorFunctionSupplier(List channels) { - this.channels = channels; - } - - @Override - public ValuesLongAggregatorFunction aggregator(DriverContext driverContext) { - return ValuesLongAggregatorFunction.create(driverContext, channels); - } - - @Override - public ValuesLongGroupingAggregatorFunction groupingAggregator(DriverContext driverContext) { - return ValuesLongGroupingAggregatorFunction.create(channels, driverContext); - } - - @Override - public String describe() { - return "values of longs"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesLongGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesLongGroupingAggregatorFunction.java deleted file mode 100644 index 06508ce360ba4..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/ValuesLongGroupingAggregatorFunction.java +++ /dev/null @@ -1,195 +0,0 @@ -// 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.aggregation; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link ValuesLongAggregator}. - * This class is generated. Do not edit it. - */ -public final class ValuesLongGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("values", ElementType.LONG) ); - - private final ValuesLongAggregator.GroupingState state; - - private final List channels; - - private final DriverContext driverContext; - - public ValuesLongGroupingAggregatorFunction(List channels, - ValuesLongAggregator.GroupingState state, DriverContext driverContext) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - } - - public static ValuesLongGroupingAggregatorFunction create(List channels, - DriverContext driverContext) { - return new ValuesLongGroupingAggregatorFunction(channels, ValuesLongAggregator.initGrouping(driverContext.bigArrays()), driverContext); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - LongBlock valuesBlock = page.getBlock(channels.get(0)); - LongVector valuesVector = valuesBlock.asVector(); - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, LongBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - ValuesLongAggregator.combine(state, groupId, values.getLong(v)); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, LongVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - ValuesLongAggregator.combine(state, groupId, values.getLong(groupPosition + positionOffset)); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, LongBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - ValuesLongAggregator.combine(state, groupId, values.getLong(v)); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, LongVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - ValuesLongAggregator.combine(state, groupId, values.getLong(groupPosition + positionOffset)); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block valuesUncast = page.getBlock(channels.get(0)); - if (valuesUncast.areAllValuesNull()) { - return; - } - LongBlock values = (LongBlock) valuesUncast; - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - ValuesLongAggregator.combineIntermediate(state, groupId, values, groupPosition + positionOffset); - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - ValuesLongAggregator.GroupingState inState = ((ValuesLongGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - ValuesLongAggregator.combineStates(state, groupId, inState, position); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = ValuesLongAggregator.evaluateFinal(state, selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidCartesianPointDocValuesAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidCartesianPointDocValuesAggregatorFunction.java deleted file mode 100644 index eef094ce2ecfa..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidCartesianPointDocValuesAggregatorFunction.java +++ /dev/null @@ -1,152 +0,0 @@ -// 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.aggregation.spatial; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.aggregation.AggregatorFunction; -import org.elasticsearch.compute.aggregation.IntermediateStateDesc; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunction} implementation for {@link SpatialCentroidCartesianPointDocValuesAggregator}. - * This class is generated. Do not edit it. - */ -public final class SpatialCentroidCartesianPointDocValuesAggregatorFunction implements AggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("xVal", ElementType.DOUBLE), - new IntermediateStateDesc("xDel", ElementType.DOUBLE), - new IntermediateStateDesc("yVal", ElementType.DOUBLE), - new IntermediateStateDesc("yDel", ElementType.DOUBLE), - new IntermediateStateDesc("count", ElementType.LONG) ); - - private final DriverContext driverContext; - - private final CentroidPointAggregator.CentroidState state; - - private final List channels; - - public SpatialCentroidCartesianPointDocValuesAggregatorFunction(DriverContext driverContext, - List channels, CentroidPointAggregator.CentroidState state) { - this.driverContext = driverContext; - this.channels = channels; - this.state = state; - } - - public static SpatialCentroidCartesianPointDocValuesAggregatorFunction create( - DriverContext driverContext, List channels) { - return new SpatialCentroidCartesianPointDocValuesAggregatorFunction(driverContext, channels, SpatialCentroidCartesianPointDocValuesAggregator.initSingle()); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public void addRawInput(Page page) { - LongBlock block = page.getBlock(channels.get(0)); - LongVector vector = block.asVector(); - if (vector != null) { - addRawVector(vector); - } else { - addRawBlock(block); - } - } - - private void addRawVector(LongVector vector) { - for (int i = 0; i < vector.getPositionCount(); i++) { - SpatialCentroidCartesianPointDocValuesAggregator.combine(state, vector.getLong(i)); - } - } - - private void addRawBlock(LongBlock block) { - for (int p = 0; p < block.getPositionCount(); p++) { - if (block.isNull(p)) { - continue; - } - int start = block.getFirstValueIndex(p); - int end = start + block.getValueCount(p); - for (int i = start; i < end; i++) { - SpatialCentroidCartesianPointDocValuesAggregator.combine(state, block.getLong(i)); - } - } - } - - @Override - public void addIntermediateInput(Page page) { - assert channels.size() == intermediateBlockCount(); - assert page.getBlockCount() >= channels.get(0) + intermediateStateDesc().size(); - Block xValUncast = page.getBlock(channels.get(0)); - if (xValUncast.areAllValuesNull()) { - return; - } - DoubleVector xVal = ((DoubleBlock) xValUncast).asVector(); - assert xVal.getPositionCount() == 1; - Block xDelUncast = page.getBlock(channels.get(1)); - if (xDelUncast.areAllValuesNull()) { - return; - } - DoubleVector xDel = ((DoubleBlock) xDelUncast).asVector(); - assert xDel.getPositionCount() == 1; - Block yValUncast = page.getBlock(channels.get(2)); - if (yValUncast.areAllValuesNull()) { - return; - } - DoubleVector yVal = ((DoubleBlock) yValUncast).asVector(); - assert yVal.getPositionCount() == 1; - Block yDelUncast = page.getBlock(channels.get(3)); - if (yDelUncast.areAllValuesNull()) { - return; - } - DoubleVector yDel = ((DoubleBlock) yDelUncast).asVector(); - assert yDel.getPositionCount() == 1; - Block countUncast = page.getBlock(channels.get(4)); - if (countUncast.areAllValuesNull()) { - return; - } - LongVector count = ((LongBlock) countUncast).asVector(); - assert count.getPositionCount() == 1; - SpatialCentroidCartesianPointDocValuesAggregator.combineIntermediate(state, xVal.getDouble(0), xDel.getDouble(0), yVal.getDouble(0), yDel.getDouble(0), count.getLong(0)); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, DriverContext driverContext) { - state.toIntermediate(blocks, offset, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, DriverContext driverContext) { - blocks[offset] = SpatialCentroidCartesianPointDocValuesAggregator.evaluateFinal(state, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidCartesianPointDocValuesAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidCartesianPointDocValuesAggregatorFunctionSupplier.java deleted file mode 100644 index 3228340beeb43..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidCartesianPointDocValuesAggregatorFunctionSupplier.java +++ /dev/null @@ -1,41 +0,0 @@ -// 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.aggregation.spatial; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.aggregation.AggregatorFunctionSupplier; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link SpatialCentroidCartesianPointDocValuesAggregator}. - * This class is generated. Do not edit it. - */ -public final class SpatialCentroidCartesianPointDocValuesAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - public SpatialCentroidCartesianPointDocValuesAggregatorFunctionSupplier(List channels) { - this.channels = channels; - } - - @Override - public SpatialCentroidCartesianPointDocValuesAggregatorFunction aggregator( - DriverContext driverContext) { - return SpatialCentroidCartesianPointDocValuesAggregatorFunction.create(driverContext, channels); - } - - @Override - public SpatialCentroidCartesianPointDocValuesGroupingAggregatorFunction groupingAggregator( - DriverContext driverContext) { - return SpatialCentroidCartesianPointDocValuesGroupingAggregatorFunction.create(channels, driverContext); - } - - @Override - public String describe() { - return "spatial_centroid_cartesian_point_doc of valuess"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidCartesianPointDocValuesGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidCartesianPointDocValuesGroupingAggregatorFunction.java deleted file mode 100644 index 795207b245023..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidCartesianPointDocValuesGroupingAggregatorFunction.java +++ /dev/null @@ -1,225 +0,0 @@ -// 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.aggregation.spatial; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.aggregation.GroupingAggregatorFunction; -import org.elasticsearch.compute.aggregation.IntermediateStateDesc; -import org.elasticsearch.compute.aggregation.SeenGroupIds; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link SpatialCentroidCartesianPointDocValuesAggregator}. - * This class is generated. Do not edit it. - */ -public final class SpatialCentroidCartesianPointDocValuesGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("xVal", ElementType.DOUBLE), - new IntermediateStateDesc("xDel", ElementType.DOUBLE), - new IntermediateStateDesc("yVal", ElementType.DOUBLE), - new IntermediateStateDesc("yDel", ElementType.DOUBLE), - new IntermediateStateDesc("count", ElementType.LONG) ); - - private final CentroidPointAggregator.GroupingCentroidState state; - - private final List channels; - - private final DriverContext driverContext; - - public SpatialCentroidCartesianPointDocValuesGroupingAggregatorFunction(List channels, - CentroidPointAggregator.GroupingCentroidState state, DriverContext driverContext) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - } - - public static SpatialCentroidCartesianPointDocValuesGroupingAggregatorFunction create( - List channels, DriverContext driverContext) { - return new SpatialCentroidCartesianPointDocValuesGroupingAggregatorFunction(channels, SpatialCentroidCartesianPointDocValuesAggregator.initGrouping(driverContext.bigArrays()), driverContext); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - LongBlock valuesBlock = page.getBlock(channels.get(0)); - LongVector valuesVector = valuesBlock.asVector(); - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, LongBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - SpatialCentroidCartesianPointDocValuesAggregator.combine(state, groupId, values.getLong(v)); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, LongVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - SpatialCentroidCartesianPointDocValuesAggregator.combine(state, groupId, values.getLong(groupPosition + positionOffset)); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, LongBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - SpatialCentroidCartesianPointDocValuesAggregator.combine(state, groupId, values.getLong(v)); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, LongVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - SpatialCentroidCartesianPointDocValuesAggregator.combine(state, groupId, values.getLong(groupPosition + positionOffset)); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block xValUncast = page.getBlock(channels.get(0)); - if (xValUncast.areAllValuesNull()) { - return; - } - DoubleVector xVal = ((DoubleBlock) xValUncast).asVector(); - Block xDelUncast = page.getBlock(channels.get(1)); - if (xDelUncast.areAllValuesNull()) { - return; - } - DoubleVector xDel = ((DoubleBlock) xDelUncast).asVector(); - Block yValUncast = page.getBlock(channels.get(2)); - if (yValUncast.areAllValuesNull()) { - return; - } - DoubleVector yVal = ((DoubleBlock) yValUncast).asVector(); - Block yDelUncast = page.getBlock(channels.get(3)); - if (yDelUncast.areAllValuesNull()) { - return; - } - DoubleVector yDel = ((DoubleBlock) yDelUncast).asVector(); - Block countUncast = page.getBlock(channels.get(4)); - if (countUncast.areAllValuesNull()) { - return; - } - LongVector count = ((LongBlock) countUncast).asVector(); - assert xVal.getPositionCount() == xDel.getPositionCount() && xVal.getPositionCount() == yVal.getPositionCount() && xVal.getPositionCount() == yDel.getPositionCount() && xVal.getPositionCount() == count.getPositionCount(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - SpatialCentroidCartesianPointDocValuesAggregator.combineIntermediate(state, groupId, xVal.getDouble(groupPosition + positionOffset), xDel.getDouble(groupPosition + positionOffset), yVal.getDouble(groupPosition + positionOffset), yDel.getDouble(groupPosition + positionOffset), count.getLong(groupPosition + positionOffset)); - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - CentroidPointAggregator.GroupingCentroidState inState = ((SpatialCentroidCartesianPointDocValuesGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - SpatialCentroidCartesianPointDocValuesAggregator.combineStates(state, groupId, inState, position); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = SpatialCentroidCartesianPointDocValuesAggregator.evaluateFinal(state, selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidCartesianPointSourceValuesAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidCartesianPointSourceValuesAggregatorFunction.java deleted file mode 100644 index bdc7c58a6c963..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidCartesianPointSourceValuesAggregatorFunction.java +++ /dev/null @@ -1,157 +0,0 @@ -// 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.aggregation.spatial; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.aggregation.AggregatorFunction; -import org.elasticsearch.compute.aggregation.IntermediateStateDesc; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunction} implementation for {@link SpatialCentroidCartesianPointSourceValuesAggregator}. - * This class is generated. Do not edit it. - */ -public final class SpatialCentroidCartesianPointSourceValuesAggregatorFunction implements AggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("xVal", ElementType.DOUBLE), - new IntermediateStateDesc("xDel", ElementType.DOUBLE), - new IntermediateStateDesc("yVal", ElementType.DOUBLE), - new IntermediateStateDesc("yDel", ElementType.DOUBLE), - new IntermediateStateDesc("count", ElementType.LONG) ); - - private final DriverContext driverContext; - - private final CentroidPointAggregator.CentroidState state; - - private final List channels; - - public SpatialCentroidCartesianPointSourceValuesAggregatorFunction(DriverContext driverContext, - List channels, CentroidPointAggregator.CentroidState state) { - this.driverContext = driverContext; - this.channels = channels; - this.state = state; - } - - public static SpatialCentroidCartesianPointSourceValuesAggregatorFunction create( - DriverContext driverContext, List channels) { - return new SpatialCentroidCartesianPointSourceValuesAggregatorFunction(driverContext, channels, SpatialCentroidCartesianPointSourceValuesAggregator.initSingle()); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public void addRawInput(Page page) { - BytesRefBlock block = page.getBlock(channels.get(0)); - BytesRefVector vector = block.asVector(); - if (vector != null) { - addRawVector(vector); - } else { - addRawBlock(block); - } - } - - private void addRawVector(BytesRefVector vector) { - BytesRef scratch = new BytesRef(); - for (int i = 0; i < vector.getPositionCount(); i++) { - SpatialCentroidCartesianPointSourceValuesAggregator.combine(state, vector.getBytesRef(i, scratch)); - } - } - - private void addRawBlock(BytesRefBlock block) { - BytesRef scratch = new BytesRef(); - for (int p = 0; p < block.getPositionCount(); p++) { - if (block.isNull(p)) { - continue; - } - int start = block.getFirstValueIndex(p); - int end = start + block.getValueCount(p); - for (int i = start; i < end; i++) { - SpatialCentroidCartesianPointSourceValuesAggregator.combine(state, block.getBytesRef(i, scratch)); - } - } - } - - @Override - public void addIntermediateInput(Page page) { - assert channels.size() == intermediateBlockCount(); - assert page.getBlockCount() >= channels.get(0) + intermediateStateDesc().size(); - Block xValUncast = page.getBlock(channels.get(0)); - if (xValUncast.areAllValuesNull()) { - return; - } - DoubleVector xVal = ((DoubleBlock) xValUncast).asVector(); - assert xVal.getPositionCount() == 1; - Block xDelUncast = page.getBlock(channels.get(1)); - if (xDelUncast.areAllValuesNull()) { - return; - } - DoubleVector xDel = ((DoubleBlock) xDelUncast).asVector(); - assert xDel.getPositionCount() == 1; - Block yValUncast = page.getBlock(channels.get(2)); - if (yValUncast.areAllValuesNull()) { - return; - } - DoubleVector yVal = ((DoubleBlock) yValUncast).asVector(); - assert yVal.getPositionCount() == 1; - Block yDelUncast = page.getBlock(channels.get(3)); - if (yDelUncast.areAllValuesNull()) { - return; - } - DoubleVector yDel = ((DoubleBlock) yDelUncast).asVector(); - assert yDel.getPositionCount() == 1; - Block countUncast = page.getBlock(channels.get(4)); - if (countUncast.areAllValuesNull()) { - return; - } - LongVector count = ((LongBlock) countUncast).asVector(); - assert count.getPositionCount() == 1; - SpatialCentroidCartesianPointSourceValuesAggregator.combineIntermediate(state, xVal.getDouble(0), xDel.getDouble(0), yVal.getDouble(0), yDel.getDouble(0), count.getLong(0)); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, DriverContext driverContext) { - state.toIntermediate(blocks, offset, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, DriverContext driverContext) { - blocks[offset] = SpatialCentroidCartesianPointSourceValuesAggregator.evaluateFinal(state, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidCartesianPointSourceValuesAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidCartesianPointSourceValuesAggregatorFunctionSupplier.java deleted file mode 100644 index 9b0d7c5f64cd7..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidCartesianPointSourceValuesAggregatorFunctionSupplier.java +++ /dev/null @@ -1,42 +0,0 @@ -// 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.aggregation.spatial; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.aggregation.AggregatorFunctionSupplier; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link SpatialCentroidCartesianPointSourceValuesAggregator}. - * This class is generated. Do not edit it. - */ -public final class SpatialCentroidCartesianPointSourceValuesAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - public SpatialCentroidCartesianPointSourceValuesAggregatorFunctionSupplier( - List channels) { - this.channels = channels; - } - - @Override - public SpatialCentroidCartesianPointSourceValuesAggregatorFunction aggregator( - DriverContext driverContext) { - return SpatialCentroidCartesianPointSourceValuesAggregatorFunction.create(driverContext, channels); - } - - @Override - public SpatialCentroidCartesianPointSourceValuesGroupingAggregatorFunction groupingAggregator( - DriverContext driverContext) { - return SpatialCentroidCartesianPointSourceValuesGroupingAggregatorFunction.create(channels, driverContext); - } - - @Override - public String describe() { - return "spatial_centroid_cartesian_point_source of valuess"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidCartesianPointSourceValuesGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidCartesianPointSourceValuesGroupingAggregatorFunction.java deleted file mode 100644 index 12c0f24ef43e3..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidCartesianPointSourceValuesGroupingAggregatorFunction.java +++ /dev/null @@ -1,232 +0,0 @@ -// 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.aggregation.spatial; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.aggregation.GroupingAggregatorFunction; -import org.elasticsearch.compute.aggregation.IntermediateStateDesc; -import org.elasticsearch.compute.aggregation.SeenGroupIds; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link SpatialCentroidCartesianPointSourceValuesAggregator}. - * This class is generated. Do not edit it. - */ -public final class SpatialCentroidCartesianPointSourceValuesGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("xVal", ElementType.DOUBLE), - new IntermediateStateDesc("xDel", ElementType.DOUBLE), - new IntermediateStateDesc("yVal", ElementType.DOUBLE), - new IntermediateStateDesc("yDel", ElementType.DOUBLE), - new IntermediateStateDesc("count", ElementType.LONG) ); - - private final CentroidPointAggregator.GroupingCentroidState state; - - private final List channels; - - private final DriverContext driverContext; - - public SpatialCentroidCartesianPointSourceValuesGroupingAggregatorFunction(List channels, - CentroidPointAggregator.GroupingCentroidState state, DriverContext driverContext) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - } - - public static SpatialCentroidCartesianPointSourceValuesGroupingAggregatorFunction create( - List channels, DriverContext driverContext) { - return new SpatialCentroidCartesianPointSourceValuesGroupingAggregatorFunction(channels, SpatialCentroidCartesianPointSourceValuesAggregator.initGrouping(driverContext.bigArrays()), driverContext); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - BytesRefBlock valuesBlock = page.getBlock(channels.get(0)); - BytesRefVector valuesVector = valuesBlock.asVector(); - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, BytesRefBlock values) { - BytesRef scratch = new BytesRef(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - SpatialCentroidCartesianPointSourceValuesAggregator.combine(state, groupId, values.getBytesRef(v, scratch)); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, BytesRefVector values) { - BytesRef scratch = new BytesRef(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - SpatialCentroidCartesianPointSourceValuesAggregator.combine(state, groupId, values.getBytesRef(groupPosition + positionOffset, scratch)); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, BytesRefBlock values) { - BytesRef scratch = new BytesRef(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - SpatialCentroidCartesianPointSourceValuesAggregator.combine(state, groupId, values.getBytesRef(v, scratch)); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, BytesRefVector values) { - BytesRef scratch = new BytesRef(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - SpatialCentroidCartesianPointSourceValuesAggregator.combine(state, groupId, values.getBytesRef(groupPosition + positionOffset, scratch)); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block xValUncast = page.getBlock(channels.get(0)); - if (xValUncast.areAllValuesNull()) { - return; - } - DoubleVector xVal = ((DoubleBlock) xValUncast).asVector(); - Block xDelUncast = page.getBlock(channels.get(1)); - if (xDelUncast.areAllValuesNull()) { - return; - } - DoubleVector xDel = ((DoubleBlock) xDelUncast).asVector(); - Block yValUncast = page.getBlock(channels.get(2)); - if (yValUncast.areAllValuesNull()) { - return; - } - DoubleVector yVal = ((DoubleBlock) yValUncast).asVector(); - Block yDelUncast = page.getBlock(channels.get(3)); - if (yDelUncast.areAllValuesNull()) { - return; - } - DoubleVector yDel = ((DoubleBlock) yDelUncast).asVector(); - Block countUncast = page.getBlock(channels.get(4)); - if (countUncast.areAllValuesNull()) { - return; - } - LongVector count = ((LongBlock) countUncast).asVector(); - assert xVal.getPositionCount() == xDel.getPositionCount() && xVal.getPositionCount() == yVal.getPositionCount() && xVal.getPositionCount() == yDel.getPositionCount() && xVal.getPositionCount() == count.getPositionCount(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - SpatialCentroidCartesianPointSourceValuesAggregator.combineIntermediate(state, groupId, xVal.getDouble(groupPosition + positionOffset), xDel.getDouble(groupPosition + positionOffset), yVal.getDouble(groupPosition + positionOffset), yDel.getDouble(groupPosition + positionOffset), count.getLong(groupPosition + positionOffset)); - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - CentroidPointAggregator.GroupingCentroidState inState = ((SpatialCentroidCartesianPointSourceValuesGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - SpatialCentroidCartesianPointSourceValuesAggregator.combineStates(state, groupId, inState, position); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = SpatialCentroidCartesianPointSourceValuesAggregator.evaluateFinal(state, selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidGeoPointDocValuesAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidGeoPointDocValuesAggregatorFunction.java deleted file mode 100644 index fcd17d4c5cd86..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidGeoPointDocValuesAggregatorFunction.java +++ /dev/null @@ -1,152 +0,0 @@ -// 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.aggregation.spatial; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.aggregation.AggregatorFunction; -import org.elasticsearch.compute.aggregation.IntermediateStateDesc; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunction} implementation for {@link SpatialCentroidGeoPointDocValuesAggregator}. - * This class is generated. Do not edit it. - */ -public final class SpatialCentroidGeoPointDocValuesAggregatorFunction implements AggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("xVal", ElementType.DOUBLE), - new IntermediateStateDesc("xDel", ElementType.DOUBLE), - new IntermediateStateDesc("yVal", ElementType.DOUBLE), - new IntermediateStateDesc("yDel", ElementType.DOUBLE), - new IntermediateStateDesc("count", ElementType.LONG) ); - - private final DriverContext driverContext; - - private final CentroidPointAggregator.CentroidState state; - - private final List channels; - - public SpatialCentroidGeoPointDocValuesAggregatorFunction(DriverContext driverContext, - List channels, CentroidPointAggregator.CentroidState state) { - this.driverContext = driverContext; - this.channels = channels; - this.state = state; - } - - public static SpatialCentroidGeoPointDocValuesAggregatorFunction create( - DriverContext driverContext, List channels) { - return new SpatialCentroidGeoPointDocValuesAggregatorFunction(driverContext, channels, SpatialCentroidGeoPointDocValuesAggregator.initSingle()); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public void addRawInput(Page page) { - LongBlock block = page.getBlock(channels.get(0)); - LongVector vector = block.asVector(); - if (vector != null) { - addRawVector(vector); - } else { - addRawBlock(block); - } - } - - private void addRawVector(LongVector vector) { - for (int i = 0; i < vector.getPositionCount(); i++) { - SpatialCentroidGeoPointDocValuesAggregator.combine(state, vector.getLong(i)); - } - } - - private void addRawBlock(LongBlock block) { - for (int p = 0; p < block.getPositionCount(); p++) { - if (block.isNull(p)) { - continue; - } - int start = block.getFirstValueIndex(p); - int end = start + block.getValueCount(p); - for (int i = start; i < end; i++) { - SpatialCentroidGeoPointDocValuesAggregator.combine(state, block.getLong(i)); - } - } - } - - @Override - public void addIntermediateInput(Page page) { - assert channels.size() == intermediateBlockCount(); - assert page.getBlockCount() >= channels.get(0) + intermediateStateDesc().size(); - Block xValUncast = page.getBlock(channels.get(0)); - if (xValUncast.areAllValuesNull()) { - return; - } - DoubleVector xVal = ((DoubleBlock) xValUncast).asVector(); - assert xVal.getPositionCount() == 1; - Block xDelUncast = page.getBlock(channels.get(1)); - if (xDelUncast.areAllValuesNull()) { - return; - } - DoubleVector xDel = ((DoubleBlock) xDelUncast).asVector(); - assert xDel.getPositionCount() == 1; - Block yValUncast = page.getBlock(channels.get(2)); - if (yValUncast.areAllValuesNull()) { - return; - } - DoubleVector yVal = ((DoubleBlock) yValUncast).asVector(); - assert yVal.getPositionCount() == 1; - Block yDelUncast = page.getBlock(channels.get(3)); - if (yDelUncast.areAllValuesNull()) { - return; - } - DoubleVector yDel = ((DoubleBlock) yDelUncast).asVector(); - assert yDel.getPositionCount() == 1; - Block countUncast = page.getBlock(channels.get(4)); - if (countUncast.areAllValuesNull()) { - return; - } - LongVector count = ((LongBlock) countUncast).asVector(); - assert count.getPositionCount() == 1; - SpatialCentroidGeoPointDocValuesAggregator.combineIntermediate(state, xVal.getDouble(0), xDel.getDouble(0), yVal.getDouble(0), yDel.getDouble(0), count.getLong(0)); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, DriverContext driverContext) { - state.toIntermediate(blocks, offset, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, DriverContext driverContext) { - blocks[offset] = SpatialCentroidGeoPointDocValuesAggregator.evaluateFinal(state, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidGeoPointDocValuesAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidGeoPointDocValuesAggregatorFunctionSupplier.java deleted file mode 100644 index 46c2777e8c77a..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidGeoPointDocValuesAggregatorFunctionSupplier.java +++ /dev/null @@ -1,41 +0,0 @@ -// 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.aggregation.spatial; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.aggregation.AggregatorFunctionSupplier; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link SpatialCentroidGeoPointDocValuesAggregator}. - * This class is generated. Do not edit it. - */ -public final class SpatialCentroidGeoPointDocValuesAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - public SpatialCentroidGeoPointDocValuesAggregatorFunctionSupplier(List channels) { - this.channels = channels; - } - - @Override - public SpatialCentroidGeoPointDocValuesAggregatorFunction aggregator( - DriverContext driverContext) { - return SpatialCentroidGeoPointDocValuesAggregatorFunction.create(driverContext, channels); - } - - @Override - public SpatialCentroidGeoPointDocValuesGroupingAggregatorFunction groupingAggregator( - DriverContext driverContext) { - return SpatialCentroidGeoPointDocValuesGroupingAggregatorFunction.create(channels, driverContext); - } - - @Override - public String describe() { - return "spatial_centroid_geo_point_doc of valuess"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidGeoPointDocValuesGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidGeoPointDocValuesGroupingAggregatorFunction.java deleted file mode 100644 index 2447939d56db9..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidGeoPointDocValuesGroupingAggregatorFunction.java +++ /dev/null @@ -1,225 +0,0 @@ -// 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.aggregation.spatial; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.elasticsearch.compute.aggregation.GroupingAggregatorFunction; -import org.elasticsearch.compute.aggregation.IntermediateStateDesc; -import org.elasticsearch.compute.aggregation.SeenGroupIds; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link SpatialCentroidGeoPointDocValuesAggregator}. - * This class is generated. Do not edit it. - */ -public final class SpatialCentroidGeoPointDocValuesGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("xVal", ElementType.DOUBLE), - new IntermediateStateDesc("xDel", ElementType.DOUBLE), - new IntermediateStateDesc("yVal", ElementType.DOUBLE), - new IntermediateStateDesc("yDel", ElementType.DOUBLE), - new IntermediateStateDesc("count", ElementType.LONG) ); - - private final CentroidPointAggregator.GroupingCentroidState state; - - private final List channels; - - private final DriverContext driverContext; - - public SpatialCentroidGeoPointDocValuesGroupingAggregatorFunction(List channels, - CentroidPointAggregator.GroupingCentroidState state, DriverContext driverContext) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - } - - public static SpatialCentroidGeoPointDocValuesGroupingAggregatorFunction create( - List channels, DriverContext driverContext) { - return new SpatialCentroidGeoPointDocValuesGroupingAggregatorFunction(channels, SpatialCentroidGeoPointDocValuesAggregator.initGrouping(driverContext.bigArrays()), driverContext); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - LongBlock valuesBlock = page.getBlock(channels.get(0)); - LongVector valuesVector = valuesBlock.asVector(); - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, LongBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - SpatialCentroidGeoPointDocValuesAggregator.combine(state, groupId, values.getLong(v)); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, LongVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - SpatialCentroidGeoPointDocValuesAggregator.combine(state, groupId, values.getLong(groupPosition + positionOffset)); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, LongBlock values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - SpatialCentroidGeoPointDocValuesAggregator.combine(state, groupId, values.getLong(v)); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, LongVector values) { - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - SpatialCentroidGeoPointDocValuesAggregator.combine(state, groupId, values.getLong(groupPosition + positionOffset)); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block xValUncast = page.getBlock(channels.get(0)); - if (xValUncast.areAllValuesNull()) { - return; - } - DoubleVector xVal = ((DoubleBlock) xValUncast).asVector(); - Block xDelUncast = page.getBlock(channels.get(1)); - if (xDelUncast.areAllValuesNull()) { - return; - } - DoubleVector xDel = ((DoubleBlock) xDelUncast).asVector(); - Block yValUncast = page.getBlock(channels.get(2)); - if (yValUncast.areAllValuesNull()) { - return; - } - DoubleVector yVal = ((DoubleBlock) yValUncast).asVector(); - Block yDelUncast = page.getBlock(channels.get(3)); - if (yDelUncast.areAllValuesNull()) { - return; - } - DoubleVector yDel = ((DoubleBlock) yDelUncast).asVector(); - Block countUncast = page.getBlock(channels.get(4)); - if (countUncast.areAllValuesNull()) { - return; - } - LongVector count = ((LongBlock) countUncast).asVector(); - assert xVal.getPositionCount() == xDel.getPositionCount() && xVal.getPositionCount() == yVal.getPositionCount() && xVal.getPositionCount() == yDel.getPositionCount() && xVal.getPositionCount() == count.getPositionCount(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - SpatialCentroidGeoPointDocValuesAggregator.combineIntermediate(state, groupId, xVal.getDouble(groupPosition + positionOffset), xDel.getDouble(groupPosition + positionOffset), yVal.getDouble(groupPosition + positionOffset), yDel.getDouble(groupPosition + positionOffset), count.getLong(groupPosition + positionOffset)); - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - CentroidPointAggregator.GroupingCentroidState inState = ((SpatialCentroidGeoPointDocValuesGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - SpatialCentroidGeoPointDocValuesAggregator.combineStates(state, groupId, inState, position); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = SpatialCentroidGeoPointDocValuesAggregator.evaluateFinal(state, selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidGeoPointSourceValuesAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidGeoPointSourceValuesAggregatorFunction.java deleted file mode 100644 index be7b8d9758d1c..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidGeoPointSourceValuesAggregatorFunction.java +++ /dev/null @@ -1,157 +0,0 @@ -// 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.aggregation.spatial; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.aggregation.AggregatorFunction; -import org.elasticsearch.compute.aggregation.IntermediateStateDesc; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunction} implementation for {@link SpatialCentroidGeoPointSourceValuesAggregator}. - * This class is generated. Do not edit it. - */ -public final class SpatialCentroidGeoPointSourceValuesAggregatorFunction implements AggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("xVal", ElementType.DOUBLE), - new IntermediateStateDesc("xDel", ElementType.DOUBLE), - new IntermediateStateDesc("yVal", ElementType.DOUBLE), - new IntermediateStateDesc("yDel", ElementType.DOUBLE), - new IntermediateStateDesc("count", ElementType.LONG) ); - - private final DriverContext driverContext; - - private final CentroidPointAggregator.CentroidState state; - - private final List channels; - - public SpatialCentroidGeoPointSourceValuesAggregatorFunction(DriverContext driverContext, - List channels, CentroidPointAggregator.CentroidState state) { - this.driverContext = driverContext; - this.channels = channels; - this.state = state; - } - - public static SpatialCentroidGeoPointSourceValuesAggregatorFunction create( - DriverContext driverContext, List channels) { - return new SpatialCentroidGeoPointSourceValuesAggregatorFunction(driverContext, channels, SpatialCentroidGeoPointSourceValuesAggregator.initSingle()); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public void addRawInput(Page page) { - BytesRefBlock block = page.getBlock(channels.get(0)); - BytesRefVector vector = block.asVector(); - if (vector != null) { - addRawVector(vector); - } else { - addRawBlock(block); - } - } - - private void addRawVector(BytesRefVector vector) { - BytesRef scratch = new BytesRef(); - for (int i = 0; i < vector.getPositionCount(); i++) { - SpatialCentroidGeoPointSourceValuesAggregator.combine(state, vector.getBytesRef(i, scratch)); - } - } - - private void addRawBlock(BytesRefBlock block) { - BytesRef scratch = new BytesRef(); - for (int p = 0; p < block.getPositionCount(); p++) { - if (block.isNull(p)) { - continue; - } - int start = block.getFirstValueIndex(p); - int end = start + block.getValueCount(p); - for (int i = start; i < end; i++) { - SpatialCentroidGeoPointSourceValuesAggregator.combine(state, block.getBytesRef(i, scratch)); - } - } - } - - @Override - public void addIntermediateInput(Page page) { - assert channels.size() == intermediateBlockCount(); - assert page.getBlockCount() >= channels.get(0) + intermediateStateDesc().size(); - Block xValUncast = page.getBlock(channels.get(0)); - if (xValUncast.areAllValuesNull()) { - return; - } - DoubleVector xVal = ((DoubleBlock) xValUncast).asVector(); - assert xVal.getPositionCount() == 1; - Block xDelUncast = page.getBlock(channels.get(1)); - if (xDelUncast.areAllValuesNull()) { - return; - } - DoubleVector xDel = ((DoubleBlock) xDelUncast).asVector(); - assert xDel.getPositionCount() == 1; - Block yValUncast = page.getBlock(channels.get(2)); - if (yValUncast.areAllValuesNull()) { - return; - } - DoubleVector yVal = ((DoubleBlock) yValUncast).asVector(); - assert yVal.getPositionCount() == 1; - Block yDelUncast = page.getBlock(channels.get(3)); - if (yDelUncast.areAllValuesNull()) { - return; - } - DoubleVector yDel = ((DoubleBlock) yDelUncast).asVector(); - assert yDel.getPositionCount() == 1; - Block countUncast = page.getBlock(channels.get(4)); - if (countUncast.areAllValuesNull()) { - return; - } - LongVector count = ((LongBlock) countUncast).asVector(); - assert count.getPositionCount() == 1; - SpatialCentroidGeoPointSourceValuesAggregator.combineIntermediate(state, xVal.getDouble(0), xDel.getDouble(0), yVal.getDouble(0), yDel.getDouble(0), count.getLong(0)); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, DriverContext driverContext) { - state.toIntermediate(blocks, offset, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, DriverContext driverContext) { - blocks[offset] = SpatialCentroidGeoPointSourceValuesAggregator.evaluateFinal(state, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidGeoPointSourceValuesAggregatorFunctionSupplier.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidGeoPointSourceValuesAggregatorFunctionSupplier.java deleted file mode 100644 index 02b975f03890f..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidGeoPointSourceValuesAggregatorFunctionSupplier.java +++ /dev/null @@ -1,41 +0,0 @@ -// 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.aggregation.spatial; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.util.List; -import org.elasticsearch.compute.aggregation.AggregatorFunctionSupplier; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link AggregatorFunctionSupplier} implementation for {@link SpatialCentroidGeoPointSourceValuesAggregator}. - * This class is generated. Do not edit it. - */ -public final class SpatialCentroidGeoPointSourceValuesAggregatorFunctionSupplier implements AggregatorFunctionSupplier { - private final List channels; - - public SpatialCentroidGeoPointSourceValuesAggregatorFunctionSupplier(List channels) { - this.channels = channels; - } - - @Override - public SpatialCentroidGeoPointSourceValuesAggregatorFunction aggregator( - DriverContext driverContext) { - return SpatialCentroidGeoPointSourceValuesAggregatorFunction.create(driverContext, channels); - } - - @Override - public SpatialCentroidGeoPointSourceValuesGroupingAggregatorFunction groupingAggregator( - DriverContext driverContext) { - return SpatialCentroidGeoPointSourceValuesGroupingAggregatorFunction.create(channels, driverContext); - } - - @Override - public String describe() { - return "spatial_centroid_geo_point_source of valuess"; - } -} diff --git a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidGeoPointSourceValuesGroupingAggregatorFunction.java b/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidGeoPointSourceValuesGroupingAggregatorFunction.java deleted file mode 100644 index 075f8749503b8..0000000000000 --- a/x-pack/plugin/esql/compute/src/main/generated/org/elasticsearch/compute/aggregation/spatial/SpatialCentroidGeoPointSourceValuesGroupingAggregatorFunction.java +++ /dev/null @@ -1,232 +0,0 @@ -// 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.aggregation.spatial; - -import java.lang.Integer; -import java.lang.Override; -import java.lang.String; -import java.lang.StringBuilder; -import java.util.List; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.aggregation.GroupingAggregatorFunction; -import org.elasticsearch.compute.aggregation.IntermediateStateDesc; -import org.elasticsearch.compute.aggregation.SeenGroupIds; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.ElementType; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; - -/** - * {@link GroupingAggregatorFunction} implementation for {@link SpatialCentroidGeoPointSourceValuesAggregator}. - * This class is generated. Do not edit it. - */ -public final class SpatialCentroidGeoPointSourceValuesGroupingAggregatorFunction implements GroupingAggregatorFunction { - private static final List INTERMEDIATE_STATE_DESC = List.of( - new IntermediateStateDesc("xVal", ElementType.DOUBLE), - new IntermediateStateDesc("xDel", ElementType.DOUBLE), - new IntermediateStateDesc("yVal", ElementType.DOUBLE), - new IntermediateStateDesc("yDel", ElementType.DOUBLE), - new IntermediateStateDesc("count", ElementType.LONG) ); - - private final CentroidPointAggregator.GroupingCentroidState state; - - private final List channels; - - private final DriverContext driverContext; - - public SpatialCentroidGeoPointSourceValuesGroupingAggregatorFunction(List channels, - CentroidPointAggregator.GroupingCentroidState state, DriverContext driverContext) { - this.channels = channels; - this.state = state; - this.driverContext = driverContext; - } - - public static SpatialCentroidGeoPointSourceValuesGroupingAggregatorFunction create( - List channels, DriverContext driverContext) { - return new SpatialCentroidGeoPointSourceValuesGroupingAggregatorFunction(channels, SpatialCentroidGeoPointSourceValuesAggregator.initGrouping(driverContext.bigArrays()), driverContext); - } - - public static List intermediateStateDesc() { - return INTERMEDIATE_STATE_DESC; - } - - @Override - public int intermediateBlockCount() { - return INTERMEDIATE_STATE_DESC.size(); - } - - @Override - public GroupingAggregatorFunction.AddInput prepareProcessPage(SeenGroupIds seenGroupIds, - Page page) { - BytesRefBlock valuesBlock = page.getBlock(channels.get(0)); - BytesRefVector valuesVector = valuesBlock.asVector(); - if (valuesVector == null) { - if (valuesBlock.mayHaveNulls()) { - state.enableGroupIdTracking(seenGroupIds); - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesBlock); - } - }; - } - return new GroupingAggregatorFunction.AddInput() { - @Override - public void add(int positionOffset, IntBlock groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - - @Override - public void add(int positionOffset, IntVector groupIds) { - addRawInput(positionOffset, groupIds, valuesVector); - } - }; - } - - private void addRawInput(int positionOffset, IntVector groups, BytesRefBlock values) { - BytesRef scratch = new BytesRef(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - SpatialCentroidGeoPointSourceValuesAggregator.combine(state, groupId, values.getBytesRef(v, scratch)); - } - } - } - - private void addRawInput(int positionOffset, IntVector groups, BytesRefVector values) { - BytesRef scratch = new BytesRef(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - SpatialCentroidGeoPointSourceValuesAggregator.combine(state, groupId, values.getBytesRef(groupPosition + positionOffset, scratch)); - } - } - - private void addRawInput(int positionOffset, IntBlock groups, BytesRefBlock values) { - BytesRef scratch = new BytesRef(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - if (values.isNull(groupPosition + positionOffset)) { - continue; - } - int valuesStart = values.getFirstValueIndex(groupPosition + positionOffset); - int valuesEnd = valuesStart + values.getValueCount(groupPosition + positionOffset); - for (int v = valuesStart; v < valuesEnd; v++) { - SpatialCentroidGeoPointSourceValuesAggregator.combine(state, groupId, values.getBytesRef(v, scratch)); - } - } - } - } - - private void addRawInput(int positionOffset, IntBlock groups, BytesRefVector values) { - BytesRef scratch = new BytesRef(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - if (groups.isNull(groupPosition)) { - continue; - } - int groupStart = groups.getFirstValueIndex(groupPosition); - int groupEnd = groupStart + groups.getValueCount(groupPosition); - for (int g = groupStart; g < groupEnd; g++) { - int groupId = Math.toIntExact(groups.getInt(g)); - SpatialCentroidGeoPointSourceValuesAggregator.combine(state, groupId, values.getBytesRef(groupPosition + positionOffset, scratch)); - } - } - } - - @Override - public void addIntermediateInput(int positionOffset, IntVector groups, Page page) { - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - assert channels.size() == intermediateBlockCount(); - Block xValUncast = page.getBlock(channels.get(0)); - if (xValUncast.areAllValuesNull()) { - return; - } - DoubleVector xVal = ((DoubleBlock) xValUncast).asVector(); - Block xDelUncast = page.getBlock(channels.get(1)); - if (xDelUncast.areAllValuesNull()) { - return; - } - DoubleVector xDel = ((DoubleBlock) xDelUncast).asVector(); - Block yValUncast = page.getBlock(channels.get(2)); - if (yValUncast.areAllValuesNull()) { - return; - } - DoubleVector yVal = ((DoubleBlock) yValUncast).asVector(); - Block yDelUncast = page.getBlock(channels.get(3)); - if (yDelUncast.areAllValuesNull()) { - return; - } - DoubleVector yDel = ((DoubleBlock) yDelUncast).asVector(); - Block countUncast = page.getBlock(channels.get(4)); - if (countUncast.areAllValuesNull()) { - return; - } - LongVector count = ((LongBlock) countUncast).asVector(); - assert xVal.getPositionCount() == xDel.getPositionCount() && xVal.getPositionCount() == yVal.getPositionCount() && xVal.getPositionCount() == yDel.getPositionCount() && xVal.getPositionCount() == count.getPositionCount(); - for (int groupPosition = 0; groupPosition < groups.getPositionCount(); groupPosition++) { - int groupId = Math.toIntExact(groups.getInt(groupPosition)); - SpatialCentroidGeoPointSourceValuesAggregator.combineIntermediate(state, groupId, xVal.getDouble(groupPosition + positionOffset), xDel.getDouble(groupPosition + positionOffset), yVal.getDouble(groupPosition + positionOffset), yDel.getDouble(groupPosition + positionOffset), count.getLong(groupPosition + positionOffset)); - } - } - - @Override - public void addIntermediateRowInput(int groupId, GroupingAggregatorFunction input, int position) { - if (input.getClass() != getClass()) { - throw new IllegalArgumentException("expected " + getClass() + "; got " + input.getClass()); - } - CentroidPointAggregator.GroupingCentroidState inState = ((SpatialCentroidGeoPointSourceValuesGroupingAggregatorFunction) input).state; - state.enableGroupIdTracking(new SeenGroupIds.Empty()); - SpatialCentroidGeoPointSourceValuesAggregator.combineStates(state, groupId, inState, position); - } - - @Override - public void evaluateIntermediate(Block[] blocks, int offset, IntVector selected) { - state.toIntermediate(blocks, offset, selected, driverContext); - } - - @Override - public void evaluateFinal(Block[] blocks, int offset, IntVector selected, - DriverContext driverContext) { - blocks[offset] = SpatialCentroidGeoPointSourceValuesAggregator.evaluateFinal(state, selected, driverContext); - } - - @Override - public String toString() { - StringBuilder sb = new StringBuilder(); - sb.append(getClass().getSimpleName()).append("["); - sb.append("channels=").append(channels); - sb.append("]"); - return sb.toString(); - } - - @Override - public void close() { - state.close(); - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/EqualsBoolsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/EqualsBoolsEvaluator.java deleted file mode 100644 index ef26fb4d000dc..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/EqualsBoolsEvaluator.java +++ /dev/null @@ -1,132 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.comparison; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Equals}. - * This class is generated. Do not edit it. - */ -public final class EqualsBoolsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public EqualsBoolsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BooleanBlock lhsBlock = (BooleanBlock) lhs.eval(page)) { - try (BooleanBlock rhsBlock = (BooleanBlock) rhs.eval(page)) { - BooleanVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - BooleanVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, BooleanBlock lhsBlock, BooleanBlock rhsBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(Equals.processBools(lhsBlock.getBoolean(lhsBlock.getFirstValueIndex(p)), rhsBlock.getBoolean(rhsBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, BooleanVector lhsVector, BooleanVector rhsVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(Equals.processBools(lhsVector.getBoolean(p), rhsVector.getBoolean(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "EqualsBoolsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public EqualsBoolsEvaluator get(DriverContext context) { - return new EqualsBoolsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "EqualsBoolsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/EqualsDoublesEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/EqualsDoublesEvaluator.java deleted file mode 100644 index d5b2e84384a03..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/EqualsDoublesEvaluator.java +++ /dev/null @@ -1,134 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.comparison; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Equals}. - * This class is generated. Do not edit it. - */ -public final class EqualsDoublesEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public EqualsDoublesEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock lhsBlock = (DoubleBlock) lhs.eval(page)) { - try (DoubleBlock rhsBlock = (DoubleBlock) rhs.eval(page)) { - DoubleVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - DoubleVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, DoubleBlock lhsBlock, DoubleBlock rhsBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(Equals.processDoubles(lhsBlock.getDouble(lhsBlock.getFirstValueIndex(p)), rhsBlock.getDouble(rhsBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, DoubleVector lhsVector, DoubleVector rhsVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(Equals.processDoubles(lhsVector.getDouble(p), rhsVector.getDouble(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "EqualsDoublesEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public EqualsDoublesEvaluator get(DriverContext context) { - return new EqualsDoublesEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "EqualsDoublesEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/EqualsGeometriesEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/EqualsGeometriesEvaluator.java deleted file mode 100644 index 025cca53ceab0..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/EqualsGeometriesEvaluator.java +++ /dev/null @@ -1,139 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.comparison; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Equals}. - * This class is generated. Do not edit it. - */ -public final class EqualsGeometriesEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public EqualsGeometriesEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock lhsBlock = (BytesRefBlock) lhs.eval(page)) { - try (BytesRefBlock rhsBlock = (BytesRefBlock) rhs.eval(page)) { - BytesRefVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - BytesRefVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, BytesRefBlock lhsBlock, BytesRefBlock rhsBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef lhsScratch = new BytesRef(); - BytesRef rhsScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(Equals.processGeometries(lhsBlock.getBytesRef(lhsBlock.getFirstValueIndex(p), lhsScratch), rhsBlock.getBytesRef(rhsBlock.getFirstValueIndex(p), rhsScratch))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, BytesRefVector lhsVector, BytesRefVector rhsVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - BytesRef lhsScratch = new BytesRef(); - BytesRef rhsScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(Equals.processGeometries(lhsVector.getBytesRef(p, lhsScratch), rhsVector.getBytesRef(p, rhsScratch))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "EqualsGeometriesEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public EqualsGeometriesEvaluator get(DriverContext context) { - return new EqualsGeometriesEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "EqualsGeometriesEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/EqualsIntsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/EqualsIntsEvaluator.java deleted file mode 100644 index c2c9c7ce2b19c..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/EqualsIntsEvaluator.java +++ /dev/null @@ -1,134 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.comparison; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Equals}. - * This class is generated. Do not edit it. - */ -public final class EqualsIntsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public EqualsIntsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (IntBlock lhsBlock = (IntBlock) lhs.eval(page)) { - try (IntBlock rhsBlock = (IntBlock) rhs.eval(page)) { - IntVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - IntVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, IntBlock lhsBlock, IntBlock rhsBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(Equals.processInts(lhsBlock.getInt(lhsBlock.getFirstValueIndex(p)), rhsBlock.getInt(rhsBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, IntVector lhsVector, IntVector rhsVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(Equals.processInts(lhsVector.getInt(p), rhsVector.getInt(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "EqualsIntsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public EqualsIntsEvaluator get(DriverContext context) { - return new EqualsIntsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "EqualsIntsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/EqualsKeywordsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/EqualsKeywordsEvaluator.java deleted file mode 100644 index 8dc15ba6d2fec..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/EqualsKeywordsEvaluator.java +++ /dev/null @@ -1,139 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.comparison; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Equals}. - * This class is generated. Do not edit it. - */ -public final class EqualsKeywordsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public EqualsKeywordsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock lhsBlock = (BytesRefBlock) lhs.eval(page)) { - try (BytesRefBlock rhsBlock = (BytesRefBlock) rhs.eval(page)) { - BytesRefVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - BytesRefVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, BytesRefBlock lhsBlock, BytesRefBlock rhsBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef lhsScratch = new BytesRef(); - BytesRef rhsScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(Equals.processKeywords(lhsBlock.getBytesRef(lhsBlock.getFirstValueIndex(p), lhsScratch), rhsBlock.getBytesRef(rhsBlock.getFirstValueIndex(p), rhsScratch))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, BytesRefVector lhsVector, BytesRefVector rhsVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - BytesRef lhsScratch = new BytesRef(); - BytesRef rhsScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(Equals.processKeywords(lhsVector.getBytesRef(p, lhsScratch), rhsVector.getBytesRef(p, rhsScratch))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "EqualsKeywordsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public EqualsKeywordsEvaluator get(DriverContext context) { - return new EqualsKeywordsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "EqualsKeywordsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/EqualsLongsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/EqualsLongsEvaluator.java deleted file mode 100644 index 870d7c546010f..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/EqualsLongsEvaluator.java +++ /dev/null @@ -1,134 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.comparison; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Equals}. - * This class is generated. Do not edit it. - */ -public final class EqualsLongsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public EqualsLongsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock lhsBlock = (LongBlock) lhs.eval(page)) { - try (LongBlock rhsBlock = (LongBlock) rhs.eval(page)) { - LongVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - LongVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, LongBlock lhsBlock, LongBlock rhsBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(Equals.processLongs(lhsBlock.getLong(lhsBlock.getFirstValueIndex(p)), rhsBlock.getLong(rhsBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, LongVector lhsVector, LongVector rhsVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(Equals.processLongs(lhsVector.getLong(p), rhsVector.getLong(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "EqualsLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public EqualsLongsEvaluator get(DriverContext context) { - return new EqualsLongsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "EqualsLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/GreaterThanDoublesEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/GreaterThanDoublesEvaluator.java deleted file mode 100644 index 051df8053417f..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/GreaterThanDoublesEvaluator.java +++ /dev/null @@ -1,134 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.comparison; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link GreaterThan}. - * This class is generated. Do not edit it. - */ -public final class GreaterThanDoublesEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public GreaterThanDoublesEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock lhsBlock = (DoubleBlock) lhs.eval(page)) { - try (DoubleBlock rhsBlock = (DoubleBlock) rhs.eval(page)) { - DoubleVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - DoubleVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, DoubleBlock lhsBlock, DoubleBlock rhsBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(GreaterThan.processDoubles(lhsBlock.getDouble(lhsBlock.getFirstValueIndex(p)), rhsBlock.getDouble(rhsBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, DoubleVector lhsVector, DoubleVector rhsVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(GreaterThan.processDoubles(lhsVector.getDouble(p), rhsVector.getDouble(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "GreaterThanDoublesEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public GreaterThanDoublesEvaluator get(DriverContext context) { - return new GreaterThanDoublesEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "GreaterThanDoublesEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/GreaterThanIntsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/GreaterThanIntsEvaluator.java deleted file mode 100644 index c6de582ef2909..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/GreaterThanIntsEvaluator.java +++ /dev/null @@ -1,134 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.comparison; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link GreaterThan}. - * This class is generated. Do not edit it. - */ -public final class GreaterThanIntsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public GreaterThanIntsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (IntBlock lhsBlock = (IntBlock) lhs.eval(page)) { - try (IntBlock rhsBlock = (IntBlock) rhs.eval(page)) { - IntVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - IntVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, IntBlock lhsBlock, IntBlock rhsBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(GreaterThan.processInts(lhsBlock.getInt(lhsBlock.getFirstValueIndex(p)), rhsBlock.getInt(rhsBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, IntVector lhsVector, IntVector rhsVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(GreaterThan.processInts(lhsVector.getInt(p), rhsVector.getInt(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "GreaterThanIntsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public GreaterThanIntsEvaluator get(DriverContext context) { - return new GreaterThanIntsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "GreaterThanIntsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/GreaterThanKeywordsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/GreaterThanKeywordsEvaluator.java deleted file mode 100644 index cf243b68e473c..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/GreaterThanKeywordsEvaluator.java +++ /dev/null @@ -1,139 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.comparison; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link GreaterThan}. - * This class is generated. Do not edit it. - */ -public final class GreaterThanKeywordsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public GreaterThanKeywordsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock lhsBlock = (BytesRefBlock) lhs.eval(page)) { - try (BytesRefBlock rhsBlock = (BytesRefBlock) rhs.eval(page)) { - BytesRefVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - BytesRefVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, BytesRefBlock lhsBlock, BytesRefBlock rhsBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef lhsScratch = new BytesRef(); - BytesRef rhsScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(GreaterThan.processKeywords(lhsBlock.getBytesRef(lhsBlock.getFirstValueIndex(p), lhsScratch), rhsBlock.getBytesRef(rhsBlock.getFirstValueIndex(p), rhsScratch))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, BytesRefVector lhsVector, BytesRefVector rhsVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - BytesRef lhsScratch = new BytesRef(); - BytesRef rhsScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(GreaterThan.processKeywords(lhsVector.getBytesRef(p, lhsScratch), rhsVector.getBytesRef(p, rhsScratch))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "GreaterThanKeywordsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public GreaterThanKeywordsEvaluator get(DriverContext context) { - return new GreaterThanKeywordsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "GreaterThanKeywordsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/GreaterThanLongsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/GreaterThanLongsEvaluator.java deleted file mode 100644 index 5f1a679c76a31..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/GreaterThanLongsEvaluator.java +++ /dev/null @@ -1,134 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.comparison; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link GreaterThan}. - * This class is generated. Do not edit it. - */ -public final class GreaterThanLongsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public GreaterThanLongsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock lhsBlock = (LongBlock) lhs.eval(page)) { - try (LongBlock rhsBlock = (LongBlock) rhs.eval(page)) { - LongVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - LongVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, LongBlock lhsBlock, LongBlock rhsBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(GreaterThan.processLongs(lhsBlock.getLong(lhsBlock.getFirstValueIndex(p)), rhsBlock.getLong(rhsBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, LongVector lhsVector, LongVector rhsVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(GreaterThan.processLongs(lhsVector.getLong(p), rhsVector.getLong(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "GreaterThanLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public GreaterThanLongsEvaluator get(DriverContext context) { - return new GreaterThanLongsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "GreaterThanLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/GreaterThanOrEqualDoublesEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/GreaterThanOrEqualDoublesEvaluator.java deleted file mode 100644 index c36031c321422..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/GreaterThanOrEqualDoublesEvaluator.java +++ /dev/null @@ -1,134 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.comparison; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link GreaterThanOrEqual}. - * This class is generated. Do not edit it. - */ -public final class GreaterThanOrEqualDoublesEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public GreaterThanOrEqualDoublesEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock lhsBlock = (DoubleBlock) lhs.eval(page)) { - try (DoubleBlock rhsBlock = (DoubleBlock) rhs.eval(page)) { - DoubleVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - DoubleVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, DoubleBlock lhsBlock, DoubleBlock rhsBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(GreaterThanOrEqual.processDoubles(lhsBlock.getDouble(lhsBlock.getFirstValueIndex(p)), rhsBlock.getDouble(rhsBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, DoubleVector lhsVector, DoubleVector rhsVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(GreaterThanOrEqual.processDoubles(lhsVector.getDouble(p), rhsVector.getDouble(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "GreaterThanOrEqualDoublesEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public GreaterThanOrEqualDoublesEvaluator get(DriverContext context) { - return new GreaterThanOrEqualDoublesEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "GreaterThanOrEqualDoublesEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/GreaterThanOrEqualIntsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/GreaterThanOrEqualIntsEvaluator.java deleted file mode 100644 index 2b64cfcf9ea49..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/GreaterThanOrEqualIntsEvaluator.java +++ /dev/null @@ -1,134 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.comparison; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link GreaterThanOrEqual}. - * This class is generated. Do not edit it. - */ -public final class GreaterThanOrEqualIntsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public GreaterThanOrEqualIntsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (IntBlock lhsBlock = (IntBlock) lhs.eval(page)) { - try (IntBlock rhsBlock = (IntBlock) rhs.eval(page)) { - IntVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - IntVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, IntBlock lhsBlock, IntBlock rhsBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(GreaterThanOrEqual.processInts(lhsBlock.getInt(lhsBlock.getFirstValueIndex(p)), rhsBlock.getInt(rhsBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, IntVector lhsVector, IntVector rhsVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(GreaterThanOrEqual.processInts(lhsVector.getInt(p), rhsVector.getInt(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "GreaterThanOrEqualIntsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public GreaterThanOrEqualIntsEvaluator get(DriverContext context) { - return new GreaterThanOrEqualIntsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "GreaterThanOrEqualIntsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/GreaterThanOrEqualKeywordsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/GreaterThanOrEqualKeywordsEvaluator.java deleted file mode 100644 index b8b2c9b6d4459..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/GreaterThanOrEqualKeywordsEvaluator.java +++ /dev/null @@ -1,139 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.comparison; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link GreaterThanOrEqual}. - * This class is generated. Do not edit it. - */ -public final class GreaterThanOrEqualKeywordsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public GreaterThanOrEqualKeywordsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock lhsBlock = (BytesRefBlock) lhs.eval(page)) { - try (BytesRefBlock rhsBlock = (BytesRefBlock) rhs.eval(page)) { - BytesRefVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - BytesRefVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, BytesRefBlock lhsBlock, BytesRefBlock rhsBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef lhsScratch = new BytesRef(); - BytesRef rhsScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(GreaterThanOrEqual.processKeywords(lhsBlock.getBytesRef(lhsBlock.getFirstValueIndex(p), lhsScratch), rhsBlock.getBytesRef(rhsBlock.getFirstValueIndex(p), rhsScratch))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, BytesRefVector lhsVector, BytesRefVector rhsVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - BytesRef lhsScratch = new BytesRef(); - BytesRef rhsScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(GreaterThanOrEqual.processKeywords(lhsVector.getBytesRef(p, lhsScratch), rhsVector.getBytesRef(p, rhsScratch))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "GreaterThanOrEqualKeywordsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public GreaterThanOrEqualKeywordsEvaluator get(DriverContext context) { - return new GreaterThanOrEqualKeywordsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "GreaterThanOrEqualKeywordsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/GreaterThanOrEqualLongsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/GreaterThanOrEqualLongsEvaluator.java deleted file mode 100644 index 907a29c8c904d..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/GreaterThanOrEqualLongsEvaluator.java +++ /dev/null @@ -1,134 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.comparison; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link GreaterThanOrEqual}. - * This class is generated. Do not edit it. - */ -public final class GreaterThanOrEqualLongsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public GreaterThanOrEqualLongsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock lhsBlock = (LongBlock) lhs.eval(page)) { - try (LongBlock rhsBlock = (LongBlock) rhs.eval(page)) { - LongVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - LongVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, LongBlock lhsBlock, LongBlock rhsBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(GreaterThanOrEqual.processLongs(lhsBlock.getLong(lhsBlock.getFirstValueIndex(p)), rhsBlock.getLong(rhsBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, LongVector lhsVector, LongVector rhsVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(GreaterThanOrEqual.processLongs(lhsVector.getLong(p), rhsVector.getLong(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "GreaterThanOrEqualLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public GreaterThanOrEqualLongsEvaluator get(DriverContext context) { - return new GreaterThanOrEqualLongsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "GreaterThanOrEqualLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/InsensitiveEqualsConstantEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/InsensitiveEqualsConstantEvaluator.java deleted file mode 100644 index 9417a2374ac12..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/InsensitiveEqualsConstantEvaluator.java +++ /dev/null @@ -1,121 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.comparison; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.apache.lucene.util.automaton.ByteRunAutomaton; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link InsensitiveEquals}. - * This class is generated. Do not edit it. - */ -public final class InsensitiveEqualsConstantEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final ByteRunAutomaton rhs; - - private final DriverContext driverContext; - - public InsensitiveEqualsConstantEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - ByteRunAutomaton rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock lhsBlock = (BytesRefBlock) lhs.eval(page)) { - BytesRefVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock); - } - return eval(page.getPositionCount(), lhsVector).asBlock(); - } - } - - public BooleanBlock eval(int positionCount, BytesRefBlock lhsBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef lhsScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(InsensitiveEquals.processConstant(lhsBlock.getBytesRef(lhsBlock.getFirstValueIndex(p), lhsScratch), rhs)); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, BytesRefVector lhsVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - BytesRef lhsScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(InsensitiveEquals.processConstant(lhsVector.getBytesRef(p, lhsScratch), rhs)); - } - return result.build(); - } - } - - @Override - public String toString() { - return "InsensitiveEqualsConstantEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final ByteRunAutomaton rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - ByteRunAutomaton rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public InsensitiveEqualsConstantEvaluator get(DriverContext context) { - return new InsensitiveEqualsConstantEvaluator(source, lhs.get(context), rhs, context); - } - - @Override - public String toString() { - return "InsensitiveEqualsConstantEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/InsensitiveEqualsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/InsensitiveEqualsEvaluator.java deleted file mode 100644 index 8832a77928aa3..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/InsensitiveEqualsEvaluator.java +++ /dev/null @@ -1,139 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.comparison; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link InsensitiveEquals}. - * This class is generated. Do not edit it. - */ -public final class InsensitiveEqualsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public InsensitiveEqualsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock lhsBlock = (BytesRefBlock) lhs.eval(page)) { - try (BytesRefBlock rhsBlock = (BytesRefBlock) rhs.eval(page)) { - BytesRefVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - BytesRefVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, BytesRefBlock lhsBlock, BytesRefBlock rhsBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef lhsScratch = new BytesRef(); - BytesRef rhsScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(InsensitiveEquals.process(lhsBlock.getBytesRef(lhsBlock.getFirstValueIndex(p), lhsScratch), rhsBlock.getBytesRef(rhsBlock.getFirstValueIndex(p), rhsScratch))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, BytesRefVector lhsVector, BytesRefVector rhsVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - BytesRef lhsScratch = new BytesRef(); - BytesRef rhsScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(InsensitiveEquals.process(lhsVector.getBytesRef(p, lhsScratch), rhsVector.getBytesRef(p, rhsScratch))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "InsensitiveEqualsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public InsensitiveEqualsEvaluator get(DriverContext context) { - return new InsensitiveEqualsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "InsensitiveEqualsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/LessThanDoublesEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/LessThanDoublesEvaluator.java deleted file mode 100644 index c3cf8293071e3..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/LessThanDoublesEvaluator.java +++ /dev/null @@ -1,134 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.comparison; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link LessThan}. - * This class is generated. Do not edit it. - */ -public final class LessThanDoublesEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public LessThanDoublesEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock lhsBlock = (DoubleBlock) lhs.eval(page)) { - try (DoubleBlock rhsBlock = (DoubleBlock) rhs.eval(page)) { - DoubleVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - DoubleVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, DoubleBlock lhsBlock, DoubleBlock rhsBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(LessThan.processDoubles(lhsBlock.getDouble(lhsBlock.getFirstValueIndex(p)), rhsBlock.getDouble(rhsBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, DoubleVector lhsVector, DoubleVector rhsVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(LessThan.processDoubles(lhsVector.getDouble(p), rhsVector.getDouble(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "LessThanDoublesEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public LessThanDoublesEvaluator get(DriverContext context) { - return new LessThanDoublesEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "LessThanDoublesEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/LessThanIntsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/LessThanIntsEvaluator.java deleted file mode 100644 index a66ac0e889090..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/LessThanIntsEvaluator.java +++ /dev/null @@ -1,134 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.comparison; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link LessThan}. - * This class is generated. Do not edit it. - */ -public final class LessThanIntsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public LessThanIntsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (IntBlock lhsBlock = (IntBlock) lhs.eval(page)) { - try (IntBlock rhsBlock = (IntBlock) rhs.eval(page)) { - IntVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - IntVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, IntBlock lhsBlock, IntBlock rhsBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(LessThan.processInts(lhsBlock.getInt(lhsBlock.getFirstValueIndex(p)), rhsBlock.getInt(rhsBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, IntVector lhsVector, IntVector rhsVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(LessThan.processInts(lhsVector.getInt(p), rhsVector.getInt(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "LessThanIntsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public LessThanIntsEvaluator get(DriverContext context) { - return new LessThanIntsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "LessThanIntsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/LessThanKeywordsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/LessThanKeywordsEvaluator.java deleted file mode 100644 index a0951d9a09382..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/LessThanKeywordsEvaluator.java +++ /dev/null @@ -1,139 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.comparison; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link LessThan}. - * This class is generated. Do not edit it. - */ -public final class LessThanKeywordsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public LessThanKeywordsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock lhsBlock = (BytesRefBlock) lhs.eval(page)) { - try (BytesRefBlock rhsBlock = (BytesRefBlock) rhs.eval(page)) { - BytesRefVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - BytesRefVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, BytesRefBlock lhsBlock, BytesRefBlock rhsBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef lhsScratch = new BytesRef(); - BytesRef rhsScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(LessThan.processKeywords(lhsBlock.getBytesRef(lhsBlock.getFirstValueIndex(p), lhsScratch), rhsBlock.getBytesRef(rhsBlock.getFirstValueIndex(p), rhsScratch))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, BytesRefVector lhsVector, BytesRefVector rhsVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - BytesRef lhsScratch = new BytesRef(); - BytesRef rhsScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(LessThan.processKeywords(lhsVector.getBytesRef(p, lhsScratch), rhsVector.getBytesRef(p, rhsScratch))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "LessThanKeywordsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public LessThanKeywordsEvaluator get(DriverContext context) { - return new LessThanKeywordsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "LessThanKeywordsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/LessThanLongsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/LessThanLongsEvaluator.java deleted file mode 100644 index f0e7ac134410b..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/LessThanLongsEvaluator.java +++ /dev/null @@ -1,134 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.comparison; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link LessThan}. - * This class is generated. Do not edit it. - */ -public final class LessThanLongsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public LessThanLongsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock lhsBlock = (LongBlock) lhs.eval(page)) { - try (LongBlock rhsBlock = (LongBlock) rhs.eval(page)) { - LongVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - LongVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, LongBlock lhsBlock, LongBlock rhsBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(LessThan.processLongs(lhsBlock.getLong(lhsBlock.getFirstValueIndex(p)), rhsBlock.getLong(rhsBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, LongVector lhsVector, LongVector rhsVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(LessThan.processLongs(lhsVector.getLong(p), rhsVector.getLong(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "LessThanLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public LessThanLongsEvaluator get(DriverContext context) { - return new LessThanLongsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "LessThanLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/LessThanOrEqualDoublesEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/LessThanOrEqualDoublesEvaluator.java deleted file mode 100644 index cf12098962599..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/LessThanOrEqualDoublesEvaluator.java +++ /dev/null @@ -1,134 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.comparison; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link LessThanOrEqual}. - * This class is generated. Do not edit it. - */ -public final class LessThanOrEqualDoublesEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public LessThanOrEqualDoublesEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock lhsBlock = (DoubleBlock) lhs.eval(page)) { - try (DoubleBlock rhsBlock = (DoubleBlock) rhs.eval(page)) { - DoubleVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - DoubleVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, DoubleBlock lhsBlock, DoubleBlock rhsBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(LessThanOrEqual.processDoubles(lhsBlock.getDouble(lhsBlock.getFirstValueIndex(p)), rhsBlock.getDouble(rhsBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, DoubleVector lhsVector, DoubleVector rhsVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(LessThanOrEqual.processDoubles(lhsVector.getDouble(p), rhsVector.getDouble(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "LessThanOrEqualDoublesEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public LessThanOrEqualDoublesEvaluator get(DriverContext context) { - return new LessThanOrEqualDoublesEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "LessThanOrEqualDoublesEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/LessThanOrEqualIntsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/LessThanOrEqualIntsEvaluator.java deleted file mode 100644 index ffa8ab38bc2eb..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/LessThanOrEqualIntsEvaluator.java +++ /dev/null @@ -1,134 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.comparison; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link LessThanOrEqual}. - * This class is generated. Do not edit it. - */ -public final class LessThanOrEqualIntsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public LessThanOrEqualIntsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (IntBlock lhsBlock = (IntBlock) lhs.eval(page)) { - try (IntBlock rhsBlock = (IntBlock) rhs.eval(page)) { - IntVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - IntVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, IntBlock lhsBlock, IntBlock rhsBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(LessThanOrEqual.processInts(lhsBlock.getInt(lhsBlock.getFirstValueIndex(p)), rhsBlock.getInt(rhsBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, IntVector lhsVector, IntVector rhsVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(LessThanOrEqual.processInts(lhsVector.getInt(p), rhsVector.getInt(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "LessThanOrEqualIntsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public LessThanOrEqualIntsEvaluator get(DriverContext context) { - return new LessThanOrEqualIntsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "LessThanOrEqualIntsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/LessThanOrEqualKeywordsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/LessThanOrEqualKeywordsEvaluator.java deleted file mode 100644 index 2e7aafeb2d805..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/LessThanOrEqualKeywordsEvaluator.java +++ /dev/null @@ -1,139 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.comparison; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link LessThanOrEqual}. - * This class is generated. Do not edit it. - */ -public final class LessThanOrEqualKeywordsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public LessThanOrEqualKeywordsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock lhsBlock = (BytesRefBlock) lhs.eval(page)) { - try (BytesRefBlock rhsBlock = (BytesRefBlock) rhs.eval(page)) { - BytesRefVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - BytesRefVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, BytesRefBlock lhsBlock, BytesRefBlock rhsBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef lhsScratch = new BytesRef(); - BytesRef rhsScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(LessThanOrEqual.processKeywords(lhsBlock.getBytesRef(lhsBlock.getFirstValueIndex(p), lhsScratch), rhsBlock.getBytesRef(rhsBlock.getFirstValueIndex(p), rhsScratch))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, BytesRefVector lhsVector, BytesRefVector rhsVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - BytesRef lhsScratch = new BytesRef(); - BytesRef rhsScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(LessThanOrEqual.processKeywords(lhsVector.getBytesRef(p, lhsScratch), rhsVector.getBytesRef(p, rhsScratch))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "LessThanOrEqualKeywordsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public LessThanOrEqualKeywordsEvaluator get(DriverContext context) { - return new LessThanOrEqualKeywordsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "LessThanOrEqualKeywordsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/LessThanOrEqualLongsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/LessThanOrEqualLongsEvaluator.java deleted file mode 100644 index 9c211610da814..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/LessThanOrEqualLongsEvaluator.java +++ /dev/null @@ -1,134 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.comparison; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link LessThanOrEqual}. - * This class is generated. Do not edit it. - */ -public final class LessThanOrEqualLongsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public LessThanOrEqualLongsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock lhsBlock = (LongBlock) lhs.eval(page)) { - try (LongBlock rhsBlock = (LongBlock) rhs.eval(page)) { - LongVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - LongVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, LongBlock lhsBlock, LongBlock rhsBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(LessThanOrEqual.processLongs(lhsBlock.getLong(lhsBlock.getFirstValueIndex(p)), rhsBlock.getLong(rhsBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, LongVector lhsVector, LongVector rhsVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(LessThanOrEqual.processLongs(lhsVector.getLong(p), rhsVector.getLong(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "LessThanOrEqualLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public LessThanOrEqualLongsEvaluator get(DriverContext context) { - return new LessThanOrEqualLongsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "LessThanOrEqualLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/NotEqualsBoolsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/NotEqualsBoolsEvaluator.java deleted file mode 100644 index 7d2067fe6bdbe..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/NotEqualsBoolsEvaluator.java +++ /dev/null @@ -1,132 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.comparison; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link NotEquals}. - * This class is generated. Do not edit it. - */ -public final class NotEqualsBoolsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public NotEqualsBoolsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BooleanBlock lhsBlock = (BooleanBlock) lhs.eval(page)) { - try (BooleanBlock rhsBlock = (BooleanBlock) rhs.eval(page)) { - BooleanVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - BooleanVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, BooleanBlock lhsBlock, BooleanBlock rhsBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(NotEquals.processBools(lhsBlock.getBoolean(lhsBlock.getFirstValueIndex(p)), rhsBlock.getBoolean(rhsBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, BooleanVector lhsVector, BooleanVector rhsVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(NotEquals.processBools(lhsVector.getBoolean(p), rhsVector.getBoolean(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "NotEqualsBoolsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public NotEqualsBoolsEvaluator get(DriverContext context) { - return new NotEqualsBoolsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "NotEqualsBoolsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/NotEqualsDoublesEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/NotEqualsDoublesEvaluator.java deleted file mode 100644 index 174d3df53853b..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/NotEqualsDoublesEvaluator.java +++ /dev/null @@ -1,134 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.comparison; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link NotEquals}. - * This class is generated. Do not edit it. - */ -public final class NotEqualsDoublesEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public NotEqualsDoublesEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock lhsBlock = (DoubleBlock) lhs.eval(page)) { - try (DoubleBlock rhsBlock = (DoubleBlock) rhs.eval(page)) { - DoubleVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - DoubleVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, DoubleBlock lhsBlock, DoubleBlock rhsBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(NotEquals.processDoubles(lhsBlock.getDouble(lhsBlock.getFirstValueIndex(p)), rhsBlock.getDouble(rhsBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, DoubleVector lhsVector, DoubleVector rhsVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(NotEquals.processDoubles(lhsVector.getDouble(p), rhsVector.getDouble(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "NotEqualsDoublesEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public NotEqualsDoublesEvaluator get(DriverContext context) { - return new NotEqualsDoublesEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "NotEqualsDoublesEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/NotEqualsGeometriesEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/NotEqualsGeometriesEvaluator.java deleted file mode 100644 index d0dd58e86babe..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/NotEqualsGeometriesEvaluator.java +++ /dev/null @@ -1,139 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.comparison; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link NotEquals}. - * This class is generated. Do not edit it. - */ -public final class NotEqualsGeometriesEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public NotEqualsGeometriesEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock lhsBlock = (BytesRefBlock) lhs.eval(page)) { - try (BytesRefBlock rhsBlock = (BytesRefBlock) rhs.eval(page)) { - BytesRefVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - BytesRefVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, BytesRefBlock lhsBlock, BytesRefBlock rhsBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef lhsScratch = new BytesRef(); - BytesRef rhsScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(NotEquals.processGeometries(lhsBlock.getBytesRef(lhsBlock.getFirstValueIndex(p), lhsScratch), rhsBlock.getBytesRef(rhsBlock.getFirstValueIndex(p), rhsScratch))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, BytesRefVector lhsVector, BytesRefVector rhsVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - BytesRef lhsScratch = new BytesRef(); - BytesRef rhsScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(NotEquals.processGeometries(lhsVector.getBytesRef(p, lhsScratch), rhsVector.getBytesRef(p, rhsScratch))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "NotEqualsGeometriesEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public NotEqualsGeometriesEvaluator get(DriverContext context) { - return new NotEqualsGeometriesEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "NotEqualsGeometriesEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/NotEqualsIntsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/NotEqualsIntsEvaluator.java deleted file mode 100644 index 03abc111d820e..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/NotEqualsIntsEvaluator.java +++ /dev/null @@ -1,134 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.comparison; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link NotEquals}. - * This class is generated. Do not edit it. - */ -public final class NotEqualsIntsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public NotEqualsIntsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (IntBlock lhsBlock = (IntBlock) lhs.eval(page)) { - try (IntBlock rhsBlock = (IntBlock) rhs.eval(page)) { - IntVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - IntVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, IntBlock lhsBlock, IntBlock rhsBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(NotEquals.processInts(lhsBlock.getInt(lhsBlock.getFirstValueIndex(p)), rhsBlock.getInt(rhsBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, IntVector lhsVector, IntVector rhsVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(NotEquals.processInts(lhsVector.getInt(p), rhsVector.getInt(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "NotEqualsIntsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public NotEqualsIntsEvaluator get(DriverContext context) { - return new NotEqualsIntsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "NotEqualsIntsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/NotEqualsKeywordsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/NotEqualsKeywordsEvaluator.java deleted file mode 100644 index 919aeb4099b1f..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/NotEqualsKeywordsEvaluator.java +++ /dev/null @@ -1,139 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.comparison; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link NotEquals}. - * This class is generated. Do not edit it. - */ -public final class NotEqualsKeywordsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public NotEqualsKeywordsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock lhsBlock = (BytesRefBlock) lhs.eval(page)) { - try (BytesRefBlock rhsBlock = (BytesRefBlock) rhs.eval(page)) { - BytesRefVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - BytesRefVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, BytesRefBlock lhsBlock, BytesRefBlock rhsBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef lhsScratch = new BytesRef(); - BytesRef rhsScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(NotEquals.processKeywords(lhsBlock.getBytesRef(lhsBlock.getFirstValueIndex(p), lhsScratch), rhsBlock.getBytesRef(rhsBlock.getFirstValueIndex(p), rhsScratch))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, BytesRefVector lhsVector, BytesRefVector rhsVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - BytesRef lhsScratch = new BytesRef(); - BytesRef rhsScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(NotEquals.processKeywords(lhsVector.getBytesRef(p, lhsScratch), rhsVector.getBytesRef(p, rhsScratch))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "NotEqualsKeywordsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public NotEqualsKeywordsEvaluator get(DriverContext context) { - return new NotEqualsKeywordsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "NotEqualsKeywordsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/NotEqualsLongsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/NotEqualsLongsEvaluator.java deleted file mode 100644 index 4ec694f918d97..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/comparison/NotEqualsLongsEvaluator.java +++ /dev/null @@ -1,134 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.comparison; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link NotEquals}. - * This class is generated. Do not edit it. - */ -public final class NotEqualsLongsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public NotEqualsLongsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock lhsBlock = (LongBlock) lhs.eval(page)) { - try (LongBlock rhsBlock = (LongBlock) rhs.eval(page)) { - LongVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - LongVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, LongBlock lhsBlock, LongBlock rhsBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(NotEquals.processLongs(lhsBlock.getLong(lhsBlock.getFirstValueIndex(p)), rhsBlock.getLong(rhsBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, LongVector lhsVector, LongVector rhsVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(NotEquals.processLongs(lhsVector.getLong(p), rhsVector.getLong(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "NotEqualsLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public NotEqualsLongsEvaluator get(DriverContext context) { - return new NotEqualsLongsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "NotEqualsLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/logical/NotEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/logical/NotEvaluator.java deleted file mode 100644 index 822d380386ee9..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/evaluator/predicate/operator/logical/NotEvaluator.java +++ /dev/null @@ -1,108 +0,0 @@ -// 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.xpack.esql.evaluator.predicate.operator.logical; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Not}. - * This class is generated. Do not edit it. - */ -public final class NotEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator v; - - private final DriverContext driverContext; - - public NotEvaluator(Source source, EvalOperator.ExpressionEvaluator v, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.v = v; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BooleanBlock vBlock = (BooleanBlock) v.eval(page)) { - BooleanVector vVector = vBlock.asVector(); - if (vVector == null) { - return eval(page.getPositionCount(), vBlock); - } - return eval(page.getPositionCount(), vVector).asBlock(); - } - } - - public BooleanBlock eval(int positionCount, BooleanBlock vBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (vBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (vBlock.getValueCount(p) != 1) { - if (vBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(Not.process(vBlock.getBoolean(vBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, BooleanVector vVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(Not.process(vVector.getBoolean(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "NotEvaluator[" + "v=" + v + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(v); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory v; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory v) { - this.source = source; - this.v = v; - } - - @Override - public NotEvaluator get(DriverContext context) { - return new NotEvaluator(source, v.get(context), context); - } - - @Override - public String toString() { - return "NotEvaluator[" + "v=" + v + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/conditional/GreatestBooleanEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/conditional/GreatestBooleanEvaluator.java deleted file mode 100644 index 75558171ab58c..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/conditional/GreatestBooleanEvaluator.java +++ /dev/null @@ -1,131 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.conditional; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import java.util.Arrays; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasable; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Greatest}. - * This class is generated. Do not edit it. - */ -public final class GreatestBooleanEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator[] values; - - private final DriverContext driverContext; - - public GreatestBooleanEvaluator(Source source, EvalOperator.ExpressionEvaluator[] values, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.values = values; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - BooleanBlock[] valuesBlocks = new BooleanBlock[values.length]; - try (Releasable valuesRelease = Releasables.wrap(valuesBlocks)) { - for (int i = 0; i < valuesBlocks.length; i++) { - valuesBlocks[i] = (BooleanBlock)values[i].eval(page); - } - BooleanVector[] valuesVectors = new BooleanVector[values.length]; - for (int i = 0; i < valuesBlocks.length; i++) { - valuesVectors[i] = valuesBlocks[i].asVector(); - if (valuesVectors[i] == null) { - return eval(page.getPositionCount(), valuesBlocks); - } - } - return eval(page.getPositionCount(), valuesVectors).asBlock(); - } - } - - public BooleanBlock eval(int positionCount, BooleanBlock[] valuesBlocks) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - boolean[] valuesValues = new boolean[values.length]; - position: for (int p = 0; p < positionCount; p++) { - for (int i = 0; i < valuesBlocks.length; i++) { - if (valuesBlocks[i].isNull(p)) { - result.appendNull(); - continue position; - } - if (valuesBlocks[i].getValueCount(p) != 1) { - if (valuesBlocks[i].getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - } - // unpack valuesBlocks into valuesValues - for (int i = 0; i < valuesBlocks.length; i++) { - int o = valuesBlocks[i].getFirstValueIndex(p); - valuesValues[i] = valuesBlocks[i].getBoolean(o); - } - result.appendBoolean(Greatest.process(valuesValues)); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, BooleanVector[] valuesVectors) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - boolean[] valuesValues = new boolean[values.length]; - position: for (int p = 0; p < positionCount; p++) { - // unpack valuesVectors into valuesValues - for (int i = 0; i < valuesVectors.length; i++) { - valuesValues[i] = valuesVectors[i].getBoolean(p); - } - result.appendBoolean(Greatest.process(valuesValues)); - } - return result.build(); - } - } - - @Override - public String toString() { - return "GreatestBooleanEvaluator[" + "values=" + Arrays.toString(values) + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(() -> Releasables.close(values)); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory[] values; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory[] values) { - this.source = source; - this.values = values; - } - - @Override - public GreatestBooleanEvaluator get(DriverContext context) { - EvalOperator.ExpressionEvaluator[] values = Arrays.stream(this.values).map(a -> a.get(context)).toArray(EvalOperator.ExpressionEvaluator[]::new); - return new GreatestBooleanEvaluator(source, values, context); - } - - @Override - public String toString() { - return "GreatestBooleanEvaluator[" + "values=" + Arrays.toString(values) + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/conditional/GreatestBytesRefEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/conditional/GreatestBytesRefEvaluator.java deleted file mode 100644 index e70d147ec19b0..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/conditional/GreatestBytesRefEvaluator.java +++ /dev/null @@ -1,140 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.conditional; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import java.util.Arrays; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasable; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Greatest}. - * This class is generated. Do not edit it. - */ -public final class GreatestBytesRefEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator[] values; - - private final DriverContext driverContext; - - public GreatestBytesRefEvaluator(Source source, EvalOperator.ExpressionEvaluator[] values, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.values = values; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - BytesRefBlock[] valuesBlocks = new BytesRefBlock[values.length]; - try (Releasable valuesRelease = Releasables.wrap(valuesBlocks)) { - for (int i = 0; i < valuesBlocks.length; i++) { - valuesBlocks[i] = (BytesRefBlock)values[i].eval(page); - } - BytesRefVector[] valuesVectors = new BytesRefVector[values.length]; - for (int i = 0; i < valuesBlocks.length; i++) { - valuesVectors[i] = valuesBlocks[i].asVector(); - if (valuesVectors[i] == null) { - return eval(page.getPositionCount(), valuesBlocks); - } - } - return eval(page.getPositionCount(), valuesVectors).asBlock(); - } - } - - public BytesRefBlock eval(int positionCount, BytesRefBlock[] valuesBlocks) { - try(BytesRefBlock.Builder result = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef[] valuesValues = new BytesRef[values.length]; - BytesRef[] valuesScratch = new BytesRef[values.length]; - for (int i = 0; i < values.length; i++) { - valuesScratch[i] = new BytesRef(); - } - position: for (int p = 0; p < positionCount; p++) { - for (int i = 0; i < valuesBlocks.length; i++) { - if (valuesBlocks[i].isNull(p)) { - result.appendNull(); - continue position; - } - if (valuesBlocks[i].getValueCount(p) != 1) { - if (valuesBlocks[i].getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - } - // unpack valuesBlocks into valuesValues - for (int i = 0; i < valuesBlocks.length; i++) { - int o = valuesBlocks[i].getFirstValueIndex(p); - valuesValues[i] = valuesBlocks[i].getBytesRef(o, valuesScratch[i]); - } - result.appendBytesRef(Greatest.process(valuesValues)); - } - return result.build(); - } - } - - public BytesRefVector eval(int positionCount, BytesRefVector[] valuesVectors) { - try(BytesRefVector.Builder result = driverContext.blockFactory().newBytesRefVectorBuilder(positionCount)) { - BytesRef[] valuesValues = new BytesRef[values.length]; - BytesRef[] valuesScratch = new BytesRef[values.length]; - for (int i = 0; i < values.length; i++) { - valuesScratch[i] = new BytesRef(); - } - position: for (int p = 0; p < positionCount; p++) { - // unpack valuesVectors into valuesValues - for (int i = 0; i < valuesVectors.length; i++) { - valuesValues[i] = valuesVectors[i].getBytesRef(p, valuesScratch[i]); - } - result.appendBytesRef(Greatest.process(valuesValues)); - } - return result.build(); - } - } - - @Override - public String toString() { - return "GreatestBytesRefEvaluator[" + "values=" + Arrays.toString(values) + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(() -> Releasables.close(values)); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory[] values; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory[] values) { - this.source = source; - this.values = values; - } - - @Override - public GreatestBytesRefEvaluator get(DriverContext context) { - EvalOperator.ExpressionEvaluator[] values = Arrays.stream(this.values).map(a -> a.get(context)).toArray(EvalOperator.ExpressionEvaluator[]::new); - return new GreatestBytesRefEvaluator(source, values, context); - } - - @Override - public String toString() { - return "GreatestBytesRefEvaluator[" + "values=" + Arrays.toString(values) + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/conditional/GreatestDoubleEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/conditional/GreatestDoubleEvaluator.java deleted file mode 100644 index 4a5d49cb5853b..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/conditional/GreatestDoubleEvaluator.java +++ /dev/null @@ -1,131 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.conditional; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import java.util.Arrays; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasable; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Greatest}. - * This class is generated. Do not edit it. - */ -public final class GreatestDoubleEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator[] values; - - private final DriverContext driverContext; - - public GreatestDoubleEvaluator(Source source, EvalOperator.ExpressionEvaluator[] values, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.values = values; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - DoubleBlock[] valuesBlocks = new DoubleBlock[values.length]; - try (Releasable valuesRelease = Releasables.wrap(valuesBlocks)) { - for (int i = 0; i < valuesBlocks.length; i++) { - valuesBlocks[i] = (DoubleBlock)values[i].eval(page); - } - DoubleVector[] valuesVectors = new DoubleVector[values.length]; - for (int i = 0; i < valuesBlocks.length; i++) { - valuesVectors[i] = valuesBlocks[i].asVector(); - if (valuesVectors[i] == null) { - return eval(page.getPositionCount(), valuesBlocks); - } - } - return eval(page.getPositionCount(), valuesVectors).asBlock(); - } - } - - public DoubleBlock eval(int positionCount, DoubleBlock[] valuesBlocks) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - double[] valuesValues = new double[values.length]; - position: for (int p = 0; p < positionCount; p++) { - for (int i = 0; i < valuesBlocks.length; i++) { - if (valuesBlocks[i].isNull(p)) { - result.appendNull(); - continue position; - } - if (valuesBlocks[i].getValueCount(p) != 1) { - if (valuesBlocks[i].getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - } - // unpack valuesBlocks into valuesValues - for (int i = 0; i < valuesBlocks.length; i++) { - int o = valuesBlocks[i].getFirstValueIndex(p); - valuesValues[i] = valuesBlocks[i].getDouble(o); - } - result.appendDouble(Greatest.process(valuesValues)); - } - return result.build(); - } - } - - public DoubleVector eval(int positionCount, DoubleVector[] valuesVectors) { - try(DoubleVector.Builder result = driverContext.blockFactory().newDoubleVectorBuilder(positionCount)) { - double[] valuesValues = new double[values.length]; - position: for (int p = 0; p < positionCount; p++) { - // unpack valuesVectors into valuesValues - for (int i = 0; i < valuesVectors.length; i++) { - valuesValues[i] = valuesVectors[i].getDouble(p); - } - result.appendDouble(Greatest.process(valuesValues)); - } - return result.build(); - } - } - - @Override - public String toString() { - return "GreatestDoubleEvaluator[" + "values=" + Arrays.toString(values) + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(() -> Releasables.close(values)); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory[] values; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory[] values) { - this.source = source; - this.values = values; - } - - @Override - public GreatestDoubleEvaluator get(DriverContext context) { - EvalOperator.ExpressionEvaluator[] values = Arrays.stream(this.values).map(a -> a.get(context)).toArray(EvalOperator.ExpressionEvaluator[]::new); - return new GreatestDoubleEvaluator(source, values, context); - } - - @Override - public String toString() { - return "GreatestDoubleEvaluator[" + "values=" + Arrays.toString(values) + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/conditional/GreatestIntEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/conditional/GreatestIntEvaluator.java deleted file mode 100644 index 6c675c3168523..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/conditional/GreatestIntEvaluator.java +++ /dev/null @@ -1,131 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.conditional; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import java.util.Arrays; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasable; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Greatest}. - * This class is generated. Do not edit it. - */ -public final class GreatestIntEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator[] values; - - private final DriverContext driverContext; - - public GreatestIntEvaluator(Source source, EvalOperator.ExpressionEvaluator[] values, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.values = values; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - IntBlock[] valuesBlocks = new IntBlock[values.length]; - try (Releasable valuesRelease = Releasables.wrap(valuesBlocks)) { - for (int i = 0; i < valuesBlocks.length; i++) { - valuesBlocks[i] = (IntBlock)values[i].eval(page); - } - IntVector[] valuesVectors = new IntVector[values.length]; - for (int i = 0; i < valuesBlocks.length; i++) { - valuesVectors[i] = valuesBlocks[i].asVector(); - if (valuesVectors[i] == null) { - return eval(page.getPositionCount(), valuesBlocks); - } - } - return eval(page.getPositionCount(), valuesVectors).asBlock(); - } - } - - public IntBlock eval(int positionCount, IntBlock[] valuesBlocks) { - try(IntBlock.Builder result = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - int[] valuesValues = new int[values.length]; - position: for (int p = 0; p < positionCount; p++) { - for (int i = 0; i < valuesBlocks.length; i++) { - if (valuesBlocks[i].isNull(p)) { - result.appendNull(); - continue position; - } - if (valuesBlocks[i].getValueCount(p) != 1) { - if (valuesBlocks[i].getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - } - // unpack valuesBlocks into valuesValues - for (int i = 0; i < valuesBlocks.length; i++) { - int o = valuesBlocks[i].getFirstValueIndex(p); - valuesValues[i] = valuesBlocks[i].getInt(o); - } - result.appendInt(Greatest.process(valuesValues)); - } - return result.build(); - } - } - - public IntVector eval(int positionCount, IntVector[] valuesVectors) { - try(IntVector.Builder result = driverContext.blockFactory().newIntVectorBuilder(positionCount)) { - int[] valuesValues = new int[values.length]; - position: for (int p = 0; p < positionCount; p++) { - // unpack valuesVectors into valuesValues - for (int i = 0; i < valuesVectors.length; i++) { - valuesValues[i] = valuesVectors[i].getInt(p); - } - result.appendInt(Greatest.process(valuesValues)); - } - return result.build(); - } - } - - @Override - public String toString() { - return "GreatestIntEvaluator[" + "values=" + Arrays.toString(values) + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(() -> Releasables.close(values)); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory[] values; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory[] values) { - this.source = source; - this.values = values; - } - - @Override - public GreatestIntEvaluator get(DriverContext context) { - EvalOperator.ExpressionEvaluator[] values = Arrays.stream(this.values).map(a -> a.get(context)).toArray(EvalOperator.ExpressionEvaluator[]::new); - return new GreatestIntEvaluator(source, values, context); - } - - @Override - public String toString() { - return "GreatestIntEvaluator[" + "values=" + Arrays.toString(values) + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/conditional/GreatestLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/conditional/GreatestLongEvaluator.java deleted file mode 100644 index 3f4f0c748db3f..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/conditional/GreatestLongEvaluator.java +++ /dev/null @@ -1,131 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.conditional; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import java.util.Arrays; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasable; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Greatest}. - * This class is generated. Do not edit it. - */ -public final class GreatestLongEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator[] values; - - private final DriverContext driverContext; - - public GreatestLongEvaluator(Source source, EvalOperator.ExpressionEvaluator[] values, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.values = values; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - LongBlock[] valuesBlocks = new LongBlock[values.length]; - try (Releasable valuesRelease = Releasables.wrap(valuesBlocks)) { - for (int i = 0; i < valuesBlocks.length; i++) { - valuesBlocks[i] = (LongBlock)values[i].eval(page); - } - LongVector[] valuesVectors = new LongVector[values.length]; - for (int i = 0; i < valuesBlocks.length; i++) { - valuesVectors[i] = valuesBlocks[i].asVector(); - if (valuesVectors[i] == null) { - return eval(page.getPositionCount(), valuesBlocks); - } - } - return eval(page.getPositionCount(), valuesVectors).asBlock(); - } - } - - public LongBlock eval(int positionCount, LongBlock[] valuesBlocks) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - long[] valuesValues = new long[values.length]; - position: for (int p = 0; p < positionCount; p++) { - for (int i = 0; i < valuesBlocks.length; i++) { - if (valuesBlocks[i].isNull(p)) { - result.appendNull(); - continue position; - } - if (valuesBlocks[i].getValueCount(p) != 1) { - if (valuesBlocks[i].getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - } - // unpack valuesBlocks into valuesValues - for (int i = 0; i < valuesBlocks.length; i++) { - int o = valuesBlocks[i].getFirstValueIndex(p); - valuesValues[i] = valuesBlocks[i].getLong(o); - } - result.appendLong(Greatest.process(valuesValues)); - } - return result.build(); - } - } - - public LongVector eval(int positionCount, LongVector[] valuesVectors) { - try(LongVector.Builder result = driverContext.blockFactory().newLongVectorBuilder(positionCount)) { - long[] valuesValues = new long[values.length]; - position: for (int p = 0; p < positionCount; p++) { - // unpack valuesVectors into valuesValues - for (int i = 0; i < valuesVectors.length; i++) { - valuesValues[i] = valuesVectors[i].getLong(p); - } - result.appendLong(Greatest.process(valuesValues)); - } - return result.build(); - } - } - - @Override - public String toString() { - return "GreatestLongEvaluator[" + "values=" + Arrays.toString(values) + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(() -> Releasables.close(values)); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory[] values; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory[] values) { - this.source = source; - this.values = values; - } - - @Override - public GreatestLongEvaluator get(DriverContext context) { - EvalOperator.ExpressionEvaluator[] values = Arrays.stream(this.values).map(a -> a.get(context)).toArray(EvalOperator.ExpressionEvaluator[]::new); - return new GreatestLongEvaluator(source, values, context); - } - - @Override - public String toString() { - return "GreatestLongEvaluator[" + "values=" + Arrays.toString(values) + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/conditional/LeastBooleanEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/conditional/LeastBooleanEvaluator.java deleted file mode 100644 index 70d4345fe197c..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/conditional/LeastBooleanEvaluator.java +++ /dev/null @@ -1,131 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.conditional; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import java.util.Arrays; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasable; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Least}. - * This class is generated. Do not edit it. - */ -public final class LeastBooleanEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator[] values; - - private final DriverContext driverContext; - - public LeastBooleanEvaluator(Source source, EvalOperator.ExpressionEvaluator[] values, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.values = values; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - BooleanBlock[] valuesBlocks = new BooleanBlock[values.length]; - try (Releasable valuesRelease = Releasables.wrap(valuesBlocks)) { - for (int i = 0; i < valuesBlocks.length; i++) { - valuesBlocks[i] = (BooleanBlock)values[i].eval(page); - } - BooleanVector[] valuesVectors = new BooleanVector[values.length]; - for (int i = 0; i < valuesBlocks.length; i++) { - valuesVectors[i] = valuesBlocks[i].asVector(); - if (valuesVectors[i] == null) { - return eval(page.getPositionCount(), valuesBlocks); - } - } - return eval(page.getPositionCount(), valuesVectors).asBlock(); - } - } - - public BooleanBlock eval(int positionCount, BooleanBlock[] valuesBlocks) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - boolean[] valuesValues = new boolean[values.length]; - position: for (int p = 0; p < positionCount; p++) { - for (int i = 0; i < valuesBlocks.length; i++) { - if (valuesBlocks[i].isNull(p)) { - result.appendNull(); - continue position; - } - if (valuesBlocks[i].getValueCount(p) != 1) { - if (valuesBlocks[i].getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - } - // unpack valuesBlocks into valuesValues - for (int i = 0; i < valuesBlocks.length; i++) { - int o = valuesBlocks[i].getFirstValueIndex(p); - valuesValues[i] = valuesBlocks[i].getBoolean(o); - } - result.appendBoolean(Least.process(valuesValues)); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, BooleanVector[] valuesVectors) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - boolean[] valuesValues = new boolean[values.length]; - position: for (int p = 0; p < positionCount; p++) { - // unpack valuesVectors into valuesValues - for (int i = 0; i < valuesVectors.length; i++) { - valuesValues[i] = valuesVectors[i].getBoolean(p); - } - result.appendBoolean(Least.process(valuesValues)); - } - return result.build(); - } - } - - @Override - public String toString() { - return "LeastBooleanEvaluator[" + "values=" + Arrays.toString(values) + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(() -> Releasables.close(values)); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory[] values; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory[] values) { - this.source = source; - this.values = values; - } - - @Override - public LeastBooleanEvaluator get(DriverContext context) { - EvalOperator.ExpressionEvaluator[] values = Arrays.stream(this.values).map(a -> a.get(context)).toArray(EvalOperator.ExpressionEvaluator[]::new); - return new LeastBooleanEvaluator(source, values, context); - } - - @Override - public String toString() { - return "LeastBooleanEvaluator[" + "values=" + Arrays.toString(values) + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/conditional/LeastBytesRefEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/conditional/LeastBytesRefEvaluator.java deleted file mode 100644 index 642ca36574cb6..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/conditional/LeastBytesRefEvaluator.java +++ /dev/null @@ -1,140 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.conditional; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import java.util.Arrays; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasable; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Least}. - * This class is generated. Do not edit it. - */ -public final class LeastBytesRefEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator[] values; - - private final DriverContext driverContext; - - public LeastBytesRefEvaluator(Source source, EvalOperator.ExpressionEvaluator[] values, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.values = values; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - BytesRefBlock[] valuesBlocks = new BytesRefBlock[values.length]; - try (Releasable valuesRelease = Releasables.wrap(valuesBlocks)) { - for (int i = 0; i < valuesBlocks.length; i++) { - valuesBlocks[i] = (BytesRefBlock)values[i].eval(page); - } - BytesRefVector[] valuesVectors = new BytesRefVector[values.length]; - for (int i = 0; i < valuesBlocks.length; i++) { - valuesVectors[i] = valuesBlocks[i].asVector(); - if (valuesVectors[i] == null) { - return eval(page.getPositionCount(), valuesBlocks); - } - } - return eval(page.getPositionCount(), valuesVectors).asBlock(); - } - } - - public BytesRefBlock eval(int positionCount, BytesRefBlock[] valuesBlocks) { - try(BytesRefBlock.Builder result = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef[] valuesValues = new BytesRef[values.length]; - BytesRef[] valuesScratch = new BytesRef[values.length]; - for (int i = 0; i < values.length; i++) { - valuesScratch[i] = new BytesRef(); - } - position: for (int p = 0; p < positionCount; p++) { - for (int i = 0; i < valuesBlocks.length; i++) { - if (valuesBlocks[i].isNull(p)) { - result.appendNull(); - continue position; - } - if (valuesBlocks[i].getValueCount(p) != 1) { - if (valuesBlocks[i].getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - } - // unpack valuesBlocks into valuesValues - for (int i = 0; i < valuesBlocks.length; i++) { - int o = valuesBlocks[i].getFirstValueIndex(p); - valuesValues[i] = valuesBlocks[i].getBytesRef(o, valuesScratch[i]); - } - result.appendBytesRef(Least.process(valuesValues)); - } - return result.build(); - } - } - - public BytesRefVector eval(int positionCount, BytesRefVector[] valuesVectors) { - try(BytesRefVector.Builder result = driverContext.blockFactory().newBytesRefVectorBuilder(positionCount)) { - BytesRef[] valuesValues = new BytesRef[values.length]; - BytesRef[] valuesScratch = new BytesRef[values.length]; - for (int i = 0; i < values.length; i++) { - valuesScratch[i] = new BytesRef(); - } - position: for (int p = 0; p < positionCount; p++) { - // unpack valuesVectors into valuesValues - for (int i = 0; i < valuesVectors.length; i++) { - valuesValues[i] = valuesVectors[i].getBytesRef(p, valuesScratch[i]); - } - result.appendBytesRef(Least.process(valuesValues)); - } - return result.build(); - } - } - - @Override - public String toString() { - return "LeastBytesRefEvaluator[" + "values=" + Arrays.toString(values) + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(() -> Releasables.close(values)); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory[] values; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory[] values) { - this.source = source; - this.values = values; - } - - @Override - public LeastBytesRefEvaluator get(DriverContext context) { - EvalOperator.ExpressionEvaluator[] values = Arrays.stream(this.values).map(a -> a.get(context)).toArray(EvalOperator.ExpressionEvaluator[]::new); - return new LeastBytesRefEvaluator(source, values, context); - } - - @Override - public String toString() { - return "LeastBytesRefEvaluator[" + "values=" + Arrays.toString(values) + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/conditional/LeastDoubleEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/conditional/LeastDoubleEvaluator.java deleted file mode 100644 index 41b0ad4d4d085..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/conditional/LeastDoubleEvaluator.java +++ /dev/null @@ -1,131 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.conditional; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import java.util.Arrays; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasable; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Least}. - * This class is generated. Do not edit it. - */ -public final class LeastDoubleEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator[] values; - - private final DriverContext driverContext; - - public LeastDoubleEvaluator(Source source, EvalOperator.ExpressionEvaluator[] values, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.values = values; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - DoubleBlock[] valuesBlocks = new DoubleBlock[values.length]; - try (Releasable valuesRelease = Releasables.wrap(valuesBlocks)) { - for (int i = 0; i < valuesBlocks.length; i++) { - valuesBlocks[i] = (DoubleBlock)values[i].eval(page); - } - DoubleVector[] valuesVectors = new DoubleVector[values.length]; - for (int i = 0; i < valuesBlocks.length; i++) { - valuesVectors[i] = valuesBlocks[i].asVector(); - if (valuesVectors[i] == null) { - return eval(page.getPositionCount(), valuesBlocks); - } - } - return eval(page.getPositionCount(), valuesVectors).asBlock(); - } - } - - public DoubleBlock eval(int positionCount, DoubleBlock[] valuesBlocks) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - double[] valuesValues = new double[values.length]; - position: for (int p = 0; p < positionCount; p++) { - for (int i = 0; i < valuesBlocks.length; i++) { - if (valuesBlocks[i].isNull(p)) { - result.appendNull(); - continue position; - } - if (valuesBlocks[i].getValueCount(p) != 1) { - if (valuesBlocks[i].getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - } - // unpack valuesBlocks into valuesValues - for (int i = 0; i < valuesBlocks.length; i++) { - int o = valuesBlocks[i].getFirstValueIndex(p); - valuesValues[i] = valuesBlocks[i].getDouble(o); - } - result.appendDouble(Least.process(valuesValues)); - } - return result.build(); - } - } - - public DoubleVector eval(int positionCount, DoubleVector[] valuesVectors) { - try(DoubleVector.Builder result = driverContext.blockFactory().newDoubleVectorBuilder(positionCount)) { - double[] valuesValues = new double[values.length]; - position: for (int p = 0; p < positionCount; p++) { - // unpack valuesVectors into valuesValues - for (int i = 0; i < valuesVectors.length; i++) { - valuesValues[i] = valuesVectors[i].getDouble(p); - } - result.appendDouble(Least.process(valuesValues)); - } - return result.build(); - } - } - - @Override - public String toString() { - return "LeastDoubleEvaluator[" + "values=" + Arrays.toString(values) + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(() -> Releasables.close(values)); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory[] values; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory[] values) { - this.source = source; - this.values = values; - } - - @Override - public LeastDoubleEvaluator get(DriverContext context) { - EvalOperator.ExpressionEvaluator[] values = Arrays.stream(this.values).map(a -> a.get(context)).toArray(EvalOperator.ExpressionEvaluator[]::new); - return new LeastDoubleEvaluator(source, values, context); - } - - @Override - public String toString() { - return "LeastDoubleEvaluator[" + "values=" + Arrays.toString(values) + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/conditional/LeastIntEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/conditional/LeastIntEvaluator.java deleted file mode 100644 index c2c80db6ca0bb..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/conditional/LeastIntEvaluator.java +++ /dev/null @@ -1,131 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.conditional; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import java.util.Arrays; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasable; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Least}. - * This class is generated. Do not edit it. - */ -public final class LeastIntEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator[] values; - - private final DriverContext driverContext; - - public LeastIntEvaluator(Source source, EvalOperator.ExpressionEvaluator[] values, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.values = values; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - IntBlock[] valuesBlocks = new IntBlock[values.length]; - try (Releasable valuesRelease = Releasables.wrap(valuesBlocks)) { - for (int i = 0; i < valuesBlocks.length; i++) { - valuesBlocks[i] = (IntBlock)values[i].eval(page); - } - IntVector[] valuesVectors = new IntVector[values.length]; - for (int i = 0; i < valuesBlocks.length; i++) { - valuesVectors[i] = valuesBlocks[i].asVector(); - if (valuesVectors[i] == null) { - return eval(page.getPositionCount(), valuesBlocks); - } - } - return eval(page.getPositionCount(), valuesVectors).asBlock(); - } - } - - public IntBlock eval(int positionCount, IntBlock[] valuesBlocks) { - try(IntBlock.Builder result = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - int[] valuesValues = new int[values.length]; - position: for (int p = 0; p < positionCount; p++) { - for (int i = 0; i < valuesBlocks.length; i++) { - if (valuesBlocks[i].isNull(p)) { - result.appendNull(); - continue position; - } - if (valuesBlocks[i].getValueCount(p) != 1) { - if (valuesBlocks[i].getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - } - // unpack valuesBlocks into valuesValues - for (int i = 0; i < valuesBlocks.length; i++) { - int o = valuesBlocks[i].getFirstValueIndex(p); - valuesValues[i] = valuesBlocks[i].getInt(o); - } - result.appendInt(Least.process(valuesValues)); - } - return result.build(); - } - } - - public IntVector eval(int positionCount, IntVector[] valuesVectors) { - try(IntVector.Builder result = driverContext.blockFactory().newIntVectorBuilder(positionCount)) { - int[] valuesValues = new int[values.length]; - position: for (int p = 0; p < positionCount; p++) { - // unpack valuesVectors into valuesValues - for (int i = 0; i < valuesVectors.length; i++) { - valuesValues[i] = valuesVectors[i].getInt(p); - } - result.appendInt(Least.process(valuesValues)); - } - return result.build(); - } - } - - @Override - public String toString() { - return "LeastIntEvaluator[" + "values=" + Arrays.toString(values) + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(() -> Releasables.close(values)); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory[] values; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory[] values) { - this.source = source; - this.values = values; - } - - @Override - public LeastIntEvaluator get(DriverContext context) { - EvalOperator.ExpressionEvaluator[] values = Arrays.stream(this.values).map(a -> a.get(context)).toArray(EvalOperator.ExpressionEvaluator[]::new); - return new LeastIntEvaluator(source, values, context); - } - - @Override - public String toString() { - return "LeastIntEvaluator[" + "values=" + Arrays.toString(values) + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/conditional/LeastLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/conditional/LeastLongEvaluator.java deleted file mode 100644 index cd8ab3a0cd06f..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/conditional/LeastLongEvaluator.java +++ /dev/null @@ -1,131 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.conditional; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import java.util.Arrays; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasable; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Least}. - * This class is generated. Do not edit it. - */ -public final class LeastLongEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator[] values; - - private final DriverContext driverContext; - - public LeastLongEvaluator(Source source, EvalOperator.ExpressionEvaluator[] values, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.values = values; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - LongBlock[] valuesBlocks = new LongBlock[values.length]; - try (Releasable valuesRelease = Releasables.wrap(valuesBlocks)) { - for (int i = 0; i < valuesBlocks.length; i++) { - valuesBlocks[i] = (LongBlock)values[i].eval(page); - } - LongVector[] valuesVectors = new LongVector[values.length]; - for (int i = 0; i < valuesBlocks.length; i++) { - valuesVectors[i] = valuesBlocks[i].asVector(); - if (valuesVectors[i] == null) { - return eval(page.getPositionCount(), valuesBlocks); - } - } - return eval(page.getPositionCount(), valuesVectors).asBlock(); - } - } - - public LongBlock eval(int positionCount, LongBlock[] valuesBlocks) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - long[] valuesValues = new long[values.length]; - position: for (int p = 0; p < positionCount; p++) { - for (int i = 0; i < valuesBlocks.length; i++) { - if (valuesBlocks[i].isNull(p)) { - result.appendNull(); - continue position; - } - if (valuesBlocks[i].getValueCount(p) != 1) { - if (valuesBlocks[i].getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - } - // unpack valuesBlocks into valuesValues - for (int i = 0; i < valuesBlocks.length; i++) { - int o = valuesBlocks[i].getFirstValueIndex(p); - valuesValues[i] = valuesBlocks[i].getLong(o); - } - result.appendLong(Least.process(valuesValues)); - } - return result.build(); - } - } - - public LongVector eval(int positionCount, LongVector[] valuesVectors) { - try(LongVector.Builder result = driverContext.blockFactory().newLongVectorBuilder(positionCount)) { - long[] valuesValues = new long[values.length]; - position: for (int p = 0; p < positionCount; p++) { - // unpack valuesVectors into valuesValues - for (int i = 0; i < valuesVectors.length; i++) { - valuesValues[i] = valuesVectors[i].getLong(p); - } - result.appendLong(Least.process(valuesValues)); - } - return result.build(); - } - } - - @Override - public String toString() { - return "LeastLongEvaluator[" + "values=" + Arrays.toString(values) + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(() -> Releasables.close(values)); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory[] values; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory[] values) { - this.source = source; - this.values = values; - } - - @Override - public LeastLongEvaluator get(DriverContext context) { - EvalOperator.ExpressionEvaluator[] values = Arrays.stream(this.values).map(a -> a.get(context)).toArray(EvalOperator.ExpressionEvaluator[]::new); - return new LeastLongEvaluator(source, values, context); - } - - @Override - public String toString() { - return "LeastLongEvaluator[" + "values=" + Arrays.toString(values) + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToBooleanFromDoubleEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToBooleanFromDoubleEvaluator.java deleted file mode 100644 index c64568251feec..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToBooleanFromDoubleEvaluator.java +++ /dev/null @@ -1,108 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToBoolean}. - * This class is generated. Do not edit it. - */ -public final class ToBooleanFromDoubleEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToBooleanFromDoubleEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToBooleanFromDouble"; - } - - @Override - public Block evalVector(Vector v) { - DoubleVector vector = (DoubleVector) v; - int positionCount = v.getPositionCount(); - if (vector.isConstant()) { - return driverContext.blockFactory().newConstantBooleanBlockWith(evalValue(vector, 0), positionCount); - } - try (BooleanBlock.Builder builder = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - builder.appendBoolean(evalValue(vector, p)); - } - return builder.build(); - } - } - - private static boolean evalValue(DoubleVector container, int index) { - double value = container.getDouble(index); - return ToBoolean.fromDouble(value); - } - - @Override - public Block evalBlock(Block b) { - DoubleBlock block = (DoubleBlock) b; - int positionCount = block.getPositionCount(); - try (BooleanBlock.Builder builder = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - boolean value = evalValue(block, i); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendBoolean(value); - valuesAppended = true; - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static boolean evalValue(DoubleBlock container, int index) { - double value = container.getDouble(index); - return ToBoolean.fromDouble(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToBooleanFromDoubleEvaluator get(DriverContext context) { - return new ToBooleanFromDoubleEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToBooleanFromDoubleEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToBooleanFromIntEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToBooleanFromIntEvaluator.java deleted file mode 100644 index daac34639c66a..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToBooleanFromIntEvaluator.java +++ /dev/null @@ -1,108 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToBoolean}. - * This class is generated. Do not edit it. - */ -public final class ToBooleanFromIntEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToBooleanFromIntEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToBooleanFromInt"; - } - - @Override - public Block evalVector(Vector v) { - IntVector vector = (IntVector) v; - int positionCount = v.getPositionCount(); - if (vector.isConstant()) { - return driverContext.blockFactory().newConstantBooleanBlockWith(evalValue(vector, 0), positionCount); - } - try (BooleanBlock.Builder builder = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - builder.appendBoolean(evalValue(vector, p)); - } - return builder.build(); - } - } - - private static boolean evalValue(IntVector container, int index) { - int value = container.getInt(index); - return ToBoolean.fromInt(value); - } - - @Override - public Block evalBlock(Block b) { - IntBlock block = (IntBlock) b; - int positionCount = block.getPositionCount(); - try (BooleanBlock.Builder builder = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - boolean value = evalValue(block, i); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendBoolean(value); - valuesAppended = true; - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static boolean evalValue(IntBlock container, int index) { - int value = container.getInt(index); - return ToBoolean.fromInt(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToBooleanFromIntEvaluator get(DriverContext context) { - return new ToBooleanFromIntEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToBooleanFromIntEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToBooleanFromLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToBooleanFromLongEvaluator.java deleted file mode 100644 index 1e6b2aefce9f3..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToBooleanFromLongEvaluator.java +++ /dev/null @@ -1,108 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToBoolean}. - * This class is generated. Do not edit it. - */ -public final class ToBooleanFromLongEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToBooleanFromLongEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToBooleanFromLong"; - } - - @Override - public Block evalVector(Vector v) { - LongVector vector = (LongVector) v; - int positionCount = v.getPositionCount(); - if (vector.isConstant()) { - return driverContext.blockFactory().newConstantBooleanBlockWith(evalValue(vector, 0), positionCount); - } - try (BooleanBlock.Builder builder = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - builder.appendBoolean(evalValue(vector, p)); - } - return builder.build(); - } - } - - private static boolean evalValue(LongVector container, int index) { - long value = container.getLong(index); - return ToBoolean.fromLong(value); - } - - @Override - public Block evalBlock(Block b) { - LongBlock block = (LongBlock) b; - int positionCount = block.getPositionCount(); - try (BooleanBlock.Builder builder = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - boolean value = evalValue(block, i); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendBoolean(value); - valuesAppended = true; - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static boolean evalValue(LongBlock container, int index) { - long value = container.getLong(index); - return ToBoolean.fromLong(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToBooleanFromLongEvaluator get(DriverContext context) { - return new ToBooleanFromLongEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToBooleanFromLongEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToBooleanFromStringEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToBooleanFromStringEvaluator.java deleted file mode 100644 index ce573a3b8d2d3..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToBooleanFromStringEvaluator.java +++ /dev/null @@ -1,111 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToBoolean}. - * This class is generated. Do not edit it. - */ -public final class ToBooleanFromStringEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToBooleanFromStringEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToBooleanFromString"; - } - - @Override - public Block evalVector(Vector v) { - BytesRefVector vector = (BytesRefVector) v; - int positionCount = v.getPositionCount(); - BytesRef scratchPad = new BytesRef(); - if (vector.isConstant()) { - return driverContext.blockFactory().newConstantBooleanBlockWith(evalValue(vector, 0, scratchPad), positionCount); - } - try (BooleanBlock.Builder builder = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - builder.appendBoolean(evalValue(vector, p, scratchPad)); - } - return builder.build(); - } - } - - private static boolean evalValue(BytesRefVector container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToBoolean.fromKeyword(value); - } - - @Override - public Block evalBlock(Block b) { - BytesRefBlock block = (BytesRefBlock) b; - int positionCount = block.getPositionCount(); - try (BooleanBlock.Builder builder = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef scratchPad = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - boolean value = evalValue(block, i, scratchPad); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendBoolean(value); - valuesAppended = true; - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static boolean evalValue(BytesRefBlock container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToBoolean.fromKeyword(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToBooleanFromStringEvaluator get(DriverContext context) { - return new ToBooleanFromStringEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToBooleanFromStringEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToBooleanFromUnsignedLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToBooleanFromUnsignedLongEvaluator.java deleted file mode 100644 index 5ec75f10c2ecb..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToBooleanFromUnsignedLongEvaluator.java +++ /dev/null @@ -1,108 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToBoolean}. - * This class is generated. Do not edit it. - */ -public final class ToBooleanFromUnsignedLongEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToBooleanFromUnsignedLongEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToBooleanFromUnsignedLong"; - } - - @Override - public Block evalVector(Vector v) { - LongVector vector = (LongVector) v; - int positionCount = v.getPositionCount(); - if (vector.isConstant()) { - return driverContext.blockFactory().newConstantBooleanBlockWith(evalValue(vector, 0), positionCount); - } - try (BooleanBlock.Builder builder = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - builder.appendBoolean(evalValue(vector, p)); - } - return builder.build(); - } - } - - private static boolean evalValue(LongVector container, int index) { - long value = container.getLong(index); - return ToBoolean.fromUnsignedLong(value); - } - - @Override - public Block evalBlock(Block b) { - LongBlock block = (LongBlock) b; - int positionCount = block.getPositionCount(); - try (BooleanBlock.Builder builder = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - boolean value = evalValue(block, i); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendBoolean(value); - valuesAppended = true; - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static boolean evalValue(LongBlock container, int index) { - long value = container.getLong(index); - return ToBoolean.fromUnsignedLong(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToBooleanFromUnsignedLongEvaluator get(DriverContext context) { - return new ToBooleanFromUnsignedLongEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToBooleanFromUnsignedLongEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToCartesianPointFromStringEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToCartesianPointFromStringEvaluator.java deleted file mode 100644 index ee5159be521d6..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToCartesianPointFromStringEvaluator.java +++ /dev/null @@ -1,125 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToCartesianPoint}. - * This class is generated. Do not edit it. - */ -public final class ToCartesianPointFromStringEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToCartesianPointFromStringEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToCartesianPointFromString"; - } - - @Override - public Block evalVector(Vector v) { - BytesRefVector vector = (BytesRefVector) v; - int positionCount = v.getPositionCount(); - BytesRef scratchPad = new BytesRef(); - if (vector.isConstant()) { - try { - return driverContext.blockFactory().newConstantBytesRefBlockWith(evalValue(vector, 0, scratchPad), positionCount); - } catch (IllegalArgumentException e) { - registerException(e); - return driverContext.blockFactory().newConstantNullBlock(positionCount); - } - } - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - try { - builder.appendBytesRef(evalValue(vector, p, scratchPad)); - } catch (IllegalArgumentException e) { - registerException(e); - builder.appendNull(); - } - } - return builder.build(); - } - } - - private static BytesRef evalValue(BytesRefVector container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToCartesianPoint.fromKeyword(value); - } - - @Override - public Block evalBlock(Block b) { - BytesRefBlock block = (BytesRefBlock) b; - int positionCount = block.getPositionCount(); - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef scratchPad = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - try { - BytesRef value = evalValue(block, i, scratchPad); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendBytesRef(value); - valuesAppended = true; - } catch (IllegalArgumentException e) { - registerException(e); - } - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static BytesRef evalValue(BytesRefBlock container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToCartesianPoint.fromKeyword(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToCartesianPointFromStringEvaluator get(DriverContext context) { - return new ToCartesianPointFromStringEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToCartesianPointFromStringEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToCartesianShapeFromStringEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToCartesianShapeFromStringEvaluator.java deleted file mode 100644 index 5ec9dcb94f67f..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToCartesianShapeFromStringEvaluator.java +++ /dev/null @@ -1,125 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToCartesianShape}. - * This class is generated. Do not edit it. - */ -public final class ToCartesianShapeFromStringEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToCartesianShapeFromStringEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToCartesianShapeFromString"; - } - - @Override - public Block evalVector(Vector v) { - BytesRefVector vector = (BytesRefVector) v; - int positionCount = v.getPositionCount(); - BytesRef scratchPad = new BytesRef(); - if (vector.isConstant()) { - try { - return driverContext.blockFactory().newConstantBytesRefBlockWith(evalValue(vector, 0, scratchPad), positionCount); - } catch (IllegalArgumentException e) { - registerException(e); - return driverContext.blockFactory().newConstantNullBlock(positionCount); - } - } - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - try { - builder.appendBytesRef(evalValue(vector, p, scratchPad)); - } catch (IllegalArgumentException e) { - registerException(e); - builder.appendNull(); - } - } - return builder.build(); - } - } - - private static BytesRef evalValue(BytesRefVector container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToCartesianShape.fromKeyword(value); - } - - @Override - public Block evalBlock(Block b) { - BytesRefBlock block = (BytesRefBlock) b; - int positionCount = block.getPositionCount(); - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef scratchPad = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - try { - BytesRef value = evalValue(block, i, scratchPad); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendBytesRef(value); - valuesAppended = true; - } catch (IllegalArgumentException e) { - registerException(e); - } - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static BytesRef evalValue(BytesRefBlock container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToCartesianShape.fromKeyword(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToCartesianShapeFromStringEvaluator get(DriverContext context) { - return new ToCartesianShapeFromStringEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToCartesianShapeFromStringEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToDatetimeFromStringEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToDatetimeFromStringEvaluator.java deleted file mode 100644 index b868fe9b950c8..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToDatetimeFromStringEvaluator.java +++ /dev/null @@ -1,126 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToDatetime}. - * This class is generated. Do not edit it. - */ -public final class ToDatetimeFromStringEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToDatetimeFromStringEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToDatetimeFromString"; - } - - @Override - public Block evalVector(Vector v) { - BytesRefVector vector = (BytesRefVector) v; - int positionCount = v.getPositionCount(); - BytesRef scratchPad = new BytesRef(); - if (vector.isConstant()) { - try { - return driverContext.blockFactory().newConstantLongBlockWith(evalValue(vector, 0, scratchPad), positionCount); - } catch (IllegalArgumentException e) { - registerException(e); - return driverContext.blockFactory().newConstantNullBlock(positionCount); - } - } - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - try { - builder.appendLong(evalValue(vector, p, scratchPad)); - } catch (IllegalArgumentException e) { - registerException(e); - builder.appendNull(); - } - } - return builder.build(); - } - } - - private static long evalValue(BytesRefVector container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToDatetime.fromKeyword(value); - } - - @Override - public Block evalBlock(Block b) { - BytesRefBlock block = (BytesRefBlock) b; - int positionCount = block.getPositionCount(); - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - BytesRef scratchPad = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - try { - long value = evalValue(block, i, scratchPad); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendLong(value); - valuesAppended = true; - } catch (IllegalArgumentException e) { - registerException(e); - } - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static long evalValue(BytesRefBlock container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToDatetime.fromKeyword(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToDatetimeFromStringEvaluator get(DriverContext context) { - return new ToDatetimeFromStringEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToDatetimeFromStringEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToDegreesEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToDegreesEvaluator.java deleted file mode 100644 index 8b581cbac5980..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToDegreesEvaluator.java +++ /dev/null @@ -1,122 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.ArithmeticException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToDegrees}. - * This class is generated. Do not edit it. - */ -public final class ToDegreesEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToDegreesEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToDegrees"; - } - - @Override - public Block evalVector(Vector v) { - DoubleVector vector = (DoubleVector) v; - int positionCount = v.getPositionCount(); - if (vector.isConstant()) { - try { - return driverContext.blockFactory().newConstantDoubleBlockWith(evalValue(vector, 0), positionCount); - } catch (ArithmeticException e) { - registerException(e); - return driverContext.blockFactory().newConstantNullBlock(positionCount); - } - } - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - try { - builder.appendDouble(evalValue(vector, p)); - } catch (ArithmeticException e) { - registerException(e); - builder.appendNull(); - } - } - return builder.build(); - } - } - - private static double evalValue(DoubleVector container, int index) { - double value = container.getDouble(index); - return ToDegrees.process(value); - } - - @Override - public Block evalBlock(Block b) { - DoubleBlock block = (DoubleBlock) b; - int positionCount = block.getPositionCount(); - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - try { - double value = evalValue(block, i); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendDouble(value); - valuesAppended = true; - } catch (ArithmeticException e) { - registerException(e); - } - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static double evalValue(DoubleBlock container, int index) { - double value = container.getDouble(index); - return ToDegrees.process(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToDegreesEvaluator get(DriverContext context) { - return new ToDegreesEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToDegreesEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToDoubleFromBooleanEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToDoubleFromBooleanEvaluator.java deleted file mode 100644 index c831e1b0a314a..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToDoubleFromBooleanEvaluator.java +++ /dev/null @@ -1,108 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToDouble}. - * This class is generated. Do not edit it. - */ -public final class ToDoubleFromBooleanEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToDoubleFromBooleanEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToDoubleFromBoolean"; - } - - @Override - public Block evalVector(Vector v) { - BooleanVector vector = (BooleanVector) v; - int positionCount = v.getPositionCount(); - if (vector.isConstant()) { - return driverContext.blockFactory().newConstantDoubleBlockWith(evalValue(vector, 0), positionCount); - } - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - builder.appendDouble(evalValue(vector, p)); - } - return builder.build(); - } - } - - private static double evalValue(BooleanVector container, int index) { - boolean value = container.getBoolean(index); - return ToDouble.fromBoolean(value); - } - - @Override - public Block evalBlock(Block b) { - BooleanBlock block = (BooleanBlock) b; - int positionCount = block.getPositionCount(); - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - double value = evalValue(block, i); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendDouble(value); - valuesAppended = true; - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static double evalValue(BooleanBlock container, int index) { - boolean value = container.getBoolean(index); - return ToDouble.fromBoolean(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToDoubleFromBooleanEvaluator get(DriverContext context) { - return new ToDoubleFromBooleanEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToDoubleFromBooleanEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToDoubleFromIntEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToDoubleFromIntEvaluator.java deleted file mode 100644 index ef1081f4ebd6a..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToDoubleFromIntEvaluator.java +++ /dev/null @@ -1,108 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToDouble}. - * This class is generated. Do not edit it. - */ -public final class ToDoubleFromIntEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToDoubleFromIntEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToDoubleFromInt"; - } - - @Override - public Block evalVector(Vector v) { - IntVector vector = (IntVector) v; - int positionCount = v.getPositionCount(); - if (vector.isConstant()) { - return driverContext.blockFactory().newConstantDoubleBlockWith(evalValue(vector, 0), positionCount); - } - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - builder.appendDouble(evalValue(vector, p)); - } - return builder.build(); - } - } - - private static double evalValue(IntVector container, int index) { - int value = container.getInt(index); - return ToDouble.fromInt(value); - } - - @Override - public Block evalBlock(Block b) { - IntBlock block = (IntBlock) b; - int positionCount = block.getPositionCount(); - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - double value = evalValue(block, i); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendDouble(value); - valuesAppended = true; - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static double evalValue(IntBlock container, int index) { - int value = container.getInt(index); - return ToDouble.fromInt(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToDoubleFromIntEvaluator get(DriverContext context) { - return new ToDoubleFromIntEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToDoubleFromIntEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToDoubleFromLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToDoubleFromLongEvaluator.java deleted file mode 100644 index fc78d9cfebc01..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToDoubleFromLongEvaluator.java +++ /dev/null @@ -1,108 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToDouble}. - * This class is generated. Do not edit it. - */ -public final class ToDoubleFromLongEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToDoubleFromLongEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToDoubleFromLong"; - } - - @Override - public Block evalVector(Vector v) { - LongVector vector = (LongVector) v; - int positionCount = v.getPositionCount(); - if (vector.isConstant()) { - return driverContext.blockFactory().newConstantDoubleBlockWith(evalValue(vector, 0), positionCount); - } - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - builder.appendDouble(evalValue(vector, p)); - } - return builder.build(); - } - } - - private static double evalValue(LongVector container, int index) { - long value = container.getLong(index); - return ToDouble.fromLong(value); - } - - @Override - public Block evalBlock(Block b) { - LongBlock block = (LongBlock) b; - int positionCount = block.getPositionCount(); - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - double value = evalValue(block, i); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendDouble(value); - valuesAppended = true; - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static double evalValue(LongBlock container, int index) { - long value = container.getLong(index); - return ToDouble.fromLong(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToDoubleFromLongEvaluator get(DriverContext context) { - return new ToDoubleFromLongEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToDoubleFromLongEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToDoubleFromStringEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToDoubleFromStringEvaluator.java deleted file mode 100644 index 209b15ef21a2f..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToDoubleFromStringEvaluator.java +++ /dev/null @@ -1,126 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.InvalidArgumentException; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToDouble}. - * This class is generated. Do not edit it. - */ -public final class ToDoubleFromStringEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToDoubleFromStringEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToDoubleFromString"; - } - - @Override - public Block evalVector(Vector v) { - BytesRefVector vector = (BytesRefVector) v; - int positionCount = v.getPositionCount(); - BytesRef scratchPad = new BytesRef(); - if (vector.isConstant()) { - try { - return driverContext.blockFactory().newConstantDoubleBlockWith(evalValue(vector, 0, scratchPad), positionCount); - } catch (InvalidArgumentException e) { - registerException(e); - return driverContext.blockFactory().newConstantNullBlock(positionCount); - } - } - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - try { - builder.appendDouble(evalValue(vector, p, scratchPad)); - } catch (InvalidArgumentException e) { - registerException(e); - builder.appendNull(); - } - } - return builder.build(); - } - } - - private static double evalValue(BytesRefVector container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToDouble.fromKeyword(value); - } - - @Override - public Block evalBlock(Block b) { - BytesRefBlock block = (BytesRefBlock) b; - int positionCount = block.getPositionCount(); - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - BytesRef scratchPad = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - try { - double value = evalValue(block, i, scratchPad); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendDouble(value); - valuesAppended = true; - } catch (InvalidArgumentException e) { - registerException(e); - } - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static double evalValue(BytesRefBlock container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToDouble.fromKeyword(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToDoubleFromStringEvaluator get(DriverContext context) { - return new ToDoubleFromStringEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToDoubleFromStringEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToDoubleFromUnsignedLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToDoubleFromUnsignedLongEvaluator.java deleted file mode 100644 index b2e4e5137543a..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToDoubleFromUnsignedLongEvaluator.java +++ /dev/null @@ -1,108 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToDouble}. - * This class is generated. Do not edit it. - */ -public final class ToDoubleFromUnsignedLongEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToDoubleFromUnsignedLongEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToDoubleFromUnsignedLong"; - } - - @Override - public Block evalVector(Vector v) { - LongVector vector = (LongVector) v; - int positionCount = v.getPositionCount(); - if (vector.isConstant()) { - return driverContext.blockFactory().newConstantDoubleBlockWith(evalValue(vector, 0), positionCount); - } - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - builder.appendDouble(evalValue(vector, p)); - } - return builder.build(); - } - } - - private static double evalValue(LongVector container, int index) { - long value = container.getLong(index); - return ToDouble.fromUnsignedLong(value); - } - - @Override - public Block evalBlock(Block b) { - LongBlock block = (LongBlock) b; - int positionCount = block.getPositionCount(); - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - double value = evalValue(block, i); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendDouble(value); - valuesAppended = true; - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static double evalValue(LongBlock container, int index) { - long value = container.getLong(index); - return ToDouble.fromUnsignedLong(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToDoubleFromUnsignedLongEvaluator get(DriverContext context) { - return new ToDoubleFromUnsignedLongEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToDoubleFromUnsignedLongEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToGeoPointFromStringEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToGeoPointFromStringEvaluator.java deleted file mode 100644 index 7ef047655b49e..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToGeoPointFromStringEvaluator.java +++ /dev/null @@ -1,125 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToGeoPoint}. - * This class is generated. Do not edit it. - */ -public final class ToGeoPointFromStringEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToGeoPointFromStringEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToGeoPointFromString"; - } - - @Override - public Block evalVector(Vector v) { - BytesRefVector vector = (BytesRefVector) v; - int positionCount = v.getPositionCount(); - BytesRef scratchPad = new BytesRef(); - if (vector.isConstant()) { - try { - return driverContext.blockFactory().newConstantBytesRefBlockWith(evalValue(vector, 0, scratchPad), positionCount); - } catch (IllegalArgumentException e) { - registerException(e); - return driverContext.blockFactory().newConstantNullBlock(positionCount); - } - } - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - try { - builder.appendBytesRef(evalValue(vector, p, scratchPad)); - } catch (IllegalArgumentException e) { - registerException(e); - builder.appendNull(); - } - } - return builder.build(); - } - } - - private static BytesRef evalValue(BytesRefVector container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToGeoPoint.fromKeyword(value); - } - - @Override - public Block evalBlock(Block b) { - BytesRefBlock block = (BytesRefBlock) b; - int positionCount = block.getPositionCount(); - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef scratchPad = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - try { - BytesRef value = evalValue(block, i, scratchPad); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendBytesRef(value); - valuesAppended = true; - } catch (IllegalArgumentException e) { - registerException(e); - } - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static BytesRef evalValue(BytesRefBlock container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToGeoPoint.fromKeyword(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToGeoPointFromStringEvaluator get(DriverContext context) { - return new ToGeoPointFromStringEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToGeoPointFromStringEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToGeoShapeFromStringEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToGeoShapeFromStringEvaluator.java deleted file mode 100644 index 68a6087d86953..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToGeoShapeFromStringEvaluator.java +++ /dev/null @@ -1,125 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToGeoShape}. - * This class is generated. Do not edit it. - */ -public final class ToGeoShapeFromStringEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToGeoShapeFromStringEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToGeoShapeFromString"; - } - - @Override - public Block evalVector(Vector v) { - BytesRefVector vector = (BytesRefVector) v; - int positionCount = v.getPositionCount(); - BytesRef scratchPad = new BytesRef(); - if (vector.isConstant()) { - try { - return driverContext.blockFactory().newConstantBytesRefBlockWith(evalValue(vector, 0, scratchPad), positionCount); - } catch (IllegalArgumentException e) { - registerException(e); - return driverContext.blockFactory().newConstantNullBlock(positionCount); - } - } - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - try { - builder.appendBytesRef(evalValue(vector, p, scratchPad)); - } catch (IllegalArgumentException e) { - registerException(e); - builder.appendNull(); - } - } - return builder.build(); - } - } - - private static BytesRef evalValue(BytesRefVector container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToGeoShape.fromKeyword(value); - } - - @Override - public Block evalBlock(Block b) { - BytesRefBlock block = (BytesRefBlock) b; - int positionCount = block.getPositionCount(); - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef scratchPad = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - try { - BytesRef value = evalValue(block, i, scratchPad); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendBytesRef(value); - valuesAppended = true; - } catch (IllegalArgumentException e) { - registerException(e); - } - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static BytesRef evalValue(BytesRefBlock container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToGeoShape.fromKeyword(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToGeoShapeFromStringEvaluator get(DriverContext context) { - return new ToGeoShapeFromStringEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToGeoShapeFromStringEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToIPFromStringEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToIPFromStringEvaluator.java deleted file mode 100644 index bd6e883a6e89e..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToIPFromStringEvaluator.java +++ /dev/null @@ -1,125 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToIP}. - * This class is generated. Do not edit it. - */ -public final class ToIPFromStringEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToIPFromStringEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToIPFromString"; - } - - @Override - public Block evalVector(Vector v) { - BytesRefVector vector = (BytesRefVector) v; - int positionCount = v.getPositionCount(); - BytesRef scratchPad = new BytesRef(); - if (vector.isConstant()) { - try { - return driverContext.blockFactory().newConstantBytesRefBlockWith(evalValue(vector, 0, scratchPad), positionCount); - } catch (IllegalArgumentException e) { - registerException(e); - return driverContext.blockFactory().newConstantNullBlock(positionCount); - } - } - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - try { - builder.appendBytesRef(evalValue(vector, p, scratchPad)); - } catch (IllegalArgumentException e) { - registerException(e); - builder.appendNull(); - } - } - return builder.build(); - } - } - - private static BytesRef evalValue(BytesRefVector container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToIP.fromKeyword(value); - } - - @Override - public Block evalBlock(Block b) { - BytesRefBlock block = (BytesRefBlock) b; - int positionCount = block.getPositionCount(); - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef scratchPad = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - try { - BytesRef value = evalValue(block, i, scratchPad); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendBytesRef(value); - valuesAppended = true; - } catch (IllegalArgumentException e) { - registerException(e); - } - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static BytesRef evalValue(BytesRefBlock container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToIP.fromKeyword(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToIPFromStringEvaluator get(DriverContext context) { - return new ToIPFromStringEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToIPFromStringEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToIntegerFromBooleanEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToIntegerFromBooleanEvaluator.java deleted file mode 100644 index f778deb32865f..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToIntegerFromBooleanEvaluator.java +++ /dev/null @@ -1,108 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToInteger}. - * This class is generated. Do not edit it. - */ -public final class ToIntegerFromBooleanEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToIntegerFromBooleanEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToIntegerFromBoolean"; - } - - @Override - public Block evalVector(Vector v) { - BooleanVector vector = (BooleanVector) v; - int positionCount = v.getPositionCount(); - if (vector.isConstant()) { - return driverContext.blockFactory().newConstantIntBlockWith(evalValue(vector, 0), positionCount); - } - try (IntBlock.Builder builder = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - builder.appendInt(evalValue(vector, p)); - } - return builder.build(); - } - } - - private static int evalValue(BooleanVector container, int index) { - boolean value = container.getBoolean(index); - return ToInteger.fromBoolean(value); - } - - @Override - public Block evalBlock(Block b) { - BooleanBlock block = (BooleanBlock) b; - int positionCount = block.getPositionCount(); - try (IntBlock.Builder builder = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - int value = evalValue(block, i); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendInt(value); - valuesAppended = true; - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static int evalValue(BooleanBlock container, int index) { - boolean value = container.getBoolean(index); - return ToInteger.fromBoolean(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToIntegerFromBooleanEvaluator get(DriverContext context) { - return new ToIntegerFromBooleanEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToIntegerFromBooleanEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToIntegerFromDoubleEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToIntegerFromDoubleEvaluator.java deleted file mode 100644 index 329269bafd9ba..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToIntegerFromDoubleEvaluator.java +++ /dev/null @@ -1,123 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.InvalidArgumentException; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToInteger}. - * This class is generated. Do not edit it. - */ -public final class ToIntegerFromDoubleEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToIntegerFromDoubleEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToIntegerFromDouble"; - } - - @Override - public Block evalVector(Vector v) { - DoubleVector vector = (DoubleVector) v; - int positionCount = v.getPositionCount(); - if (vector.isConstant()) { - try { - return driverContext.blockFactory().newConstantIntBlockWith(evalValue(vector, 0), positionCount); - } catch (InvalidArgumentException e) { - registerException(e); - return driverContext.blockFactory().newConstantNullBlock(positionCount); - } - } - try (IntBlock.Builder builder = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - try { - builder.appendInt(evalValue(vector, p)); - } catch (InvalidArgumentException e) { - registerException(e); - builder.appendNull(); - } - } - return builder.build(); - } - } - - private static int evalValue(DoubleVector container, int index) { - double value = container.getDouble(index); - return ToInteger.fromDouble(value); - } - - @Override - public Block evalBlock(Block b) { - DoubleBlock block = (DoubleBlock) b; - int positionCount = block.getPositionCount(); - try (IntBlock.Builder builder = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - try { - int value = evalValue(block, i); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendInt(value); - valuesAppended = true; - } catch (InvalidArgumentException e) { - registerException(e); - } - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static int evalValue(DoubleBlock container, int index) { - double value = container.getDouble(index); - return ToInteger.fromDouble(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToIntegerFromDoubleEvaluator get(DriverContext context) { - return new ToIntegerFromDoubleEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToIntegerFromDoubleEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToIntegerFromLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToIntegerFromLongEvaluator.java deleted file mode 100644 index f9b3cb60dad2c..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToIntegerFromLongEvaluator.java +++ /dev/null @@ -1,123 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.InvalidArgumentException; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToInteger}. - * This class is generated. Do not edit it. - */ -public final class ToIntegerFromLongEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToIntegerFromLongEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToIntegerFromLong"; - } - - @Override - public Block evalVector(Vector v) { - LongVector vector = (LongVector) v; - int positionCount = v.getPositionCount(); - if (vector.isConstant()) { - try { - return driverContext.blockFactory().newConstantIntBlockWith(evalValue(vector, 0), positionCount); - } catch (InvalidArgumentException e) { - registerException(e); - return driverContext.blockFactory().newConstantNullBlock(positionCount); - } - } - try (IntBlock.Builder builder = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - try { - builder.appendInt(evalValue(vector, p)); - } catch (InvalidArgumentException e) { - registerException(e); - builder.appendNull(); - } - } - return builder.build(); - } - } - - private static int evalValue(LongVector container, int index) { - long value = container.getLong(index); - return ToInteger.fromLong(value); - } - - @Override - public Block evalBlock(Block b) { - LongBlock block = (LongBlock) b; - int positionCount = block.getPositionCount(); - try (IntBlock.Builder builder = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - try { - int value = evalValue(block, i); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendInt(value); - valuesAppended = true; - } catch (InvalidArgumentException e) { - registerException(e); - } - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static int evalValue(LongBlock container, int index) { - long value = container.getLong(index); - return ToInteger.fromLong(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToIntegerFromLongEvaluator get(DriverContext context) { - return new ToIntegerFromLongEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToIntegerFromLongEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToIntegerFromStringEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToIntegerFromStringEvaluator.java deleted file mode 100644 index ef91bf890cd23..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToIntegerFromStringEvaluator.java +++ /dev/null @@ -1,126 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.InvalidArgumentException; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToInteger}. - * This class is generated. Do not edit it. - */ -public final class ToIntegerFromStringEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToIntegerFromStringEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToIntegerFromString"; - } - - @Override - public Block evalVector(Vector v) { - BytesRefVector vector = (BytesRefVector) v; - int positionCount = v.getPositionCount(); - BytesRef scratchPad = new BytesRef(); - if (vector.isConstant()) { - try { - return driverContext.blockFactory().newConstantIntBlockWith(evalValue(vector, 0, scratchPad), positionCount); - } catch (InvalidArgumentException e) { - registerException(e); - return driverContext.blockFactory().newConstantNullBlock(positionCount); - } - } - try (IntBlock.Builder builder = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - try { - builder.appendInt(evalValue(vector, p, scratchPad)); - } catch (InvalidArgumentException e) { - registerException(e); - builder.appendNull(); - } - } - return builder.build(); - } - } - - private static int evalValue(BytesRefVector container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToInteger.fromKeyword(value); - } - - @Override - public Block evalBlock(Block b) { - BytesRefBlock block = (BytesRefBlock) b; - int positionCount = block.getPositionCount(); - try (IntBlock.Builder builder = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - BytesRef scratchPad = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - try { - int value = evalValue(block, i, scratchPad); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendInt(value); - valuesAppended = true; - } catch (InvalidArgumentException e) { - registerException(e); - } - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static int evalValue(BytesRefBlock container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToInteger.fromKeyword(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToIntegerFromStringEvaluator get(DriverContext context) { - return new ToIntegerFromStringEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToIntegerFromStringEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToIntegerFromUnsignedLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToIntegerFromUnsignedLongEvaluator.java deleted file mode 100644 index 34128e44f1500..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToIntegerFromUnsignedLongEvaluator.java +++ /dev/null @@ -1,123 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.InvalidArgumentException; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToInteger}. - * This class is generated. Do not edit it. - */ -public final class ToIntegerFromUnsignedLongEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToIntegerFromUnsignedLongEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToIntegerFromUnsignedLong"; - } - - @Override - public Block evalVector(Vector v) { - LongVector vector = (LongVector) v; - int positionCount = v.getPositionCount(); - if (vector.isConstant()) { - try { - return driverContext.blockFactory().newConstantIntBlockWith(evalValue(vector, 0), positionCount); - } catch (InvalidArgumentException e) { - registerException(e); - return driverContext.blockFactory().newConstantNullBlock(positionCount); - } - } - try (IntBlock.Builder builder = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - try { - builder.appendInt(evalValue(vector, p)); - } catch (InvalidArgumentException e) { - registerException(e); - builder.appendNull(); - } - } - return builder.build(); - } - } - - private static int evalValue(LongVector container, int index) { - long value = container.getLong(index); - return ToInteger.fromUnsignedLong(value); - } - - @Override - public Block evalBlock(Block b) { - LongBlock block = (LongBlock) b; - int positionCount = block.getPositionCount(); - try (IntBlock.Builder builder = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - try { - int value = evalValue(block, i); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendInt(value); - valuesAppended = true; - } catch (InvalidArgumentException e) { - registerException(e); - } - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static int evalValue(LongBlock container, int index) { - long value = container.getLong(index); - return ToInteger.fromUnsignedLong(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToIntegerFromUnsignedLongEvaluator get(DriverContext context) { - return new ToIntegerFromUnsignedLongEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToIntegerFromUnsignedLongEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToLongFromBooleanEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToLongFromBooleanEvaluator.java deleted file mode 100644 index 7d6c145405e56..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToLongFromBooleanEvaluator.java +++ /dev/null @@ -1,108 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToLong}. - * This class is generated. Do not edit it. - */ -public final class ToLongFromBooleanEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToLongFromBooleanEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToLongFromBoolean"; - } - - @Override - public Block evalVector(Vector v) { - BooleanVector vector = (BooleanVector) v; - int positionCount = v.getPositionCount(); - if (vector.isConstant()) { - return driverContext.blockFactory().newConstantLongBlockWith(evalValue(vector, 0), positionCount); - } - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - builder.appendLong(evalValue(vector, p)); - } - return builder.build(); - } - } - - private static long evalValue(BooleanVector container, int index) { - boolean value = container.getBoolean(index); - return ToLong.fromBoolean(value); - } - - @Override - public Block evalBlock(Block b) { - BooleanBlock block = (BooleanBlock) b; - int positionCount = block.getPositionCount(); - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - long value = evalValue(block, i); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendLong(value); - valuesAppended = true; - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static long evalValue(BooleanBlock container, int index) { - boolean value = container.getBoolean(index); - return ToLong.fromBoolean(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToLongFromBooleanEvaluator get(DriverContext context) { - return new ToLongFromBooleanEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToLongFromBooleanEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToLongFromDoubleEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToLongFromDoubleEvaluator.java deleted file mode 100644 index 03daa257e5af2..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToLongFromDoubleEvaluator.java +++ /dev/null @@ -1,123 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.InvalidArgumentException; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToLong}. - * This class is generated. Do not edit it. - */ -public final class ToLongFromDoubleEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToLongFromDoubleEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToLongFromDouble"; - } - - @Override - public Block evalVector(Vector v) { - DoubleVector vector = (DoubleVector) v; - int positionCount = v.getPositionCount(); - if (vector.isConstant()) { - try { - return driverContext.blockFactory().newConstantLongBlockWith(evalValue(vector, 0), positionCount); - } catch (InvalidArgumentException e) { - registerException(e); - return driverContext.blockFactory().newConstantNullBlock(positionCount); - } - } - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - try { - builder.appendLong(evalValue(vector, p)); - } catch (InvalidArgumentException e) { - registerException(e); - builder.appendNull(); - } - } - return builder.build(); - } - } - - private static long evalValue(DoubleVector container, int index) { - double value = container.getDouble(index); - return ToLong.fromDouble(value); - } - - @Override - public Block evalBlock(Block b) { - DoubleBlock block = (DoubleBlock) b; - int positionCount = block.getPositionCount(); - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - try { - long value = evalValue(block, i); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendLong(value); - valuesAppended = true; - } catch (InvalidArgumentException e) { - registerException(e); - } - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static long evalValue(DoubleBlock container, int index) { - double value = container.getDouble(index); - return ToLong.fromDouble(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToLongFromDoubleEvaluator get(DriverContext context) { - return new ToLongFromDoubleEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToLongFromDoubleEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToLongFromIntEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToLongFromIntEvaluator.java deleted file mode 100644 index dc3a9578ffd9b..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToLongFromIntEvaluator.java +++ /dev/null @@ -1,108 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToLong}. - * This class is generated. Do not edit it. - */ -public final class ToLongFromIntEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToLongFromIntEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToLongFromInt"; - } - - @Override - public Block evalVector(Vector v) { - IntVector vector = (IntVector) v; - int positionCount = v.getPositionCount(); - if (vector.isConstant()) { - return driverContext.blockFactory().newConstantLongBlockWith(evalValue(vector, 0), positionCount); - } - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - builder.appendLong(evalValue(vector, p)); - } - return builder.build(); - } - } - - private static long evalValue(IntVector container, int index) { - int value = container.getInt(index); - return ToLong.fromInt(value); - } - - @Override - public Block evalBlock(Block b) { - IntBlock block = (IntBlock) b; - int positionCount = block.getPositionCount(); - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - long value = evalValue(block, i); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendLong(value); - valuesAppended = true; - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static long evalValue(IntBlock container, int index) { - int value = container.getInt(index); - return ToLong.fromInt(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToLongFromIntEvaluator get(DriverContext context) { - return new ToLongFromIntEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToLongFromIntEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToLongFromStringEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToLongFromStringEvaluator.java deleted file mode 100644 index 0d7a2cb9d7459..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToLongFromStringEvaluator.java +++ /dev/null @@ -1,126 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.InvalidArgumentException; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToLong}. - * This class is generated. Do not edit it. - */ -public final class ToLongFromStringEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToLongFromStringEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToLongFromString"; - } - - @Override - public Block evalVector(Vector v) { - BytesRefVector vector = (BytesRefVector) v; - int positionCount = v.getPositionCount(); - BytesRef scratchPad = new BytesRef(); - if (vector.isConstant()) { - try { - return driverContext.blockFactory().newConstantLongBlockWith(evalValue(vector, 0, scratchPad), positionCount); - } catch (InvalidArgumentException e) { - registerException(e); - return driverContext.blockFactory().newConstantNullBlock(positionCount); - } - } - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - try { - builder.appendLong(evalValue(vector, p, scratchPad)); - } catch (InvalidArgumentException e) { - registerException(e); - builder.appendNull(); - } - } - return builder.build(); - } - } - - private static long evalValue(BytesRefVector container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToLong.fromKeyword(value); - } - - @Override - public Block evalBlock(Block b) { - BytesRefBlock block = (BytesRefBlock) b; - int positionCount = block.getPositionCount(); - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - BytesRef scratchPad = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - try { - long value = evalValue(block, i, scratchPad); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendLong(value); - valuesAppended = true; - } catch (InvalidArgumentException e) { - registerException(e); - } - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static long evalValue(BytesRefBlock container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToLong.fromKeyword(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToLongFromStringEvaluator get(DriverContext context) { - return new ToLongFromStringEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToLongFromStringEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToLongFromUnsignedLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToLongFromUnsignedLongEvaluator.java deleted file mode 100644 index b5999d1a4e1ab..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToLongFromUnsignedLongEvaluator.java +++ /dev/null @@ -1,122 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.InvalidArgumentException; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToLong}. - * This class is generated. Do not edit it. - */ -public final class ToLongFromUnsignedLongEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToLongFromUnsignedLongEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToLongFromUnsignedLong"; - } - - @Override - public Block evalVector(Vector v) { - LongVector vector = (LongVector) v; - int positionCount = v.getPositionCount(); - if (vector.isConstant()) { - try { - return driverContext.blockFactory().newConstantLongBlockWith(evalValue(vector, 0), positionCount); - } catch (InvalidArgumentException e) { - registerException(e); - return driverContext.blockFactory().newConstantNullBlock(positionCount); - } - } - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - try { - builder.appendLong(evalValue(vector, p)); - } catch (InvalidArgumentException e) { - registerException(e); - builder.appendNull(); - } - } - return builder.build(); - } - } - - private static long evalValue(LongVector container, int index) { - long value = container.getLong(index); - return ToLong.fromUnsignedLong(value); - } - - @Override - public Block evalBlock(Block b) { - LongBlock block = (LongBlock) b; - int positionCount = block.getPositionCount(); - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - try { - long value = evalValue(block, i); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendLong(value); - valuesAppended = true; - } catch (InvalidArgumentException e) { - registerException(e); - } - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static long evalValue(LongBlock container, int index) { - long value = container.getLong(index); - return ToLong.fromUnsignedLong(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToLongFromUnsignedLongEvaluator get(DriverContext context) { - return new ToLongFromUnsignedLongEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToLongFromUnsignedLongEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToRadiansEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToRadiansEvaluator.java deleted file mode 100644 index 3bd997d0b1d38..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToRadiansEvaluator.java +++ /dev/null @@ -1,107 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToRadians}. - * This class is generated. Do not edit it. - */ -public final class ToRadiansEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToRadiansEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToRadians"; - } - - @Override - public Block evalVector(Vector v) { - DoubleVector vector = (DoubleVector) v; - int positionCount = v.getPositionCount(); - if (vector.isConstant()) { - return driverContext.blockFactory().newConstantDoubleBlockWith(evalValue(vector, 0), positionCount); - } - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - builder.appendDouble(evalValue(vector, p)); - } - return builder.build(); - } - } - - private static double evalValue(DoubleVector container, int index) { - double value = container.getDouble(index); - return ToRadians.process(value); - } - - @Override - public Block evalBlock(Block b) { - DoubleBlock block = (DoubleBlock) b; - int positionCount = block.getPositionCount(); - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - double value = evalValue(block, i); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendDouble(value); - valuesAppended = true; - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static double evalValue(DoubleBlock container, int index) { - double value = container.getDouble(index); - return ToRadians.process(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToRadiansEvaluator get(DriverContext context) { - return new ToRadiansEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToRadiansEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromBooleanEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromBooleanEvaluator.java deleted file mode 100644 index a68cd61a8c470..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromBooleanEvaluator.java +++ /dev/null @@ -1,109 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToString}. - * This class is generated. Do not edit it. - */ -public final class ToStringFromBooleanEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToStringFromBooleanEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToStringFromBoolean"; - } - - @Override - public Block evalVector(Vector v) { - BooleanVector vector = (BooleanVector) v; - int positionCount = v.getPositionCount(); - if (vector.isConstant()) { - return driverContext.blockFactory().newConstantBytesRefBlockWith(evalValue(vector, 0), positionCount); - } - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - builder.appendBytesRef(evalValue(vector, p)); - } - return builder.build(); - } - } - - private static BytesRef evalValue(BooleanVector container, int index) { - boolean value = container.getBoolean(index); - return ToString.fromBoolean(value); - } - - @Override - public Block evalBlock(Block b) { - BooleanBlock block = (BooleanBlock) b; - int positionCount = block.getPositionCount(); - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - BytesRef value = evalValue(block, i); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendBytesRef(value); - valuesAppended = true; - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static BytesRef evalValue(BooleanBlock container, int index) { - boolean value = container.getBoolean(index); - return ToString.fromBoolean(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToStringFromBooleanEvaluator get(DriverContext context) { - return new ToStringFromBooleanEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToStringFromBooleanEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromCartesianPointEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromCartesianPointEvaluator.java deleted file mode 100644 index b15f77608598d..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromCartesianPointEvaluator.java +++ /dev/null @@ -1,110 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToString}. - * This class is generated. Do not edit it. - */ -public final class ToStringFromCartesianPointEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToStringFromCartesianPointEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToStringFromCartesianPoint"; - } - - @Override - public Block evalVector(Vector v) { - BytesRefVector vector = (BytesRefVector) v; - int positionCount = v.getPositionCount(); - BytesRef scratchPad = new BytesRef(); - if (vector.isConstant()) { - return driverContext.blockFactory().newConstantBytesRefBlockWith(evalValue(vector, 0, scratchPad), positionCount); - } - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - builder.appendBytesRef(evalValue(vector, p, scratchPad)); - } - return builder.build(); - } - } - - private static BytesRef evalValue(BytesRefVector container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToString.fromCartesianPoint(value); - } - - @Override - public Block evalBlock(Block b) { - BytesRefBlock block = (BytesRefBlock) b; - int positionCount = block.getPositionCount(); - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef scratchPad = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - BytesRef value = evalValue(block, i, scratchPad); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendBytesRef(value); - valuesAppended = true; - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static BytesRef evalValue(BytesRefBlock container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToString.fromCartesianPoint(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToStringFromCartesianPointEvaluator get(DriverContext context) { - return new ToStringFromCartesianPointEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToStringFromCartesianPointEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromCartesianShapeEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromCartesianShapeEvaluator.java deleted file mode 100644 index 5e466ddfbfddc..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromCartesianShapeEvaluator.java +++ /dev/null @@ -1,110 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToString}. - * This class is generated. Do not edit it. - */ -public final class ToStringFromCartesianShapeEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToStringFromCartesianShapeEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToStringFromCartesianShape"; - } - - @Override - public Block evalVector(Vector v) { - BytesRefVector vector = (BytesRefVector) v; - int positionCount = v.getPositionCount(); - BytesRef scratchPad = new BytesRef(); - if (vector.isConstant()) { - return driverContext.blockFactory().newConstantBytesRefBlockWith(evalValue(vector, 0, scratchPad), positionCount); - } - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - builder.appendBytesRef(evalValue(vector, p, scratchPad)); - } - return builder.build(); - } - } - - private static BytesRef evalValue(BytesRefVector container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToString.fromCartesianShape(value); - } - - @Override - public Block evalBlock(Block b) { - BytesRefBlock block = (BytesRefBlock) b; - int positionCount = block.getPositionCount(); - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef scratchPad = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - BytesRef value = evalValue(block, i, scratchPad); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendBytesRef(value); - valuesAppended = true; - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static BytesRef evalValue(BytesRefBlock container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToString.fromCartesianShape(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToStringFromCartesianShapeEvaluator get(DriverContext context) { - return new ToStringFromCartesianShapeEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToStringFromCartesianShapeEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromDatetimeEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromDatetimeEvaluator.java deleted file mode 100644 index 569881ad30b61..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromDatetimeEvaluator.java +++ /dev/null @@ -1,109 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToString}. - * This class is generated. Do not edit it. - */ -public final class ToStringFromDatetimeEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToStringFromDatetimeEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToStringFromDatetime"; - } - - @Override - public Block evalVector(Vector v) { - LongVector vector = (LongVector) v; - int positionCount = v.getPositionCount(); - if (vector.isConstant()) { - return driverContext.blockFactory().newConstantBytesRefBlockWith(evalValue(vector, 0), positionCount); - } - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - builder.appendBytesRef(evalValue(vector, p)); - } - return builder.build(); - } - } - - private static BytesRef evalValue(LongVector container, int index) { - long value = container.getLong(index); - return ToString.fromDatetime(value); - } - - @Override - public Block evalBlock(Block b) { - LongBlock block = (LongBlock) b; - int positionCount = block.getPositionCount(); - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - BytesRef value = evalValue(block, i); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendBytesRef(value); - valuesAppended = true; - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static BytesRef evalValue(LongBlock container, int index) { - long value = container.getLong(index); - return ToString.fromDatetime(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToStringFromDatetimeEvaluator get(DriverContext context) { - return new ToStringFromDatetimeEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToStringFromDatetimeEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromDoubleEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromDoubleEvaluator.java deleted file mode 100644 index 69c33e07c1650..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromDoubleEvaluator.java +++ /dev/null @@ -1,109 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToString}. - * This class is generated. Do not edit it. - */ -public final class ToStringFromDoubleEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToStringFromDoubleEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToStringFromDouble"; - } - - @Override - public Block evalVector(Vector v) { - DoubleVector vector = (DoubleVector) v; - int positionCount = v.getPositionCount(); - if (vector.isConstant()) { - return driverContext.blockFactory().newConstantBytesRefBlockWith(evalValue(vector, 0), positionCount); - } - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - builder.appendBytesRef(evalValue(vector, p)); - } - return builder.build(); - } - } - - private static BytesRef evalValue(DoubleVector container, int index) { - double value = container.getDouble(index); - return ToString.fromDouble(value); - } - - @Override - public Block evalBlock(Block b) { - DoubleBlock block = (DoubleBlock) b; - int positionCount = block.getPositionCount(); - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - BytesRef value = evalValue(block, i); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendBytesRef(value); - valuesAppended = true; - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static BytesRef evalValue(DoubleBlock container, int index) { - double value = container.getDouble(index); - return ToString.fromDouble(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToStringFromDoubleEvaluator get(DriverContext context) { - return new ToStringFromDoubleEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToStringFromDoubleEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromGeoPointEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromGeoPointEvaluator.java deleted file mode 100644 index 32fe16075e046..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromGeoPointEvaluator.java +++ /dev/null @@ -1,110 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToString}. - * This class is generated. Do not edit it. - */ -public final class ToStringFromGeoPointEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToStringFromGeoPointEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToStringFromGeoPoint"; - } - - @Override - public Block evalVector(Vector v) { - BytesRefVector vector = (BytesRefVector) v; - int positionCount = v.getPositionCount(); - BytesRef scratchPad = new BytesRef(); - if (vector.isConstant()) { - return driverContext.blockFactory().newConstantBytesRefBlockWith(evalValue(vector, 0, scratchPad), positionCount); - } - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - builder.appendBytesRef(evalValue(vector, p, scratchPad)); - } - return builder.build(); - } - } - - private static BytesRef evalValue(BytesRefVector container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToString.fromGeoPoint(value); - } - - @Override - public Block evalBlock(Block b) { - BytesRefBlock block = (BytesRefBlock) b; - int positionCount = block.getPositionCount(); - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef scratchPad = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - BytesRef value = evalValue(block, i, scratchPad); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendBytesRef(value); - valuesAppended = true; - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static BytesRef evalValue(BytesRefBlock container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToString.fromGeoPoint(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToStringFromGeoPointEvaluator get(DriverContext context) { - return new ToStringFromGeoPointEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToStringFromGeoPointEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromGeoShapeEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromGeoShapeEvaluator.java deleted file mode 100644 index df8e86e58fa69..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromGeoShapeEvaluator.java +++ /dev/null @@ -1,110 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToString}. - * This class is generated. Do not edit it. - */ -public final class ToStringFromGeoShapeEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToStringFromGeoShapeEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToStringFromGeoShape"; - } - - @Override - public Block evalVector(Vector v) { - BytesRefVector vector = (BytesRefVector) v; - int positionCount = v.getPositionCount(); - BytesRef scratchPad = new BytesRef(); - if (vector.isConstant()) { - return driverContext.blockFactory().newConstantBytesRefBlockWith(evalValue(vector, 0, scratchPad), positionCount); - } - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - builder.appendBytesRef(evalValue(vector, p, scratchPad)); - } - return builder.build(); - } - } - - private static BytesRef evalValue(BytesRefVector container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToString.fromGeoShape(value); - } - - @Override - public Block evalBlock(Block b) { - BytesRefBlock block = (BytesRefBlock) b; - int positionCount = block.getPositionCount(); - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef scratchPad = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - BytesRef value = evalValue(block, i, scratchPad); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendBytesRef(value); - valuesAppended = true; - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static BytesRef evalValue(BytesRefBlock container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToString.fromGeoShape(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToStringFromGeoShapeEvaluator get(DriverContext context) { - return new ToStringFromGeoShapeEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToStringFromGeoShapeEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromIPEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromIPEvaluator.java deleted file mode 100644 index 00fb269699fe3..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromIPEvaluator.java +++ /dev/null @@ -1,110 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToString}. - * This class is generated. Do not edit it. - */ -public final class ToStringFromIPEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToStringFromIPEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToStringFromIP"; - } - - @Override - public Block evalVector(Vector v) { - BytesRefVector vector = (BytesRefVector) v; - int positionCount = v.getPositionCount(); - BytesRef scratchPad = new BytesRef(); - if (vector.isConstant()) { - return driverContext.blockFactory().newConstantBytesRefBlockWith(evalValue(vector, 0, scratchPad), positionCount); - } - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - builder.appendBytesRef(evalValue(vector, p, scratchPad)); - } - return builder.build(); - } - } - - private static BytesRef evalValue(BytesRefVector container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToString.fromIP(value); - } - - @Override - public Block evalBlock(Block b) { - BytesRefBlock block = (BytesRefBlock) b; - int positionCount = block.getPositionCount(); - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef scratchPad = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - BytesRef value = evalValue(block, i, scratchPad); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendBytesRef(value); - valuesAppended = true; - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static BytesRef evalValue(BytesRefBlock container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToString.fromIP(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToStringFromIPEvaluator get(DriverContext context) { - return new ToStringFromIPEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToStringFromIPEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromIntEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromIntEvaluator.java deleted file mode 100644 index 6e371c90adb28..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromIntEvaluator.java +++ /dev/null @@ -1,109 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToString}. - * This class is generated. Do not edit it. - */ -public final class ToStringFromIntEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToStringFromIntEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToStringFromInt"; - } - - @Override - public Block evalVector(Vector v) { - IntVector vector = (IntVector) v; - int positionCount = v.getPositionCount(); - if (vector.isConstant()) { - return driverContext.blockFactory().newConstantBytesRefBlockWith(evalValue(vector, 0), positionCount); - } - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - builder.appendBytesRef(evalValue(vector, p)); - } - return builder.build(); - } - } - - private static BytesRef evalValue(IntVector container, int index) { - int value = container.getInt(index); - return ToString.fromDouble(value); - } - - @Override - public Block evalBlock(Block b) { - IntBlock block = (IntBlock) b; - int positionCount = block.getPositionCount(); - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - BytesRef value = evalValue(block, i); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendBytesRef(value); - valuesAppended = true; - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static BytesRef evalValue(IntBlock container, int index) { - int value = container.getInt(index); - return ToString.fromDouble(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToStringFromIntEvaluator get(DriverContext context) { - return new ToStringFromIntEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToStringFromIntEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromLongEvaluator.java deleted file mode 100644 index 3dc8f738d7b1d..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromLongEvaluator.java +++ /dev/null @@ -1,109 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToString}. - * This class is generated. Do not edit it. - */ -public final class ToStringFromLongEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToStringFromLongEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToStringFromLong"; - } - - @Override - public Block evalVector(Vector v) { - LongVector vector = (LongVector) v; - int positionCount = v.getPositionCount(); - if (vector.isConstant()) { - return driverContext.blockFactory().newConstantBytesRefBlockWith(evalValue(vector, 0), positionCount); - } - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - builder.appendBytesRef(evalValue(vector, p)); - } - return builder.build(); - } - } - - private static BytesRef evalValue(LongVector container, int index) { - long value = container.getLong(index); - return ToString.fromDouble(value); - } - - @Override - public Block evalBlock(Block b) { - LongBlock block = (LongBlock) b; - int positionCount = block.getPositionCount(); - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - BytesRef value = evalValue(block, i); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendBytesRef(value); - valuesAppended = true; - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static BytesRef evalValue(LongBlock container, int index) { - long value = container.getLong(index); - return ToString.fromDouble(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToStringFromLongEvaluator get(DriverContext context) { - return new ToStringFromLongEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToStringFromLongEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromUnsignedLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromUnsignedLongEvaluator.java deleted file mode 100644 index 4bce2c1fec40f..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromUnsignedLongEvaluator.java +++ /dev/null @@ -1,109 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToString}. - * This class is generated. Do not edit it. - */ -public final class ToStringFromUnsignedLongEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToStringFromUnsignedLongEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToStringFromUnsignedLong"; - } - - @Override - public Block evalVector(Vector v) { - LongVector vector = (LongVector) v; - int positionCount = v.getPositionCount(); - if (vector.isConstant()) { - return driverContext.blockFactory().newConstantBytesRefBlockWith(evalValue(vector, 0), positionCount); - } - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - builder.appendBytesRef(evalValue(vector, p)); - } - return builder.build(); - } - } - - private static BytesRef evalValue(LongVector container, int index) { - long value = container.getLong(index); - return ToString.fromUnsignedLong(value); - } - - @Override - public Block evalBlock(Block b) { - LongBlock block = (LongBlock) b; - int positionCount = block.getPositionCount(); - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - BytesRef value = evalValue(block, i); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendBytesRef(value); - valuesAppended = true; - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static BytesRef evalValue(LongBlock container, int index) { - long value = container.getLong(index); - return ToString.fromUnsignedLong(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToStringFromUnsignedLongEvaluator get(DriverContext context) { - return new ToStringFromUnsignedLongEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToStringFromUnsignedLongEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromVersionEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromVersionEvaluator.java deleted file mode 100644 index a37696e149d4c..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToStringFromVersionEvaluator.java +++ /dev/null @@ -1,110 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToString}. - * This class is generated. Do not edit it. - */ -public final class ToStringFromVersionEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToStringFromVersionEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToStringFromVersion"; - } - - @Override - public Block evalVector(Vector v) { - BytesRefVector vector = (BytesRefVector) v; - int positionCount = v.getPositionCount(); - BytesRef scratchPad = new BytesRef(); - if (vector.isConstant()) { - return driverContext.blockFactory().newConstantBytesRefBlockWith(evalValue(vector, 0, scratchPad), positionCount); - } - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - builder.appendBytesRef(evalValue(vector, p, scratchPad)); - } - return builder.build(); - } - } - - private static BytesRef evalValue(BytesRefVector container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToString.fromVersion(value); - } - - @Override - public Block evalBlock(Block b) { - BytesRefBlock block = (BytesRefBlock) b; - int positionCount = block.getPositionCount(); - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef scratchPad = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - BytesRef value = evalValue(block, i, scratchPad); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendBytesRef(value); - valuesAppended = true; - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static BytesRef evalValue(BytesRefBlock container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToString.fromVersion(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToStringFromVersionEvaluator get(DriverContext context) { - return new ToStringFromVersionEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToStringFromVersionEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToUnsignedLongFromBooleanEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToUnsignedLongFromBooleanEvaluator.java deleted file mode 100644 index 619a4ec09d60b..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToUnsignedLongFromBooleanEvaluator.java +++ /dev/null @@ -1,108 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToUnsignedLong}. - * This class is generated. Do not edit it. - */ -public final class ToUnsignedLongFromBooleanEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToUnsignedLongFromBooleanEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToUnsignedLongFromBoolean"; - } - - @Override - public Block evalVector(Vector v) { - BooleanVector vector = (BooleanVector) v; - int positionCount = v.getPositionCount(); - if (vector.isConstant()) { - return driverContext.blockFactory().newConstantLongBlockWith(evalValue(vector, 0), positionCount); - } - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - builder.appendLong(evalValue(vector, p)); - } - return builder.build(); - } - } - - private static long evalValue(BooleanVector container, int index) { - boolean value = container.getBoolean(index); - return ToUnsignedLong.fromBoolean(value); - } - - @Override - public Block evalBlock(Block b) { - BooleanBlock block = (BooleanBlock) b; - int positionCount = block.getPositionCount(); - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - long value = evalValue(block, i); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendLong(value); - valuesAppended = true; - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static long evalValue(BooleanBlock container, int index) { - boolean value = container.getBoolean(index); - return ToUnsignedLong.fromBoolean(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToUnsignedLongFromBooleanEvaluator get(DriverContext context) { - return new ToUnsignedLongFromBooleanEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToUnsignedLongFromBooleanEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToUnsignedLongFromDoubleEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToUnsignedLongFromDoubleEvaluator.java deleted file mode 100644 index 6a45dcf907889..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToUnsignedLongFromDoubleEvaluator.java +++ /dev/null @@ -1,123 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.InvalidArgumentException; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToUnsignedLong}. - * This class is generated. Do not edit it. - */ -public final class ToUnsignedLongFromDoubleEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToUnsignedLongFromDoubleEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToUnsignedLongFromDouble"; - } - - @Override - public Block evalVector(Vector v) { - DoubleVector vector = (DoubleVector) v; - int positionCount = v.getPositionCount(); - if (vector.isConstant()) { - try { - return driverContext.blockFactory().newConstantLongBlockWith(evalValue(vector, 0), positionCount); - } catch (InvalidArgumentException e) { - registerException(e); - return driverContext.blockFactory().newConstantNullBlock(positionCount); - } - } - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - try { - builder.appendLong(evalValue(vector, p)); - } catch (InvalidArgumentException e) { - registerException(e); - builder.appendNull(); - } - } - return builder.build(); - } - } - - private static long evalValue(DoubleVector container, int index) { - double value = container.getDouble(index); - return ToUnsignedLong.fromDouble(value); - } - - @Override - public Block evalBlock(Block b) { - DoubleBlock block = (DoubleBlock) b; - int positionCount = block.getPositionCount(); - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - try { - long value = evalValue(block, i); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendLong(value); - valuesAppended = true; - } catch (InvalidArgumentException e) { - registerException(e); - } - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static long evalValue(DoubleBlock container, int index) { - double value = container.getDouble(index); - return ToUnsignedLong.fromDouble(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToUnsignedLongFromDoubleEvaluator get(DriverContext context) { - return new ToUnsignedLongFromDoubleEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToUnsignedLongFromDoubleEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToUnsignedLongFromIntEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToUnsignedLongFromIntEvaluator.java deleted file mode 100644 index 703f0729654a8..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToUnsignedLongFromIntEvaluator.java +++ /dev/null @@ -1,123 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.InvalidArgumentException; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToUnsignedLong}. - * This class is generated. Do not edit it. - */ -public final class ToUnsignedLongFromIntEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToUnsignedLongFromIntEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToUnsignedLongFromInt"; - } - - @Override - public Block evalVector(Vector v) { - IntVector vector = (IntVector) v; - int positionCount = v.getPositionCount(); - if (vector.isConstant()) { - try { - return driverContext.blockFactory().newConstantLongBlockWith(evalValue(vector, 0), positionCount); - } catch (InvalidArgumentException e) { - registerException(e); - return driverContext.blockFactory().newConstantNullBlock(positionCount); - } - } - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - try { - builder.appendLong(evalValue(vector, p)); - } catch (InvalidArgumentException e) { - registerException(e); - builder.appendNull(); - } - } - return builder.build(); - } - } - - private static long evalValue(IntVector container, int index) { - int value = container.getInt(index); - return ToUnsignedLong.fromInt(value); - } - - @Override - public Block evalBlock(Block b) { - IntBlock block = (IntBlock) b; - int positionCount = block.getPositionCount(); - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - try { - long value = evalValue(block, i); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendLong(value); - valuesAppended = true; - } catch (InvalidArgumentException e) { - registerException(e); - } - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static long evalValue(IntBlock container, int index) { - int value = container.getInt(index); - return ToUnsignedLong.fromInt(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToUnsignedLongFromIntEvaluator get(DriverContext context) { - return new ToUnsignedLongFromIntEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToUnsignedLongFromIntEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToUnsignedLongFromLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToUnsignedLongFromLongEvaluator.java deleted file mode 100644 index b43b961f5d34a..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToUnsignedLongFromLongEvaluator.java +++ /dev/null @@ -1,122 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.InvalidArgumentException; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToUnsignedLong}. - * This class is generated. Do not edit it. - */ -public final class ToUnsignedLongFromLongEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToUnsignedLongFromLongEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToUnsignedLongFromLong"; - } - - @Override - public Block evalVector(Vector v) { - LongVector vector = (LongVector) v; - int positionCount = v.getPositionCount(); - if (vector.isConstant()) { - try { - return driverContext.blockFactory().newConstantLongBlockWith(evalValue(vector, 0), positionCount); - } catch (InvalidArgumentException e) { - registerException(e); - return driverContext.blockFactory().newConstantNullBlock(positionCount); - } - } - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - try { - builder.appendLong(evalValue(vector, p)); - } catch (InvalidArgumentException e) { - registerException(e); - builder.appendNull(); - } - } - return builder.build(); - } - } - - private static long evalValue(LongVector container, int index) { - long value = container.getLong(index); - return ToUnsignedLong.fromLong(value); - } - - @Override - public Block evalBlock(Block b) { - LongBlock block = (LongBlock) b; - int positionCount = block.getPositionCount(); - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - try { - long value = evalValue(block, i); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendLong(value); - valuesAppended = true; - } catch (InvalidArgumentException e) { - registerException(e); - } - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static long evalValue(LongBlock container, int index) { - long value = container.getLong(index); - return ToUnsignedLong.fromLong(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToUnsignedLongFromLongEvaluator get(DriverContext context) { - return new ToUnsignedLongFromLongEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToUnsignedLongFromLongEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToUnsignedLongFromStringEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToUnsignedLongFromStringEvaluator.java deleted file mode 100644 index 5b46fe2bfc9bf..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToUnsignedLongFromStringEvaluator.java +++ /dev/null @@ -1,127 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.NumberFormatException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.InvalidArgumentException; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToUnsignedLong}. - * This class is generated. Do not edit it. - */ -public final class ToUnsignedLongFromStringEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToUnsignedLongFromStringEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToUnsignedLongFromString"; - } - - @Override - public Block evalVector(Vector v) { - BytesRefVector vector = (BytesRefVector) v; - int positionCount = v.getPositionCount(); - BytesRef scratchPad = new BytesRef(); - if (vector.isConstant()) { - try { - return driverContext.blockFactory().newConstantLongBlockWith(evalValue(vector, 0, scratchPad), positionCount); - } catch (InvalidArgumentException | NumberFormatException e) { - registerException(e); - return driverContext.blockFactory().newConstantNullBlock(positionCount); - } - } - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - try { - builder.appendLong(evalValue(vector, p, scratchPad)); - } catch (InvalidArgumentException | NumberFormatException e) { - registerException(e); - builder.appendNull(); - } - } - return builder.build(); - } - } - - private static long evalValue(BytesRefVector container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToUnsignedLong.fromKeyword(value); - } - - @Override - public Block evalBlock(Block b) { - BytesRefBlock block = (BytesRefBlock) b; - int positionCount = block.getPositionCount(); - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - BytesRef scratchPad = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - try { - long value = evalValue(block, i, scratchPad); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendLong(value); - valuesAppended = true; - } catch (InvalidArgumentException | NumberFormatException e) { - registerException(e); - } - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static long evalValue(BytesRefBlock container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToUnsignedLong.fromKeyword(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToUnsignedLongFromStringEvaluator get(DriverContext context) { - return new ToUnsignedLongFromStringEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToUnsignedLongFromStringEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToVersionFromStringEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToVersionFromStringEvaluator.java deleted file mode 100644 index 5945129a8ae05..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/convert/ToVersionFromStringEvaluator.java +++ /dev/null @@ -1,110 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.convert; - -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToVersion}. - * This class is generated. Do not edit it. - */ -public final class ToVersionFromStringEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public ToVersionFromStringEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "ToVersionFromString"; - } - - @Override - public Block evalVector(Vector v) { - BytesRefVector vector = (BytesRefVector) v; - int positionCount = v.getPositionCount(); - BytesRef scratchPad = new BytesRef(); - if (vector.isConstant()) { - return driverContext.blockFactory().newConstantBytesRefBlockWith(evalValue(vector, 0, scratchPad), positionCount); - } - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - builder.appendBytesRef(evalValue(vector, p, scratchPad)); - } - return builder.build(); - } - } - - private static BytesRef evalValue(BytesRefVector container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToVersion.fromKeyword(value); - } - - @Override - public Block evalBlock(Block b) { - BytesRefBlock block = (BytesRefBlock) b; - int positionCount = block.getPositionCount(); - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef scratchPad = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - BytesRef value = evalValue(block, i, scratchPad); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendBytesRef(value); - valuesAppended = true; - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static BytesRef evalValue(BytesRefBlock container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return ToVersion.fromKeyword(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public ToVersionFromStringEvaluator get(DriverContext context) { - return new ToVersionFromStringEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "ToVersionFromStringEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/date/DateDiffConstantEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/date/DateDiffConstantEvaluator.java deleted file mode 100644 index 3cb41d0028d54..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/date/DateDiffConstantEvaluator.java +++ /dev/null @@ -1,154 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.date; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.InvalidArgumentException; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link DateDiff}. - * This class is generated. Do not edit it. - */ -public final class DateDiffConstantEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final DateDiff.Part datePartFieldUnit; - - private final EvalOperator.ExpressionEvaluator startTimestamp; - - private final EvalOperator.ExpressionEvaluator endTimestamp; - - private final DriverContext driverContext; - - public DateDiffConstantEvaluator(Source source, DateDiff.Part datePartFieldUnit, - EvalOperator.ExpressionEvaluator startTimestamp, - EvalOperator.ExpressionEvaluator endTimestamp, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.datePartFieldUnit = datePartFieldUnit; - this.startTimestamp = startTimestamp; - this.endTimestamp = endTimestamp; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock startTimestampBlock = (LongBlock) startTimestamp.eval(page)) { - try (LongBlock endTimestampBlock = (LongBlock) endTimestamp.eval(page)) { - LongVector startTimestampVector = startTimestampBlock.asVector(); - if (startTimestampVector == null) { - return eval(page.getPositionCount(), startTimestampBlock, endTimestampBlock); - } - LongVector endTimestampVector = endTimestampBlock.asVector(); - if (endTimestampVector == null) { - return eval(page.getPositionCount(), startTimestampBlock, endTimestampBlock); - } - return eval(page.getPositionCount(), startTimestampVector, endTimestampVector); - } - } - } - - public IntBlock eval(int positionCount, LongBlock startTimestampBlock, - LongBlock endTimestampBlock) { - try(IntBlock.Builder result = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (startTimestampBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (startTimestampBlock.getValueCount(p) != 1) { - if (startTimestampBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (endTimestampBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (endTimestampBlock.getValueCount(p) != 1) { - if (endTimestampBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendInt(DateDiff.process(datePartFieldUnit, startTimestampBlock.getLong(startTimestampBlock.getFirstValueIndex(p)), endTimestampBlock.getLong(endTimestampBlock.getFirstValueIndex(p)))); - } catch (IllegalArgumentException | InvalidArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public IntBlock eval(int positionCount, LongVector startTimestampVector, - LongVector endTimestampVector) { - try(IntBlock.Builder result = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendInt(DateDiff.process(datePartFieldUnit, startTimestampVector.getLong(p), endTimestampVector.getLong(p))); - } catch (IllegalArgumentException | InvalidArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "DateDiffConstantEvaluator[" + "datePartFieldUnit=" + datePartFieldUnit + ", startTimestamp=" + startTimestamp + ", endTimestamp=" + endTimestamp + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(startTimestamp, endTimestamp); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final DateDiff.Part datePartFieldUnit; - - private final EvalOperator.ExpressionEvaluator.Factory startTimestamp; - - private final EvalOperator.ExpressionEvaluator.Factory endTimestamp; - - public Factory(Source source, DateDiff.Part datePartFieldUnit, - EvalOperator.ExpressionEvaluator.Factory startTimestamp, - EvalOperator.ExpressionEvaluator.Factory endTimestamp) { - this.source = source; - this.datePartFieldUnit = datePartFieldUnit; - this.startTimestamp = startTimestamp; - this.endTimestamp = endTimestamp; - } - - @Override - public DateDiffConstantEvaluator get(DriverContext context) { - return new DateDiffConstantEvaluator(source, datePartFieldUnit, startTimestamp.get(context), endTimestamp.get(context), context); - } - - @Override - public String toString() { - return "DateDiffConstantEvaluator[" + "datePartFieldUnit=" + datePartFieldUnit + ", startTimestamp=" + startTimestamp + ", endTimestamp=" + endTimestamp + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/date/DateDiffEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/date/DateDiffEvaluator.java deleted file mode 100644 index 952a819a014a9..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/date/DateDiffEvaluator.java +++ /dev/null @@ -1,176 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.date; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.InvalidArgumentException; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link DateDiff}. - * This class is generated. Do not edit it. - */ -public final class DateDiffEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator unit; - - private final EvalOperator.ExpressionEvaluator startTimestamp; - - private final EvalOperator.ExpressionEvaluator endTimestamp; - - private final DriverContext driverContext; - - public DateDiffEvaluator(Source source, EvalOperator.ExpressionEvaluator unit, - EvalOperator.ExpressionEvaluator startTimestamp, - EvalOperator.ExpressionEvaluator endTimestamp, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.unit = unit; - this.startTimestamp = startTimestamp; - this.endTimestamp = endTimestamp; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock unitBlock = (BytesRefBlock) unit.eval(page)) { - try (LongBlock startTimestampBlock = (LongBlock) startTimestamp.eval(page)) { - try (LongBlock endTimestampBlock = (LongBlock) endTimestamp.eval(page)) { - BytesRefVector unitVector = unitBlock.asVector(); - if (unitVector == null) { - return eval(page.getPositionCount(), unitBlock, startTimestampBlock, endTimestampBlock); - } - LongVector startTimestampVector = startTimestampBlock.asVector(); - if (startTimestampVector == null) { - return eval(page.getPositionCount(), unitBlock, startTimestampBlock, endTimestampBlock); - } - LongVector endTimestampVector = endTimestampBlock.asVector(); - if (endTimestampVector == null) { - return eval(page.getPositionCount(), unitBlock, startTimestampBlock, endTimestampBlock); - } - return eval(page.getPositionCount(), unitVector, startTimestampVector, endTimestampVector); - } - } - } - } - - public IntBlock eval(int positionCount, BytesRefBlock unitBlock, LongBlock startTimestampBlock, - LongBlock endTimestampBlock) { - try(IntBlock.Builder result = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - BytesRef unitScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (unitBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (unitBlock.getValueCount(p) != 1) { - if (unitBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (startTimestampBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (startTimestampBlock.getValueCount(p) != 1) { - if (startTimestampBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (endTimestampBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (endTimestampBlock.getValueCount(p) != 1) { - if (endTimestampBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendInt(DateDiff.process(unitBlock.getBytesRef(unitBlock.getFirstValueIndex(p), unitScratch), startTimestampBlock.getLong(startTimestampBlock.getFirstValueIndex(p)), endTimestampBlock.getLong(endTimestampBlock.getFirstValueIndex(p)))); - } catch (IllegalArgumentException | InvalidArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public IntBlock eval(int positionCount, BytesRefVector unitVector, - LongVector startTimestampVector, LongVector endTimestampVector) { - try(IntBlock.Builder result = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - BytesRef unitScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendInt(DateDiff.process(unitVector.getBytesRef(p, unitScratch), startTimestampVector.getLong(p), endTimestampVector.getLong(p))); - } catch (IllegalArgumentException | InvalidArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "DateDiffEvaluator[" + "unit=" + unit + ", startTimestamp=" + startTimestamp + ", endTimestamp=" + endTimestamp + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(unit, startTimestamp, endTimestamp); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory unit; - - private final EvalOperator.ExpressionEvaluator.Factory startTimestamp; - - private final EvalOperator.ExpressionEvaluator.Factory endTimestamp; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory unit, - EvalOperator.ExpressionEvaluator.Factory startTimestamp, - EvalOperator.ExpressionEvaluator.Factory endTimestamp) { - this.source = source; - this.unit = unit; - this.startTimestamp = startTimestamp; - this.endTimestamp = endTimestamp; - } - - @Override - public DateDiffEvaluator get(DriverContext context) { - return new DateDiffEvaluator(source, unit.get(context), startTimestamp.get(context), endTimestamp.get(context), context); - } - - @Override - public String toString() { - return "DateDiffEvaluator[" + "unit=" + unit + ", startTimestamp=" + startTimestamp + ", endTimestamp=" + endTimestamp + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/date/DateExtractConstantEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/date/DateExtractConstantEvaluator.java deleted file mode 100644 index 8b1804cacfc21..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/date/DateExtractConstantEvaluator.java +++ /dev/null @@ -1,123 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.date; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import java.time.ZoneId; -import java.time.temporal.ChronoField; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link DateExtract}. - * This class is generated. Do not edit it. - */ -public final class DateExtractConstantEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator value; - - private final ChronoField chronoField; - - private final ZoneId zone; - - private final DriverContext driverContext; - - public DateExtractConstantEvaluator(Source source, EvalOperator.ExpressionEvaluator value, - ChronoField chronoField, ZoneId zone, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.value = value; - this.chronoField = chronoField; - this.zone = zone; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock valueBlock = (LongBlock) value.eval(page)) { - LongVector valueVector = valueBlock.asVector(); - if (valueVector == null) { - return eval(page.getPositionCount(), valueBlock); - } - return eval(page.getPositionCount(), valueVector).asBlock(); - } - } - - public LongBlock eval(int positionCount, LongBlock valueBlock) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valueBlock.getValueCount(p) != 1) { - if (valueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendLong(DateExtract.process(valueBlock.getLong(valueBlock.getFirstValueIndex(p)), chronoField, zone)); - } - return result.build(); - } - } - - public LongVector eval(int positionCount, LongVector valueVector) { - try(LongVector.Builder result = driverContext.blockFactory().newLongVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendLong(DateExtract.process(valueVector.getLong(p), chronoField, zone)); - } - return result.build(); - } - } - - @Override - public String toString() { - return "DateExtractConstantEvaluator[" + "value=" + value + ", chronoField=" + chronoField + ", zone=" + zone + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(value); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory value; - - private final ChronoField chronoField; - - private final ZoneId zone; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory value, - ChronoField chronoField, ZoneId zone) { - this.source = source; - this.value = value; - this.chronoField = chronoField; - this.zone = zone; - } - - @Override - public DateExtractConstantEvaluator get(DriverContext context) { - return new DateExtractConstantEvaluator(source, value.get(context), chronoField, zone, context); - } - - @Override - public String toString() { - return "DateExtractConstantEvaluator[" + "value=" + value + ", chronoField=" + chronoField + ", zone=" + zone + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/date/DateExtractEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/date/DateExtractEvaluator.java deleted file mode 100644 index 65af16e2a9f5b..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/date/DateExtractEvaluator.java +++ /dev/null @@ -1,155 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.date; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import java.time.ZoneId; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link DateExtract}. - * This class is generated. Do not edit it. - */ -public final class DateExtractEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator value; - - private final EvalOperator.ExpressionEvaluator chronoField; - - private final ZoneId zone; - - private final DriverContext driverContext; - - public DateExtractEvaluator(Source source, EvalOperator.ExpressionEvaluator value, - EvalOperator.ExpressionEvaluator chronoField, ZoneId zone, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.value = value; - this.chronoField = chronoField; - this.zone = zone; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock valueBlock = (LongBlock) value.eval(page)) { - try (BytesRefBlock chronoFieldBlock = (BytesRefBlock) chronoField.eval(page)) { - LongVector valueVector = valueBlock.asVector(); - if (valueVector == null) { - return eval(page.getPositionCount(), valueBlock, chronoFieldBlock); - } - BytesRefVector chronoFieldVector = chronoFieldBlock.asVector(); - if (chronoFieldVector == null) { - return eval(page.getPositionCount(), valueBlock, chronoFieldBlock); - } - return eval(page.getPositionCount(), valueVector, chronoFieldVector); - } - } - } - - public LongBlock eval(int positionCount, LongBlock valueBlock, BytesRefBlock chronoFieldBlock) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - BytesRef chronoFieldScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (valueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valueBlock.getValueCount(p) != 1) { - if (valueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (chronoFieldBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (chronoFieldBlock.getValueCount(p) != 1) { - if (chronoFieldBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendLong(DateExtract.process(valueBlock.getLong(valueBlock.getFirstValueIndex(p)), chronoFieldBlock.getBytesRef(chronoFieldBlock.getFirstValueIndex(p), chronoFieldScratch), zone)); - } catch (IllegalArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public LongBlock eval(int positionCount, LongVector valueVector, - BytesRefVector chronoFieldVector) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - BytesRef chronoFieldScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendLong(DateExtract.process(valueVector.getLong(p), chronoFieldVector.getBytesRef(p, chronoFieldScratch), zone)); - } catch (IllegalArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "DateExtractEvaluator[" + "value=" + value + ", chronoField=" + chronoField + ", zone=" + zone + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(value, chronoField); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory value; - - private final EvalOperator.ExpressionEvaluator.Factory chronoField; - - private final ZoneId zone; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory value, - EvalOperator.ExpressionEvaluator.Factory chronoField, ZoneId zone) { - this.source = source; - this.value = value; - this.chronoField = chronoField; - this.zone = zone; - } - - @Override - public DateExtractEvaluator get(DriverContext context) { - return new DateExtractEvaluator(source, value.get(context), chronoField.get(context), zone, context); - } - - @Override - public String toString() { - return "DateExtractEvaluator[" + "value=" + value + ", chronoField=" + chronoField + ", zone=" + zone + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/date/DateFormatConstantEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/date/DateFormatConstantEvaluator.java deleted file mode 100644 index 38cc3e2809f0a..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/date/DateFormatConstantEvaluator.java +++ /dev/null @@ -1,118 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.date; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.common.time.DateFormatter; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link DateFormat}. - * This class is generated. Do not edit it. - */ -public final class DateFormatConstantEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final DateFormatter formatter; - - private final DriverContext driverContext; - - public DateFormatConstantEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - DateFormatter formatter, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.formatter = formatter; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock valBlock = (LongBlock) val.eval(page)) { - LongVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector).asBlock(); - } - } - - public BytesRefBlock eval(int positionCount, LongBlock valBlock) { - try(BytesRefBlock.Builder result = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBytesRef(DateFormat.process(valBlock.getLong(valBlock.getFirstValueIndex(p)), formatter)); - } - return result.build(); - } - } - - public BytesRefVector eval(int positionCount, LongVector valVector) { - try(BytesRefVector.Builder result = driverContext.blockFactory().newBytesRefVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendBytesRef(DateFormat.process(valVector.getLong(p), formatter)); - } - return result.build(); - } - } - - @Override - public String toString() { - return "DateFormatConstantEvaluator[" + "val=" + val + ", formatter=" + formatter + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - private final DateFormatter formatter; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val, - DateFormatter formatter) { - this.source = source; - this.val = val; - this.formatter = formatter; - } - - @Override - public DateFormatConstantEvaluator get(DriverContext context) { - return new DateFormatConstantEvaluator(source, val.get(context), formatter, context); - } - - @Override - public String toString() { - return "DateFormatConstantEvaluator[" + "val=" + val + ", formatter=" + formatter + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/date/DateFormatEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/date/DateFormatEvaluator.java deleted file mode 100644 index d517c16cb4076..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/date/DateFormatEvaluator.java +++ /dev/null @@ -1,145 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.date; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import java.util.Locale; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link DateFormat}. - * This class is generated. Do not edit it. - */ -public final class DateFormatEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final EvalOperator.ExpressionEvaluator formatter; - - private final Locale locale; - - private final DriverContext driverContext; - - public DateFormatEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - EvalOperator.ExpressionEvaluator formatter, Locale locale, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.formatter = formatter; - this.locale = locale; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock valBlock = (LongBlock) val.eval(page)) { - try (BytesRefBlock formatterBlock = (BytesRefBlock) formatter.eval(page)) { - LongVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock, formatterBlock); - } - BytesRefVector formatterVector = formatterBlock.asVector(); - if (formatterVector == null) { - return eval(page.getPositionCount(), valBlock, formatterBlock); - } - return eval(page.getPositionCount(), valVector, formatterVector).asBlock(); - } - } - } - - public BytesRefBlock eval(int positionCount, LongBlock valBlock, BytesRefBlock formatterBlock) { - try(BytesRefBlock.Builder result = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef formatterScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (formatterBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (formatterBlock.getValueCount(p) != 1) { - if (formatterBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBytesRef(DateFormat.process(valBlock.getLong(valBlock.getFirstValueIndex(p)), formatterBlock.getBytesRef(formatterBlock.getFirstValueIndex(p), formatterScratch), locale)); - } - return result.build(); - } - } - - public BytesRefVector eval(int positionCount, LongVector valVector, - BytesRefVector formatterVector) { - try(BytesRefVector.Builder result = driverContext.blockFactory().newBytesRefVectorBuilder(positionCount)) { - BytesRef formatterScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - result.appendBytesRef(DateFormat.process(valVector.getLong(p), formatterVector.getBytesRef(p, formatterScratch), locale)); - } - return result.build(); - } - } - - @Override - public String toString() { - return "DateFormatEvaluator[" + "val=" + val + ", formatter=" + formatter + ", locale=" + locale + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val, formatter); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - private final EvalOperator.ExpressionEvaluator.Factory formatter; - - private final Locale locale; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val, - EvalOperator.ExpressionEvaluator.Factory formatter, Locale locale) { - this.source = source; - this.val = val; - this.formatter = formatter; - this.locale = locale; - } - - @Override - public DateFormatEvaluator get(DriverContext context) { - return new DateFormatEvaluator(source, val.get(context), formatter.get(context), locale, context); - } - - @Override - public String toString() { - return "DateFormatEvaluator[" + "val=" + val + ", formatter=" + formatter + ", locale=" + locale + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/date/DateParseConstantEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/date/DateParseConstantEvaluator.java deleted file mode 100644 index 3a6b44d82a011..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/date/DateParseConstantEvaluator.java +++ /dev/null @@ -1,130 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.date; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.common.time.DateFormatter; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link DateParse}. - * This class is generated. Do not edit it. - */ -public final class DateParseConstantEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final DateFormatter formatter; - - private final DriverContext driverContext; - - public DateParseConstantEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - DateFormatter formatter, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.formatter = formatter; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock valBlock = (BytesRefBlock) val.eval(page)) { - BytesRefVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector); - } - } - - public LongBlock eval(int positionCount, BytesRefBlock valBlock) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - BytesRef valScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendLong(DateParse.process(valBlock.getBytesRef(valBlock.getFirstValueIndex(p), valScratch), formatter)); - } catch (IllegalArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public LongBlock eval(int positionCount, BytesRefVector valVector) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - BytesRef valScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendLong(DateParse.process(valVector.getBytesRef(p, valScratch), formatter)); - } catch (IllegalArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "DateParseConstantEvaluator[" + "val=" + val + ", formatter=" + formatter + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - private final DateFormatter formatter; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val, - DateFormatter formatter) { - this.source = source; - this.val = val; - this.formatter = formatter; - } - - @Override - public DateParseConstantEvaluator get(DriverContext context) { - return new DateParseConstantEvaluator(source, val.get(context), formatter, context); - } - - @Override - public String toString() { - return "DateParseConstantEvaluator[" + "val=" + val + ", formatter=" + formatter + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/date/DateParseEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/date/DateParseEvaluator.java deleted file mode 100644 index 2da9310b0f53a..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/date/DateParseEvaluator.java +++ /dev/null @@ -1,156 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.date; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import java.time.ZoneId; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link DateParse}. - * This class is generated. Do not edit it. - */ -public final class DateParseEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final EvalOperator.ExpressionEvaluator formatter; - - private final ZoneId zoneId; - - private final DriverContext driverContext; - - public DateParseEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - EvalOperator.ExpressionEvaluator formatter, ZoneId zoneId, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.formatter = formatter; - this.zoneId = zoneId; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock valBlock = (BytesRefBlock) val.eval(page)) { - try (BytesRefBlock formatterBlock = (BytesRefBlock) formatter.eval(page)) { - BytesRefVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock, formatterBlock); - } - BytesRefVector formatterVector = formatterBlock.asVector(); - if (formatterVector == null) { - return eval(page.getPositionCount(), valBlock, formatterBlock); - } - return eval(page.getPositionCount(), valVector, formatterVector); - } - } - } - - public LongBlock eval(int positionCount, BytesRefBlock valBlock, BytesRefBlock formatterBlock) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - BytesRef valScratch = new BytesRef(); - BytesRef formatterScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (formatterBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (formatterBlock.getValueCount(p) != 1) { - if (formatterBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendLong(DateParse.process(valBlock.getBytesRef(valBlock.getFirstValueIndex(p), valScratch), formatterBlock.getBytesRef(formatterBlock.getFirstValueIndex(p), formatterScratch), zoneId)); - } catch (IllegalArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public LongBlock eval(int positionCount, BytesRefVector valVector, - BytesRefVector formatterVector) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - BytesRef valScratch = new BytesRef(); - BytesRef formatterScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendLong(DateParse.process(valVector.getBytesRef(p, valScratch), formatterVector.getBytesRef(p, formatterScratch), zoneId)); - } catch (IllegalArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "DateParseEvaluator[" + "val=" + val + ", formatter=" + formatter + ", zoneId=" + zoneId + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val, formatter); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - private final EvalOperator.ExpressionEvaluator.Factory formatter; - - private final ZoneId zoneId; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val, - EvalOperator.ExpressionEvaluator.Factory formatter, ZoneId zoneId) { - this.source = source; - this.val = val; - this.formatter = formatter; - this.zoneId = zoneId; - } - - @Override - public DateParseEvaluator get(DriverContext context) { - return new DateParseEvaluator(source, val.get(context), formatter.get(context), zoneId, context); - } - - @Override - public String toString() { - return "DateParseEvaluator[" + "val=" + val + ", formatter=" + formatter + ", zoneId=" + zoneId + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/date/DateTruncEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/date/DateTruncEvaluator.java deleted file mode 100644 index 27a15ca19bec9..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/date/DateTruncEvaluator.java +++ /dev/null @@ -1,116 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.date; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.common.Rounding; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link DateTrunc}. - * This class is generated. Do not edit it. - */ -public final class DateTruncEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator fieldVal; - - private final Rounding.Prepared rounding; - - private final DriverContext driverContext; - - public DateTruncEvaluator(Source source, EvalOperator.ExpressionEvaluator fieldVal, - Rounding.Prepared rounding, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.fieldVal = fieldVal; - this.rounding = rounding; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock fieldValBlock = (LongBlock) fieldVal.eval(page)) { - LongVector fieldValVector = fieldValBlock.asVector(); - if (fieldValVector == null) { - return eval(page.getPositionCount(), fieldValBlock); - } - return eval(page.getPositionCount(), fieldValVector).asBlock(); - } - } - - public LongBlock eval(int positionCount, LongBlock fieldValBlock) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (fieldValBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (fieldValBlock.getValueCount(p) != 1) { - if (fieldValBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendLong(DateTrunc.process(fieldValBlock.getLong(fieldValBlock.getFirstValueIndex(p)), rounding)); - } - return result.build(); - } - } - - public LongVector eval(int positionCount, LongVector fieldValVector) { - try(LongVector.Builder result = driverContext.blockFactory().newLongVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendLong(DateTrunc.process(fieldValVector.getLong(p), rounding)); - } - return result.build(); - } - } - - @Override - public String toString() { - return "DateTruncEvaluator[" + "fieldVal=" + fieldVal + ", rounding=" + rounding + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(fieldVal); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory fieldVal; - - private final Rounding.Prepared rounding; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory fieldVal, - Rounding.Prepared rounding) { - this.source = source; - this.fieldVal = fieldVal; - this.rounding = rounding; - } - - @Override - public DateTruncEvaluator get(DriverContext context) { - return new DateTruncEvaluator(source, fieldVal.get(context), rounding, context); - } - - @Override - public String toString() { - return "DateTruncEvaluator[" + "fieldVal=" + fieldVal + ", rounding=" + rounding + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/date/NowEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/date/NowEvaluator.java deleted file mode 100644 index 45465468f7c91..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/date/NowEvaluator.java +++ /dev/null @@ -1,77 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.date; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Now}. - * This class is generated. Do not edit it. - */ -public final class NowEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final long now; - - private final DriverContext driverContext; - - public NowEvaluator(Source source, long now, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.now = now; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - return eval(page.getPositionCount()).asBlock(); - } - - public LongVector eval(int positionCount) { - try(LongVector.Builder result = driverContext.blockFactory().newLongVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendLong(Now.process(now)); - } - return result.build(); - } - } - - @Override - public String toString() { - return "NowEvaluator[" + "now=" + now + "]"; - } - - @Override - public void close() { - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final long now; - - public Factory(Source source, long now) { - this.source = source; - this.now = now; - } - - @Override - public NowEvaluator get(DriverContext context) { - return new NowEvaluator(source, now, context); - } - - @Override - public String toString() { - return "NowEvaluator[" + "now=" + now + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/ip/CIDRMatchEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/ip/CIDRMatchEvaluator.java deleted file mode 100644 index 4ac2fa7d2738e..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/ip/CIDRMatchEvaluator.java +++ /dev/null @@ -1,169 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.ip; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import java.util.Arrays; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasable; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link CIDRMatch}. - * This class is generated. Do not edit it. - */ -public final class CIDRMatchEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator ip; - - private final EvalOperator.ExpressionEvaluator[] cidrs; - - private final DriverContext driverContext; - - public CIDRMatchEvaluator(Source source, EvalOperator.ExpressionEvaluator ip, - EvalOperator.ExpressionEvaluator[] cidrs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.ip = ip; - this.cidrs = cidrs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock ipBlock = (BytesRefBlock) ip.eval(page)) { - BytesRefBlock[] cidrsBlocks = new BytesRefBlock[cidrs.length]; - try (Releasable cidrsRelease = Releasables.wrap(cidrsBlocks)) { - for (int i = 0; i < cidrsBlocks.length; i++) { - cidrsBlocks[i] = (BytesRefBlock)cidrs[i].eval(page); - } - BytesRefVector ipVector = ipBlock.asVector(); - if (ipVector == null) { - return eval(page.getPositionCount(), ipBlock, cidrsBlocks); - } - BytesRefVector[] cidrsVectors = new BytesRefVector[cidrs.length]; - for (int i = 0; i < cidrsBlocks.length; i++) { - cidrsVectors[i] = cidrsBlocks[i].asVector(); - if (cidrsVectors[i] == null) { - return eval(page.getPositionCount(), ipBlock, cidrsBlocks); - } - } - return eval(page.getPositionCount(), ipVector, cidrsVectors).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, BytesRefBlock ipBlock, BytesRefBlock[] cidrsBlocks) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef ipScratch = new BytesRef(); - BytesRef[] cidrsValues = new BytesRef[cidrs.length]; - BytesRef[] cidrsScratch = new BytesRef[cidrs.length]; - for (int i = 0; i < cidrs.length; i++) { - cidrsScratch[i] = new BytesRef(); - } - position: for (int p = 0; p < positionCount; p++) { - if (ipBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (ipBlock.getValueCount(p) != 1) { - if (ipBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - for (int i = 0; i < cidrsBlocks.length; i++) { - if (cidrsBlocks[i].isNull(p)) { - result.appendNull(); - continue position; - } - if (cidrsBlocks[i].getValueCount(p) != 1) { - if (cidrsBlocks[i].getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - } - // unpack cidrsBlocks into cidrsValues - for (int i = 0; i < cidrsBlocks.length; i++) { - int o = cidrsBlocks[i].getFirstValueIndex(p); - cidrsValues[i] = cidrsBlocks[i].getBytesRef(o, cidrsScratch[i]); - } - result.appendBoolean(CIDRMatch.process(ipBlock.getBytesRef(ipBlock.getFirstValueIndex(p), ipScratch), cidrsValues)); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, BytesRefVector ipVector, - BytesRefVector[] cidrsVectors) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - BytesRef ipScratch = new BytesRef(); - BytesRef[] cidrsValues = new BytesRef[cidrs.length]; - BytesRef[] cidrsScratch = new BytesRef[cidrs.length]; - for (int i = 0; i < cidrs.length; i++) { - cidrsScratch[i] = new BytesRef(); - } - position: for (int p = 0; p < positionCount; p++) { - // unpack cidrsVectors into cidrsValues - for (int i = 0; i < cidrsVectors.length; i++) { - cidrsValues[i] = cidrsVectors[i].getBytesRef(p, cidrsScratch[i]); - } - result.appendBoolean(CIDRMatch.process(ipVector.getBytesRef(p, ipScratch), cidrsValues)); - } - return result.build(); - } - } - - @Override - public String toString() { - return "CIDRMatchEvaluator[" + "ip=" + ip + ", cidrs=" + Arrays.toString(cidrs) + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(ip, () -> Releasables.close(cidrs)); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory ip; - - private final EvalOperator.ExpressionEvaluator.Factory[] cidrs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory ip, - EvalOperator.ExpressionEvaluator.Factory[] cidrs) { - this.source = source; - this.ip = ip; - this.cidrs = cidrs; - } - - @Override - public CIDRMatchEvaluator get(DriverContext context) { - EvalOperator.ExpressionEvaluator[] cidrs = Arrays.stream(this.cidrs).map(a -> a.get(context)).toArray(EvalOperator.ExpressionEvaluator[]::new); - return new CIDRMatchEvaluator(source, ip.get(context), cidrs, context); - } - - @Override - public String toString() { - return "CIDRMatchEvaluator[" + "ip=" + ip + ", cidrs=" + Arrays.toString(cidrs) + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/AbsDoubleEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/AbsDoubleEvaluator.java deleted file mode 100644 index d7c793b99e57b..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/AbsDoubleEvaluator.java +++ /dev/null @@ -1,108 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Abs}. - * This class is generated. Do not edit it. - */ -public final class AbsDoubleEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator fieldVal; - - private final DriverContext driverContext; - - public AbsDoubleEvaluator(Source source, EvalOperator.ExpressionEvaluator fieldVal, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.fieldVal = fieldVal; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock fieldValBlock = (DoubleBlock) fieldVal.eval(page)) { - DoubleVector fieldValVector = fieldValBlock.asVector(); - if (fieldValVector == null) { - return eval(page.getPositionCount(), fieldValBlock); - } - return eval(page.getPositionCount(), fieldValVector).asBlock(); - } - } - - public DoubleBlock eval(int positionCount, DoubleBlock fieldValBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (fieldValBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (fieldValBlock.getValueCount(p) != 1) { - if (fieldValBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendDouble(Abs.process(fieldValBlock.getDouble(fieldValBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public DoubleVector eval(int positionCount, DoubleVector fieldValVector) { - try(DoubleVector.Builder result = driverContext.blockFactory().newDoubleVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendDouble(Abs.process(fieldValVector.getDouble(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "AbsDoubleEvaluator[" + "fieldVal=" + fieldVal + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(fieldVal); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory fieldVal; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory fieldVal) { - this.source = source; - this.fieldVal = fieldVal; - } - - @Override - public AbsDoubleEvaluator get(DriverContext context) { - return new AbsDoubleEvaluator(source, fieldVal.get(context), context); - } - - @Override - public String toString() { - return "AbsDoubleEvaluator[" + "fieldVal=" + fieldVal + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/AbsIntEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/AbsIntEvaluator.java deleted file mode 100644 index 9964a95fafe0c..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/AbsIntEvaluator.java +++ /dev/null @@ -1,108 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Abs}. - * This class is generated. Do not edit it. - */ -public final class AbsIntEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator fieldVal; - - private final DriverContext driverContext; - - public AbsIntEvaluator(Source source, EvalOperator.ExpressionEvaluator fieldVal, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.fieldVal = fieldVal; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (IntBlock fieldValBlock = (IntBlock) fieldVal.eval(page)) { - IntVector fieldValVector = fieldValBlock.asVector(); - if (fieldValVector == null) { - return eval(page.getPositionCount(), fieldValBlock); - } - return eval(page.getPositionCount(), fieldValVector).asBlock(); - } - } - - public IntBlock eval(int positionCount, IntBlock fieldValBlock) { - try(IntBlock.Builder result = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (fieldValBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (fieldValBlock.getValueCount(p) != 1) { - if (fieldValBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendInt(Abs.process(fieldValBlock.getInt(fieldValBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public IntVector eval(int positionCount, IntVector fieldValVector) { - try(IntVector.Builder result = driverContext.blockFactory().newIntVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendInt(Abs.process(fieldValVector.getInt(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "AbsIntEvaluator[" + "fieldVal=" + fieldVal + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(fieldVal); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory fieldVal; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory fieldVal) { - this.source = source; - this.fieldVal = fieldVal; - } - - @Override - public AbsIntEvaluator get(DriverContext context) { - return new AbsIntEvaluator(source, fieldVal.get(context), context); - } - - @Override - public String toString() { - return "AbsIntEvaluator[" + "fieldVal=" + fieldVal + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/AbsLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/AbsLongEvaluator.java deleted file mode 100644 index 9457112aa9d81..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/AbsLongEvaluator.java +++ /dev/null @@ -1,108 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Abs}. - * This class is generated. Do not edit it. - */ -public final class AbsLongEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator fieldVal; - - private final DriverContext driverContext; - - public AbsLongEvaluator(Source source, EvalOperator.ExpressionEvaluator fieldVal, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.fieldVal = fieldVal; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock fieldValBlock = (LongBlock) fieldVal.eval(page)) { - LongVector fieldValVector = fieldValBlock.asVector(); - if (fieldValVector == null) { - return eval(page.getPositionCount(), fieldValBlock); - } - return eval(page.getPositionCount(), fieldValVector).asBlock(); - } - } - - public LongBlock eval(int positionCount, LongBlock fieldValBlock) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (fieldValBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (fieldValBlock.getValueCount(p) != 1) { - if (fieldValBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendLong(Abs.process(fieldValBlock.getLong(fieldValBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public LongVector eval(int positionCount, LongVector fieldValVector) { - try(LongVector.Builder result = driverContext.blockFactory().newLongVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendLong(Abs.process(fieldValVector.getLong(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "AbsLongEvaluator[" + "fieldVal=" + fieldVal + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(fieldVal); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory fieldVal; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory fieldVal) { - this.source = source; - this.fieldVal = fieldVal; - } - - @Override - public AbsLongEvaluator get(DriverContext context) { - return new AbsLongEvaluator(source, fieldVal.get(context), context); - } - - @Override - public String toString() { - return "AbsLongEvaluator[" + "fieldVal=" + fieldVal + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/AcosEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/AcosEvaluator.java deleted file mode 100644 index 1c86fe46e9b93..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/AcosEvaluator.java +++ /dev/null @@ -1,119 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Acos}. - * This class is generated. Do not edit it. - */ -public final class AcosEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final DriverContext driverContext; - - public AcosEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock valBlock = (DoubleBlock) val.eval(page)) { - DoubleVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector); - } - } - - public DoubleBlock eval(int positionCount, DoubleBlock valBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendDouble(Acos.process(valBlock.getDouble(valBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public DoubleBlock eval(int positionCount, DoubleVector valVector) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendDouble(Acos.process(valVector.getDouble(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "AcosEvaluator[" + "val=" + val + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val) { - this.source = source; - this.val = val; - } - - @Override - public AcosEvaluator get(DriverContext context) { - return new AcosEvaluator(source, val.get(context), context); - } - - @Override - public String toString() { - return "AcosEvaluator[" + "val=" + val + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/AsinEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/AsinEvaluator.java deleted file mode 100644 index fc73f4c475676..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/AsinEvaluator.java +++ /dev/null @@ -1,119 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Asin}. - * This class is generated. Do not edit it. - */ -public final class AsinEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final DriverContext driverContext; - - public AsinEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock valBlock = (DoubleBlock) val.eval(page)) { - DoubleVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector); - } - } - - public DoubleBlock eval(int positionCount, DoubleBlock valBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendDouble(Asin.process(valBlock.getDouble(valBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public DoubleBlock eval(int positionCount, DoubleVector valVector) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendDouble(Asin.process(valVector.getDouble(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "AsinEvaluator[" + "val=" + val + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val) { - this.source = source; - this.val = val; - } - - @Override - public AsinEvaluator get(DriverContext context) { - return new AsinEvaluator(source, val.get(context), context); - } - - @Override - public String toString() { - return "AsinEvaluator[" + "val=" + val + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/Atan2Evaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/Atan2Evaluator.java deleted file mode 100644 index b6d0a628c329c..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/Atan2Evaluator.java +++ /dev/null @@ -1,132 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Atan2}. - * This class is generated. Do not edit it. - */ -public final class Atan2Evaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator y; - - private final EvalOperator.ExpressionEvaluator x; - - private final DriverContext driverContext; - - public Atan2Evaluator(Source source, EvalOperator.ExpressionEvaluator y, - EvalOperator.ExpressionEvaluator x, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.y = y; - this.x = x; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock yBlock = (DoubleBlock) y.eval(page)) { - try (DoubleBlock xBlock = (DoubleBlock) x.eval(page)) { - DoubleVector yVector = yBlock.asVector(); - if (yVector == null) { - return eval(page.getPositionCount(), yBlock, xBlock); - } - DoubleVector xVector = xBlock.asVector(); - if (xVector == null) { - return eval(page.getPositionCount(), yBlock, xBlock); - } - return eval(page.getPositionCount(), yVector, xVector).asBlock(); - } - } - } - - public DoubleBlock eval(int positionCount, DoubleBlock yBlock, DoubleBlock xBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (yBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (yBlock.getValueCount(p) != 1) { - if (yBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (xBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (xBlock.getValueCount(p) != 1) { - if (xBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendDouble(Atan2.process(yBlock.getDouble(yBlock.getFirstValueIndex(p)), xBlock.getDouble(xBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public DoubleVector eval(int positionCount, DoubleVector yVector, DoubleVector xVector) { - try(DoubleVector.Builder result = driverContext.blockFactory().newDoubleVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendDouble(Atan2.process(yVector.getDouble(p), xVector.getDouble(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "Atan2Evaluator[" + "y=" + y + ", x=" + x + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(y, x); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory y; - - private final EvalOperator.ExpressionEvaluator.Factory x; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory y, - EvalOperator.ExpressionEvaluator.Factory x) { - this.source = source; - this.y = y; - this.x = x; - } - - @Override - public Atan2Evaluator get(DriverContext context) { - return new Atan2Evaluator(source, y.get(context), x.get(context), context); - } - - @Override - public String toString() { - return "Atan2Evaluator[" + "y=" + y + ", x=" + x + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/AtanEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/AtanEvaluator.java deleted file mode 100644 index b40a6cde6550e..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/AtanEvaluator.java +++ /dev/null @@ -1,108 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Atan}. - * This class is generated. Do not edit it. - */ -public final class AtanEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final DriverContext driverContext; - - public AtanEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock valBlock = (DoubleBlock) val.eval(page)) { - DoubleVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector).asBlock(); - } - } - - public DoubleBlock eval(int positionCount, DoubleBlock valBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendDouble(Atan.process(valBlock.getDouble(valBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public DoubleVector eval(int positionCount, DoubleVector valVector) { - try(DoubleVector.Builder result = driverContext.blockFactory().newDoubleVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendDouble(Atan.process(valVector.getDouble(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "AtanEvaluator[" + "val=" + val + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val) { - this.source = source; - this.val = val; - } - - @Override - public AtanEvaluator get(DriverContext context) { - return new AtanEvaluator(source, val.get(context), context); - } - - @Override - public String toString() { - return "AtanEvaluator[" + "val=" + val + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/CastIntToDoubleEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/CastIntToDoubleEvaluator.java deleted file mode 100644 index a13d11199c0fb..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/CastIntToDoubleEvaluator.java +++ /dev/null @@ -1,110 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Cast}. - * This class is generated. Do not edit it. - */ -public final class CastIntToDoubleEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator v; - - private final DriverContext driverContext; - - public CastIntToDoubleEvaluator(Source source, EvalOperator.ExpressionEvaluator v, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.v = v; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (IntBlock vBlock = (IntBlock) v.eval(page)) { - IntVector vVector = vBlock.asVector(); - if (vVector == null) { - return eval(page.getPositionCount(), vBlock); - } - return eval(page.getPositionCount(), vVector).asBlock(); - } - } - - public DoubleBlock eval(int positionCount, IntBlock vBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (vBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (vBlock.getValueCount(p) != 1) { - if (vBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendDouble(Cast.castIntToDouble(vBlock.getInt(vBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public DoubleVector eval(int positionCount, IntVector vVector) { - try(DoubleVector.Builder result = driverContext.blockFactory().newDoubleVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendDouble(Cast.castIntToDouble(vVector.getInt(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "CastIntToDoubleEvaluator[" + "v=" + v + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(v); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory v; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory v) { - this.source = source; - this.v = v; - } - - @Override - public CastIntToDoubleEvaluator get(DriverContext context) { - return new CastIntToDoubleEvaluator(source, v.get(context), context); - } - - @Override - public String toString() { - return "CastIntToDoubleEvaluator[" + "v=" + v + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/CastIntToLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/CastIntToLongEvaluator.java deleted file mode 100644 index cf91f080537e7..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/CastIntToLongEvaluator.java +++ /dev/null @@ -1,110 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Cast}. - * This class is generated. Do not edit it. - */ -public final class CastIntToLongEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator v; - - private final DriverContext driverContext; - - public CastIntToLongEvaluator(Source source, EvalOperator.ExpressionEvaluator v, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.v = v; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (IntBlock vBlock = (IntBlock) v.eval(page)) { - IntVector vVector = vBlock.asVector(); - if (vVector == null) { - return eval(page.getPositionCount(), vBlock); - } - return eval(page.getPositionCount(), vVector).asBlock(); - } - } - - public LongBlock eval(int positionCount, IntBlock vBlock) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (vBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (vBlock.getValueCount(p) != 1) { - if (vBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendLong(Cast.castIntToLong(vBlock.getInt(vBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public LongVector eval(int positionCount, IntVector vVector) { - try(LongVector.Builder result = driverContext.blockFactory().newLongVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendLong(Cast.castIntToLong(vVector.getInt(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "CastIntToLongEvaluator[" + "v=" + v + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(v); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory v; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory v) { - this.source = source; - this.v = v; - } - - @Override - public CastIntToLongEvaluator get(DriverContext context) { - return new CastIntToLongEvaluator(source, v.get(context), context); - } - - @Override - public String toString() { - return "CastIntToLongEvaluator[" + "v=" + v + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/CastIntToUnsignedLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/CastIntToUnsignedLongEvaluator.java deleted file mode 100644 index 15b18a91ee241..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/CastIntToUnsignedLongEvaluator.java +++ /dev/null @@ -1,110 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Cast}. - * This class is generated. Do not edit it. - */ -public final class CastIntToUnsignedLongEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator v; - - private final DriverContext driverContext; - - public CastIntToUnsignedLongEvaluator(Source source, EvalOperator.ExpressionEvaluator v, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.v = v; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (IntBlock vBlock = (IntBlock) v.eval(page)) { - IntVector vVector = vBlock.asVector(); - if (vVector == null) { - return eval(page.getPositionCount(), vBlock); - } - return eval(page.getPositionCount(), vVector).asBlock(); - } - } - - public LongBlock eval(int positionCount, IntBlock vBlock) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (vBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (vBlock.getValueCount(p) != 1) { - if (vBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendLong(Cast.castIntToUnsignedLong(vBlock.getInt(vBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public LongVector eval(int positionCount, IntVector vVector) { - try(LongVector.Builder result = driverContext.blockFactory().newLongVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendLong(Cast.castIntToUnsignedLong(vVector.getInt(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "CastIntToUnsignedLongEvaluator[" + "v=" + v + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(v); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory v; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory v) { - this.source = source; - this.v = v; - } - - @Override - public CastIntToUnsignedLongEvaluator get(DriverContext context) { - return new CastIntToUnsignedLongEvaluator(source, v.get(context), context); - } - - @Override - public String toString() { - return "CastIntToUnsignedLongEvaluator[" + "v=" + v + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/CastLongToDoubleEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/CastLongToDoubleEvaluator.java deleted file mode 100644 index 1bb63cb66eec5..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/CastLongToDoubleEvaluator.java +++ /dev/null @@ -1,110 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Cast}. - * This class is generated. Do not edit it. - */ -public final class CastLongToDoubleEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator v; - - private final DriverContext driverContext; - - public CastLongToDoubleEvaluator(Source source, EvalOperator.ExpressionEvaluator v, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.v = v; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock vBlock = (LongBlock) v.eval(page)) { - LongVector vVector = vBlock.asVector(); - if (vVector == null) { - return eval(page.getPositionCount(), vBlock); - } - return eval(page.getPositionCount(), vVector).asBlock(); - } - } - - public DoubleBlock eval(int positionCount, LongBlock vBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (vBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (vBlock.getValueCount(p) != 1) { - if (vBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendDouble(Cast.castLongToDouble(vBlock.getLong(vBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public DoubleVector eval(int positionCount, LongVector vVector) { - try(DoubleVector.Builder result = driverContext.blockFactory().newDoubleVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendDouble(Cast.castLongToDouble(vVector.getLong(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "CastLongToDoubleEvaluator[" + "v=" + v + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(v); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory v; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory v) { - this.source = source; - this.v = v; - } - - @Override - public CastLongToDoubleEvaluator get(DriverContext context) { - return new CastLongToDoubleEvaluator(source, v.get(context), context); - } - - @Override - public String toString() { - return "CastLongToDoubleEvaluator[" + "v=" + v + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/CastLongToUnsignedLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/CastLongToUnsignedLongEvaluator.java deleted file mode 100644 index 3ed067671183d..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/CastLongToUnsignedLongEvaluator.java +++ /dev/null @@ -1,108 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Cast}. - * This class is generated. Do not edit it. - */ -public final class CastLongToUnsignedLongEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator v; - - private final DriverContext driverContext; - - public CastLongToUnsignedLongEvaluator(Source source, EvalOperator.ExpressionEvaluator v, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.v = v; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock vBlock = (LongBlock) v.eval(page)) { - LongVector vVector = vBlock.asVector(); - if (vVector == null) { - return eval(page.getPositionCount(), vBlock); - } - return eval(page.getPositionCount(), vVector).asBlock(); - } - } - - public LongBlock eval(int positionCount, LongBlock vBlock) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (vBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (vBlock.getValueCount(p) != 1) { - if (vBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendLong(Cast.castLongToUnsignedLong(vBlock.getLong(vBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public LongVector eval(int positionCount, LongVector vVector) { - try(LongVector.Builder result = driverContext.blockFactory().newLongVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendLong(Cast.castLongToUnsignedLong(vVector.getLong(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "CastLongToUnsignedLongEvaluator[" + "v=" + v + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(v); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory v; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory v) { - this.source = source; - this.v = v; - } - - @Override - public CastLongToUnsignedLongEvaluator get(DriverContext context) { - return new CastLongToUnsignedLongEvaluator(source, v.get(context), context); - } - - @Override - public String toString() { - return "CastLongToUnsignedLongEvaluator[" + "v=" + v + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/CastUnsignedLongToDoubleEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/CastUnsignedLongToDoubleEvaluator.java deleted file mode 100644 index 5135aab0dcc50..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/CastUnsignedLongToDoubleEvaluator.java +++ /dev/null @@ -1,110 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Cast}. - * This class is generated. Do not edit it. - */ -public final class CastUnsignedLongToDoubleEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator v; - - private final DriverContext driverContext; - - public CastUnsignedLongToDoubleEvaluator(Source source, EvalOperator.ExpressionEvaluator v, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.v = v; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock vBlock = (LongBlock) v.eval(page)) { - LongVector vVector = vBlock.asVector(); - if (vVector == null) { - return eval(page.getPositionCount(), vBlock); - } - return eval(page.getPositionCount(), vVector).asBlock(); - } - } - - public DoubleBlock eval(int positionCount, LongBlock vBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (vBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (vBlock.getValueCount(p) != 1) { - if (vBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendDouble(Cast.castUnsignedLongToDouble(vBlock.getLong(vBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public DoubleVector eval(int positionCount, LongVector vVector) { - try(DoubleVector.Builder result = driverContext.blockFactory().newDoubleVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendDouble(Cast.castUnsignedLongToDouble(vVector.getLong(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "CastUnsignedLongToDoubleEvaluator[" + "v=" + v + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(v); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory v; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory v) { - this.source = source; - this.v = v; - } - - @Override - public CastUnsignedLongToDoubleEvaluator get(DriverContext context) { - return new CastUnsignedLongToDoubleEvaluator(source, v.get(context), context); - } - - @Override - public String toString() { - return "CastUnsignedLongToDoubleEvaluator[" + "v=" + v + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/CeilDoubleEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/CeilDoubleEvaluator.java deleted file mode 100644 index 500f108afbe39..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/CeilDoubleEvaluator.java +++ /dev/null @@ -1,108 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Ceil}. - * This class is generated. Do not edit it. - */ -public final class CeilDoubleEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final DriverContext driverContext; - - public CeilDoubleEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock valBlock = (DoubleBlock) val.eval(page)) { - DoubleVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector).asBlock(); - } - } - - public DoubleBlock eval(int positionCount, DoubleBlock valBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendDouble(Ceil.process(valBlock.getDouble(valBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public DoubleVector eval(int positionCount, DoubleVector valVector) { - try(DoubleVector.Builder result = driverContext.blockFactory().newDoubleVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendDouble(Ceil.process(valVector.getDouble(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "CeilDoubleEvaluator[" + "val=" + val + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val) { - this.source = source; - this.val = val; - } - - @Override - public CeilDoubleEvaluator get(DriverContext context) { - return new CeilDoubleEvaluator(source, val.get(context), context); - } - - @Override - public String toString() { - return "CeilDoubleEvaluator[" + "val=" + val + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/CosEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/CosEvaluator.java deleted file mode 100644 index dd3961845c244..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/CosEvaluator.java +++ /dev/null @@ -1,108 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Cos}. - * This class is generated. Do not edit it. - */ -public final class CosEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final DriverContext driverContext; - - public CosEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock valBlock = (DoubleBlock) val.eval(page)) { - DoubleVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector).asBlock(); - } - } - - public DoubleBlock eval(int positionCount, DoubleBlock valBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendDouble(Cos.process(valBlock.getDouble(valBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public DoubleVector eval(int positionCount, DoubleVector valVector) { - try(DoubleVector.Builder result = driverContext.blockFactory().newDoubleVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendDouble(Cos.process(valVector.getDouble(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "CosEvaluator[" + "val=" + val + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val) { - this.source = source; - this.val = val; - } - - @Override - public CosEvaluator get(DriverContext context) { - return new CosEvaluator(source, val.get(context), context); - } - - @Override - public String toString() { - return "CosEvaluator[" + "val=" + val + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/CoshEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/CoshEvaluator.java deleted file mode 100644 index 2f0bbaaacb40a..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/CoshEvaluator.java +++ /dev/null @@ -1,119 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Cosh}. - * This class is generated. Do not edit it. - */ -public final class CoshEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final DriverContext driverContext; - - public CoshEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock valBlock = (DoubleBlock) val.eval(page)) { - DoubleVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector); - } - } - - public DoubleBlock eval(int positionCount, DoubleBlock valBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendDouble(Cosh.process(valBlock.getDouble(valBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public DoubleBlock eval(int positionCount, DoubleVector valVector) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendDouble(Cosh.process(valVector.getDouble(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "CoshEvaluator[" + "val=" + val + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val) { - this.source = source; - this.val = val; - } - - @Override - public CoshEvaluator get(DriverContext context) { - return new CoshEvaluator(source, val.get(context), context); - } - - @Override - public String toString() { - return "CoshEvaluator[" + "val=" + val + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/FloorDoubleEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/FloorDoubleEvaluator.java deleted file mode 100644 index f8a10822a3c44..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/FloorDoubleEvaluator.java +++ /dev/null @@ -1,108 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Floor}. - * This class is generated. Do not edit it. - */ -public final class FloorDoubleEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final DriverContext driverContext; - - public FloorDoubleEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock valBlock = (DoubleBlock) val.eval(page)) { - DoubleVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector).asBlock(); - } - } - - public DoubleBlock eval(int positionCount, DoubleBlock valBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendDouble(Floor.process(valBlock.getDouble(valBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public DoubleVector eval(int positionCount, DoubleVector valVector) { - try(DoubleVector.Builder result = driverContext.blockFactory().newDoubleVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendDouble(Floor.process(valVector.getDouble(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "FloorDoubleEvaluator[" + "val=" + val + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val) { - this.source = source; - this.val = val; - } - - @Override - public FloorDoubleEvaluator get(DriverContext context) { - return new FloorDoubleEvaluator(source, val.get(context), context); - } - - @Override - public String toString() { - return "FloorDoubleEvaluator[" + "val=" + val + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/Log10DoubleEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/Log10DoubleEvaluator.java deleted file mode 100644 index d402cf7a79e68..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/Log10DoubleEvaluator.java +++ /dev/null @@ -1,119 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Log10}. - * This class is generated. Do not edit it. - */ -public final class Log10DoubleEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final DriverContext driverContext; - - public Log10DoubleEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock valBlock = (DoubleBlock) val.eval(page)) { - DoubleVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector); - } - } - - public DoubleBlock eval(int positionCount, DoubleBlock valBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendDouble(Log10.process(valBlock.getDouble(valBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public DoubleBlock eval(int positionCount, DoubleVector valVector) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendDouble(Log10.process(valVector.getDouble(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "Log10DoubleEvaluator[" + "val=" + val + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val) { - this.source = source; - this.val = val; - } - - @Override - public Log10DoubleEvaluator get(DriverContext context) { - return new Log10DoubleEvaluator(source, val.get(context), context); - } - - @Override - public String toString() { - return "Log10DoubleEvaluator[" + "val=" + val + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/Log10IntEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/Log10IntEvaluator.java deleted file mode 100644 index a1aa03af7d7f5..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/Log10IntEvaluator.java +++ /dev/null @@ -1,120 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Log10}. - * This class is generated. Do not edit it. - */ -public final class Log10IntEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final DriverContext driverContext; - - public Log10IntEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (IntBlock valBlock = (IntBlock) val.eval(page)) { - IntVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector); - } - } - - public DoubleBlock eval(int positionCount, IntBlock valBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendDouble(Log10.process(valBlock.getInt(valBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public DoubleBlock eval(int positionCount, IntVector valVector) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendDouble(Log10.process(valVector.getInt(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "Log10IntEvaluator[" + "val=" + val + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val) { - this.source = source; - this.val = val; - } - - @Override - public Log10IntEvaluator get(DriverContext context) { - return new Log10IntEvaluator(source, val.get(context), context); - } - - @Override - public String toString() { - return "Log10IntEvaluator[" + "val=" + val + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/Log10LongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/Log10LongEvaluator.java deleted file mode 100644 index 848baaea72b67..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/Log10LongEvaluator.java +++ /dev/null @@ -1,120 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Log10}. - * This class is generated. Do not edit it. - */ -public final class Log10LongEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final DriverContext driverContext; - - public Log10LongEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock valBlock = (LongBlock) val.eval(page)) { - LongVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector); - } - } - - public DoubleBlock eval(int positionCount, LongBlock valBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendDouble(Log10.process(valBlock.getLong(valBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public DoubleBlock eval(int positionCount, LongVector valVector) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendDouble(Log10.process(valVector.getLong(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "Log10LongEvaluator[" + "val=" + val + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val) { - this.source = source; - this.val = val; - } - - @Override - public Log10LongEvaluator get(DriverContext context) { - return new Log10LongEvaluator(source, val.get(context), context); - } - - @Override - public String toString() { - return "Log10LongEvaluator[" + "val=" + val + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/Log10UnsignedLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/Log10UnsignedLongEvaluator.java deleted file mode 100644 index 01812d8b1d2c9..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/Log10UnsignedLongEvaluator.java +++ /dev/null @@ -1,120 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Log10}. - * This class is generated. Do not edit it. - */ -public final class Log10UnsignedLongEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final DriverContext driverContext; - - public Log10UnsignedLongEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock valBlock = (LongBlock) val.eval(page)) { - LongVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector); - } - } - - public DoubleBlock eval(int positionCount, LongBlock valBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendDouble(Log10.processUnsignedLong(valBlock.getLong(valBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public DoubleBlock eval(int positionCount, LongVector valVector) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendDouble(Log10.processUnsignedLong(valVector.getLong(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "Log10UnsignedLongEvaluator[" + "val=" + val + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val) { - this.source = source; - this.val = val; - } - - @Override - public Log10UnsignedLongEvaluator get(DriverContext context) { - return new Log10UnsignedLongEvaluator(source, val.get(context), context); - } - - @Override - public String toString() { - return "Log10UnsignedLongEvaluator[" + "val=" + val + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/LogConstantEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/LogConstantEvaluator.java deleted file mode 100644 index ff814b530b108..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/LogConstantEvaluator.java +++ /dev/null @@ -1,119 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Log}. - * This class is generated. Do not edit it. - */ -public final class LogConstantEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator value; - - private final DriverContext driverContext; - - public LogConstantEvaluator(Source source, EvalOperator.ExpressionEvaluator value, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.value = value; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock valueBlock = (DoubleBlock) value.eval(page)) { - DoubleVector valueVector = valueBlock.asVector(); - if (valueVector == null) { - return eval(page.getPositionCount(), valueBlock); - } - return eval(page.getPositionCount(), valueVector); - } - } - - public DoubleBlock eval(int positionCount, DoubleBlock valueBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valueBlock.getValueCount(p) != 1) { - if (valueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendDouble(Log.process(valueBlock.getDouble(valueBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public DoubleBlock eval(int positionCount, DoubleVector valueVector) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendDouble(Log.process(valueVector.getDouble(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "LogConstantEvaluator[" + "value=" + value + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(value); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory value; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory value) { - this.source = source; - this.value = value; - } - - @Override - public LogConstantEvaluator get(DriverContext context) { - return new LogConstantEvaluator(source, value.get(context), context); - } - - @Override - public String toString() { - return "LogConstantEvaluator[" + "value=" + value + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/LogEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/LogEvaluator.java deleted file mode 100644 index 7fcfb37483bb7..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/LogEvaluator.java +++ /dev/null @@ -1,143 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Log}. - * This class is generated. Do not edit it. - */ -public final class LogEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator base; - - private final EvalOperator.ExpressionEvaluator value; - - private final DriverContext driverContext; - - public LogEvaluator(Source source, EvalOperator.ExpressionEvaluator base, - EvalOperator.ExpressionEvaluator value, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.base = base; - this.value = value; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock baseBlock = (DoubleBlock) base.eval(page)) { - try (DoubleBlock valueBlock = (DoubleBlock) value.eval(page)) { - DoubleVector baseVector = baseBlock.asVector(); - if (baseVector == null) { - return eval(page.getPositionCount(), baseBlock, valueBlock); - } - DoubleVector valueVector = valueBlock.asVector(); - if (valueVector == null) { - return eval(page.getPositionCount(), baseBlock, valueBlock); - } - return eval(page.getPositionCount(), baseVector, valueVector); - } - } - } - - public DoubleBlock eval(int positionCount, DoubleBlock baseBlock, DoubleBlock valueBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (baseBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (baseBlock.getValueCount(p) != 1) { - if (baseBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (valueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valueBlock.getValueCount(p) != 1) { - if (valueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendDouble(Log.process(baseBlock.getDouble(baseBlock.getFirstValueIndex(p)), valueBlock.getDouble(valueBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public DoubleBlock eval(int positionCount, DoubleVector baseVector, DoubleVector valueVector) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendDouble(Log.process(baseVector.getDouble(p), valueVector.getDouble(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "LogEvaluator[" + "base=" + base + ", value=" + value + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(base, value); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory base; - - private final EvalOperator.ExpressionEvaluator.Factory value; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory base, - EvalOperator.ExpressionEvaluator.Factory value) { - this.source = source; - this.base = base; - this.value = value; - } - - @Override - public LogEvaluator get(DriverContext context) { - return new LogEvaluator(source, base.get(context), value.get(context), context); - } - - @Override - public String toString() { - return "LogEvaluator[" + "base=" + base + ", value=" + value + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/PowEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/PowEvaluator.java deleted file mode 100644 index 33bf2b4bd0c25..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/PowEvaluator.java +++ /dev/null @@ -1,143 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Pow}. - * This class is generated. Do not edit it. - */ -public final class PowEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator base; - - private final EvalOperator.ExpressionEvaluator exponent; - - private final DriverContext driverContext; - - public PowEvaluator(Source source, EvalOperator.ExpressionEvaluator base, - EvalOperator.ExpressionEvaluator exponent, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.base = base; - this.exponent = exponent; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock baseBlock = (DoubleBlock) base.eval(page)) { - try (DoubleBlock exponentBlock = (DoubleBlock) exponent.eval(page)) { - DoubleVector baseVector = baseBlock.asVector(); - if (baseVector == null) { - return eval(page.getPositionCount(), baseBlock, exponentBlock); - } - DoubleVector exponentVector = exponentBlock.asVector(); - if (exponentVector == null) { - return eval(page.getPositionCount(), baseBlock, exponentBlock); - } - return eval(page.getPositionCount(), baseVector, exponentVector); - } - } - } - - public DoubleBlock eval(int positionCount, DoubleBlock baseBlock, DoubleBlock exponentBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (baseBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (baseBlock.getValueCount(p) != 1) { - if (baseBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (exponentBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (exponentBlock.getValueCount(p) != 1) { - if (exponentBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendDouble(Pow.process(baseBlock.getDouble(baseBlock.getFirstValueIndex(p)), exponentBlock.getDouble(exponentBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public DoubleBlock eval(int positionCount, DoubleVector baseVector, DoubleVector exponentVector) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendDouble(Pow.process(baseVector.getDouble(p), exponentVector.getDouble(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "PowEvaluator[" + "base=" + base + ", exponent=" + exponent + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(base, exponent); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory base; - - private final EvalOperator.ExpressionEvaluator.Factory exponent; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory base, - EvalOperator.ExpressionEvaluator.Factory exponent) { - this.source = source; - this.base = base; - this.exponent = exponent; - } - - @Override - public PowEvaluator get(DriverContext context) { - return new PowEvaluator(source, base.get(context), exponent.get(context), context); - } - - @Override - public String toString() { - return "PowEvaluator[" + "base=" + base + ", exponent=" + exponent + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/RoundDoubleEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/RoundDoubleEvaluator.java deleted file mode 100644 index 3b85a32fc3081..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/RoundDoubleEvaluator.java +++ /dev/null @@ -1,134 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Round}. - * This class is generated. Do not edit it. - */ -public final class RoundDoubleEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final EvalOperator.ExpressionEvaluator decimals; - - private final DriverContext driverContext; - - public RoundDoubleEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - EvalOperator.ExpressionEvaluator decimals, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.decimals = decimals; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock valBlock = (DoubleBlock) val.eval(page)) { - try (LongBlock decimalsBlock = (LongBlock) decimals.eval(page)) { - DoubleVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock, decimalsBlock); - } - LongVector decimalsVector = decimalsBlock.asVector(); - if (decimalsVector == null) { - return eval(page.getPositionCount(), valBlock, decimalsBlock); - } - return eval(page.getPositionCount(), valVector, decimalsVector).asBlock(); - } - } - } - - public DoubleBlock eval(int positionCount, DoubleBlock valBlock, LongBlock decimalsBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (decimalsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (decimalsBlock.getValueCount(p) != 1) { - if (decimalsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendDouble(Round.process(valBlock.getDouble(valBlock.getFirstValueIndex(p)), decimalsBlock.getLong(decimalsBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public DoubleVector eval(int positionCount, DoubleVector valVector, LongVector decimalsVector) { - try(DoubleVector.Builder result = driverContext.blockFactory().newDoubleVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendDouble(Round.process(valVector.getDouble(p), decimalsVector.getLong(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "RoundDoubleEvaluator[" + "val=" + val + ", decimals=" + decimals + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val, decimals); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - private final EvalOperator.ExpressionEvaluator.Factory decimals; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val, - EvalOperator.ExpressionEvaluator.Factory decimals) { - this.source = source; - this.val = val; - this.decimals = decimals; - } - - @Override - public RoundDoubleEvaluator get(DriverContext context) { - return new RoundDoubleEvaluator(source, val.get(context), decimals.get(context), context); - } - - @Override - public String toString() { - return "RoundDoubleEvaluator[" + "val=" + val + ", decimals=" + decimals + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/RoundDoubleNoDecimalsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/RoundDoubleNoDecimalsEvaluator.java deleted file mode 100644 index c36a1fe25b61f..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/RoundDoubleNoDecimalsEvaluator.java +++ /dev/null @@ -1,108 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Round}. - * This class is generated. Do not edit it. - */ -public final class RoundDoubleNoDecimalsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final DriverContext driverContext; - - public RoundDoubleNoDecimalsEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock valBlock = (DoubleBlock) val.eval(page)) { - DoubleVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector).asBlock(); - } - } - - public DoubleBlock eval(int positionCount, DoubleBlock valBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendDouble(Round.process(valBlock.getDouble(valBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public DoubleVector eval(int positionCount, DoubleVector valVector) { - try(DoubleVector.Builder result = driverContext.blockFactory().newDoubleVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendDouble(Round.process(valVector.getDouble(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "RoundDoubleNoDecimalsEvaluator[" + "val=" + val + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val) { - this.source = source; - this.val = val; - } - - @Override - public RoundDoubleNoDecimalsEvaluator get(DriverContext context) { - return new RoundDoubleNoDecimalsEvaluator(source, val.get(context), context); - } - - @Override - public String toString() { - return "RoundDoubleNoDecimalsEvaluator[" + "val=" + val + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/RoundIntEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/RoundIntEvaluator.java deleted file mode 100644 index f96f92e5d0b38..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/RoundIntEvaluator.java +++ /dev/null @@ -1,134 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Round}. - * This class is generated. Do not edit it. - */ -public final class RoundIntEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final EvalOperator.ExpressionEvaluator decimals; - - private final DriverContext driverContext; - - public RoundIntEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - EvalOperator.ExpressionEvaluator decimals, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.decimals = decimals; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (IntBlock valBlock = (IntBlock) val.eval(page)) { - try (LongBlock decimalsBlock = (LongBlock) decimals.eval(page)) { - IntVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock, decimalsBlock); - } - LongVector decimalsVector = decimalsBlock.asVector(); - if (decimalsVector == null) { - return eval(page.getPositionCount(), valBlock, decimalsBlock); - } - return eval(page.getPositionCount(), valVector, decimalsVector).asBlock(); - } - } - } - - public IntBlock eval(int positionCount, IntBlock valBlock, LongBlock decimalsBlock) { - try(IntBlock.Builder result = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (decimalsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (decimalsBlock.getValueCount(p) != 1) { - if (decimalsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendInt(Round.process(valBlock.getInt(valBlock.getFirstValueIndex(p)), decimalsBlock.getLong(decimalsBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public IntVector eval(int positionCount, IntVector valVector, LongVector decimalsVector) { - try(IntVector.Builder result = driverContext.blockFactory().newIntVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendInt(Round.process(valVector.getInt(p), decimalsVector.getLong(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "RoundIntEvaluator[" + "val=" + val + ", decimals=" + decimals + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val, decimals); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - private final EvalOperator.ExpressionEvaluator.Factory decimals; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val, - EvalOperator.ExpressionEvaluator.Factory decimals) { - this.source = source; - this.val = val; - this.decimals = decimals; - } - - @Override - public RoundIntEvaluator get(DriverContext context) { - return new RoundIntEvaluator(source, val.get(context), decimals.get(context), context); - } - - @Override - public String toString() { - return "RoundIntEvaluator[" + "val=" + val + ", decimals=" + decimals + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/RoundLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/RoundLongEvaluator.java deleted file mode 100644 index c8a2fdd384f40..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/RoundLongEvaluator.java +++ /dev/null @@ -1,132 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Round}. - * This class is generated. Do not edit it. - */ -public final class RoundLongEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final EvalOperator.ExpressionEvaluator decimals; - - private final DriverContext driverContext; - - public RoundLongEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - EvalOperator.ExpressionEvaluator decimals, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.decimals = decimals; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock valBlock = (LongBlock) val.eval(page)) { - try (LongBlock decimalsBlock = (LongBlock) decimals.eval(page)) { - LongVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock, decimalsBlock); - } - LongVector decimalsVector = decimalsBlock.asVector(); - if (decimalsVector == null) { - return eval(page.getPositionCount(), valBlock, decimalsBlock); - } - return eval(page.getPositionCount(), valVector, decimalsVector).asBlock(); - } - } - } - - public LongBlock eval(int positionCount, LongBlock valBlock, LongBlock decimalsBlock) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (decimalsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (decimalsBlock.getValueCount(p) != 1) { - if (decimalsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendLong(Round.process(valBlock.getLong(valBlock.getFirstValueIndex(p)), decimalsBlock.getLong(decimalsBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public LongVector eval(int positionCount, LongVector valVector, LongVector decimalsVector) { - try(LongVector.Builder result = driverContext.blockFactory().newLongVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendLong(Round.process(valVector.getLong(p), decimalsVector.getLong(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "RoundLongEvaluator[" + "val=" + val + ", decimals=" + decimals + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val, decimals); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - private final EvalOperator.ExpressionEvaluator.Factory decimals; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val, - EvalOperator.ExpressionEvaluator.Factory decimals) { - this.source = source; - this.val = val; - this.decimals = decimals; - } - - @Override - public RoundLongEvaluator get(DriverContext context) { - return new RoundLongEvaluator(source, val.get(context), decimals.get(context), context); - } - - @Override - public String toString() { - return "RoundLongEvaluator[" + "val=" + val + ", decimals=" + decimals + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/RoundUnsignedLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/RoundUnsignedLongEvaluator.java deleted file mode 100644 index 5c94e386d4978..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/RoundUnsignedLongEvaluator.java +++ /dev/null @@ -1,132 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Round}. - * This class is generated. Do not edit it. - */ -public final class RoundUnsignedLongEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final EvalOperator.ExpressionEvaluator decimals; - - private final DriverContext driverContext; - - public RoundUnsignedLongEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - EvalOperator.ExpressionEvaluator decimals, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.decimals = decimals; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock valBlock = (LongBlock) val.eval(page)) { - try (LongBlock decimalsBlock = (LongBlock) decimals.eval(page)) { - LongVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock, decimalsBlock); - } - LongVector decimalsVector = decimalsBlock.asVector(); - if (decimalsVector == null) { - return eval(page.getPositionCount(), valBlock, decimalsBlock); - } - return eval(page.getPositionCount(), valVector, decimalsVector).asBlock(); - } - } - } - - public LongBlock eval(int positionCount, LongBlock valBlock, LongBlock decimalsBlock) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (decimalsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (decimalsBlock.getValueCount(p) != 1) { - if (decimalsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendLong(Round.processUnsignedLong(valBlock.getLong(valBlock.getFirstValueIndex(p)), decimalsBlock.getLong(decimalsBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public LongVector eval(int positionCount, LongVector valVector, LongVector decimalsVector) { - try(LongVector.Builder result = driverContext.blockFactory().newLongVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendLong(Round.processUnsignedLong(valVector.getLong(p), decimalsVector.getLong(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "RoundUnsignedLongEvaluator[" + "val=" + val + ", decimals=" + decimals + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val, decimals); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - private final EvalOperator.ExpressionEvaluator.Factory decimals; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val, - EvalOperator.ExpressionEvaluator.Factory decimals) { - this.source = source; - this.val = val; - this.decimals = decimals; - } - - @Override - public RoundUnsignedLongEvaluator get(DriverContext context) { - return new RoundUnsignedLongEvaluator(source, val.get(context), decimals.get(context), context); - } - - @Override - public String toString() { - return "RoundUnsignedLongEvaluator[" + "val=" + val + ", decimals=" + decimals + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/SignumDoubleEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/SignumDoubleEvaluator.java deleted file mode 100644 index c7d21a7b9c5a0..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/SignumDoubleEvaluator.java +++ /dev/null @@ -1,108 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Signum}. - * This class is generated. Do not edit it. - */ -public final class SignumDoubleEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final DriverContext driverContext; - - public SignumDoubleEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock valBlock = (DoubleBlock) val.eval(page)) { - DoubleVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector).asBlock(); - } - } - - public DoubleBlock eval(int positionCount, DoubleBlock valBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendDouble(Signum.process(valBlock.getDouble(valBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public DoubleVector eval(int positionCount, DoubleVector valVector) { - try(DoubleVector.Builder result = driverContext.blockFactory().newDoubleVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendDouble(Signum.process(valVector.getDouble(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "SignumDoubleEvaluator[" + "val=" + val + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val) { - this.source = source; - this.val = val; - } - - @Override - public SignumDoubleEvaluator get(DriverContext context) { - return new SignumDoubleEvaluator(source, val.get(context), context); - } - - @Override - public String toString() { - return "SignumDoubleEvaluator[" + "val=" + val + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/SignumIntEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/SignumIntEvaluator.java deleted file mode 100644 index 939807d8deffa..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/SignumIntEvaluator.java +++ /dev/null @@ -1,110 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Signum}. - * This class is generated. Do not edit it. - */ -public final class SignumIntEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final DriverContext driverContext; - - public SignumIntEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (IntBlock valBlock = (IntBlock) val.eval(page)) { - IntVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector).asBlock(); - } - } - - public DoubleBlock eval(int positionCount, IntBlock valBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendDouble(Signum.process(valBlock.getInt(valBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public DoubleVector eval(int positionCount, IntVector valVector) { - try(DoubleVector.Builder result = driverContext.blockFactory().newDoubleVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendDouble(Signum.process(valVector.getInt(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "SignumIntEvaluator[" + "val=" + val + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val) { - this.source = source; - this.val = val; - } - - @Override - public SignumIntEvaluator get(DriverContext context) { - return new SignumIntEvaluator(source, val.get(context), context); - } - - @Override - public String toString() { - return "SignumIntEvaluator[" + "val=" + val + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/SignumLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/SignumLongEvaluator.java deleted file mode 100644 index 0c4af4671672a..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/SignumLongEvaluator.java +++ /dev/null @@ -1,110 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Signum}. - * This class is generated. Do not edit it. - */ -public final class SignumLongEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final DriverContext driverContext; - - public SignumLongEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock valBlock = (LongBlock) val.eval(page)) { - LongVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector).asBlock(); - } - } - - public DoubleBlock eval(int positionCount, LongBlock valBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendDouble(Signum.process(valBlock.getLong(valBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public DoubleVector eval(int positionCount, LongVector valVector) { - try(DoubleVector.Builder result = driverContext.blockFactory().newDoubleVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendDouble(Signum.process(valVector.getLong(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "SignumLongEvaluator[" + "val=" + val + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val) { - this.source = source; - this.val = val; - } - - @Override - public SignumLongEvaluator get(DriverContext context) { - return new SignumLongEvaluator(source, val.get(context), context); - } - - @Override - public String toString() { - return "SignumLongEvaluator[" + "val=" + val + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/SignumUnsignedLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/SignumUnsignedLongEvaluator.java deleted file mode 100644 index d3b20c98139c4..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/SignumUnsignedLongEvaluator.java +++ /dev/null @@ -1,110 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Signum}. - * This class is generated. Do not edit it. - */ -public final class SignumUnsignedLongEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final DriverContext driverContext; - - public SignumUnsignedLongEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock valBlock = (LongBlock) val.eval(page)) { - LongVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector).asBlock(); - } - } - - public DoubleBlock eval(int positionCount, LongBlock valBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendDouble(Signum.processUnsignedLong(valBlock.getLong(valBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public DoubleVector eval(int positionCount, LongVector valVector) { - try(DoubleVector.Builder result = driverContext.blockFactory().newDoubleVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendDouble(Signum.processUnsignedLong(valVector.getLong(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "SignumUnsignedLongEvaluator[" + "val=" + val + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val) { - this.source = source; - this.val = val; - } - - @Override - public SignumUnsignedLongEvaluator get(DriverContext context) { - return new SignumUnsignedLongEvaluator(source, val.get(context), context); - } - - @Override - public String toString() { - return "SignumUnsignedLongEvaluator[" + "val=" + val + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/SinEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/SinEvaluator.java deleted file mode 100644 index a3c9e1481c19e..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/SinEvaluator.java +++ /dev/null @@ -1,108 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Sin}. - * This class is generated. Do not edit it. - */ -public final class SinEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final DriverContext driverContext; - - public SinEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock valBlock = (DoubleBlock) val.eval(page)) { - DoubleVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector).asBlock(); - } - } - - public DoubleBlock eval(int positionCount, DoubleBlock valBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendDouble(Sin.process(valBlock.getDouble(valBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public DoubleVector eval(int positionCount, DoubleVector valVector) { - try(DoubleVector.Builder result = driverContext.blockFactory().newDoubleVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendDouble(Sin.process(valVector.getDouble(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "SinEvaluator[" + "val=" + val + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val) { - this.source = source; - this.val = val; - } - - @Override - public SinEvaluator get(DriverContext context) { - return new SinEvaluator(source, val.get(context), context); - } - - @Override - public String toString() { - return "SinEvaluator[" + "val=" + val + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/SinhEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/SinhEvaluator.java deleted file mode 100644 index c6020d6bd86ea..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/SinhEvaluator.java +++ /dev/null @@ -1,119 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Sinh}. - * This class is generated. Do not edit it. - */ -public final class SinhEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final DriverContext driverContext; - - public SinhEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock valBlock = (DoubleBlock) val.eval(page)) { - DoubleVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector); - } - } - - public DoubleBlock eval(int positionCount, DoubleBlock valBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendDouble(Sinh.process(valBlock.getDouble(valBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public DoubleBlock eval(int positionCount, DoubleVector valVector) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendDouble(Sinh.process(valVector.getDouble(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SinhEvaluator[" + "val=" + val + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val) { - this.source = source; - this.val = val; - } - - @Override - public SinhEvaluator get(DriverContext context) { - return new SinhEvaluator(source, val.get(context), context); - } - - @Override - public String toString() { - return "SinhEvaluator[" + "val=" + val + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/SqrtDoubleEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/SqrtDoubleEvaluator.java deleted file mode 100644 index 516d6639fb115..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/SqrtDoubleEvaluator.java +++ /dev/null @@ -1,119 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Sqrt}. - * This class is generated. Do not edit it. - */ -public final class SqrtDoubleEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final DriverContext driverContext; - - public SqrtDoubleEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock valBlock = (DoubleBlock) val.eval(page)) { - DoubleVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector); - } - } - - public DoubleBlock eval(int positionCount, DoubleBlock valBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendDouble(Sqrt.process(valBlock.getDouble(valBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public DoubleBlock eval(int positionCount, DoubleVector valVector) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendDouble(Sqrt.process(valVector.getDouble(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SqrtDoubleEvaluator[" + "val=" + val + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val) { - this.source = source; - this.val = val; - } - - @Override - public SqrtDoubleEvaluator get(DriverContext context) { - return new SqrtDoubleEvaluator(source, val.get(context), context); - } - - @Override - public String toString() { - return "SqrtDoubleEvaluator[" + "val=" + val + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/SqrtIntEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/SqrtIntEvaluator.java deleted file mode 100644 index 3719bc6bd7326..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/SqrtIntEvaluator.java +++ /dev/null @@ -1,120 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Sqrt}. - * This class is generated. Do not edit it. - */ -public final class SqrtIntEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final DriverContext driverContext; - - public SqrtIntEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (IntBlock valBlock = (IntBlock) val.eval(page)) { - IntVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector); - } - } - - public DoubleBlock eval(int positionCount, IntBlock valBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendDouble(Sqrt.process(valBlock.getInt(valBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public DoubleBlock eval(int positionCount, IntVector valVector) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendDouble(Sqrt.process(valVector.getInt(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SqrtIntEvaluator[" + "val=" + val + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val) { - this.source = source; - this.val = val; - } - - @Override - public SqrtIntEvaluator get(DriverContext context) { - return new SqrtIntEvaluator(source, val.get(context), context); - } - - @Override - public String toString() { - return "SqrtIntEvaluator[" + "val=" + val + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/SqrtLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/SqrtLongEvaluator.java deleted file mode 100644 index a9620291ddd8b..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/SqrtLongEvaluator.java +++ /dev/null @@ -1,120 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Sqrt}. - * This class is generated. Do not edit it. - */ -public final class SqrtLongEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final DriverContext driverContext; - - public SqrtLongEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock valBlock = (LongBlock) val.eval(page)) { - LongVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector); - } - } - - public DoubleBlock eval(int positionCount, LongBlock valBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendDouble(Sqrt.process(valBlock.getLong(valBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public DoubleBlock eval(int positionCount, LongVector valVector) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendDouble(Sqrt.process(valVector.getLong(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SqrtLongEvaluator[" + "val=" + val + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val) { - this.source = source; - this.val = val; - } - - @Override - public SqrtLongEvaluator get(DriverContext context) { - return new SqrtLongEvaluator(source, val.get(context), context); - } - - @Override - public String toString() { - return "SqrtLongEvaluator[" + "val=" + val + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/SqrtUnsignedLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/SqrtUnsignedLongEvaluator.java deleted file mode 100644 index 6478f0639bb9a..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/SqrtUnsignedLongEvaluator.java +++ /dev/null @@ -1,110 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Sqrt}. - * This class is generated. Do not edit it. - */ -public final class SqrtUnsignedLongEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final DriverContext driverContext; - - public SqrtUnsignedLongEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock valBlock = (LongBlock) val.eval(page)) { - LongVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector).asBlock(); - } - } - - public DoubleBlock eval(int positionCount, LongBlock valBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendDouble(Sqrt.processUnsignedLong(valBlock.getLong(valBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public DoubleVector eval(int positionCount, LongVector valVector) { - try(DoubleVector.Builder result = driverContext.blockFactory().newDoubleVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendDouble(Sqrt.processUnsignedLong(valVector.getLong(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "SqrtUnsignedLongEvaluator[" + "val=" + val + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val) { - this.source = source; - this.val = val; - } - - @Override - public SqrtUnsignedLongEvaluator get(DriverContext context) { - return new SqrtUnsignedLongEvaluator(source, val.get(context), context); - } - - @Override - public String toString() { - return "SqrtUnsignedLongEvaluator[" + "val=" + val + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/TanEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/TanEvaluator.java deleted file mode 100644 index ed410d20d122b..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/TanEvaluator.java +++ /dev/null @@ -1,108 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Tan}. - * This class is generated. Do not edit it. - */ -public final class TanEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final DriverContext driverContext; - - public TanEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock valBlock = (DoubleBlock) val.eval(page)) { - DoubleVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector).asBlock(); - } - } - - public DoubleBlock eval(int positionCount, DoubleBlock valBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendDouble(Tan.process(valBlock.getDouble(valBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public DoubleVector eval(int positionCount, DoubleVector valVector) { - try(DoubleVector.Builder result = driverContext.blockFactory().newDoubleVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendDouble(Tan.process(valVector.getDouble(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "TanEvaluator[" + "val=" + val + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val) { - this.source = source; - this.val = val; - } - - @Override - public TanEvaluator get(DriverContext context) { - return new TanEvaluator(source, val.get(context), context); - } - - @Override - public String toString() { - return "TanEvaluator[" + "val=" + val + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/TanhEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/TanhEvaluator.java deleted file mode 100644 index 94fa4fad18fd3..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/math/TanhEvaluator.java +++ /dev/null @@ -1,108 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.math; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Tanh}. - * This class is generated. Do not edit it. - */ -public final class TanhEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final DriverContext driverContext; - - public TanhEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock valBlock = (DoubleBlock) val.eval(page)) { - DoubleVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector).asBlock(); - } - } - - public DoubleBlock eval(int positionCount, DoubleBlock valBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendDouble(Tanh.process(valBlock.getDouble(valBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public DoubleVector eval(int positionCount, DoubleVector valVector) { - try(DoubleVector.Builder result = driverContext.blockFactory().newDoubleVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendDouble(Tanh.process(valVector.getDouble(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "TanhEvaluator[" + "val=" + val + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val) { - this.source = source; - this.val = val; - } - - @Override - public TanhEvaluator get(DriverContext context) { - return new TanhEvaluator(source, val.get(context), context); - } - - @Override - public String toString() { - return "TanhEvaluator[" + "val=" + val + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvAvgDoubleEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvAvgDoubleEvaluator.java deleted file mode 100644 index d87444746d2c6..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvAvgDoubleEvaluator.java +++ /dev/null @@ -1,99 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.search.aggregations.metrics.CompensatedSum; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvAvg}. - * This class is generated. Do not edit it. - */ -public final class MvAvgDoubleEvaluator extends AbstractMultivalueFunction.AbstractEvaluator { - public MvAvgDoubleEvaluator(EvalOperator.ExpressionEvaluator field, DriverContext driverContext) { - super(driverContext, field); - } - - @Override - public String name() { - return "MvAvg"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - DoubleBlock v = (DoubleBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - CompensatedSum work = new CompensatedSum(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - for (int i = first; i < end; i++) { - double value = v.getDouble(i); - MvAvg.process(work, value); - } - double result = MvAvg.finish(work, valueCount); - builder.appendDouble(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNotNullable(Block fieldVal) { - DoubleBlock v = (DoubleBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (DoubleVector.FixedBuilder builder = driverContext.blockFactory().newDoubleVectorFixedBuilder(positionCount)) { - CompensatedSum work = new CompensatedSum(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - for (int i = first; i < end; i++) { - double value = v.getDouble(i); - MvAvg.process(work, value); - } - double result = MvAvg.finish(work, valueCount); - builder.appendDouble(result); - } - return builder.build().asBlock(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field) { - this.field = field; - } - - @Override - public MvAvgDoubleEvaluator get(DriverContext context) { - return new MvAvgDoubleEvaluator(field.get(context), context); - } - - @Override - public String toString() { - return "MvAvg[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvAvgIntEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvAvgIntEvaluator.java deleted file mode 100644 index 63e6b4eb12106..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvAvgIntEvaluator.java +++ /dev/null @@ -1,158 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.search.aggregations.metrics.CompensatedSum; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvAvg}. - * This class is generated. Do not edit it. - */ -public final class MvAvgIntEvaluator extends AbstractMultivalueFunction.AbstractEvaluator { - public MvAvgIntEvaluator(EvalOperator.ExpressionEvaluator field, DriverContext driverContext) { - super(driverContext, field); - } - - @Override - public String name() { - return "MvAvg"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - IntBlock v = (IntBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - CompensatedSum work = new CompensatedSum(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - if (valueCount == 1) { - int value = v.getInt(first); - double result = MvAvg.single(value); - builder.appendDouble(result); - continue; - } - int end = first + valueCount; - for (int i = first; i < end; i++) { - int value = v.getInt(i); - MvAvg.process(work, value); - } - double result = MvAvg.finish(work, valueCount); - builder.appendDouble(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNotNullable(Block fieldVal) { - IntBlock v = (IntBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (DoubleVector.FixedBuilder builder = driverContext.blockFactory().newDoubleVectorFixedBuilder(positionCount)) { - CompensatedSum work = new CompensatedSum(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - if (valueCount == 1) { - int value = v.getInt(first); - double result = MvAvg.single(value); - builder.appendDouble(result); - continue; - } - int end = first + valueCount; - for (int i = first; i < end; i++) { - int value = v.getInt(i); - MvAvg.process(work, value); - } - double result = MvAvg.finish(work, valueCount); - builder.appendDouble(result); - } - return builder.build().asBlock(); - } - } - - /** - * Evaluate blocks containing only single valued fields. - */ - @Override - public Block evalSingleValuedNullable(Block fieldVal) { - IntBlock v = (IntBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - CompensatedSum work = new CompensatedSum(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - assert valueCount == 1; - int first = v.getFirstValueIndex(p); - int value = v.getInt(first); - double result = MvAvg.single(value); - builder.appendDouble(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing only single valued fields. - */ - @Override - public Block evalSingleValuedNotNullable(Block fieldVal) { - IntBlock v = (IntBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (DoubleVector.FixedBuilder builder = driverContext.blockFactory().newDoubleVectorFixedBuilder(positionCount)) { - CompensatedSum work = new CompensatedSum(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - assert valueCount == 1; - int first = v.getFirstValueIndex(p); - int value = v.getInt(first); - double result = MvAvg.single(value); - builder.appendDouble(result); - } - return builder.build().asBlock(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field) { - this.field = field; - } - - @Override - public MvAvgIntEvaluator get(DriverContext context) { - return new MvAvgIntEvaluator(field.get(context), context); - } - - @Override - public String toString() { - return "MvAvg[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvAvgLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvAvgLongEvaluator.java deleted file mode 100644 index d699070747b49..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvAvgLongEvaluator.java +++ /dev/null @@ -1,158 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.search.aggregations.metrics.CompensatedSum; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvAvg}. - * This class is generated. Do not edit it. - */ -public final class MvAvgLongEvaluator extends AbstractMultivalueFunction.AbstractEvaluator { - public MvAvgLongEvaluator(EvalOperator.ExpressionEvaluator field, DriverContext driverContext) { - super(driverContext, field); - } - - @Override - public String name() { - return "MvAvg"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - LongBlock v = (LongBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - CompensatedSum work = new CompensatedSum(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - if (valueCount == 1) { - long value = v.getLong(first); - double result = MvAvg.single(value); - builder.appendDouble(result); - continue; - } - int end = first + valueCount; - for (int i = first; i < end; i++) { - long value = v.getLong(i); - MvAvg.process(work, value); - } - double result = MvAvg.finish(work, valueCount); - builder.appendDouble(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNotNullable(Block fieldVal) { - LongBlock v = (LongBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (DoubleVector.FixedBuilder builder = driverContext.blockFactory().newDoubleVectorFixedBuilder(positionCount)) { - CompensatedSum work = new CompensatedSum(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - if (valueCount == 1) { - long value = v.getLong(first); - double result = MvAvg.single(value); - builder.appendDouble(result); - continue; - } - int end = first + valueCount; - for (int i = first; i < end; i++) { - long value = v.getLong(i); - MvAvg.process(work, value); - } - double result = MvAvg.finish(work, valueCount); - builder.appendDouble(result); - } - return builder.build().asBlock(); - } - } - - /** - * Evaluate blocks containing only single valued fields. - */ - @Override - public Block evalSingleValuedNullable(Block fieldVal) { - LongBlock v = (LongBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - CompensatedSum work = new CompensatedSum(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - assert valueCount == 1; - int first = v.getFirstValueIndex(p); - long value = v.getLong(first); - double result = MvAvg.single(value); - builder.appendDouble(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing only single valued fields. - */ - @Override - public Block evalSingleValuedNotNullable(Block fieldVal) { - LongBlock v = (LongBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (DoubleVector.FixedBuilder builder = driverContext.blockFactory().newDoubleVectorFixedBuilder(positionCount)) { - CompensatedSum work = new CompensatedSum(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - assert valueCount == 1; - int first = v.getFirstValueIndex(p); - long value = v.getLong(first); - double result = MvAvg.single(value); - builder.appendDouble(result); - } - return builder.build().asBlock(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field) { - this.field = field; - } - - @Override - public MvAvgLongEvaluator get(DriverContext context) { - return new MvAvgLongEvaluator(field.get(context), context); - } - - @Override - public String toString() { - return "MvAvg[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvAvgUnsignedLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvAvgUnsignedLongEvaluator.java deleted file mode 100644 index 5c63508fa3560..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvAvgUnsignedLongEvaluator.java +++ /dev/null @@ -1,159 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.search.aggregations.metrics.CompensatedSum; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvAvg}. - * This class is generated. Do not edit it. - */ -public final class MvAvgUnsignedLongEvaluator extends AbstractMultivalueFunction.AbstractEvaluator { - public MvAvgUnsignedLongEvaluator(EvalOperator.ExpressionEvaluator field, - DriverContext driverContext) { - super(driverContext, field); - } - - @Override - public String name() { - return "MvAvg"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - LongBlock v = (LongBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - CompensatedSum work = new CompensatedSum(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - if (valueCount == 1) { - long value = v.getLong(first); - double result = MvAvg.singleUnsignedLong(value); - builder.appendDouble(result); - continue; - } - int end = first + valueCount; - for (int i = first; i < end; i++) { - long value = v.getLong(i); - MvAvg.processUnsignedLong(work, value); - } - double result = MvAvg.finish(work, valueCount); - builder.appendDouble(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNotNullable(Block fieldVal) { - LongBlock v = (LongBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (DoubleVector.FixedBuilder builder = driverContext.blockFactory().newDoubleVectorFixedBuilder(positionCount)) { - CompensatedSum work = new CompensatedSum(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - if (valueCount == 1) { - long value = v.getLong(first); - double result = MvAvg.singleUnsignedLong(value); - builder.appendDouble(result); - continue; - } - int end = first + valueCount; - for (int i = first; i < end; i++) { - long value = v.getLong(i); - MvAvg.processUnsignedLong(work, value); - } - double result = MvAvg.finish(work, valueCount); - builder.appendDouble(result); - } - return builder.build().asBlock(); - } - } - - /** - * Evaluate blocks containing only single valued fields. - */ - @Override - public Block evalSingleValuedNullable(Block fieldVal) { - LongBlock v = (LongBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - CompensatedSum work = new CompensatedSum(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - assert valueCount == 1; - int first = v.getFirstValueIndex(p); - long value = v.getLong(first); - double result = MvAvg.singleUnsignedLong(value); - builder.appendDouble(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing only single valued fields. - */ - @Override - public Block evalSingleValuedNotNullable(Block fieldVal) { - LongBlock v = (LongBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (DoubleVector.FixedBuilder builder = driverContext.blockFactory().newDoubleVectorFixedBuilder(positionCount)) { - CompensatedSum work = new CompensatedSum(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - assert valueCount == 1; - int first = v.getFirstValueIndex(p); - long value = v.getLong(first); - double result = MvAvg.singleUnsignedLong(value); - builder.appendDouble(result); - } - return builder.build().asBlock(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field) { - this.field = field; - } - - @Override - public MvAvgUnsignedLongEvaluator get(DriverContext context) { - return new MvAvgUnsignedLongEvaluator(field.get(context), context); - } - - @Override - public String toString() { - return "MvAvg[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvFirstBooleanEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvFirstBooleanEvaluator.java deleted file mode 100644 index bf946aab347d2..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvFirstBooleanEvaluator.java +++ /dev/null @@ -1,89 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvFirst}. - * This class is generated. Do not edit it. - */ -public final class MvFirstBooleanEvaluator extends AbstractMultivalueFunction.AbstractEvaluator { - public MvFirstBooleanEvaluator(EvalOperator.ExpressionEvaluator field, - DriverContext driverContext) { - super(driverContext, field); - } - - @Override - public String name() { - return "MvFirst"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - BooleanBlock v = (BooleanBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (BooleanBlock.Builder builder = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - boolean result = MvFirst.process(v, first, end); - builder.appendBoolean(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNotNullable(Block fieldVal) { - BooleanBlock v = (BooleanBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (BooleanVector.FixedBuilder builder = driverContext.blockFactory().newBooleanVectorFixedBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - boolean result = MvFirst.process(v, first, end); - builder.appendBoolean(result); - } - return builder.build().asBlock(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field) { - this.field = field; - } - - @Override - public MvFirstBooleanEvaluator get(DriverContext context) { - return new MvFirstBooleanEvaluator(field.get(context), context); - } - - @Override - public String toString() { - return "MvFirst[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvFirstBytesRefEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvFirstBytesRefEvaluator.java deleted file mode 100644 index e0cb6ca4c289b..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvFirstBytesRefEvaluator.java +++ /dev/null @@ -1,92 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvFirst}. - * This class is generated. Do not edit it. - */ -public final class MvFirstBytesRefEvaluator extends AbstractMultivalueFunction.AbstractEvaluator { - public MvFirstBytesRefEvaluator(EvalOperator.ExpressionEvaluator field, - DriverContext driverContext) { - super(driverContext, field); - } - - @Override - public String name() { - return "MvFirst"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - BytesRefBlock v = (BytesRefBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef valueScratch = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - BytesRef result = MvFirst.process(v, first, end, valueScratch); - builder.appendBytesRef(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNotNullable(Block fieldVal) { - BytesRefBlock v = (BytesRefBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (BytesRefVector.Builder builder = driverContext.blockFactory().newBytesRefVectorBuilder(positionCount)) { - BytesRef valueScratch = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - BytesRef result = MvFirst.process(v, first, end, valueScratch); - builder.appendBytesRef(result); - } - return builder.build().asBlock(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field) { - this.field = field; - } - - @Override - public MvFirstBytesRefEvaluator get(DriverContext context) { - return new MvFirstBytesRefEvaluator(field.get(context), context); - } - - @Override - public String toString() { - return "MvFirst[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvFirstDoubleEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvFirstDoubleEvaluator.java deleted file mode 100644 index 584319cc1ab82..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvFirstDoubleEvaluator.java +++ /dev/null @@ -1,89 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvFirst}. - * This class is generated. Do not edit it. - */ -public final class MvFirstDoubleEvaluator extends AbstractMultivalueFunction.AbstractEvaluator { - public MvFirstDoubleEvaluator(EvalOperator.ExpressionEvaluator field, - DriverContext driverContext) { - super(driverContext, field); - } - - @Override - public String name() { - return "MvFirst"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - DoubleBlock v = (DoubleBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - double result = MvFirst.process(v, first, end); - builder.appendDouble(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNotNullable(Block fieldVal) { - DoubleBlock v = (DoubleBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (DoubleVector.FixedBuilder builder = driverContext.blockFactory().newDoubleVectorFixedBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - double result = MvFirst.process(v, first, end); - builder.appendDouble(result); - } - return builder.build().asBlock(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field) { - this.field = field; - } - - @Override - public MvFirstDoubleEvaluator get(DriverContext context) { - return new MvFirstDoubleEvaluator(field.get(context), context); - } - - @Override - public String toString() { - return "MvFirst[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvFirstIntEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvFirstIntEvaluator.java deleted file mode 100644 index 1e9c50d135559..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvFirstIntEvaluator.java +++ /dev/null @@ -1,88 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvFirst}. - * This class is generated. Do not edit it. - */ -public final class MvFirstIntEvaluator extends AbstractMultivalueFunction.AbstractEvaluator { - public MvFirstIntEvaluator(EvalOperator.ExpressionEvaluator field, DriverContext driverContext) { - super(driverContext, field); - } - - @Override - public String name() { - return "MvFirst"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - IntBlock v = (IntBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (IntBlock.Builder builder = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - int result = MvFirst.process(v, first, end); - builder.appendInt(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNotNullable(Block fieldVal) { - IntBlock v = (IntBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (IntVector.FixedBuilder builder = driverContext.blockFactory().newIntVectorFixedBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - int result = MvFirst.process(v, first, end); - builder.appendInt(result); - } - return builder.build().asBlock(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field) { - this.field = field; - } - - @Override - public MvFirstIntEvaluator get(DriverContext context) { - return new MvFirstIntEvaluator(field.get(context), context); - } - - @Override - public String toString() { - return "MvFirst[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvFirstLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvFirstLongEvaluator.java deleted file mode 100644 index 7e118a1eb9eb8..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvFirstLongEvaluator.java +++ /dev/null @@ -1,88 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvFirst}. - * This class is generated. Do not edit it. - */ -public final class MvFirstLongEvaluator extends AbstractMultivalueFunction.AbstractEvaluator { - public MvFirstLongEvaluator(EvalOperator.ExpressionEvaluator field, DriverContext driverContext) { - super(driverContext, field); - } - - @Override - public String name() { - return "MvFirst"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - LongBlock v = (LongBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - long result = MvFirst.process(v, first, end); - builder.appendLong(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNotNullable(Block fieldVal) { - LongBlock v = (LongBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (LongVector.FixedBuilder builder = driverContext.blockFactory().newLongVectorFixedBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - long result = MvFirst.process(v, first, end); - builder.appendLong(result); - } - return builder.build().asBlock(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field) { - this.field = field; - } - - @Override - public MvFirstLongEvaluator get(DriverContext context) { - return new MvFirstLongEvaluator(field.get(context), context); - } - - @Override - public String toString() { - return "MvFirst[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvLastBooleanEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvLastBooleanEvaluator.java deleted file mode 100644 index 75b49f2c3e8ee..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvLastBooleanEvaluator.java +++ /dev/null @@ -1,89 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvLast}. - * This class is generated. Do not edit it. - */ -public final class MvLastBooleanEvaluator extends AbstractMultivalueFunction.AbstractEvaluator { - public MvLastBooleanEvaluator(EvalOperator.ExpressionEvaluator field, - DriverContext driverContext) { - super(driverContext, field); - } - - @Override - public String name() { - return "MvLast"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - BooleanBlock v = (BooleanBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (BooleanBlock.Builder builder = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - boolean result = MvLast.process(v, first, end); - builder.appendBoolean(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNotNullable(Block fieldVal) { - BooleanBlock v = (BooleanBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (BooleanVector.FixedBuilder builder = driverContext.blockFactory().newBooleanVectorFixedBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - boolean result = MvLast.process(v, first, end); - builder.appendBoolean(result); - } - return builder.build().asBlock(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field) { - this.field = field; - } - - @Override - public MvLastBooleanEvaluator get(DriverContext context) { - return new MvLastBooleanEvaluator(field.get(context), context); - } - - @Override - public String toString() { - return "MvLast[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvLastBytesRefEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvLastBytesRefEvaluator.java deleted file mode 100644 index fcca356b38576..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvLastBytesRefEvaluator.java +++ /dev/null @@ -1,92 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvLast}. - * This class is generated. Do not edit it. - */ -public final class MvLastBytesRefEvaluator extends AbstractMultivalueFunction.AbstractEvaluator { - public MvLastBytesRefEvaluator(EvalOperator.ExpressionEvaluator field, - DriverContext driverContext) { - super(driverContext, field); - } - - @Override - public String name() { - return "MvLast"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - BytesRefBlock v = (BytesRefBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef valueScratch = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - BytesRef result = MvLast.process(v, first, end, valueScratch); - builder.appendBytesRef(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNotNullable(Block fieldVal) { - BytesRefBlock v = (BytesRefBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (BytesRefVector.Builder builder = driverContext.blockFactory().newBytesRefVectorBuilder(positionCount)) { - BytesRef valueScratch = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - BytesRef result = MvLast.process(v, first, end, valueScratch); - builder.appendBytesRef(result); - } - return builder.build().asBlock(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field) { - this.field = field; - } - - @Override - public MvLastBytesRefEvaluator get(DriverContext context) { - return new MvLastBytesRefEvaluator(field.get(context), context); - } - - @Override - public String toString() { - return "MvLast[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvLastDoubleEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvLastDoubleEvaluator.java deleted file mode 100644 index b0cf7bf59900b..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvLastDoubleEvaluator.java +++ /dev/null @@ -1,89 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvLast}. - * This class is generated. Do not edit it. - */ -public final class MvLastDoubleEvaluator extends AbstractMultivalueFunction.AbstractEvaluator { - public MvLastDoubleEvaluator(EvalOperator.ExpressionEvaluator field, - DriverContext driverContext) { - super(driverContext, field); - } - - @Override - public String name() { - return "MvLast"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - DoubleBlock v = (DoubleBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - double result = MvLast.process(v, first, end); - builder.appendDouble(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNotNullable(Block fieldVal) { - DoubleBlock v = (DoubleBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (DoubleVector.FixedBuilder builder = driverContext.blockFactory().newDoubleVectorFixedBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - double result = MvLast.process(v, first, end); - builder.appendDouble(result); - } - return builder.build().asBlock(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field) { - this.field = field; - } - - @Override - public MvLastDoubleEvaluator get(DriverContext context) { - return new MvLastDoubleEvaluator(field.get(context), context); - } - - @Override - public String toString() { - return "MvLast[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvLastIntEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvLastIntEvaluator.java deleted file mode 100644 index 5c2af9218308d..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvLastIntEvaluator.java +++ /dev/null @@ -1,88 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvLast}. - * This class is generated. Do not edit it. - */ -public final class MvLastIntEvaluator extends AbstractMultivalueFunction.AbstractEvaluator { - public MvLastIntEvaluator(EvalOperator.ExpressionEvaluator field, DriverContext driverContext) { - super(driverContext, field); - } - - @Override - public String name() { - return "MvLast"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - IntBlock v = (IntBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (IntBlock.Builder builder = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - int result = MvLast.process(v, first, end); - builder.appendInt(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNotNullable(Block fieldVal) { - IntBlock v = (IntBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (IntVector.FixedBuilder builder = driverContext.blockFactory().newIntVectorFixedBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - int result = MvLast.process(v, first, end); - builder.appendInt(result); - } - return builder.build().asBlock(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field) { - this.field = field; - } - - @Override - public MvLastIntEvaluator get(DriverContext context) { - return new MvLastIntEvaluator(field.get(context), context); - } - - @Override - public String toString() { - return "MvLast[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvLastLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvLastLongEvaluator.java deleted file mode 100644 index 37b95378f1f5b..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvLastLongEvaluator.java +++ /dev/null @@ -1,88 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvLast}. - * This class is generated. Do not edit it. - */ -public final class MvLastLongEvaluator extends AbstractMultivalueFunction.AbstractEvaluator { - public MvLastLongEvaluator(EvalOperator.ExpressionEvaluator field, DriverContext driverContext) { - super(driverContext, field); - } - - @Override - public String name() { - return "MvLast"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - LongBlock v = (LongBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - long result = MvLast.process(v, first, end); - builder.appendLong(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNotNullable(Block fieldVal) { - LongBlock v = (LongBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (LongVector.FixedBuilder builder = driverContext.blockFactory().newLongVectorFixedBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - long result = MvLast.process(v, first, end); - builder.appendLong(result); - } - return builder.build().asBlock(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field) { - this.field = field; - } - - @Override - public MvLastLongEvaluator get(DriverContext context) { - return new MvLastLongEvaluator(field.get(context), context); - } - - @Override - public String toString() { - return "MvLast[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMaxBooleanEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMaxBooleanEvaluator.java deleted file mode 100644 index 44b4432edbf6f..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMaxBooleanEvaluator.java +++ /dev/null @@ -1,145 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvMax}. - * This class is generated. Do not edit it. - */ -public final class MvMaxBooleanEvaluator extends AbstractMultivalueFunction.AbstractEvaluator { - public MvMaxBooleanEvaluator(EvalOperator.ExpressionEvaluator field, - DriverContext driverContext) { - super(driverContext, field); - } - - @Override - public String name() { - return "MvMax"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - if (fieldVal.mvSortedAscending()) { - return evalAscendingNullable(fieldVal); - } - BooleanBlock v = (BooleanBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (BooleanBlock.Builder builder = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - boolean value = v.getBoolean(first); - for (int i = first + 1; i < end; i++) { - boolean next = v.getBoolean(i); - value = MvMax.process(value, next); - } - boolean result = value; - builder.appendBoolean(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNotNullable(Block fieldVal) { - if (fieldVal.mvSortedAscending()) { - return evalAscendingNotNullable(fieldVal); - } - BooleanBlock v = (BooleanBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (BooleanVector.FixedBuilder builder = driverContext.blockFactory().newBooleanVectorFixedBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - boolean value = v.getBoolean(first); - for (int i = first + 1; i < end; i++) { - boolean next = v.getBoolean(i); - value = MvMax.process(value, next); - } - boolean result = value; - builder.appendBoolean(result); - } - return builder.build().asBlock(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field and all multivalued fields are in ascending order. - */ - private Block evalAscendingNullable(Block fieldVal) { - BooleanBlock v = (BooleanBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (BooleanBlock.Builder builder = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int idx = MvMax.ascendingIndex(valueCount); - boolean result = v.getBoolean(first + idx); - builder.appendBoolean(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field and all multivalued fields are in ascending order. - */ - private Block evalAscendingNotNullable(Block fieldVal) { - BooleanBlock v = (BooleanBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (BooleanVector.FixedBuilder builder = driverContext.blockFactory().newBooleanVectorFixedBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int idx = MvMax.ascendingIndex(valueCount); - boolean result = v.getBoolean(first + idx); - builder.appendBoolean(result); - } - return builder.build().asBlock(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field) { - this.field = field; - } - - @Override - public MvMaxBooleanEvaluator get(DriverContext context) { - return new MvMaxBooleanEvaluator(field.get(context), context); - } - - @Override - public String toString() { - return "MvMax[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMaxBytesRefEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMaxBytesRefEvaluator.java deleted file mode 100644 index 4e2180f2ec467..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMaxBytesRefEvaluator.java +++ /dev/null @@ -1,154 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvMax}. - * This class is generated. Do not edit it. - */ -public final class MvMaxBytesRefEvaluator extends AbstractMultivalueFunction.AbstractEvaluator { - public MvMaxBytesRefEvaluator(EvalOperator.ExpressionEvaluator field, - DriverContext driverContext) { - super(driverContext, field); - } - - @Override - public String name() { - return "MvMax"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - if (fieldVal.mvSortedAscending()) { - return evalAscendingNullable(fieldVal); - } - BytesRefBlock v = (BytesRefBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef firstScratch = new BytesRef(); - BytesRef nextScratch = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - BytesRef value = v.getBytesRef(first, firstScratch); - for (int i = first + 1; i < end; i++) { - BytesRef next = v.getBytesRef(i, nextScratch); - MvMax.process(value, next); - } - BytesRef result = value; - builder.appendBytesRef(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNotNullable(Block fieldVal) { - if (fieldVal.mvSortedAscending()) { - return evalAscendingNotNullable(fieldVal); - } - BytesRefBlock v = (BytesRefBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (BytesRefVector.Builder builder = driverContext.blockFactory().newBytesRefVectorBuilder(positionCount)) { - BytesRef firstScratch = new BytesRef(); - BytesRef nextScratch = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - BytesRef value = v.getBytesRef(first, firstScratch); - for (int i = first + 1; i < end; i++) { - BytesRef next = v.getBytesRef(i, nextScratch); - MvMax.process(value, next); - } - BytesRef result = value; - builder.appendBytesRef(result); - } - return builder.build().asBlock(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field and all multivalued fields are in ascending order. - */ - private Block evalAscendingNullable(Block fieldVal) { - BytesRefBlock v = (BytesRefBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef firstScratch = new BytesRef(); - BytesRef nextScratch = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int idx = MvMax.ascendingIndex(valueCount); - BytesRef result = v.getBytesRef(first + idx, firstScratch); - builder.appendBytesRef(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field and all multivalued fields are in ascending order. - */ - private Block evalAscendingNotNullable(Block fieldVal) { - BytesRefBlock v = (BytesRefBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (BytesRefVector.Builder builder = driverContext.blockFactory().newBytesRefVectorBuilder(positionCount)) { - BytesRef firstScratch = new BytesRef(); - BytesRef nextScratch = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int idx = MvMax.ascendingIndex(valueCount); - BytesRef result = v.getBytesRef(first + idx, firstScratch); - builder.appendBytesRef(result); - } - return builder.build().asBlock(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field) { - this.field = field; - } - - @Override - public MvMaxBytesRefEvaluator get(DriverContext context) { - return new MvMaxBytesRefEvaluator(field.get(context), context); - } - - @Override - public String toString() { - return "MvMax[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMaxDoubleEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMaxDoubleEvaluator.java deleted file mode 100644 index 4a1be1673bb7c..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMaxDoubleEvaluator.java +++ /dev/null @@ -1,144 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvMax}. - * This class is generated. Do not edit it. - */ -public final class MvMaxDoubleEvaluator extends AbstractMultivalueFunction.AbstractEvaluator { - public MvMaxDoubleEvaluator(EvalOperator.ExpressionEvaluator field, DriverContext driverContext) { - super(driverContext, field); - } - - @Override - public String name() { - return "MvMax"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - if (fieldVal.mvSortedAscending()) { - return evalAscendingNullable(fieldVal); - } - DoubleBlock v = (DoubleBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - double value = v.getDouble(first); - for (int i = first + 1; i < end; i++) { - double next = v.getDouble(i); - value = MvMax.process(value, next); - } - double result = value; - builder.appendDouble(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNotNullable(Block fieldVal) { - if (fieldVal.mvSortedAscending()) { - return evalAscendingNotNullable(fieldVal); - } - DoubleBlock v = (DoubleBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (DoubleVector.FixedBuilder builder = driverContext.blockFactory().newDoubleVectorFixedBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - double value = v.getDouble(first); - for (int i = first + 1; i < end; i++) { - double next = v.getDouble(i); - value = MvMax.process(value, next); - } - double result = value; - builder.appendDouble(result); - } - return builder.build().asBlock(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field and all multivalued fields are in ascending order. - */ - private Block evalAscendingNullable(Block fieldVal) { - DoubleBlock v = (DoubleBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int idx = MvMax.ascendingIndex(valueCount); - double result = v.getDouble(first + idx); - builder.appendDouble(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field and all multivalued fields are in ascending order. - */ - private Block evalAscendingNotNullable(Block fieldVal) { - DoubleBlock v = (DoubleBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (DoubleVector.FixedBuilder builder = driverContext.blockFactory().newDoubleVectorFixedBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int idx = MvMax.ascendingIndex(valueCount); - double result = v.getDouble(first + idx); - builder.appendDouble(result); - } - return builder.build().asBlock(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field) { - this.field = field; - } - - @Override - public MvMaxDoubleEvaluator get(DriverContext context) { - return new MvMaxDoubleEvaluator(field.get(context), context); - } - - @Override - public String toString() { - return "MvMax[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMaxIntEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMaxIntEvaluator.java deleted file mode 100644 index b0ed499efd84f..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMaxIntEvaluator.java +++ /dev/null @@ -1,144 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvMax}. - * This class is generated. Do not edit it. - */ -public final class MvMaxIntEvaluator extends AbstractMultivalueFunction.AbstractEvaluator { - public MvMaxIntEvaluator(EvalOperator.ExpressionEvaluator field, DriverContext driverContext) { - super(driverContext, field); - } - - @Override - public String name() { - return "MvMax"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - if (fieldVal.mvSortedAscending()) { - return evalAscendingNullable(fieldVal); - } - IntBlock v = (IntBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (IntBlock.Builder builder = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - int value = v.getInt(first); - for (int i = first + 1; i < end; i++) { - int next = v.getInt(i); - value = MvMax.process(value, next); - } - int result = value; - builder.appendInt(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNotNullable(Block fieldVal) { - if (fieldVal.mvSortedAscending()) { - return evalAscendingNotNullable(fieldVal); - } - IntBlock v = (IntBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (IntVector.FixedBuilder builder = driverContext.blockFactory().newIntVectorFixedBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - int value = v.getInt(first); - for (int i = first + 1; i < end; i++) { - int next = v.getInt(i); - value = MvMax.process(value, next); - } - int result = value; - builder.appendInt(result); - } - return builder.build().asBlock(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field and all multivalued fields are in ascending order. - */ - private Block evalAscendingNullable(Block fieldVal) { - IntBlock v = (IntBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (IntBlock.Builder builder = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int idx = MvMax.ascendingIndex(valueCount); - int result = v.getInt(first + idx); - builder.appendInt(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field and all multivalued fields are in ascending order. - */ - private Block evalAscendingNotNullable(Block fieldVal) { - IntBlock v = (IntBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (IntVector.FixedBuilder builder = driverContext.blockFactory().newIntVectorFixedBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int idx = MvMax.ascendingIndex(valueCount); - int result = v.getInt(first + idx); - builder.appendInt(result); - } - return builder.build().asBlock(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field) { - this.field = field; - } - - @Override - public MvMaxIntEvaluator get(DriverContext context) { - return new MvMaxIntEvaluator(field.get(context), context); - } - - @Override - public String toString() { - return "MvMax[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMaxLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMaxLongEvaluator.java deleted file mode 100644 index 24397b64c9ccc..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMaxLongEvaluator.java +++ /dev/null @@ -1,144 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvMax}. - * This class is generated. Do not edit it. - */ -public final class MvMaxLongEvaluator extends AbstractMultivalueFunction.AbstractEvaluator { - public MvMaxLongEvaluator(EvalOperator.ExpressionEvaluator field, DriverContext driverContext) { - super(driverContext, field); - } - - @Override - public String name() { - return "MvMax"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - if (fieldVal.mvSortedAscending()) { - return evalAscendingNullable(fieldVal); - } - LongBlock v = (LongBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - long value = v.getLong(first); - for (int i = first + 1; i < end; i++) { - long next = v.getLong(i); - value = MvMax.process(value, next); - } - long result = value; - builder.appendLong(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNotNullable(Block fieldVal) { - if (fieldVal.mvSortedAscending()) { - return evalAscendingNotNullable(fieldVal); - } - LongBlock v = (LongBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (LongVector.FixedBuilder builder = driverContext.blockFactory().newLongVectorFixedBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - long value = v.getLong(first); - for (int i = first + 1; i < end; i++) { - long next = v.getLong(i); - value = MvMax.process(value, next); - } - long result = value; - builder.appendLong(result); - } - return builder.build().asBlock(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field and all multivalued fields are in ascending order. - */ - private Block evalAscendingNullable(Block fieldVal) { - LongBlock v = (LongBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int idx = MvMax.ascendingIndex(valueCount); - long result = v.getLong(first + idx); - builder.appendLong(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field and all multivalued fields are in ascending order. - */ - private Block evalAscendingNotNullable(Block fieldVal) { - LongBlock v = (LongBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (LongVector.FixedBuilder builder = driverContext.blockFactory().newLongVectorFixedBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int idx = MvMax.ascendingIndex(valueCount); - long result = v.getLong(first + idx); - builder.appendLong(result); - } - return builder.build().asBlock(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field) { - this.field = field; - } - - @Override - public MvMaxLongEvaluator get(DriverContext context) { - return new MvMaxLongEvaluator(field.get(context), context); - } - - @Override - public String toString() { - return "MvMax[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMedianDoubleEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMedianDoubleEvaluator.java deleted file mode 100644 index c3ea505a29e88..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMedianDoubleEvaluator.java +++ /dev/null @@ -1,99 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvMedian}. - * This class is generated. Do not edit it. - */ -public final class MvMedianDoubleEvaluator extends AbstractMultivalueFunction.AbstractEvaluator { - public MvMedianDoubleEvaluator(EvalOperator.ExpressionEvaluator field, - DriverContext driverContext) { - super(driverContext, field); - } - - @Override - public String name() { - return "MvMedian"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - DoubleBlock v = (DoubleBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - MvMedian.Doubles work = new MvMedian.Doubles(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - for (int i = first; i < end; i++) { - double value = v.getDouble(i); - MvMedian.process(work, value); - } - double result = MvMedian.finish(work); - builder.appendDouble(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNotNullable(Block fieldVal) { - DoubleBlock v = (DoubleBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (DoubleVector.FixedBuilder builder = driverContext.blockFactory().newDoubleVectorFixedBuilder(positionCount)) { - MvMedian.Doubles work = new MvMedian.Doubles(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - for (int i = first; i < end; i++) { - double value = v.getDouble(i); - MvMedian.process(work, value); - } - double result = MvMedian.finish(work); - builder.appendDouble(result); - } - return builder.build().asBlock(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field) { - this.field = field; - } - - @Override - public MvMedianDoubleEvaluator get(DriverContext context) { - return new MvMedianDoubleEvaluator(field.get(context), context); - } - - @Override - public String toString() { - return "MvMedian[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMedianIntEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMedianIntEvaluator.java deleted file mode 100644 index d07dc41e1d04b..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMedianIntEvaluator.java +++ /dev/null @@ -1,144 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvMedian}. - * This class is generated. Do not edit it. - */ -public final class MvMedianIntEvaluator extends AbstractMultivalueFunction.AbstractEvaluator { - public MvMedianIntEvaluator(EvalOperator.ExpressionEvaluator field, DriverContext driverContext) { - super(driverContext, field); - } - - @Override - public String name() { - return "MvMedian"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - if (fieldVal.mvSortedAscending()) { - return evalAscendingNullable(fieldVal); - } - IntBlock v = (IntBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (IntBlock.Builder builder = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - MvMedian.Ints work = new MvMedian.Ints(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - for (int i = first; i < end; i++) { - int value = v.getInt(i); - MvMedian.process(work, value); - } - int result = MvMedian.finish(work); - builder.appendInt(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNotNullable(Block fieldVal) { - if (fieldVal.mvSortedAscending()) { - return evalAscendingNotNullable(fieldVal); - } - IntBlock v = (IntBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (IntVector.FixedBuilder builder = driverContext.blockFactory().newIntVectorFixedBuilder(positionCount)) { - MvMedian.Ints work = new MvMedian.Ints(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - for (int i = first; i < end; i++) { - int value = v.getInt(i); - MvMedian.process(work, value); - } - int result = MvMedian.finish(work); - builder.appendInt(result); - } - return builder.build().asBlock(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field and all multivalued fields are in ascending order. - */ - private Block evalAscendingNullable(Block fieldVal) { - IntBlock v = (IntBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (IntBlock.Builder builder = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - MvMedian.Ints work = new MvMedian.Ints(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int result = MvMedian.ascending(v, first, valueCount); - builder.appendInt(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field and all multivalued fields are in ascending order. - */ - private Block evalAscendingNotNullable(Block fieldVal) { - IntBlock v = (IntBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (IntVector.FixedBuilder builder = driverContext.blockFactory().newIntVectorFixedBuilder(positionCount)) { - MvMedian.Ints work = new MvMedian.Ints(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int result = MvMedian.ascending(v, first, valueCount); - builder.appendInt(result); - } - return builder.build().asBlock(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field) { - this.field = field; - } - - @Override - public MvMedianIntEvaluator get(DriverContext context) { - return new MvMedianIntEvaluator(field.get(context), context); - } - - @Override - public String toString() { - return "MvMedian[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMedianLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMedianLongEvaluator.java deleted file mode 100644 index f1cd87aefd3d0..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMedianLongEvaluator.java +++ /dev/null @@ -1,145 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvMedian}. - * This class is generated. Do not edit it. - */ -public final class MvMedianLongEvaluator extends AbstractMultivalueFunction.AbstractEvaluator { - public MvMedianLongEvaluator(EvalOperator.ExpressionEvaluator field, - DriverContext driverContext) { - super(driverContext, field); - } - - @Override - public String name() { - return "MvMedian"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - if (fieldVal.mvSortedAscending()) { - return evalAscendingNullable(fieldVal); - } - LongBlock v = (LongBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - MvMedian.Longs work = new MvMedian.Longs(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - for (int i = first; i < end; i++) { - long value = v.getLong(i); - MvMedian.process(work, value); - } - long result = MvMedian.finish(work); - builder.appendLong(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNotNullable(Block fieldVal) { - if (fieldVal.mvSortedAscending()) { - return evalAscendingNotNullable(fieldVal); - } - LongBlock v = (LongBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (LongVector.FixedBuilder builder = driverContext.blockFactory().newLongVectorFixedBuilder(positionCount)) { - MvMedian.Longs work = new MvMedian.Longs(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - for (int i = first; i < end; i++) { - long value = v.getLong(i); - MvMedian.process(work, value); - } - long result = MvMedian.finish(work); - builder.appendLong(result); - } - return builder.build().asBlock(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field and all multivalued fields are in ascending order. - */ - private Block evalAscendingNullable(Block fieldVal) { - LongBlock v = (LongBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - MvMedian.Longs work = new MvMedian.Longs(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - long result = MvMedian.ascending(v, first, valueCount); - builder.appendLong(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field and all multivalued fields are in ascending order. - */ - private Block evalAscendingNotNullable(Block fieldVal) { - LongBlock v = (LongBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (LongVector.FixedBuilder builder = driverContext.blockFactory().newLongVectorFixedBuilder(positionCount)) { - MvMedian.Longs work = new MvMedian.Longs(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - long result = MvMedian.ascending(v, first, valueCount); - builder.appendLong(result); - } - return builder.build().asBlock(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field) { - this.field = field; - } - - @Override - public MvMedianLongEvaluator get(DriverContext context) { - return new MvMedianLongEvaluator(field.get(context), context); - } - - @Override - public String toString() { - return "MvMedian[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMedianUnsignedLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMedianUnsignedLongEvaluator.java deleted file mode 100644 index 031280b767b41..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMedianUnsignedLongEvaluator.java +++ /dev/null @@ -1,145 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvMedian}. - * This class is generated. Do not edit it. - */ -public final class MvMedianUnsignedLongEvaluator extends AbstractMultivalueFunction.AbstractEvaluator { - public MvMedianUnsignedLongEvaluator(EvalOperator.ExpressionEvaluator field, - DriverContext driverContext) { - super(driverContext, field); - } - - @Override - public String name() { - return "MvMedian"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - if (fieldVal.mvSortedAscending()) { - return evalAscendingNullable(fieldVal); - } - LongBlock v = (LongBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - MvMedian.Longs work = new MvMedian.Longs(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - for (int i = first; i < end; i++) { - long value = v.getLong(i); - MvMedian.processUnsignedLong(work, value); - } - long result = MvMedian.finishUnsignedLong(work); - builder.appendLong(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNotNullable(Block fieldVal) { - if (fieldVal.mvSortedAscending()) { - return evalAscendingNotNullable(fieldVal); - } - LongBlock v = (LongBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (LongVector.FixedBuilder builder = driverContext.blockFactory().newLongVectorFixedBuilder(positionCount)) { - MvMedian.Longs work = new MvMedian.Longs(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - for (int i = first; i < end; i++) { - long value = v.getLong(i); - MvMedian.processUnsignedLong(work, value); - } - long result = MvMedian.finishUnsignedLong(work); - builder.appendLong(result); - } - return builder.build().asBlock(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field and all multivalued fields are in ascending order. - */ - private Block evalAscendingNullable(Block fieldVal) { - LongBlock v = (LongBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - MvMedian.Longs work = new MvMedian.Longs(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - long result = MvMedian.ascendingUnsignedLong(v, first, valueCount); - builder.appendLong(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field and all multivalued fields are in ascending order. - */ - private Block evalAscendingNotNullable(Block fieldVal) { - LongBlock v = (LongBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (LongVector.FixedBuilder builder = driverContext.blockFactory().newLongVectorFixedBuilder(positionCount)) { - MvMedian.Longs work = new MvMedian.Longs(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - long result = MvMedian.ascendingUnsignedLong(v, first, valueCount); - builder.appendLong(result); - } - return builder.build().asBlock(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field) { - this.field = field; - } - - @Override - public MvMedianUnsignedLongEvaluator get(DriverContext context) { - return new MvMedianUnsignedLongEvaluator(field.get(context), context); - } - - @Override - public String toString() { - return "MvMedian[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMinBooleanEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMinBooleanEvaluator.java deleted file mode 100644 index ea8b04cc5c4a5..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMinBooleanEvaluator.java +++ /dev/null @@ -1,145 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvMin}. - * This class is generated. Do not edit it. - */ -public final class MvMinBooleanEvaluator extends AbstractMultivalueFunction.AbstractEvaluator { - public MvMinBooleanEvaluator(EvalOperator.ExpressionEvaluator field, - DriverContext driverContext) { - super(driverContext, field); - } - - @Override - public String name() { - return "MvMin"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - if (fieldVal.mvSortedAscending()) { - return evalAscendingNullable(fieldVal); - } - BooleanBlock v = (BooleanBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (BooleanBlock.Builder builder = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - boolean value = v.getBoolean(first); - for (int i = first + 1; i < end; i++) { - boolean next = v.getBoolean(i); - value = MvMin.process(value, next); - } - boolean result = value; - builder.appendBoolean(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNotNullable(Block fieldVal) { - if (fieldVal.mvSortedAscending()) { - return evalAscendingNotNullable(fieldVal); - } - BooleanBlock v = (BooleanBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (BooleanVector.FixedBuilder builder = driverContext.blockFactory().newBooleanVectorFixedBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - boolean value = v.getBoolean(first); - for (int i = first + 1; i < end; i++) { - boolean next = v.getBoolean(i); - value = MvMin.process(value, next); - } - boolean result = value; - builder.appendBoolean(result); - } - return builder.build().asBlock(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field and all multivalued fields are in ascending order. - */ - private Block evalAscendingNullable(Block fieldVal) { - BooleanBlock v = (BooleanBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (BooleanBlock.Builder builder = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int idx = MvMin.ascendingIndex(valueCount); - boolean result = v.getBoolean(first + idx); - builder.appendBoolean(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field and all multivalued fields are in ascending order. - */ - private Block evalAscendingNotNullable(Block fieldVal) { - BooleanBlock v = (BooleanBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (BooleanVector.FixedBuilder builder = driverContext.blockFactory().newBooleanVectorFixedBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int idx = MvMin.ascendingIndex(valueCount); - boolean result = v.getBoolean(first + idx); - builder.appendBoolean(result); - } - return builder.build().asBlock(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field) { - this.field = field; - } - - @Override - public MvMinBooleanEvaluator get(DriverContext context) { - return new MvMinBooleanEvaluator(field.get(context), context); - } - - @Override - public String toString() { - return "MvMin[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMinBytesRefEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMinBytesRefEvaluator.java deleted file mode 100644 index 31d41ff61e196..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMinBytesRefEvaluator.java +++ /dev/null @@ -1,154 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvMin}. - * This class is generated. Do not edit it. - */ -public final class MvMinBytesRefEvaluator extends AbstractMultivalueFunction.AbstractEvaluator { - public MvMinBytesRefEvaluator(EvalOperator.ExpressionEvaluator field, - DriverContext driverContext) { - super(driverContext, field); - } - - @Override - public String name() { - return "MvMin"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - if (fieldVal.mvSortedAscending()) { - return evalAscendingNullable(fieldVal); - } - BytesRefBlock v = (BytesRefBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef firstScratch = new BytesRef(); - BytesRef nextScratch = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - BytesRef value = v.getBytesRef(first, firstScratch); - for (int i = first + 1; i < end; i++) { - BytesRef next = v.getBytesRef(i, nextScratch); - MvMin.process(value, next); - } - BytesRef result = value; - builder.appendBytesRef(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNotNullable(Block fieldVal) { - if (fieldVal.mvSortedAscending()) { - return evalAscendingNotNullable(fieldVal); - } - BytesRefBlock v = (BytesRefBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (BytesRefVector.Builder builder = driverContext.blockFactory().newBytesRefVectorBuilder(positionCount)) { - BytesRef firstScratch = new BytesRef(); - BytesRef nextScratch = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - BytesRef value = v.getBytesRef(first, firstScratch); - for (int i = first + 1; i < end; i++) { - BytesRef next = v.getBytesRef(i, nextScratch); - MvMin.process(value, next); - } - BytesRef result = value; - builder.appendBytesRef(result); - } - return builder.build().asBlock(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field and all multivalued fields are in ascending order. - */ - private Block evalAscendingNullable(Block fieldVal) { - BytesRefBlock v = (BytesRefBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (BytesRefBlock.Builder builder = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef firstScratch = new BytesRef(); - BytesRef nextScratch = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int idx = MvMin.ascendingIndex(valueCount); - BytesRef result = v.getBytesRef(first + idx, firstScratch); - builder.appendBytesRef(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field and all multivalued fields are in ascending order. - */ - private Block evalAscendingNotNullable(Block fieldVal) { - BytesRefBlock v = (BytesRefBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (BytesRefVector.Builder builder = driverContext.blockFactory().newBytesRefVectorBuilder(positionCount)) { - BytesRef firstScratch = new BytesRef(); - BytesRef nextScratch = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int idx = MvMin.ascendingIndex(valueCount); - BytesRef result = v.getBytesRef(first + idx, firstScratch); - builder.appendBytesRef(result); - } - return builder.build().asBlock(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field) { - this.field = field; - } - - @Override - public MvMinBytesRefEvaluator get(DriverContext context) { - return new MvMinBytesRefEvaluator(field.get(context), context); - } - - @Override - public String toString() { - return "MvMin[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMinDoubleEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMinDoubleEvaluator.java deleted file mode 100644 index 5390350751ee7..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMinDoubleEvaluator.java +++ /dev/null @@ -1,144 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvMin}. - * This class is generated. Do not edit it. - */ -public final class MvMinDoubleEvaluator extends AbstractMultivalueFunction.AbstractEvaluator { - public MvMinDoubleEvaluator(EvalOperator.ExpressionEvaluator field, DriverContext driverContext) { - super(driverContext, field); - } - - @Override - public String name() { - return "MvMin"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - if (fieldVal.mvSortedAscending()) { - return evalAscendingNullable(fieldVal); - } - DoubleBlock v = (DoubleBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - double value = v.getDouble(first); - for (int i = first + 1; i < end; i++) { - double next = v.getDouble(i); - value = MvMin.process(value, next); - } - double result = value; - builder.appendDouble(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNotNullable(Block fieldVal) { - if (fieldVal.mvSortedAscending()) { - return evalAscendingNotNullable(fieldVal); - } - DoubleBlock v = (DoubleBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (DoubleVector.FixedBuilder builder = driverContext.blockFactory().newDoubleVectorFixedBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - double value = v.getDouble(first); - for (int i = first + 1; i < end; i++) { - double next = v.getDouble(i); - value = MvMin.process(value, next); - } - double result = value; - builder.appendDouble(result); - } - return builder.build().asBlock(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field and all multivalued fields are in ascending order. - */ - private Block evalAscendingNullable(Block fieldVal) { - DoubleBlock v = (DoubleBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int idx = MvMin.ascendingIndex(valueCount); - double result = v.getDouble(first + idx); - builder.appendDouble(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field and all multivalued fields are in ascending order. - */ - private Block evalAscendingNotNullable(Block fieldVal) { - DoubleBlock v = (DoubleBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (DoubleVector.FixedBuilder builder = driverContext.blockFactory().newDoubleVectorFixedBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int idx = MvMin.ascendingIndex(valueCount); - double result = v.getDouble(first + idx); - builder.appendDouble(result); - } - return builder.build().asBlock(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field) { - this.field = field; - } - - @Override - public MvMinDoubleEvaluator get(DriverContext context) { - return new MvMinDoubleEvaluator(field.get(context), context); - } - - @Override - public String toString() { - return "MvMin[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMinIntEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMinIntEvaluator.java deleted file mode 100644 index 918b049780905..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMinIntEvaluator.java +++ /dev/null @@ -1,144 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvMin}. - * This class is generated. Do not edit it. - */ -public final class MvMinIntEvaluator extends AbstractMultivalueFunction.AbstractEvaluator { - public MvMinIntEvaluator(EvalOperator.ExpressionEvaluator field, DriverContext driverContext) { - super(driverContext, field); - } - - @Override - public String name() { - return "MvMin"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - if (fieldVal.mvSortedAscending()) { - return evalAscendingNullable(fieldVal); - } - IntBlock v = (IntBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (IntBlock.Builder builder = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - int value = v.getInt(first); - for (int i = first + 1; i < end; i++) { - int next = v.getInt(i); - value = MvMin.process(value, next); - } - int result = value; - builder.appendInt(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNotNullable(Block fieldVal) { - if (fieldVal.mvSortedAscending()) { - return evalAscendingNotNullable(fieldVal); - } - IntBlock v = (IntBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (IntVector.FixedBuilder builder = driverContext.blockFactory().newIntVectorFixedBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - int value = v.getInt(first); - for (int i = first + 1; i < end; i++) { - int next = v.getInt(i); - value = MvMin.process(value, next); - } - int result = value; - builder.appendInt(result); - } - return builder.build().asBlock(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field and all multivalued fields are in ascending order. - */ - private Block evalAscendingNullable(Block fieldVal) { - IntBlock v = (IntBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (IntBlock.Builder builder = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int idx = MvMin.ascendingIndex(valueCount); - int result = v.getInt(first + idx); - builder.appendInt(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field and all multivalued fields are in ascending order. - */ - private Block evalAscendingNotNullable(Block fieldVal) { - IntBlock v = (IntBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (IntVector.FixedBuilder builder = driverContext.blockFactory().newIntVectorFixedBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int idx = MvMin.ascendingIndex(valueCount); - int result = v.getInt(first + idx); - builder.appendInt(result); - } - return builder.build().asBlock(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field) { - this.field = field; - } - - @Override - public MvMinIntEvaluator get(DriverContext context) { - return new MvMinIntEvaluator(field.get(context), context); - } - - @Override - public String toString() { - return "MvMin[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMinLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMinLongEvaluator.java deleted file mode 100644 index 37a6709d46d4d..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvMinLongEvaluator.java +++ /dev/null @@ -1,144 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvMin}. - * This class is generated. Do not edit it. - */ -public final class MvMinLongEvaluator extends AbstractMultivalueFunction.AbstractEvaluator { - public MvMinLongEvaluator(EvalOperator.ExpressionEvaluator field, DriverContext driverContext) { - super(driverContext, field); - } - - @Override - public String name() { - return "MvMin"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - if (fieldVal.mvSortedAscending()) { - return evalAscendingNullable(fieldVal); - } - LongBlock v = (LongBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - long value = v.getLong(first); - for (int i = first + 1; i < end; i++) { - long next = v.getLong(i); - value = MvMin.process(value, next); - } - long result = value; - builder.appendLong(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNotNullable(Block fieldVal) { - if (fieldVal.mvSortedAscending()) { - return evalAscendingNotNullable(fieldVal); - } - LongBlock v = (LongBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (LongVector.FixedBuilder builder = driverContext.blockFactory().newLongVectorFixedBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - long value = v.getLong(first); - for (int i = first + 1; i < end; i++) { - long next = v.getLong(i); - value = MvMin.process(value, next); - } - long result = value; - builder.appendLong(result); - } - return builder.build().asBlock(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field and all multivalued fields are in ascending order. - */ - private Block evalAscendingNullable(Block fieldVal) { - LongBlock v = (LongBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int idx = MvMin.ascendingIndex(valueCount); - long result = v.getLong(first + idx); - builder.appendLong(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field and all multivalued fields are in ascending order. - */ - private Block evalAscendingNotNullable(Block fieldVal) { - LongBlock v = (LongBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (LongVector.FixedBuilder builder = driverContext.blockFactory().newLongVectorFixedBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int idx = MvMin.ascendingIndex(valueCount); - long result = v.getLong(first + idx); - builder.appendLong(result); - } - return builder.build().asBlock(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field) { - this.field = field; - } - - @Override - public MvMinLongEvaluator get(DriverContext context) { - return new MvMinLongEvaluator(field.get(context), context); - } - - @Override - public String toString() { - return "MvMin[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvSliceBooleanEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvSliceBooleanEvaluator.java deleted file mode 100644 index 6c4174bd9cca9..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvSliceBooleanEvaluator.java +++ /dev/null @@ -1,140 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.InvalidArgumentException; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvSlice}. - * This class is generated. Do not edit it. - */ -public final class MvSliceBooleanEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator field; - - private final EvalOperator.ExpressionEvaluator start; - - private final EvalOperator.ExpressionEvaluator end; - - private final DriverContext driverContext; - - public MvSliceBooleanEvaluator(Source source, EvalOperator.ExpressionEvaluator field, - EvalOperator.ExpressionEvaluator start, EvalOperator.ExpressionEvaluator end, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.field = field; - this.start = start; - this.end = end; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BooleanBlock fieldBlock = (BooleanBlock) field.eval(page)) { - try (IntBlock startBlock = (IntBlock) start.eval(page)) { - try (IntBlock endBlock = (IntBlock) end.eval(page)) { - return eval(page.getPositionCount(), fieldBlock, startBlock, endBlock); - } - } - } - } - - public BooleanBlock eval(int positionCount, BooleanBlock fieldBlock, IntBlock startBlock, - IntBlock endBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - boolean allBlocksAreNulls = true; - if (!fieldBlock.isNull(p)) { - allBlocksAreNulls = false; - } - if (startBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (startBlock.getValueCount(p) != 1) { - if (startBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (endBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (endBlock.getValueCount(p) != 1) { - if (endBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (allBlocksAreNulls) { - result.appendNull(); - continue position; - } - try { - MvSlice.process(result, p, fieldBlock, startBlock.getInt(startBlock.getFirstValueIndex(p)), endBlock.getInt(endBlock.getFirstValueIndex(p))); - } catch (InvalidArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "MvSliceBooleanEvaluator[" + "field=" + field + ", start=" + start + ", end=" + end + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(field, start, end); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - private final EvalOperator.ExpressionEvaluator.Factory start; - - private final EvalOperator.ExpressionEvaluator.Factory end; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory field, - EvalOperator.ExpressionEvaluator.Factory start, - EvalOperator.ExpressionEvaluator.Factory end) { - this.source = source; - this.field = field; - this.start = start; - this.end = end; - } - - @Override - public MvSliceBooleanEvaluator get(DriverContext context) { - return new MvSliceBooleanEvaluator(source, field.get(context), start.get(context), end.get(context), context); - } - - @Override - public String toString() { - return "MvSliceBooleanEvaluator[" + "field=" + field + ", start=" + start + ", end=" + end + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvSliceBytesRefEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvSliceBytesRefEvaluator.java deleted file mode 100644 index 4a4a169e45aee..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvSliceBytesRefEvaluator.java +++ /dev/null @@ -1,140 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.InvalidArgumentException; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvSlice}. - * This class is generated. Do not edit it. - */ -public final class MvSliceBytesRefEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator field; - - private final EvalOperator.ExpressionEvaluator start; - - private final EvalOperator.ExpressionEvaluator end; - - private final DriverContext driverContext; - - public MvSliceBytesRefEvaluator(Source source, EvalOperator.ExpressionEvaluator field, - EvalOperator.ExpressionEvaluator start, EvalOperator.ExpressionEvaluator end, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.field = field; - this.start = start; - this.end = end; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock fieldBlock = (BytesRefBlock) field.eval(page)) { - try (IntBlock startBlock = (IntBlock) start.eval(page)) { - try (IntBlock endBlock = (IntBlock) end.eval(page)) { - return eval(page.getPositionCount(), fieldBlock, startBlock, endBlock); - } - } - } - } - - public BytesRefBlock eval(int positionCount, BytesRefBlock fieldBlock, IntBlock startBlock, - IntBlock endBlock) { - try(BytesRefBlock.Builder result = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - boolean allBlocksAreNulls = true; - if (!fieldBlock.isNull(p)) { - allBlocksAreNulls = false; - } - if (startBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (startBlock.getValueCount(p) != 1) { - if (startBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (endBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (endBlock.getValueCount(p) != 1) { - if (endBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (allBlocksAreNulls) { - result.appendNull(); - continue position; - } - try { - MvSlice.process(result, p, fieldBlock, startBlock.getInt(startBlock.getFirstValueIndex(p)), endBlock.getInt(endBlock.getFirstValueIndex(p))); - } catch (InvalidArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "MvSliceBytesRefEvaluator[" + "field=" + field + ", start=" + start + ", end=" + end + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(field, start, end); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - private final EvalOperator.ExpressionEvaluator.Factory start; - - private final EvalOperator.ExpressionEvaluator.Factory end; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory field, - EvalOperator.ExpressionEvaluator.Factory start, - EvalOperator.ExpressionEvaluator.Factory end) { - this.source = source; - this.field = field; - this.start = start; - this.end = end; - } - - @Override - public MvSliceBytesRefEvaluator get(DriverContext context) { - return new MvSliceBytesRefEvaluator(source, field.get(context), start.get(context), end.get(context), context); - } - - @Override - public String toString() { - return "MvSliceBytesRefEvaluator[" + "field=" + field + ", start=" + start + ", end=" + end + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvSliceDoubleEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvSliceDoubleEvaluator.java deleted file mode 100644 index 3e4a83cec68b7..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvSliceDoubleEvaluator.java +++ /dev/null @@ -1,140 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.InvalidArgumentException; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvSlice}. - * This class is generated. Do not edit it. - */ -public final class MvSliceDoubleEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator field; - - private final EvalOperator.ExpressionEvaluator start; - - private final EvalOperator.ExpressionEvaluator end; - - private final DriverContext driverContext; - - public MvSliceDoubleEvaluator(Source source, EvalOperator.ExpressionEvaluator field, - EvalOperator.ExpressionEvaluator start, EvalOperator.ExpressionEvaluator end, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.field = field; - this.start = start; - this.end = end; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock fieldBlock = (DoubleBlock) field.eval(page)) { - try (IntBlock startBlock = (IntBlock) start.eval(page)) { - try (IntBlock endBlock = (IntBlock) end.eval(page)) { - return eval(page.getPositionCount(), fieldBlock, startBlock, endBlock); - } - } - } - } - - public DoubleBlock eval(int positionCount, DoubleBlock fieldBlock, IntBlock startBlock, - IntBlock endBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - boolean allBlocksAreNulls = true; - if (!fieldBlock.isNull(p)) { - allBlocksAreNulls = false; - } - if (startBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (startBlock.getValueCount(p) != 1) { - if (startBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (endBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (endBlock.getValueCount(p) != 1) { - if (endBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (allBlocksAreNulls) { - result.appendNull(); - continue position; - } - try { - MvSlice.process(result, p, fieldBlock, startBlock.getInt(startBlock.getFirstValueIndex(p)), endBlock.getInt(endBlock.getFirstValueIndex(p))); - } catch (InvalidArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "MvSliceDoubleEvaluator[" + "field=" + field + ", start=" + start + ", end=" + end + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(field, start, end); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - private final EvalOperator.ExpressionEvaluator.Factory start; - - private final EvalOperator.ExpressionEvaluator.Factory end; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory field, - EvalOperator.ExpressionEvaluator.Factory start, - EvalOperator.ExpressionEvaluator.Factory end) { - this.source = source; - this.field = field; - this.start = start; - this.end = end; - } - - @Override - public MvSliceDoubleEvaluator get(DriverContext context) { - return new MvSliceDoubleEvaluator(source, field.get(context), start.get(context), end.get(context), context); - } - - @Override - public String toString() { - return "MvSliceDoubleEvaluator[" + "field=" + field + ", start=" + start + ", end=" + end + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvSliceIntEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvSliceIntEvaluator.java deleted file mode 100644 index fc54dfb1f8336..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvSliceIntEvaluator.java +++ /dev/null @@ -1,139 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.InvalidArgumentException; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvSlice}. - * This class is generated. Do not edit it. - */ -public final class MvSliceIntEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator field; - - private final EvalOperator.ExpressionEvaluator start; - - private final EvalOperator.ExpressionEvaluator end; - - private final DriverContext driverContext; - - public MvSliceIntEvaluator(Source source, EvalOperator.ExpressionEvaluator field, - EvalOperator.ExpressionEvaluator start, EvalOperator.ExpressionEvaluator end, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.field = field; - this.start = start; - this.end = end; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (IntBlock fieldBlock = (IntBlock) field.eval(page)) { - try (IntBlock startBlock = (IntBlock) start.eval(page)) { - try (IntBlock endBlock = (IntBlock) end.eval(page)) { - return eval(page.getPositionCount(), fieldBlock, startBlock, endBlock); - } - } - } - } - - public IntBlock eval(int positionCount, IntBlock fieldBlock, IntBlock startBlock, - IntBlock endBlock) { - try(IntBlock.Builder result = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - boolean allBlocksAreNulls = true; - if (!fieldBlock.isNull(p)) { - allBlocksAreNulls = false; - } - if (startBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (startBlock.getValueCount(p) != 1) { - if (startBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (endBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (endBlock.getValueCount(p) != 1) { - if (endBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (allBlocksAreNulls) { - result.appendNull(); - continue position; - } - try { - MvSlice.process(result, p, fieldBlock, startBlock.getInt(startBlock.getFirstValueIndex(p)), endBlock.getInt(endBlock.getFirstValueIndex(p))); - } catch (InvalidArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "MvSliceIntEvaluator[" + "field=" + field + ", start=" + start + ", end=" + end + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(field, start, end); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - private final EvalOperator.ExpressionEvaluator.Factory start; - - private final EvalOperator.ExpressionEvaluator.Factory end; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory field, - EvalOperator.ExpressionEvaluator.Factory start, - EvalOperator.ExpressionEvaluator.Factory end) { - this.source = source; - this.field = field; - this.start = start; - this.end = end; - } - - @Override - public MvSliceIntEvaluator get(DriverContext context) { - return new MvSliceIntEvaluator(source, field.get(context), start.get(context), end.get(context), context); - } - - @Override - public String toString() { - return "MvSliceIntEvaluator[" + "field=" + field + ", start=" + start + ", end=" + end + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvSliceLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvSliceLongEvaluator.java deleted file mode 100644 index d6a1e7e45cabf..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvSliceLongEvaluator.java +++ /dev/null @@ -1,140 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.InvalidArgumentException; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvSlice}. - * This class is generated. Do not edit it. - */ -public final class MvSliceLongEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator field; - - private final EvalOperator.ExpressionEvaluator start; - - private final EvalOperator.ExpressionEvaluator end; - - private final DriverContext driverContext; - - public MvSliceLongEvaluator(Source source, EvalOperator.ExpressionEvaluator field, - EvalOperator.ExpressionEvaluator start, EvalOperator.ExpressionEvaluator end, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.field = field; - this.start = start; - this.end = end; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock fieldBlock = (LongBlock) field.eval(page)) { - try (IntBlock startBlock = (IntBlock) start.eval(page)) { - try (IntBlock endBlock = (IntBlock) end.eval(page)) { - return eval(page.getPositionCount(), fieldBlock, startBlock, endBlock); - } - } - } - } - - public LongBlock eval(int positionCount, LongBlock fieldBlock, IntBlock startBlock, - IntBlock endBlock) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - boolean allBlocksAreNulls = true; - if (!fieldBlock.isNull(p)) { - allBlocksAreNulls = false; - } - if (startBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (startBlock.getValueCount(p) != 1) { - if (startBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (endBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (endBlock.getValueCount(p) != 1) { - if (endBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (allBlocksAreNulls) { - result.appendNull(); - continue position; - } - try { - MvSlice.process(result, p, fieldBlock, startBlock.getInt(startBlock.getFirstValueIndex(p)), endBlock.getInt(endBlock.getFirstValueIndex(p))); - } catch (InvalidArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "MvSliceLongEvaluator[" + "field=" + field + ", start=" + start + ", end=" + end + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(field, start, end); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - private final EvalOperator.ExpressionEvaluator.Factory start; - - private final EvalOperator.ExpressionEvaluator.Factory end; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory field, - EvalOperator.ExpressionEvaluator.Factory start, - EvalOperator.ExpressionEvaluator.Factory end) { - this.source = source; - this.field = field; - this.start = start; - this.end = end; - } - - @Override - public MvSliceLongEvaluator get(DriverContext context) { - return new MvSliceLongEvaluator(source, field.get(context), start.get(context), end.get(context), context); - } - - @Override - public String toString() { - return "MvSliceLongEvaluator[" + "field=" + field + ", start=" + start + ", end=" + end + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvSumDoubleEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvSumDoubleEvaluator.java deleted file mode 100644 index b49a92404ecd1..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvSumDoubleEvaluator.java +++ /dev/null @@ -1,99 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.search.aggregations.metrics.CompensatedSum; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvSum}. - * This class is generated. Do not edit it. - */ -public final class MvSumDoubleEvaluator extends AbstractMultivalueFunction.AbstractEvaluator { - public MvSumDoubleEvaluator(EvalOperator.ExpressionEvaluator field, DriverContext driverContext) { - super(driverContext, field); - } - - @Override - public String name() { - return "MvSum"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - DoubleBlock v = (DoubleBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - CompensatedSum work = new CompensatedSum(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - for (int i = first; i < end; i++) { - double value = v.getDouble(i); - MvSum.process(work, value); - } - double result = MvSum.finish(work); - builder.appendDouble(result); - } - return builder.build(); - } - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNotNullable(Block fieldVal) { - DoubleBlock v = (DoubleBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (DoubleVector.FixedBuilder builder = driverContext.blockFactory().newDoubleVectorFixedBuilder(positionCount)) { - CompensatedSum work = new CompensatedSum(); - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - for (int i = first; i < end; i++) { - double value = v.getDouble(i); - MvSum.process(work, value); - } - double result = MvSum.finish(work); - builder.appendDouble(result); - } - return builder.build().asBlock(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field) { - this.field = field; - } - - @Override - public MvSumDoubleEvaluator get(DriverContext context) { - return new MvSumDoubleEvaluator(field.get(context), context); - } - - @Override - public String toString() { - return "MvSum[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvSumIntEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvSumIntEvaluator.java deleted file mode 100644 index 20ae9a4047385..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvSumIntEvaluator.java +++ /dev/null @@ -1,88 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.ArithmeticException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvSum}. - * This class is generated. Do not edit it. - */ -public final class MvSumIntEvaluator extends AbstractMultivalueFunction.AbstractNullableEvaluator { - private final Warnings warnings; - - public MvSumIntEvaluator(Source source, EvalOperator.ExpressionEvaluator field, - DriverContext driverContext) { - super(driverContext, field); - this.warnings = new Warnings(source); - } - - @Override - public String name() { - return "MvSum"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - IntBlock v = (IntBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (IntBlock.Builder builder = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - try { - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - int value = v.getInt(first); - for (int i = first + 1; i < end; i++) { - int next = v.getInt(i); - value = MvSum.process(value, next); - } - int result = value; - builder.appendInt(result); - } catch (ArithmeticException e) { - warnings.registerException(e); - builder.appendNull(); - } - } - return builder.build(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory field) { - this.source = source; - this.field = field; - } - - @Override - public MvSumIntEvaluator get(DriverContext context) { - return new MvSumIntEvaluator(source, field.get(context), context); - } - - @Override - public String toString() { - return "MvSum[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvSumLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvSumLongEvaluator.java deleted file mode 100644 index bff596a76d697..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvSumLongEvaluator.java +++ /dev/null @@ -1,88 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.ArithmeticException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvSum}. - * This class is generated. Do not edit it. - */ -public final class MvSumLongEvaluator extends AbstractMultivalueFunction.AbstractNullableEvaluator { - private final Warnings warnings; - - public MvSumLongEvaluator(Source source, EvalOperator.ExpressionEvaluator field, - DriverContext driverContext) { - super(driverContext, field); - this.warnings = new Warnings(source); - } - - @Override - public String name() { - return "MvSum"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - LongBlock v = (LongBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - try { - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - long value = v.getLong(first); - for (int i = first + 1; i < end; i++) { - long next = v.getLong(i); - value = MvSum.process(value, next); - } - long result = value; - builder.appendLong(result); - } catch (ArithmeticException e) { - warnings.registerException(e); - builder.appendNull(); - } - } - return builder.build(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory field) { - this.source = source; - this.field = field; - } - - @Override - public MvSumLongEvaluator get(DriverContext context) { - return new MvSumLongEvaluator(source, field.get(context), context); - } - - @Override - public String toString() { - return "MvSum[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvSumUnsignedLongEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvSumUnsignedLongEvaluator.java deleted file mode 100644 index 28ae5e5a2da3b..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvSumUnsignedLongEvaluator.java +++ /dev/null @@ -1,88 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.ArithmeticException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvSum}. - * This class is generated. Do not edit it. - */ -public final class MvSumUnsignedLongEvaluator extends AbstractMultivalueFunction.AbstractNullableEvaluator { - private final Warnings warnings; - - public MvSumUnsignedLongEvaluator(Source source, EvalOperator.ExpressionEvaluator field, - DriverContext driverContext) { - super(driverContext, field); - this.warnings = new Warnings(source); - } - - @Override - public String name() { - return "MvSum"; - } - - /** - * Evaluate blocks containing at least one multivalued field. - */ - @Override - public Block evalNullable(Block fieldVal) { - LongBlock v = (LongBlock) fieldVal; - int positionCount = v.getPositionCount(); - try (LongBlock.Builder builder = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - int valueCount = v.getValueCount(p); - if (valueCount == 0) { - builder.appendNull(); - continue; - } - try { - int first = v.getFirstValueIndex(p); - int end = first + valueCount; - long value = v.getLong(first); - for (int i = first + 1; i < end; i++) { - long next = v.getLong(i); - value = MvSum.processUnsignedLong(value, next); - } - long result = value; - builder.appendLong(result); - } catch (ArithmeticException e) { - warnings.registerException(e); - builder.appendNull(); - } - } - return builder.build(); - } - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory field) { - this.source = source; - this.field = field; - } - - @Override - public MvSumUnsignedLongEvaluator get(DriverContext context) { - return new MvSumUnsignedLongEvaluator(source, field.get(context), context); - } - - @Override - public String toString() { - return "MvSum[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvZipEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvZipEvaluator.java deleted file mode 100644 index b53a1c8f9b3c0..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/multivalue/MvZipEvaluator.java +++ /dev/null @@ -1,127 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.multivalue; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link MvZip}. - * This class is generated. Do not edit it. - */ -public final class MvZipEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftField; - - private final EvalOperator.ExpressionEvaluator rightField; - - private final EvalOperator.ExpressionEvaluator delim; - - private final DriverContext driverContext; - - public MvZipEvaluator(Source source, EvalOperator.ExpressionEvaluator leftField, - EvalOperator.ExpressionEvaluator rightField, EvalOperator.ExpressionEvaluator delim, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftField = leftField; - this.rightField = rightField; - this.delim = delim; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock leftFieldBlock = (BytesRefBlock) leftField.eval(page)) { - try (BytesRefBlock rightFieldBlock = (BytesRefBlock) rightField.eval(page)) { - try (BytesRefBlock delimBlock = (BytesRefBlock) delim.eval(page)) { - return eval(page.getPositionCount(), leftFieldBlock, rightFieldBlock, delimBlock); - } - } - } - } - - public BytesRefBlock eval(int positionCount, BytesRefBlock leftFieldBlock, - BytesRefBlock rightFieldBlock, BytesRefBlock delimBlock) { - try(BytesRefBlock.Builder result = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef delimScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - boolean allBlocksAreNulls = true; - if (!leftFieldBlock.isNull(p)) { - allBlocksAreNulls = false; - } - if (!rightFieldBlock.isNull(p)) { - allBlocksAreNulls = false; - } - if (delimBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (delimBlock.getValueCount(p) != 1) { - if (delimBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (allBlocksAreNulls) { - result.appendNull(); - continue position; - } - MvZip.process(result, p, leftFieldBlock, rightFieldBlock, delimBlock.getBytesRef(delimBlock.getFirstValueIndex(p), delimScratch)); - } - return result.build(); - } - } - - @Override - public String toString() { - return "MvZipEvaluator[" + "leftField=" + leftField + ", rightField=" + rightField + ", delim=" + delim + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftField, rightField, delim); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftField; - - private final EvalOperator.ExpressionEvaluator.Factory rightField; - - private final EvalOperator.ExpressionEvaluator.Factory delim; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftField, - EvalOperator.ExpressionEvaluator.Factory rightField, - EvalOperator.ExpressionEvaluator.Factory delim) { - this.source = source; - this.leftField = leftField; - this.rightField = rightField; - this.delim = delim; - } - - @Override - public MvZipEvaluator get(DriverContext context) { - return new MvZipEvaluator(source, leftField.get(context), rightField.get(context), delim.get(context), context); - } - - @Override - public String toString() { - return "MvZipEvaluator[" + "leftField=" + leftField + ", rightField=" + rightField + ", delim=" + delim + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialContainsCartesianPointDocValuesAndConstantEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialContainsCartesianPointDocValuesAndConstantEvaluator.java deleted file mode 100644 index 982bbd3b518d5..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialContainsCartesianPointDocValuesAndConstantEvaluator.java +++ /dev/null @@ -1,128 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.geo.Component2D; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialContains}. - * This class is generated. Do not edit it. - */ -public final class SpatialContainsCartesianPointDocValuesAndConstantEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final Component2D[] rightValue; - - private final DriverContext driverContext; - - public SpatialContainsCartesianPointDocValuesAndConstantEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, Component2D[] rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock leftValueBlock = (LongBlock) leftValue.eval(page)) { - LongVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock); - } - return eval(page.getPositionCount(), leftValueVector); - } - } - - public BooleanBlock eval(int positionCount, LongBlock leftValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendBoolean(SpatialContains.processCartesianPointDocValuesAndConstant(leftValueBlock.getLong(leftValueBlock.getFirstValueIndex(p)), rightValue)); - } catch (IllegalArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public BooleanBlock eval(int positionCount, LongVector leftValueVector) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendBoolean(SpatialContains.processCartesianPointDocValuesAndConstant(leftValueVector.getLong(p), rightValue)); - } catch (IllegalArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialContainsCartesianPointDocValuesAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final Component2D[] rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - Component2D[] rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialContainsCartesianPointDocValuesAndConstantEvaluator get(DriverContext context) { - return new SpatialContainsCartesianPointDocValuesAndConstantEvaluator(source, leftValue.get(context), rightValue, context); - } - - @Override - public String toString() { - return "SpatialContainsCartesianPointDocValuesAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialContainsCartesianPointDocValuesAndSourceEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialContainsCartesianPointDocValuesAndSourceEvaluator.java deleted file mode 100644 index 5b536707e8a0f..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialContainsCartesianPointDocValuesAndSourceEvaluator.java +++ /dev/null @@ -1,142 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialContains}. - * This class is generated. Do not edit it. - */ -public final class SpatialContainsCartesianPointDocValuesAndSourceEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final EvalOperator.ExpressionEvaluator rightValue; - - private final DriverContext driverContext; - - public SpatialContainsCartesianPointDocValuesAndSourceEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, EvalOperator.ExpressionEvaluator rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock leftValueBlock = (LongBlock) leftValue.eval(page)) { - try (BytesRefBlock rightValueBlock = (BytesRefBlock) rightValue.eval(page)) { - LongVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - BytesRefVector rightValueVector = rightValueBlock.asVector(); - if (rightValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - return eval(page.getPositionCount(), leftValueVector, rightValueVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, LongBlock leftValueBlock, - BytesRefBlock rightValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rightValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rightValueBlock.getValueCount(p) != 1) { - if (rightValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(SpatialContains.processCartesianPointDocValuesAndSource(leftValueBlock.getLong(leftValueBlock.getFirstValueIndex(p)), rightValueBlock.getBytesRef(rightValueBlock.getFirstValueIndex(p), rightValueScratch))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, LongVector leftValueVector, - BytesRefVector rightValueVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(SpatialContains.processCartesianPointDocValuesAndSource(leftValueVector.getLong(p), rightValueVector.getBytesRef(p, rightValueScratch))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialContainsCartesianPointDocValuesAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue, rightValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final EvalOperator.ExpressionEvaluator.Factory rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - EvalOperator.ExpressionEvaluator.Factory rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialContainsCartesianPointDocValuesAndSourceEvaluator get(DriverContext context) { - return new SpatialContainsCartesianPointDocValuesAndSourceEvaluator(source, leftValue.get(context), rightValue.get(context), context); - } - - @Override - public String toString() { - return "SpatialContainsCartesianPointDocValuesAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialContainsCartesianSourceAndConstantEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialContainsCartesianSourceAndConstantEvaluator.java deleted file mode 100644 index a9c3a4f887a7d..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialContainsCartesianSourceAndConstantEvaluator.java +++ /dev/null @@ -1,132 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.io.IOException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.geo.Component2D; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialContains}. - * This class is generated. Do not edit it. - */ -public final class SpatialContainsCartesianSourceAndConstantEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final Component2D[] rightValue; - - private final DriverContext driverContext; - - public SpatialContainsCartesianSourceAndConstantEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, Component2D[] rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock leftValueBlock = (BytesRefBlock) leftValue.eval(page)) { - BytesRefVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock); - } - return eval(page.getPositionCount(), leftValueVector); - } - } - - public BooleanBlock eval(int positionCount, BytesRefBlock leftValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendBoolean(SpatialContains.processCartesianSourceAndConstant(leftValueBlock.getBytesRef(leftValueBlock.getFirstValueIndex(p), leftValueScratch), rightValue)); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public BooleanBlock eval(int positionCount, BytesRefVector leftValueVector) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendBoolean(SpatialContains.processCartesianSourceAndConstant(leftValueVector.getBytesRef(p, leftValueScratch), rightValue)); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialContainsCartesianSourceAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final Component2D[] rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - Component2D[] rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialContainsCartesianSourceAndConstantEvaluator get(DriverContext context) { - return new SpatialContainsCartesianSourceAndConstantEvaluator(source, leftValue.get(context), rightValue, context); - } - - @Override - public String toString() { - return "SpatialContainsCartesianSourceAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialContainsCartesianSourceAndSourceEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialContainsCartesianSourceAndSourceEvaluator.java deleted file mode 100644 index d2456597b5761..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialContainsCartesianSourceAndSourceEvaluator.java +++ /dev/null @@ -1,152 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.io.IOException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialContains}. - * This class is generated. Do not edit it. - */ -public final class SpatialContainsCartesianSourceAndSourceEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final EvalOperator.ExpressionEvaluator rightValue; - - private final DriverContext driverContext; - - public SpatialContainsCartesianSourceAndSourceEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, EvalOperator.ExpressionEvaluator rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock leftValueBlock = (BytesRefBlock) leftValue.eval(page)) { - try (BytesRefBlock rightValueBlock = (BytesRefBlock) rightValue.eval(page)) { - BytesRefVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - BytesRefVector rightValueVector = rightValueBlock.asVector(); - if (rightValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - return eval(page.getPositionCount(), leftValueVector, rightValueVector); - } - } - } - - public BooleanBlock eval(int positionCount, BytesRefBlock leftValueBlock, - BytesRefBlock rightValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rightValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rightValueBlock.getValueCount(p) != 1) { - if (rightValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendBoolean(SpatialContains.processCartesianSourceAndSource(leftValueBlock.getBytesRef(leftValueBlock.getFirstValueIndex(p), leftValueScratch), rightValueBlock.getBytesRef(rightValueBlock.getFirstValueIndex(p), rightValueScratch))); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public BooleanBlock eval(int positionCount, BytesRefVector leftValueVector, - BytesRefVector rightValueVector) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendBoolean(SpatialContains.processCartesianSourceAndSource(leftValueVector.getBytesRef(p, leftValueScratch), rightValueVector.getBytesRef(p, rightValueScratch))); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialContainsCartesianSourceAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue, rightValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final EvalOperator.ExpressionEvaluator.Factory rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - EvalOperator.ExpressionEvaluator.Factory rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialContainsCartesianSourceAndSourceEvaluator get(DriverContext context) { - return new SpatialContainsCartesianSourceAndSourceEvaluator(source, leftValue.get(context), rightValue.get(context), context); - } - - @Override - public String toString() { - return "SpatialContainsCartesianSourceAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialContainsGeoPointDocValuesAndConstantEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialContainsGeoPointDocValuesAndConstantEvaluator.java deleted file mode 100644 index de4537e6e0a10..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialContainsGeoPointDocValuesAndConstantEvaluator.java +++ /dev/null @@ -1,128 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.geo.Component2D; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialContains}. - * This class is generated. Do not edit it. - */ -public final class SpatialContainsGeoPointDocValuesAndConstantEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final Component2D[] rightValue; - - private final DriverContext driverContext; - - public SpatialContainsGeoPointDocValuesAndConstantEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, Component2D[] rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock leftValueBlock = (LongBlock) leftValue.eval(page)) { - LongVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock); - } - return eval(page.getPositionCount(), leftValueVector); - } - } - - public BooleanBlock eval(int positionCount, LongBlock leftValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendBoolean(SpatialContains.processGeoPointDocValuesAndConstant(leftValueBlock.getLong(leftValueBlock.getFirstValueIndex(p)), rightValue)); - } catch (IllegalArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public BooleanBlock eval(int positionCount, LongVector leftValueVector) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendBoolean(SpatialContains.processGeoPointDocValuesAndConstant(leftValueVector.getLong(p), rightValue)); - } catch (IllegalArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialContainsGeoPointDocValuesAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final Component2D[] rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - Component2D[] rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialContainsGeoPointDocValuesAndConstantEvaluator get(DriverContext context) { - return new SpatialContainsGeoPointDocValuesAndConstantEvaluator(source, leftValue.get(context), rightValue, context); - } - - @Override - public String toString() { - return "SpatialContainsGeoPointDocValuesAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialContainsGeoPointDocValuesAndSourceEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialContainsGeoPointDocValuesAndSourceEvaluator.java deleted file mode 100644 index 348c343f0b005..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialContainsGeoPointDocValuesAndSourceEvaluator.java +++ /dev/null @@ -1,151 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialContains}. - * This class is generated. Do not edit it. - */ -public final class SpatialContainsGeoPointDocValuesAndSourceEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final EvalOperator.ExpressionEvaluator rightValue; - - private final DriverContext driverContext; - - public SpatialContainsGeoPointDocValuesAndSourceEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, EvalOperator.ExpressionEvaluator rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock leftValueBlock = (LongBlock) leftValue.eval(page)) { - try (BytesRefBlock rightValueBlock = (BytesRefBlock) rightValue.eval(page)) { - LongVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - BytesRefVector rightValueVector = rightValueBlock.asVector(); - if (rightValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - return eval(page.getPositionCount(), leftValueVector, rightValueVector); - } - } - } - - public BooleanBlock eval(int positionCount, LongBlock leftValueBlock, - BytesRefBlock rightValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rightValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rightValueBlock.getValueCount(p) != 1) { - if (rightValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendBoolean(SpatialContains.processGeoPointDocValuesAndSource(leftValueBlock.getLong(leftValueBlock.getFirstValueIndex(p)), rightValueBlock.getBytesRef(rightValueBlock.getFirstValueIndex(p), rightValueScratch))); - } catch (IllegalArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public BooleanBlock eval(int positionCount, LongVector leftValueVector, - BytesRefVector rightValueVector) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendBoolean(SpatialContains.processGeoPointDocValuesAndSource(leftValueVector.getLong(p), rightValueVector.getBytesRef(p, rightValueScratch))); - } catch (IllegalArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialContainsGeoPointDocValuesAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue, rightValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final EvalOperator.ExpressionEvaluator.Factory rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - EvalOperator.ExpressionEvaluator.Factory rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialContainsGeoPointDocValuesAndSourceEvaluator get(DriverContext context) { - return new SpatialContainsGeoPointDocValuesAndSourceEvaluator(source, leftValue.get(context), rightValue.get(context), context); - } - - @Override - public String toString() { - return "SpatialContainsGeoPointDocValuesAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialContainsGeoSourceAndConstantEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialContainsGeoSourceAndConstantEvaluator.java deleted file mode 100644 index a84c661df18d8..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialContainsGeoSourceAndConstantEvaluator.java +++ /dev/null @@ -1,132 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.io.IOException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.geo.Component2D; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialContains}. - * This class is generated. Do not edit it. - */ -public final class SpatialContainsGeoSourceAndConstantEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final Component2D[] rightValue; - - private final DriverContext driverContext; - - public SpatialContainsGeoSourceAndConstantEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, Component2D[] rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock leftValueBlock = (BytesRefBlock) leftValue.eval(page)) { - BytesRefVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock); - } - return eval(page.getPositionCount(), leftValueVector); - } - } - - public BooleanBlock eval(int positionCount, BytesRefBlock leftValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendBoolean(SpatialContains.processGeoSourceAndConstant(leftValueBlock.getBytesRef(leftValueBlock.getFirstValueIndex(p), leftValueScratch), rightValue)); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public BooleanBlock eval(int positionCount, BytesRefVector leftValueVector) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendBoolean(SpatialContains.processGeoSourceAndConstant(leftValueVector.getBytesRef(p, leftValueScratch), rightValue)); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialContainsGeoSourceAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final Component2D[] rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - Component2D[] rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialContainsGeoSourceAndConstantEvaluator get(DriverContext context) { - return new SpatialContainsGeoSourceAndConstantEvaluator(source, leftValue.get(context), rightValue, context); - } - - @Override - public String toString() { - return "SpatialContainsGeoSourceAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialContainsGeoSourceAndSourceEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialContainsGeoSourceAndSourceEvaluator.java deleted file mode 100644 index 6bff91629f74c..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialContainsGeoSourceAndSourceEvaluator.java +++ /dev/null @@ -1,152 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.io.IOException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialContains}. - * This class is generated. Do not edit it. - */ -public final class SpatialContainsGeoSourceAndSourceEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final EvalOperator.ExpressionEvaluator rightValue; - - private final DriverContext driverContext; - - public SpatialContainsGeoSourceAndSourceEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, EvalOperator.ExpressionEvaluator rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock leftValueBlock = (BytesRefBlock) leftValue.eval(page)) { - try (BytesRefBlock rightValueBlock = (BytesRefBlock) rightValue.eval(page)) { - BytesRefVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - BytesRefVector rightValueVector = rightValueBlock.asVector(); - if (rightValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - return eval(page.getPositionCount(), leftValueVector, rightValueVector); - } - } - } - - public BooleanBlock eval(int positionCount, BytesRefBlock leftValueBlock, - BytesRefBlock rightValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rightValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rightValueBlock.getValueCount(p) != 1) { - if (rightValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendBoolean(SpatialContains.processGeoSourceAndSource(leftValueBlock.getBytesRef(leftValueBlock.getFirstValueIndex(p), leftValueScratch), rightValueBlock.getBytesRef(rightValueBlock.getFirstValueIndex(p), rightValueScratch))); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public BooleanBlock eval(int positionCount, BytesRefVector leftValueVector, - BytesRefVector rightValueVector) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendBoolean(SpatialContains.processGeoSourceAndSource(leftValueVector.getBytesRef(p, leftValueScratch), rightValueVector.getBytesRef(p, rightValueScratch))); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialContainsGeoSourceAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue, rightValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final EvalOperator.ExpressionEvaluator.Factory rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - EvalOperator.ExpressionEvaluator.Factory rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialContainsGeoSourceAndSourceEvaluator get(DriverContext context) { - return new SpatialContainsGeoSourceAndSourceEvaluator(source, leftValue.get(context), rightValue.get(context), context); - } - - @Override - public String toString() { - return "SpatialContainsGeoSourceAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialDisjointCartesianPointDocValuesAndConstantEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialDisjointCartesianPointDocValuesAndConstantEvaluator.java deleted file mode 100644 index 62b5761cfd655..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialDisjointCartesianPointDocValuesAndConstantEvaluator.java +++ /dev/null @@ -1,128 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.geo.Component2D; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialDisjoint}. - * This class is generated. Do not edit it. - */ -public final class SpatialDisjointCartesianPointDocValuesAndConstantEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final Component2D rightValue; - - private final DriverContext driverContext; - - public SpatialDisjointCartesianPointDocValuesAndConstantEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, Component2D rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock leftValueBlock = (LongBlock) leftValue.eval(page)) { - LongVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock); - } - return eval(page.getPositionCount(), leftValueVector); - } - } - - public BooleanBlock eval(int positionCount, LongBlock leftValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendBoolean(SpatialDisjoint.processCartesianPointDocValuesAndConstant(leftValueBlock.getLong(leftValueBlock.getFirstValueIndex(p)), rightValue)); - } catch (IllegalArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public BooleanBlock eval(int positionCount, LongVector leftValueVector) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendBoolean(SpatialDisjoint.processCartesianPointDocValuesAndConstant(leftValueVector.getLong(p), rightValue)); - } catch (IllegalArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialDisjointCartesianPointDocValuesAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final Component2D rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - Component2D rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialDisjointCartesianPointDocValuesAndConstantEvaluator get(DriverContext context) { - return new SpatialDisjointCartesianPointDocValuesAndConstantEvaluator(source, leftValue.get(context), rightValue, context); - } - - @Override - public String toString() { - return "SpatialDisjointCartesianPointDocValuesAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialDisjointCartesianPointDocValuesAndSourceEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialDisjointCartesianPointDocValuesAndSourceEvaluator.java deleted file mode 100644 index 4f11da3c474a9..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialDisjointCartesianPointDocValuesAndSourceEvaluator.java +++ /dev/null @@ -1,142 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialDisjoint}. - * This class is generated. Do not edit it. - */ -public final class SpatialDisjointCartesianPointDocValuesAndSourceEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final EvalOperator.ExpressionEvaluator rightValue; - - private final DriverContext driverContext; - - public SpatialDisjointCartesianPointDocValuesAndSourceEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, EvalOperator.ExpressionEvaluator rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock leftValueBlock = (LongBlock) leftValue.eval(page)) { - try (BytesRefBlock rightValueBlock = (BytesRefBlock) rightValue.eval(page)) { - LongVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - BytesRefVector rightValueVector = rightValueBlock.asVector(); - if (rightValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - return eval(page.getPositionCount(), leftValueVector, rightValueVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, LongBlock leftValueBlock, - BytesRefBlock rightValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rightValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rightValueBlock.getValueCount(p) != 1) { - if (rightValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(SpatialDisjoint.processCartesianPointDocValuesAndSource(leftValueBlock.getLong(leftValueBlock.getFirstValueIndex(p)), rightValueBlock.getBytesRef(rightValueBlock.getFirstValueIndex(p), rightValueScratch))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, LongVector leftValueVector, - BytesRefVector rightValueVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(SpatialDisjoint.processCartesianPointDocValuesAndSource(leftValueVector.getLong(p), rightValueVector.getBytesRef(p, rightValueScratch))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialDisjointCartesianPointDocValuesAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue, rightValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final EvalOperator.ExpressionEvaluator.Factory rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - EvalOperator.ExpressionEvaluator.Factory rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialDisjointCartesianPointDocValuesAndSourceEvaluator get(DriverContext context) { - return new SpatialDisjointCartesianPointDocValuesAndSourceEvaluator(source, leftValue.get(context), rightValue.get(context), context); - } - - @Override - public String toString() { - return "SpatialDisjointCartesianPointDocValuesAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialDisjointCartesianSourceAndConstantEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialDisjointCartesianSourceAndConstantEvaluator.java deleted file mode 100644 index adb5a33b83f3b..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialDisjointCartesianSourceAndConstantEvaluator.java +++ /dev/null @@ -1,132 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.io.IOException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.geo.Component2D; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialDisjoint}. - * This class is generated. Do not edit it. - */ -public final class SpatialDisjointCartesianSourceAndConstantEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final Component2D rightValue; - - private final DriverContext driverContext; - - public SpatialDisjointCartesianSourceAndConstantEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, Component2D rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock leftValueBlock = (BytesRefBlock) leftValue.eval(page)) { - BytesRefVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock); - } - return eval(page.getPositionCount(), leftValueVector); - } - } - - public BooleanBlock eval(int positionCount, BytesRefBlock leftValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendBoolean(SpatialDisjoint.processCartesianSourceAndConstant(leftValueBlock.getBytesRef(leftValueBlock.getFirstValueIndex(p), leftValueScratch), rightValue)); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public BooleanBlock eval(int positionCount, BytesRefVector leftValueVector) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendBoolean(SpatialDisjoint.processCartesianSourceAndConstant(leftValueVector.getBytesRef(p, leftValueScratch), rightValue)); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialDisjointCartesianSourceAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final Component2D rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - Component2D rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialDisjointCartesianSourceAndConstantEvaluator get(DriverContext context) { - return new SpatialDisjointCartesianSourceAndConstantEvaluator(source, leftValue.get(context), rightValue, context); - } - - @Override - public String toString() { - return "SpatialDisjointCartesianSourceAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialDisjointCartesianSourceAndSourceEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialDisjointCartesianSourceAndSourceEvaluator.java deleted file mode 100644 index 186a1299a4a98..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialDisjointCartesianSourceAndSourceEvaluator.java +++ /dev/null @@ -1,152 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.io.IOException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialDisjoint}. - * This class is generated. Do not edit it. - */ -public final class SpatialDisjointCartesianSourceAndSourceEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final EvalOperator.ExpressionEvaluator rightValue; - - private final DriverContext driverContext; - - public SpatialDisjointCartesianSourceAndSourceEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, EvalOperator.ExpressionEvaluator rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock leftValueBlock = (BytesRefBlock) leftValue.eval(page)) { - try (BytesRefBlock rightValueBlock = (BytesRefBlock) rightValue.eval(page)) { - BytesRefVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - BytesRefVector rightValueVector = rightValueBlock.asVector(); - if (rightValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - return eval(page.getPositionCount(), leftValueVector, rightValueVector); - } - } - } - - public BooleanBlock eval(int positionCount, BytesRefBlock leftValueBlock, - BytesRefBlock rightValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rightValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rightValueBlock.getValueCount(p) != 1) { - if (rightValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendBoolean(SpatialDisjoint.processCartesianSourceAndSource(leftValueBlock.getBytesRef(leftValueBlock.getFirstValueIndex(p), leftValueScratch), rightValueBlock.getBytesRef(rightValueBlock.getFirstValueIndex(p), rightValueScratch))); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public BooleanBlock eval(int positionCount, BytesRefVector leftValueVector, - BytesRefVector rightValueVector) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendBoolean(SpatialDisjoint.processCartesianSourceAndSource(leftValueVector.getBytesRef(p, leftValueScratch), rightValueVector.getBytesRef(p, rightValueScratch))); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialDisjointCartesianSourceAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue, rightValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final EvalOperator.ExpressionEvaluator.Factory rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - EvalOperator.ExpressionEvaluator.Factory rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialDisjointCartesianSourceAndSourceEvaluator get(DriverContext context) { - return new SpatialDisjointCartesianSourceAndSourceEvaluator(source, leftValue.get(context), rightValue.get(context), context); - } - - @Override - public String toString() { - return "SpatialDisjointCartesianSourceAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialDisjointGeoPointDocValuesAndConstantEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialDisjointGeoPointDocValuesAndConstantEvaluator.java deleted file mode 100644 index 675b6cc58197e..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialDisjointGeoPointDocValuesAndConstantEvaluator.java +++ /dev/null @@ -1,128 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.geo.Component2D; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialDisjoint}. - * This class is generated. Do not edit it. - */ -public final class SpatialDisjointGeoPointDocValuesAndConstantEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final Component2D rightValue; - - private final DriverContext driverContext; - - public SpatialDisjointGeoPointDocValuesAndConstantEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, Component2D rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock leftValueBlock = (LongBlock) leftValue.eval(page)) { - LongVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock); - } - return eval(page.getPositionCount(), leftValueVector); - } - } - - public BooleanBlock eval(int positionCount, LongBlock leftValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendBoolean(SpatialDisjoint.processGeoPointDocValuesAndConstant(leftValueBlock.getLong(leftValueBlock.getFirstValueIndex(p)), rightValue)); - } catch (IllegalArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public BooleanBlock eval(int positionCount, LongVector leftValueVector) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendBoolean(SpatialDisjoint.processGeoPointDocValuesAndConstant(leftValueVector.getLong(p), rightValue)); - } catch (IllegalArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialDisjointGeoPointDocValuesAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final Component2D rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - Component2D rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialDisjointGeoPointDocValuesAndConstantEvaluator get(DriverContext context) { - return new SpatialDisjointGeoPointDocValuesAndConstantEvaluator(source, leftValue.get(context), rightValue, context); - } - - @Override - public String toString() { - return "SpatialDisjointGeoPointDocValuesAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialDisjointGeoPointDocValuesAndSourceEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialDisjointGeoPointDocValuesAndSourceEvaluator.java deleted file mode 100644 index 1b22e67d11b25..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialDisjointGeoPointDocValuesAndSourceEvaluator.java +++ /dev/null @@ -1,151 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialDisjoint}. - * This class is generated. Do not edit it. - */ -public final class SpatialDisjointGeoPointDocValuesAndSourceEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final EvalOperator.ExpressionEvaluator rightValue; - - private final DriverContext driverContext; - - public SpatialDisjointGeoPointDocValuesAndSourceEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, EvalOperator.ExpressionEvaluator rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock leftValueBlock = (LongBlock) leftValue.eval(page)) { - try (BytesRefBlock rightValueBlock = (BytesRefBlock) rightValue.eval(page)) { - LongVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - BytesRefVector rightValueVector = rightValueBlock.asVector(); - if (rightValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - return eval(page.getPositionCount(), leftValueVector, rightValueVector); - } - } - } - - public BooleanBlock eval(int positionCount, LongBlock leftValueBlock, - BytesRefBlock rightValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rightValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rightValueBlock.getValueCount(p) != 1) { - if (rightValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendBoolean(SpatialDisjoint.processGeoPointDocValuesAndSource(leftValueBlock.getLong(leftValueBlock.getFirstValueIndex(p)), rightValueBlock.getBytesRef(rightValueBlock.getFirstValueIndex(p), rightValueScratch))); - } catch (IllegalArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public BooleanBlock eval(int positionCount, LongVector leftValueVector, - BytesRefVector rightValueVector) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendBoolean(SpatialDisjoint.processGeoPointDocValuesAndSource(leftValueVector.getLong(p), rightValueVector.getBytesRef(p, rightValueScratch))); - } catch (IllegalArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialDisjointGeoPointDocValuesAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue, rightValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final EvalOperator.ExpressionEvaluator.Factory rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - EvalOperator.ExpressionEvaluator.Factory rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialDisjointGeoPointDocValuesAndSourceEvaluator get(DriverContext context) { - return new SpatialDisjointGeoPointDocValuesAndSourceEvaluator(source, leftValue.get(context), rightValue.get(context), context); - } - - @Override - public String toString() { - return "SpatialDisjointGeoPointDocValuesAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialDisjointGeoSourceAndConstantEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialDisjointGeoSourceAndConstantEvaluator.java deleted file mode 100644 index 1df80cf90bd10..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialDisjointGeoSourceAndConstantEvaluator.java +++ /dev/null @@ -1,132 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.io.IOException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.geo.Component2D; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialDisjoint}. - * This class is generated. Do not edit it. - */ -public final class SpatialDisjointGeoSourceAndConstantEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final Component2D rightValue; - - private final DriverContext driverContext; - - public SpatialDisjointGeoSourceAndConstantEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, Component2D rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock leftValueBlock = (BytesRefBlock) leftValue.eval(page)) { - BytesRefVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock); - } - return eval(page.getPositionCount(), leftValueVector); - } - } - - public BooleanBlock eval(int positionCount, BytesRefBlock leftValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendBoolean(SpatialDisjoint.processGeoSourceAndConstant(leftValueBlock.getBytesRef(leftValueBlock.getFirstValueIndex(p), leftValueScratch), rightValue)); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public BooleanBlock eval(int positionCount, BytesRefVector leftValueVector) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendBoolean(SpatialDisjoint.processGeoSourceAndConstant(leftValueVector.getBytesRef(p, leftValueScratch), rightValue)); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialDisjointGeoSourceAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final Component2D rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - Component2D rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialDisjointGeoSourceAndConstantEvaluator get(DriverContext context) { - return new SpatialDisjointGeoSourceAndConstantEvaluator(source, leftValue.get(context), rightValue, context); - } - - @Override - public String toString() { - return "SpatialDisjointGeoSourceAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialDisjointGeoSourceAndSourceEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialDisjointGeoSourceAndSourceEvaluator.java deleted file mode 100644 index 9bdc60813ad67..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialDisjointGeoSourceAndSourceEvaluator.java +++ /dev/null @@ -1,152 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.io.IOException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialDisjoint}. - * This class is generated. Do not edit it. - */ -public final class SpatialDisjointGeoSourceAndSourceEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final EvalOperator.ExpressionEvaluator rightValue; - - private final DriverContext driverContext; - - public SpatialDisjointGeoSourceAndSourceEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, EvalOperator.ExpressionEvaluator rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock leftValueBlock = (BytesRefBlock) leftValue.eval(page)) { - try (BytesRefBlock rightValueBlock = (BytesRefBlock) rightValue.eval(page)) { - BytesRefVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - BytesRefVector rightValueVector = rightValueBlock.asVector(); - if (rightValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - return eval(page.getPositionCount(), leftValueVector, rightValueVector); - } - } - } - - public BooleanBlock eval(int positionCount, BytesRefBlock leftValueBlock, - BytesRefBlock rightValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rightValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rightValueBlock.getValueCount(p) != 1) { - if (rightValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendBoolean(SpatialDisjoint.processGeoSourceAndSource(leftValueBlock.getBytesRef(leftValueBlock.getFirstValueIndex(p), leftValueScratch), rightValueBlock.getBytesRef(rightValueBlock.getFirstValueIndex(p), rightValueScratch))); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public BooleanBlock eval(int positionCount, BytesRefVector leftValueVector, - BytesRefVector rightValueVector) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendBoolean(SpatialDisjoint.processGeoSourceAndSource(leftValueVector.getBytesRef(p, leftValueScratch), rightValueVector.getBytesRef(p, rightValueScratch))); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialDisjointGeoSourceAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue, rightValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final EvalOperator.ExpressionEvaluator.Factory rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - EvalOperator.ExpressionEvaluator.Factory rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialDisjointGeoSourceAndSourceEvaluator get(DriverContext context) { - return new SpatialDisjointGeoSourceAndSourceEvaluator(source, leftValue.get(context), rightValue.get(context), context); - } - - @Override - public String toString() { - return "SpatialDisjointGeoSourceAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialIntersectsCartesianPointDocValuesAndConstantEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialIntersectsCartesianPointDocValuesAndConstantEvaluator.java deleted file mode 100644 index e32357c42bf71..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialIntersectsCartesianPointDocValuesAndConstantEvaluator.java +++ /dev/null @@ -1,128 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.geo.Component2D; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialIntersects}. - * This class is generated. Do not edit it. - */ -public final class SpatialIntersectsCartesianPointDocValuesAndConstantEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final Component2D rightValue; - - private final DriverContext driverContext; - - public SpatialIntersectsCartesianPointDocValuesAndConstantEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, Component2D rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock leftValueBlock = (LongBlock) leftValue.eval(page)) { - LongVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock); - } - return eval(page.getPositionCount(), leftValueVector); - } - } - - public BooleanBlock eval(int positionCount, LongBlock leftValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendBoolean(SpatialIntersects.processCartesianPointDocValuesAndConstant(leftValueBlock.getLong(leftValueBlock.getFirstValueIndex(p)), rightValue)); - } catch (IllegalArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public BooleanBlock eval(int positionCount, LongVector leftValueVector) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendBoolean(SpatialIntersects.processCartesianPointDocValuesAndConstant(leftValueVector.getLong(p), rightValue)); - } catch (IllegalArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialIntersectsCartesianPointDocValuesAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final Component2D rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - Component2D rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialIntersectsCartesianPointDocValuesAndConstantEvaluator get(DriverContext context) { - return new SpatialIntersectsCartesianPointDocValuesAndConstantEvaluator(source, leftValue.get(context), rightValue, context); - } - - @Override - public String toString() { - return "SpatialIntersectsCartesianPointDocValuesAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialIntersectsCartesianPointDocValuesAndSourceEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialIntersectsCartesianPointDocValuesAndSourceEvaluator.java deleted file mode 100644 index 7bf47b766bd95..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialIntersectsCartesianPointDocValuesAndSourceEvaluator.java +++ /dev/null @@ -1,142 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialIntersects}. - * This class is generated. Do not edit it. - */ -public final class SpatialIntersectsCartesianPointDocValuesAndSourceEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final EvalOperator.ExpressionEvaluator rightValue; - - private final DriverContext driverContext; - - public SpatialIntersectsCartesianPointDocValuesAndSourceEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, EvalOperator.ExpressionEvaluator rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock leftValueBlock = (LongBlock) leftValue.eval(page)) { - try (BytesRefBlock rightValueBlock = (BytesRefBlock) rightValue.eval(page)) { - LongVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - BytesRefVector rightValueVector = rightValueBlock.asVector(); - if (rightValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - return eval(page.getPositionCount(), leftValueVector, rightValueVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, LongBlock leftValueBlock, - BytesRefBlock rightValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rightValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rightValueBlock.getValueCount(p) != 1) { - if (rightValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(SpatialIntersects.processCartesianPointDocValuesAndSource(leftValueBlock.getLong(leftValueBlock.getFirstValueIndex(p)), rightValueBlock.getBytesRef(rightValueBlock.getFirstValueIndex(p), rightValueScratch))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, LongVector leftValueVector, - BytesRefVector rightValueVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(SpatialIntersects.processCartesianPointDocValuesAndSource(leftValueVector.getLong(p), rightValueVector.getBytesRef(p, rightValueScratch))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialIntersectsCartesianPointDocValuesAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue, rightValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final EvalOperator.ExpressionEvaluator.Factory rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - EvalOperator.ExpressionEvaluator.Factory rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialIntersectsCartesianPointDocValuesAndSourceEvaluator get(DriverContext context) { - return new SpatialIntersectsCartesianPointDocValuesAndSourceEvaluator(source, leftValue.get(context), rightValue.get(context), context); - } - - @Override - public String toString() { - return "SpatialIntersectsCartesianPointDocValuesAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialIntersectsCartesianSourceAndConstantEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialIntersectsCartesianSourceAndConstantEvaluator.java deleted file mode 100644 index 979869dc86c56..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialIntersectsCartesianSourceAndConstantEvaluator.java +++ /dev/null @@ -1,132 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.io.IOException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.geo.Component2D; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialIntersects}. - * This class is generated. Do not edit it. - */ -public final class SpatialIntersectsCartesianSourceAndConstantEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final Component2D rightValue; - - private final DriverContext driverContext; - - public SpatialIntersectsCartesianSourceAndConstantEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, Component2D rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock leftValueBlock = (BytesRefBlock) leftValue.eval(page)) { - BytesRefVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock); - } - return eval(page.getPositionCount(), leftValueVector); - } - } - - public BooleanBlock eval(int positionCount, BytesRefBlock leftValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendBoolean(SpatialIntersects.processCartesianSourceAndConstant(leftValueBlock.getBytesRef(leftValueBlock.getFirstValueIndex(p), leftValueScratch), rightValue)); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public BooleanBlock eval(int positionCount, BytesRefVector leftValueVector) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendBoolean(SpatialIntersects.processCartesianSourceAndConstant(leftValueVector.getBytesRef(p, leftValueScratch), rightValue)); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialIntersectsCartesianSourceAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final Component2D rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - Component2D rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialIntersectsCartesianSourceAndConstantEvaluator get(DriverContext context) { - return new SpatialIntersectsCartesianSourceAndConstantEvaluator(source, leftValue.get(context), rightValue, context); - } - - @Override - public String toString() { - return "SpatialIntersectsCartesianSourceAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialIntersectsCartesianSourceAndSourceEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialIntersectsCartesianSourceAndSourceEvaluator.java deleted file mode 100644 index 6c47745d6af37..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialIntersectsCartesianSourceAndSourceEvaluator.java +++ /dev/null @@ -1,152 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.io.IOException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialIntersects}. - * This class is generated. Do not edit it. - */ -public final class SpatialIntersectsCartesianSourceAndSourceEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final EvalOperator.ExpressionEvaluator rightValue; - - private final DriverContext driverContext; - - public SpatialIntersectsCartesianSourceAndSourceEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, EvalOperator.ExpressionEvaluator rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock leftValueBlock = (BytesRefBlock) leftValue.eval(page)) { - try (BytesRefBlock rightValueBlock = (BytesRefBlock) rightValue.eval(page)) { - BytesRefVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - BytesRefVector rightValueVector = rightValueBlock.asVector(); - if (rightValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - return eval(page.getPositionCount(), leftValueVector, rightValueVector); - } - } - } - - public BooleanBlock eval(int positionCount, BytesRefBlock leftValueBlock, - BytesRefBlock rightValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rightValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rightValueBlock.getValueCount(p) != 1) { - if (rightValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendBoolean(SpatialIntersects.processCartesianSourceAndSource(leftValueBlock.getBytesRef(leftValueBlock.getFirstValueIndex(p), leftValueScratch), rightValueBlock.getBytesRef(rightValueBlock.getFirstValueIndex(p), rightValueScratch))); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public BooleanBlock eval(int positionCount, BytesRefVector leftValueVector, - BytesRefVector rightValueVector) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendBoolean(SpatialIntersects.processCartesianSourceAndSource(leftValueVector.getBytesRef(p, leftValueScratch), rightValueVector.getBytesRef(p, rightValueScratch))); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialIntersectsCartesianSourceAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue, rightValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final EvalOperator.ExpressionEvaluator.Factory rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - EvalOperator.ExpressionEvaluator.Factory rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialIntersectsCartesianSourceAndSourceEvaluator get(DriverContext context) { - return new SpatialIntersectsCartesianSourceAndSourceEvaluator(source, leftValue.get(context), rightValue.get(context), context); - } - - @Override - public String toString() { - return "SpatialIntersectsCartesianSourceAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialIntersectsGeoPointDocValuesAndConstantEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialIntersectsGeoPointDocValuesAndConstantEvaluator.java deleted file mode 100644 index 8d87884d04077..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialIntersectsGeoPointDocValuesAndConstantEvaluator.java +++ /dev/null @@ -1,128 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.geo.Component2D; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialIntersects}. - * This class is generated. Do not edit it. - */ -public final class SpatialIntersectsGeoPointDocValuesAndConstantEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final Component2D rightValue; - - private final DriverContext driverContext; - - public SpatialIntersectsGeoPointDocValuesAndConstantEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, Component2D rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock leftValueBlock = (LongBlock) leftValue.eval(page)) { - LongVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock); - } - return eval(page.getPositionCount(), leftValueVector); - } - } - - public BooleanBlock eval(int positionCount, LongBlock leftValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendBoolean(SpatialIntersects.processGeoPointDocValuesAndConstant(leftValueBlock.getLong(leftValueBlock.getFirstValueIndex(p)), rightValue)); - } catch (IllegalArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public BooleanBlock eval(int positionCount, LongVector leftValueVector) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendBoolean(SpatialIntersects.processGeoPointDocValuesAndConstant(leftValueVector.getLong(p), rightValue)); - } catch (IllegalArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialIntersectsGeoPointDocValuesAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final Component2D rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - Component2D rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialIntersectsGeoPointDocValuesAndConstantEvaluator get(DriverContext context) { - return new SpatialIntersectsGeoPointDocValuesAndConstantEvaluator(source, leftValue.get(context), rightValue, context); - } - - @Override - public String toString() { - return "SpatialIntersectsGeoPointDocValuesAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialIntersectsGeoPointDocValuesAndSourceEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialIntersectsGeoPointDocValuesAndSourceEvaluator.java deleted file mode 100644 index 45e9daf5bc453..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialIntersectsGeoPointDocValuesAndSourceEvaluator.java +++ /dev/null @@ -1,151 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialIntersects}. - * This class is generated. Do not edit it. - */ -public final class SpatialIntersectsGeoPointDocValuesAndSourceEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final EvalOperator.ExpressionEvaluator rightValue; - - private final DriverContext driverContext; - - public SpatialIntersectsGeoPointDocValuesAndSourceEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, EvalOperator.ExpressionEvaluator rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock leftValueBlock = (LongBlock) leftValue.eval(page)) { - try (BytesRefBlock rightValueBlock = (BytesRefBlock) rightValue.eval(page)) { - LongVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - BytesRefVector rightValueVector = rightValueBlock.asVector(); - if (rightValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - return eval(page.getPositionCount(), leftValueVector, rightValueVector); - } - } - } - - public BooleanBlock eval(int positionCount, LongBlock leftValueBlock, - BytesRefBlock rightValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rightValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rightValueBlock.getValueCount(p) != 1) { - if (rightValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendBoolean(SpatialIntersects.processGeoPointDocValuesAndSource(leftValueBlock.getLong(leftValueBlock.getFirstValueIndex(p)), rightValueBlock.getBytesRef(rightValueBlock.getFirstValueIndex(p), rightValueScratch))); - } catch (IllegalArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public BooleanBlock eval(int positionCount, LongVector leftValueVector, - BytesRefVector rightValueVector) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendBoolean(SpatialIntersects.processGeoPointDocValuesAndSource(leftValueVector.getLong(p), rightValueVector.getBytesRef(p, rightValueScratch))); - } catch (IllegalArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialIntersectsGeoPointDocValuesAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue, rightValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final EvalOperator.ExpressionEvaluator.Factory rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - EvalOperator.ExpressionEvaluator.Factory rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialIntersectsGeoPointDocValuesAndSourceEvaluator get(DriverContext context) { - return new SpatialIntersectsGeoPointDocValuesAndSourceEvaluator(source, leftValue.get(context), rightValue.get(context), context); - } - - @Override - public String toString() { - return "SpatialIntersectsGeoPointDocValuesAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialIntersectsGeoSourceAndConstantEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialIntersectsGeoSourceAndConstantEvaluator.java deleted file mode 100644 index f043ff4104bbb..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialIntersectsGeoSourceAndConstantEvaluator.java +++ /dev/null @@ -1,132 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.io.IOException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.geo.Component2D; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialIntersects}. - * This class is generated. Do not edit it. - */ -public final class SpatialIntersectsGeoSourceAndConstantEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final Component2D rightValue; - - private final DriverContext driverContext; - - public SpatialIntersectsGeoSourceAndConstantEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, Component2D rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock leftValueBlock = (BytesRefBlock) leftValue.eval(page)) { - BytesRefVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock); - } - return eval(page.getPositionCount(), leftValueVector); - } - } - - public BooleanBlock eval(int positionCount, BytesRefBlock leftValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendBoolean(SpatialIntersects.processGeoSourceAndConstant(leftValueBlock.getBytesRef(leftValueBlock.getFirstValueIndex(p), leftValueScratch), rightValue)); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public BooleanBlock eval(int positionCount, BytesRefVector leftValueVector) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendBoolean(SpatialIntersects.processGeoSourceAndConstant(leftValueVector.getBytesRef(p, leftValueScratch), rightValue)); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialIntersectsGeoSourceAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final Component2D rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - Component2D rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialIntersectsGeoSourceAndConstantEvaluator get(DriverContext context) { - return new SpatialIntersectsGeoSourceAndConstantEvaluator(source, leftValue.get(context), rightValue, context); - } - - @Override - public String toString() { - return "SpatialIntersectsGeoSourceAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialIntersectsGeoSourceAndSourceEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialIntersectsGeoSourceAndSourceEvaluator.java deleted file mode 100644 index 9f5f1c7cc9674..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialIntersectsGeoSourceAndSourceEvaluator.java +++ /dev/null @@ -1,152 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.io.IOException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialIntersects}. - * This class is generated. Do not edit it. - */ -public final class SpatialIntersectsGeoSourceAndSourceEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final EvalOperator.ExpressionEvaluator rightValue; - - private final DriverContext driverContext; - - public SpatialIntersectsGeoSourceAndSourceEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, EvalOperator.ExpressionEvaluator rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock leftValueBlock = (BytesRefBlock) leftValue.eval(page)) { - try (BytesRefBlock rightValueBlock = (BytesRefBlock) rightValue.eval(page)) { - BytesRefVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - BytesRefVector rightValueVector = rightValueBlock.asVector(); - if (rightValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - return eval(page.getPositionCount(), leftValueVector, rightValueVector); - } - } - } - - public BooleanBlock eval(int positionCount, BytesRefBlock leftValueBlock, - BytesRefBlock rightValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rightValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rightValueBlock.getValueCount(p) != 1) { - if (rightValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendBoolean(SpatialIntersects.processGeoSourceAndSource(leftValueBlock.getBytesRef(leftValueBlock.getFirstValueIndex(p), leftValueScratch), rightValueBlock.getBytesRef(rightValueBlock.getFirstValueIndex(p), rightValueScratch))); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public BooleanBlock eval(int positionCount, BytesRefVector leftValueVector, - BytesRefVector rightValueVector) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendBoolean(SpatialIntersects.processGeoSourceAndSource(leftValueVector.getBytesRef(p, leftValueScratch), rightValueVector.getBytesRef(p, rightValueScratch))); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialIntersectsGeoSourceAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue, rightValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final EvalOperator.ExpressionEvaluator.Factory rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - EvalOperator.ExpressionEvaluator.Factory rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialIntersectsGeoSourceAndSourceEvaluator get(DriverContext context) { - return new SpatialIntersectsGeoSourceAndSourceEvaluator(source, leftValue.get(context), rightValue.get(context), context); - } - - @Override - public String toString() { - return "SpatialIntersectsGeoSourceAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialWithinCartesianPointDocValuesAndConstantEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialWithinCartesianPointDocValuesAndConstantEvaluator.java deleted file mode 100644 index 6deb7133fcf13..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialWithinCartesianPointDocValuesAndConstantEvaluator.java +++ /dev/null @@ -1,128 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.geo.Component2D; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialWithin}. - * This class is generated. Do not edit it. - */ -public final class SpatialWithinCartesianPointDocValuesAndConstantEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final Component2D rightValue; - - private final DriverContext driverContext; - - public SpatialWithinCartesianPointDocValuesAndConstantEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, Component2D rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock leftValueBlock = (LongBlock) leftValue.eval(page)) { - LongVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock); - } - return eval(page.getPositionCount(), leftValueVector); - } - } - - public BooleanBlock eval(int positionCount, LongBlock leftValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendBoolean(SpatialWithin.processCartesianPointDocValuesAndConstant(leftValueBlock.getLong(leftValueBlock.getFirstValueIndex(p)), rightValue)); - } catch (IllegalArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public BooleanBlock eval(int positionCount, LongVector leftValueVector) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendBoolean(SpatialWithin.processCartesianPointDocValuesAndConstant(leftValueVector.getLong(p), rightValue)); - } catch (IllegalArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialWithinCartesianPointDocValuesAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final Component2D rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - Component2D rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialWithinCartesianPointDocValuesAndConstantEvaluator get(DriverContext context) { - return new SpatialWithinCartesianPointDocValuesAndConstantEvaluator(source, leftValue.get(context), rightValue, context); - } - - @Override - public String toString() { - return "SpatialWithinCartesianPointDocValuesAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialWithinCartesianPointDocValuesAndSourceEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialWithinCartesianPointDocValuesAndSourceEvaluator.java deleted file mode 100644 index d2470583c3a7c..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialWithinCartesianPointDocValuesAndSourceEvaluator.java +++ /dev/null @@ -1,142 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialWithin}. - * This class is generated. Do not edit it. - */ -public final class SpatialWithinCartesianPointDocValuesAndSourceEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final EvalOperator.ExpressionEvaluator rightValue; - - private final DriverContext driverContext; - - public SpatialWithinCartesianPointDocValuesAndSourceEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, EvalOperator.ExpressionEvaluator rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock leftValueBlock = (LongBlock) leftValue.eval(page)) { - try (BytesRefBlock rightValueBlock = (BytesRefBlock) rightValue.eval(page)) { - LongVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - BytesRefVector rightValueVector = rightValueBlock.asVector(); - if (rightValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - return eval(page.getPositionCount(), leftValueVector, rightValueVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, LongBlock leftValueBlock, - BytesRefBlock rightValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rightValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rightValueBlock.getValueCount(p) != 1) { - if (rightValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(SpatialWithin.processCartesianPointDocValuesAndSource(leftValueBlock.getLong(leftValueBlock.getFirstValueIndex(p)), rightValueBlock.getBytesRef(rightValueBlock.getFirstValueIndex(p), rightValueScratch))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, LongVector leftValueVector, - BytesRefVector rightValueVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(SpatialWithin.processCartesianPointDocValuesAndSource(leftValueVector.getLong(p), rightValueVector.getBytesRef(p, rightValueScratch))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialWithinCartesianPointDocValuesAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue, rightValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final EvalOperator.ExpressionEvaluator.Factory rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - EvalOperator.ExpressionEvaluator.Factory rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialWithinCartesianPointDocValuesAndSourceEvaluator get(DriverContext context) { - return new SpatialWithinCartesianPointDocValuesAndSourceEvaluator(source, leftValue.get(context), rightValue.get(context), context); - } - - @Override - public String toString() { - return "SpatialWithinCartesianPointDocValuesAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialWithinCartesianSourceAndConstantEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialWithinCartesianSourceAndConstantEvaluator.java deleted file mode 100644 index 45c8f60d12b03..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialWithinCartesianSourceAndConstantEvaluator.java +++ /dev/null @@ -1,132 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.io.IOException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.geo.Component2D; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialWithin}. - * This class is generated. Do not edit it. - */ -public final class SpatialWithinCartesianSourceAndConstantEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final Component2D rightValue; - - private final DriverContext driverContext; - - public SpatialWithinCartesianSourceAndConstantEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, Component2D rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock leftValueBlock = (BytesRefBlock) leftValue.eval(page)) { - BytesRefVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock); - } - return eval(page.getPositionCount(), leftValueVector); - } - } - - public BooleanBlock eval(int positionCount, BytesRefBlock leftValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendBoolean(SpatialWithin.processCartesianSourceAndConstant(leftValueBlock.getBytesRef(leftValueBlock.getFirstValueIndex(p), leftValueScratch), rightValue)); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public BooleanBlock eval(int positionCount, BytesRefVector leftValueVector) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendBoolean(SpatialWithin.processCartesianSourceAndConstant(leftValueVector.getBytesRef(p, leftValueScratch), rightValue)); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialWithinCartesianSourceAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final Component2D rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - Component2D rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialWithinCartesianSourceAndConstantEvaluator get(DriverContext context) { - return new SpatialWithinCartesianSourceAndConstantEvaluator(source, leftValue.get(context), rightValue, context); - } - - @Override - public String toString() { - return "SpatialWithinCartesianSourceAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialWithinCartesianSourceAndSourceEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialWithinCartesianSourceAndSourceEvaluator.java deleted file mode 100644 index 958ac825eeb0b..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialWithinCartesianSourceAndSourceEvaluator.java +++ /dev/null @@ -1,152 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.io.IOException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialWithin}. - * This class is generated. Do not edit it. - */ -public final class SpatialWithinCartesianSourceAndSourceEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final EvalOperator.ExpressionEvaluator rightValue; - - private final DriverContext driverContext; - - public SpatialWithinCartesianSourceAndSourceEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, EvalOperator.ExpressionEvaluator rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock leftValueBlock = (BytesRefBlock) leftValue.eval(page)) { - try (BytesRefBlock rightValueBlock = (BytesRefBlock) rightValue.eval(page)) { - BytesRefVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - BytesRefVector rightValueVector = rightValueBlock.asVector(); - if (rightValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - return eval(page.getPositionCount(), leftValueVector, rightValueVector); - } - } - } - - public BooleanBlock eval(int positionCount, BytesRefBlock leftValueBlock, - BytesRefBlock rightValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rightValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rightValueBlock.getValueCount(p) != 1) { - if (rightValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendBoolean(SpatialWithin.processCartesianSourceAndSource(leftValueBlock.getBytesRef(leftValueBlock.getFirstValueIndex(p), leftValueScratch), rightValueBlock.getBytesRef(rightValueBlock.getFirstValueIndex(p), rightValueScratch))); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public BooleanBlock eval(int positionCount, BytesRefVector leftValueVector, - BytesRefVector rightValueVector) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendBoolean(SpatialWithin.processCartesianSourceAndSource(leftValueVector.getBytesRef(p, leftValueScratch), rightValueVector.getBytesRef(p, rightValueScratch))); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialWithinCartesianSourceAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue, rightValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final EvalOperator.ExpressionEvaluator.Factory rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - EvalOperator.ExpressionEvaluator.Factory rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialWithinCartesianSourceAndSourceEvaluator get(DriverContext context) { - return new SpatialWithinCartesianSourceAndSourceEvaluator(source, leftValue.get(context), rightValue.get(context), context); - } - - @Override - public String toString() { - return "SpatialWithinCartesianSourceAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialWithinGeoPointDocValuesAndConstantEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialWithinGeoPointDocValuesAndConstantEvaluator.java deleted file mode 100644 index 680cf7b38445b..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialWithinGeoPointDocValuesAndConstantEvaluator.java +++ /dev/null @@ -1,128 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.geo.Component2D; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialWithin}. - * This class is generated. Do not edit it. - */ -public final class SpatialWithinGeoPointDocValuesAndConstantEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final Component2D rightValue; - - private final DriverContext driverContext; - - public SpatialWithinGeoPointDocValuesAndConstantEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, Component2D rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock leftValueBlock = (LongBlock) leftValue.eval(page)) { - LongVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock); - } - return eval(page.getPositionCount(), leftValueVector); - } - } - - public BooleanBlock eval(int positionCount, LongBlock leftValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendBoolean(SpatialWithin.processGeoPointDocValuesAndConstant(leftValueBlock.getLong(leftValueBlock.getFirstValueIndex(p)), rightValue)); - } catch (IllegalArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public BooleanBlock eval(int positionCount, LongVector leftValueVector) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendBoolean(SpatialWithin.processGeoPointDocValuesAndConstant(leftValueVector.getLong(p), rightValue)); - } catch (IllegalArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialWithinGeoPointDocValuesAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final Component2D rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - Component2D rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialWithinGeoPointDocValuesAndConstantEvaluator get(DriverContext context) { - return new SpatialWithinGeoPointDocValuesAndConstantEvaluator(source, leftValue.get(context), rightValue, context); - } - - @Override - public String toString() { - return "SpatialWithinGeoPointDocValuesAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialWithinGeoPointDocValuesAndSourceEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialWithinGeoPointDocValuesAndSourceEvaluator.java deleted file mode 100644 index 624b9243a62c4..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialWithinGeoPointDocValuesAndSourceEvaluator.java +++ /dev/null @@ -1,151 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialWithin}. - * This class is generated. Do not edit it. - */ -public final class SpatialWithinGeoPointDocValuesAndSourceEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final EvalOperator.ExpressionEvaluator rightValue; - - private final DriverContext driverContext; - - public SpatialWithinGeoPointDocValuesAndSourceEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, EvalOperator.ExpressionEvaluator rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock leftValueBlock = (LongBlock) leftValue.eval(page)) { - try (BytesRefBlock rightValueBlock = (BytesRefBlock) rightValue.eval(page)) { - LongVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - BytesRefVector rightValueVector = rightValueBlock.asVector(); - if (rightValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - return eval(page.getPositionCount(), leftValueVector, rightValueVector); - } - } - } - - public BooleanBlock eval(int positionCount, LongBlock leftValueBlock, - BytesRefBlock rightValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rightValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rightValueBlock.getValueCount(p) != 1) { - if (rightValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendBoolean(SpatialWithin.processGeoPointDocValuesAndSource(leftValueBlock.getLong(leftValueBlock.getFirstValueIndex(p)), rightValueBlock.getBytesRef(rightValueBlock.getFirstValueIndex(p), rightValueScratch))); - } catch (IllegalArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public BooleanBlock eval(int positionCount, LongVector leftValueVector, - BytesRefVector rightValueVector) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendBoolean(SpatialWithin.processGeoPointDocValuesAndSource(leftValueVector.getLong(p), rightValueVector.getBytesRef(p, rightValueScratch))); - } catch (IllegalArgumentException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialWithinGeoPointDocValuesAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue, rightValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final EvalOperator.ExpressionEvaluator.Factory rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - EvalOperator.ExpressionEvaluator.Factory rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialWithinGeoPointDocValuesAndSourceEvaluator get(DriverContext context) { - return new SpatialWithinGeoPointDocValuesAndSourceEvaluator(source, leftValue.get(context), rightValue.get(context), context); - } - - @Override - public String toString() { - return "SpatialWithinGeoPointDocValuesAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialWithinGeoSourceAndConstantEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialWithinGeoSourceAndConstantEvaluator.java deleted file mode 100644 index 3647594337c57..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialWithinGeoSourceAndConstantEvaluator.java +++ /dev/null @@ -1,132 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.io.IOException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.geo.Component2D; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialWithin}. - * This class is generated. Do not edit it. - */ -public final class SpatialWithinGeoSourceAndConstantEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final Component2D rightValue; - - private final DriverContext driverContext; - - public SpatialWithinGeoSourceAndConstantEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, Component2D rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock leftValueBlock = (BytesRefBlock) leftValue.eval(page)) { - BytesRefVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock); - } - return eval(page.getPositionCount(), leftValueVector); - } - } - - public BooleanBlock eval(int positionCount, BytesRefBlock leftValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendBoolean(SpatialWithin.processGeoSourceAndConstant(leftValueBlock.getBytesRef(leftValueBlock.getFirstValueIndex(p), leftValueScratch), rightValue)); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public BooleanBlock eval(int positionCount, BytesRefVector leftValueVector) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendBoolean(SpatialWithin.processGeoSourceAndConstant(leftValueVector.getBytesRef(p, leftValueScratch), rightValue)); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialWithinGeoSourceAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final Component2D rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - Component2D rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialWithinGeoSourceAndConstantEvaluator get(DriverContext context) { - return new SpatialWithinGeoSourceAndConstantEvaluator(source, leftValue.get(context), rightValue, context); - } - - @Override - public String toString() { - return "SpatialWithinGeoSourceAndConstantEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialWithinGeoSourceAndSourceEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialWithinGeoSourceAndSourceEvaluator.java deleted file mode 100644 index 8794c3d0488b3..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/SpatialWithinGeoSourceAndSourceEvaluator.java +++ /dev/null @@ -1,152 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.io.IOException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link SpatialWithin}. - * This class is generated. Do not edit it. - */ -public final class SpatialWithinGeoSourceAndSourceEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator leftValue; - - private final EvalOperator.ExpressionEvaluator rightValue; - - private final DriverContext driverContext; - - public SpatialWithinGeoSourceAndSourceEvaluator(Source source, - EvalOperator.ExpressionEvaluator leftValue, EvalOperator.ExpressionEvaluator rightValue, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.leftValue = leftValue; - this.rightValue = rightValue; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock leftValueBlock = (BytesRefBlock) leftValue.eval(page)) { - try (BytesRefBlock rightValueBlock = (BytesRefBlock) rightValue.eval(page)) { - BytesRefVector leftValueVector = leftValueBlock.asVector(); - if (leftValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - BytesRefVector rightValueVector = rightValueBlock.asVector(); - if (rightValueVector == null) { - return eval(page.getPositionCount(), leftValueBlock, rightValueBlock); - } - return eval(page.getPositionCount(), leftValueVector, rightValueVector); - } - } - } - - public BooleanBlock eval(int positionCount, BytesRefBlock leftValueBlock, - BytesRefBlock rightValueBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (leftValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (leftValueBlock.getValueCount(p) != 1) { - if (leftValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rightValueBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rightValueBlock.getValueCount(p) != 1) { - if (rightValueBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendBoolean(SpatialWithin.processGeoSourceAndSource(leftValueBlock.getBytesRef(leftValueBlock.getFirstValueIndex(p), leftValueScratch), rightValueBlock.getBytesRef(rightValueBlock.getFirstValueIndex(p), rightValueScratch))); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public BooleanBlock eval(int positionCount, BytesRefVector leftValueVector, - BytesRefVector rightValueVector) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef leftValueScratch = new BytesRef(); - BytesRef rightValueScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendBoolean(SpatialWithin.processGeoSourceAndSource(leftValueVector.getBytesRef(p, leftValueScratch), rightValueVector.getBytesRef(p, rightValueScratch))); - } catch (IllegalArgumentException | IOException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SpatialWithinGeoSourceAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(leftValue, rightValue); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory leftValue; - - private final EvalOperator.ExpressionEvaluator.Factory rightValue; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory leftValue, - EvalOperator.ExpressionEvaluator.Factory rightValue) { - this.source = source; - this.leftValue = leftValue; - this.rightValue = rightValue; - } - - @Override - public SpatialWithinGeoSourceAndSourceEvaluator get(DriverContext context) { - return new SpatialWithinGeoSourceAndSourceEvaluator(source, leftValue.get(context), rightValue.get(context), context); - } - - @Override - public String toString() { - return "SpatialWithinGeoSourceAndSourceEvaluator[" + "leftValue=" + leftValue + ", rightValue=" + rightValue + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/StXFromWKBEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/StXFromWKBEvaluator.java deleted file mode 100644 index 937eedc1d8fe0..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/StXFromWKBEvaluator.java +++ /dev/null @@ -1,127 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.esql.expression.function.scalar.convert.AbstractConvertFunction; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link StX}. - * This class is generated. Do not edit it. - */ -public final class StXFromWKBEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public StXFromWKBEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "StXFromWKB"; - } - - @Override - public Block evalVector(Vector v) { - BytesRefVector vector = (BytesRefVector) v; - int positionCount = v.getPositionCount(); - BytesRef scratchPad = new BytesRef(); - if (vector.isConstant()) { - try { - return driverContext.blockFactory().newConstantDoubleBlockWith(evalValue(vector, 0, scratchPad), positionCount); - } catch (IllegalArgumentException e) { - registerException(e); - return driverContext.blockFactory().newConstantNullBlock(positionCount); - } - } - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - try { - builder.appendDouble(evalValue(vector, p, scratchPad)); - } catch (IllegalArgumentException e) { - registerException(e); - builder.appendNull(); - } - } - return builder.build(); - } - } - - private static double evalValue(BytesRefVector container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return StX.fromWellKnownBinary(value); - } - - @Override - public Block evalBlock(Block b) { - BytesRefBlock block = (BytesRefBlock) b; - int positionCount = block.getPositionCount(); - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - BytesRef scratchPad = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - try { - double value = evalValue(block, i, scratchPad); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendDouble(value); - valuesAppended = true; - } catch (IllegalArgumentException e) { - registerException(e); - } - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static double evalValue(BytesRefBlock container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return StX.fromWellKnownBinary(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public StXFromWKBEvaluator get(DriverContext context) { - return new StXFromWKBEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "StXFromWKBEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/StYFromWKBEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/StYFromWKBEvaluator.java deleted file mode 100644 index 33405f6db5998..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/spatial/StYFromWKBEvaluator.java +++ /dev/null @@ -1,127 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.spatial; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.Vector; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.xpack.esql.expression.function.scalar.convert.AbstractConvertFunction; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link StY}. - * This class is generated. Do not edit it. - */ -public final class StYFromWKBEvaluator extends AbstractConvertFunction.AbstractEvaluator { - public StYFromWKBEvaluator(EvalOperator.ExpressionEvaluator field, Source source, - DriverContext driverContext) { - super(driverContext, field, source); - } - - @Override - public String name() { - return "StYFromWKB"; - } - - @Override - public Block evalVector(Vector v) { - BytesRefVector vector = (BytesRefVector) v; - int positionCount = v.getPositionCount(); - BytesRef scratchPad = new BytesRef(); - if (vector.isConstant()) { - try { - return driverContext.blockFactory().newConstantDoubleBlockWith(evalValue(vector, 0, scratchPad), positionCount); - } catch (IllegalArgumentException e) { - registerException(e); - return driverContext.blockFactory().newConstantNullBlock(positionCount); - } - } - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - for (int p = 0; p < positionCount; p++) { - try { - builder.appendDouble(evalValue(vector, p, scratchPad)); - } catch (IllegalArgumentException e) { - registerException(e); - builder.appendNull(); - } - } - return builder.build(); - } - } - - private static double evalValue(BytesRefVector container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return StY.fromWellKnownBinary(value); - } - - @Override - public Block evalBlock(Block b) { - BytesRefBlock block = (BytesRefBlock) b; - int positionCount = block.getPositionCount(); - try (DoubleBlock.Builder builder = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - BytesRef scratchPad = new BytesRef(); - for (int p = 0; p < positionCount; p++) { - int valueCount = block.getValueCount(p); - int start = block.getFirstValueIndex(p); - int end = start + valueCount; - boolean positionOpened = false; - boolean valuesAppended = false; - for (int i = start; i < end; i++) { - try { - double value = evalValue(block, i, scratchPad); - if (positionOpened == false && valueCount > 1) { - builder.beginPositionEntry(); - positionOpened = true; - } - builder.appendDouble(value); - valuesAppended = true; - } catch (IllegalArgumentException e) { - registerException(e); - } - } - if (valuesAppended == false) { - builder.appendNull(); - } else if (positionOpened) { - builder.endPositionEntry(); - } - } - return builder.build(); - } - } - - private static double evalValue(BytesRefBlock container, int index, BytesRef scratchPad) { - BytesRef value = container.getBytesRef(index, scratchPad); - return StY.fromWellKnownBinary(value); - } - - public static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory field; - - public Factory(EvalOperator.ExpressionEvaluator.Factory field, Source source) { - this.field = field; - this.source = source; - } - - @Override - public StYFromWKBEvaluator get(DriverContext context) { - return new StYFromWKBEvaluator(field.get(context), source, context); - } - - @Override - public String toString() { - return "StYFromWKBEvaluator[field=" + field + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/AutomataMatchEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/AutomataMatchEvaluator.java deleted file mode 100644 index fb95bbc1acef9..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/AutomataMatchEvaluator.java +++ /dev/null @@ -1,127 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.string; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.apache.lucene.util.automaton.ByteRunAutomaton; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link AutomataMatch}. - * This class is generated. Do not edit it. - */ -public final class AutomataMatchEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator input; - - private final ByteRunAutomaton automaton; - - private final String pattern; - - private final DriverContext driverContext; - - public AutomataMatchEvaluator(Source source, EvalOperator.ExpressionEvaluator input, - ByteRunAutomaton automaton, String pattern, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.input = input; - this.automaton = automaton; - this.pattern = pattern; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock inputBlock = (BytesRefBlock) input.eval(page)) { - BytesRefVector inputVector = inputBlock.asVector(); - if (inputVector == null) { - return eval(page.getPositionCount(), inputBlock); - } - return eval(page.getPositionCount(), inputVector).asBlock(); - } - } - - public BooleanBlock eval(int positionCount, BytesRefBlock inputBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef inputScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (inputBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (inputBlock.getValueCount(p) != 1) { - if (inputBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(AutomataMatch.process(inputBlock.getBytesRef(inputBlock.getFirstValueIndex(p), inputScratch), automaton, pattern)); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, BytesRefVector inputVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - BytesRef inputScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(AutomataMatch.process(inputVector.getBytesRef(p, inputScratch), automaton, pattern)); - } - return result.build(); - } - } - - @Override - public String toString() { - return "AutomataMatchEvaluator[" + "input=" + input + ", pattern=" + pattern + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(input); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory input; - - private final ByteRunAutomaton automaton; - - private final String pattern; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory input, - ByteRunAutomaton automaton, String pattern) { - this.source = source; - this.input = input; - this.automaton = automaton; - this.pattern = pattern; - } - - @Override - public AutomataMatchEvaluator get(DriverContext context) { - return new AutomataMatchEvaluator(source, input.get(context), automaton, pattern, context); - } - - @Override - public String toString() { - return "AutomataMatchEvaluator[" + "input=" + input + ", pattern=" + pattern + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/ConcatEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/ConcatEvaluator.java deleted file mode 100644 index 99e87ce490eb1..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/ConcatEvaluator.java +++ /dev/null @@ -1,149 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.string; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import java.util.Arrays; -import java.util.function.Function; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.BreakingBytesRefBuilder; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasable; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Concat}. - * This class is generated. Do not edit it. - */ -public final class ConcatEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final BreakingBytesRefBuilder scratch; - - private final EvalOperator.ExpressionEvaluator[] values; - - private final DriverContext driverContext; - - public ConcatEvaluator(Source source, BreakingBytesRefBuilder scratch, - EvalOperator.ExpressionEvaluator[] values, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.scratch = scratch; - this.values = values; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - BytesRefBlock[] valuesBlocks = new BytesRefBlock[values.length]; - try (Releasable valuesRelease = Releasables.wrap(valuesBlocks)) { - for (int i = 0; i < valuesBlocks.length; i++) { - valuesBlocks[i] = (BytesRefBlock)values[i].eval(page); - } - BytesRefVector[] valuesVectors = new BytesRefVector[values.length]; - for (int i = 0; i < valuesBlocks.length; i++) { - valuesVectors[i] = valuesBlocks[i].asVector(); - if (valuesVectors[i] == null) { - return eval(page.getPositionCount(), valuesBlocks); - } - } - return eval(page.getPositionCount(), valuesVectors).asBlock(); - } - } - - public BytesRefBlock eval(int positionCount, BytesRefBlock[] valuesBlocks) { - try(BytesRefBlock.Builder result = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef[] valuesValues = new BytesRef[values.length]; - BytesRef[] valuesScratch = new BytesRef[values.length]; - for (int i = 0; i < values.length; i++) { - valuesScratch[i] = new BytesRef(); - } - position: for (int p = 0; p < positionCount; p++) { - for (int i = 0; i < valuesBlocks.length; i++) { - if (valuesBlocks[i].isNull(p)) { - result.appendNull(); - continue position; - } - if (valuesBlocks[i].getValueCount(p) != 1) { - if (valuesBlocks[i].getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - } - // unpack valuesBlocks into valuesValues - for (int i = 0; i < valuesBlocks.length; i++) { - int o = valuesBlocks[i].getFirstValueIndex(p); - valuesValues[i] = valuesBlocks[i].getBytesRef(o, valuesScratch[i]); - } - result.appendBytesRef(Concat.process(scratch, valuesValues)); - } - return result.build(); - } - } - - public BytesRefVector eval(int positionCount, BytesRefVector[] valuesVectors) { - try(BytesRefVector.Builder result = driverContext.blockFactory().newBytesRefVectorBuilder(positionCount)) { - BytesRef[] valuesValues = new BytesRef[values.length]; - BytesRef[] valuesScratch = new BytesRef[values.length]; - for (int i = 0; i < values.length; i++) { - valuesScratch[i] = new BytesRef(); - } - position: for (int p = 0; p < positionCount; p++) { - // unpack valuesVectors into valuesValues - for (int i = 0; i < valuesVectors.length; i++) { - valuesValues[i] = valuesVectors[i].getBytesRef(p, valuesScratch[i]); - } - result.appendBytesRef(Concat.process(scratch, valuesValues)); - } - return result.build(); - } - } - - @Override - public String toString() { - return "ConcatEvaluator[" + "values=" + Arrays.toString(values) + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(scratch, () -> Releasables.close(values)); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final Function scratch; - - private final EvalOperator.ExpressionEvaluator.Factory[] values; - - public Factory(Source source, Function scratch, - EvalOperator.ExpressionEvaluator.Factory[] values) { - this.source = source; - this.scratch = scratch; - this.values = values; - } - - @Override - public ConcatEvaluator get(DriverContext context) { - EvalOperator.ExpressionEvaluator[] values = Arrays.stream(this.values).map(a -> a.get(context)).toArray(EvalOperator.ExpressionEvaluator[]::new); - return new ConcatEvaluator(source, scratch.apply(context), values, context); - } - - @Override - public String toString() { - return "ConcatEvaluator[" + "values=" + Arrays.toString(values) + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/EndsWithEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/EndsWithEvaluator.java deleted file mode 100644 index 8d1d197aae9ad..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/EndsWithEvaluator.java +++ /dev/null @@ -1,140 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.string; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link EndsWith}. - * This class is generated. Do not edit it. - */ -public final class EndsWithEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator str; - - private final EvalOperator.ExpressionEvaluator suffix; - - private final DriverContext driverContext; - - public EndsWithEvaluator(Source source, EvalOperator.ExpressionEvaluator str, - EvalOperator.ExpressionEvaluator suffix, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.str = str; - this.suffix = suffix; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock strBlock = (BytesRefBlock) str.eval(page)) { - try (BytesRefBlock suffixBlock = (BytesRefBlock) suffix.eval(page)) { - BytesRefVector strVector = strBlock.asVector(); - if (strVector == null) { - return eval(page.getPositionCount(), strBlock, suffixBlock); - } - BytesRefVector suffixVector = suffixBlock.asVector(); - if (suffixVector == null) { - return eval(page.getPositionCount(), strBlock, suffixBlock); - } - return eval(page.getPositionCount(), strVector, suffixVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, BytesRefBlock strBlock, BytesRefBlock suffixBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef strScratch = new BytesRef(); - BytesRef suffixScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (strBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (strBlock.getValueCount(p) != 1) { - if (strBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (suffixBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (suffixBlock.getValueCount(p) != 1) { - if (suffixBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(EndsWith.process(strBlock.getBytesRef(strBlock.getFirstValueIndex(p), strScratch), suffixBlock.getBytesRef(suffixBlock.getFirstValueIndex(p), suffixScratch))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, BytesRefVector strVector, - BytesRefVector suffixVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - BytesRef strScratch = new BytesRef(); - BytesRef suffixScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(EndsWith.process(strVector.getBytesRef(p, strScratch), suffixVector.getBytesRef(p, suffixScratch))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "EndsWithEvaluator[" + "str=" + str + ", suffix=" + suffix + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(str, suffix); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory str; - - private final EvalOperator.ExpressionEvaluator.Factory suffix; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory str, - EvalOperator.ExpressionEvaluator.Factory suffix) { - this.source = source; - this.str = str; - this.suffix = suffix; - } - - @Override - public EndsWithEvaluator get(DriverContext context) { - return new EndsWithEvaluator(source, str.get(context), suffix.get(context), context); - } - - @Override - public String toString() { - return "EndsWithEvaluator[" + "str=" + str + ", suffix=" + suffix + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/LTrimEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/LTrimEvaluator.java deleted file mode 100644 index 0f68955507d50..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/LTrimEvaluator.java +++ /dev/null @@ -1,111 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.string; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link LTrim}. - * This class is generated. Do not edit it. - */ -public final class LTrimEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final DriverContext driverContext; - - public LTrimEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock valBlock = (BytesRefBlock) val.eval(page)) { - BytesRefVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector).asBlock(); - } - } - - public BytesRefBlock eval(int positionCount, BytesRefBlock valBlock) { - try(BytesRefBlock.Builder result = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef valScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBytesRef(LTrim.process(valBlock.getBytesRef(valBlock.getFirstValueIndex(p), valScratch))); - } - return result.build(); - } - } - - public BytesRefVector eval(int positionCount, BytesRefVector valVector) { - try(BytesRefVector.Builder result = driverContext.blockFactory().newBytesRefVectorBuilder(positionCount)) { - BytesRef valScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - result.appendBytesRef(LTrim.process(valVector.getBytesRef(p, valScratch))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "LTrimEvaluator[" + "val=" + val + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val) { - this.source = source; - this.val = val; - } - - @Override - public LTrimEvaluator get(DriverContext context) { - return new LTrimEvaluator(source, val.get(context), context); - } - - @Override - public String toString() { - return "LTrimEvaluator[" + "val=" + val + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/LeftEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/LeftEvaluator.java deleted file mode 100644 index 13e7cbe9ece92..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/LeftEvaluator.java +++ /dev/null @@ -1,154 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.string; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import java.util.function.Function; -import org.apache.lucene.util.BytesRef; -import org.apache.lucene.util.UnicodeUtil; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Left}. - * This class is generated. Do not edit it. - */ -public final class LeftEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final BytesRef out; - - private final UnicodeUtil.UTF8CodePoint cp; - - private final EvalOperator.ExpressionEvaluator str; - - private final EvalOperator.ExpressionEvaluator length; - - private final DriverContext driverContext; - - public LeftEvaluator(Source source, BytesRef out, UnicodeUtil.UTF8CodePoint cp, - EvalOperator.ExpressionEvaluator str, EvalOperator.ExpressionEvaluator length, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.out = out; - this.cp = cp; - this.str = str; - this.length = length; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock strBlock = (BytesRefBlock) str.eval(page)) { - try (IntBlock lengthBlock = (IntBlock) length.eval(page)) { - BytesRefVector strVector = strBlock.asVector(); - if (strVector == null) { - return eval(page.getPositionCount(), strBlock, lengthBlock); - } - IntVector lengthVector = lengthBlock.asVector(); - if (lengthVector == null) { - return eval(page.getPositionCount(), strBlock, lengthBlock); - } - return eval(page.getPositionCount(), strVector, lengthVector).asBlock(); - } - } - } - - public BytesRefBlock eval(int positionCount, BytesRefBlock strBlock, IntBlock lengthBlock) { - try(BytesRefBlock.Builder result = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef strScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (strBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (strBlock.getValueCount(p) != 1) { - if (strBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (lengthBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lengthBlock.getValueCount(p) != 1) { - if (lengthBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBytesRef(Left.process(out, cp, strBlock.getBytesRef(strBlock.getFirstValueIndex(p), strScratch), lengthBlock.getInt(lengthBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public BytesRefVector eval(int positionCount, BytesRefVector strVector, IntVector lengthVector) { - try(BytesRefVector.Builder result = driverContext.blockFactory().newBytesRefVectorBuilder(positionCount)) { - BytesRef strScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - result.appendBytesRef(Left.process(out, cp, strVector.getBytesRef(p, strScratch), lengthVector.getInt(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "LeftEvaluator[" + "str=" + str + ", length=" + length + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(str, length); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final Function out; - - private final Function cp; - - private final EvalOperator.ExpressionEvaluator.Factory str; - - private final EvalOperator.ExpressionEvaluator.Factory length; - - public Factory(Source source, Function out, - Function cp, - EvalOperator.ExpressionEvaluator.Factory str, - EvalOperator.ExpressionEvaluator.Factory length) { - this.source = source; - this.out = out; - this.cp = cp; - this.str = str; - this.length = length; - } - - @Override - public LeftEvaluator get(DriverContext context) { - return new LeftEvaluator(source, out.apply(context), cp.apply(context), str.get(context), length.get(context), context); - } - - @Override - public String toString() { - return "LeftEvaluator[" + "str=" + str + ", length=" + length + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/LengthEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/LengthEvaluator.java deleted file mode 100644 index 890b56e78ca13..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/LengthEvaluator.java +++ /dev/null @@ -1,113 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.string; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Length}. - * This class is generated. Do not edit it. - */ -public final class LengthEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final DriverContext driverContext; - - public LengthEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock valBlock = (BytesRefBlock) val.eval(page)) { - BytesRefVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector).asBlock(); - } - } - - public IntBlock eval(int positionCount, BytesRefBlock valBlock) { - try(IntBlock.Builder result = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - BytesRef valScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendInt(Length.process(valBlock.getBytesRef(valBlock.getFirstValueIndex(p), valScratch))); - } - return result.build(); - } - } - - public IntVector eval(int positionCount, BytesRefVector valVector) { - try(IntVector.Builder result = driverContext.blockFactory().newIntVectorBuilder(positionCount)) { - BytesRef valScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - result.appendInt(Length.process(valVector.getBytesRef(p, valScratch))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "LengthEvaluator[" + "val=" + val + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val) { - this.source = source; - this.val = val; - } - - @Override - public LengthEvaluator get(DriverContext context) { - return new LengthEvaluator(source, val.get(context), context); - } - - @Override - public String toString() { - return "LengthEvaluator[" + "val=" + val + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/LocateEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/LocateEvaluator.java deleted file mode 100644 index 24055ad44f624..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/LocateEvaluator.java +++ /dev/null @@ -1,166 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.string; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Locate}. - * This class is generated. Do not edit it. - */ -public final class LocateEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator str; - - private final EvalOperator.ExpressionEvaluator substr; - - private final EvalOperator.ExpressionEvaluator start; - - private final DriverContext driverContext; - - public LocateEvaluator(Source source, EvalOperator.ExpressionEvaluator str, - EvalOperator.ExpressionEvaluator substr, EvalOperator.ExpressionEvaluator start, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.str = str; - this.substr = substr; - this.start = start; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock strBlock = (BytesRefBlock) str.eval(page)) { - try (BytesRefBlock substrBlock = (BytesRefBlock) substr.eval(page)) { - try (IntBlock startBlock = (IntBlock) start.eval(page)) { - BytesRefVector strVector = strBlock.asVector(); - if (strVector == null) { - return eval(page.getPositionCount(), strBlock, substrBlock, startBlock); - } - BytesRefVector substrVector = substrBlock.asVector(); - if (substrVector == null) { - return eval(page.getPositionCount(), strBlock, substrBlock, startBlock); - } - IntVector startVector = startBlock.asVector(); - if (startVector == null) { - return eval(page.getPositionCount(), strBlock, substrBlock, startBlock); - } - return eval(page.getPositionCount(), strVector, substrVector, startVector).asBlock(); - } - } - } - } - - public IntBlock eval(int positionCount, BytesRefBlock strBlock, BytesRefBlock substrBlock, - IntBlock startBlock) { - try(IntBlock.Builder result = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - BytesRef strScratch = new BytesRef(); - BytesRef substrScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (strBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (strBlock.getValueCount(p) != 1) { - if (strBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (substrBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (substrBlock.getValueCount(p) != 1) { - if (substrBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (startBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (startBlock.getValueCount(p) != 1) { - if (startBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendInt(Locate.process(strBlock.getBytesRef(strBlock.getFirstValueIndex(p), strScratch), substrBlock.getBytesRef(substrBlock.getFirstValueIndex(p), substrScratch), startBlock.getInt(startBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public IntVector eval(int positionCount, BytesRefVector strVector, BytesRefVector substrVector, - IntVector startVector) { - try(IntVector.Builder result = driverContext.blockFactory().newIntVectorBuilder(positionCount)) { - BytesRef strScratch = new BytesRef(); - BytesRef substrScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - result.appendInt(Locate.process(strVector.getBytesRef(p, strScratch), substrVector.getBytesRef(p, substrScratch), startVector.getInt(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "LocateEvaluator[" + "str=" + str + ", substr=" + substr + ", start=" + start + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(str, substr, start); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory str; - - private final EvalOperator.ExpressionEvaluator.Factory substr; - - private final EvalOperator.ExpressionEvaluator.Factory start; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory str, - EvalOperator.ExpressionEvaluator.Factory substr, - EvalOperator.ExpressionEvaluator.Factory start) { - this.source = source; - this.str = str; - this.substr = substr; - this.start = start; - } - - @Override - public LocateEvaluator get(DriverContext context) { - return new LocateEvaluator(source, str.get(context), substr.get(context), start.get(context), context); - } - - @Override - public String toString() { - return "LocateEvaluator[" + "str=" + str + ", substr=" + substr + ", start=" + start + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/LocateNoStartEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/LocateNoStartEvaluator.java deleted file mode 100644 index 947b1ecb49d0c..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/LocateNoStartEvaluator.java +++ /dev/null @@ -1,139 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.string; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Locate}. - * This class is generated. Do not edit it. - */ -public final class LocateNoStartEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator str; - - private final EvalOperator.ExpressionEvaluator substr; - - private final DriverContext driverContext; - - public LocateNoStartEvaluator(Source source, EvalOperator.ExpressionEvaluator str, - EvalOperator.ExpressionEvaluator substr, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.str = str; - this.substr = substr; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock strBlock = (BytesRefBlock) str.eval(page)) { - try (BytesRefBlock substrBlock = (BytesRefBlock) substr.eval(page)) { - BytesRefVector strVector = strBlock.asVector(); - if (strVector == null) { - return eval(page.getPositionCount(), strBlock, substrBlock); - } - BytesRefVector substrVector = substrBlock.asVector(); - if (substrVector == null) { - return eval(page.getPositionCount(), strBlock, substrBlock); - } - return eval(page.getPositionCount(), strVector, substrVector).asBlock(); - } - } - } - - public IntBlock eval(int positionCount, BytesRefBlock strBlock, BytesRefBlock substrBlock) { - try(IntBlock.Builder result = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - BytesRef strScratch = new BytesRef(); - BytesRef substrScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (strBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (strBlock.getValueCount(p) != 1) { - if (strBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (substrBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (substrBlock.getValueCount(p) != 1) { - if (substrBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendInt(Locate.process(strBlock.getBytesRef(strBlock.getFirstValueIndex(p), strScratch), substrBlock.getBytesRef(substrBlock.getFirstValueIndex(p), substrScratch))); - } - return result.build(); - } - } - - public IntVector eval(int positionCount, BytesRefVector strVector, BytesRefVector substrVector) { - try(IntVector.Builder result = driverContext.blockFactory().newIntVectorBuilder(positionCount)) { - BytesRef strScratch = new BytesRef(); - BytesRef substrScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - result.appendInt(Locate.process(strVector.getBytesRef(p, strScratch), substrVector.getBytesRef(p, substrScratch))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "LocateNoStartEvaluator[" + "str=" + str + ", substr=" + substr + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(str, substr); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory str; - - private final EvalOperator.ExpressionEvaluator.Factory substr; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory str, - EvalOperator.ExpressionEvaluator.Factory substr) { - this.source = source; - this.str = str; - this.substr = substr; - } - - @Override - public LocateNoStartEvaluator get(DriverContext context) { - return new LocateNoStartEvaluator(source, str.get(context), substr.get(context), context); - } - - @Override - public String toString() { - return "LocateNoStartEvaluator[" + "str=" + str + ", substr=" + substr + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/RTrimEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/RTrimEvaluator.java deleted file mode 100644 index fdd1c2a23357f..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/RTrimEvaluator.java +++ /dev/null @@ -1,111 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.string; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link RTrim}. - * This class is generated. Do not edit it. - */ -public final class RTrimEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final DriverContext driverContext; - - public RTrimEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock valBlock = (BytesRefBlock) val.eval(page)) { - BytesRefVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector).asBlock(); - } - } - - public BytesRefBlock eval(int positionCount, BytesRefBlock valBlock) { - try(BytesRefBlock.Builder result = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef valScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBytesRef(RTrim.process(valBlock.getBytesRef(valBlock.getFirstValueIndex(p), valScratch))); - } - return result.build(); - } - } - - public BytesRefVector eval(int positionCount, BytesRefVector valVector) { - try(BytesRefVector.Builder result = driverContext.blockFactory().newBytesRefVectorBuilder(positionCount)) { - BytesRef valScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - result.appendBytesRef(RTrim.process(valVector.getBytesRef(p, valScratch))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "RTrimEvaluator[" + "val=" + val + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val) { - this.source = source; - this.val = val; - } - - @Override - public RTrimEvaluator get(DriverContext context) { - return new RTrimEvaluator(source, val.get(context), context); - } - - @Override - public String toString() { - return "RTrimEvaluator[" + "val=" + val + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/ReplaceConstantEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/ReplaceConstantEvaluator.java deleted file mode 100644 index 71f8724d17a80..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/ReplaceConstantEvaluator.java +++ /dev/null @@ -1,156 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.string; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import java.util.regex.Pattern; -import java.util.regex.PatternSyntaxException; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Replace}. - * This class is generated. Do not edit it. - */ -public final class ReplaceConstantEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator str; - - private final Pattern regex; - - private final EvalOperator.ExpressionEvaluator newStr; - - private final DriverContext driverContext; - - public ReplaceConstantEvaluator(Source source, EvalOperator.ExpressionEvaluator str, - Pattern regex, EvalOperator.ExpressionEvaluator newStr, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.str = str; - this.regex = regex; - this.newStr = newStr; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock strBlock = (BytesRefBlock) str.eval(page)) { - try (BytesRefBlock newStrBlock = (BytesRefBlock) newStr.eval(page)) { - BytesRefVector strVector = strBlock.asVector(); - if (strVector == null) { - return eval(page.getPositionCount(), strBlock, newStrBlock); - } - BytesRefVector newStrVector = newStrBlock.asVector(); - if (newStrVector == null) { - return eval(page.getPositionCount(), strBlock, newStrBlock); - } - return eval(page.getPositionCount(), strVector, newStrVector); - } - } - } - - public BytesRefBlock eval(int positionCount, BytesRefBlock strBlock, BytesRefBlock newStrBlock) { - try(BytesRefBlock.Builder result = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef strScratch = new BytesRef(); - BytesRef newStrScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (strBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (strBlock.getValueCount(p) != 1) { - if (strBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (newStrBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (newStrBlock.getValueCount(p) != 1) { - if (newStrBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendBytesRef(Replace.process(strBlock.getBytesRef(strBlock.getFirstValueIndex(p), strScratch), regex, newStrBlock.getBytesRef(newStrBlock.getFirstValueIndex(p), newStrScratch))); - } catch (PatternSyntaxException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public BytesRefBlock eval(int positionCount, BytesRefVector strVector, - BytesRefVector newStrVector) { - try(BytesRefBlock.Builder result = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef strScratch = new BytesRef(); - BytesRef newStrScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendBytesRef(Replace.process(strVector.getBytesRef(p, strScratch), regex, newStrVector.getBytesRef(p, newStrScratch))); - } catch (PatternSyntaxException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "ReplaceConstantEvaluator[" + "str=" + str + ", regex=" + regex + ", newStr=" + newStr + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(str, newStr); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory str; - - private final Pattern regex; - - private final EvalOperator.ExpressionEvaluator.Factory newStr; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory str, Pattern regex, - EvalOperator.ExpressionEvaluator.Factory newStr) { - this.source = source; - this.str = str; - this.regex = regex; - this.newStr = newStr; - } - - @Override - public ReplaceConstantEvaluator get(DriverContext context) { - return new ReplaceConstantEvaluator(source, str.get(context), regex, newStr.get(context), context); - } - - @Override - public String toString() { - return "ReplaceConstantEvaluator[" + "str=" + str + ", regex=" + regex + ", newStr=" + newStr + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/ReplaceEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/ReplaceEvaluator.java deleted file mode 100644 index 8d4deb878f117..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/ReplaceEvaluator.java +++ /dev/null @@ -1,177 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.string; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import java.util.regex.PatternSyntaxException; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Replace}. - * This class is generated. Do not edit it. - */ -public final class ReplaceEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator str; - - private final EvalOperator.ExpressionEvaluator regex; - - private final EvalOperator.ExpressionEvaluator newStr; - - private final DriverContext driverContext; - - public ReplaceEvaluator(Source source, EvalOperator.ExpressionEvaluator str, - EvalOperator.ExpressionEvaluator regex, EvalOperator.ExpressionEvaluator newStr, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.str = str; - this.regex = regex; - this.newStr = newStr; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock strBlock = (BytesRefBlock) str.eval(page)) { - try (BytesRefBlock regexBlock = (BytesRefBlock) regex.eval(page)) { - try (BytesRefBlock newStrBlock = (BytesRefBlock) newStr.eval(page)) { - BytesRefVector strVector = strBlock.asVector(); - if (strVector == null) { - return eval(page.getPositionCount(), strBlock, regexBlock, newStrBlock); - } - BytesRefVector regexVector = regexBlock.asVector(); - if (regexVector == null) { - return eval(page.getPositionCount(), strBlock, regexBlock, newStrBlock); - } - BytesRefVector newStrVector = newStrBlock.asVector(); - if (newStrVector == null) { - return eval(page.getPositionCount(), strBlock, regexBlock, newStrBlock); - } - return eval(page.getPositionCount(), strVector, regexVector, newStrVector); - } - } - } - } - - public BytesRefBlock eval(int positionCount, BytesRefBlock strBlock, BytesRefBlock regexBlock, - BytesRefBlock newStrBlock) { - try(BytesRefBlock.Builder result = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef strScratch = new BytesRef(); - BytesRef regexScratch = new BytesRef(); - BytesRef newStrScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (strBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (strBlock.getValueCount(p) != 1) { - if (strBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (regexBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (regexBlock.getValueCount(p) != 1) { - if (regexBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (newStrBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (newStrBlock.getValueCount(p) != 1) { - if (newStrBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendBytesRef(Replace.process(strBlock.getBytesRef(strBlock.getFirstValueIndex(p), strScratch), regexBlock.getBytesRef(regexBlock.getFirstValueIndex(p), regexScratch), newStrBlock.getBytesRef(newStrBlock.getFirstValueIndex(p), newStrScratch))); - } catch (PatternSyntaxException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public BytesRefBlock eval(int positionCount, BytesRefVector strVector, BytesRefVector regexVector, - BytesRefVector newStrVector) { - try(BytesRefBlock.Builder result = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef strScratch = new BytesRef(); - BytesRef regexScratch = new BytesRef(); - BytesRef newStrScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendBytesRef(Replace.process(strVector.getBytesRef(p, strScratch), regexVector.getBytesRef(p, regexScratch), newStrVector.getBytesRef(p, newStrScratch))); - } catch (PatternSyntaxException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "ReplaceEvaluator[" + "str=" + str + ", regex=" + regex + ", newStr=" + newStr + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(str, regex, newStr); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory str; - - private final EvalOperator.ExpressionEvaluator.Factory regex; - - private final EvalOperator.ExpressionEvaluator.Factory newStr; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory str, - EvalOperator.ExpressionEvaluator.Factory regex, - EvalOperator.ExpressionEvaluator.Factory newStr) { - this.source = source; - this.str = str; - this.regex = regex; - this.newStr = newStr; - } - - @Override - public ReplaceEvaluator get(DriverContext context) { - return new ReplaceEvaluator(source, str.get(context), regex.get(context), newStr.get(context), context); - } - - @Override - public String toString() { - return "ReplaceEvaluator[" + "str=" + str + ", regex=" + regex + ", newStr=" + newStr + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/RightEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/RightEvaluator.java deleted file mode 100644 index 96473a2deefd2..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/RightEvaluator.java +++ /dev/null @@ -1,154 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.string; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import java.util.function.Function; -import org.apache.lucene.util.BytesRef; -import org.apache.lucene.util.UnicodeUtil; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Right}. - * This class is generated. Do not edit it. - */ -public final class RightEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final BytesRef out; - - private final UnicodeUtil.UTF8CodePoint cp; - - private final EvalOperator.ExpressionEvaluator str; - - private final EvalOperator.ExpressionEvaluator length; - - private final DriverContext driverContext; - - public RightEvaluator(Source source, BytesRef out, UnicodeUtil.UTF8CodePoint cp, - EvalOperator.ExpressionEvaluator str, EvalOperator.ExpressionEvaluator length, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.out = out; - this.cp = cp; - this.str = str; - this.length = length; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock strBlock = (BytesRefBlock) str.eval(page)) { - try (IntBlock lengthBlock = (IntBlock) length.eval(page)) { - BytesRefVector strVector = strBlock.asVector(); - if (strVector == null) { - return eval(page.getPositionCount(), strBlock, lengthBlock); - } - IntVector lengthVector = lengthBlock.asVector(); - if (lengthVector == null) { - return eval(page.getPositionCount(), strBlock, lengthBlock); - } - return eval(page.getPositionCount(), strVector, lengthVector).asBlock(); - } - } - } - - public BytesRefBlock eval(int positionCount, BytesRefBlock strBlock, IntBlock lengthBlock) { - try(BytesRefBlock.Builder result = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef strScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (strBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (strBlock.getValueCount(p) != 1) { - if (strBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (lengthBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lengthBlock.getValueCount(p) != 1) { - if (lengthBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBytesRef(Right.process(out, cp, strBlock.getBytesRef(strBlock.getFirstValueIndex(p), strScratch), lengthBlock.getInt(lengthBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public BytesRefVector eval(int positionCount, BytesRefVector strVector, IntVector lengthVector) { - try(BytesRefVector.Builder result = driverContext.blockFactory().newBytesRefVectorBuilder(positionCount)) { - BytesRef strScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - result.appendBytesRef(Right.process(out, cp, strVector.getBytesRef(p, strScratch), lengthVector.getInt(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "RightEvaluator[" + "str=" + str + ", length=" + length + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(str, length); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final Function out; - - private final Function cp; - - private final EvalOperator.ExpressionEvaluator.Factory str; - - private final EvalOperator.ExpressionEvaluator.Factory length; - - public Factory(Source source, Function out, - Function cp, - EvalOperator.ExpressionEvaluator.Factory str, - EvalOperator.ExpressionEvaluator.Factory length) { - this.source = source; - this.out = out; - this.cp = cp; - this.str = str; - this.length = length; - } - - @Override - public RightEvaluator get(DriverContext context) { - return new RightEvaluator(source, out.apply(context), cp.apply(context), str.get(context), length.get(context), context); - } - - @Override - public String toString() { - return "RightEvaluator[" + "str=" + str + ", length=" + length + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/SplitSingleByteEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/SplitSingleByteEvaluator.java deleted file mode 100644 index 7081f22606112..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/SplitSingleByteEvaluator.java +++ /dev/null @@ -1,125 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.string; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import java.util.function.Function; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Split}. - * This class is generated. Do not edit it. - */ -public final class SplitSingleByteEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator str; - - private final byte delim; - - private final BytesRef scratch; - - private final DriverContext driverContext; - - public SplitSingleByteEvaluator(Source source, EvalOperator.ExpressionEvaluator str, byte delim, - BytesRef scratch, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.str = str; - this.delim = delim; - this.scratch = scratch; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock strBlock = (BytesRefBlock) str.eval(page)) { - BytesRefVector strVector = strBlock.asVector(); - if (strVector == null) { - return eval(page.getPositionCount(), strBlock); - } - return eval(page.getPositionCount(), strVector); - } - } - - public BytesRefBlock eval(int positionCount, BytesRefBlock strBlock) { - try(BytesRefBlock.Builder result = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef strScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (strBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (strBlock.getValueCount(p) != 1) { - if (strBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - Split.process(result, strBlock.getBytesRef(strBlock.getFirstValueIndex(p), strScratch), delim, scratch); - } - return result.build(); - } - } - - public BytesRefBlock eval(int positionCount, BytesRefVector strVector) { - try(BytesRefBlock.Builder result = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef strScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - Split.process(result, strVector.getBytesRef(p, strScratch), delim, scratch); - } - return result.build(); - } - } - - @Override - public String toString() { - return "SplitSingleByteEvaluator[" + "str=" + str + ", delim=" + delim + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(str); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory str; - - private final byte delim; - - private final Function scratch; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory str, byte delim, - Function scratch) { - this.source = source; - this.str = str; - this.delim = delim; - this.scratch = scratch; - } - - @Override - public SplitSingleByteEvaluator get(DriverContext context) { - return new SplitSingleByteEvaluator(source, str.get(context), delim, scratch.apply(context), context); - } - - @Override - public String toString() { - return "SplitSingleByteEvaluator[" + "str=" + str + ", delim=" + delim + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/SplitVariableEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/SplitVariableEvaluator.java deleted file mode 100644 index 82feca1b79053..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/SplitVariableEvaluator.java +++ /dev/null @@ -1,145 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.string; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import java.util.function.Function; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Split}. - * This class is generated. Do not edit it. - */ -public final class SplitVariableEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator str; - - private final EvalOperator.ExpressionEvaluator delim; - - private final BytesRef scratch; - - private final DriverContext driverContext; - - public SplitVariableEvaluator(Source source, EvalOperator.ExpressionEvaluator str, - EvalOperator.ExpressionEvaluator delim, BytesRef scratch, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.str = str; - this.delim = delim; - this.scratch = scratch; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock strBlock = (BytesRefBlock) str.eval(page)) { - try (BytesRefBlock delimBlock = (BytesRefBlock) delim.eval(page)) { - BytesRefVector strVector = strBlock.asVector(); - if (strVector == null) { - return eval(page.getPositionCount(), strBlock, delimBlock); - } - BytesRefVector delimVector = delimBlock.asVector(); - if (delimVector == null) { - return eval(page.getPositionCount(), strBlock, delimBlock); - } - return eval(page.getPositionCount(), strVector, delimVector); - } - } - } - - public BytesRefBlock eval(int positionCount, BytesRefBlock strBlock, BytesRefBlock delimBlock) { - try(BytesRefBlock.Builder result = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef strScratch = new BytesRef(); - BytesRef delimScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (strBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (strBlock.getValueCount(p) != 1) { - if (strBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (delimBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (delimBlock.getValueCount(p) != 1) { - if (delimBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - Split.process(result, strBlock.getBytesRef(strBlock.getFirstValueIndex(p), strScratch), delimBlock.getBytesRef(delimBlock.getFirstValueIndex(p), delimScratch), scratch); - } - return result.build(); - } - } - - public BytesRefBlock eval(int positionCount, BytesRefVector strVector, - BytesRefVector delimVector) { - try(BytesRefBlock.Builder result = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef strScratch = new BytesRef(); - BytesRef delimScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - Split.process(result, strVector.getBytesRef(p, strScratch), delimVector.getBytesRef(p, delimScratch), scratch); - } - return result.build(); - } - } - - @Override - public String toString() { - return "SplitVariableEvaluator[" + "str=" + str + ", delim=" + delim + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(str, delim); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory str; - - private final EvalOperator.ExpressionEvaluator.Factory delim; - - private final Function scratch; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory str, - EvalOperator.ExpressionEvaluator.Factory delim, Function scratch) { - this.source = source; - this.str = str; - this.delim = delim; - this.scratch = scratch; - } - - @Override - public SplitVariableEvaluator get(DriverContext context) { - return new SplitVariableEvaluator(source, str.get(context), delim.get(context), scratch.apply(context), context); - } - - @Override - public String toString() { - return "SplitVariableEvaluator[" + "str=" + str + ", delim=" + delim + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/StartsWithEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/StartsWithEvaluator.java deleted file mode 100644 index 9eb1c488f52dd..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/StartsWithEvaluator.java +++ /dev/null @@ -1,140 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.string; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BooleanBlock; -import org.elasticsearch.compute.data.BooleanVector; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link StartsWith}. - * This class is generated. Do not edit it. - */ -public final class StartsWithEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator str; - - private final EvalOperator.ExpressionEvaluator prefix; - - private final DriverContext driverContext; - - public StartsWithEvaluator(Source source, EvalOperator.ExpressionEvaluator str, - EvalOperator.ExpressionEvaluator prefix, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.str = str; - this.prefix = prefix; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock strBlock = (BytesRefBlock) str.eval(page)) { - try (BytesRefBlock prefixBlock = (BytesRefBlock) prefix.eval(page)) { - BytesRefVector strVector = strBlock.asVector(); - if (strVector == null) { - return eval(page.getPositionCount(), strBlock, prefixBlock); - } - BytesRefVector prefixVector = prefixBlock.asVector(); - if (prefixVector == null) { - return eval(page.getPositionCount(), strBlock, prefixBlock); - } - return eval(page.getPositionCount(), strVector, prefixVector).asBlock(); - } - } - } - - public BooleanBlock eval(int positionCount, BytesRefBlock strBlock, BytesRefBlock prefixBlock) { - try(BooleanBlock.Builder result = driverContext.blockFactory().newBooleanBlockBuilder(positionCount)) { - BytesRef strScratch = new BytesRef(); - BytesRef prefixScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (strBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (strBlock.getValueCount(p) != 1) { - if (strBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (prefixBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (prefixBlock.getValueCount(p) != 1) { - if (prefixBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBoolean(StartsWith.process(strBlock.getBytesRef(strBlock.getFirstValueIndex(p), strScratch), prefixBlock.getBytesRef(prefixBlock.getFirstValueIndex(p), prefixScratch))); - } - return result.build(); - } - } - - public BooleanVector eval(int positionCount, BytesRefVector strVector, - BytesRefVector prefixVector) { - try(BooleanVector.Builder result = driverContext.blockFactory().newBooleanVectorBuilder(positionCount)) { - BytesRef strScratch = new BytesRef(); - BytesRef prefixScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - result.appendBoolean(StartsWith.process(strVector.getBytesRef(p, strScratch), prefixVector.getBytesRef(p, prefixScratch))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "StartsWithEvaluator[" + "str=" + str + ", prefix=" + prefix + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(str, prefix); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory str; - - private final EvalOperator.ExpressionEvaluator.Factory prefix; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory str, - EvalOperator.ExpressionEvaluator.Factory prefix) { - this.source = source; - this.str = str; - this.prefix = prefix; - } - - @Override - public StartsWithEvaluator get(DriverContext context) { - return new StartsWithEvaluator(source, str.get(context), prefix.get(context), context); - } - - @Override - public String toString() { - return "StartsWithEvaluator[" + "str=" + str + ", prefix=" + prefix + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/SubstringEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/SubstringEvaluator.java deleted file mode 100644 index 9da104137ba94..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/SubstringEvaluator.java +++ /dev/null @@ -1,164 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.string; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Substring}. - * This class is generated. Do not edit it. - */ -public final class SubstringEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator str; - - private final EvalOperator.ExpressionEvaluator start; - - private final EvalOperator.ExpressionEvaluator length; - - private final DriverContext driverContext; - - public SubstringEvaluator(Source source, EvalOperator.ExpressionEvaluator str, - EvalOperator.ExpressionEvaluator start, EvalOperator.ExpressionEvaluator length, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.str = str; - this.start = start; - this.length = length; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock strBlock = (BytesRefBlock) str.eval(page)) { - try (IntBlock startBlock = (IntBlock) start.eval(page)) { - try (IntBlock lengthBlock = (IntBlock) length.eval(page)) { - BytesRefVector strVector = strBlock.asVector(); - if (strVector == null) { - return eval(page.getPositionCount(), strBlock, startBlock, lengthBlock); - } - IntVector startVector = startBlock.asVector(); - if (startVector == null) { - return eval(page.getPositionCount(), strBlock, startBlock, lengthBlock); - } - IntVector lengthVector = lengthBlock.asVector(); - if (lengthVector == null) { - return eval(page.getPositionCount(), strBlock, startBlock, lengthBlock); - } - return eval(page.getPositionCount(), strVector, startVector, lengthVector).asBlock(); - } - } - } - } - - public BytesRefBlock eval(int positionCount, BytesRefBlock strBlock, IntBlock startBlock, - IntBlock lengthBlock) { - try(BytesRefBlock.Builder result = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef strScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (strBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (strBlock.getValueCount(p) != 1) { - if (strBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (startBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (startBlock.getValueCount(p) != 1) { - if (startBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (lengthBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lengthBlock.getValueCount(p) != 1) { - if (lengthBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBytesRef(Substring.process(strBlock.getBytesRef(strBlock.getFirstValueIndex(p), strScratch), startBlock.getInt(startBlock.getFirstValueIndex(p)), lengthBlock.getInt(lengthBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public BytesRefVector eval(int positionCount, BytesRefVector strVector, IntVector startVector, - IntVector lengthVector) { - try(BytesRefVector.Builder result = driverContext.blockFactory().newBytesRefVectorBuilder(positionCount)) { - BytesRef strScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - result.appendBytesRef(Substring.process(strVector.getBytesRef(p, strScratch), startVector.getInt(p), lengthVector.getInt(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "SubstringEvaluator[" + "str=" + str + ", start=" + start + ", length=" + length + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(str, start, length); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory str; - - private final EvalOperator.ExpressionEvaluator.Factory start; - - private final EvalOperator.ExpressionEvaluator.Factory length; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory str, - EvalOperator.ExpressionEvaluator.Factory start, - EvalOperator.ExpressionEvaluator.Factory length) { - this.source = source; - this.str = str; - this.start = start; - this.length = length; - } - - @Override - public SubstringEvaluator get(DriverContext context) { - return new SubstringEvaluator(source, str.get(context), start.get(context), length.get(context), context); - } - - @Override - public String toString() { - return "SubstringEvaluator[" + "str=" + str + ", start=" + start + ", length=" + length + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/SubstringNoLengthEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/SubstringNoLengthEvaluator.java deleted file mode 100644 index 08d12ac049837..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/SubstringNoLengthEvaluator.java +++ /dev/null @@ -1,137 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.string; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Substring}. - * This class is generated. Do not edit it. - */ -public final class SubstringNoLengthEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator str; - - private final EvalOperator.ExpressionEvaluator start; - - private final DriverContext driverContext; - - public SubstringNoLengthEvaluator(Source source, EvalOperator.ExpressionEvaluator str, - EvalOperator.ExpressionEvaluator start, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.str = str; - this.start = start; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock strBlock = (BytesRefBlock) str.eval(page)) { - try (IntBlock startBlock = (IntBlock) start.eval(page)) { - BytesRefVector strVector = strBlock.asVector(); - if (strVector == null) { - return eval(page.getPositionCount(), strBlock, startBlock); - } - IntVector startVector = startBlock.asVector(); - if (startVector == null) { - return eval(page.getPositionCount(), strBlock, startBlock); - } - return eval(page.getPositionCount(), strVector, startVector).asBlock(); - } - } - } - - public BytesRefBlock eval(int positionCount, BytesRefBlock strBlock, IntBlock startBlock) { - try(BytesRefBlock.Builder result = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef strScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (strBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (strBlock.getValueCount(p) != 1) { - if (strBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (startBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (startBlock.getValueCount(p) != 1) { - if (startBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBytesRef(Substring.process(strBlock.getBytesRef(strBlock.getFirstValueIndex(p), strScratch), startBlock.getInt(startBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public BytesRefVector eval(int positionCount, BytesRefVector strVector, IntVector startVector) { - try(BytesRefVector.Builder result = driverContext.blockFactory().newBytesRefVectorBuilder(positionCount)) { - BytesRef strScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - result.appendBytesRef(Substring.process(strVector.getBytesRef(p, strScratch), startVector.getInt(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "SubstringNoLengthEvaluator[" + "str=" + str + ", start=" + start + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(str, start); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory str; - - private final EvalOperator.ExpressionEvaluator.Factory start; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory str, - EvalOperator.ExpressionEvaluator.Factory start) { - this.source = source; - this.str = str; - this.start = start; - } - - @Override - public SubstringNoLengthEvaluator get(DriverContext context) { - return new SubstringNoLengthEvaluator(source, str.get(context), start.get(context), context); - } - - @Override - public String toString() { - return "SubstringNoLengthEvaluator[" + "str=" + str + ", start=" + start + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/ToLowerEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/ToLowerEvaluator.java deleted file mode 100644 index 23f28385916c7..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/ToLowerEvaluator.java +++ /dev/null @@ -1,118 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.string; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import java.util.Locale; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToLower}. - * This class is generated. Do not edit it. - */ -public final class ToLowerEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final Locale locale; - - private final DriverContext driverContext; - - public ToLowerEvaluator(Source source, EvalOperator.ExpressionEvaluator val, Locale locale, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.locale = locale; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock valBlock = (BytesRefBlock) val.eval(page)) { - BytesRefVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector).asBlock(); - } - } - - public BytesRefBlock eval(int positionCount, BytesRefBlock valBlock) { - try(BytesRefBlock.Builder result = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef valScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBytesRef(ToLower.process(valBlock.getBytesRef(valBlock.getFirstValueIndex(p), valScratch), locale)); - } - return result.build(); - } - } - - public BytesRefVector eval(int positionCount, BytesRefVector valVector) { - try(BytesRefVector.Builder result = driverContext.blockFactory().newBytesRefVectorBuilder(positionCount)) { - BytesRef valScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - result.appendBytesRef(ToLower.process(valVector.getBytesRef(p, valScratch), locale)); - } - return result.build(); - } - } - - @Override - public String toString() { - return "ToLowerEvaluator[" + "val=" + val + ", locale=" + locale + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - private final Locale locale; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val, Locale locale) { - this.source = source; - this.val = val; - this.locale = locale; - } - - @Override - public ToLowerEvaluator get(DriverContext context) { - return new ToLowerEvaluator(source, val.get(context), locale, context); - } - - @Override - public String toString() { - return "ToLowerEvaluator[" + "val=" + val + ", locale=" + locale + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/ToUpperEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/ToUpperEvaluator.java deleted file mode 100644 index 5c3e86184d460..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/ToUpperEvaluator.java +++ /dev/null @@ -1,118 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.string; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import java.util.Locale; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link ToUpper}. - * This class is generated. Do not edit it. - */ -public final class ToUpperEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final Locale locale; - - private final DriverContext driverContext; - - public ToUpperEvaluator(Source source, EvalOperator.ExpressionEvaluator val, Locale locale, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.locale = locale; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock valBlock = (BytesRefBlock) val.eval(page)) { - BytesRefVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector).asBlock(); - } - } - - public BytesRefBlock eval(int positionCount, BytesRefBlock valBlock) { - try(BytesRefBlock.Builder result = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef valScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBytesRef(ToUpper.process(valBlock.getBytesRef(valBlock.getFirstValueIndex(p), valScratch), locale)); - } - return result.build(); - } - } - - public BytesRefVector eval(int positionCount, BytesRefVector valVector) { - try(BytesRefVector.Builder result = driverContext.blockFactory().newBytesRefVectorBuilder(positionCount)) { - BytesRef valScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - result.appendBytesRef(ToUpper.process(valVector.getBytesRef(p, valScratch), locale)); - } - return result.build(); - } - } - - @Override - public String toString() { - return "ToUpperEvaluator[" + "val=" + val + ", locale=" + locale + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - private final Locale locale; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val, Locale locale) { - this.source = source; - this.val = val; - this.locale = locale; - } - - @Override - public ToUpperEvaluator get(DriverContext context) { - return new ToUpperEvaluator(source, val.get(context), locale, context); - } - - @Override - public String toString() { - return "ToUpperEvaluator[" + "val=" + val + ", locale=" + locale + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/TrimEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/TrimEvaluator.java deleted file mode 100644 index 1ecb6b3bd578f..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/function/scalar/string/TrimEvaluator.java +++ /dev/null @@ -1,111 +0,0 @@ -// 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.xpack.esql.expression.function.scalar.string; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.apache.lucene.util.BytesRef; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.BytesRefBlock; -import org.elasticsearch.compute.data.BytesRefVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Trim}. - * This class is generated. Do not edit it. - */ -public final class TrimEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator val; - - private final DriverContext driverContext; - - public TrimEvaluator(Source source, EvalOperator.ExpressionEvaluator val, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.val = val; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (BytesRefBlock valBlock = (BytesRefBlock) val.eval(page)) { - BytesRefVector valVector = valBlock.asVector(); - if (valVector == null) { - return eval(page.getPositionCount(), valBlock); - } - return eval(page.getPositionCount(), valVector).asBlock(); - } - } - - public BytesRefBlock eval(int positionCount, BytesRefBlock valBlock) { - try(BytesRefBlock.Builder result = driverContext.blockFactory().newBytesRefBlockBuilder(positionCount)) { - BytesRef valScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - if (valBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (valBlock.getValueCount(p) != 1) { - if (valBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendBytesRef(Trim.process(valBlock.getBytesRef(valBlock.getFirstValueIndex(p), valScratch))); - } - return result.build(); - } - } - - public BytesRefVector eval(int positionCount, BytesRefVector valVector) { - try(BytesRefVector.Builder result = driverContext.blockFactory().newBytesRefVectorBuilder(positionCount)) { - BytesRef valScratch = new BytesRef(); - position: for (int p = 0; p < positionCount; p++) { - result.appendBytesRef(Trim.process(valVector.getBytesRef(p, valScratch))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "TrimEvaluator[" + "val=" + val + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(val); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory val; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory val) { - this.source = source; - this.val = val; - } - - @Override - public TrimEvaluator get(DriverContext context) { - return new TrimEvaluator(source, val.get(context), context); - } - - @Override - public String toString() { - return "TrimEvaluator[" + "val=" + val + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/AddDatetimesEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/AddDatetimesEvaluator.java deleted file mode 100644 index 04b433ecde34a..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/AddDatetimesEvaluator.java +++ /dev/null @@ -1,128 +0,0 @@ -// 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.xpack.esql.expression.predicate.operator.arithmetic; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import java.time.DateTimeException; -import java.time.temporal.TemporalAmount; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Add}. - * This class is generated. Do not edit it. - */ -public final class AddDatetimesEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator datetime; - - private final TemporalAmount temporalAmount; - - private final DriverContext driverContext; - - public AddDatetimesEvaluator(Source source, EvalOperator.ExpressionEvaluator datetime, - TemporalAmount temporalAmount, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.datetime = datetime; - this.temporalAmount = temporalAmount; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock datetimeBlock = (LongBlock) datetime.eval(page)) { - LongVector datetimeVector = datetimeBlock.asVector(); - if (datetimeVector == null) { - return eval(page.getPositionCount(), datetimeBlock); - } - return eval(page.getPositionCount(), datetimeVector); - } - } - - public LongBlock eval(int positionCount, LongBlock datetimeBlock) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (datetimeBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (datetimeBlock.getValueCount(p) != 1) { - if (datetimeBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendLong(Add.processDatetimes(datetimeBlock.getLong(datetimeBlock.getFirstValueIndex(p)), temporalAmount)); - } catch (ArithmeticException | DateTimeException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public LongBlock eval(int positionCount, LongVector datetimeVector) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendLong(Add.processDatetimes(datetimeVector.getLong(p), temporalAmount)); - } catch (ArithmeticException | DateTimeException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "AddDatetimesEvaluator[" + "datetime=" + datetime + ", temporalAmount=" + temporalAmount + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(datetime); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory datetime; - - private final TemporalAmount temporalAmount; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory datetime, - TemporalAmount temporalAmount) { - this.source = source; - this.datetime = datetime; - this.temporalAmount = temporalAmount; - } - - @Override - public AddDatetimesEvaluator get(DriverContext context) { - return new AddDatetimesEvaluator(source, datetime.get(context), temporalAmount, context); - } - - @Override - public String toString() { - return "AddDatetimesEvaluator[" + "datetime=" + datetime + ", temporalAmount=" + temporalAmount + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/AddDoublesEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/AddDoublesEvaluator.java deleted file mode 100644 index 071369c29f333..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/AddDoublesEvaluator.java +++ /dev/null @@ -1,132 +0,0 @@ -// 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.xpack.esql.expression.predicate.operator.arithmetic; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Add}. - * This class is generated. Do not edit it. - */ -public final class AddDoublesEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public AddDoublesEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock lhsBlock = (DoubleBlock) lhs.eval(page)) { - try (DoubleBlock rhsBlock = (DoubleBlock) rhs.eval(page)) { - DoubleVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - DoubleVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public DoubleBlock eval(int positionCount, DoubleBlock lhsBlock, DoubleBlock rhsBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendDouble(Add.processDoubles(lhsBlock.getDouble(lhsBlock.getFirstValueIndex(p)), rhsBlock.getDouble(rhsBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public DoubleVector eval(int positionCount, DoubleVector lhsVector, DoubleVector rhsVector) { - try(DoubleVector.Builder result = driverContext.blockFactory().newDoubleVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendDouble(Add.processDoubles(lhsVector.getDouble(p), rhsVector.getDouble(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "AddDoublesEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public AddDoublesEvaluator get(DriverContext context) { - return new AddDoublesEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "AddDoublesEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/AddIntsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/AddIntsEvaluator.java deleted file mode 100644 index bf9157540ea55..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/AddIntsEvaluator.java +++ /dev/null @@ -1,143 +0,0 @@ -// 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.xpack.esql.expression.predicate.operator.arithmetic; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Add}. - * This class is generated. Do not edit it. - */ -public final class AddIntsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public AddIntsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (IntBlock lhsBlock = (IntBlock) lhs.eval(page)) { - try (IntBlock rhsBlock = (IntBlock) rhs.eval(page)) { - IntVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - IntVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector); - } - } - } - - public IntBlock eval(int positionCount, IntBlock lhsBlock, IntBlock rhsBlock) { - try(IntBlock.Builder result = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendInt(Add.processInts(lhsBlock.getInt(lhsBlock.getFirstValueIndex(p)), rhsBlock.getInt(rhsBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public IntBlock eval(int positionCount, IntVector lhsVector, IntVector rhsVector) { - try(IntBlock.Builder result = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendInt(Add.processInts(lhsVector.getInt(p), rhsVector.getInt(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "AddIntsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public AddIntsEvaluator get(DriverContext context) { - return new AddIntsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "AddIntsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/AddLongsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/AddLongsEvaluator.java deleted file mode 100644 index 51199df88fb9b..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/AddLongsEvaluator.java +++ /dev/null @@ -1,143 +0,0 @@ -// 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.xpack.esql.expression.predicate.operator.arithmetic; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Add}. - * This class is generated. Do not edit it. - */ -public final class AddLongsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public AddLongsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock lhsBlock = (LongBlock) lhs.eval(page)) { - try (LongBlock rhsBlock = (LongBlock) rhs.eval(page)) { - LongVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - LongVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector); - } - } - } - - public LongBlock eval(int positionCount, LongBlock lhsBlock, LongBlock rhsBlock) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendLong(Add.processLongs(lhsBlock.getLong(lhsBlock.getFirstValueIndex(p)), rhsBlock.getLong(rhsBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public LongBlock eval(int positionCount, LongVector lhsVector, LongVector rhsVector) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendLong(Add.processLongs(lhsVector.getLong(p), rhsVector.getLong(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "AddLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public AddLongsEvaluator get(DriverContext context) { - return new AddLongsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "AddLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/AddUnsignedLongsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/AddUnsignedLongsEvaluator.java deleted file mode 100644 index 10b21fb5898e8..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/AddUnsignedLongsEvaluator.java +++ /dev/null @@ -1,143 +0,0 @@ -// 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.xpack.esql.expression.predicate.operator.arithmetic; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Add}. - * This class is generated. Do not edit it. - */ -public final class AddUnsignedLongsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public AddUnsignedLongsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock lhsBlock = (LongBlock) lhs.eval(page)) { - try (LongBlock rhsBlock = (LongBlock) rhs.eval(page)) { - LongVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - LongVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector); - } - } - } - - public LongBlock eval(int positionCount, LongBlock lhsBlock, LongBlock rhsBlock) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendLong(Add.processUnsignedLongs(lhsBlock.getLong(lhsBlock.getFirstValueIndex(p)), rhsBlock.getLong(rhsBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public LongBlock eval(int positionCount, LongVector lhsVector, LongVector rhsVector) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendLong(Add.processUnsignedLongs(lhsVector.getLong(p), rhsVector.getLong(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "AddUnsignedLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public AddUnsignedLongsEvaluator get(DriverContext context) { - return new AddUnsignedLongsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "AddUnsignedLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/DivDoublesEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/DivDoublesEvaluator.java deleted file mode 100644 index bb9f55f2b5b85..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/DivDoublesEvaluator.java +++ /dev/null @@ -1,132 +0,0 @@ -// 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.xpack.esql.expression.predicate.operator.arithmetic; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Div}. - * This class is generated. Do not edit it. - */ -public final class DivDoublesEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public DivDoublesEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock lhsBlock = (DoubleBlock) lhs.eval(page)) { - try (DoubleBlock rhsBlock = (DoubleBlock) rhs.eval(page)) { - DoubleVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - DoubleVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public DoubleBlock eval(int positionCount, DoubleBlock lhsBlock, DoubleBlock rhsBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendDouble(Div.processDoubles(lhsBlock.getDouble(lhsBlock.getFirstValueIndex(p)), rhsBlock.getDouble(rhsBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public DoubleVector eval(int positionCount, DoubleVector lhsVector, DoubleVector rhsVector) { - try(DoubleVector.Builder result = driverContext.blockFactory().newDoubleVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendDouble(Div.processDoubles(lhsVector.getDouble(p), rhsVector.getDouble(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "DivDoublesEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public DivDoublesEvaluator get(DriverContext context) { - return new DivDoublesEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "DivDoublesEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/DivIntsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/DivIntsEvaluator.java deleted file mode 100644 index de3fb03fe4405..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/DivIntsEvaluator.java +++ /dev/null @@ -1,143 +0,0 @@ -// 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.xpack.esql.expression.predicate.operator.arithmetic; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Div}. - * This class is generated. Do not edit it. - */ -public final class DivIntsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public DivIntsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (IntBlock lhsBlock = (IntBlock) lhs.eval(page)) { - try (IntBlock rhsBlock = (IntBlock) rhs.eval(page)) { - IntVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - IntVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector); - } - } - } - - public IntBlock eval(int positionCount, IntBlock lhsBlock, IntBlock rhsBlock) { - try(IntBlock.Builder result = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendInt(Div.processInts(lhsBlock.getInt(lhsBlock.getFirstValueIndex(p)), rhsBlock.getInt(rhsBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public IntBlock eval(int positionCount, IntVector lhsVector, IntVector rhsVector) { - try(IntBlock.Builder result = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendInt(Div.processInts(lhsVector.getInt(p), rhsVector.getInt(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "DivIntsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public DivIntsEvaluator get(DriverContext context) { - return new DivIntsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "DivIntsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/DivLongsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/DivLongsEvaluator.java deleted file mode 100644 index 9eb02cbd47614..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/DivLongsEvaluator.java +++ /dev/null @@ -1,143 +0,0 @@ -// 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.xpack.esql.expression.predicate.operator.arithmetic; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Div}. - * This class is generated. Do not edit it. - */ -public final class DivLongsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public DivLongsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock lhsBlock = (LongBlock) lhs.eval(page)) { - try (LongBlock rhsBlock = (LongBlock) rhs.eval(page)) { - LongVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - LongVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector); - } - } - } - - public LongBlock eval(int positionCount, LongBlock lhsBlock, LongBlock rhsBlock) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendLong(Div.processLongs(lhsBlock.getLong(lhsBlock.getFirstValueIndex(p)), rhsBlock.getLong(rhsBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public LongBlock eval(int positionCount, LongVector lhsVector, LongVector rhsVector) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendLong(Div.processLongs(lhsVector.getLong(p), rhsVector.getLong(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "DivLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public DivLongsEvaluator get(DriverContext context) { - return new DivLongsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "DivLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/DivUnsignedLongsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/DivUnsignedLongsEvaluator.java deleted file mode 100644 index 50e3c933fec41..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/DivUnsignedLongsEvaluator.java +++ /dev/null @@ -1,143 +0,0 @@ -// 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.xpack.esql.expression.predicate.operator.arithmetic; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Div}. - * This class is generated. Do not edit it. - */ -public final class DivUnsignedLongsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public DivUnsignedLongsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock lhsBlock = (LongBlock) lhs.eval(page)) { - try (LongBlock rhsBlock = (LongBlock) rhs.eval(page)) { - LongVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - LongVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector); - } - } - } - - public LongBlock eval(int positionCount, LongBlock lhsBlock, LongBlock rhsBlock) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendLong(Div.processUnsignedLongs(lhsBlock.getLong(lhsBlock.getFirstValueIndex(p)), rhsBlock.getLong(rhsBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public LongBlock eval(int positionCount, LongVector lhsVector, LongVector rhsVector) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendLong(Div.processUnsignedLongs(lhsVector.getLong(p), rhsVector.getLong(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "DivUnsignedLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public DivUnsignedLongsEvaluator get(DriverContext context) { - return new DivUnsignedLongsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "DivUnsignedLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/ModDoublesEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/ModDoublesEvaluator.java deleted file mode 100644 index 8d441ffe10a48..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/ModDoublesEvaluator.java +++ /dev/null @@ -1,132 +0,0 @@ -// 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.xpack.esql.expression.predicate.operator.arithmetic; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Mod}. - * This class is generated. Do not edit it. - */ -public final class ModDoublesEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public ModDoublesEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock lhsBlock = (DoubleBlock) lhs.eval(page)) { - try (DoubleBlock rhsBlock = (DoubleBlock) rhs.eval(page)) { - DoubleVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - DoubleVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public DoubleBlock eval(int positionCount, DoubleBlock lhsBlock, DoubleBlock rhsBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendDouble(Mod.processDoubles(lhsBlock.getDouble(lhsBlock.getFirstValueIndex(p)), rhsBlock.getDouble(rhsBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public DoubleVector eval(int positionCount, DoubleVector lhsVector, DoubleVector rhsVector) { - try(DoubleVector.Builder result = driverContext.blockFactory().newDoubleVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendDouble(Mod.processDoubles(lhsVector.getDouble(p), rhsVector.getDouble(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "ModDoublesEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public ModDoublesEvaluator get(DriverContext context) { - return new ModDoublesEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "ModDoublesEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/ModIntsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/ModIntsEvaluator.java deleted file mode 100644 index c2c44dba5207d..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/ModIntsEvaluator.java +++ /dev/null @@ -1,143 +0,0 @@ -// 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.xpack.esql.expression.predicate.operator.arithmetic; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Mod}. - * This class is generated. Do not edit it. - */ -public final class ModIntsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public ModIntsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (IntBlock lhsBlock = (IntBlock) lhs.eval(page)) { - try (IntBlock rhsBlock = (IntBlock) rhs.eval(page)) { - IntVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - IntVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector); - } - } - } - - public IntBlock eval(int positionCount, IntBlock lhsBlock, IntBlock rhsBlock) { - try(IntBlock.Builder result = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendInt(Mod.processInts(lhsBlock.getInt(lhsBlock.getFirstValueIndex(p)), rhsBlock.getInt(rhsBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public IntBlock eval(int positionCount, IntVector lhsVector, IntVector rhsVector) { - try(IntBlock.Builder result = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendInt(Mod.processInts(lhsVector.getInt(p), rhsVector.getInt(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "ModIntsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public ModIntsEvaluator get(DriverContext context) { - return new ModIntsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "ModIntsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/ModLongsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/ModLongsEvaluator.java deleted file mode 100644 index 58b3f055db6b2..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/ModLongsEvaluator.java +++ /dev/null @@ -1,143 +0,0 @@ -// 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.xpack.esql.expression.predicate.operator.arithmetic; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Mod}. - * This class is generated. Do not edit it. - */ -public final class ModLongsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public ModLongsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock lhsBlock = (LongBlock) lhs.eval(page)) { - try (LongBlock rhsBlock = (LongBlock) rhs.eval(page)) { - LongVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - LongVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector); - } - } - } - - public LongBlock eval(int positionCount, LongBlock lhsBlock, LongBlock rhsBlock) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendLong(Mod.processLongs(lhsBlock.getLong(lhsBlock.getFirstValueIndex(p)), rhsBlock.getLong(rhsBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public LongBlock eval(int positionCount, LongVector lhsVector, LongVector rhsVector) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendLong(Mod.processLongs(lhsVector.getLong(p), rhsVector.getLong(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "ModLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public ModLongsEvaluator get(DriverContext context) { - return new ModLongsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "ModLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/ModUnsignedLongsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/ModUnsignedLongsEvaluator.java deleted file mode 100644 index 5b79aa8653923..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/ModUnsignedLongsEvaluator.java +++ /dev/null @@ -1,143 +0,0 @@ -// 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.xpack.esql.expression.predicate.operator.arithmetic; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Mod}. - * This class is generated. Do not edit it. - */ -public final class ModUnsignedLongsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public ModUnsignedLongsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock lhsBlock = (LongBlock) lhs.eval(page)) { - try (LongBlock rhsBlock = (LongBlock) rhs.eval(page)) { - LongVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - LongVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector); - } - } - } - - public LongBlock eval(int positionCount, LongBlock lhsBlock, LongBlock rhsBlock) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendLong(Mod.processUnsignedLongs(lhsBlock.getLong(lhsBlock.getFirstValueIndex(p)), rhsBlock.getLong(rhsBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public LongBlock eval(int positionCount, LongVector lhsVector, LongVector rhsVector) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendLong(Mod.processUnsignedLongs(lhsVector.getLong(p), rhsVector.getLong(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "ModUnsignedLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public ModUnsignedLongsEvaluator get(DriverContext context) { - return new ModUnsignedLongsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "ModUnsignedLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/MulDoublesEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/MulDoublesEvaluator.java deleted file mode 100644 index 1b9d10bff58e9..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/MulDoublesEvaluator.java +++ /dev/null @@ -1,132 +0,0 @@ -// 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.xpack.esql.expression.predicate.operator.arithmetic; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Mul}. - * This class is generated. Do not edit it. - */ -public final class MulDoublesEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public MulDoublesEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock lhsBlock = (DoubleBlock) lhs.eval(page)) { - try (DoubleBlock rhsBlock = (DoubleBlock) rhs.eval(page)) { - DoubleVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - DoubleVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public DoubleBlock eval(int positionCount, DoubleBlock lhsBlock, DoubleBlock rhsBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendDouble(Mul.processDoubles(lhsBlock.getDouble(lhsBlock.getFirstValueIndex(p)), rhsBlock.getDouble(rhsBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public DoubleVector eval(int positionCount, DoubleVector lhsVector, DoubleVector rhsVector) { - try(DoubleVector.Builder result = driverContext.blockFactory().newDoubleVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendDouble(Mul.processDoubles(lhsVector.getDouble(p), rhsVector.getDouble(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "MulDoublesEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public MulDoublesEvaluator get(DriverContext context) { - return new MulDoublesEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "MulDoublesEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/MulIntsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/MulIntsEvaluator.java deleted file mode 100644 index 7501d0fc505a1..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/MulIntsEvaluator.java +++ /dev/null @@ -1,143 +0,0 @@ -// 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.xpack.esql.expression.predicate.operator.arithmetic; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Mul}. - * This class is generated. Do not edit it. - */ -public final class MulIntsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public MulIntsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (IntBlock lhsBlock = (IntBlock) lhs.eval(page)) { - try (IntBlock rhsBlock = (IntBlock) rhs.eval(page)) { - IntVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - IntVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector); - } - } - } - - public IntBlock eval(int positionCount, IntBlock lhsBlock, IntBlock rhsBlock) { - try(IntBlock.Builder result = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendInt(Mul.processInts(lhsBlock.getInt(lhsBlock.getFirstValueIndex(p)), rhsBlock.getInt(rhsBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public IntBlock eval(int positionCount, IntVector lhsVector, IntVector rhsVector) { - try(IntBlock.Builder result = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendInt(Mul.processInts(lhsVector.getInt(p), rhsVector.getInt(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "MulIntsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public MulIntsEvaluator get(DriverContext context) { - return new MulIntsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "MulIntsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/MulLongsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/MulLongsEvaluator.java deleted file mode 100644 index 383e55755917d..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/MulLongsEvaluator.java +++ /dev/null @@ -1,143 +0,0 @@ -// 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.xpack.esql.expression.predicate.operator.arithmetic; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Mul}. - * This class is generated. Do not edit it. - */ -public final class MulLongsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public MulLongsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock lhsBlock = (LongBlock) lhs.eval(page)) { - try (LongBlock rhsBlock = (LongBlock) rhs.eval(page)) { - LongVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - LongVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector); - } - } - } - - public LongBlock eval(int positionCount, LongBlock lhsBlock, LongBlock rhsBlock) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendLong(Mul.processLongs(lhsBlock.getLong(lhsBlock.getFirstValueIndex(p)), rhsBlock.getLong(rhsBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public LongBlock eval(int positionCount, LongVector lhsVector, LongVector rhsVector) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendLong(Mul.processLongs(lhsVector.getLong(p), rhsVector.getLong(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "MulLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public MulLongsEvaluator get(DriverContext context) { - return new MulLongsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "MulLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/MulUnsignedLongsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/MulUnsignedLongsEvaluator.java deleted file mode 100644 index 95ecaee6b34ac..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/MulUnsignedLongsEvaluator.java +++ /dev/null @@ -1,143 +0,0 @@ -// 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.xpack.esql.expression.predicate.operator.arithmetic; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Mul}. - * This class is generated. Do not edit it. - */ -public final class MulUnsignedLongsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public MulUnsignedLongsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock lhsBlock = (LongBlock) lhs.eval(page)) { - try (LongBlock rhsBlock = (LongBlock) rhs.eval(page)) { - LongVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - LongVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector); - } - } - } - - public LongBlock eval(int positionCount, LongBlock lhsBlock, LongBlock rhsBlock) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendLong(Mul.processUnsignedLongs(lhsBlock.getLong(lhsBlock.getFirstValueIndex(p)), rhsBlock.getLong(rhsBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public LongBlock eval(int positionCount, LongVector lhsVector, LongVector rhsVector) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendLong(Mul.processUnsignedLongs(lhsVector.getLong(p), rhsVector.getLong(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "MulUnsignedLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public MulUnsignedLongsEvaluator get(DriverContext context) { - return new MulUnsignedLongsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "MulUnsignedLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/NegDoublesEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/NegDoublesEvaluator.java deleted file mode 100644 index 5915d4d476f19..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/NegDoublesEvaluator.java +++ /dev/null @@ -1,108 +0,0 @@ -// 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.xpack.esql.expression.predicate.operator.arithmetic; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Neg}. - * This class is generated. Do not edit it. - */ -public final class NegDoublesEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator v; - - private final DriverContext driverContext; - - public NegDoublesEvaluator(Source source, EvalOperator.ExpressionEvaluator v, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.v = v; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock vBlock = (DoubleBlock) v.eval(page)) { - DoubleVector vVector = vBlock.asVector(); - if (vVector == null) { - return eval(page.getPositionCount(), vBlock); - } - return eval(page.getPositionCount(), vVector).asBlock(); - } - } - - public DoubleBlock eval(int positionCount, DoubleBlock vBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (vBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (vBlock.getValueCount(p) != 1) { - if (vBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendDouble(Neg.processDoubles(vBlock.getDouble(vBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public DoubleVector eval(int positionCount, DoubleVector vVector) { - try(DoubleVector.Builder result = driverContext.blockFactory().newDoubleVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendDouble(Neg.processDoubles(vVector.getDouble(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "NegDoublesEvaluator[" + "v=" + v + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(v); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory v; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory v) { - this.source = source; - this.v = v; - } - - @Override - public NegDoublesEvaluator get(DriverContext context) { - return new NegDoublesEvaluator(source, v.get(context), context); - } - - @Override - public String toString() { - return "NegDoublesEvaluator[" + "v=" + v + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/NegIntsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/NegIntsEvaluator.java deleted file mode 100644 index 1821406f061bd..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/NegIntsEvaluator.java +++ /dev/null @@ -1,119 +0,0 @@ -// 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.xpack.esql.expression.predicate.operator.arithmetic; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Neg}. - * This class is generated. Do not edit it. - */ -public final class NegIntsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator v; - - private final DriverContext driverContext; - - public NegIntsEvaluator(Source source, EvalOperator.ExpressionEvaluator v, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.v = v; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (IntBlock vBlock = (IntBlock) v.eval(page)) { - IntVector vVector = vBlock.asVector(); - if (vVector == null) { - return eval(page.getPositionCount(), vBlock); - } - return eval(page.getPositionCount(), vVector); - } - } - - public IntBlock eval(int positionCount, IntBlock vBlock) { - try(IntBlock.Builder result = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (vBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (vBlock.getValueCount(p) != 1) { - if (vBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendInt(Neg.processInts(vBlock.getInt(vBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public IntBlock eval(int positionCount, IntVector vVector) { - try(IntBlock.Builder result = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendInt(Neg.processInts(vVector.getInt(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "NegIntsEvaluator[" + "v=" + v + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(v); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory v; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory v) { - this.source = source; - this.v = v; - } - - @Override - public NegIntsEvaluator get(DriverContext context) { - return new NegIntsEvaluator(source, v.get(context), context); - } - - @Override - public String toString() { - return "NegIntsEvaluator[" + "v=" + v + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/NegLongsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/NegLongsEvaluator.java deleted file mode 100644 index 49a0096665112..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/NegLongsEvaluator.java +++ /dev/null @@ -1,119 +0,0 @@ -// 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.xpack.esql.expression.predicate.operator.arithmetic; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Neg}. - * This class is generated. Do not edit it. - */ -public final class NegLongsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator v; - - private final DriverContext driverContext; - - public NegLongsEvaluator(Source source, EvalOperator.ExpressionEvaluator v, - DriverContext driverContext) { - this.warnings = new Warnings(source); - this.v = v; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock vBlock = (LongBlock) v.eval(page)) { - LongVector vVector = vBlock.asVector(); - if (vVector == null) { - return eval(page.getPositionCount(), vBlock); - } - return eval(page.getPositionCount(), vVector); - } - } - - public LongBlock eval(int positionCount, LongBlock vBlock) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (vBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (vBlock.getValueCount(p) != 1) { - if (vBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendLong(Neg.processLongs(vBlock.getLong(vBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public LongBlock eval(int positionCount, LongVector vVector) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendLong(Neg.processLongs(vVector.getLong(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "NegLongsEvaluator[" + "v=" + v + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(v); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory v; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory v) { - this.source = source; - this.v = v; - } - - @Override - public NegLongsEvaluator get(DriverContext context) { - return new NegLongsEvaluator(source, v.get(context), context); - } - - @Override - public String toString() { - return "NegLongsEvaluator[" + "v=" + v + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/SubDatetimesEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/SubDatetimesEvaluator.java deleted file mode 100644 index 88d94573b7562..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/SubDatetimesEvaluator.java +++ /dev/null @@ -1,128 +0,0 @@ -// 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.xpack.esql.expression.predicate.operator.arithmetic; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import java.time.DateTimeException; -import java.time.temporal.TemporalAmount; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Sub}. - * This class is generated. Do not edit it. - */ -public final class SubDatetimesEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator datetime; - - private final TemporalAmount temporalAmount; - - private final DriverContext driverContext; - - public SubDatetimesEvaluator(Source source, EvalOperator.ExpressionEvaluator datetime, - TemporalAmount temporalAmount, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.datetime = datetime; - this.temporalAmount = temporalAmount; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock datetimeBlock = (LongBlock) datetime.eval(page)) { - LongVector datetimeVector = datetimeBlock.asVector(); - if (datetimeVector == null) { - return eval(page.getPositionCount(), datetimeBlock); - } - return eval(page.getPositionCount(), datetimeVector); - } - } - - public LongBlock eval(int positionCount, LongBlock datetimeBlock) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (datetimeBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (datetimeBlock.getValueCount(p) != 1) { - if (datetimeBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendLong(Sub.processDatetimes(datetimeBlock.getLong(datetimeBlock.getFirstValueIndex(p)), temporalAmount)); - } catch (ArithmeticException | DateTimeException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public LongBlock eval(int positionCount, LongVector datetimeVector) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendLong(Sub.processDatetimes(datetimeVector.getLong(p), temporalAmount)); - } catch (ArithmeticException | DateTimeException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SubDatetimesEvaluator[" + "datetime=" + datetime + ", temporalAmount=" + temporalAmount + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(datetime); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory datetime; - - private final TemporalAmount temporalAmount; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory datetime, - TemporalAmount temporalAmount) { - this.source = source; - this.datetime = datetime; - this.temporalAmount = temporalAmount; - } - - @Override - public SubDatetimesEvaluator get(DriverContext context) { - return new SubDatetimesEvaluator(source, datetime.get(context), temporalAmount, context); - } - - @Override - public String toString() { - return "SubDatetimesEvaluator[" + "datetime=" + datetime + ", temporalAmount=" + temporalAmount + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/SubDoublesEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/SubDoublesEvaluator.java deleted file mode 100644 index d479d0fe751c9..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/SubDoublesEvaluator.java +++ /dev/null @@ -1,132 +0,0 @@ -// 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.xpack.esql.expression.predicate.operator.arithmetic; - -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.DoubleBlock; -import org.elasticsearch.compute.data.DoubleVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Sub}. - * This class is generated. Do not edit it. - */ -public final class SubDoublesEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public SubDoublesEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (DoubleBlock lhsBlock = (DoubleBlock) lhs.eval(page)) { - try (DoubleBlock rhsBlock = (DoubleBlock) rhs.eval(page)) { - DoubleVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - DoubleVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector).asBlock(); - } - } - } - - public DoubleBlock eval(int positionCount, DoubleBlock lhsBlock, DoubleBlock rhsBlock) { - try(DoubleBlock.Builder result = driverContext.blockFactory().newDoubleBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - result.appendDouble(Sub.processDoubles(lhsBlock.getDouble(lhsBlock.getFirstValueIndex(p)), rhsBlock.getDouble(rhsBlock.getFirstValueIndex(p)))); - } - return result.build(); - } - } - - public DoubleVector eval(int positionCount, DoubleVector lhsVector, DoubleVector rhsVector) { - try(DoubleVector.Builder result = driverContext.blockFactory().newDoubleVectorBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - result.appendDouble(Sub.processDoubles(lhsVector.getDouble(p), rhsVector.getDouble(p))); - } - return result.build(); - } - } - - @Override - public String toString() { - return "SubDoublesEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public SubDoublesEvaluator get(DriverContext context) { - return new SubDoublesEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "SubDoublesEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/SubIntsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/SubIntsEvaluator.java deleted file mode 100644 index 72bd7e4b6848a..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/SubIntsEvaluator.java +++ /dev/null @@ -1,143 +0,0 @@ -// 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.xpack.esql.expression.predicate.operator.arithmetic; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.IntBlock; -import org.elasticsearch.compute.data.IntVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Sub}. - * This class is generated. Do not edit it. - */ -public final class SubIntsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public SubIntsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (IntBlock lhsBlock = (IntBlock) lhs.eval(page)) { - try (IntBlock rhsBlock = (IntBlock) rhs.eval(page)) { - IntVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - IntVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector); - } - } - } - - public IntBlock eval(int positionCount, IntBlock lhsBlock, IntBlock rhsBlock) { - try(IntBlock.Builder result = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendInt(Sub.processInts(lhsBlock.getInt(lhsBlock.getFirstValueIndex(p)), rhsBlock.getInt(rhsBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public IntBlock eval(int positionCount, IntVector lhsVector, IntVector rhsVector) { - try(IntBlock.Builder result = driverContext.blockFactory().newIntBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendInt(Sub.processInts(lhsVector.getInt(p), rhsVector.getInt(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SubIntsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public SubIntsEvaluator get(DriverContext context) { - return new SubIntsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "SubIntsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/SubLongsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/SubLongsEvaluator.java deleted file mode 100644 index 88cb6bf287d8d..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/SubLongsEvaluator.java +++ /dev/null @@ -1,143 +0,0 @@ -// 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.xpack.esql.expression.predicate.operator.arithmetic; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Sub}. - * This class is generated. Do not edit it. - */ -public final class SubLongsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public SubLongsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock lhsBlock = (LongBlock) lhs.eval(page)) { - try (LongBlock rhsBlock = (LongBlock) rhs.eval(page)) { - LongVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - LongVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector); - } - } - } - - public LongBlock eval(int positionCount, LongBlock lhsBlock, LongBlock rhsBlock) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendLong(Sub.processLongs(lhsBlock.getLong(lhsBlock.getFirstValueIndex(p)), rhsBlock.getLong(rhsBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public LongBlock eval(int positionCount, LongVector lhsVector, LongVector rhsVector) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendLong(Sub.processLongs(lhsVector.getLong(p), rhsVector.getLong(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SubLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public SubLongsEvaluator get(DriverContext context) { - return new SubLongsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "SubLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -} diff --git a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/SubUnsignedLongsEvaluator.java b/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/SubUnsignedLongsEvaluator.java deleted file mode 100644 index 1ef9034d76f62..0000000000000 --- a/x-pack/plugin/esql/src/main/java/generated/org/elasticsearch/xpack/esql/expression/predicate/operator/arithmetic/SubUnsignedLongsEvaluator.java +++ /dev/null @@ -1,143 +0,0 @@ -// 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.xpack.esql.expression.predicate.operator.arithmetic; - -import java.lang.ArithmeticException; -import java.lang.IllegalArgumentException; -import java.lang.Override; -import java.lang.String; -import org.elasticsearch.compute.data.Block; -import org.elasticsearch.compute.data.LongBlock; -import org.elasticsearch.compute.data.LongVector; -import org.elasticsearch.compute.data.Page; -import org.elasticsearch.compute.operator.DriverContext; -import org.elasticsearch.compute.operator.EvalOperator; -import org.elasticsearch.core.Releasables; -import org.elasticsearch.xpack.esql.expression.function.Warnings; -import org.elasticsearch.xpack.ql.tree.Source; - -/** - * {@link EvalOperator.ExpressionEvaluator} implementation for {@link Sub}. - * This class is generated. Do not edit it. - */ -public final class SubUnsignedLongsEvaluator implements EvalOperator.ExpressionEvaluator { - private final Warnings warnings; - - private final EvalOperator.ExpressionEvaluator lhs; - - private final EvalOperator.ExpressionEvaluator rhs; - - private final DriverContext driverContext; - - public SubUnsignedLongsEvaluator(Source source, EvalOperator.ExpressionEvaluator lhs, - EvalOperator.ExpressionEvaluator rhs, DriverContext driverContext) { - this.warnings = new Warnings(source); - this.lhs = lhs; - this.rhs = rhs; - this.driverContext = driverContext; - } - - @Override - public Block eval(Page page) { - try (LongBlock lhsBlock = (LongBlock) lhs.eval(page)) { - try (LongBlock rhsBlock = (LongBlock) rhs.eval(page)) { - LongVector lhsVector = lhsBlock.asVector(); - if (lhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - LongVector rhsVector = rhsBlock.asVector(); - if (rhsVector == null) { - return eval(page.getPositionCount(), lhsBlock, rhsBlock); - } - return eval(page.getPositionCount(), lhsVector, rhsVector); - } - } - } - - public LongBlock eval(int positionCount, LongBlock lhsBlock, LongBlock rhsBlock) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - if (lhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (lhsBlock.getValueCount(p) != 1) { - if (lhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - if (rhsBlock.isNull(p)) { - result.appendNull(); - continue position; - } - if (rhsBlock.getValueCount(p) != 1) { - if (rhsBlock.getValueCount(p) > 1) { - warnings.registerException(new IllegalArgumentException("single-value function encountered multi-value")); - } - result.appendNull(); - continue position; - } - try { - result.appendLong(Sub.processUnsignedLongs(lhsBlock.getLong(lhsBlock.getFirstValueIndex(p)), rhsBlock.getLong(rhsBlock.getFirstValueIndex(p)))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - public LongBlock eval(int positionCount, LongVector lhsVector, LongVector rhsVector) { - try(LongBlock.Builder result = driverContext.blockFactory().newLongBlockBuilder(positionCount)) { - position: for (int p = 0; p < positionCount; p++) { - try { - result.appendLong(Sub.processUnsignedLongs(lhsVector.getLong(p), rhsVector.getLong(p))); - } catch (ArithmeticException e) { - warnings.registerException(e); - result.appendNull(); - } - } - return result.build(); - } - } - - @Override - public String toString() { - return "SubUnsignedLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - - @Override - public void close() { - Releasables.closeExpectNoException(lhs, rhs); - } - - static class Factory implements EvalOperator.ExpressionEvaluator.Factory { - private final Source source; - - private final EvalOperator.ExpressionEvaluator.Factory lhs; - - private final EvalOperator.ExpressionEvaluator.Factory rhs; - - public Factory(Source source, EvalOperator.ExpressionEvaluator.Factory lhs, - EvalOperator.ExpressionEvaluator.Factory rhs) { - this.source = source; - this.lhs = lhs; - this.rhs = rhs; - } - - @Override - public SubUnsignedLongsEvaluator get(DriverContext context) { - return new SubUnsignedLongsEvaluator(source, lhs.get(context), rhs.get(context), context); - } - - @Override - public String toString() { - return "SubUnsignedLongsEvaluator[" + "lhs=" + lhs + ", rhs=" + rhs + "]"; - } - } -}