/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.jena.sparql.function.library.leviathan;
import java.util.List;
import org.apache.jena.atlas.lib.Lib ;
import org.apache.jena.atlas.lib.RandomLib;
import org.apache.jena.query.QueryBuildException ;
import org.apache.jena.sparql.expr.ExprEvalException ;
import org.apache.jena.sparql.expr.ExprList ;
import org.apache.jena.sparql.expr.NodeValue ;
import org.apache.jena.sparql.function.FunctionBase ;
public class rnd extends FunctionBase {
@Override
public NodeValue exec(List<NodeValue> args) {
if (args.size() > 2)
throw new ExprEvalException("Too many arguments");
switch (args.size()) {
case 0:
return NodeValue.makeDouble(RandomLib.random.nextDouble());
case 1: {
double max = args.get(0).getDouble();
if (max <= 0d)
throw new ExprEvalException("Max must be > 0");
return NodeValue.makeDouble(RandomLib.random.nextDouble() * max);
}
case 2: {
double min = args.get(0).getDouble();
double max = args.get(1).getDouble();
if (min > max)
throw new ExprEvalException("Min cannot be greater than the max");
double range = max - min;
double value = min + (RandomLib.random.nextDouble() * range);
return NodeValue.makeDouble(value);
}
default:
throw new ExprEvalException("Too many arguments");
}
}
@Override
public void checkBuild(String uri, ExprList args) {
if (args.size() > 2)
throw new QueryBuildException("Function '" + Lib.className(this)
+ "' takes between 0, 1 or 2 argument(s)");
}
}