itkDiscretePrincipalCurvaturesQuadEdgeMeshFilter.h

Go to the documentation of this file.

/*=========================================================================
 *
 *  Copyright NumFOCUS
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *         https://www.apache.org/licenses/LICENSE-2.0.txt
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 *
 *=========================================================================*/
#ifndef itkDiscretePrincipalCurvaturesQuadEdgeMeshFilter_h
#define itkDiscretePrincipalCurvaturesQuadEdgeMeshFilter_h
 
#include "itkDiscreteCurvatureQuadEdgeMeshFilter.h"
#include "itkQuadEdgeMeshParamMatrixCoefficients.h"
 
namespace itk
{
template <typename TInputMesh, typename TOutputMesh = TInputMesh>
class DiscretePrincipalCurvaturesQuadEdgeMeshFilter
  : public DiscreteCurvatureQuadEdgeMeshFilter<TInputMesh, TOutputMesh>
{
public:
  ITK_DISALLOW_COPY_AND_MOVE(DiscretePrincipalCurvaturesQuadEdgeMeshFilter);
 
  using Self = DiscretePrincipalCurvaturesQuadEdgeMeshFilter;
  using Pointer = SmartPointer<Self>;
  using ConstPointer = SmartPointer<const Self>;
  using Superclass = DiscreteCurvatureQuadEdgeMeshFilter<TInputMesh, TOutputMesh>;
 
  using typename Superclass::InputMeshType;
  using typename Superclass::InputMeshPointer;
  using typename Superclass::OutputMeshType;
  using typename Superclass::OutputMeshPointer;
  using typename Superclass::OutputPointsContainerPointer;
  using typename Superclass::OutputPointsContainerIterator;
  using typename Superclass::OutputPointType;
  using typename Superclass::OutputVectorType;
  using typename Superclass::OutputCoordType;
  using typename Superclass::OutputPointIdentifier;
  using typename Superclass::OutputCellIdentifier;
  using typename Superclass::OutputQEType;
  using typename Superclass::OutputMeshTraits;
  using typename Superclass::OutputCurvatureType;
 
  using typename Superclass::TriangleType;
 
  itkOverrideGetNameOfClassMacro(DiscretePrincipalCurvaturesQuadEdgeMeshFilter);
 
  using CoefficientType = ConformalMatrixCoefficients<OutputMeshType>;
 
#ifdef ITK_USE_CONCEPT_CHECKING
  // Begin concept checking
  itkConceptMacro(OutputIsFloatingPointCheck, (Concept::IsFloatingPoint<OutputCurvatureType>));
  // End concept checking
#endif
 
protected:
  DiscretePrincipalCurvaturesQuadEdgeMeshFilter()
    : m_Gaussian(0.0)
    , m_Mean(0.0)
  {}
  ~DiscretePrincipalCurvaturesQuadEdgeMeshFilter() override = default;
 
  OutputCurvatureType m_Gaussian{};
  OutputCurvatureType m_Mean{};
 
  void
  ComputeMeanAndGaussianCurvatures(const OutputPointType & iP)
  {
    const OutputMeshPointer output = this->GetOutput();
 
    OutputQEType * qe = iP.GetEdge();
 
    m_Mean = 0.;
    m_Gaussian = 0.;
 
    if (qe != nullptr)
    {
      OutputVectorType Laplace{};
 
      OutputQEType * qe_it = qe;
 
      OutputCurvatureType area(0.);
      OutputCurvatureType sum_theta(0.);
 
      if (qe_it != qe_it->GetOnext())
      {
        qe_it = qe;
 
        OutputVectorType      normal{};
        const CoefficientType coefficent;
        do
        {
          OutputQEType *        qe_it2 = qe_it->GetOnext();
          const OutputPointType q0 = output->GetPoint(qe_it->GetDestination());
          const OutputPointType q1 = output->GetPoint(qe_it2->GetDestination());
 
          const OutputCoordType temp_coeff = coefficent(output, qe_it);
          Laplace += temp_coeff * (iP - q0);
 
          // Compute Angle;
          sum_theta += static_cast<OutputCurvatureType>(TriangleType::ComputeAngle(q0, iP, q1));
 
          const OutputCurvatureType temp_area = this->ComputeMixedArea(qe_it, qe_it2);
          area += temp_area;
 
          const OutputVectorType face_normal = TriangleType::ComputeNormal(q0, iP, q1);
          normal += face_normal;
 
          qe_it = qe_it2;
        } while (qe_it != qe);
 
        if (area > 1e-10)
        {
          area = 1. / area;
          Laplace *= 0.25 * area;
          m_Mean = Laplace * normal;
          m_Gaussian = (2. * itk::Math::pi - sum_theta) * area;
        }
      }
    }
  }
 
  virtual OutputCurvatureType
  ComputeDelta()
  {
    return std::max(static_cast<OutputCurvatureType>(0.), m_Mean * m_Mean - m_Gaussian);
  }
};
} // namespace itk
 
#endif