itkFEMLoadLandmark.h

Go to the documentation of this file.

/*=========================================================================

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkFEMLoadLandmark.h,v $
  Language:  C++
  Date:  $Date: 2009-01-30 21:10:18 $
  Version:   $Revision: 1.12 $

  Copyright (c) Insight Software Consortium. All rights reserved.
  See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm for details.

     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notices for more information.

=========================================================================*/

#ifndef __itkFEMLoadLandmark_h
#define __itkFEMLoadLandmark_h

#include "itkFEMLoadElementBase.h"
#include "vnl/vnl_vector.h"

namespace itk {
namespace fem {

class LoadLandmark : public LoadElement {
  FEM_CLASS(LoadLandmark,LoadElement)
public:

  double eta;

  vnl_vector<Float> m_pt;

  vnl_vector<Float> m_target;

  vnl_vector<Float> m_source;

  vnl_vector<Float> m_force;


  //Element::ConstPointer m_element;

  Solution::ConstPointer m_Solution;

  void SetSolution(Solution::ConstPointer ptr) { m_Solution = ptr; }
  Solution::ConstPointer GetSolution() { return m_Solution; }
  Float GetSolution(unsigned int i, unsigned int v=0) { return m_Solution->GetSolutionValue(i,v); }

  const Element::VectorType& GetPoint() const  { return m_pt; }

  void SetPoint( const  vnl_vector<Float>& pt) { m_pt=pt; }

  Element::VectorType& GetSource()
    {
    return m_source;
    }

  Element::VectorType& GetForce()
    {
    return m_force;
    }

  void SetForce( const  vnl_vector<Float>& force)
    {
    if (m_force.size() != force.size())
      {
      m_force.set_size(force.size());
      }
    for (unsigned int i=0; i<force.size(); i++)
      {
      m_force[i]=force[i];
      }
    }


  void SetSource( const  vnl_vector<Float>& source)
    {
    if (m_source.size() != source.size())
      {
      m_source.set_size(source.size());
      }
    for (unsigned int i=0; i<source.size(); i++)
      {
      m_source[i]=source[i];
      }
    }

  const Element::VectorType& GetTarget() const
    {
    return m_target;
    }

  void SetTarget( const  vnl_vector<Float>& target)
    {
    if (m_target.size() != target.size())
      {
      m_target.set_size(target.size());
      }
    for (unsigned int i=0; i<target.size(); i++)
      {
      m_target[i]=target[i];
      }
    }

  void ScalePointAndForce( double* spacing, double fwt)
    {
    for (unsigned int i=0; i<m_target.size(); i++)
      {
      m_target[i] /= spacing[i];
      m_source[i] /= spacing[i];
      this->eta *= fwt;
      }
    }

  virtual void Read( std::istream& f, void* info );

  virtual void AssignToElement( Element::ArrayType::Pointer elements );

  virtual void Write( std::ostream& f ) const;

  LoadLandmark() {}
};

FEM_CLASS_INIT(LoadLandmark)

}} // end namespace itk::fem

#endif // #ifndef __itkFEMLoadLandmark_h