#!/usr/bin/env python
|
# -*- coding: utf-8 -*-
|
#
|
# by Panos Mavrogiorgos, email: pmav99 <> gmail
|
|
from vtk import *
|
|
# The source file
|
#file_name = "uGridEx.vtk"
|
file_name="timestep.vtu"
|
# Read the source file.
|
#reader = vtkUnstructuredGridReader()
|
reader=vtkXMLUnstructuredGridReader()
|
reader.SetFileName(file_name)
|
reader.Update() # Needed because of GetScalarRange
|
output = reader.GetOutput()
|
scalar_range = output.GetScalarRange()
|
|
# Create the mapper that corresponds the objects of the vtk file
|
# into graphics elements
|
mapper = vtkDataSetMapper()
|
mapper.SetInput(output)
|
mapper.SetScalarRange(scalar_range)
|
|
# Create the Actor
|
actor = vtkActor()
|
actor.SetMapper(mapper)
|
|
# Create the Renderer
|
renderer = vtkRenderer()
|
renderer.AddActor(actor)
|
renderer.SetBackground(0, 0, 0) # Set background to white
|
|
# Create the RendererWindow
|
renderer_window = vtkRenderWindow()
|
renderer_window.AddRenderer(renderer)
|
|
# Create the RendererWindowInteractor and display the vtk_file
|
interactor = vtkRenderWindowInteractor()
|
interactor.SetRenderWindow(renderer_window)
|
interactor.Initialize()
|
interactor.Start()
|