itkFEMLoadBCMFC.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 __itkFEMLoadBCMFC_h
#define __itkFEMLoadBCMFC_h

#include "itkFEMLoadBase.h"

namespace itk
{
namespace fem
{

// forward declaratons...
// class Solver;

class LoadBCMFC : public Load
{
public:
  typedef LoadBCMFC                Self;
  typedef Load                     Superclass;
  typedef SmartPointer<Self>       Pointer;
  typedef SmartPointer<const Self> ConstPointer;

  itkSimpleNewMacro(Self);

  itkTypeMacro(LoadBCMFC, Load);

  virtual::itk::LightObject::Pointer CreateAnother(void) const ITK_OVERRIDE;

  class MFCTerm
  {
  public:

    Element::ConstPointer m_element;

    unsigned int dof;

    Element::Float value;

    MFCTerm(Element::ConstPointer element_, int dof_,
            Element::Float value_) : m_element(element_), dof(dof_), value(value_)
    {
    }
  };

  typedef std::vector<MFCTerm> LhsType;

  LoadBCMFC() : m_Index(0), m_LeftHandSide(), m_RightHandSide()  {}

  LoadBCMFC(Element::ConstPointer element, int dof, vnl_vector<Element::Float> val);

  void SetIndex(int ind);

  int GetIndex();

  void AddLeftHandSideTerm(LoadBCMFC::MFCTerm term);

  void AddRightHandSideTerm(Element::Float term);

  int GetNumberOfLeftHandSideTerms() const;

  int GetNumberOfRightHandSideTerms() const;

  const MFCTerm GetLeftHandSideTerm(int lhs) const;

  Element::Float GetRightHandSideTerm(int rhs) const;

  const std::vector<MFCTerm> & GetLeftHandSideArray() const;
  std::vector<MFCTerm> & GetLeftHandSideArray();

  vnl_vector<Element::Float> & GetRightHandSideArray();

//  friend class Solver;

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

  // private:  // FIXME: CrankNicolsonSolver class, which is derived from Solver
  // class also needs access to Index.
  int m_Index;

  LhsType m_LeftHandSide;

  vnl_vector<Element::Float> m_RightHandSide;
};

}
}  // end namespace itk::fem

#endif // #ifndef __itkFEMLoadBCMFC_h