113 lines
4.1 KiB
C#
113 lines
4.1 KiB
C#
/*=========================================================================
|
|
|
|
Library: iMSTK-Unity
|
|
|
|
Copyright (c) Kitware, Inc.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0.txt
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
|
|
=========================================================================*/
|
|
|
|
using UnityEngine;
|
|
|
|
namespace ImstkUnity
|
|
{
|
|
/// <summary>
|
|
/// Handles drawing of meshes in the editor, can draw line or triangle meshes
|
|
/// </summary>
|
|
public static class ImstkGizmos
|
|
{
|
|
static public void DrawPointMesh(Mesh mesh, Vector3 position, Quaternion rotation, Vector3 localScale, Color? color = null)
|
|
{
|
|
var actualColor = color ?? Color.white;
|
|
for (int i = 0; i < mesh.vertices.Length - 1; ++i)
|
|
{
|
|
Vector3 v0 = mesh.vertices[i];
|
|
v0 = rotation * v0;
|
|
for (int j = 0; j < 3; ++j)
|
|
{
|
|
v0[j] = v0[j] * localScale[j];
|
|
}
|
|
v0 = v0 + position;
|
|
Gizmos.color = actualColor;
|
|
Gizmos.DrawLine(v0, v0);
|
|
}
|
|
}
|
|
static public void DrawLineMesh(Mesh mesh, Vector3 position, Quaternion rotation, Vector3 localScale, Color? color = null)
|
|
{
|
|
var actualColor = color ?? Color.white;
|
|
for (int i = 0; i < mesh.vertices.Length - 1; ++i)
|
|
{
|
|
Vector3 v0 = mesh.vertices[i];
|
|
Vector3 v1 = mesh.vertices[i + 1];
|
|
v0 = rotation * v0;
|
|
v1 = rotation * v1;
|
|
for (int j = 0; j < 3; ++j)
|
|
{
|
|
v0[j] = v0[j] * localScale[j];
|
|
v1[j] = v1[j] * localScale[j];
|
|
}
|
|
v0 = v0 + position;
|
|
v1 = v1 + position;
|
|
|
|
Gizmos.color = actualColor;
|
|
Gizmos.DrawLine(v0, v1);
|
|
}
|
|
}
|
|
|
|
static public void DrawLineMesh(Mesh mesh, Color? color = null)
|
|
{
|
|
var actualColor = color ?? Color.white;
|
|
for (int i = 0; i < mesh.vertices.Length - 1; ++i)
|
|
{
|
|
Vector3 v0 = mesh.vertices[i];
|
|
Vector3 v1 = mesh.vertices[i + 1];
|
|
Gizmos.color = actualColor;
|
|
Gizmos.DrawLine(v0, v1);
|
|
}
|
|
}
|
|
|
|
static public void DrawMesh(Mesh mesh, Color? color = null)
|
|
{
|
|
DrawMesh(mesh, Vector3.zero, Quaternion.identity, Vector3.one, color);
|
|
}
|
|
|
|
static public void DrawMesh(Mesh mesh, Vector3 position, Quaternion rotation, Vector3 localScale, Color? color = null)
|
|
{
|
|
if (mesh)
|
|
{
|
|
// Only check for the first sub-mesh
|
|
switch (mesh.GetTopology(0))
|
|
{
|
|
case MeshTopology.Quads:
|
|
case MeshTopology.Triangles:
|
|
Gizmos.color = color ?? Color.white;
|
|
Gizmos.DrawWireMesh(mesh, position, rotation, localScale);
|
|
break;
|
|
case MeshTopology.Lines:
|
|
case MeshTopology.LineStrip:
|
|
DrawLineMesh(mesh, position, rotation, localScale, color);
|
|
break;
|
|
case MeshTopology.Points:
|
|
DrawPointMesh(mesh, position, rotation, localScale, color);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
static public void DrawMesh(Imstk.PointSet pointSet, Color? color = null)
|
|
{
|
|
var mesh = pointSet.ToMesh();
|
|
DrawMesh(mesh, color);
|
|
}
|
|
}
|
|
} |