mitkSurfaceDeformationDataInteractor3D.h

Go to the documentation of this file.

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

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center,
Division of Medical and Biological Informatics.
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 or http://www.mitk.org for details.

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

#ifndef mitkSurfaceDeformationDataInteractor3D_h_
#define mitkSurfaceDeformationDataInteractor3D_h_

#include "MitkDataTypesExtExports.h"
#include "mitkDataInteractor.h"
#include "mitkSurface.h"

namespace mitk
{
  // Inherit from DataInteratcor, this provides functionality of a state machine and configurable inputs.
  class MITKDATATYPESEXT_EXPORT SurfaceDeformationDataInteractor3D : public DataInteractor
  {
  public:
    mitkClassMacro(SurfaceDeformationDataInteractor3D, DataInteractor);
    itkFactorylessNewMacro(Self) itkCloneMacro(Self)

      protected : SurfaceDeformationDataInteractor3D();
    virtual ~SurfaceDeformationDataInteractor3D();
    virtual void ConnectActionsAndFunctions() override;
    virtual void DataNodeChanged() override;

    virtual bool CheckOverObject(const InteractionEvent *);
    virtual void SelectObject(StateMachineAction *, InteractionEvent *);
    virtual void DeselectObject(StateMachineAction *, InteractionEvent *);
    virtual void InitDeformation(StateMachineAction *, InteractionEvent *);
    virtual void DeformObject(StateMachineAction *, InteractionEvent *);
    virtual void ScaleRadius(StateMachineAction *, InteractionEvent *);

    enum
    {
      COLORIZATION_GAUSS,
      COLORIZATION_CONSTANT
    };

  private:
    void ColorizeSurface(
      vtkPolyData *polyData, int timeStep, const Point3D &pickedPoint, int mode, double scalar = 0.0);

    double m_InitialPickedWorldPoint[4];
    Point3D m_InitialPickedPoint;
    Point3D m_SurfaceColorizationCenter;

    Surface *m_Surface;
    vtkPolyData *m_OriginalPolyData;

    double m_GaussSigma;
    Vector3D m_ObjectNormal;
  };
}
#endif