mitk::Surface::Pointer mitk::PlanarFigureSegmentationController::CreateSurfaceFromPlanarFigure( mitk::PlanarFigure::Pointer figure )
{
if ( figure.IsNull() )
{
MITK_ERROR << "Given PlanarFigure is NULL. Please provide valid PlanarFigure.";
return NULL;
}
mitk::Surface::Pointer newSurface = mitk::Surface::New();
vtkSmartPointer<vtkPoints> points = vtkSmartPointer<vtkPoints>::New();
vtkSmartPointer<vtkPolygon> polygon = vtkSmartPointer<vtkPolygon>::New();
vtkSmartPointer<vtkCellArray> cells = vtkSmartPointer<vtkCellArray>::New();
vtkSmartPointer<vtkPolyData> polyData = vtkSmartPointer<vtkPolyData>::New();
const mitk::Geometry2D* figureGeometry = figure->GetGeometry2D();
// Get the polyline
mitk::PlanarFigure::PolyLineType planarPolyLine = figure->GetPolyLine(0);
mitk::PlanarFigure::PolyLineType::iterator iter;
// iterate over the polyline, ...
int pointCounter = 0;
for( iter = planarPolyLine.begin(); iter != planarPolyLine.end(); iter++ )
{
// ... determine the world-coordinates
mitk::Point2D polyLinePoint = iter->Point;
mitk::Point3D pointInWorldCoordiantes;
figureGeometry->Map( polyLinePoint, pointInWorldCoordiantes );
// and add them as new points to the vtkPoints
points->InsertNextPoint( pointInWorldCoordiantes[0], pointInWorldCoordiantes[1], pointInWorldCoordiantes[2] );
++pointCounter;
}
// create a polygon with the points of the polyline
polygon->GetPointIds()->SetNumberOfIds( pointCounter );
for(int i = 0; i < pointCounter; i++)
{
polygon->GetPointIds()->SetId(i,i);
}
// initialize the vtkCellArray and vtkPolyData
cells->InsertNextCell(polygon);
polyData->SetPoints(points);
polyData->SetPolys( cells );
// set the polydata to the surface
newSurface->SetVtkPolyData( polyData );
return newSurface;
}
void mitk::PlanarFigureSegmentationController::AddPlanarFigure( mitk::PlanarFigure::Pointer planarFigure )
{
if ( planarFigure.IsNull() )
return;
bool newFigure = true;
std::size_t indexOfFigure = 0;
for( std::size_t i=0; i<m_PlanarFigureList.size(); i++ )
{
if( m_PlanarFigureList.at(i) == planarFigure )
{
indexOfFigure = i;
newFigure = false;
break;
}
}
mitk::Surface::Pointer figureAsSurface = NULL;
if ( newFigure )
{
m_PlanarFigureList.push_back( planarFigure );
figureAsSurface = this->CreateSurfaceFromPlanarFigure( planarFigure );
m_SurfaceList.push_back( figureAsSurface );
if (!m_PlanarFigureList.empty())
{
indexOfFigure = m_PlanarFigureList.size() -1 ;
}
}
else
{
figureAsSurface = this->CreateSurfaceFromPlanarFigure( planarFigure );
m_SurfaceList.at(indexOfFigure) = figureAsSurface;
}
if ( m_ReduceFilter.IsNull() )
{
InitializeFilters();
}
m_ReduceFilter->SetInput( indexOfFigure, figureAsSurface );
m_NormalsFilter->SetInput( indexOfFigure, m_ReduceFilter->GetOutput( indexOfFigure ) );
m_DistanceImageCreator->SetInput( indexOfFigure, m_NormalsFilter->GetOutput( indexOfFigure ) );
}
void mitk::PlanarFigureSegmentationController::RemovePlanarFigure( mitk::PlanarFigure::Pointer planarFigure )
{
if ( planarFigure.IsNull() )
return;
bool figureFound = false;
std::size_t indexOfFigure = 0;
for( std::size_t i=0; i<m_PlanarFigureList.size(); i++ )
{
if( m_PlanarFigureList.at(i) == planarFigure )
{
indexOfFigure = i;
figureFound = true;
break;
}
}
if ( !figureFound )
return;
// TODO: fix this! The following code had to be removed as the method
// RemoveInputs() has been removed in ITK 4
// The remaining code works correctly but is slower
if ( false && indexOfFigure == m_PlanarFigureList.size()-1 )
{
// Ff the removed figure was the last one in the list, we can simply
// remove the last input from each filter.
// m_DistanceImageCreator->RemoveInputs( m_NormalsFilter->GetOutput( indexOfFigure ) );
// m_NormalsFilter->RemoveInput( m_ReduceFilter->GetOutput( indexOfFigure ) );
// m_ReduceFilter->RemoveInput( const_cast<mitk::Surface*>(m_ReduceFilter->GetInput(indexOfFigure)) );
}
else
{
// this is not very nice! If the figure that has been removed is NOT the last
// one in the list we have to create new filters and add all remaining
// inputs again.
//
// Has to be done as the filters do not work when removing an input
// other than the last one.
// create new filters
InitializeFilters();
// and add all existing surfaces
SurfaceListType::iterator surfaceIter = m_SurfaceList.begin();
for ( surfaceIter = m_SurfaceList.begin(); surfaceIter!=m_SurfaceList.end(); surfaceIter++ )
{
m_ReduceFilter->SetInput( indexOfFigure, (*surfaceIter) );
m_NormalsFilter->SetInput( indexOfFigure, m_ReduceFilter->GetOutput( indexOfFigure ) );
m_DistanceImageCreator->SetInput( indexOfFigure, m_NormalsFilter->GetOutput( indexOfFigure ) );
}
}
PlanarFigureListType::iterator whereIter = m_PlanarFigureList.begin();
whereIter += indexOfFigure;
m_PlanarFigureList.erase( whereIter );
SurfaceListType::iterator surfaceIter = m_SurfaceList.begin();
surfaceIter += indexOfFigure;
m_SurfaceList.erase( surfaceIter );
}