IntenseFinance/nosqlbench
3
0
Fork 0
mirror of https://github.com/nosqlbench/nosqlbench.git synced 2025-02-25 18:55:28 -06:00
Code Issues Packages Projects Releases Wiki Activity

add experimental optimo into project

Browse Source
This commit is contained in:
Jonathan Shook 2020-09-10 09:37:07 -05:00
parent 9e4551a7cf
commit 43f41caecd
2 changed files with 14 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
engine-extensions/src/main/java/io/nosqlbench/engine/extensions/optimizers/MVResult.java
View File

@ -7,17 +7,23 @@ public class MVResult {
private final double[] vars; private final double[] vars;
private final MVParams params; private final MVParams params;
private final double[][] datalog;
public MVResult(double[] vars, MVParams params, double[][] datalog) { public MVResult(double[] vars, MVParams params, double[][] datalog) {
this.params = params; this.params = params;
this.vars = vars; this.vars = vars;
this.datalog = datalog;
} }
public double[] getVarArray() { public double[] getVarArray() {
return vars; return vars;
} }
public Map<String,Double> getVarMap() { public double[][] getDatalog() {
return datalog;
}
public Map<String, Double> getVars() {
Map<String, Double> result = new HashMap<>(params.size()); Map<String, Double> result = new HashMap<>(params.size());
for (int i = 0; i < vars.length; i++) { for (int i = 0; i < vars.length; i++) {
result.put(params.get(i).name, vars[i]); result.put(params.get(i).name, vars[i]);

4
engine-extensions/src/main/java/io/nosqlbench/engine/extensions/optimizers/PolyglotMultivariateObjectScript.java
View File

@ -13,7 +13,7 @@ public class PolyglotMultivariateObjectScript implements MultivariateFunction {
private final MVParams params; private final MVParams params;
private final Object function; private final Object function;
private Logger logger; private final Logger logger;
public PolyglotMultivariateObjectScript(Logger logger, MVParams params, Object function) { public PolyglotMultivariateObjectScript(Logger logger, MVParams params, Object function) {
this.logger = logger; this.logger = logger;
@ -37,6 +37,8 @@ public class PolyglotMultivariateObjectScript implements MultivariateFunction {
if (result instanceof Double) { if (result instanceof Double) {
return (Double) result; return (Double) result;
} else if (result instanceof Integer) {
return (double) ((Integer) result);
} else { } else {
throw new RuntimeException( throw new RuntimeException(
"Unable to case result of polyglot function return value as a double:" + "Unable to case result of polyglot function return value as a double:" +