QmitkNiftyRegView.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 QmitkNiftyRegView_h
#define QmitkNiftyRegView_h

// broken
#undef _USE_CUDA

#include "ui_QmitkNiftyRegViewControls.h"
#include <berryISelectionListener.h>
#include <QmitkAbstractView.h>

// ITK
#include <itkMultiThreader.h>

#include "NiftyRegParameters.h"

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

#include <_reg_f3d2.h>

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


#ifdef _USE_NR_DOUBLE
#define PrecisionTYPE double
#else
#define PrecisionTYPE float
#endif


class QmitkNiftyRegView : public QmitkAbstractView
{  
  // this is needed for all Qt objects that should have a Qt meta-object
  // (everything that derives from QObject and wants to have signal/slots)
  Q_OBJECT
  
  friend class RegistrationExecution;

  public:  

    static const std::string VIEW_ID;

    QmitkNiftyRegView();
    virtual ~QmitkNiftyRegView();

    void OnNodeAdded(const mitk::DataNode* node);
    void OnNodeRemoved(const mitk::DataNode* node);
    void OnNodeChanged(const mitk::DataNode* node);


  protected slots:

    void OnSourceImageComboBoxChanged( int index );
    void OnTargetImageComboBoxChanged( int index );

    void OnTargetMaskImageComboBoxChanged( int index );

    // Multi-Scale Options
    
    void OnNumberOfLevelsSpinBoxValueChanged( int value );
    void OnLevelsToPerformSpinBoxValueChanged( int value );
    
    // Input Image Options
    
    void OnSmoothSourceImageDoubleSpinBoxValueChanged( double value );
    void OnSmoothTargetImageDoubleSpinBoxValueChanged( double value );
    void OnNoSmoothingPushButtonPressed( void );

    void OnDoBlockMatchingOnlyRadioButtonToggled( bool checked );
    void OnDoNonRigidOnlyRadioButtonToggled( bool checked );
    void OnDoBlockMatchingThenNonRigidRadioButtonToggled( bool checked );

    // Initial Affine Transformation

    void OnInputAffineCheckBoxToggled( bool checked );
    void OnInputAffineBrowsePushButtonPressed( void );

    void OnInputFlirtCheckBoxStateChanged( int state );

     // Aladin - Initialisation

   void OnUseNiftyHeaderCheckBoxStateChanged( int state );

    // Aladin - Method

    void OnRigidOnlyRadioButtonToggled( bool checked );
    void OnRigidThenAffineRadioButtonToggled( bool checked );
    void OnDirectAffineRadioButtonToggled( bool checked );

    void OnAladinUseSymmetricAlgorithmCheckBoxStateChanged( int state );
    
    void OnAladinIterationsMaxSpinBoxValueChanged( int value );
    void OnPercentBlockSpinBoxValueChanged( int value );
    void OnPercentInliersSpinBoxValueChanged( int value );

    // Aladin - Advanced

    void OnAladinInterpolationNearestRadioButtonToggled( bool checked );
    void OnAladinInterpolationLinearRadioButtonToggled( bool checked );
    void OnAladinInterpolationCubicRadioButtonToggled( bool checked );

    
    
    // Non-Rigid - Initialisation
    
    void OnNonRigidInputControlPointCheckBoxStateChanged( int state );
    void OnNonRigidInputControlPointBrowsePushButtonPressed( void );


    // Non-Rigid - Input Image
    
    void OnLowerThresholdTargetImageDoubleSpinBoxValueChanged( double value );
    void OnUpperThresholdTargetImageDoubleSpinBoxValueChanged( double value );

    void OnUpperThresholdTargetImageAutoPushButtonPressed( void );
    void OnLowerThresholdTargetImageAutoPushButtonPressed( void );

    void OnLowerThresholdSourceImageDoubleSpinBoxValueChanged( double value );
    void OnUpperThresholdSourceImageDoubleSpinBoxValueChanged( double value );

