Initial Commit

Assets/Imstk/Scripts/Editor/GeometryImporter.cs

@@ -0,0 +1,137 @@
+/*=========================================================================
+
+   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>
+    /// This class reads in various types of geometry utilized by Imstk
+    /// </summary>
+    [UnityEditor.AssetImporters.ScriptedImporter(1, new string[] { "vtk", "vtu", "vtp", "stl", "ply", "veg", "msh", "mhd" })]
+    public class GeometryImporter : UnityEditor.AssetImporters.ScriptedImporter
+    {
+        /// <summary>
+        /// When on, reverses winding of cells (currently only works for surface meshes)
+        /// </summary>
+        public bool reverseWinding = false;
+
+        // Default is mirror along the X-Axis
+        public Matrix4x4 transform = Matrix4x4.Scale(new Vector3(-1, 1, 1));
+
+        [Tooltip("If true, will center the mesh around the origin")]
+        public bool centerMesh = false;
+
+        public override void OnImportAsset(UnityEditor.AssetImporters.AssetImportContext ctx)
+        {
+            string assetsPath = Application.dataPath;
+            assetsPath = assetsPath.Replace("Assets", "");
+            string filePath = assetsPath + ctx.assetPath;
+            string fileName = System.IO.Path.GetFileNameWithoutExtension(ctx.assetPath);
+
+            // Read the geometry using imstk, warning: type not correct use conversion funcs
+            Imstk.PointSet pointSet = Imstk.MeshIO.read(filePath);
+
+            // Create resource to load into
+            // Setup a default cube, remove BoxCollider, replace mesh, add Geometry (Imstk), replace Mesh (Unity)
+            GameObject obj = GameObject.CreatePrimitive(PrimitiveType.Cube);
+            DestroyImmediate(obj.GetComponent<BoxCollider>());
+            obj.name = fileName;
+
+            if (centerMesh)
+            {
+                var vertices = pointSet.getVertexPositions();
+                var center = new Imstk.Vec3d(0, 0, 0);
+                for (uint i = 0; i < vertices.size(); ++i)
+                {
+                    center += vertices[i];
+                }
+                center = center / vertices.size();
+                pointSet.translate(center*-1.0, Imstk.Geometry.TransformType.ApplyToData);
+            }
+
+            if (pointSet.getTypeName() == Imstk.LineMesh.getStaticTypeName())
+            {
+                Imstk.LineMesh lineMesh = Imstk.Utils.CastTo<Imstk.LineMesh>(pointSet);
+                lineMesh.transform(transform.ToMat4d(), Imstk.Geometry.TransformType.ApplyToData);
+                lineMesh.updatePostTransformData();
+
+                Mesh mesh = lineMesh.ToMesh();
+                foreach (var v in mesh.vertices)
+                {
+                    Debug.Log(v);
+                }
+                mesh.name = fileName + "_mesh";
+                obj.GetComponent<MeshFilter>().sharedMesh = mesh;
+
+                ctx.AddObjectToAsset(obj.name, obj);
+                ctx.AddObjectToAsset(mesh.name, mesh); // Add to the load asset
+            }
+            else if (pointSet.getTypeName() == Imstk.SurfaceMesh.getStaticTypeName())
+            {
+                Imstk.SurfaceMesh surfMesh = Imstk.Utils.CastTo<Imstk.SurfaceMesh>(pointSet);
+                surfMesh.transform(transform.ToMat4d(), Imstk.Geometry.TransformType.ApplyToData);
+                surfMesh.updatePostTransformData();
+
+                if (reverseWinding)
+                {
+                    surfMesh.flipNormals();
+                }
+
+                Mesh mesh = surfMesh.ToMesh();
+                mesh.name = fileName + "_mesh";
+                obj.GetComponent<MeshFilter>().sharedMesh = mesh;
+
+                ctx.AddObjectToAsset(obj.name, obj);
+                ctx.AddObjectToAsset(mesh.name, mesh); // Add to the load asset
+            }
+            else if (pointSet.getTypeName() == Imstk.TetrahedralMesh.getStaticTypeName())
+            {
+                Imstk.TetrahedralMesh tetMesh = Imstk.Utils.CastTo<Imstk.TetrahedralMesh>(pointSet);
+                if (tetMesh == null)
+                {
+                    return;
+                }
+                tetMesh.transform(transform.ToMat4d(), Imstk.Geometry.TransformType.ApplyToData);
+                tetMesh.updatePostTransformData();
+
+                Imstk.SurfaceMesh surfMesh = tetMesh.extractSurfaceMesh();
+                
+                if (reverseWinding)
+                {
+                    surfMesh.flipNormals();
+                }
+
+                Geometry tetGeom = tetMesh.ToImstkMesh();
+                tetGeom.name = fileName + "_mesh";
+                Mesh mesh = surfMesh.ToMesh();
+                mesh.name = fileName + "_mesh_surface";
+                obj.GetComponent<MeshFilter>().sharedMesh = mesh;
+                ctx.AddObjectToAsset(obj.name, obj);
+                ctx.AddObjectToAsset(mesh.name, mesh); // Add to the load asset
+                // \todo: Add a custom thumbnail for the tetrahedral mesh
+                ctx.AddObjectToAsset(tetGeom.name, tetGeom);
+            }
+
+            ctx.SetMainObject(obj);
+        }
+    }
+}