cmtkImageOperationReplace.h

Go to the documentation of this file.

/*
//
//  Copyright 2010 Torsten Rohlfing
//
//  Copyright 2011 SRI International
//
//  This file is part of the Computational Morphometry Toolkit.
//
//  http://www.nitrc.org/projects/cmtk/
//
//  The Computational Morphometry Toolkit is free software: you can
//  redistribute it and/or modify it under the terms of the GNU General Public
//  License as published by the Free Software Foundation, either version 3 of
//  the License, or (at your option) any later version.
//
//  The Computational Morphometry Toolkit is distributed in the hope that it
//  will be useful, but WITHOUT ANY WARRANTY; without even the implied
//  warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//  GNU General Public License for more details.
//
//  You should have received a copy of the GNU General Public License along
//  with the Computational Morphometry Toolkit.  If not, see
//  <http://www.gnu.org/licenses/>.
//
//  $Revision: 2755 $
//
//  $LastChangedDate: 2011-01-18 11:42:28 -0800 (Tue, 18 Jan 2011) $
//
//  $LastChangedBy: torstenrohlfing $
//
*/

#ifndef __cmtkImageOperationReplace_h_included_
#define __cmtkImageOperationReplace_h_included_

#include <cmtkconfig.h>

#include <Base/cmtkImageOperation.h>

namespace
cmtk
{

class ImageOperationReplace
  : public ImageOperation
{
public:
  typedef ImageOperationReplace Self;

  typedef ImageOperation Superclass;

  typedef enum
  {
    REPLACE_PADDING,
    REPLACE_INF_NAN
  } Mode;

  ImageOperationReplace( const Self::Mode mode , const Types::DataItem value  )
    : m_Mode( mode ),
      m_ReplacementValue( value )
  {}

  virtual cmtk::UniformVolume::SmartPtr  Apply( cmtk::UniformVolume::SmartPtr& volume )
  {
    TypedArray& volumeData = *(volume->GetData());
    switch ( this->m_Mode ) 
      {
      case Self::REPLACE_PADDING:
        volumeData.ReplacePaddingData( this->m_ReplacementValue );
        break;
      case Self::REPLACE_INF_NAN:
#pragma omp parallel for
        for ( size_t i = 0; i < volumeData.GetDataSize(); ++i )
          {
          cmtk::Types::DataItem value = 0;
          if ( volumeData.Get( value, i ) )
            {
            if ( !finite( value ) )
              {
              volumeData.Set( this->m_ReplacementValue, i );
              }
            }
          }
        break;
      }
    return volume;
  }
  
  static void NewReplacePadding( const double value  )
  {
    ImageOperation::m_ImageOperationList.push_back( SmartPtr( new Self( Self::REPLACE_PADDING, value ) ) );
  }

  static void NewReplaceInfNaN( const double value  )
  {
    ImageOperation::m_ImageOperationList.push_back( SmartPtr( new Self( Self::REPLACE_INF_NAN, value ) ) );
  }
  
private:
  Self::Mode m_Mode;

  Types::DataItem m_ReplacementValue;
};

} // namespace cmtk

#endif // #ifndef __cmtkImageOperationReplace_h_included_