itkPSMShapeMatrixAttribute.h

/*=========================================================================
 *
 *  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 __itkPSMShapeMatrixAttribute_h
#define __itkPSMShapeMatrixAttribute_h

#include "itkDataObject.h"
#include "itkWeakPointer.h"
#include "itkPSMAttribute.h"
#include "itkPSMContainer.h"
#include "itkPSMParticleSystem.h"
#include "vnl/vnl_matrix.h"

namespace itk
{
template <class T, unsigned int VDimension>
class ITK_EXPORT PSMShapeMatrixAttribute
  : public vnl_matrix<T>, public PSMAttribute<VDimension>
{
public:
  typedef T DataType;
  typedef PSMShapeMatrixAttribute Self;
  typedef PSMAttribute<VDimension> Superclass;
  typedef SmartPointer<Self>  Pointer;
  typedef SmartPointer<const Self>  ConstPointer;
  typedef WeakPointer<const Self>  ConstWeakPointer;
  
  itkNewMacro(Self);
  
  itkTypeMacro(PSMShapeMatrixAttribute, PSMAttribute);
  
  virtual void Initialize() {};

  virtual void DomainAddEventCallback(Object *, const EventObject &e);

  virtual void ResizeMatrix(int , int );

  virtual void PositionAddEventCallback(Object *o, const EventObject &e);
  
  virtual void PositionSetEventCallback(Object *o, const EventObject &e);
  
  virtual void PositionRemoveEventCallback(Object *, const EventObject &) 
  {
    itkExceptionMacro("This Attribute class does not support particle position removal.");
  }
  
  void SetDomainsPerShape(int i)
  { m_DomainsPerShape = i; }
  int GetDomainsPerShape() const
  { return m_DomainsPerShape; }

  virtual void BeforeIteration() {}

  virtual void AfterIteration() {}

  virtual void SetMatrix(const vnl_matrix<T> &m)
  {
    vnl_matrix<T>::operator=(m);
  }
  
protected:
  PSMShapeMatrixAttribute() : m_DomainsPerShape(1)
  {
    this->m_DefinedCallbacks.DomainAddEvent = true;
    this->m_DefinedCallbacks.PositionAddEvent = true;
    this->m_DefinedCallbacks.PositionSetEvent = true;
    this->m_DefinedCallbacks.PositionRemoveEvent = true;
  }
  virtual ~PSMShapeMatrixAttribute() {};

  void PrintSelf(std::ostream& os, Indent indent) const
  {  Superclass::PrintSelf(os,indent);  }

  int m_DomainsPerShape;

private:
  PSMShapeMatrixAttribute(const Self&); //purposely not implemented
  void operator=(const Self&); //purposely not implemented

};

} // end namespace

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

#endif