itkFeatureAggregator.h

Go to the documentation of this file.

/*=========================================================================

  Program:   Insight Segmentation & Registration Toolkit
  Module:    itkFeatureAggregator.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 itkFeatureAggregator_h
#define itkFeatureAggregator_h

#include "itkFeatureGenerator.h"
#include "itkImage.h"
#include "itkImageSpatialObject.h"
#include "itkProgressAccumulator.h"

namespace itk
{

template <unsigned int NDimension>
class ITK_TEMPLATE_EXPORT FeatureAggregator : public FeatureGenerator<NDimension>
{
public:
  ITK_DISALLOW_COPY_AND_ASSIGN(FeatureAggregator);

  using Self = FeatureAggregator;
  using Superclass = FeatureGenerator<NDimension>;
  using Pointer = SmartPointer<Self>;
  using ConstPointer = SmartPointer<const Self>;

  itkTypeMacro(FeatureAggregator, FeatureGenerator);

  static constexpr unsigned int Dimension = NDimension;

  using SpatialObjectType = SpatialObject< NDimension >;
  using SpatialObjectPointer = typename SpatialObjectType::Pointer;
  using SpatialObjectConstPointer = typename SpatialObjectType::ConstPointer;

  using OutputPixelType = float;
  using OutputImageType = Image< OutputPixelType, NDimension >;
  using OutputImageSpatialObjectType = ImageSpatialObject< NDimension, OutputPixelType >;

  using FeatureGeneratorType = FeatureGenerator< Dimension >;

  void AddFeatureGenerator( FeatureGeneratorType * generator );

  unsigned long GetMTime() const override;

protected:
  FeatureAggregator();
  ~FeatureAggregator() override;
  void PrintSelf(std::ostream& os, Indent indent) const override;

  void  GenerateData() override;

  unsigned int GetNumberOfInputFeatures() const;

  using InputFeatureType = typename FeatureGeneratorType::SpatialObjectType;

  const InputFeatureType * GetInputFeature( unsigned int featureId ) const;

  ProgressAccumulator::Pointer              m_ProgressAccumulator;

private:
  using FeatureGeneratorPointer = typename FeatureGeneratorType::Pointer;
  using FeatureGeneratorArrayType = std::vector< FeatureGeneratorPointer >;
  using FeatureGeneratorIterator = typename FeatureGeneratorArrayType::iterator;
  using FeatureGeneratorConstIterator = typename FeatureGeneratorArrayType::const_iterator;

  FeatureGeneratorArrayType                 m_FeatureGenerators;

  void UpdateAllFeatureGenerators();

  void virtual ConsolidateFeatures() = 0;

};

} // end namespace itk

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

#endif