mitkUnstructuredGridClusteringFilter.h

Go to the documentation of this file.

/*============================================================================
 
The Medical Imaging Interaction Toolkit (MITK)
 
Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.
 
Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.
 
============================================================================*/
 
#ifndef mitkUnstructuredGridClusteringFilter_h
#define mitkUnstructuredGridClusteringFilter_h
 
#include <MitkAlgorithmsExtExports.h>
 
#include <mitkCommon.h>
 
#include <mitkUnstructuredGrid.h>
#include <mitkUnstructuredGridToUnstructuredGridFilter.h>
#include <vtkIdList.h>
#include <vtkPoints.h>
#include <vtkSmartPointer.h>
 
namespace mitk
{
  class MITKALGORITHMSEXT_EXPORT UnstructuredGridClusteringFilter : public UnstructuredGridToUnstructuredGridFilter
  {
  public:
    mitkClassMacro(UnstructuredGridClusteringFilter, UnstructuredGridToUnstructuredGridFilter);
    itkFactorylessNewMacro(Self);
    itkCloneMacro(Self);
 
      itkSetMacro(eps, double);
      itkGetMacro(eps, double);
 
      itkSetMacro(MinPts, int);
      itkGetMacro(MinPts, int);
 
      itkSetMacro(Meshing, bool);
 
      virtual std::vector<mitk::UnstructuredGrid::Pointer> GetAllClusters();
 
    virtual int GetNumberOfFoundClusters();
 
  protected:
    UnstructuredGridClusteringFilter();
 
    ~UnstructuredGridClusteringFilter() override;
 
    void GenerateOutputInformation() override;
 
    void GenerateData() override;
 
  private:
    void ExpandCluster(int id, vtkIdList *pointIDs, vtkPoints *cluster, vtkPoints *inpPoints);
 
    mitk::UnstructuredGrid::Pointer m_UnstructGrid;
 
    std::vector<vtkSmartPointer<vtkPoints>> m_Clusters;
 
    std::vector<vtkSmartPointer<vtkDoubleArray>> m_DistanceArrays;
 
    double m_eps;
 
    int m_MinPts;
 
    bool m_Meshing;
 
    bool m_DistCalc;
  };
 
} // namespace mitk
 
#endif