Shapeworks Studio  2.1
Shape analysis software suite
List of all members | Public Types | Public Member Functions | Protected Member Functions
itk::PSMProcrustesRegistration< VDimension > Class Template Reference

This class interfaces with the PSMProcrustesFunction class to register a list of point sets. It transforms the registration transforms generated by PSMProcrustesFunction from Configuration space to Procrustes space. The resulting transforms are then applied to the point sets. More...

#include <itkPSMProcrustesRegistration.h>

+ Inheritance diagram for itk::PSMProcrustesRegistration< VDimension >:
+ Collaboration diagram for itk::PSMProcrustesRegistration< VDimension >:

Public Types

typedef PSMProcrustesRegistration Self
 
typedef DataObject Superclass
 
typedef SmartPointer< Self > Pointer
 
typedef SmartPointer< const Self > ConstPointer
 
typedef WeakPointer< const Self > ConstWeakPointer
 
typedef PSMParticleSystem< VDimension > PSMParticleSystemType
 
typedef PSMParticleSystemType::TransformType TransformType
 
typedef PSMProcrustesFunction< VDimension > PSMProcrustesFunctionType
 
typedef PSMProcrustesFunctionType::ShapeListType ShapeListType
 
typedef PSMProcrustesFunctionType::ShapeType ShapeType
 
typedef PSMProcrustesFunctionType::PointType PointType
 
typedef PSMProcrustesFunctionType::SimilarityTransformListType SimilarityTransformListType
 

Public Member Functions

 itkNewMacro (Self)
 
 itkTypeMacro (PSMProcrustesRegistration, DataObject)
 
void SetPSMParticleSystem (PSMParticleSystemType *p)
 
const PSMParticleSystemType * GetPSMParticleSystem () const
 
PSMParticleSystemType * GetPSMParticleSystem ()
 
void IterationCallback (itk::Object *, const itk::EventObject &)
 
void RunRegistration (int i)
 
void RunRegistration ()
 
void SetDomainsPerShape (int i)
 
int GetDomainsPerShape () const
 
bool GetScaling () const
 
void ScalingOn ()
 
void ScalingOff ()
 
void SetProcrustesInterval (int i)
 
int GetProcrustesInterval ()
 

Protected Member Functions

void PrintSelf (std::ostream &os, Indent indent) const
 

Detailed Description

template<unsigned int VDimension>
class itk::PSMProcrustesRegistration< VDimension >

This class interfaces with the PSMProcrustesFunction class to register a list of point sets. It transforms the registration transforms generated by PSMProcrustesFunction from Configuration space to Procrustes space. The resulting transforms are then applied to the point sets.

Definition at line 36 of file itkPSMProcrustesRegistration.h.

Member Typedef Documentation

template<unsigned int VDimension>
typedef PSMParticleSystem<VDimension> itk::PSMProcrustesRegistration< VDimension >::PSMParticleSystemType

Particle system typedefs.

Definition at line 48 of file itkPSMProcrustesRegistration.h.

template<unsigned int VDimension>
typedef PSMProcrustesFunction<VDimension> itk::PSMProcrustesRegistration< VDimension >::PSMProcrustesFunctionType

Procrustes Function typedef

Definition at line 52 of file itkPSMProcrustesRegistration.h.

template<unsigned int VDimension>
typedef PSMProcrustesRegistration itk::PSMProcrustesRegistration< VDimension >::Self

Standard class typedefs

Definition at line 41 of file itkPSMProcrustesRegistration.h.

Member Function Documentation

template<unsigned int VDimension>
bool itk::PSMProcrustesRegistration< VDimension >::GetScaling ( ) const
inline

Turn on/off the scaling. Shapes will only be rotated and translated if scaling is turned off.

Definition at line 99 of file itkPSMProcrustesRegistration.h.

100  { return m_Scaling; }
template<unsigned int VDimension>
void itk::PSMProcrustesRegistration< VDimension >::IterationCallback ( itk::Object *  ,
const itk::EventObject &   
)
inline

Callback suitable for adding as an observer of itk object iteration events.

Definition at line 73 of file itkPSMProcrustesRegistration.h.

74  {
75  this->RunRegistration();
76  }
template<unsigned int VDimension>
itk::PSMProcrustesRegistration< VDimension >::itkNewMacro ( Self  )

Method for creation through the object factory.

