mitkPlaneProposer.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 MITKPLANEPROPOSER_H
#define MITKPLANEPROPOSER_H

#include <MitkSurfaceInterpolationExports.h>

#include <mitkPoint.h>
#include <mitkVector.h>

#include <itkSmartPointer.h>

#include <array>
#include <vector>

#include <mitkDataStorage.h>

namespace mitk
{
  class SliceNavigationController;
  class UnstructuredGrid;

  class MITKSURFACEINTERPOLATION_EXPORT PlaneProposer
  {
  public:
    struct PlaneInfo
    {
      mitk::Vector3D normal;
      mitk::Vector3D x;
      mitk::Vector3D y;
      mitk::Point3D pointOnPlane;
    };

    PlaneProposer();
    ~PlaneProposer();

    void SetUnstructuredGrids(std::vector<itk::SmartPointer<mitk::UnstructuredGrid>> &grids);

    void SetUseDistances(bool);

    void SetUseLeastSquares(bool);

    void SetNumberOfClustersToUse(unsigned int);
    void SetSliceNavigationController(itk::SmartPointer<mitk::SliceNavigationController> &snc);

    void CreatePlaneInfo();

    PlaneInfo GetProposedPlaneInfo();

    std::array<std::array<double, 3>, 3> GetCentroids();

  private:
    PlaneProposer(const PlaneProposer &);            // not implemented on purpose
    PlaneProposer &operator=(const PlaneProposer &); // not implemented on purpose

    PlaneInfo CreatePlaneByCentroids(const std::vector<std::pair<int, int>> &sizeIDs,
                                     const std::vector<std::pair<double, int>> &avgDistances);

    PlaneInfo CreatePlaneByLeastSquares(const std::vector<std::pair<int, int>> &sizeIDs,
                                        const std::vector<std::pair<double, int>> &avgDistances);

    std::vector<itk::SmartPointer<mitk::UnstructuredGrid>> m_Grids;
    bool m_UseDistances;
    bool m_UseLeastSquares;
    unsigned int m_NumberOfClustersToUse;
    std::array<std::array<double, 3>, 3> m_Centroids;
    itk::SmartPointer<mitk::SliceNavigationController> m_SNC;
    PlaneInfo m_ProposedPlaneInfo;
  };

} // namespace

#endif // MITKPLANEPROPOSER_H