2020-01-18 01:07:11 +00:00
|
|
|
#include <Functions/IFunctionImpl.h>
|
|
|
|
#include <Functions/FunctionHelpers.h>
|
|
|
|
#include <Functions/FunctionFactory.h>
|
|
|
|
#include <DataTypes/DataTypesNumber.h>
|
|
|
|
#include <DataTypes/getLeastSupertype.h>
|
|
|
|
#include <Core/ColumnNumbers.h>
|
|
|
|
|
|
|
|
|
|
|
|
namespace DB
|
|
|
|
{
|
2020-09-07 18:00:37 +00:00
|
|
|
namespace
|
|
|
|
{
|
2020-01-18 01:07:11 +00:00
|
|
|
|
|
|
|
/// ifNotFinite(x, y) is equivalent to isFinite(x) ? x : y.
|
|
|
|
class FunctionIfNotFinite : public IFunction
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
static constexpr auto name = "ifNotFinite";
|
|
|
|
|
2020-03-18 03:27:32 +00:00
|
|
|
explicit FunctionIfNotFinite(const Context & context_) : context(context_) {}
|
2020-01-18 01:07:11 +00:00
|
|
|
|
|
|
|
static FunctionPtr create(const Context & context)
|
|
|
|
{
|
|
|
|
return std::make_shared<FunctionIfNotFinite>(context);
|
|
|
|
}
|
|
|
|
|
|
|
|
std::string getName() const override
|
|
|
|
{
|
|
|
|
return name;
|
|
|
|
}
|
|
|
|
|
|
|
|
size_t getNumberOfArguments() const override { return 2; }
|
|
|
|
bool useDefaultImplementationForNulls() const override { return false; }
|
|
|
|
bool useDefaultImplementationForConstants() const override { return true; }
|
|
|
|
ColumnNumbers getArgumentsThatDontImplyNullableReturnType(size_t /*number_of_arguments*/) const override { return {0}; }
|
|
|
|
|
2020-01-18 13:06:02 +00:00
|
|
|
DataTypePtr getReturnTypeImpl(const ColumnsWithTypeAndName & arguments) const override
|
2020-01-18 01:07:11 +00:00
|
|
|
{
|
2020-01-18 13:06:02 +00:00
|
|
|
auto is_finite_type = FunctionFactory::instance().get("isFinite", context)->build({arguments[0]})->getReturnType();
|
|
|
|
auto if_type = FunctionFactory::instance().get("if", context)->build({{nullptr, is_finite_type, ""}, arguments[0], arguments[1]})->getReturnType();
|
|
|
|
return if_type;
|
2020-01-18 01:07:11 +00:00
|
|
|
}
|
|
|
|
|
2020-10-14 14:04:50 +00:00
|
|
|
void executeImpl(ColumnsWithTypeAndName & columns, const ColumnNumbers & arguments, size_t result, size_t input_rows_count) const override
|
2020-01-18 01:07:11 +00:00
|
|
|
{
|
2020-10-14 14:04:50 +00:00
|
|
|
ColumnsWithTypeAndName temp_columns = columns;
|
2020-01-18 01:07:11 +00:00
|
|
|
|
2020-10-14 14:04:50 +00:00
|
|
|
auto is_finite = FunctionFactory::instance().get("isFinite", context)->build({temp_columns[arguments[0]]});
|
2020-01-18 01:07:11 +00:00
|
|
|
|
2020-10-14 14:04:50 +00:00
|
|
|
size_t is_finite_pos = temp_columns.size();
|
|
|
|
temp_columns.emplace_back(ColumnWithTypeAndName{nullptr, is_finite->getReturnType(), ""});
|
2020-01-18 13:06:02 +00:00
|
|
|
|
2020-01-18 01:07:11 +00:00
|
|
|
auto func_if = FunctionFactory::instance().get("if", context)->build(
|
2020-10-14 14:04:50 +00:00
|
|
|
{temp_columns[is_finite_pos], temp_columns[arguments[0]], temp_columns[arguments[1]]});
|
2020-01-18 01:07:11 +00:00
|
|
|
|
2020-10-14 14:04:50 +00:00
|
|
|
is_finite->execute(temp_columns, {arguments[0]}, is_finite_pos, input_rows_count);
|
2020-01-18 13:06:02 +00:00
|
|
|
|
2020-10-14 14:04:50 +00:00
|
|
|
func_if->execute(temp_columns, {is_finite_pos, arguments[0], arguments[1]}, result, input_rows_count);
|
2020-01-18 01:07:11 +00:00
|
|
|
|
2020-10-14 14:04:50 +00:00
|
|
|
columns[result].column = std::move(temp_columns[result].column);
|
2020-01-18 01:07:11 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
private:
|
|
|
|
const Context & context;
|
|
|
|
};
|
|
|
|
|
2020-09-07 18:00:37 +00:00
|
|
|
}
|
2020-01-18 01:07:11 +00:00
|
|
|
|
|
|
|
void registerFunctionIfNotFinite(FunctionFactory & factory)
|
|
|
|
{
|
|
|
|
factory.registerFunction<FunctionIfNotFinite>();
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|