mitkTubeGraphDataInteractor.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 mitkTubeGraphDataInteractor_h
#define mitkTubeGraphDataInteractor_h
 
#include <MitkTubeGraphExports.h>
 
#include <mitkBaseRenderer.h>
#include <mitkDataInteractor.h>
 
#include "mitkTubeGraph.h"
#include "mitkTubeGraphProperty.h"
 
namespace mitk
{
  // Define events for TubeGraph interaction notifications
  itkEventMacroDeclaration(SelectionChangedTubeGraphEvent, itk::AnyEvent);
 
  // Inherit from DataInteratcor, this provides functionality of a state machine and configurable inputs.
  class MITKTUBEGRAPH_EXPORT TubeGraphDataInteractor : public DataInteractor
  {
  public:
    mitkClassMacro(TubeGraphDataInteractor, DataInteractor);
    itkNewMacro(Self);
 
    enum ActivationMode
    {
      None = 0,
      Single,
      ToRoot,
      ToPeriphery,
      Points,
      Multiple
    };
 
    enum ActionMode
    {
      AttributationMode = 0,
      AnnotationMode,
      EditMode,
      RootMode,
      InformationMode
    };
 
    void SetActivationMode(const ActivationMode &activationMode);
    ActivationMode GetActivationMode();
 
    void SetActionMode(const ActionMode &actionMode);
    ActionMode GetActionMode();
 
    void ResetPickedTubes();
 
        mitk::Point3D GetLastPickedPosition();
 
  protected:
    TubeGraphDataInteractor();
    ~TubeGraphDataInteractor() override;
 
    void ConnectActionsAndFunctions() override;
 
    void DataNodeChanged() override;
 
    virtual bool CheckOverTube(const InteractionEvent *);
    virtual void SelectTube(StateMachineAction *, InteractionEvent *);
    virtual void DeselectTube(StateMachineAction *, InteractionEvent *);
 
    void SelectTubesByActivationModus();
    void UpdateActivation();
 
  private:
    std::vector<TubeGraph::TubeDescriptorType> GetTubesToRoot();
    std::vector<TubeGraph::TubeDescriptorType> GetTubesBetweenPoints();
    std::vector<TubeGraph::TubeDescriptorType> GetPathToPeriphery();
    std::vector<TubeGraph::TubeDescriptorType> GetPathBetweenTubes(const TubeGraph::TubeDescriptorType &start,
                                                                   const TubeGraph::TubeDescriptorType &end);
 
    TubeGraph::Pointer m_TubeGraph;
    TubeGraphProperty::Pointer m_TubeGraphProperty;
    TubeGraph::TubeDescriptorType m_LastPickedTube;
    TubeGraph::TubeDescriptorType m_SecondLastPickedTube;
    ActivationMode m_ActivationMode;
    ActionMode m_ActionMode;
    mitk::TubeElement *m_LastPickedElement = nullptr;
  };
}
#endif