SQLite R*Tree 模块测试
SQLite R*Tree 模块测试
相关参考：
我另外做了GEOS STRtree/Quadtree 空间检索的性能，测试代码和数据可见Spatial_Index_Test
1、SQLite R*Tree 模块特性简介
关于SQLite的空间索引相关介绍可以查看官方文档 The SQLite R*Tree Module ，这里只做简单的介绍。
1、SQLite R *Tree模块实现部分在其源代码内（源码下载页面），无需另外合并。但是默认是没有启用的，启用需要定义SQLITE_ENABLE_RTREE=1宏再编译。
2、SQLite R *Tree模块采用虚拟表实现，每个R *Tree索引都是一个虚拟表。对于这个表，其第一列必须是64位有符号整数类型，作为主键。其它的列（2-12列）根据空间维度确定，每个维度包含一对（两列），分别是该维度的最小和最大值。例如：一维R *Tree索引虚拟表包含3列，分别是Int64主键| 最小值| 最大值；二维R*Tree索引虚拟表包含5列，分别是Int64主键| 第一维最小值| 第一维最大值| 第二维最小值| 第二维最大值；3、4、5维R*Tree索引虚拟表列数情况的以此论推，SQLite R *Tree实现不支持宽度超过5维的R *树。
3、对于各个维度的最大最小值列，SQLite中可以使用int32或者float32类型进行数据存储。与其它常规表中的列不同，这里存储就是二进制类型的值，而不是转换为字符串。如果在插入数据的时候，使用了这两者之外的类型，则会进行隐式转换。
-- 创建整型坐标rtree索引虚拟表
CREATE VIRTUAL TABLE intrtree USING rtree_i32(id,x0,x1,y0,y1,z0,z1);
-- 创建浮点型坐标rtree索引虚拟表
CREATE VIRTUAL TABLE floatrtree USING rtree(id,x0,x1,y0,y1,z0,z1);
4、SQLite R *Tree中查询并不限制查询的维度一定要与所查询的表中的维度一致，可以仅查询其中的某几个维度（如3维空间仅查询2个维度）。一般来说，约束（维度）越多，查询的范围框越小，速度越快。
5、默认情况下使用float32存储坐标值，当无法精确表示传入值时，下限坐标向下舍入，上限坐标向上舍入，因此边界框可能略大于指定，但永远不会变小。这在查询某个范围之外的数据时，可能会有极小的误差。
6、对于3.24.0之前的版本，SQLite R *Tree索引虚拟表仅能存储整数主键和坐标值列，其它的信息需要另存于其它表中（通过主键进行关联）。从3.24.0版本开始，SQLite R *Tree索引虚拟表可以存储任意类型数据的辅助列，辅助列必须以+开头，最多可以存储100个辅助列。
CREATE VIRTUAL TABLE demo_index2 USING rtree(
id, -- 64位整型主键
minX, maxX, -- X方向最小最大值
minY, maxY, -- Y方向最小最大值
+objname TEXT, -- 辅助列 文本类型
+objtype TEXT, -- 辅助列 文本类型
+boundary BLOB -- 辅助列 二进制数据
);
7、可以自定义R-Tree查询，以便实现非矩形框碰撞。这需要通过sqlite3_rtree_query_callback（新，3.8.5开始提供）或sqlite3_rtree_geometry_callback（旧）注册查询SQL语句和匹配检测回调。相关信息在SQLite网站上有详细介绍。
8、一个SQLite R *Tree会附带三个影子表，用于存储数据，分别是虚拟表名_node（存储节点） 虚拟表名_parent（存储父节点） 虚拟表名_rowid（存储节点的rowid）。
9、可以使用SELECT rtreecheck('虚拟表名')来对R-Tree索引进行完整性和正确性检查。
2、SQLite R*Tree 模块简单测试代码
写了一个简单的测试程序来测试一下R *Tree树的速度，结果还是可以的。（可用，并不是最佳）
我的机器环境是：
Windows 10 1903 x64专业版
AMD 锐龙 2600X
DDR4 2400 8G
编译器：VS2017 Native x64
使用本地文件的时候，十万条数据插入时间大概在2秒以内，查询一个5x5度大小的范围，时间基本在0.07秒以内；使用内存模式时，插入时间大概在1.8秒以内，查询一个5x5度大小的范围，时间基本在0.04秒以内。
注意：编译SQLite的时候要定义SQLITE_ENABLE_RTREE宏，开启RTree索引支持。
#include "sqlite/sqlite3.h"
#include<time.h>
#include <stdlib.h>
#include <stdio.h>
// 因为仅仅是进行一下试用测试，所以有些地方就没有处理，包括close
int main()
{
sqlite3* db = NULL;
int rc = sqlite3_open(":memory:", &db);
// int rc = sqlite3_open("D:/sqlite_rtree/test.db", &db);
if (rc != SQLITE_OK)
{
return -1;
}
char* errmsg;
// 创建RTree索引虚拟表
rc = sqlite3_exec(db,
"CREATE VIRTUAL TABLE demo_index USING rtree(id,minX, maxX,minY, maxY,+axucol INTEGER NOT NULL)",
NULL, NULL, &errmsg);
if (rc != SQLITE_OK)
{
printf("%4d Error:%sn", __LINE__, errmsg);
return -2;
}
// 开始计时
clock_t start = clock();
// 开启事物
if (sqlite3_exec(db, "begin", NULL, NULL, &errmsg) != SQLITE_OK) {
printf("%4d Error:%sn", __LINE__, errmsg);
return -2;
}
// 生成十万个大小在 边长在[0.002,0.202]度大小以内的数据(0.2~22.5公里左右)
srand(time(NULL)); // 初始化随机数种子
sqlite3_stmt *pStmt = NULL;
// 预处理SQL语句
if(sqlite3_prepare_v2(db,
"INSERT INTO demo_index VALUES(?,?,?,?,?,?)",
-1, &pStmt, NULL) != SQLITE_OK) {
printf("%4d Error:%sn", __LINE__, errmsg);
return -3;
}
// 逐个插入
for (int i = 0; i < 100000; ++i) {
// 生成在经纬度范围内的x,y
double x0 = ((double)rand() / (double)RAND_MAX) * 360 - 180;
double y0 = ((double)rand() / (double)RAND_MAX) * 180 - 90;
double x1 = x0 + 0.002 + ((double)rand() / (double)RAND_MAX)*0.2;
double y1 = y0 + 0.002 + ((double)rand() / (double)RAND_MAX)*0.2;
// 绑定数据
sqlite3_bind_int64(pStmt, 1, i);
sqlite3_bind_double(pStmt, 2, x0);
sqlite3_bind_double(pStmt, 3, x1);
sqlite3_bind_double(pStmt, 4, y0);
sqlite3_bind_double(pStmt, 5, y1);
sqlite3_bind_int(pStmt, 6, rand()%3);
// 执行
sqlite3_step(pStmt);
// 重置
sqlite3_reset(pStmt);
}
sqlite3_finalize(pStmt); //结束语句，释放语句句柄
// 结束事物
if (sqlite3_exec(db, "commit", NULL, NULL, &errmsg) != SQLITE_OK){
printf("%4d Error:%sn", __LINE__, errmsg);
return -2;
}
// 结束计时
clock_t end = clock();
double hs = (double)(end - start) * 1000 / CLOCKS_PER_SEC;
printf("插入总耗时: %lf msn", hs);
// 查询
// select * from test where NOT(maxX<74.254915 OR minX>79.765758 OR maxY< 24.214285 OR minY>29.725129) AND auxcol==2 ORDER BY id;
// 预处理SQL语句
pStmt = NULL;
if (sqlite3_prepare_v2(db,
"SELECT id,minX,minY,auxcol FROM demo_index WHERE NOT(maxX<? OR minX>? OR maxY<? OR minY>?) AND auxcol==1;",
-1, &pStmt, NULL) != SQLITE_OK) {
printf("%4d Error:%sn", __LINE__, errmsg);
return -4;
}
//-------------------------------------------------------------------------
// 输入查询的范围框数据
puts("Input x0,x1,y0,y1:");
double x0, x1, y0, y1;
scanf("%lf,%lf,%lf,%lf", &x0, &x1, &y0, &y1);
printf("-----------[%lf,%lf,%lf,%lf]-------------n", x0, x1, y0, y1);
// 开始计时
start = clock();
// 绑定查询范围数据
sqlite3_bind_double(pStmt, 1, x0);
sqlite3_bind_double(pStmt, 2, x1);
sqlite3_bind_double(pStmt, 3, y0);
sqlite3_bind_double(pStmt, 4, y1);
while (sqlite3_step(pStmt) == SQLITE_ROW) {
int id = sqlite3_column_int(pStmt, 0);
double x = sqlite3_column_double(pStmt, 1);
double y = sqlite3_column_double(pStmt, 2);
int auxcol = sqlite3_column_int(pStmt, 3);
// 可以把输出去掉，减少对时间统计的影响
printf("%dt %lf,%lfn", id, x, y);
}
sqlite3_reset(pStmt); // 这里只查询一次可以没有，如果需要多次使用这个查询语句，则必须有，不然查出数据不对
sqlite3_finalize(pStmt); //结束语句，释放语句句柄
// 结束计时
end = clock();
hs = (double)(end - start) * 1000 / CLOCKS_PER_SEC;
printf("本次查询总耗时: %lf msn", hs);
sqlite3_close(db);
system("pause");
return 0;
}
