Skip to content

Libs/Optimize/ParticleSystem/itkParticleShapeMatrixAttribute.h

Namespaces

Name
itk

Classes

Name
class itk::ParticleShapeMatrixAttribute
Each column describes a shape. A shape may be composed of m_DomainsPerShape domains (default 1). ALL DOMAINS ARE NOT ASSUMED TO HAVE THE SAME NUMBER OF PARTICLES!

Source code

#pragma once

#include "itkDataObject.h"
#include "itkWeakPointer.h"
#include "itkParticleAttribute.h"
#include "itkParticleContainer.h"
#include "vnl/vnl_matrix.h"

namespace itk
{
template <class T, unsigned int VDimension>
class ITK_EXPORT ParticleShapeMatrixAttribute
        : public vnl_matrix<T>, public ParticleAttribute<VDimension>
{
public:
    typedef T DataType;
    typedef ParticleShapeMatrixAttribute Self;
    typedef ParticleAttribute<VDimension> Superclass;
    typedef SmartPointer<Self>  Pointer;
    typedef SmartPointer<const Self>  ConstPointer;
    typedef WeakPointer<const Self>  ConstWeakPointer;

    itkNewMacro(Self)


    itkTypeMacro(ParticleShapeMatrixAttribute, ParticleAttribute)


    virtual void DomainAddEventCallback(Object *, const EventObject &e)
    {
        const itk::ParticleDomainAddEvent &event = dynamic_cast<const itk::ParticleDomainAddEvent &>(e);
        unsigned int d = event.GetDomainIndex();

        if ( d % m_DomainsPerShape  == 0 )
            this->ResizeMatrix(this->rows(), this->cols()+1);
    }

    virtual void ResizeMatrix(int rs, int cs)
    {
        vnl_matrix<T> tmp(*this); // copy existing  matrix

        // Create new column (shape)
        this->set_size(rs, cs);

        // Copy old data into new matrix.
        for (unsigned int c = 0; c < tmp.cols(); c++)
        {
            for (unsigned int r = 0; r < tmp.rows(); r++)
                this->operator()(r,c) = tmp(r,c);
        }
    }

    virtual void PositionAddEventCallback(Object *o, const EventObject &e)
    {
        const itk::ParticlePositionAddEvent &event = dynamic_cast<const itk::ParticlePositionAddEvent &>(e);
        const itk::ParticleSystem *ps= dynamic_cast<const itk::ParticleSystem *>(o);
        const int d = event.GetDomainIndex();
        const unsigned int idx = event.GetPositionIndex();
        const typename itk::ParticleSystem::PointType pos = ps->GetTransformedPosition(idx, d);

        int numRows = 0;
        for (int i = 0; i < m_DomainsPerShape; i++)
            numRows += VDimension * ps->GetNumberOfParticles(i);

        if (numRows > this->rows())
            this->ResizeMatrix(numRows, this->cols());

        unsigned int k = 0;
        int dom = d % m_DomainsPerShape;
        for (int i = 0; i < dom; i++)
            k += VDimension * ps->GetNumberOfParticles(i);
        k += idx * VDimension;

        for (unsigned int i = 0; i < VDimension; i++)
            this->operator()(i+k, d / m_DomainsPerShape) = pos[i];
    }

    virtual void PositionSetEventCallback(Object *o, const EventObject &e)
    {
        const itk::ParticlePositionSetEvent &event = dynamic_cast<const itk::ParticlePositionSetEvent &>(e);
        const itk::ParticleSystem *ps= dynamic_cast<const itk::ParticleSystem *>(o);
        const int d = event.GetDomainIndex();
        const unsigned int idx = event.GetPositionIndex();
        const typename itk::ParticleSystem::PointType pos = ps->GetTransformedPosition(idx, d);

        unsigned int k = 0;
        int dom = d % m_DomainsPerShape;
        for (int i = 0; i < dom; i++)
            k += VDimension * ps->GetNumberOfParticles(i);
        k += idx * VDimension;

        for (unsigned int i = 0; i < VDimension; i++)
            this->operator()(i+k, d / m_DomainsPerShape) = pos[i];
    }

    virtual void PositionRemoveEventCallback(Object *, const EventObject &)
    {
        // NEED TO IMPLEMENT THIS
    }

    // For debugging purposes.
      void PrintMatrix()
      {
        std::cout << std::endl;
        for (unsigned int r = 0; r < this->rows(); r++)
          {
          for (unsigned int c = 0; c < this->cols(); c++)
            {
            std::cout << this->operator()(r, c) << "  ";
            }
        std::cout << std::endl;
          }
      }

      bool CheckForNans()
      {
        bool flag = false;
        for (unsigned int r = 0; r < this->rows(); r++)
          {
          for (unsigned int c = 0; c < this->cols(); c++)
            {
            if (isnan(this->operator()(r, c)))
            {
                flag = true;
                break;
            }
            }
          }
        return flag;
      }

    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:
    ParticleShapeMatrixAttribute() : m_DomainsPerShape(1)
    {
        this->m_DefinedCallbacks.DomainAddEvent = true;
        this->m_DefinedCallbacks.PositionAddEvent = true;
        this->m_DefinedCallbacks.PositionSetEvent = true;
        this->m_DefinedCallbacks.PositionRemoveEvent = true;
    }
    virtual ~ParticleShapeMatrixAttribute() {}

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

    int m_DomainsPerShape;
private:
    ParticleShapeMatrixAttribute(const Self&); //purposely not implemented
    void operator=(const Self&); //purposely not implemented


};

} // end namespace

Updated on 2022-07-23 at 17:50:04 -0600