2

我正在尝试在 Postgres 中实现一个自定义聚合函数，该函数将以度为单位平均方向 - 即我希望能够做到：

SELECT circavg(direction) FROM sometable;

这可以使用以下公式来完成：

xbar = atan2(sum(sin(xi), sum(cos(xi)))

我想我需要定义sfunc一个方向，并将其正弦和余弦添加到两个累加器中。最后一个函数然后使用 atan2 将这两个分量转换回一个方向。

我不知道如何定义，sfunc以便当前状态由两个组件组成，例如(float, float). 该文档在具体示例上有点短，因此感谢您提供任何帮助。

4

2 回答 2

4

您可以在内部使用ARRAY类型。参数类型仍然可以是任何数字类型。float用(= )演示double precision：

CREATE OR REPLACE FUNCTION f_circavg (float[], float)
  RETURNS float[] LANGUAGE sql STRICT AS
'SELECT ARRAY[$1[1] + sin($2), $1[2] + cos($2), 1]';

CREATE OR REPLACE FUNCTION f_circavg_final (float[])
  RETURNS float  LANGUAGE sql AS
'SELECT CASE WHEN $1[3] > 0 THEN atan2($1[1], $1[2]) END';

CREATE AGGREGATE circavg (float) (
   sfunc     = f_circavg
 , stype     = float[]
 , finalfunc = f_circavg_final
 , initcond  = '{0,0,0}'
);

转换函数f_circavg()已定义STRICT，因此它会忽略NULL输入行。它还设置第三个数组元素以标识具有一个或多个输入行的集合 - 否则CASE最终函数返回NULL。

测试临时表：

CREATE TEMP TABLE t (x float);
INSERT INTO t VALUES (2), (NULL), (3), (4), (5);

我输入了一个NULL值来测试STRICT魔法。称呼：

SELECT circavg(x) FROM t;

       circavg
-------------------
 -2.78318530717959

交叉检查：

SELECT atan2(sum(sin(x)), sum(cos(x))) FROM t;

       atan2
-------------------
 -2.78318530717959

返回相同。似乎工作。在使用更大表的测试中，使用常规聚合函数的最后一个表达式比自定义聚合快 4 倍。

测试零输入行/仅 NULL 输入：

SELECT circavg(x) FROM t WHERE false;     -- no input rows
SELECT circavg(x) FROM t WHERE x IS NULL; -- only NULL input

NULL在两种情况下都返回。

于 2012-04-19T10:29:56.120 回答
0

PostgreSQL 为您提供了一组几何类型，POINT是最基本的。
将此类型用作函数的输入参数。

如果您愿意，可以创建自定义类型作为替代。

于 2012-04-19T10:02:36.737 回答