neighbour_searcher.h

Go to the documentation of this file.

#ifndef __NEIGHBOUR_SEARCHER_H__
#define __NEIGHBOUR_SEARCHER_H__



#ifdef _MSC_VER  
typedef __int32 int32_t; 
typedef unsigned __int32 uint32_t; 
typedef __int64 int64_t; 
typedef unsigned __int64 uint64_t;  
#else 
#include <stdint.h> 
#endif 


#include <cstdlib>
#include <vector>
#include <deque>
#include "octree.h"


class Initializer;


class NeighbourSearcher {

public:
  
    virtual ~NeighbourSearcher() {} 

    virtual int buildSearchStructure(const double* x, const double* y, const double* z, size_t numParticles) = 0;
    
    virtual int buildSearchStructure(const double* x, const double* y, const double* z, size_t begin, size_t numParticles) = 0;
 
#ifdef _OPENMP

    virtual int searchNeighbour(const double x, const double y, const double z, const double radius, int* result, double* distance, size_t& result_length, int tid, int index = -1) = 0;
#else

    virtual int searchNeighbour(const double x, const double y, const double z, const double radius, int* result, double* distance, size_t& result_length, int index = -1) = 0;
#endif

//virtual int searchNeighbour(const double x, const double y, const double z, const double radius, int* result, size_t& result_length, int index = -1) = 0;

  //virtual int searchNeighbour(const double x, const double y, const double z, const double radius, int* result, size_t& result_length) = 0;


};

class OctreeSearcher : public NeighbourSearcher {
public:
    OctreeSearcher(const Initializer& init);
    

    OctreeSearcher(size_t maxParticleNum, size_t maxNeighbourNum, int treedepth);
    

    virtual ~OctreeSearcher() {
        delete m_pOctree;
#ifdef _OPENMP
        for(int i=0 ; i<m_iTheadNum ; i++) {
            delete[] m_pSearchResult[i];
        }
        delete[] m_pSearchResult;
#else
        delete[] m_pSearchResult;
#endif
    }

public:
    
    virtual int buildSearchStructure(const double* x, const double* y, const double* z, size_t numParticles) {
        return buildSearchStructure(x, y, z, 0, numParticles);
    }
    

    virtual int buildSearchStructure(const double* x, const double* y, const double* z, size_t begin, size_t numParticles);
   
   
#ifdef _OPENMP

    virtual int searchNeighbour(const double x, const double y, const double z, const double radius, int* result, double* distance, size_t& result_length, int tid, int index = -1);
#else

    virtual int searchNeighbour(const double x, const double y, const double z, const double radius, int* result, double* distance, size_t& result_length, int index = -1);
#endif



private:
    Octree* m_pOctree;
    size_t m_iMaxParticleNum;
    size_t m_iMaxNeighborNum;

#ifdef _OPENMP
    int m_iTheadNum;
    SearchResult** m_pSearchResult;
    //vector<SearchResult*> m_pSearchResult;
#else
    SearchResult* m_pSearchResult;
#endif

};

#endif