--! qt:dataset:src
DESCRIBE FUNCTION percentile_cont;
DESCRIBE FUNCTION EXTENDED percentile_cont;


set hive.map.aggr = false;
set hive.groupby.skewindata = false;

-- SORT_QUERY_RESULTS

SELECT CAST(key AS INT) DIV 10,
       percentile_cont(CAST(substr(value, 5) AS INT), 0.0),
       percentile_cont(CAST(substr(value, 5) AS DOUBLE), 0.5),
       percentile_cont(0.5) WITHIN GROUP (ORDER BY CAST(substr(value, 5) AS DOUBLE)),
       percentile_cont(CAST(substr(value, 5) AS DECIMAL), 1.0),
       percentile_cont(array(0.0, 0.5, 1.0)) WITHIN GROUP (ORDER BY CAST(substr(value, 5) AS DOUBLE))
FROM src
GROUP BY CAST(key AS INT) DIV 10;


set hive.map.aggr = true;
set hive.groupby.skewindata = false;

SELECT CAST(key AS INT) DIV 10,
       percentile_cont(CAST(substr(value, 5) AS INT), 0.0),
       percentile_cont(CAST(substr(value, 5) AS DOUBLE), 0.5),
       percentile_cont(0.5) WITHIN GROUP (ORDER BY CAST(substr(value, 5) AS DOUBLE)),
       percentile_cont(CAST(substr(value, 5) AS DECIMAL), 1.0),
       percentile_cont(array(0.0, 0.1, 0.5, 0.8, 1.0)) WITHIN GROUP (ORDER BY CAST(substr(value, 5) AS DOUBLE))
FROM src
GROUP BY CAST(key AS INT) DIV 10;


set hive.map.aggr = false;
set hive.groupby.skewindata = true;

SELECT CAST(key AS INT) DIV 10,
       percentile_cont(CAST(substr(value, 5) AS INT), 0.0),
       percentile_cont(CAST(substr(value, 5) AS DOUBLE), 0.5),
       percentile_cont(0.5) WITHIN GROUP (ORDER BY CAST(substr(value, 5) AS DOUBLE)),
       percentile_cont(CAST(substr(value, 5) AS DECIMAL), 1.0),
       percentile_cont(array(0.5, 1.0)) WITHIN GROUP (ORDER BY CAST(substr(value, 5) AS DOUBLE))
FROM src
GROUP BY CAST(key AS INT) DIV 10;


set hive.map.aggr = true;
set hive.groupby.skewindata = true;

SELECT CAST(key AS INT) DIV 10,
       percentile_cont(CAST(substr(value, 5) AS INT), 0.0),
       percentile_cont(CAST(substr(value, 5) AS DOUBLE), 0.5),
       percentile_cont(0.5) WITHIN GROUP (ORDER BY CAST(substr(value, 5) AS DOUBLE)),
       percentile_cont(CAST(substr(value, 5) AS DECIMAL), 1.0),
       percentile_cont(array(0.0, 0.5, 0.7, 1.0)) WITHIN GROUP (ORDER BY CAST(substr(value, 5) AS DOUBLE))
FROM src
GROUP BY CAST(key AS INT) DIV 10;


set hive.map.aggr = true;
set hive.groupby.skewindata = false;

-- test null handling
SELECT CAST(key AS INT) DIV 10,
       percentile_cont(NULL, 0.0),
       percentile_cont(0.0) WITHIN GROUP (ORDER BY NULL)
FROM src
GROUP BY CAST(key AS INT) DIV 10;


-- test empty array handling
SELECT CAST(key AS INT) DIV 10,
       percentile_cont(IF(CAST(key AS INT) DIV 10 < 5, 1, NULL), 0.5),
       percentile_cont(0.5) WITHIN GROUP (ORDER BY IF(CAST(key AS INT) DIV 10 < 5, 1, NULL))
FROM src
GROUP BY CAST(key AS INT) DIV 10;

select percentile_cont(cast(key as bigint), 0.5),
       percentile_cont(0.5) within group (order by cast(key as bigint))
from src where false;

CREATE TABLE t_test (value int);
INSERT INTO t_test VALUES (NULL), (3), (8), (13), (7), (6), (20), (NULL), (NULL), (10), (7), (15), (16), (8), (7), (8), (NULL);

EXPLAIN SELECT
percentile_cont(value, 0.0),
percentile_cont(value, 0.2),
percentile_cont(0.2) WITHIN GROUP (ORDER BY value),
percentile_cont(0.2) WITHIN GROUP (ORDER BY value NULLS FIRST),
percentile_cont(0.2) WITHIN GROUP (ORDER BY value NULLS LAST),
percentile_cont(0.2) WITHIN GROUP (ORDER BY value) = percentile_cont(value, 0.2),
percentile_cont(0.2) WITHIN GROUP (ORDER BY value ASC),
percentile_cont(0.2) WITHIN GROUP (ORDER BY value ASC) = percentile_cont(value, 0.2),
percentile_cont(0.2) WITHIN GROUP (ORDER BY value DESC),
percentile_cont(0.2) WITHIN GROUP (ORDER BY value DESC NULLS FIRST),
percentile_cont(0.2) WITHIN GROUP (ORDER BY value DESC NULLS LAST)
FROM t_test;

SELECT
percentile_cont(value, 0.0),
percentile_cont(value, 0.2),
percentile_cont(0.2) WITHIN GROUP (ORDER BY value),
percentile_cont(0.2) WITHIN GROUP (ORDER BY value NULLS FIRST),
percentile_cont(0.2) WITHIN GROUP (ORDER BY value NULLS LAST),
percentile_cont(0.2) WITHIN GROUP (ORDER BY value) = percentile_cont(value, 0.2),
percentile_cont(0.2) WITHIN GROUP (ORDER BY value ASC),
percentile_cont(0.2) WITHIN GROUP (ORDER BY value ASC) = percentile_cont(value, 0.2),
percentile_cont(0.2) WITHIN GROUP (ORDER BY value DESC),
percentile_cont(0.2) WITHIN GROUP (ORDER BY value DESC NULLS FIRST),
percentile_cont(0.2) WITHIN GROUP (ORDER BY value DESC NULLS LAST)
FROM t_test;

DROP TABLE t_test;