mitkUnstructuredGridClusteringFilter.h

Go to the documentation of this file.

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

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center,
Division of Medical and Biological Informatics.
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 or http://www.mitk.org for details.

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

#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();

    virtual ~UnstructuredGridClusteringFilter();

    virtual void GenerateOutputInformation();

    virtual void GenerateData();

  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__