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 //_MITKUNSTRUCTUREDGRIDCLUSTERINGFILTER_h__