itkImageToRectilinearFEMObjectFilter.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 itkImageToRectilinearFEMObjectFilter_h
#define itkImageToRectilinearFEMObjectFilter_h

#include "vnl/vnl_vector.h"
#include "itkFEMObject.h"
#include "itkProcessObject.h"

namespace itk
{
namespace fem
{
template <typename TInputImage>
class ITK_TEMPLATE_EXPORT ImageToRectilinearFEMObjectFilter : public ProcessObject
{
public:
  typedef ImageToRectilinearFEMObjectFilter Self;
  typedef ProcessObject                     Superclass;
  typedef SmartPointer<Self>                Pointer;
  typedef SmartPointer<const Self>          ConstPointer;

  itkNewMacro(Self);

  itkTypeMacro(ImageToRectilinearFEMObjectFilter, ProcessObject);

  itkStaticConstMacro(NDimensions, unsigned int, TInputImage::ImageDimension);

  typedef TInputImage                           InputImageType;
  typedef typename InputImageType::Pointer      ImagePointer;
  typedef typename InputImageType::ConstPointer ImageConstPointer;
  typedef typename InputImageType::RegionType   ImageRegionType;
  typedef typename InputImageType::SizeType     ImageSizeType;
  typedef typename InputImageType::PointType    ImagePointType;
  typedef typename InputImageType::IndexType    ImageIndexType;

  typedef typename itk::fem::FEMObject<NDimensions> FEMObjectType;
  typedef typename FEMObjectType::Pointer           FEMObjectPointer;
  typedef typename FEMObjectType::ConstPointer      FEMObjectConstPointer;
  typedef typename DataObject::Pointer              DataObjectPointer;

  typedef itk::fem::MaterialLinearElasticity MaterialType;
  typedef MaterialType::Pointer              MaterialPointerType;
  // typedef itk::fem::Element2DC0LinearQuadrilateral  QuadElementBaseType;
  // typedef itk::fem::Element3DC0LinearHexahedron     HexElementBaseType;
  typedef itk::fem::Element               ElementBaseType;
  typedef itk::fem::Element::ConstPointer ElementBasePointerType;

#ifdef ITK_USE_CONCEPT_CHECKING
  // Begin concept checking
//  itkConceptMacro(SameDimensionOrMinusOne,
//    (Concept::SameDimensionOrMinusOne<NDimensions, 3>));
// End concept checking
#endif

  itkGetMacro(PixelsPerElement, vnl_vector<unsigned int> );
  itkSetMacro(PixelsPerElement, vnl_vector<unsigned int> );
  void SetPixelsPerElement( unsigned int numPixels )
  {
    this->m_PixelsPerElement.fill( numPixels );
  }

  itkGetMacro(NumberOfElements, vnl_vector<unsigned int> );

  itkGetMacro(Material, MaterialPointerType);
  itkSetMacro(Material, MaterialPointerType);

  itkGetMacro(Element, ElementBasePointerType);
  itkSetMacro(Element, ElementBasePointerType);

  using Superclass::SetInput;
  void SetInput( InputImageType *image);

  void SetInput( unsigned int, InputImageType *image);

  InputImageType * GetInput();

  InputImageType * GetInput(unsigned int idx);

  typedef ProcessObject::DataObjectPointerArraySizeType DataObjectPointerArraySizeType;
  using Superclass::MakeOutput;
  virtual DataObjectPointer MakeOutput(DataObjectPointerArraySizeType idx) ITK_OVERRIDE;

  FEMObjectType * GetOutput();

  FEMObjectType * GetOutput(unsigned int idx);

protected:
  ImageToRectilinearFEMObjectFilter();
  virtual ~ImageToRectilinearFEMObjectFilter() ITK_OVERRIDE {}
  virtual void PrintSelf(std::ostream& os, Indent indent) const ITK_OVERRIDE;

  virtual void  GenerateData() ITK_OVERRIDE;

  void Generate2DRectilinearMesh();

  void Generate3DRectilinearMesh();

private:
  ITK_DISALLOW_COPY_AND_ASSIGN(ImageToRectilinearFEMObjectFilter);

  vnl_vector<unsigned int> m_NumberOfElements;
  vnl_vector<unsigned int> m_PixelsPerElement;
  MaterialPointerType      m_Material;
  ElementBasePointerType   m_Element;
};

}
}  // end namespace itk::fem

#ifndef ITK_MANUAL_INSTANTIATION
#include "itkImageToRectilinearFEMObjectFilter.hxx"
#endif

#endif // #ifndef itkImageToRectilinearFEMObjectFilter_h