VTK/Examples/Cxx/Broken/Images/ImageSobel3D
From KitwarePublic
Jump to navigationJump to searchContents
ImageSobel3D.cxx
#include <vtkSmartPointer.h>
#include <vtkImageMathematics.h>
#include <vtkPointData.h>
#include <vtkImageData.h>
#include <vtkImageCanvasSource2D.h>
#include <vtkImageSobel3D.h>
#include <vtkImageMagnitude.h>
#include <vtkImageExtractComponents.h>
#include <vtkImageShiftScale.h>
#include <vtkRenderWindow.h>
#include <vtkRenderWindowInteractor.h>
#include <vtkInteractorStyleImage.h>
#include <vtkRenderer.h>
#include <vtkImageActor.h>
#include <vtkImageEllipsoidSource.h>
#include <vtkImageCast.h>
#include <vtkGlyph3DMapper.h>
#include <vtkArrowSource.h>
int main(int argc, char *argv[])
{
// Create an image of a rectangle
vtkSmartPointer<vtkImageCanvasSource2D> source =
vtkSmartPointer<vtkImageCanvasSource2D>::New();
source->SetScalarTypeToUnsignedChar();
source->SetExtent(0, 200, 0, 200, 0, 200);
source->SetDrawColor(0,0,0);
source->FillBox(0,200,0,200);
source->SetDrawColor(255,0,0);
source->FillBox(100,120,100,120);
source->Update();
// Find the gradient using a sobel filter
vtkSmartPointer<vtkImageSobel3D> sobelFilter =
vtkSmartPointer<vtkImageSobel3D>::New();
sobelFilter->SetInputConnection(source->GetOutputPort());
sobelFilter->Update();
// Extract the x component of the gradient
vtkSmartPointer<vtkImageExtractComponents> extractXFilter =
vtkSmartPointer<vtkImageExtractComponents>::New();
extractXFilter->SetComponents(0);
extractXFilter->SetInputConnection(sobelFilter->GetOutputPort());
extractXFilter->Update();
double xRange[2];
extractXFilter->GetOutput()->GetPointData()->GetScalars()->GetRange( xRange );
vtkSmartPointer<vtkImageMathematics> xImageAbs =
vtkSmartPointer<vtkImageMathematics>::New();
xImageAbs->SetOperationToAbsoluteValue();
xImageAbs->SetInputConnection(extractXFilter->GetOutputPort());
xImageAbs->Update();
vtkSmartPointer<vtkImageShiftScale> xShiftScale =
vtkSmartPointer<vtkImageShiftScale>::New();
xShiftScale->SetOutputScalarTypeToUnsignedChar();
xShiftScale->SetScale( 255 / xRange[1] );
xShiftScale->SetInputConnection(xImageAbs->GetOutputPort());
xShiftScale->Update();
// Extract the y component of the gradient
vtkSmartPointer<vtkImageExtractComponents> extractYFilter =
vtkSmartPointer<vtkImageExtractComponents>::New();
extractYFilter->SetComponents(1);
extractYFilter->SetInputConnection(sobelFilter->GetOutputPort());
extractYFilter->Update();
double yRange[2];
extractYFilter->GetOutput()->GetPointData()->GetScalars()->GetRange(yRange);
vtkSmartPointer<vtkImageMathematics> yImageAbs =
vtkSmartPointer<vtkImageMathematics>::New();
yImageAbs->SetOperationToAbsoluteValue();
yImageAbs->SetInputConnection(extractYFilter->GetOutputPort());
yImageAbs->Update();
vtkSmartPointer<vtkImageShiftScale> yShiftScale =
vtkSmartPointer<vtkImageShiftScale>::New();
yShiftScale->SetOutputScalarTypeToUnsignedChar();
yShiftScale->SetScale( 255 / yRange[1] );
yShiftScale->SetInputConnection(yImageAbs->GetOutputPort());
yShiftScale->Update();
// Create actors
vtkSmartPointer<vtkImageActor> originalActor =
vtkSmartPointer<vtkImageActor>::New();
originalActor->SetInput(source->GetOutput());
vtkSmartPointer<vtkImageActor> xActor =
vtkSmartPointer<vtkImageActor>::New();
xActor->SetInput(xShiftScale->GetOutput());
vtkSmartPointer<vtkImageActor> yActor =
vtkSmartPointer<vtkImageActor>::New();
yActor->SetInput(yShiftScale->GetOutput());
vtkSmartPointer<vtkArrowSource> arrowSource =
vtkSmartPointer<vtkArrowSource>::New();
sobelFilter->GetOutput()->GetPointData()->SetActiveVectors("ImageScalarsGradient");
vtkSmartPointer<vtkGlyph3DMapper> sobelMapper =
vtkSmartPointer<vtkGlyph3DMapper>::New();
sobelMapper->ScalingOn();
sobelMapper->SetScaleFactor(.05);
sobelMapper->SetSourceConnection(arrowSource->GetOutputPort());
sobelMapper->SetInputConnection(sobelFilter->GetOutputPort());
sobelMapper->Update();
vtkSmartPointer<vtkActor> sobelActor =
vtkSmartPointer<vtkActor>::New();
sobelActor->SetMapper(sobelMapper);
// Define viewport ranges
// (xmin, ymin, xmax, ymax)
double originalViewport[4] = {0.0, 0.0, 0.25, 1.0};
double xViewport[4] = {0.25, 0.0, 0.5, 1.0};
double yViewport[4] = {0.5, 0.0, 0.75, 1.0};
double sobelViewport[4] = {0.75, 0.0, 1.0, 1.0};
// Setup renderers
vtkSmartPointer<vtkRenderer> originalRenderer =
vtkSmartPointer<vtkRenderer>::New();
originalRenderer->SetViewport(originalViewport);
originalRenderer->AddActor(originalActor);
originalRenderer->ResetCamera();
originalRenderer->SetBackground(.4, .5, .6);
vtkSmartPointer<vtkRenderer> xRenderer =
vtkSmartPointer<vtkRenderer>::New();
xRenderer->SetViewport(xViewport);
xRenderer->AddActor(xActor);
xRenderer->ResetCamera();
xRenderer->SetBackground(.4, .5, .7);
vtkSmartPointer<vtkRenderer> yRenderer =
vtkSmartPointer<vtkRenderer>::New();
yRenderer->SetViewport(yViewport);
yRenderer->AddActor(yActor);
yRenderer->ResetCamera();
yRenderer->SetBackground(.4, .5, .7);
vtkSmartPointer<vtkRenderer> sobelRenderer =
vtkSmartPointer<vtkRenderer>::New();
sobelRenderer->SetViewport(sobelViewport);
sobelRenderer->AddActor(sobelActor);
sobelRenderer->ResetCamera();
sobelRenderer->SetBackground(.4, .5, .7);
vtkSmartPointer<vtkRenderWindow> renderWindow =
vtkSmartPointer<vtkRenderWindow>::New();
renderWindow->SetSize(1200, 300);
renderWindow->AddRenderer(originalRenderer);
renderWindow->AddRenderer(xRenderer);
renderWindow->AddRenderer(yRenderer);
renderWindow->AddRenderer(sobelRenderer);
vtkSmartPointer<vtkRenderWindowInteractor> renderWindowInteractor =
vtkSmartPointer<vtkRenderWindowInteractor>::New();
vtkSmartPointer<vtkInteractorStyleImage> style =
vtkSmartPointer<vtkInteractorStyleImage>::New();
renderWindowInteractor->SetInteractorStyle(style);
renderWindowInteractor->SetRenderWindow(renderWindow);
renderWindowInteractor->Initialize();
renderWindowInteractor->Start();
return EXIT_SUCCESS;
}
Please try the new VTKExamples website.
CMakeLists.txt
cmake_minimum_required(VERSION 2.8)
PROJECT(ImageSobel3D)
find_package(VTK REQUIRED)
include(${VTK_USE_FILE})
add_executable(ImageSobel3D MACOSX_BUNDLE ImageSobel3D.cxx)
if(VTK_LIBRARIES)
target_link_libraries(ImageSobel3D ${VTK_LIBRARIES})
else()
target_link_libraries(ImageSobel3D vtkHybrid vtkWidgets)
endif()
Download and Build ImageSobel3D
Click here to download ImageSobel3D. and its CMakeLists.txt file.
Once the tarball ImageSobel3D.tar has been downloaded and extracted,
cd ImageSobel3D/build
- If VTK is installed:
cmake ..
- If VTK is not installed but compiled on your system, you will need to specify the path to your VTK build:
cmake -DVTK_DIR:PATH=/home/me/vtk_build ..
Build the project:
make
and run it:
./ImageSobel3D
WINDOWS USERS PLEASE NOTE: Be sure to add the VTK bin directory to your path. This will resolve the VTK dll's at run time.