def main():
if not vtk_version_ok(8, 90, 0):
print('You need VTK version 8.90 or greater to run this program.')
return
path, surface = get_program_parameters()
cube_path = Path(path)
if not cube_path.is_dir():
print('This path does not exist:', cube_path)
return
surface = surface.lower()
available_surfaces = {'boy', 'mobius', 'randomhills', 'torus', 'sphere', 'cube'}
if surface not in available_surfaces:
surface = 'boy'
if surface == 'mobius':
source = GetMobius()
elif surface == 'randomhills':
source = GetRandomHills()
elif surface == 'torus':
source = GetTorus()
elif surface == 'sphere':
source = GetSphere()
elif surface == 'cube':
source = GetCube()
else:
source = GetBoy()
# Load the cube map
# cubemap = ReadCubeMap(cube_path, '/', '.jpg', 0)
cubemap = ReadCubeMap(cube_path, '/', '.jpg', 1)
# cubemap = ReadCubeMap(cube_path/'skybox','', '.jpg', 2)
# Load the skybox
# Read it again as there is no deep copy for vtkTexture
# skybox = ReadCubeMap(cube_path, '/', '.jpg', 0)
skybox = ReadCubeMap(cube_path, '/', '.jpg', 1)
# skybox = ReadCubeMap(cube_path, '/skybox', '.jpg', 2)
skybox.InterpolateOn()
skybox.MipmapOn()
skybox.RepeatOff()
skybox.EdgeClampOn()
colors = vtk.vtkNamedColors()
# Set the background color.
colors.SetColor('BkgColor', [26, 51, 102, 255])
renderer = vtk.vtkOpenGLRenderer()
renderWindow = vtk.vtkRenderWindow()
renderWindow.AddRenderer(renderer)
interactor = vtk.vtkRenderWindowInteractor()
interactor.SetRenderWindow(renderWindow)
# Lets use a smooth metallic surface
metallicCoefficient = 1.0
roughnessCoefficient = 0.05
slwP = SliderProperties()
slwP.initialValue = metallicCoefficient
slwP.title = 'Metallicity'
sliderWidgetMetallic = MakeSliderWidget(slwP)
sliderWidgetMetallic.SetInteractor(interactor)
sliderWidgetMetallic.SetAnimationModeToAnimate()
sliderWidgetMetallic.EnabledOn()
slwP.initialValue = roughnessCoefficient
slwP.title = 'Roughness'
slwP.p1 = [0.1, 0.9]
slwP.p2 = [0.9, 0.9]
sliderWidgetRoughness = MakeSliderWidget(slwP)
sliderWidgetRoughness.SetInteractor(interactor)
sliderWidgetRoughness.SetAnimationModeToAnimate()
sliderWidgetRoughness.EnabledOn()
# Build the pipeline
mapper = vtk.vtkPolyDataMapper()
mapper.SetInputData(source)
actor = vtk.vtkActor()
actor.SetMapper(mapper)
renderer.UseImageBasedLightingOn()
if vtk_version_ok(9, 0, 0):
renderer.SetEnvironmentTexture(cubemap)
else:
renderer.SetEnvironmentCubeMap(cubemap)
actor.GetProperty().SetInterpolationToPBR()
# configure the basic properties
actor.GetProperty().SetColor(colors.GetColor3d('White'))
actor.GetProperty().SetMetallic(metallicCoefficient)
actor.GetProperty().SetRoughness(roughnessCoefficient)
# Create the slider callbacks to manipulate metallicity and roughness
sliderWidgetMetallic.AddObserver(vtk.vtkCommand.InteractionEvent, SliderCallbackMetallic(actor.GetProperty()))
sliderWidgetRoughness.AddObserver(vtk.vtkCommand.InteractionEvent, SliderCallbackRoughness(actor.GetProperty()))
renderer.SetBackground(colors.GetColor3d("BkgColor"))
renderer.AddActor(actor)
skyboxActor = vtk.vtkSkybox()
skyboxActor.SetTexture(skybox)
renderer.AddActor(skyboxActor)
renderer.UseSphericalHarmonicsOff()
renderWindow.SetSize(640, 480)
renderWindow.Render()
renderWindow.SetWindowName("Skybox-PBR")
axes = vtk.vtkAxesActor()
widget = vtk.vtkOrientationMarkerWidget()
rgba = [0.0, 0.0, 0.0, 0.0]
colors.GetColor("Carrot", rgba)
widget.SetOutlineColor(rgba[0], rgba[1], rgba[2])
widget.SetOrientationMarker(axes)
widget.SetInteractor(interactor)
widget.SetViewport(0.0, 0.2, 0.2, 0.4)
widget.SetEnabled(1)
widget.InteractiveOn()
interactor.SetRenderWindow(renderWindow)
renderWindow.Render()
interactor.Start()
def main():
if not vtk_version_ok(8, 90, 0):
print('You need VTK version 8.90 or greater to run this program.')
return
cube_path, material_fn, albedo_fn, normal_fn, emissive_fn, surface = get_program_parameters()
if not os.path.isdir(cube_path):
print('This path does not exist:', cube_path)
return
# Load the cube map
# cubemap = ReadCubeMap(cube_path, '/', '.jpg', 0)
cubemap = ReadCubeMap(cube_path, '/', '.jpg', 1)
# cubemap = ReadCubeMap(cube_path, '/skybox', '.jpg', 2)
# Load the skybox
# Read it again as there is no deep copy for vtkTexture
# skybox = ReadCubeMap(cube_path, '/', '.jpg', 0)
skybox = ReadCubeMap(cube_path, '/', '.jpg', 1)
# skybox = ReadCubeMap(cube_path, '/skybox', '.jpg', 2)
skybox.InterpolateOn()
skybox.RepeatOff()
skybox.EdgeClampOn()
# Get the textures
material = GetTexture(material_fn)
albedo = GetTexture(albedo_fn)
albedo.UseSRGBColorSpaceOn()
normal = GetTexture(normal_fn)
emissive = GetTexture(emissive_fn)
emissive.UseSRGBColorSpaceOn()
# Get the surface
surface = surface.lower()
available_surfaces = {'boy', 'mobius', 'randomhills', 'torus', 'sphere', 'cube'}
if surface not in available_surfaces:
surface = 'boy'
if surface == 'mobius':
source = GetMobius()
elif surface == 'randomhills':
source = GetRandomHills()
elif surface == 'torus':
source = GetTorus()
elif surface == 'sphere':
source = GetSphere()
elif surface == 'cube':
source = GetCube()
else:
source = GetBoy()
colors = vtk.vtkNamedColors()
# Set the background color.
colors.SetColor('BkgColor', [26, 51, 102, 255])
colors.SetColor('VTKBlue', [6, 79, 141, 255])
# Let's make a complementary colour to VTKBlue
colors.SetColor('VTKBlueComp', [249, 176, 114, 255])
renderer = vtk.vtkRenderer()
renderWindow = vtk.vtkRenderWindow()
renderWindow.AddRenderer(renderer)
interactor = vtk.vtkRenderWindowInteractor()
interactor.SetRenderWindow(renderWindow)
# Lets use a rough metallic surface
metallicCoefficient = 1.0
roughnessCoefficient = 0.8
# Other parameters
occlusionStrength = 10.0
normalScale = 10.0
emissiveCol = colors.GetColor3d('VTKBlueComp')
emissiveFactor = emissiveCol
# emissiveFactor = [1.0, 1.0, 1.0]
slwP = SliderProperties()
slwP.initialValue = metallicCoefficient
slwP.title = 'Metallicity'
sliderWidgetMetallic = MakeSliderWidget(slwP)
sliderWidgetMetallic.SetInteractor(interactor)
sliderWidgetMetallic.SetAnimationModeToAnimate()
sliderWidgetMetallic.EnabledOn()
slwP.initialValue = roughnessCoefficient
slwP.title = 'Roughness'
slwP.p1 = [0.2, 0.9]
slwP.p2 = [0.8, 0.9]
sliderWidgetRoughness = MakeSliderWidget(slwP)
sliderWidgetRoughness.SetInteractor(interactor)
sliderWidgetRoughness.SetAnimationModeToAnimate()
sliderWidgetRoughness.EnabledOn()
slwP.initialValue = occlusionStrength
slwP.title = 'Occlusion'
slwP.p1 = [0.1, 0.1]
slwP.p2 = [0.1, 0.9]
sliderWidgetOcclusionStrength = MakeSliderWidget(slwP)
sliderWidgetOcclusionStrength.SetInteractor(interactor)
sliderWidgetOcclusionStrength.SetAnimationModeToAnimate()
sliderWidgetOcclusionStrength.EnabledOn()
slwP.initialValue = normalScale
slwP.title = 'Normal'
slwP.p1 = [0.85, 0.1]
slwP.p2 = [0.85, 0.9]
sliderWidgetNormal = MakeSliderWidget(slwP)
sliderWidgetNormal.SetInteractor(interactor)
sliderWidgetNormal.SetAnimationModeToAnimate()
sliderWidgetNormal.EnabledOn()
# Build the pipeline
mapper = vtk.vtkPolyDataMapper()
mapper.SetInputData(source)
actor = vtk.vtkActor()
actor.SetMapper(mapper)
actor.GetProperty().SetInterpolationToPBR()
# configure the basic properties
actor.GetProperty().SetColor(colors.GetColor3d('White'))
actor.GetProperty().SetMetallic(metallicCoefficient)
actor.GetProperty().SetRoughness(roughnessCoefficient)
# configure textures (needs tcoords on the mesh)
actor.GetProperty().SetBaseColorTexture(albedo)
actor.GetProperty().SetORMTexture(material)
actor.GetProperty().SetOcclusionStrength(occlusionStrength)
actor.GetProperty().SetEmissiveTexture(emissive)
actor.GetProperty().SetEmissiveFactor(emissiveFactor)
# needs tcoords, normals and tangents on the mesh
actor.GetProperty().SetNormalTexture(normal)
actor.GetProperty().SetNormalScale(normalScale)
renderer.UseImageBasedLightingOn()
if vtk.VTK_VERSION_NUMBER >= 90000000000:
renderer.SetEnvironmentTexture(cubemap)
else:
renderer.SetEnvironmentCubeMap(cubemap)
renderer.SetBackground(colors.GetColor3d("BkgColor"))
renderer.AddActor(actor)
# Comment out if you don't want a skybox
skyboxActor = vtk.vtkSkybox()
skyboxActor.SetTexture(skybox)
renderer.AddActor(skyboxActor)
# Create the slider callbacks to manipulate metallicity, roughness
# occlusion strength and normal scaling
sliderWidgetMetallic.AddObserver(vtk.vtkCommand.InteractionEvent, SliderCallbackMetallic(actor.GetProperty()))
sliderWidgetRoughness.AddObserver(vtk.vtkCommand.InteractionEvent, SliderCallbackRoughness(actor.GetProperty()))
sliderWidgetOcclusionStrength.AddObserver(vtk.vtkCommand.InteractionEvent,
SliderCallbackOcclusionStrength(actor.GetProperty()))
sliderWidgetNormal.AddObserver(vtk.vtkCommand.InteractionEvent, SliderCallbackNormalScale(actor.GetProperty()))
renderWindow.SetSize(640, 480)
renderWindow.Render()
renderWindow.SetWindowName('PhysicallyBasedRendering')
axes = vtk.vtkAxesActor()
widget = vtk.vtkOrientationMarkerWidget()
rgba = [0.0, 0.0, 0.0, 0.0]
colors.GetColor("Carrot", rgba)
widget.SetOutlineColor(rgba[0], rgba[1], rgba[2])
widget.SetOrientationMarker(axes)
widget.SetInteractor(interactor)
widget.SetViewport(0.0, 0.0, 0.2, 0.2)
widget.SetEnabled(1)
widget.InteractiveOn()
interactor.SetRenderWindow(renderWindow)
renderWindow.Render()
interactor.Start()