    void OnUpperThresholdSourceImageAutoPushButtonPressed( void );
    void OnLowerThresholdSourceImageAutoPushButtonPressed( void );

    // Non-Rigid - Spline
    
    void OnControlPointSpacingXDoubleSpinBoxValueChanged( double value );
    void OnControlPointSpacingYDoubleSpinBoxValueChanged( double value );
    void OnControlPointSpacingZDoubleSpinBoxValueChanged( double value );

    // Non-Rigid - Objective Function
    
    void OnNumberSourceHistogramBinsSpinBoxValueChanged( int value );
    void OnNumberTargetHistogramBinsSpinBoxValueChanged( int value );

    void OnWeightBendingEnergyDoubleSpinBoxValueChanged( double value );
    void OnWeightLogJacobianDoubleSpinBoxValueChanged( double value );

    void OnLinearEnergyWeightsDoubleSpinBox_1ValueChanged( double value );
    void OnLinearEnergyWeightsDoubleSpinBox_2ValueChanged( double value );

    void OnApproxJacobianLogCheckBoxStateChanged( int state );

    void OnSimilarityNMIRadioButtonToggled( bool checked );
    void OnSimilaritySSDRadioButtonToggled( bool checked );
    void OnSimilarityKLDivRadioButtonToggled( bool checked );
    
    // Non-Rigid - Optimisation

    void OnUseSimpleGradientAscentCheckBoxStateChanged( int state );
    void OnNonRigidIterationsMaxSpinBoxValueChanged( int value );
    void OnUsePyramidalCheckBoxStateChanged( int state );
    
    // Non-Rigid - Advanced

    void OnSmoothingMetricDoubleSpinBoxValueChanged( double value );
    void OnWarpedPaddingValueDoubleSpinBoxValueChanged( double value );
    void OnWarpedPaddingValuePushButtonPressed( void );

    void OnNonRigidNearestInterpolationRadioButtonToggled( bool checked );
    void OnNonRigidLinearInterpolationRadioButtonToggled( bool checked );
    void OnNonRigidCubicInterpolationRadioButtonToggled( bool checked );


    // Execution


    void OnCancelPushButtonPressed( void );

    void OnResetParametersPushButtonPressed( void );
    void OnSaveTransformationPushButtonPressed( void );
    void OnExecutePushButtonPressed( void );

    void OnSaveRegistrationParametersPushButtonPressed( void );
    void OnLoadRegistrationParametersPushButtonPressed( void );

    friend void UpdateProgressBar( float pcntProgress, void *param );

    friend ITK_THREAD_RETURN_TYPE ExecuteRegistration( void *param );


  protected:

    mitk::DataNode::Pointer GetDataNode( QString searchName );

    mitk::DataStorage::SetOfObjects::ConstPointer GetNodes();
    
    void UpdateAladinResultImageFilename();
    void UpdateNonRigidResultImageFilename();

    virtual void CreateQtPartControl(QWidget *parent);

    virtual void SetDefaultParameters();

    virtual void SetGuiToParameterValues();

    virtual void CreateConnections();

    virtual void SetFocus();

    void WriteRegistrationParametersToFile( QString &filename );

    void ReadRegistrationParametersFromFile( QString &filename );

    void PrintSelf( std::ostream& os );

    void Modified();


    Ui::QmitkNiftyRegViewControls m_Controls;

    bool m_Modified;


    float m_ProgressBarOffset;
    float m_ProgressBarRange;

    NiftyRegParameters<PrecisionTYPE> m_RegParameters;

    reg_aladin<PrecisionTYPE> *m_RegAladin;

    reg_f3d<PrecisionTYPE> *m_RegNonRigid;

};


void UpdateProgressBar( float pcntProgress, void *param );
ITK_THREAD_RETURN_TYPE ExecuteRegistration( void *param );

#endif // QmitkNiftyRegView_h