mitkInteractionTestHelper.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 mitkInteractionTestHelper_h
#define mitkInteractionTestHelper_h

#include <mitkDataStorage.h>
#include <mitkMouseModeSwitcher.h>
#include <mitkRenderWindow.h>
#include <mitkXML2EventParser.h>

#include <MitkTestingHelperExports.h>

class vtkRenderWindow;
class vtkRenderer;

namespace mitk
{
  class MITKTESTINGHELPER_EXPORT InteractionTestHelper
  {
  public:
    InteractionTestHelper(const std::string &interactionXmlFilePath);

    // unregisters all render windows and its renderers.
    virtual ~InteractionTestHelper();

    mitk::DataStorage::Pointer GetDataStorage();

    void AddNodeToStorage(mitk::DataNode::Pointer node);

    void PlaybackInteraction();

    void SetTimeStep(int newTimeStep);

    typedef std::vector<mitk::RenderWindow::Pointer> RenderWindowListType;

    const RenderWindowListType &GetRenderWindowList() { return m_RenderWindowList; }
    RenderWindow *GetRenderWindowByName(const std::string &name);

    RenderWindow *GetRenderWindowByDefaultViewDirection(mitk::SliceNavigationController::ViewDirection viewDirection);

    RenderWindow *GetRenderWindow(unsigned int index);

    void AddDisplayPlaneSubTree();

    void Set3dCameraSettings();

  protected:
    void Initialize(const std::string &interactionXmlFilePath);

    void LoadInteraction();

    mitk::XML2EventParser::EventContainerType m_Events; // List with loaded interaction events

    std::string m_InteractionFilePath;

    RenderWindowListType m_RenderWindowList;
    mitk::DataStorage::Pointer m_DataStorage;
    mitk::MouseModeSwitcher::Pointer m_MouseModeSwitcher;
  };
} // namespace mitk
#endif