mitkPhotoacousticVesselTest.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 <mitkTestFixture.h>
#include <mitkTestingMacros.h>

#include <mitkImageReadAccessor.h>

#include "mitkPAInSilicoTissueVolume.h"
#include "mitkPAVector.h"
#include "mitkPAVessel.h"

class mitkPhotoacousticVesselTestSuite : public mitk::TestFixture
{
  CPPUNIT_TEST_SUITE(mitkPhotoacousticVesselTestSuite);
  MITK_TEST(testEmptyInitializationProperties);
  MITK_TEST(testWalkInStraightLine);
  MITK_TEST(testBifurcate);
  CPPUNIT_TEST_SUITE_END();

private:
  mitk::pa::Vessel::Pointer m_TestVessel;
  mitk::pa::Vessel::CalculateNewVesselPositionCallback m_StraightLine;
  mitk::pa::Vessel::CalculateNewVesselPositionCallback m_Diverging;
  mitk::pa::InSilicoTissueVolume::Pointer m_TestInSilicoVolume;
  mitk::pa::TissueGeneratorParameters::Pointer m_TestVolumeParameters;

public:

  void setUp() override
  {
    auto params = mitk::pa::VesselProperties::New();
    m_TestVessel = mitk::pa::Vessel::New(params);
    m_StraightLine = &mitk::pa::VesselMeanderStrategy::CalculateNewDirectionVectorInStraightLine;
    m_Diverging = &mitk::pa::VesselMeanderStrategy::CalculateNewRandomlyDivergingDirectionVector;
    m_TestVolumeParameters = createTestVolumeParameters();
    auto rng = std::mt19937();
    m_TestInSilicoVolume = mitk::pa::InSilicoTissueVolume::New(m_TestVolumeParameters, &rng);
  }

  mitk::pa::TissueGeneratorParameters::Pointer createTestVolumeParameters()
  {
    auto returnParameters =
      mitk::pa::TissueGeneratorParameters::New();

    returnParameters->SetXDim(10);
    returnParameters->SetYDim(10);
    returnParameters->SetZDim(10);
    returnParameters->SetMinBackgroundAbsorption(0);
    returnParameters->SetMaxBackgroundAbsorption(0);
    returnParameters->SetBackgroundScattering(0);
    returnParameters->SetBackgroundAnisotropy(0);
    return returnParameters;
  }

  void testEmptyInitializationProperties()
  {
    CPPUNIT_ASSERT(m_TestVessel->CanBifurcate() == false);
    CPPUNIT_ASSERT(m_TestVessel->IsFinished() == true);
  }

