itkFEMPArray.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkFEMPArray.h,v $
  Language:  C++
  Date:      $Date: 2009-01-30 21:53:03 $
  Version:   $Revision: 1.16 $

  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 __itkFEMPArray_h
#define __itkFEMPArray_h

#include "itkFEMP.h"
#include "itkFEMException.h"
#include <vector>

namespace itk {
namespace fem {

template<class T>
class FEMPArray : public std::vector<FEMP<T> >
{
public:

  typedef FEMPArray Self;

  typedef std::vector<FEMP<T> > Superclass;

  typedef Self*       Pointer;
  typedef const Self* ConstPointer;

  typedef T                                ClassType;
  typedef typename ClassType::Pointer      ClassTypePointer;
  typedef typename ClassType::ConstPointer ClassTypeConstPointer;

  ClassTypePointer Find(int gn);
  ClassTypeConstPointer Find(int gn) const;

  ClassTypePointer operator() (int i) 
    {
    return &(*this->operator[](i));
    }

  ClassTypeConstPointer operator() (int i) const 
    {  
    return &(*this->operator[](i));
    }


  int Renumber();

};

template<class T>
typename FEMPArray<T>::ClassTypePointer 
FEMPArray<T>::Find(int gn)
{

  typedef typename Superclass::iterator Iterator;

  Iterator it   = this->begin(); 
  Iterator iend = this->end();
  while( it != iend )
    {
    if( (*it)->GN == gn )
      {
      break;
      }
    it++;
    }  

  if( it == this->end() )
    {
    throw FEMExceptionObjectNotFound(__FILE__,__LINE__,"FEMPArray::Find() const",typeid(T).name(),gn);
    }

  return &(*(*it));

} 


template<class T>
typename FEMPArray<T>::ClassTypeConstPointer 
FEMPArray<T>::Find( int gn ) const
{

  typedef typename Superclass::const_iterator ConstIterator;

  ConstIterator it   = this->begin(); 
  ConstIterator iend = this->end();
  while( it != iend )
    {
    if( (*it)->GN == gn )
      {
      break;
      }
    it++;
    }  

  if( it == this->end() )
    {
    throw FEMExceptionObjectNotFound(__FILE__,__LINE__,"FEMPArray::Find() const",typeid(T).name(),gn);
    }

  return &(*(*it));

}

template<class T>
int FEMPArray<T>::Renumber() 
{

  typename Superclass::iterator i;
  int j=0;

  for(i = this->begin(); i != this->end(); i++)
    {
    (*i)->GN=j;
    j++;
    }

  return j;

}

}} // end namespace itk::fem

#endif // #ifndef __itkFEMPArray_h