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
|
/*=========================================================================
*
* 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 itkCorrelationImageToImageMetricv4HelperThreader_h
#define itkCorrelationImageToImageMetricv4HelperThreader_h
#include "itkImageToImageMetricv4GetValueAndDerivativeThreader.h"
namespace itk
{
/** \class CorrelationImageToImageMetricv4GetValueAndDerivativeThreader
* \brief Helper class for CorrelationImageToImageMetricv4 \c
* To compute the average pixel intensities of the fixed image and the moving image
* on the sampled points or inside the virtual image region:
* \f$ \bar f (CorrelationImageToImageMetricv4::m_AverageFix ) \f$
* \f$ \bar m (CorrelationImageToImageMetricv4::m_AverageMov ) \f$
*
* \ingroup ITKMetricsv4
*/
template < typename TDomainPartitioner, typename TImageToImageMetric, typename TCorrelationMetric >
class ITK_TEMPLATE_EXPORT CorrelationImageToImageMetricv4HelperThreader
: public ImageToImageMetricv4GetValueAndDerivativeThreader< TDomainPartitioner, TImageToImageMetric >
{
public:
/** Standard class typedefs. */
typedef CorrelationImageToImageMetricv4HelperThreader Self;
typedef ImageToImageMetricv4GetValueAndDerivativeThreader< TDomainPartitioner, TImageToImageMetric > Superclass;
typedef SmartPointer< Self > Pointer;
typedef SmartPointer< const Self > ConstPointer;
itkTypeMacro( CorrelationImageToImageMetricv4HelperThreader, ImageToImageMetricv4GetValueAndDerivativeThreader );
itkNewMacro( Self );
typedef typename Superclass::DomainType DomainType;
typedef typename Superclass::AssociateType AssociateType;
typedef typename Superclass::ImageToImageMetricv4Type ImageToImageMetricv4Type;
typedef typename Superclass::VirtualIndexType VirtualIndexType;
typedef typename Superclass::VirtualPointType VirtualPointType;
typedef typename Superclass::FixedImagePointType FixedImagePointType;
typedef typename Superclass::FixedImagePixelType FixedImagePixelType;
typedef typename Superclass::FixedImageGradientType FixedImageGradientType;
typedef typename Superclass::MovingImagePointType MovingImagePointType;
typedef typename Superclass::MovingImagePixelType MovingImagePixelType;
typedef typename Superclass::MovingImageGradientType MovingImageGradientType;
typedef typename Superclass::MeasureType MeasureType;
typedef typename Superclass::DerivativeType DerivativeType;
typedef typename Superclass::DerivativeValueType DerivativeValueType;
typedef typename Superclass::InternalComputationValueType InternalComputationValueType;
typedef typename Superclass::NumberOfParametersType NumberOfParametersType;
typedef typename Superclass::FixedOutputPointType FixedOutputPointType;
typedef typename Superclass::MovingOutputPointType MovingOutputPointType;
protected:
CorrelationImageToImageMetricv4HelperThreader();
virtual ~CorrelationImageToImageMetricv4HelperThreader() ITK_OVERRIDE;
/** Overload: Resize and initialize per thread objects. */
virtual void BeforeThreadedExecution() ITK_OVERRIDE;
/** Overload:
* Collects the results from each thread and sums them. Results are stored
* in the enclosing class \c m_Value and \c m_DerivativeResult. Behavior
* depends on m_AverageValueAndDerivativeByNumberOfValuePoints,
* m_NumberOfValidPoints, to average the value sum, and to average
* derivative sums for global transforms only (i.e. transforms without local
* support). */
virtual void AfterThreadedExecution() ITK_OVERRIDE;
/* Overload: don't need to compute the image gradients and store derivatives
*
* Method called by the threaders to process the given virtual point. This
* in turn calls \c TransformAndEvaluateFixedPoint, \c
* TransformAndEvaluateMovingPoint, and \c ProcessPoint.
*/
virtual bool ProcessVirtualPoint( const VirtualIndexType & virtualIndex,
const VirtualPointType & virtualPoint,
const ThreadIdType threadId ) ITK_OVERRIDE;
/**
* Not using. All processing is done in ProcessVirtualPoint.
*/
virtual bool ProcessPoint(
const VirtualIndexType & ,
const VirtualPointType & ,
const FixedImagePointType & ,
const FixedImagePixelType & ,
const FixedImageGradientType & ,
const MovingImagePointType & ,
const MovingImagePixelType & ,
const MovingImageGradientType & ,
MeasureType & ,
DerivativeType & ,
const ThreadIdType ) const ITK_OVERRIDE
{
return false;
}
private:
ITK_DISALLOW_COPY_AND_ASSIGN(CorrelationImageToImageMetricv4HelperThreader);
struct CorrelationMetricPerThreadStruct
{
InternalComputationValueType FixSum;
InternalComputationValueType MovSum;
};
itkPadStruct( ITK_CACHE_LINE_ALIGNMENT, CorrelationMetricPerThreadStruct,
PaddedCorrelationMetricPerThreadStruct);
itkAlignedTypedef( ITK_CACHE_LINE_ALIGNMENT, PaddedCorrelationMetricPerThreadStruct,
AlignedCorrelationMetricPerThreadStruct );
/* per thread variables for correlation and its derivatives */
mutable AlignedCorrelationMetricPerThreadStruct * m_CorrelationMetricPerThreadVariables;
/** Internal pointer to the metric object in use by this threader.
* This will avoid costly dynamic casting in tight loops. */
TCorrelationMetric * m_CorrelationAssociate;
};
} // end namespace itk
#ifndef ITK_MANUAL_INSTANTIATION
#include "itkCorrelationImageToImageMetricv4HelperThreader.hxx"
#endif
#endif
|