public void ColorBy(string fieldname, VtkColorType type)
{
colorFieldName = fieldname;
colorDataType = type;
if (colorFieldName != "")
{
triangleFilter.Update();
Kitware.VTK.vtkPolyData pd = triangleFilter.GetOutput();
if (colorDataType == VtkColorType.POINT_DATA)
{
colorArray = pd.GetPointData().GetScalars(colorFieldName);
}
else if (colorDataType == VtkColorType.CELL_DATA)
{
colorArray = pd.GetCellData().GetScalars(colorFieldName);
}
go.GetComponent <Renderer>().materials = new Material[2] {
new Material(Shader.Find("UFZ/Vertex Color Front")),
new Material(Shader.Find("UFZ/Vertex Color Back"))
};
}
else
{
colorArray = null;
mat = new Material(Shader.Find("Diffuse"));
mat.color = Color.magenta;
go.GetComponent <Renderer>().material = mat;
Debug.Log("Color array " + fieldname + " not found!");
}
}
void PolyDataToMesh()
{
mesh.Clear();
triangleFilter.Update();
Kitware.VTK.vtkPolyData pd = triangleFilter.GetOutput();
// Points / Vertices
int numVertices = pd.GetNumberOfPoints();
Vector3[] vertices = new Vector3[numVertices];
for (int i = 0; i < numVertices; ++i)
{
double[] pnt = pd.GetPoint(i);
// Flip z-up to y-up
vertices[i] = new Vector3(-(float)pnt[0], (float)pnt[2], (float)pnt[1]);
}
mesh.vertices = vertices;
// Texture coordinates
Kitware.VTK.vtkDataArray vtkTexCoords = pd.GetPointData().GetTCoords();
if (vtkTexCoords != null)
{
int numCoords = vtkTexCoords.GetNumberOfTuples();
Vector2[] uvs = new Vector2[numCoords];
for (int i = 0; i < numCoords; ++i)
{
double[] texCoords = vtkTexCoords.GetTuple2(i);
uvs[i] = new Vector2((float)texCoords[0], (float)texCoords[1]);
}
mesh.uv = uvs;
}
// Vertex colors
if (numVertices > 0 && colorArray != null)
{
Color32[] colors = new Color32[numVertices];
for (int i = 0; i < numVertices; ++i)
{
colors[i] = GetColor32AtIndex(i);
}
mesh.colors32 = colors;
}
// Triangles / Cells
int numTriangles = pd.GetNumberOfPolys();
Kitware.VTK.vtkCellArray polys = pd.GetPolys();
if (polys.GetNumberOfCells() > 0)
{
int[] triangles = new int[numTriangles * 3];
int prim = 0;
Kitware.VTK.vtkIdList pts = Kitware.VTK.vtkIdList.New();
polys.InitTraversal();
while (polys.GetNextCell(pts) != 0)
{
for (int i = 0; i < pts.GetNumberOfIds(); ++i)
{
triangles[prim * 3 + i] = pts.GetId(i);
}
++prim;
}
mesh.SetTriangles(triangles, 0);
mesh.RecalculateNormals();
mesh.RecalculateBounds();
return;
}
// Lines
Kitware.VTK.vtkCellArray lines = pd.GetLines();
if (lines.GetNumberOfCells() > 0)
{
ArrayList idList = new ArrayList();
Kitware.VTK.vtkIdList pts = Kitware.VTK.vtkIdList.New();
lines.InitTraversal();
while (lines.GetNextCell(pts) != 0)
{
for (int i = 0; i < pts.GetNumberOfIds() - 1; ++i)
{
idList.Add(pts.GetId(i));
idList.Add(pts.GetId(i + 1));
}
}
mesh.SetIndices(idList.ToArray(typeof(int)) as int[], MeshTopology.Lines, 0);
mesh.RecalculateBounds();
return;
}
// Points
Kitware.VTK.vtkCellArray points = pd.GetVerts();
int numPointCells = points.GetNumberOfCells();
if (numPointCells > 0)
{
ArrayList idList = new ArrayList();
Kitware.VTK.vtkIdList pts = Kitware.VTK.vtkIdList.New();
points.InitTraversal();
while (points.GetNextCell(pts) != 0)
{
for (int i = 0; i < pts.GetNumberOfIds(); ++i)
{
idList.Add(pts.GetId(i));
}
}
mesh.SetIndices(idList.ToArray(typeof(int)) as int[], MeshTopology.Points, 0);
mesh.RecalculateBounds();
return;
}
//Debug.Log("Number of point data arrays: " + pd.GetPointData().GetNumberOfArrays());
//Debug.Log(" - " + pd.GetPointData().GetArrayName(0));
//Debug.Log("Number of cell data arrays: " + pd.GetCellData().GetNumberOfArrays());
//Debug.Log(" - " + pd.GetCellData().GetArrayName(0));
//Debug.Log(name + " - Vertices: " + numPoints + ", triangle: " + numTriangles + ", UVs: " + numCoords);
}
void PolyDataToMesh()
{
// mesh.MarkDynamic();
mesh.Clear();
triangleFilter.Update();
Kitware.VTK.vtkPolyData pd = triangleFilter.GetOutput();
// Points / Vertices
int numVertices = pd.GetNumberOfPoints();
Vector3[] vertices = new Vector3[numVertices];
for (int i = 0; i < numVertices; ++i)
{
double[] pnt = pd.GetPoint(i);
// Flip z-up to y-up
vertices[i] = new Vector3(-(float)pnt[0], (float)pnt[2], (float)pnt[1]);
}
mesh.vertices = vertices;
// Triangles / Cells
int numTriangles = pd.GetNumberOfPolys();
int[] triangles = new int[numTriangles * 3];
Kitware.VTK.vtkCellArray polys = pd.GetPolys();
if (polys.GetNumberOfCells() > 0)
{
int prim = 0;
Kitware.VTK.vtkIdList pts = Kitware.VTK.vtkIdList.New();
polys.InitTraversal();
while (polys.GetNextCell(pts) != 0)
{
for (int i = 0; i < pts.GetNumberOfIds(); ++i)
{
triangles[prim * 3 + i] = pts.GetId(i);
}
++prim;
}
}
mesh.triangles = triangles;
// Lines
Kitware.VTK.vtkCellArray lines = pd.GetLines();
if (lines.GetNumberOfCells() > 0)
{
Kitware.VTK.vtkIdList pts = Kitware.VTK.vtkIdList.New();
lines.InitTraversal();
int prim = 0;
while (lines.GetNextCell(pts) != 0)
{
Color lineColor = solidColor;
if (colorArray != null)
{
if (colorDataType == VtkColorType.CELL_DATA)
{
lineColor = GetColorAtIndex(prim);
}
else if (colorDataType == VtkColorType.POINT_DATA)
{
lineColor = GetColorAtIndex(pts.GetId(0));
}
}
Vector3[] linePoints = new Vector3[2];
linePoints[0] = vertices[pts.GetId(0)];
linePoints[1] = vertices[pts.GetId(1)];
Vectrosity.VectorLine line = new Vectrosity.VectorLine(name + "-Line", linePoints, lineColor, null, 1.0f);
line.Draw3DAuto(go.transform);
++prim;
}
}
// Points
Kitware.VTK.vtkCellArray points = pd.GetVerts();
int numPointCells = points.GetNumberOfCells();
if (numPointCells > 0)
{
ArrayList list = new ArrayList();
ArrayList colorList = new ArrayList();
Kitware.VTK.vtkIdList pts = Kitware.VTK.vtkIdList.New();
points.InitTraversal();
while (points.GetNextCell(pts) != 0)
{
ArrayList pointsList = new ArrayList();
ArrayList colorsList = new ArrayList();
for (int i = 0; i < pts.GetNumberOfIds(); ++i)
{
pointsList.Add(vertices[pts.GetId(i)]);
Color pointColor = solidColor;
if (colorArray != null && colorDataType == VtkColorType.POINT_DATA)
{
pointColor = GetColorAtIndex(pts.GetId(i));
}
colorsList.Add(pointColor);
}
list.AddRange(pointsList);
colorList.AddRange(colorsList);
}
Vectrosity.VectorPoints pnt =
new Vectrosity.VectorPoints(name + "-Point " + list.Count,
list.ToArray(typeof(Vector3)) as Vector3[],
colorList.ToArray(typeof(Color)) as Color[], null, 1f);
pnt.Draw3DAuto(go.transform);
}
// Texture coordinates
Vector2[] uvs;
int numCoords = 0;
Kitware.VTK.vtkDataArray vtkTexCoords = pd.GetPointData().GetTCoords();
if (vtkTexCoords != null)
{
numCoords = vtkTexCoords.GetNumberOfTuples();
uvs = new Vector2[numCoords];
for (int i = 0; i < numCoords; ++i)
{
double[] texCoords = vtkTexCoords.GetTuple2(i);
uvs[i] = new Vector2((float)texCoords[0], (float)texCoords[1]);
}
mesh.uv = uvs;
}
// Vertex colors
if (numTriangles > 0 && colorArray != null)
{
Color32[] colors = new Color32[numVertices];
for (int i = 0; i < numVertices; ++i)
{
colors[i] = GetColor32AtIndex(i);
}
mesh.colors32 = colors;
}
//Debug.Log("Number of point data arrays: " + pd.GetPointData().GetNumberOfArrays());
//Debug.Log(" - " + pd.GetPointData().GetArrayName(0));
//Debug.Log("Number of cell data arrays: " + pd.GetCellData().GetNumberOfArrays());
//Debug.Log(" - " + pd.GetCellData().GetArrayName(0));
//Debug.Log(name + " - Vertices: " + numPoints + ", triangle: " + numTriangles + ", UVs: " + numCoords);
mesh.RecalculateNormals();
mesh.RecalculateBounds();
//mesh.Optimize();
}