我有一堆点,我需要对它们进行最近邻搜索,所以我使用了libSpatialIndex。代码非常简单,库让我可以选择将数据存储在磁盘上,但我无法加载它。
代码:
int main(){
Tools::PropertySet* ps = GetDefaults();
Tools::Variant var;
// set index type to R*-Tree
var.m_varType = Tools::VT_ULONG;
var.m_val.ulVal = RT_RTree;
ps->setProperty("IndexType", var);
// Set index to store in disk
var.m_varType = Tools::VT_ULONG;
var.m_val.ulVal = RT_Disk;
ps->setProperty("IndexStorageType", var);
char filename[] = "indexTeste";
var.m_varType = Tools::VT_PCHAR;
var.m_val.pcVal = filename;
ps->setProperty("FileName", var);
var.m_varType = Tools::VT_BOOL;
var.m_val.blVal = false;
ps->setProperty("Overwrite", var);
cout << (*ps) << endl;
// initalise index
idx = new Index(*ps);
delete ps;
// Now there's specific code for point loading so I've shortened it - this part is working
for (...) { // all points
double pt[] = {point.getX(), point.getY()};
SpatialIndex::IShape* shape = 0;
shape = new SpatialIndex::Point(pt, 2);
// insert into index along with the an object and an ID
idx->index().insertData(nDataLength,(unsigned char*)&lineID,*shape,id);
}
// Now the search - working as well
ObjVisitor* visitor = new ObjVisitor;
SpatialIndex::Point* r = new SpatialIndex::Point(inter, 2);
idx->index().nearestNeighborQuery(1,*r,*visitor);
int64_t nResultCount;
nResultCount = visitor->GetResultCount();
// get actual results
vector<SpatialIndex::IData*>& results = visitor->GetResults();
SpatialIndex::IShape* shape;
results[0]->getShape(&shape);
unsigned char * dataAddr;
unsigned int length = sizeof(int);
results[0]->getData(length,&dataAddr);
int lineId = ((int*)dataAddr)[0];
SpatialIndex::Point center;
shape->getCenter(center);
}
该计划在此之后完成。确实在内存中创建了两个文件,&#34; indexTest.dat&#34; 8.8MB和&#34; indexTest.idx&#34; 0kB但是如果我在初始化之后立即执行查询或检查索引中的元素数量,它将失败并且树上只有一个节点。
我已经看过这些问题了: (Re)loading the R Tree with spatialindex library
C++ spatialindex library: loading/storing main memory RTree from/to disk
但是我没有成功,因为我使用了索引,当我直接使用RTree时,数据插入速度就慢了1000倍。
答案 0 :(得分:0)
我找到了解决方案。索引实例化树的ID,并且必须在创建索引时使用它才能正确加载文件。
代码:
//Example
// When storing
Tools::PropertySet* ps = GetDefaults();
Index* idx;
idx = new Index(*ps);
Tools::PropertySet properties = idx->GetProperties();
Tools::Variant vari = properties.getProperty("IndexIdentifier");
cout << "ID: " << vari.m_val.llVal << endl;
// when loading
Tools::PropertySet* ps = GetDefaults();
Tools::Variant var;
// Important
var.m_varType = Tools::VT_BOOL;
var.m_val.blVal = false;
ps->setProperty("Overwrite", var);
var.m_varType = Tools::VT_LONGLONG;
var.m_val.llVal = ID; // The number "couted" before
ps->setProperty("IndexIdentifier", var);
Index* idx;
idx = new Index(*ps);