mitkSurfaceEqualTest.cpp

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.

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

#include "mitkSurface.h"
#include "mitkTestFixture.h"
#include "mitkTestingMacros.h"

#include <vtkCellArray.h>
#include <vtkPolyData.h>
#include <vtkPolyLine.h>
#include <vtkPolygon.h>
#include <vtkSmartPointer.h>

class mitkSurfaceEqualTestSuite : public mitk::TestFixture
{
  CPPUNIT_TEST_SUITE(mitkSurfaceEqualTestSuite);
  MITK_TEST(Equal_CloneAndOriginalOneTimestep_ReturnsTrue);
  MITK_TEST(Equal_CloneAndOriginalTwoTimesteps_ReturnsTrue);
  MITK_TEST(Equal_OneTimeStepVSTwoTimeStep_ReturnsFalse);
  MITK_TEST(Equal_TwoTimeStepsDifferentPoints_ReturnsFalse);
  MITK_TEST(Equal_DifferentPoints_ReturnsFalse);
  MITK_TEST(Equal_SurfaceWithPolygonSurfaceWithPolyLine_ReturnsFalse);
  CPPUNIT_TEST_SUITE_END();

private:
  mitk::Surface::Pointer m_Surface3D;
  mitk::Surface::Pointer m_Surface3DTwoTimeSteps;

  vtkSmartPointer<vtkPoints> m_PointsOne;
  vtkSmartPointer<vtkPoints> m_PointsTwo;
  vtkSmartPointer<vtkCellArray> m_PolygonArrayTwo;
  vtkSmartPointer<vtkPolyData> m_PolyDataOne;

public:
  void setUp() override
  {
    // generate two sets of points
    m_PointsOne = vtkSmartPointer<vtkPoints>::New();
    m_PointsOne->InsertNextPoint(0.0, 0.0, 0.0);
    m_PointsOne->InsertNextPoint(1.0, 0.0, 0.0);
    m_PointsOne->InsertNextPoint(0.0, 1.0, 0.0);
    m_PointsOne->InsertNextPoint(1.0, 1.0, 0.0);

    m_PointsTwo = vtkSmartPointer<vtkPoints>::New();
    m_PointsTwo->InsertNextPoint(0.0, 0.0, 0.0);
    m_PointsTwo->InsertNextPoint(0.0, 0.0, 2.0);
    m_PointsTwo->InsertNextPoint(0.0, 1.0, 0.0);
    m_PointsTwo->InsertNextPoint(0.0, 1.0, 2.0);

    // generate two polygons
    vtkSmartPointer<vtkPolygon> polygonOne = vtkSmartPointer<vtkPolygon>::New();
    polygonOne->GetPointIds()->SetNumberOfIds(4);
    polygonOne->GetPointIds()->SetId(0, 0);
    polygonOne->GetPointIds()->SetId(1, 1);
    polygonOne->GetPointIds()->SetId(2, 2);
    polygonOne->GetPointIds()->SetId(3, 3);

    vtkSmartPointer<vtkPolygon> polygonTwo = vtkSmartPointer<vtkPolygon>::New();
    polygonTwo->GetPointIds()->SetNumberOfIds(4);
    polygonTwo->GetPointIds()->SetId(0, 3);
    polygonTwo->GetPointIds()->SetId(1, 2);
    polygonTwo->GetPointIds()->SetId(2, 0);
    polygonTwo->GetPointIds()->SetId(3, 1);

    // generate polydatas
    vtkSmartPointer<vtkCellArray> polygonArrayOne = vtkSmartPointer<vtkCellArray>::New();
    polygonArrayOne->InsertNextCell(polygonOne);

    m_PolyDataOne = vtkSmartPointer<vtkPolyData>::New();
    m_PolyDataOne->SetPoints(m_PointsOne);
    m_PolyDataOne->SetPolys(polygonArrayOne);

    m_PolygonArrayTwo = vtkSmartPointer<vtkCellArray>::New();
    m_PolygonArrayTwo->InsertNextCell(polygonTwo);

    vtkSmartPointer<vtkPolyData> polyDataTwo = vtkSmartPointer<vtkPolyData>::New();
    polyDataTwo->SetPoints(m_PointsOne);
    polyDataTwo->SetPolys(m_PolygonArrayTwo);

    // generate surfaces
    m_Surface3D = mitk::Surface::New();
    m_Surface3D->SetVtkPolyData(m_PolyDataOne);

    m_Surface3DTwoTimeSteps = mitk::Surface::New();
    m_Surface3DTwoTimeSteps->SetVtkPolyData(m_PolyDataOne, 0);
    m_Surface3DTwoTimeSteps->SetVtkPolyData(polyDataTwo, 1);
  }

