#include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include namespace DB { namespace ErrorCodes { extern const int ILLEGAL_TYPE_OF_ARGUMENT; } template class FunctionPolygonsWithin : public IFunction { public: static inline const char * name; explicit FunctionPolygonsWithin() = default; static FunctionPtr create(ContextPtr) { return std::make_shared(); } String getName() const override { return name; } bool isVariadic() const override { return false; } size_t getNumberOfArguments() const override { return 2; } DataTypePtr getReturnTypeImpl(const DataTypes &) const override { return std::make_shared(); } bool isSuitableForShortCircuitArgumentsExecution(const DataTypesWithConstInfo & /*arguments*/) const override { return true; } ColumnPtr executeImpl(const ColumnsWithTypeAndName & arguments, const DataTypePtr & /*result_type*/, size_t input_rows_count) const override { auto res_column = ColumnUInt8::create(); auto & res_data = res_column->getData(); res_data.reserve(input_rows_count); callOnTwoGeometryDataTypes(arguments[0].type, arguments[1].type, [&](const auto & left_type, const auto & right_type) { using LeftConverterType = std::decay_t; using RightConverterType = std::decay_t; using LeftConverter = typename LeftConverterType::Type; using RightConverter = typename RightConverterType::Type; if constexpr (std::is_same_v, LeftConverter> || std::is_same_v, RightConverter>) throw Exception(ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT, "Any argument of function {} must not be Point", getName()); else { auto first = LeftConverter::convert(arguments[0].column->convertToFullColumnIfConst()); auto second = RightConverter::convert(arguments[1].column->convertToFullColumnIfConst()); /// NOLINTNEXTLINE(clang-analyzer-core.uninitialized.Assign) for (size_t i = 0; i < input_rows_count; ++i) { boost::geometry::correct(first[i]); boost::geometry::correct(second[i]); res_data.emplace_back(boost::geometry::within(first[i], second[i])); } } }); return res_column; } bool useDefaultImplementationForConstants() const override { return true; } }; template <> const char * FunctionPolygonsWithin::name = "polygonsWithinCartesian"; template <> const char * FunctionPolygonsWithin::name = "polygonsWithinSpherical"; REGISTER_FUNCTION(PolygonsWithin) { factory.registerFunction>(); factory.registerFunction>(); } }