1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
|
#include "math_ir.h"
#include <util/system/byteorder.h>
#include <yql/essentials/public/langver/yql_langver.h>
#include <yql/essentials/public/udf/udf_helpers.h>
extern const char TagRoundingMode[] = "MathRoundingMode";
using TTaggedRoundingMode = NYql::NUdf::TTagged<ui32, TagRoundingMode>;
#define MATH_UDF_MAP(XX, XXL) \
XX(Pi, double(), 0) \
XX(E, double(), 0) \
XX(Eps, double(), 0) \
XX(RoundDownward, TTaggedRoundingMode(), 0) \
XX(RoundToNearest, TTaggedRoundingMode(), 0) \
XX(RoundTowardZero, TTaggedRoundingMode(), 0) \
XX(RoundUpward, TTaggedRoundingMode(), 0) \
XX(Abs, double(TAutoMap<double>), 0) \
XX(Acos, double(TAutoMap<double>), 0) \
XX(Asin, double(TAutoMap<double>), 0) \
XX(Asinh, double(TAutoMap<double>), 0) \
XX(Atan, double(TAutoMap<double>), 0) \
XX(Cbrt, double(TAutoMap<double>), 0) \
XX(Ceil, double(TAutoMap<double>), 0) \
XX(Cos, double(TAutoMap<double>), 0) \
XX(Cosh, double(TAutoMap<double>), 0) \
XX(Erf, double(TAutoMap<double>), 0) \
XX(ErfInv, double(TAutoMap<double>), 0) \
XX(ErfcInv, double(TAutoMap<double>), 0) \
XX(Exp, double(TAutoMap<double>), 0) \
XX(Exp2, double(TAutoMap<double>), 0) \
XX(Fabs, double(TAutoMap<double>), 0) \
XX(Floor, double(TAutoMap<double>), 0) \
XX(Lgamma, double(TAutoMap<double>), 0) \
XX(Rint, double(TAutoMap<double>), 0) \
XX(Sin, double(TAutoMap<double>), 0) \
XX(Sinh, double(TAutoMap<double>), 0) \
XX(Sqrt, double(TAutoMap<double>), 0) \
XX(Tan, double(TAutoMap<double>), 0) \
XX(Tanh, double(TAutoMap<double>), 0) \
XX(Tgamma, double(TAutoMap<double>), 0) \
XX(Trunc, double(TAutoMap<double>), 0) \
XX(Log, double(TAutoMap<double>), 0) \
XX(Log2, double(TAutoMap<double>), 0) \
XX(Log10, double(TAutoMap<double>), 0) \
XX(Atan2, double(TAutoMap<double>, TAutoMap<double>), 0) \
XX(Fmod, double(TAutoMap<double>, TAutoMap<double>), 0) \
XX(Hypot, double(TAutoMap<double>, TAutoMap<double>), 0) \
XX(Remainder, double(TAutoMap<double>, TAutoMap<double>), 0) \
XX(Pow, double(TAutoMap<double>, TAutoMap<double>), 0) \
XX(Ldexp, double(TAutoMap<double>, TAutoMap<int>), 0) \
XX(IsFinite, bool(TAutoMap<double>), 0) \
XX(IsInf, bool(TAutoMap<double>), 0) \
XX(IsNaN, bool(TAutoMap<double>), 0) \
XX(Sigmoid, double(TAutoMap<double>), 0) \
XX(FuzzyEquals, bool(TAutoMap<double>, TAutoMap<double>, TEpsilon), 1) \
XX(Mod, TOptional<i64>(TAutoMap<i64>, i64), 0) \
XX(Rem, TOptional<i64>(TAutoMap<i64>, i64), 0) \
XXL(Round, double(TAutoMap<double>, TPrecision), 1)
#define MATH_UDF_MAP_WITHOUT_IR(XX) \
XX(NearbyInt, TOptional<i64>(TAutoMap<double>, TTaggedRoundingMode), 0)
#ifdef DISABLE_IR
#define MATH_STRICT_UDF(name, signature, optionalArgsCount) \
SIMPLE_STRICT_UDF_WITH_OPTIONAL_ARGS(T##name, signature, optionalArgsCount) { \
TUnboxedValuePod res; \
name##IR(this, &res, valueBuilder, args); \
return res; \
}
#else
#define MATH_STRICT_UDF(name, signature, optionalArgsCount) \
SIMPLE_STRICT_UDF_WITH_IR(T##name, signature, optionalArgsCount, "/llvm_bc/Math", #name "IR") { \
TUnboxedValuePod res; \
name##IR(this, &res, valueBuilder, args); \
return res; \
}
#endif
#define MATH_STRICT_UDF_WITHOUT_IR(name, signature, optionalArgsCount) \
SIMPLE_STRICT_UDF_WITH_OPTIONAL_ARGS(T##name, signature, optionalArgsCount) { \
TUnboxedValuePod res; \
name##IR(this, &res, valueBuilder, args); \
return res; \
}
#define REGISTER_MATH_UDF(udfName, ...) T##udfName,
#define REGISTER_MATH_UDF_LAST(udfName, ...) T##udfName
using namespace NKikimr;
using namespace NUdf;
namespace {
const char SwapBytesUDF[] = "SwapBytes";
template <class TUserType>
class TSwapBytesFunc: public TBoxedValue {
private:
TSourcePosition Pos_;
TSwapBytesFunc(TSourcePosition pos)
: Pos_(pos)
{
}
TUnboxedValue Run(const IValueBuilder* valueBuilder, const TUnboxedValuePod* args) const override {
Y_UNUSED(valueBuilder);
if constexpr (sizeof(TUserType) == 1) {
return args[0];
}
return TUnboxedValuePod(SwapBytes(args[0].Get<TUserType>()));
}
public:
static void DeclareSignature(
TStringRef name,
TType* userType,
IFunctionTypeInfoBuilder& builder,
bool typesOnly)
{
Y_UNUSED(name);
Y_UNUSED(userType);
builder.SimpleSignature<TUserType(TAutoMap<TUserType>)>()
.IsStrict()
.SetMinLangVer(NYql::MakeLangVersion(2025, 3));
if (!typesOnly) {
builder.Implementation(new TSwapBytesFunc<TUserType>(builder.GetSourcePosition()));
}
}
};
extern const char epsilon[] = "Epsilon";
using TEpsilon = TNamedArg<double, epsilon>;
extern const char precision[] = "Precision";
using TPrecision = TNamedArg<int, precision>;
MATH_UDF_MAP(MATH_STRICT_UDF, MATH_STRICT_UDF)
MATH_UDF_MAP_WITHOUT_IR(MATH_STRICT_UDF_WITHOUT_IR)
SIMPLE_MODULE(TMathModule,
MATH_UDF_MAP_WITHOUT_IR(REGISTER_MATH_UDF)
TUserDataTypeFuncFactory<true, false, SwapBytesUDF, TSwapBytesFunc, ui8, ui16, ui32, ui64>,
MATH_UDF_MAP(REGISTER_MATH_UDF, REGISTER_MATH_UDF_LAST))
} // namespace
REGISTER_MODULES(TMathModule)
|
