niftkAffineTransformer.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 niftkAffineTransformer_h
#define niftkAffineTransformer_h

#include <itkObject.h>
#include <mitkBaseData.h>
#include <mitkDataStorage.h>
#include <mitkImage.h>
#include <vtkSmartPointer.h>
#include <vtkLinearTransform.h>

#include "niftkAffineTransformParametersDataNodeProperty.h"
#include "niftkAffineTransformDataNodeProperty.h"

#include "niftkCoreExports.h"

namespace niftk
{

class NIFTKCORE_EXPORT AffineTransformer : public itk::Object
{

public:

  static const std::string VIEW_ID;

  static const std::string INITIAL_TRANSFORM_KEY;

  static const std::string INCREMENTAL_TRANSFORM_KEY;

  static const std::string PRELOADED_TRANSFORM_KEY;

  static const std::string DISPLAYED_TRANSFORM_KEY;

  static const std::string DISPLAYED_PARAMETERS_KEY;

  mitkClassMacroItkParent(AffineTransformer, itk::Object)
  itkNewMacro(AffineTransformer)

  itkGetMacro(RotateAroundCenter, bool);
  itkSetMacro(RotateAroundCenter, bool);

  void SetDataStorage(mitk::DataStorage::Pointer dataStorage);

  mitk::DataStorage::Pointer GetDataStorage() const;

  AffineTransformParametersDataNodeProperty::Pointer GetCurrentTransformParameters() const;

  vtkSmartPointer<vtkMatrix4x4> GetTransformMatrixFromNode(std::string which) const;

  vtkSmartPointer<vtkMatrix4x4> GetCurrentTransformMatrix() const;

  void ResetTransform();

  void OnNodeChanged(mitk::DataNode::Pointer node);

  void OnParametersChanged(AffineTransformParametersDataNodeProperty::Pointer paramsProperty);

  void OnSaveTransform(std::string filename);

  void OnLoadTransform(std::string filename);

  void OnApplyTransform();

  void OnResampleTransform();

  void InitialiseTransformProperty(std::string name, mitk::DataNode::Pointer node);

  void InitialiseNodeProperties(mitk::DataNode::Pointer node);

  void UpdateNodeProperties(const vtkSmartPointer<vtkMatrix4x4> displayedTransformFromParameters,
                            const vtkSmartPointer<vtkMatrix4x4> incrementalTransformToBeComposed,
                            mitk::DataNode::Pointer);

  void UpdateTransformProperty(std::string name, vtkSmartPointer<vtkMatrix4x4> transform, mitk::DataNode::Pointer node);

  void ApplyTransformToNode(const vtkSmartPointer<vtkMatrix4x4> transformFromFile, mitk::DataNode::Pointer node);

  void ApplyResampleToCurrentNode();

protected:

  AffineTransformer();
  virtual ~AffineTransformer();

  AffineTransformer(const AffineTransformer&); // Purposefully not implemented.
  AffineTransformer& operator=(const AffineTransformer&); // Purposefully not implemented.

  virtual vtkSmartPointer<vtkMatrix4x4> ComputeTransformFromParameters(void) const;

  void UpdateTransformationGeometry();

private:
  AffineTransformParametersDataNodeProperty::Pointer m_CurrDispTransfProp;

  mitk::DataStorage::Pointer m_DataStorage;

  mitk::DataNode::Pointer    m_CurrentDataNode;

  bool                       m_RotateAroundCenter;

  double                     m_CentreOfRotation[3];

  double                     m_Translation[3];
  double                     m_Rotation[3];
  double                     m_Scaling[3];
  double                     m_Shearing[3];
};

}

#endif