itkFastMarchingQuadEdgeMeshFilterBase.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 __itkFastMarchingQuadEdgeMeshFilterBase_h
#define __itkFastMarchingQuadEdgeMeshFilterBase_h

#include "itkFastMarchingBase.h"
#include "itkFastMarchingTraits.h"

namespace itk
{
template< typename TInput, typename TOutput >
class FastMarchingQuadEdgeMeshFilterBase :
    public FastMarchingBase< TInput, TOutput >
{
public:
  typedef FastMarchingQuadEdgeMeshFilterBase     Self;
  typedef FastMarchingBase< TInput, TOutput >    Superclass;
  typedef SmartPointer< Self >                   Pointer;
  typedef SmartPointer< const Self >             ConstPointer;
  typedef typename Superclass::Traits            Traits;

  itkNewMacro(Self);

  itkTypeMacro(FastMarchingQuadEdgeMeshFilterBase, FastMarchingBase);

  typedef typename Superclass::InputDomainType     InputMeshType;
  typedef typename Superclass::InputDomainPointer  InputMeshPointer;
  typedef typename Superclass::InputPixelType      InputPixelType;
  typedef typename InputMeshType::PointType        InputPointType;
  typedef typename InputMeshType::PointIdentifier  InputPointIdentifierType;

  typedef typename Superclass::OutputDomainType     OutputMeshType;
  typedef typename Superclass::OutputDomainPointer  OutputMeshPointer;
  typedef typename Superclass::OutputPixelType      OutputPixelType;
  typedef typename OutputMeshType::PointType        OutputPointType;
  typedef typename OutputPointType::VectorType      OutputVectorType;
  typedef typename OutputVectorType::RealValueType  OutputVectorRealType;
  typedef typename OutputMeshType::QEType           OutputQEType;
  typedef typename OutputMeshType::PointIdentifier  OutputPointIdentifierType;
  typedef typename OutputMeshType::PointsContainer  OutputPointsContainer;
  typedef typename OutputPointsContainer::Pointer   OutputPointsContainerPointer;
  typedef typename OutputPointsContainer::Iterator  OutputPointsContainerIterator;
  typedef typename OutputMeshType::PointDataContainer
                                                    OutputPointDataContainer;
  typedef typename OutputPointDataContainer::Pointer
                                                    OutputPointDataContainerPointer;

  typedef typename OutputMeshType::CellsContainer   OutputCellsContainer;
  typedef typename OutputCellsContainer::Pointer    OutputCellsContainerPointer;
  typedef typename OutputCellsContainer::ConstIterator
                                                    OutputCellsContainerConstIterator;
  typedef typename OutputMeshType::CellType         OutputCellType;


  typedef typename Traits::NodeType                 NodeType;
  typedef typename Traits::NodePairType             NodePairType;
  typedef typename Traits::NodePairContainerType    NodePairContainerType;
  typedef typename Traits::NodePairContainerPointer NodePairContainerPointer;
  typedef typename Traits::NodePairContainerConstIterator
    NodePairContainerConstIterator;

//  typedef typename Traits::NodeContainerType        NodeContainerType;
//  typedef typename Traits::NodeContainerPointer     NodeContainerPointer;
//  typedef typename Traits::NodeContainerConstIterator
//    NodeContainerConstIterator;

  typedef typename Superclass::LabelType LabelType;

  typedef std::map< NodeType, LabelType >           NodeLabelMapType;
  typedef typename NodeLabelMapType::iterator       NodeLabelMapIterator;
  typedef typename NodeLabelMapType::const_iterator NodeLabelMapConstIterator;

protected:

  FastMarchingQuadEdgeMeshFilterBase();
  virtual ~FastMarchingQuadEdgeMeshFilterBase();

  NodeLabelMapType m_Label;

  IdentifierType GetTotalNumberOfNodes() const;

  void SetOutputValue( OutputMeshType* oMesh,
                      const NodeType& iNode,
                      const OutputPixelType& iValue );

  const OutputPixelType GetOutputValue( OutputMeshType* oMesh,
                                  const NodeType& iNode ) const;

  unsigned char GetLabelValueForGivenNode( const NodeType& iNode ) const;

  void SetLabelValueForGivenNode( const NodeType& iNode,
                                  const LabelType& iLabel );

  void UpdateNeighbors( OutputMeshType* oMesh,
                        const NodeType& iNode );

  void UpdateValue( OutputMeshType* oMesh,
                    const NodeType& iNode );

  const OutputVectorRealType
  Solve( OutputMeshType* oMesh,
         const NodeType& iId, const OutputPointType& iCurrentPoint,
         const OutputVectorRealType& iF,
         const NodeType& iId1, const OutputPointType& iP1,
         const bool& iIsFar1, const OutputVectorRealType iVal1,
         const NodeType& iId2, const OutputPointType& iP2,
         const bool& iIsFar2, const OutputVectorRealType& iVal2 )
  const;


  const OutputVectorRealType
  ComputeUpdate(
    const OutputVectorRealType& iVal1, const OutputVectorRealType& iVal2,
    const OutputVectorRealType& iNorm1, const OutputVectorRealType& iSqNorm1,
    const OutputVectorRealType& iNorm2, const OutputVectorRealType& iSqNorm2,
    const OutputVectorRealType& iDot, const OutputVectorRealType& iF )
    const;

  bool UnfoldTriangle(
    OutputMeshType* oMesh,
    const OutputPointIdentifierType& iId, const OutputPointType& iP,
    const OutputPointIdentifierType& iId1, const OutputPointType& iP1,
    const OutputPointIdentifierType& iId2, const OutputPointType &iP2,
    OutputVectorRealType& oNorm, OutputVectorRealType& oSqNorm,
    OutputVectorRealType& oDot1, OutputVectorRealType& oDot2,
    OutputPointIdentifierType& oId ) const;

  bool CheckTopology( OutputMeshType* oMesh,
                      const NodeType& iNode );

  void InitializeOutput( OutputMeshType* oMesh );

private:
  FastMarchingQuadEdgeMeshFilterBase( const Self& );
  void operator = ( const Self& );
  const InputMeshType *m_InputMesh;
};
}

#include "itkFastMarchingQuadEdgeMeshFilterBase.hxx"
#endif // __itkFastMarchingQuadEdgeMeshFilterBase_h