mitkPointSetFileIOTest.cpp

Go to the documentation of this file.

/*=========================================================================
 
 Program:   Medical Imaging & Interaction Toolkit
 Language:  C++
 Date:      $Date$
 Version:   $Revision$
 
 Copyright (c) German Cancer Research Center, Division of Medical and
 Biological Informatics. All rights reserved.
 See MITKCopyright.txt or https://www.mitk.org/copyright.html for details.
 
 This software is distributed WITHOUT ANY WARRANTY; without even
 the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
 PURPOSE.  See the above copyright notices for more information.
 
 =========================================================================*/

#include "mitkPointSet.h"
#include "mitkPointSetWriter.h"
#include "mitkPointSetReader.h"
#include "mitkTestingMacros.h"
#include <vector>
#include <itksys/SystemTools.hxx>
#include <time.h>

//unsigned int numberOfTestPointSets = 1;
unsigned int numberOfTimeSeries = 5;

// create one test PointSet
class mitkPointSetFileIOTestClass
{
public:

  std::vector<mitk::PointSet::Pointer> m_SavedPointSet;

  mitkPointSetFileIOTestClass()
  {

  }

  mitk::PointSet::Pointer CreateTestPointSet()
  {
    mitk::PointSet::Pointer pointSet = mitk::PointSet::New();

    for (unsigned int t = 0; t < numberOfTimeSeries; t++)
    {
      unsigned int position(0);
      mitk::Point3D point;
      mitk::FillVector3D(point, (rand()%1000) /1000.0 , (rand()%1000) /1000.0, (rand()%1000)/1000.0);
      pointSet->SetPoint(position, point, t);

      mitk::FillVector3D(point, (rand()%1000) /1000.0 , (rand()%1000) /1000.0, (rand()%1000)/1000.0);
      ++position;
      pointSet->SetPoint(position, point, t);

      mitk::FillVector3D(point, (rand()%1000) /1000.0 , (rand()%1000) /1000.0, (rand()%1000)/1000.0);
      ++position;
      pointSet->SetPoint(position, point, t);
    }
    m_SavedPointSet.push_back(pointSet);

    return pointSet;

  }

  void PointSetCompare(mitk::PointSet::Pointer pointSet2,
      mitk::PointSet::Pointer pointSet1, bool& /*identical*/)
  {

    MITK_TEST_CONDITION(pointSet1->GetSize() == pointSet2->GetSize(), "Testing if PointSet size is correct" );

    for (unsigned int t = 0; t < numberOfTimeSeries; t++)
    {
      for (unsigned int i = 0; i < (unsigned int) pointSet1->GetSize(t); ++i)
      {
        mitk::Point3D p1 = pointSet1->GetPoint(i);
        mitk::Point3D p2 = pointSet2->GetPoint(i);

          //test
          std::cout << "r point: " << p2 << std::endl;
          std::cout << "w point: " << p1 << std::endl;

          //test end

        MITK_TEST_CONDITION((p1[0] - p2[0]) <= 0.0001, "Testing if X coordinates of the Point are at the same Position" );
        MITK_TEST_CONDITION((p1[1] - p2[1]) <= 0.0001, "Testing if Y coordinates of the Point are at the same Position" );
        MITK_TEST_CONDITION((p1[2] - p2[2]) <= 0.0001, "Testing if Z coordinates of the Point are at the same Position" );

      }
    }

  }

  bool PointSetWrite(unsigned int numberOfPointSets)
  {
    try
    {
      m_SavedPointSet.clear();

      mitk::PointSetWriter::Pointer pointSetWriter = mitk::PointSetWriter::New();
      pointSetWriter->SetFileName("test_pointset_new.mps");
      
      for (unsigned int i = 0; i < numberOfPointSets; i++)
      {
        pointSetWriter->SetInput(i, CreateTestPointSet());
      }

      pointSetWriter->Write();
    } 
    catch (std::exception& /*e*/)
    {
      return false;
    }

    return true;
  }

  void PointSetLoadAndCompareTest(unsigned int numberOfPointSets)
  {
    try
    {
      mitk::PointSetReader::Pointer pointSetReader =
          mitk::PointSetReader::New();
      mitk::PointSet::Pointer pointSet;

      pointSetReader->SetFileName("test_pointset_new.mps");
      for (unsigned int i = 0; i < numberOfPointSets; i++)
      {
        pointSetReader->Update();
        pointSet = pointSetReader->GetOutput(i);
        MITK_TEST_CONDITION(pointSet.IsNotNull(), "Testing if the loaded Data are NULL" );

        bool identical(true);
        PointSetCompare(pointSet.GetPointer(), m_SavedPointSet.at(i).GetPointer(),
            identical);
      }
    } catch (std::exception& /*e*/)
    {
    }
  }

}; //mitkPointSetFileIOTestClass


int mitkPointSetFileIOTest(int, char*[])
{
  MITK_TEST_BEGIN("PointSet");
  unsigned int numberOfPointSets(5);

  mitkPointSetFileIOTestClass* test = new mitkPointSetFileIOTestClass();

  // write
  MITK_TEST_CONDITION(test->PointSetWrite(numberOfPointSets), "Testing if the PointSetWriter writes Data" );

  // load - compare
  test->PointSetLoadAndCompareTest(numberOfPointSets);
  
  //Delete correctly
  delete test;

  MITK_TEST_END();
}