@ScalarFunction("value_at_quantile") @Description("Given an input q between [0, 1], find the value whose rank in the sorted sequence of the n values represented by the qdigest is qn.") @SqlType(StandardTypes.REAL) public static long valueAtQuantileReal(@SqlType("qdigest(real)") Slice input, @SqlType(StandardTypes.DOUBLE) double quantile) { return floatToRawIntBits(sortableIntToFloat((int) valueAtQuantileBigint(input, quantile))); }
@OutputFunction("array(real)") public static void output(@AggregationState DigestAndPercentileArrayState state, BlockBuilder out) { QuantileDigest digest = state.getDigest(); List<Double> percentiles = state.getPercentiles(); if (percentiles == null || digest == null) { out.appendNull(); return; } BlockBuilder blockBuilder = out.beginBlockEntry(); for (int i = 0; i < percentiles.size(); i++) { Double percentile = percentiles.get(i); REAL.writeLong(blockBuilder, floatToRawIntBits(sortableIntToFloat((int) digest.getQuantile(percentile)))); } out.closeEntry(); } }
@OutputFunction(StandardTypes.REAL) public static void output(@AggregationState DigestAndPercentileState state, BlockBuilder out) { QuantileDigest digest = state.getDigest(); double percentile = state.getPercentile(); if (digest == null || digest.getCount() == 0.0) { out.appendNull(); } else { checkState(percentile != -1.0, "Percentile is missing"); checkCondition(0 <= percentile && percentile <= 1, INVALID_FUNCTION_ARGUMENT, "Percentile must be between 0 and 1"); REAL.writeLong(out, floatToRawIntBits(sortableIntToFloat((int) digest.getQuantile(percentile)))); } } }
@ScalarFunction("values_at_quantiles") @Description("For each input q between [0, 1], find the value whose rank in the sorted sequence of the n values represented by the qdigest is qn.") @SqlType("array(real)") public static Block valuesAtQuantilesReal(@SqlType("qdigest(real)") Slice input, @SqlType("array(double)") Block percentilesArrayBlock) { QuantileDigest digest = new QuantileDigest(input); BlockBuilder output = REAL.createBlockBuilder(null, percentilesArrayBlock.getPositionCount()); for (int i = 0; i < percentilesArrayBlock.getPositionCount(); i++) { REAL.writeLong(output, floatToRawIntBits(sortableIntToFloat((int) digest.getQuantile(DOUBLE.getDouble(percentilesArrayBlock, i))))); } return output.build(); }