QmitkPointBasedRegistrationView.h

Go to the documentation of this file.

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

Program:   Medical Imaging & Interaction Toolkit
Language:  C++
Date:      $Date$
Version:   $Revision$

Copyright (c) German Cancer Research Center, Division of Medical and
Biological Informatics. All rights reserved.
See MITKCopyright.txt or https://www.mitk.org/copyright.html 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.

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

#if !defined(QMITK_POINTBASEDREGISTRATION_H__INCLUDED)
#define QMITK_POINTBASEDREGISTRATION_H__INCLUDED

#include "QmitkFunctionality.h"

#include "berryISelectionListener.h"
#include "berryIStructuredSelection.h"

//#include "mitkTestingConfig.h" // IMPORTANT: this defines or undefines BUILD_TESTING !

#include <mitkPointSetInteractor.h>
#include <mitkGlobalInteraction.h>
#include <mitkAffineInteractor.h>
#include <mitkPointSet.h>
#include <vtkCellArray.h>
#include <vtkLandmarkTransform.h>
//#include "QmitkMessageBoxHelper.h"
#include "ui_QmitkPointBasedRegistrationViewControls.h"

#include "../PointbasedregistrationDll.h"

class POINTBASEDREGISTRATION_EXPORT QmitkPointBasedRegistrationView : public QObject, public QmitkFunctionality
{

  friend struct SelListenerPointBasedRegistration;

  Q_OBJECT

public:

  static const std::string VIEW_ID;

  QmitkPointBasedRegistrationView(QObject *parent=0, const char *name=0);

  virtual ~QmitkPointBasedRegistrationView();

  virtual void CreateQtPartControl(QWidget *parent);

  virtual void StdMultiWidgetAvailable (QmitkStdMultiWidget &stdMultiWidget);

  virtual void StdMultiWidgetNotAvailable();

  virtual void CreateConnections();

  virtual void Activated();
  virtual void Deactivated();
  virtual void Visible();
  virtual void Hidden();

  //
  //     #ifdef BUILD_TESTING
  //         / **
  //           \brief Testing entry point
  //         * /
  //         virtual int TestYourself();
  //
  //         / **
  //            \brief Helper method for testing
  //         * /
  //         bool TestAllTools();
  //
  //
  //   protected slots:
  //     /**
  //        \brief Helper method for testing
  //     */
  //     void RegistrationErrorDialogFound( QWidget* widget );
  //
  //     /**
  //     \brief Helper method for testing
  //     */
  //     void ClearPointSetDialogFound( QWidget* widget );
  //
  //   private:
  //     bool m_MessageBox;
  //
  //
  //   public:
  // #else
  //     // slot function is needed, because moc ignores our #ifdefs
  //     void RegistrationErrorDialogFound( QWidget* widget ) {}
  //     // slot function is needed, because moc ignores our #ifdefs
  //     void ClearPointSetDialogFound(QWidget* widget){}
  // #endif

    void DataNodeHasBeenRemoved(const mitk::DataNode* node);

  protected slots:

    void FixedSelected(mitk::DataNode::Pointer fixedImage);

    void MovingSelected(mitk::DataNode::Pointer movingImage);

    void calculateLandmarkbased();

    void calculateLandmarkWarping();

    void calculateLandmarkbasedWithICP();

    void HideMovingImage(bool hide);

    void HideFixedImage(bool hide);

    bool CheckCalculate();

    void UndoTransformation();

    void RedoTransformation();

    void showRedGreen(bool show);

    void OpacityUpdate(float opacity);

    void OpacityUpdate(int opacity);

    void updateMovingLandmarksList();

    void updateFixedLandmarksList();

    void setImageColor(bool redGreen);

    void clearTransformationLists();

    void checkLandmarkError();

    void transformationChanged(int transform);

    bool checkCalculateEnabled();

    void calculate();

    void SetImagesVisible(berry::ISelection::ConstPointer /*selection*/);

    void SwitchImages();
    
protected:

  berry::ISelectionListener::Pointer m_SelListener;
  berry::IStructuredSelection::ConstPointer m_CurrentSelection;

  QmitkStdMultiWidget * m_MultiWidget;

  Ui::QmitkPointBasedRegistrationControls m_Controls;
  mitk::PointSet::Pointer m_FixedLandmarks;
  mitk::PointSet::Pointer m_MovingLandmarks;
  mitk::DataNode::Pointer m_MovingPointSetNode;
  mitk::DataNode::Pointer m_FixedPointSetNode;
  mitk::DataNode::Pointer m_MovingNode;
  mitk::DataNode::Pointer m_FixedNode;
  std::list<mitk::Geometry3D::Pointer> m_UndoGeometryList;
  std::list<mitk::Geometry3D::Pointer> m_UndoPointsGeometryList;
  std::list<mitk::Geometry3D::Pointer> m_RedoGeometryList;
  std::list<mitk::Geometry3D::Pointer> m_RedoPointsGeometryList;
  bool m_ShowRedGreen;
  float m_Opacity;
  float m_OriginalOpacity;
  mitk::Color m_FixedColor;
  mitk::Color m_MovingColor;
  int m_Transformation;
  bool m_HideFixedImage;
  bool m_HideMovingImage;
  std::string m_OldFixedLabel;
  std::string m_OldMovingLabel;
  bool m_Deactivated;
  int m_CurrentFixedLandmarksObserverID;
  int m_CurrentMovingLandmarksObserverID;
  itk::SimpleMemberCommand<QmitkPointBasedRegistrationView>::Pointer m_FixedLandmarksChangedCommand;
  itk::SimpleMemberCommand<QmitkPointBasedRegistrationView>::Pointer m_MovingLandmarksChangedCommand;
};
#endif // !defined(QMITK_POINTBASEDREGISTRATION_H__INCLUDED)