itkLaplacianDeformationQuadEdgeMeshFilter.h

Go to the documentation of this file.

/*=========================================================================
 *
 *  Copyright Insight Software Consortium
 *
 *  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
 *
 *         http://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 "itkConceptChecking.h"

#include "itksys/hash_map.hxx"

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

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

  itkTypeMacro(LaplacianDeformationQuadEdgeMeshFilter, QuadEdgeMeshToQuadEdgeMeshFilter)


  using InputMeshType = TInputMesh;
  using InputPointType = typename Superclass::InputPointType;

  static constexpr unsigned int InputPointDimension = InputMeshType::PointDimension;

  using OutputMeshType = TOutputMesh;
  using OutputCoordRepType = typename Superclass::OutputCoordRepType;
  using OutputPointType = typename Superclass::OutputPointType;
  using OutputPointIdentifier = typename Superclass::OutputPointIdentifier;
  using OutputQEPrimal = typename Superclass::OutputQEPrimal;
  using OutputVectorType = typename Superclass::OutputVectorType;
  using OutputQEIterator = 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);

  enum AreaType
  {
    NONE = 0,

    MIXEDAREA
  };

  itkSetMacro(AreaComputationType, AreaType );
  itkGetMacro(AreaComputationType, AreaType );

#ifdef ITK_USE_CONCEPT_CHECKING
  itkConceptMacro( SameDimensionCheck1,
                   ( Concept::SameDimension< InputPointDimension, OutputPointDimension > ) );
  itkConceptMacro( SameDimensionCheck2,
                   ( Concept::SameDimension< InputPointDimension, 3 > ) );
#endif

protected:

  LaplacianDeformationQuadEdgeMeshFilter();
  ~LaplacianDeformationQuadEdgeMeshFilter() override = default;

  using OutputMapPointIdentifier = itksys::hash_map< OutputPointIdentifier, OutputPointIdentifier >;
  using OutputMapPointIdentifierIterator = typename OutputMapPointIdentifier::iterator;
  using OutputMapPointIdentifierConstIterator = typename OutputMapPointIdentifier::const_iterator;

  using ConstraintMapType = itksys::hash_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 = itksys::hash_map< OutputQEPrimal*, OutputCoordRepType, HashOutputQEPrimal >;
  using CoefficientMapConstIterator = typename CoefficientMapType::const_iterator;

  using AreaMapType = itksys::hash_map< OutputPointIdentifier, OutputCoordRepType >;
  using AreaMapConstIterator = typename AreaMapType::const_iterator;

  using RowType = itksys::hash_map< OutputPointIdentifier, OutputCoordRepType >;
  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};
  AreaType                  m_AreaComputationType;

  void PrintSelf(std::ostream & os, Indent indent) const override;

  OutputCoordRepType ComputeMixedAreaForGivenVertex(OutputPointIdentifier vId);
  OutputCoordRepType ComputeMixedArea(OutputQEPrimal *iQE1, OutputQEPrimal *iQE2);

  virtual void ComputeVertexIdMapping();

  void ComputeLaplacianMatrix( MatrixType &ioL );

  void
  FillMatrixRow(OutputPointIdentifier iId,
                unsigned int iDegree,
                OutputCoordRepType 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, OutputCoordRepType iWeight, unsigned int iDegree):
      m_Id(iV), m_Weight(iWeight), m_Degree(iDegree) {}

    OutputPointIdentifier m_Id;
    OutputCoordRepType    m_Weight;
    unsigned int          m_Degree;
  };
};
} // end namespace itk

#include "itkLaplacianDeformationQuadEdgeMeshFilter.hxx"

#endif