FMEwork/ITK/Sources/ITK/MLITK/ITKSupport/mlITKSpecialFieldsSupport.h

Go to the documentation of this file.

// **InsertLicense** code
//----------------------------------------------------------------------------------

//----------------------------------------------------------------------------------

#ifndef __mlITKSpecialFieldsSupport_H
#define __mlITKSpecialFieldsSupport_H

// Include dll-specific settings.
#include "mlInitSystemITKSupport.h"

#ifndef __mlModuleIncludes_H
#include "mlModuleIncludes.h"
#endif

#ifndef __mlMultiFields_H
#include "mlMultiFields.h"
#endif

#ifndef __mlPointList_H
#include "mlPointList.h"
#endif
#ifndef __mlVectorList_H
#include "mlVectorList.h"
#endif
#ifndef __mlXMarkerList_H
#include "mlXMarkerList.h"
#endif
#ifndef __mlITKMultiBaseConnect_H
#include "mlITKMultiBaseConnect.h"
#endif

#include <itkPolyLineParametricPath.h>
#include <itkFiniteDifferenceFunction.h>

ML_START_NAMESPACE

//---------------------------------------------------------------------------
//---------------------------------------------------------------------------
template<class POLYLINEPATHTYPE>
typename POLYLINEPATHTYPE::Pointer ITKPolylineFromBasePointer(Base *baseVal, bool emptyDefaultToOneZeroVal=true)
{
  ML_TRACE_IN("ITKPointSetFromBasePointer()");

  XMarkerListContainer *xmlc = NULL;
  XMarkerList          *xml  = NULL;
  PointList            *pl   = NULL;
  VectorList           *vl   = NULL;

  // Get base field.
  if (baseVal){
    // Check for different base object types.
    if (ML_BASE_IS_A(baseVal, XMarkerListContainer)){ xmlc = static_cast<XMarkerListContainer*>(baseVal); }
    else if (ML_BASE_IS_A(baseVal, XMarkerList))    { xml  = static_cast<XMarkerList*>(baseVal);          }
    else if (ML_BASE_IS_A(baseVal, PointList))      { pl   = static_cast<PointList*>(baseVal);            }
    else if (ML_BASE_IS_A(baseVal, VectorList))     { vl   = static_cast<VectorList*>(baseVal);           }
    else{ /* No valid type in base field. That can happen. */ }
  }

  // Get number of points from list.
  size_t numVals = (xmlc ? xmlc->getList()->size() : 
                   (xml  ? xml->size()            : 
                   (pl   ? static_cast<size_t>(pl->getNum()) : 
                   (vl   ? static_cast<size_t>(vl->getNum()) : 0))));


  // Determine maximum point dimension, clamp it to 6.
  int maxDim = POLYLINEPATHTYPE::VertexType::IndexDimension;
  if (maxDim > 6){
    maxDim = 6;
    ML_PRINT_WARNING("ITKPolylineFromBasePointer", ML_BAD_DIMENSION, "Too high dimension of itk Polyline. Only 6 dimensions will be converted.");
  }

  if (numVals > 0){

    // Create point set object.
    typename POLYLINEPATHTYPE::Pointer outputPolyline = POLYLINEPATHTYPE::New();                                             

    // A point to be added.
    typename POLYLINEPATHTYPE::VertexType point;

    // Insert all list elements into the NodeContainer.
    for (size_t c=0; c < numVals; ++c){
      // Get point list value if we have such a list.
      Vector6 plVal(0);
      if (pl){
        float px=0, py=0, pz=0; 
        pl->getValue(c, px, py, pz); 
        plVal[0] = px;
        plVal[1] = py;
        plVal[2] = pz;
      }

      // Get point list value if we have such a list.
      Vector6 vlVal(0);
      int vecType=0;
      if (vl){
        float px=0, py=0, pz=0; 
        vl->getPoint(c, vecType, px, py, pz);
        vlVal[0]=px;
        vlVal[1]=py;
        vlVal[2]=pz;
      }

      // get position from any of the lists.
      Vector6 mlVec((xmlc ? (static_cast<XMarker*>(xmlc->getList()->getItemAt(c)))->pos :
                    (xml  ? (static_cast<XMarker*>(xml->getItemAt(c)))->pos             :
                    (pl   ? plVal                                            :
                    (vl   ? vlVal                                            : Vector6(0))))));

      // Copy point components.
      for (int j=0;j< maxDim ;j++){ point[j] = mlVec[j]; }
      outputPolyline->AddVertex(point);
    }

    return outputPolyline;
  }
  else{
    // Create and return default (empty) point set object.
    typename POLYLINEPATHTYPE::Pointer outputPolyline = POLYLINEPATHTYPE::New();
    if (emptyDefaultToOneZeroVal){
      // Add at least one point at coordinates (0,...,0).
      typename POLYLINEPATHTYPE::VertexType point;
      outputPolyline->AddVertex(point);
    }
    return outputPolyline;
  }
}


//---------------------------------------------------------------------------
//---------------------------------------------------------------------------
template<class FINITE_DIFFERENCE_FUNCTION_TYPE>
typename FINITE_DIFFERENCE_FUNCTION_TYPE::Pointer ITKDifferenceFunctionFromBasePointer(Base *baseVal)
{
  ML_TRACE_IN("ITKDifferenceFunctionFromBasePointer()");

  // Return value to be validated with function from connection.
  typename FINITE_DIFFERENCE_FUNCTION_TYPE::Pointer defaultFunction = NULL;

  // Pointer to MultiBaseObject is there is one connected.
  MultiBaseType *multiBaseObject=NULL;

  // Get base field.
  if (baseVal){
    // Check for different base object types.
    if (ML_BASE_IS_A(baseVal, MultiBaseTypeWrapper)){
      multiBaseObject = static_cast<MultiBaseTypeWrapper*>(baseVal)->getWrappedOutputObject();
      if (multiBaseObject){
        defaultFunction = multiBaseObject->GetFiniteDifferenceFunction((FINITE_DIFFERENCE_FUNCTION_TYPE*)NULL);
      }
    }
  }
  else{
    // Create and return a default difference function.
    //defaultFunction = FINITE_DIFFERENCE_FUNCTION_TYPE::New();                                             
  }

  // Return the found or the default function.
  return defaultFunction;
}

ML_END_NAMESPACE

#endif // __mlITKSpecialFieldsSupport_H