cmtkDeblurringVolumeReconstruction.h

Go to the documentation of this file.

/*
//
//  Copyright 1997-2009 Torsten Rohlfing
//
//  Copyright 2004-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: 2752 $
//
//  $LastChangedDate: 2011-01-17 11:33:31 -0800 (Mon, 17 Jan 2011) $
//
//  $LastChangedBy: torstenrohlfing $
//
*/

#ifndef __cmtkDeblurringVolumeReconstruction_h_included_
#define __cmtkDeblurringVolumeReconstruction_h_included_

#include <cmtkconfig.h>

#include <Recon/cmtkInverseInterpolationVolumeReconstructionBase.h>

#include <Registration/cmtkAffineRegistration.h>

#include <Base/cmtkAffineXform.h>
#include <Base/cmtkUniformVolume.h>

#include "Numerics/ap.h"

#include <vector>

namespace
cmtk
{

template<class TPSF>
class DeblurringVolumeReconstruction :
  public InverseInterpolationVolumeReconstructionBase
{
public:
  typedef DeblurringVolumeReconstruction<TPSF> Self;

  typedef InverseInterpolationVolumeReconstructionBase Superclass;

  DeblurringVolumeReconstruction( const UniformVolume* originalImage, const size_t numberOfPasses, const int interleaveAxis, const Types::Coordinate psfScale = 1.0 )
    : InverseInterpolationVolumeReconstructionBase( originalImage, numberOfPasses, interleaveAxis )
  { 
    this->m_FunctionAndGradient = new typename Self::FunctionAndGradient( this );
    
    cmtk::FixedVector<3,cmtk::Types::Coordinate> scale( originalImage->m_Delta );
    scale *= psfScale;
    scale[interleaveAxis] *= numberOfPasses;
    for ( size_t i = 0; i < numberOfPasses; ++i )
      {
      typename TPSF::SmartPtr psf( new TPSF( scale ) );
      this->m_PassImagePSF.push_back( psf );
      }
  }

  DeblurringVolumeReconstruction( const UniformVolume* reconstructionGrid, std::vector<UniformVolume::SmartPtr>& images, const Vector3D& psfSize )
    : InverseInterpolationVolumeReconstructionBase( reconstructionGrid, images )
  { 
    this->m_FunctionAndGradient = new typename Self::FunctionAndGradient( this );

    for ( size_t i = 0; i < images.size(); ++i )
      {
      const Vector3D scale( psfSize );
      typename TPSF::SmartPtr psf( new TPSF( scale ) );
      this->m_PassImagePSF.push_back( psf );
      }
  }
  
private:
  std::vector<typename TPSF::SmartPtr> m_PassImagePSF;

  void Blur( const ap::real_1d_array& reconstructedPixelArray );

  void ComputeErrorGradientImage( ap::real_1d_array& g );

  void GetBoundingBoxOfXformedPassNeighborhood( int* region, 
                                                const UniformVolume* correctedImage, 
                                                const Vector3D& currentPassVoxel, 
                                                const TPSF* psf, 
                                                const AffineXform* passToCorrectedXform, 
                                                const DataGrid::IndexType& correctedImageDims ) const;

  class FunctionAndGradient : 
    public ap::FunctionAndGradient
  {
  public:
    typedef DeblurringVolumeReconstruction FunctionType;

    FunctionAndGradient( FunctionType* function )
    {
      this->m_Function = function;
    }

    virtual void Evaluate( const ap::real_1d_array& x, ap::real_value_type& f, ap::real_1d_array& g );

  private:
    FunctionType* m_Function;
  };

  friend class Self::FunctionAndGradient;
};

#include "cmtkDeblurringVolumeReconstruction.txx"


} // namespace cmtk

#endif // #ifndef __cmtkDeblurringVolumeReconstruction_h_included_