itkOctree.h

Go to the documentation of this file.

/*=========================================================================

Program:   Insight Segmentation & Registration Toolkit
Module:    $RCSfile: itkOctree.h,v $
Language:  C++
Date:      $Date: 2003/09/10 14:29:19 $
Version:   $Revision: 1.4 $

Copyright (c) Insight Software Consortium. All rights reserved.
See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm for details.

This software is distributed WITHOUT ANY WARRANTY; without even 
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR 
PURPOSE.  See the above copyright notices for more information.

=========================================================================*/

#ifndef __ITKOCTREE_H__
#define __ITKOCTREE_H__

#include "itkOctreeNode.h"
#include "itkImage.h"
/*
 * Octree data structure
 */
namespace itk {

  enum
    {
      B2_MASKFILE_BLACK = 0,
      B2_MASKFILE_WHITE = 1,
      B2_MASKFILE_GRAY = 2
    };
  enum OctreePlaneType 
    {
      UNKNOWN_PLANE, 
      SAGITAL_PLANE, 
      CORONAL_PLANE, 
      TRANSVERSE_PLANE  
    };

  class OctreeBase : public Object
  {
  public:
    typedef OctreeBase Self;
    typedef SmartPointer<Self>  Pointer;

    virtual OctreeNode *GetTree() = 0;
    virtual unsigned int GetDepth() = 0;
    virtual unsigned int GetWidth() = 0;

    virtual void SetDepth(unsigned int depth) = 0;

    virtual void SetWidth(unsigned int width) = 0;
    virtual void BuildFromBuffer(const void *buffer,
                                 const int xsize,const int ysize,const int zsize) = 0;
    virtual const char *GetColorTable() const = 0;

    virtual int GetColorTableSize() const = 0;
  };

  template <class TPixel,unsigned int ColorTableSize,class MappingFunctionType>
  class Octree: public OctreeBase
  {
  public:
    typedef Octree            Self;
    typedef OctreeBase  Superclass;
    typedef SmartPointer<Self>  Pointer;
    typedef Image<TPixel,3> ImageType;
    typedef typename ImageType::Pointer ImageTypePointer;
    itkNewMacro(Self);

    itkTypeMacro(AnalyzeImageIO, Superclass);

    ImageTypePointer GetImage();
    virtual void BuildFromBuffer(const void *buffer,const int xsize,const int ysize,const int zsize);
    void BuildFromImage(Image<TPixel,3> *fromImage);

    Octree(void);
    ~Octree(void);
    void SetColor(unsigned int color) { m_Tree.SetColor(color); }
    void SetTree(OctreeNodeBranch *branch) { m_Tree.SetBranch(branch); }
    void SetTrueDims(const unsigned int Dim0, const unsigned int Dim1, 
                     const unsigned int Dim2);

    unsigned int GetValue(const unsigned int Dim0, const unsigned int Dim1, 
                           const unsigned int Dim2);
    
    virtual void SetWidth(unsigned int width);
    virtual void SetDepth(unsigned int depth);
    virtual unsigned int GetWidth();
    virtual unsigned int GetDepth();

    virtual OctreeNode *GetTree();
    virtual const char *GetColorTable() const;
    virtual int GetColorTableSize() const;
  private:
    Octree(const Self&);         // purposely not implemented
    void operator=(const Self&); // purposely not implemented

    OctreeNodeBranch *maskToOctree (const TPixel* Mask, unsigned width, unsigned x, 
                                    unsigned y, unsigned z, unsigned xsize, 
                                    unsigned ysize, unsigned zsize);
    enum OctreePlaneType m_Plane; // The orientation of the plane for this octree
    unsigned int m_Width;         // The width of the Octree 
                                  // ( This is always a power of 2, and large 
                                  // enough to contain MAX(DIMS[1,2,3]))
    unsigned int m_Depth;         // < The depth of the Octree
    unsigned int m_TrueDims[3];   // The true dimensions of the image
    char m_ColorTable[ColorTableSize];
    OctreeNode m_Tree;
    // OctreeColorMapFunction m_ColorMapFunction;
    MappingFunctionType m_MappingFunction;
  };

}

#ifndef ITK_MANUAL_INSTANTIATION
#include "itkOctree.txx"
#endif

#endif                          /* __ITKOCTREE_H__ */

---