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

   1 /*******************************************************************************\r
   2  * Copyright (c) 2012, 2013 Association for Decentralized Information Management in\r
   3  * Industry THTH ry.\r
   4  * All rights reserved. This program and the accompanying materials\r
   5  * are made available under the terms of the Eclipse Public License v1.0\r
   6  * which accompanies this distribution, and is available at\r
   7  * http://www.eclipse.org/legal/epl-v10.html\r
   8  *\r
   9  * Contributors:\r
  10  *     VTT Technical Research Centre of Finland - initial API and implementation\r
  11  *******************************************************************************/\r
  12 package org.simantics.g3d.vtk.shape;\r
  13 \r
  14 import javax.vecmath.Vector3d;\r
  15 \r
  16 import org.simantics.g3d.shape.Color4d;\r
  17 import org.simantics.g3d.shape.Mesh;\r
  18 \r
  19 import vtk.vtkActor;\r
  20 import vtk.vtkDataSetMapper;\r
  21 import vtk.vtkIdList;\r
  22 import vtk.vtkPoints;\r
  23 import vtk.vtkPolyData;\r
  24 import vtk.vtkPolyDataMapper;\r
  25 import vtk.vtkPolyDataNormals;\r
  26 import vtk.vtkTriangle;\r
  27 import vtk.vtkUnsignedCharArray;\r
  28 \r
  29 public class MeshActor extends vtkActor {\r
  30         \r
  31         public void setMesh(Mesh mesh) {\r
  32                 \r
  33                 vtkPolyDataMapper mapper = new vtkPolyDataMapper();\r
  34                 \r
  35                 vtkPolyData polyData = new vtkPolyData();\r
  36                 polyData.Allocate(mesh.getIndices().size()/3, mesh.getIndices().size()/3);\r
  37                 \r
  38                 vtkTriangle triangle = new vtkTriangle();\r
  39         vtkIdList list = triangle.GetPointIds();\r
  40         for (int i = 0; i < mesh.getIndices().size(); i+=3) {\r
  41                 list.SetId(0, mesh.getIndices().get(i));\r
  42                 list.SetId(1, mesh.getIndices().get(i+1));\r
  43                 list.SetId(2, mesh.getIndices().get(i+2));\r
  44                 polyData.InsertNextCell(triangle.GetCellType(), list);\r
  45         }\r
  46         list.Delete();\r
  47         triangle.Delete();\r
  48                 \r
  49         \r
  50         vtkPoints points = new vtkPoints();\r
  51         for (int i = 0; i < mesh.getVertices().size(); i++) {\r
  52                 Vector3d p = mesh.getVertices().get(i);\r
  53                 points.InsertPoint(i, p.x, p.y, p.z);\r
  54         }\r
  55         \r
  56         polyData.SetPoints(points);\r
  57         points.Delete();\r
  58         \r
  59         if (mesh.getColors() != null) {\r
  60                 vtkUnsignedCharArray colors = new vtkUnsignedCharArray();\r
  61                 colors.SetName("Colors");\r
  62                 colors.SetNumberOfComponents(3);\r
  63                 colors.SetNumberOfTuples(mesh.getColors().size());\r
  64                 for (int i = 0; i < mesh.getColors().size(); i++) {\r
  65                         Color4d c = mesh.getColors().get(i);\r
  66                         colors.InsertTuple3(i, 255.0* c.x, 255.0 * c.y, 255.0 * c.z);\r
  67                 }\r
  68                 polyData.GetPointData().AddArray(colors);\r
  69                 colors.Delete();\r
  70                 \r
  71         }\r
  72         \r
  73         boolean computeNormals = true;\r
  74         if (computeNormals) {\r
  75                 vtkPolyDataNormals normals = new vtkPolyDataNormals();\r
  76                 normals.SetInput(polyData);\r
  77                 mapper.SetInputConnection(normals.GetOutputPort());\r
  78                 normals.GetOutputPort().Delete();\r
  79                 normals.Delete();\r
  80         } else {\r
  81                 mapper.SetInput(polyData);\r
  82         }\r
  83         \r
  84         if (mesh.getColors() != null) {\r
  85                  mapper.ScalarVisibilityOn();\r
  86                  mapper.SetScalarModeToUsePointFieldData();\r
  87                  mapper.SelectColorArray("Colors");\r
  88         }\r
  89         \r
  90         SetMapper(mapper);\r
  91         mapper.Delete();\r
  92         polyData.GetPointData().Delete();\r
  93         polyData.Delete();\r
  94         \r
  95         }\r
  96 \r
  97 }\r