mitkSphereInterpolator.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.

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

#include "mitkSphereInterpolator.h"

#include <usGetModuleContext.h>
#include <usModuleContext.h>
#include <usModule.h>
#include <usModuleResource.h>
#include <usModuleResourceStream.h>

#include <mitkQBallImage.h>

#include <fstream>
#include <exception>

static const std::string BaryCoordsFileName = "FiberTrackingLUTBaryCoords.bin";
static const std::string IndicesFileName = "FiberTrackingLUTIndices.bin";

SphereInterpolator::SphereInterpolator(const string& lutPath)
{
  m_ValidState = true;
  if (lutPath.length()==0)
  {
    if (!LoadLookuptables())
    {
      m_ValidState = false;
      return;
    }
  }
  else
  {
    if (!LoadLookuptables(lutPath))
    {
      m_ValidState = false;
      return;
    }
  }

  size = 301;
  sN = (size-1)/2;
  nverts = QBALL_ODFSIZE;
  beta = 0.5;

  inva = (sqrt(1+beta)-sqrt(beta));
  b = 1/(1-sqrt(1/beta + 1));
}

SphereInterpolator::~SphereInterpolator()
{

}

bool SphereInterpolator::LoadLookuptables(const string& lutPath)
{
  MITK_INFO << "SphereInterpolator: loading lookuptables from custom path: " << lutPath;

  string path = lutPath; path.append(BaryCoordsFileName);
  std::ifstream BaryCoordsStream;
  BaryCoordsStream.open(path.c_str(), ios::in | ios::binary);
  MITK_INFO << "SphereInterpolator: 1 " << path;
  if (!BaryCoordsStream.is_open())
  {
    MITK_INFO << "SphereInterpolator: could not load FiberTrackingLUTBaryCoords.bin from " << path;
    return false;
  }

  ifstream IndicesStream;
  path = lutPath; path.append("FiberTrackingLUTIndices.bin");
  IndicesStream.open(path.c_str(), ios::in | ios::binary);
  MITK_INFO << "SphereInterpolator: 1 " << path;
  if (!IndicesStream.is_open())
  {
    MITK_INFO << "SphereInterpolator: could not load FiberTrackingLUTIndices.bin from " << path;
    return false;
  }

  if (LoadLookuptables(BaryCoordsStream, IndicesStream))
  {
    MITK_INFO << "SphereInterpolator: first and second lut loaded successfully";
    return true;
  }

  return false;
}

bool SphereInterpolator::LoadLookuptables()
{
  MITK_INFO << "SphereInterpolator: loading lookuptables";

  us::Module* module = us::GetModuleContext()->GetModule();
  us::ModuleResource BaryCoordsRes = module->GetResource(BaryCoordsFileName);
  if (!BaryCoordsRes.IsValid())
  {
    MITK_INFO << "Could not retrieve resource " << BaryCoordsFileName;
    return false;
  }

  us::ModuleResource IndicesRes = module->GetResource(IndicesFileName);
  if (!IndicesRes)
  {
    MITK_INFO << "Could not retrieve resource " << IndicesFileName;
    return false;
  }

  us::ModuleResourceStream BaryCoordsStream(BaryCoordsRes, std::ios_base::binary);
  us::ModuleResourceStream IndicesStream(IndicesRes, std::ios_base::binary);
  return LoadLookuptables(BaryCoordsStream, IndicesStream);
}

bool SphereInterpolator::LoadLookuptables(std::istream& BaryCoordsStream, std::istream& IndicesStream)
{
  if (BaryCoordsStream)
  {
    try
    {
      float tmp;
      BaryCoordsStream.seekg (0, ios::beg);
      while (!BaryCoordsStream.eof())
      {
        BaryCoordsStream.read((char *)&tmp, sizeof(tmp));
        barycoords.push_back(tmp);
      }
    }
    catch (const std::exception& e)
    {
      MITK_INFO << e.what();
    }
  }
  else
  {
    MITK_INFO << "SphereInterpolator: could not load FiberTrackingLUTBaryCoords.bin";
    return false;
  }

  if (IndicesStream)
  {
    try
    {
      int tmp;
      IndicesStream.seekg (0, ios::beg);
      while (!IndicesStream.eof())
      {
        IndicesStream.read((char *)&tmp, sizeof(tmp));
        indices.push_back(tmp);
      }
    }
    catch (const std::exception& e)
    {
      MITK_INFO << e.what();
    }
  }
  else
  {
    MITK_INFO << "SphereInterpolator: could not load FiberTrackingLUTIndices.bin";
    return false;
  }

  return true;
}