Smart.py

'''A module for wrapping geometry with UserString and Attribute Dictionaries'''

import Rhino

class SmartFeature(object):
    def __init__(self, rhinoObjectOrTuple):
        self._parseConstructor(rhinoObjectOrTuple)

    def _parseConstructor(self, rhinoObjectOrTuple):
        # determine if it is a tuple
        kind = type(rhinoObjectOrTuple)
        if kind == tuple or kind == list:
            # build from geom, user string pair
            pair = rhinoObjectOrTuple
            self.geom = self._filterGeometry(pair[0]) # geometry
            self.attributes = pair[1] # properties (as dictionary)
        else: # assume RhinoObject
            rhObj = rhinoObjectOrTuple
            self.geom = self._filterGeom(rhObj.Geometry)
            self.attributes = {}
            numAtts = rhObj.Attributes.UserStringCount
            rawAtts = rhObj.Attributes.GetUserStrings()
            keys = rawAtts.AllKeys
            for key in keys:
                self.attributes[key] = rhObj.Attributes.GetUserString(key)

    def _filterGeom(self, geometry):
        if type(geometry) == Rhino.Geometry.Point:
            return geometry.Location
        else:
            return geometry

    def objAttributes(self, objectAttributes):
        for key in self.attributes:
            objectAttributes.SetUserString(key, self.attributes[key])
        return objectAttributes


def RhinoObjectsToSmartFeatures(RhinoObjectList):
    return [SmartFeature(obj) for obj in RhinoObjectList]

def replaceGeometries(smartFeatures, geometries):
    out = []
    for i in range(len(smartFeatures)):
        feature = smartFeatures[i]
        geometry = geometries[i]
        feature.geom = geometry
        out.append(feature)
    return out