PlusCommon.txx

Go to the documentation of this file.

/*=Plus=header=begin======================================================
Program: Plus
Copyright (c) Laboratory for Percutaneous Surgery. All rights reserved.
See License.txt for details.
=========================================================Plus=header=end*/

namespace PlusCommon
{
  namespace XML
  {
    //----------------------------------------------------------------------------
    template<typename T>
    PlusStatus SafeGetAttributeValueInsensitive(vtkXMLDataElement& element, const std::wstring& attributeName, T& value)
    {
      return PlusCommon::XML::SafeGetAttributeValueInsensitive(element, std::string(begin(attributeName), end(attributeName)), value);
    }

    //----------------------------------------------------------------------------
    template<typename T>
    PlusStatus SafeGetAttributeValueInsensitive(vtkXMLDataElement& element, const std::string& attributeName, T& value)
    {
      if (element.GetAttribute(attributeName.c_str()) == NULL)
      {
        return PLUS_FAIL;
      }
      element.GetScalarAttribute(attributeName.c_str(), value);
      return PLUS_SUCCESS;
    }
  }

  //----------------------------------------------------------------------------
  template<typename ElemType>
  void JoinTokensIntoString(const std::vector<ElemType>& elems, std::string& output, char separator)
  {
    output = "";
    std::stringstream ss;
    typedef std::vector<ElemType> List;

    for (typename List::const_iterator it = elems.begin(); it != elems.end(); ++it)
    {
      ss << *it;
      if (it != elems.end() - 1)
      {
        ss << separator;
      }
    }

    output = ss.str();
  }

  //----------------------------------------------------------------------------
  template<typename ScalarType>
  PlusStatus DeepCopyVtkVolumeToItkVolume( vtkImageData* inFrame, typename itk::Image< ScalarType, 3 >::Pointer outFrame )
  {
    LOG_TRACE("PlusCommon::ConvertVtkImageToItkImage"); 
  
    if ( inFrame == NULL )
    {
      LOG_ERROR("Failed to convert vtk image to itk image - input image is NULL!"); 
      return PLUS_FAIL; 
    }
  
    if ( outFrame.IsNull() )
    {
      LOG_ERROR("Failed to convert vtk image to itk image - output image is NULL!"); 
      return PLUS_FAIL; 
    }
  
    if( igsioVideoFrame::GetNumberOfBytesPerScalar(inFrame->GetScalarType()) != sizeof(ScalarType) )
    {
      LOG_ERROR("Mismatch between input and output scalar types. In: " << igsioVideoFrame::GetStringFromVTKPixelType(inFrame->GetScalarType()));
      return PLUS_FAIL;
    }
  
    // convert vtkImageData to itkImage 
    vtkSmartPointer<vtkImageExport> imageExport = vtkSmartPointer<vtkImageExport>::New(); 
    imageExport->SetInputData(inFrame); 
    imageExport->Update(); 
  
    int extent[6]={0,0,0,0,0,0}; 
    inFrame->GetExtent(extent); 
  
    double width = extent[1] - extent[0] + 1; 
    double height = extent[3] - extent[2] + 1;
    double thickness = extent[5] - extent[4] + 1;
    typename itk::Image< ScalarType, 3 >::SizeType size;
    size[0] = width;
    size[1] = height;
    size[2] = thickness;
    typename itk::Image< ScalarType, 3 >::IndexType start;
    start[0]=0;
    start[1]=0;
    start[2]=0;
    typename itk::Image< ScalarType, 3 >::RegionType region;
    region.SetSize(size);
    region.SetIndex(start);
    outFrame->SetRegions(region);
    try 
    {
      outFrame->Allocate();
    }
    catch(itk::ExceptionObject & err)
    {
      LOG_ERROR("Failed to allocate memory for the image conversion: " << err.GetDescription() ); 
      return PLUS_FAIL; 
    }
  
    imageExport->Export( outFrame->GetBufferPointer() ); 
  
    return PLUS_SUCCESS;
  }
  
  //----------------------------------------------------------------------------
  template<typename ScalarType>
  PlusStatus DeepCopyVtkVolumeToItkImage( vtkImageData* inFrame, typename itk::Image< ScalarType, 2 >::Pointer outFrame )
  {
    LOG_TRACE("PlusCommon::ConvertVtkImageToItkImage"); 
  
    if ( inFrame == NULL )
    {
      LOG_ERROR("Failed to convert vtk image to itk image - input image is NULL!"); 
      return PLUS_FAIL; 
    }
  
    if ( outFrame.IsNull() )
    {
      LOG_ERROR("Failed to convert vtk image to itk image - output image is NULL!"); 
      return PLUS_FAIL; 
    }
  
    if( igsioVideoFrame::GetNumberOfBytesPerScalar(inFrame->GetScalarType()) != sizeof(ScalarType) )
    {
      LOG_ERROR("Mismatch between input and output scalar types. In: " << igsioVideoFrame::GetStringFromVTKPixelType(inFrame->GetScalarType()));
      return PLUS_FAIL;
    }
  
    int extent[6]={0,0,0,0,0,0}; 
    inFrame->GetExtent(extent);
    
    if( extent[5]-extent[4] > 1 )
    {
   LOG_WARNING("3D volume sent in to PlusCommon::DeepCopyVtkVolumeToItkImage. Only first slice will be copied.");
    }
  
    // convert vtkImageData to itkImage 
    vtkSmartPointer<vtkImageExport> imageExport = vtkSmartPointer<vtkImageExport>::New(); 
    imageExport->SetInputData(inFrame); 
    imageExport->Update(); 
  
    double width = extent[1] - extent[0] + 1; 
    double height = extent[3] - extent[2] + 1;
  
    typename itk::Image< ScalarType, 2 >::SizeType size;
    size[0] = width;
    size[1] = height;
    typename itk::Image< ScalarType, 2 >::IndexType start;
    start[0]=0;
    start[1]=0;
    typename itk::Image< ScalarType, 2 >::RegionType region;
    region.SetSize(size);
    region.SetIndex(start);
    outFrame->SetRegions(region);
    try 
    {
      outFrame->Allocate();
    }
    catch(itk::ExceptionObject & err)
    {
      LOG_ERROR("Failed to allocate memory for the image conversion: " << err.GetDescription() ); 
      return PLUS_FAIL; 
    }
  
    imageExport->Export( outFrame->GetBufferPointer() ); 
  
    return PLUS_SUCCESS;
  }

}