Merge pull request #32 from moethu/develop

Writing Skp files from GH

Author: moethu

SketchUpForGrasshopper/SketchUp.cs

@@ -42,7 +42,7 @@ namespace SketchUpForGrasshopper
     /// </summary>
     public class SketchUpModel : GH_Component
     {
-        public SketchUpModel() : base("Load SketchUp Model", "Load SketchUp Model", "Loads a SketchUp Model from file", "SketchUpSharp", "Model") { }
+        public SketchUpModel() : base("Load SketchUp Model", "Load SketchUp Model", "Loads a SketchUp Model from file", "GrassUp", "Model") { }
 
         protected override void RegisterInputParams(GH_Component.GH_InputParamManager pManager)
         {
@@ -106,12 +106,62 @@ protected override Bitmap Internal_Icon_24x24
         }
     }
 
+    /// <summary>
+    /// Save SketchUp Model Component
+    /// </summary>
+    public class SaveSketchUpModel : GH_Component
+    {
+        public SaveSketchUpModel() : base("Save SketchUp Model", "Save SketchUp Model", "Save a SketchUp Model to file", "GrassUp", "Model") { }
+
+        protected override void RegisterInputParams(GH_Component.GH_InputParamManager pManager)
+        {
+            pManager.AddTextParameter("Path", "P", "Path to Sketchup File (skp)", GH_ParamAccess.item);
+            int a = pManager.AddCurveParameter("Curves", "C", "Curves", GH_ParamAccess.list);
+            int b = pManager.AddSurfaceParameter("Surfaces", "S", "Surfaces", GH_ParamAccess.list);
+            pManager[a].Optional = true;
+            pManager[b].Optional = true;
+        }
+
+        protected override void RegisterOutputParams(GH_Component.GH_OutputParamManager pManager)
+        {
+
+        }
+
+        protected override void SolveInstance(IGH_DataAccess DA)
+        {
+            GH_String path = new GH_String();
+            DA.GetData<GH_String>(0, ref path);
+
+            List<GH_Surface> surfaces = new List<GH_Surface>();
+            DA.GetDataList<GH_Surface>(1, surfaces);
+            List<GH_Curve> curves = new List<GH_Curve>();
+            DA.GetDataList<GH_Curve>(2, curves);
+
+            Geometry.WriteModel(path.Value, surfaces, curves, false);
+        }
+
+        public override Guid ComponentGuid
+        {
+            get
+            {
+                return new Guid("{5ea8ce3d-d262-4d7f-a733-1573beeb4b5d}");
+            }
+        }
+        protected override Bitmap Internal_Icon_24x24
+        {
+            get
+            {
+                return Properties.Resources.Skp;
+            }
+        }
+    }
+
     /// <summary>
     /// Decomposes a SketchUp Model Instance
     /// </summary>
     public class SketchUpInstance : GH_Component
     {
-        public SketchUpInstance() : base("Decompose SketchUp Instance", "Decompose SketchUp Instance", "Decomposes a SketchUp Instance", "SketchUpSharp", "Elements") { }
+        public SketchUpInstance() : base("Decompose SketchUp Instance", "Decompose SketchUp Instance", "Decomposes a SketchUp Instance", "GrassUp", "Elements") { }
 
         protected override void RegisterInputParams(GH_Component.GH_InputParamManager pManager)
         {
@@ -126,6 +176,7 @@ protected override void RegisterOutputParams(GH_Component.GH_OutputParamManager
             pManager.AddBrepParameter("Surfaces", "Sf", "Surfaces", GH_ParamAccess.list);
             pManager.AddTextParameter("Parent Name", "PN", "Parent Name", GH_ParamAccess.item);
             pManager.AddBrepParameter("Inner", "I", "Inner", GH_ParamAccess.list);
+            pManager.AddCurveParameter("Curves", "C", "Curves", GH_ParamAccess.list);
         }
 
         protected override void SolveInstance(IGH_DataAccess DA)
@@ -139,6 +190,7 @@ protected override void SolveInstance(IGH_DataAccess DA)
 
             List<GH_Brep> surfaces = new List<GH_Brep>();
             List<GH_Brep> inner = new List<GH_Brep>();
+            List<GH_Curve> curves = new List<GH_Curve>();
 
             GH_String parentName = new GH_String("");
             SketchUpNET.Component parentComponent = i.Parent as Component;
@@ -152,6 +204,9 @@ protected override void SolveInstance(IGH_DataAccess DA)
                         surfaces.Add(new GH_Brep(brep));
                     }
                 }
