forked from InsightSoftwareConsortium/ITK
-
Notifications
You must be signed in to change notification settings - Fork 0
/
itkMedialNodePairCorrespondenceProcess.h
181 lines (133 loc) · 6.36 KB
/
itkMedialNodePairCorrespondenceProcess.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
/*=========================================================================
Program: Insight Segmentation & Registration Toolkit
Module: itkMedialNodePairCorrespondenceProcess.h
Language: C++
Date: $Date$
Version: $Revision$
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 __itkMedialNodePairCorrespondenceProcess_h
#define __itkMedialNodePairCorrespondenceProcess_h
#include "itkImage.h"
#include "itkProcessObject.h"
#include "itkDataObject.h"
#include "itkBloxCoreAtomImage.h"
#include "itkBloxCoreAtomPixel.h"
#include "itkMatrixResizeableDataObject.h"
#include "itkBinaryMedialNodeMetric.h"
#include "itkCorrespondenceDataStructure.h"
#include "itkSecondaryNodeList.h"
#include "itkCorrespondingList.h"
#include "itkCorrespondingMedialNodeClique.h"
#include "itkCorrespondenceDataStructureIterator.h"
namespace itk
{
/** \class MedialNodePairCorrespondenceProcess
* \brief This process takes as inputs two core atom images, the distance matrices
* of the two images, and the unary correspondence matrix between the two images
* in order to produce an itkCorrespondenceDataStructure containing
* correspondences between pairs (node cliques of size 2) in the images.
*
* \ingroup
*/
template< typename TSourceImage >
class MedialNodePairCorrespondenceProcess : public ProcessObject
{
public:
/** Number of dimensions */
itkStaticConstMacro(NDimensions, unsigned int, TSourceImage::ImageDimension);
/** Standard class typedefs */
typedef MedialNodePairCorrespondenceProcess Self;
typedef ProcessObject Superclass;
typedef SmartPointer<Self> Pointer;
typedef SmartPointer<const Self> ConstPointer;
/** Smart Pointer type to a DataObject. */
typedef DataObject::Pointer DataObjectPointer;
/** Method for creation through the object factory. */
itkNewMacro(Self);
/** Run-time type information. (and related methods) */
itkTypeMacro(MedialNodePairCorrespondenceProcess, ProcessObject);
/** Typedef for core atom image. */
typedef TSourceImage CoreAtomImageType;
typedef typename CoreAtomImageType::Pointer CoreAtomImagePointer;
typedef typename CoreAtomImageType::RegionType CoreAtomImageRegionType;
typedef typename CoreAtomImageType::PixelType CoreAtomImagePixelType;
typedef typename CoreAtomImageType::ConstPointer CoreAtomImageConstPointer;
/** Typedef for distance matrix. */
typedef MatrixResizeableDataObject<double> DistanceMatrixType;
typedef typename DistanceMatrixType::Pointer DistanceMatrixPointer;
/** Typedef for correspondence matrix */
typedef MatrixResizeableDataObject<double> CorrespondenceMatrixType;
typedef typename CorrespondenceMatrixType::Pointer CorrespondenceMatrixPointer;
/** Typedef for correspondence data structure. (output) */
typedef CorrespondingMedialNodeClique<itkGetStaticConstMacro(NDimensions),2> NodeType;
typedef CorrespondenceDataStructure<NodeType,2> DataStructureType;
typedef typename CorrespondenceDataStructure<NodeType,2>::Pointer DataStructurePointerType;
typedef CorrespondenceDataStructureIterator<DataStructureType> IteratorType;
/** Typedef for binary node metric. */
typedef BinaryMedialNodeMetric<itkGetStaticConstMacro(NDimensions)> BinaryMetricType;
typedef typename BinaryMedialNodeMetric<itkGetStaticConstMacro(NDimensions)>::Pointer BinaryMetricPointer;
/** Typedef for medial nodes. */
typedef BloxCoreAtomPixel<itkGetStaticConstMacro(NDimensions)> MedialNodeType;
/** The type used to store the position of the BloxPixel. */
typedef Point<double, itkGetStaticConstMacro(NDimensions)> PositionType;
/** Get the image output of this process object. */
DataStructureType * GetOutput(void);
DataStructureType * GetOutput(unsigned int idx);
/* This is important to note...the inputs for this process are somewhat
complicated, and there are a lot of them. You need all 5 inputs for
this process to perform its task */
/** Set the first core atom image. */
void SetCoreAtomImageA( const CoreAtomImageType * CoreAtomImageA );
/** Set the second core atom image. */
void SetCoreAtomImageB( const CoreAtomImageType * CoreAtomImageB );
/** Set the first distance matrix. */
void SetDistanceMatrixA( const DistanceMatrixType * DistanceMatrixA );
/** Set the second distance matrix. */
void SetDistanceMatrixB( const DistanceMatrixType * DistanceMatrixB );
/** Get number of node pairs. */
int GetNumberOfNodePairs() {return m_NumberOfNodePairs;}
/** Get number of node basepairs. */
int GetNumberOfNodeBasePairs() {return m_NumberOfNodeBasePairs;}
/** Set the correspondence matrix. */
void SetCorrespondenceMatrix( const CorrespondenceMatrixType * CorrespondenceMatrix );
virtual void Update() {this->GenerateData();}
virtual DataObjectPointer MakeOutput(unsigned int idx);
protected:
MedialNodePairCorrespondenceProcess();
virtual ~MedialNodePairCorrespondenceProcess(){}
void PrintSelf(std::ostream& os, Indent indent) const;
/** Method for forming the Pair Correspondence Data Structure. */
void GenerateData();
/** Methods to get core atom images. */
TSourceImage * GetCoreAtomImageA();
TSourceImage * GetCoreAtomImageB();
/** Methods to get distance matrices. */
DistanceMatrixType * GetDistanceMatrixA();
DistanceMatrixType * GetDistanceMatrixB();
/** Method to get correspondence matrix. */
CorrespondenceMatrixType * GetCorrespondenceMatrix();
private:
MedialNodePairCorrespondenceProcess(const Self&); //purposely not implemented
void operator=(const Self&); //purposely not implemented
CoreAtomImagePointer m_CoreAtomImageA;
CoreAtomImagePointer m_CoreAtomImageB;
DistanceMatrixPointer m_DistanceMatrixA;
DistanceMatrixPointer m_DistanceMatrixB;
CorrespondenceMatrixPointer m_CorrespondenceMatrix;
DataStructurePointerType m_DataStructure;
BinaryMetricPointer m_BinaryMetric;
int m_Rows;
int m_Columns;
int m_NumberOfNodePairs;
int m_NumberOfNodeBasePairs;
};
} // end namespace itk
#ifndef ITK_MANUAL_INSTANTIATION
#include "itkMedialNodePairCorrespondenceProcess.txx"
#endif
#endif