mitkSurfaceDeformationDataInteractor3D.h

Go to the documentation of this file.

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

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.

Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.

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

#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();
    ~SurfaceDeformationDataInteractor3D() override;
    void ConnectActionsAndFunctions() override;
    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