1#include <AggregateFunctions/AggregateFunctionFactory.h>
2#include <AggregateFunctions/Helpers.h>
3#include <AggregateFunctions/FactoryHelpers.h>
4#include <AggregateFunctions/AggregateFunctionStatistics.h>
5#include "registerAggregateFunctions.h"
6
7
8namespace DB
9{
10
11namespace ErrorCodes
12{
13 extern const int ILLEGAL_TYPE_OF_ARGUMENT;
14}
15
16namespace
17{
18
19template <template <typename> class FunctionTemplate>
20AggregateFunctionPtr createAggregateFunctionStatisticsUnary(const std::string & name, const DataTypes & argument_types, const Array & parameters)
21{
22 assertNoParameters(name, parameters);
23 assertUnary(name, argument_types);
24
25 AggregateFunctionPtr res(createWithNumericType<FunctionTemplate>(*argument_types[0], argument_types[0]));
26
27 if (!res)
28 throw Exception("Illegal type " + argument_types[0]->getName() + " of argument for aggregate function " + name, ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT);
29
30 return res;
31}
32
33template <template <typename, typename> class FunctionTemplate>
34AggregateFunctionPtr createAggregateFunctionStatisticsBinary(const std::string & name, const DataTypes & argument_types, const Array & parameters)
35{
36 assertNoParameters(name, parameters);
37 assertBinary(name, argument_types);
38
39 AggregateFunctionPtr res(createWithTwoNumericTypes<FunctionTemplate>(*argument_types[0], *argument_types[1], argument_types));
40 if (!res)
41 throw Exception("Illegal types " + argument_types[0]->getName() + " and " + argument_types[1]->getName()
42 + " of arguments for aggregate function " + name, ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT);
43
44 return res;
45}
46
47}
48
49void registerAggregateFunctionsStatisticsStable(AggregateFunctionFactory & factory)
50{
51 factory.registerFunction("varSampStable", createAggregateFunctionStatisticsUnary<AggregateFunctionVarSampStable>);
52 factory.registerFunction("varPopStable", createAggregateFunctionStatisticsUnary<AggregateFunctionVarPopStable>);
53 factory.registerFunction("stddevSampStable", createAggregateFunctionStatisticsUnary<AggregateFunctionStddevSampStable>);
54 factory.registerFunction("stddevPopStable", createAggregateFunctionStatisticsUnary<AggregateFunctionStddevPopStable>);
55 factory.registerFunction("covarSampStable", createAggregateFunctionStatisticsBinary<AggregateFunctionCovarSampStable>);
56 factory.registerFunction("covarPopStable", createAggregateFunctionStatisticsBinary<AggregateFunctionCovarPopStable>);
57 factory.registerFunction("corrStable", createAggregateFunctionStatisticsBinary<AggregateFunctionCorrStable>);
58}
59
60}
61