开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 并行计算 > 查看源码
NearestNeighborSearch3.h
资源名称：fastgrid-4.2.1-141-src.tar.gz　[点击查看]
上传用户：chinafayin
上传日期：2022-04-05
资源大小：153k
文件大小：3k
源码类别：

并行计算

开发平台：

Visual C++

NearestNeighborSearch3.h：源码内容
  1. /*
  2.     FastGrid (formerly AutoGrid)
  4.     Copyright (C) 2009 The Scripps Research Institute. All rights reserved.
  5.     Copyright (C) 2009 Masaryk University. All rights reserved.
  7.     AutoGrid is a Trade Mark of The Scripps Research Institute.
  9.     This program is free software; you can redistribute it and/or
  10.     modify it under the terms of the GNU General Public License
  11.     as published by the Free Software Foundation; either version 2
  12.     of the License, or (at your option) any later version.
  14.     This program is distributed in the hope that it will be useful,
  15.     but WITHOUT ANY WARRANTY; without even the implied warranty of
  16.     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17.     GNU General Public License for more details.
  19.     You should have received a copy of the GNU General Public License
  20.     along with this program; if not, write to the Free Software
  21.     Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  22. */
  24. #pragma once
  25. #include "ann/ANN.h"
  27. // Given a set of points, this class can find the nearest neighbor point from the given position in space.
  28. template<typename T>
  29. class NearestNeighborSearch3
  30. {
  31. public:
  32.     NearestNeighborSearch3(): releaseCoords(false), coords(0), tree(0), pointarray(0) {}
  34.     ~NearestNeighborSearch3()
  35.     {
  36.         if (tree)
  37.             delete tree;
  38.         if (pointarray)
  39.             delete [] pointarray;
  40.         if (releaseCoords && coords)
  41.             switch (stride)
  42.             {
  43.             case 3:
  44.                 delete [] (Vec3<T>*)coords;
  45.                 break;
  47.             case 4:
  48.                 delete [] (Vec4<T>*)coords;
  49.                 break;
  50.             }
  51.     }
  53.     void create(const Vec3<T> *points, int num, bool releaseCoordsOnDestroy = false)
  54.     {
  55.         create(&points->x, 3, num, releaseCoordsOnDestroy);
  56.     }
  58.     void create(const Vec4<T> *points, int num, bool releaseCoordsOnDestroy = false)
  59.     {
  60.         create(&points->x, 4, num, releaseCoordsOnDestroy);
  61.     }
  63.     int searchNearest(const Vec3<T> &point) const
  64.     {
  65.         int result;
  66.         double d;
  67.         const_cast<ANNkd_tree*>(tree)->annkSearch(const_cast<double*>(&point.x), 1, &result, &d, 0);
  68.         return result;
  69.     }
  71.     T getDistanceSqrOfNearest2(const Vec3<T> &point) const
  72.     {
  73.         int result[2];
  74.         double d[2];
  75.         const_cast<ANNkd_tree*>(tree)->annkSearch(const_cast<double*>(&point.x), 2, result, d, 0);
  76.         return Vec3<T>::DistanceSqr(*(Vec3<T>*)&coords[result[0]*stride], *(Vec3<T>*)&coords[result[1]*stride]);
  77.     }
  79. private:
  80.     bool releaseCoords;
  81.     int stride;
  82.     const T *coords;
  83.     ANNkd_tree *tree;
  84.     ANNpointArray pointarray;
  86.     void create(const T *points, int stride, int num, bool releaseCoordsOnDestroy)
  87.     {
  88.         this->stride = stride;
  89.         releaseCoords = releaseCoordsOnDestroy;
  90.         coords = points;
  91.         pointarray = new ANNpoint[num];
  92.         for (int i = 0; i < num; i++)
  93.             pointarray[i] = const_cast<double*>(&coords[i*stride]);
  95.         tree = new ANNkd_tree(pointarray, num, 3);
  96.     }
  97. };
  99. typedef NearestNeighborSearch3<int32> NearestNeighborSearch3i;
  100. typedef NearestNeighborSearch3<float> NearestNeighborSearch3f;
  101. typedef NearestNeighborSearch3<double> NearestNeighborSearch3d;