itkOctree.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkOctree.h,v $
  Language:  C++
  Date:      $Date: 2009-02-06 20:53:14 $
  Version:   $Revision: 1.7 $

  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"
namespace itk {

enum
{
  B2_MASKFILE_BLACK = 0,
  B2_MASKFILE_WHITE = 1,
  B2_MASKFILE_GRAY = 2
};
enum OctreePlaneType 
{

};

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(Octree, 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 */