// 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

466

            : IAggregateFunctionDataHelper<typename StatFunc::Data,

58

466

                                           AggregateFunctionBinary<StatFunc>>(argument_types_) {}

_ZN5doris23AggregateFunctionBinaryINS_8StatFuncILNS_13PrimitiveTypeE9ENS_17CorrMomentWelfordEEEEC2ERKSt6vectorISt10shared_ptrIKNS_9IDataTypeEESaISA_EE

Line

Count

Source

57

35

            : IAggregateFunctionDataHelper<typename StatFunc::Data,

58

35

                                           AggregateFunctionBinary<StatFunc>>(argument_types_) {}

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

_ZNK5doris23AggregateFunctionBinaryINS_8StatFuncILNS_13PrimitiveTypeE9ENS_10CorrMomentEEEE8get_nameB5cxx11Ev

Line

Count

Source

60

31

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

_ZNK5doris23AggregateFunctionBinaryINS_8StatFuncILNS_13PrimitiveTypeE9ENS_17CorrMomentWelfordEEEE8get_nameB5cxx11Ev

Line

Count

Source

60

1

    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

39

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

_ZNK5doris23AggregateFunctionBinaryINS_8StatFuncILNS_13PrimitiveTypeE9ENS_17CorrMomentWelfordEEEE5resetEPc

Line

Count

Source

62

1

    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

281

    DataTypePtr get_return_type() const override {

65

281

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

66

281

    }

_ZNK5doris23AggregateFunctionBinaryINS_8StatFuncILNS_13PrimitiveTypeE9ENS_17CorrMomentWelfordEEEE15get_return_typeEv

Line

Count

Source

64

180

    DataTypePtr get_return_type() const override {

65

180

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

66

180

    }

    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

303

             Arena&) const override {

70

303

        this->data(place).add(

71

303

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

72

303

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

73

303

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

74

303

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

75

303

    }

_ZNK5doris23AggregateFunctionBinaryINS_8StatFuncILNS_13PrimitiveTypeE9ENS_17CorrMomentWelfordEEEE3addEPcPPKNS_7IColumnElRNS_5ArenaE

Line

Count

Source

69

131

             Arena&) const override {

70

131

        this->data(place).add(

71

131

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

72

131

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

73

131

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

74

131

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

75

131

    }

    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

171

               Arena&) const override {

79

171

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

80

171

    }

_ZNK5doris23AggregateFunctionBinaryINS_8StatFuncILNS_13PrimitiveTypeE9ENS_17CorrMomentWelfordEEEE5mergeEPcPKcRNS_5ArenaE

Line

Count

Source

78

52

               Arena&) const override {

79

52

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

80

52

    }

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

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

    }

_ZNK5doris23AggregateFunctionBinaryINS_8StatFuncILNS_13PrimitiveTypeE9ENS_10CorrMomentEEEE9serializeEPKcRNS_14BufferWritableE

Line

Count

Source

82

211

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

83

211

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

84

211

    }

_ZNK5doris23AggregateFunctionBinaryINS_8StatFuncILNS_13PrimitiveTypeE9ENS_17CorrMomentWelfordEEEE9serializeEPKcRNS_14BufferWritableE

Line

Count

Source

82

53

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

83

53

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

84

53

    }

    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

171

                     Arena&) const override {

88

171

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

89

171

    }

_ZNK5doris23AggregateFunctionBinaryINS_8StatFuncILNS_13PrimitiveTypeE9ENS_17CorrMomentWelfordEEEE11deserializeEPcRNS_14BufferReadableERNS_5ArenaE

Line

Count

Source

87

52

                     Arena&) const override {

88

52

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

89

52

    }

    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

78

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

92

78

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

93

78

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

94

78

        dst.push_back(data.get());

95

78

    }

_ZNK5doris23AggregateFunctionBinaryINS_8StatFuncILNS_13PrimitiveTypeE9ENS_17CorrMomentWelfordEEEE18insert_result_intoEPKcRNS_7IColumnE

Line

Count

Source

91

29

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

92

29

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

93

29

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

94

29

        dst.push_back(data.get());

95

29

    }

};

} // namespace doris