[VTK] MFC에서 VTK ScalrBar 사용하기

이번 포스팅에서는 VTK Window에 ScalarBar를 띄우는 예제를 소개하고자 합니다.

아래 링크는 VTK에서 제공하는 공식 예제 사이트입니다.

링크: https://lorensen.github.io/VTKExamples/site/Cxx

동작은 아래와 같습니다.

<MFC ScalarBar 예제>

먼저 예제는 아래 링크에서 확인할 수 있습니다.

링크: https://lorensen.github.io/VTKExamples/site/Cxx/Visualization/ScalarBarActorColorSeries/

ScalarBar는 Point Cloud Data의 Z축 높이 범위에 따라 색상을 입힐 때, 높이의 범위를 보여주기 위해 사용됩니다.

자, 본격적으로 예제 만들기를 실습해 보겠습니다.
(핵심 부분만 설명합니다. 자세한 내용은 본문 하단의 예제 코드를 참고해주세요.)

먼저 Dialog 기반으로 프로젝트를 생성합니다.

Dialog 기반 프로젝트 생성하기.

그리고 MFC에서 VTK 사용을 위해 아래 링크를 참고하여 설정합니다.

[VTK] Visual Studio 2015 MFC에서 VTK 사용 준비하기

1. VTK 초기화

Main Dialog header 파일에 아래의 초기화 코드를 추가합니다.

#pragma once
#include <vtkAutoInit.h>

#define vtkRenderingCore_AUTOINIT 3(vtkRenderingOpenGL2,vtkInteractionStyle, vtkRenderingFreeType)
#define vtkRenderingContext2D_AUTOINIT 1(vtkRenderingContextOpenGL2)

#include <vtkActor.h>
#include <vtkFloatArray.h>
#include <vtkLookupTable.h>
#include <vtkPointData.h>
#include <vtkPolyData.h>
#include <vtkPolyDataMapper.h>
#include <vtkProperty.h>
#include <vtkRenderer.h>
#include <vtkRenderWindow.h>
#include <vtkRenderWindowInteractor.h>
#include <vtkInteractorStyleTrackballCamera.h>
#include <vtkScalarBarActor.h>
#include <vtkSmartPointer.h>
#include <vtkSphereSource.h>
#include <vtkColorSeries.h>
#include <vtkNamedColors.h>

2. Picture Control에 그려질 vtkRenderWindow를 초기화합니다.

<Header File>

public :
	vtkNew<vtkRenderWindow> m_vtkRenderWindow;

	void InitializeVTKWindow(void* hWnd);
	void ResizeVTKWindow();

<Source File>

void CExampleBarChartDlg::InitializeVTKWindow(void* hWnd)
{
	vtkNewGetClientRect(rc);
	m_vtkRenderWindow->SetSize(rc.Width(), rc.Height());
}

3. Button Event에 ScalarBar data를 불러오는 코드를 정의합니다.

void CExampleScalarBarDlg::OnBnClickedBtnInitScalar()
{
	vtkNew<vtkNamedColors> colors;

	// Create a sphere fora some geometry
	vtkNew<vtkSphereSource> sphere;
	sphere->SetCenter(0, 0, 0);
	sphere->SetRadius(1);
	sphere->SetPhiResolution(30);
	sphere->SetThetaResolution(60);
	sphere->Update();

	// Create scalar data to associate with the vertices of the sphere
	int numPts = sphere->GetOutput()->GetPoints()->GetNumberOfPoints();
	vtkNew<vtkFloatArray> scalars;
	scalars->SetNumberOfValues(numPts);
	for (int i = 0; i < numPts; ++i)
	{
		scalars->SetValue(i, static_cast<float>(i) / numPts);
	}
	vtkNew<vtkPolyData> poly;
	poly->DeepCopy(sphere->GetOutput());
	poly->GetPointData()->SetScalars(scalars);

	vtkNew<vtkPolyDataMapper> mapper;
	mapper->SetInputData(poly);
	mapper->ScalarVisibilityOn();
	mapper->SetScalarModeToUsePointData();
	mapper->SetColorModeToMapScalars();

	vtkNew<vtkActor> actor;
	actor->SetMapper(mapper);
	actor->GetProperty()->SetInterpolationToFlat();

	vtkNew<vtkScalarBarActor> scalarBar;
	scalarBar->SetLookupTable(mapper->GetLookupTable());
	scalarBar->SetTitle("Beachball");
	scalarBar->SetNumberOfLabels(4);

	// Create a lookup table to share between the mapper and the scalarbar
	vtkNew<vtkLookupTable> seriesLut;

	vtkNew<vtkColorSeries> series;
	int seriesEnum = series->BREWER_QUALITATIVE_SET3;
	series->SetColorScheme(seriesEnum);
	series->BuildLookupTable(seriesLut, series->ORDINAL);

	mapper->SetLookupTable(seriesLut);
	scalarBar->SetLookupTable(seriesLut);

	// Create an interactor
	vtkNew<vtkRenderWindowInteractor> renderWindowInteractor;
	renderWindowInteractor->SetRenderWindow(m_vtkRenderWindow);

	vtkNew<vtkRenderer> renderer;
	renderer->AddActor(actor);
	renderer->AddActor2D(scalarBar);
	renderer->GradientBackgroundOn();
	renderer->SetBackground(.1, .2, .3);
	renderer->ResetCamera();

	m_vtkRenderWindow->AddRenderer(renderer);
	m_vtkRenderWindow->Render();

	renderer->ResetCamera();
	m_vtkRenderWindow->Render();
}

기본 C++ 예제 코드를 MFC Dialog 프로젝트로 변경한 것 밖에 없으니 간단하죠?

자세한 내용은 첨부된 코드를 참고해주세요.

ExampleScalarBar.zip