mitkPlaneFit.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 mitkPlaneFit_h
#define mitkPlaneFit_h
 
#include <MitkAlgorithmsExtExports.h>
#include <mitkGeometryDataSource.h>
#include <mitkPlaneGeometry.h>
#include <mitkPointSet.h>
#include <mitkTimeGeometry.h>
 
namespace mitk
{
  class MITKALGORITHMSEXT_EXPORT PlaneFit : public GeometryDataSource
  {
  public:
    mitkClassMacro(PlaneFit, GeometryDataSource);
    itkNewMacro(Self);
 
    typedef mitk::PointSet::PointDataType PointDataType;
    typedef mitk::PointSet::PointDataIterator PointDataIterator;
 
    void GenerateOutputInformation() override;
 
    void GenerateData() override;
 
    const mitk::PointSet *GetInput();
 
    using mitk::GeometryDataSource::SetInput;
    virtual void SetInput(const mitk::PointSet *ps);
 
    virtual const mitk::Point3D &GetCentroid(int t = 0) const;
 
    virtual mitk::PlaneGeometry::Pointer GetPlaneGeometry(int t = 0);
 
    virtual const mitk::Vector3D &GetPlaneNormal(int t = 0) const;
 
  protected:
    PlaneFit();
    ~PlaneFit() override;
 
    void CalculateCentroid(int t = 0);
 
    void ProcessPointSet(int t = 0);
 
    void InitializePlane(int t = 0);
 
  private:
    const mitk::PointSet *m_PointSet;
 
    /* output object - a time sliced geometry.*/
    mitk::TimeGeometry::Pointer m_TimeGeometry;
 
    std::vector<mitk::PlaneGeometry::Pointer> m_Planes;
 
    std::vector<mitk::Point3D> m_Centroids;
 
    /* the normal vector to describe a plane geometry.*/
    std::vector<mitk::Vector3D> m_PlaneVectors;
  };
} // namespace mitk
#endif