+
+                foreach (Edge c in parentComponent.Edges)
+                    curves.Add(new GH_Curve(c.ToRhinoGeo().ToNurbsCurve()));
             }
 
             DA.SetData(0, location);
@@ -160,6 +215,7 @@ protected override void SolveInstance(IGH_DataAccess DA)
             DA.SetDataList(3, surfaces);
             DA.SetData(4, parentName);
             DA.SetDataList(5, inner);
+            DA.SetDataList(6, curves);
         }
 
         public override Guid ComponentGuid
@@ -183,6 +239,8 @@ protected override Bitmap Internal_Icon_24x24
     /// </summary>
     public static class Geometry
     {
+        public static string DefaultLayer = "Default";
+
         /// <summary>
         /// Converts a SketchUp Vertex to a Rhino Point
         /// </summary>
@@ -197,6 +255,22 @@ public static Rhino.Geometry.Point3d ToRhinoGeo(this SketchUpNET.Vertex v, Trans
             }
         }
 
+        /// <summary>
+        /// Converts a Rhino Point to a SketchUp Point
+        /// </summary>
+        public static SketchUpNET.Vertex ToSkpGeo(this Rhino.Geometry.Point3d v)
+        {
+            return new SketchUpNET.Vertex(v.X, v.Y, v.Z);
+        }
+
+        /// <summary>
+        /// Converts a Rhino Vector to a SketchUp Vector
+        /// </summary>
+        public static SketchUpNET.Vector ToSkpGeo(this Rhino.Geometry.Vector3d v)
+        {
+            return new SketchUpNET.Vector(v.X, v.Y, v.Z);
+        }
+
         /// <summary>
         /// Converts a SketchUp Vector to a Rhino Vector
         /// </summary>
@@ -213,17 +287,71 @@ public static Rhino.Geometry.Line ToRhinoGeo(this SketchUpNET.Edge v, Transform
             return new Rhino.Geometry.Line(v.Start.ToRhinoGeo(t), v.End.ToRhinoGeo(t));
         }
 
