add Giulio's mod #RH-30298

hstehling · May 23, 2015 · 04da973 · 04da973
1 parent 946c8ed
commit 04da973
Showing 1 changed file with 13 additions and 16 deletions.
diff --git a/Scripts/rhinoscript/surface.py b/Scripts/rhinoscript/surface.py
@@ -2053,23 +2053,20 @@ def TrimBrep(object_id, cutter, tolerance=None):
     else: cutter = rhutil.coerceplane(cutter, True)
     if tolerance is None: tolerance = scriptcontext.doc.ModelAbsoluteTolerance
     breps = brep.Trim(cutter, tolerance)
-    if scriptcontext.id == 2: #gh
-        return breps
+    rhobj = rhutil.coercerhinoobject(object_id)
+    if rhobj:
+        attr = rhobj.Attributes if scriptcontext.id == 1 else None #no attributes in gh
+        rc = []
+        for i in range(len(breps)):
+            if i==0:
+                scriptcontext.doc.Objects.Replace(rhobj.Id, breps[i])
+                rc.append(rhobj.Id)
+            else:
+                rc.append(scriptcontext.doc.Objects.AddBrep(breps[i], attr))
     else:
-        rhobj = rhutil.coercerhinoobject(object_id)
-        if rhobj:
-            attr = rhobj.Attributes
-            rc = []
-            for i in range(len(breps)):
-                if i==0:
-                    scriptcontext.doc.Objects.Replace(rhobj.Id, breps[i])
-                    rc.append(rhobj.Id)
-                else:
-                    rc.append(scriptcontext.doc.Objects.AddBrep(breps[i], attr))
-        else:
-            rc = [scriptcontext.doc.Objects.AddBrep(brep) for brep in breps]
-        scriptcontext.doc.Views.Redraw()
-        return rc
+        rc = [scriptcontext.doc.Objects.AddBrep(brep) for brep in breps]
+    scriptcontext.doc.Views.Redraw()
+    return rc
 
 
 def TrimSurface( surface_id, direction, interval, delete_input=False):