mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-09-08 13:23:51 +00:00
Originally, most of the sources used constructor parameters to set the various options of the source. Although convenient, it was difficult to keep track of what each parameter meant. To make the code more clear, source parameters are now set with accessor functions (e.g. `SetPointDimensions`). Although this makes code more verbose, it helps prevent mistakes and makes the changes more resilient to future changes.
95 lines
3.6 KiB
C++
95 lines
3.6 KiB
C++
//============================================================================
|
|
// Copyright (c) Kitware, Inc.
|
|
// All rights reserved.
|
|
// See LICENSE.txt for details.
|
|
//
|
|
// This software is distributed WITHOUT ANY WARRANTY; without even
|
|
// the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
|
|
// PURPOSE. See the above copyright notice for more information.
|
|
//============================================================================
|
|
|
|
#include <vtkm/cont/Initialize.h>
|
|
#include <vtkm/source/Tangle.h>
|
|
|
|
#include <vtkm/rendering/Actor.h>
|
|
#include <vtkm/rendering/CanvasRayTracer.h>
|
|
#include <vtkm/rendering/MapperRayTracer.h>
|
|
#include <vtkm/rendering/MapperVolume.h>
|
|
#include <vtkm/rendering/MapperWireframer.h>
|
|
#include <vtkm/rendering/Scene.h>
|
|
#include <vtkm/rendering/View3D.h>
|
|
|
|
#include <vtkm/filter/contour/Contour.h>
|
|
|
|
// This example creates a simple data set and uses VTK-m's rendering engine to render an image and
|
|
// write that image to a file. It then computes an isosurface on the input data set and renders
|
|
// this output data set in a separate image file
|
|
|
|
using vtkm::rendering::CanvasRayTracer;
|
|
using vtkm::rendering::MapperRayTracer;
|
|
using vtkm::rendering::MapperVolume;
|
|
using vtkm::rendering::MapperWireframer;
|
|
|
|
int main(int argc, char* argv[])
|
|
{
|
|
vtkm::cont::Initialize(argc, argv, vtkm::cont::InitializeOptions::Strict);
|
|
|
|
vtkm::source::Tangle tangle;
|
|
tangle.SetPointDimensions({ 50, 50, 50 });
|
|
vtkm::cont::DataSet tangleData = tangle.Execute();
|
|
std::string fieldName = "tangle";
|
|
|
|
// Set up a camera for rendering the input data
|
|
vtkm::rendering::Camera camera;
|
|
camera.SetLookAt(vtkm::Vec3f_32(0.5, 0.5, 0.5));
|
|
camera.SetViewUp(vtkm::make_Vec(0.f, 1.f, 0.f));
|
|
camera.SetClippingRange(1.f, 10.f);
|
|
camera.SetFieldOfView(60.f);
|
|
camera.SetPosition(vtkm::Vec3f_32(1.5, 1.5, 1.5));
|
|
vtkm::cont::ColorTable colorTable("inferno");
|
|
|
|
// Background color:
|
|
vtkm::rendering::Color bg(0.2f, 0.2f, 0.2f, 1.0f);
|
|
vtkm::rendering::Actor actor(tangleData.GetCellSet(),
|
|
tangleData.GetCoordinateSystem(),
|
|
tangleData.GetField(fieldName),
|
|
colorTable);
|
|
vtkm::rendering::Scene scene;
|
|
scene.AddActor(actor);
|
|
// 2048x2048 pixels in the canvas:
|
|
CanvasRayTracer canvas(2048, 2048);
|
|
// Create a view and use it to render the input data using OS Mesa
|
|
|
|
vtkm::rendering::View3D view(scene, MapperVolume(), canvas, camera, bg);
|
|
view.Paint();
|
|
view.SaveAs("volume.png");
|
|
|
|
// Compute an isosurface:
|
|
vtkm::filter::contour::Contour filter;
|
|
// [min, max] of the tangle field is [-0.887, 24.46]:
|
|
filter.SetIsoValue(3.0);
|
|
filter.SetActiveField(fieldName);
|
|
vtkm::cont::DataSet isoData = filter.Execute(tangleData);
|
|
// Render a separate image with the output isosurface
|
|
vtkm::rendering::Actor isoActor(
|
|
isoData.GetCellSet(), isoData.GetCoordinateSystem(), isoData.GetField(fieldName), colorTable);
|
|
// By default, the actor will automatically scale the scalar range of the color table to match
|
|
// that of the data. However, we are coloring by the scalar that we just extracted a contour
|
|
// from, so we want the scalar range to match that of the previous image.
|
|
isoActor.SetScalarRange(actor.GetScalarRange());
|
|
vtkm::rendering::Scene isoScene;
|
|
isoScene.AddActor(isoActor);
|
|
|
|
// Wireframe surface:
|
|
vtkm::rendering::View3D isoView(isoScene, MapperWireframer(), canvas, camera, bg);
|
|
isoView.Paint();
|
|
isoView.SaveAs("isosurface_wireframer.png");
|
|
|
|
// Smooth surface:
|
|
vtkm::rendering::View3D solidView(isoScene, MapperRayTracer(), canvas, camera, bg);
|
|
solidView.Paint();
|
|
solidView.SaveAs("isosurface_raytracer.png");
|
|
|
|
return 0;
|
|
}
|