Skip to content


# Spring

VTKEx/Python/Modelling/Spring

Other languages

See (Cxx), (Java)

Question

If you have a simple question about this example contact us at VTKExProject If your question is more complex and may require extended discussion, please use the VTK Discourse Forum

### Code

Spring.py

    #!/usr/bin/env python

    import vtk

    def main():
        colors = vtk.vtkNamedColors()

        # Create the RenderWindow, Renderer and Interactor.
        #
        renderer = vtk.vtkRenderer()
        renWin = vtk.vtkRenderWindow()
        renWin.AddRenderer(renderer)

        iren = vtk.vtkRenderWindowInteractor()
        iren.SetRenderWindow(renWin)

        # Create the spring profile (a circle).
        #
        points = vtk.vtkPoints()
        points.InsertPoint(0, 1.0, 0.0, 0.0)
        points.InsertPoint(1, 1.0732, 0.0, -0.1768)
        points.InsertPoint(2, 1.25, 0.0, -0.25)
        points.InsertPoint(3, 1.4268, 0.0, -0.1768)
        points.InsertPoint(4, 1.5, 0.0, 0.00)
        points.InsertPoint(5, 1.4268, 0.0, 0.1768)
        points.InsertPoint(6, 1.25, 0.0, 0.25)
        points.InsertPoint(7, 1.0732, 0.0, 0.1768)

        poly = vtk.vtkCellArray()
        poly.InsertNextCell(8)  # The number of points.
        poly.InsertCellPoint(0)
        poly.InsertCellPoint(1)
        poly.InsertCellPoint(2)
        poly.InsertCellPoint(3)
        poly.InsertCellPoint(4)
        poly.InsertCellPoint(5)
        poly.InsertCellPoint(6)
        poly.InsertCellPoint(7)

        profile = vtk.vtkPolyData()
        profile.SetPoints(points)
        profile.SetPolys(poly)

        # Extrude the profile to make a spring.
        #
        extrude = vtk.vtkRotationalExtrusionFilter()
        extrude.SetInputData(profile)
        extrude.SetResolution(360)
        extrude.SetTranslation(6)
        extrude.SetDeltaRadius(1.0)
        extrude.SetAngle(2160.0)  # six revolutions

        normals = vtk.vtkPolyDataNormals()
        normals.SetInputConnection(extrude.GetOutputPort())
        normals.SetFeatureAngle(60)

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

        spring = vtk.vtkActor()
        spring.SetMapper(mapper)
        spring.GetProperty().SetColor(colors.GetColor3d("PowderBlue"))
        spring.GetProperty().SetDiffuse(0.7)
        spring.GetProperty().SetSpecular(0.4)
        spring.GetProperty().SetSpecularPower(20)
        spring.GetProperty().BackfaceCullingOn()

        # Add the actors to the renderer, set the background and size.
        #
        renderer.AddActor(spring)
        renderer.SetBackground(colors.GetColor3d("Burlywood"))
        renWin.SetSize(640, 512)

        renderer.ResetCamera()
        renderer.GetActiveCamera().Azimuth(90)

        # Render the image.
        #
        renWin.Render()
        iren.Start()

    if __name__ == '__main__':
        main()