org.simantics.g3d.vtk/src/org/simantics/g3d/vtk/shape/MeshActor.java

   1 package org.simantics.g3d.vtk.shape;\r
   2 \r
   3 import javax.vecmath.Vector3d;\r
   4 \r
   5 import org.simantics.g3d.shape.Color4d;\r
   6 import org.simantics.g3d.shape.Mesh;\r
   7 \r
   8 import vtk.vtkActor;\r
   9 import vtk.vtkDataSetMapper;\r
  10 import vtk.vtkIdList;\r
  11 import vtk.vtkPoints;\r
  12 import vtk.vtkPolyData;\r
  13 import vtk.vtkPolyDataMapper;\r
  14 import vtk.vtkPolyDataNormals;\r
  15 import vtk.vtkTriangle;\r
  16 import vtk.vtkUnsignedCharArray;\r
  17 \r
  18 public class MeshActor extends vtkActor {\r
  19         \r
  20         public void setMesh(Mesh mesh) {\r
  21                 \r
  22                 vtkPolyDataMapper mapper = new vtkPolyDataMapper();\r
  23                 \r
  24                 vtkPolyData polyData = new vtkPolyData();\r
  25                 polyData.Allocate(mesh.getIndices().size()/3, mesh.getIndices().size()/3);\r
  26                 \r
  27                 vtkTriangle triangle = new vtkTriangle();\r
  28         vtkIdList list = triangle.GetPointIds();\r
  29         for (int i = 0; i < mesh.getIndices().size(); i+=3) {\r
  30                 list.SetId(0, mesh.getIndices().get(i));\r
  31                 list.SetId(1, mesh.getIndices().get(i+1));\r
  32                 list.SetId(2, mesh.getIndices().get(i+2));\r
  33                 polyData.InsertNextCell(triangle.GetCellType(), list);\r
  34         }\r
  35         list.Delete();\r
  36         triangle.Delete();\r
  37                 \r
  38         \r
  39         vtkPoints points = new vtkPoints();\r
  40         for (int i = 0; i < mesh.getVertices().size(); i++) {\r
  41                 Vector3d p = mesh.getVertices().get(i);\r
  42                 points.InsertPoint(i, p.x, p.y, p.z);\r
  43         }\r
  44         \r
  45         polyData.SetPoints(points);\r
  46         points.Delete();\r
  47         \r
  48         if (mesh.getColors() != null) {\r
  49                 vtkUnsignedCharArray colors = new vtkUnsignedCharArray();\r
  50                 colors.SetName("Colors");\r
  51                 colors.SetNumberOfComponents(3);\r
  52                 colors.SetNumberOfTuples(mesh.getColors().size());\r
  53                 for (int i = 0; i < mesh.getColors().size(); i++) {\r
  54                         Color4d c = mesh.getColors().get(i);\r
  55                         colors.InsertTuple3(i, 255.0* c.x, 255.0 * c.y, 255.0 * c.z);\r
  56                 }\r
  57                 polyData.GetPointData().AddArray(colors);\r
  58                 colors.Delete();\r
  59                 \r
  60         }\r
  61         \r
  62         boolean computeNormals = true;\r
  63         if (computeNormals) {\r
  64                 vtkPolyDataNormals normals = new vtkPolyDataNormals();\r
  65                 normals.SetInput(polyData);\r
  66                 mapper.SetInputConnection(normals.GetOutputPort());\r
  67                 normals.GetOutputPort().Delete();\r
  68                 normals.Delete();\r
  69         } else {\r
  70                 mapper.SetInput(polyData);\r
  71         }\r
  72         \r
  73         if (mesh.getColors() != null) {\r
  74                  mapper.ScalarVisibilityOn();\r
  75                  mapper.SetScalarModeToUsePointFieldData();\r
  76                  mapper.SelectColorArray("Colors");\r
  77         }\r
  78         \r
  79         SetMapper(mapper);\r
  80         mapper.Delete();\r
  81         polyData.GetPointData().Delete();\r
  82         polyData.Delete();\r
  83         \r
  84         }\r
  85 \r
  86 }\r