Physically Based Rendering (PBR) property for vtkPolyDataMapper

pbr.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import vtk


def sphere(center, radius=1.0, color='Gray', metallic=0.5, roughness=1.0):
    colors = vtk.vtkNamedColors()
    cx, cy, cz = center

    # Create a sphere
    sphereSource = vtk.vtkSphereSource()
    sphereSource.SetCenter(cx, cy, cz)
    sphereSource.SetRadius(radius)
    # Make the surface smooth.
    sphereSource.SetPhiResolution(100)
    sphereSource.SetThetaResolution(100)

    mapper = vtk.vtkPolyDataMapper()
    mapper.SetInputConnection(sphereSource.GetOutputPort())

    actor = vtk.vtkActor()
    actor.SetMapper(mapper)
    actor.GetProperty().SetInterpolationToPBR()
    actor.GetProperty().SetMetallic(metallic)
    actor.GetProperty().SetRoughness(roughness)
    actor.GetProperty().SetColor(colors.GetColor3d(color))
    return actor


def main():
    renderer = vtk.vtkRenderer()
    renderWindow = vtk.vtkRenderWindow()
    renderWindow.SetWindowName("Sphere")
    renderWindow.AddRenderer(renderer)
    renderWindowInteractor = vtk.vtkRenderWindowInteractor()
    renderWindowInteractor.SetRenderWindow(renderWindow)

    colors = ['Red', 'Blue', 'Green', 'Silver', 'Gold']

    for i in range(8):
        for j, color in zip(range(5), colors):
            s = sphere(
                center=[i, j, 0],
                radius=0.5,
                color=color,
                metallic=j/4.,
                roughness=i/7.
            )
            renderer.AddActor(s)
    renderer.SetBackground([0.5, 0.5, 0.5])

    renderWindow.Render()
    renderWindowInteractor.Start()


if __name__ == '__main__':
    main()