  void tearDown() override
  {
    m_Surface3D = nullptr;
    m_Surface3DTwoTimeSteps = nullptr;
    m_PolyDataOne = nullptr;
    m_PolygonArrayTwo = nullptr;
    m_PointsOne = nullptr;
    m_PointsTwo = nullptr;
  }

  void Equal_CloneAndOriginalOneTimestep_ReturnsTrue()
  {
    MITK_ASSERT_EQUAL(m_Surface3D, m_Surface3D->Clone(), "A one timestep clone should be equal to its original.");
  }

  void Equal_CloneAndOriginalTwoTimesteps_ReturnsTrue()
  {
    MITK_ASSERT_EQUAL(m_Surface3DTwoTimeSteps,
                      m_Surface3DTwoTimeSteps->Clone(),
                      "A two timestep clone should be equal to its original.");
  }

  void Equal_OneTimeStepVSTwoTimeStep_ReturnsFalse()
  {
    MITK_ASSERT_NOT_EQUAL(
      m_Surface3D, m_Surface3DTwoTimeSteps, "A one timestep and two timestep surface should not be equal.");
  }

  void Equal_TwoTimeStepsDifferentPoints_ReturnsFalse()
  {
    vtkSmartPointer<vtkPolyData> polyDataDifferentPoints = vtkSmartPointer<vtkPolyData>::New();
    polyDataDifferentPoints->SetPoints(m_PointsTwo);
    polyDataDifferentPoints->SetPolys(m_PolygonArrayTwo);

    mitk::Surface::Pointer surface3DTwoTimeStepsDifferentPoints = mitk::Surface::New();
    surface3DTwoTimeStepsDifferentPoints->SetVtkPolyData(m_PolyDataOne, 0);
    surface3DTwoTimeStepsDifferentPoints->SetVtkPolyData(polyDataDifferentPoints, 1);

    // The geometry also changes, because the second pointset has a different geometry/extent.
    MITK_ASSERT_NOT_EQUAL(surface3DTwoTimeStepsDifferentPoints,
                          m_Surface3DTwoTimeSteps,
                          "A surface with the same timesteps and different points should not be equal.");
  }

  void Equal_SurfaceWithPolygonSurfaceWithPolyLine_ReturnsFalse()
  {
    // generate a line
    vtkSmartPointer<vtkPolyLine> polyLineOne = vtkSmartPointer<vtkPolyLine>::New();
    polyLineOne->GetPointIds()->SetNumberOfIds(2);
    polyLineOne->GetPointIds()->SetId(0, 0);
    polyLineOne->GetPointIds()->SetId(1, 1);

    vtkSmartPointer<vtkCellArray> polyLineArrayOne = vtkSmartPointer<vtkCellArray>::New();
    polyLineArrayOne->InsertNextCell(polyLineOne);

    vtkSmartPointer<vtkPolyData> polyDataLine = vtkSmartPointer<vtkPolyData>::New();
    polyDataLine->SetPoints(m_PointsOne);
    polyDataLine->SetLines(polyLineArrayOne);

    mitk::Surface::Pointer surface3DLine = mitk::Surface::New();
    surface3DLine->SetVtkPolyData(polyDataLine);

    MITK_ASSERT_NOT_EQUAL(m_Surface3D,
                          surface3DLine,
                          "A surface with the same timesteps and points and the same "
                          "number of cells, but different types of cells should not be "
                          "equal.");
  }

  void Equal_DifferentPoints_ReturnsFalse()
  {
    mitk::Surface::Pointer surfaceWithADifferentPoint = m_Surface3D->Clone();
    // modify points. m_Surface3D contains m_PointsOne
    surfaceWithADifferentPoint->GetVtkPolyData()->SetPoints(m_PointsTwo);

    MITK_ASSERT_NOT_EQUAL(m_Surface3D,
                          surfaceWithADifferentPoint,
                          "A surface with a single timestep and different points should not be equal.");
  }
};

MITK_TEST_SUITE_REGISTRATION(mitkSurfaceEqual)