io.nosqlbench.virtdata.annotations.ExampleData Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of virtdata-lib-curves4 Show documentation
Statistical sampling library for use in virtdata libraries, based on apache commons math 4
There is a newer version: 5.17.0
package io.nosqlbench.virtdata.annotations;

import java.util.Arrays;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

/**
 * Formatting conventions
 * The following example formats are supported:
 *
 * 
 *     CtorName(param,...)
 - as the constructor would be called.
 *     [1,2,3]
 - a sequence of integer values
 *     [longs1]
 - a name of a pre-defined set of sample inputs
 * 
 */
public class ExampleData {

    public static Pattern CTOR_PATTERN = Pattern.compile("(?[^)]+)\\((?.+)\\)");
    public static Pattern VALS_PATTERN = Pattern.compile("\\[(?-?\\d+([,-. ]+-?\\d+)*)]");
    private Pattern COMMA_VALS = Pattern.compile("\\[(?-?\\d+(,-?\\d+)*)]");
    private Pattern RANGE_VALS = Pattern.compile("\\[(?-?\\d+)\\.\\.(?-?\\d+)( +(?-?\\d+))?]");

    public String[] parts;

    public ExampleData(String[] parts) {
        this.parts = parts;
    }

    public static void validateExamples(List> examples) {
        for (List example : examples) {
            for (String examplePart : example) {
                if (CTOR_PATTERN.matcher(examplePart).matches()) {
                    continue;
                }
                if (VALS_PATTERN.matcher(examplePart).matches()) {
                    continue;
                }
                if (!examplePart.startsWith("[")) {
                    return;
                }
                throw new RuntimeException("Unable to match a valid pattern for example fragment '"+examplePart +"')" +
                        ". See javadoc for the Example annotation for details.");
            }
        }
    }


    public long[] getLongInputs() {
        long[] vals = new long[0];
        for (String part : parts) {
            if (VALS_PATTERN.matcher(part).matches()) {
                String[] specs = part.split(";");
                long[][] genvals = new long[specs.length][];
                int len =0;
                for (int i = 0; i < genvals.length; i++) {
                    String spec = specs[i];
                    genvals[i] = parseRange(spec);
                    len+=genvals[i].length;
                }
                vals = new long[len];
                int atlen=0;
                for (int i = 0; i < genvals.length; i++) {
                    System.arraycopy(genvals[i],0,vals,atlen,genvals[i].length);
                    atlen+=genvals[i].length;
                }
            }
        }
        return vals;
    }


    private long[] parseRange(String spec) {
        Matcher comma_matcher = COMMA_VALS.matcher(spec);
        if (comma_matcher.matches()) {
            String vals = comma_matcher.group("vals");
            long[] longvals = Arrays.stream(vals.split(",")).mapToLong(Long::valueOf).toArray();
            return longvals;
        }
        Matcher range_matcher = RANGE_VALS.matcher(spec);
        if (range_matcher.matches()) {
            long from=Long.parseLong(range_matcher.group("from"));
            long to=Long.parseLong(range_matcher.group("to"));
            String step_size = range_matcher.group("step");
            long step = 1L;
            if (step_size!=null) {
                step = Long.parseLong(step_size);
            }
            if (fromto && step>=0) {
                throw new RuntimeException("for decreasing from-to of (" +from+"-"+to+":, stepsize must be negative.");
            }
            long sizeL = (Math.abs(from-to) / Math.abs(step))+1;
            if (sizeL>Integer.MAX_VALUE) {
                throw new RuntimeException("example size " + sizeL + " is too big.");
            }
            long[] vals = new long[(int)sizeL];
            long sign = (step>0 ? 1L : -1L);
            for (int i = 0; i < vals.length; i++) {
                vals[i]=from+(step*i);
            }
            return vals;
        }
        throw new RuntimeException("Unable to parse spec pattern: '"+spec+"'");
    }
}