// Licensed to the Apache Software Foundation (ASF) under one

// or more contributor license agreements.  See the NOTICE file

// distributed with this work for additional information

// regarding copyright ownership.  The ASF licenses this file

// to you under the Apache License, Version 2.0 (the

// "License"); you may not use this file except in compliance

// with the License.  You may obtain a copy of the License at

//

//   http://www.apache.org/licenses/LICENSE-2.0

//

// Unless required by applicable law or agreed to in writing,

// software distributed under the License is distributed on an

// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY

// KIND, either express or implied.  See the License for the

// specific language governing permissions and limitations

// under the License.

#pragma once

#include <glog/logging.h>

#include <cmath>

#include "core/column/column_decimal.h"

#include "core/column/column_vector.h"

#include "core/data_type/data_type_decimal.h"

#include "core/data_type/data_type_nullable.h"

#include "core/data_type/data_type_number.h"

#include "core/string_buffer.hpp"

#include "core/types.h"

#include "exprs/aggregate/aggregate_function.h"

namespace doris {

template <PrimitiveType T, template <PrimitiveType> typename Moments>

struct StatFunc {

    using ColVecT1 = typename PrimitiveTypeTraits<T>::ColumnType;

    using ColVecT2 = ColVecT1;

    using Data = Moments<TYPE_DOUBLE>;

    static constexpr PrimitiveType ResultPrimitiveType = TYPE_DOUBLE;

};

template <typename StatFunc>

struct AggregateFunctionBinary final

        : public IAggregateFunctionDataHelper<typename StatFunc::Data,

                                              AggregateFunctionBinary<StatFunc>>,

          MultiExpression,

          NullableAggregateFunction {

    static constexpr PrimitiveType ResultType = StatFunc::ResultPrimitiveType;

    using ColVecT1 = typename StatFunc::ColVecT1;

    using ColVecT2 = typename StatFunc::ColVecT2;

    using ColVecResult = ColumnVector<ResultType>;

    static constexpr UInt32 num_args = 2;

    AggregateFunctionBinary(const DataTypes& argument_types_)

            : IAggregateFunctionDataHelper<typename StatFunc::Data,

                                           AggregateFunctionBinary<StatFunc>>(argument_types_) {}

_ZN5doris23AggregateFunctionBinaryINS_8StatFuncILNS_13PrimitiveTypeE9ENS_10CorrMomentEEEEC2ERKSt6vectorISt10shared_ptrIKNS_9IDataTypeEESaISA_EE

Line

Count

Source

57

357

            : IAggregateFunctionDataHelper<typename StatFunc::Data,

58

357

                                           AggregateFunctionBinary<StatFunc>>(argument_types_) {}

_ZN5doris23AggregateFunctionBinaryINS_8StatFuncILNS_13PrimitiveTypeE9ENS_17CorrMomentWelfordEEEEC2ERKSt6vectorISt10shared_ptrIKNS_9IDataTypeEESaISA_EE

Line

Count

Source

57

36

            : IAggregateFunctionDataHelper<typename StatFunc::Data,

58

36

                                           AggregateFunctionBinary<StatFunc>>(argument_types_) {}

    String get_name() const override { return StatFunc::Data::name(); }

_ZNK5doris23AggregateFunctionBinaryINS_8StatFuncILNS_13PrimitiveTypeE9ENS_10CorrMomentEEEE8get_nameB5cxx11Ev

Line

Count

Source

60

33

    String get_name() const override { return StatFunc::Data::name(); }

_ZNK5doris23AggregateFunctionBinaryINS_8StatFuncILNS_13PrimitiveTypeE9ENS_17CorrMomentWelfordEEEE8get_nameB5cxx11Ev

Line

Count

Source

60

2

    String get_name() const override { return StatFunc::Data::name(); }

    void reset(AggregateDataPtr __restrict place) const override { this->data(place).reset(); }

_ZNK5doris23AggregateFunctionBinaryINS_8StatFuncILNS_13PrimitiveTypeE9ENS_10CorrMomentEEEE5resetEPc

Line

Count

Source

62

40

    void reset(AggregateDataPtr __restrict place) const override { this->data(place).reset(); }

_ZNK5doris23AggregateFunctionBinaryINS_8StatFuncILNS_13PrimitiveTypeE9ENS_17CorrMomentWelfordEEEE5resetEPc

Line

Count

Source

62

2

    void reset(AggregateDataPtr __restrict place) const override { this->data(place).reset(); }

    DataTypePtr get_return_type() const override {

        return std::make_shared<DataTypeNumber<StatFunc::ResultPrimitiveType>>();

    }

_ZNK5doris23AggregateFunctionBinaryINS_8StatFuncILNS_13PrimitiveTypeE9ENS_10CorrMomentEEEE15get_return_typeEv

Line

Count

Source

64

237

    DataTypePtr get_return_type() const override {

65

237

        return std::make_shared<DataTypeNumber<StatFunc::ResultPrimitiveType>>();

66

237

    }

_ZNK5doris23AggregateFunctionBinaryINS_8StatFuncILNS_13PrimitiveTypeE9ENS_17CorrMomentWelfordEEEE15get_return_typeEv

Line

Count

Source

64

132

    DataTypePtr get_return_type() const override {

65

132

        return std::make_shared<DataTypeNumber<StatFunc::ResultPrimitiveType>>();

66

132

    }

    void add(AggregateDataPtr __restrict place, const IColumn** columns, ssize_t row_num,

             Arena&) const override {

        this->data(place).add(

                static_cast<typename PrimitiveTypeTraits<ResultType>::CppType>(

                        static_cast<const ColVecT1&>(*columns[0]).get_data()[row_num]),

                static_cast<typename PrimitiveTypeTraits<ResultType>::CppType>(

                        static_cast<const ColVecT2&>(*columns[1]).get_data()[row_num]));

    }

_ZNK5doris23AggregateFunctionBinaryINS_8StatFuncILNS_13PrimitiveTypeE9ENS_10CorrMomentEEEE3addEPcPPKNS_7IColumnElRNS_5ArenaE

Line

Count

Source

69

338

             Arena&) const override {

70

338

        this->data(place).add(

71

338

                static_cast<typename PrimitiveTypeTraits<ResultType>::CppType>(

72

338

                        static_cast<const ColVecT1&>(*columns[0]).get_data()[row_num]),

73

338

                static_cast<typename PrimitiveTypeTraits<ResultType>::CppType>(

74

338

                        static_cast<const ColVecT2&>(*columns[1]).get_data()[row_num]));

75

338

    }

_ZNK5doris23AggregateFunctionBinaryINS_8StatFuncILNS_13PrimitiveTypeE9ENS_17CorrMomentWelfordEEEE3addEPcPPKNS_7IColumnElRNS_5ArenaE

Line

Count

Source

69

166