template<unsigned int VDimension>
itk::PSMProcrustesRegistration< VDimension >::itkTypeMacro ( PSMProcrustesRegistration< VDimension >  ,
DataObject   
)

Run-time type information (and related methods).

template<unsigned int VDimension>
template void itk::PSMProcrustesRegistration< VDimension >::RunRegistration ( int  i)

Performs a procrustes registration and modifies the transforms of the particle system accordingly. Assumes m_ParticleSystem has been set to point to a valid object.

Definition at line 22 of file itkPSMProcrustesRegistration.cxx.

23 {
24  // Assume all domains have the same number of particles.
25  const int totalDomains = m_PSMParticleSystem->GetNumberOfDomains();
26  const int numPoints = m_PSMParticleSystem->GetNumberOfParticles(0);
27  const int numShapes = totalDomains / m_DomainsPerShape;
28  //PSMProcrustesFunctionType typedefs
29  ShapeListType shapelist;
30  ShapeType shapevector;
31  PointType point;
32 
33  // Read input shapes from file list
34  for (int i = d % m_DomainsPerShape; i < totalDomains; i+=m_DomainsPerShape)
35  {
36  int j = 0;
37  shapevector.clear();
38  for (j = 0; j < numPoints; j++)
39  {
40  for (unsigned int k = 0; k < VDimension; k++)
41  {
42  point(k) = m_PSMParticleSystem->GetPosition(j,i)[k];
43  }
44  shapevector.push_back(point);
45  }
46  shapelist.push_back(shapevector);
47  }
48 
49  // Run alignment
50  SimilarityTransformListType transforms;
51 
52  typename PSMProcrustesFunctionType::Pointer procrustes = PSMProcrustesFunctionType::New();
53  procrustes->RunGeneralizedProcrustes(transforms, shapelist);
54  // Construct transform matrices for each particle system.
55  int k = d % m_DomainsPerShape;
56  for (int i = 0; i < numShapes; i++, k += m_DomainsPerShape)
57  {
58  // Transform from Configuration space to Procrustes space. Translation
59  // followed by rotation and scaling.
60  TransformType R; //PSMParticleSystemType typedef
61  if( VDimension == 3 )
62  {
63  if (m_Scaling == true)
64  {
65  R(0,0) = transforms[i].rotation(0,0) * transforms[i].scale;
66  R(1,0) = transforms[i].rotation(1,0) * transforms[i].scale;
67  R(2,0) = transforms[i].rotation(2,0) * transforms[i].scale;
68  R(3,0) = 0.0;
69 
70  R(0,1) = transforms[i].rotation(0,1) * transforms[i].scale;
71  R(1,1) = transforms[i].rotation(1,1) * transforms[i].scale;
72  R(2,1) = transforms[i].rotation(2,1) * transforms[i].scale;
73  R(3,1) = 0.0;
74 
75  R(0,2) = transforms[i].rotation(0,2) * transforms[i].scale;
76  R(1,2) = transforms[i].rotation(1,2) * transforms[i].scale;
77  R(2,2) = transforms[i].rotation(2,2) * transforms[i].scale;
78  R(3,2) = 0.0;
79 
80  R(0,3) = transforms[i].translation(0) * R(0,0) + transforms[i].translation(1) * R(0,1) + transforms[i].translation(2) * R(0,2);
81  R(1,3) = transforms[i].translation(0) * R(1,0) + transforms[i].translation(1) * R(1,1) + transforms[i].translation(2) * R(1,2);
82  R(2,3) = transforms[i].translation(0) * R(2,0) + transforms[i].translation(1) * R(2,1) + transforms[i].translation(2) * R(2,2);
83  R(3,3) = 1.0;
84  }
85  else // do not do scaling
86  {
87  R(0,0) = transforms[i].rotation(0,0);
88  R(1,0) = transforms[i].rotation(1,0);
89  R(2,0) = transforms[i].rotation(2,0);
90  R(3,0) = 0.0;
91 
92  R(0,1) = transforms[i].rotation(0,1);
93  R(1,1) = transforms[i].rotation(1,1);
94  R(2,1) = transforms[i].rotation(2,1);
95  R(3,1) = 0.0;
96 
97  R(0,2) = transforms[i].rotation(0,2);
98  R(1,2) = transforms[i].rotation(1,2);
99  R(2,2) = transforms[i].rotation(2,2);
100  R(3,2) = 0.0;
101 
102  R(0,3) = transforms[i].translation(0) * R(0,0) + transforms[i].translation(1) * R(0,1) + transforms[i].translation(2) * R(0,2);
103  R(1,3) = transforms[i].translation(0) * R(1,0) + transforms[i].translation(1) * R(1,1) + transforms[i].translation(2) * R(1,2);
104  R(2,3) = transforms[i].translation(0) * R(2,0) + transforms[i].translation(1) * R(2,1) + transforms[i].translation(2) * R(2,2);
105  R(3,3) = 1.0;
106  }
107 
108  m_PSMParticleSystem->SetTransform(k, R);
109  std::cout << "R" << std::endl;
110  std::cout << R << std::endl;
111  std::cout << std::endl;
112  }
113  else if( VDimension == 2 )
114  {
115  if (m_Scaling == true)
116  {
117  R(0,0) = transforms[i].rotation(0,0) * transforms[i].scale;
118  R(1,0) = transforms[i].rotation(1,0) * transforms[i].scale;
119  R(2,0) = 0.0;
120 
121  R(0,1) = transforms[i].rotation(0,1) * transforms[i].scale;
122  R(1,1) = transforms[i].rotation(1,1) * transforms[i].scale;
123  R(2,1) = 0.0;
124 
125  R(0,2) = transforms[i].translation(0) * R(0,0) + transforms[i].translation(1) * R(0,1);
126  R(1,2) = transforms[i].translation(0) * R(1,0) + transforms[i].translation(1) * R(1,1);
127  R(2,2) = 1.0;
128  }
129  else // do not do scaling
130  {
131  R(0,0) = transforms[i].rotation(0,0);
132  R(1,0) = transforms[i].rotation(1,0);
133  R(2,0) = 0.0;
134 
135  R(0,1) = transforms[i].rotation(0,1);
136  R(1,1) = transforms[i].rotation(1,1);
137  R(2,1) = 0.0;
138 
139  R(0,2) = transforms[i].translation(0) * R(0,0) + transforms[i].translation(1) * R(0,1);
140  R(1,2) = transforms[i].translation(0) * R(1,0) + transforms[i].translation(1) * R(1,1);
141  R(2,2) = 1.0;
142  }
143 
144  m_PSMParticleSystem->SetTransform(k, R);
145  std::cout << "R" << std::endl;
146  std::cout << R << std::endl;
147  std::cout << std::endl;
148  }
149  } // end for loop
150 } // end RunRegistration
itk::PSMParticleSystem::GetNumberOfDomains
unsigned int GetNumberOfDomains() const
Definition: itkPSMParticleSystem.h:270
itk::PSMParticleSystem::GetPosition
PointType & GetPosition(unsigned long int k, unsigned int d=0)
Definition: itkPSMParticleSystem.h:169
itk::PSMParticleSystem::SetTransform
void SetTransform(unsigned int i, const TransformType &, int threadId=0)
Definition: itkPSMParticleSystem.hxx:102
itk::PSMParticleSystem::GetNumberOfParticles
unsigned long int GetNumberOfParticles(unsigned int d=0) const
Definition: itkPSMParticleSystem.h:136
template<unsigned int VDimension>
void itk::PSMProcrustesRegistration< VDimension >::SetDomainsPerShape ( int  i)
inline

Set/Get the number of Domains in each shape. Procrustes will be performed separately for each domain.

Definition at line 92 of file itkPSMProcrustesRegistration.h.

93  { m_DomainsPerShape = i; }
template<unsigned int VDimension>
void itk::PSMProcrustesRegistration< VDimension >::SetProcrustesInterval ( int  i)
inline

Set/Get the number (interval) of iterations at which GPA is run.

Definition at line 107 of file itkPSMProcrustesRegistration.h.

108  { m_Procrustes_Interval = i; }
template<unsigned int VDimension>
void itk::PSMProcrustesRegistration< VDimension >::SetPSMParticleSystem ( PSMParticleSystemType *  p)
inline

Set/Get the target particle system.

Definition at line 64 of file itkPSMProcrustesRegistration.h.

65  { m_PSMParticleSystem = p; }
The documentation for this class was generated from the following files:
Generated on Mon Aug 15 2016 13:48:36 for Shapeworks Studio by   doxygen 1.8.11