Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
2dot5
ClickHouse
提交
ba5d0a97
C
ClickHouse
项目概览
2dot5
/
ClickHouse
通知
3
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
C
ClickHouse
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
ba5d0a97
编写于
5月 23, 2020
作者:
A
Alexey Milovidov
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Avoid virtual calls
上级
d2cd941f
变更
1
显示空白变更内容
内联
并排
Showing
1 changed file
with
42 addition
and
19 deletion
+42
-19
src/Functions/pointInPolygon.cpp
src/Functions/pointInPolygon.cpp
+42
-19
未找到文件。
src/Functions/pointInPolygon.cpp
浏览文件 @
ba5d0a97
...
...
@@ -10,6 +10,7 @@
#include <Columns/ColumnFixedString.h>
#include <Columns/ColumnString.h>
#include <Columns/ColumnTuple.h>
#include <Columns/ColumnsNumber.h>
#include <Common/ObjectPool.h>
#include <Common/ProfileEvents.h>
#include <DataTypes/DataTypeArray.h>
...
...
@@ -19,6 +20,7 @@
#include <IO/WriteHelpers.h>
#include <Interpreters/ExpressionActions.h>
#include <Interpreters/Context.h>
#include <Interpreters/castColumn.h>
#include <string>
#include <memory>
...
...
@@ -212,10 +214,18 @@ public:
if
(
isTwoDimensionalArray
(
*
block
.
getByPosition
(
arguments
[
1
]).
type
))
{
ColumnPtr
polygon_column_float64
=
castColumn
(
block
.
getByPosition
(
arguments
[
1
]),
std
::
make_shared
<
DataTypeArray
>
(
std
::
make_shared
<
DataTypeArray
>
(
std
::
make_shared
<
DataTypeTuple
>
(
DataTypes
{
std
::
make_shared
<
DataTypeFloat64
>
(),
std
::
make_shared
<
DataTypeFloat64
>
()}))));
for
(
size_t
i
=
0
;
i
<
input_rows_count
;
++
i
)
{
polygon
.
clear
();
parsePolygonFromSingleColumn2D
(
*
poly
_col
,
i
,
polygon
);
parsePolygonFromSingleColumn2D
(
*
poly
gon_column_float64
,
i
,
polygon
);
PointInNonConstPolygonImpl
impl
(
polygon
);
size_t
point_index
=
point_is_const
?
0
:
i
;
...
...
@@ -224,10 +234,17 @@ public:
}
else
{
ColumnPtr
polygon_column_float64
=
castColumn
(
block
.
getByPosition
(
arguments
[
1
]),
std
::
make_shared
<
DataTypeArray
>
(
std
::
make_shared
<
DataTypeTuple
>
(
DataTypes
{
std
::
make_shared
<
DataTypeFloat64
>
(),
std
::
make_shared
<
DataTypeFloat64
>
()})));
for
(
size_t
i
=
0
;
i
<
input_rows_count
;
++
i
)
{
polygon
.
clear
();
parsePolygonFromSingleColumn1D
(
*
poly
_col
,
i
,
polygon
);
parsePolygonFromSingleColumn1D
(
*
poly
gon_column_float64
,
i
,
polygon
);
PointInNonConstPolygonImpl
impl
(
polygon
);
size_t
point_index
=
point_is_const
?
0
:
i
;
...
...
@@ -249,8 +266,14 @@ private:
void
parseConstPolygonFromSingleColumn
(
Block
&
block
,
const
ColumnNumbers
&
arguments
,
Polygon
&
out_polygon
)
const
{
const
auto
&
poly
=
*
block
.
getByPosition
(
arguments
[
1
]).
column
;
const
ColumnConst
&
column_const
=
typeid_cast
<
const
ColumnConst
&>
(
poly
);
ColumnPtr
polygon_column_float64
=
castColumn
(
block
.
getByPosition
(
arguments
[
1
]),
std
::
make_shared
<
DataTypeArray
>
(
std
::
make_shared
<
DataTypeTuple
>
(
DataTypes
{
std
::
make_shared
<
DataTypeFloat64
>
(),
std
::
make_shared
<
DataTypeFloat64
>
()})));
const
ColumnConst
&
column_const
=
typeid_cast
<
const
ColumnConst
&>
(
*
polygon_column_float64
);
const
IColumn
&
column_const_data
=
column_const
.
getDataColumn
();
if
(
isTwoDimensionalArray
(
*
block
.
getByPosition
(
arguments
[
1
]).
type
))
...
...
@@ -261,41 +284,41 @@ private:
template
<
typename
T
>
void
parsePolygonPart
(
const
IColumn
&
x_column
,
const
IColumn
&
y_column
,
const
Float64
*
x_data
,
const
Float64
*
y_data
,
size_t
begin
,
size_t
end
,
T
&
out_container
)
const
{
out_container
.
reserve
(
end
-
begin
);
for
(
size_t
i
=
begin
;
i
<
end
;
++
i
)
{
Float64
x
=
x_column
.
getFloat64
(
i
);
Float64
y
=
y_column
.
getFloat64
(
i
);
out_container
.
emplace_back
(
x
,
y
);
}
out_container
.
emplace_back
(
x_data
[
i
],
y_data
[
i
]);
}
void
parsePolygonFromSingleColumn1D
(
const
IColumn
&
column
,
size_t
i
,
Polygon
&
out_polygon
)
const
{
const
auto
&
array_col
=
static_cast
<
const
ColumnArray
&>
(
column
);
const
auto
&
tuple_columns
=
static_cast
<
const
ColumnTuple
&>
(
array_col
.
getData
()).
getColumns
();
size_t
begin
=
array_col
.
getOffsets
()[
i
-
1
];
size_t
end
=
array_col
.
getOffsets
()[
i
];
parsePolygonPart
(
*
tuple_columns
[
0
],
*
tuple_columns
[
1
],
begin
,
end
,
out_polygon
.
outer
());
const
auto
&
tuple_columns
=
static_cast
<
const
ColumnTuple
&>
(
array_col
.
getData
()).
getColumns
();
const
auto
*
x_data
=
static_cast
<
const
ColumnFloat64
&>
(
*
tuple_columns
[
0
]).
getData
().
data
();
const
auto
*
y_data
=
static_cast
<
const
ColumnFloat64
&>
(
*
tuple_columns
[
1
]).
getData
().
data
();
parsePolygonPart
(
x_data
,
y_data
,
begin
,
end
,
out_polygon
.
outer
());
}
void
parsePolygonFromSingleColumn2D
(
const
IColumn
&
column
,
size_t
i
,
Polygon
&
out_polygon
)
const
{
const
auto
&
array_col
=
static_cast
<
const
ColumnArray
&>
(
column
);
const
auto
&
nested_array_col
=
static_cast
<
const
ColumnArray
&>
(
array_col
.
getData
());
const
auto
&
tuple_columns
=
static_cast
<
const
ColumnTuple
&>
(
nested_array_col
.
getData
()).
getColumns
();
size_t
rings_begin
=
array_col
.
getOffsets
()[
i
-
1
];
size_t
rings_end
=
array_col
.
getOffsets
()[
i
];
const
auto
&
nested_array_col
=
static_cast
<
const
ColumnArray
&>
(
array_col
.
getData
());
const
auto
&
tuple_columns
=
static_cast
<
const
ColumnTuple
&>
(
nested_array_col
.
getData
()).
getColumns
();
const
auto
*
x_data
=
static_cast
<
const
ColumnFloat64
&>
(
*
tuple_columns
[
0
]).
getData
().
data
();
const
auto
*
y_data
=
static_cast
<
const
ColumnFloat64
&>
(
*
tuple_columns
[
1
]).
getData
().
data
();
for
(
size_t
j
=
rings_begin
;
j
<
rings_end
;
++
j
)
{
size_t
begin
=
nested_array_col
.
getOffsets
()[
j
-
1
];
...
...
@@ -303,12 +326,12 @@ private:
if
(
out_polygon
.
outer
().
empty
())
{
parsePolygonPart
(
*
tuple_columns
[
0
],
*
tuple_columns
[
1
]
,
begin
,
end
,
out_polygon
.
outer
());
parsePolygonPart
(
x_data
,
y_data
,
begin
,
end
,
out_polygon
.
outer
());
}
else
{
out_polygon
.
inners
().
emplace_back
();
parsePolygonPart
(
*
tuple_columns
[
0
],
*
tuple_columns
[
1
]
,
begin
,
end
,
out_polygon
.
inners
().
back
());
parsePolygonPart
(
x_data
,
y_data
,
begin
,
end
,
out_polygon
.
inners
().
back
());
}
}
}
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录