             Arena&) const override {

70

166

        this->data(place).add(

71

166

                static_cast<typename PrimitiveTypeTraits<ResultType>::CppType>(

72

166

                        static_cast<const ColVecT1&>(*columns[0]).get_data()[row_num]),

73

166

                static_cast<typename PrimitiveTypeTraits<ResultType>::CppType>(

74

166

                        static_cast<const ColVecT2&>(*columns[1]).get_data()[row_num]));

75

166

    }

    void merge(AggregateDataPtr __restrict place, ConstAggregateDataPtr rhs,

               Arena&) const override {

        this->data(place).merge(this->data(rhs));

    }

_ZNK5doris23AggregateFunctionBinaryINS_8StatFuncILNS_13PrimitiveTypeE9ENS_10CorrMomentEEEE5mergeEPcPKcRNS_5ArenaE

Line

Count

Source

78

179

               Arena&) const override {

79

179

        this->data(place).merge(this->data(rhs));

80

179

    }

_ZNK5doris23AggregateFunctionBinaryINS_8StatFuncILNS_13PrimitiveTypeE9ENS_17CorrMomentWelfordEEEE5mergeEPcPKcRNS_5ArenaE

Line

Count

Source

78

46

               Arena&) const override {

79

46

        this->data(place).merge(this->data(rhs));

80

46

    }

    void serialize(ConstAggregateDataPtr __restrict place, BufferWritable& buf) const override {

        this->data(place).write(buf);

    }

_ZNK5doris23AggregateFunctionBinaryINS_8StatFuncILNS_13PrimitiveTypeE9ENS_10CorrMomentEEEE9serializeEPKcRNS_14BufferWritableE

Line

Count

Source

82

220

    void serialize(ConstAggregateDataPtr __restrict place, BufferWritable& buf) const override {

83

220

        this->data(place).write(buf);

84

220

    }

_ZNK5doris23AggregateFunctionBinaryINS_8StatFuncILNS_13PrimitiveTypeE9ENS_17CorrMomentWelfordEEEE9serializeEPKcRNS_14BufferWritableE

Line

Count

Source

82

48

    void serialize(ConstAggregateDataPtr __restrict place, BufferWritable& buf) const override {

83

48

        this->data(place).write(buf);

84

48

    }

    void deserialize(AggregateDataPtr __restrict place, BufferReadable& buf,

                     Arena&) const override {

        this->data(place).read(buf);

    }

_ZNK5doris23AggregateFunctionBinaryINS_8StatFuncILNS_13PrimitiveTypeE9ENS_10CorrMomentEEEE11deserializeEPcRNS_14BufferReadableERNS_5ArenaE

Line

Count

Source

87

179

                     Arena&) const override {

88

179

        this->data(place).read(buf);

89

179

    }

_ZNK5doris23AggregateFunctionBinaryINS_8StatFuncILNS_13PrimitiveTypeE9ENS_17CorrMomentWelfordEEEE11deserializeEPcRNS_14BufferReadableERNS_5ArenaE

Line

Count

Source

87

46

                     Arena&) const override {

88

46

        this->data(place).read(buf);

89

46

    }

    void insert_result_into(ConstAggregateDataPtr __restrict place, IColumn& to) const override {

        const auto& data = this->data(place);

        auto& dst = static_cast<ColVecResult&>(to).get_data();

        dst.push_back(data.get());

    }

_ZNK5doris23AggregateFunctionBinaryINS_8StatFuncILNS_13PrimitiveTypeE9ENS_10CorrMomentEEEE18insert_result_intoEPKcRNS_7IColumnE

Line

Count

Source

91

84

    void insert_result_into(ConstAggregateDataPtr __restrict place, IColumn& to) const override {

92

84

        const auto& data = this->data(place);

93

84

        auto& dst = static_cast<ColVecResult&>(to).get_data();

94

84

        dst.push_back(data.get());

95

84

    }

_ZNK5doris23AggregateFunctionBinaryINS_8StatFuncILNS_13PrimitiveTypeE9ENS_17CorrMomentWelfordEEEE18insert_result_intoEPKcRNS_7IColumnE

Line

Count

Source

91

35

    void insert_result_into(ConstAggregateDataPtr __restrict place, IColumn& to) const override {

92

35

        const auto& data = this->data(place);

93

35

        auto& dst = static_cast<ColVecResult&>(to).get_data();

94

35

        dst.push_back(data.get());

95

35

    }

};

} // namespace doris