niftkMeshMerger.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 niftkMeshMerger_h
#define niftkMeshMerger_h

#include <niftkInvalidArgumentException.h>

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

#include <vector>
#include <string>

#include <niftkCommonWin32ExportHeader.h>

namespace niftk {
  class NIFTKCOMMON_WINEXPORT MeshMerger {
  private:
    std::vector<int> m_DesiredLabels;
    bool m_UseImageLabels;
    const std::vector<std::vector<std::pair<int,int> > > *mpc_SubMeshLabels;

  public:
    void SetDesiredLabels(const std::vector<int> &desiredLabels) {
      m_DesiredLabels = desiredLabels;
    }

    const std::vector<int>& GetDesiredLabels(void) const {
      return m_DesiredLabels;
    }

    void SetMeshLabels(const std::vector<std::vector<std::pair<int,int> > > &subMeshLabels) {
      mpc_SubMeshLabels = &subMeshLabels;
    }


    void ToggleUseImageLabels(void) {
      m_UseImageLabels = !m_UseImageLabels;
    }
  private:
    vtkSmartPointer<vtkMultiBlockDataSet> *mpp_SubMeshes;
    vtkSmartPointer<vtkUnstructuredGrid> mp_OutputMesh;

  public:
    void SetInput(vtkSmartPointer<vtkMultiBlockDataSet> &inputSubMeshes) {
      mpp_SubMeshes = &inputSubMeshes;
    }

    vtkSmartPointer<vtkUnstructuredGrid> GetOutput(void) {
      return mp_OutputMesh;
    }

    void Update(void) throw (niftk::InvalidArgumentException);
  };
}
#endif /* MESHMERGER_H_ */