mitkNodeDisplacementFilter.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 NODEDISPLACEMENTFILTER_H_INCLUDED
#define NODEDISPLACEMENTFILTER_H_INCLUDED

#include "MitkUSNavigationExports.h"

#include <itkProcessObject.h>

#include "mitkBaseRenderer.h"
#include "mitkCommon.h"
#include "mitkNavigationData.h"
#include "mitkNavigationDataPassThroughFilter.h"

//MITK

namespace mitk {
  class MITKUSNAVIGATION_EXPORT NodeDisplacementFilter : public NavigationDataPassThroughFilter
  {
  public:
    mitkClassMacro(NodeDisplacementFilter, NavigationDataPassThroughFilter);
    itkNewMacro(Self);

    bool AddNode(mitk::DataNode::Pointer node);

    bool RemoveNode(unsigned int i);

    virtual int GetNumberOfNodes();

    virtual mitk::DataNode::Pointer GetNode (unsigned int i = 0);

    virtual mitk::NavigationData::Pointer GetRawDisplacementNavigationData(unsigned int i = 0);

    virtual std::vector< mitk::DataNode::Pointer > GetNodes();

    virtual void SelectInput(int i);

    itkSetMacro(InitialReferencePose,mitk::NavigationData::Pointer);

    virtual void ResetNodes();

  protected:
    NodeDisplacementFilter();
    ~NodeDisplacementFilter() override;

    void GenerateData() override;

    mitk::Geometry3D::Pointer TransformToGeometry(mitk::AffineTransform3D::Pointer transform);

    std::vector<mitk::DataNode::Pointer> m_Nodes;

    std::vector<mitk::AffineTransform3D::Pointer> m_Transforms;

    int m_SelectedInput;

    mitk::NavigationData::Pointer m_InitialReferencePose;
  };
} // namespace mitk

#endif