subprojects/store-query/src/main/java/tools/refinery/store/query/view/FunctionView.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36

/*
 * SPDX-FileCopyrightText: 2021-2023 The Refinery Authors <https://refinery.tools/>
 *
 * SPDX-License-Identifier: EPL-2.0
 */
package tools.refinery.store.query.view;

import tools.refinery.store.query.term.*;
import tools.refinery.store.representation.Symbol;

import java.util.ArrayList;
import java.util.List;

public final class FunctionView<T> extends AbstractFunctionView<T> {
	public FunctionView(Symbol<T> symbol, String name) {
		super(symbol, name, new Parameter(symbol.valueType(), ParameterDirection.OUT));
	}

	public FunctionView(Symbol<T> symbol) {
		this(symbol, "function");
	}

	public <R> AssignedValue<R> aggregate(Aggregator<R, T> aggregator, List<NodeVariable> arguments) {
		return targetVariable -> {
			var placeholderVariable = Variable.of(getSymbol().valueType());
			var argumentsWithPlaceholder = new ArrayList<Variable>(arguments.size() + 1);
			argumentsWithPlaceholder.addAll(arguments);
			argumentsWithPlaceholder.add(placeholderVariable);
			return aggregateBy(placeholderVariable, aggregator, argumentsWithPlaceholder).toLiteral(targetVariable);
		};
	}

	public <R> AssignedValue<R> aggregate(Aggregator<R, T> aggregator, NodeVariable... arguments) {
		return aggregate(aggregator, List.of(arguments));
	}
}