niftkMeshGenerator.h

Go to the documentation of this file.

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

  NifTK: A software platform for medical image computing.

  Copyright (c) University College London (UCL). All rights reserved.

  This software is distributed WITHOUT ANY WARRANTY; without even
  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  PURPOSE.

  See LICENSE.txt in the top level directory for details.

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

#ifndef niftkMeshGenerator_h
#define niftkMeshGenerator_h

#include <string>
#include <vector>
#include <utility>

#include <vtkSmartPointer.h>
#include <vtkMultiBlockDataSet.h>
#include <itkImage.h>

#include <niftkIOException.h>

#include <niftkCommonWin32ExportHeader.h>

namespace niftk {
  class NIFTKCOMMON_WINEXPORT MeshGenerator {
  public:
    typedef itk::Image<unsigned char, 3> ITKImageType;

  private:
    std::string m_InputFileName;
    vtkSmartPointer<vtkMultiBlockDataSet> mp_OutputMeshes;
    std::vector<std::vector<std::pair<int,int> > > m_SubMeshLabels;
    bool m_DoSurface;

  public:
    void SetFileName(const std::string &fileName) {
      m_InputFileName = fileName;
    }

    vtkSmartPointer<vtkMultiBlockDataSet>& GetOutput(void) {
      return mp_OutputMeshes;
    }

    const std::vector<std::vector<std::pair<int, int> > >& GetMeshLabels(void) const {
      return m_SubMeshLabels;
    }
  private:
     float m_facetAngle, m_facetEdgeLength, m_facetApproximationError;
     float m_cellSize, m_cellEdgeRadiusRatio;

  public:
    void SetFacetMinAngle(const float facetAngle) {
      m_facetAngle = facetAngle;
    }

    void SetFacetMaxEdgeLength(const float facetEdgeLength) {
      m_facetEdgeLength = facetEdgeLength;
    }

    void SetBoundaryApproximationError(const float facetApproximationError) {
      m_facetApproximationError = facetApproximationError;
    }

    void SetCellMaxSize(const float cellSize) {
      m_cellSize = cellSize;
    }

    void SetCellMaxRadiusEdgeRatio(const float cellEdgeRadiusRatio) {
      m_cellEdgeRadiusRatio = cellEdgeRadiusRatio;
    }
  private:
    void _ComputeMeshLabels(void);

  public:
    void SetDoSurface(const bool doSurface)
    {
      m_DoSurface = doSurface;
    }

    void Update(void) throw (niftk::IOException);
  public:
    MeshGenerator(void);
  };
} /* namespace niftk */

#endif /* MESHGENERATOR_H_ */