ImportPolyDataScene
Repository source: ImportPolyDataScene
Other languages
See (PythonicAPI)
Question
If you have a question about this example, please use the VTK Discourse Forum
Code¶
ImportPolyDataScene.cxx
#include <vtkActor.h>
#include <vtkCamera.h>
#include <vtkCompositeDataSet.h>
#include <vtkDataObjectTreeIterator.h>
#include <vtkFieldData.h>
#include <vtkMultiBlockDataSet.h>
#include <vtkNamedColors.h>
#include <vtkNew.h>
#include <vtkPolyData.h>
#include <vtkPolyDataMapper.h>
#include <vtkProperty.h>
#include <vtkRenderWindow.h>
#include <vtkRenderWindowInteractor.h>
#include <vtkRenderer.h>
#include <vtkVersion.h>
#include <vtkXMLMultiBlockDataReader.h>
#if VTK_VERSION_NUMBER >= 89000000000ULL
#define VTK890 1
#include <vtkDataObjectTreeRange.h>
#endif
namespace {
void ImportMultiBlockScene(vtkRenderer* renderer, std::string const& fileName);
void RestoreCameraFromFieldData(std::string const&, vtkCamera*, vtkPolyData*);
void RestorePropertyFromFieldData(std::string const&, vtkProperty*,
vtkPolyData*);
void RestoreActorFromFieldData(std::string const&, vtkActor*, vtkPolyData*);
// These functions need to be written.
// void RestoreMapperFromFieldData(std::string const&, vtkPolyDataMapper*,
// vtkPolyData*);
// void RestoreLookupTableFromFieldData(std::string const&, vtkScalarsToColors*,
// vtkPolyData*);
} // namespace
int main(int argc, char* argv[])
{
if (argc < 2)
{
std::cout << "Usage: " << argv[0] << " file.vtp e.g. ExportBunny.vtp"
<< std::endl;
return EXIT_FAILURE;
}
// Visualization
vtkNew<vtkNamedColors> colors;
vtkNew<vtkRenderer> renderer;
vtkNew<vtkRenderWindow> renderWindow;
renderWindow->AddRenderer(renderer);
renderWindow->SetWindowName("ImportPolyDataScene");
vtkNew<vtkRenderWindowInteractor> renderWindowInteractor;
renderWindowInteractor->SetRenderWindow(renderWindow);
ImportMultiBlockScene(renderer.GetPointer(), std::string(argv[1]));
renderWindow->Render();
renderer->SetBackground(colors->GetColor3d("Silver").GetData());
renderWindow->SetSize(640, 480);
renderWindow->Render();
renderWindowInteractor->Start();
return EXIT_SUCCESS;
}
namespace {
void ImportMultiBlockScene(vtkRenderer* renderer, std::string const& fileName)
{
vtkCamera* camera = renderer->GetActiveCamera();
// Read the multiblock data
// auto reader = vtkSmartPointer<vtkXMLMultiBlockDataReader>::New();
vtkNew<vtkXMLMultiBlockDataReader> reader;
reader->SetFileName(fileName.c_str());
reader->Update();
std::cout << "Importing "
<< dynamic_cast<vtkMultiBlockDataSet*>(reader->GetOutput())
->GetNumberOfBlocks()
<< " actors" << std::endl;
#if VTK890
vtkDataObjectTree* input =
dynamic_cast<vtkDataObjectTree*>(reader->GetOutput());
using Opts = vtk::DataObjectTreeOptions;
for (vtkDataObject* dso :
vtk::Range(input, Opts::SkipEmptyNodes | Opts::VisitOnlyLeaves))
{
vtkPolyData* pd = dynamic_cast<vtkPolyData*>(dso);
RestoreCameraFromFieldData("Camera", camera, pd);
vtkNew<vtkPolyDataMapper> mapper;
mapper->SetInputData(pd);
vtkNew<vtkActor> actor;
actor->SetMapper(mapper);
RestorePropertyFromFieldData("Property", actor->GetProperty(), pd);
vtkNew<vtkProperty> backProperty;
actor->SetBackfaceProperty(backProperty);
RestorePropertyFromFieldData("BackfaceProperty",
actor->GetBackfaceProperty(), pd);
RestoreActorFromFieldData("Actor", actor, pd);
renderer->AddActor(actor);
}
#else
vtkCompositeDataSet* input =
dynamic_cast<vtkCompositeDataSet*>(reader->GetOutput());
vtkNew<vtkDataObjectTreeIterator> iter;
iter->SetDataSet(input);
iter->SkipEmptyNodesOn();
iter->VisitOnlyLeavesOn();
for (iter->InitTraversal(); !iter->IsDoneWithTraversal();
iter->GoToNextItem())
{
vtkDataObject* dso = iter->GetCurrentDataObject();
vtkPolyData* pd = dynamic_cast<vtkPolyData*>(dso);
RestoreCameraFromFieldData("Camera", camera, pd);
vtkNew<vtkPolyDataMapper> mapper;
mapper->SetInputData(pd);
vtkNew<vtkActor> actor;
actor->SetMapper(mapper);
RestorePropertyFromFieldData("Property", actor->GetProperty(), pd);
vtkNew<vtkProperty> backProperty;
actor->SetBackfaceProperty(backProperty);
RestorePropertyFromFieldData("BackfaceProperty",
actor->GetBackfaceProperty(), pd);
RestoreActorFromFieldData("Actor", actor, pd);
renderer->AddActor(actor);
}
#endif
}
void RestoreCameraFromFieldData(std::string const& arrayPrefix,
vtkCamera* camera, vtkPolyData* pd)
{
vtkFieldData* fd = pd->GetFieldData();
camera->SetFocalPoint(
fd->GetArray(std::string(arrayPrefix + ":" + "FocalPoint").c_str())
->GetTuple(0));
camera->SetPosition(
fd->GetArray(std::string(arrayPrefix + ":" + "Position").c_str())
->GetTuple(0));
camera->SetViewUp(
fd->GetArray(std::string(arrayPrefix + ":" + "ViewUp").c_str())
->GetTuple(0));
camera->SetClippingRange(
fd->GetArray(std::string(arrayPrefix + ":" + "ClippingRange").c_str())
->GetTuple(0));
camera->SetViewAngle(
fd->GetArray(std::string(arrayPrefix + ":" + "ViewAngle").c_str())
->GetTuple1(0));
}
void RestorePropertyFromFieldData(std::string const& arrayPrefix,
vtkProperty* property, vtkPolyData* pd)
{
if (property)
{
vtkFieldData* fd = pd->GetFieldData();
property->SetAmbient(
fd->GetArray(std::string(arrayPrefix + ":" + "Ambient").c_str())
->GetTuple1(0));
property->SetAmbientColor(
fd->GetArray(std::string(arrayPrefix + ":" + "AmbientColor").c_str())
->GetTuple(0));
property->SetDiffuse(
fd->GetArray(std::string(arrayPrefix + ":" + "Diffuse").c_str())
->GetTuple1(0));
property->SetDiffuseColor(
fd->GetArray(std::string(arrayPrefix + ":" + "DiffuseColor").c_str())
->GetTuple(0));
property->SetSpecular(
fd->GetArray(std::string(arrayPrefix + ":" + "Specular").c_str())
->GetTuple1(0));
property->SetSpecularColor(
fd->GetArray(std::string(arrayPrefix + ":" + "SpecularColor").c_str())
->GetTuple(0));
property->SetSpecularPower(
fd->GetArray(std::string(arrayPrefix + ":" + "SpecularPower").c_str())
->GetTuple1(0));
property->SetEdgeColor(
fd->GetArray(std::string(arrayPrefix + ":" + "EdgeColor").c_str())
->GetTuple(0));
property->SetEdgeVisibility(
fd->GetArray(std::string(arrayPrefix + ":" + "EdgeVisibility").c_str())
->GetTuple1(0));
property->SetVertexColor(
fd->GetArray(std::string(arrayPrefix + ":" + "VertexColor").c_str())
->GetTuple(0));
property->SetVertexVisibility(
fd->GetArray(
std::string(arrayPrefix + ":" + "VertexVisibility").c_str())
->GetTuple1(0));
property->SetInterpolation(
fd->GetArray(std::string(arrayPrefix + ":" + "Interpolation").c_str())
->GetTuple1(0));
property->SetOpacity(
fd->GetArray(std::string(arrayPrefix + ":" + "Opacity").c_str())
->GetTuple1(0));
property->SetRepresentation(
fd->GetArray(std::string(arrayPrefix + ":" + "Representation").c_str())
->GetTuple1(0));
property->SetBackfaceCulling(
fd->GetArray(std::string(arrayPrefix + ":" + "BackfaceCulling").c_str())
->GetTuple1(0));
property->SetFrontfaceCulling(
fd->GetArray(
std::string(arrayPrefix + ":" + "FrontfaceCulling").c_str())
->GetTuple1(0));
property->SetPointSize(
fd->GetArray(std::string(arrayPrefix + ":" + "PointSize").c_str())
->GetTuple1(0));
property->SetLineWidth(
fd->GetArray(std::string(arrayPrefix + ":" + "LineWidth").c_str())
->GetTuple1(0));
property->SetLineStipplePattern(
fd->GetArray(
std::string(arrayPrefix + ":" + "LineStipplePattern").c_str())
->GetTuple1(0));
property->SetLineStippleRepeatFactor(
fd->GetArray(std::string(arrayPrefix + ":" + "LineStippleRepeatFactor")
.c_str())
->GetTuple1(0));
property->SetLighting(
fd->GetArray(std::string(arrayPrefix + ":" + "Lighting").c_str())
->GetTuple1(0) == 1.0
? true
: false);
property->SetRenderPointsAsSpheres(
fd->GetArray(
std::string(arrayPrefix + ":" + "RenderPointsAsSpheres").c_str())
->GetTuple1(0) == 1.0
? true
: false);
property->SetRenderLinesAsTubes(
fd->GetArray(
std::string(arrayPrefix + ":" + "RenderLinesAsTubes").c_str())
->GetTuple1(0) == 1.0
? true
: false);
property->SetShading(
fd->GetArray(std::string(arrayPrefix + ":" + "Shading").c_str())
->GetTuple1(0) == 1.0
? true
: false);
}
}
void RestoreActorFromFieldData(std::string const& arrayPrefix, vtkActor* actor,
vtkPolyData* pd)
{
vtkFieldData* fd = pd->GetFieldData();
actor->SetDragable(
fd->GetArray(std::string(arrayPrefix + ":" + "Dragable").c_str())
->GetTuple1(0) == 1.0
? true
: false);
actor->SetPickable(
fd->GetArray(std::string(arrayPrefix + ":" + "Pickable").c_str())
->GetTuple1(0) == 1.0
? true
: false);
actor->SetVisibility(
fd->GetArray(std::string(arrayPrefix + ":" + "Visibility").c_str())
->GetTuple1(0));
actor->SetPosition(
fd->GetArray(std::string(arrayPrefix + ":" + "Position").c_str())
->GetTuple(0));
actor->SetOrientation(
fd->GetArray(std::string(arrayPrefix + ":" + "Orientation").c_str())
->GetTuple(0));
actor->SetOrigin(
fd->GetArray(std::string(arrayPrefix + ":" + "Origin").c_str())
->GetTuple(0));
actor->SetScale(fd->GetArray(std::string(arrayPrefix + ":" + "Scale").c_str())
->GetTuple(0));
actor->SetForceOpaque(
fd->GetArray(std::string(arrayPrefix + ":" + "ForceOpaque").c_str())
->GetTuple1(0) == 1.0
? true
: false);
actor->SetForceTranslucent(
fd->GetArray(std::string(arrayPrefix + ":" + "ForceTranslucent").c_str())
->GetTuple1(0) == 1.0
? true
: false);
}
} // namespace
CMakeLists.txt¶
cmake_minimum_required(VERSION 3.12 FATAL_ERROR)
project(ImportPolyDataScene)
find_package(VTK COMPONENTS
CommonColor
CommonCore
CommonDataModel
IOXML
InteractionStyle
RenderingContextOpenGL2
RenderingCore
RenderingFreeType
RenderingGL2PSOpenGL2
RenderingOpenGL2
)
if (NOT VTK_FOUND)
message(FATAL_ERROR "ImportPolyDataScene: 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(ImportPolyDataScene MACOSX_BUNDLE ImportPolyDataScene.cxx )
target_link_libraries(ImportPolyDataScene PRIVATE ${VTK_LIBRARIES}
)
# vtk_module_autoinit is needed
vtk_module_autoinit(
TARGETS ImportPolyDataScene
MODULES ${VTK_LIBRARIES}
)
Download and Build ImportPolyDataScene¶
Click here to download ImportPolyDataScene and its CMakeLists.txt file. Once the tarball ImportPolyDataScene.tar has been downloaded and extracted,
cd ImportPolyDataScene/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:
./ImportPolyDataScene
WINDOWS USERS
Be sure to add the VTK bin directory to your path. This will resolve the VTK dll's at run time.