rhinopythonscripts/GeomTools.py at master · localcode/rhinopythonscripts · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
'''A module of tools for short simpel geometry conversions and calculations. This module is likely to be split up into pieces in the future.'''
import Rhino
import scriptcontext
WorldXY = Rhino.Geometry.Plane.WorldXY

def getObjsByGeomType(geomType):
    filtr  = Rhino.DocObjects.ObjectEnumeratorSettings()
    filtr.VisibleFilter = True
    all_objs = scriptcontext.doc.Objects.FindByFilter(filtr)
    right_objs = [o for o in all_objs if isinstance(o.Geometry, geomType)]
    return right_objs

def getSelected():
    filtr  = Rhino.DocObjects.ObjectEnumeratorSettings()
    filtr.SelectedObjectsFilter = True
    all_objs = scriptcontext.doc.Objects.FindByFilter(filtr)
    return all_objs

def brepToCurves(brep):
    edges = brep.Edges
    return [e.DuplicateCurve() for e in edges]

def pointsToCircles(pointList, radii):
    circles = []
    if type(radii) == float or type(radii) == int:
        radii = [radii for n in range(len(pointList))]
    elif len(radii) == 1:
        radii = [radii[0] for n in range(len(pointList))]
    for i in range(len(pointList)):
        point = pointList[i]
        radius = radii[i]
        circles.append(Rhino.Geometry.Circle(point, radius))
    return circles

def pointGrid(xrows=10, ycols=10, xspacing=1, yspacing=1,
        plane=WorldXY):
    xA = range(0, (xrows*xspacing), xspacing)
    yA = range(0, (ycols*yspacing), yspacing)
    xVals = [float(x-((xrows-1)*xspacing/2)) for x in xA]
    yVals = [float(y-((ycols-1)*yspacing/2)) for y in yA]
    pts = []
    for y in yVals:
        for x in xVals:
            pts.append(Rhino.Geometry.Point3d(x, y, 0.0))
    if plane != WorldXY:
        xform = Rhino.Geometry.Transform.PlaneToPlane(WorldXY,
                plane)
        # I think Transform modifies in place
        [pt.Transform(xform) for pt in pts]
    return pts

def curveClosestPoint3d(curve, point, searchDistance=1000.0):
    '''enter curve and point (and optional max distance),
    and returns a point3d or None (if no points within distance)'''
    result = curve.ClosestPoint(point, searchDistance)
    # if something could be found within the search distance
    if result[0]:
        t = result[1]
        return curve.PointAt(t) # get the point
    else: # no point found
        return None

def vectorToClosestCurve(point, curves, searchDistance=1000.0):
    '''get the vector from a point to the closest curve in a
    set of curves. If none are within searchDistance, return None.'''
    testDistance = searchDistance
    vector = None
    for curve in curves:
        closestPt = curveClosestPoint3d(curve, point, searchDistance)
        if closestPt:
            newVector = point.Subtract(closestPt, point)
            if newVector.Length < testDistance:
                testDistance = newVector.Length
                vector = newVector
    return vector

def moveMany(listOfThings, vector):
    '''Translate (move) a list of geometries.'''
    out = []
    for thing in listOfThings:
        out.append(thing.Translate(vector))
    return out

def rotateMany(listOfThings, radiansAngle, axisVector, origin):
    '''Rotate a list of geometries.'''
    out = []
    for thing in listOfThings:
        out.append(thing.Rotate(radiansAngle, axisVector, origin))
    return out

def bakeMany(listOfThings, objectAttributes=None, translationVector=None):
    idsOut = []
    if not objectAttributes:
        objectAttributes = Rhino.DocObjects.ObjectAttributes()
    if translationVector:
        movedList = []
        for thing in listOfThings:
            if thing:
                thing.Translate(translationVector)
            movedList.append(thing)
        listOfThings = movedList
    for thing in listOfThings:
        # move from specific to broad
        if isinstance(thing, Rhino.Geometry.Point3d):
            idsOut.append(scriptcontext.doc.Objects.AddPoint(thing, objectAttributes))
        elif isinstance(thing, Rhino.Geometry.Curve):
            idsOut.append(scriptcontext.doc.Objects.AddCurve(thing, objectAttributes))
        elif isinstance(thing, Rhino.Geometry.Brep):
            idsOut.append(scriptcontext.doc.Objects.AddBrep(thing, objectAttributes))
        elif isinstance(thing, Rhino.Geometry.Surface):
            idsOut.append(scriptcontext.doc.Objects.AddSurface(thing, objectAttributes))
        elif isinstance(thing, Rhino.Geometry.Mesh):
            idsOut.append(scriptcontext.doc.Objects.AddMesh(thing, objectAttributes))
        elif isinstance(thing, Rhino.Geometry.Hatch):
            idsOut.append(scriptcontext.doc.Objects.AddHatch(thing, objectAttributes))
        elif isinstance(thing, Rhino.Display.Text3d):
            idsOut.append(scriptcontext.doc.Objects.AddText(thing, objectAttributes))
        else:
            print 'this here was a thang I couldnay bake:'
            print thing
    return idsOut