  void testWalkInStraightLine()
  {
    auto testPosition = mitk::pa::Vector::New();
    testPosition->SetElement(0, 0);
    testPosition->SetElement(1, 4);
    testPosition->SetElement(2, 4);
    auto testDirection = mitk::pa::Vector::New();
    testDirection->SetElement(0, 1);
    testDirection->SetElement(1, 0);
    testDirection->SetElement(2, 0);
    auto params = mitk::pa::VesselProperties::New();
    params->SetDoPartialVolume(false);
    params->SetRadiusInVoxel(1);
    params->SetBifurcationFrequency(100);
    params->SetAbsorptionCoefficient(10);
    params->SetScatteringCoefficient(10);
    params->SetAnisotopyCoefficient(10);
    params->SetPositionVector(testPosition);
    params->SetDirectionVector(testDirection);
    m_TestVessel = mitk::pa::Vessel::New(params);

    CPPUNIT_ASSERT(m_TestVessel->CanBifurcate() == false);
    CPPUNIT_ASSERT(m_TestVessel->IsFinished() == false);

    CPPUNIT_ASSERT_MESSAGE(std::to_string(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(0, 4, 4)),
      std::abs(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(0, 4, 4)) <= mitk::eps);

    m_TestVessel->ExpandVessel(m_TestInSilicoVolume, m_StraightLine, 0, nullptr);

    CPPUNIT_ASSERT_MESSAGE(std::to_string(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(0, 4, 4)),
      std::abs(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(0, 4, 4) - 10) <= mitk::eps);
    CPPUNIT_ASSERT_MESSAGE(std::to_string(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(0, 5, 4)),
      std::abs(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(0, 5, 4)) <= mitk::eps);
    CPPUNIT_ASSERT_MESSAGE(std::to_string(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(0, 6, 4)),
      std::abs(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(0, 6, 4)) <= mitk::eps);
    CPPUNIT_ASSERT_MESSAGE(std::to_string(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(0, 4, 5)),
      std::abs(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(0, 4, 5)) <= mitk::eps);
    CPPUNIT_ASSERT_MESSAGE(std::to_string(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(0, 4, 6)),
      std::abs(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(0, 4, 6)) <= mitk::eps);

    CPPUNIT_ASSERT_MESSAGE(std::to_string(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(1, 4, 4)),
      std::abs(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(1, 4, 4)) <= mitk::eps);
    CPPUNIT_ASSERT_MESSAGE(std::to_string(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(2, 4, 4)),
      std::abs(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(2, 4, 4)) <= mitk::eps);

    m_TestVessel->ExpandVessel(m_TestInSilicoVolume, m_StraightLine, 0, nullptr);

    CPPUNIT_ASSERT_MESSAGE(std::to_string(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(1, 4, 4)),
      std::abs(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(1, 4, 4) - 10) <= mitk::eps);
    CPPUNIT_ASSERT_MESSAGE(std::to_string(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(1, 5, 4)),
      std::abs(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(1, 5, 4)) <= mitk::eps);
    CPPUNIT_ASSERT_MESSAGE(std::to_string(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(1, 6, 4)),
      std::abs(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(1, 6, 4)) <= mitk::eps);
    CPPUNIT_ASSERT_MESSAGE(std::to_string(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(1, 4, 5)),
      std::abs(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(1, 4, 5)) <= mitk::eps);
    CPPUNIT_ASSERT_MESSAGE(std::to_string(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(1, 4, 6)),
      std::abs(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(1, 4, 6)) <= mitk::eps);

    CPPUNIT_ASSERT_MESSAGE(std::to_string(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(1, 4, 4)),
      std::abs(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(0, 4, 4) - 10) <= mitk::eps);
    CPPUNIT_ASSERT_MESSAGE(std::to_string(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(1, 4, 4)),
      std::abs(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(1, 4, 4) - 10) <= mitk::eps);
    CPPUNIT_ASSERT_MESSAGE(std::to_string(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(2, 4, 4)),
      std::abs(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(2, 4, 4)) <= mitk::eps);
    CPPUNIT_ASSERT_MESSAGE(std::to_string(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(3, 4, 4)),
      std::abs(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(3, 4, 4)) <= mitk::eps);
  }

  void testBifurcate()
  {
    auto testPosition = mitk::pa::Vector::New();
    testPosition->SetElement(0, 0);
    testPosition->SetElement(1, 4);
    testPosition->SetElement(2, 4);
    auto testDirection = mitk::pa::Vector::New();
    testDirection->SetElement(0, 1);
    testDirection->SetElement(1, 0);
    testDirection->SetElement(2, 0);
    auto params = mitk::pa::VesselProperties::New();
    params->SetRadiusInVoxel(1);
    params->SetBifurcationFrequency(1);
    params->SetAbsorptionCoefficient(10);
    params->SetScatteringCoefficient(10);
    params->SetAnisotopyCoefficient(10);
    params->SetPositionVector(testPosition);
    params->SetDirectionVector(testDirection);
    m_TestVessel = mitk::pa::Vessel::New(params);

    CPPUNIT_ASSERT(m_TestVessel->CanBifurcate() == false);
    CPPUNIT_ASSERT(m_TestVessel->IsFinished() == false);

    CPPUNIT_ASSERT(std::abs(m_TestInSilicoVolume->GetAbsorptionVolume()->GetData(0, 4, 4)) <= mitk::eps);

    m_TestVessel->ExpandVessel(m_TestInSilicoVolume, m_StraightLine, 0, nullptr);
    m_TestVessel->ExpandVessel(m_TestInSilicoVolume, m_StraightLine, 0, nullptr);

    CPPUNIT_ASSERT(m_TestVessel->CanBifurcate() == true);
    std::mt19937 rng;
    auto bifurcationVessel = m_TestVessel->Bifurcate(&rng);

    CPPUNIT_ASSERT(m_TestVessel->CanBifurcate() == false);
    CPPUNIT_ASSERT(bifurcationVessel->CanBifurcate() == false);
  }

  void tearDown() override
  {
  }
};

MITK_TEST_SUITE_REGISTRATION(mitkPhotoacousticVessel)