mitkSurfaceVtkMapper3DTexturedSphereTest.cpp

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.

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

// MITK
#include "mitkRenderingTestHelper.h"
#include "mitkTestingMacros.h"
#include <mitkNodePredicateDataType.h>
#include <mitkSmartPointerProperty.h>
#include <mitkSurface.h>

// VTK
#include <mitkIOUtil.h>
#include <vtkFloatArray.h>
#include <vtkPointData.h>
#include <vtkPolyData.h>
#include <vtkRegressionTestImage.h>
#include <vtkSphereSource.h>
#include <vtkTextureMapToSphere.h>

int mitkSurfaceVtkMapper3DTexturedSphereTest(int argc, char *argv[])
{
  // load all arguments into a datastorage, take last argument as reference rendering
  // setup a renderwindow of fixed size X*Y
  // render the datastorage
  // compare rendering to reference image
  MITK_TEST_BEGIN("mitkSurfaceVtkMapper3DTexturedSphereTest")

  mitk::RenderingTestHelper renderingHelper(640, 480, argc, argv);
  // This is a test for a 3D surface, thus we need to set the mapper ID to 3D
  renderingHelper.SetMapperID(mitk::BaseRenderer::Standard3D);

  //######## Exmaple code begin ########
  // Generate a sphere in order to map texture on it
  vtkSmartPointer<vtkSphereSource> sphere = vtkSmartPointer<vtkSphereSource>::New();
  sphere->SetThetaResolution(12);
  sphere->SetPhiResolution(12);
  sphere->SetRadius(50.0);     // just to make it huge
  sphere->SetCenter(50, 0, 0); // just to center the sphere in the screen

  // taken from VTK example: http://www.vtk.org/Wiki/VTK/Examples/Python/Visualization/SphereTexture
  vtkSmartPointer<vtkTextureMapToSphere> mapToSphere = vtkSmartPointer<vtkTextureMapToSphere>::New();
  mapToSphere->SetInputConnection(sphere->GetOutputPort());
  mapToSphere->PreventSeamOn();

  // get the texture image from the helper's data storage
  mitk::Image::Pointer textureImage = static_cast<mitk::Image *>(
    renderingHelper.GetDataStorage()->GetNode(mitk::NodePredicateDataType::New("Image"))->GetData());

  // Generate MITK surface
  mitk::Surface::Pointer surfaceToPutTextureOn = mitk::Surface::New();
  surfaceToPutTextureOn->SetVtkPolyData(static_cast<vtkPolyData *>(mapToSphere->GetOutput()));
  // Generate a node
  mitk::DataNode::Pointer surfaceNode = mitk::DataNode::New();
  surfaceNode->SetData(surfaceToPutTextureOn);
  // Make a Property and add to the node
  mitk::SmartPointerProperty::Pointer textureProperty = mitk::SmartPointerProperty::New(textureImage);
  surfaceNode->SetProperty("Surface.Texture", textureProperty);
  // add to data storage
  renderingHelper.AddNodeToStorage(surfaceNode);
  //######## Exmaple code end ########

  // use this to generate a reference screenshot or save the file:
  bool generateReferenceScreenshot = false;
  if (generateReferenceScreenshot)
  {
    renderingHelper.SaveReferenceScreenShot("/home/kilgus/Pictures/RenderingTestData/output.png");
  }

  //### Usage of CompareRenderWindowAgainstReference: See docu of mitkRrenderingTestHelper
  MITK_TEST_CONDITION(renderingHelper.CompareRenderWindowAgainstReference(argc, argv, 50.0) == true,
                      "CompareRenderWindowAgainstReference test result positive?");

  MITK_TEST_END();
}