NiftyRegParameters.h

Go to the documentation of this file.

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

  NifTK: A software platform for medical image computing.

  Copyright (c) University College London (UCL). All rights reserved.

  This software is distributed WITHOUT ANY WARRANTY; without even
  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  PURPOSE.

  See LICENSE.txt in the top level directory for details.

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

#ifndef NiftyRegParameters_h
#define NiftyRegParameters_h

// broken
#undef _USE_CUDA

#include "RegAladinParameters.h"
#include "RegF3dParameters.h"

#include <QString>
#include <mitkImage.h>

#include <_reg_aladin.h>
#include <_reg_aladin_sym.h>
#include <_reg_tools.h>

#include <_reg_f3d2.h>

#ifdef _USE_CUDA
#include <_reg_f3d_gpu.h>
#endif


template <class PRECISION_TYPE>
class NiftyRegParameters
{  

  public:  

    NiftyRegParameters();
    virtual ~NiftyRegParameters();

    void SetDefaultParameters();

    void PrintSelf( std::ostream& os );

    NiftyRegParameters<PRECISION_TYPE> &operator=(const NiftyRegParameters<PRECISION_TYPE> &p);

    reg_aladin<PRECISION_TYPE> 
      *CreateAladinRegistrationObject( mitk::Image *mitkSourceImage, 
               mitk::Image *mitkTargetImage, 
               mitk::Image *mitkTargetMaskImage );

    reg_f3d<PRECISION_TYPE> 
      *CreateNonRigidRegistrationObject( mitk::Image *mitkSourceImage, 
           mitk::Image *mitkTargetImage, 
           mitk::Image *mitkTargetMaskImage );
    

    void DeallocateImages( void );


    int m_LevelNumber;
    int m_Level2Perform;    

    // Smooth the target image using the specified sigma (mm) 
    float m_TargetSigmaValue;
    // Smooth the source image using the specified sigma (mm)
    float m_SourceSigmaValue;

    bool m_FlagDoInitialRigidReg;
    bool m_FlagDoNonRigidReg;

    QString m_InputAffineName;  // -inaff
    bool m_FlagInputAffine;
    bool m_FlagFlirtAffine;   // -affFlirt


    RegAladinParameters m_AladinParameters;

    RegF3dParameters<PRECISION_TYPE> m_F3dParameters;


    nifti_image *m_ReferenceImage;
    nifti_image *m_FloatingImage;
    nifti_image *m_ReferenceMaskImage;
    nifti_image *m_ControlPointGridImage;


};

#ifndef ITK_MANUAL_INSTANTIATION
#include "NiftyRegParameters.txx"
#endif

#endif // NiftyRegParameters_h