itkOctreeNode.h

Go to the documentation of this file.

#ifndef __ITKOCTREENODE_H__
#define __ITKOCTREENODE_H__
/*********************************** MH-CRC IPL *****************************************
 * Iowa MH-CRC IPL C Header File
 * Copyright (C) 1998 Nancy C. Andreasen and Vincent A. Magnotta
 * File Name:        itkOctree.h
 * \author Hans J. Johnson
 * \brief Includes the basic class definitions for using Octtrees.
 *****************************************************************************************/
#include "itkMacro.h"
namespace itk {
  enum LeafIdentifier { ZERO=0,ONE=1,TWO=2,THREE=3,FOUR=4,FIVE=5,SIX=6,SEVEN=7 };


  //A forward-declaration
  class OctreeNodeBranch;
  class OctreeBase;
  class ITKCommon_EXPORT OctreeNode
  {
  public:
    OctreeNode(void);
    virtual ~OctreeNode(void);

    OctreeNode & GetChild(const enum LeafIdentifier ChildID) const;
    OctreeNode & GetChild(const enum LeafIdentifier ChildID);
    int GetColor(void) const;
    void SetColor( int NodeColor);
    void SetBranch(OctreeNodeBranch * NewBranch);
    bool IsNodeColored(void) const;
    inline void SetParentOctree(OctreeBase *parent)
    {
      m_Parent = parent;
    }
  protected:
  private:
    void RemoveChildren(void);

    OctreeNodeBranch * m_Branch; 
    OctreeBase *m_Parent;
  };

  class OctreeNodeBranch
  {
  public:
    OctreeNodeBranch(OctreeBase *parent)
    {
      for(int i = 0; i < 8; i++)
        m_Leaves[i].SetParentOctree(parent);
    }
    inline OctreeNode *GetLeaf(enum LeafIdentifier LeafID)
    {
      return &m_Leaves[LeafID];
    }
  private:
    OctreeNode m_Leaves[8];
  };
} //End of itk Namespace
#endif                          /* __ITKOCTREENODE_H__ */

---