+        /// <summary>
+        /// Converts a Rhino Line to a SketchUp Edge
+        /// </summary>
+        public static SketchUpNET.Edge ToSkpGeo(this Rhino.Geometry.Line v)
+        {
+            return new SketchUpNET.Edge(v.PointAt(0).ToSkpGeo(), v.PointAt(1.0).ToSkpGeo(), DefaultLayer);
+        }
+
+
+        /// <summary>
+        /// Converts a Rhino Curve to a SketchUp Curve
+        /// </summary>
+        public static SketchUpNET.Curve ToSkpGeo(this Rhino.Geometry.Curve v)
+        {
+            List<Edge> edges = new List<Edge>();
+            if (v.IsLinear())
+            {
+                edges.Add(new Edge(v.PointAt(0).ToSkpGeo(), v.PointAt(1.0).ToSkpGeo(), DefaultLayer));
+            }
+            else
+            {
+                for (double i = 0; i < 1.0; i = i + 0.1)
+                {
+                    edges.Add(new Edge(v.PointAt(i).ToSkpGeo(), v.PointAt(i + 0.1).ToSkpGeo(), DefaultLayer));
+                }
+            }
+            return new SketchUpNET.Curve(edges,v.IsArc());
+        }
+
+        /// <summary>
+        /// Converts a Rhino Surface to a SketchUp Surface
+        /// </summary>
+        public static SketchUpNET.Surface ToSkpGeo(this Rhino.Geometry.Brep surface)
+        {
+            Surface srf = new Surface();
+            srf.Vertices = new List<Vertex>();
+            
+            foreach (var curve in surface.Edges)
+            {
+                if (curve.IsLinear())
+                {
+                    srf.Vertices.Add(curve.PointAt(0.0).ToSkpGeo());
+                }
+                else
+                {
+                    for (double i = 0; i < 1.0; i = i + 0.1)
+                    {
+                        srf.Vertices.Add(curve.PointAt(i).ToSkpGeo());
+                    }
+                }
+            }
+            return srf;
+        }
+
         /// <summary>
         /// Converts a SketchUp Surface to a Rhino Surface
         /// </summary>
         public static Rhino.Geometry.Brep[] ToRhinoGeo(this SketchUpNET.Surface v, Transform t = null)
         {
             List<Rhino.Geometry.Curve> curves = new List<Rhino.Geometry.Curve>();
-
+            var tol = Rhino.RhinoDoc.ActiveDoc.ModelAbsoluteTolerance;
             foreach (SketchUpNET.Edge c in v.OuterEdges.Edges)
                 curves.Add(c.ToRhinoGeo(t).ToNurbsCurve());
 
-            Rhino.Geometry.Brep[] b = Rhino.Geometry.Brep.CreatePlanarBreps(curves,0.1);
+            Rhino.Geometry.Brep[] b = Rhino.Geometry.Brep.CreatePlanarBreps(curves,tol);
             if (b == null) return new Rhino.Geometry.Brep[] { };
 
             List<Rhino.Geometry.Brep> breps = v.InnerLoops(t);
@@ -243,12 +371,13 @@ public static Rhino.Geometry.Brep[] ToRhinoGeo(this SketchUpNET.Surface v, Trans
         public static List<Rhino.Geometry.Brep> InnerLoops(this SketchUpNET.Surface v, Transform t = null)
         {
             List<Rhino.Geometry.Brep> breps = new List<Rhino.Geometry.Brep>();
+            var tol = Rhino.RhinoDoc.ActiveDoc.ModelAbsoluteTolerance;
 
             foreach (Loop loop in v.InnerEdges)
             {
                 List<Rhino.Geometry.Curve> curves = new List<Rhino.Geometry.Curve>();
                 foreach (SketchUpNET.Edge c in loop.Edges) curves.Add(c.ToRhinoGeo(t).ToNurbsCurve());
-                Rhino.Geometry.Brep[] b = Rhino.Geometry.Brep.CreatePlanarBreps(curves, 0.1);
+                Rhino.Geometry.Brep[] b = Rhino.Geometry.Brep.CreatePlanarBreps(curves,tol);
                 if (b != null)
                 {
                     foreach (var brep in b)
@@ -257,5 +386,38 @@ public static Rhino.Geometry.Brep[] ToRhinoGeo(this SketchUpNET.Surface v, Trans
             }
             return breps;
         }
+
+        public static void WriteModel(string path, List<GH_Surface> surfaces = null, List<GH_Curve> curves = null, bool append = false)
+        {
+            SketchUpNET.SketchUp skp = new SketchUpNET.SketchUp();
+            skp.Surfaces = new List<Surface>();
+            skp.Edges = new List<Edge>();
+            skp.Curves = new List<Curve>();
+
+            if (curves != null)
+                foreach (var c in curves)
+                {
+                    var curve = c.Value;
+                    if (curve.IsLinear())
+                    {
+                        var line = new SketchUpNET.Edge(curve.PointAt(0).ToSkpGeo(), curve.PointAt(1.0).ToSkpGeo(), DefaultLayer);
+                        skp.Edges.Add(line);
+                    }
+                    else
+                    {
+                        skp.Curves.Add(curve.ToSkpGeo());
+                    }
+                }
+
+            if (surfaces != null)
+                foreach (var surface in surfaces)
+                    skp.Surfaces.Add(surface.Value.ToSkpGeo());
+
+            if (System.IO.File.Exists(path) && append)
+                skp.AppendToModel(path);
+            else
+                skp.WriteNewModel(path);
+        }
+
     }
 }