itkFEMSolverHyperbolic.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 itkFEMSolverHyperbolic_h
#define itkFEMSolverHyperbolic_h

#include "itkFEMSolver.h"

namespace itk {
namespace fem {

template <unsigned int TDimension = 3>
class ITK_TEMPLATE_EXPORT SolverHyperbolic : public Solver<TDimension>
{
public:
  typedef SolverHyperbolic         Self;
  typedef Solver<TDimension>       Superclass;
  typedef SmartPointer<Self>       Pointer;
  typedef SmartPointer<const Self> ConstPointer;

  itkNewMacro(Self);

  itkTypeMacro(SolverHyperbolic, Solver<TDimension> );

  typedef Element::Float Float;

  itkSetMacro(Gamma, Float);
  itkGetMacro(Gamma, Float);

  itkSetMacro(Beta, Float);
  itkGetMacro(Beta, Float);

  itkSetMacro(NumberOfIterations, unsigned int);
  itkGetMacro(NumberOfIterations, unsigned int);

  virtual Float GetTimeStep(void) const ITK_OVERRIDE
  {
    return this->m_TimeStep;
  }

  virtual void SetTimeStep(Float dt) ITK_OVERRIDE
  {
    this->m_TimeStep = dt;
  }

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

  virtual void InitializeLinearSystemWrapper(void) ITK_OVERRIDE;

  virtual void AssembleElementMatrix(Element::Pointer e) ITK_OVERRIDE;

  virtual void InitializeMatrixForAssembly(unsigned int N) ITK_OVERRIDE;

  virtual void FinalizeMatrixAfterAssembly( void ) ITK_OVERRIDE;


  void  GenerateData() ITK_OVERRIDE;

  virtual void RunSolver(void) ITK_OVERRIDE;

  void Solve();

  enum { matrix_K=1, matrix_M=2, matrix_C=3, matrix_tmp=4 };

  enum { solution_d=0, solution_v=1, solution_a=2};
  enum { vector_dhat=2, vector_vhat=3, vector_ahat=4, vector_tmp=5 };

  Float          m_TimeStep;
  Float          m_Gamma;
  Float          m_Beta;
  unsigned int   m_NumberOfIterations;

private:
  ITK_DISALLOW_COPY_AND_ASSIGN(SolverHyperbolic);
};

} // end namespace fem
} // end namespace itk

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


#endif // #ifndef itkFEMSolverHyperbolic_h