mitkNodeDisplacementFilter.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 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();
    virtual ~NodeDisplacementFilter();

    virtual 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