DrawOnAnImage
Repository source: DrawOnAnImage
Description¶
This example draws a circle in the center of the input image using vtkImageCanvasSource2D's DrawCircle method.
Question
If you have a question about this example, please use the VTK Discourse Forum
Code¶
DrawOnAnImage.cxx
#include <vtkImageBlend.h>
#include <vtkImageCanvasSource2D.h>
#include <vtkImageData.h>
#include <vtkImageReader2.h>
#include <vtkImageReader2Factory.h>
#include <vtkImageViewer2.h>
#include <vtkNamedColors.h>
#include <vtkNew.h>
#include <vtkRenderWindow.h>
#include <vtkRenderWindowInteractor.h>
#include <vtkRenderer.h>
#include <vtkSmartPointer.h>
#include <array>
int main(int argc, char* argv[])
{
vtkNew<vtkNamedColors> colors;
// Verify input arguments.
if (argc != 2)
{
std::cout << "Usage: " << argv[0] << " InputImageFilename e.g. Gourds2.jpg"
<< std::endl;
return EXIT_FAILURE;
}
// Read the image.
vtkNew<vtkImageReader2Factory> readerFactory;
vtkSmartPointer<vtkImageReader2> imgReader;
imgReader.TakeReference(readerFactory->CreateImageReader2(argv[1]));
imgReader->SetFileName(argv[1]);
imgReader->Update();
vtkImageData* image = imgReader->GetOutput();
// Find center of image.
int center[2];
center[0] = (image->GetExtent()[1] + image->GetExtent()[0]) / 2;
center[1] = (image->GetExtent()[3] + image->GetExtent()[2]) / 2;
// Pick a radius for the circle.
int radius;
radius = (image->GetExtent()[1] < image->GetExtent()[3])
? image->GetExtent()[1] * 2 / 5
: image->GetExtent()[3] * 2 / 5;
std::array<double, 3> drawColor1{0, 0, 0};
auto color1 = colors->GetColor3ub("Black").GetData();
std::array<double, 3> drawColor2{0, 0, 0};
auto color2 = colors->GetColor3ub("Seashell").GetData();
for (auto i = 0; i < 3; ++i)
{
drawColor1[i] = color1[i];
drawColor2[i] = color2[i];
}
// Draw a circle in the center of the image.
vtkNew<vtkImageCanvasSource2D> drawing;
drawing->SetNumberOfScalarComponents(3);
drawing->SetScalarTypeToUnsignedChar();
drawing->SetExtent(image->GetExtent());
drawing->SetDrawColor(drawColor1.data());
drawing->FillBox(image->GetExtent()[0], image->GetExtent()[1],
image->GetExtent()[2], image->GetExtent()[3]);
drawing->SetDrawColor(drawColor2.data());
drawing->DrawCircle(center[0], center[1], radius);
// Combine the images (blend takes multiple connections on the 0th
// input port).
vtkNew<vtkImageBlend> blend;
blend->AddInputConnection(imgReader->GetOutputPort());
blend->AddInputConnection(drawing->GetOutputPort());
blend->SetOpacity(0, .6);
blend->SetOpacity(1, .4);
// Display the result.
vtkNew<vtkRenderWindowInteractor> renderWindowInteractor;
vtkNew<vtkImageViewer2> imageViewer;
imageViewer->SetInputConnection(blend->GetOutputPort());
imageViewer->SetSize(640, 512);
imageViewer->SetupInteractor(renderWindowInteractor);
imageViewer->GetRenderer()->ResetCamera();
imageViewer->GetRenderer()->SetBackground(
colors->GetColor3d("LightSlateGray").GetData());
imageViewer->GetRenderWindow()->SetWindowName("DrawOnAnImage");
imageViewer->GetRenderWindow()->Render();
renderWindowInteractor->Initialize();
renderWindowInteractor->Start();
return EXIT_SUCCESS;
}
CMakeLists.txt¶
cmake_minimum_required(VERSION 3.12 FATAL_ERROR)
project(DrawOnAnImage)
find_package(VTK COMPONENTS
CommonColor
CommonCore
CommonDataModel
IOImage
ImagingCore
ImagingSources
InteractionImage
InteractionStyle
RenderingContextOpenGL2
RenderingCore
RenderingFreeType
RenderingGL2PSOpenGL2
RenderingOpenGL2
)
if (NOT VTK_FOUND)
message(FATAL_ERROR "DrawOnAnImage: Unable to find the VTK build folder.")
endif()
# Prevent a "command line is too long" failure in Windows.
set(CMAKE_NINJA_FORCE_RESPONSE_FILE "ON" CACHE BOOL "Force Ninja to use response files.")
add_executable(DrawOnAnImage MACOSX_BUNDLE DrawOnAnImage.cxx )
target_link_libraries(DrawOnAnImage PRIVATE ${VTK_LIBRARIES}
)
# vtk_module_autoinit is needed
vtk_module_autoinit(
TARGETS DrawOnAnImage
MODULES ${VTK_LIBRARIES}
)
Download and Build DrawOnAnImage¶
Click here to download DrawOnAnImage and its CMakeLists.txt file. Once the tarball DrawOnAnImage.tar has been downloaded and extracted,
cd DrawOnAnImage/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:
./DrawOnAnImage
WINDOWS USERS
Be sure to add the VTK bin directory to your path. This will resolve the VTK dll's at run time.