itkLaplacianDeformationQuadEdgeMeshFilter.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 itkLaplacianDeformationQuadEdgeMeshFilter_h
#define itkLaplacianDeformationQuadEdgeMeshFilter_h
 
#include "itkQuadEdgeMeshParamMatrixCoefficients.h"
#include "itkQuadEdgeMeshToQuadEdgeMeshFilter.h"
#include "ITKQuadEdgeMeshFilteringExport.h"
 
#include "itkConceptChecking.h"
 
#include <unordered_map>
 
namespace itk
{
class LaplacianDeformationQuadEdgeMeshFilterEnums
{
public:
  enum class Area : uint8_t
  {
    NONE = 0,

    MIXEDAREA
  };
};
// Define how to print enumeration
extern ITKQuadEdgeMeshFiltering_EXPORT std::ostream &
operator<<(std::ostream & out, const LaplacianDeformationQuadEdgeMeshFilterEnums::Area value);

template <class TInputMesh, class TOutputMesh, class TSolverTraits>
class ITK_TEMPLATE_EXPORT LaplacianDeformationQuadEdgeMeshFilter
  : public QuadEdgeMeshToQuadEdgeMeshFilter<TInputMesh, TOutputMesh>
{
public:
  ITK_DISALLOW_COPY_AND_MOVE(LaplacianDeformationQuadEdgeMeshFilter);

  using Self = LaplacianDeformationQuadEdgeMeshFilter;
  using Superclass = QuadEdgeMeshToQuadEdgeMeshFilter<TInputMesh, TOutputMesh>;
  using Pointer = SmartPointer<Self>;
  using ConstPointer = SmartPointer<const Self>;
 
  itkOverrideGetNameOfClassMacro(LaplacianDeformationQuadEdgeMeshFilter);

  using InputMeshType = TInputMesh;
  using typename Superclass::InputPointType;
 
  static constexpr unsigned int InputPointDimension = InputMeshType::PointDimension;

  using OutputMeshType = TOutputMesh;
  using typename Superclass::OutputCoordinateType;
  using typename Superclass::OutputPointType;
  using typename Superclass::OutputPointIdentifier;
  using typename Superclass::OutputQEPrimal;
  using typename Superclass::OutputVectorType;
  using typename Superclass::OutputQEIterator;
 
  static constexpr unsigned int OutputPointDimension = OutputMeshType::PointDimension;
 
  using SolverTraits = TSolverTraits;
  using ValueType = typename SolverTraits::ValueType;
  using MatrixType = typename SolverTraits::MatrixType;
  using VectorType = typename SolverTraits::VectorType;
 
  using CoefficientsComputationType = MatrixCoefficients<OutputMeshType>;

  void
  SetCoefficientsMethod(CoefficientsComputationType * iMethod)
  {
    this->m_CoefficientsMethod = iMethod;
    this->Modified();
  }

 
  using TriangleType = TriangleHelper<OutputPointType>;

  void
  SetConstrainedNode(OutputPointIdentifier vId, const OutputPointType & iP);

  void
  SetDisplacement(OutputPointIdentifier vId, const OutputVectorType & iV);

  bool
  GetDisplacement(OutputPointIdentifier vId, OutputVectorType & oV) const;

  void
  ClearConstraints();

  itkSetMacro(Order, unsigned int);
  itkGetMacro(Order, unsigned int);
 
  using AreaEnum = LaplacianDeformationQuadEdgeMeshFilterEnums::Area;
#if !defined(ITK_LEGACY_REMOVE)
  static constexpr AreaEnum NONE = AreaEnum::NONE;
  static constexpr AreaEnum MIXEDAREA = AreaEnum::MIXEDAREA;
#endif

  itkSetEnumMacro(AreaComputationType, AreaEnum);
  itkGetMacro(AreaComputationType, AreaEnum);
 
  itkConceptMacro(SameDimensionCheck1, (Concept::SameDimension<InputPointDimension, OutputPointDimension>));
  itkConceptMacro(SameDimensionCheck2, (Concept::SameDimension<InputPointDimension, 3>));
 
protected:
  LaplacianDeformationQuadEdgeMeshFilter();
  ~LaplacianDeformationQuadEdgeMeshFilter() override = default;
 
  using OutputMapPointIdentifier = std::unordered_map<OutputPointIdentifier, OutputPointIdentifier>;
  using OutputMapPointIdentifierIterator = typename OutputMapPointIdentifier::iterator;
  using OutputMapPointIdentifierConstIterator = typename OutputMapPointIdentifier::const_iterator;
 
  using ConstraintMapType = std::unordered_map<OutputPointIdentifier, OutputVectorType>;
  using ConstraintMapConstIterator = typename ConstraintMapType::const_iterator;
 
  struct HashOutputQEPrimal
  {
    size_t
    operator()(OutputQEPrimal * qe) const
    {
      return reinterpret_cast<size_t>(qe);
    }
  };
 
  using CoefficientMapType = std::unordered_map<OutputQEPrimal *, OutputCoordinateType, HashOutputQEPrimal>;
  using CoefficientMapConstIterator = typename CoefficientMapType::const_iterator;
 
  using AreaMapType = std::unordered_map<OutputPointIdentifier, OutputCoordinateType>;
  using AreaMapConstIterator = typename AreaMapType::const_iterator;
 
  using RowType = std::unordered_map<OutputPointIdentifier, OutputCoordinateType>;
  using RowIterator = typename RowType::iterator;
  using RowConstIterator = typename RowType::const_iterator;
 
  OutputMapPointIdentifier m_InternalMap{};
  ConstraintMapType        m_Constraints{};
  CoefficientMapType       m_CoefficientMap{};
  AreaMapType              m_MixedAreaMap{};
 
  CoefficientsComputationType * m_CoefficientsMethod{};
 
  unsigned int m_Order{ 1 };
  AreaEnum     m_AreaComputationType{ AreaEnum::NONE };
 
  void
  PrintSelf(std::ostream & os, Indent indent) const override;
 
  OutputCoordinateType
  ComputeMixedAreaForGivenVertex(OutputPointIdentifier iId);
  OutputCoordinateType
  ComputeMixedArea(OutputQEPrimal * iQE1, OutputQEPrimal * iQE2);
 
  virtual void
  ComputeVertexIdMapping();
 
  void
  ComputeLaplacianMatrix(MatrixType & ioL);
 
  void
  FillMatrixRow(OutputPointIdentifier iId, unsigned int iDegree, OutputCoordinateType iWeight, RowType & ioRow);

  void
  FillMatrix(MatrixType & iM, VectorType & iBx, VectorType & iBy, VectorType & iBz);

  void
  SolveLinearSystems(const MatrixType & iM,
                     const VectorType & iBx,
                     const VectorType & iBy,
                     const VectorType & iBz,
                     VectorType &       oX,
                     VectorType &       oY,
                     VectorType &       oZ);
 
 
private:
  struct Triple
  {
    Triple() = default;
    Triple(OutputPointIdentifier iV, OutputCoordinateType iWeight, unsigned int iDegree)
      : m_Id(iV)
      , m_Weight(iWeight)
      , m_Degree(iDegree)
    {}
 
    OutputPointIdentifier m_Id;
    OutputCoordinateType  m_Weight;
    unsigned int          m_Degree;
  };
};
} // end namespace itk
 
#include "itkLaplacianDeformationQuadEdgeMeshFilter.hxx"
 
#endif