vcglib/apps/sample/hashing_2D/test_hash2D.cpp

383 lines
12 KiB
C++

/****************************************************************************
* VCGLib o o *
* Visual and Computer Graphics Library o o *
* _ O _ *
* Copyright(C) 2004-2009 \/)\/ *
* Visual Computing Lab /\/| *
* ISTI - Italian National Research Council | *
* \ *
* All rights reserved. *
* *
* This program is free software; you can redistribute it and/or modify *
* it under the terms of the GNU General Public License as published by *
* the Free Software Foundation; either version 2 of the License, or *
* (at your option) any later version. *
* *
* This program is distributed in the hope that it will be useful, *
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
* GNU General Public License (http://www.gnu.org/licenses/gpl.txt) *
* for more details. *
* *
****************************************************************************/
#include <stdio.h>
#include <time.h>
#include <vcg/complex/used_types.h>
#include <vcg/space/distance2.h>
#include<vcg/space/index/index2D/spatial_hashing_2D.h>
#include<vcg/space/intersection2.h>
typedef double MyScalarType;
class MySegmentType:public vcg::Segment2<MyScalarType>
{
public:
int mark;
bool deleted;
bool IsD(){return deleted;}
typedef vcg::Point2<ScalarType> CoordType;
MySegmentType(const vcg::Point2<MyScalarType> &_P0,
const vcg::Point2<MyScalarType> &_P1)
{
P0()=_P0;
P1()=_P1;
mark=0;
}
void GetBBox(vcg::Box2<ScalarType> &BB2)
{
//BB2.SetNull();
BB2.Set(P0());
BB2.Add(P1());
}
void GetSubBBox(const ScalarType &step_size,
std::vector<vcg::Box2<ScalarType> > &RasterBox)
{
//RasterBox.clear();
ScalarType lenght=(P1()-P0()).Norm();
CoordType dir=(P1()-P0());
dir.Normalize();
int steps= (int)ceil(lenght/(ScalarType)step_size);
RasterBox.resize(steps);
CoordType currP0=P0();
CoordType currP1;
for (int i=0;i<steps-1;i++)
{
currP1=currP0+dir*step_size;
RasterBox[i]=(vcg::Box2<ScalarType>(currP0,currP1));
currP0=currP1;
}
RasterBox[steps-1]=(vcg::Box2<ScalarType>(currP0,P1()));
}
MySegmentType(){}
MySegmentType(const MySegmentType &s1):vcg::Segment2<MyScalarType>(s1)
{
P0()=s1.P0();
P1()=s1.P1();
mark=s1.mark;
deleted=s1.deleted;
}
};
//**MARKER CLASSES**//
class MyMarker
{
public:
int mark;
MyMarker(){mark=0;}
//MyMarker( MESH_TYPE *m) {SetMesh(m);}
void UnMarkAll(){mark++;}
bool IsMarked(MySegmentType* obj)
{return(obj->mark==mark);}
void Mark(MySegmentType* obj)
{obj->mark=mark;}
/*void SetMesh(MESH_TYPE *_m)
{m=_m;}*/
};
vcg::SpatialHashTable2D<MySegmentType,MyScalarType> Hash2D;
std::vector<MySegmentType> Allocated;
MyMarker MyMark;
void RandomSeg(vcg::Point2<MyScalarType> &P0,
vcg::Point2<MyScalarType> &P1,
MyScalarType SpaceSize=100,
MyScalarType maxdim=0.01)
{
MyScalarType dimAbs=SpaceSize*maxdim;
int dimension=RAND_MAX;
int X=rand();
int Y=rand();
int dX=rand();
int dY=rand();
MyScalarType size=((MyScalarType)(rand()))/(MyScalarType)dimension;
P0=vcg::Point2<MyScalarType>((MyScalarType)X/dimension,(MyScalarType)Y/dimension);
P0*=SpaceSize;
vcg::Point2<MyScalarType> D=vcg::Point2<MyScalarType>((MyScalarType)dX/dimension,(MyScalarType)dY/dimension);
D.Normalize();
D*=size*dimAbs;
P1=P0+D;
}
void InitRandom(int num,
MyScalarType SpaceSize=100,
MyScalarType maxdim=0.01)
{
Allocated.clear();
Allocated.resize(num);
srand(clock());
for (int i=0;i<num;i++)
{
vcg::Point2<MyScalarType> P0,P1;
RandomSeg(P0,P1,SpaceSize,maxdim);
Allocated[i]=MySegmentType(P0,P1);
Allocated[i].deleted=false;
}
}
MyScalarType TestBox(int num_test=100000,
MyScalarType SpaceSize=100,
MyScalarType maxdim=0.02)
{
//GetInBox(OBJMARKER & _marker,const Box2x _bbox,OBJPTRCONTAINER & _objectPtrs)
MyMark.UnMarkAll();
//int t0=clock();
int num=0;
for (int i=0;i<num_test;i++)
{
vcg::Point2<MyScalarType> P0,P1;
RandomSeg(P0,P1,SpaceSize,maxdim);
vcg::Box2<MyScalarType> bbox;
bbox.Add(P0);
bbox.Add(P1);
std::vector<MySegmentType*> result;
num+=Hash2D.GetInBox<MyMarker,std::vector<MySegmentType*> >(MyMark,bbox,result);
}
//int t1=clock();
MyScalarType numd=(double)num/(double)num_test;
return numd;
}
MyScalarType GetIntersectingSegments(MySegmentType *S,
std::vector<MySegmentType*> &result,
bool subdivide=false)
{
///get the bbox
result.clear();
///then get into the grid
std::vector<MySegmentType*> inbox;
int num=0;
if (!subdivide)
{
vcg::Box2<MyScalarType> bbox;
S->GetBBox(bbox);
num=Hash2D.GetInBox<MyMarker,std::vector<MySegmentType*> >(MyMark,bbox,inbox);
}
else
{
std::vector<vcg::Box2<MyScalarType> > bbox;
MyScalarType size_cell=Hash2D.cell_size;
S->GetSubBBox(size_cell,bbox);
num=Hash2D.GetInBoxes<MyMarker,std::vector<MySegmentType*> >(MyMark,bbox,inbox);
}
///then test intersection
for (int j=0;j<num;j++)
{
if (inbox[j]==S)continue;
vcg::Point2<MyScalarType> p_inters;
if (vcg::SegmentSegmentIntersection<MyScalarType>(*S,*inbox[j],p_inters))
result.push_back(inbox[j]);
}
return (((MyScalarType)num-result.size())/(MyScalarType)num);
}
MyScalarType GetCloseSegments(MySegmentType *S,
const MyScalarType &radius,
std::vector<MySegmentType*> &result,
bool use_sub=false)
{
///get the bbox
result.clear();
std::vector<MySegmentType*> inbox;
int num=0;
if (!use_sub)
{
vcg::Box2<MyScalarType> bbox;
S->GetBBox(bbox);
bbox.Offset(radius);//*1.02);
///then get into the grid
num=Hash2D.GetInBox<MyMarker,std::vector<MySegmentType*> >(MyMark,bbox,inbox);
}
else
{
std::vector<vcg::Box2<MyScalarType> > bbox;
MyScalarType size_cell=Hash2D.cell_size;
S->GetSubBBox(size_cell,bbox);
for (int i=0;i<bbox.size();i++)
bbox[i].Offset(radius);//*1.02);
///then get into the grid
num=Hash2D.GetInBoxes<MyMarker,std::vector<MySegmentType*> >(MyMark,bbox,inbox);
}
///then test intersection
for (int j=0;j<num;j++)
{
if (inbox[j]==S)continue;
vcg::Point2<MyScalarType> p_clos;
MyScalarType dist=vcg::Segment2DSegment2DDistance<MyScalarType>(*S,*inbox[j],p_clos);
if (dist<radius)
result.push_back(inbox[j]);
}
return (((MyScalarType)num-result.size())/(MyScalarType)num);
}
MyScalarType TestIntersection(unsigned int num_test=1000000,bool use_sub=false)
{
MyScalarType false_pos=0;
for (unsigned int i=0;i<num_test;i++)
{
assert(i<Allocated.size());
std::vector<MySegmentType*> result;
MyScalarType false_pos_t=GetIntersectingSegments(&Allocated[i],result,use_sub);
false_pos+=false_pos_t;
}
return (false_pos/(MyScalarType)num_test);
}
MyScalarType TestClosest(unsigned int num_test=1000000,
MyScalarType radius=0.1,
bool use_sub=false)
{
MyScalarType false_pos=0;
for (unsigned int i=0;i<num_test;i++)
{
assert(i<Allocated.size());
//get the segment
MySegmentType *S=&Allocated[i];
MyScalarType absRadius=S->Length()*radius;
///get the segments closer than a radius
std::vector<MySegmentType*> closer;
MyScalarType false_pos_t=GetCloseSegments(S,absRadius,closer,use_sub);
false_pos+=false_pos_t;
}
return (false_pos/(MyScalarType)num_test);
}
int TestCorrectIntersect(int num_test=1000,bool use_sub=false)
{
int num=0;
for (int i=0;i<num_test;i++)
{
MySegmentType S0=Allocated[i];
std::vector<MySegmentType*> result0,result1;
for (int j=0;j<num_test;j++)
{
if (j==i) continue;
MySegmentType *S1=&Allocated[j];
vcg::Point2<MyScalarType> p_inters;
if (vcg::SegmentSegmentIntersection<MyScalarType>(S0,*S1,p_inters))
result0.push_back(S1);
/*num+=result0.size();*/
}
GetIntersectingSegments(&Allocated[i],result1,use_sub);
///then see if equal number
if (result1.size()==result0.size())num++;
}
return (num);
}
int TestCorrectClosest(int num_test=1000,
MyScalarType radius=0.1,
bool use_sub=false)
{
int num=0;
for (int i=0;i<num_test;i++)
{
MySegmentType *S0=&Allocated[i];
std::vector<MySegmentType*> result0,result1;
MyScalarType absRadius=S0->Length()*radius;
for (int j=0;j<num_test;j++)
{
if (j==i) continue;
MySegmentType *S1=&Allocated[j];
vcg::Point2<MyScalarType> p_clos;
MyScalarType dist=vcg::Segment2DSegment2DDistance<MyScalarType>(*S0,*S1,p_clos);
if (dist<absRadius)
result0.push_back(S1);
/*num+=result0.size();*/
}
GetCloseSegments(S0,absRadius,result1,use_sub);
///then see if equal number
if (result1.size()==result0.size())num++;
}
return (num);
}
int main( int argc, char **argv )
{
bool use_sub=true;
(void) argc;
(void) argv;
int num_sample=20000;
int t0=clock();
InitRandom(num_sample,100,0.3);
int t1=clock();
///Initialization performance
printf("** Time elapsed for initialization of %d sample is %d\n \n",num_sample,t1-t0);
Hash2D.Set(Allocated.begin(),Allocated.end(),use_sub);
///Box Query performance
t0=clock();
MyScalarType avg_test=TestBox(num_sample);
t1=clock();
printf("** Time elapsed for %d BOX queries is %d\n, average found %5.5f \n \n",num_sample,t1-t0,avg_test);
///Intersecting segment performance
t0=clock();
MyScalarType perc_int=TestIntersection(num_sample,use_sub);
t1=clock();
printf("** Time elapsed for %d INTERSECTION queries is %d\n, false positive perc found %5.5f \n \n",num_sample,t1-t0,perc_int);
///closest test
t0=clock();
MyScalarType perc_clos=TestClosest(num_sample,0.1,use_sub);
t1=clock();
printf("** Time elapsed for %d CLOSEST queries is %d\n, false positive perc found %5.5f \n \n",num_sample,t1-t0,perc_clos);
///reinitialize structure
MyMark.mark=0;
Hash2D.Clear();
int n_test=1000;
InitRandom(n_test,100,0.1);
Hash2D.Set(Allocated.begin(),Allocated.end(),use_sub);
int tested_int=TestCorrectIntersect(n_test,use_sub);
printf("** Correct Intersect on %d test are %d \n",n_test,tested_int);
int tested_clos=TestCorrectClosest(n_test,0.1,use_sub);
printf("** Correct Closest on %d test are %d \n",n_test,tested_clos);
return 0;
}