octatree.h

快速fft变换的c实现

// OctaTree.h: interface for the OctaTree class.
//
//////////////////////////////////////////////////////////////////////

#if !defined(AFX_OCTATREE_H__61A3C994_7092_4683_9A61_C6AF09861772__INCLUDED_)
#define AFX_OCTATREE_H__61A3C994_7092_4683_9A61_C6AF09861772__INCLUDED_

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000

#include "ell.h"

class OctaTree  
{
private:
	float m_fOriginX;
	float m_fOriginY;
	float m_fOriginZ;
	float m_fSizeX;
	float m_fSizeY;
	float m_fSizeZ;
	int *m_pPointTable;
	Cell *m_pRoot;
public:
	void FitSubdomains(PointSet *ps, float *solution,  int leavesNum, float *b);
	int GetLeavesNum();
	void Draw();
	void SetSize(float sx, float sy, float sz);
	void SetOriginCoord(float ox, float oy, float oz);
	Cell* GetRoot();
	int* GetCoarseGrid(int num, int *indexTable, int n=0);
	int* GetPointTable();
	void GetPointIndexTable(PointSet *ps, int &tableSize, float x, float y, float  z, float r);
	void BuildTree(PointSet *ps, int MaxLevel);
	void GetSize(float &sx, float &sy, float &sz);
	void GetOriginCoord(float &ox, float &oy, float &oz);
	void SetPointSet(PointSet* ps);
	OctaTree();
	virtual ~OctaTree();

};

#endif // !defined(AFX_OCTATREE_H__61A3C994_7092_4683_9A61_C6AF09861772__INCLUDED_)